configd.tproj/_configlist.c

   1 /*
   2  * Copyright (c) 2000-2003 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  * March 24, 2000               Allan Nathanson <ajn@apple.com>
  31  * - initial revision
  32  */
  33
  34 #include "configd.h"
  35 #include "session.h"
  36 #include "pattern.h"
  37
  38 #define N_QUICK 64
  39
  40 __private_extern__
  41 int
  42 __SCDynamicStoreCopyKeyList(SCDynamicStoreRef store, CFStringRef key, Boolean isRegex, CFArrayRef *subKeys)
  43 {
  44         SCDynamicStorePrivateRef        storePrivate    = (SCDynamicStorePrivateRef)store;
  45         CFMutableArrayRef               keyArray;
  46         regex_t                         preg;
  47         CFIndex                         storeCnt;
  48         CFStringRef                     storeStr;
  49         CFDictionaryRef                 storeValue;
  50
  51         if (!store || (storePrivate->server == MACH_PORT_NULL)) {
  52                 return kSCStatusNoStoreSession; /* you must have an open session to play */
  53         }
  54
  55         if (isRegex) {
  56                 CFStringRef     reErrStr;
  57
  58                 if (!patternCompile(key, &preg, &reErrStr)) {
  59                         *subKeys = CFArrayCreate(NULL, (const void **)&reErrStr, 1, &kCFTypeArrayCallBacks);
  60                         CFRelease(reErrStr);
  61                         return kSCStatusFailed;
  62                 }
  63         }
  64
  65         storeCnt = CFDictionaryGetCount(storeData);
  66         keyArray = CFArrayCreateMutable(NULL, storeCnt, &kCFTypeArrayCallBacks);
  67         if (storeCnt > 0) {
  68                 int             i;
  69                 const void *    storeKeys_q[N_QUICK];
  70                 const void **   storeKeys       = storeKeys_q;
  71                 const void *    storeValues_q[N_QUICK];
  72                 const void **   storeValues     = storeValues_q;
  73
  74                 if (storeCnt > (CFIndex)(sizeof(storeKeys_q) / sizeof(CFStringRef))) {
  75                         storeKeys   = CFAllocatorAllocate(NULL, storeCnt * sizeof(CFStringRef), 0);
  76                         storeValues = CFAllocatorAllocate(NULL, storeCnt * sizeof(CFStringRef), 0);
  77                 }
  78
  79                 CFDictionaryGetKeysAndValues(storeData, storeKeys, storeValues);
  80                 for (i = 0; i < storeCnt; i++) {
  81                         storeStr   = (CFStringRef)storeKeys[i];
  82                         storeValue = (CFDictionaryRef)storeValues[i];
  83                         if (isRegex) {
  84                                 /*
  85                                  * only return those keys which match the regular
  86                                  * expression specified in the provided key.
  87                                  */
  88
  89                                 int     reError;
  90                                 char    storeKey_q[256];
  91                                 char *  storeKey        = storeKey_q;
  92                                 CFIndex storeKeyLen     = CFStringGetLength(storeStr) + 1;
  93
  94                                 if (storeKeyLen > (CFIndex)sizeof(storeKey_q))
  95                                         storeKey = CFAllocatorAllocate(NULL, storeKeyLen, 0);
  96                                 if (_SC_cfstring_to_cstring(storeStr, storeKey, storeKeyLen, kCFStringEncodingASCII) == NULL) {
  97                                         SCLog(TRUE, LOG_DEBUG, CFSTR("__SCDynamicStoreCopyKeyList(): could not convert store key to C string"));
  98                                         if (storeKey != storeKey_q) CFAllocatorDeallocate(NULL, storeKey);
  99                                         continue;
 100                                 }
 101
 102                                 reError = regexec(&preg, storeKey, 0, NULL, 0);
 103                                 switch (reError) {
 104                                         case 0 :
 105                                                 /* we've got a match */
 106                                                 if (CFDictionaryContainsKey(storeValue, kSCDData))
 107                                                         CFArrayAppendValue(keyArray, storeStr);
 108                                                 break;
 109                                         case REG_NOMATCH :
 110                                                 /* no match */
 111                                                 break;
 112                                         default : {
 113                                                 char    reErrBuf[256];
 114                                                 int     reErrStrLen;
 115
 116                                                 reErrStrLen = regerror(reError,
 117                                                                        &preg,
 118                                                                        reErrBuf,
 119                                                                        sizeof(reErrBuf));
 120                                                 SCLog(TRUE, LOG_DEBUG, CFSTR("__SCDynamicStoreCopyKeyList regexec() failed: %s"), reErrBuf);
 121                                                 break;
 122                                         }
 123                                 }
 124                                 if (storeKey != storeKey_q) CFAllocatorDeallocate(NULL, storeKey);
 125                         } else {
 126                                 /*
 127                                  * only return those keys which are prefixed by the
 128                                  * provided key string and have data.
 129                                  */
 130                                 if (((CFStringGetLength(key) == 0) || CFStringHasPrefix(storeStr, key)) &&
 131                                     CFDictionaryContainsKey(storeValue, kSCDData)) {
 132                                         CFArrayAppendValue(keyArray, storeStr);
 133                                 }
 134                         }
 135                 }
 136
 137                 if (storeKeys != storeKeys_q) {
 138                         CFAllocatorDeallocate(NULL, storeKeys);
 139                         CFAllocatorDeallocate(NULL, storeValues);
 140                 }
 141         }
 142
 143         if (isRegex) {
 144                 regfree(&preg);
 145         }
 146
 147         *subKeys = CFArrayCreateCopy(NULL, keyArray);
 148         CFRelease(keyArray);
 149         return kSCStatusOK;
 150 }
 151
 152
 153 __private_extern__
 154 kern_return_t
 155 _configlist(mach_port_t                 server,
 156             xmlData_t                   keyRef,         /* raw XML bytes */
 157             mach_msg_type_number_t      keyLen,
 158             int                         isRegex,
 159             xmlDataOut_t                *listRef,       /* raw XML bytes */
 160             mach_msg_type_number_t      *listLen,
 161             int                         *sc_status
 162 )
 163 {
 164         CFStringRef             key             = NULL;         /* key  (un-serialized) */
 165         serverSessionRef        mySession       = getSession(server);
 166         Boolean                 ok;
 167         CFArrayRef              subKeys;                        /* array of CFStringRef's */
 168
 169         *listRef = NULL;
 170         *listLen = 0;
 171
 172         /* un-serialize the key */
 173         if (!_SCUnserializeString(&key, NULL, (void *)keyRef, keyLen)) {
 174                 *sc_status = kSCStatusFailed;
 175                 goto done;
 176         }
 177
 178         if (!isA_CFString(key)) {
 179                 *sc_status = kSCStatusInvalidArgument;
 180                 goto done;
 181         }
 182
 183         if (!mySession) {
 184                 *sc_status = kSCStatusNoStoreSession;   /* you must have an open session to play */
 185                 goto done;
 186         }
 187
 188         *sc_status = __SCDynamicStoreCopyKeyList(mySession->store, key, isRegex != 0, &subKeys);
 189         if (*sc_status != kSCStatusOK) {
 190                 goto done;
 191         }
 192
 193         /* serialize the list of keys */
 194         ok = _SCSerialize(subKeys, NULL, (void **)listRef, (CFIndex *)listLen);
 195         CFRelease(subKeys);
 196         if (!ok) {
 197                 *sc_status = kSCStatusFailed;
 198                 goto done;
 199         }
 200
 201     done :
 202
 203         if (key)        CFRelease(key);
 204         return KERN_SUCCESS;
 205 }