umount.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini umount implementation for busybox
   4  *
   5  *
   6  * Copyright (C) 1999,2000,2001 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 <stdio.h>
  26 #include <mntent.h>
  27 #include <errno.h>
  28 #include <string.h>
  29 #include <stdlib.h>
  30 #include "busybox.h"
  31
  32
  33 static const int MNT_FORCE = 1;
  34 static const int MS_MGC_VAL = 0xc0ed0000; /* Magic number indicatng "new" flags */
  35 static const int MS_REMOUNT = 32;       /* Alter flags of a mounted FS.  */
  36 static const int MS_RDONLY = 1; /* Mount read-only.  */
  37
  38 extern int mount (__const char *__special_file, __const char *__dir,
  39                         __const char *__fstype, unsigned long int __rwflag,
  40                         __const void *__data);
  41 extern int umount (__const char *__special_file);
  42 extern int umount2 (__const char *__special_file, int __flags);
  43
  44 struct _mtab_entry_t {
  45         char *device;
  46         char *mountpt;
  47         struct _mtab_entry_t *next;
  48 };
  49
  50 static struct _mtab_entry_t *mtab_cache = NULL;
  51
  52
  53
  54 #if defined BB_FEATURE_MOUNT_FORCE
  55 static int doForce = FALSE;
  56 #endif
  57 #if defined BB_FEATURE_MOUNT_LOOP
  58 static int freeLoop = TRUE;
  59 #endif
  60 #if defined BB_FEATURE_MTAB_SUPPORT
  61 static int useMtab = TRUE;
  62 #endif
  63 static int umountAll = FALSE;
  64 static int doRemount = FALSE;
  65 extern const char mtab_file[];  /* Defined in utility.c */
  66
  67
  68
  69 /* These functions are here because the getmntent functions do not appear
  70  * to be re-entrant, which leads to all sorts of problems when we try to
  71  * use them recursively - randolph
  72  *
  73  * TODO: Perhaps switch to using Glibc's getmntent_r
  74  *        -Erik
  75  */
  76 void mtab_read(void)
  77 {
  78         struct _mtab_entry_t *entry = NULL;
  79         struct mntent *e;
  80         FILE *fp;
  81
  82         if (mtab_cache != NULL)
  83                 return;
  84
  85         if ((fp = setmntent(mtab_file, "r")) == NULL) {
  86                 error_msg("Cannot open %s", mtab_file);
  87                 return;
  88         }
  89         while ((e = getmntent(fp))) {
  90                 entry = xmalloc(sizeof(struct _mtab_entry_t));
  91                 entry->device = strdup(e->mnt_fsname);
  92                 entry->mountpt = strdup(e->mnt_dir);
  93                 entry->next = mtab_cache;
  94                 mtab_cache = entry;
  95         }
  96         endmntent(fp);
  97 }
  98
  99 char *mtab_getinfo(const char *match, const char which)
 100 {
 101         struct _mtab_entry_t *cur = mtab_cache;
 102
 103         while (cur) {
 104                 if (strcmp(cur->mountpt, match) == 0 ||
 105                         strcmp(cur->device, match) == 0) {
 106                         if (which == MTAB_GETMOUNTPT) {
 107                                 return cur->mountpt;
 108                         } else {
 109 #if !defined BB_FEATURE_MTAB_SUPPORT
 110                                 if (strcmp(cur->device, "/dev/root") == 0) {
 111                                         /* Adjusts device to be the real root device,
 112                                          * or leaves device alone if it can't find it */
 113                                         find_real_root_device_name( cur->device);
 114                                         return ( cur->device);
 115                                 }
 116 #endif
 117                                 return cur->device;
 118                         }
 119                 }
 120                 cur = cur->next;
 121         }
 122         return NULL;
 123 }
 124
 125 char *mtab_first(void **iter)
 126 {
 127         struct _mtab_entry_t *mtab_iter;
 128
 129         if (!iter)
 130                 return NULL;
 131         mtab_iter = mtab_cache;
 132         *iter = (void *) mtab_iter;
 133         return mtab_next(iter);
 134 }
 135
 136 char *mtab_next(void **iter)
 137 {
 138         char *mp;
 139
 140         if (iter == NULL || *iter == NULL)
 141                 return NULL;
 142         mp = ((struct _mtab_entry_t *) (*iter))->mountpt;
 143         *iter = (void *) ((struct _mtab_entry_t *) (*iter))->next;
 144         return mp;
 145 }
 146
 147 /* Don't bother to clean up, since exit() does that
 148  * automagically, so we can save a few bytes */
 149 #ifdef BB_FEATURE_CLEAN_UP
 150 void mtab_free(void)
 151 {
 152         struct _mtab_entry_t *this, *next;
 153
 154         this = mtab_cache;
 155         while (this) {
 156                 next = this->next;
 157                 if (this->device)
 158                         free(this->device);
 159                 if (this->mountpt)
 160                         free(this->mountpt);
 161                 free(this);
 162                 this = next;
 163         }
 164 }
 165 #endif
 166
 167 static int do_umount(const char *name)
 168 {
 169         int status;
 170         char *blockDevice = mtab_getinfo(name, MTAB_GETDEVICE);
 171
 172         if (blockDevice && strcmp(blockDevice, name) == 0)
 173                 name = mtab_getinfo(blockDevice, MTAB_GETMOUNTPT);
 174
 175         status = umount(name);
 176
 177 #if defined BB_FEATURE_MOUNT_LOOP
 178         if (freeLoop == TRUE && blockDevice != NULL && !strncmp("/dev/loop", blockDevice, 9))
 179                 /* this was a loop device, delete it */
 180                 del_loop(blockDevice);
 181 #endif
 182 #if defined BB_FEATURE_MOUNT_FORCE
 183         if (status != 0 && doForce == TRUE) {
 184                 status = umount2(blockDevice, MNT_FORCE);
 185                 if (status != 0) {
 186                         error_msg_and_die("forced umount of %s failed!", blockDevice);
 187                 }
 188         }
 189 #endif
 190         if (status != 0 && doRemount == TRUE && errno == EBUSY) {
 191                 status = mount(blockDevice, name, NULL,
 192                                            MS_MGC_VAL | MS_REMOUNT | MS_RDONLY, NULL);
 193                 if (status == 0) {
 194                         error_msg("%s busy - remounted read-only", blockDevice);
 195                 } else {
 196                         error_msg("Cannot remount %s read-only", blockDevice);
 197                 }
 198         }
 199         if (status == 0) {
 200 #if defined BB_FEATURE_MTAB_SUPPORT
 201                 if (useMtab == TRUE)
 202                         erase_mtab(name);
 203 #endif
 204                 return (TRUE);
 205         }
 206         return (FALSE);
 207 }
 208
 209 static int umount_all(void)
 210 {
 211         int status = TRUE;
 212         char *mountpt;
 213         void *iter;
 214
 215         for (mountpt = mtab_first(&iter); mountpt; mountpt = mtab_next(&iter)) {
 216                 /* Never umount /proc on a umount -a */
 217                 if (strstr(mountpt, "proc")!= NULL)
 218                         continue;
 219                 if (!do_umount(mountpt)) {
 220                         /* Don't bother retrying the umount on busy devices */
 221                         if (errno == EBUSY) {
 222                                 perror_msg("%s", mountpt);
 223                                 status = FALSE;
 224                                 continue;
 225                         }
 226                         if (!do_umount(mountpt)) {
 227                                 printf("Couldn't umount %s on %s: %s\n",
 228                                            mountpt, mtab_getinfo(mountpt, MTAB_GETDEVICE),
 229                                            strerror(errno));
 230                                 status = FALSE;
 231                         }
 232                 }
 233         }
 234         return (status);
 235 }
 236
 237 extern int umount_main(int argc, char **argv)
 238 {
 239         if (argc < 2) {
 240                 show_usage();
 241         }
 242 #ifdef BB_FEATURE_CLEAN_UP
 243         atexit(mtab_free);
 244 #endif
 245
 246         /* Parse any options */
 247         while (--argc > 0 && **(++argv) == '-') {
 248                 while (*++(*argv))
 249                         switch (**argv) {
 250                         case 'a':
 251                                 umountAll = TRUE;
 252                                 break;
 253 #if defined BB_FEATURE_MOUNT_LOOP
 254                         case 'l':
 255                                 freeLoop = FALSE;
 256                                 break;
 257 #endif
 258 #ifdef BB_FEATURE_MTAB_SUPPORT
 259                         case 'n':
 260                                 useMtab = FALSE;
 261                                 break;
 262 #endif
 263 #ifdef BB_FEATURE_MOUNT_FORCE
 264                         case 'f':
 265                                 doForce = TRUE;
 266                                 break;
 267 #endif
 268                         case 'r':
 269                                 doRemount = TRUE;
 270                                 break;
 271                         case 'v':
 272                                 break; /* ignore -v */
 273                         default:
 274                                 show_usage();
 275                         }
 276         }
 277
 278         mtab_read();
 279         if (umountAll == TRUE) {
 280                 if (umount_all() == TRUE)
 281                         return EXIT_SUCCESS;
 282                 else
 283                         return EXIT_FAILURE;
 284         }
 285         if (do_umount(*argv) == TRUE)
 286                 return EXIT_SUCCESS;
 287         perror_msg_and_die("%s", *argv);
 288 }
 289