1 files changed, 64 insertions, 8 deletions

diff --git a/umount.c b/umount.c
index ab76be4da..af1b3a43e 100644
--- a/umount.c
+++ b/umount.c
@@ -28,6 +28,14 @@
 #include <fstab.h>
 #include <errno.h>
 
+#if defined BB_FEATURE_MOUNT_LOOP
+#include <fcntl.h>
+#include <sys/ioctl.h>
+#include <linux/loop.h>
+
+static int del_loop(const char *device);
+#endif
+
 static const char umount_usage[] = 
 "Usage: umount [flags] filesystem|directory\n\n"
 "Flags:\n"
@@ -44,23 +52,54 @@ 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);
+    int status;
+
+#if defined BB_FEATURE_MOUNT_LOOP
+    /* check to see if this is a loop device */
+    struct stat fst;
+    char dev[20];
+    const char *oldname = NULL;
+    int i;
+    
+    if (stat(name, &fst)) {
+        fprintf(stderr, "umount: %s: %s\n", name, strerror(errno));
+        exit(1);
+    }
+    for (i = 0 ; i <= 7 ; i++) {
+        struct stat lst;
+        sprintf(dev, "/dev/loop%d", i);
+        if (stat(dev, &lst))
+            continue;
+        if (lst.st_dev == fst.st_dev) {
+            oldname = name;
+            name = dev;
+            break;
+        }
+    }
+#endif
+
+    status = umount(name);
 
+#if defined BB_FEATURE_MOUNT_LOOP
+    if (!strncmp("/dev/loop", name, 9)) { /* this was a loop device, delete it */
+        del_loop(name);
+        if (oldname != NULL)
+            name = oldname;
+    }
+#endif
+#if defined BB_MTAB
     if ( status == 0 ) {
         if ( useMtab==TRUE )
             erase_mtab(name);
         return 0;
     }
-    else 
-        return( status);
-}
+    else
 #endif
+        return(status);
+}
 
 static int
 umount_all(int useMtab)
@@ -135,7 +174,24 @@ umount_main(int argc, char** argv)
         exit (TRUE);
     else {
         perror("umount");
-        exit( FALSE);
+        exit(FALSE);
     }
 }
 
+#if defined BB_FEATURE_MOUNT_LOOP
+static int del_loop(const char *device)
+{
+        int fd;
+
+        if ((fd = open(device, O_RDONLY)) < 0) {
+                perror(device);
+                exit(1);
+        }
+        if (ioctl(fd, LOOP_CLR_FD, 0) < 0) {
+                perror("ioctl: LOOP_CLR_FD");
+                exit(1);
+        }
+        close(fd);
+        return(0);
+}
+#endif