SystemConfiguration.fproj/SCDNotifierAdd.c

   1 /*
   2  * Copyright (c) 2000-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 24, 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
  43 static void
  44 addKey(CFMutableArrayRef *keysP, CFStringRef key)
  45 {
  46         if (*keysP == NULL) {
  47                 *keysP = CFArrayCreateMutable(NULL, 0, &kCFTypeArrayCallBacks);
  48         }
  49
  50         CFArrayAppendValue(*keysP, key);
  51         return;
  52 }
  53
  54
  55 Boolean
  56 SCDynamicStoreAddWatchedKey(SCDynamicStoreRef store, CFStringRef key, Boolean isRegex)
  57 {
  58         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
  59         kern_return_t                   status;
  60         CFDataRef                       utfKey;         /* serialized key */
  61         xmlData_t                       myKeyRef;
  62         CFIndex                         myKeyLen;
  63         int                             sc_status;
  64
  65         if (store == NULL) {
  66                 /* sorry, you must provide a session */
  67                 _SCErrorSet(kSCStatusNoStoreSession);
  68                 return FALSE;
  69         }
  70
  71         if (storePrivate->server == MACH_PORT_NULL) {
  72                 /* sorry, you must have an open session to play */
  73                 _SCErrorSet(kSCStatusNoStoreServer);
  74                 return FALSE;
  75         }
  76
  77         /* serialize the key */
  78         if (!_SCSerializeString(key, &utfKey, (void **)&myKeyRef, &myKeyLen)) {
  79                 _SCErrorSet(kSCStatusFailed);
  80                 return FALSE;
  81         }
  82
  83     retry :
  84
  85         /* send the key to the server */
  86         status = notifyadd(storePrivate->server,
  87                            myKeyRef,
  88                            myKeyLen,
  89                            isRegex,
  90                            (int *)&sc_status);
  91
  92         if (status != KERN_SUCCESS) {
  93                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
  94                         /* the server's gone and our session port's dead, remove the dead name right */
  95                         (void) mach_port_deallocate(mach_task_self(), storePrivate->server);
  96                 } else {
  97                         /* we got an unexpected error, leave the [session] port alone */
  98                         SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreAddWatchedKey notifyadd(): %s"), mach_error_string(status));
  99                 }
 100                 storePrivate->server = MACH_PORT_NULL;
 101                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 102                         if (__SCDynamicStoreReconnect(store)) {
 103                                 goto retry;
 104                         }
 105                 }
 106                 sc_status = status;
 107         }
 108
 109         /* clean up */
 110         CFRelease(utfKey);
 111
 112         if (sc_status != kSCStatusOK) {
 113                 _SCErrorSet(sc_status);
 114                 return FALSE;
 115         }
 116
 117         if (isRegex) {
 118                 addKey(&storePrivate->patterns, key);
 119         } else {
 120                 addKey(&storePrivate->keys, key);
 121         }
 122         return TRUE;
 123 }