diff options
Diffstat (limited to 'util-linux')
| -rw-r--r-- | util-linux/mount.c | 165 | ||||
| -rw-r--r-- | util-linux/umount.c | 70 |
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 |