• source navigation  • diff markup  • identifier search  • freetext search  • 

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

This page was automatically generated by LXR 0.3.1.  •  OpenWrt