xnu-517.3.15.tar.gz

[apple/xnu.git] / bsd / kern / kern_shutdown.c

diff --git a/bsd/kern/kern_shutdown.c b/bsd/kern/kern_shutdown.c
index 809b9c6d353f55610aff0c1a8a28a27c98bd4c53..b4d4396092eb4f4e80d56aa3476635a1ea1318b0 100644 (file)
--- a/bsd/kern/kern_shutdown.c
+++ b/bsd/kern/kern_shutdown.c
@@ -59,6 +59,7 @@
 #include <mach/vm_param.h>
 #include <sys/filedesc.h>
 #include <mach/host_reboot.h>
 #include <mach/vm_param.h>
 #include <sys/filedesc.h>
 #include <mach/host_reboot.h>

+#include <sys/kern_audit.h>

 
 int    waittime = -1;
 
 
 int    waittime = -1;
 


@@ -94,6 +95,8 @@ boot(paniced, howto, command)
                /* handle live procs (deallocate their root and current directories). */                
                proc_shutdown();
 
                /* 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 */    
                sync(p, (void *)NULL, (int *)NULL);
 
                /* Release vnodes from the VM object cache */    


@@ -179,10 +182,15 @@ proc_shutdown()
        /*
         * send SIGTERM to those procs interested in catching one
         */
        /*
         * send SIGTERM to those procs interested in catching one
         */

+sigterm_loop:

        for (p = allproc.lh_first; p; p = p->p_list.le_next) {
        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_shutdownstate == 0)) {
+                       if (p->p_sigcatch & sigmask(SIGTERM)) {
+                               p->p_shutdownstate = 1;

                                psignal(p, SIGTERM);
                                psignal(p, SIGTERM);

+
+                               goto sigterm_loop;
+                       }

                }
        }
        /*
                }
        }
        /*


@@ -200,21 +208,34 @@ proc_shutdown()
                TERM_catch = 0;
 
                for (p = allproc.lh_first; p; p = p->p_list.le_next) {
                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++;
-                       }
+                       if (p->p_shutdownstate == 1)
+                               TERM_catch++;

                }
                if (TERM_catch == 0)
                        break;
        }
                }
                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
         */
 
        /*
         * send a SIGKILL to all the procs still hanging around
         */

+sigkill_loop:

        for (p = allproc.lh_first; p; p = p->p_list.le_next) {
        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_flag&P_SYSTEM) == 0) && (p->p_pptr->p_pid != 0) && (p != self) && (p->p_shutdownstate != 2)) {

                        psignal(p, SIGKILL);
                        psignal(p, SIGKILL);

+                       p->p_shutdownstate = 2;
+                       
+                       goto sigkill_loop;
+               }

        }
        /*
         * wait for up to 60 seconds to allow these procs to exit normally
        }
        /*
         * wait for up to 60 seconds to allow these procs to exit normally


@@ -223,7 +244,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) {
                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)
                                break;
                }
                if (!p)


@@ -251,7 +272,7 @@ proc_shutdown()
                                thread_block(THREAD_CONTINUE_NULL);
                        }
                        else {
                                thread_block(THREAD_CONTINUE_NULL);
                        }
                        else {

-                               p->exit_thread = current_thread();
+                               p->exit_thread = current_act();

                                printf(".");
                                exit1(p, 1, (int *)NULL);
                        }
                                printf(".");
                                exit1(p, 1, (int *)NULL);
                        }


@@ -262,18 +283,19 @@ proc_shutdown()
        /*
         *      Forcibly free resources of what's left.
         */
        /*
         *      Forcibly free resources of what's left.
         */

+#ifdef notyet

        p = allproc.lh_first;
        while (p) {
        /*
         * Close open files and release open-file table.
         * This may block!
         */
        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);
        /* panics on reboot due to "zfree: non-allocated memory in collectable zone" message */
        fdfree(p);

-#endif /* notyet */

        p = p->p_list.le_next;
        }
        p = p->p_list.le_next;
        }

+#endif /* notyet */

        /* 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.
        /* 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.