- Fix two minor warnings here (catch (ModuleException& e) -> catch (ModuleException...

[user/henk/code/inspircd.git] / win / inspircd_win32wrapper.cpp

diff --git a/win/inspircd_win32wrapper.cpp b/win/inspircd_win32wrapper.cpp
index ef04b4fc86935bb3f71fd367d6b804e24da6d8b1..1cdd46444096c2b981a81fca4375328728fce334 100644 (file)
--- a/win/inspircd_win32wrapper.cpp
+++ b/win/inspircd_win32wrapper.cpp
@@ -9,6 +9,7 @@ using namespace std;
 #define INADDR_NONE 0xffffffff\r
 #endif\r
 \r
+\r
 HANDLE hIPCPipe;\r
 \r
 int inet_aton(const char *cp, struct in_addr *addr)\r
@@ -209,7 +210,7 @@ int printf_c(const char * format, ...)
                        \r
                        temp[t] = 0;\r
                        p++;\r
-                       if (!stricmp(temp, "[0"))\r
+                       if (!_stricmp(temp, "[0"))\r
                        {\r
                                // Returning to normal colour.\r
                                SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE);\r
@@ -366,3 +367,74 @@ void CloseIPC()
        CloseHandle(hIPCPipe);\r
 }\r
 \r
+\r
+/* These three functions were created from looking at how ares does it\r
+ * (...and they look far tidier in C++)\r
+ */\r
+\r
+/* Get active nameserver */\r
+bool GetNameServer(HKEY regkey, const char *key, char* &output)\r
+{\r
+       DWORD size = 0;\r
+       DWORD result = RegQueryValueEx(regkey, key, 0, NULL, NULL, &size);\r
+       if (((result != ERROR_SUCCESS) && (result != ERROR_MORE_DATA)) || (!size))\r
+               return false;\r
+\r
+       output = new char[size+1];\r
+\r
+       if ((RegQueryValueEx(regkey, key, 0, NULL, (LPBYTE)output, &size) != ERROR_SUCCESS) || (!*output))\r
+       {\r
+               delete output;\r
+               return false;\r
+       }\r
+       return true;\r
+}\r
+\r
+/* Check a network interface for its nameserver */\r
+bool GetInterface(HKEY regkey, const char *key, char* &output)\r
+{\r
+       char buf[39];\r
+       DWORD size = 39;\r
+       int idx = 0;\r
+       HKEY top;\r
+\r
+       while (RegEnumKeyEx(regkey, idx++, buf, &size, 0, NULL, NULL, NULL) != ERROR_NO_MORE_ITEMS)\r
+       {\r
+               size = 39;\r
+               if (RegOpenKeyEx(regkey, buf, 0, KEY_QUERY_VALUE, &top) != ERROR_SUCCESS)\r
+                       continue;\r
+               int rc = GetNameServer(top, key, output);\r
+               RegCloseKey(top);\r
+               if (rc)\r
+                       return true;\r
+       }\r
+       return false;\r
+}\r
+\r
+\r
+std::string FindNameServerWin()\r
+{\r
+       std::string returnval;\r
+       HKEY top, key;\r
+       char* dns = NULL;\r
+\r
+       /* Lets see if the correct registry hive and tree exist */\r
+       if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, "System\\CurrentControlSet\\Services\\Tcpip\\Parameters", 0, KEY_READ, &top) == ERROR_SUCCESS)\r
+       {\r
+               /* If they do, attempt to get the nameserver name */\r
+               RegOpenKeyEx(top, "Interfaces", 0, KEY_QUERY_VALUE|KEY_ENUMERATE_SUB_KEYS, &key);\r
+               if ((GetNameServer(top, "NameServer", dns)) || (GetNameServer(top, "DhcpNameServer", dns))\r
+                       || (GetInterface(key, "NameServer", dns)) || (GetInterface(key, "DhcpNameServer", dns)))\r
+               {\r
+                       if (dns)\r
+                       {\r
+                               returnval = dns;\r
+                               delete dns;\r
+                       }\r
+               }\r
+               RegCloseKey(key);\r
+               RegCloseKey(top);\r
+       }\r
+       return returnval;\r
+}\r
+\r