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     /* Probe for all relevant devices on the system */
 133     deviceGetAll();
 134
 135     /* Prompt for the driver floppy if appropriate. */
 136     if (!pvariable_get("driverFloppyCheck")) {
 137         driverFloppyCheck();
 138         pvariable_set("driverFloppyCheck=1");
 139     }
 140
 141     /* First, see if we have any arguments to process (and argv[0] counts if it's not "sysinstall") */
 142     if (!RunningAsInit) {
 143         int i, start_arg;
 144
 145         if (!strstr(argv[0], "sysinstall"))
 146             start_arg = 0;
 147         else if (Fake || Restarting)
 148             start_arg = 2;
 149         else
 150             start_arg = 1;
 151         for (i = start_arg; i < argc; i++) {
 152             if (DITEM_STATUS(dispatchCommand(argv[i])) != DITEM_SUCCESS)
 153                 systemShutdown(1);
 154         }
 155         if (argc > start_arg)
 156             systemShutdown(0);
 157     }
 158     else
 159         dispatch_load_file_int(TRUE);
 160
 161     status = setjmp(BailOut);
 162     if (status) {
 163         msgConfirm("A signal %d was caught - I'm saving what I can and shutting\n"
 164                    "down.  If you can reproduce the problem, please turn Debug on\n"
 165                    "in the Options menu for the extra information it provides\n"
 166                    "in debugging problems like this.", status);
 167         systemShutdown(status);
 168     }
 169
 170     /* Get user's country and keymap */
 171     if (RunningAsInit)
 172         configCountry(NULL);
 173
 174     /* Add FreeBSD version info to the menu title */
 175     arch = getsysctlbyname("hw.machine_arch");
 176     osrel = getsysctlbyname("kern.osrelease");
 177     ostype = getsysctlbyname("kern.ostype");
 178     snprintf(titlestr, sizeof(titlestr), "%s/%s %s - %s", ostype, arch,
 179              osrel, MenuInitial.title);
 180     free(arch);
 181     free(osrel);
 182     free(ostype);
 183     MenuInitial.title = titlestr;
 184
 185     /* Begin user dialog at outer menu */
 186     dialog_clear();
 187     while (1) {
 188         choice = scroll = curr = max = 0;
 189         dmenuOpen(&MenuInitial, &choice, &scroll, &curr, &max, TRUE);
 190         if (getpid() != 1
 191 #if defined(__sparc64__)
 192             || !msgNoYes("Are you sure you wish to exit?  The system will halt.")
 193 #else
 194             || !msgNoYes("Are you sure you wish to exit?  The system will reboot.")
 195 #endif
 196             )
 197             break;
 198     }
 199
 200     /* Say goodnight, Gracie */
 201     systemShutdown(0);
 202
 203     return 0; /* We should never get here */
 204 }