2a9d86a8bb728150598422293a86ce34c3dd8064
[oweals/gnunet.git] / src / gns / gnunet-service-gns.c
1 /*
2      This file is part of GNUnet.
3      Copyright (C) 2011-2013 GNUnet e.V.
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., 51 Franklin Street, Fifth Floor,
18      Boston, MA 02110-1301, USA.
19 */
20 /**
21  * @file gns/gnunet-service-gns.c
22  * @brief GNU Name System (main service)
23  * @author Martin Schanzenbach
24  * @author Christian Grothoff
25  */
26 #include "platform.h"
27 #include "gnunet_util_lib.h"
28 #include "gnunet_dns_service.h"
29 #include "gnunet_dnsparser_lib.h"
30 #include "gnunet_dht_service.h"
31 #include "gnunet_namecache_service.h"
32 #include "gnunet_namestore_service.h"
33 #include "gnunet_identity_service.h"
34 #include "gnunet_gns_service.h"
35 #include "gnunet_statistics_service.h"
36 #include "gns.h"
37 #include "gnunet-service-gns_resolver.h"
38 #include "gnunet-service-gns_shorten.h"
39 #include "gnunet-service-gns_interceptor.h"
40 #include "gnunet_protocols.h"
41
42 /**
43  * The initial interval in milliseconds btween puts in
44  * a zone iteration
45  */
46 #define INITIAL_PUT_INTERVAL GNUNET_TIME_UNIT_MILLISECONDS
47
48 /**
49  * The upper bound for the zone iteration interval in milliseconds
50  */
51 #define MINIMUM_ZONE_ITERATION_INTERVAL GNUNET_TIME_UNIT_SECONDS
52
53 /**
54  * The default put interval for the zone iteration. In case
55  * no option is found
56  */
57 #define DEFAULT_ZONE_PUBLISH_TIME_WINDOW GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_HOURS, 4)
58
59 /**
60  * The factor the current zone iteration interval is divided by for each
61  * additional new record
62  */
63 #define LATE_ITERATION_SPEEDUP_FACTOR 2
64
65 /**
66  * How long until a DHT PUT attempt should time out?
67  */
68 #define DHT_OPERATION_TIMEOUT GNUNET_TIME_relative_multiply (GNUNET_TIME_UNIT_SECONDS, 60)
69
70 /**
71  * What replication level do we use for DHT PUT operations?
72  */
73 #define DHT_GNS_REPLICATION_LEVEL 5
74
75
76 /**
77  * Handle to a lookup operation from api
78  */
79 struct ClientLookupHandle
80 {
81
82   /**
83    * We keep these in a DLL.
84    */
85   struct ClientLookupHandle *next;
86
87   /**
88    * We keep these in a DLL.
89    */
90   struct ClientLookupHandle *prev;
91
92   /**
93    * Handle to the requesting client
94    */
95   struct GNUNET_SERVER_Client *client;
96
97   /**
98    * Active handle for the lookup.
99    */
100   struct GNS_ResolverHandle *lookup;
101
102   /**
103    * request id
104    */
105   uint32_t request_id;
106
107 };
108
109
110 /**
111  * Handle for DHT PUT activity triggered from the namestore monitor.
112  */
113 struct MonitorActivity
114 {
115   /**
116    * Kept in a DLL.
117    */
118   struct MonitorActivity *next;
119
120   /**
121    * Kept in a DLL.
122    */
123   struct MonitorActivity *prev;
124
125   /**
126    * Handle for the DHT PUT operation.
127    */
128   struct GNUNET_DHT_PutHandle *ph;
129 };
130
131
132 /**
133  * Our handle to the DHT
134  */
135 static struct GNUNET_DHT_Handle *dht_handle;
136
137 /**
138  * Active DHT put operation (or NULL)
139  */
140 static struct GNUNET_DHT_PutHandle *active_put;
141
142 /**
143  * Our handle to the namestore service
144  */
145 static struct GNUNET_NAMESTORE_Handle *namestore_handle;
146
147 /**
148  * Our handle to the namecache service
149  */
150 static struct GNUNET_NAMECACHE_Handle *namecache_handle;
151
152 /**
153  * Our handle to the identity service
154  */
155 static struct GNUNET_IDENTITY_Handle *identity_handle;
156
157 /**
158  * Our handle to the identity operation to find the master zone
159  * for intercepted queries.
160  */
161 static struct GNUNET_IDENTITY_Operation *identity_op;
162
163 /**
164  * Handle to iterate over our authoritative zone in namestore
165  */
166 static struct GNUNET_NAMESTORE_ZoneIterator *namestore_iter;
167
168 /**
169  * Handle to monitor namestore changes to instant propagation.
170  */
171 static struct GNUNET_NAMESTORE_ZoneMonitor *zmon;
172
173 /**
174  * Our notification context.
175  */
176 static struct GNUNET_SERVER_NotificationContext *nc;
177
178 /**
179  * Head of the DLL.
180  */
181 static struct ClientLookupHandle *clh_head;
182
183 /**
184  * Tail of the DLL.
185  */
186 static struct ClientLookupHandle *clh_tail;
187
188 /**
189  * Head of monitor activities; kept in a DLL.
190  */
191 static struct MonitorActivity *ma_head;
192
193 /**
194  * Tail of monitor activities; kept in a DLL.
195  */
196 static struct MonitorActivity *ma_tail;
197
198 /**
199  * Useful for zone update for DHT put
200  */
201 static unsigned long long num_public_records;
202
203 /**
204  * Last seen record count
205  */
206 static unsigned long long last_num_public_records;
207
208 /**
209  * Minimum relative expiration time of records seem during the current
210  * zone iteration.
211  */
212 static struct GNUNET_TIME_Relative min_relative_record_time;
213
214 /**
215  * Zone iteration PUT interval.
216  */
217 static struct GNUNET_TIME_Relative put_interval;
218
219 /**
220  * Default time window for zone iteration
221  */
222 static struct GNUNET_TIME_Relative zone_publish_time_window_default;
223
224 /**
225  * Time window for zone iteration, adjusted based on relative record
226  * expiration times in our zone.
227  */
228 static struct GNUNET_TIME_Relative zone_publish_time_window;
229
230 /**
231  * zone publish task
232  */
233 static struct GNUNET_SCHEDULER_Task *zone_publish_task;
234
235 /**
236  * #GNUNET_YES if zone has never been published before
237  */
238 static int first_zone_iteration;
239
240 /**
241  * #GNUNET_YES if ipv6 is supported
242  */
243 static int v6_enabled;
244
245 /**
246  * #GNUNET_YES if ipv4 is supported
247  */
248 static int v4_enabled;
249
250 /**
251  * Handle to the statistics service
252  */
253 static struct GNUNET_STATISTICS_Handle *statistics;
254
255
256 /**
257  * Task run during shutdown.
258  *
259  * @param cls unused
260  * @param tc unused
261  */
262 static void
263 shutdown_task (void *cls)
264 {
265   struct ClientLookupHandle *clh;
266   struct MonitorActivity *ma;
267
268   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
269               "Shutting down!\n");
270   if (NULL != nc)
271   {
272     GNUNET_SERVER_notification_context_destroy (nc);
273     nc = NULL;
274   }
275   while (NULL != (clh = clh_head))
276   {
277     GNUNET_SERVER_client_set_user_context (clh->client,
278                                            NULL);
279     GNS_resolver_lookup_cancel (clh->lookup);
280     GNUNET_CONTAINER_DLL_remove (clh_head,
281                                  clh_tail,
282                                  clh);
283     GNUNET_free (clh);
284   }
285
286   GNS_interceptor_done ();
287   if (NULL != identity_op)
288   {
289     GNUNET_IDENTITY_cancel (identity_op);
290     identity_op = NULL;
291   }
292   if (NULL != identity_handle)
293   {
294     GNUNET_IDENTITY_disconnect (identity_handle);
295     identity_handle = NULL;
296   }
297   GNS_resolver_done ();
298   GNS_shorten_done ();
299   while (NULL != (ma = ma_head))
300   {
301     GNUNET_DHT_put_cancel (ma->ph);
302     GNUNET_CONTAINER_DLL_remove (ma_head,
303                                  ma_tail,
304                                  ma);
305     GNUNET_free (ma);
306   }
307   if (NULL != statistics)
308   {
309     GNUNET_STATISTICS_destroy (statistics,
310                                GNUNET_NO);
311     statistics = NULL;
312   }
313   if (NULL != zone_publish_task)
314   {
315     GNUNET_SCHEDULER_cancel (zone_publish_task);
316     zone_publish_task = NULL;
317   }
318   if (NULL != namestore_iter)
319   {
320     GNUNET_NAMESTORE_zone_iteration_stop (namestore_iter);
321     namestore_iter = NULL;
322   }
323   if (NULL != zmon)
324   {
325     GNUNET_NAMESTORE_zone_monitor_stop (zmon);
326     zmon = NULL;
327   }
328   if (NULL != namestore_handle)
329   {
330     GNUNET_NAMESTORE_disconnect (namestore_handle);
331     namestore_handle = NULL;
332   }
333   if (NULL != namecache_handle)
334   {
335     GNUNET_NAMECACHE_disconnect (namecache_handle);
336     namecache_handle = NULL;
337   }
338   if (NULL != active_put)
339   {
340     GNUNET_DHT_put_cancel (active_put);
341     active_put = NULL;
342   }
343   if (NULL != dht_handle)
344   {
345     GNUNET_DHT_disconnect (dht_handle);
346     dht_handle = NULL;
347   }
348 }
349
350
351 /**
352  * Method called periodically that triggers iteration over authoritative records
353  *
354  * @param cls closure
355  */
356 static void
357 publish_zone_dht_next (void *cls)
358 {
359   zone_publish_task = NULL;
360   GNUNET_assert (NULL != namestore_iter);
361   GNUNET_NAMESTORE_zone_iterator_next (namestore_iter);
362 }
363
364
365 /**
366  * Periodically iterate over our zone and store everything in dht
367  *
368  * @param cls NULL
369  */
370 static void
371 publish_zone_dht_start (void *cls);
372
373
374 /**
375  * Continuation called from DHT once the PUT operation is done.
376  *
377  * @param cls closure, NULL if called from regular iteration,
378  *        `struct MonitorActivity` if called from #handle_monitor_event.
379  * @param success #GNUNET_OK on success
380  */
381 static void
382 dht_put_continuation (void *cls,
383                       int success)
384 {
385   struct MonitorActivity *ma = cls;
386   struct GNUNET_TIME_Relative next_put_interval;
387
388   num_public_records++;
389   if (NULL == ma)
390   {
391     active_put = NULL;
392     if ( (num_public_records > last_num_public_records) &&
393          (GNUNET_NO == first_zone_iteration) )
394     {
395       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
396                   "Last record count was lower than current record count.  Reducing interval.\n");
397       put_interval = GNUNET_TIME_relative_divide (zone_publish_time_window,
398                                                   num_public_records);
399       next_put_interval = GNUNET_TIME_relative_divide (put_interval,
400                                                        LATE_ITERATION_SPEEDUP_FACTOR);
401     }
402     else
403       next_put_interval = put_interval;
404
405     GNUNET_STATISTICS_set (statistics,
406                            "Current zone iteration interval (ms)",
407                            next_put_interval.rel_value_us / 1000LL,
408                            GNUNET_NO);
409     GNUNET_assert (NULL == zone_publish_task);
410     zone_publish_task = GNUNET_SCHEDULER_add_delayed (next_put_interval,
411                                                       &publish_zone_dht_next,
412                                                       NULL);
413   }
414   else
415   {
416     GNUNET_CONTAINER_DLL_remove (ma_head,
417                                  ma_tail,
418                                  ma);
419     GNUNET_free (ma);
420   }
421 }
422
423
424 /**
425  * Convert namestore records from the internal format to that
426  * suitable for publication (removes private records, converts
427  * to absolute expiration time).
428  *
429  * @param rd input records
430  * @param rd_count size of the @a rd and @a rd_public arrays
431  * @param rd_public where to write the converted records
432  * @return number of records written to @a rd_public
433  */
434 static unsigned int
435 convert_records_for_export (const struct GNUNET_GNSRECORD_Data *rd,
436                             unsigned int rd_count,
437                             struct GNUNET_GNSRECORD_Data *rd_public)
438 {
439   struct GNUNET_TIME_Absolute now;
440   unsigned int rd_public_count;
441   unsigned int i;
442
443   rd_public_count = 0;
444   now = GNUNET_TIME_absolute_get ();
445   for (i=0;i<rd_count;i++)
446     if (0 == (rd[i].flags & GNUNET_GNSRECORD_RF_PRIVATE))
447     {
448       rd_public[rd_public_count] = rd[i];
449       if (0 != (rd[i].flags & GNUNET_GNSRECORD_RF_RELATIVE_EXPIRATION))
450       {
451         /* GNUNET_GNSRECORD_block_create will convert to absolute time;
452            we just need to adjust our iteration frequency */
453         min_relative_record_time.rel_value_us =
454           GNUNET_MIN (rd_public[rd_public_count].expiration_time,
455                       min_relative_record_time.rel_value_us);
456       }
457       else if (rd_public[rd_public_count].expiration_time < now.abs_value_us)
458       {
459         /* record already expired, skip it */
460         continue;
461       }
462       rd_public_count++;
463     }
464   return rd_public_count;
465 }
466
467
468 /**
469  * Store GNS records in the DHT.
470  *
471  * @param key key of the zone
472  * @param label label to store under
473  * @param rd_public public record data
474  * @param rd_public_count number of records in @a rd_public
475  * @param pc_arg closure argument to pass to the #dht_put_continuation
476  * @return DHT PUT handle, NULL on error
477  */
478 static struct GNUNET_DHT_PutHandle *
479 perform_dht_put (const struct GNUNET_CRYPTO_EcdsaPrivateKey *key,
480                  const char *label,
481                  const struct GNUNET_GNSRECORD_Data *rd_public,
482                  unsigned int rd_public_count,
483                  void *pc_arg)
484 {
485   struct GNUNET_GNSRECORD_Block *block;
486   struct GNUNET_HashCode query;
487   struct GNUNET_TIME_Absolute expire;
488   size_t block_size;
489   struct GNUNET_DHT_PutHandle *ret;
490
491   expire = GNUNET_GNSRECORD_record_get_expiration_time (rd_public_count,
492                                                         rd_public);
493   block = GNUNET_GNSRECORD_block_create (key,
494                                          expire,
495                                          label,
496                                          rd_public,
497                                          rd_public_count);
498   if (NULL == block)
499     return NULL; /* whoops */
500   block_size = ntohl (block->purpose.size)
501     + sizeof (struct GNUNET_CRYPTO_EcdsaSignature)
502     + sizeof (struct GNUNET_CRYPTO_EcdsaPublicKey);
503   GNUNET_GNSRECORD_query_from_private_key (key,
504                                            label,
505                                            &query);
506   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
507               "Storing %u record(s) for label `%s' in DHT with expiration `%s' under key %s\n",
508               rd_public_count,
509               label,
510               GNUNET_STRINGS_absolute_time_to_string (expire),
511               GNUNET_h2s (&query));
512   ret = GNUNET_DHT_put (dht_handle, &query,
513                         DHT_GNS_REPLICATION_LEVEL,
514                         GNUNET_DHT_RO_DEMULTIPLEX_EVERYWHERE,
515                         GNUNET_BLOCK_TYPE_GNS_NAMERECORD,
516                         block_size,
517                         block,
518                         expire,
519                         &dht_put_continuation,
520                         pc_arg);
521   GNUNET_free (block);
522   return ret;
523 }
524
525
526 /**
527  * Function used to put all records successively into the DHT.
528  *
529  * @param cls the closure (NULL)
530  * @param key the private key of the authority (ours)
531  * @param label the name of the records, NULL once the iteration is done
532  * @param rd_count the number of records in @a rd
533  * @param rd the record data
534  */
535 static void
536 put_gns_record (void *cls,
537                 const struct GNUNET_CRYPTO_EcdsaPrivateKey *key,
538                 const char *label,
539                 unsigned int rd_count,
540                 const struct GNUNET_GNSRECORD_Data *rd)
541 {
542   struct GNUNET_GNSRECORD_Data rd_public[rd_count];
543   unsigned int rd_public_count;
544
545   if ( (NULL == key) &&
546        (NULL == label) &&
547        (0 == rd_count) )
548   {
549     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
550                 "Got disconnected from namestore database, retrying.\n");
551     namestore_iter = NULL;
552     if (NULL != zone_publish_task)
553     {
554       GNUNET_SCHEDULER_cancel (zone_publish_task);
555       zone_publish_task = NULL;
556     }
557     zone_publish_task = GNUNET_SCHEDULER_add_now (&publish_zone_dht_start,
558                                                   NULL);
559     return;
560   }
561   if (NULL == label)
562   {
563     /* we're done with one iteration, calculate when to do the next one */
564     namestore_iter = NULL;
565     last_num_public_records = num_public_records;
566     first_zone_iteration = GNUNET_NO;
567     if (0 == num_public_records)
568     {
569       /**
570        * If no records are known (startup) or none present
571        * we can safely set the interval to the value for a single
572        * record
573        */
574       put_interval = zone_publish_time_window;
575       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK,
576                   "No records in namestore database.\n");
577     }
578     else
579     {
580       /* If records are present, next publication is based on the minimum
581        * relative expiration time of the records published divided by 4
582        */
583       zone_publish_time_window = GNUNET_TIME_relative_min (
584           GNUNET_TIME_relative_divide (min_relative_record_time, 4),
585           zone_publish_time_window_default);
586       put_interval = GNUNET_TIME_relative_divide (zone_publish_time_window,
587                                                   num_public_records);
588     }
589     /* reset for next iteration */
590     min_relative_record_time = GNUNET_TIME_UNIT_FOREVER_REL;
591     put_interval = GNUNET_TIME_relative_max (MINIMUM_ZONE_ITERATION_INTERVAL,
592                                              put_interval);
593
594     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
595                 "Zone iteration finished. Adjusted zone iteration interval to %s\n",
596                 GNUNET_STRINGS_relative_time_to_string (put_interval,
597                                                         GNUNET_YES));
598     GNUNET_STATISTICS_set (statistics,
599                            "Current zone iteration interval (in ms)",
600                            put_interval.rel_value_us / 1000LL,
601                            GNUNET_NO);
602     GNUNET_STATISTICS_update (statistics,
603                               "Number of zone iterations",
604                               1,
605                               GNUNET_NO);
606     GNUNET_STATISTICS_set (statistics,
607                            "Number of public records in DHT",
608                            last_num_public_records,
609                            GNUNET_NO);
610     GNUNET_assert (NULL == zone_publish_task);
611     if (0 == num_public_records)
612       zone_publish_task = GNUNET_SCHEDULER_add_delayed (put_interval,
613                                                         &publish_zone_dht_start,
614                                                         NULL);
615     else
616       zone_publish_task = GNUNET_SCHEDULER_add_now (&publish_zone_dht_start,
617                                                     NULL);
618     return;
619   }
620
621   rd_public_count = convert_records_for_export (rd,
622                                                 rd_count,
623                                                 rd_public);
624
625   /* We got a set of records to publish */
626   if (0 == rd_public_count)
627   {
628     GNUNET_assert (NULL == zone_publish_task);
629     zone_publish_task = GNUNET_SCHEDULER_add_now (&publish_zone_dht_next,
630                                                    NULL);
631     return;
632   }
633
634   active_put = perform_dht_put (key,
635                                 label,
636                                 rd_public,
637                                 rd_public_count,
638                                 NULL);
639   if (NULL == active_put)
640   {
641     GNUNET_break (0);
642     dht_put_continuation (NULL, GNUNET_NO);
643   }
644 }
645
646
647 /**
648  * Periodically iterate over all zones and store everything in DHT
649  *
650  * @param cls NULL
651  */
652 static void
653 publish_zone_dht_start (void *cls)
654 {
655   zone_publish_task = NULL;
656
657   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
658               "Starting DHT zone update!\n");
659   /* start counting again */
660   num_public_records = 0;
661   GNUNET_assert (NULL == namestore_iter);
662   namestore_iter
663     = GNUNET_NAMESTORE_zone_iteration_start (namestore_handle,
664                                              NULL, /* All zones */
665                                              &put_gns_record,
666                                              NULL);
667 }
668
669
670 /**
671  * Process a record that was stored in the namestore
672  * (invoked by the monitor).
673  *
674  * @param cls closure, NULL
675  * @param zone private key of the zone; NULL on disconnect
676  * @param label label of the records; NULL on disconnect
677  * @param rd_count number of entries in @a rd array, 0 if label was deleted
678  * @param rd array of records with data to store
679  */
680 static void
681 handle_monitor_event (void *cls,
682                       const struct GNUNET_CRYPTO_EcdsaPrivateKey *zone,
683                       const char *label,
684                       unsigned int rd_count,
685                       const struct GNUNET_GNSRECORD_Data *rd)
686 {
687   struct GNUNET_GNSRECORD_Data rd_public[rd_count];
688   unsigned int rd_public_count;
689   struct MonitorActivity *ma;
690
691   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
692               "Received %u records for label `%s' via namestore monitor\n",
693               rd_count,
694               label);
695   /* filter out records that are not public, and convert to
696      absolute expiration time. */
697   rd_public_count = convert_records_for_export (rd, rd_count,
698                                                 rd_public);
699   if (0 == rd_public_count)
700     return; /* nothing to do */
701   ma = GNUNET_new (struct MonitorActivity);
702   ma->ph = perform_dht_put (zone, label,
703                             rd, rd_count,
704                             ma);
705   if (NULL == ma->ph)
706   {
707     /* PUT failed, do not remember operation */
708     GNUNET_free (ma);
709     return;
710   }
711   GNUNET_CONTAINER_DLL_insert (ma_head,
712                                ma_tail,
713                                ma);
714 }
715
716
717 /* END DHT ZONE PROPAGATION */
718
719
720 /**
721  * Reply to client with the result from our lookup.
722  *
723  * @param cls the closure (our client lookup handle)
724  * @param rd_count the number of records in @a rd
725  * @param rd the record data
726  */
727 static void
728 send_lookup_response (void* cls,
729                       uint32_t rd_count,
730                       const struct GNUNET_GNSRECORD_Data *rd)
731 {
732   struct ClientLookupHandle *clh = cls;
733   struct GNUNET_GNS_ClientLookupResultMessage *rmsg;
734   size_t len;
735
736   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
737               "Sending LOOKUP_RESULT message with %u results\n",
738               (unsigned int) rd_count);
739
740   len = GNUNET_GNSRECORD_records_get_size (rd_count, rd);
741   rmsg = GNUNET_malloc (len + sizeof (struct GNUNET_GNS_ClientLookupResultMessage));
742   rmsg->header.type = htons (GNUNET_MESSAGE_TYPE_GNS_LOOKUP_RESULT);
743   rmsg->header.size = htons (len + sizeof(struct GNUNET_GNS_ClientLookupResultMessage));
744   rmsg->id = clh->request_id;
745   rmsg->rd_count = htonl (rd_count);
746   GNUNET_GNSRECORD_records_serialize (rd_count, rd, len,
747                                       (char*) &rmsg[1]);
748   GNUNET_SERVER_notification_context_unicast (nc,
749                                               clh->client,
750                                               &rmsg->header,
751                                               GNUNET_NO);
752   GNUNET_free (rmsg);
753   GNUNET_CONTAINER_DLL_remove (clh_head, clh_tail, clh);
754   GNUNET_SERVER_client_set_user_context (clh->client, NULL);
755   GNUNET_free (clh);
756   GNUNET_STATISTICS_update (statistics,
757                             "Completed lookups", 1,
758                             GNUNET_NO);
759   GNUNET_STATISTICS_update (statistics,
760                             "Records resolved",
761                             rd_count,
762                             GNUNET_NO);
763 }
764
765
766 /**
767  * Handle lookup requests from client
768  *
769  * @param cls the closure
770  * @param client the client
771  * @param message the message
772  */
773 static void
774 handle_lookup (void *cls,
775                struct GNUNET_SERVER_Client *client,
776                const struct GNUNET_MessageHeader *message)
777 {
778   char name[GNUNET_DNSPARSER_MAX_NAME_LENGTH + 1];
779   struct ClientLookupHandle *clh;
780   char *nameptr = name;
781   const char *utf_in;
782   const struct GNUNET_CRYPTO_EcdsaPrivateKey *key;
783   uint16_t msg_size;
784   const struct GNUNET_GNS_ClientLookupMessage *sh_msg;
785
786   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
787               "Received LOOKUP message\n");
788   msg_size = ntohs (message->size);
789   if (msg_size < sizeof (struct GNUNET_GNS_ClientLookupMessage))
790   {
791     GNUNET_break (0);
792     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
793     return;
794   }
795   sh_msg = (const struct GNUNET_GNS_ClientLookupMessage *) message;
796   GNUNET_SERVER_notification_context_add (nc, client);
797   if (GNUNET_YES == ntohs (sh_msg->have_key))
798     key = &sh_msg->shorten_key;
799   else
800     key = NULL;
801   utf_in = (const char *) &sh_msg[1];
802   if ( ('\0' != utf_in[msg_size - sizeof (struct GNUNET_GNS_ClientLookupMessage) - 1]) ||
803        (strlen (utf_in) > GNUNET_DNSPARSER_MAX_NAME_LENGTH) )
804   {
805     GNUNET_break (0);
806     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
807     return;
808   }
809   GNUNET_STRINGS_utf8_tolower (utf_in, nameptr);
810   GNUNET_SERVER_receive_done (client, GNUNET_OK);
811
812   clh = GNUNET_new (struct ClientLookupHandle);
813   GNUNET_SERVER_client_set_user_context (client, clh);
814   GNUNET_CONTAINER_DLL_insert (clh_head, clh_tail, clh);
815   clh->client = client;
816   clh->request_id = sh_msg->id;
817   if ( (GNUNET_DNSPARSER_TYPE_A == ntohl (sh_msg->type)) &&
818        (GNUNET_OK != v4_enabled) )
819   {
820     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
821                 "LOOKUP: Query for A record but AF_INET not supported!");
822     send_lookup_response (clh, 0, NULL);
823     return;
824   }
825   if ( (GNUNET_DNSPARSER_TYPE_AAAA == ntohl (sh_msg->type)) &&
826        (GNUNET_OK != v6_enabled) )
827   {
828     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
829                 "LOOKUP: Query for AAAA record but AF_INET6 not supported!");
830     send_lookup_response (clh, 0, NULL);
831     return;
832   }
833   clh->lookup = GNS_resolver_lookup (&sh_msg->zone,
834                                      ntohl (sh_msg->type),
835                                      name,
836                                      key,
837                                      (enum GNUNET_GNS_LocalOptions) ntohs (sh_msg->options),
838                                      &send_lookup_response, clh);
839   GNUNET_STATISTICS_update (statistics,
840                             "Lookup attempts",
841                             1, GNUNET_NO);
842 }
843
844
845 /**
846  * One of our clients disconnected, clean up after it.
847  *
848  * @param cls NULL
849  * @param client the client that disconnected
850  */
851 static void
852 notify_client_disconnect (void *cls,
853                           struct GNUNET_SERVER_Client *client)
854 {
855   struct ClientLookupHandle *clh;
856
857   if (NULL == client)
858     return;
859   clh = GNUNET_SERVER_client_get_user_context (client, struct ClientLookupHandle);
860   if (NULL == clh)
861     return;
862   GNS_resolver_lookup_cancel (clh->lookup);
863   GNUNET_CONTAINER_DLL_remove (clh_head, clh_tail, clh);
864   GNUNET_free (clh);
865 }
866
867
868 /**
869  * The zone monitor is now in SYNC with the current state of the
870  * name store.  Start to perform periodic iterations.
871  *
872  * @param cls NULL
873  */
874 static void
875 monitor_sync_event (void *cls)
876 {
877   GNUNET_assert (NULL == zone_publish_task);
878   zone_publish_task = GNUNET_SCHEDULER_add_now (&publish_zone_dht_start,
879                                                 NULL);
880 }
881
882
883 /**
884  * Method called to inform about the ego to be used for the master zone
885  * for DNS interceptions.
886  *
887  * This function is only called ONCE, and 'NULL' being passed in
888  * @a ego does indicate that interception is not configured.
889  * If @a ego is non-NULL, we should start to intercept DNS queries
890  * and resolve ".gnu" queries using the given ego as the master zone.
891  *
892  * @param cls closure, our `const struct GNUNET_CONFIGURATION_Handle *c`
893  * @param ego ego handle
894  * @param ctx context for application to store data for this ego
895  *                 (during the lifetime of this process, initially NULL)
896  * @param name name assigned by the user for this ego,
897  *                   NULL if the user just deleted the ego and it
898  *                   must thus no longer be used
899  */
900 static void
901 identity_intercept_cb (void *cls,
902                        struct GNUNET_IDENTITY_Ego *ego,
903                        void **ctx,
904                        const char *name)
905 {
906   const struct GNUNET_CONFIGURATION_Handle *cfg = cls;
907   struct GNUNET_CRYPTO_EcdsaPublicKey dns_root;
908
909   identity_op = NULL;
910   if (NULL == ego)
911   {
912     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
913                 _("No ego configured for `%s`\n"),
914                 "gns-intercept");
915     return;
916   }
917   GNUNET_IDENTITY_ego_get_public_key (ego,
918                                       &dns_root);
919   if (GNUNET_SYSERR ==
920       GNS_interceptor_init (&dns_root, cfg))
921   {
922     GNUNET_break (0);
923     GNUNET_SCHEDULER_add_now (&shutdown_task, NULL);
924     return;
925   }
926 }
927
928
929 /**
930  * Process GNS requests.
931  *
932  * @param cls closure
933  * @param server the initialized server
934  * @param c configuration to use
935  */
936 static void
937 run (void *cls,
938      struct GNUNET_SERVER_Handle *server,
939      const struct GNUNET_CONFIGURATION_Handle *c)
940 {
941   static const struct GNUNET_SERVER_MessageHandler handlers[] = {
942     { &handle_lookup, NULL, GNUNET_MESSAGE_TYPE_GNS_LOOKUP, 0},
943     {NULL, NULL, 0, 0}
944   };
945   unsigned long long max_parallel_bg_queries = 0;
946
947   v6_enabled = GNUNET_NETWORK_test_pf (PF_INET6);
948   v4_enabled = GNUNET_NETWORK_test_pf (PF_INET);
949   min_relative_record_time = GNUNET_TIME_UNIT_FOREVER_REL;
950   namestore_handle = GNUNET_NAMESTORE_connect (c);
951   if (NULL == namestore_handle)
952   {
953     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
954                 _("Failed to connect to the namestore!\n"));
955     GNUNET_SCHEDULER_shutdown ();
956     return;
957   }
958   namecache_handle = GNUNET_NAMECACHE_connect (c);
959   if (NULL == namecache_handle)
960   {
961     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
962                 _("Failed to connect to the namecache!\n"));
963     GNUNET_SCHEDULER_shutdown ();
964     return;
965   }
966
967   put_interval = INITIAL_PUT_INTERVAL;
968   zone_publish_time_window_default = DEFAULT_ZONE_PUBLISH_TIME_WINDOW;
969   if (GNUNET_OK ==
970       GNUNET_CONFIGURATION_get_value_time (c, "gns",
971                                            "ZONE_PUBLISH_TIME_WINDOW",
972                                            &zone_publish_time_window_default))
973   {
974     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
975                 "Time window for zone iteration: %s\n",
976                 GNUNET_STRINGS_relative_time_to_string (zone_publish_time_window,
977                                                         GNUNET_YES));
978   }
979   zone_publish_time_window = zone_publish_time_window_default;
980   if (GNUNET_OK ==
981       GNUNET_CONFIGURATION_get_value_number (c, "gns",
982                                             "MAX_PARALLEL_BACKGROUND_QUERIES",
983                                             &max_parallel_bg_queries))
984   {
985     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
986                 "Number of allowed parallel background queries: %llu\n",
987                 max_parallel_bg_queries);
988   }
989
990   dht_handle = GNUNET_DHT_connect (c,
991                                    (unsigned int) max_parallel_bg_queries);
992   if (NULL == dht_handle)
993   {
994     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
995                 _("Could not connect to DHT!\n"));
996     GNUNET_SCHEDULER_add_now (&shutdown_task, NULL);
997     return;
998   }
999
1000   identity_handle = GNUNET_IDENTITY_connect (c,
1001                                              NULL,
1002                                              NULL);
1003   if (NULL == identity_handle)
1004   {
1005     GNUNET_log (GNUNET_ERROR_TYPE_INFO,
1006                 "Could not connect to identity service!\n");
1007   }
1008   else
1009   {
1010     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1011                 "Looking for gns-intercept ego\n");
1012     identity_op = GNUNET_IDENTITY_get (identity_handle,
1013                                        "gns-intercept",
1014                                        &identity_intercept_cb,
1015                                        (void *) c);
1016   }
1017   GNS_resolver_init (namecache_handle,
1018                      dht_handle,
1019                      c,
1020                      max_parallel_bg_queries);
1021   GNS_shorten_init (namestore_handle,
1022                     namecache_handle,
1023                     dht_handle);
1024   GNUNET_SERVER_disconnect_notify (server,
1025                                    &notify_client_disconnect,
1026                                    NULL);
1027   /* Schedule periodic put for our records. */
1028   first_zone_iteration = GNUNET_YES;
1029   GNUNET_SERVER_add_handlers (server, handlers);
1030   statistics = GNUNET_STATISTICS_create ("gns", c);
1031   nc = GNUNET_SERVER_notification_context_create (server, 1);
1032   zmon = GNUNET_NAMESTORE_zone_monitor_start (c,
1033                                               NULL,
1034                                               GNUNET_NO,
1035                                               &handle_monitor_event,
1036                                               &monitor_sync_event,
1037                                               NULL);
1038   GNUNET_break (NULL != zmon);
1039   GNUNET_SCHEDULER_add_shutdown (&shutdown_task, NULL);
1040 }
1041
1042
1043 /**
1044  * The main function for the GNS service.
1045  *
1046  * @param argc number of arguments from the command line
1047  * @param argv command line arguments
1048  * @return 0 ok, 1 on error
1049  */
1050 int
1051 main (int argc, char *const *argv)
1052 {
1053   int ret;
1054
1055   ret =
1056       (GNUNET_OK ==
1057        GNUNET_SERVICE_run (argc, argv, "gns", GNUNET_SERVICE_OPTION_NONE, &run,
1058                            NULL)) ? 0 : 1;
1059   return ret;
1060 }
1061
1062 /* end of gnunet-service-gns.c */