src/util/gnunet-service-resolver.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2007, 2008, 2009 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 2, 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 /**
  22  * @file util/gnunet-service-resolver.c
  23  * @brief code to do DNS resolution
  24  * @author Christian Grothoff
  25  */
  26
  27 #include <stdlib.h>
  28 #include "platform.h"
  29 #include "gnunet_disk_lib.h"
  30 #include "gnunet_getopt_lib.h"
  31 #include "gnunet_protocols.h"
  32 #include "gnunet_service_lib.h"
  33 #include "gnunet_statistics_service.h"
  34 #include "gnunet_strings_lib.h"
  35 #include "gnunet_time_lib.h"
  36 #include "resolver.h"
  37
  38 /**
  39  * A cached DNS lookup result.
  40  */
  41 struct IPCache
  42 {
  43   /**
  44    * This is a linked list.
  45    */
  46   struct IPCache *next;
  47
  48   /**
  49    * Hostname in human-readable form.
  50    */
  51   char *addr;
  52
  53   /**
  54    * Hostname in binary format.
  55    */
  56   struct sockaddr *sa;
  57
  58   /**
  59    * Last time this entry was updated.
  60    */
  61   struct GNUNET_TIME_Absolute last_refresh;
  62
  63   /**
  64    * Last time this entry was requested.
  65    */
  66   struct GNUNET_TIME_Absolute last_request;
  67
  68   /**
  69    * Number of bytes in sa.
  70    */
  71   socklen_t salen;
  72 };
  73
  74
  75 /**
  76  * Start of the linked list of cached DNS lookup results.
  77  */
  78 static struct IPCache *head;
  79
  80
  81 #if HAVE_GETNAMEINFO
  82 /**
  83  * Resolve the given request using getnameinfo
  84  *
  85  * @param cache the request to resolve (and where to store the result)
  86  */
  87 static void
  88 getnameinfo_resolve (struct IPCache *cache)
  89 {
  90   char hostname[256];
  91
  92   if (0 == getnameinfo (cache->sa,
  93                         cache->salen,
  94                         hostname, sizeof (hostname), NULL, 0, 0))
  95     cache->addr = GNUNET_strdup (hostname);
  96 }
  97 #endif
  98
  99
 100 #if HAVE_GETHOSTBYADDR
 101 /**
 102  * Resolve the given request using gethostbyaddr
 103  *
 104  * @param cache the request to resolve (and where to store the result)
 105  */
 106 static void
 107 gethostbyaddr_resolve (struct IPCache *cache)
 108 {
 109   struct hostent *ent;
 110
 111   switch (cache->sa->sa_family)
 112     {
 113     case AF_INET:
 114       ent = gethostbyaddr (&((struct sockaddr_in *) cache->sa)->sin_addr,
 115                            sizeof (struct in_addr), AF_INET);
 116       break;
 117     case AF_INET6:
 118       ent = gethostbyaddr (&((struct sockaddr_in6 *) cache->sa)->sin6_addr,
 119                            sizeof (struct in6_addr), AF_INET6);
 120       break;
 121     default:
 122       ent = NULL;
 123     }
 124   if (ent != NULL)
 125     cache->addr = GNUNET_strdup (ent->h_name);
 126 }
 127 #endif
 128
 129 /**
 130  * Resolve the given request using the available methods.
 131  *
 132  * @param cache the request to resolve (and where to store the result)
 133  */
 134 static void
 135 cache_resolve (struct IPCache *cache)
 136 {
 137 #if HAVE_GETNAMEINFO
 138   if (cache->addr == NULL)
 139     getnameinfo_resolve (cache);
 140 #endif
 141 #if HAVE_GETHOSTBYADDR
 142   if (cache->addr == NULL)
 143     gethostbyaddr_resolve (cache);
 144 #endif
 145 }
 146
 147
 148
 149 /**
 150  * Get an IP address as a string (works for both IPv4 and IPv6).  Note
 151  * that the resolution happens asynchronously and that the first call
 152  * may not immediately result in the FQN (but instead in a
 153  * human-readable IP address).
 154  *
 155  * @param client handle to the client making the request (for sending the reply)
 156  * @param sa should be of type "struct sockaddr*"
 157  * @param salen number of bytes in sa
 158  */
 159 static void
 160 get_ip_as_string (struct GNUNET_SERVER_Client *client,
 161                   const struct sockaddr *sa, socklen_t salen)
 162 {
 163   struct IPCache *cache;
 164   struct IPCache *prev;
 165   struct GNUNET_TIME_Absolute now;
 166   struct GNUNET_SERVER_TransmitContext *tc;
 167
 168   if (salen < sizeof (struct sockaddr))
 169     {
 170       GNUNET_break (0);
 171       return;
 172     }
 173   now = GNUNET_TIME_absolute_get ();
 174   cache = head;
 175   prev = NULL;
 176   while ((cache != NULL) &&
 177          ((cache->salen != salen) || (0 != memcmp (cache->sa, sa, salen))))
 178     {
 179       if (GNUNET_TIME_absolute_get_duration (cache->last_request).value <
 180           60 * 60 * 1000)
 181         {
 182           if (prev != NULL)
 183             {
 184               prev->next = cache->next;
 185               GNUNET_free_non_null (cache->addr);
 186               GNUNET_free (cache->sa);
 187               GNUNET_free (cache);
 188               cache = prev->next;
 189             }
 190           else
 191             {
 192               head = cache->next;
 193               GNUNET_free_non_null (cache->addr);
 194               GNUNET_free (cache->sa);
 195               GNUNET_free (cache);
 196               cache = head;
 197             }
 198           continue;
 199         }
 200       prev = cache;
 201       cache = cache->next;
 202     }
 203   if (cache != NULL)
 204     {
 205       cache->last_request = now;
 206       if (GNUNET_TIME_absolute_get_duration (cache->last_request).value <
 207           60 * 60 * 1000)
 208         {
 209           GNUNET_free_non_null (cache->addr);
 210           cache->addr = NULL;
 211           cache->salen = 0;
 212           cache_resolve (cache);
 213         }
 214     }
 215   else
 216     {
 217       cache = GNUNET_malloc (sizeof (struct IPCache));
 218       cache->next = head;
 219       cache->salen = salen;
 220       cache->sa = GNUNET_malloc (salen);
 221       memcpy (cache->sa, sa, salen);
 222       cache->last_request = GNUNET_TIME_absolute_get ();
 223       cache->last_refresh = GNUNET_TIME_absolute_get ();
 224       cache->addr = NULL;
 225       cache_resolve (cache);
 226       head = cache;
 227     }
 228   tc = GNUNET_SERVER_transmit_context_create (client);
 229   if (cache->addr != NULL)
 230     GNUNET_SERVER_transmit_context_append (tc,
 231                                            cache->addr,
 232                                            strlen (cache->addr) + 1,
 233                                            GNUNET_MESSAGE_TYPE_RESOLVER_RESPONSE);
 234   GNUNET_SERVER_transmit_context_append (tc, NULL, 0,
 235                                          GNUNET_MESSAGE_TYPE_RESOLVER_RESPONSE);
 236   GNUNET_SERVER_transmit_context_run (tc, GNUNET_TIME_UNIT_FOREVER_REL);
 237 }
 238
 239
 240 #if HAVE_GETADDRINFO
 241 static int
 242 getaddrinfo_resolve (struct GNUNET_SERVER_TransmitContext *tc,
 243                      const char *hostname, int domain)
 244 {
 245   int s;
 246   struct addrinfo hints;
 247   struct addrinfo *result;
 248   struct addrinfo *pos;
 249
 250   memset (&hints, 0, sizeof (struct addrinfo));
 251 // FIXME in PlibC
 252 #ifndef MINGW
 253   hints.ai_family = domain;
 254 #else
 255   hints.ai_family = AF_INET;
 256 #endif
 257   hints.ai_socktype = SOCK_STREAM;      /* go for TCP */
 258
 259   if (0 != (s = getaddrinfo (hostname, NULL, &hints, &result)))
 260     {
 261       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
 262                   _("Could not resolve `%s' (%s): %s\n"), hostname,
 263                   (domain ==
 264                    AF_INET) ? "IPv4" : ((domain ==
 265                                          AF_INET6) ? "IPv6" : "any"),
 266                   gai_strerror (s));
 267       if ((s == EAI_BADFLAGS) || (s == EAI_MEMORY) ||
 268 #ifndef MINGW
 269           (s == EAI_SYSTEM)
 270 #else
 271           // FIXME NILS
 272           1
 273 #endif
 274         )
 275         return GNUNET_NO;       /* other function may still succeed */
 276       return GNUNET_SYSERR;
 277     }
 278   if (result == NULL)
 279     return GNUNET_SYSERR;
 280   pos = result;
 281   while (pos != NULL)
 282     {
 283       GNUNET_SERVER_transmit_context_append (tc,
 284                                              result->ai_addr,
 285                                              result->ai_addrlen,
 286                                              GNUNET_MESSAGE_TYPE_RESOLVER_RESPONSE);
 287       pos = pos->ai_next;
 288     }
 289   freeaddrinfo (result);
 290   return GNUNET_OK;
 291 }
 292 #endif
 293
 294 #if HAVE_GETHOSTBYNAME2
 295 static int
 296 gethostbyname2_resolve (struct GNUNET_SERVER_TransmitContext *tc,
 297                         const char *hostname, int domain)
 298 {
 299   struct hostent *hp;
 300   struct sockaddr_in a4;
 301   struct sockaddr_in6 a6;
 302   int ret1;
 303   int ret2;
 304
 305   if (domain == AF_UNSPEC)
 306     {
 307       ret1 = gethostbyname2_resolve (tc, hostname, AF_INET);
 308       ret2 = gethostbyname2_resolve (tc, hostname, AF_INET6);
 309       if ((ret1 == GNUNET_OK) || (ret2 == GNUNET_OK))
 310         return GNUNET_OK;
 311       if ((ret1 == GNUNET_SYSERR) || (ret2 == GNUNET_SYSERR))
 312         return GNUNET_SYSERR;
 313       return GNUNET_NO;
 314     }
 315   hp = gethostbyname2 (hostname, domain);
 316   if (hp == NULL)
 317     {
 318       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
 319                   _("Could not find IP of host `%s': %s\n"),
 320                   hostname, hstrerror (h_errno));
 321       return GNUNET_SYSERR;
 322     }
 323   GNUNET_assert (hp->h_addrtype == domain);
 324   if (domain == AF_INET)
 325     {
 326       GNUNET_assert (hp->h_length == sizeof (struct in_addr));
 327       memset (&a4, 0, sizeof (a4));
 328       a4.sin_family = AF_INET;
 329       memcpy (&a4.sin_addr, hp->h_addr_list[0], hp->h_length);
 330       GNUNET_SERVER_transmit_context_append (tc,
 331                                              &a4,
 332                                              sizeof (a4),
 333                                              GNUNET_MESSAGE_TYPE_RESOLVER_RESPONSE);
 334     }
 335   else
 336     {
 337       GNUNET_assert (hp->h_length == sizeof (struct in6_addr));
 338       memset (&a6, 0, sizeof (a6));
 339       a6.sin6_family = AF_INET6;
 340       memcpy (&a6.sin6_addr, hp->h_addr_list[0], hp->h_length);
 341       GNUNET_SERVER_transmit_context_append (tc,
 342                                              &a6,
 343                                              sizeof (a6),
 344                                              GNUNET_MESSAGE_TYPE_RESOLVER_RESPONSE);
 345     }
 346   return GNUNET_OK;
 347 }
 348 #endif
 349
 350 #if HAVE_GETHOSTBYNAME
 351 static int
 352 gethostbyname_resolve (struct GNUNET_SERVER_TransmitContext *tc,
 353                        const char *hostname)
 354 {
 355   struct hostent *hp;
 356   struct sockaddr_in addr;
 357
 358   hp = GETHOSTBYNAME (hostname);
 359   if (hp == NULL)
 360     {
 361       GNUNET_log (GNUNET_ERROR_TYPE_INFO,
 362                   _("Could not find IP of host `%s': %s\n"),
 363                   hostname, hstrerror (h_errno));
 364       return GNUNET_SYSERR;
 365     }
 366   if (hp->h_addrtype != AF_INET)
 367     {
 368       GNUNET_break (0);
 369       return GNUNET_SYSERR;
 370     }
 371   GNUNET_assert (hp->h_length == sizeof (struct in_addr));
 372   memset (&addr, 0, sizeof (addr));
 373   addr.sin_family = AF_INET;
 374   memcpy (&addr.sin_addr, hp->h_addr_list[0], hp->h_length);
 375   GNUNET_SERVER_transmit_context_append (tc,
 376                                          &addr,
 377                                          sizeof (addr),
 378                                          GNUNET_MESSAGE_TYPE_RESOLVER_RESPONSE);
 379   return GNUNET_OK;
 380 }
 381 #endif
 382
 383
 384 /**
 385  * Convert a string to an IP address.
 386  *
 387  * @param client where to send the IP address
 388  * @param hostname the hostname to resolve
 389  * @param domain AF_INET or AF_INET6; use AF_UNSPEC for "any"
 390  */
 391 static void
 392 get_ip_from_hostname (struct GNUNET_SERVER_Client *client,
 393                       const char *hostname, int domain)
 394 {
 395   int ret;
 396   struct GNUNET_SERVER_TransmitContext *tc;
 397
 398   tc = GNUNET_SERVER_transmit_context_create (client);
 399   ret = GNUNET_NO;
 400 #if HAVE_GETADDRINFO
 401   if (ret == GNUNET_NO)
 402     ret = getaddrinfo_resolve (tc, hostname, domain);
 403 #endif
 404 #if HAVE_GETHOSTBYNAME2
 405   if (ret == GNUNET_NO)
 406     ret = gethostbyname2_resolve (tc, hostname, domain);
 407 #endif
 408 #if HAVE_GETHOSTBYNAME
 409   if ((ret == GNUNET_NO) && ((domain == AF_UNSPEC) || (domain == PF_INET)))
 410     gethostbyname_resolve (tc, hostname);
 411 #endif
 412   GNUNET_SERVER_transmit_context_append (tc, NULL, 0,
 413                                          GNUNET_MESSAGE_TYPE_RESOLVER_RESPONSE);
 414   GNUNET_SERVER_transmit_context_run (tc, GNUNET_TIME_UNIT_FOREVER_REL);
 415 }
 416
 417
 418 /**
 419  * Handle GET-message.
 420  *
 421  * @param cls closure
 422  * @param client identification of the client
 423  * @param message the actual message
 424  */
 425 static void
 426 handle_get (void *cls,
 427             struct GNUNET_SERVER_Client *client,
 428             const struct GNUNET_MessageHeader *message)
 429 {
 430   uint16_t msize;
 431   const struct GNUNET_RESOLVER_GetMessage *msg;
 432   const char *hostname;
 433   uint16_t size;
 434   int direction;
 435   int domain;
 436
 437   msize = ntohs (message->size);
 438   if (msize < sizeof (struct GNUNET_RESOLVER_GetMessage))
 439     {
 440       GNUNET_break (0);
 441       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 442       return;
 443     }
 444   msg = (const struct GNUNET_RESOLVER_GetMessage *) message;
 445   size = msize - sizeof (struct GNUNET_RESOLVER_GetMessage);
 446   direction = ntohl (msg->direction);
 447   domain = ntohl (msg->domain);
 448   if (direction == GNUNET_NO)
 449     {
 450       /* IP from hostname */
 451       hostname = (const char *) &msg[1];
 452       if (hostname[size - 1] != '\0')
 453         {
 454           GNUNET_break (0);
 455           GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
 456           return;
 457         }
 458 #if DEBUG_RESOLVER
 459       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 460                   _("Resolver asked to look up `%s'.\n"), hostname);
 461 #endif
 462       get_ip_from_hostname (client, hostname, domain);
 463     }
 464   else
 465     {
 466 #if DEBUG_RESOLVER
 467       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 468                   _("Resolver asked to look up IP address.\n"));
 469 #endif
 470       get_ip_as_string (client, (const struct sockaddr *) &msg[1], size);
 471     }
 472 }
 473
 474
 475 /**
 476  * List of handlers for the messages understood by this
 477  * service.
 478  */
 479 static struct GNUNET_SERVER_MessageHandler handlers[] = {
 480   {&handle_get, NULL, GNUNET_MESSAGE_TYPE_RESOLVER_REQUEST, 0},
 481   {NULL, NULL, 0, 0}
 482 };
 483
 484
 485 /**
 486  * Process resolver requests.
 487  *
 488  * @param cls closure
 489  * @param sched scheduler to use
 490  * @param server the initialized server
 491  * @param cfg configuration to use
 492  */
 493 static void
 494 run (void *cls,
 495      struct GNUNET_SCHEDULER_Handle *sched,
 496      struct GNUNET_SERVER_Handle *server,
 497      const struct GNUNET_CONFIGURATION_Handle *cfg)
 498 {
 499   GNUNET_SERVER_add_handlers (server, handlers);
 500 }
 501
 502
 503 /**
 504  * The main function for the resolver service.
 505  *
 506  * @param argc number of arguments from the command line
 507  * @param argv command line arguments
 508  * @return 0 ok, 1 on error
 509  */
 510 int
 511 main (int argc, char *const *argv)
 512 {
 513   int ret;
 514   struct IPCache *pos;
 515
 516   ret = (GNUNET_OK ==
 517          GNUNET_SERVICE_run (argc,
 518                              argv,
 519                              "resolver", GNUNET_SERVICE_OPTION_NONE,
 520                              &run, NULL)) ? 0 : 1;
 521
 522   while (head != NULL)
 523     {
 524       pos = head->next;
 525       GNUNET_free_non_null (head->addr);
 526       GNUNET_free (head->sa);
 527       GNUNET_free (head);
 528       head = pos;
 529     }
 530   return ret;
 531 }
 532
 533 /* end of gnunet-service-resolver.c */