]> git.saurik.com Git - apple/xnu.git/blobdiff - bsd/kern/kern_shutdown.c
xnu-7195.81.3.tar.gz
[apple/xnu.git] / bsd / kern / kern_shutdown.c
index 936fd61e28212a1dc46596999941ee4df3be1b2a..3bd774db5e2dbc530ba01bd8ae1d0cdc8938f1e3 100644 (file)
@@ -2,7 +2,7 @@
  * Copyright (c) 2000-2006 Apple Computer, Inc. All rights reserved.
  *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
- * 
+ *
  * This file contains Original Code and/or Modifications of Original Code
  * as defined in and that are subject to the Apple Public Source License
  * Version 2.0 (the 'License'). You may not use this file except in
  * unlawful or unlicensed copies of an Apple operating system, or to
  * circumvent, violate, or enable the circumvention or violation of, any
  * terms of an Apple operating system software license agreement.
- * 
+ *
  * Please obtain a copy of the License at
  * http://www.opensource.apple.com/apsl/ and read it before using this file.
- * 
+ *
  * The Original Code and all software distributed under the License are
  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
@@ -22,7 +22,7 @@
  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  * Please see the License for the specific language governing rights and
  * limitations under the License.
- * 
+ *
  * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
  */
 /*
@@ -42,8 +42,6 @@
 #include <sys/conf.h>
 #include <sys/vnode_internal.h>
 #include <sys/file_internal.h>
-#include <sys/clist.h>
-#include <sys/callout.h>
 #include <sys/mbuf.h>
 #include <sys/msgbuf.h>
 #include <sys/ioctl.h>
 
 #include <security/audit/audit.h>
 
-#include <kern/sched_prim.h>           /* for thread_block() */
-#include <kern/host.h>                 /* for host_priv_self() */
-#include <net/if_var.h>                        /* for if_down_all() */
-#include <sys/buf_internal.h>          /* for count_busy_buffers() */
-#include <sys/mount_internal.h>                /* for vfs_unmountall() */
-#include <mach/task.h>                 /* for task_suspend() */
-#include <sys/sysproto.h>              /* abused for sync() */
-#include <kern/clock.h>                        /* for delay_for_interval() */
+#include <kern/sched_prim.h>            /* for thread_block() */
+#include <kern/host.h>                  /* for host_priv_self() */
+#include <net/if_var.h>                 /* for if_down_all() */
+#include <sys/buf_internal.h>           /* for count_busy_buffers() */
+#include <sys/mount_internal.h>         /* for vfs_unmountall() */
+#include <mach/task.h>                  /* for task_suspend() */
+#include <sys/sysproto.h>               /* abused for sync() */
+#include <kern/clock.h>                 /* for delay_for_interval() */
 #include <libkern/OSAtomic.h>
 #include <IOKit/IOPlatformExpert.h>
 
 
 uint32_t system_inshutdown = 0;
 
+#if XNU_TARGET_OS_OSX
 /* XXX should be in a header file somewhere, but isn't */
 extern void (*unmountroot_pre_hook)(void);
+#endif
 
 unsigned int proc_shutdown_exitcount = 0;
 
@@ -90,17 +90,17 @@ extern void halt_log_enter(const char * what, const void * pc, uint64_t time);
 extern boolean_t kdp_has_polled_corefile(void);
 #endif /* DEVELOPMENT || DEBUG */
 
