More stuff.

2 files changed, 109 insertions, 30 deletions

diff --git a/util-linux/mount.c b/util-linux/mount.c
index 1efbdf407..8b5efe14f 100644
--- a/util-linux/mount.c
+++ b/util-linux/mount.c
@@ -41,11 +41,17 @@
 #include <ctype.h>
 #include <fstab.h>
 
+extern const char mtab_file[]; /* Defined in utility.c */
+
 static const char mount_usage[] = "Usage:\tmount [flags]\n"
     "\tmount [flags] device directory [-o options,more-options]\n"
     "\n"
     "Flags:\n"
     "\t-a:\tMount all file systems in fstab.\n"
+#ifdef BB_MTAB
+    "\t-f:\t\"Fake\" mount. Add entry to mount table but don't mount it.\n"
+    "\t-n:\tDon't write a mount table entry.\n"
+#endif
     "\t-o option:\tOne of many filesystem options, listed below.\n"
     "\t-r:\tMount the filesystem read-only.\n"
     "\t-t filesystem-type:\tSpecify the filesystem type.\n"
@@ -62,6 +68,7 @@ static const char mount_usage[] = "Usage:\tmount [flags]\n"
     "There are EVEN MORE flags that are specific to each filesystem.\n"
     "You'll have to see the written documentation for those.\n";
 
+
 struct mount_options {
     const char *name;
     unsigned long and;
@@ -84,6 +91,29 @@ static const struct mount_options mount_options[] = {
     {0, 0, 0}
 };
 
+#if ! defined BB_MTAB
+#define do_mount(specialfile, dir, filesystemtype, flags, string_flags, useMtab, fakeIt) \
+        mount(specialfile, dir, filesystemtype, flags, string_flags)
+#else
+static int
+do_mount(char* specialfile, char* dir, char* filesystemtype, 
+        long flags, void* string_flags, int useMtab, int fakeIt)
+{
+    int status=0;
+
+    if (fakeIt==FALSE)
+        status=mount(specialfile, dir, filesystemtype, flags, string_flags);
+
+    if ( status == 0 ) {
+        if ( useMtab==TRUE )
+            write_mtab(specialfile, dir, filesystemtype, flags, string_flags);
+        return 0;
+    }
+    else 
+        return( status);
+}
+#endif
+
 
 /* Seperate standard mount options from the nonstandard string options */
 static void
@@ -126,9 +156,8 @@ parse_mount_options ( char *options, unsigned long *flags, char *strflags)
 }
 
 int
