#include "configd.h"
#include "session.h"
#include "pattern.h"
#define N_QUICK 64
__private_extern__
int
__SCDynamicStoreCopyKeyList(SCDynamicStoreRef store, CFStringRef key, Boolean isRegex, CFArrayRef *subKeys)
{
SCDynamicStorePrivateRef storePrivate = (SCDynamicStorePrivateRef)store;
CFMutableArrayRef keyArray;
regex_t preg;
CFIndex storeCnt;
CFStringRef storeStr;
CFDictionaryRef storeValue;
if (_configd_verbose) {
SCLog(TRUE, LOG_DEBUG, CFSTR("__SCDynamicStoreCopyKeyList:"));
SCLog(TRUE, LOG_DEBUG, CFSTR(" key = %@"), key);
SCLog(TRUE, LOG_DEBUG, CFSTR(" isRegex = %s"), isRegex ? "TRUE" : "FALSE");
}
if (!store || (storePrivate->server == MACH_PORT_NULL)) {
return kSCStatusNoStoreSession;
}
storeCnt = CFDictionaryGetCount(storeData);
keyArray = CFArrayCreateMutable(NULL, storeCnt, &kCFTypeArrayCallBacks);
if (isRegex) {
CFStringRef reErrStr;
if (!patternCompile(key, &preg, &reErrStr)) {
CFArrayAppendValue(keyArray, reErrStr);
CFRelease(reErrStr);
*subKeys = CFArrayCreateCopy(NULL, keyArray);
CFRelease(keyArray);
return kSCStatusFailed;
}
}
if (storeCnt > 0) {
int i;
const void * storeKeys_q[N_QUICK];
const void ** storeKeys = storeKeys_q;
const void * storeValues_q[N_QUICK];
const void ** storeValues = storeValues_q;
if (storeCnt > (CFIndex)(sizeof(storeKeys_q) / sizeof(CFStringRef))) {
storeKeys = CFAllocatorAllocate(NULL, storeCnt * sizeof(CFStringRef), 0);
storeValues = CFAllocatorAllocate(NULL, storeCnt * sizeof(CFStringRef), 0);
}
CFDictionaryGetKeysAndValues(storeData, storeKeys, storeValues);
for (i = 0; i < storeCnt; i++) {
storeStr = (CFStringRef)storeKeys[i];
storeValue = (CFDictionaryRef)storeValues[i];
if (isRegex) {
int reError;
char storeKey_q[256];
char * storeKey = storeKey_q;
CFIndex storeKeyLen = CFStringGetLength(storeStr) + 1;
if (storeKeyLen > (CFIndex)sizeof(storeKey_q))
storeKey = CFAllocatorAllocate(NULL, storeKeyLen, 0);
if (_SC_cfstring_to_cstring(storeStr, storeKey, storeKeyLen, kCFStringEncodingASCII) == NULL) {
SCLog(_configd_verbose, LOG_DEBUG, CFSTR("could not convert store key to C string"));
if (storeKey != storeKey_q) CFAllocatorDeallocate(NULL, storeKey);
continue;
}
reError = regexec(&preg, storeKey, 0, NULL, 0);
switch (reError) {
case 0 :
if (CFDictionaryContainsKey(storeValue, kSCDData))
CFArrayAppendValue(keyArray, storeStr);
break;
case REG_NOMATCH :
break;
default : {
char reErrBuf[256];
int reErrStrLen;
reErrStrLen = regerror(reError,
&preg,
reErrBuf,
sizeof(reErrBuf));
SCLog(_configd_verbose, LOG_DEBUG, CFSTR("regexec(): %s"), reErrBuf);
break;
}
}
if (storeKey != storeKey_q) CFAllocatorDeallocate(NULL, storeKey);
} else {
if (((CFStringGetLength(key) == 0) || CFStringHasPrefix(storeStr, key)) &&
CFDictionaryContainsKey(storeValue, kSCDData)) {
CFArrayAppendValue(keyArray, storeStr);
}
}
}
if (storeKeys != storeKeys_q) {
CFAllocatorDeallocate(NULL, storeKeys);
CFAllocatorDeallocate(NULL, storeValues);
}
}
if (isRegex) {
regfree(&preg);
}
*subKeys = keyArray;
return kSCStatusOK;
}
__private_extern__
kern_return_t
_configlist(mach_port_t server,
xmlData_t keyRef,
mach_msg_type_number_t keyLen,
int isRegex,
xmlDataOut_t *listRef,
mach_msg_type_number_t *listLen,
int *sc_status
)
{
CFStringRef key;
serverSessionRef mySession = getSession(server);
Boolean ok;
CFArrayRef subKeys;
if (_configd_verbose) {
SCLog(TRUE, LOG_DEBUG, CFSTR("List keys in configuration database."));
SCLog(TRUE, LOG_DEBUG, CFSTR(" server = %d"), server);
}
*listRef = NULL;
*listLen = 0;
if (!_SCUnserializeString(&key, NULL, (void *)keyRef, keyLen)) {
*sc_status = kSCStatusFailed;
return KERN_SUCCESS;
}
if (!isA_CFString(key)) {
*sc_status = kSCStatusInvalidArgument;
CFRelease(key);
return KERN_SUCCESS;
}
if (!mySession) {
*sc_status = kSCStatusNoStoreSession;
CFRelease(key);
return KERN_SUCCESS;
}
*sc_status = __SCDynamicStoreCopyKeyList(mySession->store, key, isRegex != 0, &subKeys);
CFRelease(key);
if (*sc_status != kSCStatusOK) {
return KERN_SUCCESS;
}
ok = _SCSerialize(subKeys, NULL, (void **)listRef, (CFIndex *)listLen);
CFRelease(subKeys);
if (!ok) {
*sc_status = kSCStatusFailed;
return KERN_SUCCESS;
}
return KERN_SUCCESS;
}