-struct sd_filterargs{
+struct sd_filterargs {
        int delayterm;
        int shutdownstate;
 };
 
 
 struct sd_iterargs {
-       int signo;              /* the signal to be posted */
-       int setsdstate;         /* shutdown state to be set */
-       int countproc;          /* count processes on action */
-       int activecount;        /* number of processes on which action was done */
+       int signo;              /* the signal to be posted */
+       int setsdstate;         /* shutdown state to be set */
+       int countproc;          /* count processes on action */
+       int activecount;        /* number of processes on which action was done */
 };
 
 static vnode_t sd_logvp = NULLVP;
@@ -142,9 +142,10 @@ zprint_panic_info(void)
 int
 get_system_inshutdown()
 {
-       return (system_inshutdown);
+       return system_inshutdown;
 }
 
+__abortlike
 static void
 panic_kernel(int howto, char *message)
 {
@@ -154,10 +155,15 @@ panic_kernel(int howto, char *message)
        panic("userspace panic: %s", message);
 }
 
+extern boolean_t compressor_store_stop_compaction;
+extern lck_mtx_t vm_swap_data_lock;
+extern int vm_swapfile_create_thread_running;
+extern int vm_swapfile_gc_thread_running;
+
 int
 reboot_kernel(int howto, char *message)
 {
-       int hostboot_option=0;
+       int hostboot_option = 0;
        uint64_t startTime;
 
        if ((howto & (RB_PANIC | RB_QUICK)) == (RB_PANIC | RB_QUICK)) {
@@ -165,22 +171,41 @@ reboot_kernel(int howto, char *message)
        }
 
        if (!OSCompareAndSwap(0, 1, &system_inshutdown)) {
-               if ( (howto&RB_QUICK) == RB_QUICK)
+               if ((howto & RB_QUICK) == RB_QUICK) {
                        goto force_reboot;
-               return (EBUSY);
+               }
+               return EBUSY;
        }
+
+       lck_mtx_lock(&vm_swap_data_lock);
+
+       /* Turn OFF future swapfile reclaimation / compaction etc.*/
+       compressor_store_stop_compaction = TRUE;
+
+       /* wait for any current swapfile work to end */
+       while (vm_swapfile_create_thread_running || vm_swapfile_gc_thread_running) {
+               assert_wait((event_t)&compressor_store_stop_compaction, THREAD_UNINT);
+
+               lck_mtx_unlock(&vm_swap_data_lock);
+
+               thread_block(THREAD_CONTINUE_NULL);
+
+               lck_mtx_lock(&vm_swap_data_lock);
+       }
+
+       lck_mtx_unlock(&vm_swap_data_lock);
+
        /*
         * Notify the power management root domain that the system will shut down.
         */
        IOSystemShutdownNotification(kIOSystemShutdownNotificationStageProcessExit);
 
-       if ((howto&RB_QUICK)==RB_QUICK) {
+       if ((howto & RB_QUICK) == RB_QUICK) {
                printf("Quick reboot...\n");
-               if ((howto&RB_NOSYNC)==0) {
+               if ((howto & RB_NOSYNC) == 0) {
                        sync((proc_t)NULL, (void *)NULL, (int *)NULL);
                }
-       }
-       else if ((howto&RB_NOSYNC)==0) {
+       } else if ((howto & RB_NOSYNC) == 0) {
                int iter, nbusy;
 
                printf("syncing disks... ");
@@ -201,8 +226,11 @@ reboot_kernel(int howto, char *message)
                halt_log_enter("audit_shutdown", 0, mach_absolute_time() - startTime);
 #endif
 
-               if (unmountroot_pre_hook != NULL)
+#if XNU_TARGET_OS_OSX
+               if (unmountroot_pre_hook != NULL) {
                        unmountroot_pre_hook();
+               }
+#endif
 
                startTime = mach_absolute_time();
                sync((proc_t)NULL, (void *)NULL, (int *)NULL);
@@ -232,15 +260,17 @@ reboot_kernel(int howto, char *message)
                startTime = mach_absolute_time();
                for (iter = 0; iter < 100; iter++) {
                        nbusy = count_busy_buffers();
-                       if (nbusy == 0)
+                       if (nbusy == 0) {
                                break;
+                       }
                        printf("%d ", nbusy);
                        delay_for_interval( 1 * nbusy, 1000 * 1000);
                }
-               if (nbusy)
+               if (nbusy) {
                        printf("giving up\n");
-               else
+               } else {
                        printf("done\n");
+               }
                halt_log_enter("bufferclean", 0, mach_absolute_time() - startTime);
        }
 #if NETWORKING
@@ -260,10 +290,9 @@ force_reboot:
                panic_kernel(howto, message);
        }
 
-       if (howto & RB_POWERDOWN)
-               hostboot_option = HOST_REBOOT_HALT;
-       if (howto & RB_HALT)
+       if (howto & RB_HALT) {
                hostboot_option = HOST_REBOOT_HALT;
+       }
 
        if (howto & RB_UPSDELAY) {
                hostboot_option = HOST_REBOOT_UPSDELAY;
@@ -273,7 +302,7 @@ force_reboot:
        /*
         * should not be reached
         */
-       return (0);
+       return 0;
 }
 
 static int
@@ -281,7 +310,7 @@ sd_openlog(vfs_context_t ctx)
 {
        int error = 0;
        struct timeval tv;
-       
+
        /* Open shutdown log */
        if ((error = vnode_open(PROC_SHUTDOWN_LOG, (O_CREAT | FWRITE | O_NOFOLLOW), 0644, 0, &sd_logvp, ctx))) {
                printf("Failed to open %s: error %d\n", PROC_SHUTDOWN_LOG, error);
@@ -311,7 +340,7 @@ sd_closelog(vfs_context_t ctx)
 }
 
 static void
-sd_log(vfs_context_t ctx, const char *fmt, ...) 
+sd_log(vfs_context_t ctx, const char *fmt, ...)
 {
        int resid, log_error, len;
        char logbuf[100];
@@ -328,13 +357,12 @@ sd_log(vfs_context_t ctx, const char *fmt, ...)
        va_start(arglist, fmt);
        len = vsnprintf(logbuf, sizeof(logbuf), fmt, arglist);
        log_error = vn_rdwr(UIO_WRITE, sd_logvp, (caddr_t)logbuf, len, sd_log_offset,
-                       UIO_SYSSPACE, IO_UNIT | IO_NOAUTH, vfs_context_ucred(ctx), &resid, vfs_context_proc(ctx));
+           UIO_SYSSPACE, IO_UNIT | IO_NOAUTH, vfs_context_ucred(ctx), &resid, vfs_context_proc(ctx));
        if (log_error == EIO || log_error == 0) {
                sd_log_offset += (len - resid);
        }
 
        va_end(arglist);
-
 }
 
 static int
@@ -342,18 +370,18 @@ sd_filt1(proc_t p, void * args)
 {
        proc_t self = current_proc();
        struct sd_filterargs * sf = (struct sd_filterargs *)args;
-       int delayterm = sf-> delayterm;
+       int delayterm = sf->delayterm;
        int shutdownstate = sf->shutdownstate;
 
-       if (((p->p_flag&P_SYSTEM) != 0) || (p->p_ppid == 0) 
-               ||(p == self) || (p->p_stat == SZOMB) 
-               || (p->p_shutdownstate != shutdownstate) 
-               ||((delayterm == 0) && ((p->p_lflag& P_LDELAYTERM) == P_LDELAYTERM))
-               || ((p->p_sigcatch & sigmask(SIGTERM))== 0)) {
-                       return(0);
-               }
-        else 
-                return(1);
+       if (((p->p_flag & P_SYSTEM) != 0) || (p->p_ppid == 0)
+           || (p == self) || (p->p_stat == SZOMB)
+           || (p->p_shutdownstate != shutdownstate)
+           || ((delayterm == 0) && ((p->p_lflag & P_LDELAYTERM) == P_LDELAYTERM))
+           || ((p->p_sigcatch & sigmask(SIGTERM)) == 0)) {
+               return 0;
+       } else {
+               return 1;
+       }
 }
 
 
@@ -366,7 +394,7 @@ sd_callback1(proc_t p, void * args)
        int countproc = sd->countproc;
 
        proc_lock(p);
-       p->p_shutdownstate = setsdstate;
+       p->p_shutdownstate = (char)setsdstate;
        if (p->p_stat != SZOMB) {
                proc_unlock(p);
                if (countproc != 0) {
@@ -377,8 +405,9 @@ sd_callback1(proc_t p, void * args)
                }
 
                psignal(p, signo);
-               if (countproc !=  0)
+               if (countproc != 0) {
                        sd->activecount++;
+               }
        } else {
                proc_unlock(p);
        }
@@ -391,17 +420,17 @@ sd_filt2(proc_t p, void * args)
 {
        proc_t self = current_proc();
        struct sd_filterargs * sf = (struct sd_filterargs *)args;
-       int delayterm = sf-> delayterm;
+       int delayterm = sf->delayterm;
        int shutdownstate = sf->shutdownstate;
 
-       if (((p->p_flag&P_SYSTEM) != 0) || (p->p_ppid == 0) 
-               ||(p == self) || (p->p_stat == SZOMB) 
-               || (p->p_shutdownstate == shutdownstate) 
-               ||((delayterm == 0) && ((p->p_lflag& P_LDELAYTERM) == P_LDELAYTERM))) {
-                       return(0);
-               }
-        else
-                return(1);
+       if (((p->p_flag & P_SYSTEM) != 0) || (p->p_ppid == 0)
+           || (p == self) || (p->p_stat == SZOMB)
+           || (p->p_shutdownstate == shutdownstate)
+           || ((delayterm == 0) && ((p->p_lflag & P_LDELAYTERM) == P_LDELAYTERM))) {
+               return 0;
+       } else {
+               return 1;
+       }
 }
 
 static int
@@ -413,18 +442,19 @@ sd_callback2(proc_t p, void * args)
        int countproc = sd->countproc;
 
        proc_lock(p);
-       p->p_shutdownstate = setsdstate;
+       p->p_shutdownstate = (char)setsdstate;
        if (p->p_stat != SZOMB) {
                proc_unlock(p);
-               if (countproc !=  0) {
+               if (countproc != 0) {
                        proc_list_lock();
                        p->p_listflag |= P_LIST_EXITCOUNT;
                        proc_shutdown_exitcount++;
                        proc_list_unlock();
                }
                psignal(p, signo);
-               if (countproc !=  0)
+               if (countproc != 0) {
                        sd->activecount++;
+               }
        } else {
                proc_unlock(p);
        }
@@ -441,16 +471,16 @@ sd_callback3(proc_t p, void * args)
        int setsdstate = sd->setsdstate;
 
        proc_lock(p);
-       p->p_shutdownstate = setsdstate;
+       p->p_shutdownstate = (char)setsdstate;
        if (p->p_stat != SZOMB) {
-              /*
-               * NOTE: following code ignores sig_lock and plays
-               * with exit_thread correctly.  This is OK unless we
-               * are a multiprocessor, in which case I do not
-               * understand the sig_lock.  This needs to be fixed.
-               * XXX
-               */
-               if (p->exit_thread) {   /* someone already doing it */
+               /*
+                * NOTE: following code ignores sig_lock and plays
+                * with exit_thread correctly.  This is OK unless we
+                * are a multiprocessor, in which case I do not
+                * understand the sig_lock.  This needs to be fixed.
+                * XXX
+                */
+               if (p->exit_thread) {   /* someone already doing it */
                        proc_unlock(p);
                        /* give him a chance */
                        thread_block(THREAD_CONTINUE_NULL);
@@ -462,7 +492,7 @@ sd_callback3(proc_t p, void * args)
 
                        proc_unlock(p);
                        KERNEL_DEBUG_CONSTANT(BSDDBG_CODE(DBG_BSD_PROC, BSD_PROC_FRCEXIT) | DBG_FUNC_NONE,
-                                             p->p_pid, 0, 1, 0, 0);
+                           p->p_pid, 0, 1, 0, 0);
                        sd->activecount++;
                        exit1(p, 1, (int *)NULL);
                }
@@ -529,7 +559,7 @@ sigterm_loop:
        /* post a SIGTERM to all that catch SIGTERM and not marked for delay */
        proc_rebootscan(sd_callback1, (void *)&sdargs, sd_filt1, (void *)&sfargs);
 
-       if (sdargs.activecount != 0 && proc_shutdown_exitcount!= 0) {
+       if (sdargs.activecount != 0 && proc_shutdown_exitcount != 0) {
                proc_list_lock();
                if (proc_shutdown_exitcount != 0) {
                        /*
@@ -542,12 +572,14 @@ sigterm_loop:
                        error = msleep(&proc_shutdown_exitcount, proc_list_mlock, PWAIT, "shutdownwait", &ts);
                        if (error != 0) {
                                for (p = allproc.lh_first; p; p = p->p_list.le_next) {
-                                       if ((p->p_listflag & P_LIST_EXITCOUNT) == P_LIST_EXITCOUNT)
+                                       if ((p->p_listflag & P_LIST_EXITCOUNT) == P_LIST_EXITCOUNT) {
                                                p->p_listflag &= ~P_LIST_EXITCOUNT;
+                                       }
                                }
                                for (p = zombproc.lh_first; p; p = p->p_list.le_next) {
-                                       if ((p->p_listflag & P_LIST_EXITCOUNT) == P_LIST_EXITCOUNT)
+                                       if ((p->p_listflag & P_LIST_EXITCOUNT) == P_LIST_EXITCOUNT) {
                                                p->p_listflag &= ~P_LIST_EXITCOUNT;
+                                       }
                                }
                        }
                }
@@ -585,26 +617,28 @@ sigterm_loop:
 
        error = 0;
 
-       if (sdargs.activecount != 0 && proc_shutdown_exitcount!= 0) {
+       if (sdargs.activecount != 0 && proc_shutdown_exitcount != 0) {
                proc_list_lock();
                if (proc_shutdown_exitcount != 0) {
                        /*
-                       * wait for up to 60 seconds to allow these procs to exit normally
-                       *
-                       * History:      The delay interval was changed from 100 to 200
-                                     for NFS requests in particular.
-                       */
+                        * wait for up to 60 seconds to allow these procs to exit normally
+                        *
+                        * History:     The delay interval was changed from 100 to 200
+                        *              for NFS requests in particular.
+                        */
                        ts.tv_sec = 10;
                        ts.tv_nsec = 0;
                        error = msleep(&proc_shutdown_exitcount, proc_list_mlock, PWAIT, "shutdownwait", &ts);
                        if (error != 0) {
                                for (p = allproc.lh_first; p; p = p->p_list.le_next) {
-                                       if ((p->p_listflag & P_LIST_EXITCOUNT) == P_LIST_EXITCOUNT)
+                                       if ((p->p_listflag & P_LIST_EXITCOUNT) == P_LIST_EXITCOUNT) {
                                                p->p_listflag &= ~P_LIST_EXITCOUNT;
+                                       }
                                }
                                for (p = zombproc.lh_first; p; p = p->p_list.le_next) {
-                                       if ((p->p_listflag & P_LIST_EXITCOUNT) == P_LIST_EXITCOUNT)
+                                       if ((p->p_listflag & P_LIST_EXITCOUNT) == P_LIST_EXITCOUNT) {
                                                p->p_listflag &= ~P_LIST_EXITCOUNT;
+                                       }
                                }
                        }
                }
@@ -661,4 +695,3 @@ sigterm_loop:
        proc_rele(initproc);
        printf("continuing\n");
 }
-