diff options
| -rw-r--r-- | loader/init.c | 56 | ||||
| -rw-r--r-- | loader/minilibc.h | 1 |
2 files changed, 57 insertions, 0 deletions
diff --git a/loader/init.c b/loader/init.c index cf84790d8..08107b42e 100644 --- a/loader/init.c +++ b/loader/init.c @@ -445,6 +445,60 @@ void disableSwap(void) { } } +int handleCleanup(void) { + int i; + char buf[4096]; + int fd; + int loopfd; + char * start, * end; + + fd = open("/tmp/cleanup", O_RDONLY, 0); + if (fd < 0) return 1; + + i = read(fd, buf, sizeof(buf) - 1); + buf[i] = '\0'; + close(fd); + + start = buf; + while (*start) { + end = start; + while (*end && *end != '\n') end++; + if (!*end) return 0; + *end = '\0'; + + if (!strncmp(start, "lounsetup ", 10)) { + start += 10; + while (isspace(*start) && start < end) start++; + if (start == end) return 1; + + loopfd = open(start, O_RDONLY, 0); + if (loopfd < 0) { + printf("\terror opening %s: %d\n", start, errno); + } else { + printf("disabling %s", start); + if (ioctl(loopfd, LOOP_CLR_FD, 0)) + printf(" failed: %d", errno); + printf("\n"); + } + + close(loopfd); + } else if (!strncmp(start, "umount ", 7)) { + start += 7; + while (isspace(*start) && start < end) start++; + if (start == end) return 1; + + printf("unmounting %s...", start); + if (umount(start) < 0) + printf(" failed (%d)", errno); + printf("\n"); + } + + start = end + 1; + } + + return 0; +} + int main(int argc, char **argv) { pid_t installpid, childpid; int waitStatus; @@ -663,6 +717,8 @@ int main(int argc, char **argv) { printf("disabling swap...\n"); disableSwap(); + handleCleanup(); + printf("unmounting filesystems...\n"); unmountFilesystems(); diff --git a/loader/minilibc.h b/loader/minilibc.h index e0bab86e3..21db34cd8 100644 --- a/loader/minilibc.h +++ b/loader/minilibc.h @@ -27,6 +27,7 @@ extern int errno; #include <linux/time.h> #include <linux/if.h> #include <linux/un.h> +#include <linux/loop.h> #include <linux/net.h> #include <asm/posix_types.h> #include <asm/termios.h> |