Current Path : /usr/src/usr.sbin/sysinstall/ |
FreeBSD hs32.drive.ne.jp 9.1-RELEASE FreeBSD 9.1-RELEASE #1: Wed Jan 14 12:18:08 JST 2015 root@hs32.drive.ne.jp:/sys/amd64/compile/hs32 amd64 |
Current File : //usr/src/usr.sbin/sysinstall/main.c |
/* * The new sysinstall program. * * This is probably the last attempt in the `sysinstall' line, the next * generation being slated for what's essentially a complete rewrite. * * $FreeBSD: release/9.1.0/usr.sbin/sysinstall/main.c 211417 2010-08-17 09:39:06Z brucec $ * * Copyright (c) 1995 * Jordan Hubbard. All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer, * verbatim and that no modifications are made prior to this * point in the file. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY JORDAN HUBBARD ``AS IS'' AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL JORDAN HUBBARD OR HIS PETS BE LIABLE * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS * OR SERVICES; LOSS OF USE, DATA, LIFE OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. * */ #include "sysinstall.h" #include <sys/signal.h> #include <sys/fcntl.h> #include <sys/time.h> #include <sys/resource.h> const char *StartName; /* Initial contents of argv[0] */ const char *ProgName = "sysinstall"; static void screech(int sig) { msgDebug("\007Signal %d caught! That's bad!\n", sig); longjmp(BailOut, sig); } int main(int argc, char **argv) { int choice, scroll, curr, max, status; char titlestr[80], *arch, *osrel, *ostype; struct rlimit rlim; char *arg; int i; int optionArgs = 0; /* Record name to be able to restart */ StartName = argv[0]; Restarting = FALSE; RunningAsInit = FALSE; Fake = FALSE; for (i = 1; i < argc; i++) { arg = argv[i]; if (arg[0] != '-') break; optionArgs++; if (!strcmp(arg, "-fake")) { variable_set2(VAR_DEBUG, "YES", 0); Fake = TRUE; } else if (!strcmp(arg, "-restart")) { Restarting = TRUE; } else if (!strcmp(arg, "-fakeInit")) { RunningAsInit = TRUE; } arg = argv[optionArgs+1]; } if (getpid() == 1) RunningAsInit = TRUE; /* Catch fatal signals and complain about them if running as init */ if (RunningAsInit) { signal(SIGBUS, screech); signal(SIGSEGV, screech); } signal(SIGPIPE, SIG_IGN); /* We don't work too well when running as non-root anymore */ if (geteuid() != 0) { fprintf(stderr, "Error: This utility should only be run as root.\n"); return 1; } /* * Given what it does sysinstall (and stuff sysinstall runs like * pkg_add) shouldn't be subject to process limits. Better to just * let them have what they think they need than have them blow * their brains out during an install (in sometimes strange and * mysterious ways). */ rlim.rlim_cur = rlim.rlim_max = RLIM_INFINITY; if (setrlimit(RLIMIT_DATA, &rlim) != 0) fprintf(stderr, "Warning: setrlimit() of datasize failed.\n"); if (setrlimit(RLIMIT_STACK, &rlim) != 0) fprintf(stderr, "Warning: setrlimit() of stacksize failed.\n"); #ifdef PC98 { /* XXX */ char *p = getenv("TERM"); if (p && strcmp(p, "cons25") == 0) setenv("TERM", "cons25w", 1); } #endif /* Set up whatever things need setting up */ systemInitialize(argc, argv); /* Set default flag and variable values */ installVarDefaults(NULL); /* only when multi-user is it reasonable to do this here */ if (!RunningAsInit) installEnvironment(); if (Fake) msgConfirm("I'll be just faking it from here on out, OK?"); /* Try to preserve our scroll-back buffer */ if (OnVTY) { for (curr = 0; curr < 25; curr++) putchar('\n'); } /* Move stderr aside */ if (DebugFD) dup2(DebugFD, 2); /* Initialize driver modules, if we haven't already done so (ie, the user hit Ctrl-C -> Restart. */ if (!pvariable_get("modulesInitialize")) { moduleInitialize(); pvariable_set("modulesInitialize=1"); } /* Probe for all relevant devices on the system */ deviceGetAll(); /* Prompt for the driver floppy if appropriate. */ if (!pvariable_get("driverFloppyCheck")) { driverFloppyCheck(); pvariable_set("driverFloppyCheck=1"); } /* First, see if we have any arguments to process (and argv[0] counts if it's not "sysinstall") */ if (!RunningAsInit) { for (i = optionArgs+1; i < argc; i++) { if (DITEM_STATUS(dispatchCommand(argv[i])) != DITEM_SUCCESS) systemShutdown(1); } /* If we were given commands to process on the command line, just exit * now */ if (argc > optionArgs+1) systemShutdown(0); } else dispatch_load_file_int(TRUE); status = setjmp(BailOut); if (status) { msgConfirm("A signal %d was caught - I'm saving what I can and shutting\n" "down. If you can reproduce the problem, please turn Debug on\n" "in the Options menu for the extra information it provides\n" "in debugging problems like this.", status); systemShutdown(status); } /* Get user's country and keymap */ if (RunningAsInit) configCountry(NULL); /* Add FreeBSD version info to the menu title */ arch = getsysctlbyname("hw.machine_arch"); osrel = getsysctlbyname("kern.osrelease"); ostype = getsysctlbyname("kern.ostype"); snprintf(titlestr, sizeof(titlestr), "%s/%s %s - %s", ostype, arch, osrel, MenuInitial.title); free(arch); free(osrel); free(ostype); MenuInitial.title = titlestr; /* Begin user dialog at outer menu */ dialog_clear(); while (1) { choice = scroll = curr = max = 0; dmenuOpen(&MenuInitial, &choice, &scroll, &curr, &max, TRUE); if (!RunningAsInit #if defined(__sparc64__) || !msgNoYes("Are you sure you wish to exit? The system will halt.") #else || !msgNoYes("Are you sure you wish to exit? The system will reboot.") #endif ) break; } /* Say goodnight, Gracie */ systemShutdown(0); return 0; /* We should never get here */ }