configd.tproj/_snapshot.c

   1 /*
   2  * Copyright (c) 2000-2006 Apple Computer, 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  * April 14, 2000               Allan Nathanson <ajn@apple.com>
  31  * - initial revision
  32  */
  33
  34 #include <fcntl.h>
  35 #include <paths.h>
  36 #include <unistd.h>
  37
  38 #include "configd.h"
  39 #include "configd_server.h"
  40 #include "session.h"
  41 #include "plugin_support.h"
  42
  43
  44 #define SNAPSHOT_PATH_STATE     _PATH_VARTMP "configd-state"
  45 #define SNAPSHOT_PATH_STORE     _PATH_VARTMP "configd-store.xml"
  46 #define SNAPSHOT_PATH_PATTERN   _PATH_VARTMP "configd-pattern.xml"
  47 #define SNAPSHOT_PATH_SESSION   _PATH_VARTMP "configd-session.xml"
  48
  49
  50 #define N_QUICK 100
  51
  52 static CFDictionaryRef
  53 _expandStore(CFDictionaryRef storeData)
  54 {
  55         const void *            keys_q[N_QUICK];
  56         const void **           keys            = keys_q;
  57         CFIndex                 nElements;
  58         CFDictionaryRef         newStoreData    = NULL;
  59         const void *            nValues_q[N_QUICK];
  60         const void **           nValues         = nValues_q;
  61         const void *            oValues_q[N_QUICK];
  62         const void **           oValues         = oValues_q;
  63
  64         nElements = CFDictionaryGetCount(storeData);
  65         if (nElements > 0) {
  66                 CFIndex i;
  67
  68                 if (nElements > (CFIndex)(sizeof(keys_q) / sizeof(CFTypeRef))) {
  69                         keys    = CFAllocatorAllocate(NULL, nElements * sizeof(CFTypeRef), 0);
  70                         oValues = CFAllocatorAllocate(NULL, nElements * sizeof(CFTypeRef), 0);
  71                         nValues = CFAllocatorAllocate(NULL, nElements * sizeof(CFTypeRef), 0);
  72                 }
  73                 bzero(nValues, nElements * sizeof(CFTypeRef));
  74
  75                 CFDictionaryGetKeysAndValues(storeData, keys, oValues);
  76                 for (i = 0; i < nElements; i++) {
  77                         CFDataRef               data;
  78
  79                         data = CFDictionaryGetValue(oValues[i], kSCDData);
  80                         if (data) {
  81                                 CFPropertyListRef       plist;
  82
  83                                 if (!_SCUnserialize(&plist, data, NULL, 0)) {
  84                                         goto done;
  85                                 }
  86
  87                                 nValues[i] = CFDictionaryCreateMutableCopy(NULL, 0, oValues[i]);
  88                                 CFDictionarySetValue((CFMutableDictionaryRef)nValues[i],
  89                                                      kSCDData,
  90                                                      plist);
  91                                 CFRelease(plist);
  92                         } else {
  93                                 nValues[i] = CFRetain(oValues[i]);
  94                         }
  95                 }
  96         }
  97
  98         newStoreData = CFDictionaryCreate(NULL,
  99                                      keys,
 100                                      nValues,
 101                                      nElements,
 102                                      &kCFTypeDictionaryKeyCallBacks,
 103                                      &kCFTypeDictionaryValueCallBacks);
 104
 105     done :
 106
 107         if (nElements > 0) {
 108                 CFIndex i;
 109
 110                 for (i = 0; i < nElements; i++) {
 111                         if (nValues[i]) CFRelease(nValues[i]);
 112                 }
 113
 114                 if (keys != keys_q) {
 115                         CFAllocatorDeallocate(NULL, keys);
 116                         CFAllocatorDeallocate(NULL, oValues);
 117                         CFAllocatorDeallocate(NULL, nValues);
 118                 }
 119         }
 120
 121         return newStoreData;
 122 }
 123
 124
 125 __private_extern__
 126 int
 127 __SCDynamicStoreSnapshot(SCDynamicStoreRef store)
 128 {
 129         CFDictionaryRef                 expandedStoreData;
 130         FILE                            *f;
 131         int                             fd;
 132         serverSessionRef                mySession;
 133         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
 134         CFDataRef                       xmlData;
 135
 136         /* check credentials */
 137
 138         mySession = getSession(storePrivate->server);
 139         if (mySession->callerEUID != 0) {
 140                 return kSCStatusAccessError;
 141         }
 142
 143         /* Save a snapshot of configd's "state" */
 144
 145         (void) unlink(SNAPSHOT_PATH_STATE);
 146         fd = open(SNAPSHOT_PATH_STATE, O_WRONLY|O_CREAT|O_TRUNC|O_EXCL, 0644);
 147         if (fd == -1) {
 148                 return kSCStatusFailed;
 149         }
 150         f = fdopen(fd, "w");
 151         if (f == NULL) {
 152                 return kSCStatusFailed;
 153         }
 154         SCPrint(TRUE, f, CFSTR("Main thread :\n\n"));
 155         SCPrint(TRUE, f, CFSTR("%@\n"), CFRunLoopGetCurrent());
 156         if (plugin_runLoop != NULL) {
 157                 SCPrint(TRUE, f, CFSTR("Plug-in thread :\n\n"));
 158                 SCPrint(TRUE, f, CFSTR("%@\n"), plugin_runLoop);
 159         }
 160         (void) fclose(f);
 161
 162         /* Save a snapshot of the "store" data */
 163
 164         (void) unlink(SNAPSHOT_PATH_STORE);
 165         fd = open(SNAPSHOT_PATH_STORE, O_WRONLY|O_CREAT|O_TRUNC|O_EXCL, 0644);
 166         if (fd == -1) {
 167                 return kSCStatusFailed;
 168         }
 169
 170         expandedStoreData = _expandStore(storeData);
 171         xmlData = CFPropertyListCreateXMLData(NULL, expandedStoreData);
 172         CFRelease(expandedStoreData);
 173         if (!xmlData) {
 174                 SCLog(TRUE, LOG_ERR, CFSTR("__SCDynamicStoreSnapshot CFPropertyListCreateXMLData() failed"));
 175                 close(fd);
 176                 return kSCStatusFailed;
 177         }
 178         (void) write(fd, CFDataGetBytePtr(xmlData), CFDataGetLength(xmlData));
 179         (void) close(fd);
 180         CFRelease(xmlData);
 181
 182         /* Save a snapshot of the "pattern" data */
 183
 184         (void) unlink(SNAPSHOT_PATH_PATTERN);
 185         fd = open(SNAPSHOT_PATH_PATTERN, O_WRONLY|O_CREAT|O_TRUNC|O_EXCL, 0644);
 186         if (fd == -1) {
 187                 return kSCStatusFailed;
 188         }
 189
 190         xmlData = CFPropertyListCreateXMLData(NULL, patternData);
 191         if (!xmlData) {
 192                 SCLog(TRUE, LOG_ERR, CFSTR("__SCDynamicStoreSnapshot CFPropertyListCreateXMLData() failed"));
 193                 close(fd);
 194                 return kSCStatusFailed;
 195         }
 196         (void) write(fd, CFDataGetBytePtr(xmlData), CFDataGetLength(xmlData));
 197         (void) close(fd);
 198         CFRelease(xmlData);
 199
 200         /* Save a snapshot of the "session" data */
 201
 202         (void) unlink(SNAPSHOT_PATH_SESSION);
 203         fd = open(SNAPSHOT_PATH_SESSION, O_WRONLY|O_CREAT|O_TRUNC|O_EXCL, 0644);
 204         if (fd == -1) {
 205                 return kSCStatusFailed;
 206         }
 207
 208         xmlData = CFPropertyListCreateXMLData(NULL, sessionData);
 209         if (!xmlData) {
 210                 SCLog(TRUE, LOG_ERR, CFSTR("__SCDynamicStoreSnapshot CFPropertyListCreateXMLData() failed"));
 211                 close(fd);
 212                 return kSCStatusFailed;
 213         }
 214         (void) write(fd, CFDataGetBytePtr(xmlData), CFDataGetLength(xmlData));
 215         (void) close(fd);
 216         CFRelease(xmlData);
 217
 218         return kSCStatusOK;
 219 }
 220
 221
 222 __private_extern__
 223 kern_return_t
 224 _snapshot(mach_port_t server, int *sc_status)
 225 {
 226         serverSessionRef        mySession = getSession(server);
 227
 228         if (mySession == NULL) {
 229                 *sc_status = kSCStatusNoStoreSession;   /* you must have an open session to play */
 230                 return KERN_SUCCESS;
 231         }
 232
 233         *sc_status = __SCDynamicStoreSnapshot(mySession->store);
 234         return KERN_SUCCESS;
 235 }