ubase/umount.c

91 lines
1.5 KiB
C
Raw Normal View History

/* See LICENSE file for copyright and license details. */
2014-03-07 08:49:40 -05:00
#include <mntent.h>
2013-08-07 04:52:10 -04:00
#include <stdio.h>
2013-10-07 14:11:40 -04:00
#include <stdlib.h>
2014-03-12 10:35:20 -04:00
#include <string.h>
2014-03-07 08:49:40 -05:00
#include <sys/mount.h>
2013-08-07 04:52:10 -04:00
#include "util.h"
static int umountall(int);
2013-08-07 04:52:10 -04:00
static void
usage(void)
{
2014-03-12 10:28:56 -04:00
weprintf("usage: %s [-lfn] target\n", argv0);
weprintf("usage: %s -a [-lfn]\n", argv0);
exit(EXIT_FAILURE);
2013-08-07 04:52:10 -04:00
}
int
main(int argc, char *argv[])
{
2013-08-07 04:52:10 -04:00
int i;
2014-03-07 08:49:40 -05:00
int aflag = 0;
int flags = 0;
2013-10-07 14:11:40 -04:00
int ret = EXIT_SUCCESS;
2013-08-07 04:52:10 -04:00
ARGBEGIN {
2014-03-07 08:49:40 -05:00
case 'a':
aflag = 1;
break;
2013-08-07 04:52:10 -04:00
case 'f':
flags |= MNT_FORCE;
break;
case 'l':
flags |= MNT_DETACH;
2013-08-07 04:52:10 -04:00
break;
case 'n':
break;
2013-08-07 04:52:10 -04:00
default:
usage();
} ARGEND;
2013-08-10 06:38:30 -04:00
2014-03-07 08:49:40 -05:00
if (argc < 1 && aflag == 0)
2013-08-07 04:52:10 -04:00
usage();
2013-08-10 06:38:30 -04:00
if (aflag == 1)
return umountall(flags);
2014-03-07 08:49:40 -05:00
2013-08-07 04:52:10 -04:00
for (i = 0; i < argc; i++) {
if (umount2(argv[i], flags) < 0) {
weprintf("umount2 %s:", argv[i]);
ret = EXIT_FAILURE;
}
2013-08-07 04:52:10 -04:00
}
return ret;
}
static int
umountall(int flags)
{
FILE *fp;
struct mntent *me;
int ret;
char **mntdirs = NULL;
int len = 0;
fp = setmntent("/proc/mounts", "r");
if (!fp)
eprintf("setmntent %s:", "/proc/mounts");
while ((me = getmntent(fp))) {
if (strcmp(me->mnt_type, "proc") == 0)
continue;
mntdirs = realloc(mntdirs, ++len * sizeof(*mntdirs));
if (!mntdirs)
eprintf("realloc:");
mntdirs[len - 1] = strdup(me->mnt_dir);
2014-03-15 14:48:01 -04:00
if (!mntdirs[len - 1])
eprintf("strdup:");
}
endmntent(fp);
while (--len >= 0) {
if (umount2(mntdirs[len], flags) < 0) {
weprintf("umount2 %s:", mntdirs[len]);
ret = EXIT_FAILURE;
}
free(mntdirs[len]);
}
free(mntdirs);
return ret;
}