src/util/gnunet-service-resolver.c

   1 /*
   2      This file is part of GNUnet.
   3      Copyright (C) 2007-2016 GNUnet e.V.
   4
   5      GNUnet is free software: you can redistribute it and/or modify it
   6      under the terms of the GNU Affero General Public License as published
   7      by the Free Software Foundation, either version 3 of the License,
   8      or (at your 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      Affero General Public License for more details.
  14
  15      You should have received a copy of the GNU Affero General Public License
  16      along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17
  18      SPDX-License-Identifier: AGPL3.0-or-later
  19 */
  20
  21 /**
  22  * @file util/gnunet-service-resolver.c
  23  * @brief code to do DNS resolution
  24  * @author Christian Grothoff
  25  */
  26 #include "platform.h"
  27 #include "gnunet_util_lib.h"
  28 #include "gnunet_protocols.h"
  29 #include "gnunet_statistics_service.h"
  30 #include "resolver.h"
  31
  32
  33 /**
  34  * How long do we wait for DNS answers?
  35  */
  36 #define DNS_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 30)
  37
  38 /**
  39  * Maximum number of hostnames we cache results for.
  40  */
  41 #define MAX_CACHE 1024
  42
  43 /**
  44  * Entry in list of cached DNS records for a hostname.
  45  */
  46 struct RecordListEntry
  47 {
  48   /**
  49    * This is a doubly linked list.
  50    */
  51   struct RecordListEntry *next;
  52
  53   /**
  54    * This is a doubly linked list.
  55    */
  56   struct RecordListEntry *prev;
  57
  58   /**
  59    * Cached data.
  60    */
  61   struct GNUNET_DNSPARSER_Record *record;
  62
  63 };
  64
  65
  66 /**
  67  * A cached DNS lookup result.
  68  */
  69 struct ResolveCache
  70 {
  71   /**
  72    * This is a doubly linked list.
  73    */
  74   struct ResolveCache *next;
  75
  76   /**
  77    * This is a doubly linked list.
  78    */
  79   struct ResolveCache *prev;
  80
  81   /**
  82    * Which hostname is this cache for?
  83    */
  84   char *hostname;
  85
  86   /**
  87    * head of a double linked list containing the lookup results
  88    */
  89   struct RecordListEntry *records_head;
  90
  91   /**
  92    * tail of a double linked list containing the lookup results
  93    */
  94   struct RecordListEntry *records_tail;
  95
  96 };
  97
  98
  99 /**
 100  * Information about pending lookups.
 101  */
 102 struct ActiveLookup
 103 {
 104   /**
 105    * Stored in a DLL.
 106    */
 107   struct ActiveLookup *next;
 108
 109   /**
 110    * Stored in a DLL.
 111    */
 112   struct ActiveLookup *prev;
 113
 114   /**
 115    * The client that queried the records contained in this cache entry.
 116    */
 117   struct GNUNET_SERVICE_Client *client;
 118
 119   /**
 120    * handle for cancelling a request
 121    */
 122   struct GNUNET_DNSSTUB_RequestSocket *resolve_handle;
 123
 124   /**
 125    * handle for the resolution timeout task
 126    */
 127   struct GNUNET_SCHEDULER_Task *timeout_task;
 128
 129   /**
 130    * Which hostname are we resolving?
 131    */
 132   char *hostname;
 133
 134   /**
 135    * If @a record_type is #GNUNET_DNSPARSER_TYPE_ALL, did we go again
 136    * for the AAAA records yet?
 137    */
 138   int did_aaaa;
 139
 140   /**
 141    * type of queried DNS record
 142    */
 143   uint16_t record_type;
 144
 145   /**
 146    * Unique request ID of a client if a query for this hostname/record_type
 147    * is currently pending, undefined otherwise.
 148    */
 149   uint32_t client_request_id;
 150
 151   /**
 152    * Unique DNS request ID of a client if a query for this hostname/record_type
 153    * is currently pending, undefined otherwise.
 154    */
 155   uint16_t dns_id;
 156
 157 };
 158
 159
 160 /**
 161  * Start of the linked list of cached DNS lookup results.
 162  */
 163 static struct ResolveCache *cache_head;
 164
 165 /**
 166  * Tail of the linked list of cached DNS lookup results.
 167  */
 168 static struct ResolveCache *cache_tail;
 169
 170 /**
 171  * Head of the linked list of DNS lookup results from /etc/hosts.
 172  */
 173 static struct ResolveCache *hosts_head;
 174
 175 /**
 176  * Tail of the linked list of DNS lookup results from /etc/hosts.
 177  */
 178 static struct ResolveCache *hosts_tail;
 179
 180 /**
 181  * Start of the linked list of active DNS lookups.
 182  */
 183 static struct ActiveLookup *lookup_head;
 184
 185 /**
 186  * Tail of the linked list of active DNS lookups.
 187  */
 188 static struct ActiveLookup *lookup_tail;
 189
 190 /**
 191  * context of dnsstub library
 192  */
 193 static struct GNUNET_DNSSTUB_Context *dnsstub_ctx;
 194
 195 /**
 196  * My domain, to be appended to the hostname to get a FQDN.
 197  */
 198 static char *my_domain;
 199
 200 /**
 201  * How many entries do we have in #cache_head DLL?
 202  */
 203 static unsigned int cache_size;
 204
 205
 206 /**
 207  * Remove @a entry from cache.
 208  *
 209  * @param rc entry to free
 210  */
 211 static void
 212 free_cache_entry (struct ResolveCache *rc)
 213 {
 214   struct RecordListEntry *pos;
 215
 216   while (NULL != (pos = rc->records_head))
 217   {
 218     GNUNET_CONTAINER_DLL_remove (rc->records_head,
 219                                  rc->records_tail,
 220                                  pos);
 221     GNUNET_DNSPARSER_free_record (pos->record);
 222     GNUNET_free (pos->record);
 223     GNUNET_free (pos);
 224   }
 225   GNUNET_free_non_null (rc->hostname);
 226   GNUNET_CONTAINER_DLL_remove (cache_head,
 227                                cache_tail,
 228                                rc);
 229   cache_size--;
 230   GNUNET_free (rc);
 231 }
 232
 233
 234 /**
 235  * Remove @a entry from cache.
 236  *
 237  * @param rc entry to free
 238  */
 239 static void
 240 free_hosts_entry (struct ResolveCache *rc)
 241 {
 242   struct RecordListEntry *pos;
 243
 244   while (NULL != (pos = rc->records_head))
 245   {
 246     GNUNET_CONTAINER_DLL_remove (rc->records_head,
 247                                  rc->records_tail,
 248                                  pos);
 249     GNUNET_DNSPARSER_free_record (pos->record);
 250     GNUNET_free (pos->record);
 251     GNUNET_free (pos);
 252   }
 253   GNUNET_free_non_null (rc->hostname);
 254   GNUNET_CONTAINER_DLL_remove (hosts_head,
 255                                hosts_tail,
 256                                rc);
 257   cache_size--;
 258   GNUNET_free (rc);
 259 }
 260
 261
 262 /**
 263  * Release resources associated with @a al
 264  *
 265  * @param al an active lookup
 266  */
 267 static void
 268 free_active_lookup (struct ActiveLookup *al)
 269 {
 270   GNUNET_CONTAINER_DLL_remove (lookup_head,
 271                                lookup_tail,
 272                                al);
 273   if (NULL != al->resolve_handle)
 274   {
 275     GNUNET_DNSSTUB_resolve_cancel (al->resolve_handle);
 276     al->resolve_handle = NULL;
 277   }
 278   if (NULL != al->timeout_task)
 279   {
 280     GNUNET_SCHEDULER_cancel (al->timeout_task);
 281     al->timeout_task = NULL;
 282   }
 283   GNUNET_free_non_null (al->hostname);
 284   GNUNET_free (al);
 285 }
 286
 287
 288
 289 /**
 290  * Find out if the configuration file line contains a string
 291  * starting with "nameserver ", and if so, return a copy of
 292  * the nameserver's IP.
 293  *
 294  * @param line line to parse
 295  * @param line_len number of characters in @a line
 296  * @return NULL if no nameserver is configured in this @a line
 297  */
 298 static char *
 299 extract_dns_server (const char* line,
 300                     size_t line_len)
 301 {
 302   if (0 == strncmp (line,
 303                     "nameserver ",
 304                     strlen ("nameserver ")))
 305     return GNUNET_strndup (line + strlen ("nameserver "),
 306                            line_len - strlen ("nameserver "));
 307   return NULL;
 308 }
 309
 310
 311 /**
 312  * Find out if the configuration file line contains a string
 313  * starting with "search ", and if so, return a copy of
 314  * the machine's search domain.
 315  *
 316  * @param line line to parse
 317  * @param line_len number of characters in @a line
 318  * @return NULL if no nameserver is configured in this @a line
 319  */
 320 static char *
 321 extract_search_domain (const char* line,
 322                        size_t line_len)
 323 {
 324   if (0 == strncmp (line,
 325                     "search ",
 326                     strlen ("search ")))
 327     return GNUNET_strndup (line + strlen ("search "),
 328                            line_len - strlen ("search "));
 329   return NULL;
 330 }
 331
 332
 333 /**
 334  * Reads the list of nameservers from /etc/resolve.conf
 335  *
 336  * @param server_addrs[out] a list of null-terminated server address strings
 337  * @return the number of server addresses in @server_addrs, -1 on error
 338  */
 339 static int
 340 lookup_dns_servers (char ***server_addrs)
 341 {
 342   struct GNUNET_DISK_FileHandle *fh;
 343   struct GNUNET_DISK_MapHandle *mh;
 344   off_t bytes_read;
 345   const char *buf;
 346   size_t read_offset;
 347   unsigned int num_dns_servers;
 348
 349   fh = GNUNET_DISK_file_open ("/etc/resolv.conf",
 350                               GNUNET_DISK_OPEN_READ,
 351                               GNUNET_DISK_PERM_NONE);
 352   if (NULL == fh)
 353   {
 354     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 355                 "Could not open /etc/resolv.conf. "
 356                 "DNS resolution will not be possible.\n");
 357     return -1;
 358   }
 359   if (GNUNET_OK !=
 360       GNUNET_DISK_file_handle_size (fh,
 361                                     &bytes_read))
 362   {
 363     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 364                 "Could not determine size of /etc/resolv.conf. "
 365                 "DNS resolution will not be possible.\n");
 366     GNUNET_DISK_file_close (fh);
 367     return -1;
 368   }
 369   if ((size_t) bytes_read > SIZE_MAX)
 370   {
 371     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 372                 "/etc/resolv.conf file too large to mmap. "
 373                 "DNS resolution will not be possible.\n");
 374     GNUNET_DISK_file_close (fh);
 375     return -1;
 376   }
 377   buf = GNUNET_DISK_file_map (fh,
 378                               &mh,
 379                               GNUNET_DISK_MAP_TYPE_READ,
 380                               (size_t) bytes_read);
 381   *server_addrs = NULL;
 382   read_offset = 0;
 383   num_dns_servers = 0;
 384   while (read_offset < (size_t) bytes_read)
 385   {
 386     const char *newline;
 387     size_t line_len;
 388     char *dns_server;
 389
 390     newline = strchr (buf + read_offset,
 391                       '\n');
 392     if (NULL == newline)
 393       break;
 394     line_len = newline - buf - read_offset;
 395     dns_server = extract_dns_server (buf + read_offset,
 396                                      line_len);
 397     if (NULL != dns_server)
 398     {
 399       GNUNET_array_append (*server_addrs,
 400                            num_dns_servers,
 401                            dns_server);
 402     }
 403     else if (NULL == my_domain)
 404     {
 405       my_domain = extract_search_domain (buf + read_offset,
 406                                          line_len);
 407     }
 408     read_offset += line_len + 1;
 409   }
 410   GNUNET_DISK_file_unmap (mh);
 411   GNUNET_DISK_file_close (fh);
 412   return (int) num_dns_servers;
 413 }
 414
 415
 416 /**
 417  * Compute name to use for DNS reverse lookups from @a ip.
 418  *
 419  * @param ip IP address to resolve, in binary format, network byte order
 420  * @param af address family of @a ip, AF_INET or AF_INET6
 421  */
 422 static char *
 423 make_reverse_hostname (const void *ip,
 424                        int af)
 425 {
 426   char *buf = GNUNET_new_array (80,
 427                                 char);
 428   int pos = 0;
 429
 430   if (AF_INET == af)
 431   {
 432     struct in_addr *addr = (struct in_addr *)ip;
 433     uint32_t ip_int = addr->s_addr;
 434
 435     for (int i = 3; i >= 0; i--)
 436     {
 437       int n = GNUNET_snprintf (buf + pos,
 438                                80 - pos,
 439                                "%u.",
 440                                ((uint8_t *)&ip_int)[i]);
 441       if (n < 0)
 442       {
 443         GNUNET_free (buf);
 444         return NULL;
 445       }
 446       pos += n;
 447     }
 448     pos += GNUNET_snprintf (buf + pos,
 449                             80 - pos,
 450                             "in-addr.arpa");
 451   }
 452   else if (AF_INET6 == af)
 453   {
 454     struct in6_addr *addr = (struct in6_addr *)ip;
 455     for (int i = 15; i >= 0; i--)
 456     {
 457       int n = GNUNET_snprintf (buf + pos,
 458                                80 - pos,
 459                                "%x.",
 460                                addr->s6_addr[i] & 0xf);
 461       if (n < 0)
 462       {
 463         GNUNET_free (buf);
 464         return NULL;
 465       }
 466       pos += n;
 467       n = GNUNET_snprintf (buf + pos,
 468                            80 - pos,
 469                            "%x.",
 470                            addr->s6_addr[i] >> 4);
 471       if (n < 0)
 472       {
 473         GNUNET_free (buf);
 474         return NULL;
 475       }
 476       pos += n;
 477     }
 478     pos += GNUNET_snprintf (buf + pos,
 479                             80 - pos,
 480                             "ip6.arpa");
 481   }
 482   buf[pos] = '\0';
 483   return buf;
 484 }
 485
 486
 487 /**
 488  * Send DNS @a record back to our @a client.
 489  *
 490  * @param record information to transmit
 491  * @param record_type requested record type from client
 492  * @param client_request_id to which request are we responding
 493  * @param client where to send @a record
 494  * @return #GNUNET_YES if we sent a reply,
 495  *         #GNUNET_NO if the record type is not understood or
 496  *         does not match @a record_type
 497  */
 498 static int
 499 send_reply (struct GNUNET_DNSPARSER_Record *record,
 500             uint16_t record_type,
 501             uint32_t client_request_id,
 502             struct GNUNET_SERVICE_Client *client)
 503 {
 504   struct GNUNET_RESOLVER_ResponseMessage *msg;
 505   struct GNUNET_MQ_Envelope *env;
 506   const void *payload;
 507   size_t payload_len;
 508
 509   switch (record->type)
 510   {
 511   case GNUNET_DNSPARSER_TYPE_CNAME:
 512     if (GNUNET_DNSPARSER_TYPE_CNAME != record_type)
 513       return GNUNET_NO;
 514     payload = record->data.hostname;
 515     payload_len = strlen (record->data.hostname) + 1;
 516     break;
 517   case GNUNET_DNSPARSER_TYPE_PTR:
 518     if (GNUNET_DNSPARSER_TYPE_PTR != record_type)
 519       return GNUNET_NO;
 520     payload = record->data.hostname;
 521     payload_len = strlen (record->data.hostname) + 1;
 522     break;
 523   case GNUNET_DNSPARSER_TYPE_A:
 524     if ( (GNUNET_DNSPARSER_TYPE_A != record_type) &&
 525          (GNUNET_DNSPARSER_TYPE_ALL != record_type) )
 526       return GNUNET_NO;
 527     payload = record->data.raw.data;
 528     payload_len = record->data.raw.data_len;
 529     break;
 530   case GNUNET_DNSPARSER_TYPE_AAAA:
 531     if ( (GNUNET_DNSPARSER_TYPE_AAAA != record_type) &&
 532          (GNUNET_DNSPARSER_TYPE_ALL != record_type) )
 533       return GNUNET_NO;
 534     payload = record->data.raw.data;
 535     payload_len = record->data.raw.data_len;
 536     break;
 537   default:
 538     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 539                 "Cannot handle DNS response type %u: not supported here\n",
 540                 record->type);
 541     return GNUNET_NO;
 542   }
 543   env = GNUNET_MQ_msg_extra (msg,
 544                              payload_len,
 545                              GNUNET_MESSAGE_TYPE_RESOLVER_RESPONSE);
 546   msg->client_id = client_request_id;
 547   GNUNET_memcpy (&msg[1],
 548                  payload,
 549                  payload_len);
 550   GNUNET_MQ_send (GNUNET_SERVICE_client_get_mq (client),
 551                   env);
 552   return GNUNET_YES;
 553 }
 554
 555
 556 /**
 557  * Send message to @a client that we transmitted all
 558  * responses for @a client_request_id
 559  *
 560  * @param client_request_id to which request are we responding
 561  * @param client where to send @a record
 562  */
 563 static void
 564 send_end_msg (uint32_t client_request_id,
 565               struct GNUNET_SERVICE_Client *client)
 566 {
 567   struct GNUNET_RESOLVER_ResponseMessage *msg;
 568   struct GNUNET_MQ_Envelope *env;
 569
 570   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 571               "Sending END message\n");
 572   env = GNUNET_MQ_msg (msg,
 573                        GNUNET_MESSAGE_TYPE_RESOLVER_RESPONSE);
 574   msg->client_id = client_request_id;
 575   GNUNET_MQ_send (GNUNET_SERVICE_client_get_mq (client),
 576                   env);
 577 }
 578
 579
 580 /**
 581  * Remove expired entries from @a rc
 582  *
 583  * @param rc entry in resolver cache
 584  * @return #GNUNET_YES if @a rc was completely expired
 585  *         #GNUNET_NO if some entries are left
 586  */
 587 static int
 588 remove_expired (struct ResolveCache *rc)
 589 {
 590   struct GNUNET_TIME_Absolute now = GNUNET_TIME_absolute_get ();
 591   struct RecordListEntry *n;
 592
 593   for (struct RecordListEntry *pos = rc->records_head;
 594        NULL != pos;
 595        pos = n)
 596   {
 597     n = pos->next;
 598     if (now.abs_value_us > pos->record->expiration_time.abs_value_us)
 599     {
 600       GNUNET_CONTAINER_DLL_remove (rc->records_head,
 601                                    rc->records_tail,
 602                                    pos);
 603       GNUNET_DNSPARSER_free_record (pos->record);
 604       GNUNET_free (pos->record);
 605       GNUNET_free (pos);
 606     }
 607   }
 608   if (NULL == rc->records_head)
 609   {
 610     free_cache_entry (rc);
 611     return GNUNET_YES;
 612   }
 613   return GNUNET_NO;
 614 }
 615
 616
 617 /**
 618  * Process DNS request for @a hostname with request ID @a request_id
 619  * from @a client demanding records of type @a record_type.
 620  *
 621  * @param hostname DNS name to resolve
 622  * @param record_type desired record type
 623  * @param client_request_id client's request ID
 624  * @param client who should get the result?
 625  */
 626 static void
 627 process_get (const char *hostname,
 628              uint16_t record_type,
 629              uint32_t client_request_id,
 630              struct GNUNET_SERVICE_Client *client);
 631
 632
 633 /**
 634  * Get an IP address as a string (works for both IPv4 and IPv6).  Note
 635  * that the resolution happens asynchronously and that the first call
 636  * may not immediately result in the FQN (but instead in a
 637  * human-readable IP address).
 638  *
 639  * @param hostname what hostname was to be resolved
 640  * @param record_type what type of record was requested
 641  * @param client_request_id unique identification of the client's request
 642  * @param client handle to the client making the request (for sending the reply)
 643  */
 644 static int
 645 try_cache (const char *hostname,
 646            uint16_t record_type,
 647            uint32_t client_request_id,
 648            struct GNUNET_SERVICE_Client *client)
 649 {
 650   struct ResolveCache *pos;
 651   struct ResolveCache *next;
 652   int found;
 653   int in_hosts;
 654
 655   in_hosts = GNUNET_NO;
 656   for (pos = hosts_head; NULL != pos; pos = pos->next)
 657     if (0 == strcmp (pos->hostname,
 658                      hostname))
 659     {
 660       in_hosts = GNUNET_YES;
 661       break;
 662     }
 663   if (NULL == pos)
 664   {
 665     next = cache_head;
 666     for (pos = next; NULL != pos; pos = next)
 667     {
 668       next = pos->next;
 669       if (GNUNET_YES == remove_expired (pos))
 670         continue;
 671       if (0 == strcmp (pos->hostname,
 672                        hostname))
 673         break;
 674     }
 675   }
 676   if (NULL == pos)
 677   {
 678     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 679                 "No cache entry for '%s'\n",
 680                 hostname);
 681     return GNUNET_NO;
 682   }
 683   if ( (GNUNET_NO == in_hosts) &&
 684        (cache_head != pos) )
 685   {
 686     /* move result to head to achieve LRU for cache eviction */
 687     GNUNET_CONTAINER_DLL_remove (cache_head,
 688                                  cache_tail,
 689                                  pos);
 690     GNUNET_CONTAINER_DLL_insert (cache_head,
 691                                  cache_tail,
 692                                  pos);
 693   }
 694   found = GNUNET_NO;
 695   for (struct RecordListEntry *rle = pos->records_head;
 696        NULL != rle;
 697        rle = rle->next)
 698   {
 699     const struct GNUNET_DNSPARSER_Record *record = rle->record;
 700
 701     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 702                 "Found cache entry for '%s', record type '%u'\n",
 703                 hostname,
 704                 record_type);
 705     if ( (GNUNET_DNSPARSER_TYPE_CNAME == record->type) &&
 706          (GNUNET_DNSPARSER_TYPE_CNAME != record_type) &&
 707          (GNUNET_NO == found) )
 708     {
 709       const char *hostname = record->data.hostname;
 710
 711       process_get (hostname,
 712                    record_type,
 713                    client_request_id,
 714                    client);
 715       return GNUNET_YES; /* counts as a cache "hit" */
 716     }
 717     found |= send_reply (rle->record,
 718                          record_type,
 719                          client_request_id,
 720                          client);
 721   }
 722   if (GNUNET_NO == found)
 723     return GNUNET_NO; /* had records, but none matched! */
 724   send_end_msg (client_request_id,
 725                 client);
 726   return GNUNET_YES;
 727 }
 728
 729
 730 /**
 731  * Create DNS query for @a hostname of type @a type
 732  * with DNS request ID @a dns_id.
 733  *
 734  * @param hostname DNS name to query
 735  * @param type requested DNS record type
 736  * @param dns_id what should be the DNS request ID
 737  * @param packet_buf[out] where to write the request packet
 738  * @param packet_size[out] set to size of @a packet_buf on success
 739  * @return #GNUNET_OK on success
 740  */
 741 static int
 742 pack (const char *hostname,
 743       uint16_t type,
 744       uint16_t dns_id,
 745       char **packet_buf,
 746       size_t *packet_size)
 747 {
 748   struct GNUNET_DNSPARSER_Query query;
 749   struct GNUNET_DNSPARSER_Packet packet;
 750
 751   query.name = (char *)hostname;
 752   query.type = type;
 753   query.dns_traffic_class = GNUNET_TUN_DNS_CLASS_INTERNET;
 754   memset (&packet,
 755           0,
 756           sizeof (packet));
 757   packet.num_queries = 1;
 758   packet.queries = &query;
 759   packet.id = htons (dns_id);
 760   packet.flags.recursion_desired = 1;
 761   if (GNUNET_OK !=
 762       GNUNET_DNSPARSER_pack (&packet,
 763                              UINT16_MAX,
 764                              packet_buf,
 765                              packet_size))
 766   {
 767     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 768                 "Failed to pack query for hostname `%s'\n",
 769                 hostname);
 770     packet_buf = NULL;
 771     return GNUNET_SYSERR;
 772   }
 773   return GNUNET_OK;
 774 }
 775
 776 static void
 777 cache_answers(const char* name,
 778               struct GNUNET_DNSPARSER_Record *records,
 779               unsigned int num_records)
 780 {
 781   struct ResolveCache *rc;
 782   struct GNUNET_DNSPARSER_Record *record;
 783   struct RecordListEntry *rle;
 784
 785   for (unsigned int i = 0; i != num_records; i++)
 786   {
 787     record = &records[i];
 788
 789     for (rc = cache_head; NULL != rc; rc = rc->next)
 790       if (0 == strcasecmp (rc->hostname,
 791                            name))
 792         break;
 793     if (NULL == rc)
 794     {
 795       rc = GNUNET_new (struct ResolveCache);
 796       rc->hostname = GNUNET_strdup (name);
 797       GNUNET_CONTAINER_DLL_insert (cache_head,
 798                                    cache_tail,
 799                                    rc);
 800       cache_size++;
 801     }
 802     /* TODO: ought to check first if we have this exact record
 803        already in the cache! */
 804     rle = GNUNET_new (struct RecordListEntry);
 805     rle->record = GNUNET_DNSPARSER_duplicate_record (record);
 806     GNUNET_CONTAINER_DLL_insert (rc->records_head,
 807                                  rc->records_tail,
 808                                  rle);
 809   }
 810 }
 811
 812 /**
 813  * We got a result from DNS. Add it to the cache and
 814  * see if we can make our client happy...
 815  *
 816  * @param cls the `struct ActiveLookup`
 817  * @param dns the DNS response
 818  * @param dns_len number of bytes in @a dns
 819  */
 820 static void
 821 handle_resolve_result (void *cls,
 822                        const struct GNUNET_TUN_DnsHeader *dns,
 823                        size_t dns_len)
 824 {
 825   struct ActiveLookup *al = cls;
 826   struct GNUNET_DNSPARSER_Packet *parsed;
 827
 828   parsed = GNUNET_DNSPARSER_parse ((const char *)dns,
 829                                    dns_len);
 830   if (NULL == parsed)
 831   {
 832     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 833                 "Failed to parse DNS reply (hostname %s, request ID %u)\n",
 834                 al->hostname,
 835                 al->dns_id);
 836     return;
 837   }
 838   if (al->dns_id != ntohs (parsed->id))
 839   {
 840     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 841                 "Request ID in DNS reply does not match\n");
 842     GNUNET_DNSPARSER_free_packet (parsed);
 843     return;
 844   }
 845   if (0 == parsed->num_answers + parsed->num_authority_records + parsed->num_additional_records)
 846   {
 847     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 848                 "DNS reply (hostname %s, request ID %u) contains no answers\n",
 849                 al->hostname,
 850                 (unsigned int) al->client_request_id);
 851     /* resume by trying again from cache */
 852     if (GNUNET_NO ==
 853         try_cache (al->hostname,
 854                    al->record_type,
 855                    al->client_request_id,
 856                    al->client))
 857       /* cache failed, tell client we could not get an answer */
 858     {
 859       send_end_msg (al->client_request_id,
 860                     al->client);
 861     }
 862     GNUNET_DNSPARSER_free_packet (parsed);
 863     free_active_lookup (al);
 864     return;
 865   }
 866   /* LRU-based cache eviction: we remove from tail */
 867   while (cache_size > MAX_CACHE)
 868     free_cache_entry (cache_tail);
 869
 870   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 871               "Got reply for hostname %s and request ID %u\n",
 872               al->hostname,
 873               (unsigned int) al->client_request_id);
 874   /* add to cache */
 875   cache_answers(al->hostname,
 876                 parsed->answers, parsed->num_answers);
 877   cache_answers(al->hostname,
 878                 parsed->authority_records, parsed->num_authority_records);
 879   cache_answers(al->hostname,
 880                 parsed->additional_records, parsed->num_additional_records);
 881
 882   /* see if we need to do the 2nd request for AAAA records */
 883   if ( (GNUNET_DNSPARSER_TYPE_ALL == al->record_type) &&
 884        (GNUNET_NO == al->did_aaaa) )
 885   {
 886     char *packet_buf;
 887     size_t packet_size;
 888     uint16_t dns_id;
 889
 890     dns_id = (uint16_t) GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_NONCE,
 891                                                   UINT16_MAX);
 892     if (GNUNET_OK ==
 893         pack (al->hostname,
 894               GNUNET_DNSPARSER_TYPE_AAAA,
 895               dns_id,
 896               &packet_buf,
 897               &packet_size))
 898     {
 899       al->did_aaaa = GNUNET_YES;
 900       al->dns_id = dns_id;
 901       GNUNET_DNSSTUB_resolve_cancel (al->resolve_handle);
 902       al->resolve_handle =
 903         GNUNET_DNSSTUB_resolve (dnsstub_ctx,
 904                                 packet_buf,
 905                                 packet_size,
 906                                 &handle_resolve_result,
 907                                 al);
 908       GNUNET_free (packet_buf);
 909       GNUNET_DNSPARSER_free_packet (parsed);
 910       return;
 911     }
 912   }
 913
 914   /* resume by trying again from cache */
 915   if (GNUNET_NO ==
 916       try_cache (al->hostname,
 917                  al->record_type,
 918                  al->client_request_id,
 919                  al->client))
 920     /* cache failed, tell client we could not get an answer */
 921   {
 922     send_end_msg (al->client_request_id,
 923                   al->client);
 924   }
 925   free_active_lookup (al);
 926   GNUNET_DNSPARSER_free_packet (parsed);
 927 }
 928
 929
 930 /**
 931  * We encountered a timeout trying to perform a
 932  * DNS lookup.
 933  *
 934  * @param cls a `struct ActiveLookup`
 935  */
 936 static void
 937 handle_resolve_timeout (void *cls)
 938 {
 939   struct ActiveLookup *al = cls;
 940
 941   al->timeout_task = NULL;
 942   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 943               "DNS lookup timeout!\n");
 944   send_end_msg (al->client_request_id,
 945                 al->client);
 946   free_active_lookup (al);
 947 }
 948
 949
 950 /**
 951  * Initiate an active lookup, then cache the result and
 952  * try to then complete the resolution.
 953  *
 954  * @param hostname DNS name to resolve
 955  * @param record_type record type to locate
 956  * @param client_request_id client request ID
 957  * @param client handle to the client
 958  * @return #GNUNET_OK if the DNS query is now pending
 959  */
 960 static int
 961 resolve_and_cache (const char* hostname,
 962                    uint16_t record_type,
 963                    uint32_t client_request_id,
 964                    struct GNUNET_SERVICE_Client *client)
 965 {
 966   char *packet_buf;
 967   size_t packet_size;
 968   struct ActiveLookup *al;
 969   uint16_t dns_id;
 970   uint16_t type;
 971
 972   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
 973               "resolve_and_cache `%s'\n",
 974               hostname);
 975   dns_id = (uint16_t) GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_NONCE,
 976                                                 UINT16_MAX);
 977
 978   if (GNUNET_DNSPARSER_TYPE_ALL == record_type)
 979     type = GNUNET_DNSPARSER_TYPE_A;
 980   else
 981     type = record_type;
 982   if (GNUNET_OK !=
 983       pack (hostname,
 984             type,
 985             dns_id,
 986             &packet_buf,
 987             &packet_size))
 988   {
 989     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
 990                 "Failed to pack query for hostname `%s'\n",
 991                 hostname);
 992     return GNUNET_SYSERR;
 993   }
 994
 995   al = GNUNET_new (struct ActiveLookup);
 996   al->hostname = GNUNET_strdup (hostname);
 997   al->record_type = record_type;
 998   al->client_request_id = client_request_id;
 999   al->dns_id = dns_id;
