Author: andreas Date: 2010-07-27 20:33:19 -0700 (Tue, 27 Jul 2010) New Revision: 2249

Added: trunk/platforms/win32/vm/sqWin32DnsInfo.c Modified: trunk/platforms/win32/vm/sqWin32Exports.c Log: Add primitive for obtaining Win32 DNS info.

Added: trunk/platforms/win32/vm/sqWin32DnsInfo.c =================================================================== --- trunk/platforms/win32/vm/sqWin32DnsInfo.c (rev 0) +++ trunk/platforms/win32/vm/sqWin32DnsInfo.c 2010-07-28 03:33:19 UTC (rev 2249) @@ -0,0 +1,180 @@ +/* sqWin32DnsInfo.c: Gathers DNS information on Windows */ +#include <windows.h> +#include "sq.h" + +extern struct VirtualMachine *interpreterProxy; +#define vm interpreterProxy + +/* Large enough since we only use 3 entries with 1k bytes max */ +static char buf[20000]; + +static char* lookup(HKEY hk, char *key) { + static char buf[1000]; + DWORD dwLength, ret; + dwLength = sizeof(buf)-1; + *buf = 0; + ret = RegQueryValueEx(hk, key, NULL, NULL, buf, &dwLength); + if(buf[dwLength]) buf[dwLength] = 0; + return buf; +} + +/* Print base registry info. Return true if DNS info was provided. */ +static int printBaseInfo(char *hkeyName) { + HKEY hkey; + DWORD ret; + char *value; + int result = 0; + + ret = RegOpenKeyEx(HKEY_LOCAL_MACHINE, hkeyName, 0, KEY_READ, &hkey); + if(ret != ERROR_SUCCESS) { + printf("RegOpenKeyEx failed\n"); + return vm->primitiveFail(); + } + + /* Provide Hostname info if present */ + value = lookup(hkey, "Hostname"); + if(*value) { + strcat(buf, "\nhostname "); + strcat(buf, value); + } + + /* The list of suffixes to search */ + value = lookup(hkey, "SearchList"); + if(*value) { + strcat(buf, "\nsearch "); + strcat(buf, value); + } + + /* The (static or dhcp) domain */ + value = lookup(hkey, "Domain"); + if(*value) { + strcat(buf, "\ndomain "); + strcat(buf, value); + strcat(buf, " # static"); + } else { + value = lookup(hkey, "DhcpDomain"); + if(*value) { + strcat(buf, "\ndomain "); + strcat(buf, value); + strcat(buf, " # dhcp"); + } + } + + /* And the (static or dhcp) configured name servers */ + value = lookup(hkey, "NameServer"); + if(*value) { + strcat(buf, "\nnameserver "); + strcat(buf, value); + strcat(buf, " # static"); + result = 1; + } else { + value = lookup(hkey, "DhcpNameServer"); + if(*value) { + strcat(buf, "\nnameserver "); + strcat(buf, value); + strcat(buf, " # dhcp"); + result = 1; + } + } + RegCloseKey(hkey); + return result; +} + +static int printAdapterInfo(char *hkeyName) { + HKEY hkey; + DWORD ret; + char *value; + int result = 0; + + ret = RegOpenKeyEx(HKEY_LOCAL_MACHINE, hkeyName, 0, KEY_READ, &hkey); + if(ret != ERROR_SUCCESS) { + printf("RegOpenKeyEx failed\n"); + return vm->primitiveFail(); + } + + /* Provide interface address if present */ + value = lookup(hkey, "IPAddress"); + if(*value && strcmp(value, "0.0.0.0")) { + /* Skip interface info since we can't currently determine + if the information is correct or a dead adapter */ +#if 0 + strcat(buf, "\ninterface "); + strcat(buf, value); + strcat(buf, " # static"); +#endif + } else { + value = lookup(hkey, "DhcpIPAddress"); + if(*value && strcmp(value, "0.0.0.0")) { +#if 0 + strcat(buf, "\ninterface "); + strcat(buf, value); + strcat(buf, " # dhcp "); +#endif + } else { + /* Don't bother printing DNS for an adapter w/o IP */ + goto done; + } + } + + /* Print the (static or dhcp) configured name servers */ + value = lookup(hkey, "NameServer"); + if(*value) { + strcat(buf, "\nnameserver "); + strcat(buf, value); + strcat(buf, " # static"); + result = 1; + } else { + value = lookup(hkey, "DhcpNameServer"); + if(*value) { + strcat(buf, "\nnameserver "); + strcat(buf, value); + strcat(buf, " # dhcp"); + result = 1; + } + } + + done: + RegCloseKey(hkey); + return result; +} + +#define BASE_KEY \ + "SYSTEM\CurrentControlSet\services\Tcpip\Parameters\Interfaces" + +EXPORT(int) primitiveDnsInfo(void) { + HKEY hkey; + sqInt sz, stroop; + DWORD dwLength, ret, index; + char adapter[1024]; + + *buf = 0; + strcat(buf, "# Global config settings"); + + /* Print the global options */ + if(printBaseInfo("SYSTEM\CurrentControlSet\services\Tcpip\Parameters")) { + /* If we have DNS entries globally, skip enumerating all interfaces */ + goto done; + } + + /* Enumerate the available interfaces */ + ret = RegOpenKeyEx(HKEY_LOCAL_MACHINE, BASE_KEY, 0, KEY_READ, &hkey); + for(index=0;;index++) { + char keyName[1024]; + dwLength = sizeof(adapter); + ret = RegEnumKeyEx(hkey, index, adapter, &dwLength, NULL, NULL, NULL, NULL); + if(ret != ERROR_SUCCESS) break; + strcpy(keyName, BASE_KEY); + strcat(keyName, "\"); + strcat(keyName, adapter); + /* Should look up and print the adapter name */ + printAdapterInfo(keyName); + } + + done: + sz = strlen(buf); + stroop = vm->instantiateClassindexableSize(vm->classString(), sz); + if(!vm->failed()) { + memcpy(vm->firstIndexableField(stroop), buf, sz); + vm->popthenPush(vm->methodArgumentCount()+1, stroop); + } +}

Modified: trunk/platforms/win32/vm/sqWin32Exports.c =================================================================== --- trunk/platforms/win32/vm/sqWin32Exports.c 2010-07-23 01:09:10 UTC (rev 2248) +++ trunk/platforms/win32/vm/sqWin32Exports.c 2010-07-28 03:33:19 UTC (rev 2249) @@ -11,6 +11,7 @@ int primitivePluginRequestFileHandle(void); int primitivePluginDestroyRequest(void); int primitivePluginRequestState(void); +int primitiveDnsInfo(void);

extern void* stWindow; extern void* firstMessageHook; @@ -29,6 +30,7 @@ {"","primitivePluginRequestFileHandle", primitivePluginRequestFileHandle}, {"","primitivePluginDestroyRequest", primitivePluginDestroyRequest}, {"","primitivePluginRequestState", primitivePluginRequestState}, + {"","primitiveDnsInfo", primitiveDnsInfo}, {"", "printf", printf}, {"","stWindow", &stWindow}, {"","firstMessageHook", &firstMessageHook},