main.c revision 22102
1128355Sru/* 2128355Sru * The new sysinstall program. 3128355Sru * 4128355Sru * This is probably the last attempt in the `sysinstall' line, the next 5128355Sru * generation being slated for what's essentially a complete rewrite. 6128355Sru * 7 * $FreeBSD: head/usr.sbin/sade/main.c 22102 1997-01-29 21:45:30Z jkh $ 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 41static void 42screech(int sig) 43{ 44 printf("\007Fatal signal %d caught! I'm dead..\n", sig); 45 if (RunningAsInit) 46 pause(); 47 else 48 exit(1); 49} 50 51int 52main(int argc, char **argv) 53{ 54 int choice, scroll, curr, max; 55 56 /* Catch fatal signals and complain about them if running as init */ 57 if (getpid() == 1) { 58 signal(SIGBUS, screech); 59 signal(SIGSEGV, screech); 60 } 61 62 /* We don't work too well when running as non-root anymore */ 63 if (geteuid() != 0) { 64 fprintf(stderr, "Error: This utility should only be run as root.\n"); 65 return 1; 66 } 67 68 /* Set up whatever things need setting up */ 69 systemInitialize(argc, argv); 70 71 /* Set default flag and variable values */ 72 installVarDefaults(NULL); 73 74 if (argc > 1 && !strcmp(argv[1], "-fake")) { 75 variable_set2(VAR_DEBUG, "YES"); 76 Fake = TRUE; 77 msgConfirm("I'll be just faking it from here on out, OK?"); 78 } 79 80 /* Try to preserve our scroll-back buffer */ 81 if (OnVTY) { 82 for (curr = 0; curr < 25; curr++) 83 putchar('\n'); 84 } 85 /* Move stderr aside */ 86 if (DebugFD) 87 dup2(DebugFD, 2); 88 89 /* Probe for all relevant devices on the system */ 90 deviceGetAll(); 91 92 /* First, see if we have any arguments to process (and argv[0] counts if it's not "sysinstall") */ 93 if (!RunningAsInit) { 94 int i, start_arg; 95 96 if (!strstr(argv[0], "sysinstall")) 97 start_arg = 0; 98 else if (Fake) 99 start_arg = 2; 100 else 101 start_arg = 1; 102 for (i = start_arg; i < argc; i++) { 103 if (DITEM_STATUS(dispatchCommand(argv[i])) != DITEM_SUCCESS) 104 systemShutdown(1); 105 } 106 if (argc > start_arg) 107 systemShutdown(0); 108 } 109 else { 110 FILE *fp; 111 char buf[BUFSIZ]; 112 113 fp = fopen("install.cfg", "r"); 114 if (fp) { 115 msgNotify("Loading pre-configuration file"); 116 while (fgets(buf, sizeof buf, fp)) { 117 if (DITEM_STATUS(dispatchCommand(buf)) != DITEM_SUCCESS) { 118 msgDebug("Command `%s' failed - rest of script aborted.\n", buf); 119 break; 120 } 121 } 122 fclose(fp); 123 } 124#if defined(LOAD_CONFIG_FILE) 125 else { 126 /* If we have a compiled-in startup config file name on 127 the floppy, look for it and try to load it on startup */ 128 extern char *distWanted; 129 130 /* Tell mediaSetFloppy() to try floppy now */ 131 distWanted = LOAD_CONFIG_FILE; 132 133 /* Try to open the floppy drive if we can do that first */ 134 if (DITEM_STATUS(mediaSetFloppy(NULL)) != DITEM_FAILURE && mediaDevice->init(mediaDevice)) { 135 fp = mediaDevice->get(mediaDevice, LOAD_CONFIG_FILE, TRUE); 136 if (fp) { 137 msgNotify("Loading %s pre-configuration file", LOAD_CONFIG_FILE); 138 while (fgets(buf, sizeof buf, fp)) { 139 if (DITEM_STATUS(dispatchCommand(buf)) != DITEM_SUCCESS) { 140 msgDebug("Command `%s' failed - rest of script aborted.\n", buf); 141 break; 142 } 143 } 144 fclose(fp); 145 } 146 mediaDevice->shutdown(mediaDevice); 147 } 148 } 149#endif 150 } 151 152 /* Begin user dialog at outer menu */ 153 dialog_clear(); 154 while (1) { 155 choice = scroll = curr = max = 0; 156 dmenuOpen(&MenuInitial, &choice, &scroll, &curr, &max, TRUE); 157 if (getpid() != 1 || !msgYesNo("Are you sure you wish to exit? The system will reboot\n" 158 "(be sure to remove any floppies from the drives).")) 159 break; 160 } 161 162 /* Say goodnight, Gracie */ 163 systemShutdown(0); 164 165 return 0; /* We should never get here */ 166} 167