w32nsp.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705
  1. /*
  2. This file is part of GNUnet.
  3. Copyright (C) 2012 GNUnet e.V.
  4. GNUnet is free software: you can redistribute it and/or modify it
  5. under the terms of the GNU Affero General Public License as published
  6. by the Free Software Foundation, either version 3 of the License,
  7. or (at your option) any later version.
  8. GNUnet is distributed in the hope that it will be useful, but
  9. WITHOUT ANY WARRANTY; without even the implied warranty of
  10. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  11. Affero General Public License for more details.
  12. You should have received a copy of the GNU Affero General Public License
  13. along with this program. If not, see <http://www.gnu.org/licenses/>.
  14. SPDX-License-Identifier: AGPL3.0-or-later
  15. */
  16. /**
  17. * @file gns/w32nsp.c
  18. * @brief W32 integration for GNS
  19. * @author LRN
  20. */
  21. /* This code is partially based upon samples from the book
  22. * "Network Programming For Microsoft Windows, 2Nd Edition".
  23. */
  24. #define VERBOSE 0
  25. #if !VERBOSE
  26. # define DEBUGLOG(s, ...)
  27. #endif
  28. #if VERBOSE
  29. # define __printf__ printf
  30. # define DEBUGLOG(s, ...) printf (s, ##__VA_ARGS__)
  31. #endif
  32. #include <stdint.h>
  33. #include <ws2tcpip.h>
  34. #include <ws2spi.h>
  35. #include <windows.h>
  36. #include <nspapi.h>
  37. #define WINDOWS 1
  38. #define MINGW 1
  39. #ifndef __BYTE_ORDER
  40. #ifdef _BYTE_ORDER
  41. #define __BYTE_ORDER _BYTE_ORDER
  42. #else
  43. #ifdef BYTE_ORDER
  44. #define __BYTE_ORDER BYTE_ORDER
  45. #endif
  46. #endif
  47. #endif
  48. #ifndef __BIG_ENDIAN
  49. #ifdef _BIG_ENDIAN
  50. #define __BIG_ENDIAN _BIG_ENDIAN
  51. #else
  52. #ifdef BIG_ENDIAN
  53. #define __BIG_ENDIAN BIG_ENDIAN
  54. #endif
  55. #endif
  56. #endif
  57. #ifndef __LITTLE_ENDIAN
  58. #ifdef _LITTLE_ENDIAN
  59. #define __LITTLE_ENDIAN _LITTLE_ENDIAN
  60. #else
  61. #ifdef LITTLE_ENDIAN
  62. #define __LITTLE_ENDIAN LITTLE_ENDIAN
  63. #endif
  64. #endif
  65. #endif
  66. #include "w32resolver.h"
  67. #include <initguid.h>
  68. #include "gnunet_w32nsp_lib.h"
  69. #undef INITGUID
  70. #define NSPAPI_VERSION_MAJOR 4
  71. #define NSPAPI_VERSION_MINOR 4
  72. static CRITICAL_SECTION records_cs;
  73. struct record
  74. {
  75. SOCKET s;
  76. DWORD flags;
  77. uint8_t state;
  78. char *buf;
  79. wchar_t *name;
  80. };
  81. static struct record *records = NULL;
  82. static size_t records_len = 0;
  83. static size_t records_size = 0;
  84. static int
  85. resize_records ()
  86. {
  87. size_t new_size = records_len > 0 ? records_len * 2 : 5;
  88. struct record *new_records = malloc (new_size * sizeof (struct record));
  89. if (new_records == NULL)
  90. {
  91. SetLastError (WSA_NOT_ENOUGH_MEMORY);
  92. return 0;
  93. }
  94. GNUNET_memcpy (new_records, records, records_len * sizeof (struct record));
  95. memset (&new_records[records_len], 0, sizeof (struct record) * (new_size - records_len));
  96. records_size = new_size;
  97. free (records);
  98. records = new_records;
  99. return 1;
  100. }
  101. static int
  102. add_record (SOCKET s, const wchar_t *name, DWORD flags)
  103. {
  104. int res = 1;
  105. int i;
  106. int empty = -1;
  107. //EnterCriticalSection (&records_cs);
  108. for (i = 0; i < records_len; i++)
  109. if (records[i].state == 0)
  110. break;
  111. empty = i;
  112. if (i == records_len)
  113. {
  114. res = resize_records ();
  115. if (res)
  116. empty = records_len++;
  117. }
  118. if (res)
  119. {
  120. struct record r;
  121. r.s = s;
  122. r.flags = flags;
  123. r.name = (wchar_t *) name;
  124. r.state = 1;
  125. r.buf = NULL;
  126. if (name)
  127. r.name = wcsdup (name);
  128. records[empty] = r;
  129. }
  130. //LeaveCriticalSection (&records_cs);
  131. return res;
  132. }
  133. /* These are not defined by mingw.org headers at the moment*/
  134. typedef INT (WSPAPI *LPNSPIOCTL) (HANDLE,DWORD,LPVOID,DWORD,LPVOID,DWORD,LPDWORD,LPWSACOMPLETION,LPWSATHREADID);
  135. typedef struct _NSP_ROUTINE_XP {
  136. DWORD cbSize;
  137. DWORD dwMajorVersion;
  138. DWORD dwMinorVersion;
  139. LPNSPCLEANUP NSPCleanup;
  140. LPNSPLOOKUPSERVICEBEGIN NSPLookupServiceBegin;
  141. LPNSPLOOKUPSERVICENEXT NSPLookupServiceNext;
  142. LPNSPLOOKUPSERVICEEND NSPLookupServiceEnd;
  143. LPNSPSETSERVICE NSPSetService;
  144. LPNSPINSTALLSERVICECLASS NSPInstallServiceClass;
  145. LPNSPREMOVESERVICECLASS NSPRemoveServiceClass;
  146. LPNSPGETSERVICECLASSINFO NSPGetServiceClassInfo;
  147. LPNSPIOCTL NSPIoctl;
  148. } NSP_ROUTINE_XP;
  149. static SOCKET
  150. connect_to_dns_resolver ()
  151. {
  152. struct sockaddr_in addr;
  153. SOCKET r;
  154. int ret;
  155. r = socket (AF_INET, SOCK_STREAM, IPPROTO_TCP);
  156. if (INVALID_SOCKET == r)
  157. {
  158. SetLastError (16004);
  159. return r;
  160. }
  161. addr.sin_family = AF_INET;
  162. addr.sin_port = htons (5353); /* TCP 5353 is not registered; UDP 5353 is */
  163. addr.sin_addr.s_addr = inet_addr ("127.0.0.1");
  164. ret = connect (r, (struct sockaddr *) &addr, sizeof (addr));
  165. if (SOCKET_ERROR == ret)
  166. {
  167. DWORD err = GetLastError ();
  168. closesocket (r);
  169. SetLastError (err);
  170. SetLastError (16005);
  171. r = INVALID_SOCKET;
  172. }
  173. return r;
  174. }
  175. static int
  176. send_name_to_ip_request (LPWSAQUERYSETW lpqsRestrictions,
  177. LPWSASERVICECLASSINFOW lpServiceClassInfo, DWORD dwControlFlags,
  178. SOCKET *resolver)
  179. {
  180. struct GNUNET_W32RESOLVER_GetMessage *msg;
  181. int af4 = 0;
  182. int af6 = 0;
  183. char *buf;
  184. int ret = 1;
  185. int i;
  186. size_t size = sizeof (struct GNUNET_W32RESOLVER_GetMessage);
  187. size_t namelen = 0;
  188. if (lpqsRestrictions->lpszServiceInstanceName)
  189. namelen = sizeof (wchar_t) * (wcslen (lpqsRestrictions->lpszServiceInstanceName) + 1);
  190. size += namelen;
  191. buf = malloc (size);
  192. msg = (struct GNUNET_W32RESOLVER_GetMessage *) buf;
  193. msg->header.size = htons (size);
  194. msg->header.type = htons (GNUNET_MESSAGE_TYPE_W32RESOLVER_REQUEST);
  195. if (lpqsRestrictions->dwNumberOfProtocols > 0)
  196. {
  197. int i;
  198. for (i = 0; i < lpqsRestrictions->dwNumberOfProtocols; i++)
  199. {
  200. if (lpqsRestrictions->lpafpProtocols[0].iAddressFamily == AF_INET)
  201. af4 = 1;
  202. if (lpqsRestrictions->lpafpProtocols[0].iAddressFamily == AF_INET6)
  203. af6 = 1;
  204. }
  205. }
  206. if (af4 && !af6)
  207. msg->af = htonl (AF_INET);
  208. else if (af6 && !af4)
  209. msg->af = htonl (AF_INET6);
  210. else
  211. msg->af = htonl (AF_UNSPEC);
  212. if (lpqsRestrictions->lpszServiceInstanceName)
  213. GNUNET_memcpy (&msg[1], lpqsRestrictions->lpszServiceInstanceName, namelen);
  214. msg->sc_data1 = htonl (lpqsRestrictions->lpServiceClassId->Data1);
  215. msg->sc_data2 = htons (lpqsRestrictions->lpServiceClassId->Data2);
  216. msg->sc_data3 = htons (lpqsRestrictions->lpServiceClassId->Data3);
  217. for (i = 0; i < 8; i++)
  218. msg->sc_data4[i] = lpqsRestrictions->lpServiceClassId->Data4[i];
  219. *resolver = connect_to_dns_resolver ();
  220. if (*resolver != INVALID_SOCKET)
  221. {
  222. if (size != send (*resolver, buf, size, 0))
  223. {
  224. #if VERBOSE
  225. DWORD err = GetLastError ();
  226. #endif
  227. closesocket (*resolver);
  228. *resolver = INVALID_SOCKET;
  229. DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: failed to send request: %lu\n", err);
  230. SetLastError (WSATRY_AGAIN);
  231. ret = 0;
  232. }
  233. }
  234. else
  235. ret = 0;
  236. free (buf);
  237. return ret;
  238. }
  239. static int WSPAPI
  240. NSPCleanup (LPGUID lpProviderId)
  241. {
  242. DEBUGLOG ("NSPCleanup\n");
  243. if (IsEqualGUID (lpProviderId, &GNUNET_NAMESPACE_PROVIDER_DNS))
  244. {
  245. return NO_ERROR;
  246. }
  247. SetLastError (WSAEINVALIDPROVIDER);
  248. return SOCKET_ERROR;
  249. }
  250. BOOL WINAPI
  251. DllMain (HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
  252. {
  253. switch (fdwReason)
  254. {
  255. case DLL_PROCESS_ATTACH:
  256. if (!InitializeCriticalSectionAndSpinCount (&records_cs, 0x00000400))
  257. {
  258. return FALSE;
  259. }
  260. break;
  261. case DLL_THREAD_ATTACH:
  262. break;
  263. case DLL_THREAD_DETACH:
  264. break;
  265. case DLL_PROCESS_DETACH:
  266. DeleteCriticalSection (&records_cs);
  267. break;
  268. }
  269. return TRUE;
  270. }
  271. static int WSPAPI
  272. GNUNET_W32NSP_LookupServiceBegin (LPGUID lpProviderId, LPWSAQUERYSETW lpqsRestrictions,
  273. LPWSASERVICECLASSINFOW lpServiceClassInfo, DWORD dwControlFlags,
  274. LPHANDLE lphLookup)
  275. {
  276. DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin\n");
  277. if (IsEqualGUID (lpProviderId, &GNUNET_NAMESPACE_PROVIDER_DNS))
  278. {
  279. SOCKET s;
  280. if (lpqsRestrictions->dwNameSpace != NS_DNS && lpqsRestrictions->dwNameSpace != NS_ALL)
  281. {
  282. DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: wrong namespace\n");
  283. SetLastError (WSAEINVAL);
  284. return SOCKET_ERROR;
  285. }
  286. if (lpqsRestrictions->lpszServiceInstanceName != NULL)
  287. {
  288. wchar_t *s = lpqsRestrictions->lpszServiceInstanceName;
  289. size_t len = wcslen (s);
  290. if (len >= 5 && wcscmp (&s[len - 5], L".zkey") == 0)
  291. {
  292. }
  293. else if (len >= 4 && wcscmp (&s[len - 4], L".gnu") == 0)
  294. {
  295. }
  296. else
  297. {
  298. DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: unsupported TLD\n");
  299. SetLastError (WSAEINVAL);
  300. return SOCKET_ERROR;
  301. }
  302. }
  303. if (send_name_to_ip_request (lpqsRestrictions,
  304. lpServiceClassInfo, dwControlFlags, &s))
  305. {
  306. if (!(add_record (s, lpqsRestrictions->lpszServiceInstanceName, dwControlFlags)))
  307. {
  308. DWORD err = GetLastError ();
  309. DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: failed to add a record\n");
  310. closesocket (s);
  311. SetLastError (err);
  312. return SOCKET_ERROR;
  313. }
  314. *lphLookup = (HANDLE) s;
  315. DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: OK (%lu)\n", GetLastError ());
  316. return NO_ERROR;
  317. }
  318. return SOCKET_ERROR;
  319. }
  320. DEBUGLOG ("GNUNET_W32NSP_LookupServiceBegin: wrong provider\n");
  321. SetLastError (WSAEINVALIDPROVIDER);
  322. return SOCKET_ERROR;
  323. }
  324. #define UnmarshallPtr(ptr, ptrtype, base) \
  325. if (ptr) \
  326. ptr = (ptrtype *) (base + (uintptr_t) ptr)
  327. static void
  328. UnmarshallWSAQUERYSETW (LPWSAQUERYSETW req)
  329. {
  330. int i;
  331. char *base = (char *) req;
  332. UnmarshallPtr (req->lpszServiceInstanceName, wchar_t, base);
  333. UnmarshallPtr (req->lpServiceClassId, GUID, base);
  334. UnmarshallPtr (req->lpVersion, WSAVERSION, base);
  335. UnmarshallPtr (req->lpszComment, wchar_t, base);
  336. UnmarshallPtr (req->lpNSProviderId, GUID, base);
  337. UnmarshallPtr (req->lpszContext, wchar_t, base);
  338. UnmarshallPtr (req->lpafpProtocols, AFPROTOCOLS, base);
  339. UnmarshallPtr (req->lpszQueryString, wchar_t, base);
  340. UnmarshallPtr (req->lpcsaBuffer, CSADDR_INFO, base);
  341. for (i = 0; i < req->dwNumberOfCsAddrs; i++)
  342. {
  343. UnmarshallPtr (req->lpcsaBuffer[i].LocalAddr.lpSockaddr, SOCKADDR, base);
  344. UnmarshallPtr (req->lpcsaBuffer[i].RemoteAddr.lpSockaddr, SOCKADDR, base);
  345. }
  346. UnmarshallPtr (req->lpBlob, BLOB, base);
  347. if (req->lpBlob)
  348. UnmarshallPtr (req->lpBlob->pBlobData, BYTE, base);
  349. }
  350. static int WSAAPI
  351. GNUNET_W32NSP_LookupServiceNext (HANDLE hLookup, DWORD dwControlFlags,
  352. LPDWORD lpdwBufferLength, LPWSAQUERYSETW lpqsResults)
  353. {
  354. /*DWORD effective_flags;*/
  355. int i;
  356. struct GNUNET_MessageHeader header = {0, 0};
  357. int rec = -1;
  358. int rc;
  359. int to_receive;
  360. int t;
  361. char *buf;
  362. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext\n");
  363. //EnterCriticalSection (&records_cs);
  364. for (i = 0; i < records_len; i++)
  365. {
  366. if (records[i].s == (SOCKET) hLookup)
  367. {
  368. rec = i;
  369. break;
  370. }
  371. }
  372. if (rec == -1)
  373. {
  374. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: invalid handle\n");
  375. SetLastError (WSA_INVALID_HANDLE);
  376. //LeaveCriticalSection (&records_cs);
  377. return SOCKET_ERROR;
  378. }
  379. if (records[rec].state & 4)
  380. {
  381. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: session is closed\n");
  382. SetLastError (WSA_E_NO_MORE);
  383. //LeaveCriticalSection (&records_cs);
  384. return SOCKET_ERROR;
  385. }
  386. /*effective_flags = dwControlFlags & records[rec].flags;*/
  387. if (records[rec].buf)
  388. {
  389. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: checking buffer\n");
  390. header = *((struct GNUNET_MessageHeader *) records[rec].buf);
  391. if (*lpdwBufferLength < header.size - sizeof (struct GNUNET_W32RESOLVER_GetMessage))
  392. {
  393. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: client buffer is too small\n");
  394. SetLastError (WSAEFAULT);
  395. //LeaveCriticalSection (&records_cs);
  396. return SOCKET_ERROR;
  397. }
  398. GNUNET_memcpy (lpqsResults, &((struct GNUNET_W32RESOLVER_GetMessage *)records[rec].buf)[1], header.size - sizeof (struct GNUNET_W32RESOLVER_GetMessage));
  399. free (records[rec].buf);
  400. records[rec].buf = NULL;
  401. //LeaveCriticalSection (&records_cs);
  402. UnmarshallWSAQUERYSETW ((LPWSAQUERYSETW) lpqsResults);
  403. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: OK (from buffer)\n");
  404. return NO_ERROR;
  405. }
  406. records[rec].state |= 8;
  407. //LeaveCriticalSection (&records_cs);
  408. to_receive = sizeof (header);
  409. rc = 0;
  410. #if VERBOSE
  411. {
  412. unsigned long have;
  413. int ior = ioctlsocket ((SOCKET) hLookup, FIONREAD, &have);
  414. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: reading %d bytes as a header from %p, %lu bytes available\n", to_receive, hLookup, have);
  415. }
  416. #endif
  417. while (to_receive > 0)
  418. {
  419. t = recv ((SOCKET) hLookup, &((char *) &header)[rc], to_receive, 0);
  420. if (t > 0)
  421. {
  422. rc += t;
  423. to_receive -= t;
  424. }
  425. else
  426. break;
  427. }
  428. #if VERBOSE
  429. {
  430. unsigned long have;
  431. int ior = ioctlsocket ((SOCKET) hLookup, FIONREAD, &have);
  432. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: read %d bytes as a header from %p, %lu bytes available\n", rc, hLookup, have);
  433. }
  434. #endif
  435. //EnterCriticalSection (&records_cs);
  436. records[rec].state &= ~8;
  437. if (rc != sizeof (header))
  438. {
  439. if (records[rec].state & 2)
  440. {
  441. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: call cancelled\n");
  442. SetLastError (WSA_E_CANCELLED);
  443. }
  444. else
  445. {
  446. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: failed to receive enough data for a header (rc %d != %u, state is 0x%0X)\n", rc, sizeof (header), records[rec].state);
  447. SetLastError (WSA_E_NO_MORE);
  448. }
  449. records[rec].state |= 4;
  450. //LeaveCriticalSection (&records_cs);
  451. return SOCKET_ERROR;
  452. }
  453. records[rec].state &= ~8;
  454. header.type = ntohs (header.type);
  455. header.size = ntohs (header.size);
  456. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: header type %d, header size %u\n", header.type, header.size);
  457. if (header.type != GNUNET_MESSAGE_TYPE_W32RESOLVER_RESPONSE ||
  458. (header.type == GNUNET_MESSAGE_TYPE_W32RESOLVER_RESPONSE &&
  459. header.size == sizeof (header)))
  460. {
  461. records[rec].state |= 4;
  462. if (header.type != GNUNET_MESSAGE_TYPE_W32RESOLVER_RESPONSE)
  463. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: header type is wrong\n");
  464. else
  465. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: empty header - no data\n");
  466. //LeaveCriticalSection (&records_cs);
  467. SetLastError (WSA_E_NO_MORE);
  468. return SOCKET_ERROR;
  469. }
  470. buf = malloc (header.size);
  471. if (buf == NULL)
  472. {
  473. records[rec].state |= 4;
  474. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: malloc() failed\n");
  475. //LeaveCriticalSection (&records_cs);
  476. SetLastError (WSA_E_NO_MORE);
  477. return SOCKET_ERROR;
  478. }
  479. records[rec].state |= 8;
  480. //LeaveCriticalSection (&records_cs);
  481. GNUNET_memcpy (buf, &header, sizeof (header));
  482. to_receive = header.size - sizeof (header);
  483. rc = 0;
  484. #if VERBOSE
  485. {
  486. unsigned long have;
  487. int ior = ioctlsocket ((SOCKET) hLookup, FIONREAD, &have);
  488. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: reading %d bytes as a body from %p, %lu bytes available\n", to_receive, hLookup, have);
  489. }
  490. #endif
  491. while (to_receive > 0)
  492. {
  493. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: recv (%d)\n", to_receive);
  494. t = recv ((SOCKET) hLookup, &((char *) &((struct GNUNET_MessageHeader *) buf)[1])[rc], to_receive, 0);
  495. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: recv returned %d\n", t);
  496. if (t > 0)
  497. {
  498. rc += t;
  499. to_receive -= t;
  500. }
  501. else
  502. break;
  503. }
  504. #if VERBOSE
  505. {
  506. unsigned long have;
  507. int ior = ioctlsocket ((SOCKET) hLookup, FIONREAD, &have);
  508. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: read %d bytes as a body from %p, %lu bytes available\n", rc, hLookup, have);
  509. }
  510. #endif
  511. //EnterCriticalSection (&records_cs);
  512. records[rec].state &= ~8;
  513. if (rc != header.size - sizeof (header))
  514. {
  515. free (buf);
  516. if (records[rec].state & 2)
  517. {
  518. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: call cancelled\n");
  519. SetLastError (WSA_E_CANCELLED);
  520. }
  521. else
  522. {
  523. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: failed to receive enough data for the rest (rc %d != %d, state is 0x%0X)\n", rc, header.size - sizeof (header), records[rec].state);
  524. SetLastError (WSA_E_NO_MORE);
  525. }
  526. records[rec].state |= 4;
  527. //LeaveCriticalSection (&records_cs);
  528. return SOCKET_ERROR;
  529. }
  530. if (*lpdwBufferLength < header.size - sizeof (struct GNUNET_W32RESOLVER_GetMessage))
  531. {
  532. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: client buffer is too small\n");
  533. SetLastError (WSAEFAULT);
  534. records[rec].buf = buf;
  535. //LeaveCriticalSection (&records_cs);
  536. return SOCKET_ERROR;
  537. }
  538. //LeaveCriticalSection (&records_cs);
  539. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: writing %d bytes into result buffer\n", header.size - sizeof (struct GNUNET_W32RESOLVER_GetMessage));
  540. GNUNET_memcpy (lpqsResults, &((struct GNUNET_W32RESOLVER_GetMessage *)buf)[1], header.size - sizeof (struct GNUNET_W32RESOLVER_GetMessage));
  541. free (buf);
  542. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: OK\n");
  543. UnmarshallWSAQUERYSETW ((LPWSAQUERYSETW) lpqsResults);
  544. DEBUGLOG ("GNUNET_W32NSP_LookupServiceNext: returning (%lu)\n", GetLastError ());
  545. return NO_ERROR;
  546. }
  547. static int WSPAPI
  548. GNUNET_W32NSP_LookupServiceEnd (HANDLE hLookup)
  549. {
  550. int i;
  551. int rec = -1;
  552. DEBUGLOG ("GNUNET_W32NSP_LookupServiceEnd\n");
  553. //EnterCriticalSection (&records_cs);
  554. for (i = 0; i < records_len; i++)
  555. {
  556. if (records[i].s == (SOCKET) hLookup)
  557. {
  558. rec = i;
  559. break;
  560. }
  561. }
  562. if (rec == -1)
  563. {
  564. SetLastError (WSA_INVALID_HANDLE);
  565. //LeaveCriticalSection (&records_cs);
  566. DEBUGLOG ("GNUNET_W32NSP_LookupServiceEnd: invalid handle\n");
  567. return SOCKET_ERROR;
  568. }
  569. records[rec].state |= 2;
  570. closesocket (records[rec].s);
  571. while (records[rec].state & 8)
  572. {
  573. //LeaveCriticalSection (&records_cs);
  574. Sleep (10);
  575. //EnterCriticalSection (&records_cs);
  576. }
  577. if (records[rec].buf)
  578. free (records[rec].buf);
  579. records[rec].buf = NULL;
  580. records[rec].state = 0;
  581. if (records[rec].name)
  582. free (records[rec].name);
  583. //LeaveCriticalSection (&records_cs);
  584. DEBUGLOG ("GNUNET_W32NSP_LookupServiceEnd: OK\n");
  585. return NO_ERROR;
  586. }
  587. static int WSAAPI
  588. GNUNET_W32NSP_SetService (LPGUID lpProviderId,
  589. LPWSASERVICECLASSINFOW lpServiceClassInfo, LPWSAQUERYSETW lpqsRegInfo,
  590. WSAESETSERVICEOP essOperation, DWORD dwControlFlags)
  591. {
  592. DEBUGLOG ("GNUNET_W32NSP_SetService\n");
  593. SetLastError (WSAEOPNOTSUPP);
  594. return SOCKET_ERROR;
  595. }
  596. static int WSAAPI
  597. GNUNET_W32NSP_InstallServiceClass (LPGUID lpProviderId,
  598. LPWSASERVICECLASSINFOW lpServiceClassInfo)
  599. {
  600. DEBUGLOG ("GNUNET_W32NSP_InstallServiceClass\n");
  601. SetLastError (WSAEOPNOTSUPP);
  602. return SOCKET_ERROR;
  603. }
  604. static int WSAAPI
  605. GNUNET_W32NSP_RemoveServiceClass (LPGUID lpProviderId, LPGUID lpServiceClassId)
  606. {
  607. DEBUGLOG ("GNUNET_W32NSP_RemoveServiceClass\n");
  608. SetLastError (WSAEOPNOTSUPP);
  609. return SOCKET_ERROR;
  610. }
  611. static int WSAAPI
  612. GNUNET_W32NSP_GetServiceClassInfo (LPGUID lpProviderId, LPDWORD lpdwBufSize,
  613. LPWSASERVICECLASSINFOW lpServiceClassInfo)
  614. {
  615. DEBUGLOG ("GNUNET_W32NSP_GetServiceClassInfo\n");
  616. SetLastError (WSAEOPNOTSUPP);
  617. return SOCKET_ERROR;
  618. }
  619. static int WSAAPI
  620. GNUNET_W32NSP_Ioctl (HANDLE hLookup, DWORD dwControlCode, LPVOID lpvInBuffer,
  621. DWORD cbInBuffer, LPVOID lpvOutBuffer, DWORD cbOutBuffer,
  622. LPDWORD lpcbBytesReturned, LPWSACOMPLETION lpCompletion,
  623. LPWSATHREADID lpThreadId)
  624. {
  625. DEBUGLOG ("GNUNET_W32NSP_Ioctl\n");
  626. SetLastError (WSAEOPNOTSUPP);
  627. return SOCKET_ERROR;
  628. }
  629. /**
  630. * This function is called by Winsock to hook up our provider.
  631. * It is the only function that [should be/is] exported by the
  632. * provider. All other routines are passed as pointers in lpnspRoutines.
  633. */
  634. int WSAAPI
  635. GNUNET_W32NSP_NSPStartup (LPGUID lpProviderId, LPNSP_ROUTINE lpnspRoutines)
  636. {
  637. if (IsEqualGUID (lpProviderId, &GNUNET_NAMESPACE_PROVIDER_DNS))
  638. {
  639. if (!connect_to_dns_resolver ())
  640. {
  641. return SOCKET_ERROR;
  642. }
  643. /* This assumes that NSP_ROUTINE struct doesn't have a NSPIoctl member.
  644. * If it does, you need to use FIELD_OFFSET() macro to get offset of NSPIoctl
  645. * and use that offset as cbSize.
  646. */
  647. lpnspRoutines->cbSize = sizeof(NSP_ROUTINE);
  648. lpnspRoutines->dwMajorVersion = NSPAPI_VERSION_MAJOR;
  649. lpnspRoutines->dwMinorVersion = NSPAPI_VERSION_MINOR;
  650. lpnspRoutines->NSPCleanup = NSPCleanup;
  651. lpnspRoutines->NSPLookupServiceBegin = GNUNET_W32NSP_LookupServiceBegin;
  652. lpnspRoutines->NSPLookupServiceNext = GNUNET_W32NSP_LookupServiceNext;
  653. lpnspRoutines->NSPLookupServiceEnd = GNUNET_W32NSP_LookupServiceEnd;
  654. lpnspRoutines->NSPSetService = GNUNET_W32NSP_SetService;
  655. lpnspRoutines->NSPInstallServiceClass = GNUNET_W32NSP_InstallServiceClass;
  656. lpnspRoutines->NSPRemoveServiceClass = GNUNET_W32NSP_RemoveServiceClass;
  657. lpnspRoutines->NSPGetServiceClassInfo = GNUNET_W32NSP_GetServiceClassInfo;
  658. /*((NSP_ROUTINE_XP *) lpnspRoutines)->NSPIoctl = GNUNET_W32NSP_Ioctl;*/
  659. lpnspRoutines->NSPIoctl = GNUNET_W32NSP_Ioctl;
  660. return NO_ERROR;
  661. }
  662. SetLastError (WSAEINVALIDPROVIDER);
  663. return SOCKET_ERROR;
  664. }