summaryrefslogtreecommitdiff
path: root/umount.c
blob: 4efc9f9d90b9422a28ef9648b92072dc152fab2e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
#include "internal.h"
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <string.h>
#include <stdio.h>
#include <mntent.h>
#include <sys/mount.h>

const char	umount_usage[] = "umount {filesystem|directory}\n"
"\tumount -a\n"
"\n"
"\tUnmount a filesystem.\n"
"\t-a:\tUnmounts all mounted filesystems.\n";

static char *
stralloc(const char * string)
{
	int	length = strlen(string) + 1;
	char *	n = malloc(length);
	memcpy(n, string, length);
	return n;
}

extern void
erase_mtab(const char * name)
{
	struct mntent	entries[100];
	int		count = 0;
	FILE *		mountTable = setmntent("/etc/mtab", "r");
	struct mntent *	m;

	if ( mountTable == 0
	 && (mountTable = setmntent("/proc/mounts", "r")) == 0 ) {
		name_and_error("/etc/mtab");
		return;
	}

	while ( (m = getmntent(mountTable)) != 0 ) {
		entries[count].mnt_fsname = stralloc(m->mnt_fsname);
		entries[count].mnt_dir = stralloc(m->mnt_dir);
		entries[count].mnt_type = stralloc(m->mnt_type);
		entries[count].mnt_opts = stralloc(m->mnt_opts);
		entries[count].mnt_freq = m->mnt_freq;
		entries[count].mnt_passno = m->mnt_passno;
		count++;
	}
	endmntent(mountTable);
	if ( (mountTable = setmntent("/etc/mtab", "w")) ) {
		int	i;
		for ( i = 0; i < count; i++ ) {
			int result = ( strcmp(entries[i].mnt_fsname, name) == 0
			 || strcmp(entries[i].mnt_dir, name) == 0 );

			if ( result )
				continue;
			else
				addmntent(mountTable, &entries[i]);
		}
		endmntent(mountTable);
	}
	else if ( errno != EROFS )
		name_and_error("/etc/mtab");
}

static int
umount_all(int noMtab)
{
	struct mntent	entries[100];
	int				count = 0;
	FILE *			mountTable = setmntent("/etc/mtab", "r");
	struct mntent *	m;
	int				status = 0;

	if ( mountTable == 0
	 && (mountTable = setmntent("/proc/mounts", "r")) == 0 ) {
		name_and_error("/etc/mtab");
		return 1;
	}

	while ( (m = getmntent(mountTable)) != 0 ) {
		entries[count].mnt_fsname = stralloc(m->mnt_fsname);
		count++;
	}
	endmntent(mountTable);

	while ( count > 0 ) {
		int result = umount(entries[--count].mnt_fsname) == 0;
		/* free(entries[count].mnt_fsname); */
		if ( result ) {
			if ( !noMtab )
				erase_mtab(entries[count].mnt_fsname);
		}
		else {
			status = 1;
			name_and_error(entries[count].mnt_fsname);
		}
	}
	return status;
}

extern int
do_umount(const char * name, int noMtab)
{
	if ( umount(name) == 0 ) {
		if ( !noMtab )
			erase_mtab(name);
		return 0;
	}
	return 1;
}

extern int
umount_main(struct FileInfo * i, int argc, char * * argv)
{
	int	noMtab = 0;

	if ( argv[1][0] == '-' ) {
		switch ( argv[1][1] ) {
		case 'a':
			return umount_all(noMtab);
		case 'n':
			noMtab = 1;
			break;
		default:
			usage(umount_usage);
			return 1;
		}
	}
	if ( do_umount(argv[1],noMtab) != 0 ) {
		fprintf(stderr, "%s: %s.\n", argv[1], strerror(errno));
		return 1;
	}
	return 0;
}