Use errorMsg rather than fprintf.
[oweals/busybox.git] / util-linux / umount.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini umount implementation for busybox
4  *
5  *
6  * Copyright (C) 1999,2000 by Lineo, inc.
7  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17  * General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22  *
23  */
24
25 #include "internal.h"
26 #include <stdio.h>
27 #include <mntent.h>
28 #include <errno.h>
29
30
31 #define MNT_FORCE               1
32 #define MS_MGC_VAL              0xc0ed0000 /* Magic number indicatng "new" flags */
33 #define MS_REMOUNT              32      /* Alter flags of a mounted FS.  */
34 #define MS_RDONLY               1       /* Mount read-only.  */
35
36 extern int mount (__const char *__special_file, __const char *__dir,
37                         __const char *__fstype, unsigned long int __rwflag,
38                         __const void *__data);
39 extern int umount (__const char *__special_file);
40 extern int umount2 (__const char *__special_file, int __flags);
41
42
43
44 static const char umount_usage[] =
45         "umount [flags] filesystem|directory\n"
46 #ifndef BB_FEATURE_TRIVIAL_HELP
47         "Unmount file systems\n"
48         "\nFlags:\n" "\t-a:\tUnmount all file systems"
49 #ifdef BB_MTAB
50         " in /etc/mtab\n\t-n:\tDon't erase /etc/mtab entries\n"
51 #else
52         "\n"
53 #endif
54         "\t-r:\tTry to remount devices as read-only if mount is busy\n"
55 #if defined BB_FEATURE_MOUNT_FORCE
56         "\t-f:\tForce filesystem umount (i.e. unreachable NFS server)\n"
57 #endif
58 #if defined BB_FEATURE_MOUNT_LOOP
59         "\t-l:\tDo not free loop device (if a loop device has been used)\n"
60 #endif
61 #endif
62 ;
63
64 struct _mtab_entry_t {
65         char *device;
66         char *mountpt;
67         struct _mtab_entry_t *next;
68 };
69
70 static struct _mtab_entry_t *mtab_cache = NULL;
71
72
73
74 #if defined BB_FEATURE_MOUNT_FORCE
75 static int doForce = FALSE;
76 #endif
77 #if defined BB_FEATURE_MOUNT_LOOP
78 static int freeLoop = TRUE;
79 #endif
80 static int useMtab = TRUE;
81 static int umountAll = FALSE;
82 static int doRemount = FALSE;
83 extern const char mtab_file[];  /* Defined in utility.c */
84
85
86
87 /* These functions are here because the getmntent functions do not appear
88  * to be re-entrant, which leads to all sorts of problems when we try to
89  * use them recursively - randolph
90  *
91  * TODO: Perhaps switch to using Glibc's getmntent_r
92  *        -Erik
93  */
94 void mtab_read(void)
95 {
96         struct _mtab_entry_t *entry = NULL;
97         struct mntent *e;
98         FILE *fp;
99
100         if (mtab_cache != NULL)
101                 return;
102
103         if ((fp = setmntent(mtab_file, "r")) == NULL) {
104                 errorMsg("Cannot open %s\n", mtab_file);
105                 return;
106         }
107         while ((e = getmntent(fp))) {
108                 entry = xmalloc(sizeof(struct _mtab_entry_t));
109                 entry->device = strdup(e->mnt_fsname);
110                 entry->mountpt = strdup(e->mnt_dir);
111                 entry->next = mtab_cache;
112                 mtab_cache = entry;
113         }
114         endmntent(fp);
115 }
116
117 char *mtab_getinfo(const char *match, const char which)
118 {
119         struct _mtab_entry_t *cur = mtab_cache;
120
121         while (cur) {
122                 if (strcmp(cur->mountpt, match) == 0 ||
123                         strcmp(cur->device, match) == 0) {
124                         if (which == MTAB_GETMOUNTPT) {
125                                 return cur->mountpt;
126                         } else {
127 #if !defined BB_MTAB
128                                 if (strcmp(cur->device, "/dev/root") == 0) {
129                                         /* Adjusts device to be the real root device,
130                                          * or leaves device alone if it can't find it */
131                                         find_real_root_device_name( cur->device);
132                                         return ( cur->device);
133                                 }
134 #endif
135                                 return cur->device;
136                         }
137                 }
138                 cur = cur->next;
139         }
140         return NULL;
141 }
142
143 char *mtab_first(void **iter)
144 {
145         struct _mtab_entry_t *mtab_iter;
146
147         if (!iter)
148                 return NULL;
149         mtab_iter = mtab_cache;
150         *iter = (void *) mtab_iter;
151         return mtab_next(iter);
152 }
153
154 char *mtab_next(void **iter)
155 {
156         char *mp;
157
158         if (iter == NULL || *iter == NULL)
159                 return NULL;
160         mp = ((struct _mtab_entry_t *) (*iter))->mountpt;
161         *iter = (void *) ((struct _mtab_entry_t *) (*iter))->next;
162         return mp;
163 }
164
165 /* Don't bother to clean up, since exit() does that 
166  * automagically, so we can save a few bytes */
167 #if 0
168 void mtab_free(void)
169 {
170         struct _mtab_entry_t *this, *next;
171
172         this = mtab_cache;
173         while (this) {
174                 next = this->next;
175                 if (this->device)
176                         free(this->device);
177                 if (this->mountpt)
178                         free(this->mountpt);
179                 free(this);
180                 this = next;
181         }
182 }
183 #endif
184
185 static int do_umount(const char *name, int useMtab)
186 {
187         int status;
188         char *blockDevice = mtab_getinfo(name, MTAB_GETDEVICE);
189
190         if (blockDevice && strcmp(blockDevice, name) == 0)
191                 name = mtab_getinfo(blockDevice, MTAB_GETMOUNTPT);
192
193         status = umount(name);
194
195 #if defined BB_FEATURE_MOUNT_LOOP
196         if (freeLoop == TRUE && blockDevice != NULL && !strncmp("/dev/loop", blockDevice, 9))
197                 /* this was a loop device, delete it */
198                 del_loop(blockDevice);
199 #endif
200 #if defined BB_FEATURE_MOUNT_FORCE
201         if (status != 0 && doForce == TRUE) {
202                 status = umount2(blockDevice, MNT_FORCE);
203                 if (status != 0) {
204                         fatalError("forced umount of %s failed!\n", blockDevice);
205                 }
206         }
207 #endif
208         if (status != 0 && doRemount == TRUE && errno == EBUSY) {
209                 status = mount(blockDevice, name, NULL,
210                                            MS_MGC_VAL | MS_REMOUNT | MS_RDONLY, NULL);
211                 if (status == 0) {
212                         errorMsg("%s busy - remounted read-only\n", blockDevice);
213                 } else {
214                         errorMsg("Cannot remount %s read-only\n", blockDevice);
215                 }
216         }
217         if (status == 0) {
218 #if defined BB_MTAB
219                 if (useMtab == TRUE)
220                         erase_mtab(name);
221 #endif
222                 return (TRUE);
223         }
224         return (FALSE);
225 }
226
227 static int umount_all(int useMtab)
228 {
229         int status = TRUE;
230         char *mountpt;
231         void *iter;
232
233         for (mountpt = mtab_first(&iter); mountpt; mountpt = mtab_next(&iter)) {
234                 /* Never umount /proc on a umount -a */
235                 if (strstr(mountpt, "proc")!= NULL)
236                         continue;
237                 status = do_umount(mountpt, useMtab);
238                 if (status != 0) {
239                         /* Don't bother retrying the umount on busy devices */
240                         if (errno == EBUSY) {
241                                 perror(mountpt);
242                                 continue;
243                         }
244                         status = do_umount(mountpt, useMtab);
245                         if (status != 0) {
246                                 printf("Couldn't umount %s on %s: %s\n",
247                                            mountpt, mtab_getinfo(mountpt, MTAB_GETDEVICE),
248                                            strerror(errno));
249                         }
250                 }
251         }
252         return (status);
253 }
254
255 extern int umount_main(int argc, char **argv)
256 {
257         if (argc < 2) {
258                 usage(umount_usage);
259         }
260
261         /* Parse any options */
262         while (--argc > 0 && **(++argv) == '-') {
263                 while (*++(*argv))
264                         switch (**argv) {
265                         case 'a':
266                                 umountAll = TRUE;
267                                 break;
268 #if defined BB_FEATURE_MOUNT_LOOP
269                         case 'l':
270                                 freeLoop = FALSE;
271                                 break;
272 #endif
273 #ifdef BB_MTAB
274                         case 'n':
275                                 useMtab = FALSE;
276                                 break;
277 #endif
278 #ifdef BB_FEATURE_MOUNT_FORCE
279                         case 'f':
280                                 doForce = TRUE;
281                                 break;
282 #endif
283                         case 'r':
284                                 doRemount = TRUE;
285                                 break;
286                         case 'v':
287                                 break; /* ignore -v */
288                         default:
289                                 usage(umount_usage);
290                         }
291         }
292
293         mtab_read();
294         if (umountAll == TRUE) {
295                 exit(umount_all(useMtab));
296         }
297         if (do_umount(*argv, useMtab) == 0)
298                 exit(TRUE);
299         perror("umount");
300         return(FALSE);
301 }
302