SystemConfiguration.fproj/SCDSet.c

   1 /*
   2  * Copyright (c) 2000-2006, 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 Boolean
  44 SCDynamicStoreSetMultiple(SCDynamicStoreRef     store,
  45                           CFDictionaryRef       keysToSet,
  46                           CFArrayRef            keysToRemove,
  47                           CFArrayRef            keysToNotify)
  48 {
  49         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
  50         kern_return_t                   status;
  51         CFDataRef                       xmlSet          = NULL; /* key/value pairs to set (XML serialized) */
  52         xmlData_t                       mySetRef        = NULL; /* key/value pairs to set (serialized) */
  53         CFIndex                         mySetLen        = 0;
  54         CFDataRef                       xmlRemove       = NULL; /* keys to remove (XML serialized) */
  55         xmlData_t                       myRemoveRef     = NULL; /* keys to remove (serialized) */
  56         CFIndex                         myRemoveLen     = 0;
  57         CFDataRef                       xmlNotify       = NULL; /* keys to notify (XML serialized) */
  58         xmlData_t                       myNotifyRef     = NULL; /* keys to notify (serialized) */
  59         CFIndex                         myNotifyLen     = 0;
  60         int                             sc_status;
  61
  62         if (store == NULL) {
  63                 /* sorry, you must provide a session */
  64                 _SCErrorSet(kSCStatusNoStoreSession);
  65                 return FALSE;
  66         }
  67
  68         if (storePrivate->server == MACH_PORT_NULL) {
  69                 _SCErrorSet(kSCStatusNoStoreServer);
  70                 return FALSE;   /* you must have an open session to play */
  71         }
  72
  73         /* serialize the key/value pairs to set*/
  74         if (keysToSet != NULL) {
  75                 CFDictionaryRef newInfo;
  76                 Boolean         ok;
  77
  78                 newInfo = _SCSerializeMultiple(keysToSet);
  79                 if (newInfo == NULL) {
  80                         _SCErrorSet(kSCStatusInvalidArgument);
  81                         return FALSE;
  82                 }
  83
  84                 ok = _SCSerialize(newInfo, &xmlSet, (void **)&mySetRef, &mySetLen);
  85                 CFRelease(newInfo);
  86                 if (!ok) {
  87                         _SCErrorSet(kSCStatusInvalidArgument);
  88                         return FALSE;
  89                 }
  90         }
  91
  92         /* serialize the keys to remove */
  93         if (keysToRemove != NULL) {
  94                 if (!_SCSerialize(keysToRemove, &xmlRemove, (void **)&myRemoveRef, &myRemoveLen)) {
  95                         if (xmlSet != NULL)     CFRelease(xmlSet);
  96                         _SCErrorSet(kSCStatusInvalidArgument);
  97                         return FALSE;
  98                 }
  99         }
 100
 101         /* serialize the keys to notify */
 102         if (keysToNotify != NULL) {
 103                 if (!_SCSerialize(keysToNotify, &xmlNotify, (void **)&myNotifyRef, &myNotifyLen)) {
 104                         if (xmlSet != NULL)     CFRelease(xmlSet);
 105                         if (xmlRemove != NULL)  CFRelease(xmlRemove);
 106                         _SCErrorSet(kSCStatusInvalidArgument);
 107                         return FALSE;
 108                 }
 109         }
 110
 111     retry :
 112
 113         /* send the keys and patterns, fetch the associated result from the server */
 114         status = configset_m(storePrivate->server,
 115                              mySetRef,
 116                              mySetLen,
 117                              myRemoveRef,
 118                              myRemoveLen,
 119                              myNotifyRef,
 120                              myNotifyLen,
 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("SCDynamicStoreSetMultiple configset_m(): %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                 }
 137                 sc_status = status;
 138         }
 139
 140         /* clean up */
 141         if (xmlSet != NULL)     CFRelease(xmlSet);
 142         if (xmlRemove != NULL)  CFRelease(xmlRemove);
 143         if (xmlNotify != NULL)  CFRelease(xmlNotify);
 144
 145         if (sc_status != kSCStatusOK) {
 146                 _SCErrorSet(sc_status);
 147                 return FALSE;
 148         }
 149
 150         return TRUE;
 151 }
 152
 153 Boolean
 154 SCDynamicStoreSetValue(SCDynamicStoreRef store, CFStringRef key, CFPropertyListRef value)
 155 {
 156         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
 157         kern_return_t                   status;
 158         CFDataRef                       utfKey;         /* serialized key */
 159         xmlData_t                       myKeyRef;
 160         CFIndex                         myKeyLen;
 161         CFDataRef                       xmlData;        /* serialized data */
 162         xmlData_t                       myDataRef;
 163         CFIndex                         myDataLen;
 164         int                             sc_status;
 165         int                             newInstance;
 166
 167         if (store == NULL) {
 168                 /* sorry, you must provide a session */
 169                 _SCErrorSet(kSCStatusNoStoreSession);
 170                 return FALSE;
 171         }
 172
 173         if (storePrivate->server == MACH_PORT_NULL) {
 174                 /* sorry, you must have an open session to play */
 175                 _SCErrorSet(kSCStatusNoStoreServer);
 176                 return FALSE;
 177         }
 178
 179         /* serialize the key */
 180         if (!_SCSerializeString(key, &utfKey, (void **)&myKeyRef, &myKeyLen)) {
 181                 _SCErrorSet(kSCStatusInvalidArgument);
 182                 return FALSE;
 183         }
 184
 185         /* serialize the data */
 186         if (!_SCSerialize(value, &xmlData, (void **)&myDataRef, &myDataLen)) {
 187                 CFRelease(utfKey);
 188                 _SCErrorSet(kSCStatusInvalidArgument);
 189                 return FALSE;
 190         }
 191
 192     retry :
 193
 194         /* send the key & data to the server, get new instance id */
 195         status = configset(storePrivate->server,
 196                            myKeyRef,
 197                            myKeyLen,
 198                            myDataRef,
 199                            myDataLen,
 200                            0,
 201                            &newInstance,
 202                            (int *)&sc_status);
 203
 204         if (status != KERN_SUCCESS) {
 205                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 206                         /* the server's gone and our session port's dead, remove the dead name right */
 207                         (void) mach_port_deallocate(mach_task_self(), storePrivate->server);
 208                 } else {
 209                         /* we got an unexpected error, leave the [session] port alone */
 210                         SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreSetValue configset(): %s"), mach_error_string(status));
 211                 }
 212                 storePrivate->server = MACH_PORT_NULL;
 213                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 214                         if (__SCDynamicStoreReconnect(store)) {
 215                                 goto retry;
 216                         }
 217                 }
 218                 sc_status = status;
 219         }
 220
 221         /* clean up */
 222         CFRelease(utfKey);
 223         CFRelease(xmlData);
 224
 225         if (sc_status != kSCStatusOK) {
 226                 _SCErrorSet(sc_status);
 227                 return FALSE;
 228         }
 229
 230         return TRUE;
 231 }