usr.sbin/sysinstall/main.c

   1 /*
   2  * The new sysinstall program.
   3  *
   4  * This is probably the last attempt in the `sysinstall' line, the next
   5  * generation being slated for what's essentially a complete rewrite.
   6  *
   7  * $FreeBSD$
   8  *
   9  * Copyright (c) 1995
  10  *      Jordan Hubbard.  All rights reserved.
  11  *
  12  * Redistribution and use in source and binary forms, with or without
  13  * modification, are permitted provided that the following conditions
  14  * are met:
  15  * 1. Redistributions of source code must retain the above copyright
  16  *    notice, this list of conditions and the following disclaimer,
  17  *    verbatim and that no modifications are made prior to this
  18  *    point in the file.
  19  * 2. Redistributions in binary form must reproduce the above copyright
  20  *    notice, this list of conditions and the following disclaimer in the
  21  *    documentation and/or other materials provided with the distribution.
  22  *
  23  * THIS SOFTWARE IS PROVIDED BY JORDAN HUBBARD ``AS IS'' AND
  24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  26  * ARE DISCLAIMED.  IN NO EVENT SHALL JORDAN HUBBARD OR HIS PETS BE LIABLE
  27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  29  * OR SERVICES; LOSS OF USE, DATA, LIFE OR PROFITS; OR BUSINESS INTERRUPTION)
  30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  33  * SUCH DAMAGE.
  34  *
  35  */
  36
  37 #include "sysinstall.h"
  38 #include <sys/signal.h>
  39 #include <sys/fcntl.h>
  40 #include <sys/time.h>
  41 #include <sys/resource.h>
  42
  43 const char *StartName;          /* Initial contents of argv[0] */
  44 const char *ProgName = "sysinstall";
  45
  46 static void
  47 screech(int sig)
  48 {
  49     msgDebug("\007Signal %d caught!  That's bad!\n", sig);
  50     longjmp(BailOut, sig);
  51 }
  52
  53 int
  54 main(int argc, char **argv)
  55 {
  56     int choice, scroll, curr, max, status;
  57     char titlestr[80], *arch, *osrel, *ostype;
  58     struct rlimit rlim;
  59
  60     /* Record name to be able to restart */
  61     StartName = argv[0];
  62
  63     /* Catch fatal signals and complain about them if running as init */
  64     if (getpid() == 1) {
  65         signal(SIGBUS, screech);
  66         signal(SIGSEGV, screech);
  67     }
  68     signal(SIGPIPE, SIG_IGN);
  69
  70     /* We don't work too well when running as non-root anymore */
  71     if (geteuid() != 0) {
  72         fprintf(stderr, "Error: This utility should only be run as root.\n");
  73         return 1;
  74     }
  75
  76     /*
  77      * Given what it does sysinstall (and stuff sysinstall runs like
  78      * pkg_add) shouldn't be subject to process limits.  Better to just
  79      * let them have what they think they need than have them blow
  80      * their brains out during an install (in sometimes strange and
  81      * mysterious ways).
  82      */
  83
  84     rlim.rlim_cur = rlim.rlim_max = RLIM_INFINITY;
  85     if (setrlimit(RLIMIT_DATA, &rlim) != 0)
  86         fprintf(stderr, "Warning: setrlimit() of datasize failed.\n");
  87     if (setrlimit(RLIMIT_STACK, &rlim) != 0)
  88         fprintf(stderr, "Warning: setrlimit() of stacksize failed.\n");
  89
  90 #ifdef PC98
  91     {
  92         /* XXX */
  93         char *p = getenv("TERM");
  94         if (p && strcmp(p, "cons25") == 0)
  95             setenv("TERM", "cons25w", 1);
  96     }
  97 #endif
  98
  99     /* Set up whatever things need setting up */
 100     systemInitialize(argc, argv);
 101
 102     /* Set default flag and variable values */
 103     installVarDefaults(NULL);
 104     /* only when multi-user is it reasonable to do this here */
 105     if (!RunningAsInit)
 106         installEnvironment();
 107
 108     if (argc > 1 && !strcmp(argv[1], "-fake")) {
 109         variable_set2(VAR_DEBUG, "YES", 0);
 110         Fake = TRUE;
 111         msgConfirm("I'll be just faking it from here on out, OK?");
 112     }
 113     if (argc > 1 && !strcmp(argv[1], "-restart"))
 114         Restarting = TRUE;
 115
 116     /* Try to preserve our scroll-back buffer */
 117     if (OnVTY) {
 118         for (curr = 0; curr < 25; curr++)
 119             putchar('\n');
 120     }
 121     /* Move stderr aside */
 122     if (DebugFD)
 123         dup2(DebugFD, 2);
 124
 125     /* Initialize driver modules, if we haven't already done so (ie,
 126        the user hit Ctrl-C -> Restart. */
 127     if (!pvariable_get("modulesInitialize")) {
 128         moduleInitialize();
 129         pvariable_set("modulesInitialize=1");
 130     }
 131
 132     /* Initialize PC Card, if we haven't already done so. */
 133 #ifdef PCCARD_ARCH
 134     if (!variable_cmp(VAR_SKIP_PCCARD, "YES") &&
 135       variable_get(VAR_SKIP_PCCARD)!=1 &&
 136        !pvariable_get("pccardInitialize")) {
 137         pccardInitialize();
 138         pvariable_set("pccardInitialize=1");
 139     }
 140 #endif
 141
 142     /* Probe for all relevant devices on the system */
 143     deviceGetAll();
 144
 145     /* Prompt for the driver floppy if appropriate. */
 146     if (!pvariable_get("driverFloppyCheck")) {
 147         driverFloppyCheck();
 148         pvariable_set("driverFloppyCheck=1");
 149     }
 150
 151     /* First, see if we have any arguments to process (and argv[0] counts if it's not "sysinstall") */
 152     if (!RunningAsInit) {
 153         int i, start_arg;
 154
 155         if (!strstr(argv[0], "sysinstall"))
 156             start_arg = 0;
 157         else if (Fake || Restarting)
 158             start_arg = 2;
 159         else
 160             start_arg = 1;
 161         for (i = start_arg; i < argc; i++) {
 162             if (DITEM_STATUS(dispatchCommand(argv[i])) != DITEM_SUCCESS)
 163                 systemShutdown(1);
 164         }
 165         if (argc > start_arg)
 166             systemShutdown(0);
 167     }
 168     else
 169         dispatch_load_file_int(TRUE);
 170
 171     status = setjmp(BailOut);
 172     if (status) {
 173         msgConfirm("A signal %d was caught - I'm saving what I can and shutting\n"
 174                    "down.  If you can reproduce the problem, please turn Debug on\n"
 175                    "in the Options menu for the extra information it provides\n"
 176                    "in debugging problems like this.", status);
 177         systemShutdown(status);
 178     }
 179
 180     /* Get user's country and keymap */
 181     if (RunningAsInit)
 182         configCountry(NULL);
 183
 184     /* Add FreeBSD version info to the menu title */
 185     arch = getsysctlbyname("hw.machine_arch");
 186     osrel = getsysctlbyname("kern.osrelease");
 187     ostype = getsysctlbyname("kern.ostype");
 188     snprintf(titlestr, sizeof(titlestr), "%s/%s %s - %s", ostype, arch,
 189              osrel, MenuInitial.title);
 190     free(arch);
 191     free(osrel);
 192     free(ostype);
 193     MenuInitial.title = titlestr;
 194
 195     /* Begin user dialog at outer menu */
 196     dialog_clear();
 197     while (1) {
 198         choice = scroll = curr = max = 0;
 199         dmenuOpen(&MenuInitial, &choice, &scroll, &curr, &max, TRUE);
 200         if (getpid() != 1
 201 #if defined(__alpha__) || defined(__sparc64__)
 202             || !msgNoYes("Are you sure you wish to exit?  The system will halt.")
 203 #else
 204             || !msgNoYes("Are you sure you wish to exit?  The system will reboot\n"
 205                          "(be sure to remove any floppies/CDs/DVDs from the drives).")
 206 #endif
 207             )
 208             break;
 209     }
 210
 211     /* Say goodnight, Gracie */
 212     systemShutdown(0);
 213
 214     return 0; /* We should never get here */
 215 }