summaryrefslogtreecommitdiffstats
path: root/mdk-stage1
diff options
context:
space:
mode:
Diffstat (limited to 'mdk-stage1')
-rw-r--r--mdk-stage1/disk.c8
-rw-r--r--mdk-stage1/lomount.c21
2 files changed, 17 insertions, 12 deletions
diff --git a/mdk-stage1/disk.c b/mdk-stage1/disk.c
index ac38f82e0..df2552bd5 100644
--- a/mdk-stage1/disk.c
+++ b/mdk-stage1/disk.c
@@ -108,7 +108,7 @@ static enum return_type try_with_device(char *dev_name)
strcpy(device_fullname, "/dev/");
strcat(device_fullname, choice);
-
+
if (my_mount(device_fullname, disk_own_mount, "ext2") == -1 &&
my_mount(device_fullname, disk_own_mount, "vfat") == -1 &&
my_mount(device_fullname, disk_own_mount, "reiserfs") == -1) {
@@ -140,7 +140,7 @@ static enum return_type try_with_device(char *dev_name)
log_message("%s exists and is not a directory, assuming this is an ISO image", location_full);
if (lomount(location_full, IMAGE_LOCATION)) {
error_message("Could not mount ISO image.");
- umount(IMAGE_LOCATION);
+ umount(disk_own_mount);
return try_with_device(dev_name);
}
} else
@@ -154,11 +154,11 @@ static enum return_type try_with_device(char *dev_name)
"Here's a short extract of the files in the directory:\n"
"%s", list_directory(IMAGE_LOCATION));
umount(disk_own_mount);
- unlink(IMAGE_LOCATION);
return try_with_device(dev_name);
}
if (load_ramdisk() != RETURN_OK) {
error_message("Could not load program into memory.");
+ umount(disk_own_mount);
return try_with_device(dev_name);
}
} else {
@@ -170,7 +170,6 @@ static enum return_type try_with_device(char *dev_name)
"Here's a short extract of the files in the directory:\n"
"%s", list_directory(IMAGE_LOCATION));
umount(disk_own_mount);
- unlink(IMAGE_LOCATION);
return try_with_device(dev_name);
}
if (readlink(IMAGE_LOCATION LIVE_LOCATION "/usr/bin/runinstall2", &p, 1) != 1) {
@@ -178,7 +177,6 @@ static enum return_type try_with_device(char *dev_name)
"You need more memory to perform an installation from a Windows partition. "
"Another solution if to copy the " DISTRIB_NAME " Distribution on a Linux partition.");
umount(disk_own_mount);
- unlink(IMAGE_LOCATION);
return try_with_device(dev_name);
}
log_message("found the " DISTRIB_NAME " Installation, good news!");
diff --git a/mdk-stage1/lomount.c b/mdk-stage1/lomount.c
index ebb107e56..bf152af26 100644
--- a/mdk-stage1/lomount.c
+++ b/mdk-stage1/lomount.c
@@ -70,8 +70,10 @@ set_loop (const char *device, const char *file)
if ((ffd = open (file, mode)) < 0)
return 1;
- if ((fd = open (device, mode)) < 0)
+ if ((fd = open (device, mode)) < 0) {
+ close(ffd);
return 1;
+ }
memset(&loopinfo, 0, sizeof (loopinfo));
strncpy(loopinfo.lo_name, file, LO_NAME_SIZE);
@@ -84,17 +86,22 @@ set_loop (const char *device, const char *file)
* passwd etc being swapped out and left somewhere on disk.
*/
- if(mlockall(MCL_CURRENT|MCL_FUTURE)) {
- log_message("CRITICAL Couldn't lock into memory! %s (memlock)", strerror(errno));
- return 1;
- }
+ if(mlockall(MCL_CURRENT|MCL_FUTURE)) {
+ log_message("CRITICAL Couldn't lock into memory! %s (memlock)", strerror(errno));
+ return 1;
+ }
#endif
- if (ioctl(fd, LOOP_SET_FD, ffd) < 0)
+ if (ioctl(fd, LOOP_SET_FD, ffd) < 0) {
+ close(fd);
+ close(ffd);
return 1;
-
+ }
+
if (ioctl(fd, LOOP_SET_STATUS, &loopinfo) < 0) {
(void) ioctl (fd, LOOP_CLR_FD, 0);
+ close(fd);
+ close(ffd);
return 1;
}