Implement WinAPI based TCP socket support.
This commit is contained in:
parent
ab796477e2
commit
ca11fdec82
4
dub.sdl
4
dub.sdl
|
@ -33,13 +33,13 @@ configuration "select" {
|
||||||
configuration "winapi-optlink" {
|
configuration "winapi-optlink" {
|
||||||
platforms "windows-x86-dmd"
|
platforms "windows-x86-dmd"
|
||||||
versions "EventcoreWinAPIDriver"
|
versions "EventcoreWinAPIDriver"
|
||||||
sourceFiles "lib/ws2_32.lib"
|
sourceFiles "lib/ws2_32.lib" "lib/kernel32.lib"
|
||||||
}
|
}
|
||||||
|
|
||||||
configuration "select-optlink" {
|
configuration "select-optlink" {
|
||||||
platforms "windows-x86-dmd"
|
platforms "windows-x86-dmd"
|
||||||
versions "EventcoreSelectDriver"
|
versions "EventcoreSelectDriver"
|
||||||
sourceFiles "lib/ws2_32.lib"
|
sourceFiles "lib/ws2_32.lib" "lib/kernel32.lib"
|
||||||
}
|
}
|
||||||
|
|
||||||
configuration "libasync" {
|
configuration "libasync" {
|
||||||
|
|
BIN
lib/kernel32.lib
Normal file
BIN
lib/kernel32.lib
Normal file
Binary file not shown.
|
@ -53,7 +53,7 @@ final class WinAPIEventDriver : EventDriver {
|
||||||
m_core = new WinAPIEventDriverCore(m_timers);
|
m_core = new WinAPIEventDriverCore(m_timers);
|
||||||
m_events = new WinAPIEventDriverEvents(m_core);
|
m_events = new WinAPIEventDriverEvents(m_core);
|
||||||
m_files = new WinAPIEventDriverFiles(m_core);
|
m_files = new WinAPIEventDriverFiles(m_core);
|
||||||
m_sockets = new WinAPIEventDriverSockets();
|
m_sockets = new WinAPIEventDriverSockets(m_core);
|
||||||
m_dns = new WinAPIEventDriverDNS();
|
m_dns = new WinAPIEventDriverDNS();
|
||||||
m_watchers = new WinAPIEventDriverWatchers(m_core);
|
m_watchers = new WinAPIEventDriverWatchers(m_core);
|
||||||
}
|
}
|
||||||
|
|
|
@ -3,66 +3,346 @@ module eventcore.drivers.winapi.sockets;
|
||||||
version (Windows):
|
version (Windows):
|
||||||
|
|
||||||
import eventcore.driver;
|
import eventcore.driver;
|
||||||
|
import eventcore.drivers.winapi.core;
|
||||||
import eventcore.internal.win32;
|
import eventcore.internal.win32;
|
||||||
|
import eventcore.internal.utils : AlgebraicChoppedVector, print;
|
||||||
import std.socket : Address;
|
import std.socket : Address;
|
||||||
|
|
||||||
|
private enum WM_USER_SOCKET = WM_USER + 1;
|
||||||
|
|
||||||
|
|
||||||
final class WinAPIEventDriverSockets : EventDriverSockets {
|
final class WinAPIEventDriverSockets : EventDriverSockets {
|
||||||
@safe: /*@nogc:*/ nothrow:
|
@safe: /*@nogc:*/ nothrow:
|
||||||
|
private {
|
||||||
|
alias SocketVector = AlgebraicChoppedVector!(SocketSlot, StreamSocketSlot, StreamListenSocketSlot, DgramSocketSlot);
|
||||||
|
SocketVector m_sockets;
|
||||||
|
WinAPIEventDriverCore m_core;
|
||||||
|
DWORD m_tid;
|
||||||
|
HWND m_hwnd;
|
||||||
|
size_t m_waiters;
|
||||||
|
}
|
||||||
|
|
||||||
|
this(WinAPIEventDriverCore core)
|
||||||
|
@trusted {
|
||||||
|
m_tid = GetCurrentThreadId();
|
||||||
|
m_core = core;
|
||||||
|
|
||||||
|
// setup socket event message window
|
||||||
|
setupWindowClass();
|
||||||
|
m_hwnd = CreateWindowA("VibeWin32MessageWindow", "VibeWin32MessageWindow", 0, 0,0,0,0, HWND_MESSAGE,null,null,null);
|
||||||
|
SetWindowLongPtrA(m_hwnd, GWLP_USERDATA, cast(ULONG_PTR)cast(void*)this);
|
||||||
|
assert(cast(WinAPIEventDriverSockets)cast(void*)GetWindowLongPtrA(m_hwnd, GWLP_USERDATA) is this);
|
||||||
|
}
|
||||||
|
|
||||||
|
package @property size_t waiterCount() const { return m_waiters; }
|
||||||
|
|
||||||
override StreamSocketFD connectStream(scope Address peer_address, scope Address bind_address, ConnectCallback on_connect)
|
override StreamSocketFD connectStream(scope Address peer_address, scope Address bind_address, ConnectCallback on_connect)
|
||||||
{
|
@trusted {
|
||||||
assert(false, "TODO!");
|
assert(m_tid == GetCurrentThreadId());
|
||||||
|
|
||||||
|
auto fd = WSASocketW(peer_address.addressFamily, SOCK_STREAM, IPPROTO_TCP, null, 0, WSA_FLAG_OVERLAPPED);
|
||||||
|
if (fd == INVALID_SOCKET)
|
||||||
|
return StreamSocketFD.invalid;
|
||||||
|
|
||||||
|
void invalidateSocket() @nogc @trusted nothrow { closesocket(fd); fd = INVALID_SOCKET; }
|
||||||
|
|
||||||
|
int bret;
|
||||||
|
if (bind_address !is null)
|
||||||
|
bret = bind(fd, bind_address.name, bind_address.nameLen);
|
||||||
|
if (bret != 0) {
|
||||||
|
invalidateSocket();
|
||||||
|
on_connect(StreamSocketFD.invalid, ConnectStatus.bindFailure);
|
||||||
|
return StreamSocketFD.invalid;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto sock = adoptStreamInternal(fd);
|
||||||
|
|
||||||
|
auto ret = .connect(fd, peer_address.name, peer_address.nameLen);
|
||||||
|
//auto ret = WSAConnect(m_socket, peer_address.name, peer_address.nameLen, null, null, null, null);
|
||||||
|
|
||||||
|
if (ret == 0) {
|
||||||
|
m_sockets[sock].specific.state = ConnectionState.connected;
|
||||||
|
on_connect(sock, ConnectStatus.connected);
|
||||||
|
return sock;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto err = WSAGetLastError();
|
||||||
|
if (err == WSAEWOULDBLOCK) {
|
||||||
|
with (m_sockets[sock].streamSocket) {
|
||||||
|
connectCallback = on_connect;
|
||||||
|
state = ConnectionState.connecting;
|
||||||
|
}
|
||||||
|
return sock;
|
||||||
|
} else {
|
||||||
|
clearSocketSlot(sock);
|
||||||
|
invalidateSocket();
|
||||||
|
on_connect(StreamSocketFD.invalid, ConnectStatus.unknownError);
|
||||||
|
return StreamSocketFD.invalid;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
override StreamSocketFD adoptStream(int socket)
|
override StreamSocketFD adoptStream(int socket)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
return adoptStreamInternal(socket);
|
||||||
|
}
|
||||||
|
|
||||||
|
private StreamSocketFD adoptStreamInternal(SOCKET socket)
|
||||||
|
{
|
||||||
|
auto fd = StreamSocketFD(socket);
|
||||||
|
if (m_sockets[fd].common.refCount) // FD already in use?
|
||||||
|
return StreamSocketFD.invalid;
|
||||||
|
|
||||||
|
uint enable = 1;
|
||||||
|
() @trusted { ioctlsocket(socket, FIONBIO, &enable); } ();
|
||||||
|
|
||||||
|
void setupOverlapped(ref WSAOVERLAPPEDX overlapped) @trusted @nogc nothrow {
|
||||||
|
overlapped.Internal = 0;
|
||||||
|
overlapped.InternalHigh = 0;
|
||||||
|
overlapped.Offset = 0;
|
||||||
|
overlapped.OffsetHigh = 0;
|
||||||
|
overlapped.hEvent = cast(HANDLE)cast(void*)&m_sockets[socket];
|
||||||
|
}
|
||||||
|
|
||||||
|
initSocketSlot(fd);
|
||||||
|
with (m_sockets[socket]) {
|
||||||
|
specific = StreamSocketSlot.init;
|
||||||
|
setupOverlapped(streamSocket.write.overlapped);
|
||||||
|
setupOverlapped(streamSocket.read.overlapped);
|
||||||
|
}
|
||||||
|
|
||||||
|
() @trusted { WSAAsyncSelect(socket, m_hwnd, WM_USER_SOCKET, FD_READ|FD_WRITE|FD_CONNECT|FD_CLOSE); } ();
|
||||||
|
|
||||||
|
return fd;
|
||||||
}
|
}
|
||||||
|
|
||||||
alias listenStream = EventDriverSockets.listenStream;
|
alias listenStream = EventDriverSockets.listenStream;
|
||||||
override StreamListenSocketFD listenStream(scope Address bind_address, StreamListenOptions options, AcceptCallback on_accept)
|
override StreamListenSocketFD listenStream(scope Address bind_address, StreamListenOptions options, AcceptCallback on_accept)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
auto fd = () @trusted { return WSASocketW(bind_address.addressFamily, SOCK_STREAM, IPPROTO_TCP, null, 0, WSA_FLAG_OVERLAPPED); } ();
|
||||||
|
if (fd == INVALID_SOCKET)
|
||||||
|
return StreamListenSocketFD.invalid;
|
||||||
|
|
||||||
|
void invalidateSocket() @nogc @trusted nothrow { closesocket(fd); fd = INVALID_SOCKET; }
|
||||||
|
|
||||||
|
() @trusted {
|
||||||
|
int tmp_reuse = 1;
|
||||||
|
if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &tmp_reuse, tmp_reuse.sizeof) != 0) {
|
||||||
|
invalidateSocket();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// FIXME: should SO_EXCLUSIVEADDRUSE be used of StreamListenOptions.reuseAddress isn't set?
|
||||||
|
|
||||||
|
if (bind(fd, bind_address.name, bind_address.nameLen) != 0) {
|
||||||
|
invalidateSocket();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (listen(fd, 128) != 0) {
|
||||||
|
invalidateSocket();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
} ();
|
||||||
|
|
||||||
|
if (fd == INVALID_SOCKET)
|
||||||
|
return StreamListenSocketFD.invalid;
|
||||||
|
|
||||||
|
auto sock = cast(StreamListenSocketFD)fd;
|
||||||
|
initSocketSlot(sock);
|
||||||
|
m_sockets[sock].specific = StreamListenSocketSlot.init;
|
||||||
|
|
||||||
|
if (on_accept) waitForConnections(sock, on_accept);
|
||||||
|
|
||||||
|
return sock;
|
||||||
}
|
}
|
||||||
|
|
||||||
override void waitForConnections(StreamListenSocketFD sock, AcceptCallback on_accept)
|
override void waitForConnections(StreamListenSocketFD sock, AcceptCallback on_accept)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
assert(!m_sockets[sock].streamListen.acceptCallback);
|
||||||
|
m_sockets[sock].streamListen.acceptCallback = on_accept;
|
||||||
|
() @trusted { WSAAsyncSelect(sock, m_hwnd, WM_USER_SOCKET, FD_ACCEPT); } ();
|
||||||
|
m_core.addWaiter();
|
||||||
}
|
}
|
||||||
|
|
||||||
override ConnectionState getConnectionState(StreamSocketFD sock)
|
override ConnectionState getConnectionState(StreamSocketFD sock)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
return m_sockets[sock].streamSocket.state;
|
||||||
}
|
}
|
||||||
|
|
||||||
override bool getLocalAddress(SocketFD sock, scope RefAddress dst)
|
override bool getLocalAddress(SocketFD sock, scope RefAddress dst)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
socklen_t addr_len = dst.nameLen;
|
||||||
|
if (() @trusted { return getsockname(sock, dst.name, &addr_len); } () != 0)
|
||||||
|
return false;
|
||||||
|
dst.cap(addr_len);
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
override bool getRemoteAddress(SocketFD sock, scope RefAddress dst)
|
override bool getRemoteAddress(SocketFD sock, scope RefAddress dst)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
socklen_t addr_len = dst.nameLen;
|
||||||
|
if (() @trusted { return getpeername(sock, dst.name, &addr_len); } () != 0)
|
||||||
|
return false;
|
||||||
|
dst.cap(addr_len);
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
override void setTCPNoDelay(StreamSocketFD socket, bool enable)
|
override void setTCPNoDelay(StreamSocketFD socket, bool enable)
|
||||||
{
|
@trusted {
|
||||||
assert(false, "TODO!");
|
BOOL eni = enable;
|
||||||
|
setsockopt(INVALID_SOCKET, IPPROTO_TCP, TCP_NODELAY, &eni, eni.sizeof);
|
||||||
}
|
}
|
||||||
|
|
||||||
override void setKeepAlive(StreamSocketFD socket, bool enable)
|
override void setKeepAlive(StreamSocketFD socket, bool enable)
|
||||||
{
|
@trusted {
|
||||||
assert(false, "TODO!");
|
BOOL eni = enable;
|
||||||
|
setsockopt(INVALID_SOCKET, SOL_SOCKET, SO_KEEPALIVE, &eni, eni.sizeof);
|
||||||
}
|
}
|
||||||
|
|
||||||
override void read(StreamSocketFD socket, ubyte[] buffer, IOMode mode, IOCallback on_read_finish)
|
override void read(StreamSocketFD socket, ubyte[] buffer, IOMode mode, IOCallback on_read_finish)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
WSABUF buf;
|
||||||
|
buf.len = buffer.length;
|
||||||
|
buf.buf = () @trusted { return buffer.ptr; } ();
|
||||||
|
|
||||||
|
m_sockets[socket].streamSocket.read.buffer = buffer;
|
||||||
|
m_sockets[socket].streamSocket.read.mode = mode;
|
||||||
|
|
||||||
|
auto ovl = mode == IOMode.immediate ? null : &m_sockets[socket].streamSocket.read.overlapped;
|
||||||
|
DWORD flags = 0;
|
||||||
|
auto ret = () @trusted { return WSARecv(socket, &buf, 1, null, &flags, ovl, &onIOReadCompleted); } ();
|
||||||
|
if (ret == SOCKET_ERROR) {
|
||||||
|
auto err = WSAGetLastError();
|
||||||
|
if (err == WSA_IO_PENDING) {
|
||||||
|
if (mode == IOMode.immediate) {
|
||||||
|
on_read_finish(socket, IOStatus.wouldBlock, 0);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
on_read_finish(socket, IOStatus.error, 0);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
m_sockets[socket].streamSocket.read.callback = on_read_finish;
|
||||||
|
m_core.addWaiter();
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
private static extern(System) nothrow
|
||||||
|
void onIOReadCompleted(DWORD dwError, DWORD cbTransferred, WSAOVERLAPPEDX* lpOverlapped, DWORD dwFlags)
|
||||||
|
{
|
||||||
|
auto slot = () @trusted { return cast(SocketVector.FullField*)lpOverlapped.hEvent; } ();
|
||||||
|
|
||||||
|
if (!slot.streamSocket.read.callback) return;
|
||||||
|
|
||||||
|
void invokeCallback(IOStatus status, size_t nsent)
|
||||||
|
@safe nothrow {
|
||||||
|
slot.common.core.removeWaiter();
|
||||||
|
auto cb = slot.streamSocket.read.callback;
|
||||||
|
slot.streamSocket.read.callback = null;
|
||||||
|
cb(cast(StreamSocketFD)slot.common.fd, status, nsent);
|
||||||
|
}
|
||||||
|
|
||||||
|
slot.streamSocket.read.bytesTransferred += cbTransferred;
|
||||||
|
slot.streamSocket.read.buffer = slot.streamSocket.read.buffer[cbTransferred .. $];
|
||||||
|
|
||||||
|
if (dwError) {
|
||||||
|
invokeCallback(IOStatus.error, 0);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (slot.streamSocket.read.mode == IOMode.once || !slot.streamSocket.read.buffer.length) {
|
||||||
|
invokeCallback(IOStatus.ok, cbTransferred);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
WSABUF buf;
|
||||||
|
buf.len = slot.streamSocket.read.buffer.length;
|
||||||
|
buf.buf = () @trusted { return cast(ubyte*)slot.streamSocket.read.buffer.ptr; } ();
|
||||||
|
auto ovl = slot.streamSocket.read.mode == IOMode.immediate ? null : &slot.streamSocket.read.overlapped;
|
||||||
|
DWORD flags = 0;
|
||||||
|
auto ret = () @trusted { return WSARecv(slot.common.fd, &buf, 1, null, &flags, ovl, &onIOReadCompleted); } ();
|
||||||
|
if (ret == SOCKET_ERROR) {
|
||||||
|
auto err = WSAGetLastError();
|
||||||
|
if (err == WSA_IO_PENDING) {
|
||||||
|
if (slot.streamSocket.read.mode == IOMode.immediate) {
|
||||||
|
invokeCallback(IOStatus.wouldBlock, 0);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
invokeCallback(IOStatus.error, 0);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
override void write(StreamSocketFD socket, const(ubyte)[] buffer, IOMode mode, IOCallback on_write_finish)
|
override void write(StreamSocketFD socket, const(ubyte)[] buffer, IOMode mode, IOCallback on_write_finish)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
WSABUF buf;
|
||||||
|
buf.len = buffer.length;
|
||||||
|
buf.buf = () @trusted { return cast(ubyte*)buffer.ptr; } ();
|
||||||
|
|
||||||
|
m_sockets[socket].streamSocket.write.buffer = buffer;
|
||||||
|
m_sockets[socket].streamSocket.write.mode = mode;
|
||||||
|
|
||||||
|
auto ovl = mode == IOMode.immediate ? null : &m_sockets[socket].streamSocket.write.overlapped;
|
||||||
|
auto ret = () @trusted { return WSASend(socket, &buf, 1, null, 0, ovl, &onIOWriteCompleted); } ();
|
||||||
|
if (ret == SOCKET_ERROR) {
|
||||||
|
auto err = WSAGetLastError();
|
||||||
|
if (err == WSA_IO_PENDING) {
|
||||||
|
if (mode == IOMode.immediate) {
|
||||||
|
on_write_finish(socket, IOStatus.wouldBlock, 0);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
on_write_finish(socket, IOStatus.error, 0);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
m_sockets[socket].streamSocket.write.callback = on_write_finish;
|
||||||
|
m_core.addWaiter();
|
||||||
|
}
|
||||||
|
|
||||||
|
private static extern(System) nothrow
|
||||||
|
void onIOWriteCompleted(DWORD dwError, DWORD cbTransferred, WSAOVERLAPPEDX* lpOverlapped, DWORD dwFlags)
|
||||||
|
{
|
||||||
|
auto slot = () @trusted { return cast(SocketVector.FullField*)lpOverlapped.hEvent; } ();
|
||||||
|
|
||||||
|
if (!slot.streamSocket.write.callback) return;
|
||||||
|
|
||||||
|
void invokeCallback(IOStatus status, size_t nsent)
|
||||||
|
@safe nothrow {
|
||||||
|
slot.common.core.removeWaiter();
|
||||||
|
auto cb = slot.streamSocket.write.callback;
|
||||||
|
slot.streamSocket.write.callback = null;
|
||||||
|
cb(cast(StreamSocketFD)slot.common.fd, status, nsent);
|
||||||
|
}
|
||||||
|
|
||||||
|
slot.streamSocket.write.bytesTransferred += cbTransferred;
|
||||||
|
slot.streamSocket.write.buffer = slot.streamSocket.write.buffer[cbTransferred .. $];
|
||||||
|
|
||||||
|
if (dwError) {
|
||||||
|
invokeCallback(IOStatus.error, 0);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (slot.streamSocket.write.mode == IOMode.once || !slot.streamSocket.write.buffer.length) {
|
||||||
|
invokeCallback(IOStatus.ok, cbTransferred);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
WSABUF buf;
|
||||||
|
buf.len = slot.streamSocket.write.buffer.length;
|
||||||
|
buf.buf = () @trusted { return cast(ubyte*)slot.streamSocket.write.buffer.ptr; } ();
|
||||||
|
auto ovl = slot.streamSocket.write.mode == IOMode.immediate ? null : &slot.streamSocket.write.overlapped;
|
||||||
|
auto ret = () @trusted { return WSASend(slot.common.fd, &buf, 1, null, 0, ovl, &onIOWriteCompleted); } ();
|
||||||
|
if (ret == SOCKET_ERROR) {
|
||||||
|
auto err = WSAGetLastError();
|
||||||
|
if (err == WSA_IO_PENDING) {
|
||||||
|
if (slot.streamSocket.write.mode == IOMode.immediate) {
|
||||||
|
invokeCallback(IOStatus.wouldBlock, 0);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
invokeCallback(IOStatus.error, 0);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
override void waitForData(StreamSocketFD socket, IOCallback on_data_available)
|
override void waitForData(StreamSocketFD socket, IOCallback on_data_available)
|
||||||
|
@ -72,17 +352,26 @@ final class WinAPIEventDriverSockets : EventDriverSockets {
|
||||||
|
|
||||||
override void shutdown(StreamSocketFD socket, bool shut_read = true, bool shut_write = true)
|
override void shutdown(StreamSocketFD socket, bool shut_read = true, bool shut_write = true)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
() @trusted { WSASendDisconnect(socket, null); } ();
|
||||||
|
with (m_sockets[socket].streamSocket) {
|
||||||
|
state = ConnectionState.closed;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
override void cancelRead(StreamSocketFD socket)
|
override void cancelRead(StreamSocketFD socket)
|
||||||
{
|
@trusted {
|
||||||
assert(false, "TODO!");
|
if (!m_sockets[socket].streamSocket.read.callback) return;
|
||||||
|
CancelIoEx(cast(HANDLE)cast(SOCKET)socket, cast(LPOVERLAPPED)&m_sockets[socket].streamSocket.read.overlapped);
|
||||||
|
m_sockets[socket].streamSocket.read.callback = null;
|
||||||
|
m_core.removeWaiter();
|
||||||
}
|
}
|
||||||
|
|
||||||
override void cancelWrite(StreamSocketFD socket)
|
override void cancelWrite(StreamSocketFD socket)
|
||||||
{
|
@trusted {
|
||||||
assert(false, "TODO!");
|
if (!m_sockets[socket].streamSocket.write.callback) return;
|
||||||
|
CancelIoEx(cast(HANDLE)cast(SOCKET)socket, cast(LPOVERLAPPED)&m_sockets[socket].streamSocket.write.overlapped);
|
||||||
|
m_sockets[socket].streamSocket.write.callback = null;
|
||||||
|
m_core.removeWaiter();
|
||||||
}
|
}
|
||||||
|
|
||||||
override DatagramSocketFD createDatagramSocket(scope Address bind_address, scope Address target_address)
|
override DatagramSocketFD createDatagramSocket(scope Address bind_address, scope Address target_address)
|
||||||
|
@ -125,23 +414,206 @@ final class WinAPIEventDriverSockets : EventDriverSockets {
|
||||||
assert(false, "TODO!");
|
assert(false, "TODO!");
|
||||||
}
|
}
|
||||||
|
|
||||||
override void addRef(SocketFD descriptor)
|
override void addRef(SocketFD fd)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
assert(m_sockets[fd].common.refCount > 0, "Adding reference to unreferenced socket FD.");
|
||||||
|
m_sockets[fd].common.refCount++;
|
||||||
}
|
}
|
||||||
|
|
||||||
override bool releaseRef(SocketFD descriptor)
|
override bool releaseRef(SocketFD fd)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
import taggedalgebraic : hasType;
|
||||||
|
assert(m_sockets[fd].common.refCount > 0, "Releasing reference to unreferenced socket FD.");
|
||||||
|
if (--m_sockets[fd].common.refCount == 0) {
|
||||||
|
final switch (m_sockets[fd].specific.kind) with (SocketVector.FieldType) {
|
||||||
|
case Kind.none: break;
|
||||||
|
case Kind.streamSocket:
|
||||||
|
cancelRead(cast(StreamSocketFD)fd);
|
||||||
|
cancelWrite(cast(StreamSocketFD)fd);
|
||||||
|
break;
|
||||||
|
case Kind.streamListen:
|
||||||
|
if (m_sockets[fd].streamListen.acceptCallback)
|
||||||
|
m_core.removeWaiter();
|
||||||
|
break;
|
||||||
|
case Kind.datagramSocket:
|
||||||
|
cancelReceive(cast(DatagramSocketFD)fd);
|
||||||
|
cancelSend(cast(DatagramSocketFD)fd);
|
||||||
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
protected override void* rawUserData(StreamSocketFD descriptor, size_t size, DataInitializer initialize, DataInitializer destroy) @system
|
clearSocketSlot(fd);
|
||||||
{
|
() @trusted { closesocket(fd); } ();
|
||||||
assert(false, "TODO!");
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
protected override void* rawUserData(DatagramSocketFD descriptor, size_t size, DataInitializer initialize, DataInitializer destroy) @system
|
final protected override void* rawUserData(StreamSocketFD descriptor, size_t size, DataInitializer initialize, DataInitializer destroy)
|
||||||
|
@system {
|
||||||
|
return rawUserDataImpl(descriptor, size, initialize, destroy);
|
||||||
|
}
|
||||||
|
|
||||||
|
final protected override void* rawUserData(DatagramSocketFD descriptor, size_t size, DataInitializer initialize, DataInitializer destroy)
|
||||||
|
@system {
|
||||||
|
return rawUserDataImpl(descriptor, size, initialize, destroy);
|
||||||
|
}
|
||||||
|
|
||||||
|
private void* rawUserDataImpl(FD descriptor, size_t size, DataInitializer initialize, DataInitializer destroy)
|
||||||
|
@system {
|
||||||
|
SocketSlot* fds = &m_sockets[descriptor].common;
|
||||||
|
assert(fds.userDataDestructor is null || fds.userDataDestructor is destroy,
|
||||||
|
"Requesting user data with differing type (destructor).");
|
||||||
|
assert(size <= SocketSlot.userData.length, "Requested user data is too large.");
|
||||||
|
if (size > SocketSlot.userData.length) assert(false);
|
||||||
|
if (!fds.userDataDestructor) {
|
||||||
|
initialize(fds.userData.ptr);
|
||||||
|
fds.userDataDestructor = destroy;
|
||||||
|
}
|
||||||
|
return m_sockets[descriptor].common.userData.ptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
private void initSocketSlot(SocketFD fd)
|
||||||
{
|
{
|
||||||
assert(false, "TODO!");
|
m_sockets[fd.value].common.refCount = 1;
|
||||||
|
m_sockets[fd.value].common.fd = fd;
|
||||||
|
m_sockets[fd.value].common.core = m_core;
|
||||||
|
}
|
||||||
|
|
||||||
|
package void clearSocketSlot(FD fd)
|
||||||
|
{
|
||||||
|
auto slot = () @trusted { return &m_sockets[fd]; } ();
|
||||||
|
if (slot.common.userDataDestructor)
|
||||||
|
() @trusted { slot.common.userDataDestructor(slot.common.userData.ptr); } ();
|
||||||
|
*slot = m_sockets.FullField.init;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static nothrow extern(System)
|
||||||
|
LRESULT onMessage(HWND wnd, UINT msg, WPARAM wparam, LPARAM lparam)
|
||||||
|
{
|
||||||
|
auto driver = () @trusted { return cast(WinAPIEventDriverSockets)cast(void*)GetWindowLongPtrA(wnd, GWLP_USERDATA); } ();
|
||||||
|
switch(msg){
|
||||||
|
default: break;
|
||||||
|
case WM_USER_SOCKET:
|
||||||
|
SOCKET sock = cast(SOCKET)wparam;
|
||||||
|
auto evt = () @trusted { return LOWORD(lparam); } ();
|
||||||
|
auto err = () @trusted { return HIWORD(lparam); } ();
|
||||||
|
auto slot = () @trusted { return &driver.m_sockets[sock]; } ();
|
||||||
|
final switch (slot.specific.kind) with (SocketVector.FieldType) {
|
||||||
|
case Kind.none: break;
|
||||||
|
case Kind.streamSocket:
|
||||||
|
switch (evt) {
|
||||||
|
default: break;
|
||||||
|
case FD_CONNECT:
|
||||||
|
auto cb = slot.streamSocket.connectCallback;
|
||||||
|
slot.streamSocket.connectCallback = null;
|
||||||
|
if (err) {
|
||||||
|
slot.streamSocket.state = ConnectionState.closed;
|
||||||
|
cb(cast(StreamSocketFD)sock, ConnectStatus.refused);
|
||||||
|
} else {
|
||||||
|
slot.streamSocket.state = ConnectionState.connected;
|
||||||
|
cb(cast(StreamSocketFD)sock, ConnectStatus.connected);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case FD_READ:
|
||||||
|
break;
|
||||||
|
case FD_WRITE:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case Kind.streamListen:
|
||||||
|
if (evt == FD_ACCEPT) {
|
||||||
|
/*
|
||||||
|
sock_t sockfd;
|
||||||
|
sockaddr_storage addr;
|
||||||
|
socklen_t addr_len = addr.sizeof;
|
||||||
|
() @trusted { sockfd = accept(cast(sock_t)listenfd, () @trusted { return cast(sockaddr*)&addr; } (), &addr_len); } ();
|
||||||
|
if (sockfd == -1) break;
|
||||||
|
|
||||||
|
setSocketNonBlocking(cast(SocketFD)sockfd);
|
||||||
|
auto fd = cast(StreamSocketFD)sockfd;
|
||||||
|
initSocketSlot(fd);
|
||||||
|
m_sockets[fd].specific = StreamSocketSlot.init;
|
||||||
|
m_sockets[fd].streamSocket.state = ConnectionState.connected;
|
||||||
|
m_loop.registerFD(fd, EventMask.read|EventMask.write|EventMask.status);
|
||||||
|
m_loop.setNotifyCallback!(EventType.status)(fd, &onConnectError);
|
||||||
|
releaseRef(fd); // setNotifyCallback adds a reference, but waiting for status/disconnect should not affect the ref count
|
||||||
|
//print("accept %d", sockfd);
|
||||||
|
scope RefAddress addrc = new RefAddress(() @trusted { return cast(sockaddr*)&addr; } (), addr_len);
|
||||||
|
m_sockets[listenfd].streamListen.acceptCallback(cast(StreamListenSocketFD)listenfd, fd, addrc);
|
||||||
|
*/
|
||||||
|
SOCKADDR_STORAGE addr;
|
||||||
|
socklen_t addr_len = addr.sizeof;
|
||||||
|
auto clientsockfd = () @trusted { return WSAAccept(sock, cast(sockaddr*)&addr, &addr_len, null, 0); } ();
|
||||||
|
if (clientsockfd == INVALID_SOCKET) return 0;
|
||||||
|
auto clientsock = driver.adoptStreamInternal(clientsockfd);
|
||||||
|
scope RefAddress addrc = new RefAddress(() @trusted { return cast(sockaddr*)&addr; } (), addr_len);
|
||||||
|
slot.streamListen.acceptCallback(cast(StreamListenSocketFD)sock, clientsock, addrc);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case Kind.datagramSocket:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
return () @trusted { return DefWindowProcA(wnd, msg, wparam, lparam); } ();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void setupWindowClass() nothrow
|
||||||
|
@trusted {
|
||||||
|
static __gshared registered = false;
|
||||||
|
|
||||||
|
if (registered) return;
|
||||||
|
|
||||||
|
WNDCLASSA wc;
|
||||||
|
wc.lpfnWndProc = &WinAPIEventDriverSockets.onMessage;
|
||||||
|
wc.lpszClassName = "VibeWin32MessageWindow";
|
||||||
|
RegisterClassA(&wc);
|
||||||
|
registered = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
static struct SocketSlot {
|
||||||
|
SocketFD fd; // redundant, but needed by the current IO Completion Routines based approach
|
||||||
|
WinAPIEventDriverCore core; // redundant, but needed by the current IO Completion Routines based approach
|
||||||
|
int refCount;
|
||||||
|
DataInitializer userDataDestructor;
|
||||||
|
ubyte[16*size_t.sizeof] userData;
|
||||||
|
}
|
||||||
|
|
||||||
|
private struct StreamSocketSlot {
|
||||||
|
alias Handle = StreamSocketFD;
|
||||||
|
StreamDirection!true write;
|
||||||
|
StreamDirection!false read;
|
||||||
|
ConnectCallback connectCallback;
|
||||||
|
ConnectionState state;
|
||||||
|
}
|
||||||
|
|
||||||
|
static struct StreamDirection(bool RO) {
|
||||||
|
WSAOVERLAPPEDX overlapped;
|
||||||
|
static if (RO) const(ubyte)[] buffer;
|
||||||
|
else ubyte[] buffer;
|
||||||
|
size_t bytesTransferred;
|
||||||
|
IOMode mode;
|
||||||
|
IOCallback callback;
|
||||||
|
}
|
||||||
|
|
||||||
|
private struct StreamListenSocketSlot {
|
||||||
|
alias Handle = StreamListenSocketFD;
|
||||||
|
AcceptCallback acceptCallback;
|
||||||
|
}
|
||||||
|
|
||||||
|
private struct DgramSocketSlot {
|
||||||
|
alias Handle = DatagramSocketFD;
|
||||||
|
DgramDirection!true write;
|
||||||
|
DgramDirection!false read;
|
||||||
|
Address targetAddr;
|
||||||
|
}
|
||||||
|
|
||||||
|
static struct DgramDirection(bool RO) {
|
||||||
|
WSAOVERLAPPEDX overlapped;
|
||||||
|
static if (RO) const(ubyte)[] buffer;
|
||||||
|
else ubyte[] buffer;
|
||||||
|
size_t bytesTransferred;
|
||||||
|
IOMode mode;
|
||||||
|
DatagramIOCallback callback;
|
||||||
|
}
|
||||||
|
|
|
@ -5,7 +5,7 @@ version(Windows):
|
||||||
public import core.sys.windows.windows;
|
public import core.sys.windows.windows;
|
||||||
public import core.sys.windows.winsock2;
|
public import core.sys.windows.winsock2;
|
||||||
|
|
||||||
extern(System) nothrow @nogc:
|
extern(System) nothrow:
|
||||||
|
|
||||||
enum {
|
enum {
|
||||||
FD_READ = 0x0001,
|
FD_READ = 0x0001,
|
||||||
|
@ -130,22 +130,25 @@ alias LPLOOKUPSERVICE_COMPLETION_ROUTINE = void function(DWORD, DWORD, WSAOVERLA
|
||||||
alias LPCONDITIONPROC = void*;
|
alias LPCONDITIONPROC = void*;
|
||||||
alias LPTRANSMIT_FILE_BUFFERS = void*;
|
alias LPTRANSMIT_FILE_BUFFERS = void*;
|
||||||
|
|
||||||
|
int WSARecv(SOCKET s, WSABUF* lpBuffers, DWORD dwBufferCount, DWORD* lpNumberOfBytesRecvd, DWORD* lpFlags, in WSAOVERLAPPEDX* lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINEX lpCompletionRoutine);
|
||||||
|
int WSASend(SOCKET s, in WSABUF* lpBuffers, DWORD dwBufferCount, DWORD* lpNumberOfBytesSent, DWORD dwFlags, in WSAOVERLAPPEDX* lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINEX lpCompletionRoutine);
|
||||||
|
int GetAddrInfoExW(LPCWSTR pName, LPCWSTR pServiceName, DWORD dwNameSpace, GUID* lpNspId, const ADDRINFOEXW *pHints, ADDRINFOEXW **ppResult, timeval *timeout, WSAOVERLAPPEDX* lpOverlapped, LPLOOKUPSERVICE_COMPLETION_ROUTINE lpCompletionRoutine, HANDLE* lpNameHandle);
|
||||||
|
|
||||||
|
@nogc:
|
||||||
SOCKET WSAAccept(SOCKET s, sockaddr *addr, INT* addrlen, LPCONDITIONPROC lpfnCondition, DWORD_PTR dwCallbackData);
|
SOCKET WSAAccept(SOCKET s, sockaddr *addr, INT* addrlen, LPCONDITIONPROC lpfnCondition, DWORD_PTR dwCallbackData);
|
||||||
int WSAAsyncSelect(SOCKET s, HWND hWnd, uint wMsg, sizediff_t lEvent);
|
int WSAAsyncSelect(SOCKET s, HWND hWnd, uint wMsg, sizediff_t lEvent);
|
||||||
SOCKET WSASocketW(int af, int type, int protocol, WSAPROTOCOL_INFOW *lpProtocolInfo, uint g, DWORD dwFlags);
|
SOCKET WSASocketW(int af, int type, int protocol, WSAPROTOCOL_INFOW *lpProtocolInfo, uint g, DWORD dwFlags);
|
||||||
int WSARecv(SOCKET s, WSABUF* lpBuffers, DWORD dwBufferCount, DWORD* lpNumberOfBytesRecvd, DWORD* lpFlags, in WSAOVERLAPPEDX* lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINEX lpCompletionRoutine);
|
|
||||||
int WSASend(SOCKET s, in WSABUF* lpBuffers, DWORD dwBufferCount, DWORD* lpNumberOfBytesSent, DWORD dwFlags, in WSAOVERLAPPEDX* lpOverlapped, LPWSAOVERLAPPED_COMPLETION_ROUTINEX lpCompletionRoutine);
|
|
||||||
int WSASendDisconnect(SOCKET s, WSABUF* lpOutboundDisconnectData);
|
int WSASendDisconnect(SOCKET s, WSABUF* lpOutboundDisconnectData);
|
||||||
INT WSAStringToAddressA(in LPTSTR AddressString, INT AddressFamily, in WSAPROTOCOL_INFO* lpProtocolInfo, SOCKADDR* lpAddress, INT* lpAddressLength);
|
INT WSAStringToAddressA(in LPTSTR AddressString, INT AddressFamily, in WSAPROTOCOL_INFO* lpProtocolInfo, SOCKADDR* lpAddress, INT* lpAddressLength);
|
||||||
INT WSAStringToAddressW(in LPWSTR AddressString, INT AddressFamily, in WSAPROTOCOL_INFOW* lpProtocolInfo, SOCKADDR* lpAddress, INT* lpAddressLength);
|
INT WSAStringToAddressW(in LPWSTR AddressString, INT AddressFamily, in WSAPROTOCOL_INFOW* lpProtocolInfo, SOCKADDR* lpAddress, INT* lpAddressLength);
|
||||||
INT WSAAddressToStringW(in SOCKADDR* lpsaAddress, DWORD dwAddressLength, in WSAPROTOCOL_INFO* lpProtocolInfo, LPWSTR lpszAddressString, DWORD* lpdwAddressStringLength);
|
INT WSAAddressToStringW(in SOCKADDR* lpsaAddress, DWORD dwAddressLength, in WSAPROTOCOL_INFO* lpProtocolInfo, LPWSTR lpszAddressString, DWORD* lpdwAddressStringLength);
|
||||||
int GetAddrInfoExW(LPCWSTR pName, LPCWSTR pServiceName, DWORD dwNameSpace, GUID* lpNspId, const ADDRINFOEXW *pHints, ADDRINFOEXW **ppResult, timeval *timeout, WSAOVERLAPPEDX* lpOverlapped, LPLOOKUPSERVICE_COMPLETION_ROUTINE lpCompletionRoutine, HANDLE* lpNameHandle);
|
|
||||||
int GetAddrInfoW(LPCWSTR pName, LPCWSTR pServiceName, const ADDRINFOW *pHints, ADDRINFOW **ppResult);
|
int GetAddrInfoW(LPCWSTR pName, LPCWSTR pServiceName, const ADDRINFOW *pHints, ADDRINFOW **ppResult);
|
||||||
int getaddrinfo(LPCSTR pName, LPCSTR pServiceName, const ADDRINFOA *pHints, ADDRINFOA **ppResult);
|
int getaddrinfo(LPCSTR pName, LPCSTR pServiceName, const ADDRINFOA *pHints, ADDRINFOA **ppResult);
|
||||||
void FreeAddrInfoW(ADDRINFOW* pAddrInfo);
|
void FreeAddrInfoW(ADDRINFOW* pAddrInfo);
|
||||||
void FreeAddrInfoExW(ADDRINFOEXW* pAddrInfo);
|
void FreeAddrInfoExW(ADDRINFOEXW* pAddrInfo);
|
||||||
void freeaddrinfo(ADDRINFOA* ai);
|
void freeaddrinfo(ADDRINFOA* ai);
|
||||||
BOOL TransmitFile(SOCKET hSocket, HANDLE hFile, DWORD nNumberOfBytesToWrite, DWORD nNumberOfBytesPerSend, OVERLAPPED* lpOverlapped, LPTRANSMIT_FILE_BUFFERS lpTransmitBuffers, DWORD dwFlags);
|
BOOL TransmitFile(SOCKET hSocket, HANDLE hFile, DWORD nNumberOfBytesToWrite, DWORD nNumberOfBytesPerSend, OVERLAPPED* lpOverlapped, LPTRANSMIT_FILE_BUFFERS lpTransmitBuffers, DWORD dwFlags);
|
||||||
|
BOOL CancelIoEx(HANDLE hFile, LPOVERLAPPED lpOverlapped);
|
||||||
|
|
||||||
struct WSAOVERLAPPEDX {
|
struct WSAOVERLAPPEDX {
|
||||||
ULONG_PTR Internal;
|
ULONG_PTR Internal;
|
||||||
|
|
Loading…
Reference in a new issue