ubase/switch_root.c

132 lines
2.8 KiB
C
Raw Normal View History

2014-04-13 18:07:06 -04:00
/* See LICENSE file for copyright and license details. */
2014-06-30 14:03:41 -04:00
#include <sys/mount.h>
#include <sys/stat.h>
#include <sys/vfs.h>
#include <dirent.h>
#include <fcntl.h>
#include <limits.h>
2014-04-13 18:07:06 -04:00
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
2014-06-30 14:03:41 -04:00
2014-04-13 18:07:06 -04:00
#include "util.h"
#define RAMFS_MAGIC 0x858458f6 /* some random number */
#define TMPFS_MAGIC 0x01021994
2014-04-13 18:07:06 -04:00
static void
delete_content(const char *dir, dev_t curdevice)
{
2014-06-27 06:57:26 -04:00
char path[PATH_MAX];
2014-04-13 18:07:06 -04:00
DIR *d;
struct stat st;
struct dirent *dent;
/* don't dive into other filesystems */
2014-06-27 06:57:26 -04:00
if (lstat(dir, &st) < 0 || st.st_dev != curdevice)
2014-04-13 18:07:06 -04:00
return;
2014-06-27 06:57:26 -04:00
if (!(d = opendir(dir)))
return;
while ((dent = readdir(d))) {
if (strcmp(dent->d_name, ".") == 0 ||
strcmp(dent->d_name, "..") == 0)
continue;
2014-04-13 18:07:06 -04:00
2014-06-27 06:57:26 -04:00
/* build path and dive deeper */
if (strlcpy(path, dir, sizeof(path)) >= sizeof(path))
eprintf("path too long\n");
if (path[strlen(path) - 1] != '/')
if (strlcat(path, "/", sizeof(path)) >= sizeof(path))
eprintf("path too long\n");
if (strlcat(path, dent->d_name, sizeof(path)) >= sizeof(path))
eprintf("path too long\n");
2014-04-13 18:07:06 -04:00
2014-06-27 06:57:26 -04:00
if (lstat(path, &st) < 0)
weprintf("lstat %s:", path);
2014-04-13 18:07:06 -04:00
2014-06-27 06:57:26 -04:00
if (S_ISDIR(st.st_mode)) {
delete_content(path, curdevice);
if (rmdir(path) < 0)
weprintf("rmdir %s:", path);
} else {
if (unlink(path) < 0)
weprintf("unlink %s:", path);
2014-04-13 18:07:06 -04:00
}
}
2014-06-27 06:57:26 -04:00
closedir(d);
2014-04-13 18:07:06 -04:00
}
static void
usage(void)
{
eprintf("usage: %s [-c console] [newroot] [init] (PID 1)\n", argv0);
}
int
2014-04-18 06:49:10 -04:00
main(int argc, char *argv[])
2014-04-13 18:07:06 -04:00
{
char *console = NULL;
dev_t curdev;
struct stat st;
struct statfs stfs;
ARGBEGIN {
case 'c':
console = EARGF(usage());
break;
default:
usage();
} ARGEND;
/* check number of args and if we are PID 1 */
if (argc != 2 || getpid() != 1)
2014-04-13 18:07:06 -04:00
usage();
/* chdir to newroot and make sure it's a different fs */
if (chdir(argv[0]))
2014-04-13 18:07:06 -04:00
eprintf("chdir %s:", argv[0]);
if (stat("/", &st))
2014-04-13 18:07:06 -04:00
eprintf("stat %s:", "/");
2014-04-13 18:07:06 -04:00
curdev = st.st_dev;
if (stat(".", &st))
2014-04-13 18:07:06 -04:00
eprintf("stat %s:", ".");
if (st.st_dev == curdev)
2014-04-13 18:07:06 -04:00
usage();
/* avoids trouble with real filesystems */
if (stat("/init", &st) || !S_ISREG(st.st_mode))
2014-04-13 18:07:06 -04:00
eprintf("/init is not a regular file\n");
2014-04-13 18:07:06 -04:00
statfs("/", &stfs);
if ((unsigned)stfs.f_type != RAMFS_MAGIC && (unsigned)stfs.f_type != TMPFS_MAGIC)
2014-04-13 18:07:06 -04:00
eprintf("current filesystem is not a RAMFS or TMPFS\n");
/* wipe / */
delete_content("/", curdev);
/* overmount / with newroot and chroot into it */
if (mount(".", "/", NULL, MS_MOVE, NULL))
2014-04-13 18:07:06 -04:00
eprintf("mount %s:", ".");
if (chroot("."))
2014-04-13 18:07:06 -04:00
eprintf("chroot failed\n");
/* if -c is set, redirect stdin/stdout/stderr to console */
if (console) {
2014-11-30 08:12:15 -05:00
close(0);
2014-07-09 11:39:32 -04:00
if (open(console, O_RDWR) == -1)
2014-04-13 18:07:06 -04:00
eprintf("open %s:", console);
2014-11-30 08:12:15 -05:00
dup2(0, 1);
dup2(0, 2);
2014-04-13 18:07:06 -04:00
}
/* execute init */
execv(argv[1], argv);
eprintf("can't execute '%s'\n", argv[1]);
2014-10-02 18:45:25 -04:00
return 1;
2014-04-13 18:07:06 -04:00
}