mirror of
https://github.com/jart/cosmopolitan.git
synced 2025-01-31 11:37:35 +00:00
93 lines
4.7 KiB
C
93 lines
4.7 KiB
C
/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
|
|
│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8 :vi│
|
|
╞══════════════════════════════════════════════════════════════════════════════╡
|
|
│ Copyright 2020 Justine Alexandra Roberts Tunney │
|
|
│ │
|
|
│ Permission to use, copy, modify, and/or distribute this software for │
|
|
│ any purpose with or without fee is hereby granted, provided that the │
|
|
│ above copyright notice and this permission notice appear in all copies. │
|
|
│ │
|
|
│ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL │
|
|
│ WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED │
|
|
│ WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE │
|
|
│ AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL │
|
|
│ DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR │
|
|
│ PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER │
|
|
│ TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR │
|
|
│ PERFORMANCE OF THIS SOFTWARE. │
|
|
╚─────────────────────────────────────────────────────────────────────────────*/
|
|
#include "libc/calls/calls.h"
|
|
#include "libc/calls/syscall_support-nt.internal.h"
|
|
#include "libc/dns/dns.h"
|
|
#include "libc/dns/resolvconf.h"
|
|
#include "libc/mem/arraylist.internal.h"
|
|
#include "libc/nt/enum/keyaccess.h"
|
|
#include "libc/nt/enum/reggetvalueflags.h"
|
|
#include "libc/nt/registry.h"
|
|
#include "libc/nt/runtime.h"
|
|
#include "libc/runtime/runtime.h"
|
|
#include "libc/sock/sock.h"
|
|
#include "libc/str/str.h"
|
|
#include "libc/sysv/consts/af.h"
|
|
|
|
/**
|
|
* Extracts DNS nameserver IPs from Windows Registry.
|
|
*
|
|
* @param resolv points to a ResolvConf object, which should be zero
|
|
* initialized by the caller; or if it already contains items,
|
|
* this function will append
|
|
* @return number of nameservers appended, or -1 w/ errno
|
|
*/
|
|
textwindows int GetNtNameServers(struct ResolvConf *resolv) {
|
|
int rc;
|
|
char value8[128];
|
|
int64_t hkInterfaces;
|
|
char *state, *addr, *tmp;
|
|
struct sockaddr_in nameserver;
|
|
char16_t value[128], uuid[64];
|
|
uint32_t i, keycount, valuebytes, uuidlen;
|
|
keycount = 0;
|
|
hkInterfaces = kNtInvalidHandleValue;
|
|
if (!RegOpenKeyEx(
|
|
kNtHkeyLocalMachine,
|
|
u"SYSTEM\\CurrentControlSet\\Services\\Tcpip\\Parameters\\Interfaces",
|
|
0, kNtKeyRead, &hkInterfaces) &&
|
|
!RegQueryInfoKey(hkInterfaces, 0, 0, 0, &keycount, 0, 0, 0, 0, 0, 0, 0)) {
|
|
nameserver.sin_family = AF_INET;
|
|
nameserver.sin_port = htons(DNS_PORT);
|
|
rc = 0;
|
|
for (i = 0; i < keycount; ++i) {
|
|
uuidlen = sizeof(uuid);
|
|
if (!RegEnumKeyEx(hkInterfaces, i, uuid, &uuidlen, 0, 0, 0, 0) &&
|
|
((!RegGetValue(hkInterfaces, uuid, u"DhcpIpAddress",
|
|
kNtRrfRtRegSz | kNtRrfRtRegMultiSz, NULL, value,
|
|
((valuebytes = sizeof(value)), &valuebytes)) &&
|
|
valuebytes > 2 * sizeof(char16_t)) ||
|
|
(!RegGetValue(hkInterfaces, uuid, u"IpAddress",
|
|
kNtRrfRtRegSz | kNtRrfRtRegMultiSz, NULL, value,
|
|
((valuebytes = sizeof(value)), &valuebytes)) &&
|
|
valuebytes > 2 * sizeof(char16_t))) &&
|
|
((!RegGetValue(hkInterfaces, uuid, u"DhcpNameServer",
|
|
kNtRrfRtRegSz | kNtRrfRtRegMultiSz, NULL, value,
|
|
((valuebytes = sizeof(value)), &valuebytes)) &&
|
|
valuebytes > 2 * sizeof(char16_t)) ||
|
|
(!RegGetValue(hkInterfaces, uuid, u"NameServer",
|
|
kNtRrfRtRegSz | kNtRrfRtRegMultiSz, NULL, value,
|
|
((valuebytes = sizeof(value)), &valuebytes)) &&
|
|
valuebytes > 2 * sizeof(char16_t)))) {
|
|
tprecode16to8(value8, sizeof(value8), value);
|
|
tmp = value8;
|
|
while ((addr = strtok_r(tmp, ", ", &state))) {
|
|
if (inet_pton(AF_INET, addr, &nameserver.sin_addr.s_addr) == 1) {
|
|
if (append(&resolv->nameservers, &nameserver) != -1) ++rc;
|
|
}
|
|
tmp = NULL;
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
rc = __winerr();
|
|
}
|
|
RegCloseKey(hkInterfaces);
|
|
return rc;
|
|
}
|