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=0744")) {
 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                         sethostname(optarg, strlen(optarg));
 317                         break;
 318                 }
 319         }
 320
 321         asprintf(&mpoint, "%s/old", path);
 322         mkdir_p(mpoint, 0755);
 323         if (pivot_root(path, mpoint) == -1) {
 324                 ERROR("pivot_root failed:%s\n", strerror(errno));
 325                 return -1;
 326         }
 327         free(mpoint);
 328         umount2("/old", MNT_DETACH);
 329         rmdir("/old");
 330         if (procfs) {
 331                 mkdir("/proc", 0755);
 332                 mount("proc", "/proc", "proc", MS_NOATIME, 0);
 333         }
 334         if (sysfs) {
 335                 mkdir("/sys", 0755);
 336                 mount("sysfs", "/sys", "sysfs", MS_NOATIME, 0);
 337         }
 338         if (ronly)
 339                 mount(NULL, "/", NULL, MS_RDONLY | MS_REMOUNT, 0);
 340
 341         uloop_init();
 342
 343         child_process.pid = spawn(path, &argv[optind], seccomp);
 344         uloop_process_add(&child_process);
 345         uloop_run();
 346         uloop_done();
 347         if (child_running) {
 348                 kill(child_process.pid, SIGTERM);
 349                 waitpid(child_process.pid, NULL, 0);
 350         }
 351
 352         return 0;
 353 }
 354
 355 static int namespace_running = 1;
 356
 357 static void namespace_process_handler(struct uloop_process *c, int ret)
 358 {
 359         INFO("namespace (%d) exited: %d\n", c->pid, ret);
 360         uloop_end();
 361         namespace_running = 0;
 362 }
 363
 364 struct uloop_process namespace_process = {
 365         .cb = namespace_process_handler,
 366 };
 367
 368 static void spawn_namespace(const char *path, int argc, char **argv)
 369 {
 370         char *dir = get_current_dir_name();
 371
 372         uloop_init();
 373         chdir(path);
 374         namespace_process.pid = clone(spawn_child,
 375                         child_stack + STACK_SIZE,
 376                         CLONE_NEWUTS | CLONE_NEWPID | CLONE_NEWNS | SIGCHLD, argv);
 377
 378         if (namespace_process.pid != -1) {
 379                 chdir(dir);
 380                 free(dir);
 381                 uloop_process_add(&namespace_process);
 382                 uloop_run();
 383                 uloop_done();
 384                 if (namespace_running) {
 385                         kill(namespace_process.pid, SIGTERM);
 386                         waitpid(namespace_process.pid, NULL, 0);
 387                 }
 388         } else {
 389                 ERROR("failed to spawn namespace: %s\n", strerror(errno));
 390         }
 391 }
 392
 393 static void add_extra(char *name, int readonly)
 394 {
 395         struct extra *f;
 396
 397         if (*name != '/') {
 398                 ERROR("%s is not an absolute path\n", name);
 399                 return;
 400         }
 401
 402         f = calloc(1, sizeof(struct extra));
 403
 404         f->name = basename(name);
 405         f->path = dirname(strdup(name));
 406         f->readonly = readonly;
 407
 408         list_add_tail(&f->list, &extras);
 409 }
 410
 411 int main(int argc, char **argv)
 412 {
 413         uid_t uid = getuid();
 414         const char *name = NULL;
 415         char *path = NULL;
 416         struct stat s;
 417         int ch, ret;
 418         char log[] = "/dev/log";
 419         char ubus[] = "/var/run/ubus.sock";
 420
 421         if (uid) {
 422                 ERROR("not root, aborting: %s\n", strerror(errno));
 423                 return -1;
 424         }
 425
 426         umask(022);
 427
 428         while ((ch = getopt(argc, argv, OPT_ARGS)) != -1) {
 429                 switch (ch) {
 430                 case 'd':
 431                         debug = 1;
 432                         break;
 433                 case 'P':
 434                         path = optarg;
 435                         break;
 436                 case 'n':
 437                         name = optarg;
 438                         break;
 439                 case 'S':
 440                 case 'r':
 441                         add_extra(optarg, 1);
 442                         break;
 443                 case 'w':
 444                         add_extra(optarg, 0);
 445                         break;
 446                 case 'u':
 447                         add_extra(ubus, 0);
 448                         break;
 449                 case 'l':
 450                         add_extra(log, 0);
 451                         break;
 452                 }
 453         }
 454
 455         if (argc - optind < 1)
 456                 return usage();
 457
 458         if (!path && asprintf(&path, "/tmp/%s", basename(argv[optind])) == -1) {
 459                 ERROR("failed to set root path\n: %s", strerror(errno));
 460                 return -1;
 461         }
 462
 463         if (!stat(path, &s)) {
 464                 ERROR("%s already exists: %s\n", path, strerror(errno));
 465                 return -1;
 466         }
 467
 468         if (name)
 469                 prctl(PR_SET_NAME, name, NULL, NULL, NULL);
 470
 471         avl_init(&libraries, avl_strcmp, false, NULL);
 472         alloc_library_path("/lib64");
 473         alloc_library_path("/lib");
 474         alloc_library_path("/usr/lib");
 475         load_ldso_conf("/etc/ld.so.conf");
 476
 477         if (elf_load_deps(argv[optind])) {
 478                 ERROR("failed to load dependencies\n");
 479                 return -1;
 480         }
 481
 482         if (elf_load_deps("libpreload-seccomp.so")) {
 483                 ERROR("failed to load libpreload-seccomp.so\n");
 484                 return -1;
 485         }
 486
 487         ret = build_jail(path);
 488
 489         if (!ret)
 490                 spawn_namespace(path, argc, argv);
 491         else
 492                 ERROR("failed to build jail\n");
 493
 494         stop_jail(path);
 495
 496         return ret;
 497 }