]> git.saurik.com Git - apple/xnu.git/blobdiff - bsd/kern/kern_shutdown.c
xnu-792.1.5.tar.gz
[apple/xnu.git] / bsd / kern / kern_shutdown.c
index e9a2a2c524efb05651ceb686cc6413406cb0b121..7c8eb53b7005051ded64b2ca3fca056b5ce7165e 100644 (file)
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2000 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 2000-2004 Apple Computer, Inc. All rights reserved.
  *
  * @APPLE_LICENSE_HEADER_START@
  * 
 #include <sys/systm.h>
 #include <sys/kernel.h>
 #include <sys/vm.h>
-#include <sys/proc.h>
+#include <sys/proc_internal.h>
 #include <sys/user.h>
-#include <sys/buf.h>
 #include <sys/reboot.h>
 #include <sys/conf.h>
-#include <sys/vnode.h>
-#include <sys/file.h>
+#include <sys/vnode_internal.h>
+#include <sys/file_internal.h>
 #include <sys/clist.h>
 #include <sys/callout.h>
 #include <sys/mbuf.h>
@@ -45,7 +44,7 @@
 #include <sys/signal.h>
 #include <sys/tty.h>
 #include <kern/task.h>
-#include <ufs/ufs/quota.h>
+#include <sys/quota.h>
 #include <ufs/ufs/inode.h>
 #if    NCPUS > 1
 #include <kern/processor.h>
 #include <vm/vm_kern.h>
 #include <mach/vm_param.h>
 #include <sys/filedesc.h>
+#include <mach/host_priv.h>
 #include <mach/host_reboot.h>
 
+#include <bsm/audit_kernel.h>
+
 int    waittime = -1;
 
 void
@@ -69,12 +71,14 @@ boot(paniced, howto, command)
        struct proc *p = current_proc();        /* XXX */
        int hostboot_option=0;
        int funnel_state;
+       struct proc  *launchd_proc;
 
-        static void proc_shutdown();
+       static void proc_shutdown();
+    extern void md_prepare_for_shutdown(int paniced, int howto, char * command);
 
        funnel_state = thread_funnel_set(kernel_flock, TRUE);
 
