diff options
Diffstat (limited to 'rescue-Flash/flash-rescue-gui.c')
| -rw-r--r-- | rescue-Flash/flash-rescue-gui.c | 274 |
1 files changed, 274 insertions, 0 deletions
diff --git a/rescue-Flash/flash-rescue-gui.c b/rescue-Flash/flash-rescue-gui.c new file mode 100644 index 000000000..f2e17d6a3 --- /dev/null +++ b/rescue-Flash/flash-rescue-gui.c @@ -0,0 +1,274 @@ +/* + * Guillaume Cottenceau (gc@mandrakesoft.com) + * + * Copyright 2001-2007 Mandrakesoft + * + * + * This software may be freely redistributed under the terms of the GNU + * public license. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + * + */ + +#include <stdlib.h> +#include <strings.h> +#define _USE_BSD +#include <sys/types.h> +#include <sys/time.h> +#include <sys/resource.h> +#include <sys/wait.h> +#include <unistd.h> +#include <string.h> +#include <stdio.h> +#include <sys/mount.h> +#include <fcntl.h> +#include <sys/ioctl.h> +#include <linux/unistd.h> +#include <sys/select.h> + +#include "config-stage1.h" +#include "frontend.h" +#include "tools.h" + +#ifndef LINE_MAX +#define LINE_MAX 2048 +#endif + +char * env[] = { + "PATH=/usr/bin:/bin:/sbin:/usr/sbin:/mnt/sbin:/mnt/usr/sbin:/mnt/bin:/mnt/usr/bin", + "LD_LIBRARY_PATH=/lib:/usr/lib:/mnt/lib:/mnt/usr/lib:/usr/X11R6/lib:/mnt/usr/X11R6/lib", + "HOME=/", + "TERM=linux", + "TERMINFO=/etc/terminfo", + NULL +}; + +/* pause() already exists and causes the invoking process to sleep + until a signal is received */ +static void PAUSE(void) +{ + unsigned char t; + fflush(stdout); + read(0, &t, 1); +} + +/* this is duplicated from `init.c', don't edit here */ +static inline _syscall3(int, reboot, int, magic, int, magic2, int, flag); + +#define LOOP_CLR_FD 0x4C01 + +void del_loop(char *device) +{ + int fd; + if ((fd = open(device, O_RDONLY, 0)) < 0) { + printf("del_loop open failed\n"); + return; + } + + if (ioctl(fd, LOOP_CLR_FD, 0) < 0) { + printf("del_loop ioctl failed"); + return; + } + + close(fd); +} + +struct filesystem { + char * dev; + char * name; + char * fs; + int mounted; +}; + +void unmount_filesystems(void) +{ + struct filesystem fs[500]; + int i, nb, fd, size, numfs = 0; + char *buf, *p; + + printf("unmounting filesystems...\n"); + + fd = open("/proc/mounts", O_RDONLY, 0); + if (fd == -1) { + printf("ERROR: failed to open /proc/mounts"); + sleep(2); + return; + } + + buf = (char *) malloc(LINE_MAX); + if (buf == NULL) { + printf("ERROR: not enough memory"); + sleep(2); + return; + } + bzero(buf, LINE_MAX); + size = read(fd, buf, LINE_MAX - 1); + close(fd); + + p = buf; + while (*p) { + + fs[numfs].mounted = 1; + fs[numfs].dev = p; + while (*p != ' ') + p++; + + *p++ = '\0'; + fs[numfs].name = p; + while (*p != ' ') + p++; + + *p++ = '\0'; + fs[numfs].fs = p; + while (*p != ' ') + p++; + + *p++ = '\0'; + while (*p != '\n') + p++; + p++; + if (strcmp(fs[numfs].name, "/") != 0) + /* skip if root, no need to take initrd root + in account */ + numfs++; + } + + /* multiple passes trying to umount everything */ + do { + nb = 0; + for (i = 0; i < numfs; i++) { + if (fs[i].mounted && umount(fs[i].name) == 0) { + if (strncmp(fs[i].dev + sizeof("/dev/") - 1, + "loop", sizeof("loop") - 1) == 0) + del_loop(fs[i].dev); + + printf("\t%s\n", fs[i].name); + fs[i].mounted = 0; + nb++; + } + } + } while (nb); + + for (i = nb = 0; i < numfs; i++) { + if (fs[i].mounted) { + printf("\t%s umount failed\n", fs[i].name); + if (strcmp(fs[i].fs, "ext2") == 0) + /* don't count not-ext2 umount failed */ + nb++; + } + } + + if (nb) { + printf("failed to umount some filesystems\n"); + while (1); + } +} + +void probe_that_type(void) { } + +int main(int argc __attribute__ ((unused)), char **argv __attribute__ ((unused))) +{ + enum return_type results; + + char rootpass[] = "Reset Root Password"; + char userpass[] = "Reset User Password"; + char factory[] = "Reset to Factory Defaults"; + char backup[] = "Backup User Files"; + char restore[] = "Restore User Files from Backup"; + char badblocks[] = "Test Key for Badblocks"; + char reboot_[] = "Reboot"; + + char * actions[] = { rootpass, userpass, factory, backup, restore, + badblocks, reboot_, NULL }; + char * choice; + + init_frontend("Welcome to " DISTRIB_NAME " Rescue (" + DISTRIB_VERSION ") " __DATE__ " " __TIME__); + + do { + int pid; + char * binary = NULL; + + choice = ""; + results = ask_from_list("Please choose the desired action.", + actions, &choice); + + if (ptr_begins_static_str(choice, rootpass)) { + binary = "/usr/bin/reset_rootpass"; + } + if (ptr_begins_static_str(choice, userpass)) { + binary = "/usr/bin/reset_userpass"; + } + if (ptr_begins_static_str(choice, factory)) { + binary = "/usr/bin/clear_systemloop"; + } + if (ptr_begins_static_str(choice, backup)) { + binary = "/usr/bin/backup_systemloop"; + } + if (ptr_begins_static_str(choice, restore)) { + binary = "/usr/bin/restore_systemloop"; + } + if (ptr_begins_static_str(choice, badblocks)) { + binary = "/usr/bin/test_badblocks"; + } + if (ptr_begins_static_str(choice, reboot_)) { + finish_frontend(); + sync(); sync(); + sleep(2); + unmount_filesystems(); + sync(); sync(); + printf("rebooting system\n"); + sleep(2); + reboot(0xfee1dead, 672274793, 0x01234567); + } + + if (binary) { + + int wait_status; + + suspend_to_console(); + pid = fork(); + if (pid == -1) { /* error forking */ + + printf("Can't fork()\n"); + return 33; + + } else if (pid == 0) { /* child */ + + char * child_argv[2]; + child_argv[0] = binary; + child_argv[1] = NULL; + + execve(child_argv[0], child_argv, env); + printf("Can't execute binary (%s)\n<press Enter>\n", binary); + PAUSE(); + + return 33; + } else { /* parent */ + + while (wait4(-1, &wait_status, 0, NULL) != pid) + ; + + printf("<press Enter to return to Rescue GUI>"); + PAUSE(); + resume_from_suspend(); + if (!WIFEXITED(wait_status) || WEXITSTATUS(wait_status) != 0) { + error_message("Program exited abnormally (return code %d).", + WEXITSTATUS(wait_status)); + if (WIFSIGNALED(wait_status)) + error_message("(received signal %d)", + WTERMSIG(wait_status)); + } + } + } + + } while (results == RETURN_OK); + + finish_frontend(); + printf("Bye.\n"); + + return 0; +} |