SystemConfiguration.fproj/SCDNotifierSetKeys.c

   1 /*
   2  * Copyright (c) 2000, 2001, 2003-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  * - initial revision
  29  */
  30
  31 #include <mach/mach.h>
  32 #include <mach/mach_error.h>
  33
  34 #include <SystemConfiguration/SystemConfiguration.h>
  35 #include <SystemConfiguration/SCPrivate.h>
  36 #include "SCDynamicStoreInternal.h"
  37 #include "config.h"             /* MiG generated file */
  38
  39
  40 Boolean
  41 SCDynamicStoreSetNotificationKeys(SCDynamicStoreRef     store,
  42                                   CFArrayRef            keys,
  43                                   CFArrayRef            patterns)
  44 {
  45         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
  46         kern_return_t                   status;
  47         CFDataRef                       xmlKeys         = NULL; /* keys (XML serialized) */
  48         xmlData_t                       myKeysRef       = NULL; /* keys (serialized) */
  49         CFIndex                         myKeysLen       = 0;
  50         CFDataRef                       xmlPatterns     = NULL; /* patterns (XML serialized) */
  51         xmlData_t                       myPatternsRef   = NULL; /* patterns (serialized) */
  52         CFIndex                         myPatternsLen   = 0;
  53         int                             sc_status;
  54         CFMutableArrayRef               tmp;
  55
  56         if (store == NULL) {
  57                 /* sorry, you must provide a session */
  58                 _SCErrorSet(kSCStatusNoStoreSession);
  59                 return FALSE;
  60         }
  61
  62         if (storePrivate->server == MACH_PORT_NULL) {
  63                 _SCErrorSet(kSCStatusNoStoreServer);
  64                 return FALSE;   /* you must have an open session to play */
  65         }
  66
  67         /* serialize the keys */
  68         if (keys != NULL) {
  69                 if (!_SCSerialize(keys, &xmlKeys, (void **)&myKeysRef, &myKeysLen)) {
  70                         _SCErrorSet(kSCStatusFailed);
  71                         return FALSE;
  72                 }
  73         }
  74
  75         /* serialize the patterns */
  76         if (patterns != NULL) {
  77                 if (!_SCSerialize(patterns, &xmlPatterns, (void **)&myPatternsRef, &myPatternsLen)) {
  78                         if (xmlKeys != NULL) CFRelease(xmlKeys);
  79                         _SCErrorSet(kSCStatusFailed);
  80                         return FALSE;
  81                 }
  82         }
  83
  84     retry :
  85
  86         /* send the keys and patterns, fetch the associated result from the server */
  87         status = notifyset(storePrivate->server,
  88                            myKeysRef,
  89                            myKeysLen,
  90                            myPatternsRef,
  91                            myPatternsLen,
  92                            (int *)&sc_status);
  93
  94         if (status != KERN_SUCCESS) {
  95                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
  96                         /* the server's gone and our session port's dead, remove the dead name right */
  97                         (void) mach_port_deallocate(mach_task_self(), storePrivate->server);
  98                 } else {
  99                         /* we got an unexpected error, leave the [session] port alone */
 100                         SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreSetNotificationKeys notifyset(): %s"), mach_error_string(status));
 101                 }
 102                 storePrivate->server = MACH_PORT_NULL;
 103                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 104                         if (__SCDynamicStoreReconnect(store)) {
 105                                 goto retry;
 106                         }
 107                 }
 108                 sc_status = status;
 109         }
 110
 111         /* clean up */
 112         if (xmlKeys != NULL)            CFRelease(xmlKeys);
 113         if (xmlPatterns != NULL)        CFRelease(xmlPatterns);
 114
 115         if (sc_status != kSCStatusOK) {
 116                 _SCErrorSet(sc_status);
 117                 return FALSE;
 118         }
 119
 120         /* in case we need to re-connect, save the keys/patterns */
 121         tmp = (keys != NULL) ? CFArrayCreateMutableCopy(NULL, 0, keys) : NULL;
 122         if (storePrivate->keys != NULL) CFRelease(storePrivate->keys);
 123         storePrivate->keys = tmp;
 124
 125         tmp = (patterns != NULL) ? CFArrayCreateMutableCopy(NULL, 0, patterns) : NULL;
 126         if (storePrivate->patterns != NULL) CFRelease(storePrivate->patterns);
 127         storePrivate->patterns = tmp;
 128
 129         return TRUE;
 130 }