| 1 | /* ------------------------------------------------------------------------- |
| 2 | * Project: GSocket (Generic Socket) |
| 3 | * Name: gsockmsw.c |
| 4 | * Author: Guillermo Rodriguez Garcia <guille@iies.es> |
| 5 | * Purpose: GSocket GUI-specific MSW code |
| 6 | * CVSID: $Id$ |
| 7 | * ------------------------------------------------------------------------- |
| 8 | */ |
| 9 | |
| 10 | /* |
| 11 | * PLEASE don't put C++ comments here - this is a C source file. |
| 12 | */ |
| 13 | |
| 14 | /* including rasasync.h (included from windows.h itself included from |
| 15 | * wx/setup.h and/or winsock.h results in this warning for |
| 16 | * RPCNOTIFICATION_ROUTINE |
| 17 | */ |
| 18 | #ifdef _MSC_VER |
| 19 | # pragma warning(disable:4115) /* named type definition in parentheses */ |
| 20 | #endif |
| 21 | |
| 22 | #ifndef __GSOCKET_STANDALONE__ |
| 23 | #include "wx/defs.h" |
| 24 | #include "wx/setup.h" |
| 25 | #endif |
| 26 | |
| 27 | #if wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__) |
| 28 | |
| 29 | #ifndef __GSOCKET_STANDALONE__ |
| 30 | |
| 31 | #include "wx/msw/gsockmsw.h" |
| 32 | #include "wx/gsocket.h" |
| 33 | |
| 34 | #define INSTANCE wxGetInstance() |
| 35 | |
| 36 | #else |
| 37 | |
| 38 | #include "gsockmsw.h" |
| 39 | #include "gsocket.h" |
| 40 | |
| 41 | /* If not using wxWindows, a global var called hInst must |
| 42 | * be available and it must contain the app's instance |
| 43 | * handle. |
| 44 | */ |
| 45 | #define INSTANCE hInst |
| 46 | |
| 47 | #endif /* __GSOCKET_STANDALONE__ */ |
| 48 | |
| 49 | #include <assert.h> |
| 50 | #include <string.h> |
| 51 | #include <stdio.h> |
| 52 | #include <stdlib.h> |
| 53 | #include <stddef.h> |
| 54 | #include <ctype.h> |
| 55 | |
| 56 | #include <winsock.h> |
| 57 | |
| 58 | #ifdef _MSC_VER |
| 59 | # pragma warning(default:4115) /* named type definition in parentheses */ |
| 60 | #endif |
| 61 | |
| 62 | #define CLASSNAME TEXT("_GSocket_Internal_Window_Class") |
| 63 | #define WINDOWNAME TEXT("_GSocket_Internal_Window_Name") |
| 64 | |
| 65 | /* Maximum number of different GSocket objects at a given time. |
| 66 | * This value can be modified at will, but it CANNOT be greater |
| 67 | * than (0x7FFF - WM_USER + 1) |
| 68 | */ |
| 69 | #define MAXSOCKETS 1024 |
| 70 | |
| 71 | #if (MAXSOCKETS > (0x7FFF - WM_USER + 1)) |
| 72 | #error "MAXSOCKETS is too big!" |
| 73 | #endif |
| 74 | |
| 75 | |
| 76 | /* Global variables */ |
| 77 | |
| 78 | extern HINSTANCE INSTANCE; |
| 79 | static HWND hWin; |
| 80 | static CRITICAL_SECTION critical; |
| 81 | static GSocket* socketList[MAXSOCKETS]; |
| 82 | static int firstAvailable; |
| 83 | |
| 84 | /* Global initializers */ |
| 85 | |
| 86 | int GSocket_Init(void) |
| 87 | { |
| 88 | WSADATA wsaData; |
| 89 | WNDCLASS winClass; |
| 90 | int i; |
| 91 | |
| 92 | /* Create internal window for event notifications */ |
| 93 | winClass.style = 0; |
| 94 | winClass.lpfnWndProc = _GSocket_Internal_WinProc; |
| 95 | winClass.cbClsExtra = 0; |
| 96 | winClass.cbWndExtra = 0; |
| 97 | winClass.hInstance = INSTANCE; |
| 98 | winClass.hIcon = (HICON) NULL; |
| 99 | winClass.hCursor = (HCURSOR) NULL; |
| 100 | winClass.hbrBackground = (HBRUSH) NULL; |
| 101 | winClass.lpszMenuName = (LPCTSTR) NULL; |
| 102 | winClass.lpszClassName = CLASSNAME; |
| 103 | |
| 104 | RegisterClass(&winClass); |
| 105 | hWin = CreateWindow(CLASSNAME, |
| 106 | WINDOWNAME, |
| 107 | 0, 0, 0, 0, 0, |
| 108 | (HWND) NULL, (HMENU) NULL, INSTANCE, (LPVOID) NULL); |
| 109 | |
| 110 | if (!hWin) return FALSE; |
| 111 | |
| 112 | /* Initialize socket list */ |
| 113 | InitializeCriticalSection(&critical); |
| 114 | |
| 115 | for (i = 0; i < MAXSOCKETS; i++) |
| 116 | { |
| 117 | socketList[i] = NULL; |
| 118 | } |
| 119 | firstAvailable = 0; |
| 120 | |
| 121 | /* Initialize WinSocket */ |
| 122 | return (WSAStartup((1 << 8) | 1, &wsaData) == 0); |
| 123 | } |
| 124 | |
| 125 | void GSocket_Cleanup(void) |
| 126 | { |
| 127 | /* Destroy internal window */ |
| 128 | DestroyWindow(hWin); |
| 129 | UnregisterClass(CLASSNAME, INSTANCE); |
| 130 | |
| 131 | /* Delete critical section */ |
| 132 | DeleteCriticalSection(&critical); |
| 133 | |
| 134 | /* Cleanup WinSocket */ |
| 135 | WSACleanup(); |
| 136 | } |
| 137 | |
| 138 | /* Per-socket GUI initialization / cleanup */ |
| 139 | |
| 140 | int _GSocket_GUI_Init(GSocket *socket) |
| 141 | { |
| 142 | int i; |
| 143 | |
| 144 | /* Allocate a new message number for this socket */ |
| 145 | EnterCriticalSection(&critical); |
| 146 | |
| 147 | i = firstAvailable; |
| 148 | while (socketList[i] != NULL) |
| 149 | { |
| 150 | i = (i + 1) % MAXSOCKETS; |
| 151 | |
| 152 | if (i == firstAvailable) /* abort! */ |
| 153 | { |
| 154 | LeaveCriticalSection(&critical); |
| 155 | return FALSE; |
| 156 | } |
| 157 | } |
| 158 | socketList[i] = socket; |
| 159 | firstAvailable = (i + 1) % MAXSOCKETS; |
| 160 | socket->m_msgnumber = (i + WM_USER); |
| 161 | |
| 162 | LeaveCriticalSection(&critical); |
| 163 | |
| 164 | return TRUE; |
| 165 | } |
| 166 | |
| 167 | void _GSocket_GUI_Destroy(GSocket *socket) |
| 168 | { |
| 169 | /* Remove the socket from the list */ |
| 170 | EnterCriticalSection(&critical); |
| 171 | socketList[(socket->m_msgnumber - WM_USER)] = NULL; |
| 172 | LeaveCriticalSection(&critical); |
| 173 | } |
| 174 | |
| 175 | /* Windows proc for asynchronous event handling */ |
| 176 | |
| 177 | LRESULT CALLBACK _GSocket_Internal_WinProc(HWND hWnd, |
| 178 | UINT uMsg, |
| 179 | WPARAM wParam, |
| 180 | LPARAM lParam) |
| 181 | { |
| 182 | GSocket *socket; |
| 183 | GSocketEvent event; |
| 184 | GSocketCallback cback; |
| 185 | char *data; |
| 186 | |
| 187 | if (uMsg >= WM_USER && uMsg <= (WM_USER + MAXSOCKETS - 1)) |
| 188 | { |
| 189 | EnterCriticalSection(&critical); |
| 190 | socket = socketList[(uMsg - WM_USER)]; |
| 191 | event = -1; |
| 192 | cback = NULL; |
| 193 | data = NULL; |
| 194 | |
| 195 | /* Check that the socket still exists (it has not been |
| 196 | * destroyed) and for safety, check that the m_fd field |
| 197 | * is what we expect it to be. |
| 198 | */ |
| 199 | if ((socket != NULL) && (socket->m_fd == wParam)) |
| 200 | { |
| 201 | switch WSAGETSELECTEVENT(lParam) |
| 202 | { |
| 203 | case FD_READ: event = GSOCK_INPUT; break; |
| 204 | case FD_WRITE: event = GSOCK_OUTPUT; break; |
| 205 | case FD_ACCEPT: event = GSOCK_CONNECTION; break; |
| 206 | case FD_CONNECT: |
| 207 | { |
| 208 | if (WSAGETSELECTERROR(lParam) != 0) |
| 209 | event = GSOCK_LOST; |
| 210 | else |
| 211 | event = GSOCK_CONNECTION; |
| 212 | break; |
| 213 | } |
| 214 | case FD_CLOSE: event = GSOCK_LOST; break; |
| 215 | } |
| 216 | |
| 217 | if (event != -1) |
| 218 | { |
| 219 | cback = socket->m_cbacks[event]; |
| 220 | data = socket->m_data[event]; |
| 221 | |
| 222 | if (event == GSOCK_LOST) |
| 223 | socket->m_detected = GSOCK_LOST_FLAG; |
| 224 | else |
| 225 | socket->m_detected |= (1 << event); |
| 226 | } |
| 227 | } |
| 228 | |
| 229 | /* OK, we can now leave the critical section because we have |
| 230 | * already obtained the callback address (we make no further |
| 231 | * accesses to socket->whatever). However, the app should |
| 232 | * be prepared to handle events from a socket that has just |
| 233 | * been closed! |
| 234 | */ |
| 235 | LeaveCriticalSection(&critical); |
| 236 | |
| 237 | if (cback != NULL) |
| 238 | (cback)(socket, event, data); |
| 239 | |
| 240 | return (LRESULT) 0; |
| 241 | } |
| 242 | else |
| 243 | return DefWindowProc(hWnd, uMsg, wParam, lParam); |
| 244 | } |
| 245 | |
| 246 | /* _GSocket_Enable_Events: |
| 247 | * Enable all event notifications; we need to be notified of all |
| 248 | * events for internal processing, but we will only notify users |
| 249 | * when an appropiate callback function has been installed. |
| 250 | */ |
| 251 | void _GSocket_Enable_Events(GSocket *socket) |
| 252 | { |
| 253 | assert (socket != NULL); |
| 254 | |
| 255 | if (socket->m_fd != INVALID_SOCKET) |
| 256 | { |
| 257 | /* We could probably just subscribe to all events regardless |
| 258 | * of the socket type, but MS recommends to do it this way. |
| 259 | */ |
| 260 | long lEvent = socket->m_server? |
| 261 | FD_ACCEPT : (FD_READ | FD_WRITE | FD_CONNECT | FD_CLOSE); |
| 262 | |
| 263 | WSAAsyncSelect(socket->m_fd, hWin, socket->m_msgnumber, lEvent); |
| 264 | } |
| 265 | } |
| 266 | |
| 267 | /* _GSocket_Disable_Events: |
| 268 | * Disable event notifications (when shutdowning the socket) |
| 269 | */ |
| 270 | void _GSocket_Disable_Events(GSocket *socket) |
| 271 | { |
| 272 | assert (socket != NULL); |
| 273 | |
| 274 | if (socket->m_fd != INVALID_SOCKET) |
| 275 | { |
| 276 | WSAAsyncSelect(socket->m_fd, hWin, socket->m_msgnumber, 0); |
| 277 | } |
| 278 | } |
| 279 | |
| 280 | #else /* !wxUSE_SOCKETS */ |
| 281 | |
| 282 | /* |
| 283 | * Translation unit shouldn't be empty, so include this typedef to make the |
| 284 | * compiler (VC++ 6.0, for example) happy |
| 285 | */ |
| 286 | typedef void (*wxDummy)(); |
| 287 | |
| 288 | #endif /* wxUSE_SOCKETS || defined(__GSOCKET_STANDALONE__) */ |