SystemStarter/IPC.c

   1 /**
   2  * IPC.c - System Starter IPC routines
   3  * Wilfredo Sanchez  | wsanchez@opensource.apple.com
   4  * Kevin Van Vechten | kevinvv@uclink4.berkeley.edu
   5  * $Apple$
   6  **
   7  * Copyright (c) 1999-2002 Apple Computer, Inc. All rights reserved.
   8  *
   9  * @APPLE_APACHE_LICENSE_HEADER_START@
  10  *
  11  * Licensed under the Apache License, Version 2.0 (the "License");
  12  * you may not use this file except in compliance with the License.
  13  * You may obtain a copy of the License at
  14  *
  15  *     http://www.apache.org/licenses/LICENSE-2.0
  16  *
  17  * Unless required by applicable law or agreed to in writing, software
  18  * distributed under the License is distributed on an "AS IS" BASIS,
  19  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  20  * See the License for the specific language governing permissions and
  21  * limitations under the License.
  22  *
  23  * @APPLE_APACHE_LICENSE_HEADER_END@
  24  **/
  25
  26 #include <sys/wait.h>
  27 #include <mach/mach.h>
  28 #include <mach/message.h>
  29 #include <mach/mach_error.h>
  30 #include <CoreFoundation/CoreFoundation.h>
  31 #include <syslog.h>
  32
  33 #include "bootstrap.h"
  34
  35 #include "IPC.h"
  36 #include "StartupItems.h"
  37 #include "SystemStarter.h"
  38 #include "SystemStarterIPC.h"
  39
  40 /* Structure to pass StartupContext and anItem to the termination handler. */
  41 typedef struct TerminationContextStorage {
  42         StartupContext  aStartupContext;
  43         CFMutableDictionaryRef anItem;
  44 }              *TerminationContext;
  45
  46 /**
  47  * A CFMachPort invalidation callback that records the termination of
  48  * a startup item task.  Stops the current run loop to give system_starter
  49  * another go at running items.
  50  **/
  51 static void
  52 startupItemTerminated(CFMachPortRef aMachPort, void *anInfo)
  53 {
  54         TerminationContext aTerminationContext = (TerminationContext) anInfo;
  55
  56         if (aMachPort) {
  57                 mach_port_deallocate(mach_task_self(), CFMachPortGetPort(aMachPort));
  58         }
  59         if (aTerminationContext && aTerminationContext->anItem) {
  60                 pid_t           aPID = 0;
  61                 pid_t           rPID = 0;
  62                 int             aStatus = 0;
  63                 CFMutableDictionaryRef anItem = aTerminationContext->anItem;
  64                 StartupContext  aStartupContext = aTerminationContext->aStartupContext;
  65
  66                 /* Get the exit status */
  67                 if (anItem) {
  68                         aPID = StartupItemGetPID(anItem);
  69                         if (aPID > 0)
  70                                 rPID = waitpid(aPID, &aStatus, 0);
  71                 }
  72                 if (aStartupContext) {
  73                         --aStartupContext->aRunningCount;
  74
  75                         /* Record the item's status */
  76                         if (aStartupContext->aStatusDict) {
  77                                 StartupItemExit(aStartupContext->aStatusDict, anItem, (WIFEXITED(aStatus) && WEXITSTATUS(aStatus) == 0));
  78                                 if (aStatus) {
  79                                         CF_syslog(LOG_WARNING, CFSTR("%@ (%d) did not complete successfully"), CFDictionaryGetValue(anItem, CFSTR("Description")), aPID);
  80                                 } else {
  81                                         CF_syslog(LOG_DEBUG, CFSTR("Finished %@ (%d)"), CFDictionaryGetValue(anItem, CFSTR("Description")), aPID);
  82                                 }
  83                         }
  84                         /*
  85                          * If the item failed to start, then add it to the
  86                          * failed list
  87                          */
  88                         if (WEXITSTATUS(aStatus) || WTERMSIG(aStatus) || WCOREDUMP(aStatus)) {
  89                                 CFDictionarySetValue(anItem, kErrorKey, kErrorReturnNonZero);
  90                                 AddItemToFailedList(aStartupContext, anItem);
  91                         }
  92                         /*
  93                          * Remove the item from the waiting list regardless
  94                          * if it was successful or it failed.
  95                          */
  96                         RemoveItemFromWaitingList(aStartupContext, anItem);
  97                 }
  98         }
  99         if (aTerminationContext)
 100                 free(aTerminationContext);
 101 }
 102
 103 void
 104 MonitorStartupItem(StartupContext aStartupContext, CFMutableDictionaryRef anItem)
 105 {
 106         pid_t           aPID = StartupItemGetPID(anItem);
 107         if (anItem && aPID > 0) {
 108                 mach_port_t     aPort;
 109                 kern_return_t   aResult;
 110                 CFMachPortContext aContext;
 111                 CFMachPortRef   aMachPort;
 112                 CFRunLoopSourceRef aSource;
 113                 TerminationContext aTerminationContext = (TerminationContext) malloc(sizeof(struct TerminationContextStorage));
 114
 115                 aTerminationContext->aStartupContext = aStartupContext;
 116                 aTerminationContext->anItem = anItem;
 117
 118                 aContext.version = 0;
 119                 aContext.info = aTerminationContext;
 120                 aContext.retain = 0;
 121                 aContext.release = 0;
 122
 123                 if ((aResult = task_name_for_pid(mach_task_self(), aPID, &aPort)) != KERN_SUCCESS)
 124                         goto out_bad;
 125
 126                 if (!(aMachPort = CFMachPortCreateWithPort(NULL, aPort, NULL, &aContext, NULL)))
 127                         goto out_bad;
 128
 129                 if (!(aSource = CFMachPortCreateRunLoopSource(NULL, aMachPort, 0))) {
 130                         CFRelease(aMachPort);
 131                         goto out_bad;
 132                 }
 133                 CFMachPortSetInvalidationCallBack(aMachPort, startupItemTerminated);
 134                 CFRunLoopAddSource(CFRunLoopGetCurrent(), aSource, kCFRunLoopCommonModes);
 135                 CFRelease(aSource);
 136                 CFRelease(aMachPort);
 137                 return;
 138 out_bad:
 139                 /*
 140                  * The assumption is something failed, the task already
 141                  * terminated.
 142                  */
 143                 startupItemTerminated(NULL, aTerminationContext);
 144         }
 145 }