9 #include <sys/reboot.h>
10 #include <sys/types.h>
11 #include <sys/socket.h>
18 #include "control-cmds.h"
19 #include "service-constants.h"
21 // shutdown: shut down the system
22 // This utility communicates with the dinit daemon via a unix socket (/dev/initctl).
24 void do_system_shutdown(shutdown_type_t shutdown_type);
25 static void unmount_disks();
26 static void swap_off();
27 static void wait_for_reply(cpbuffer<1024> &rbuffer, int fd);
34 ReadCPException(int err) : errcode(err) { }
38 int main(int argc, char **argv)
42 bool show_help = false;
43 bool sys_shutdown = false;
44 bool use_passed_cfd = false;
46 auto shutdown_type = shutdown_type_t::POWEROFF;
48 for (int i = 1; i < argc; i++) {
49 if (argv[i][0] == '-') {
50 if (strcmp(argv[i], "--help") == 0) {
55 if (strcmp(argv[i], "--system") == 0) {
58 else if (strcmp(argv[i], "-r") == 0) {
59 shutdown_type = shutdown_type_t::REBOOT;
61 else if (strcmp(argv[i], "-h") == 0) {
62 shutdown_type = shutdown_type_t::HALT;
64 else if (strcmp(argv[i], "-p") == 0) {
65 shutdown_type = shutdown_type_t::POWEROFF;
67 else if (strcmp(argv[i], "--use-passed-cfd") == 0) {
68 use_passed_cfd = true;
71 cerr << "Unrecognized command-line parameter: " << argv[i] << endl;
76 // time argument? TODO
82 cout << "dinit-shutdown : shutdown the system" << endl;
83 cout << " --help : show this help" << endl;
84 cout << " -r : reboot" << endl;
85 cout << " -h : halt system" << endl;
86 cout << " -p : power down (default)" << endl;
87 cout << " --use-passed-cfd : use the socket file descriptor identified by the DINIT_CS_FD" << endl;
88 cout << " environment variable to communicate with the init daemon." << endl;
89 cout << " --system : perform shutdown immediately, instead of issuing shutdown" << endl;
90 cout << " command to the init program. Not recommended for use" << endl;
91 cout << " by users." << endl;
96 do_system_shutdown(shutdown_type);
100 signal(SIGPIPE, SIG_IGN);
104 if (use_passed_cfd) {
105 char * dinit_cs_fd_env = getenv("DINIT_CS_FD");
106 if (dinit_cs_fd_env != nullptr) {
108 long int cfdnum = strtol(dinit_cs_fd_env, &endptr, 10);
109 if (endptr != dinit_cs_fd_env) {
110 socknum = (int) cfdnum;
113 use_passed_cfd = false;
117 use_passed_cfd = false;
121 if (! use_passed_cfd) {
122 socknum = socket(AF_UNIX, SOCK_STREAM, 0);
128 const char *naddr = "/dev/dinitctl";
130 struct sockaddr_un name;
131 name.sun_family = AF_UNIX;
132 strcpy(name.sun_path, naddr);
133 int sunlen = offsetof(struct sockaddr_un, sun_path) + strlen(naddr) + 1; // family, (string), nul
135 int connr = connect(socknum, (struct sockaddr *) &name, sunlen);
143 constexpr int bufsize = 2;
146 buf[0] = DINIT_CP_SHUTDOWN;
147 buf[1] = static_cast<char>(shutdown_type);
149 cout << "Issuing shutdown command..." << endl;
151 // TODO make sure to write the whole buffer
152 int r = write(socknum, buf, bufsize);
159 // r = read(socknum, buf, 1);
161 // cout << "Received acknowledgement. System should now shut down." << endl;
164 cpbuffer<1024> rbuffer;
166 wait_for_reply(rbuffer, socknum);
168 if (rbuffer[0] != DINIT_RP_ACK) {
169 cerr << "shutdown: control socket protocol error" << endl;
173 catch (ReadCPException &exc)
175 cerr << "shutdown: control socket read failure or protocol error" << endl;
186 // Fill a circular buffer from a file descriptor, reading at least _rlength_ bytes.
187 // Throws ReadException if the requested number of bytes cannot be read, with:
188 // errcode = 0 if end of stream (remote end closed)
189 // errcode = errno if another error occurred
190 // Note that EINTR is ignored (i.e. the read will be re-tried).
191 static void fillBufferTo(cpbuffer<1024> *buf, int fd, int rlength)
194 int r = buf->fill_to(fd, rlength);
196 if (errno != EINTR) {
197 throw ReadCPException(errno);
201 throw ReadCPException(0);
210 // Wait for a reply packet, skipping over any information packets
211 // that are received in the meantime.
212 static void wait_for_reply(cpbuffer<1024> &rbuffer, int fd)
214 fillBufferTo(&rbuffer, fd, 1);
216 while (rbuffer[0] >= 100) {
217 // Information packet; discard.
218 fillBufferTo(&rbuffer, fd, 2);
219 int pktlen = (unsigned char) rbuffer[1];
221 rbuffer.consume(1); // Consume one byte so we'll read one byte of the next packet
222 fillBufferTo(&rbuffer, fd, pktlen);
223 rbuffer.consume(pktlen - 1);
227 // Actually shut down the system.
228 void do_system_shutdown(shutdown_type_t shutdown_type)
232 // Mask all signals to prevent death of our parent etc from terminating us
234 sigfillset(&allsigs);
235 sigprocmask(SIG_SETMASK, &allsigs, nullptr);
238 if (shutdown_type == shutdown_type_t::REBOOT) reboot_type = RB_AUTOBOOT;
239 else if (shutdown_type == shutdown_type_t::POWEROFF) reboot_type = RB_POWER_OFF;
240 else reboot_type = RB_HALT_SYSTEM;
242 // Write to console rather than any terminal, since we lose the terminal it seems:
243 close(STDOUT_FILENO);
244 int consfd = open("/dev/console", O_WRONLY);
245 if (consfd != STDOUT_FILENO) {
246 dup2(consfd, STDOUT_FILENO);
249 cout << "Sending TERM/KILL to all processes..." << endl;
251 // Send TERM/KILL to all (remaining) processes
257 cout << "Turning off swap..." << endl;
259 cout << "Unmounting disks..." << endl;
263 cout << "Issuing shutdown via kernel..." << endl;
267 static void unmount_disks()
269 pid_t chpid = fork();
272 // -a : all filesystems (except proc)
273 // -r : mount readonly if can't unmount
274 execl("/bin/umount", "/bin/umount", "-a", "-r", nullptr);
276 else if (chpid > 0) {
278 waitpid(chpid, &status, 0);
282 static void swap_off()
284 pid_t chpid = fork();
287 execl("/sbin/swapoff", "/sbin/swapoff", "-a", nullptr);
289 else if (chpid > 0) {
291 waitpid(chpid, &status, 0);