-mount_one (
-           char *blockDevice, char *directory, char *filesystemType,
-           unsigned long flags, char *string_flags)
+mount_one(char *blockDevice, char *directory, char *filesystemType,
+           unsigned long flags, char *string_flags, int useMtab, int fakeIt)
 {
     int status = 0;
 
@@ -152,16 +181,16 @@ mount_one (
                 filesystemType = buf;
                 filesystemType++;       // hop past tab
 
-                status = mount (blockDevice, directory, filesystemType,
-                                flags | MS_MGC_VAL, string_flags);
+                status = do_mount (blockDevice, directory, filesystemType,
+                                flags | MS_MGC_VAL, string_flags, useMtab, fakeIt);
                 if (status == 0)
                     break;
             }
         }
         fclose (f);
     } else {
-        status = mount (blockDevice, directory, filesystemType,
-                        flags | MS_MGC_VAL, string_flags);
+        status = do_mount (blockDevice, directory, filesystemType,
+                        flags | MS_MGC_VAL, string_flags, useMtab, fakeIt);
     }
 
     if (status) {
@@ -180,15 +209,17 @@ extern int mount_main (int argc, char **argv)
     char *device = NULL;
     char *directory = NULL;
     struct stat statBuf;
-    int all = 0;
+    int all = FALSE;
+    int fakeIt = FALSE;
+    int useMtab = TRUE;
     int i;
 
     if (stat("/etc/fstab", &statBuf) < 0) 
         fprintf(stderr, "/etc/fstab file missing -- Please install one.\n\n");
 
     if (argc == 1) {
-        FILE *mountTable;
-        if ((mountTable = setmntent ("/proc/mounts", "r"))) {
+        FILE *mountTable = setmntent (mtab_file, "r");
+        if (mountTable) {
             struct mntent *m;
             while ((m = getmntent (mountTable)) != 0) {
                 struct fstab* fstabItem;
@@ -203,6 +234,8 @@ extern int mount_main (int argc, char **argv)
                         m->mnt_type, m->mnt_opts);
             }
             endmntent (mountTable);
+        } else {
+            perror(mtab_file);
         }
         exit( TRUE);
     }
@@ -241,6 +274,14 @@ extern int mount_main (int argc, char **argv)
             case 'a':
                 all = TRUE;
                 break;
+#ifdef BB_MTAB
+            case 'f':
+                fakeIt = TRUE;
+                break;
+            case 'n':
+                useMtab = FALSE;
+                break;
+#endif
             case 'v':
             case 'h':
             case '-':
@@ -263,7 +304,6 @@ extern int mount_main (int argc, char **argv)
     }
 
     if (all == TRUE) {
-        long newFlags;
         struct mntent *m;
         FILE *f = setmntent ("/etc/fstab", "r");
 
@@ -279,17 +319,18 @@ extern int mount_main (int argc, char **argv)
                     (!strstr (m->mnt_type, "swap")) && 
                     (!strstr (m->mnt_type, "nfs"))) 
             {
-                newFlags = flags;
+                flags = 0;
                 *string_flags = '\0';
-                parse_mount_options(m->mnt_opts, &newFlags, string_flags);
-                mount_one (m->mnt_fsname, m->mnt_dir, m->mnt_type, newFlags, string_flags);
+                parse_mount_options(m->mnt_opts, &flags, string_flags);
+                mount_one (m->mnt_fsname, m->mnt_dir, m->mnt_type, 
+                        flags, string_flags, useMtab, fakeIt);
             }
         }
         endmntent (f);
     } else {
         if (device && directory) {
             exit (mount_one (device, directory, filesystemType, 
-                        flags, string_flags));
+                        flags, string_flags, useMtab, fakeIt));
         } else {
             fprintf (stderr, "%s\n", mount_usage);
             exit( FALSE);
diff --git a/util-linux/umount.c b/util-linux/umount.c
index 95f7dfb3c..e749c5f0f 100644
--- a/util-linux/umount.c
+++ b/util-linux/umount.c
@@ -29,24 +29,54 @@
 #include <errno.h>
 
 static const char umount_usage[] = 
-"Usage: umount filesystem\n"
-"   or: umount directory\n"
-"   or: umount -a"
-"to unmount all mounted file systems.\n";
+"Usage: umount [flags] filesystem|directory\n"
+"Optional Flags:\n"
+"\t-a:\tUnmount all file systems"
+#ifdef BB_MTAB
+" in /etc/mtab\n\t-n:\tDon't erase /etc/mtab entries\n"
+#else
+"\n"
+#endif
+;
+
+
+static int useMtab = TRUE;
+static int umountAll = FALSE;
+extern const char mtab_file[]; /* Defined in utility.c */
+
+#if ! defined BB_MTAB
+#define do_umount( blockDevice, useMtab) umount( blockDevice)
+#else
+static int 
+do_umount(const char* name, int useMtab)
+{
+    int status = umount(name);
+
+    if ( status == 0 ) {
+        if ( useMtab==TRUE )
+            erase_mtab(name);
+        return 0;
+    }
+    else 
+        return( status);
+}
+#endif
 
 static int
-umount_all()
+umount_all(int useMtab)
 {
         int status;
         struct mntent *m;
         FILE *mountTable;
 
-        if ((mountTable = setmntent ("/proc/mounts", "r"))) {
+        if ((mountTable = setmntent (mtab_file, "r"))) {
             while ((m = getmntent (mountTable)) != 0) {
                 char *blockDevice = m->mnt_fsname;
+#if ! defined BB_MTAB
                 if (strcmp (blockDevice, "/dev/root") == 0)
                     blockDevice = (getfsfile ("/"))->fs_spec;
-                status=umount (m->mnt_dir);
+#endif
+                status=do_umount (m->mnt_dir, useMtab);
                 if (status!=0) {
                     /* Don't bother retrying the umount on busy devices */
                     if (errno==EBUSY) {
@@ -56,7 +86,7 @@ umount_all()
                     printf ("Trying to umount %s failed: %s\n", 
                             m->mnt_dir, strerror(errno)); 
                     printf ("Instead trying to umount %s\n", blockDevice); 
-                    status=umount (blockDevice);
+                    status=do_umount (blockDevice, useMtab);
                     if (status!=0) {
                         printf ("Couldn't umount %s on %s (type %s): %s\n", 
                                 blockDevice, m->mnt_dir, m->mnt_type, strerror(errno));
@@ -69,27 +99,35 @@ umount_all()
 }
 
 extern int
-umount_main(int argc, char * * argv)
+umount_main(int argc, char** argv)
 {
 
     if (argc < 2) {
         usage( umount_usage); 
     }
-    argc--;
-    argv++;
 
     /* Parse any options */
-    while (**argv == '-') {
+    while (argc-- > 0 && **(++argv) == '-') {
         while (*++(*argv)) switch (**argv) {
             case 'a':
-                exit ( umount_all() );
+                umountAll = TRUE;
+                break;
+#ifdef BB_MTAB
+            case 'n':
+                useMtab = FALSE;
                 break;
+#endif
             default:
                 usage( umount_usage);
         }
     }
-    if ( umount(*argv) == 0 )
-             exit (TRUE);
+
+
+    if(umountAll) {
+        exit(umount_all(useMtab));
+    }
+    if ( do_umount(*argv,useMtab) == 0 )
+        exit (TRUE);
     else {
         perror("umount");
         exit( FALSE);