jail/jail.c

   1 /*
   2  * Copyright (C) 2015 John Crispin <blogic@openwrt.org>
   3  *
   4  * This program is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU Lesser General Public License version 2.1
   6  * as published by the Free Software Foundation
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  * GNU General Public License for more details.
  12  */
  13
  14 #define _GNU_SOURCE
  15 #include <sys/syscall.h>
  16 #include <sys/mman.h>
  17 #include <sys/utsname.h>
  18 #include <sys/types.h>
  19 #include <sys/syscall.h>
  20 #include <sys/types.h>
  21 #include <sys/mount.h>
  22 #include <sys/prctl.h>
  23 #include <sys/wait.h>
  24
  25 #include <stdlib.h>
  26 #include <unistd.h>
  27 #include <values.h>
  28 #include <errno.h>
  29 #include <stdio.h>
  30 #include <string.h>
  31 #include <sys/stat.h>
  32 #include <fcntl.h>
  33 #include <syslog.h>
  34 #include <libgen.h>
  35 #include <glob.h>
  36 #include <elf.h>
  37 #include <sched.h>
  38
  39 #include "elf.h"
  40
  41 #include <libubox/utils.h>
  42 #include <libubox/list.h>
  43 #include <libubox/uloop.h>
  44
  45 #define STACK_SIZE      (1024 * 1024)
  46 #define OPT_ARGS        "P:S:n:r:w:psuldo"
  47
  48 struct extra {
  49         struct list_head list;
  50
  51         const char *path;
  52         const char *name;
  53         int readonly;
  54 };
  55
  56 static LIST_HEAD(extras);
  57
  58 extern int pivot_root(const char *new_root, const char *put_old);
  59
  60 int debug = 0;
  61
  62 static char child_stack[STACK_SIZE];
  63
  64 static int mkdir_p(char *dir, mode_t mask)
  65 {
  66         char *l = strrchr(dir, '/');
  67         int ret;
  68
  69         if (!l)
  70                 return 0;
  71
  72         *l = '\0';
  73
  74         if (mkdir_p(dir, mask))
  75                 return -1;
  76
  77         *l = '/';
  78
  79         ret = mkdir(dir, mask);
  80         if (ret && errno == EEXIST)
  81                 return 0;
  82
  83         if (ret)
  84                 ERROR("mkdir failed on %s: %s\n", dir, strerror(errno));
  85
  86         return ret;
  87 }
  88
  89 static int mount_bind(const char *root, const char *path, const char *name, int readonly, int error)
  90 {
  91         const char *p = path;
  92         struct stat s;
  93         char old[256];
  94         char new[256];
  95         int fd;
  96
  97         if (strstr(p, "local"))
  98                 p = "/lib";
  99
 100         snprintf(old, sizeof(old), "%s/%s", path, name);
 101         snprintf(new, sizeof(new), "%s%s", root, p);
 102
 103         mkdir_p(new, 0755);
 104
 105         snprintf(new, sizeof(new), "%s%s/%s", root, p, name);
 106
 107         if (stat(old, &s)) {
 108                 ERROR("%s does not exist\n", old);
 109                 return error;
 110         }
 111
 112         if (S_ISDIR(s.st_mode)) {
 113                 mkdir_p(new, 0755);
 114         } else {
 115                 fd = creat(new, 0644);
 116                 if (fd == -1) {
 117                         ERROR("failed to create %s: %s\n", new, strerror(errno));
 118                         return -1;
 119                 }
 120                 close(fd);
 121         }
 122
 123         if (mount(old, new, NULL, MS_BIND, NULL)) {
 124                 ERROR("failed to mount -B %s %s: %s\n", old, new, strerror(errno));
 125                 return -1;
 126         }
 127
 128         if (readonly && mount(old, new, NULL, MS_BIND | MS_REMOUNT | MS_RDONLY, NULL)) {
 129                 ERROR("failed to remount ro %s: %s\n", new, strerror(errno));
 130                 return -1;
 131         }
 132
 133         DEBUG("mount -B %s %s\n", old, new);
 134
 135         return 0;
 136 }
 137
 138 static int build_jail(const char *path)
 139 {
 140         struct library *l;
 141         struct extra *m;
 142         int ret = 0;
 143
 144         mkdir(path, 0755);
 145
 146         if (mount("tmpfs", path, "tmpfs", MS_NOATIME, "mode=0755")) {
 147                 ERROR("tmpfs mount failed %s\n", strerror(errno));
 148                 return -1;
 149         }
 150
 151         avl_for_each_element(&libraries, l, avl)
 152                 if (mount_bind(path, l->path, l->name, 1, -1))
 153                         return -1;
 154
 155         list_for_each_entry(m, &extras, list)
 156                 if (mount_bind(path, m->path, m->name, m->readonly, 0))
 157                         return -1;
 158
 159         return ret;
 160 }
 161
 162 static void _umount(const char *root, const char *path)
 163 {
 164         char *buf = NULL;
 165
 166         if (asprintf(&buf, "%s%s", root, path) < 0) {
 167                 ERROR("failed to alloc umount buffer: %s\n", strerror(errno));
 168         } else {
 169                 DEBUG("umount %s\n", buf);
 170                 umount(buf);
 171                 free(buf);
 172         }
 173 }
 174
 175 static int stop_jail(const char *root)
 176 {
 177         struct library *l;
 178         struct extra *m;
 179
 180         avl_for_each_element(&libraries, l, avl) {
 181                 char path[256];
 182                 char *p = l->path;
 183
 184                 if (strstr(p, "local"))
 185                         p = "/lib";
 186
 187                 snprintf(path, sizeof(path), "%s%s/%s", root, p, l->name);
 188                 DEBUG("umount %s\n", path);
 189                 umount(path);
 190         }
 191
 192         list_for_each_entry(m, &extras, list) {
 193                 char path[256];
 194
 195                 snprintf(path, sizeof(path), "%s%s/%s", root, m->path, m->name);
 196                 DEBUG("umount %s\n", path);
 197                 umount(path);
 198         }
 199
 200         _umount(root, "/proc");
 201         _umount(root, "/sys");
 202
 203         DEBUG("umount %s\n", root);
 204         umount(root);
 205         rmdir(root);
 206
 207         return 0;
 208 }
 209
 210 #define MAX_ENVP        8
 211 static char** build_envp(const char *seccomp, int debug)
 212 {
 213         static char *envp[MAX_ENVP];
 214         static char preload_var[64];
 215         static char seccomp_var[64];
 216         static char debug_var[] = "LD_DEBUG=all";
 217         char *preload_lib = find_lib("libpreload-seccomp.so");
 218         int count = 0;
 219
 220         if (seccomp && !preload_lib) {
 221                 ERROR("failed to add preload-lib to env\n");
 222                 return NULL;
 223         }
 224         if (seccomp) {
 225                 snprintf(seccomp_var, sizeof(seccomp_var), "SECCOMP_FILE=%s", seccomp);
 226                 envp[count++] = seccomp_var;
 227                 snprintf(preload_var, sizeof(preload_var), "LD_PRELOAD=%s", preload_lib);
 228                 envp[count++] = preload_var;
 229         }
 230         if (debug)
 231                 envp[count++] = debug_var;
 232
 233         return envp;
 234 }
 235
 236 static int spawn(const char *path, char **argv, const char *seccomp)
 237 {
 238         pid_t pid = fork();
 239
 240         if (pid < 0) {
 241                 ERROR("failed to spawn %s: %s\n", *argv, strerror(errno));
 242                 return -1;
 243         } else if (!pid) {
 244                 char **envp = build_envp(seccomp, 0);
 245
 246                 INFO("spawning %s\n", *argv);
 247                 execve(*argv, argv, envp);
 248                 ERROR("failed to spawn child %s: %s\n", *argv, strerror(errno));
 249                 exit(-1);
 250         }
 251
 252         return pid;
 253 }
 254
 255 static int usage(void)
 256 {
 257         fprintf(stderr, "jail <options> -D <binary> <params ...>\n");
 258         fprintf(stderr, "  -P <path>\tpath where the jail will be staged\n");
 259         fprintf(stderr, "  -S <file>\tseccomp filter\n");
 260         fprintf(stderr, "  -n <name>\tthe name of the jail\n");
 261         fprintf(stderr, "  -r <file>\treadonly files that should be staged\n");
 262         fprintf(stderr, "  -w <file>\twriteable files that should be staged\n");
 263         fprintf(stderr, "  -p\t\tjail has /proc\t\n");
 264         fprintf(stderr, "  -s\t\tjail has /sys\t\n");
 265         fprintf(stderr, "  -l\t\tjail has /dev/log\t\n");
 266         fprintf(stderr, "  -u\t\tjail has a ubus socket\t\n");
 267
 268         return -1;
 269 }
 270
 271 static int child_running = 1;
 272
 273 static void child_process_handler(struct uloop_process *c, int ret)
 274 {
 275         INFO("child (%d) exited: %d\n", c->pid, ret);
 276         uloop_end();
 277         child_running = 0;
 278 }
 279
 280 struct uloop_process child_process = {
 281         .cb = child_process_handler,
 282 };
 283
 284 static int spawn_child(void *arg)
 285 {
 286         char *path = get_current_dir_name();
 287         int procfs = 0, sysfs = 0;
 288         char *seccomp = NULL;
 289         char **argv = arg;
 290         int argc = 0, ch;
 291         char *mpoint;
 292         int ronly = 0;
 293
 294         while (argv[argc])
 295                 argc++;
 296
 297         optind = 0;
 298         while ((ch = getopt(argc, argv, OPT_ARGS)) != -1) {
 299                 switch (ch) {
 300                 case 'd':
 301                         debug = 1;
 302                         break;
 303                 case 'S':
 304                         seccomp = optarg;
 305                         break;
 306                 case 'p':
 307                         procfs = 1;
 308                         break;
 309                 case 'o':
 310                         ronly = 1;
 311                         break;
 312                 case 's':
 313                         sysfs = 1;
 314                         break;
 315                 case 'n':
 316                         if (sethostname(optarg, strlen(optarg)))
 317                                 ERROR("failed to sethostname: %s\n", strerror(errno));
 318                         break;
 319                 }
 320         }
 321
 322         if (asprintf(&mpoint, "%s/old", path) < 0) {
 323                 ERROR("failed to alloc pivot path: %s\n", strerror(errno));
 324                 return -1;
 325         }
 326         mkdir_p(mpoint, 0755);
 327         if (pivot_root(path, mpoint) == -1) {
 328                 ERROR("pivot_root failed:%s\n", strerror(errno));
 329                 return -1;
 330         }
 331         free(mpoint);
 332         umount2("/old", MNT_DETACH);
 333         rmdir("/old");
 334         if (procfs) {
 335                 mkdir("/proc", 0755);
 336                 mount("proc", "/proc", "proc", MS_NOATIME, 0);
 337         }
 338         if (sysfs) {
 339                 mkdir("/sys", 0755);
 340                 mount("sysfs", "/sys", "sysfs", MS_NOATIME, 0);
 341         }
 342         if (ronly)
 343                 mount(NULL, "/", NULL, MS_RDONLY | MS_REMOUNT, 0);
 344
 345         uloop_init();
 346
 347         child_process.pid = spawn(path, &argv[optind], seccomp);
 348         uloop_process_add(&child_process);
 349         uloop_run();
 350         uloop_done();
 351         if (child_running) {
 352                 kill(child_process.pid, SIGTERM);
 353                 waitpid(child_process.pid, NULL, 0);
 354         }
 355
 356         return 0;
 357 }
 358
 359 static int namespace_running = 1;
 360
 361 static void namespace_process_handler(struct uloop_process *c, int ret)
 362 {
 363         INFO("namespace (%d) exited: %d\n", c->pid, ret);
 364         uloop_end();
 365         namespace_running = 0;
 366 }
 367
 368 struct uloop_process namespace_process = {
 369         .cb = namespace_process_handler,
 370 };
 371
 372 static void spawn_namespace(const char *path, int argc, char **argv)
 373 {
 374         char *dir = get_current_dir_name();
 375
 376         uloop_init();
 377         if (chdir(path)) {
 378                 ERROR("failed to chdir() into the jail\n");
 379                 return;
 380         }
 381         namespace_process.pid = clone(spawn_child,
 382                         child_stack + STACK_SIZE,
 383                         CLONE_NEWUTS | CLONE_NEWPID | CLONE_NEWNS | SIGCHLD, argv);
 384
 385         if (namespace_process.pid != -1) {
 386                 if (chdir(dir))
 387                         ERROR("failed to chdir() out of the jail\n");
 388                 free(dir);
 389                 uloop_process_add(&namespace_process);
 390                 uloop_run();
 391                 uloop_done();
 392                 if (namespace_running) {
 393                         kill(namespace_process.pid, SIGTERM);
 394                         waitpid(namespace_process.pid, NULL, 0);
 395                 }
 396         } else {
 397                 ERROR("failed to spawn namespace: %s\n", strerror(errno));
 398         }
 399 }
 400
 401 static void add_extra(char *name, int readonly)
 402 {
 403         struct extra *f;
 404
 405         if (*name != '/') {
 406                 ERROR("%s is not an absolute path\n", name);
 407                 return;
 408         }
 409
 410         f = calloc(1, sizeof(struct extra));
 411
 412         f->name = basename(name);
 413         f->path = dirname(strdup(name));
 414         f->readonly = readonly;
 415
 416         list_add_tail(&f->list, &extras);
 417 }
 418
 419 int main(int argc, char **argv)
 420 {
 421         uid_t uid = getuid();
 422         const char *name = NULL;
 423         char *path = NULL;
 424         struct stat s;
 425         int ch, ret;
 426         char log[] = "/dev/log";
 427         char ubus[] = "/var/run/ubus.sock";
 428
 429         if (uid) {
 430                 ERROR("not root, aborting: %s\n", strerror(errno));
 431                 return -1;
 432         }
 433
 434         umask(022);
 435
 436         while ((ch = getopt(argc, argv, OPT_ARGS)) != -1) {
 437                 switch (ch) {
 438                 case 'd':
 439                         debug = 1;
 440                         break;
 441                 case 'P':
 442                         path = optarg;
 443                         break;
 444                 case 'n':
 445                         name = optarg;
 446                         break;
 447                 case 'S':
 448                 case 'r':
 449                         add_extra(optarg, 1);
 450                         break;
 451                 case 'w':
 452                         add_extra(optarg, 0);
 453                         break;
 454                 case 'u':
 455                         add_extra(ubus, 0);
 456                         break;
 457                 case 'l':
 458                         add_extra(log, 0);
 459                         break;
 460                 }
 461         }
 462
 463         if (argc - optind < 1)
 464                 return usage();
 465
 466         if (!path && asprintf(&path, "/tmp/%s", basename(argv[optind])) == -1) {
 467                 ERROR("failed to set root path\n: %s", strerror(errno));
 468                 return -1;
 469         }
 470
 471         if (!stat(path, &s)) {
 472                 ERROR("%s already exists: %s\n", path, strerror(errno));
 473                 return -1;
 474         }
 475
 476         if (name)
 477                 prctl(PR_SET_NAME, name, NULL, NULL, NULL);
 478
 479         avl_init(&libraries, avl_strcmp, false, NULL);
 480         alloc_library_path("/lib64");
 481         alloc_library_path("/lib");
 482         alloc_library_path("/usr/lib");
 483         load_ldso_conf("/etc/ld.so.conf");
 484
 485         if (elf_load_deps(argv[optind])) {
 486                 ERROR("failed to load dependencies\n");
 487                 return -1;
 488         }
 489
 490         if (elf_load_deps("libpreload-seccomp.so")) {
 491                 ERROR("failed to load libpreload-seccomp.so\n");
 492                 return -1;
 493         }
 494
 495         ret = build_jail(path);
 496
 497         if (!ret)
 498                 spawn_namespace(path, argc, argv);
 499         else
 500                 ERROR("failed to build jail\n");
 501
 502         stop_jail(path);
 503
 504         return ret;
 505 }