SystemConfiguration.fproj/SCDSet.c

   1 /*
   2  * Copyright (c) 2000-2006, 2009 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         /* send the keys and patterns, fetch the associated result from the server */
 112         status = configset_m(storePrivate->server,
 113                              mySetRef,
 114                              mySetLen,
 115                              myRemoveRef,
 116                              myRemoveLen,
 117                              myNotifyRef,
 118                              myNotifyLen,
 119                              (int *)&sc_status);
 120
 121         /* clean up */
 122         if (xmlSet != NULL)     CFRelease(xmlSet);
 123         if (xmlRemove != NULL)  CFRelease(xmlRemove);
 124         if (xmlNotify != NULL)  CFRelease(xmlNotify);
 125
 126         if (status != KERN_SUCCESS) {
 127                 if (status == MACH_SEND_INVALID_DEST) {
 128                         /* the server's gone and our session port's dead, remove the dead name right */
 129                         (void) mach_port_deallocate(mach_task_self(), storePrivate->server);
 130                 } else {
 131                         /* we got an unexpected error, leave the [session] port alone */
 132                         SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreSetMultiple configset_m(): %s"), mach_error_string(status));
 133                 }
 134                 storePrivate->server = MACH_PORT_NULL;
 135                 _SCErrorSet(status);
 136                 return FALSE;
 137         }
 138
 139         if (sc_status != kSCStatusOK) {
 140                 _SCErrorSet(sc_status);
 141                 return FALSE;
 142         }
 143
 144         return TRUE;
 145 }
 146
 147 Boolean
 148 SCDynamicStoreSetValue(SCDynamicStoreRef store, CFStringRef key, CFPropertyListRef value)
 149 {
 150         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
 151         kern_return_t                   status;
 152         CFDataRef                       utfKey;         /* serialized key */
 153         xmlData_t                       myKeyRef;
 154         CFIndex                         myKeyLen;
 155         CFDataRef                       xmlData;        /* serialized data */
 156         xmlData_t                       myDataRef;
 157         CFIndex                         myDataLen;
 158         int                             sc_status;
 159         int                             newInstance;
 160
 161         if (store == NULL) {
 162                 /* sorry, you must provide a session */
 163                 _SCErrorSet(kSCStatusNoStoreSession);
 164                 return FALSE;
 165         }
 166
 167         if (storePrivate->server == MACH_PORT_NULL) {
 168                 /* sorry, you must have an open session to play */
 169                 _SCErrorSet(kSCStatusNoStoreServer);
 170                 return FALSE;
 171         }
 172
 173         /* serialize the key */
 174         if (!_SCSerializeString(key, &utfKey, (void **)&myKeyRef, &myKeyLen)) {
 175                 _SCErrorSet(kSCStatusInvalidArgument);
 176                 return FALSE;
 177         }
 178
 179         /* serialize the data */
 180         if (!_SCSerialize(value, &xmlData, (void **)&myDataRef, &myDataLen)) {
 181                 CFRelease(utfKey);
 182                 _SCErrorSet(kSCStatusInvalidArgument);
 183                 return FALSE;
 184         }
 185
 186         /* send the key & data to the server, get new instance id */
 187         status = configset(storePrivate->server,
 188                            myKeyRef,
 189                            myKeyLen,
 190                            myDataRef,
 191                            myDataLen,
 192                            0,
 193                            &newInstance,
 194                            (int *)&sc_status);
 195
 196         /* clean up */
 197         CFRelease(utfKey);
 198         CFRelease(xmlData);
 199
 200         if (status != KERN_SUCCESS) {
 201                 if (status == MACH_SEND_INVALID_DEST) {
 202                         /* the server's gone and our session port's dead, remove the dead name right */
 203                         (void) mach_port_deallocate(mach_task_self(), storePrivate->server);
 204                 } else {
 205                         /* we got an unexpected error, leave the [session] port alone */
 206                         SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreSetValue configset(): %s"), mach_error_string(status));
 207                 }
 208                 storePrivate->server = MACH_PORT_NULL;
 209                 _SCErrorSet(status);
 210                 return FALSE;
 211         }
 212
 213         if (sc_status != kSCStatusOK) {
 214                 _SCErrorSet(sc_status);
 215                 return FALSE;
 216         }
 217
 218         return TRUE;
 219 }