More fixes for #3522
[oweals/gnunet.git] / src / namestore / gnunet-service-namestore.c
1 /*
2      This file is part of GNUnet.
3      (C) 2012, 2013, 2014 Christian Grothoff (and other contributing authors)
4
5      GNUnet is free software; you can redistribute it and/or modify
6      it under the terms of the GNU General Public License as published
7      by the Free Software Foundation; either version 3, or (at your
8      option) any later version.
9
10      GNUnet is distributed in the hope that it will be useful, but
11      WITHOUT ANY WARRANTY; without even the implied warranty of
12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13      General Public License for more details.
14
15      You should have received a copy of the GNU General Public License
16      along with GNUnet; see the file COPYING.  If not, write to the
17      Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18      Boston, MA 02111-1307, USA.
19 */
20
21 /**
22  * @file namestore/gnunet-service-namestore.c
23  * @brief namestore for the GNUnet naming system
24  * @author Matthias Wachs
25  * @author Christian Grothoff
26  */
27 #include "platform.h"
28 #include "gnunet_util_lib.h"
29 #include "gnunet_dnsparser_lib.h"
30 #include "gnunet_gns_service.h"
31 #include "gnunet_namecache_service.h"
32 #include "gnunet_namestore_service.h"
33 #include "gnunet_namestore_plugin.h"
34 #include "gnunet_signatures.h"
35 #include "namestore.h"
36
37 #define LOG_STRERROR_FILE(kind,syscall,filename) GNUNET_log_from_strerror_file (kind, "util", syscall, filename)
38
39
40 /**
41  * A namestore client
42  */
43 struct NamestoreClient;
44
45
46 /**
47  * A namestore iteration operation.
48  */
49 struct ZoneIteration
50 {
51   /**
52    * Next element in the DLL
53    */
54   struct ZoneIteration *next;
55
56   /**
57    * Previous element in the DLL
58    */
59   struct ZoneIteration *prev;
60
61   /**
62    * Namestore client which intiated this zone iteration
63    */
64   struct NamestoreClient *client;
65
66   /**
67    * The nick to add to the records
68    */
69   struct GNUNET_GNSRECORD_Data *nick;
70
71   /**
72    * Key of the zone we are iterating over.
73    */
74   struct GNUNET_CRYPTO_EcdsaPrivateKey zone;
75
76   /**
77    * The operation id fot the zone iteration in the response for the client
78    */
79   uint32_t request_id;
80
81   /**
82    * Offset of the zone iteration used to address next result of the zone
83    * iteration in the store
84    *
85    * Initialy set to 0 in handle_iteration_start
86    * Incremented with by every call to handle_iteration_next
87    */
88   uint32_t offset;
89
90 };
91
92
93 /**
94  * A namestore client
95  */
96 struct NamestoreClient
97 {
98   /**
99    * Next element in the DLL
100    */
101   struct NamestoreClient *next;
102
103   /**
104    * Previous element in the DLL
105    */
106   struct NamestoreClient *prev;
107
108   /**
109    * The client
110    */
111   struct GNUNET_SERVER_Client *client;
112
113   /**
114    * Head of the DLL of
115    * Zone iteration operations in progress initiated by this client
116    */
117   struct ZoneIteration *op_head;
118
119   /**
120    * Tail of the DLL of
121    * Zone iteration operations in progress initiated by this client
122    */
123   struct ZoneIteration *op_tail;
124 };
125
126
127 /**
128  * A namestore monitor.
129  */
130 struct ZoneMonitor
131 {
132   /**
133    * Next element in the DLL
134    */
135   struct ZoneMonitor *next;
136
137   /**
138    * Previous element in the DLL
139    */
140   struct ZoneMonitor *prev;
141
142   /**
143    * Namestore client which intiated this zone monitor
144    */
145   struct NamestoreClient *nc;
146
147   /**
148    * Private key of the zone.
149    */
150   struct GNUNET_CRYPTO_EcdsaPrivateKey zone;
151
152   /**
153    * Task active during initial iteration.
154    */
155   GNUNET_SCHEDULER_TaskIdentifier task;
156
157   /**
158    * Offset of the zone iteration used to address next result of the zone
159    * iteration in the store
160    *
161    * Initialy set to 0.
162    * Incremented with by every call to #handle_iteration_next
163    */
164   uint32_t offset;
165
166 };
167
168
169 /**
170  * Pending operation on the namecache.
171  */
172 struct CacheOperation
173 {
174
175   /**
176    * Kept in a DLL.
177    */
178   struct CacheOperation *prev;
179
180   /**
181    * Kept in a DLL.
182    */
183   struct CacheOperation *next;
184
185   /**
186    * Handle to namecache queue.
187    */
188   struct GNUNET_NAMECACHE_QueueEntry *qe;
189
190   /**
191    * Client to notify about the result.
192    */
193   struct GNUNET_SERVER_Client *client;
194
195   /**
196    * Client's request ID.
197    */
198   uint32_t rid;
199 };
200
201
202 /**
203  * Public key of all zeros.
204  */
205 static const struct GNUNET_CRYPTO_EcdsaPrivateKey zero;
206
207 /**
208  * Configuration handle.
209  */
210 static const struct GNUNET_CONFIGURATION_Handle *GSN_cfg;
211
212 /**
213  * Namecache handle.
214  */
215 static struct GNUNET_NAMECACHE_Handle *namecache;
216
217 /**
218  * Database handle
219  */
220 static struct GNUNET_NAMESTORE_PluginFunctions *GSN_database;
221
222 /**
223  * Name of the database plugin
224  */
225 static char *db_lib_name;
226
227 /**
228  * Our notification context.
229  */
230 static struct GNUNET_SERVER_NotificationContext *snc;
231
232 /**
233  * Head of the Client DLL
234  */
235 static struct NamestoreClient *client_head;
236
237 /**
238  * Tail of the Client DLL
239  */
240 static struct NamestoreClient *client_tail;
241
242 /**
243  * Head of cop DLL.
244  */
245 static struct CacheOperation *cop_head;
246
247 /**
248  * Tail of cop DLL.
249  */
250 static struct CacheOperation *cop_tail;
251
252 /**
253  * First active zone monitor.
254  */
255 static struct ZoneMonitor *monitor_head;
256
257 /**
258  * Last active zone monitor.
259  */
260 static struct ZoneMonitor *monitor_tail;
261
262 /**
263  * Notification context shared by all monitors.
264  */
265 static struct GNUNET_SERVER_NotificationContext *monitor_nc;
266
267
268
269 /**
270  * Task run during shutdown.
271  *
272  * @param cls unused
273  * @param tc unused
274  */
275 static void
276 cleanup_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
277 {
278   struct ZoneIteration *no;
279   struct NamestoreClient *nc;
280   struct CacheOperation *cop;
281
282   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
283               "Stopping namestore service\n");
284   if (NULL != snc)
285   {
286     GNUNET_SERVER_notification_context_destroy (snc);
287     snc = NULL;
288   }
289   while (NULL != (cop = cop_head))
290   {
291     GNUNET_NAMECACHE_cancel (cop->qe);
292     GNUNET_CONTAINER_DLL_remove (cop_head,
293                                  cop_tail,
294                                  cop);
295     GNUNET_free (cop);
296   }
297   GNUNET_NAMECACHE_disconnect (namecache);
298   namecache = NULL;
299   while (NULL != (nc = client_head))
300   {
301     while (NULL != (no = nc->op_head))
302     {
303       GNUNET_CONTAINER_DLL_remove (nc->op_head, nc->op_tail, no);
304       GNUNET_free (no);
305     }
306     GNUNET_CONTAINER_DLL_remove (client_head, client_tail, nc);
307     GNUNET_SERVER_client_set_user_context (nc->client, (void *)NULL);
308     GNUNET_free (nc);
309   }
310   GNUNET_break (NULL == GNUNET_PLUGIN_unload (db_lib_name, GSN_database));
311   GNUNET_free (db_lib_name);
312   db_lib_name = NULL;
313   if (NULL != monitor_nc)
314   {
315     GNUNET_SERVER_notification_context_destroy (monitor_nc);
316     monitor_nc = NULL;
317   }
318 }
319
320
321 /**
322  * Called whenever a client is disconnected.
323  * Frees our resources associated with that client.
324  *
325  * @param cls closure
326  * @param client identification of the client
327  */
328 static void
329 client_disconnect_notification (void *cls,
330                                 struct GNUNET_SERVER_Client *client)
331 {
332   struct ZoneIteration *no;
333   struct NamestoreClient *nc;
334   struct ZoneMonitor *zm;
335   struct CacheOperation *cop;
336
337   if (NULL == client)
338     return;
339   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
340               "Client %p disconnected\n",
341               client);
342   if (NULL == (nc = GNUNET_SERVER_client_get_user_context (client, struct NamestoreClient)))
343     return;
344   while (NULL != (no = nc->op_head))
345   {
346     GNUNET_CONTAINER_DLL_remove (nc->op_head, nc->op_tail, no);
347     GNUNET_free (no);
348   }
349   GNUNET_CONTAINER_DLL_remove (client_head, client_tail, nc);
350   GNUNET_free (nc);
351   for (zm = monitor_head; NULL != zm; zm = zm->next)
352   {
353     if (client == zm->nc->client)
354     {
355       GNUNET_CONTAINER_DLL_remove (monitor_head,
356                                    monitor_tail,
357                                    zm);
358       if (GNUNET_SCHEDULER_NO_TASK != zm->task)
359       {
360         GNUNET_SCHEDULER_cancel (zm->task);
361         zm->task = GNUNET_SCHEDULER_NO_TASK;
362       }
363       GNUNET_free (zm);
364       break;
365     }
366   }
367   for (cop = cop_head; NULL != cop; cop = cop->next)
368     if (client == cop->client)
369       cop->client = NULL;
370 }
371
372
373 /**
374  * Add a client to our list of active clients, if it is not yet
375  * in there.
376  *
377  * @param client client to add
378  * @return internal namestore client structure for this client
379  */
380 static struct NamestoreClient *
381 client_lookup (struct GNUNET_SERVER_Client *client)
382 {
383   struct NamestoreClient *nc;
384
385   nc = GNUNET_SERVER_client_get_user_context (client, struct NamestoreClient);
386   if (NULL != nc)
387     return nc;
388   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
389               "Client %p connected\n",
390               client);
391   nc = GNUNET_new (struct NamestoreClient);
392   nc->client = client;
393   GNUNET_SERVER_notification_context_add (snc, client);
394   GNUNET_CONTAINER_DLL_insert (client_head, client_tail, nc);
395   GNUNET_SERVER_client_set_user_context (client, nc);
396   return nc;
397 }
398
399
400 /**
401  * Function called with the records for the #GNUNET_GNS_MASTERZONE_STR
402  * label in the zone.  Used to locate the #GNUNET_GNSRECORD_TYPE_NICK
403  * record, which (if found) is then copied to @a cls for future use.
404  *
405  * @param cls a `struct GNUNET_GNSRECORD_Data **` for storing the nick (if found)
406  * @param private_key the private key of the zone (unused)
407  * @param label should be #GNUNET_GNS_MASTERZONE_STR
408  * @param rd_count number of records in @a rd
409  * @param rd records stored under @a label in the zone
410  */
411 static void
412 lookup_nick_it (void *cls,
413                 const struct GNUNET_CRYPTO_EcdsaPrivateKey *private_key,
414                 const char *label,
415                 unsigned int rd_count,
416                 const struct GNUNET_GNSRECORD_Data *rd)
417 {
418   struct GNUNET_GNSRECORD_Data **res = cls;
419   int c;
420
421   if (0 != strcmp (label, GNUNET_GNS_MASTERZONE_STR))
422   {
423     GNUNET_break (0);
424     return;
425   }
426   for (c = 0; c < rd_count; c++)
427   {
428     if (GNUNET_GNSRECORD_TYPE_NICK == rd[c].record_type)
429     {
430       (*res) = GNUNET_malloc (rd[c].data_size + sizeof (struct GNUNET_GNSRECORD_Data));
431       (*res)->data = &(*res)[1];
432       memcpy ((char *)(*res)->data, rd[c].data, rd[c].data_size);
433       (*res)->data_size = rd[c].data_size;
434       (*res)->expiration_time = rd[c].expiration_time;
435       (*res)->flags = rd[c].flags;
436       (*res)->record_type = GNUNET_GNSRECORD_TYPE_NICK;
437       return;
438     }
439   }
440   (*res) = NULL;
441 }
442
443
444 /**
445  * Return the NICK record for the zone (if it exists).
446  *
447  * @param zone private key for the zone to look for nick
448  * @return NULL if no NICK record was found
449  */
450 static struct GNUNET_GNSRECORD_Data *
451 get_nick_record (const struct GNUNET_CRYPTO_EcdsaPrivateKey *zone)
452 {
453   struct GNUNET_CRYPTO_EcdsaPublicKey pub;
454   struct GNUNET_GNSRECORD_Data *nick;
455   int res;
456
457   res = GSN_database->lookup_records (GSN_database->cls, zone,
458                                       GNUNET_GNS_MASTERZONE_STR,
459                                       &lookup_nick_it, &nick);
460   if ((NULL == nick) || (GNUNET_OK != res))
461   {
462     GNUNET_CRYPTO_ecdsa_key_get_public (zone, &pub);
463     GNUNET_log (GNUNET_ERROR_TYPE_INFO | GNUNET_ERROR_TYPE_BULK,
464                 "No nick name set for zone `%s'\n",
465                 GNUNET_GNSRECORD_z2s (&pub));
466     return NULL;
467   }
468   return nick;
469 }
470
471
472 static void
473 merge_with_nick_records ( const struct GNUNET_GNSRECORD_Data *nick_rd,
474                           unsigned int rdc2,
475                           const struct GNUNET_GNSRECORD_Data *rd2,
476                           unsigned int *rdc_res,
477                           struct GNUNET_GNSRECORD_Data **rd_res)
478 {
479   uint64_t latest_expiration;
480   int c;
481   size_t req;
482   char *data;
483   int record_offset;
484   size_t data_offset;
485   (*rdc_res) = 1 + rdc2;
486
487   if (0 == 1 + rdc2)
488   {
489     (*rd_res) = NULL;
490     return;
491   }
492
493   req = 0;
494   for (c=0; c< 1; c++)
495     req += sizeof (struct GNUNET_GNSRECORD_Data) + nick_rd[c].data_size;
496   for (c=0; c< rdc2; c++)
497     req += sizeof (struct GNUNET_GNSRECORD_Data) + rd2[c].data_size;
498   (*rd_res) = GNUNET_malloc (req);
499   data = (char *) &(*rd_res)[1 + rdc2];
500   data_offset = 0;
501   latest_expiration = 0;
502
503   for (c=0; c< rdc2; c++)
504   {
505     if (0 != (rd2[c].flags & GNUNET_GNSRECORD_RF_RELATIVE_EXPIRATION))
506     {
507       if ((GNUNET_TIME_absolute_get().abs_value_us + rd2[c].expiration_time) >
508         latest_expiration)
509           latest_expiration = rd2[c].expiration_time;
510     }
511     else if (rd2[c].expiration_time > latest_expiration)
512       latest_expiration = rd2[c].expiration_time;
513     (*rd_res)[c] = rd2[c];
514     (*rd_res)[c].data = (void *) &data[data_offset];
515     // WTF?
516     memcpy ((void *) (*rd_res)[c].data, rd2[c].data, rd2[c].data_size);
517     data_offset += (*rd_res)[c].data_size;
518   }
519   record_offset = rdc2;
520   for (c=0; c< 1; c++)
521   {
522     (*rd_res)[c+record_offset] = nick_rd[c];
523     (*rd_res)[c+record_offset].expiration_time = latest_expiration;
524     (*rd_res)[c+record_offset].data = (void *) &data[data_offset];
525     // WTF?
526     memcpy ((void *) (*rd_res)[c+record_offset].data,
527             nick_rd[c].data,
528             nick_rd[c].data_size);
529     data_offset += (*rd_res)[c+record_offset].data_size;
530   }
531   GNUNET_assert (req == (sizeof (struct GNUNET_GNSRECORD_Data)) * (*rdc_res) + data_offset);
532 }
533
534
535 /**
536  * Generate a `struct LookupNameResponseMessage` and send it to the
537  * given client using the given notification context.
538  *
539  * @param nc notification context to use
540  * @param client client to unicast to
541  * @param request_id request ID to use
542  * @param zone_key zone key of the zone
543  * @param name name
544  * @param rd_count number of records in @a rd
545  * @param rd array of records
546  */
547 static void
548 send_lookup_response (struct GNUNET_SERVER_NotificationContext *nc,
549                       struct GNUNET_SERVER_Client *client,
550                       uint32_t request_id,
551                       const struct GNUNET_CRYPTO_EcdsaPrivateKey *zone_key,
552                       const char *name,
553                       unsigned int rd_count,
554                       const struct GNUNET_GNSRECORD_Data *rd)
555 {
556   struct RecordResultMessage *zir_msg;
557   struct GNUNET_GNSRECORD_Data *nick;
558   struct GNUNET_GNSRECORD_Data *res;
559   unsigned int res_count;
560   size_t name_len;
561   size_t rd_ser_len;
562   size_t msg_size;
563   char *name_tmp;
564   char *rd_ser;
565
566   nick = get_nick_record (zone_key);
567   if ((NULL != nick) && (0 != strcmp(name, GNUNET_GNS_MASTERZONE_STR)))
568   {
569     nick->flags = (nick->flags | GNUNET_GNSRECORD_RF_PRIVATE) ^ GNUNET_GNSRECORD_RF_PRIVATE;
570     merge_with_nick_records (nick, rd_count, rd, &res_count, &res);
571     //fprintf (stderr, "Merging %u records for `%s' with NICK records\n",rd_count, name);
572     GNUNET_free (nick);
573   }
574   else
575   {
576     res_count = rd_count;
577     res = (struct GNUNET_GNSRECORD_Data *) rd;
578     //fprintf (stderr, "Not merging %u records for `%s' with NICK records\n",rd_count, name);
579   }
580
581   name_len = strlen (name) + 1;
582   rd_ser_len = GNUNET_GNSRECORD_records_get_size (res_count, res);
583   msg_size = sizeof (struct RecordResultMessage) + name_len + rd_ser_len;
584   (void) client_lookup (client);
585   zir_msg = GNUNET_malloc (msg_size);
586   zir_msg->gns_header.header.type = htons (GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_RESULT);
587   zir_msg->gns_header.header.size = htons (msg_size);
588   zir_msg->gns_header.r_id = htonl (request_id);
589   zir_msg->name_len = htons (name_len);
590   zir_msg->rd_count = htons (res_count);
591   zir_msg->rd_len = htons (rd_ser_len);
592   zir_msg->private_key = *zone_key;
593   name_tmp = (char *) &zir_msg[1];
594   memcpy (name_tmp, name, name_len);
595   rd_ser = &name_tmp[name_len];
596   GNUNET_GNSRECORD_records_serialize (res_count, res, rd_ser_len, rd_ser);
597   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
598               "Sending `%s' message with %u records and size %u\n",
599               "RECORD_RESULT",
600               res_count,
601               msg_size);
602   GNUNET_SERVER_notification_context_unicast (nc,
603                                               client,
604                                               &zir_msg->gns_header.header,
605                                               GNUNET_NO);
606   if (rd != res)
607     GNUNET_free (res);
608   GNUNET_free (zir_msg);
609 }
610
611
612 /**
613  * Send response to the store request to the client.
614  *
615  * @param client client to talk to
616  * @param res status of the operation
617  * @param rid client's request ID
618  */
619 static void
620 send_store_response (struct GNUNET_SERVER_Client *client,
621                      int res,
622                      uint32_t rid)
623 {
624   struct RecordStoreResponseMessage rcr_msg;
625
626   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
627               "Sending `%s' message\n",
628               "RECORD_STORE_RESPONSE");
629   rcr_msg.gns_header.header.type = htons (GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_STORE_RESPONSE);
630   rcr_msg.gns_header.header.size = htons (sizeof (struct RecordStoreResponseMessage));
631   rcr_msg.gns_header.r_id = htonl (rid);
632   rcr_msg.op_result = htonl (res);
633   GNUNET_SERVER_notification_context_unicast (snc, client,
634                                               &rcr_msg.gns_header.header,
635                                               GNUNET_NO);
636 }
637
638
639 /**
640  * Cache operation complete, clean up.
641  *
642  * @param cls the `struct CacheOperation`
643  * @param success success
644  * @param emsg error messages
645  */
646 static void
647 finish_cache_operation (void *cls,
648                         int32_t success,
649                         const char *emsg)
650 {
651   struct CacheOperation *cop = cls;
652
653   if (NULL != emsg)
654     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
655                 _("Failed to replicate block in namecache: %s\n"),
656                 emsg);
657   else
658     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
659                 "CACHE operation completed\n");
660   GNUNET_CONTAINER_DLL_remove (cop_head,
661                                cop_tail,
662                                cop);
663   if (NULL != cop->client)
664     send_store_response (cop->client,
665                          success,
666                          cop->rid);
667   GNUNET_free (cop);
668 }
669
670
671 /**
672  * We just touched the plaintext information about a name in our zone;
673  * refresh the corresponding (encrypted) block in the namecache.
674  *
675  * @param client client responsible for the request
676  * @param rid request ID of the client
677  * @param zone_key private key of the zone
678  * @param name label for the records
679  * @param rd_count number of records
680  * @param rd records stored under the given @a name
681  */
682 static void
683 refresh_block (struct GNUNET_SERVER_Client *client,
684                uint32_t rid,
685                const struct GNUNET_CRYPTO_EcdsaPrivateKey *zone_key,
686                const char *name,
687                unsigned int rd_count,
688                const struct GNUNET_GNSRECORD_Data *rd)
689 {
690   struct GNUNET_GNSRECORD_Block *block;
691   struct CacheOperation *cop;
692   struct GNUNET_CRYPTO_EcdsaPublicKey pkey;
693   struct GNUNET_GNSRECORD_Data *nick;
694   struct GNUNET_GNSRECORD_Data *res;
695   unsigned int res_count;
696
697   nick = get_nick_record (zone_key);
698   res_count = rd_count;
699   res = (struct GNUNET_GNSRECORD_Data *) rd; /* fixme: a bit unclean... */
700   if (NULL != nick)
701   {
702     nick->flags = (nick->flags | GNUNET_GNSRECORD_RF_PRIVATE) ^ GNUNET_GNSRECORD_RF_PRIVATE;
703     merge_with_nick_records (nick, rd_count,rd, &res_count, &res);
704     GNUNET_free (nick);
705   }
706
707   if (0 == res_count)
708     block = GNUNET_GNSRECORD_block_create (zone_key,
709                                            GNUNET_TIME_UNIT_ZERO_ABS,
710                                            name,
711                                            res, rd_count);
712   else
713     block = GNUNET_GNSRECORD_block_create (zone_key,
714                                            GNUNET_GNSRECORD_record_get_expiration_time (res_count,
715                                                res),
716                                            name,
717                                            res, res_count);
718   GNUNET_assert (NULL != block);
719   GNUNET_CRYPTO_ecdsa_key_get_public (zone_key,
720                                       &pkey);
721   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
722               "Caching block for label `%s' with %u records in zone `%s' in namecache\n",
723               name,
724               res_count,
725               GNUNET_GNSRECORD_z2s (&pkey));
726   cop = GNUNET_new (struct CacheOperation);
727   cop->client = client;
728   cop->rid = rid;
729   GNUNET_CONTAINER_DLL_insert (cop_head,
730                                cop_tail,
731                                cop);
732   cop->qe = GNUNET_NAMECACHE_block_cache (namecache,
733                                           block,
734                                           &finish_cache_operation,
735                                           cop);
736   GNUNET_free (block);
737 }
738
739
740 /**
741  * Closure for #lookup_it().
742  */
743 struct RecordLookupContext
744 {
745   const char *label;
746
747   int found;
748
749   unsigned int res_rd_count;
750
751   size_t rd_ser_len;
752
753   char *res_rd;
754
755   struct GNUNET_GNSRECORD_Data *nick;
756 };
757
758
759 static void
760 lookup_it (void *cls,
761            const struct GNUNET_CRYPTO_EcdsaPrivateKey *private_key,
762            const char *label,
763            unsigned int rd_count,
764            const struct GNUNET_GNSRECORD_Data *rd)
765 {
766   struct RecordLookupContext *rlc = cls;
767   struct GNUNET_GNSRECORD_Data *rd_res;
768   unsigned int rdc_res;
769
770   if (0 == strcmp (label, rlc->label))
771   {
772     rlc->found = GNUNET_YES;
773     if (0 != rd_count)
774     {
775       if ((NULL != rlc->nick) && (0 != strcmp(label, GNUNET_GNS_MASTERZONE_STR)))
776       {
777         /* Merge */
778         rd_res = NULL;
779         rdc_res = 0;
780         rlc->nick->flags = (rlc->nick->flags | GNUNET_GNSRECORD_RF_PRIVATE) ^ GNUNET_GNSRECORD_RF_PRIVATE;
781         merge_with_nick_records (rlc->nick,
782                                  rd_count, rd,
783                                  &rdc_res, &rd_res);
784
785         rlc->rd_ser_len = GNUNET_GNSRECORD_records_get_size (rdc_res, rd_res);
786         rlc->res_rd_count = rdc_res;
787         rlc->res_rd = GNUNET_malloc (rlc->rd_ser_len);
788         GNUNET_GNSRECORD_records_serialize (rdc_res, rd_res, rlc->rd_ser_len , rlc->res_rd);
789
790         GNUNET_free  (rd_res);
791         GNUNET_free  (rlc->nick);
792         rlc->nick = NULL;
793       }
794       else
795       {
796         rlc->rd_ser_len = GNUNET_GNSRECORD_records_get_size (rd_count, rd);
797         rlc->res_rd_count = rd_count;
798         rlc->res_rd = GNUNET_malloc (rlc->rd_ser_len);
799         GNUNET_GNSRECORD_records_serialize (rd_count, rd, rlc->rd_ser_len , rlc->res_rd);
800       }
801     }
802     else
803     {
804       rlc->rd_ser_len = 0;
805       rlc->res_rd_count = 0;
806       rlc->res_rd = NULL;
807     }
808   }
809 }
810
811
812 /**
813  * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_LOOKUP message
814  *
815  * @param cls unused
816  * @param client client sending the message
817  * @param message message of type 'struct RecordCreateMessage'
818  */
819 static void
820 handle_record_lookup (void *cls,
821                      struct GNUNET_SERVER_Client *client,
822                      const struct GNUNET_MessageHeader *message)
823 {
824   const struct LabelLookupMessage *ll_msg;
825   struct LabelLookupResponseMessage *llr_msg;
826   struct RecordLookupContext rlc;
827   const char *name_tmp;
828   char *res_name;
829   uint32_t name_len;
830   size_t src_size;
831   size_t res_size;
832   int res;
833
834   if (ntohs (message->size) < sizeof (struct LabelLookupMessage))
835   {
836     GNUNET_break (0);
837     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
838     return;
839   }
840
841   ll_msg = (const struct LabelLookupMessage *) message;
842   name_len = ntohl (ll_msg->label_len);
843   src_size = ntohs (message->size);
844
845   if (name_len !=  src_size - sizeof (struct LabelLookupMessage))
846   {
847     GNUNET_break (0);
848     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
849     return;
850   }
851
852   name_tmp = (const char *) &ll_msg[1];
853   if ('\0' != name_tmp[name_len -1])
854   {
855     GNUNET_break (0);
856     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
857     return;
858   }
859
860   GNUNET_SERVER_receive_done (client, GNUNET_OK);
861   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
862               "Received `%s' message for name `%s'\n",
863               "NAMESTORE_RECORD_LOOKUP", name_tmp);
864
865   if (NULL == (client_lookup (client)))
866   {
867     GNUNET_break (0);
868     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
869     return;
870   }
871
872   rlc.label = name_tmp;
873   rlc.found = GNUNET_NO;
874   rlc.res_rd_count = 0;
875   rlc.res_rd = NULL;
876   rlc.rd_ser_len = 0;
877   rlc.nick = get_nick_record (&ll_msg->zone);
878
879   res = GSN_database->lookup_records (GSN_database->cls,
880         &ll_msg->zone, name_tmp, &lookup_it, &rlc);
881
882   res_size = sizeof (struct LabelLookupResponseMessage) + name_len + rlc.rd_ser_len;
883   llr_msg = GNUNET_malloc (res_size);
884   llr_msg->gns_header.header.size = htons (res_size);
885   llr_msg->gns_header.header.type = htons (GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_LOOKUP_RESPONSE);
886   llr_msg->gns_header.r_id = ll_msg->gns_header.r_id;
887   llr_msg->private_key = ll_msg->zone;
888   llr_msg->name_len = htons (name_len);
889   llr_msg->rd_count = htons (rlc.res_rd_count);
890   llr_msg->rd_len = htons (rlc.rd_ser_len);
891   res_name = (char *) &llr_msg[1];
892   if  ((GNUNET_YES == rlc.found) && (GNUNET_OK == res))
893     llr_msg->found = ntohs (GNUNET_YES);
894   else
895     llr_msg->found = ntohs (GNUNET_NO);
896   memcpy (&llr_msg[1], name_tmp, name_len);
897   memcpy (&res_name[name_len], rlc.res_rd, rlc.rd_ser_len);
898
899   GNUNET_SERVER_notification_context_unicast (snc, client, &llr_msg->gns_header.header,
900       GNUNET_NO);
901
902   GNUNET_free_non_null (rlc.res_rd);
903   GNUNET_free (llr_msg);
904 }
905
906
907 /**
908  * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_STORE message
909  *
910  * @param cls unused
911  * @param client client sending the message
912  * @param message message of type 'struct RecordCreateMessage'
913  */
914 static void
915 handle_record_store (void *cls,
916                      struct GNUNET_SERVER_Client *client,
917                      const struct GNUNET_MessageHeader *message)
918 {
919   const struct RecordStoreMessage *rp_msg;
920   size_t name_len;
921   size_t msg_size;
922   size_t msg_size_exp;
923   size_t rd_ser_len;
924   uint32_t rid;
925   const char *name_tmp;
926   char *conv_name;
927   const char *rd_ser;
928   unsigned int rd_count;
929   int res;
930   struct GNUNET_CRYPTO_EcdsaPublicKey pubkey;
931   struct ZoneMonitor *zm;
932
933   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
934               "Received `%s' message\n",
935               "NAMESTORE_RECORD_STORE");
936   if (ntohs (message->size) < sizeof (struct RecordStoreMessage))
937   {
938     GNUNET_break (0);
939     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
940     return;
941   }
942   rp_msg = (const struct RecordStoreMessage *) message;
943   rid = ntohl (rp_msg->gns_header.r_id);
944   name_len = ntohs (rp_msg->name_len);
945   msg_size = ntohs (message->size);
946   rd_count = ntohs (rp_msg->rd_count);
947   rd_ser_len = ntohs (rp_msg->rd_len);
948   GNUNET_break (0 == ntohs (rp_msg->reserved));
949   msg_size_exp = sizeof (struct RecordStoreMessage) + name_len + rd_ser_len;
950   if (msg_size != msg_size_exp)
951   {
952     GNUNET_break (0);
953     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
954     return;
955   }
956   if ((0 == name_len) || (name_len > MAX_NAME_LEN))
957   {
958     GNUNET_break (0);
959     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
960     return;
961   }
962   name_tmp = (const char *) &rp_msg[1];
963   rd_ser = &name_tmp[name_len];
964   if ('\0' != name_tmp[name_len -1])
965   {
966     GNUNET_break (0);
967     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
968     return;
969   }
970   (void) client_lookup (client);
971   {
972     struct GNUNET_GNSRECORD_Data rd[rd_count];
973
974     if (GNUNET_OK !=
975         GNUNET_GNSRECORD_records_deserialize (rd_ser_len, rd_ser, rd_count, rd))
976     {
977       GNUNET_break (0);
978       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
979       return;
980     }
981
982     /* Extracting and converting private key */
983     GNUNET_CRYPTO_ecdsa_key_get_public (&rp_msg->private_key,
984                                       &pubkey);
985     conv_name = GNUNET_GNSRECORD_string_to_lowercase (name_tmp);
986     if (NULL == conv_name)
987     {
988       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
989                   "Error converting name `%s'\n", name_tmp);
990       GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
991       return;
992     }
993     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
994                 "Creating %u records for name `%s' in zone `%s'\n",
995                 (unsigned int) rd_count,
996                 conv_name,
997                 GNUNET_GNSRECORD_z2s (&pubkey));
998
999     if ( (0 == rd_count) &&
1000          (GNUNET_NO ==
1001           GSN_database->iterate_records (GSN_database->cls,
1002                                          &rp_msg->private_key, 0, NULL, 0)) )
1003     {
1004       /* This name does not exist, so cannot be removed */
1005       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1006                   "Name `%s' does not exist, no deletion required\n",
1007                   conv_name);
1008       res = GNUNET_NO;
1009     }
1010     else
1011     {
1012       struct GNUNET_GNSRECORD_Data rd_clean[rd_count];
1013       unsigned int i;
1014       unsigned int rd_clean_off;
1015
1016       /* remove "NICK" records, unless this is for the "+" label */
1017       rd_clean_off = 0;
1018       for (i=0;i<rd_count;i++)
1019       {
1020         rd_clean[rd_clean_off] = rd[i];
1021         if ( (0 == strcmp (GNUNET_GNS_MASTERZONE_STR,
1022                            conv_name)) ||
1023              (GNUNET_GNSRECORD_TYPE_NICK != rd[i].record_type) )
1024           rd_clean_off++;
1025       }
1026       res = GSN_database->store_records (GSN_database->cls,
1027                                          &rp_msg->private_key,
1028                                          conv_name,
1029                                          rd_clean_off, rd_clean);
1030       if (GNUNET_OK == res)
1031       {
1032         for (zm = monitor_head; NULL != zm; zm = zm->next)
1033         {
1034           if ( (0 == memcmp (&rp_msg->private_key, &zm->zone,
1035                              sizeof (struct GNUNET_CRYPTO_EcdsaPrivateKey))) ||
1036                (0 == memcmp (&zm->zone,
1037                              &zero,
1038                              sizeof (struct GNUNET_CRYPTO_EcdsaPrivateKey))) )
1039           {
1040             GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1041                         "Notifying monitor about changes under label `%s'\n",
1042                         conv_name);
1043             send_lookup_response (monitor_nc,
1044                                   zm->nc->client,
1045                                   0,
1046                                   &rp_msg->private_key,
1047                                   conv_name,
1048                                   rd_count, rd);
1049           }
1050           else
1051             GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1052                         "Monitor is for another zone\n");
1053         }
1054         if (NULL == monitor_head)
1055           GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1056                       "No monitors active\n");
1057       }
1058       else
1059       {
1060         GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1061                     "Error storing record: %d\n",
1062                     res);
1063       }
1064     }
1065     if (GNUNET_OK == res)
1066     {
1067       refresh_block (client, rid,
1068                      &rp_msg->private_key,
1069                      conv_name,
1070                      rd_count, rd);
1071       GNUNET_SERVER_receive_done (client, GNUNET_OK);
1072       GNUNET_free (conv_name);
1073       return;
1074     }
1075     GNUNET_free (conv_name);
1076   }
1077   send_store_response (client, res, rid);
1078   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1079 }
1080
1081
1082 /**
1083  * Context for record remove operations passed from #handle_zone_to_name to
1084  * #handle_zone_to_name_it as closure
1085  */
1086 struct ZoneToNameCtx
1087 {
1088   /**
1089    * Namestore client
1090    */
1091   struct NamestoreClient *nc;
1092
1093   /**
1094    * Request id (to be used in the response to the client).
1095    */
1096   uint32_t rid;
1097
1098   /**
1099    * Set to #GNUNET_OK on success, #GNUNET_SYSERR on error.  Note that
1100    * not finding a name for the zone still counts as a 'success' here,
1101    * as this field is about the success of executing the IPC protocol.
1102    */
1103   int success;
1104 };
1105
1106
1107 /**
1108  * Zone to name iterator
1109  *
1110  * @param cls struct ZoneToNameCtx *
1111  * @param zone_key the zone key
1112  * @param name name
1113  * @param rd_count number of records in @a rd
1114  * @param rd record data
1115  */
1116 static void
1117 handle_zone_to_name_it (void *cls,
1118                         const struct GNUNET_CRYPTO_EcdsaPrivateKey *zone_key,
1119                         const char *name,
1120                         unsigned int rd_count,
1121                         const struct GNUNET_GNSRECORD_Data *rd)
1122 {
1123   struct ZoneToNameCtx *ztn_ctx = cls;
1124   struct ZoneToNameResponseMessage *ztnr_msg;
1125   int16_t res;
1126   size_t name_len;
1127   size_t rd_ser_len;
1128   size_t msg_size;
1129   char *name_tmp;
1130   char *rd_tmp;
1131
1132   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1133               "Found result for zone-to-name lookup: `%s'\n",
1134               name);
1135   res = GNUNET_YES;
1136   name_len = (NULL == name) ? 0 : strlen (name) + 1;
1137   rd_ser_len = GNUNET_GNSRECORD_records_get_size (rd_count, rd);
1138   msg_size = sizeof (struct ZoneToNameResponseMessage) + name_len + rd_ser_len;
1139   if (msg_size >= GNUNET_SERVER_MAX_MESSAGE_SIZE)
1140   {
1141     GNUNET_break (0);
1142     ztn_ctx->success = GNUNET_SYSERR;
1143     return;
1144   }
1145   ztnr_msg = GNUNET_malloc (msg_size);
1146   ztnr_msg->gns_header.header.type = htons (GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_TO_NAME_RESPONSE);
1147   ztnr_msg->gns_header.header.size = htons (msg_size);
1148   ztnr_msg->gns_header.r_id = htonl (ztn_ctx->rid);
1149   ztnr_msg->res = htons (res);
1150   ztnr_msg->rd_len = htons (rd_ser_len);
1151   ztnr_msg->rd_count = htons (rd_count);
1152   ztnr_msg->name_len = htons (name_len);
1153   ztnr_msg->zone = *zone_key;
1154   name_tmp = (char *) &ztnr_msg[1];
1155   if (NULL != name)
1156     memcpy (name_tmp, name, name_len);
1157   rd_tmp = &name_tmp[name_len];
1158   GNUNET_GNSRECORD_records_serialize (rd_count, rd, rd_ser_len, rd_tmp);
1159   ztn_ctx->success = GNUNET_OK;
1160   GNUNET_SERVER_notification_context_unicast (snc, ztn_ctx->nc->client,
1161                                               &ztnr_msg->gns_header.header,
1162                                               GNUNET_NO);
1163   GNUNET_free (ztnr_msg);
1164 }
1165
1166
1167 /**
1168  * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_TO_NAME message
1169  *
1170  * @param cls unused
1171  * @param client client sending the message
1172  * @param message message of type 'struct ZoneToNameMessage'
1173  */
1174 static void
1175 handle_zone_to_name (void *cls,
1176                      struct GNUNET_SERVER_Client *client,
1177                      const struct GNUNET_MessageHeader *message)
1178 {
1179   struct NamestoreClient *nc;
1180   const struct ZoneToNameMessage *ztn_msg;
1181   struct ZoneToNameCtx ztn_ctx;
1182   struct ZoneToNameResponseMessage ztnr_msg;
1183
1184   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1185               "Received `%s' message\n",
1186               "ZONE_TO_NAME");
1187   ztn_msg = (const struct ZoneToNameMessage *) message;
1188   nc = client_lookup (client);
1189   ztn_ctx.rid = ntohl (ztn_msg->gns_header.r_id);
1190   ztn_ctx.nc = nc;
1191   ztn_ctx.success = GNUNET_NO;
1192   if (GNUNET_SYSERR ==
1193       GSN_database->zone_to_name (GSN_database->cls,
1194                                   &ztn_msg->zone,
1195                                   &ztn_msg->value_zone,
1196                                   &handle_zone_to_name_it, &ztn_ctx))
1197   {
1198     /* internal error, hang up instead of signalling something
1199        that might be wrong */
1200     GNUNET_break (0);
1201     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1202     return;
1203   }
1204   if (GNUNET_NO == ztn_ctx.success)
1205   {
1206     /* no result found, send empty response */
1207     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1208                 "Found no result for zone-to-name lookup.\n");
1209     memset (&ztnr_msg, 0, sizeof (ztnr_msg));
1210     ztnr_msg.gns_header.header.type = htons (GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_TO_NAME_RESPONSE);
1211     ztnr_msg.gns_header.header.size = htons (sizeof (ztnr_msg));
1212     ztnr_msg.gns_header.r_id = ztn_msg->gns_header.r_id;
1213     ztnr_msg.res = htons (GNUNET_NO);
1214     GNUNET_SERVER_notification_context_unicast (snc,
1215                                                 client,
1216                                                 &ztnr_msg.gns_header.header,
1217                                                 GNUNET_NO);
1218   }
1219   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1220 }
1221
1222
1223 /**
1224  * Zone iteration processor result
1225  */
1226 enum ZoneIterationResult
1227 {
1228   /**
1229    * Iteration start.
1230    */
1231   IT_START = 0,
1232
1233   /**
1234    * Found records,
1235    * Continue to iterate with next iteration_next call
1236    */
1237   IT_SUCCESS_MORE_AVAILABLE = 1,
1238
1239   /**
1240    * Iteration complete
1241    */
1242   IT_SUCCESS_NOT_MORE_RESULTS_AVAILABLE = 2
1243 };
1244
1245
1246 /**
1247  * Context for record remove operations passed from
1248  * #run_zone_iteration_round to #zone_iterate_proc as closure
1249  */
1250 struct ZoneIterationProcResult
1251 {
1252   /**
1253    * The zone iteration handle
1254    */
1255   struct ZoneIteration *zi;
1256
1257   /**
1258    * Iteration result: iteration done?
1259    * #IT_SUCCESS_MORE_AVAILABLE:  if there may be more results overall but
1260    * we got one for now and have sent it to the client
1261    * #IT_SUCCESS_NOT_MORE_RESULTS_AVAILABLE: if there are no further results,
1262    * #IT_START: if we are still trying to find a result.
1263    */
1264   int res_iteration_finished;
1265
1266 };
1267
1268
1269 /**
1270  * Process results for zone iteration from database
1271  *
1272  * @param cls struct ZoneIterationProcResult *proc
1273  * @param zone_key the zone key
1274  * @param name name
1275  * @param rd_count number of records for this name
1276  * @param rd record data
1277  */
1278 static void
1279 zone_iterate_proc (void *cls,
1280                        const struct GNUNET_CRYPTO_EcdsaPrivateKey *zone_key,
1281                        const char *name,
1282                        unsigned int rd_count,
1283                        const struct GNUNET_GNSRECORD_Data *rd)
1284 {
1285   struct ZoneIterationProcResult *proc = cls;
1286   unsigned int i;
1287   int do_refresh_block;
1288
1289   if ((NULL == zone_key) && (NULL == name))
1290   {
1291     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1292                 "Iteration done\n");
1293     proc->res_iteration_finished = IT_SUCCESS_NOT_MORE_RESULTS_AVAILABLE;
1294     return;
1295   }
1296   if ((NULL == zone_key) || (NULL == name))
1297   {
1298     /* what is this!? should never happen */
1299     proc->res_iteration_finished = IT_START;
1300     GNUNET_break (0);
1301     return;
1302   }
1303   proc->res_iteration_finished = IT_SUCCESS_MORE_AVAILABLE;
1304   send_lookup_response (snc,
1305                         proc->zi->client->client,
1306                         proc->zi->request_id,
1307                         zone_key,
1308                         name,
1309                         rd_count,
1310                         rd);
1311   do_refresh_block = GNUNET_NO;
1312   for (i=0;i<rd_count;i++)
1313     if(  (0 != (rd[i].flags & GNUNET_GNSRECORD_RF_RELATIVE_EXPIRATION)) &&
1314          (0 == (rd[i].flags & GNUNET_GNSRECORD_RF_PENDING)) )
1315     {
1316       do_refresh_block = GNUNET_YES;
1317       break;
1318     }
1319   if (GNUNET_YES == do_refresh_block)
1320     refresh_block (NULL, 0,
1321                    zone_key,
1322                    name,
1323                    rd_count,
1324                    rd);
1325
1326 }
1327
1328
1329 /**
1330  * Perform the next round of the zone iteration.
1331  *
1332  * @param zi zone iterator to process
1333  */
1334 static void
1335 run_zone_iteration_round (struct ZoneIteration *zi)
1336 {
1337   struct ZoneIterationProcResult proc;
1338   struct RecordResultMessage rrm;
1339   int ret;
1340
1341   memset (&proc, 0, sizeof (proc));
1342   proc.zi = zi;
1343   proc.res_iteration_finished = IT_START;
1344   while (IT_START == proc.res_iteration_finished)
1345   {
1346     if (GNUNET_SYSERR ==
1347         (ret = GSN_database->iterate_records (GSN_database->cls,
1348                                               (0 == memcmp (&zi->zone, &zero, sizeof (zero)))
1349                                               ? NULL
1350                                               : &zi->zone,
1351                                               zi->offset,
1352                                               &zone_iterate_proc, &proc)))
1353     {
1354       GNUNET_break (0);
1355       break;
1356     }
1357     if (GNUNET_NO == ret)
1358       proc.res_iteration_finished = IT_SUCCESS_NOT_MORE_RESULTS_AVAILABLE;
1359     zi->offset++;
1360   }
1361   if (IT_SUCCESS_MORE_AVAILABLE == proc.res_iteration_finished)
1362   {
1363     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1364                 "More results available\n");
1365     return; /* more results later */
1366   }
1367   /* send empty response to indicate end of list */
1368   memset (&rrm, 0, sizeof (rrm));
1369   rrm.gns_header.header.type = htons (GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_RESULT);
1370   rrm.gns_header.header.size = htons (sizeof (rrm));
1371   rrm.gns_header.r_id = htonl (zi->request_id);
1372   GNUNET_SERVER_notification_context_unicast (snc,
1373                                               zi->client->client,
1374                                               &rrm.gns_header.header,
1375                                               GNUNET_NO);
1376   GNUNET_CONTAINER_DLL_remove (zi->client->op_head,
1377                                zi->client->op_tail,
1378                                zi);
1379   GNUNET_free (zi);
1380 }
1381
1382
1383 /**
1384  * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_ITERATION_START message
1385  *
1386  * @param cls unused
1387  * @param client the client sending the message
1388  * @param message message of type 'struct ZoneIterationStartMessage'
1389  */
1390 static void
1391 handle_iteration_start (void *cls,
1392                         struct GNUNET_SERVER_Client *client,
1393                         const struct GNUNET_MessageHeader *message)
1394 {
1395   const struct ZoneIterationStartMessage *zis_msg;
1396   struct NamestoreClient *nc;
1397   struct ZoneIteration *zi;
1398
1399   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received `%s' message\n", "ZONE_ITERATION_START");
1400   if (NULL == (nc = client_lookup (client)))
1401   {
1402     GNUNET_break (0);
1403     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1404     return;
1405   }
1406   zis_msg = (const struct ZoneIterationStartMessage *) message;
1407   zi = GNUNET_new (struct ZoneIteration);
1408   zi->request_id = ntohl (zis_msg->gns_header.r_id);
1409   zi->offset = 0;
1410   zi->client = nc;
1411   zi->zone = zis_msg->zone;
1412
1413
1414   GNUNET_CONTAINER_DLL_insert (nc->op_head, nc->op_tail, zi);
1415   run_zone_iteration_round (zi);
1416   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1417 }
1418
1419
1420 /**
1421  * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_ITERATION_STOP message
1422  *
1423  * @param cls unused
1424  * @param client GNUNET_SERVER_Client sending the message
1425  * @param message message of type 'struct ZoneIterationStopMessage'
1426  */
1427 static void
1428 handle_iteration_stop (void *cls,
1429                        struct GNUNET_SERVER_Client *client,
1430                        const struct GNUNET_MessageHeader *message)
1431 {
1432   struct NamestoreClient *nc;
1433   struct ZoneIteration *zi;
1434   const struct ZoneIterationStopMessage *zis_msg;
1435   uint32_t rid;
1436
1437   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1438               "Received `%s' message\n",
1439               "ZONE_ITERATION_STOP");
1440   if (NULL == (nc = client_lookup(client)))
1441   {
1442     GNUNET_break (0);
1443     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1444     return;
1445   }
1446   zis_msg = (const struct ZoneIterationStopMessage *) message;
1447   rid = ntohl (zis_msg->gns_header.r_id);
1448   for (zi = nc->op_head; NULL != zi; zi = zi->next)
1449     if (zi->request_id == rid)
1450       break;
1451   if (NULL == zi)
1452   {
1453     GNUNET_break (0);
1454     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1455     return;
1456   }
1457   GNUNET_CONTAINER_DLL_remove (nc->op_head, nc->op_tail, zi);
1458   GNUNET_free (zi);
1459   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1460 }
1461
1462
1463 /**
1464  * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_ITERATION_NEXT message
1465  *
1466  * @param cls unused
1467  * @param client GNUNET_SERVER_Client sending the message
1468  * @param message message of type 'struct ZoneIterationNextMessage'
1469  */
1470 static void
1471 handle_iteration_next (void *cls,
1472                        struct GNUNET_SERVER_Client *client,
1473                        const struct GNUNET_MessageHeader *message)
1474 {
1475   struct NamestoreClient *nc;
1476   struct ZoneIteration *zi;
1477   const struct ZoneIterationNextMessage *zis_msg;
1478   uint32_t rid;
1479
1480   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1481               "Received `%s' message\n",
1482               "ZONE_ITERATION_NEXT");
1483   if (NULL == (nc = client_lookup(client)))
1484   {
1485     GNUNET_break (0);
1486     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1487     return;
1488   }
1489   zis_msg = (const struct ZoneIterationNextMessage *) message;
1490   rid = ntohl (zis_msg->gns_header.r_id);
1491   for (zi = nc->op_head; NULL != zi; zi = zi->next)
1492     if (zi->request_id == rid)
1493       break;
1494   if (NULL == zi)
1495   {
1496     GNUNET_break (0);
1497     GNUNET_SERVER_receive_done (client, GNUNET_SYSERR);
1498     return;
1499   }
1500   run_zone_iteration_round (zi);
1501   GNUNET_SERVER_receive_done (client, GNUNET_OK);
1502 }
1503
1504
1505 /**
1506  * Send 'sync' message to zone monitor, we're now in sync.
1507  *
1508  * @param zm monitor that is now in sync
1509  */
1510 static void
1511 monitor_sync (struct ZoneMonitor *zm)
1512 {
1513   struct GNUNET_MessageHeader sync;
1514
1515   sync.size = htons (sizeof (struct GNUNET_MessageHeader));
1516   sync.type = htons (GNUNET_MESSAGE_TYPE_NAMESTORE_MONITOR_SYNC);
1517   GNUNET_SERVER_notification_context_unicast (monitor_nc,
1518                                               zm->nc->client,
1519                                               &sync,
1520                                               GNUNET_NO);
1521 }
1522
1523
1524 /**
1525  * Obtain the next datum during the zone monitor's zone intiial iteration.
1526  *
1527  * @param cls zone monitor that does its initial iteration
1528  * @param tc scheduler context
1529  */
1530 static void
1531 monitor_next (void *cls,
1532               const struct GNUNET_SCHEDULER_TaskContext *tc);
1533
1534
1535 /**
1536  * A #GNUNET_NAMESTORE_RecordIterator for monitors.
1537  *
1538  * @param cls a 'struct ZoneMonitor *' with information about the monitor
1539  * @param zone_key zone key of the zone
1540  * @param name name
1541  * @param rd_count number of records in @a rd
1542  * @param rd array of records
1543  */
1544 static void
1545 monitor_iterate_cb (void *cls,
1546                     const struct GNUNET_CRYPTO_EcdsaPrivateKey *zone_key,
1547                     const char *name,
1548                     unsigned int rd_count,
1549                     const struct GNUNET_GNSRECORD_Data *rd)
1550 {
1551   struct ZoneMonitor *zm = cls;
1552
1553   if (NULL == name)
1554   {
1555     /* finished with iteration */
1556     monitor_sync (zm);
1557     return;
1558   }
1559   send_lookup_response (monitor_nc,
1560                         zm->nc->client,
1561                         0,
1562                         zone_key,
1563                         name,
1564                         rd_count,
1565                         rd);
1566   zm->task = GNUNET_SCHEDULER_add_now (&monitor_next, zm);
1567 }
1568
1569
1570 /**
1571  * Handles a #GNUNET_MESSAGE_TYPE_NAMESTORE_MONITOR_START message
1572  *
1573  * @param cls unused
1574  * @param client the client sending the message
1575  * @param message message of type 'struct ZoneMonitorStartMessage'
1576  */
1577 static void
1578 handle_monitor_start (void *cls,
1579                       struct GNUNET_SERVER_Client *client,
1580                       const struct GNUNET_MessageHeader *message)
1581 {
1582   const struct ZoneMonitorStartMessage *zis_msg;
1583   struct ZoneMonitor *zm;
1584
1585   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
1586               "Received `%s' message\n",
1587               "ZONE_MONITOR_START");
1588   zis_msg = (const struct ZoneMonitorStartMessage *) message;
1589   zm = GNUNET_new (struct ZoneMonitor);
1590   zm->offset = 0;
1591   zm->nc = client_lookup (client);
1592   zm->zone = zis_msg->zone;
1593   GNUNET_CONTAINER_DLL_insert (monitor_head, monitor_tail, zm);
1594   GNUNET_SERVER_client_mark_monitor (client);
1595   GNUNET_SERVER_disable_receive_done_warning (client);
1596   GNUNET_SERVER_notification_context_add (monitor_nc,
1597                                           client);
1598   if (GNUNET_YES == ntohl (zis_msg->iterate_first))
1599     zm->task = GNUNET_SCHEDULER_add_now (&monitor_next, zm);
1600   else
1601     monitor_sync (zm);
1602 }
1603
1604
1605 /**
1606  * Obtain the next datum during the zone monitor's zone intiial iteration.
1607  *
1608  * @param cls zone monitor that does its initial iteration
1609  * @param tc scheduler context
1610  */
1611 static void
1612 monitor_next (void *cls,
1613               const struct GNUNET_SCHEDULER_TaskContext *tc)
1614 {
1615   struct ZoneMonitor *zm = cls;
1616   int ret;
1617
1618   zm->task = GNUNET_SCHEDULER_NO_TASK;
1619   ret = GSN_database->iterate_records (GSN_database->cls,
1620                                        (0 == memcmp (&zm->zone, &zero, sizeof (zero)))
1621                                        ? NULL
1622                                        : &zm->zone,
1623                                        zm->offset++,
1624                                        &monitor_iterate_cb, zm);
1625   if (GNUNET_SYSERR == ret)
1626   {
1627     GNUNET_SERVER_client_disconnect (zm->nc->client);
1628     return;
1629   }
1630   if (GNUNET_NO == ret)
1631   {
1632     /* empty zone */
1633     monitor_sync (zm);
1634     return;
1635   }
1636 }
1637
1638
1639 /**
1640  * Process namestore requests.
1641  *
1642  * @param cls closure
1643  * @param server the initialized server
1644  * @param cfg configuration to use
1645  */
1646 static void
1647 run (void *cls, struct GNUNET_SERVER_Handle *server,
1648      const struct GNUNET_CONFIGURATION_Handle *cfg)
1649 {
1650   static const struct GNUNET_SERVER_MessageHandler handlers[] = {
1651     {&handle_record_store, NULL,
1652      GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_STORE, 0},
1653     {&handle_record_lookup, NULL,
1654      GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_LOOKUP, 0},
1655     {&handle_zone_to_name, NULL,
1656      GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_TO_NAME, sizeof (struct ZoneToNameMessage) },
1657     {&handle_iteration_start, NULL,
1658      GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_ITERATION_START, sizeof (struct ZoneIterationStartMessage) },
1659     {&handle_iteration_next, NULL,
1660      GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_ITERATION_NEXT, sizeof (struct ZoneIterationNextMessage) },
1661     {&handle_iteration_stop, NULL,
1662      GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_ITERATION_STOP, sizeof (struct ZoneIterationStopMessage) },
1663     {&handle_monitor_start, NULL,
1664      GNUNET_MESSAGE_TYPE_NAMESTORE_MONITOR_START, sizeof (struct ZoneMonitorStartMessage) },
1665     {NULL, NULL, 0, 0}
1666   };
1667   char *database;
1668
1669   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Starting namestore service\n");
1670   GSN_cfg = cfg;
1671   monitor_nc = GNUNET_SERVER_notification_context_create (server, 1);
1672   namecache = GNUNET_NAMECACHE_connect (cfg);
1673   /* Loading database plugin */
1674   if (GNUNET_OK !=
1675       GNUNET_CONFIGURATION_get_value_string (cfg, "namestore", "database",
1676                                              &database))
1677     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "No database backend configured\n");
1678
1679   GNUNET_asprintf (&db_lib_name, "libgnunet_plugin_namestore_%s", database);
1680   GSN_database = GNUNET_PLUGIN_load (db_lib_name, (void *) GSN_cfg);
1681   GNUNET_free (database);
1682   if (NULL == GSN_database)
1683   {
1684     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
1685                 "Could not load database backend `%s'\n",
1686                 db_lib_name);
1687     GNUNET_SCHEDULER_add_now (&cleanup_task, NULL);
1688     return;
1689   }
1690
1691   /* Configuring server handles */
1692   GNUNET_SERVER_add_handlers (server, handlers);
1693   snc = GNUNET_SERVER_notification_context_create (server, 16);
1694   GNUNET_SERVER_disconnect_notify (server,
1695                                    &client_disconnect_notification,
1696                                    NULL);
1697   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &cleanup_task,
1698                                 NULL);
1699 }
1700
1701
1702 /**
1703  * The main function for the template service.
1704  *
1705  * @param argc number of arguments from the command line
1706  * @param argv command line arguments
1707  * @return 0 ok, 1 on error
1708  */
1709 int
1710 main (int argc, char *const *argv)
1711 {
1712   return (GNUNET_OK ==
1713           GNUNET_SERVICE_run (argc, argv, "namestore",
1714                               GNUNET_SERVICE_OPTION_NONE, &run, NULL)) ? 0 : 1;
1715 }
1716
1717 /* end of gnunet-service-namestore.c */
1718