src/gns/gnunet-gns-helper-service-w32.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2012 Christian Grothoff (and other contributing authors)
   4
   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.
   9
  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.
  14
  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.
  19 */
  20 /**
  21  * @file gnunet-gns-helper-service-w32.c
  22  * @brief an intermediary service to access distributed GNS
  23  * @author Christian Grothoff
  24  * @author LRN
  25  */
  26 #define INITGUID
  27 #include "platform.h"
  28 #include <gnunet_util_lib.h>
  29 #include <gnunet_dnsparser_lib.h>
  30 #include <gnunet_namestore_service.h>
  31 #include <gnunet_gns_service.h>
  32 #include <gnunet_w32nsp_lib.h>
  33 #include "w32resolver.h"
  34 #include <nspapi.h>
  35 #include <unistr.h>
  36
  37 #define DEFINE_DNS_GUID(a,x) DEFINE_GUID(a, 0x00090035, 0x0000, x, 0xc0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x46)
  38 DEFINE_DNS_GUID(SVCID_DNS_TYPE_A, 0x0001);
  39 DEFINE_DNS_GUID(SVCID_DNS_TYPE_NS, 0x0002);
  40 DEFINE_DNS_GUID(SVCID_DNS_TYPE_CNAME, 0x0005);
  41 DEFINE_DNS_GUID(SVCID_DNS_TYPE_SOA, 0x0006);
  42 DEFINE_DNS_GUID(SVCID_DNS_TYPE_PTR, 0x000c);
  43 DEFINE_DNS_GUID(SVCID_DNS_TYPE_MX, 0x000f);
  44 DEFINE_DNS_GUID(SVCID_DNS_TYPE_TEXT, 0x0010);
  45 DEFINE_DNS_GUID(SVCID_DNS_TYPE_AAAA, 0x001c);
  46 DEFINE_DNS_GUID(SVCID_DNS_TYPE_SRV, 0x0021);
  47 DEFINE_GUID(SVCID_HOSTNAME, 0x0002a800, 0x0000, 0x0000, 0xc0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x46);
  48 DEFINE_GUID(SVCID_INET_HOSTADDRBYNAME, 0x0002a803, 0x0000, 0x0000, 0xc0, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x46);
  49
  50 struct request
  51 {
  52   struct GNUNET_SERVER_Client *client;
  53   GUID sc;
  54   int af;
  55   wchar_t *name;
  56   char *u8name;
  57 };
  58
  59 /**
  60  * Handle to GNS service.
  61  */
  62 static struct GNUNET_GNS_Handle *gns;
  63
  64 static struct GNUNET_CRYPTO_ShortHashCode *zone = NULL;
  65 static struct GNUNET_CRYPTO_ShortHashCode user_zone;
  66 struct GNUNET_CRYPTO_RsaPrivateKey *shorten_key = NULL;
  67
  68
  69 /**
  70  * Task run on shutdown.  Cleans up everything.
  71  *
  72  * @param cls unused
  73  * @param tc scheduler context
  74  */
  75 static void
  76 do_shutdown (void *cls,
  77              const struct GNUNET_SCHEDULER_TaskContext *tc)
  78 {
  79   if (NULL != gns)
  80   {
  81     GNUNET_GNS_disconnect (gns);
  82     gns = NULL;
  83   }
  84 }
  85
  86 /**
  87  * Context for transmitting replies to clients.
  88  */
  89 struct TransmitCallbackContext
  90 {
  91
  92   /**
  93    * We keep these in a doubly-linked list (for cleanup).
  94    */
  95   struct TransmitCallbackContext *next;
  96
  97   /**
  98    * We keep these in a doubly-linked list (for cleanup).
  99    */
 100   struct TransmitCallbackContext *prev;
 101
 102   /**
 103    * The message that we're asked to transmit.
 104    */
 105   struct GNUNET_MessageHeader *msg;
 106
 107   size_t msgsize;
 108
 109   /**
 110    * Handle for the transmission request.
 111    */
 112   struct GNUNET_SERVER_TransmitHandle *th;
 113
 114   /**
 115    * Client that we are transmitting to.
 116    */
 117   struct GNUNET_SERVER_Client *client;
 118
 119 };
 120
 121
 122 /**
 123  * Head of the doubly-linked list (for cleanup).
 124  */
 125 static struct TransmitCallbackContext *tcc_head;
 126
 127 /**
 128  * Tail of the doubly-linked list (for cleanup).
 129  */
 130 static struct TransmitCallbackContext *tcc_tail;
 131
 132 /**
 133  * Have we already cleaned up the TCCs and are hence no longer
 134  * willing (or able) to transmit anything to anyone?
 135  */
 136 static int cleaning_done;
 137
 138 /**
 139  * Function called to notify a client about the socket
 140  * begin ready to queue more data.  "buf" will be
 141  * NULL and "size" zero if the socket was closed for
 142  * writing in the meantime.
 143  *
 144  * @param cls closure
 145  * @param size number of bytes available in buf
 146  * @param buf where the callee should write the message
 147  * @return number of bytes written to buf
 148  */
 149 static size_t
 150 transmit_callback (void *cls, size_t size, void *buf)
 151 {
 152   struct TransmitCallbackContext *tcc = cls;
 153   size_t msize;
 154
 155   tcc->th = NULL;
 156   GNUNET_CONTAINER_DLL_remove (tcc_head, tcc_tail, tcc);
 157   msize = tcc->msgsize;
 158   if (size == 0)
 159   {
 160     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 161                 _("Transmission to client failed!\n"));
 162     GNUNET_SERVER_client_drop (tcc->client);
 163     GNUNET_free (tcc->msg);
 164     GNUNET_free (tcc);
 165     return 0;
 166   }
 167   GNUNET_assert (size >= msize);
 168   memcpy (buf, tcc->msg, msize);
 169   GNUNET_SERVER_client_drop (tcc->client);
 170   GNUNET_free (tcc->msg);
 171   GNUNET_free (tcc);
 172   return msize;
 173 }
 174
 175
 176 /**
 177  * Transmit the given message to the client.
 178  *
 179  * @param client target of the message
 180  * @param msg message to transmit, will be freed!
 181  */
 182 static void
 183 transmit (struct GNUNET_SERVER_Client *client, struct GNUNET_MessageHeader *msg, size_t msgsize)
 184 {
 185   struct TransmitCallbackContext *tcc;
 186
 187   if (GNUNET_YES == cleaning_done)
 188   {
 189     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
 190                 _("Shutdown in progress, aborting transmission.\n"));
 191     GNUNET_SERVER_client_drop (client);
 192     GNUNET_free (msg);
 193     return;
 194   }
 195   tcc = GNUNET_malloc (sizeof (struct TransmitCallbackContext));
 196   tcc->msg = msg;
 197   tcc->msgsize = msgsize;
 198   tcc->client = client;
 199   if (NULL ==
 200       (tcc->th =
 201        GNUNET_SERVER_notify_transmit_ready (client, msgsize,
 202                                             GNUNET_TIME_UNIT_FOREVER_REL,
 203                                             &transmit_callback, tcc)))
 204   {
 205     GNUNET_break (0);
 206     GNUNET_SERVER_client_drop (client);
 207     GNUNET_free (msg);
 208     GNUNET_free (tcc);
 209     return;
 210   }
 211   GNUNET_SERVER_client_keep (client);
 212   GNUNET_CONTAINER_DLL_insert (tcc_head, tcc_tail, tcc);
 213 }
 214
 215 #define MarshallPtr(ptr, base, type) \
 216   if (ptr) \
 217     ptr = (type *) ((char *) ptr - (char *) base)
 218
 219 void
 220 MarshallWSAQUERYSETW (WSAQUERYSETW *qs, GUID *sc)
 221 {
 222   int i;
 223   MarshallPtr (qs->lpszServiceInstanceName, qs, wchar_t);
 224   MarshallPtr (qs->lpServiceClassId, qs, GUID);
 225   MarshallPtr (qs->lpVersion, qs, WSAVERSION);
 226   MarshallPtr (qs->lpNSProviderId, qs, GUID);
 227   MarshallPtr (qs->lpszContext, qs, wchar_t);
 228   MarshallPtr (qs->lpafpProtocols, qs, AFPROTOCOLS);
 229   MarshallPtr (qs->lpszQueryString, qs, wchar_t);
 230   for (i = 0; i < qs->dwNumberOfCsAddrs; i++)
 231   {
 232     MarshallPtr (qs->lpcsaBuffer[i].LocalAddr.lpSockaddr, qs, SOCKADDR);
 233     MarshallPtr (qs->lpcsaBuffer[i].RemoteAddr.lpSockaddr, qs, SOCKADDR);
 234   }
 235   MarshallPtr (qs->lpcsaBuffer, qs, CSADDR_INFO);
 236   if (IsEqualGUID (&SVCID_INET_HOSTADDRBYNAME, sc) && qs->lpBlob != NULL && qs->lpBlob->pBlobData != NULL)
 237   {
 238     struct hostent *he;
 239     he = (struct hostent *) qs->lpBlob->pBlobData;
 240     for (i = 0; he->h_aliases[i] != NULL; i++)
 241       MarshallPtr (he->h_aliases[i], he, char);
 242     MarshallPtr (he->h_aliases, he, char *);
 243     MarshallPtr (he->h_name, he, char);
 244     for (i = 0; he->h_addr_list[i] != NULL; i++)
 245       MarshallPtr (he->h_addr_list[i], he, void);
 246     MarshallPtr (he->h_addr_list, he, char *);
 247     MarshallPtr (qs->lpBlob->pBlobData, qs, void);
 248   }
 249   MarshallPtr (qs->lpBlob, qs, BLOB);
 250 }
 251
 252 static void
 253 process_ip_lookup_result (void* cls, uint32_t rd_count,
 254     const struct GNUNET_NAMESTORE_RecordData *rd)
 255 {
 256   int i, j, csanum;
 257   struct request *rq = (struct request *) cls;
 258   struct GNUNET_W32RESOLVER_GetMessage *msg;
 259   struct GNUNET_MessageHeader *msgend;
 260   WSAQUERYSETW *qs;
 261   size_t size;
 262   size_t size_recalc;
 263   char *ptr;
 264   size_t blobsize = 0;
 265   size_t blobaddrcount = 0;
 266
 267   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 268               "Got lookup result with count %u for rq %p with client %p\n",
 269               rd_count, rq, rq->client);
 270
 271   if (rd_count == 0)
 272   {
 273     size = sizeof (struct GNUNET_MessageHeader);
 274     msg = GNUNET_malloc (size);
 275     msg->header.size = htons (size);
 276     msg->header.type = htons (GNUNET_MESSAGE_TYPE_W32RESOLVER_RESPONSE);
 277     transmit (rq->client, &msg->header, msg->header.size);
 278     return;
 279   }
 280
 281   size = sizeof (struct GNUNET_W32RESOLVER_GetMessage) + sizeof (WSAQUERYSETW);
 282   size += (wcslen (rq->name) + 1) * sizeof (wchar_t);
 283   size += sizeof (GUID);
 284   /* lpszComment ? a TXT record? */
 285   size += sizeof (GUID);
 286   /* lpszContext ? Not sure what it is */
 287   csanum = 0;
 288   for (i = 0; i < rd_count; i++)
 289   {
 290     switch (rd[i].record_type)
 291     {
 292     case GNUNET_GNS_RECORD_A:
 293       if (rd[i].data_size != sizeof (struct in_addr))
 294         continue;
 295       size += sizeof (CSADDR_INFO) + sizeof (struct sockaddr_in) * 2;
 296       csanum++;
 297       break;
 298     case GNUNET_GNS_RECORD_AAAA:
 299       if (rd[i].data_size != sizeof (struct in6_addr))
 300         continue;
 301       size += sizeof (CSADDR_INFO) + sizeof (struct sockaddr_in6) * 2;
 302       csanum++;
 303       break;
 304     }
 305   }
 306   if (IsEqualGUID (&SVCID_INET_HOSTADDRBYNAME, &rq->sc))
 307   {
 308     size += sizeof (BLOB);
 309     blobsize += sizeof (struct hostent);
 310     blobsize += strlen (rq->u8name) + 1;
 311     blobsize += sizeof (void *); /* For aliases */
 312     blobsize += sizeof (void *); /* For addresses */
 313     for (i = 0; i < rd_count; i++)
 314     {
 315       if ((rq->af == AF_INET || rq->af == AF_UNSPEC) && rd[i].record_type == GNUNET_GNS_RECORD_A)
 316       {
 317         blobsize += sizeof (void *);
 318         blobsize += sizeof (struct in_addr);
 319         blobaddrcount++;
 320       }
 321       else if (rq->af == AF_INET6 && rd[i].record_type == GNUNET_GNS_RECORD_AAAA)
 322       {
 323         blobsize += sizeof (void *);
 324         blobsize += sizeof (struct in6_addr);
 325         blobaddrcount++;
 326       }
 327     }
 328     size += blobsize;
 329   }
 330   size += sizeof (struct GNUNET_MessageHeader);
 331   size_recalc = sizeof (struct GNUNET_W32RESOLVER_GetMessage) + sizeof (WSAQUERYSETW);
 332   msg = GNUNET_malloc (size);
 333   msg->header.size = htons (size - sizeof (struct GNUNET_MessageHeader));
 334   msg->header.type = htons (GNUNET_MESSAGE_TYPE_W32RESOLVER_RESPONSE);
 335   msg->af = htonl (rq->af);
 336   msg->sc_data1 = htonl (rq->sc.Data1);
 337   msg->sc_data2 = htons (rq->sc.Data2);
 338   msg->sc_data3 = htons (rq->sc.Data3);
 339   msg->sc_data4 = 0;
 340   for (i = 0; i < 8; i++)
 341     msg->sc_data4 |= rq->sc.Data4[i] << ((7 - i) * 8);
 342   msg->sc_data4 = GNUNET_htonll (msg->sc_data4);
 343   qs = (WSAQUERYSETW *) &msg[1];
 344   ptr = (char *) &qs[1];
 345   GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 346   qs->dwSize = sizeof (WSAQUERYSETW);
 347   qs->lpszServiceInstanceName = (wchar_t *) ptr;
 348   ptr += (wcslen (rq->name) + 1) * sizeof (wchar_t);
 349   size_recalc += (wcslen (rq->name) + 1) * sizeof (wchar_t);
 350   GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 351   wcscpy (qs->lpszServiceInstanceName, rq->name);
 352   qs->lpServiceClassId = (GUID *) ptr;
 353   ptr += sizeof (GUID);
 354   size_recalc += sizeof (GUID);
 355   GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 356   memcpy (qs->lpServiceClassId, &rq->sc, sizeof (GUID));
 357   qs->lpVersion = NULL;
 358   qs->dwNameSpace = NS_DNS;
 359   qs->lpNSProviderId = (GUID *) ptr;
 360   ptr += sizeof (GUID);
 361   size_recalc += sizeof (GUID);
 362   GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 363   memcpy (qs->lpNSProviderId, &GNUNET_NAMESPACE_PROVIDER_DNS, sizeof (GUID));
 364   qs->lpszContext = NULL;
 365   qs->dwNumberOfProtocols = 0;
 366   qs->lpafpProtocols = NULL;
 367   /* Don't bother with this... */
 368   qs->lpszQueryString = NULL;
 369   qs->dwNumberOfCsAddrs = rd_count;
 370   qs->lpcsaBuffer = (CSADDR_INFO *) ptr;
 371   ptr += sizeof (CSADDR_INFO) * csanum;
 372   j = 0;
 373   for (i = 0; i < rd_count; i++)
 374   {
 375     switch (rd[i].record_type)
 376     {
 377     case GNUNET_GNS_RECORD_A:
 378       if (rd[i].data_size != sizeof (struct in_addr))
 379         continue;
 380       qs->lpcsaBuffer[j].iSocketType = SOCK_STREAM;
 381       qs->lpcsaBuffer[j].iProtocol = IPPROTO_TCP;
 382
 383       qs->lpcsaBuffer[j].LocalAddr.iSockaddrLength = sizeof (struct sockaddr_in);
 384       qs->lpcsaBuffer[j].LocalAddr.lpSockaddr = (SOCKADDR *) ptr;
 385       ptr += qs->lpcsaBuffer[j].LocalAddr.iSockaddrLength;
 386       memset (qs->lpcsaBuffer[j].LocalAddr.lpSockaddr, 0, qs->lpcsaBuffer[j].LocalAddr.iSockaddrLength);
 387       ((struct sockaddr_in *)qs->lpcsaBuffer[j].LocalAddr.lpSockaddr)->sin_family = AF_INET;
 388
 389       qs->lpcsaBuffer[j].RemoteAddr.iSockaddrLength = sizeof (struct sockaddr_in);
 390       qs->lpcsaBuffer[j].RemoteAddr.lpSockaddr = (SOCKADDR *) ptr;
 391       ptr += qs->lpcsaBuffer[j].RemoteAddr.iSockaddrLength;
 392       memset (qs->lpcsaBuffer[j].RemoteAddr.lpSockaddr, 0, qs->lpcsaBuffer[j].RemoteAddr.iSockaddrLength);
 393       ((struct sockaddr_in *)qs->lpcsaBuffer[j].RemoteAddr.lpSockaddr)->sin_family = AF_INET;
 394       ((struct sockaddr_in *)qs->lpcsaBuffer[j].RemoteAddr.lpSockaddr)->sin_port = htonl (53); /* Don't ask why it's 53 */
 395       ((struct sockaddr_in *)qs->lpcsaBuffer[j].RemoteAddr.lpSockaddr)->sin_addr = *(struct in_addr *) rd[i].data;
 396       size_recalc += sizeof (CSADDR_INFO) + sizeof (struct sockaddr_in) * 2;
 397       j++;
 398       break;
 399     case GNUNET_GNS_RECORD_AAAA:
 400       if (rd[i].data_size != sizeof (struct in6_addr))
 401         continue;
 402       qs->lpcsaBuffer[j].iSocketType = SOCK_STREAM;
 403       qs->lpcsaBuffer[j].iProtocol = IPPROTO_TCP;
 404
 405       qs->lpcsaBuffer[j].LocalAddr.iSockaddrLength = sizeof (struct sockaddr_in6);
 406       qs->lpcsaBuffer[j].LocalAddr.lpSockaddr = (SOCKADDR *) ptr;
 407       ptr += qs->lpcsaBuffer[j].LocalAddr.iSockaddrLength;
 408       memset (qs->lpcsaBuffer[j].LocalAddr.lpSockaddr, 0, qs->lpcsaBuffer[j].LocalAddr.iSockaddrLength);
 409       ((struct sockaddr_in6 *)qs->lpcsaBuffer[j].LocalAddr.lpSockaddr)->sin6_family = AF_INET6;
 410
 411       qs->lpcsaBuffer[j].RemoteAddr.iSockaddrLength = sizeof (struct sockaddr_in6);
 412       qs->lpcsaBuffer[j].RemoteAddr.lpSockaddr = (SOCKADDR *) ptr;
 413       ptr += qs->lpcsaBuffer[j].RemoteAddr.iSockaddrLength;
 414       memset (qs->lpcsaBuffer[j].RemoteAddr.lpSockaddr, 0, qs->lpcsaBuffer[j].RemoteAddr.iSockaddrLength);
 415       ((struct sockaddr_in6 *)qs->lpcsaBuffer[j].RemoteAddr.lpSockaddr)->sin6_family = AF_INET6;
 416       ((struct sockaddr_in6 *)qs->lpcsaBuffer[j].RemoteAddr.lpSockaddr)->sin6_port = htonl (53); /* Don't ask why it's 53 */
 417       ((struct sockaddr_in6 *)qs->lpcsaBuffer[j].RemoteAddr.lpSockaddr)->sin6_addr = *(struct in6_addr *) rd[i].data;
 418       size_recalc += sizeof (CSADDR_INFO) + sizeof (struct sockaddr_in6) * 2;
 419       j++;
 420       break;
 421     default:
 422       break;
 423     }
 424   }
 425   GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 426   qs->dwOutputFlags = 0;
 427   if (IsEqualGUID (&SVCID_INET_HOSTADDRBYNAME, &rq->sc))
 428   {
 429     struct hostent *he;
 430     qs->lpBlob = (BLOB *) ptr;
 431     ptr += sizeof (BLOB);
 432
 433     size_recalc += sizeof (BLOB);
 434     GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 435
 436     qs->lpBlob->cbSize = blobsize;
 437     qs->lpBlob->pBlobData = (BYTE *) ptr;
 438     ptr += sizeof (struct hostent);
 439
 440     size_recalc += sizeof (struct hostent);
 441     GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 442
 443     he = (struct hostent *) qs->lpBlob->pBlobData;
 444     he->h_name = (char *) ptr;
 445     ptr += strlen (rq->u8name) + 1;
 446
 447     size_recalc += strlen (rq->u8name) + 1;
 448     GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 449
 450     strcpy (he->h_name, rq->u8name);
 451     he->h_aliases = (char **) ptr;
 452     ptr += sizeof (void *);
 453
 454     size_recalc += sizeof (void *); /* For aliases */
 455     GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 456
 457     he->h_aliases[0] = NULL;
 458     he->h_addrtype = rq->af;
 459     he->h_length = rq->af == AF_INET || rq->af == AF_UNSPEC ? sizeof (struct in_addr) : sizeof (struct in6_addr);
 460     he->h_addr_list = (char **) ptr;
 461     ptr += sizeof (void *) * (blobaddrcount + 1);
 462
 463     size_recalc += sizeof (void *) * (blobaddrcount + 1); /* For addresses */
 464     GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 465
 466     j = 0;
 467     for (i = 0; i < rd_count; i++)
 468     {
 469       if ((rq->af == AF_INET || rq->af == AF_UNSPEC) &&
 470           rd[i].record_type == GNUNET_GNS_RECORD_A)
 471       {
 472         he->h_addr_list[j] = (char *) ptr;
 473         ptr += sizeof (struct in_addr);
 474
 475         size_recalc += sizeof (struct in_addr);
 476         GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 477
 478         memcpy (he->h_addr_list[j], rd[i].data, sizeof (struct in_addr));
 479         j++;
 480       }
 481       else if (rq->af == AF_INET6 && rd[i].record_type == GNUNET_GNS_RECORD_AAAA)
 482       {
 483         he->h_addr_list[j] = (char *) ptr;
 484         ptr += sizeof (struct in6_addr);
 485
 486         size_recalc += sizeof (struct in6_addr);
 487         GNUNET_break (size_recalc == (size_t) ((char *) ptr - (char *) msg));
 488
 489         memcpy (he->h_addr_list[j], rd[i].data, sizeof (struct in6_addr));
 490         j++;
 491       }
 492     }
 493     he->h_addr_list[j] = NULL;
 494   }
 495   msgend = (struct GNUNET_MessageHeader *) ptr;
 496   ptr += sizeof (struct GNUNET_MessageHeader);
 497   size_recalc += sizeof (struct GNUNET_MessageHeader);
 498
 499   msgend->type = htons (GNUNET_MESSAGE_TYPE_W32RESOLVER_RESPONSE);
 500   msgend->size = htons (sizeof (struct GNUNET_MessageHeader));
 501
 502   if ((char *) ptr - (char *) msg != size || size_recalc != size || size_recalc != ((char *) ptr - (char *) msg))
 503   {
 504     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Error in WSAQUERYSETW size calc: expected %lu, got %lu (recalc %lu)\n", size, (unsigned long) ((char *) ptr - (char *) msg), size_recalc);
 505   }
 506   MarshallWSAQUERYSETW (qs, &rq->sc);
 507   transmit (rq->client, &msg->header, size);
 508 }
 509
 510 static void
 511 get_ip_from_hostname (struct GNUNET_SERVER_Client *client,
 512     const wchar_t *name, int af, GUID sc)
 513 {
 514   struct request *rq;
 515   char *hostname;
 516   size_t strl;
 517   size_t namelen;
 518   uint32_t rtype;
 519
 520   if (IsEqualGUID (&SVCID_DNS_TYPE_A, &sc))
 521     rtype = GNUNET_GNS_RECORD_A;
 522   else if (IsEqualGUID (&SVCID_DNS_TYPE_NS, &sc))
 523     rtype = GNUNET_GNS_RECORD_NS;
 524   else if (IsEqualGUID (&SVCID_DNS_TYPE_CNAME, &sc))
 525     rtype = GNUNET_GNS_RECORD_CNAME;
 526   else if (IsEqualGUID (&SVCID_DNS_TYPE_SOA, &sc))
 527     rtype = GNUNET_GNS_RECORD_SOA;
 528   else if (IsEqualGUID (&SVCID_DNS_TYPE_PTR, &sc))
 529     rtype = GNUNET_GNS_RECORD_PTR;
 530   else if (IsEqualGUID (&SVCID_DNS_TYPE_MX, &sc))
 531     rtype = GNUNET_GNS_RECORD_MX;
 532   else if (IsEqualGUID (&SVCID_DNS_TYPE_TEXT, &sc))
 533     rtype = GNUNET_GNS_RECORD_TXT;
 534   else if (IsEqualGUID (&SVCID_DNS_TYPE_AAAA, &sc))
 535     rtype = GNUNET_GNS_RECORD_AAAA;
 536   else if (IsEqualGUID (&SVCID_DNS_TYPE_SRV, &sc))
 537     rtype = GNUNET_GNS_RECORD_SRV;
 538   else if (IsEqualGUID (&SVCID_INET_HOSTADDRBYNAME, &sc))
 539     rtype = GNUNET_GNS_RECORD_A;
 540   else
 541   {
 542     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 543                 "Unknown GUID: %08X-%04X-%04X-%02X-%02X-%02X-%02X-%02X-%02X-%02X-%02X\n",
 544                 sc.Data1, sc.Data2, sc.Data3, sc.Data4[0], sc.Data4[1], sc.Data4[2],
 545                 sc.Data4[3], sc.Data4[4], sc.Data4[5], sc.Data4[6], sc.Data4[7]);
 546     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 547     return;
 548   }
 549
 550   if (name)
 551     namelen = wcslen (name);
 552   else
 553     namelen = 0;
 554   if (namelen > 0)
 555     hostname = (char *) u16_to_u8 (name, namelen + 1, NULL, &strl);
 556
 557   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 558               "W32 DNS resolver asked to look up %s for `%s'.\n",
 559               af == AF_INET ? "IPv4" : af == AF_INET6 ? "IPv6" : "anything",
 560               hostname);
 561   rq = GNUNET_malloc (sizeof (struct request));
 562   rq->sc = sc;
 563   rq->client = client;
 564   rq->af = af;
 565   if (rq->af != AF_INET && rq->af != AF_INET6)
 566     rq->af = AF_INET;
 567   if (namelen)
 568   {
 569     rq->name = GNUNET_malloc ((namelen + 1) * sizeof (wchar_t));
 570     memcpy (rq->name, name, (namelen + 1) * sizeof (wchar_t));
 571     rq->u8name = hostname;
 572   }
 573
 574   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 575               "Launching a lookup for client %p with rq %p\n",
 576               client, rq);
 577
 578   if (NULL != GNUNET_GNS_lookup_zone (gns, hostname, zone, rtype,
 579       GNUNET_YES, shorten_key, &process_ip_lookup_result, rq))
 580   {
 581     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 582                 "Lookup launched, waiting for a reply\n");
 583     GNUNET_SERVER_client_keep (client);
 584     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 585   }
 586   else
 587   {
 588     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 589                 "Lookup was not, disconnecting the client\n");
 590     if (namelen)
 591     {
 592       GNUNET_free (rq->name);
 593       free (rq->u8name);
 594     }
 595     GNUNET_free (rq);
 596     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 597   }
 598 }
 599
 600 /**
 601  * Handle GET-message.
 602  *
 603  * @param cls closure
 604  * @param client identification of the client
 605  * @param message the actual message
 606  */
 607 static void
 608 handle_get (void *cls, struct GNUNET_SERVER_Client *client,
 609             const struct GNUNET_MessageHeader *message)
 610 {
 611   uint16_t msize;
 612   const struct GNUNET_W32RESOLVER_GetMessage *msg;
 613   GUID sc;
 614   uint16_t size;
 615   uint64_t data4;
 616   int i;
 617   const wchar_t *hostname;
 618   int af;
 619
 620   msize = ntohs (message->size);
 621   if (msize < sizeof (struct GNUNET_W32RESOLVER_GetMessage))
 622   {
 623     GNUNET_break (0);
 624     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 625     return;
 626   }
 627   msg = (const struct GNUNET_W32RESOLVER_GetMessage *) message;
 628   size = msize - sizeof (struct GNUNET_W32RESOLVER_GetMessage);
 629   af = ntohl (msg->af);
 630   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got NBO GUID: %08X-%04X-%04X-%016llX\n",
 631       msg->sc_data1, msg->sc_data2, msg->sc_data3, msg->sc_data4);
 632   sc.Data1 = ntohl (msg->sc_data1);
 633   sc.Data2 = ntohs (msg->sc_data2);
 634   sc.Data3 = ntohs (msg->sc_data3);
 635   data4 = GNUNET_ntohll (msg->sc_data4);
 636   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Got GUID: %08X-%04X-%04X-%016llX\n",
 637       sc.Data1, sc.Data2, sc.Data3, data4);
 638   for (i = 0; i < 8; i++)
 639     sc.Data4[i] = 0xFF & (data4 >> ((7 - i) * 8));
 640
 641   hostname = (const wchar_t *) &msg[1];
 642   if (hostname[size - 1] != L'\0')
 643   {
 644     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "name of length %u, not 0-terminated: %*S\n",
 645         size, size, hostname);
 646     GNUNET_break (0);
 647     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 648     return;
 649   }
 650   get_ip_from_hostname (client, hostname, af, sc);
 651   return;
 652 }
 653
 654
 655 /**
 656  * Start up gns-helper-w32 service.
 657  *
 658  * @param cls closure
 659  * @param server the initialized server
 660  * @param cfg configuration to use
 661  */
 662 static void
 663 run (void *cls, struct GNUNET_SERVER_Handle *server,
 664      const struct GNUNET_CONFIGURATION_Handle *cfg)
 665 {
 666   static const struct GNUNET_SERVER_MessageHandler handlers[] = {
 667     {&handle_get, NULL, GNUNET_MESSAGE_TYPE_W32RESOLVER_REQUEST, 0},
 668     {NULL, NULL, 0, 0}
 669   };
 670
 671   char* keyfile;
 672   struct GNUNET_CRYPTO_RsaPrivateKey *key = NULL;
 673   struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded pkey;
 674   struct GNUNET_CRYPTO_ShortHashAsciiEncoded zonename;
 675
 676   if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (cfg, "gns",
 677                                                            "ZONEKEY", &keyfile))
 678   {
 679     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
 680                 "No private key for root zone found, using default!\n");
 681     zone = NULL;
 682   }
 683   else
 684   {
 685     if (GNUNET_YES == GNUNET_DISK_file_test (keyfile))
 686     {
 687       key = GNUNET_CRYPTO_rsa_key_create_from_file (keyfile);
 688       GNUNET_CRYPTO_rsa_key_get_public (key, &pkey);
 689       GNUNET_CRYPTO_short_hash(&pkey,
 690                          sizeof(struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded),
 691                          &user_zone);
 692       zone = &user_zone;
 693       GNUNET_CRYPTO_short_hash_to_enc (zone, &zonename);
 694       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 695                   "Using zone: %s!\n", &zonename);
 696       GNUNET_CRYPTO_rsa_key_free(key);
 697     }
 698     GNUNET_free(keyfile);
 699   }
 700
 701   if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_filename (cfg, "gns",
 702                                                    "SHORTEN_ZONEKEY", &keyfile))
 703   {
 704     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
 705                 "No shorten key found!\n");
 706     shorten_key = NULL;
 707   }
 708   else
 709   {
 710     if (GNUNET_YES == GNUNET_DISK_file_test (keyfile))
 711     {
 712       shorten_key = GNUNET_CRYPTO_rsa_key_create_from_file (keyfile);
 713     }
 714     GNUNET_free(keyfile);
 715   }
 716
 717   gns = GNUNET_GNS_connect (cfg);
 718   if (gns == NULL)
 719     return;
 720
 721   GNUNET_SERVER_add_handlers (server, handlers);
 722   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &do_shutdown,
 723                                 NULL);
 724
 725 }
 726
 727
 728 /**
 729  * The main function for gns-helper-w32.
 730  *
 731  * @param argc number of arguments from the command line
 732  * @param argv command line arguments
 733  * @return 0 ok, 1 on error
 734  */
 735 int
 736 main (int argc, char *const *argv)
 737 {
 738   int ret;
 739
 740   ret =
 741       (GNUNET_OK ==
 742        GNUNET_SERVICE_run (argc, argv, "gns-helper-service-w32", GNUNET_SERVICE_OPTION_NONE,
 743                            &run, NULL)) ? 0 : 1;
 744
 745   return ret;
 746 }
 747
 748 /* end of gnunet-gns.c */