mount and umount could leak loop device allocations causing the system to

quickly run out.  Also disable init's SIGHUP handler during shutdown.
 -Erik

2 files changed, 107 insertions, 128 deletions

diff --git a/util-linux/mount.c b/util-linux/mount.c
index 709c7fc..3c1568a 100644
--- a/util-linux/mount.c
+++ b/util-linux/mount.c
@@ -127,11 +127,11 @@ do_mount(char* specialfile, char* dir, char* filesystemtype,
 	    specialfile = find_unused_loop_device();
 	    if (specialfile == NULL) {
 		fprintf(stderr, "Could not find a spare loop device\n");
-		exit(1);
+		return( FALSE);
 	    }
 	    if (set_loop (specialfile, lofile, 0, &loro)) {
 		fprintf(stderr, "Could not setup loop device\n");
-		exit(1);
+		return( FALSE);
 	    }
 	    if (!(flags & MS_RDONLY) && loro) { /* loop is ro, but wanted rw */
 		fprintf(stderr, "WARNING: loop device is read-only\n");
@@ -141,15 +141,26 @@ do_mount(char* specialfile, char* dir, char* filesystemtype,
 #endif
 	status=mount(specialfile, dir, filesystemtype, flags, string_flags);
     }
-#if defined BB_MTAB
+
+
+    /* If the mount was sucessful, do anything needed, then return TRUE */
     if (status == 0) {
-	if (useMtab==TRUE)
+
+#if defined BB_MTAB
+	if (useMtab==TRUE) {
 	    write_mtab(specialfile, dir, filesystemtype, flags, mtab_opts);
-	return 0;
+	}
+#endif
+	return( TRUE);
+    }
+
+    /* Bummer.  mount failed.  Clean up */
+#if defined BB_FEATURE_MOUNT_LOOP
+    if (specialfile != NULL) {
+	del_loop(specialfile);
     }
-    else
 #endif
-	return(status);
+    return( FALSE);
 }
 
 
@@ -166,6 +177,75 @@ extern void whine_if_fstab_is_missing()
 #endif
 
 
+#if defined BB_FEATURE_MOUNT_LOOP
+static int set_loop(const char *device, const char *file, int offset, int *loopro)
+{
+	struct loop_info loopinfo;
+	int	fd, ffd, mode;
+	
+	mode = *loopro ? O_RDONLY : O_RDWR;
+	if ((ffd = open (file, mode)) < 0 && !*loopro
+	    && (errno != EROFS || (ffd = open (file, mode = O_RDONLY)) < 0)) {
+	  perror (file);
+	  return 1;
+	}
+	if ((fd = open (device, mode)) < 0) {
+	  close(ffd);
+	  perror (device);
+	  return 1;
+	}
+	*loopro = (mode == O_RDONLY);
+
+	memset(&loopinfo, 0, sizeof(loopinfo));
+	strncpy(loopinfo.lo_name, file, LO_NAME_SIZE);
+	loopinfo.lo_name[LO_NAME_SIZE-1] = 0;
+
+	loopinfo.lo_offset = offset;
+
+	loopinfo.lo_encrypt_key_size = 0;
+	if (ioctl(fd, LOOP_SET_FD, ffd) < 0) {
+		perror("ioctl: LOOP_SET_FD");
+		close(fd);
+		close(ffd);
+		return 1;
+	}
+	if (ioctl(fd, LOOP_SET_STATUS, &loopinfo) < 0) {
+		(void) ioctl(fd, LOOP_CLR_FD, 0);
+		perror("ioctl: LOOP_SET_STATUS");
+		close(fd);
+		close(ffd);
+		return 1;
+	}
+	close(fd);
+	close(ffd);
+	return 0;
+}
+
+char *find_unused_loop_device (void)
+{
+	char dev[20];
+	int i, fd;
+	struct stat statbuf;
+	struct loop_info loopinfo;
+
+	for(i = 0; i <= 7; i++) {
+	    sprintf(dev, "/dev/loop%d", i);
+	    if (stat (dev, &statbuf) == 0 && S_ISBLK(statbuf.st_mode)) {
+		if ((fd = open (dev, O_RDONLY)) >= 0) {
+		    if(ioctl (fd, LOOP_GET_STATUS, &loopinfo) == -1) {
+			if (errno == ENXIO) { /* probably free */
+			    close (fd);
+			    return strdup(dev);
+			}
+		    }
+		    close (fd);
+		}
+	    }
+	}
+        return NULL;
+}
+#endif /* BB_FEATURE_MOUNT_LOOP */
+
 /* Seperate standard mount options from the nonstandard string options */
 static void
 parse_mount_options ( char *options, unsigned long *flags, char *strflags)
@@ -240,7 +320,7 @@ mount_one(char *blockDevice, char *directory, char *filesystemType,
 		status = do_mount (blockDevice, directory, filesystemType,
 				flags | MS_MGC_VAL, string_flags, useMtab, 
 				fakeIt, mtab_opts);
-		if (status == 0)
+		if (status == TRUE)
 		    break;
 	    }
 	}
