SystemConfiguration.fproj/SCDGet.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 CFDictionaryRef
  44 SCDynamicStoreCopyMultiple(SCDynamicStoreRef    store,
  45                            CFArrayRef           keys,
  46                            CFArrayRef           patterns)
  47 {
  48         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
  49         kern_return_t                   status;
  50         CFDataRef                       xmlKeys         = NULL; /* keys (XML serialized) */
  51         xmlData_t                       myKeysRef       = NULL; /* keys (serialized) */
  52         CFIndex                         myKeysLen       = 0;
  53         CFDataRef                       xmlPatterns     = NULL; /* patterns (XML serialized) */
  54         xmlData_t                       myPatternsRef   = NULL; /* patterns (serialized) */
  55         CFIndex                         myPatternsLen   = 0;
  56         xmlDataOut_t                    xmlDictRef      = NULL; /* dict (serialized) */
  57         mach_msg_type_number_t          xmlDictLen      = 0;
  58         CFDictionaryRef                 dict            = NULL; /* dict (un-serialized) */
  59         CFDictionaryRef                 expDict         = NULL; /* dict (un-serialized / expanded) */
  60         int                             sc_status;
  61
  62         if (store == NULL) {
  63                 /* sorry, you must provide a session */
  64                 _SCErrorSet(kSCStatusNoStoreSession);
  65                 return NULL;
  66         }
  67
  68         if (storePrivate->server == MACH_PORT_NULL) {
  69                 _SCErrorSet(kSCStatusNoStoreServer);
  70                 return NULL;    /* you must have an open session to play */
  71         }
  72
  73         /* serialize the keys */
  74         if (keys != NULL) {
  75                 if (!_SCSerialize(keys, &xmlKeys, (void **)&myKeysRef, &myKeysLen)) {
  76                         _SCErrorSet(kSCStatusFailed);
  77                         return NULL;
  78                 }
  79         }
  80
  81         /* serialize the patterns */
  82         if (patterns != NULL) {
  83                 if (!_SCSerialize(patterns, &xmlPatterns, (void **)&myPatternsRef, &myPatternsLen)) {
  84                         if (xmlKeys != NULL) CFRelease(xmlKeys);
  85                         _SCErrorSet(kSCStatusFailed);
  86                         return NULL;
  87                 }
  88         }
  89
  90     retry :
  91
  92         /* send the keys and patterns, fetch the associated result from the server */
  93         status = configget_m(storePrivate->server,
  94                              myKeysRef,
  95                              myKeysLen,
  96                              myPatternsRef,
  97                              myPatternsLen,
  98                              &xmlDictRef,
  99                              &xmlDictLen,
 100                              (int *)&sc_status);
 101
 102         if (status != KERN_SUCCESS) {
 103                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 104                         /* the server's gone and our session port's dead, remove the dead name right */
 105                         (void) mach_port_deallocate(mach_task_self(), storePrivate->server);
 106                 } else {
 107                         /* we got an unexpected error, leave the [session] port alone */
 108                         SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreCopyMultiple configget_m(): %s"), mach_error_string(status));
 109                 }
 110                 storePrivate->server = MACH_PORT_NULL;
 111                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 112                         if (__SCDynamicStoreReconnect(store)) {
 113                                 goto retry;
 114                         }
 115                 }
 116                 sc_status = status;
 117         }
 118
 119         /* clean up */
 120         if (xmlKeys != NULL)            CFRelease(xmlKeys);
 121         if (xmlPatterns != NULL)        CFRelease(xmlPatterns);
 122
 123         if (sc_status != kSCStatusOK) {
 124                 if (xmlDictRef != NULL) {
 125                         (void) vm_deallocate(mach_task_self(), (vm_address_t)xmlDictRef, xmlDictLen);
 126                 }
 127                 _SCErrorSet(sc_status);
 128                 return NULL;
 129         }
 130
 131         /* un-serialize the dictionary */
 132         if (!_SCUnserialize((CFPropertyListRef *)&dict, NULL, xmlDictRef, xmlDictLen)) {
 133                 _SCErrorSet(kSCStatusFailed);
 134                 return NULL;
 135         }
 136
 137         expDict = _SCUnserializeMultiple(dict);
 138         CFRelease(dict);
 139
 140         return expDict;
 141 }
 142
 143
 144 CFPropertyListRef
 145 SCDynamicStoreCopyValue(SCDynamicStoreRef store, CFStringRef key)
 146 {
 147         SCDynamicStorePrivateRef        storePrivate    = (SCDynamicStorePrivateRef)store;
 148         kern_return_t                   status;
 149         CFDataRef                       utfKey;                 /* key (XML serialized) */
 150         xmlData_t                       myKeyRef;               /* key (serialized) */
 151         CFIndex                         myKeyLen;
 152         xmlDataOut_t                    xmlDataRef      = NULL; /* data (serialized) */
 153         mach_msg_type_number_t          xmlDataLen      = 0;
 154         CFPropertyListRef               data;                   /* data (un-serialized) */
 155         int                             newInstance;
 156         int                             sc_status;
 157
 158         if (store == NULL) {
 159                 /* sorry, you must provide a session */
 160                 _SCErrorSet(kSCStatusNoStoreSession);
 161                 return NULL;
 162         }
 163
 164         if (storePrivate->server == MACH_PORT_NULL) {
 165                 _SCErrorSet(kSCStatusNoStoreServer);
 166                 return NULL;    /* you must have an open session to play */
 167         }
 168
 169         /* serialize the key */
 170         if (!_SCSerializeString(key, &utfKey, (void **)&myKeyRef, &myKeyLen)) {
 171                 _SCErrorSet(kSCStatusFailed);
 172                 return NULL;
 173         }
 174
 175     retry :
 176
 177         /* send the key & fetch the associated data from the server */
 178         status = configget(storePrivate->server,
 179                            myKeyRef,
 180                            myKeyLen,
 181                            &xmlDataRef,
 182                            &xmlDataLen,
 183                            &newInstance,
 184                            (int *)&sc_status);
 185
 186         if (status != KERN_SUCCESS) {
 187                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 188                         /* the server's gone and our session port's dead, remove the dead name right */
 189                         (void) mach_port_deallocate(mach_task_self(), storePrivate->server);
 190                 } else {
 191                         /* we got an unexpected error, leave the [session] port alone */
 192                         SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreCopyValue configget(): %s"), mach_error_string(status));
 193                 }
 194                 storePrivate->server = MACH_PORT_NULL;
 195                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 196                         if (__SCDynamicStoreReconnect(store)) {
 197                                 goto retry;
 198                         }
 199                 }
 200                 sc_status = status;
 201         }
 202
 203         /* clean up */
 204         CFRelease(utfKey);
 205
 206         if (sc_status != kSCStatusOK) {
 207                 if (xmlDataRef != NULL) {
 208                         (void) vm_deallocate(mach_task_self(), (vm_address_t)xmlDataRef, xmlDataLen);
 209                 }
 210                 _SCErrorSet(sc_status);
 211                 return NULL;
 212         }
 213
 214         /* un-serialize the data */
 215         if (!_SCUnserialize(&data, NULL, xmlDataRef, xmlDataLen)) {
 216                 _SCErrorSet(kSCStatusFailed);
 217                 return NULL;
 218         }
 219
 220         return data;
 221 }