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