@@ -253,7 +333,7 @@ mount_one(char *blockDevice, char *directory, char *filesystemType,
 			fakeIt, mtab_opts);
     }
 
-    if (status) {
+    if (status==FALSE) {
 	fprintf (stderr, "Mounting %s on %s failed: %s\n",
 		 blockDevice, directory, strerror(errno));
 	return (FALSE);
@@ -400,70 +480,3 @@ goodbye:
     usage( mount_usage);
 }
 
-#if defined BB_FEATURE_MOUNT_LOOP
-static int set_loop(const char *device, const char *file, int offset, int *loopro)
-{
-	struct loop_info loopinfo;
-	int	fd, ffd, mode;
-	
-	mode = *loopro ? O_RDONLY : O_RDWR;
-	if ((ffd = open (file, mode)) < 0 && !*loopro
-	    && (errno != EROFS || (ffd = open (file, mode = O_RDONLY)) < 0)) {
-	  perror (file);
-	  return 1;
-	}
-	if ((fd = open (device, mode)) < 0) {
-	  close(ffd);
-	  perror (device);
-	  return 1;
-	}
-	*loopro = (mode == O_RDONLY);
-
-	memset(&loopinfo, 0, sizeof(loopinfo));
-	strncpy(loopinfo.lo_name, file, LO_NAME_SIZE);
-	loopinfo.lo_name[LO_NAME_SIZE-1] = 0;
-
-	loopinfo.lo_offset = offset;
-
-	loopinfo.lo_encrypt_key_size = 0;
-	if (ioctl(fd, LOOP_SET_FD, ffd) < 0) {
-		perror("ioctl: LOOP_SET_FD");
-		close(fd);
-		close(ffd);
-		return 1;
-	}
-	if (ioctl(fd, LOOP_SET_STATUS, &loopinfo) < 0) {
-		(void) ioctl(fd, LOOP_CLR_FD, 0);
-		perror("ioctl: LOOP_SET_STATUS");
-		close(fd);
-		close(ffd);
-		return 1;
-	}
-	close(fd);
-	close(ffd);
-	return 0;
-}
-
-char *find_unused_loop_device (void)
-{
-	char dev[20];
-	int i, fd;
-	struct stat statbuf;
-	struct loop_info loopinfo;
-
-	for(i = 0; i <= 7; i++) {
-		sprintf(dev, "/dev/loop%d", i);
-		if (stat (dev, &statbuf) == 0 && S_ISBLK(statbuf.st_mode)) {
-			if ((fd = open (dev, O_RDONLY)) >= 0) {
-				if(ioctl (fd, LOOP_GET_STATUS, &loopinfo) == -1 &&
-				   errno == ENXIO) { /* probably free */
-					close (fd);
-					return strdup(dev);
-				}
-				close (fd);
-			}
-		}
-	}
-        return NULL;
-}
-#endif /* BB_FEATURE_MOUNT_LOOP */
diff --git a/util-linux/umount.c b/util-linux/umount.c
index 9ad6f26..68b27e3 100644
--- a/util-linux/umount.c
+++ b/util-linux/umount.c
@@ -28,14 +28,6 @@
 #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[] = 
 "umount [flags] filesystem|directory\n\n"
 "Flags:\n"
@@ -52,43 +44,34 @@ static int useMtab = TRUE;
 static int umountAll = FALSE;
 extern const char mtab_file[]; /* Defined in utility.c */
 
+#define MIN(x,y) (x > y ? x : y)
+
 static int 
 do_umount(const char* name, int useMtab)
 {
     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;
+    struct mntent *m;
+    FILE *mountTable;
+    const char *blockDevice = NULL;
+
+    if ((mountTable = setmntent (mtab_file, "r"))) {
+	while ((m = getmntent (mountTable)) != 0) {
+	    if (strncmp(m->mnt_dir, name,
+			MIN(strlen(m->mnt_dir),strlen(name))) == 0)
+		blockDevice = m->mnt_fsname;
+	    else if (strcmp(m->mnt_fsname, name) == 0) {
+		blockDevice = name;
+		name = m->mnt_dir;
+	    }
 	}
     }
-#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;
-    }
+    if (blockDevice != NULL && !strncmp("/dev/loop", blockDevice, 9))
+	/* this was a loop device, delete it */
+	del_loop(blockDevice);
 #endif
 #if defined BB_MTAB
     if ( status == 0 ) {
@@ -178,20 +161,3 @@ umount_main(int argc, char** argv)
     }
 }
 
-#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