only add the nandupgrade method when /sbin/upgraded exists
[project/procd.git] / initd / init.c
1 /*
2  * Copyright (C) 2013 Felix Fietkau <nbd@openwrt.org>
3  * Copyright (C) 2013 John Crispin <blogic@openwrt.org>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU Lesser General Public License version 2.1
7  * as published by the Free Software Foundation
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  */
14
15 #include <sys/wait.h>
16 #include <sys/types.h>
17 #include <sys/stat.h>
18 #include <sys/reboot.h>
19
20 #include <libubox/uloop.h>
21 #include <libubus.h>
22
23 #include <stdlib.h>
24 #include <fcntl.h>
25 #include <getopt.h>
26 #include <libgen.h>
27 #include <regex.h>
28 #include <unistd.h>
29 #include <stdio.h>
30
31 #include "init.h"
32 #include "../watchdog.h"
33
34 unsigned int debug = 0;
35
36 static void
37 signal_shutdown(int signal, siginfo_t *siginfo, void *data)
38 {
39         fprintf(stderr, "reboot\n");
40         fflush(stderr);
41         sync();
42         sleep(2);
43         reboot(RB_AUTOBOOT);
44         while (1)
45                 ;
46 }
47
48 static struct sigaction sa_shutdown = {
49         .sa_sigaction = signal_shutdown,
50         .sa_flags = SA_SIGINFO
51 };
52
53 static void
54 cmdline(void)
55 {
56         char line[256];
57         int r, fd = open("/proc/cmdline", O_RDONLY);
58         regex_t pat_cmdline;
59         regmatch_t matches[2];
60
61         if (fd < 0)
62                 return;
63
64         r = read(fd, line, sizeof(line) - 1);
65         line[r] = '\0';
66         close(fd);
67
68         regcomp(&pat_cmdline, "init_debug=([0-9]+)", REG_EXTENDED);
69         if (!regexec(&pat_cmdline, line, 2, matches, 0)) {
70                 line[matches[1].rm_eo] = '\0';
71                 debug = atoi(&line[matches[1].rm_so]);
72         }
73         regfree(&pat_cmdline);
74 }
75
76 int
77 main(int argc, char **argv)
78 {
79         pid_t pid;
80
81         sigaction(SIGTERM, &sa_shutdown, NULL);
82         sigaction(SIGUSR1, &sa_shutdown, NULL);
83         sigaction(SIGUSR2, &sa_shutdown, NULL);
84
85         early();
86         cmdline();
87         watchdog_init(1);
88
89         pid = fork();
90         if (!pid) {
91                 char *kmod[] = { "/sbin/kmodloader", "/etc/modules-boot.d/", NULL };
92
93                 if (debug < 3) {
94                         int fd = open("/dev/null", O_RDWR);
95
96                         if (fd > -1) {
97                                 dup2(fd, STDIN_FILENO);
98                                 dup2(fd, STDOUT_FILENO);
99                                 dup2(fd, STDERR_FILENO);
100                                 if (fd > STDERR_FILENO)
101                                         close(fd);
102                         }
103                 }
104                 execvp(kmod[0], kmod);
105                 ERROR("Failed to start kmodloader\n");
106                 exit(-1);
107         }
108         if (pid <= 0)
109                 ERROR("Failed to start kmodloader instance\n");
110         else
111                 waitpid(pid, NULL, 0);
112         uloop_init();
113         preinit();
114         uloop_run();
115
116         return 0;
117 }