1 /*
2 * Copyright (c) 2007 The DragonFly Project. All rights reserved.
3 *
4 * This code is derived from software contributed to The DragonFly Project
5 * by Matthew Dillon <dillon@backplane.com>
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 *
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 * notice, this list of conditions and the following disclaimer in
15 * the documentation and/or other materials provided with the
16 * distribution.
17 * 3. Neither the name of The DragonFly Project nor the names of its
18 * contributors may be used to endorse or promote products derived
19 * from this software without specific, prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
22 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
23 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
24 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
25 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
26 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
27 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
28 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
29 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
30 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
31 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
33 */
34 /*
35 * Install a signal handler for SIGTERM which shuts down the virtual kernel
36 */
37
38 #include <sys/types.h>
39 #include <sys/systm.h>
40 #include <sys/reboot.h>
41 #include <sys/kernel.h>
42 #include <sys/kthread.h>
43 #include <sys/interrupt.h>
44 #include <sys/bus.h>
45 #include <ddb/ddb.h>
46
47 #include <machine/trap.h>
48 #include <machine/md_var.h>
49 #include <machine/segments.h>
50 #include <machine/cpu.h>
51
52 #include <err.h>
53 #include <signal.h>
54 #include <unistd.h>
55
56 static void shutdownsig(int signo);
57 static void shutdown_intr(void *arg __unused, void *frame __unused);
58
59 static
60 void
initshutdown(void * arg __unused)61 initshutdown(void *arg __unused)
62 {
63 struct sigaction sa;
64
65 bzero(&sa, sizeof(sa));
66 sigemptyset(&sa.sa_mask);
67 sa.sa_flags |= SA_NODEFER;
68 sa.sa_handler = shutdownsig;
69 sigaction(SIGTERM, &sa, NULL);
70
71 register_int_virtual(2, shutdown_intr, NULL, "shutdown", NULL,
72 INTR_MPSAFE);
73 }
74
75 static
76 void
shutdownsig(int signo)77 shutdownsig(int signo)
78 {
79 signalintr(2);
80 }
81
82 SYSINIT(initshutdown, SI_BOOT2_PROC0, SI_ORDER_ANY,
83 initshutdown, NULL);
84
85 /*
86 * DragonFly-safe interrupt thread. We are the only handler on interrupt
87 * #2 so we can just steal the thread's context forever.
88 */
89 static
90 void
shutdown_intr(void * arg __unused,void * frame __unused)91 shutdown_intr(void *arg __unused, void *frame __unused)
92 {
93 kprintf("Caught SIGTERM from host system. Shutting down...\n");
94 if (initproc != NULL) {
95 ksignal(initproc, SIGUSR2);
96 } else {
97 reboot(RB_POWEROFF);
98 }
99 }
100