#include <VBox/com/string.h>

#include <VBox/com/ptr.h>

#include <iprt/err.h>

#include <iprt/thread.h>

#include <iprt/semaphore.h>

#include <CoreFoundation/CoreFoundation.h>

#include <SystemConfiguration/SCDynamicStore.h>

#include <string>

#include <vector>

#include "../HostDnsService.h"

SCDynamicStoreRef g_store;

CFRunLoopSourceRef g_DnsWatcher;

CFRunLoopRef g_RunLoopRef;

RTTHREAD g_DnsMonitoringThread;

RTSEMEVENT g_DnsInitEvent;

static const CFStringRef kStateNetworkGlobalDNSKey = CFSTR("State:/Network/Global/DNS");

static int hostMonitoringRoutine(RTTHREAD ThreadSelf, void *pvUser)

{

NOREF(ThreadSelf);

NOREF(pvUser);

g_RunLoopRef = CFRunLoopGetCurrent();

AssertReturn(g_RunLoopRef, VERR_INTERNAL_ERROR);

CFRetain(g_RunLoopRef);

CFArrayRef watchingArrayRef = CFArrayCreate(NULL,

(const void **)&kStateNetworkGlobalDNSKey,

1, &kCFTypeArrayCallBacks);

if (!watchingArrayRef)

{

CFRelease(g_DnsWatcher);

return E_OUTOFMEMORY;

}

if(SCDynamicStoreSetNotificationKeys(g_store, watchingArrayRef, NULL))

CFRunLoopAddSource(CFRunLoopGetCurrent(), g_DnsWatcher, kCFRunLoopCommonModes);

CFRelease(watchingArrayRef);

RTSemEventSignal(g_DnsInitEvent);

CFRunLoopRun();

CFRelease(g_RunLoopRef);

return VINF_SUCCESS;

}

HostDnsServiceDarwin::HostDnsServiceDarwin(){}

HostDnsServiceDarwin::~HostDnsServiceDarwin()

{

if (g_RunLoopRef)

CFRunLoopStop(g_RunLoopRef);

CFRelease(g_DnsWatcher);

CFRelease(g_store);

}

void HostDnsServiceDarwin::hostDnsServiceStoreCallback(void *arg0, void *arg1, void *info)

{

HostDnsServiceDarwin *pThis = (HostDnsServiceDarwin *)info;

NOREF(arg0); /* SCDynamicStore */

NOREF(arg1); /* CFArrayRef */

ALock l(pThis);

pThis->updateInfo();

pThis->notifyAll();

}

HRESULT HostDnsServiceDarwin::init()

{

SCDynamicStoreContext ctx;

RT_ZERO(ctx);

ctx.info = this;

g_store = SCDynamicStoreCreate(NULL, CFSTR("org.virtualbox.VBoxSVC"),

(SCDynamicStoreCallBack)HostDnsServiceDarwin::hostDnsServiceStoreCallback,

&ctx);

AssertReturn(g_store, E_FAIL);

g_DnsWatcher = SCDynamicStoreCreateRunLoopSource(NULL, g_store, 0);

if (!g_DnsWatcher)

return E_OUTOFMEMORY;

HRESULT hrc = HostDnsMonitor::init();

AssertComRCReturn(hrc, hrc);

int rc = RTSemEventCreate(&g_DnsInitEvent);

AssertRCReturn(rc, E_FAIL);

rc = RTThreadCreate(&g_DnsMonitoringThread, hostMonitoringRoutine,

this, 128 * _1K, RTTHREADTYPE_IO, 0, "dns-monitor");

AssertRCReturn(rc, E_FAIL);

RTSemEventWait(g_DnsInitEvent, RT_INDEFINITE_WAIT);

return updateInfo();

}

HRESULT HostDnsServiceDarwin::updateInfo()

{

CFPropertyListRef propertyRef = SCDynamicStoreCopyValue(g_store,

kStateNetworkGlobalDNSKey);

/**

if (!propertyRef)

return S_OK;

HostDnsInformation info;

CFStringRef domainNameRef = (CFStringRef)CFDictionaryGetValue(

static_cast<CFDictionaryRef>(propertyRef), CFSTR("DomainName"));

if (domainNameRef)

{

const char *pszDomainName = CFStringGetCStringPtr(domainNameRef,

CFStringGetSystemEncoding());

if (pszDomainName)

info.domain = pszDomainName;

}

int i, arrayCount;

CFArrayRef serverArrayRef = (CFArrayRef)CFDictionaryGetValue(

static_cast<CFDictionaryRef>(propertyRef), CFSTR("ServerAddresses"));

if (serverArrayRef)

{

arrayCount = CFArrayGetCount(serverArrayRef);

for (i = 0; i < arrayCount; ++i)

{

CFStringRef serverAddressRef = (CFStringRef)CFArrayGetValueAtIndex(serverArrayRef, i);

if (!serverArrayRef)

continue;

const char *pszServerAddress = CFStringGetCStringPtr(serverAddressRef,

CFStringGetSystemEncoding());

if (!pszServerAddress)

continue;

info.servers.push_back(std::string(pszServerAddress));

}

}

CFArrayRef searchArrayRef = (CFArrayRef)CFDictionaryGetValue(

static_cast<CFDictionaryRef>(propertyRef), CFSTR("SearchDomains"));

if (searchArrayRef)

{

arrayCount = CFArrayGetCount(searchArrayRef);

for (i = 0; i < arrayCount; ++i)

{

CFStringRef searchStringRef = (CFStringRef)CFArrayGetValueAtIndex(searchArrayRef, i);

if (!searchArrayRef)

continue;

const char *pszSearchString = CFStringGetCStringPtr(searchStringRef,

CFStringGetSystemEncoding());

if (!pszSearchString)

continue;

info.searchList.push_back(std::string(pszSearchString));

}

}

CFRelease(propertyRef);

setInfo(info);

return S_OK;

}