1 /* IOEngine_win32.c - IOMultiplexer
2 * Copyright (C) 2014 Philipp Kreil (pk910)
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
17 #define _IOHandler_internals
18 #include "IOInternal.h"
19 #include "IOHandler.h"
21 #include "IOSockets.h"
26 #define _WIN32_WINNT 0x501
30 /* This is massively kludgy. Unfortunately, the only performant I/O
31 * multiplexer with halfway decent semantics under Windows is
32 * WSAAsyncSelect() -- which requires a window that can receive
35 * So ioset_win32_init() creates a hidden window and sets it up for
36 * asynchronous socket notifications.
39 #define IDT_TIMER1 1000
40 #define IDT_TIMER2 1001
41 #define IDT_SOCKET 1002
43 static HWND ioset_window;
45 static struct _IOSocket *engine_win32_get_iosock(int fd) {
46 struct _IOSocket *iosock;
47 for(iosock = iosocket_first; iosock; iosock = iosock->next) {
54 static LRESULT CALLBACK engine_win32_wndproc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam) {
55 struct _IOSocket *iosock;
58 if (hWnd == ioset_window) {
67 iosock = engine_win32_get_iosock(wParam);
68 events = WSAGETSELECTEVENT(lParam);
70 if((events & FD_CONNECT)) {
72 if((err = WSAGETSELECTERROR(lParam)))
73 iosocket_events_callback(iosock, err, 0);
75 iosocket_events_callback(iosock, 0, 1);
77 iosocket_events_callback(iosock, (events & (FD_READ | FD_ACCEPT | FD_CLOSE)) != 0, (events & FD_WRITE) != 0);
83 return DefWindowProc(hWnd, uMsg, wParam, lParam);
86 static int engine_win32_init() {
91 // Start Windows Sockets.
92 if (WSAStartup(MAKEWORD(2, 0), &wsadata)) {
93 iolog_trigger(IOLOG_FATAL, "Unable to start Windows Sockets");
97 // Get Windows HINSTANCE.
98 hinst = GetModuleHandle(NULL);
100 // Describe and register a window class.
101 memset(&wcx, 0, sizeof(wcx));
102 wcx.cbSize = sizeof(wcx);
103 wcx.lpfnWndProc = engine_win32_wndproc;
104 wcx.hInstance = hinst;
105 wcx.lpszClassName = "IOMultiplexerMainWindow";
106 if (!RegisterClassEx(&wcx))
109 ioset_window = CreateWindow("IOMultiplexerMainWindow", "IOMultiplexer", WS_OVERLAPPEDWINDOW, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, NULL, NULL, hinst, NULL);
115 static long engine_win32_events(struct _IOSocket *iosock) {
116 if(iosock->socket_flags & IOSOCKETFLAG_LISTENING)
118 if(iosock->socket_flags & IOSOCKETFLAG_CONNECTING)
121 return FD_READ | FD_CLOSE | (iosocket_wants_writes(iosock) ? FD_WRITE : 0);
124 static void engine_win32_update(struct _IOSocket *iosock) {
126 events = engine_win32_events(iosock);
127 WSAAsyncSelect(iosock->fd, ioset_window, IDT_SOCKET, events);
130 static void engine_win32_add(struct _IOSocket *iosock) {
131 engine_win32_update(iosock);
134 static void engine_win32_remove(struct _IOSocket *iosock) {
135 unsigned long ulong = 0;
136 WSAAsyncSelect(iosock->fd, ioset_window, IDT_SOCKET, 0);
137 ioctlsocket(iosock->fd, FIONBIO, &ulong);
140 static void engine_win32_loop(struct timeval *timeout) {
147 gettimeofday(&now, NULL);
148 if(iotimer_sorted_descriptors && timeval_is_bigger(now, iotimer_sorted_descriptors->timeout))
151 //get timeout (timer or given timeout)
152 if(iotimer_sorted_descriptors) {
153 msec = (iotimer_sorted_descriptors->timeout.tv_sec - now.tv_sec) * 1000;
154 msec += (iotimer_sorted_descriptors->timeout.tv_usec - now.tv_usec) / 1000;
157 msec2 = (timeout->tv_sec * 1000 + timeout->tv_usec / 1000);
158 if(!iotimer_sorted_descriptors || msec2 < msec)
160 } else if(!iotimer_sorted_descriptors)
164 SetTimer(ioset_window, IDT_TIMER1, 1000, NULL);
166 SetTimer(ioset_window, IDT_TIMER2, msec, NULL);
168 //GetMessage system call
169 res = GetMessage(&msg, NULL, 0, 0);
172 KillTimer(ioset_window, IDT_TIMER1);
174 KillTimer(ioset_window, IDT_TIMER2);
179 TranslateMessage(&msg);
180 DispatchMessage(&msg);
184 static void engine_win32_cleanup() {
185 DestroyWindow(ioset_window);
190 struct IOEngine engine_win32 = {
192 .init = engine_win32_init,
193 .add = engine_win32_add,
194 .remove = engine_win32_remove,
195 .update = engine_win32_update,
196 .loop = engine_win32_loop,
197 .cleanup = engine_win32_cleanup,
202 struct IOEngine engine_win32 = {