#include <SystemConfiguration/SystemConfiguration.h>
#include <SystemConfiguration/SCValidation.h>
#include <SystemConfiguration/SCPrivate.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <net/if.h>
#include <net/if_dl.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <mach/mach.h>
#include <mach/notify.h>
#include <mach/mach_error.h>
#include <pthread.h>
#define N_QUICK 32
char *
_SC_cfstring_to_cstring(CFStringRef cfstr, char *buf, int bufLen, CFStringEncoding encoding)
{
CFIndex last;
CFIndex len = CFStringGetLength(cfstr);
(void)CFStringGetBytes(cfstr,
CFRangeMake(0, len),
encoding,
0,
FALSE,
NULL,
0,
&len);
if (buf) {
if (bufLen < (len + 1)) {
return NULL;
}
} else {
bufLen = len + 1;
buf = CFAllocatorAllocate(NULL, bufLen, 0);
if (!buf) {
return NULL;
}
}
(void)CFStringGetBytes(cfstr,
CFRangeMake(0, len),
encoding,
0,
FALSE,
(UInt8 *)buf,
bufLen,
&last);
buf[last] = '\0';
return buf;
}
void
_SC_sockaddr_to_string(const struct sockaddr *address, char *buf, size_t bufLen)
{
bzero(buf, bufLen);
switch (address->sa_family) {
case AF_INET :
(void)inet_ntop(((struct sockaddr_in *)address)->sin_family,
&((struct sockaddr_in *)address)->sin_addr,
buf,
bufLen);
break;
case AF_INET6 : {
(void)inet_ntop(((struct sockaddr_in6 *)address)->sin6_family,
&((struct sockaddr_in6 *)address)->sin6_addr,
buf,
bufLen);
if (((struct sockaddr_in6 *)address)->sin6_scope_id != 0) {
int n;
n = strlen(buf);
if ((n+IF_NAMESIZE+1) <= (int)bufLen) {
buf[n++] = '%';
if_indextoname(((struct sockaddr_in6 *)address)->sin6_scope_id, &buf[n]);
}
}
break;
}
case AF_LINK :
if (((struct sockaddr_dl *)address)->sdl_len < bufLen) {
bufLen = ((struct sockaddr_dl *)address)->sdl_len;
} else {
bufLen = bufLen - 1;
}
bcopy(((struct sockaddr_dl *)address)->sdl_data, buf, bufLen);
break;
default :
snprintf(buf, bufLen, "unexpected address family %d", address->sa_family);
break;
}
return;
}
Boolean
_SCSerialize(CFPropertyListRef obj, CFDataRef *xml, void **dataRef, CFIndex *dataLen)
{
CFDataRef myXml;
CFWriteStreamRef stream;
if (!xml && !(dataRef && dataLen)) {
return FALSE;
}
stream = CFWriteStreamCreateWithAllocatedBuffers(NULL, NULL);
CFWriteStreamOpen(stream);
CFPropertyListWriteToStream(obj, stream, kCFPropertyListBinaryFormat_v1_0, NULL);
CFWriteStreamClose(stream);
myXml = CFWriteStreamCopyProperty(stream, kCFStreamPropertyDataWritten);
CFRelease(stream);
if (!myXml) {
SCLog(TRUE, LOG_ERR, CFSTR("_SCSerialize() failed"));
if (xml) *xml = NULL;
if (dataRef) *dataRef = NULL;
if (dataLen) *dataLen = 0;
return FALSE;
}
if (xml) {
*xml = myXml;
if (dataRef) {
*dataRef = (void *)CFDataGetBytePtr(myXml);
}
if (dataLen) {
*dataLen = CFDataGetLength(myXml);
}
} else {
kern_return_t status;
*dataLen = CFDataGetLength(myXml);
status = vm_allocate(mach_task_self(), (void *)dataRef, *dataLen, TRUE);
if (status != KERN_SUCCESS) {
SCLog(TRUE, LOG_ERR, CFSTR("_SCSerialize(): %s"), mach_error_string(status));
CFRelease(myXml);
*dataRef = NULL;
*dataLen = 0;
return FALSE;
}
bcopy((char *)CFDataGetBytePtr(myXml), *dataRef, *dataLen);
CFRelease(myXml);
}
return TRUE;
}
Boolean
_SCUnserialize(CFPropertyListRef *obj, CFDataRef xml, void *dataRef, CFIndex dataLen)
{
CFStringRef xmlError;
if (!xml) {
kern_return_t status;
xml = CFDataCreateWithBytesNoCopy(NULL, (void *)dataRef, dataLen, kCFAllocatorNull);
*obj = CFPropertyListCreateFromXMLData(NULL,
xml,
kCFPropertyListImmutable,
&xmlError);
CFRelease(xml);
status = vm_deallocate(mach_task_self(), (vm_address_t)dataRef, dataLen);
if (status != KERN_SUCCESS) {
SCLog(_sc_verbose, LOG_DEBUG, CFSTR("_SCUnserialize(): %s"), mach_error_string(status));
}
} else {
*obj = CFPropertyListCreateFromXMLData(NULL,
xml,
kCFPropertyListImmutable,
&xmlError);
}
if (*obj == NULL) {
if (xmlError) {
SCLog(TRUE, LOG_ERR, CFSTR("_SCUnserialize(): %@"), xmlError);
CFRelease(xmlError);
}
_SCErrorSet(kSCStatusFailed);
return FALSE;
}
return TRUE;
}
Boolean
_SCSerializeString(CFStringRef str, CFDataRef *data, void **dataRef, CFIndex *dataLen)
{
CFDataRef myData;
if (!isA_CFString(str)) {
return FALSE;
}
if (!data && !(dataRef && dataLen)) {
return FALSE;
}
myData = CFStringCreateExternalRepresentation(NULL, str, kCFStringEncodingUTF8, 0);
if (!myData) {
SCLog(TRUE, LOG_ERR, CFSTR("_SCSerializeString() failed"));
if (data) *data = NULL;
if (dataRef) *dataRef = NULL;
if (dataLen) *dataLen = 0;
return FALSE;
}
if (data) {
*data = myData;
if (dataRef) {
*dataRef = (void *)CFDataGetBytePtr(myData);
}
if (dataLen) {
*dataLen = CFDataGetLength(myData);
}
} else {
kern_return_t status;
*dataLen = CFDataGetLength(myData);
status = vm_allocate(mach_task_self(), (void *)dataRef, *dataLen, TRUE);
if (status != KERN_SUCCESS) {
SCLog(TRUE, LOG_ERR, CFSTR("_SCSerializeString(): %s"), mach_error_string(status));
CFRelease(myData);
*dataRef = NULL;
*dataLen = 0;
return FALSE;
}
bcopy((char *)CFDataGetBytePtr(myData), *dataRef, *dataLen);
CFRelease(myData);
}
return TRUE;
}
Boolean
_SCUnserializeString(CFStringRef *str, CFDataRef utf8, void *dataRef, CFIndex dataLen)
{
if (!utf8) {
kern_return_t status;
utf8 = CFDataCreateWithBytesNoCopy(NULL, dataRef, dataLen, kCFAllocatorNull);
*str = CFStringCreateFromExternalRepresentation(NULL, utf8, kCFStringEncodingUTF8);
CFRelease(utf8);
status = vm_deallocate(mach_task_self(), (vm_address_t)dataRef, dataLen);
if (status != KERN_SUCCESS) {
SCLog(_sc_verbose, LOG_DEBUG, CFSTR("_SCUnserializeString(): %s"), mach_error_string(status));
}
} else {
*str = CFStringCreateFromExternalRepresentation(NULL, utf8, kCFStringEncodingUTF8);
}
if (*str == NULL) {
SCLog(TRUE, LOG_ERR, CFSTR("_SCUnserializeString() failed"));
return FALSE;
}
return TRUE;
}
Boolean
_SCSerializeData(CFDataRef data, void **dataRef, CFIndex *dataLen)
{
kern_return_t status;
if (!isA_CFData(data)) {
return FALSE;
}
*dataLen = CFDataGetLength(data);
status = vm_allocate(mach_task_self(), (void *)dataRef, *dataLen, TRUE);
if (status != KERN_SUCCESS) {
SCLog(TRUE, LOG_ERR, CFSTR("_SCSerializeData(): %s"), mach_error_string(status));
*dataRef = NULL;
*dataLen = 0;
return FALSE;
}
bcopy((char *)CFDataGetBytePtr(data), *dataRef, *dataLen);
return TRUE;
}
Boolean
_SCUnserializeData(CFDataRef *data, void *dataRef, CFIndex dataLen)
{
kern_return_t status;
*data = CFDataCreate(NULL, dataRef, dataLen);
status = vm_deallocate(mach_task_self(), (vm_address_t)dataRef, dataLen);
if (status != KERN_SUCCESS) {
SCLog(_sc_verbose, LOG_DEBUG, CFSTR("_SCUnserializeData(): %s"), mach_error_string(status));
_SCErrorSet(kSCStatusFailed);
return FALSE;
}
return TRUE;
}
CFDictionaryRef
_SCSerializeMultiple(CFDictionaryRef dict)
{
const void * keys_q[N_QUICK];
const void ** keys = keys_q;
CFIndex nElements;
CFDictionaryRef newDict = NULL;
const void * pLists_q[N_QUICK];
const void ** pLists = pLists_q;
const void * values_q[N_QUICK];
const void ** values = values_q;
nElements = CFDictionaryGetCount(dict);
if (nElements > 0) {
CFIndex i;
if (nElements > (CFIndex)(sizeof(keys_q) / sizeof(CFTypeRef))) {
keys = CFAllocatorAllocate(NULL, nElements * sizeof(CFTypeRef), 0);
values = CFAllocatorAllocate(NULL, nElements * sizeof(CFTypeRef), 0);
pLists = CFAllocatorAllocate(NULL, nElements * sizeof(CFTypeRef), 0);
}
bzero(pLists, nElements * sizeof(CFTypeRef));
CFDictionaryGetKeysAndValues(dict, keys, values);
for (i = 0; i < nElements; i++) {
if (!_SCSerialize((CFPropertyListRef)values[i], (CFDataRef *)&pLists[i], NULL, NULL)) {
goto done;
}
}
}
newDict = CFDictionaryCreate(NULL,
keys,
pLists,
nElements,
&kCFTypeDictionaryKeyCallBacks,
&kCFTypeDictionaryValueCallBacks);
done :
if (nElements > 0) {
CFIndex i;
for (i = 0; i < nElements; i++) {
if (pLists[i]) CFRelease(pLists[i]);
}
if (keys != keys_q) {
CFAllocatorDeallocate(NULL, keys);
CFAllocatorDeallocate(NULL, values);
CFAllocatorDeallocate(NULL, pLists);
}
}
return newDict;
}
CFDictionaryRef
_SCUnserializeMultiple(CFDictionaryRef dict)
{
const void * keys_q[N_QUICK];
const void ** keys = keys_q;
CFIndex nElements;
CFDictionaryRef newDict = NULL;
const void * pLists_q[N_QUICK];
const void ** pLists = pLists_q;
const void * values_q[N_QUICK];
const void ** values = values_q;
nElements = CFDictionaryGetCount(dict);
if (nElements > 0) {
CFIndex i;
if (nElements > (CFIndex)(sizeof(keys_q) / sizeof(CFTypeRef))) {
keys = CFAllocatorAllocate(NULL, nElements * sizeof(CFTypeRef), 0);
values = CFAllocatorAllocate(NULL, nElements * sizeof(CFTypeRef), 0);
pLists = CFAllocatorAllocate(NULL, nElements * sizeof(CFTypeRef), 0);
}
bzero(pLists, nElements * sizeof(CFTypeRef));
CFDictionaryGetKeysAndValues(dict, keys, values);
for (i = 0; i < nElements; i++) {
if (!_SCUnserialize((CFPropertyListRef *)&pLists[i], values[i], NULL, 0)) {
goto done;
}
}
}
newDict = CFDictionaryCreate(NULL,
keys,
pLists,
nElements,
&kCFTypeDictionaryKeyCallBacks,
&kCFTypeDictionaryValueCallBacks);
done :
if (nElements > 0) {
CFIndex i;
for (i = 0; i < nElements; i++) {
if (pLists[i]) CFRelease(pLists[i]);
}
if (keys != keys_q) {
CFAllocatorDeallocate(NULL, keys);
CFAllocatorDeallocate(NULL, values);
CFAllocatorDeallocate(NULL, pLists);
}
}
return newDict;
}
__private_extern__ void
_SC_signalRunLoop(CFTypeRef obj, CFRunLoopSourceRef rls, CFArrayRef rlList)
{
CFRunLoopRef rl = NULL;
CFRunLoopRef rl1 = NULL;
CFIndex i;
CFIndex n = CFArrayGetCount(rlList);
if (n == 0) {
return;
}
for (i = 0; i < n; i += 3) {
if (!CFEqual(obj, CFArrayGetValueAtIndex(rlList, i))) {
continue;
}
rl1 = (CFRunLoopRef)CFArrayGetValueAtIndex(rlList, i+1);
break;
}
if (!rl1) {
return;
}
rl = rl1;
for (i = i+3; i < n; i += 3) {
CFRunLoopRef rl2;
if (!CFEqual(obj, CFArrayGetValueAtIndex(rlList, i))) {
continue;
}
rl2 = (CFRunLoopRef)CFArrayGetValueAtIndex(rlList, i+1);
if (!CFEqual(rl1, rl2)) {
rl = NULL;
break;
}
}
if (rl) {
CFRunLoopWakeUp(rl);
return;
}
for (i = 0; i < n; i+=3) {
CFStringRef rlMode;
if (!CFEqual(obj, CFArrayGetValueAtIndex(rlList, i))) {
continue;
}
rl = (CFRunLoopRef)CFArrayGetValueAtIndex(rlList, i+1);
rlMode = CFRunLoopCopyCurrentMode(rl);
if (rlMode && CFRunLoopIsWaiting(rl) && CFRunLoopContainsSource(rl, rls, rlMode)) {
CFRelease(rlMode);
CFRunLoopWakeUp(rl);
return;
}
if (rlMode) CFRelease(rlMode);
}
CFRunLoopWakeUp(rl1);
return;
}
__private_extern__ Boolean
_SC_isScheduled(CFTypeRef obj, CFRunLoopRef runLoop, CFStringRef runLoopMode, CFMutableArrayRef rlList)
{
CFIndex i;
CFIndex n = CFArrayGetCount(rlList);
for (i = 0; i < n; i += 3) {
if (obj && !CFEqual(obj, CFArrayGetValueAtIndex(rlList, i))) {
continue;
}
if (runLoop && !CFEqual(runLoop, CFArrayGetValueAtIndex(rlList, i+1))) {
continue;
}
if (runLoopMode && !CFEqual(runLoopMode, CFArrayGetValueAtIndex(rlList, i+2))) {
continue;
}
return TRUE;
}
return FALSE;
}
__private_extern__ void
_SC_schedule(CFTypeRef obj, CFRunLoopRef runLoop, CFStringRef runLoopMode, CFMutableArrayRef rlList)
{
CFArrayAppendValue(rlList, obj);
CFArrayAppendValue(rlList, runLoop);
CFArrayAppendValue(rlList, runLoopMode);
return;
}
__private_extern__ Boolean
_SC_unschedule(CFTypeRef obj, CFRunLoopRef runLoop, CFStringRef runLoopMode, CFMutableArrayRef rlList, Boolean all)
{
CFIndex i = 0;
Boolean found = FALSE;
CFIndex n = CFArrayGetCount(rlList);
while (i < n) {
if (obj && !CFEqual(obj, CFArrayGetValueAtIndex(rlList, i))) {
i += 3;
continue;
}
if (runLoop && !CFEqual(runLoop, CFArrayGetValueAtIndex(rlList, i+1))) {
i += 3;
continue;
}
if (runLoopMode && !CFEqual(runLoopMode, CFArrayGetValueAtIndex(rlList, i+2))) {
i += 3;
continue;
}
found = TRUE;
CFArrayRemoveValueAtIndex(rlList, i + 2);
CFArrayRemoveValueAtIndex(rlList, i + 1);
CFArrayRemoveValueAtIndex(rlList, i);
if (!all) {
return found;
}
n -= 3;
}
return found;
}
void
__showMachPortStatus()
{
#ifdef DEBUG
if (_sc_debug) {
kern_return_t status;
mach_port_name_array_t ports;
mach_port_type_array_t types;
int pi, pn, tn;
CFMutableStringRef str;
SCLog(_sc_verbose, LOG_DEBUG, CFSTR("----------"));
status = mach_port_names(mach_task_self(), &ports, &pn, &types, &tn);
if (status == MACH_MSG_SUCCESS) {
str = CFStringCreateMutable(NULL, 0);
for (pi = 0; pi < pn; pi++) {
char rights[16], *rp = &rights[0];
if (types[pi] != MACH_PORT_TYPE_NONE) {
*rp++ = ' ';
*rp++ = '(';
if (types[pi] & MACH_PORT_TYPE_SEND)
*rp++ = 'S';
if (types[pi] & MACH_PORT_TYPE_RECEIVE)
*rp++ = 'R';
if (types[pi] & MACH_PORT_TYPE_SEND_ONCE)
*rp++ = 'O';
if (types[pi] & MACH_PORT_TYPE_PORT_SET)
*rp++ = 'P';
if (types[pi] & MACH_PORT_TYPE_DEAD_NAME)
*rp++ = 'D';
*rp++ = ')';
}
*rp = '\0';
CFStringAppendFormat(str, NULL, CFSTR(" %d%s"), ports[pi], rights);
}
SCLog(_sc_verbose, LOG_DEBUG, CFSTR("Task ports (n=%d):%@"), pn, str);
CFRelease(str);
} else {
SCLog(_sc_verbose, LOG_DEBUG, CFSTR("mach_port_names(): %s"), mach_error_string(status));
}
}
#endif
return;
}
void
__showMachPortReferences(mach_port_t port)
{
#ifdef DEBUG
kern_return_t status;
mach_port_urefs_t refs_send = 0;
mach_port_urefs_t refs_recv = 0;
mach_port_urefs_t refs_once = 0;
mach_port_urefs_t refs_pset = 0;
mach_port_urefs_t refs_dead = 0;
SCLog(_sc_verbose, LOG_DEBUG, CFSTR("user references for mach port %d"), port);
status = mach_port_get_refs(mach_task_self(), port, MACH_PORT_RIGHT_SEND, &refs_send);
if (status != KERN_SUCCESS) {
SCLog(_sc_verbose, LOG_DEBUG, CFSTR(" mach_port_get_refs(MACH_PORT_RIGHT_SEND): %s"), mach_error_string(status));
return;
}
status = mach_port_get_refs(mach_task_self(), port, MACH_PORT_RIGHT_RECEIVE, &refs_recv);
if (status != KERN_SUCCESS) {
SCLog(_sc_verbose, LOG_DEBUG, CFSTR(" mach_port_get_refs(MACH_PORT_RIGHT_RECEIVE): %s"), mach_error_string(status));
return;
}
status = mach_port_get_refs(mach_task_self(), port, MACH_PORT_RIGHT_SEND_ONCE, &refs_once);
if (status != KERN_SUCCESS) {
SCLog(_sc_verbose, LOG_DEBUG, CFSTR(" mach_port_get_refs(MACH_PORT_RIGHT_SEND_ONCE): %s"), mach_error_string(status));
return;
}
status = mach_port_get_refs(mach_task_self(), port, MACH_PORT_RIGHT_PORT_SET, &refs_pset);
if (status != KERN_SUCCESS) {
SCLog(_sc_verbose, LOG_DEBUG, CFSTR(" mach_port_get_refs(MACH_PORT_RIGHT_PORT_SET): %s"), mach_error_string(status));
return;
}
status = mach_port_get_refs(mach_task_self(), port, MACH_PORT_RIGHT_DEAD_NAME, &refs_dead);
if (status != KERN_SUCCESS) {
SCLog(_sc_verbose, LOG_DEBUG, CFSTR(" mach_port_get_refs(MACH_PORT_RIGHT_DEAD_NAME): %s"), mach_error_string(status));
return;
}
SCLog(_sc_verbose, LOG_DEBUG,
CFSTR(" send = %d, receive = %d, send once = %d, port set = %d, dead name = %d"),
refs_send,
refs_recv,
refs_once,
refs_pset,
refs_dead);
#endif
return;
}