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