From 1affb5c169632b67e25e68017236d423d56d2c41 Mon Sep 17 00:00:00 2001
From: Tau <tau@nowhere>
Date: Wed, 4 Sep 2019 13:43:15 -0400
Subject: [PATCH] platform/dns.c: Add platform DNS hook

---
 platform/config.c    | 45 ++++++++++++++++++++++++++++++++++
 platform/config.h    | 10 ++++++++
 platform/dns.c       | 57 ++++++++++++++++++++++++++++++++++++++++++++
 platform/dns.h       |  8 +++++++
 platform/meson.build |  2 ++
 5 files changed, 122 insertions(+)
 create mode 100644 platform/dns.c
 create mode 100644 platform/dns.h

diff --git a/platform/config.c b/platform/config.c
index 754e14e..e722824 100644
--- a/platform/config.c
+++ b/platform/config.c
@@ -7,6 +7,7 @@
 #include <stdint.h>
 #include <stdio.h>
 #include <stdlib.h>
+#include <string.h>
 
 #include "platform/config.h"
 
@@ -16,6 +17,7 @@ void alls_config_load(struct alls_config *cfg, const wchar_t *filename)
     assert(filename != NULL);
 
     amvideo_config_load(&cfg->amvideo, filename);
+    dns_config_load(&cfg->dns, filename);
     hwmon_config_load(&cfg->hwmon, filename);
     misc_config_load(&cfg->misc, filename);
     pcbid_config_load(&cfg->pcbid, filename);
@@ -29,6 +31,7 @@ void nu_config_load(struct nu_config *cfg, const wchar_t *filename)
     assert(filename != NULL);
 
     amvideo_config_load(&cfg->amvideo, filename);
+    dns_config_load(&cfg->dns, filename);
     hwmon_config_load(&cfg->hwmon, filename);
     misc_config_load(&cfg->misc, filename);
     nusec_config_load(&cfg->nusec, filename);
@@ -43,6 +46,48 @@ void amvideo_config_load(struct amvideo_config *cfg, const wchar_t *filename)
     cfg->enable = GetPrivateProfileIntW(L"amvideo", L"enable", 1, filename);
 }
 
+void dns_config_load(struct dns_config *cfg, const wchar_t *filename)
+{
+    wchar_t default_[128];
+
+    assert(cfg != NULL);
+    assert(filename != NULL);
+
+    cfg->enable = GetPrivateProfileIntW(L"dns", L"enable", 1, filename);
+
+    GetPrivateProfileStringW(
+            L"dns",
+            L"default",
+            L"localhost",
+            default_,
+            _countof(default_),
+            filename);
+
+    GetPrivateProfileStringW(
+            L"dns",
+            L"startup",
+            default_,
+            cfg->startup,
+            _countof(cfg->startup),
+            filename);
+
+    GetPrivateProfileStringW(
+            L"dns",
+            L"billing",
+            default_,
+            cfg->billing,
+            _countof(cfg->billing),
+            filename);
+
+    GetPrivateProfileStringW(
+            L"dns",
+            L"aimedb",
+            default_,
+            cfg->aimedb,
+            _countof(cfg->aimedb),
+            filename);
+}
+
 void hwmon_config_load(struct hwmon_config *cfg, const wchar_t *filename)
 {
     assert(cfg != NULL);
diff --git a/platform/config.h b/platform/config.h
index fccc661..37ae555 100644
--- a/platform/config.h
+++ b/platform/config.h
@@ -10,6 +10,13 @@ struct amvideo_config {
     bool enable;
 };
 
+struct dns_config {
+    bool enable;
+    wchar_t startup[128];
+    wchar_t billing[128];
+    wchar_t aimedb[128];
+};
+
 struct hwmon_config {
     bool enable;
 };
@@ -43,6 +50,7 @@ struct vfs_config {
 
 struct nu_config {
     struct amvideo_config amvideo;
+    struct dns_config dns;
     struct hwmon_config hwmon;
     struct misc_config misc;
     struct nusec_config nusec;
@@ -51,6 +59,7 @@ struct nu_config {
 
 struct alls_config {
     struct amvideo_config amvideo;
+    struct dns_config dns;
     struct hwmon_config hwmon;
     struct misc_config misc;
     struct pcbid_config pcbid;
@@ -62,6 +71,7 @@ void alls_config_load(struct alls_config *cfg, const wchar_t *filename);
 void nu_config_load(struct nu_config *cfg, const wchar_t *filename);
 
 void amvideo_config_load(struct amvideo_config *cfg, const wchar_t *filename);
+void dns_config_load(struct dns_config *cfg, const wchar_t *filename);
 void hwmon_config_load(struct hwmon_config *cfg, const wchar_t *filename);
 void misc_config_load(struct misc_config *cfg, const wchar_t *filename);
 void nusec_config_load(struct nusec_config *cfg, const wchar_t *filename);
diff --git a/platform/dns.c b/platform/dns.c
new file mode 100644
index 0000000..8a8ea46
--- /dev/null
+++ b/platform/dns.c
@@ -0,0 +1,57 @@
+#include <windows.h>
+
+#include <assert.h>
+
+#include "hooklib/dns.h"
+
+#include "platform/config.h"
+#include "platform/dns.h"
+
+HRESULT dns_platform_hook_init(const struct dns_config *cfg)
+{
+    HRESULT hr;
+
+    assert(cfg != NULL);
+
+    if (!cfg->enable) {
+        return S_FALSE;
+    }
+
+    hr = dns_hook_push(L"naominet.jp", cfg->startup);
+
+    if (FAILED(hr)) {
+        return hr;
+    }
+
+    hr = dns_hook_push(L"anbzvarg.wc", cfg->startup);
+
+    if (FAILED(hr)) {
+        return hr;
+    }
+
+    hr = dns_hook_push(L"ib.naominet.jp", cfg->billing);
+
+    if (FAILED(hr)) {
+        return hr;
+    }
+
+    hr = dns_hook_push(L"vo.anbzvarg.wc", cfg->billing);
+
+    if (FAILED(hr)) {
+        return hr;
+    }
+
+    hr = dns_hook_push(L"aime.naominet.jp", cfg->aimedb);
+
+    if (FAILED(hr)) {
+        return hr;
+    }
+
+    hr = dns_hook_push(L"nvzr.anbzvarg.wc", cfg->aimedb);
+
+    if (FAILED(hr)) {
+        return hr;
+    }
+
+    return S_OK;
+}
diff --git a/platform/dns.h b/platform/dns.h
new file mode 100644
index 0000000..ab12edb
--- /dev/null
+++ b/platform/dns.h
@@ -0,0 +1,8 @@
+#pragma once
+
+#include <windows.h>
+
+#include "platform/config.h"
+
+HRESULT dns_platform_hook_init(const struct dns_config *cfg);
+
diff --git a/platform/meson.build b/platform/meson.build
index dffa012..67d0a6e 100644
--- a/platform/meson.build
+++ b/platform/meson.build
@@ -11,6 +11,8 @@ platform_lib = static_library(
         'amvideo.h',
         'config.c',
         'config.h',
+        'dns.c',
+        'dns.h',
         'hwmon.c',
         'hwmon.h',
         'misc.c',