debian-packages/eventcore
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Implement simple (synchronous) DNS lookup for the WinAPI driver.

Browse source
This commit is contained in:
Sönke Ludwig 2017-01-23 21:37:35 +01:00
parent e8a0968ded
commit 356a34fad2
5 changed files with 251 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

87
source/eventcore/drivers/winapi.d
View file

@ -13,6 +13,7 @@ import eventcore.driver;
import eventcore.drivers.timer;
import eventcore.internal.consumablequeue : ConsumableQueue;
import eventcore.internal.utils;
import eventcore.internal.win32;
import taggedalgebraic;
import core.sys.windows.windows;
import core.sys.windows.winsock2;
@ -357,7 +358,91 @@ final class WinAPIEventDriverDNS : EventDriverDNS {
DNSLookupID lookupHost(string name, DNSLookupCallback on_lookup_finished)
{
assert(false, "TODO!");
import std.typecons : scoped;
import std.utf : toUTF16z;
auto id = DNSLookupID(0);
static immutable ushort[] addrfamilies = [AF_INET, AF_INET6];
const(WCHAR)* namew;
try namew = name.toUTF16z;
catch (Exception e) return DNSLookupID.invalid;
foreach (af; addrfamilies) {
//if (family != af && family != AF_UNSPEC) continue;
SOCKADDR_STORAGE sa;
INT addrlen = sa.sizeof;
auto ret = () @trusted { return WSAStringToAddressW(namew, af, null, cast(sockaddr*)&sa, &addrlen); } ();
if (ret != 0) continue;
scope addr = new RefAddress(() @trusted { return cast(sockaddr*)&sa; } (), addrlen);
RefAddress[1] addrs;
addrs[0] = addr;
on_lookup_finished(id, DNSStatus.ok, addrs);
return id;
}
version(none){ // Windows 8+
LookupStatus status;
status.task = Task.getThis();
status.driver = this;
status.finished = false;
WSAOVERLAPPEDX overlapped;
overlapped.Internal = 0;
overlapped.InternalHigh = 0;
overlapped.hEvent = cast(HANDLE)cast(void*)&status;
void* aif;
ADDRINFOEXW addr_hint;
ADDRINFOEXW* addr_ret;
addr_hint.ai_family = family;
addr_hint.ai_socktype = SOCK_STREAM;
addr_hint.ai_protocol = IPPROTO_TCP;
enforce(GetAddrInfoExW(namew, null, NS_DNS, null, &addr_hint, &addr_ret, null, &overlapped, &onDnsResult, null) == 0, "Failed to lookup host");
while( !status.finished ) m_core.yieldForEvent();
enforce(!status.error, "Failed to lookup host: "~to!string(status.error));
aif = addr_ret;
addr.family = cast(ubyte)addr_ret.ai_family;
switch(addr.family){
default: assert(false, "Invalid address family returned from host lookup.");
case AF_INET: addr.sockAddrInet4 = *cast(sockaddr_in*)addr_ret.ai_addr; break;
case AF_INET6: addr.sockAddrInet6 = *cast(sockaddr_in6*)addr_ret.ai_addr; break;
}
FreeAddrInfoExW(addr_ret);
} else {
ADDRINFOW* results;
if (auto ret = () @trusted { return GetAddrInfoW(namew, null, null, &results); } ()) {
on_lookup_finished(id, DNSStatus.error, null);
return id;
}
scope(failure) assert(false);
() @trusted {
typeof(scoped!RefAddress())[16] addr_storage = [
scoped!RefAddress(), scoped!RefAddress(), scoped!RefAddress(), scoped!RefAddress(),
scoped!RefAddress(), scoped!RefAddress(), scoped!RefAddress(), scoped!RefAddress(),
scoped!RefAddress(), scoped!RefAddress(), scoped!RefAddress(), scoped!RefAddress(),
scoped!RefAddress(), scoped!RefAddress(), scoped!RefAddress(), scoped!RefAddress()
];
RefAddress[16] buf;
size_t naddr = 0;
while (results) {
RefAddress addr = addr_storage[naddr];
addr.set(results.ai_addr, results.ai_addrlen);
buf[naddr++] = addr;
results = results.ai_next;
}
on_lookup_finished(id, DNSStatus.ok, buf[0 .. naddr]);
} ();
}
return id;
}
void cancelLookup(DNSLookupID handle)