configd.tproj/_notifyviaport.c

   1 /*
   2  * Copyright (c) 2000, 2001, 2003, 2004, 2006, 2009, 2011 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 24, 2000               Allan Nathanson <ajn@apple.com>
  31  * - initial revision
  32  */
  33
  34 #include "configd.h"
  35 #include "session.h"
  36
  37 __private_extern__
  38 int
  39 __SCDynamicStoreNotifyMachPort(SCDynamicStoreRef        store,
  40                                mach_msg_id_t            identifier,
  41                                mach_port_t              port)
  42 {
  43         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
  44         CFStringRef                     sessionKey;
  45         CFDictionaryRef                 info;
  46
  47         if (storePrivate->notifyStatus != NotifierNotRegistered) {
  48                 /* sorry, you can only have one notification registered at once */
  49                 return kSCStatusNotifierActive;
  50         }
  51
  52         if (port == MACH_PORT_NULL) {
  53                 /* sorry, you must specify a valid mach port */
  54                 return kSCStatusInvalidArgument;
  55         }
  56
  57         /* push out a notification if any changes are pending */
  58         sessionKey = CFStringCreateWithFormat(NULL, NULL, CFSTR("%d"), storePrivate->server);
  59         info = CFDictionaryGetValue(sessionData, sessionKey);
  60         CFRelease(sessionKey);
  61         if (info && CFDictionaryContainsKey(info, kSCDChangedKeys)) {
  62                 CFNumberRef     sessionNum;
  63
  64                 if (needsNotification == NULL)
  65                         needsNotification = CFSetCreateMutable(NULL,
  66                                                                0,
  67                                                                &kCFTypeSetCallBacks);
  68
  69                 sessionNum = CFNumberCreate(NULL, kCFNumberIntType, &storePrivate->server);
  70                 CFSetAddValue(needsNotification, sessionNum);
  71                 CFRelease(sessionNum);
  72         }
  73
  74         return kSCStatusOK;
  75 }
  76
  77
  78 __private_extern__
  79 kern_return_t
  80 _notifyviaport(mach_port_t      server,
  81                mach_port_t      port,
  82                mach_msg_id_t    identifier,
  83                int              *sc_status
  84 )
  85 {
  86         serverSessionRef                mySession       = getSession(server);
  87         SCDynamicStorePrivateRef        storePrivate;
  88
  89         if (mySession == NULL) {
  90                 /* sorry, you must have an open session to play */
  91                 *sc_status = kSCStatusNoStoreSession;
  92                 if (port != MACH_PORT_NULL) {
  93                         (void) mach_port_deallocate(mach_task_self(), port);
  94                 }
  95                 return KERN_SUCCESS;
  96         }
  97         storePrivate = (SCDynamicStorePrivateRef)mySession->store;
  98
  99         *sc_status = __SCDynamicStoreNotifyMachPort(mySession->store, identifier, port);
 100         if (*sc_status != kSCStatusOK) {
 101                 // if we can't enable the notification, release the provided callback port
 102                 if (port != MACH_PORT_NULL) {
 103                         __MACH_PORT_DEBUG(TRUE, "*** _notifyviaport __SCDynamicStoreNotifyMachPort failed: releasing port", port);
 104                         (void) mach_port_deallocate(mach_task_self(), port);
 105                 }
 106                 return KERN_SUCCESS;
 107         }
 108
 109         /* save notification port, requested identifier, and set notifier active */
 110         __MACH_PORT_DEBUG(TRUE, "*** _notifyviaport", port);
 111         storePrivate->notifyStatus         = Using_NotifierInformViaMachPort;
 112         storePrivate->notifyPort           = port;
 113         storePrivate->notifyPortIdentifier = identifier;
 114
 115         return KERN_SUCCESS;
 116 }