From 2864d22df22fd01c39406d695cfb99eb97a1cfb6 Mon Sep 17 00:00:00 2001
From: Tau <tau@nowhere>
Date: Sat, 7 Sep 2019 16:26:10 -0400
Subject: [PATCH] hooklib/dns.c: Hook classic getaddrinfo() API too

---
 hooklib/dns.c | 101 ++++++++++++++++++++++++++++++++++++++++++++++++--
 precompiled.h |   1 +
 2 files changed, 99 insertions(+), 3 deletions(-)

diff --git a/hooklib/dns.c b/hooklib/dns.c
index 221c7fc..f75085d 100644
--- a/hooklib/dns.c
+++ b/hooklib/dns.c
@@ -2,6 +2,7 @@
 
 #include <windows.h>
 #include <windns.h>
+#include <ws2tcpip.h>
 
 #include <assert.h>
 #include <stdbool.h>
@@ -59,6 +60,12 @@ static DNS_STATUS WINAPI hook_DnsQueryEx(
         void *pQueryResults,
         void *pCancelHandle);
 
+static int WSAAPI hook_getaddrinfo(
+        const char *pNodeName,
+        const char *pServiceName,
+        const ADDRINFOA *pHints,
+        ADDRINFOA **ppResult);
+
 /* Link pointers */
 
 static DNS_STATUS WINAPI (*next_DnsQuery_A)(
@@ -82,7 +89,13 @@ static DNS_STATUS WINAPI (*next_DnsQueryEx)(
         void *pQueryResults,
         void *pCancelHandle);
 
-static const struct hook_symbol dns_hook_syms[] = {
+static int WSAAPI (*next_getaddrinfo)(
+        const char *pNodeName,
+        const char *pServiceName,
+        const ADDRINFOA *pHints,
+        ADDRINFOA **ppResult);
+
+static const struct hook_symbol dns_hook_syms_dnsapi[] = {
     {
         .name       = "DnsQuery_A",
         .patch      = hook_DnsQuery_A,
@@ -98,6 +111,15 @@ static const struct hook_symbol dns_hook_syms[] = {
     }
 };
 
+static const struct hook_symbol dns_hook_syms_ws2[] = {
+    {
+        .name       = "getaddrinfo",
+        .ordinal    = 176,
+        .patch      = hook_getaddrinfo,
+        .link       = (void **) &next_getaddrinfo,
+    }
+};
+
 static bool dns_hook_initted;
 static CRITICAL_SECTION dns_hook_lock;
 static struct dns_hook_entry *dns_hook_entries;
@@ -115,8 +137,14 @@ static void dns_hook_init(void)
     hook_table_apply(
             NULL,
             "dnsapi.dll",
-            dns_hook_syms,
-            _countof(dns_hook_syms));
+            dns_hook_syms_dnsapi,
+            _countof(dns_hook_syms_dnsapi));
+
+    hook_table_apply(
+            NULL,
+            "ws2_32.dll",
+            dns_hook_syms_ws2,
+            _countof(dns_hook_syms_ws2));
 }
 
 HRESULT dns_hook_push(const wchar_t *from_src, const wchar_t *to_src)
@@ -340,3 +368,70 @@ static DNS_STATUS WINAPI hook_DnsQueryEx(
 
     return code;
 }
+
+static int WSAAPI hook_getaddrinfo(
+        const char *pNodeName,
+        const char *pServiceName,
+        const ADDRINFOA *pHints,
+        ADDRINFOA **ppResult)
+{
+    const struct dns_hook_entry *pos;
+    char *str;
+    size_t str_c;
+    wchar_t *wstr;
+    size_t wstr_c;
+    int result;
+    size_t i;
+
+    str = NULL;
+    wstr = NULL;
+
+    if (pNodeName == NULL) {
+        result = WSA_INVALID_PARAMETER;
+
+        goto end;
+    }
+
+    mbstowcs_s(&wstr_c, NULL, 0, pNodeName, 0);
+    wstr = malloc(wstr_c * sizeof(wchar_t));
+
+    if (wstr == NULL) {
+        result = WSA_NOT_ENOUGH_MEMORY;
+
+        goto end;
+    }
+
+    mbstowcs_s(NULL, wstr, wstr_c, pNodeName, wstr_c - 1);
+    EnterCriticalSection(&dns_hook_lock);
+
+    for (i = 0 ; i < dns_hook_nentries ; i++) {
+        pos = &dns_hook_entries[i];
+
+        if (_wcsicmp(wstr, pos->from) == 0) {
+            wcstombs_s(&str_c, NULL, 0, pos->to, 0);
+            str = malloc(str_c * sizeof(char));
+
+            if (str == NULL) {
+                LeaveCriticalSection(&dns_hook_lock);
+                result = WSA_NOT_ENOUGH_MEMORY;
+
+                goto end;
+            }
+
+            wcstombs_s(NULL, str, str_c, pos->to, str_c - 1);
+            pNodeName = str;
+
+            break;
+        }
+    }
+
+    LeaveCriticalSection(&dns_hook_lock);
+
+    result = next_getaddrinfo(pNodeName, pServiceName, pHints, ppResult);
+
+end:
+    free(wstr);
+    free(str);
+
+    return result;
+}
diff --git a/precompiled.h b/precompiled.h
index 63338a7..7c85da7 100644
--- a/precompiled.h
+++ b/precompiled.h
@@ -11,6 +11,7 @@
 #include <setupapi.h>
 #include <unknwn.h>
 #include <windns.h>
+#include <ws2tcpip.h>
 #include <dinput.h>
 #include <xinput.h>
 #include <d3d9.h>