22a06a14f9be2c61714e89494cd82bde97845281
[project/procd.git] / state.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 <fcntl.h>
16 #include <sys/reboot.h>
17 #include <stdio.h>
18 #include <stdlib.h>
19 #include <unistd.h>
20 #include <sys/types.h>
21 #include <signal.h>
22
23 #include "procd.h"
24 #include "syslog.h"
25 #include "plug/hotplug.h"
26 #include "watchdog.h"
27 #include "service/service.h"
28 #include "utils/utils.h"
29
30 enum {
31 STATE_NONE = 0,
32 STATE_EARLY,
33 STATE_UBUS,
34 STATE_INIT,
35 STATE_RUNNING,
36 STATE_SHUTDOWN,
37 STATE_HALT,
38 __STATE_MAX,
39 };
40
41 static int state = STATE_NONE;
42 static int reboot_event;
43
44 static void set_stdio(const char* tty)
45 {
46 chdir("/dev");
47 freopen(tty, "r", stdin);
48 freopen(tty, "w", stdout);
49 freopen(tty, "w", stderr);
50 chdir("/");
51 fcntl(STDERR_FILENO, F_SETFL, fcntl(STDERR_FILENO, F_GETFL) | O_NONBLOCK);
52 }
53
54 static void set_console(void)
55 {
56 const char* tty;
57 char* split;
58 char line[ 20 ];
59 const char* try[] = { "tty0", "console", NULL }; /* Try the most common outputs */
60 int f, i = 0;
61
62 tty = get_cmdline_val("console",line,sizeof(line));
63 if (tty != NULL) {
64 split = strchr(tty, ',');
65 if ( split != NULL )
66 *split = '\0';
67 } else {
68 // Try a default
69 tty=try[i];
70 i++;
71 }
72
73 chdir("/dev");
74 while (tty!=NULL) {
75 f = open(tty, O_RDONLY);
76 if (f >= 0) {
77 close(f);
78 break;
79 }
80
81 tty=try[i];
82 i++;
83 }
84 chdir("/");
85
86 if (tty != NULL)
87 set_stdio(tty);
88 }
89
90 static void state_enter(void)
91 {
92 char ubus_cmd[] = "/sbin/ubusd";
93
94 switch (state) {
95 case STATE_EARLY:
96 LOG("- early -\n");
97 watchdog_init(0);
98 hotplug("/etc/hotplug.json");
99 procd_coldplug();
100 break;
101
102 case STATE_UBUS:
103 // try to reopen incase the wdt was not available before coldplug
104 watchdog_init(0);
105 set_stdio("console");
106 LOG("- ubus -\n");
107 procd_connect_ubus();
108 service_init();
109 service_start_early("ubus", ubus_cmd);
110 break;
111
112 case STATE_INIT:
113 LOG("- init -\n");
114 procd_inittab();
115 procd_inittab_run("respawn");
116 procd_inittab_run("askconsole");
117 procd_inittab_run("askfirst");
118 procd_inittab_run("sysinit");
119
120 // switch to syslog log channel
121 ulog_open(ULOG_SYSLOG, LOG_DAEMON, "procd");
122 break;
123
124 case STATE_RUNNING:
125 LOG("- init complete -\n");
126 break;
127
128 case STATE_SHUTDOWN:
129 /* Redirect output to the console for the users' benefit */
130 set_console();
131 LOG("- shutdown -\n");
132 procd_inittab_run("shutdown");
133 sync();
134 break;
135
136 case STATE_HALT:
137 // To prevent killed processes from interrupting the sleep
138 signal(SIGCHLD, SIG_IGN);
139 LOG("- SIGTERM processes -\n");
140 kill(-1, SIGTERM);
141 sync();
142 sleep(1);
143 LOG("- SIGKILL processes -\n");
144 kill(-1, SIGKILL);
145 sync();
146 sleep(1);
147 if (reboot_event == RB_POWER_OFF)
148 LOG("- power down -\n");
149 else
150 LOG("- reboot -\n");
151
152 /* Allow time for last message to reach serial console, etc */
153 sleep(1);
154
155 /* We have to fork here, since the kernel calls do_exit(EXIT_SUCCESS)
156 * in linux/kernel/sys.c, which can cause the machine to panic when
157 * the init process exits... */
158 if (!vfork( )) { /* child */
159 reboot(reboot_event);
160 _exit(EXIT_SUCCESS);
161 }
162
163 while (1)
164 sleep(1);
165 break;
166
167 default:
168 ERROR("Unhandled state %d\n", state);
169 return;
170 };
171 }
172
173 void procd_state_next(void)
174 {
175 DEBUG(4, "Change state %d -> %d\n", state, state + 1);
176 state++;
177 state_enter();
178 }
179
180 void procd_state_ubus_connect(void)
181 {
182 if (state == STATE_UBUS)
183 procd_state_next();
184 }
185
186 void procd_shutdown(int event)
187 {
188 if (state >= STATE_SHUTDOWN)
189 return;
190 DEBUG(2, "Shutting down system with event %x\n", event);
191 reboot_event = event;
192 state = STATE_SHUTDOWN;
193 state_enter();
194 }