X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=init.c;h=bf97fe5d6df2b300a52f0ac35c6d9e34c3411c77;hb=e18c75a563cf202a24effa8a4631dbf46bcc1575;hp=0676bf76d62edbf829322777137079b48bb2e560;hpb=0460ff2e5d443b485c2f6447c8c2ece1b3bec9c3;p=oweals%2Fbusybox.git diff --git a/init.c b/init.c index 0676bf76d..bf97fe5d6 100644 --- a/init.c +++ b/init.c @@ -37,212 +37,215 @@ #include #include #include -#include /* for serial_struct */ -#include /* for vt_stat */ +#include /* for serial_struct */ +#include /* for vt_stat */ #include -#define DEBUG_INIT - -#define CONSOLE "/dev/console" /* Logical system console */ -#define VT_PRIMARY "/dev/tty0" /* Virtual console master */ -#define VT_SECONDARY "/dev/tty1" /* Virtual console master */ -#define VT_LOG "/dev/tty2" /* Virtual console master */ -#define SHELL "/bin/sh" /* Default shell */ -#define INITSCRIPT "/etc/init.d/rcS" /* Initscript. */ +#define VT_CONSOLE "/dev/console" /* Logical system console */ +#define VT_PRIMARY "/dev/tty1" /* Primary virtual console */ +#define VT_SECONDARY "/dev/tty2" /* Virtual console */ +#define VT_LOG "/dev/tty3" /* Virtual console */ +#define SERIAL_CON0 "/dev/ttyS0" /* Primary serial console */ +#define SERIAL_CON1 "/dev/ttyS1" /* Serial console */ +#define SHELL "/bin/sh" /* Default shell */ +#define INITSCRIPT "/etc/init.d/rcS" /* Initscript. */ #define PATH_DEFAULT "PATH=/usr/local/sbin:/sbin:/bin:/usr/sbin:/usr/bin" -static int maxproclen=0; -static char* argv0; - -static char* console = CONSOLE; -static char* second_terminal = "/dev/tty2"; -static char* log = "/dev/tty3"; - +#define LOG 0x1 +#define CONSOLE 0x2 +static char *console = VT_CONSOLE; +static char *second_console = VT_SECONDARY; +static char *log = VT_LOG; +static int kernel_version = 0; /* try to open up the specified device */ -int device_open(char* device, int mode) +int device_open(char *device, int mode) { int m, f, fd = -1; - - mode = m | O_NONBLOCK; - + + m = mode | O_NONBLOCK; + /* Retry up to 5 times */ - for(f = 0; f < 5; f++) - if ((fd = open(device, m)) >= 0) break; - if (fd < 0) return fd; - /* Set original flags. */ + for (f = 0; f < 5; f++) + if ((fd = open(device, m)) >= 0) + break; + if (fd < 0) + return fd; + /* Reset original flags. */ if (m != mode) fcntl(fd, F_SETFL, mode); return fd; } -/* print a message to the specified device */ -void message(char* device, char *fmt, ...) +/* print a message to the specified device: + * device may be bitwise-or'd from LOG | CONSOLE */ +void message(int device, char *fmt, ...) { int fd; + static int log_fd=-1; va_list arguments; - if ((fd = device_open(device, O_WRONLY|O_NOCTTY|O_NDELAY)) >= 0) { + + /* Take full control of the log tty, and never close it. + * It's mine, all mine! Muhahahaha! */ + if (log_fd==-1) { + if ((log_fd = device_open(log, O_RDWR|O_NDELAY)) < 0) { + log_fd=-1; + fprintf(stderr, "Bummer, can't write to log on %s!\r\n", log); + fflush(stderr); + return; + } + } + + if ( (device & LOG) && (log_fd != -1) ) { va_start(arguments, fmt); - vdprintf(fd, fmt, arguments); + vdprintf(log_fd, fmt, arguments); va_end(arguments); } - close( fd); + if (device & CONSOLE) { + if ((fd = device_open(console, O_WRONLY|O_NOCTTY|O_NDELAY)) >= 0) { + va_start(arguments, fmt); + vdprintf(fd, fmt, arguments); + va_end(arguments); + close(fd); + } else { + fprintf(stderr, "Bummer, can't print: "); + va_start(arguments, fmt); + vfprintf(stderr, fmt, arguments); + fflush(stderr); + va_end(arguments); + } + } } + /* Set terminal settings to reasonable defaults */ -void set_term() +void set_term( int fd) { - int fd; struct termios tty; + static const char control_characters[] = { + '\003', '\034', '\177', '\030', '\004', '\0', + '\1', '\0', '\021', '\023', '\032', '\0', '\022', + '\017', '\027', '\026', '\0' + }; - if ((fd = device_open(console, O_RDWR|O_NOCTTY)) < 0) { - message(log, "can't open %s", console); - return; - } - ioctl(fd, TCGETS, &tty); + tcgetattr(fd, &tty); + + /* Make it be sane */ tty.c_cflag &= CBAUD|CBAUDEX|CSIZE|CSTOPB|PARENB|PARODD; tty.c_cflag |= HUPCL|CLOCAL; - tty.c_cc[VINTR] = 3; - tty.c_cc[VQUIT] = 28; - tty.c_cc[VERASE] = 127; - tty.c_cc[VKILL] = 24; - tty.c_cc[VEOF] = 4; - tty.c_cc[VTIME] = 0; - tty.c_cc[VMIN] = 1; - tty.c_cc[VSTART] = 17; - tty.c_cc[VSTOP] = 19; - tty.c_cc[VSUSP] = 26; - - /* Set pre and post processing */ - tty.c_iflag = IGNPAR|ICRNL|IXON|IXANY; + /* input modes */ + tty.c_iflag = IGNPAR|ICRNL|IXON|IXOFF|IXANY; + + /* use line dicipline 0 */ + tty.c_line = 0; + + /* output modes */ tty.c_oflag = OPOST|ONLCR; - tty.c_lflag = ISIG|ICANON|ECHO|ECHOCTL|ECHOPRT|ECHOKE; - /* Now set the terminal line. */ - ioctl(fd, TCSETS, &tty); - close( fd); + /* local modes */ + tty.c_lflag = ISIG|ICANON|ECHO|ECHOE|ECHOK|ECHOCTL|ECHOPRT|ECHOKE|IEXTEN; + + /* control chars */ + memcpy(tty.c_cc, control_characters, sizeof(control_characters)); + + tcsetattr(fd, TCSANOW, &tty); } -static int -mem_total() +/* How much memory does this machine have? */ +static int mem_total() { char s[80]; - char *p; + char *p = "/proc/meminfo"; FILE *f; - const char pattern[]="MemTotal:"; + const char pattern[] = "MemTotal:"; - f=fopen("/proc/meminfo","r"); - while (NULL != fgets(s,79,f)) { - p=strstr(s, pattern); + if ((f = fopen(p, "r")) < 0) { + message(LOG, "Error opening %s: %s\n", p, strerror( errno)); + return -1; + } + while (NULL != fgets(s, 79, f)) { + p = strstr(s, pattern); if (NULL != p) { fclose(f); - return(atoi(p+strlen(pattern))); + return (atoi(p + strlen(pattern))); } } return -1; } -static void -set_free_pages() -{ - char s[80]; - FILE *f; - - f=fopen("/proc/sys/vm/freepages","r"); - fgets(s,79,f); - if (atoi(s) < 32) { - fclose(f); - f=fopen("/proc/sys/vm/freepages","w"); - fprintf(f,"30\t40\t50\n"); - printf("\nIncreased /proc/sys/vm/freepages values to 30/40/50\n"); - } - fclose(f); -} - - -static void -console_init() +static void console_init() { int fd; int tried_devcons = 0; - int tried_vtmaster = 0; + int tried_vtprimary = 0; char *s; - if ((s = getenv("CONSOLE")) != NULL) + if ((s = getenv("CONSOLE")) != NULL) { console = s; - else { - console = CONSOLE; +/* Apparently the sparc does wierd things... */ +#if defined (__sparc__) + if (strncmp( s, "/dev/tty", 8 )==0) { + switch( s[8]) { + case 'a': + s=SERIAL_CON0; + break; + case 'b': + s=SERIAL_CON1; + } + } +#endif + } else { + console = VT_CONSOLE; tried_devcons++; } - while ((fd = open(console, O_RDONLY|O_NONBLOCK)) < 0) { + + while ((fd = open(console, O_RDONLY | O_NONBLOCK)) < 0) { + /* Can't open selected console -- try /dev/console */ if (!tried_devcons) { tried_devcons++; - console = CONSOLE; + console = VT_CONSOLE; continue; } - if (!tried_vtmaster) { - tried_vtmaster++; + /* Can't open selected console -- try vt1 */ + if (!tried_vtprimary) { + tried_vtprimary++; console = VT_PRIMARY; continue; } break; } if (fd < 0) + /* Perhaps we should panic here? */ console = "/dev/null"; else close(fd); + message(LOG, "console=%s\n", console ); } -static int -waitfor(int pid) +static int waitfor(int pid) { int status, wpid; - message(log, "Waiting for process %d.\n", pid); - while ( (wpid = wait(&status)) != pid ) { - if ( wpid > 0 ) - message(log,"pid %d exited, status=%x.\n", wpid, status); + message(LOG, "Waiting for process %d.\n", pid); + while ((wpid = wait(&status)) != pid) { + if (wpid > 0) + message(LOG, "pid %d exited, status=0x%x.\n", wpid, status); } return wpid; } -static int -run(const char* command, char* terminal, int get_enter) + +static pid_t run(const char * const* command, + char *terminal, int get_enter) { - int f, pid; - char *args[16]; - char buf[256]; - char* ptr; - static const char press_enter[] = + int fd; + pid_t pid; + const char * const* cmd = command+1; + static const char press_enter[] = "\nPlease press Enter to activate this console. "; - - /* Make a proper command from the command string */ - strcpy(buf, command); - ptr = buf; - for(f = 1; f < 15; f++) { - /* Skip white space */ - while(*ptr == ' ' || *ptr == '\t') ptr++; - args[f] = ptr; - /* May be trailing space.. */ - if (*ptr == 0) break; - /* Skip this `word' */ - while(*ptr && *ptr != ' ' && *ptr != '\t' && *ptr != '#') - ptr++; - /* If end-of-line, break */ - if (*ptr == '#' || *ptr == 0) { - f++; - *ptr = 0; - break; - } - /* End word with \0 and continue */ - args[f] = NULL; - } - args[0] = args[1]; - - if ((pid = fork()) == 0) { /* Clean up */ close(0); @@ -250,16 +253,22 @@ run(const char* command, char* terminal, int get_enter) close(2); setsid(); - if ((f = device_open(terminal, O_RDWR|O_NOCTTY)) < 0) { - message( log, "open(%s) failed: %s", terminal, strerror(errno)); - return -1; + /* Reset signal handlers set for parent process */ + signal(SIGUSR1, SIG_DFL); + signal(SIGUSR2, SIG_DFL); + signal(SIGINT, SIG_DFL); + signal(SIGTERM, SIG_DFL); + + if ((fd = device_open(terminal, O_RDWR)) < 0) { + message(LOG, "Bummer, can't open %s\r\n", terminal); + exit(-1); } - dup(f); - dup(f); + dup(fd); + dup(fd); tcsetpgrp(0, getpgrp()); - set_term(); + set_term(0); - if ( get_enter ) { + if (get_enter==TRUE) { /* * Save memory by not exec-ing anything large (like a shell) * before the user wants it. This is critical if swap is not @@ -268,219 +277,225 @@ run(const char* command, char* terminal, int get_enter) * be allowed to start a shell or whatever an init script * specifies. */ - char c; + char c; + message(LOG, "Waiting for enter to start '%s' (pid %d, console %s)\r\n", + *cmd, getpid(), terminal ); write(1, press_enter, sizeof(press_enter) - 1); read(0, &c, 1); } /* Log the process name and args */ - message(log, "Executing '%s'\n", command); - - /* Now run it. This should take over the PID, so nothing - * further in init.c should be run by this PID. */ - execvp(args[1], args + 1); - - /* If shell scripts are not executed, force the issue */ - if (errno == ENOEXEC) { - char buf[256]; - args[1] = SHELL; - args[2] = "-c"; - strcpy(buf, "exec "); - strcat(buf, command); - args[3] = buf; - args[4] = NULL; - execvp(args[1], args + 1); - } - message(log, "Could not execute '%s'\n", command, strerror(errno)); + message(LOG, "Starting pid %d, console %s: '", getpid(), terminal); + while ( *cmd) message(LOG, "%s ", *cmd++); + message(LOG, "'\r\n"); + + /* Now run it. The new program will take over this PID, + * so nothing further in init.c should be run. */ + execvp(*command, (char**)command+1); + + message(LOG, "Bummer, could not run '%s'\n", command); exit(-1); } return pid; } -#ifndef DEBUG_INIT -static void -shutdown_system(void) +/* Make sure there is enough memory to do something useful. * + * Calls swapon if needed so be sure /proc is mounted. */ +static void check_memory() { + struct stat statbuf; + const char* const swap_on_cmd[] = + { "/bin/swapon", "swapon", "-a", 0}; - message(console, "The system is going down NOW !!"); - sync(); + if (mem_total() > 3500) + return; + + if (stat("/etc/fstab", &statbuf) == 0) { + /* Try to turn on swap */ + waitfor(run(swap_on_cmd, log, FALSE)); + if (mem_total() < 3500) + goto goodnight; + } else + goto goodnight; + return; + +goodnight: + message(CONSOLE, "Sorry, your computer does not have enough memory.\r\n"); + while (1) sleep(1); +} + +static void shutdown_system(void) +{ + const char* const swap_off_cmd[] = { "swapoff", "swapoff", "-a", 0}; + const char* const umount_cmd[] = { "umount", "umount", "-a", "-n", 0}; + +#ifndef DEBUG_INIT /* Allow Ctrl-Alt-Del to reboot system. */ reboot(RB_ENABLE_CAD); - +#endif + message(CONSOLE, "The system is going down NOW !!\r\n"); + sync(); /* Send signals to every process _except_ pid 1 */ - message(console, "Sending SIGHUP to all processes.\r\n"); + message(CONSOLE, "Sending SIGHUP to all processes.\r\n"); +#ifndef DEBUG_INIT kill(-1, SIGHUP); +#endif sleep(2); sync(); - message(console, "Sending SIGKILL to all processes.\r\n"); + message(CONSOLE, "Sending SIGKILL to all processes.\r\n"); +#ifndef DEBUG_INIT kill(-1, SIGKILL); +#endif sleep(1); - waitfor(run("/bin/swapoff -a", console, 0)); - waitfor(run("/bin/umount -a -n", console, 0)); + waitfor(run( swap_off_cmd, console, FALSE)); + waitfor(run( umount_cmd, console, FALSE)); sync(); - if (get_kernel_revision() <= 2*65536+2*256+11) { - /* Removed bdflush call, kupdate in kernels >2.2.11 */ + if (kernel_version > 0 && kernel_version <= 2 * 65536 + 2 * 256 + 11) { + /* bdflush, kupdate not needed for kernels >2.2.11 */ + message(CONSOLE, "Flushing buffers.\r\n"); bdflush(1, 0); sync(); } } -static void -halt_signal(int sig) +static void halt_signal(int sig) { shutdown_system(); - message(console, "The system is halted. Press CTRL-ALT-DEL or turn off power\r\n"); - reboot( RB_POWER_OFF); + message(CONSOLE, + "The system is halted. Press CTRL-ALT-DEL or turn off power\r\n"); + sync(); +#ifndef DEBUG_INIT + reboot(RB_HALT_SYSTEM); + //reboot(RB_POWER_OFF); +#endif exit(0); } -static void -reboot_signal(int sig) +static void reboot_signal(int sig) { shutdown_system(); - message(console, "Please stand by while rebooting the system.\r\n"); - reboot( RB_AUTOBOOT); - exit(0); -} + message(CONSOLE, "Please stand by while rebooting the system.\r\n"); + sync(); +#ifndef DEBUG_INIT + reboot(RB_AUTOBOOT); #endif - -int setproctitle(char *fmt, ...) -{ - va_list ap; - int len; - char buf[256]; - - buf[0] = 0; - va_start(ap, fmt); - len = vsprintf(buf, fmt, ap); - va_end(ap); - memset(argv0, 0, maxproclen + 1); - strncpy(argv0, buf, maxproclen); - return len; + exit(0); } -extern int -init_main(int argc, char * * argv) +extern int init_main(int argc, char **argv) { - int run_rc = TRUE; - int pid1 = 0; - int pid2 = 0; - struct stat statbuf; - const char* init_commands = SHELL "-c exec " INITSCRIPT; - const char* shell_commands = SHELL; - const char* tty0_commands = init_commands; - const char* tty1_commands = shell_commands; - const char* no_memory = - "Sorry, your computer does not have enough memory.\n"; - - /* For later use */ - argv0 = argv[0]; - maxproclen = strlen(argv[0]); - setproctitle("init [boot]"); + int run_rc = TRUE; + int wait_for_enter = TRUE; + pid_t pid1 = 0; + pid_t pid2 = 0; + struct stat statbuf; + const char* const init_commands[] = { INITSCRIPT, INITSCRIPT, 0}; + const char* const shell_commands[] = { SHELL, "-" SHELL, 0}; + const char* const* tty0_commands = shell_commands; + const char* const* tty1_commands = shell_commands; +#ifdef DEBUG_INIT + char *hello_msg_format = + "init(%d) started: BusyBox v%s (%s) multi-call binary\r\n"; +#else + char *hello_msg_format = + "init started: BusyBox v%s (%s) multi-call binary\r\n"; +#endif + + /* Check if we are supposed to be in single user mode */ + if ( argc > 1 && (!strcmp(argv[1], "single") || + !strcmp(argv[1], "-s") || !strcmp(argv[1], "1"))) { + run_rc = FALSE; + } + + /* Set up sig handlers -- be sure to + * clear all of these in run() */ + signal(SIGUSR1, halt_signal); + signal(SIGUSR2, reboot_signal); + signal(SIGINT, reboot_signal); + signal(SIGTERM, reboot_signal); + + /* Turn off rebooting via CTL-ALT-DEL -- we get a + * SIGINT on CAD so we can shut things down gracefully... */ #ifndef DEBUG_INIT - /* Set up sig handlers */ - signal(SIGUSR1, halt_signal); - signal(SIGSEGV, halt_signal); - signal(SIGPWR, halt_signal); - signal(SIGALRM, halt_signal); - signal(SIGHUP, halt_signal); - signal(SIGUSR2, reboot_signal); - signal(SIGINT, reboot_signal); - signal(SIGTERM, reboot_signal); -#endif - /* Figure out where the default console should be */ - console_init(); + reboot(RB_DISABLE_CAD); +#endif + + /* Figure out where the default console should be */ + console_init(); - /* Turn off rebooting via CTL-ALT-DEL -- we get a - * SIGINT on CAD so we can shut things down gracefully... */ + /* Close whatever files are open, and reset the console. */ + close(0); + close(1); + close(2); + set_term(0); + setsid(); + + /* Make sure PATH is set to something sane */ + putenv(PATH_DEFAULT); + + + /* Hello world */ #ifndef DEBUG_INIT - reboot(RB_DISABLE_CAD); + message(CONSOLE|LOG, hello_msg_format, BB_VER, BB_BT); +#else + message(CONSOLE|LOG, hello_msg_format, getpid(), BB_VER, BB_BT); #endif - /* Close whatever files are open, and reset the console. */ - close(0); - close(1); - close(2); - set_term(); - setsid(); + + /* Mount /proc */ + if (mount ("proc", "/proc", "proc", 0, 0) == 0) { + message(CONSOLE|LOG, "Mounting /proc: done.\n"); + kernel_version = get_kernel_revision(); + } else + message(CONSOLE|LOG, "Mounting /proc: failed!\n"); + + /* Make sure there is enough memory to do something useful. */ + check_memory(); + - /* Make sure PATH is set to something sane */ - if (getenv("PATH") == NULL) - putenv(PATH_DEFAULT); + /* Make sure an init script exists before trying to run it */ + if (run_rc == TRUE && stat(INITSCRIPT, &statbuf)==0) { + wait_for_enter = FALSE; + tty0_commands = init_commands; + } - /* Hello world */ - message(console, "%s started: BusyBox v%s (%s) multi-call binary", - argv[0], BB_VER, BB_BT); - message(log, "%s started: BusyBox v%s (%s) multi-call binary", - argv[0], BB_VER, BB_BT); + /* Ok, now launch the rc script and/or prepare to + * start up some VTs if somebody hits enter... + */ + for (;;) { + pid_t wpid; + int status; - /* Mount /proc */ - message(console, "Mounting /proc: \n"); - if (mount("/proc","/proc","proc",0,0)) { - message(log, "%s: could not mount /proc!\n", argv[0]); - message(console, "failed!\n"); + if (pid1 == 0 && tty0_commands) { + pid1 = run(tty0_commands, console, wait_for_enter); } - message(console, "done.\n"); - - - /* Make sure there is enough memory to do something useful*/ - set_free_pages(); - if (mem_total() < 2000) { - int retval; - retval= stat("/etc/fstab",&statbuf); - if (retval) { - message(console, "%s", no_memory); - while (1) { sleep(1);} - } else { - /* Try to turn on swap */ - waitfor(run("/bin/swapon -a", console, 0)); - if (mem_total() < 2000) { - message(console, "%s", no_memory); - while (1) { sleep(1);} - } - } + if (pid2 == 0 && tty1_commands) { + pid2 = run(tty1_commands, second_console, TRUE); } - - /* Check if we are supposed to be in single user mode */ - if (!strcmp(argv[1], "single") || !strcmp(argv[1], "-s") || !strcmp(argv[1], "1")) { - run_rc = FALSE; - tty0_commands = shell_commands; - tty1_commands = 0; - setproctitle("init [S]"); - } else { - setproctitle("init [1]"); + wpid = wait(&status); + if (wpid > 0 ) { + message(LOG, "pid %d exited, status=%x.\n", wpid, status); } - - /* Make sure an init script exists before trying to run it */ - if ( run_rc == TRUE && stat( INITSCRIPT, &statbuf)) { - tty0_commands = shell_commands; - tty1_commands = shell_commands; - } - - /* Ok, now launch the rc script and/or prepare to - * start up some VTs if somebody hits enter... - */ - for ( ; ; ) { - int wpid; - int status; - - if ( pid1 == 0 && *tty0_commands ) { - pid1 = run(tty0_commands, console, 1); - } - if ( pid2 == 0 && *tty1_commands ) { - pid2 = run(tty1_commands, second_terminal, 1); - } - wpid = wait(&status); - if ( wpid > 0 && wpid != pid1) { - message(log, "pid %d exited, status=%x.\n", wpid, status); - } - if ( wpid == pid2 ) { - pid2 = 0; - } - sleep(1); + /* Don't respawn init script if it exits */ + if (wpid == pid1) { + if (run_rc == FALSE) { + pid1 = 0; + } +#if 0 +/* Turn this on to start a shell on the console if the init script exits.... */ + else { + run_rc=FALSE; + wait_for_enter=TRUE; + tty0_commands=shell_commands; + } +#endif } + if (wpid == pid2) { + pid2 = 0; + } + sleep(1); + } } -