src-server/win/HostDnsServiceWin.cpp

	HostDnsServiceWin.cpp revision 3f2299591cfeb8a37a3578bef9c44c1790860948
/* -*- indent-tabs-mode: nil; -*- */
#include <VBox/com/string.h>
#include <VBox/com/ptr.h>

#include "../HostDnsService.h"

#include <iprt/assert.h>
#include <iprt/err.h>

#include <Windows.h>

static HKEY g_hKeyTcpipParameters;

HostDnsServiceWin::HostDnsServiceWin()
{
    RegOpenKeyEx(HKEY_LOCAL_MACHINE,
                 TEXT("SYSTEM\\CurrentControlSet\\Services\\Tcpip\\Parameters"),
                 0, KEY_READ, &g_hKeyTcpipParameters);
}


HostDnsServiceWin::~HostDnsServiceWin()
{
    if (!g_hKeyTcpipParameters)
    {
        RegCloseKey(g_hKeyTcpipParameters);
        g_hKeyTcpipParameters = 0;
    }
}


HRESULT HostDnsServiceWin::init()
{
    HRESULT hrc = HostDnsMonitor::init(aParent);
    AssertComRCReturn(hrc, hrc);

    hrc = update();
    AssertComRCReturn(hrc, hrc);

    return S_OK;
}


HRESULT HostDnsServiceWin::update()
{
    HRESULT hrc;
    DWORD regIndex;
    BYTE abDomain[256];
    BYTE abNameServers[256];
    BYTE abSearchList[256];

    regIndex = 0;
    do {
        CHAR keyName[256];
        DWORD cbKeyName = 256;
        DWORD keyType = 0;
        BYTE keyData[1024];
        DWORD cbKeyData = 1024;

        hrc = RegEnumValueA(g_hKeyTcpipParameters, regIndex, keyName, &cbKeyName, 0,
                            &keyType, keyData, &cbKeyData);
        if (   hrc == ERROR_SUCCESS
            || hrc == ERROR_MORE_DATA)
        {
            if (   RTStrICmp("Domain", keyName) == 0
                && cbKeyData > 1
                && cbKeyData < 256)
                memcpy(abDomain, keyData, cbKeyData);

            else if (   RTStrICmp("DhcpDomain", keyName) == 0
                     && cbKeyData > 1
                     && abDomain[0] == 0
                     && cbKeyData < 256)
                memcpy(abDomain, keyData, cbKeyData);

            else if (   RTStrICmp("NameServer", keyName) == 0
                     && cbKeyData > 1
                     && cbKeyData < 256)
                memcpy(abNameServers, keyData, cbKeyData);

            else if (   RTStrICmp("DhcpNameServer", keyName) == 0
                     && cbKeyData > 1
                     && abNameServers[0] == 0
                     && cbKeyData < 256)
                memcpy(abNameServers, keyData, cbKeyData);

            else if (   RTStrICmp("SearchList", keyName) == 0
                     && cbKeyData > 1
                     && cbKeyData < 256)
              memcpy(abSearchList, keyData, cbKeyData);
        }
        regIndex++;
    } while (hrc != ERROR_NO_MORE_ITEMS);

    /* OK, now parse and update DNS structures. */
    /* domain name */
    HostDnsInformation info;
    info.domain = static_cast<char*>(abDomain);

    /* server list */
    strList2List(info.servers, static_cast<char *>(abNameServers));
    /* search list */
    strList2List(info.searchList, static_cast<char *>(abSearchList));

    setInfo(info);

    return S_OK;
}


void HostDnsServiceWin::strList2List(std::vector<std::string>& lst, char *strLst)
{
    char *next, *current;
    char address[512];

    AssertPtrReturnVoid(strLst);

    if (strlen(strLst) == 0)
      return;

    current = strLst;
    do {
        RT_ZERO(address);
        next = RTStrStr(current, " ");

        if (next)
          strncpy(address, current, next - current);
        else
          strcpy(address, current);

        lst.push_back(std::string(address));

        current = next + 1;
    } while(next);

}