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(ShutdownType 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;
45 auto shutdown_type = ShutdownType::POWEROFF;
47 for (int i = 1; i < argc; i++) {
48 if (argv[i][0] == '-') {
49 if (strcmp(argv[i], "--help") == 0) {
54 if (strcmp(argv[i], "--system") == 0) {
57 else if (strcmp(argv[i], "-r") == 0) {
58 shutdown_type = ShutdownType::REBOOT;
60 else if (strcmp(argv[i], "-h") == 0) {
61 shutdown_type = ShutdownType::HALT;
63 else if (strcmp(argv[i], "-p") == 0) {
64 shutdown_type = ShutdownType::POWEROFF;
67 cerr << "Unrecognized command-line parameter: " << argv[i] << endl;
72 // time argument? TODO
78 cout << "dinit-shutdown : shutdown the system" << endl;
79 cout << " --help : show this help" << endl;
80 cout << " -r : reboot" << endl;
81 cout << " -h : halt system" << endl;
82 cout << " -p : power down (default)" << endl;
83 cout << " --system : perform shutdown immediately, instead of issuing shutdown" << endl;
84 cout << " command to the init program. Not recommended for use" << endl;
85 cout << " by users." << endl;
90 do_system_shutdown(shutdown_type);
94 int socknum = socket(AF_UNIX, SOCK_STREAM, 0);
100 signal(SIGPIPE, SIG_IGN);
102 const char *naddr = "/dev/dinitctl";
104 struct sockaddr_un name;
105 name.sun_family = AF_UNIX;
106 strcpy(name.sun_path, naddr);
107 int sunlen = offsetof(struct sockaddr_un, sun_path) + strlen(naddr) + 1; // family, (string), nul
109 int connr = connect(socknum, (struct sockaddr *) &name, sunlen);
116 //uint16_t sname_len = strlen(service_name);
118 char * buf = new char[bufsize];
120 buf[0] = DINIT_CP_SHUTDOWN;
121 buf[1] = static_cast<char>(shutdown_type);
123 cout << "Issuing shutdown command..." << endl;
125 // TODO make sure to write the whole buffer
126 int r = write(socknum, buf, bufsize);
133 // r = read(socknum, buf, 1);
135 // cout << "Received acknowledgement. System should now shut down." << endl;
138 CPBuffer<1024> rbuffer;
140 wait_for_reply(rbuffer, socknum);
142 if (rbuffer[0] != DINIT_RP_ACK) {
143 cerr << "shutdown: control socket protocol error" << endl;
147 catch (ReadCPException &exc)
149 cerr << "shutdown: control socket read failure or protocol error" << endl;
160 // Fill a circular buffer from a file descriptor, reading at least _rlength_ bytes.
161 // Throws ReadException if the requested number of bytes cannot be read, with:
162 // errcode = 0 if end of stream (remote end closed)
163 // errcode = errno if another error occurred
164 // Note that EINTR is ignored (i.e. the read will be re-tried).
165 static void fillBufferTo(CPBuffer<1024> *buf, int fd, int rlength)
168 int r = buf->fill_to(fd, rlength);
170 if (errno != EINTR) {
171 throw ReadCPException(errno);
175 throw ReadCPException(0);
184 // Wait for a reply packet, skipping over any information packets
185 // that are received in the meantime.
186 static void wait_for_reply(CPBuffer<1024> &rbuffer, int fd)
188 fillBufferTo(&rbuffer, fd, 1);
190 while (rbuffer[0] >= 100) {
191 // Information packet; discard.
192 fillBufferTo(&rbuffer, fd, 1);
193 int pktlen = (unsigned char) rbuffer[1];
195 rbuffer.consume(1); // Consume one byte so we'll read one byte of the next packet
196 fillBufferTo(&rbuffer, fd, pktlen);
197 rbuffer.consume(pktlen - 1);
201 // Actually shut down the system.
202 void do_system_shutdown(ShutdownType shutdown_type)
207 if (shutdown_type == ShutdownType::REBOOT) reboot_type = RB_AUTOBOOT;
208 else if (shutdown_type == ShutdownType::POWEROFF) reboot_type = RB_POWER_OFF;
209 else reboot_type = RB_HALT_SYSTEM;
211 // Write to console rather than any terminal, since we lose the terminal it seems:
212 close(STDOUT_FILENO);
213 int consfd = open("/dev/console", O_WRONLY);
214 if (consfd != STDOUT_FILENO) {
215 dup2(consfd, STDOUT_FILENO);
218 cout << "Sending TERM/KILL to all processes..." << endl; // DAV
220 // Send TERM/KILL to all (remaining) processes
226 cout << "Turning off swap..." << endl;
228 cout << "Unmounting disks..." << endl;
232 cout << "Issuing shutdown via kernel..." << endl;
236 static void unmount_disks()
238 pid_t chpid = fork();
241 // -a : all filesystems (except proc)
242 // -r : mount readonly if can't unmount
243 execl("/bin/umount", "/bin/umount", "-a", "-r", nullptr);
245 else if (chpid > 0) {
247 waitpid(chpid, &status, 0);
251 static void swap_off()
253 pid_t chpid = fork();
256 execl("/sbin/swapoff", "/sbin/swapoff", "-a", nullptr);
258 else if (chpid > 0) {
260 waitpid(chpid, &status, 0);