-       /* md_prepare_for_shutdown(paniced, howto, command); */
+       md_prepare_for_shutdown(paniced, howto, command);
 
        if ((howto&RB_NOSYNC)==0 && waittime < 0) {
                int iter, nbusy;
@@ -90,26 +94,32 @@ boot(paniced, howto, command)
                /* handle live procs (deallocate their root and current directories). */                
                proc_shutdown();
 
+               audit_shutdown();
+
                sync(p, (void *)NULL, (int *)NULL);
 
-               /* Release vnodes from the VM object cache */    
-               ubc_unmountall();
+               /*
+                * Now that all processes have been  termianted and system is sync'ed up, 
+                * suspend launchd
+                */
 
-               IOSleep( 1 * 1000 );
+               launchd_proc = pfind(1);
+               if (launchd_proc && p != launchd_proc) {
+                       task_suspend(launchd_proc->task);
+               }
 
                /*
                 * Unmount filesystems
                 */
-               if (panicstr == 0)
-                       vfs_unmountall();
+               vfs_unmountall();
 
                /* Wait for the buffer cache to clean remaining dirty buffers */
-               for (iter = 0; iter < 20; iter++) {
+               for (iter = 0; iter < 100; iter++) {
                        nbusy = count_busy_buffers();
                        if (nbusy == 0)
                                break;
                        printf("%d ", nbusy);
-                       IOSleep( 4 * nbusy );
+                       IOSleep( 1 * nbusy );
                }
                if (nbusy)
                        printf("giving up\n");
@@ -131,6 +141,16 @@ boot(paniced, howto, command)
        if (paniced == RB_PANIC)
                hostboot_option = HOST_REBOOT_HALT;
 
+       /*
+        * if we're going to power down due to a halt,
+        * give the disks a chance to finish getting
+        * the track cache flushed to the media... 
+        * unfortunately, some of our earlier drives
+        * don't properly hold off on returning 
+        * from the track flush command (issued by
+        * the unmounts) until it's actully fully
+        * committed.
+        */
        if (hostboot_option == HOST_REBOOT_HALT)
                IOSleep( 1 * 1000 );
 
@@ -157,37 +177,42 @@ proc_shutdown()
        struct proc     *p, *self;
        struct vnode    **cdirp, **rdirp, *vp;
        int             restart, i, TERM_catch;
+       int delayterm = 0;
 
        /*
         *      Kill as many procs as we can.  (Except ourself...)
         */
-       self = (struct proc *)(get_bsdtask_info(current_task()));
+       self = (struct proc *)current_proc();
        
        /*
-        * Suspend /etc/init
+        * Signal the init with SIGTERM so that he does not launch
+        * new processes 
         */
        p = pfind(1);
-       if (p && p != self)
-               task_suspend(p->task);          /* stop init */
-
-       /*
-        * Suspend mach_init
-        */
-       p = pfind(2);
-       if (p && p != self)
-               task_suspend(p->task);          /* stop mach_init */
+       if (p && p != self) {
+               psignal(p, SIGTERM);
+       }
 
        printf("Killing all processes ");
 
        /*
         * send SIGTERM to those procs interested in catching one
         */
+sigterm_loop:
        for (p = allproc.lh_first; p; p = p->p_list.le_next) {
-               if (((p->p_flag&P_SYSTEM) == 0) && (p->p_pptr->p_pid != 0) && (p != self)) {
-                       if (p->p_sigcatch & sigmask(SIGTERM))
+               if (((p->p_flag&P_SYSTEM) == 0) && (p->p_pptr->p_pid != 0) && (p != self) && (p->p_stat != SZOMB) && (p->p_shutdownstate == 0)) {
+
+                       if ((delayterm == 0) && ((p->p_lflag& P_LDELAYTERM) == P_LDELAYTERM)) {
+                               continue;
+                       }
+                       if (p->p_sigcatch & sigmask(SIGTERM)) {
+                                       p->p_shutdownstate = 1;
                                psignal(p, SIGTERM);
+
+                               goto sigterm_loop;
                }
        }
+       }
        /*
         * now wait for up to 30 seconds to allow those procs catching SIGTERM
         * to digest it
@@ -199,25 +224,44 @@ proc_shutdown()
                 * and then check to see if the tasks that were sent a
                 * SIGTERM have exited
                 */
-               IOSleep(100);   
+               IOSleep(100);   
                TERM_catch = 0;
 
-               for (p = allproc.lh_first; p; p = p->p_list.le_next) {
-                       if (((p->p_flag&P_SYSTEM) == 0) && (p->p_pptr->p_pid != 0) && (p != self)) {
-                               if (p->p_sigcatch & sigmask(SIGTERM))
-                                       TERM_catch++;
+               for (p = allproc.lh_first; p; p = p->p_list.le_next) {
+                       if (p->p_shutdownstate == 1) {
+                               TERM_catch++;
                        }
                }
                if (TERM_catch == 0)
                        break;
        }
+       if (TERM_catch) {
+               /*
+                * log the names of the unresponsive tasks
+                */
+
+               for (p = allproc.lh_first; p; p = p->p_list.le_next) {
+                       if (p->p_shutdownstate == 1) {
+                                 printf("%s[%d]: didn't act on SIGTERM\n", p->p_comm, p->p_pid);
+                       }
+               }
+               IOSleep(1000 * 5);
+       }
 
        /*
         * send a SIGKILL to all the procs still hanging around
         */
+sigkill_loop:
        for (p = allproc.lh_first; p; p = p->p_list.le_next) {
-               if (((p->p_flag&P_SYSTEM) == 0) && (p->p_pptr->p_pid != 0) && (p != self))
-                       psignal(p, SIGKILL);
+               if (((p->p_flag&P_SYSTEM) == 0) && (p->p_pptr->p_pid != 0) && (p != self) && (p->p_stat != SZOMB) && (p->p_shutdownstate != 2)) {
+
+                       if ((delayterm == 0) && ((p->p_lflag& P_LDELAYTERM) == P_LDELAYTERM)) {
+                               continue;
+                       }
+                       psignal(p, SIGKILL);
+                       p->p_shutdownstate = 2;
+                       goto sigkill_loop;
+               }
        }
        /*
         * wait for up to 60 seconds to allow these procs to exit normally
@@ -226,7 +270,7 @@ proc_shutdown()
                IOSleep(200);  /* double the time from 100 to 200 for NFS requests in particular */
 
                for (p = allproc.lh_first; p; p = p->p_list.le_next) {
-                       if (((p->p_flag&P_SYSTEM) == 0) && (p->p_pptr->p_pid != 0) && (p != self))
+                               if (p->p_shutdownstate == 2)
                                break;
                }
                if (!p)
@@ -238,7 +282,8 @@ proc_shutdown()
         */
        p = allproc.lh_first;
        while (p) {
-               if ((p->p_flag&P_SYSTEM) || (p->p_pptr->p_pid == 0) || (p == self)) {
+               if ((p->p_flag&P_SYSTEM) || (!delayterm && ((p->p_lflag& P_LDELAYTERM))) 
+                               || (p->p_pptr->p_pid == 0) || (p == self)) {
                        p = p->p_list.le_next;
                }
                else {
@@ -249,39 +294,25 @@ proc_shutdown()
                         * understand the sig_lock.  This needs to be fixed.
                         * XXX
                         */
-                       if (p->exit_thread) {   /* someone already doing it */
-                               thread_block(0);/* give him a chance */
-                       }
-                       else {
-                               p->exit_thread = current_thread();
+                       if (p->exit_thread) {   /* someone already doing it */
+                               /* give him a chance */
+                               thread_block(THREAD_CONTINUE_NULL);
+                       else {
+                               p->exit_thread = current_thread();
                                printf(".");
-                               exit1(p, 1);
+                               exit1(p, 1, (int *)NULL);
                        }
                        p = allproc.lh_first;
                }
        }
        printf("\n");
-       /*
-        *      Forcibly free resources of what's left.
-        */
-       p = allproc.lh_first;
-       while (p) {
-       /*
-        * Close open files and release open-file table.
-        * This may block!
-        */
-#ifdef notyet
-       /* panics on reboot due to "zfree: non-allocated memory in collectable zone" message */
-       fdfree(p);
-#endif /* notyet */
-       p = p->p_list.le_next;
+
+
+       /* Now start the termination of processes that are marked for delayed termn */
+       if (delayterm == 0) {
+               delayterm = 1;
+               goto  sigterm_loop;
        }
-       /* Wait for the reaper thread to run, and clean up what we have done 
-        * before we proceed with the hardcore shutdown. This reduces the race
-        * between kill_tasks and the reaper thread.
-        */
-       /* thread_wakeup(&reaper_queue); */
-       /*      IOSleep( 1 * 1000);      */
        printf("continuing\n");
 }