SystemConfiguration.fproj/SCDNotifierInformViaMachPort.c

   1 /*
   2  * Copyright (c) 2000, 2001, 2004, 2005, 2009, 2010 Apple Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. Please obtain a copy of the License at
  10  * http://www.opensource.apple.com/apsl/ and read it before using this
  11  * file.
  12  *
  13  * The Original Code and all software distributed under the License are
  14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  18  * Please see the License for the specific language governing rights and
  19  * limitations under the License.
  20  *
  21  * @APPLE_LICENSE_HEADER_END@
  22  */
  23
  24 /*
  25  * Modification History
  26  *
  27  * June 1, 2001                 Allan Nathanson <ajn@apple.com>
  28  * - public API conversion
  29  *
  30  * March 31, 2000               Allan Nathanson <ajn@apple.com>
  31  * - initial revision
  32  */
  33
  34 #include <mach/mach.h>
  35 #include <mach/mach_error.h>
  36
  37 #include <SystemConfiguration/SystemConfiguration.h>
  38 #include <SystemConfiguration/SCPrivate.h>
  39 #include "SCDynamicStoreInternal.h"
  40 #include "config.h"             /* MiG generated file */
  41
  42 Boolean
  43 SCDynamicStoreNotifyMachPort(SCDynamicStoreRef store, mach_msg_id_t identifier, mach_port_t *port)
  44 {
  45         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
  46         kern_return_t                   status;
  47         mach_port_t                     oldNotify;
  48         int                             sc_status;
  49
  50         if (store == NULL) {
  51                 /* sorry, you must provide a session */
  52                 _SCErrorSet(kSCStatusNoStoreSession);
  53                 return FALSE;
  54         }
  55
  56         if (storePrivate->server == MACH_PORT_NULL) {
  57                 /* sorry, you must have an open session to play */
  58                 _SCErrorSet(kSCStatusNoStoreServer);
  59                 return FALSE;
  60         }
  61
  62         if (storePrivate->notifyStatus != NotifierNotRegistered) {
  63                 /* sorry, you can only have one notification registered at once */
  64                 _SCErrorSet(kSCStatusNotifierActive);
  65                 return FALSE;
  66         }
  67
  68         /* Allocating port (for server response) */
  69         status = mach_port_allocate(mach_task_self(), MACH_PORT_RIGHT_RECEIVE, port);
  70         if (status != KERN_SUCCESS) {
  71                 SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreNotifyMachPort mach_port_allocate(): %s"), mach_error_string(status));
  72                 _SCErrorSet(status);
  73                 return FALSE;
  74         }
  75
  76         status = mach_port_insert_right(mach_task_self(),
  77                                         *port,
  78                                         *port,
  79                                         MACH_MSG_TYPE_MAKE_SEND);
  80         if (status != KERN_SUCCESS) {
  81                 /*
  82                  * We can't insert a send right into our own port!  This should
  83                  * only happen if someone stomped on OUR port (so let's leave
  84                  * the port alone).
  85                  */
  86                 SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreNotifyMachPort mach_port_insert_right(): %s"), mach_error_string(status));
  87                 *port = MACH_PORT_NULL;
  88                 _SCErrorSet(status);
  89                 return FALSE;
  90         }
  91
  92         /* Request a notification when/if the server dies */
  93         status = mach_port_request_notification(mach_task_self(),
  94                                                 *port,
  95                                                 MACH_NOTIFY_NO_SENDERS,
  96                                                 1,
  97                                                 *port,
  98                                                 MACH_MSG_TYPE_MAKE_SEND_ONCE,
  99                                                 &oldNotify);
 100         if (status != KERN_SUCCESS) {
 101                 /*
 102                  * We can't request a notification for our own port!  This should
 103                  * only happen if someone stomped on OUR port (so let's leave
 104                  * the port alone).
 105                  */
 106                 SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreNotifyMachPort mach_port_request_notification(): %s"), mach_error_string(status));
 107                 *port = MACH_PORT_NULL;
 108                 _SCErrorSet(status);
 109                 return FALSE;
 110         }
 111
 112         if (oldNotify != MACH_PORT_NULL) {
 113                 SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreNotifyMachPort(): oldNotify != MACH_PORT_NULL"));
 114         }
 115
 116     retry :
 117
 118         status = notifyviaport(storePrivate->server,
 119                                *port,
 120                                identifier,
 121                                (int *)&sc_status);
 122
 123         if (status != KERN_SUCCESS) {
 124                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 125                         /* the server's gone and our session port's dead, remove the dead name right */
 126                         (void) mach_port_deallocate(mach_task_self(), storePrivate->server);
 127                 } else {
 128                         /* we got an unexpected error, leave the [session] port alone */
 129                         SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreNotifyMachPort notifyviaport(): %s"), mach_error_string(status));
 130                 }
 131                 storePrivate->server = MACH_PORT_NULL;
 132                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 133                         if (__SCDynamicStoreReconnect(store)) {
 134                                 goto retry;
 135                         }
 136                         /* remove the send right that we tried (but failed) to pass to the server */
 137                         (void) mach_port_deallocate(mach_task_self(), *port);
 138                 }
 139
 140                 /* remove our receive right  */
 141                 (void) mach_port_mod_refs(mach_task_self(), *port, MACH_PORT_RIGHT_RECEIVE, -1);
 142                 *port = MACH_PORT_NULL;
 143                 _SCErrorSet(status);
 144                 return FALSE;
 145         }
 146
 147         /* set notifier active */
 148         __MACH_PORT_DEBUG(TRUE, "*** SCDynamicStoreNotifyMachPort", *port);
 149         storePrivate->notifyStatus = Using_NotifierInformViaMachPort;
 150
 151         return TRUE;
 152 }