The tendency of vi to auto-indent can be really annoying at times.
[oweals/busybox.git] / util-linux / umount.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini umount implementation for busybox
4  *
5  * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
6  * Copyright (C) 2005 by Rob Landley <rob@landley.net>
7  *
8  * This program is licensed under the GNU General Public license (GPL)
9  * version 2 or later, see http://www.fsf.org/licensing/licenses/gpl.html
10  * or the file "LICENSE" in the busybox source tarball for the full text.
11  *
12  */
13
14 #include <limits.h>
15 #include <stdio.h>
16 #include <mntent.h>
17 #include <errno.h>
18 #include <string.h>
19 #include <stdlib.h>
20 #include <sys/mount.h>
21 #include <getopt.h>
22 #include "busybox.h"
23
24 #define OPTION_STRING           "flDnrva"
25 #define OPT_FORCE                       1
26 #define OPT_LAZY                        2
27 #define OPT_DONTFREELOOP        4
28 #define OPT_NO_MTAB                     8
29 #define OPT_REMOUNT                     16
30 #define OPT_IGNORED                     32      // -v is ignored
31 #define OPT_ALL                         (ENABLE_FEATURE_UMOUNT_ALL ? 64 : 0)
32
33 extern int umount_main(int argc, char **argv)
34 {
35         int doForce;
36         char path[2*PATH_MAX];
37         struct mntent me;
38         FILE *fp;
39         int status=EXIT_SUCCESS;
40         unsigned long opt;
41         struct mtab_list {
42                 char *dir;
43                 char *device;
44                 struct mtab_list *next;
45         } *mtl, *m;
46
47         /* Parse any options */
48
49         opt = bb_getopt_ulflags (argc, argv, OPTION_STRING);
50
51         argc -= optind;
52         argv += optind;
53
54         doForce = MAX((opt & OPT_FORCE), (opt & OPT_LAZY));
55
56         /* Get a list of mount points from mtab.  We read them all in now mostly
57          * for umount -a (so we don't have to worry about the list changing while
58          * we iterate over it, or about getting stuck in a loop on the same failing
59          * entry.  Notice that this also naturally reverses the list so that -a
60          * umounts the most recent entries first. */
61
62         m=mtl=0;
63         if(opt & OPT_ALL) {
64
65                 /* If we're umounting all, then m points to the start of the list and
66                  * the argument list should be empty (which will match all). */
67
68                 if(!(fp = setmntent(bb_path_mtab_file, "r")))
69                         bb_error_msg_and_die("Cannot open %s", bb_path_mtab_file);
70                 while (getmntent_r(fp,&me,path,sizeof(path))) {
71                         m=xmalloc(sizeof(struct mtab_list));
72                         m->next=mtl;
73                         m->device=bb_xstrdup(me.mnt_fsname);
74                         m->dir=bb_xstrdup(me.mnt_dir);
75                         mtl=m;
76                 }
77                 endmntent(fp);
78
79         /* If we're not mounting all, we need at least one argument. */
80         } else if(argc <= 0) bb_show_usage();
81         
82         // Loop through everything we're supposed to umount, and do so.
83         for(;;) {
84                 int curstat;
85
86                 // Do we already know what to umount this time through the loop?
87                 if(m) safe_strncpy(path,m->dir,PATH_MAX);
88                 // For umount -a, end of mtab means time to exit.
89                 else if(opt & OPT_ALL) break;
90                 // Get next command line argument (and look it up in mtab list)
91                 else if(!argc--) break;
92                 else {
93                         // Get next command line argument (and look it up in mtab list)
94                         realpath(*argv++, path);
95                         if (ENABLE_FEATURE_MTAB_SUPPORT)
96                                 for(m = mtl; m; m = m->next)
97                                         if(!strcmp(path, m->dir) || !strcmp(path, m->device))
98                                                 break;
99                 }
100
101                 // Let's ask the thing nicely to unmount.
102                 curstat = umount(path);
103
104                 // Force the unmount, if necessary.
105                 if(curstat && doForce) {
106                         curstat = umount2(path, doForce);
107                         if(curstat)
108                                 bb_error_msg_and_die("forced umount of %s failed!", path);
109                 }
110
111                 // If still can't umount, maybe remount read-only?
112                 if (curstat && (opt & OPT_REMOUNT) && errno == EBUSY && m) {
113                         curstat = mount(m->device, path, NULL, MS_REMOUNT|MS_RDONLY, NULL);
114                         bb_error_msg(curstat ? "Cannot remount %s read-only" :
115                                                  "%s busy - remounted read-only", m->device);
116                 }
117
118                 /* De-allocate the loop device.  This ioctl should be ignored on any
119                  * non-loop block devices. */
120                 if(ENABLE_FEATURE_MOUNT_LOOP && !(opt & OPT_DONTFREELOOP) && m)
121                         del_loop(m->device);
122
123                 if(curstat) {
124                         /* Yes, the ENABLE is redundant here, but the optimizer for ARM
125                          * can't do simple constant propagation in local variables... */
126                         if(ENABLE_FEATURE_MTAB_SUPPORT && !(opt & OPT_NO_MTAB) && m)
127                                 erase_mtab(m->dir);
128                         status = EXIT_FAILURE;
129                         bb_perror_msg("Couldn't umount %s", path);
130                 }
131                 // Find next matching mtab entry for -a or umount /dev
132                 while(m && (m = m->next))
133                         if((opt & OPT_ALL) || !strcmp(path,m->device))
134                                 break;
135         }
136
137         // Free mtab list if necessary
138
139         if(ENABLE_FEATURE_CLEAN_UP) {
140                 while(mtl) {
141                         m=mtl->next;
142                         free(mtl->device);
143                         free(mtl->dir);
144                         free(mtl);
145                         mtl=m;
146                 }
147         }
148
149         return status;
150 }