123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371 |
- #define _GNU_SOURCE
- #include <sys/types.h>
- #include <sys/stat.h>
- #include <sys/ioctl.h>
- #include <fcntl.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <unistd.h>
- #include <regex.h>
- #include <ctype.h>
- #include <libubox/utils.h>
- #include <libubox/list.h>
- #include "utils/utils.h"
- #include "procd.h"
- #include "rcS.h"
- #ifndef O_PATH
- #define O_PATH 010000000
- #endif
- #define TAG_ID 0
- #define TAG_RUNLVL 1
- #define TAG_ACTION 2
- #define TAG_PROCESS 3
- #define MAX_ARGS 8
- struct init_action;
- char *console = NULL;
- struct init_handler {
- const char *name;
- void (*cb) (struct init_action *a);
- int multi;
- };
- struct init_action {
- struct list_head list;
- char *id;
- char *argv[MAX_ARGS];
- char *line;
- struct init_handler *handler;
- struct uloop_process proc;
- int respawn;
- struct uloop_timeout tout;
- };
- static const char *tab = "/etc/inittab";
- static char *ask = "/sbin/askfirst";
- static LIST_HEAD(actions);
- static int dev_exist(const char *dev)
- {
- int dfd, fd;
- dfd = open("/dev", O_PATH|O_DIRECTORY);
- if (dfd < 0)
- return 0;
- fd = openat(dfd, dev, O_RDONLY|O_NOCTTY);
- close(dfd);
- if (fd < 0)
- return 0;
- close(fd);
- return 1;
- }
- static void fork_worker(struct init_action *a)
- {
- pid_t p;
- a->proc.pid = fork();
- if (!a->proc.pid) {
- p = setsid();
- if (patch_stdio(a->id))
- ERROR("Failed to setup i/o redirection\n");
- ioctl(STDIN_FILENO, TIOCSCTTY, 1);
- tcsetpgrp(STDIN_FILENO, p);
- execvp(a->argv[0], a->argv);
- ERROR("Failed to execute %s: %m\n", a->argv[0]);
- exit(-1);
- }
- if (a->proc.pid > 0) {
- DEBUG(4, "Launched new %s action, pid=%d\n",
- a->handler->name,
- (int) a->proc.pid);
- uloop_process_add(&a->proc);
- }
- }
- static void child_exit(struct uloop_process *proc, int ret)
- {
- struct init_action *a = container_of(proc, struct init_action, proc);
- DEBUG(4, "pid:%d, exitcode:%d\n", proc->pid, ret);
- proc->pid = 0;
- if (a->respawn < 0)
- return;
- if (!dev_exist(a->id)) {
- DEBUG(4, "Skipping respawn: device '%s' does not exist anymore\n", a->id);
- return;
- }
- uloop_timeout_set(&a->tout, a->respawn);
- }
- static void respawn(struct uloop_timeout *tout)
- {
- struct init_action *a = container_of(tout, struct init_action, tout);
- if (!a->proc.pid)
- fork_worker(a);
- }
- static void rcdone(struct runqueue *q)
- {
- procd_state_next();
- }
- static void runrc(struct init_action *a)
- {
- if (!a->argv[1] || !a->argv[2]) {
- ERROR("valid format is rcS <S|K> <param>\n");
- return;
- }
-
- if (rcS(a->argv[1], a->argv[2], rcdone))
- rcdone(NULL);
- }
- static void askfirst(struct init_action *a)
- {
- int i;
- if (!dev_exist(a->id) || (console && !strcmp(console, a->id))) {
- DEBUG(4, "Skipping %s\n", a->id);
- return;
- }
- a->tout.cb = respawn;
-
- if (a->argv[0] != ask) {
- for (i = MAX_ARGS - 1; i >= 1; i--)
- a->argv[i] = a->argv[i - 1];
- a->argv[0] = ask;
- }
- a->respawn = 500;
- a->proc.cb = child_exit;
- if (!a->proc.pid)
- fork_worker(a);
- }
- static void askconsole(struct init_action *a)
- {
- char line[256], *tty, *split;
- int i;
-
- tty = get_cmdline_val("console", line, sizeof(line));
- if (tty == NULL ||
- get_cmdline_val_offset("console", line, sizeof(line), 1)) {
- if (dev_exist("console"))
- tty = "console";
- else
- tty = get_active_console(line, sizeof(line));
- }
- if (tty != NULL) {
- split = strchr(tty, ',');
- if (split != NULL)
- *split = '\0';
- if (!dev_exist(tty)) {
- DEBUG(4, "skipping %s\n", tty);
- return;
- }
- console = strdup(tty);
- a->id = strdup(tty);
- }
- else {
- console = NULL;
- a->id = NULL;
- }
- a->tout.cb = respawn;
-
- if (a->argv[0] != ask) {
- for (i = MAX_ARGS - 1; i >= 1; i--)
- a->argv[i] = a->argv[i - 1];
- a->argv[0] = ask;
- }
- a->respawn = 500;
- a->proc.cb = child_exit;
- if (!a->proc.pid)
- fork_worker(a);
- }
- static void rcrespawn(struct init_action *a)
- {
- a->tout.cb = respawn;
- a->respawn = 500;
- a->proc.cb = child_exit;
- if (!a->proc.pid)
- fork_worker(a);
- }
- static struct init_handler handlers[] = {
- {
- .name = "sysinit",
- .cb = runrc,
- }, {
- .name = "shutdown",
- .cb = runrc,
- }, {
- .name = "askfirst",
- .cb = askfirst,
- .multi = 1,
- }, {
- .name = "askconsole",
- .cb = askconsole,
- .multi = 1,
- }, {
- .name = "respawn",
- .cb = rcrespawn,
- .multi = 1,
- }, {
- .name = "askconsolelate",
- .cb = askconsole,
- .multi = 1,
- }, {
- .name = "respawnlate",
- .cb = rcrespawn,
- .multi = 1,
- }
- };
- static int add_action(struct init_action *a, const char *name)
- {
- int i;
- for (i = 0; i < ARRAY_SIZE(handlers); i++)
- if (!strcmp(handlers[i].name, name)) {
- a->handler = &handlers[i];
- list_add_tail(&a->list, &actions);
- return 0;
- }
- ERROR("Unknown init handler %s\n", name);
- return -1;
- }
- void procd_inittab_run(const char *handler)
- {
- struct init_action *a;
- list_for_each_entry(a, &actions, list)
- if (!strcmp(a->handler->name, handler)) {
- a->handler->cb(a);
- if (!a->handler->multi)
- break;
- }
- }
- void procd_inittab_kill(void)
- {
- struct init_action *a;
- list_for_each_entry(a, &actions, list) {
- a->respawn = -1;
- if (a->proc.pid)
- kill(a->proc.pid, SIGKILL);
- }
- }
- void procd_inittab(void)
- {
- #define LINE_LEN 128
- FILE *fp = fopen(tab, "r");
- struct init_action *a;
- regex_t pat_inittab;
- regmatch_t matches[5];
- char *line;
- if (!fp) {
- ERROR("Failed to open %s: %m\n", tab);
- return;
- }
- regcomp(&pat_inittab, "([a-zA-Z0-9]*):([a-zA-Z0-9]*):([a-zA-Z0-9]*):(.*)", REG_EXTENDED);
- line = malloc(LINE_LEN);
- a = calloc(1, sizeof(struct init_action));
- while (fgets(line, LINE_LEN, fp)) {
- char *tags[TAG_PROCESS + 1];
- char *tok;
- int i;
- int len = strlen(line);
- while (isspace(line[len - 1]))
- len--;
- line[len] = 0;
- if (*line == '#')
- continue;
- if (regexec(&pat_inittab, line, 5, matches, 0))
- continue;
- DEBUG(4, "Parsing inittab - %s\n", line);
- for (i = TAG_ID; i <= TAG_PROCESS; i++) {
- line[matches[i].rm_eo] = '\0';
- tags[i] = &line[matches[i + 1].rm_so];
- };
- tok = strtok(tags[TAG_PROCESS], " ");
- for (i = 0; i < (MAX_ARGS - 1) && tok; i++) {
- a->argv[i] = tok;
- tok = strtok(NULL, " ");
- }
- a->argv[i] = NULL;
- a->id = tags[TAG_ID];
- a->line = line;
- if (add_action(a, tags[TAG_ACTION]))
- continue;
- line = malloc(LINE_LEN);
- a = calloc(1, sizeof(struct init_action));
- }
- fclose(fp);
- free(line);
- free(a);
- regfree(&pat_inittab);
- }
|