SystemConfiguration.fproj/SCDAdd.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 Boolean
  43 SCDynamicStoreAddTemporaryValue(SCDynamicStoreRef store, CFStringRef key, CFPropertyListRef value)
  44 {
  45         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
  46         kern_return_t                   status;
  47         CFDataRef                       utfKey;         /* serialized key */
  48         xmlData_t                       myKeyRef;
  49         CFIndex                         myKeyLen;
  50         CFDataRef                       xmlData;        /* serialized data */
  51         xmlData_t                       myDataRef;
  52         CFIndex                         myDataLen;
  53         int                             newInstance;
  54         int                             sc_status;
  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                 /* sorry, you must have an open session to play */
  64                 _SCErrorSet(kSCStatusNoStoreServer);
  65                 return FALSE;
  66         }
  67
  68         /* serialize the key */
  69         if (!_SCSerializeString(key, &utfKey, (void **)&myKeyRef, &myKeyLen)) {
  70                 _SCErrorSet(kSCStatusFailed);
  71                 return FALSE;
  72         }
  73
  74         /* serialize the data */
  75         if (!_SCSerialize(value, &xmlData, (void **)&myDataRef, &myDataLen)) {
  76                 CFRelease(utfKey);
  77                 _SCErrorSet(kSCStatusFailed);
  78                 return FALSE;
  79         }
  80
  81     retry :
  82
  83         /* send the key & data to the server */
  84         status = configadd_s(storePrivate->server,
  85                              myKeyRef,
  86                              myKeyLen,
  87                              myDataRef,
  88                              myDataLen,
  89                              &newInstance,
  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("SCDynamicStoreAddTemporaryValue configadd_s(): %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         CFRelease(xmlData);
 112
 113         if (sc_status != kSCStatusOK) {
 114                 _SCErrorSet(sc_status);
 115                 return FALSE;
 116         }
 117
 118         return TRUE;
 119 }
 120
 121 Boolean
 122 SCDynamicStoreAddValue(SCDynamicStoreRef store, CFStringRef key, CFPropertyListRef value)
 123 {
 124         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
 125         kern_return_t                   status;
 126         CFDataRef                       utfKey;         /* serialized key */
 127         xmlData_t                       myKeyRef;
 128         CFIndex                         myKeyLen;
 129         CFDataRef                       xmlData;        /* serialized data */
 130         xmlData_t                       myDataRef;
 131         CFIndex                         myDataLen;
 132         int                             newInstance;
 133         int                             sc_status;
 134
 135         if (store == NULL) {
 136                 /* sorry, you must provide a session */
 137                 _SCErrorSet(kSCStatusNoStoreSession);
 138                 return FALSE;
 139         }
 140
 141         if (storePrivate->server == MACH_PORT_NULL) {
 142                 /* sorry, you must have an open session to play */
 143                 _SCErrorSet(kSCStatusNoStoreServer);
 144                 return FALSE;
 145         }
 146
 147         /* serialize the key */
 148         if (!_SCSerializeString(key, &utfKey, (void **)&myKeyRef, &myKeyLen)) {
 149                 _SCErrorSet(kSCStatusFailed);
 150                 return FALSE;
 151         }
 152
 153         /* serialize the data */
 154         if (!_SCSerialize(value, &xmlData, (void **)&myDataRef, &myDataLen)) {
 155                 CFRelease(utfKey);
 156                 _SCErrorSet(kSCStatusFailed);
 157                 return FALSE;
 158         }
 159
 160     retry :
 161
 162         /* send the key & data to the server */
 163         status = configadd(storePrivate->server,
 164                            myKeyRef,
 165                            myKeyLen,
 166                            myDataRef,
 167                            myDataLen,
 168                            &newInstance,
 169                            (int *)&sc_status);
 170
 171         if (status != KERN_SUCCESS) {
 172                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 173                         /* the server's gone and our session port's dead, remove the dead name right */
 174                         (void) mach_port_deallocate(mach_task_self(), storePrivate->server);
 175                 } else {
 176                         /* we got an unexpected error, leave the [session] port alone */
 177                         SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreAddValue configadd(): %s"), mach_error_string(status));
 178                 }
 179                 storePrivate->server = MACH_PORT_NULL;
 180                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 181                         if (__SCDynamicStoreReconnect(store)) {
 182                                 goto retry;
 183                         }
 184                 }
 185                 sc_status = status;
 186         }
 187
 188         /* clean up */
 189         CFRelease(utfKey);
 190         CFRelease(xmlData);
 191
 192         if (sc_status != kSCStatusOK) {
 193                 _SCErrorSet(sc_status);
 194                 return FALSE;
 195         }
 196
 197         return TRUE;
 198 }