2 This file is part of GNUnet.
3 (C) 2012 Christian Grothoff (and other contributing authors)
5 GNUnet is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published
7 by the Free Software Foundation; either version 3, or (at your
8 option) any later version.
10 GNUnet is distributed in the hope that it will be useful, but
11 WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with GNUnet; see the file COPYING. If not, write to the
17 Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18 Boston, MA 02111-1307, USA.
22 * @brief W32 integration for GNS
25 /* This code is partially based upon samples from the book
26 * "Network Programming For Microsoft Windows, 2Nd Edition".
37 # define DEBUGLOG(s, ...)
40 # define DEBUGLOG(s, ...) printf (s, ##__VA_ARGS__)
47 #define __BYTE_ORDER _BYTE_ORDER
50 #define __BYTE_ORDER BYTE_ORDER
56 #define __BIG_ENDIAN _BIG_ENDIAN
59 #define __BIG_ENDIAN BIG_ENDIAN
63 #ifndef __LITTLE_ENDIAN
65 #define __LITTLE_ENDIAN _LITTLE_ENDIAN
68 #define __LITTLE_ENDIAN LITTLE_ENDIAN
72 #include "gnunet_w32nsp_lib.h"
73 #include "w32resolver.h"
75 #define NSPAPI_VERSION_MAJOR 4
76 #define NSPAPI_VERSION_MINOR 4
78 #define REPLY_LIFETIME 60*5
80 #define STATE_BEGIN 0x01
81 #define STATE_END 0x02
82 #define STATE_REPLY 0x04
83 #define STATE_GHBN 0x08
85 CRITICAL_SECTION records_cs;
96 static struct record *records = NULL;
97 static size_t records_len = 0;
98 static size_t records_size = 0;
103 size_t new_size = records_len > 0 ? records_len * 2 : 5;
104 struct record *new_records = malloc (new_size * sizeof (struct record));
105 if (new_records == NULL)
107 SetLastError (WSA_NOT_ENOUGH_MEMORY);
110 memcpy (new_records, records, records_len * sizeof (struct record));
111 memset (&new_records[records_len], 0, sizeof (struct record) * (new_size - records_len));
112 records_size = new_size;
114 records = new_records;
119 add_record (SOCKET s, const wchar_t *name, DWORD flags)
124 //EnterCriticalSection (&records_cs);
125 for (i = 0; i < records_len; i++)
126 if (records[i].state == 0)
129 if (i == records_len)
131 res = resize_records ();
133 empty = records_len++;
140 r.name = (wchar_t *) name;
144 r.name = wcsdup (name);
147 //LeaveCriticalSection (&records_cs);
155 free (records[i].name);
156 records[i].state = 0;
159 /* These are not defined by mingw.org headers at the moment*/
160 typedef INT (WSPAPI *LPNSPIOCTL) (HANDLE,DWORD,LPVOID,DWORD,LPVOID,DWORD,LPDWORD,LPWSACOMPLETION,LPWSATHREADID);
161 typedef struct _NSP_ROUTINE_XP {
163 DWORD dwMajorVersion;
164 DWORD dwMinorVersion;
165 LPNSPCLEANUP NSPCleanup;
166 LPNSPLOOKUPSERVICEBEGIN NSPLookupServiceBegin;
167 LPNSPLOOKUPSERVICENEXT NSPLookupServiceNext;
168 LPNSPLOOKUPSERVICEEND NSPLookupServiceEnd;
169 LPNSPSETSERVICE NSPSetService;
170 LPNSPINSTALLSERVICECLASS NSPInstallServiceClass;
171 LPNSPREMOVESERVICECLASS NSPRemoveServiceClass;
172 LPNSPGETSERVICECLASSINFO NSPGetServiceClassInfo;
177 connect_to_dns_resolver ()
179 struct sockaddr_in addr;
183 r = socket (AF_INET, SOCK_STREAM, IPPROTO_TCP);
184 if (INVALID_SOCKET == r)
186 SetLastError (16004);
190 addr.sin_family = AF_INET;
191 addr.sin_port = htons (5353); /* TCP 5353 is not registered; UDP 5353 is */
192 addr.sin_addr.s_addr = inet_addr ("127.0.0.1");
194 ret = connect (r, (struct sockaddr *) &addr, sizeof (addr));
195 if (SOCKET_ERROR == ret)
197 DWORD err = GetLastError ();
200 SetLastError (16005);
207 send_name_to_ip_request (LPWSAQUERYSETW lpqsRestrictions,
208 LPWSASERVICECLASSINFOW lpServiceClassInfo, DWORD dwControlFlags,
211 struct GNUNET_W32RESOLVER_GetMessage *msg;
218 size_t size = sizeof (struct GNUNET_W32RESOLVER_GetMessage);
220 if (lpqsRestrictions->lpszServiceInstanceName)
221 namelen = sizeof (wchar_t) * (wcslen (lpqsRestrictions->lpszServiceInstanceName) + 1);
224 msg = (struct GNUNET_W32RESOLVER_GetMessage *) buf;
225 msg->header.size = htons (size);
226 msg->header.type = htons (GNUNET_MESSAGE_TYPE_W32RESOLVER_REQUEST);
227 if (lpqsRestrictions->dwNumberOfProtocols > 0)
230 for (i = 0; i < lpqsRestrictions->dwNumberOfProtocols; i++)
232 if (lpqsRestrictions->lpafpProtocols[0].iAddressFamily == AF_INET)
234 if (lpqsRestrictions->lpafpProtocols[0].iAddressFamily == AF_INET6)
239 msg->af = htonl (AF_INET);
240 else if (af6 && !af4)
241 msg->af = htonl (AF_INET6);
243 msg->af = htonl (AF_UNSPEC);
244 if (lpqsRestrictions->lpszServiceInstanceName)
245 memcpy (&msg[1], lpqsRestrictions->lpszServiceInstanceName, namelen);
246 msg->sc_data1 = htonl (lpqsRestrictions->lpServiceClassId->Data1);
247 msg->sc_data2 = htons (lpqsRestrictions->lpServiceClassId->Data2);
248 msg->sc_data3 = htons (lpqsRestrictions->lpServiceClassId->Data3);
249 for (i = 0; i < 8; i++)
250 msg->sc_data4[i] = lpqsRestrictions->lpServiceClassId->Data4[i];
251 *resolver = connect_to_dns_resolver ();
252 if (*resolver != INVALID_SOCKET)
254 if (size != send (*resolver, buf, size, 0))
256 DWORD err = GetLastError ();
257 closesocket (*resolver);
258 *resolver = INVALID_SOCKET;
259 DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: failed to send request: %lu\n", err);
260 SetLastError (WSATRY_AGAIN);
271 NSPCleanup (LPGUID lpProviderId)
273 DEBUGLOG ("NSPCleanup\n");
274 if (IsEqualGUID (lpProviderId, &GNUNET_NAMESPACE_PROVIDER_DNS))
278 SetLastError (WSAEINVALIDPROVIDER);
283 DllMain (HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
287 case DLL_PROCESS_ATTACH:
288 if (!InitializeCriticalSectionAndSpinCount (&records_cs, 0x00000400))
293 case DLL_THREAD_ATTACH:
295 case DLL_THREAD_DETACH:
297 case DLL_PROCESS_DETACH:
298 DeleteCriticalSection (&records_cs);
308 GNUNET_W32NSP_LookupServiceBegin (LPGUID lpProviderId, LPWSAQUERYSETW lpqsRestrictions,
309 LPWSASERVICECLASSINFOW lpServiceClassInfo, DWORD dwControlFlags,
312 DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin\n");
313 if (IsEqualGUID (lpProviderId, &GNUNET_NAMESPACE_PROVIDER_DNS))
316 if (lpqsRestrictions->dwNameSpace != NS_DNS && lpqsRestrictions->dwNameSpace != NS_ALL)
318 DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: wrong namespace\n");
319 SetLastError (WSANO_DATA);
322 if (lpqsRestrictions->lpszServiceInstanceName != NULL)
324 wchar_t *s = lpqsRestrictions->lpszServiceInstanceName;
325 size_t len = wcslen (s);
326 if (len >= 5 && wcscmp (&s[len - 5], L".zkey") == 0)
329 else if (len >= 4 && wcscmp (&s[len - 4], L".gnu") == 0)
334 DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: unsupported TLD\n");
335 SetLastError (WSANO_DATA);
340 if (send_name_to_ip_request (lpqsRestrictions,
341 lpServiceClassInfo, dwControlFlags, &s))
343 if (!(add_record (s, lpqsRestrictions->lpszServiceInstanceName, dwControlFlags)))
345 DWORD err = GetLastError ();
346 DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: failed to add a record\n");
351 *lphLookup = (HANDLE) s;
352 DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: OK (%lu)\n", GetLastError ());
357 DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: wrong provider\n");
358 SetLastError (WSAEINVALIDPROVIDER);
362 #define UnmarshallPtr(ptr, ptrtype, base) \
364 ptr = (ptrtype *) (base + (uintptr_t) ptr)
367 UnmarshallWSAQUERYSETW (LPWSAQUERYSETW req)
370 char *base = (char *) req;
371 UnmarshallPtr (req->lpszServiceInstanceName, wchar_t, base);
372 UnmarshallPtr (req->lpServiceClassId, GUID, base);
373 UnmarshallPtr (req->lpVersion, WSAVERSION, base);
374 UnmarshallPtr (req->lpszComment, wchar_t, base);
375 UnmarshallPtr (req->lpNSProviderId, GUID, base);
376 UnmarshallPtr (req->lpszContext, wchar_t, base);
377 UnmarshallPtr (req->lpafpProtocols, AFPROTOCOLS, base);
378 UnmarshallPtr (req->lpszQueryString, wchar_t, base);
379 UnmarshallPtr (req->lpcsaBuffer, CSADDR_INFO, base);
380 for (i = 0; i < req->dwNumberOfCsAddrs; i++)
382 UnmarshallPtr (req->lpcsaBuffer[i].LocalAddr.lpSockaddr, SOCKADDR, base);
383 UnmarshallPtr (req->lpcsaBuffer[i].RemoteAddr.lpSockaddr, SOCKADDR, base);
385 UnmarshallPtr (req->lpBlob, BLOB, base);
387 UnmarshallPtr (req->lpBlob->pBlobData, BYTE, base);
391 GNUNET_W32NSP_LookupServiceNext (HANDLE hLookup, DWORD dwControlFlags,
392 LPDWORD lpdwBufferLength, LPWSAQUERYSETW lpqsResults)
394 DWORD effective_flags;
396 struct GNUNET_MessageHeader header = {0, 0};
403 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext\n");
404 //EnterCriticalSection (&records_cs);
405 for (i = 0; i < records_len; i++)
407 if (records[i].s == (SOCKET) hLookup)
415 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: invalid handle\n");
416 SetLastError (WSA_INVALID_HANDLE);
417 //LeaveCriticalSection (&records_cs);
420 if (records[rec].state & 4)
422 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: session is closed\n");
423 SetLastError (WSA_E_NO_MORE);
424 //LeaveCriticalSection (&records_cs);
427 effective_flags = dwControlFlags & records[rec].flags;
428 if (records[rec].buf)
430 header = *((struct GNUNET_MessageHeader *) records[rec].buf);
431 if (dwControlFlags & LUP_FLUSHCACHE)
433 free (records[rec].buf);
434 records[rec].buf = NULL;
438 if (*lpdwBufferLength < header.size - sizeof (struct GNUNET_W32RESOLVER_GetMessage))
440 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: client buffer is too small\n");
441 SetLastError (WSAEFAULT);
442 //LeaveCriticalSection (&records_cs);
445 memcpy (lpqsResults, &((struct GNUNET_W32RESOLVER_GetMessage *)records[rec].buf)[1], header.size - sizeof (struct GNUNET_W32RESOLVER_GetMessage));
446 free (records[rec].buf);
447 records[rec].buf = NULL;
448 //LeaveCriticalSection (&records_cs);
449 UnmarshallWSAQUERYSETW ((LPWSAQUERYSETW) lpqsResults);
450 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: OK (from buffer)\n");
454 records[rec].state |= 8;
455 //LeaveCriticalSection (&records_cs);
456 to_receive = sizeof (header);
458 while (to_receive > 0)
460 t = recv ((SOCKET) hLookup, &((char *) &header)[rc], to_receive, 0);
469 //EnterCriticalSection (&records_cs);
470 records[rec].state &= ~8;
471 if (rc != sizeof (header))
473 if (records[rec].state & 2)
475 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: call cancelled\n");
476 SetLastError (WSA_E_CANCELLED);
480 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: failed to receive enough data\n");
481 SetLastError (WSA_E_NO_MORE);
483 records[rec].state |= 4;
484 //LeaveCriticalSection (&records_cs);
487 records[rec].state &= ~8;
488 header.type = ntohs (header.type);
489 header.size = ntohs (header.size);
490 if (header.type != GNUNET_MESSAGE_TYPE_W32RESOLVER_RESPONSE ||
491 (header.type == GNUNET_MESSAGE_TYPE_W32RESOLVER_RESPONSE &&
492 header.size == sizeof (header)))
494 records[rec].state |= 4;
495 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: header is wrong or type is wrong or no data\n");
496 //LeaveCriticalSection (&records_cs);
497 SetLastError (WSA_E_NO_MORE);
500 buf = malloc (header.size);
503 records[rec].state |= 4;
504 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: malloc() failed\n");
505 //LeaveCriticalSection (&records_cs);
506 SetLastError (WSA_E_NO_MORE);
509 records[rec].state |= 8;
510 //LeaveCriticalSection (&records_cs);
511 memcpy (buf, &header, sizeof (header));
512 to_receive = header.size - sizeof (header);
514 while (to_receive > 0)
516 t = recv ((SOCKET) hLookup, &((char *) &((struct GNUNET_MessageHeader *) buf)[1])[rc], to_receive, 0);
525 //EnterCriticalSection (&records_cs);
526 records[rec].state &= ~8;
527 if (rc != header.size - sizeof (header))
530 if (records[rec].state & 2)
532 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: call cancelled\n");
533 SetLastError (WSA_E_CANCELLED);
537 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: failed to receive enough data\n");
538 SetLastError (WSA_E_NO_MORE);
540 records[rec].state |= 4;
541 //LeaveCriticalSection (&records_cs);
544 if (*lpdwBufferLength < header.size - sizeof (struct GNUNET_W32RESOLVER_GetMessage))
546 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: client buffer is too small\n");
547 SetLastError (WSAEFAULT);
548 records[rec].buf = buf;
549 //LeaveCriticalSection (&records_cs);
552 //LeaveCriticalSection (&records_cs);
553 memcpy (lpqsResults, &((struct GNUNET_W32RESOLVER_GetMessage *)buf)[1], header.size - sizeof (struct GNUNET_W32RESOLVER_GetMessage));
555 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: OK\n");
556 UnmarshallWSAQUERYSETW ((LPWSAQUERYSETW) lpqsResults);
557 DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: returning (%lu)\n", GetLastError ());
562 GNUNET_W32NSP_LookupServiceEnd (HANDLE hLookup)
564 DWORD effective_flags;
566 struct GNUNET_MessageHeader header = {0, 0};
571 DEBUGLOG ("GNUNET_W32NSP_LookupServiceEnd\n");
572 //EnterCriticalSection (&records_cs);
573 for (i = 0; i < records_len; i++)
575 if (records[i].s == (SOCKET) hLookup)
583 SetLastError (WSA_INVALID_HANDLE);
584 //LeaveCriticalSection (&records_cs);
585 DEBUGLOG ("GNUNET_W32NSP_LookupServiceEnd: invalid handle\n");
588 records[rec].state |= 2;
589 closesocket (records[rec].s);
590 while (records[rec].state & 8)
592 //LeaveCriticalSection (&records_cs);
594 //EnterCriticalSection (&records_cs);
596 if (records[rec].buf)
597 free (records[rec].buf);
598 records[rec].buf = NULL;
599 records[rec].state = 0;
600 if (records[rec].name)
601 free (records[rec].name);
602 //LeaveCriticalSection (&records_cs);
603 DEBUGLOG ("GNUNET_W32NSP_LookupServiceEnd: OK\n");
608 GNUNET_W32NSP_SetService (LPGUID lpProviderId,
609 LPWSASERVICECLASSINFOW lpServiceClassInfo, LPWSAQUERYSETW lpqsRegInfo,
610 WSAESETSERVICEOP essOperation, DWORD dwControlFlags)
612 DEBUGLOG ("GNUNET_W32NSP_SetService\n");
613 SetLastError (WSAEOPNOTSUPP);
618 GNUNET_W32NSP_InstallServiceClass (LPGUID lpProviderId,
619 LPWSASERVICECLASSINFOW lpServiceClassInfo)
621 DEBUGLOG ("GNUNET_W32NSP_InstallServiceClass\n");
622 SetLastError (WSAEOPNOTSUPP);
628 GNUNET_W32NSP_RemoveServiceClass (LPGUID lpProviderId, LPGUID lpServiceClassId)
630 DEBUGLOG ("GNUNET_W32NSP_RemoveServiceClass\n");
631 SetLastError (WSAEOPNOTSUPP);
636 GNUNET_W32NSP_GetServiceClassInfo (LPGUID lpProviderId, LPDWORD lpdwBufSize,
637 LPWSASERVICECLASSINFOW lpServiceClassInfo)
639 DEBUGLOG ("GNUNET_W32NSP_GetServiceClassInfo\n");
640 SetLastError (WSAEOPNOTSUPP);
645 GNUNET_W32NSP_Ioctl (HANDLE hLookup, DWORD dwControlCode, LPVOID lpvInBuffer,
646 DWORD cbInBuffer, LPVOID lpvOutBuffer, DWORD cbOutBuffer,
647 LPDWORD lpcbBytesReturned, LPWSACOMPLETION lpCompletion,
648 LPWSATHREADID lpThreadId)
650 DEBUGLOG ("GNUNET_W32NSP_Ioctl\n");
651 SetLastError (WSAEOPNOTSUPP);
656 * This function is called by Winsock to hook up our provider.
657 * It is the only function that [should be/is] exported by the
658 * provider. All other routines are passed as pointers in lpnspRoutines.
661 NSPStartup (LPGUID lpProviderId, LPNSP_ROUTINE lpnspRoutines)
663 if (IsEqualGUID (lpProviderId, &GNUNET_NAMESPACE_PROVIDER_DNS))
665 if (!connect_to_dns_resolver ())
669 /* This assumes that NSP_ROUTINE struct doesn't have a NSPIoctl member.
670 * If it does, you need to use FIELD_OFFSET() macro to get offset of NSPIoctl
671 * and use that offset as cbSize.
673 lpnspRoutines->cbSize = sizeof(NSP_ROUTINE_XP);
675 lpnspRoutines->dwMajorVersion = NSPAPI_VERSION_MAJOR;
676 lpnspRoutines->dwMinorVersion = NSPAPI_VERSION_MINOR;
677 lpnspRoutines->NSPCleanup = NSPCleanup;
678 lpnspRoutines->NSPLookupServiceBegin = GNUNET_W32NSP_LookupServiceBegin;
679 lpnspRoutines->NSPLookupServiceNext = GNUNET_W32NSP_LookupServiceNext;
680 lpnspRoutines->NSPLookupServiceEnd = GNUNET_W32NSP_LookupServiceEnd;
681 lpnspRoutines->NSPSetService = GNUNET_W32NSP_SetService;
682 lpnspRoutines->NSPInstallServiceClass = GNUNET_W32NSP_InstallServiceClass;
683 lpnspRoutines->NSPRemoveServiceClass = GNUNET_W32NSP_RemoveServiceClass;
684 lpnspRoutines->NSPGetServiceClassInfo = GNUNET_W32NSP_GetServiceClassInfo;
685 ((NSP_ROUTINE_XP *) lpnspRoutines)->NSPIoctl = GNUNET_W32NSP_Ioctl;
688 SetLastError (WSAEINVALIDPROVIDER);