1000   al->client = client;
1001   al->timeout_task = GNUNET_SCHEDULER_add_delayed (DNS_TIMEOUT,
1002                                                    &handle_resolve_timeout,
1003                                                    al);
1004   al->resolve_handle =
1005     GNUNET_DNSSTUB_resolve (dnsstub_ctx,
1006                             packet_buf,
1007                             packet_size,
1008                             &handle_resolve_result,
1009                             al);
1010   GNUNET_free (packet_buf);
1011   GNUNET_CONTAINER_DLL_insert (lookup_head,
1012                                lookup_tail,
1013                                al);
1014   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1015               "Resolving %s, client_request_id = %u, dns_id = %u\n",
1016               hostname,
1017               (unsigned int) client_request_id,
1018               (unsigned int) dns_id);
1019   return GNUNET_OK;
1020 }
1021
1022
1023 /**
1024  * Process DNS request for @a hostname with request ID @a client_request_id
1025  * from @a client demanding records of type @a record_type.
1026  *
1027  * @param hostname DNS name to resolve
1028  * @param record_type desired record type
1029  * @param client_request_id client's request ID
1030  * @param client who should get the result?
1031  */
1032 static void
1033 process_get (const char *hostname,
1034              uint16_t record_type,
1035              uint32_t client_request_id,
1036              struct GNUNET_SERVICE_Client *client)
1037 {
1038   char fqdn[255];
1039
1040   if (  (NULL != my_domain) &&
1041         (NULL == strchr (hostname,
1042                          (unsigned char) '.')) &&
1043         (strlen (hostname) + strlen (my_domain) <= 253) )
1044   {
1045     GNUNET_snprintf (fqdn,
1046                      sizeof (fqdn),
1047                      "%s.%s",
1048                      hostname,
1049                      my_domain);
1050   }
1051   else if (strlen (hostname) < 255)
1052   {
1053     GNUNET_snprintf (fqdn,
1054                      sizeof (fqdn),
1055                      "%s",
1056                      hostname);
1057   }
1058   else
1059   {
1060     GNUNET_break (0);
1061     GNUNET_SERVICE_client_drop (client);
1062     return;
1063   }
1064   if (GNUNET_NO ==
1065       try_cache (fqdn,
1066                  record_type,
1067                  client_request_id,
1068                  client))
1069   {
1070     if (GNUNET_OK !=
1071         resolve_and_cache (fqdn,
1072                            record_type,
1073                            client_request_id,
1074                            client))
1075     {
1076       send_end_msg (client_request_id,
1077                     client);
1078     }
1079   }
1080 }
1081
1082
1083 /**
1084  * Verify well-formedness of GET-message.
1085  *
1086  * @param cls closure, unused
1087  * @param get the actual message
1088  * @return #GNUNET_OK if @a get is well-formed
1089  */
1090 static int
1091 check_get (void *cls,
1092            const struct GNUNET_RESOLVER_GetMessage *get)
1093 {
1094   uint16_t size;
1095   int direction;
1096   int af;
1097
1098   (void) cls;
1099   size = ntohs (get->header.size) - sizeof (*get);
1100   direction = ntohl (get->direction);
1101   if (GNUNET_NO == direction)
1102   {
1103     GNUNET_MQ_check_zero_termination (get);
1104     return GNUNET_OK;
1105   }
1106   af = ntohl (get->af);
1107   switch (af)
1108   {
1109     case AF_INET:
1110       if (size != sizeof (struct in_addr))
1111       {
1112         GNUNET_break (0);
1113         return GNUNET_SYSERR;
1114       }
1115       break;
1116     case AF_INET6:
1117       if (size != sizeof (struct in6_addr))
1118       {
1119         GNUNET_break (0);
1120         return GNUNET_SYSERR;
1121       }
1122       break;
1123     default:
1124       GNUNET_break (0);
1125       return GNUNET_SYSERR;
1126   }
1127   return GNUNET_OK;
1128 }
1129
1130
1131 /**
1132  * Handle GET-message.
1133  *
1134  * @param cls identification of the client
1135  * @param msg the actual message
1136  */
1137 static void
1138 handle_get (void *cls,
1139             const struct GNUNET_RESOLVER_GetMessage *msg)
1140 {
1141   struct GNUNET_SERVICE_Client *client = cls;
1142   int direction;
1143   int af;
1144   uint32_t client_request_id;
1145   char *hostname;
1146
1147   direction = ntohl (msg->direction);
1148   af = ntohl (msg->af);
1149   client_request_id = msg->client_id;
1150   GNUNET_SERVICE_client_continue (client);
1151   if (GNUNET_NO == direction)
1152   {
1153     /* IP from hostname */
1154     hostname = GNUNET_strdup ((const char *) &msg[1]);
1155     switch (af)
1156     {
1157       case AF_UNSPEC:
1158         {
1159           process_get (hostname,
1160                        GNUNET_DNSPARSER_TYPE_ALL,
1161                        client_request_id,
1162                        client);
1163           break;
1164         }
1165       case AF_INET:
1166         {
1167           process_get (hostname,
1168                        GNUNET_DNSPARSER_TYPE_A,
1169                        client_request_id,
1170                        client);
1171           break;
1172         }
1173       case AF_INET6:
1174         {
1175           process_get (hostname,
1176                        GNUNET_DNSPARSER_TYPE_AAAA,
1177                        client_request_id,
1178                        client);
1179           break;
1180         }
1181       default:
1182         {
1183           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1184                       "got invalid af: %d\n",
1185                       af);
1186           GNUNET_assert (0);
1187         }
1188     }
1189   }
1190   else
1191   {
1192     /* hostname from IP */
1193     hostname = make_reverse_hostname (&msg[1],
1194                                       af);
1195     process_get (hostname,
1196                  GNUNET_DNSPARSER_TYPE_PTR,
1197                  client_request_id,
1198                  client);
1199   }
1200   GNUNET_free_non_null (hostname);
1201 }
1202
1203
1204 /**
1205  * Service is shutting down, clean up.
1206  *
1207  * @param cls NULL, unused
1208  */
1209 static void
1210 shutdown_task (void *cls)
1211 {
1212   (void) cls;
1213
1214   while (NULL != lookup_head)
1215     free_active_lookup (lookup_head);
1216   while (NULL != cache_head)
1217     free_cache_entry (cache_head);
1218   while (NULL != hosts_head)
1219     free_hosts_entry (hosts_head);
1220   GNUNET_DNSSTUB_stop (dnsstub_ctx);
1221   GNUNET_free_non_null (my_domain);
1222 }
1223
1224
1225 /**
1226  * Add information about a host from /etc/hosts
1227  * to our cache.
1228  *
1229  * @param hostname the name of the host
1230  * @param rec_type DNS record type to use
1231  * @param data payload
1232  * @param data_size number of bytes in @a data
1233  */
1234 static void
1235 add_host (const char *hostname,
1236           uint16_t rec_type,
1237           const void *data,
1238           size_t data_size)
1239 {
1240   struct ResolveCache *rc;
1241   struct RecordListEntry *rle;
1242   struct GNUNET_DNSPARSER_Record *rec;
1243
1244   rec = GNUNET_malloc (sizeof (struct GNUNET_DNSPARSER_Record));
1245   rec->expiration_time = GNUNET_TIME_UNIT_FOREVER_ABS;
1246   rec->type = rec_type;
1247   rec->dns_traffic_class = GNUNET_TUN_DNS_CLASS_INTERNET;
1248   rec->name = GNUNET_strdup (hostname);
1249   rec->data.raw.data = GNUNET_memdup (data,
1250                                       data_size);
1251   rec->data.raw.data_len = data_size;
1252   rle = GNUNET_new (struct RecordListEntry);
1253   rle->record = rec;
1254   rc = GNUNET_new (struct ResolveCache);
1255   rc->hostname = GNUNET_strdup (hostname);
1256   GNUNET_CONTAINER_DLL_insert (rc->records_head,
1257                                rc->records_tail,
1258                                rle);
1259   GNUNET_CONTAINER_DLL_insert (hosts_head,
1260                                hosts_tail,
1261                                rc);
1262 }
1263
1264
1265 /**
1266  * Extract host information from a line in /etc/hosts
1267  *
1268  * @param line the line to parse
1269  * @param line_len number of bytes in @a line
1270  */
1271 static void
1272 extract_hosts (const char *line,
1273                size_t line_len)
1274 {
1275   const char *c;
1276   struct in_addr v4;
1277   struct in6_addr v6;
1278   char *tbuf;
1279   char *tok;
1280
1281   /* ignore everything after '#' */
1282   c = memrchr (line,
1283                (unsigned char) '#',
1284                line_len);
1285   if (NULL != c)
1286     line_len = c - line;
1287   /* ignore leading whitespace */
1288   while ( (0 < line_len) &&
1289           isspace ((unsigned char) *line) )
1290   {
1291     line++;
1292     line_len--;
1293   }
1294   tbuf = GNUNET_strndup (line,
1295                          line_len);
1296   tok = strtok (tbuf, " \t");
1297   if (NULL == tok)
1298   {
1299     GNUNET_free (tbuf);
1300     return;
1301   }
1302   if (1 == inet_pton (AF_INET,
1303                       tok,
1304                       &v4))
1305   {
1306     while (NULL != (tok = strtok (NULL, " \t")))
1307       add_host (tok,
1308                 GNUNET_DNSPARSER_TYPE_A,
1309                 &v4,
1310                 sizeof (struct in_addr));
1311   }
1312   else if (1 == inet_pton (AF_INET6,
1313                            tok,
1314                            &v6))
1315   {
1316     while (NULL != (tok = strtok (NULL, " \t")))
1317       add_host (tok,
1318                 GNUNET_DNSPARSER_TYPE_AAAA,
1319                 &v6,
1320                 sizeof (struct in6_addr));
1321   }
1322   GNUNET_free (tbuf);
1323 }
1324
1325
1326 /**
1327  * Reads the list of hosts from /etc/hosts.
1328  */
1329 static void
1330 load_etc_hosts (void)
1331 {
1332   struct GNUNET_DISK_FileHandle *fh;
1333   struct GNUNET_DISK_MapHandle *mh;
1334   off_t bytes_read;
1335   const char *buf;
1336   size_t read_offset;
1337
1338   fh = GNUNET_DISK_file_open ("/etc/hosts",
1339                               GNUNET_DISK_OPEN_READ,
1340                               GNUNET_DISK_PERM_NONE);
1341   if (NULL == fh)
1342   {
1343     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1344                 "Failed to open /etc/hosts");
1345     return;
1346   }
1347   if (GNUNET_OK !=
1348       GNUNET_DISK_file_handle_size (fh,
1349                                     &bytes_read))
1350   {
1351     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1352                 "Could not determin size of /etc/hosts. "
1353                 "DNS resolution will not be possible.\n");
1354     GNUNET_DISK_file_close (fh);
1355     return;
1356   }
1357   if ((size_t) bytes_read > SIZE_MAX)
1358   {
1359     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1360                 "/etc/hosts file too large to mmap. "
1361                 "DNS resolution will not be possible.\n");
1362     GNUNET_DISK_file_close (fh);
1363     return;
1364   }
1365   buf = GNUNET_DISK_file_map (fh,
1366                               &mh,
1367                               GNUNET_DISK_MAP_TYPE_READ,
1368                               (size_t) bytes_read);
1369   read_offset = 0;
1370   while (read_offset < (size_t) bytes_read)
1371   {
1372     const char *newline;
1373     size_t line_len;
1374
1375     newline = strchr (buf + read_offset,
1376                       '\n');
1377     if (NULL == newline)
1378       break;
1379     line_len = newline - buf - read_offset;
1380     extract_hosts (buf + read_offset,
1381                    line_len);
1382     read_offset += line_len + 1;
1383   }
1384   GNUNET_DISK_file_unmap (mh);
1385   GNUNET_DISK_file_close (fh);
1386 }
1387
1388
1389 /**
1390  * Service is starting, initialize everything.
1391  *
1392  * @param cls NULL, unused
1393  * @param cfg our configuration
1394  * @param sh service handle
1395  */
1396 static void
1397 init_cb (void *cls,
1398          const struct GNUNET_CONFIGURATION_Handle *cfg,
1399          struct GNUNET_SERVICE_Handle *sh)
1400 {
1401   char **dns_servers;
1402   int num_dns_servers;
1403
1404   (void) cfg;
1405   (void) sh;
1406   load_etc_hosts ();
1407   GNUNET_SCHEDULER_add_shutdown (&shutdown_task,
1408                                  cls);
1409   dnsstub_ctx = GNUNET_DNSSTUB_start (128);
1410   dns_servers = NULL;
1411   num_dns_servers = lookup_dns_servers (&dns_servers);
1412   if (0 >= num_dns_servers)
1413   {
1414     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1415                 _("No DNS server available. DNS resolution will not be possible.\n"));
1416     return;
1417   }
1418   for (int i = 0; i < num_dns_servers; i++)
1419   {
1420     int result = GNUNET_DNSSTUB_add_dns_ip (dnsstub_ctx, dns_servers[i]);
1421     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1422                 "Adding DNS server '%s': %s\n",
1423                 dns_servers[i],
1424                 GNUNET_OK == result ? "success" : "failure");
1425     GNUNET_free (dns_servers[i]);
1426   }
1427   GNUNET_free_non_null (dns_servers);
1428 }
1429
1430
1431 /**
1432  * Callback called when a client connects to the service.
1433  *
1434  * @param cls closure for the service, unused
1435  * @param c the new client that connected to the service
1436  * @param mq the message queue used to send messages to the client
1437  * @return @a c
1438  */
1439 static void *
1440 connect_cb (void *cls,
1441             struct GNUNET_SERVICE_Client *c,
1442             struct GNUNET_MQ_Handle *mq)
1443 {
1444   (void) cls;
1445   (void) mq;
1446
1447   return c;
1448 }
1449
1450
1451 /**
1452  * Callback called when a client disconnected from the service
1453  *
1454  * @param cls closure for the service
1455  * @param c the client that disconnected
1456  * @param internal_cls should be equal to @a c
1457  */
1458 static void
1459 disconnect_cb (void *cls,
1460                struct GNUNET_SERVICE_Client *c,
1461                void *internal_cls)
1462 {
1463   struct ActiveLookup *n;
1464   (void) cls;
1465
1466   GNUNET_assert (c == internal_cls);
1467   n = lookup_head;
1468   for (struct ActiveLookup *al = n;
1469        NULL != al;
1470        al = n)
1471   {
1472     n = al->next;
1473     if (al->client == c)
1474       free_active_lookup (al);
1475   }
1476 }
1477
1478
1479 /**
1480  * Define "main" method using service macro.
1481  */
1482 GNUNET_SERVICE_MAIN
1483 ("resolver",
1484  GNUNET_SERVICE_OPTION_NONE,
1485  &init_cb,
1486  &connect_cb,
1487  &disconnect_cb,
1488  NULL,
1489  GNUNET_MQ_hd_var_size (get,
1490                         GNUNET_MESSAGE_TYPE_RESOLVER_REQUEST,
1491                         struct GNUNET_RESOLVER_GetMessage,
1492                         NULL),
1493  GNUNET_MQ_handler_end ());
1494
1495
1496 #if defined(LINUX) && defined(__GLIBC__)
1497 #include <malloc.h>
1498
1499 /**
1500  * MINIMIZE heap size (way below 128k) since this process doesn't need much.
1501  */
1502 void __attribute__ ((constructor))
1503 GNUNET_RESOLVER_memory_init ()
1504 {
1505   mallopt (M_TRIM_THRESHOLD, 4 * 1024);
1506   mallopt (M_TOP_PAD, 1 * 1024);
1507   malloc_trim (0);
1508 }
1509 #endif
1510
1511
1512 /* end of gnunet-service-resolver.c */