diff options
Diffstat (limited to 'usr.sbin/sysinstall/system.c')
| -rw-r--r-- | usr.sbin/sysinstall/system.c | 542 |
1 files changed, 542 insertions, 0 deletions
diff --git a/usr.sbin/sysinstall/system.c b/usr.sbin/sysinstall/system.c new file mode 100644 index 0000000000000..65c353a41c9e5 --- /dev/null +++ b/usr.sbin/sysinstall/system.c @@ -0,0 +1,542 @@ +/* + * The new sysinstall program. + * + * This is probably the last program in the `sysinstall' line - the next + * generation being essentially a complete rewrite. + * + * $FreeBSD$ + * + * Jordan Hubbard + * + * My contributions are in the public domain. + * + * Parts of this file are also blatently stolen from Poul-Henning Kamp's + * previous version of sysinstall, and as such fall under his "BEERWARE license" + * so buy him a beer if you like it! Buy him a beer for me, too! + * Heck, get him completely drunk and send me pictures! :-) + */ + +#include "sysinstall.h" +#include <signal.h> +#include <termios.h> +#include <sys/param.h> +#include <sys/reboot.h> +#include <sys/consio.h> +#include <sys/fcntl.h> +#include <sys/ioctl.h> +#include <sys/mount.h> +#include <sys/stat.h> +#include <sys/sysctl.h> +#include <ufs/ufs/ufsmount.h> + + +/* Where we stick our temporary expanded doc file */ +#define DOC_TMP_DIR "/tmp/.doc" +#define DOC_TMP_FILE "/tmp/.doc/doc.tmp" + +static pid_t ehs_pid; + +/* + * Handle interrupt signals - this probably won't work in all cases + * due to our having bogotified the internal state of dialog or curses, + * but we'll give it a try. + */ +static int +intr_continue(dialogMenuItem *self) +{ + return DITEM_LEAVE_MENU; +} + +static int +intr_reboot(dialogMenuItem *self) +{ + systemShutdown(-1); + /* NOTREACHED */ + return 0; +} + +static int +intr_restart(dialogMenuItem *self) +{ + int ret, fd, fdmax; + + mediaClose(); + free_variables(); + fdmax = getdtablesize(); + for (fd = 3; fd < fdmax; fd++) + close(fd); + ret = execl(StartName, StartName, "-restart", (char *)NULL); + msgDebug("execl failed (%s)\n", strerror(errno)); + /* NOTREACHED */ + return -1; +} + +static dialogMenuItem intrmenu[] = { + { "Abort", "Abort the installation", NULL, intr_reboot }, + { "Restart", "Restart the installation program", NULL, intr_restart }, + { "Continue", "Continue the installation", NULL, intr_continue }, +}; + + +static void +handle_intr(int sig) +{ + WINDOW *save = savescr(); + + use_helpline(NULL); + use_helpfile(NULL); + if (OnVTY) { + ioctl(0, VT_ACTIVATE, 1); /* Switch back */ + msgInfo(NULL); + } + (void)dialog_menu("Installation interrupt", + "Do you want to abort the installation?", + -1, -1, 3, -3, intrmenu, NULL, NULL, NULL); + restorescr(save); +} + +/* + * Harvest children if we are init. + */ +static void +reap_children(int sig) +{ + int errbak = errno; + + while (waitpid(-1, NULL, WNOHANG) > 0) + ; + errno = errbak; +} + +/* Expand a file into a convenient location, nuking it each time */ +static char * +expand(char *fname) +{ + char *gunzip = RunningAsInit ? "/stand/gunzip" : "/usr/bin/gunzip"; + + if (!directory_exists(DOC_TMP_DIR)) { + Mkdir(DOC_TMP_DIR); + if (chown(DOC_TMP_DIR, 0, 0) < 0) + return NULL; + if (chmod(DOC_TMP_DIR, S_IRWXU) < 0) + return NULL; + } + else + unlink(DOC_TMP_FILE); + if (!file_readable(fname) || vsystem("%s < %s > %s", gunzip, fname, DOC_TMP_FILE)) + return NULL; + return DOC_TMP_FILE; +} + +/* Initialize system defaults */ +void +systemInitialize(int argc, char **argv) +{ + size_t i; + int boothowto; + sigset_t signalset; + + signal(SIGINT, SIG_IGN); + globalsInit(); + + i = sizeof(boothowto); + if (!sysctlbyname("debug.boothowto", &boothowto, &i, NULL, 0) && + (i == sizeof(boothowto)) && (boothowto & RB_VERBOSE)) + variable_set2(VAR_DEBUG, "YES", 0); + + /* Are we running as init? */ + if (getpid() == 1) { + struct ufs_args ufs_args; + int fd; + + RunningAsInit = 1; + setsid(); + close(0); + fd = open("/dev/ttyv0", O_RDWR); + if (fd == -1) { + fd = open("/dev/console", O_RDWR); /* fallback */ + variable_set2(VAR_FIXIT_TTY, "serial", 0); /* give fixit a hint */ + } else + OnVTY = TRUE; + /* + * To make _sure_ we're on a VTY and don't have /dev/console switched + * away to a serial port or something, attempt to set the cursor appearance. + */ + if (OnVTY) { + int fd2, type; + + type = 0; /* normal */ + if ((fd2 = open("/dev/console", O_RDWR)) != -1) { + if (ioctl(fd2, CONS_CURSORTYPE, &type) == -1) { + OnVTY = FALSE; + variable_set2(VAR_FIXIT_TTY, "serial", 0); /* Tell Fixit + the console + type */ + close(fd); close(fd2); + open("/dev/console", O_RDWR); + } + else + close(fd2); + } + } + close(1); dup(0); + close(2); dup(0); + printf("%s running as init on %s\n", argv[0], OnVTY ? "vty0" : "serial console"); + ioctl(0, TIOCSCTTY, (char *)NULL); + setlogin("root"); + setenv("PATH", "/stand:/bin:/sbin:/usr/sbin:/usr/bin:/mnt/bin:/mnt/sbin:/mnt/usr/sbin:/mnt/usr/bin:/usr/X11R6/bin", 1); + setbuf(stdin, 0); + setbuf(stderr, 0); +#ifdef __alpha__ + i = 0; + sysctlbyname("machdep.unaligned_print", NULL, 0, &i, sizeof(i)); +#endif +#if 0 + signal(SIGCHLD, reap_children); +#endif + memset(&ufs_args, 0, sizeof(ufs_args)); + mount("ufs", "/", MNT_UPDATE, &ufs_args); + } + else { + char hname[256]; + + /* Initalize various things for a multi-user environment */ + if (!gethostname(hname, sizeof hname)) + variable_set2(VAR_HOSTNAME, hname, 0); + } + + if (set_termcap() == -1) { + printf("Can't find terminal entry\n"); + exit(-1); + } + + /* XXX - libdialog has particularly bad return value checking */ + init_dialog(); + + /* If we haven't crashed I guess dialog is running ! */ + DialogActive = TRUE; + + /* Make sure HOME is set for those utilities that need it */ + if (!getenv("HOME")) + setenv("HOME", "/", 1); + signal(SIGINT, handle_intr); + /* + * Make sure we can be interrupted even if we were re-executed + * from an interrupt. + */ + sigemptyset(&signalset); + sigaddset(&signalset, SIGINT); + sigprocmask(SIG_UNBLOCK, &signalset, NULL); + + (void)vsystem("rm -rf %s", DOC_TMP_DIR); +} + +/* Close down and prepare to exit */ +void +systemShutdown(int status) +{ + /* If some media is open, close it down */ + if (status >=0) + mediaClose(); + + /* write out any changes to rc.conf .. */ + configRC_conf(); + + /* Shut down the dialog library */ + if (DialogActive) { + end_dialog(); + DialogActive = FALSE; + } + + /* Shut down curses */ + endwin(); + + /* If we have a temporary doc dir lying around, nuke it */ + (void)vsystem("rm -rf %s", DOC_TMP_DIR); + + /* REALLY exit! */ + if (RunningAsInit) { + /* Put the console back */ + ioctl(0, VT_ACTIVATE, 2); +#if defined(__alpha__) || defined(__sparc64__) + reboot(RB_HALT); +#else + reboot(0); +#endif + } + else + exit(status); +} + +/* Run some general command */ +int +systemExecute(char *command) +{ + int status; + struct termios foo; + WINDOW *w = savescr(); + + dialog_clear(); + dialog_update(); + end_dialog(); + DialogActive = FALSE; + if (tcgetattr(0, &foo) != -1) { + foo.c_cc[VERASE] = '\010'; + tcsetattr(0, TCSANOW, &foo); + } + if (!Fake) + status = system(command); + else { + status = 0; + msgDebug("systemExecute: Faked execution of `%s'\n", command); + } + DialogActive = TRUE; + restorescr(w); + return status; +} + +/* suspend/resume libdialog/curses screen */ +static WINDOW *oldW; + +void +systemSuspendDialog(void) +{ + + oldW = savescr(); + dialog_clear(); + dialog_update(); + end_dialog(); + DialogActive = FALSE; +} + +void +systemResumeDialog(void) +{ + + DialogActive = TRUE; + restorescr(oldW); +} + +/* Display a help file in a filebox */ +int +systemDisplayHelp(char *file) +{ + char *fname = NULL; + char buf[FILENAME_MAX]; + int ret = 0; + WINDOW *w = savescr(); + + fname = systemHelpFile(file, buf); + if (!fname) { + snprintf(buf, FILENAME_MAX, "The %s file is not provided on this particular floppy image.", file); + use_helpfile(NULL); + use_helpline(NULL); + dialog_mesgbox("Sorry!", buf, -1, -1); + ret = 1; + } + else { + use_helpfile(NULL); + use_helpline(NULL); + dialog_textbox(file, fname, LINES, COLS); + } + restorescr(w); + return ret; +} + +char * +systemHelpFile(char *file, char *buf) +{ + if (!file) + return NULL; + if (file[0] == '/') + return file; + snprintf(buf, FILENAME_MAX, "/stand/help/%s.hlp.gz", file); + if (file_readable(buf)) + return expand(buf); + snprintf(buf, FILENAME_MAX, "/stand/help/%s.hlp", file); + if (file_readable(buf)) + return expand(buf); + snprintf(buf, FILENAME_MAX, "/stand/help/%s.TXT.gz", file); + if (file_readable(buf)) + return expand(buf); + snprintf(buf, FILENAME_MAX, "/stand/help/%s.TXT", file); + if (file_readable(buf)) + return expand(buf); + snprintf(buf, FILENAME_MAX, "/usr/src/usr.sbin/sysinstall/help/%s.hlp", file); + if (file_readable(buf)) + return buf; + snprintf(buf, FILENAME_MAX, "/usr/src/usr.sbin/sysinstall/help/%s.TXT", file); + if (file_readable(buf)) + return buf; + return NULL; +} + +void +systemChangeTerminal(char *color, const u_char c_term[], + char *mono, const u_char m_term[]) +{ + if (OnVTY) { + int setupterm(char *color, int, int *); + + if (ColorDisplay) { + setenv("TERM", color, 1); + setenv("TERMCAP", c_term, 1); + reset_shell_mode(); + setterm(color); + cbreak(); noecho(); + } + else { + setenv("TERM", mono, 1); + setenv("TERMCAP", m_term, 1); + reset_shell_mode(); + setterm(mono); + cbreak(); noecho(); + } + } + clear(); + refresh(); + dialog_clear(); +} + +int +vsystem(char *fmt, ...) +{ + va_list args; + int pstat; + pid_t pid; + int omask; + sig_t intsave, quitsave; + char *cmd; + int i; + + cmd = (char *)alloca(FILENAME_MAX); + cmd[0] = '\0'; + va_start(args, fmt); + vsnprintf(cmd, FILENAME_MAX, fmt, args); + va_end(args); + + omask = sigblock(sigmask(SIGCHLD)); + if (Fake) { + msgDebug("vsystem: Faked execution of `%s'\n", cmd); + return 0; + } + if (isDebug()) + msgDebug("Executing command `%s'\n", cmd); + pid = fork(); + if (pid == -1) { + (void)sigsetmask(omask); + i = 127; + } + else if (!pid) { /* Junior */ + (void)sigsetmask(omask); + if (DebugFD != -1) { + dup2(DebugFD, 0); + dup2(DebugFD, 1); + dup2(DebugFD, 2); + } + else { + close(1); open("/dev/null", O_WRONLY); + dup2(1, 2); + } + if (!RunningAsInit) + execl("/bin/sh", "/bin/sh", "-c", cmd, (char *)NULL); + else + execl("/stand/sh", "/stand/sh", "-c", cmd, (char *)NULL); + exit(1); + } + else { + intsave = signal(SIGINT, SIG_IGN); + quitsave = signal(SIGQUIT, SIG_IGN); + pid = waitpid(pid, &pstat, 0); + (void)sigsetmask(omask); + (void)signal(SIGINT, intsave); + (void)signal(SIGQUIT, quitsave); + i = (pid == -1) ? -1 : WEXITSTATUS(pstat); + if (isDebug()) + msgDebug("Command `%s' returns status of %d\n", cmd, i); + } + return i; +} + +void +systemCreateHoloshell(void) +{ + int waitstatus; + + if ((FixItMode || OnVTY) && RunningAsInit) { + + if (ehs_pid != 0) { + int pstat; + + if (kill(ehs_pid, 0) == 0) { + + if (msgNoYes("There seems to be an emergency holographic shell\n" + "already running on VTY 4.\n\n" + "Kill it and start a new one?")) + return; + + /* try cleaning up as much as possible */ + (void) kill(ehs_pid, SIGHUP); + sleep(1); + (void) kill(ehs_pid, SIGKILL); + } + + /* avoid too many zombies */ + (void) waitpid(ehs_pid, &pstat, WNOHANG); + } + + if (strcmp(variable_get(VAR_FIXIT_TTY), "serial") == 0) + systemSuspendDialog(); /* must be before the fork() */ + if ((ehs_pid = fork()) == 0) { + int i, fd; + struct termios foo; + extern int login_tty(int); + + ioctl(0, TIOCNOTTY, NULL); + for (i = getdtablesize(); i >= 0; --i) + close(i); + if (strcmp(variable_get(VAR_FIXIT_TTY), "serial") == 0) + fd = open("/dev/console", O_RDWR); + else + fd = open("/dev/ttyv3", O_RDWR); + ioctl(0, TIOCSCTTY, &fd); + dup2(0, 1); + dup2(0, 2); + DebugFD = 2; + if (login_tty(fd) == -1) + msgDebug("Doctor: I can't set the controlling terminal.\n"); + signal(SIGTTOU, SIG_IGN); + if (tcgetattr(fd, &foo) != -1) { + foo.c_cc[VERASE] = '\010'; + if (tcsetattr(fd, TCSANOW, &foo) == -1) + msgDebug("Doctor: I'm unable to set the erase character.\n"); + } + else + msgDebug("Doctor: I'm unable to get the terminal attributes!\n"); + if (strcmp(variable_get(VAR_FIXIT_TTY), "serial") == 0) { + printf("Type ``exit'' in this fixit shell to resume sysinstall.\n\n"); + fflush(stdout); + } + execlp("sh", "-sh", 0); + msgDebug("Was unable to execute sh for Holographic shell!\n"); + exit(1); + } + else { + if (strcmp(variable_get(VAR_FIXIT_TTY), "standard") == 0) { + WINDOW *w = savescr(); + + msgNotify("Starting an emergency holographic shell on VTY4"); + sleep(2); + restorescr(w); + } + else { + (void)waitpid(ehs_pid, &waitstatus, 0); /* we only wait for + shell to finish + it serial mode + since there is no + virtual console */ + systemResumeDialog(); + } + } + } +} |