mirror of
https://github.com/freebsd/freebsd-src.git
synced 2024-11-28 19:52:44 +00:00
546 lines
12 KiB
C
546 lines
12 KiB
C
/*
|
|
* 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 blatantly 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);
|
|
}
|
|
|
|
#if 0
|
|
/*
|
|
* Harvest children if we are init.
|
|
*/
|
|
static void
|
|
reap_children(int sig)
|
|
{
|
|
int errbak = errno;
|
|
|
|
while (waitpid(-1, NULL, WNOHANG) > 0)
|
|
;
|
|
errno = errbak;
|
|
}
|
|
#endif
|
|
|
|
/* 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;
|
|
struct stat sb;
|
|
|
|
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 (stat("/stand/sh", &sb) == 0)
|
|
execl("/stand/sh", "/stand/sh", "-c", cmd, (char *)NULL);
|
|
else
|
|
execl("/bin/sh", "/bin/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();
|
|
}
|
|
}
|
|
}
|
|
}
|