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