ubase/mount.c

240 lines
5.2 KiB
C
Raw Normal View History

2013-08-12 07:32:34 -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/types.h>
#include <limits.h>
#include <mntent.h>
2013-08-12 07:32:34 -04:00
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
2014-06-30 14:03:41 -04:00
2013-08-12 07:32:34 -04:00
#include "util.h"
struct {
const char *opt;
const char *notopt;
unsigned long v;
} optnames[] = {
{ "defaults", NULL, 0 },
2014-03-15 09:26:32 -04:00
{ "remount", NULL, MS_REMOUNT },
{ "ro", "rw", MS_RDONLY },
{ "sync", "async", MS_SYNCHRONOUS },
{ "dirsync", NULL, MS_DIRSYNC },
{ "nodev", "dev", MS_NODEV },
{ "noatime", "atime", MS_NOATIME },
{ "nodiratime", "diratime", MS_NODIRATIME },
{ "noexec", "exec", MS_NOEXEC },
{ "nosuid", "suid", MS_NOSUID },
{ "mand", "nomand", MS_MANDLOCK },
{ "relatime", "norelatime", MS_RELATIME },
{ "bind", NULL, MS_BIND },
{ NULL, NULL, 0 }
2013-08-12 07:32:34 -04:00
};
2014-03-15 09:26:32 -04:00
static void
parseopts(char *popts, unsigned long *flags, char *data, size_t datasiz)
2014-03-15 09:26:32 -04:00
{
unsigned int i, validopt;
size_t optlen, dlen = 0;
char *name;
2014-03-15 09:26:32 -04:00
data[0] = '\0';
2014-07-09 11:39:32 -04:00
for (name = strtok(popts, ","); name; name = strtok(NULL, ",")) {
2014-03-15 09:26:32 -04:00
validopt = 0;
2014-07-09 11:39:32 -04:00
for (i = 0; optnames[i].opt; i++) {
if (optnames[i].opt && strcmp(name, optnames[i].opt) == 0) {
2014-03-15 09:26:32 -04:00
*flags |= optnames[i].v;
validopt = 1;
break;
}
2014-07-09 11:39:32 -04:00
if (optnames[i].notopt && strcmp(name, optnames[i].notopt) == 0) {
2014-03-15 09:26:32 -04:00
*flags &= ~optnames[i].v;
validopt = 1;
break;
}
}
2014-07-09 11:39:32 -04:00
if (!validopt) {
2014-03-15 09:26:32 -04:00
/* unknown option, pass as data option to mount() */
2014-07-09 11:39:32 -04:00
if ((optlen = strlen(name))) {
if (dlen + optlen + 2 >= datasiz)
2014-03-15 09:26:32 -04:00
return; /* prevent overflow */
2014-07-09 11:39:32 -04:00
if (dlen)
2014-03-15 09:26:32 -04:00
data[dlen++] = ',';
memcpy(&data[dlen], name, optlen);
dlen += optlen;
data[dlen] = '\0';
}
}
}
}
static int
mounted(const char *dir)
{
FILE *fp;
struct mntent *me;
struct stat st1, st2;
if (stat(dir, &st1) < 0) {
weprintf("stat %s:", dir);
return 0;
}
fp = setmntent("/proc/mounts", "r");
if (!fp)
eprintf("setmntent %s:", "/proc/mounts");
while ((me = getmntent(fp)) != NULL) {
if (stat(me->mnt_dir, &st2) < 0) {
weprintf("stat %s:", me->mnt_dir);
continue;
}
if (st1.st_dev == st2.st_dev &&
st1.st_ino == st2.st_ino)
return 1;
}
endmntent(fp);
return 0;
}
2013-08-12 07:32:34 -04:00
static void
usage(void)
{
2014-03-15 09:26:32 -04:00
eprintf("usage: %s [-BMRan] [-t fstype] [-o options] [source] [target]\n",
2013-08-12 07:32:34 -04:00
argv0);
}
static int
catfile(FILE *in, FILE *out)
{
char buf[BUFSIZ];
size_t bytesread;
2014-07-09 11:39:32 -04:00
while (!feof(in)) {
bytesread = fread(buf, 1, sizeof(buf), in);
2014-07-09 11:39:32 -04:00
if (ferror(in))
return 0;
fwrite(buf, 1, bytesread, out);
}
return 1;
}
2013-08-12 07:32:34 -04:00
int
main(int argc, char *argv[])
{
int aflag = 0, oflag = 0, status = EXIT_SUCCESS, i;
2013-08-12 07:32:34 -04:00
unsigned long flags = 0;
char *types = NULL, data[512] = "", *resolvpath = NULL;
2014-03-15 09:26:32 -04:00
char *files[] = { "/proc/mounts", "/etc/fstab", NULL };
size_t datasiz = sizeof(data);
const char *source, *target;
struct mntent *me = NULL;
FILE *fp;
2014-03-15 09:26:32 -04:00
2013-08-12 07:32:34 -04:00
ARGBEGIN {
case 'B':
flags |= MS_BIND;
break;
case 'M':
flags |= MS_MOVE;
break;
case 'R':
flags |= MS_REC;
break;
case 'a':
aflag = 1;
break;
2013-08-12 07:32:34 -04:00
case 'o':
oflag = 1;
2014-03-15 09:26:32 -04:00
parseopts(EARGF(usage()), &flags, data, datasiz);
2013-08-12 07:32:34 -04:00
break;
case 't':
types = EARGF(usage());
break;
case 'n':
break;
2013-08-12 07:32:34 -04:00
default:
usage();
} ARGEND;
2014-07-09 11:39:32 -04:00
if (argc < 1 && aflag == 0) {
if (!(fp = fopen(files[0], "r")))
eprintf("fopen %s:", files[0]);
2014-07-09 11:39:32 -04:00
if (catfile(fp, stdout) != 1) {
weprintf("error while reading %s:", files[0]);
status = EXIT_FAILURE;
}
fclose(fp);
return status;
}
2013-08-12 07:32:34 -04:00
2014-07-09 11:39:32 -04:00
if (aflag == 1)
2014-03-15 09:26:32 -04:00
goto mountall;
2013-08-12 07:32:34 -04:00
source = argv[0];
target = argv[1];
2014-07-09 11:39:32 -04:00
if (!target) {
target = argv[0];
2013-08-12 07:32:34 -04:00
source = NULL;
2014-07-09 11:39:32 -04:00
if (!(resolvpath = realpath(target, NULL)))
eprintf("realpath %s:", target);
target = resolvpath;
2014-03-15 09:26:32 -04:00
}
2014-07-09 11:39:32 -04:00
for (i = 0; files[i]; i++) {
if (!(fp = setmntent(files[i], "r"))) {
if (strcmp(files[i], "/proc/mounts") != 0)
weprintf("setmntent %s:", files[i]);
continue;
}
2014-07-09 11:39:32 -04:00
while ((me = getmntent(fp))) {
if (strcmp(me->mnt_dir, target) == 0 ||
strcmp(me->mnt_fsname, target) == 0 ||
(source && strcmp(me->mnt_dir, source) == 0) ||
(source && strcmp(me->mnt_fsname, source) == 0)) {
2014-07-09 11:39:32 -04:00
if (!source) {
target = me->mnt_dir;
source = me->mnt_fsname;
}
2014-07-09 11:39:32 -04:00
if (!oflag)
parseopts(me->mnt_opts, &flags, data, datasiz);
2014-07-09 11:39:32 -04:00
if (!types)
types = me->mnt_type;
goto mountsingle;
}
}
endmntent(fp);
fp = NULL;
2013-08-12 07:32:34 -04:00
}
2014-07-09 11:39:32 -04:00
if (!source)
eprintf("can't find %s in /etc/fstab\n", target);
2013-08-12 07:32:34 -04:00
2014-03-15 09:26:32 -04:00
mountsingle:
2014-07-09 11:39:32 -04:00
if (mount(source, target, types, flags, data) < 0) {
weprintf("mount: %s:", source);
status = EXIT_FAILURE;
}
2014-07-09 11:39:32 -04:00
if (fp)
endmntent(fp);
free(resolvpath);
return status;
2014-03-15 09:26:32 -04:00
mountall:
2014-07-09 11:39:32 -04:00
if (!(fp = setmntent("/etc/fstab", "r")))
2014-03-15 09:26:32 -04:00
eprintf("setmntent %s:", "/etc/fstab");
2014-07-09 11:39:32 -04:00
while ((me = getmntent(fp))) {
2014-03-15 09:26:32 -04:00
flags = 0;
parseopts(me->mnt_opts, &flags, data, datasiz);
2014-07-09 11:39:32 -04:00
if (mount(me->mnt_fsname, me->mnt_dir, me->mnt_type, flags, data) < 0) {
if (mounted(me->mnt_dir) == 0) {
weprintf("mount: %s:", me->mnt_fsname);
status = EXIT_FAILURE;
}
}
}
2014-03-15 09:26:32 -04:00
endmntent(fp);
return status;
2013-08-12 07:32:34 -04:00
}