src/namestore/gnunet-service-namestore.c

   1 /*
   2      This file is part of GNUnet.
   3      (C) 2009 Christian Grothoff (and other contributing authors)
   4
   5      GNUnet is free software; you can redistribute it and/or modify
   6      it under the terms of the GNU General Public License as published
   7      by the Free Software Foundation; either version 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  */
  26 #include "platform.h"
  27 #include "gnunet_getopt_lib.h"
  28 #include "gnunet_service_lib.h"
  29 #include "gnunet_namestore_service.h"
  30 #include "gnunet_namestore_plugin.h"
  31 #include "gnunet_signatures.h"
  32 #include "namestore.h"
  33
  34
  35
  36 /**
  37  * A namestore operation.
  38  */
  39 struct GNUNET_NAMESTORE_ZoneIteration
  40 {
  41   struct GNUNET_NAMESTORE_ZoneIteration *next;
  42   struct GNUNET_NAMESTORE_ZoneIteration *prev;
  43
  44   struct GNUNET_NAMESTORE_Client * client;
  45
  46   GNUNET_HashCode zone;
  47
  48   uint64_t request_id;
  49   uint32_t offset;
  50
  51 };
  52
  53
  54 /**
  55  * A namestore client
  56  */
  57 struct GNUNET_NAMESTORE_Client
  58 {
  59   struct GNUNET_NAMESTORE_Client *next;
  60   struct GNUNET_NAMESTORE_Client *prev;
  61
  62   struct GNUNET_SERVER_Client * client;
  63
  64   struct GNUNET_NAMESTORE_ZoneIteration *op_head;
  65   struct GNUNET_NAMESTORE_ZoneIteration *op_tail;
  66 };
  67
  68
  69
  70 /**
  71  * Configuration handle.
  72  */
  73 const struct GNUNET_CONFIGURATION_Handle *GSN_cfg;
  74
  75 static struct GNUNET_NAMESTORE_PluginFunctions *GSN_database;
  76
  77 /**
  78  * Our notification context.
  79  */
  80 static struct GNUNET_SERVER_NotificationContext *snc;
  81
  82 static char *db_lib_name;
  83
  84 static struct GNUNET_NAMESTORE_Client *client_head;
  85 static struct GNUNET_NAMESTORE_Client *client_tail;
  86
  87
  88 /**
  89  * Task run during shutdown.
  90  *
  91  * @param cls unused
  92  * @param tc unused
  93  */
  94 static void
  95 cleanup_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
  96 {
  97   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Stopping namestore service\n");
  98
  99   struct GNUNET_NAMESTORE_ZoneIteration * no;
 100   struct GNUNET_NAMESTORE_ZoneIteration * tmp;
 101   struct GNUNET_NAMESTORE_Client * nc;
 102   struct GNUNET_NAMESTORE_Client * next;
 103
 104   for (nc = client_head; nc != NULL; nc = next)
 105   {
 106     next = nc->next;
 107     for (no = nc->op_head; no != NULL; no = tmp)
 108     {
 109       GNUNET_CONTAINER_DLL_remove (nc->op_head, nc->op_tail, no);
 110       tmp = no->next;
 111       GNUNET_free (no);
 112     }
 113
 114     GNUNET_CONTAINER_DLL_remove (client_head, client_tail, nc);
 115     GNUNET_free (nc);
 116
 117   }
 118
 119   GNUNET_SERVER_notification_context_destroy (snc);
 120   snc = NULL;
 121
 122   GNUNET_break (NULL == GNUNET_PLUGIN_unload (db_lib_name, GSN_database));
 123   GNUNET_free (db_lib_name);
 124 }
 125
 126 static struct GNUNET_NAMESTORE_Client *
 127 client_lookup (struct GNUNET_SERVER_Client *client)
 128 {
 129   struct GNUNET_NAMESTORE_Client * nc;
 130
 131   GNUNET_assert (NULL != client);
 132
 133   for (nc = client_head; nc != NULL; nc = nc->next)
 134   {
 135     if (client == nc->client)
 136       break;
 137   }
 138   return nc;
 139 }
 140
 141
 142 /**
 143  * Called whenever a client is disconnected.  Frees our
 144  * resources associated with that client.
 145  *
 146  * @param cls closure
 147  * @param client identification of the client
 148  */
 149 static void
 150 client_disconnect_notification (void *cls, struct GNUNET_SERVER_Client *client)
 151 {
 152   struct GNUNET_NAMESTORE_ZoneIteration * no;
 153   struct GNUNET_NAMESTORE_Client * nc;
 154   if (NULL == client)
 155     return;
 156
 157   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Client %p disconnected \n", client);
 158
 159   nc = client_lookup (client);
 160
 161   if ((NULL == client) || (NULL == nc))
 162     return;
 163
 164   for (no = nc->op_head; no != NULL; no = no->next)
 165   {
 166     GNUNET_CONTAINER_DLL_remove (nc->op_head, nc->op_tail, no);
 167     GNUNET_free (no);
 168   }
 169
 170   GNUNET_CONTAINER_DLL_remove (client_head, client_tail, nc);
 171   GNUNET_free (nc);
 172 }
 173
 174 static void handle_start (void *cls,
 175                           struct GNUNET_SERVER_Client * client,
 176                           const struct GNUNET_MessageHeader * message)
 177 {
 178   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Client %p connected\n", client);
 179
 180   struct GNUNET_NAMESTORE_Client * nc = GNUNET_malloc (sizeof (struct GNUNET_NAMESTORE_Client));
 181   nc->client = client;
 182   GNUNET_SERVER_notification_context_add (snc, client);
 183   GNUNET_CONTAINER_DLL_insert(client_head, client_tail, nc);
 184
 185   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 186 }
 187
 188 struct LookupNameContext
 189 {
 190   struct GNUNET_NAMESTORE_Client *nc;
 191   uint32_t request_id;
 192   uint32_t record_type;
 193 };
 194
 195
 196
 197
 198 static void
 199 handle_lookup_name_it (void *cls,
 200     const struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded *zone_key,
 201     struct GNUNET_TIME_Absolute expire,
 202     const char *name,
 203     unsigned int rd_count,
 204     const struct GNUNET_NAMESTORE_RecordData *rd,
 205     const struct GNUNET_CRYPTO_RsaSignature *signature)
 206 {
 207   /* send response */
 208   struct LookupNameContext *lnc = cls;
 209   struct LookupNameResponseMessage *lnr_msg;
 210
 211   struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded *zone_key_tmp;
 212   struct GNUNET_NAMESTORE_RecordData *rd_selected = NULL;
 213   char *rd_tmp;
 214   char *name_tmp;
 215   size_t rd_ser_len;
 216   struct GNUNET_CRYPTO_RsaSignature *signature_tmp;
 217
 218   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending `%s' message\n", "NAMESTORE_LOOKUP_NAME_RESPONSE");
 219
 220   size_t r_size = 0;
 221
 222   size_t name_len = 0;
 223   if (NULL != name)
 224     name_len = strlen(name) + 1;
 225
 226   int copied_elements = 0;
 227   int contains_signature = 0;
 228   int c;
 229
 230   /* count records to copy */
 231   if (rd_count != 0)
 232   {
 233     if (lnc->record_type != 0)
 234     {
 235       /* special record type needed */
 236       for (c = 0; c < rd_count; c ++)
 237         if (rd[c].record_type == lnc->record_type)
 238           copied_elements++; /* found matching record */
 239       rd_selected = GNUNET_malloc (copied_elements * sizeof (struct GNUNET_NAMESTORE_RecordData));
 240       copied_elements = 0;
 241       for (c = 0; c < rd_count; c ++)
 242       {
 243         if (rd[c].record_type == lnc->record_type)
 244         {
 245           /* found matching record */
 246           memcpy (&rd_selected[copied_elements], &rd[c], sizeof (struct GNUNET_NAMESTORE_RecordData));
 247           copied_elements++;
 248         }
 249       }
 250     }
 251     else
 252     {
 253       copied_elements = rd_count;
 254       rd_selected = (struct GNUNET_NAMESTORE_RecordData *) rd;
 255     }
 256   }
 257   else
 258   {
 259     /* No results */
 260     copied_elements = 0;
 261     rd_selected = NULL;
 262     expire = GNUNET_TIME_UNIT_ZERO_ABS;
 263   }
 264
 265
 266
 267   rd_ser_len = GNUNET_NAMESTORE_records_get_size(copied_elements, rd_selected);
 268   char rd_ser[rd_ser_len];
 269   GNUNET_NAMESTORE_records_serialize(copied_elements, rd_selected, rd_ser_len, rd_ser);
 270
 271
 272   if ((copied_elements == rd_count) && (signature != NULL))
 273       contains_signature = GNUNET_YES;
 274
 275   if (rd_selected != rd)
 276     GNUNET_free (rd_selected);
 277
 278   r_size = sizeof (struct LookupNameResponseMessage) +
 279            sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded) +
 280            name_len +
 281            rd_ser_len +
 282            contains_signature * sizeof (struct GNUNET_CRYPTO_RsaSignature);
 283
 284   lnr_msg = GNUNET_malloc (r_size);
 285
 286   lnr_msg->gns_header.header.type = ntohs (GNUNET_MESSAGE_TYPE_NAMESTORE_LOOKUP_NAME_RESPONSE);
 287   lnr_msg->gns_header.header.size = ntohs (r_size);
 288   lnr_msg->gns_header.r_id = htonl (lnc->request_id);
 289   lnr_msg->rd_count = htons (rd_count);
 290   lnr_msg->rd_len = htons (rd_ser_len);
 291   lnr_msg->name_len = htons (name_len);
 292   lnr_msg->expire = GNUNET_TIME_absolute_hton(expire);
 293   lnr_msg->contains_sig = htons (contains_signature);
 294
 295   zone_key_tmp =  (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded *) &lnr_msg[1];
 296   name_tmp = (char *) &zone_key_tmp[1];
 297   rd_tmp = &name_tmp[name_len];
 298   signature_tmp = (struct GNUNET_CRYPTO_RsaSignature *) &rd_tmp[rd_ser_len];
 299
 300   if (zone_key != NULL)
 301     memcpy (zone_key_tmp, zone_key, sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded));
 302   else
 303   {
 304     struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded dummy;
 305     memset (&dummy, '0', sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded));
 306     memcpy (zone_key_tmp, &dummy, sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded));
 307   }
 308   memcpy (name_tmp, name, name_len);
 309   memcpy (rd_tmp, rd_ser, rd_ser_len);
 310
 311   if (GNUNET_YES == contains_signature)
 312     memcpy (signature_tmp, signature, sizeof (struct GNUNET_CRYPTO_RsaSignature));
 313   GNUNET_SERVER_notification_context_unicast (snc, lnc->nc->client, (const struct GNUNET_MessageHeader *) lnr_msg, GNUNET_NO);
 314
 315   GNUNET_free (lnr_msg);
 316 }
 317
 318 static void handle_lookup_name (void *cls,
 319                           struct GNUNET_SERVER_Client * client,
 320                           const struct GNUNET_MessageHeader * message)
 321 {
 322   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received `%s' message\n", "NAMESTORE_LOOKUP_NAME");
 323   struct LookupNameContext lnc;
 324   struct GNUNET_NAMESTORE_Client *nc;
 325   size_t name_len;
 326   char * name;
 327   uint32_t rid = 0;
 328   uint32_t type = 0;
 329   int res;
 330
 331   if (ntohs (message->size) < sizeof (struct LookupNameMessage))
 332   {
 333     GNUNET_break_op (0);
 334     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 335     return;
 336   }
 337
 338   nc = client_lookup(client);
 339   if (nc == NULL)
 340   {
 341     GNUNET_break_op (0);
 342     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 343     return;
 344   }
 345
 346   struct LookupNameMessage * ln_msg = (struct LookupNameMessage *) message;
 347   rid = ntohl (ln_msg->gns_header.r_id);
 348   name_len = ntohl (ln_msg->name_len);
 349   type = ntohl (ln_msg->record_type);
 350
 351   if ((name_len == 0) || (name_len > 256))
 352   {
 353     GNUNET_break_op (0);
 354     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 355     return;
 356   }
 357
 358   name = (char *) &ln_msg[1];
 359   if (name[name_len -1] != '\0')
 360   {
 361     GNUNET_break_op (0);
 362     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 363     return;
 364   }
 365
 366   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Looking up record for name `%s'\n", name);
 367
 368   /* do the actual lookup */
 369   lnc.request_id = rid;
 370   lnc.nc = nc;
 371   lnc.record_type = type;
 372   res = GSN_database->iterate_records(GSN_database->cls, &ln_msg->zone, name, 0, &handle_lookup_name_it, &lnc);
 373
 374   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 375 }
 376
 377 static void handle_record_put (void *cls,
 378                           struct GNUNET_SERVER_Client * client,
 379                           const struct GNUNET_MessageHeader * message)
 380 {
 381   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received `%s' message\n", "NAMESTORE_RECORD_PUT");
 382   struct GNUNET_NAMESTORE_Client *nc;
 383   struct GNUNET_TIME_Absolute expire;
 384   struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded *zone_key;
 385   struct GNUNET_CRYPTO_RsaSignature *signature;
 386   struct RecordPutResponseMessage rpr_msg;
 387   size_t name_len;
 388   size_t msg_size;
 389   size_t msg_size_exp;
 390   size_t key_len;
 391   char * name;
 392   char * rd_ser;
 393   uint32_t rid = 0;
 394   uint32_t rd_ser_len;
 395   uint32_t rd_count;
 396   int res = GNUNET_SYSERR;
 397
 398   if (ntohs (message->size) < sizeof (struct RecordPutMessage))
 399   {
 400     GNUNET_break_op (0);
 401     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 402     return;
 403   }
 404
 405   nc = client_lookup (client);
 406   if (nc == NULL)
 407   {
 408     GNUNET_break_op (0);
 409     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 410     return;
 411   }
 412
 413   struct RecordPutMessage * rp_msg = (struct RecordPutMessage *) message;
 414
 415   rid = ntohl (rp_msg->gns_header.r_id);
 416   msg_size = ntohs (rp_msg->gns_header.header.size);
 417   key_len = sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded);
 418   name_len = ntohs (rp_msg->name_len);
 419   rd_count = ntohs (rp_msg->rd_count);
 420   rd_ser_len = ntohs(rp_msg->rd_len);
 421   msg_size_exp = sizeof (struct RecordPutMessage) + key_len + name_len  + rd_ser_len;
 422
 423   if (msg_size != msg_size_exp)
 424   {
 425     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Expected message %u size but message size is %u \n", msg_size_exp, msg_size);
 426     GNUNET_break_op (0);
 427     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 428     return;
 429   }
 430
 431
 432   if ((name_len == 0) || (name_len > 256))
 433   {
 434     GNUNET_break_op (0);
 435     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 436     return;
 437   }
 438
 439   zone_key = (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded *) &rp_msg[1];
 440   name =  &((char *)zone_key)[key_len];
 441
 442   if (name[name_len -1] != '\0')
 443   {
 444     GNUNET_break_op (0);
 445     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 446     return;
 447   }
 448
 449   expire = GNUNET_TIME_absolute_ntoh(rp_msg->expire);
 450   signature = (struct GNUNET_CRYPTO_RsaSignature *) &rp_msg->signature;
 451
 452   rd_ser = &name[name_len];
 453   struct GNUNET_NAMESTORE_RecordData rd[rd_count];
 454   GNUNET_NAMESTORE_records_deserialize(rd_ser_len, rd_ser, rd_count, rd);
 455
 456   GNUNET_HashCode zone_hash;
 457   GNUNET_CRYPTO_hash (zone_key, key_len, &zone_hash);
 458   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "PUT ZONE HASH: `%s'\n", GNUNET_h2s_full(&zone_hash));
 459
 460
 461   /* Database operation */
 462   res = GSN_database->put_records(GSN_database->cls,
 463                                 zone_key,
 464                                 expire,
 465                                 name,
 466                                 rd_count, rd,
 467                                 signature);
 468
 469   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Putting record for name `%s': %s\n",
 470       name, (res == GNUNET_OK) ? "OK" : "FAIL");
 471
 472   /* Send response */
 473
 474   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending `%s' message\n", "RECORD_PUT_RESPONSE");
 475   rpr_msg.header.type = htons (GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_PUT_RESPONSE);
 476   rpr_msg.op_id = rp_msg->gns_header.r_id;
 477   rpr_msg.header.size = htons (sizeof (struct RecordPutResponseMessage));
 478   if (GNUNET_OK == res)
 479     rpr_msg.op_result = htons (GNUNET_OK);
 480   else
 481     rpr_msg.op_result = htons (GNUNET_NO);
 482   GNUNET_SERVER_notification_context_unicast (snc, nc->client, (const struct GNUNET_MessageHeader *) &rpr_msg, GNUNET_NO);
 483
 484   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 485 }
 486
 487 struct CreateRecordContext
 488 {
 489   struct GNUNET_NAMESTORE_RecordData *rd;
 490   struct GNUNET_CRYPTO_RsaPrivateKey *pkey;
 491   struct GNUNET_TIME_Absolute expire;
 492   uint32_t op_id;
 493   struct GNUNET_NAMESTORE_Client *nc;
 494 };
 495
 496 struct GNUNET_CRYPTO_RsaSignature *
 497 GNUNET_NAMESTORE_create_signature (const struct GNUNET_CRYPTO_RsaPrivateKey *key, struct GNUNET_NAMESTORE_RecordData *rd, unsigned int rd_count)
 498 {
 499   struct GNUNET_CRYPTO_RsaSignature *sig = GNUNET_malloc(sizeof (struct GNUNET_CRYPTO_RsaSignature));
 500   struct GNUNET_CRYPTO_RsaSignaturePurpose *sig_purpose;
 501   size_t rd_ser_len;
 502
 503   rd_ser_len = GNUNET_NAMESTORE_records_get_size(rd_count, rd);
 504   char rd_ser[rd_ser_len];
 505   GNUNET_NAMESTORE_records_serialize(rd_count, rd, rd_ser_len, rd_ser);
 506
 507   sig_purpose = GNUNET_malloc(sizeof (struct GNUNET_CRYPTO_RsaSignaturePurpose) + rd_ser_len);
 508
 509   sig_purpose->size = htonl (sizeof (struct GNUNET_CRYPTO_RsaSignaturePurpose)+ rd_ser_len);
 510   sig_purpose->purpose = htonl (GNUNET_SIGNATURE_PURPOSE_GNS_RECORD_SIGN);
 511   memcpy (&sig_purpose[1], rd_ser, rd_ser_len);
 512
 513   GNUNET_CRYPTO_rsa_sign (key, sig_purpose, sig);
 514
 515   GNUNET_free (rd_ser);
 516   GNUNET_free (sig_purpose);
 517   return sig;
 518 }
 519
 520 static void
 521 handle_create_record_it (void *cls,
 522     const struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded *zone_key,
 523     struct GNUNET_TIME_Absolute expire,
 524     const char *name,
 525     unsigned int rd_count,
 526     const struct GNUNET_NAMESTORE_RecordData *rd,
 527     const struct GNUNET_CRYPTO_RsaSignature *signature)
 528 {
 529   struct CreateRecordContext * crc = cls;
 530   struct GNUNET_CRYPTO_RsaSignature *signature_new;
 531   struct RecordCreateResponseMessage rcr_msg;
 532   int res;
 533
 534   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Found %u existing records for `%s'\n", rd_count, name);
 535   struct GNUNET_NAMESTORE_RecordData *rd_new = GNUNET_malloc ((rd_count+1) * sizeof (struct GNUNET_NAMESTORE_RecordData));
 536   memcpy (rd_new, rd, rd_count * sizeof (struct GNUNET_NAMESTORE_RecordData));
 537
 538   rd_new[rd_count] = *(crc->rd);
 539
 540   signature_new = GNUNET_NAMESTORE_create_signature (crc->pkey, rd_new, rd_count+1);
 541
 542   /* Database operation */
 543   res = GSN_database->put_records(GSN_database->cls,
 544                                 zone_key,
 545                                 expire,
 546                                 name,
 547                                 rd_count +1, rd_new,
 548                                 signature_new);
 549
 550   GNUNET_free (rd_new);
 551   GNUNET_free (signature_new);
 552
 553   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Update result for name %u\n", res);
 554   /* Send response */
 555
 556   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending `%s' message\n", "RECORD_CREATE_RESPONSE");
 557   rcr_msg.gns_header.header.type = htons (GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_CREATE_RESPONSE);
 558   rcr_msg.gns_header.r_id = htonl (crc->op_id);
 559   rcr_msg.gns_header.header.size = htons (sizeof (struct RecordCreateResponseMessage));
 560   if (GNUNET_OK == res)
 561     rcr_msg.op_result = htons (GNUNET_OK);
 562   else
 563     rcr_msg.op_result = htons (GNUNET_NO);
 564   GNUNET_SERVER_notification_context_unicast (snc, crc->nc->client, (const struct GNUNET_MessageHeader *) &rcr_msg, GNUNET_NO);
 565
 566 }
 567
 568 static void handle_record_create (void *cls,
 569                           struct GNUNET_SERVER_Client * client,
 570                           const struct GNUNET_MessageHeader * message)
 571 {
 572   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received `%s' message\n", "NAMESTORE_RECORD_CREATE");
 573   struct GNUNET_NAMESTORE_Client *nc;
 574   struct CreateRecordContext crc;
 575   struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded pub;
 576   GNUNET_HashCode pubkey_hash;
 577   size_t name_len;
 578   size_t msg_size;
 579   size_t msg_size_exp;
 580   size_t rd_ser_len;
 581   size_t key_len;
 582   uint32_t rid = 0;
 583   char *pkey_tmp;
 584   char *name_tmp;
 585   char *rd_ser;
 586   int rd_count;
 587
 588
 589   int res = GNUNET_SYSERR;
 590
 591   if (ntohs (message->size) < sizeof (struct RecordCreateMessage))
 592   {
 593     GNUNET_break_op (0);
 594     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 595     return;
 596   }
 597
 598   nc = client_lookup(client);
 599   if (nc == NULL)
 600   {
 601     GNUNET_break_op (0);
 602     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 603     return;
 604   }
 605
 606   struct RecordCreateMessage * rp_msg = (struct RecordCreateMessage *) message;
 607   rid = ntohl (rp_msg->gns_header.r_id);
 608   name_len = ntohs (rp_msg->name_len);
 609   msg_size = ntohs (message->size);
 610   rd_count = ntohs (rp_msg->rd_count);
 611   rd_ser_len = ntohs (rp_msg->rd_len);
 612   key_len = ntohs (rp_msg->pkey_len);
 613   msg_size_exp = sizeof (struct RecordCreateMessage) + key_len + name_len + rd_ser_len;
 614
 615   if (msg_size != msg_size_exp)
 616   {
 617     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Expected message %u size but message size is %u \n", msg_size_exp, msg_size);
 618     GNUNET_break_op (0);
 619     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 620     return;
 621   }
 622
 623   if ((name_len == 0) || (name_len > 256))
 624   {
 625     GNUNET_break_op (0);
 626     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 627     return;
 628   }
 629
 630   pkey_tmp = (char *) &rp_msg[1];
 631   name_tmp = &pkey_tmp[key_len];
 632   rd_ser = &name_tmp[name_len];
 633
 634   if (name_tmp[name_len -1] != '\0')
 635   {
 636     GNUNET_break_op (0);
 637     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 638     return;
 639   }
 640
 641   struct GNUNET_NAMESTORE_RecordData rd[rd_count];
 642   GNUNET_NAMESTORE_records_deserialize(rd_ser_len, rd_ser, rd_count, rd);
 643   GNUNET_assert (rd_count == 1);
 644
 645   /* Extracting and converting private key */
 646   struct GNUNET_CRYPTO_RsaPrivateKey *pkey = GNUNET_CRYPTO_rsa_decode_key((char *) pkey_tmp, key_len);
 647   GNUNET_assert (pkey != NULL);
 648   GNUNET_CRYPTO_rsa_key_get_public(pkey, &pub);
 649   GNUNET_CRYPTO_hash (&pub, sizeof (struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded), &pubkey_hash);
 650
 651   crc.pkey = pkey;
 652   crc.rd = rd;
 653   crc.nc = nc;
 654   crc.op_id = rid;
 655
 656   /* Get existing records for name */
 657   res = GSN_database->iterate_records(GSN_database->cls, &pubkey_hash, name_tmp, 0, &handle_create_record_it, &crc);
 658
 659   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 660 }
 661
 662
 663
 664 static void handle_record_remove (void *cls,
 665                           struct GNUNET_SERVER_Client * client,
 666                           const struct GNUNET_MessageHeader * message)
 667 {
 668   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received `%s' message\n", "NAMESTORE_RECORD_REMOVE");
 669   struct GNUNET_NAMESTORE_Client *nc;
 670   struct RecordRemoveResponseMessage rrr_msg;
 671   size_t name_len;
 672   size_t msg_size;
 673   size_t msg_size_exp;
 674   uint32_t rid = 0;
 675
 676   int res = GNUNET_SYSERR;
 677
 678   if (ntohs (message->size) < sizeof (struct RecordRemoveMessage))
 679   {
 680     GNUNET_break_op (0);
 681     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 682     return;
 683   }
 684
 685   nc = client_lookup(client);
 686   if (nc == NULL)
 687   {
 688     GNUNET_break_op (0);
 689     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 690     return;
 691   }
 692
 693   struct RecordRemoveMessage * rp_msg = (struct RecordRemoveMessage *) message;
 694   rid = ntohl (rp_msg->gns_header.r_id);
 695   name_len = ntohs (rp_msg->name_len);
 696   msg_size = ntohs (message->size);
 697   msg_size_exp = sizeof (struct RecordRemoveMessage) + name_len + sizeof (struct GNUNET_NAMESTORE_RecordData);
 698
 699   if (msg_size != msg_size_exp)
 700   {
 701     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Expected message %u size but message size is %u \n", msg_size_exp, msg_size);
 702     GNUNET_break_op (0);
 703     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 704     return;
 705   }
 706
 707
 708   if ((name_len == 0) || (name_len > 256))
 709   {
 710     GNUNET_break_op (0);
 711     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 712     return;
 713   }
 714 /*
 715   if (name[name_len -1] != '\0')
 716   {
 717     GNUNET_break_op (0);
 718     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 719     return;
 720   }
 721   */
 722   /* DO WORK HERE */
 723
 724   /* Send response */
 725
 726   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending `%s' message\n", "RECORD_REMOVE_RESPONSE");
 727   rrr_msg.gns_header.header.type = htons (GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_REMOVE_RESPONSE);
 728   rrr_msg.gns_header.r_id = rp_msg->gns_header.r_id;
 729   rrr_msg.gns_header.header.size = htons (sizeof (struct RecordRemoveResponseMessage));
 730   if (GNUNET_OK == res)
 731     rrr_msg.op_result = htons (GNUNET_OK);
 732   else
 733     rrr_msg.op_result = htons (GNUNET_NO);
 734   GNUNET_SERVER_notification_context_unicast (snc, nc->client, (const struct GNUNET_MessageHeader *) &rrr_msg, GNUNET_NO);
 735
 736   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 737 }
 738
 739 struct ZoneIterationProcResult
 740 {
 741   int have_zone_key;
 742   struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded zone_key;
 743
 744   int have_signature;
 745   struct GNUNET_CRYPTO_RsaSignature signature;
 746   struct GNUNET_TIME_Absolute expire;
 747
 748   int have_name;
 749   char name[256];
 750
 751   size_t rd_ser_len;
 752   char *rd_ser;
 753 };
 754
 755
 756 void zone_iteration_proc (void *cls,
 757                          const struct GNUNET_CRYPTO_RsaPublicKeyBinaryEncoded *zone_key,
 758                          struct GNUNET_TIME_Absolute expire,
 759                          const char *name,
 760                          unsigned int rd_count,
 761                          const struct GNUNET_NAMESTORE_RecordData *rd,
 762                          const struct GNUNET_CRYPTO_RsaSignature *signature)
 763 {
 764   struct ZoneIterationProcResult *zipr = cls;
 765   size_t len;
 766   if (zone_key != NULL)
 767   {
 768     zipr->zone_key = *zone_key;
 769     zipr->have_zone_key = GNUNET_YES;
 770   }
 771   else
 772     zipr->have_zone_key = GNUNET_NO;
 773
 774   zipr->expire = expire;
 775
 776   if (name != NULL)
 777   {
 778     memcpy (zipr->name, name, strlen(name) + 1);
 779     zipr->have_name = GNUNET_YES;
 780   }
 781   else
 782     zipr->have_name = GNUNET_NO;
 783
 784   if (signature != NULL)
 785   {
 786     zipr->signature = *signature;
 787     zipr->have_signature = GNUNET_YES;
 788   }
 789   else
 790     zipr->have_signature = GNUNET_NO;
 791
 792   if ((rd_count > 0) && (rd != NULL))
 793   {
 794     len = GNUNET_NAMESTORE_records_get_size(rd_count, rd);
 795     zipr->rd_ser = GNUNET_malloc (len);
 796     GNUNET_NAMESTORE_records_serialize(rd_count, rd, len, zipr->rd_ser);
 797     zipr->rd_ser_len = len;
 798   }
 799 }
 800
 801 static void handle_iteration_start (void *cls,
 802                           struct GNUNET_SERVER_Client * client,
 803                           const struct GNUNET_MessageHeader * message)
 804 {
 805   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received `%s' message\n", "ZONE_ITERATION_START");
 806
 807   struct ZoneIterationStartMessage * zis_msg = (struct ZoneIterationStartMessage *) message;
 808   struct GNUNET_NAMESTORE_Client *nc;
 809   struct GNUNET_NAMESTORE_ZoneIteration *zi;
 810   struct ZoneIterationResponseMessage zir_msg;
 811   struct ZoneIterationProcResult zipr;
 812   int res;
 813
 814   nc = client_lookup(client);
 815   if (nc == NULL)
 816   {
 817     GNUNET_break_op (0);
 818     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 819     return;
 820   }
 821
 822   zi = GNUNET_malloc (sizeof (struct GNUNET_NAMESTORE_ZoneIteration));
 823   zi->request_id = ntohl (zis_msg->gns_header.r_id);
 824   zi->offset = 0;
 825   zi->client = nc;
 826   zi->zone = zis_msg->zone;
 827
 828   GNUNET_CONTAINER_DLL_insert (nc->op_head, nc->op_tail, zi);
 829
 830   res = GSN_database->iterate_records (GSN_database->cls, &zis_msg->zone, NULL, zi->offset , &zone_iteration_proc, &zipr);
 831   switch (res) {
 832     case GNUNET_OK:
 833       /* GNUNET_OK on success */
 834
 835       break;
 836     case GNUNET_SYSERR:
 837       /* GNUNET_SYSERR on error */
 838       break;
 839     case GNUNET_NO:
 840       /* GNUNET_NO if there were no results, */
 841       break;
 842     default:
 843       break;
 844   }
 845
 846
 847
 848   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Sending `%s' message\n", "ZONE_ITERATION_RESPONSE");
 849   zir_msg.gns_header.header.type = htons (GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_ITERATION_RESPONSE);
 850   zir_msg.gns_header.r_id = htonl(zi->request_id);
 851   zir_msg.gns_header.header.size = htons (sizeof (struct ZoneIterationResponseMessage));
 852
 853
 854   GNUNET_SERVER_notification_context_unicast (snc, zi->client->client, (const struct GNUNET_MessageHeader *) &zir_msg, GNUNET_NO);
 855
 856
 857   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 858 }
 859
 860 static void handle_iteration_stop (void *cls,
 861                           struct GNUNET_SERVER_Client * client,
 862                           const struct GNUNET_MessageHeader * message)
 863 {
 864   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received `%s' message\n", "ZONE_ITERATION_STOP");
 865
 866   struct GNUNET_NAMESTORE_Client *nc;
 867   struct GNUNET_NAMESTORE_ZoneIteration *zi;
 868   struct ZoneIterationStopMessage * zis_msg = (struct ZoneIterationStopMessage *) message;
 869   uint32_t rid;
 870
 871   nc = client_lookup(client);
 872   if (nc == NULL)
 873   {
 874     GNUNET_break_op (0);
 875     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 876     return;
 877   }
 878
 879   rid = ntohl (zis_msg->gns_header.r_id);
 880   for (zi = nc->op_head; zi != NULL; zi = zi->next)
 881   {
 882     if (zi->request_id == rid)
 883       break;
 884   }
 885   if (zi == NULL)
 886   {
 887     GNUNET_break_op (0);
 888     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 889     return;
 890   }
 891
 892   GNUNET_CONTAINER_DLL_remove(nc->op_head, nc->op_tail, zi);
 893   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Stopped zone iteration for zone `%s'\n", GNUNET_h2s (&zi->zone));
 894   GNUNET_free (zi);
 895
 896   GNUNET_SERVER_receive_done (client, GNUNET_OK);
 897 }
 898
 899 static void handle_iteration_next (void *cls,
 900                           struct GNUNET_SERVER_Client * client,
 901                           const struct GNUNET_MessageHeader * message)
 902 {
 903   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Received `%s' message\n", "ZONE_ITERATION_NEXT");
 904
 905   struct GNUNET_NAMESTORE_Client *nc;
 906   struct GNUNET_NAMESTORE_ZoneIteration *zi;
 907   struct ZoneIterationStopMessage * zis_msg = (struct ZoneIterationStopMessage *) message;
 908   uint32_t rid;
 909   int res;
 910
 911   nc = client_lookup(client);
 912   if (nc == NULL)
 913   {
 914     GNUNET_break_op (0);
 915     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 916     return;
 917   }
 918
 919   rid = ntohl (zis_msg->gns_header.r_id);
 920   for (zi = nc->op_head; zi != NULL; zi = zi->next)
 921   {
 922     if (zi->request_id == rid)
 923       break;
 924   }
 925   if (zi == NULL)
 926   {
 927     GNUNET_break_op (0);
 928     GNUNET_SERVER_receive_done (client, GNUNET_OK);
 929     return;
 930   }
 931
 932   zi->offset++;
 933   res = GSN_database->iterate_records (GSN_database->cls, &zi->zone, NULL, zi->offset , &zone_iteration_proc, zi);
 934 }
 935
 936
 937
 938 /**
 939  * Process template requests.
 940  *
 941  * @param cls closure
 942  * @param server the initialized server
 943  * @param cfg configuration to use
 944  */
 945 static void
 946 run (void *cls, struct GNUNET_SERVER_Handle *server,
 947      const struct GNUNET_CONFIGURATION_Handle *cfg)
 948 {
 949   char * database;
 950
 951   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Starting namestore service\n");
 952
 953   static const struct GNUNET_SERVER_MessageHandler handlers[] = {
 954     {&handle_start, NULL,
 955      GNUNET_MESSAGE_TYPE_NAMESTORE_START, sizeof (struct StartMessage)},
 956     {&handle_lookup_name, NULL,
 957      GNUNET_MESSAGE_TYPE_NAMESTORE_LOOKUP_NAME, 0},
 958     {&handle_record_put, NULL,
 959     GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_PUT, 0},
 960     {&handle_record_create, NULL,
 961      GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_CREATE, 0},
 962     {&handle_record_remove, NULL,
 963      GNUNET_MESSAGE_TYPE_NAMESTORE_RECORD_REMOVE, 0},
 964     {&handle_iteration_start, NULL,
 965      GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_ITERATION_START, sizeof (struct ZoneIterationStartMessage)},
 966     {&handle_iteration_stop, NULL,
 967      GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_ITERATION_STOP, sizeof (struct ZoneIterationStopMessage)},
 968     {&handle_iteration_next, NULL,
 969      GNUNET_MESSAGE_TYPE_NAMESTORE_ZONE_ITERATION_NEXT, 0},
 970     {NULL, NULL, 0, 0}
 971   };
 972
 973   GSN_cfg = cfg;
 974
 975   /* Loading database plugin */
 976   if (GNUNET_OK !=
 977       GNUNET_CONFIGURATION_get_value_string (cfg, "namestore", "database",
 978                                              &database))
 979     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "No database backend configured\n");
 980
 981   GNUNET_asprintf (&db_lib_name, "libgnunet_plugin_namestore_%s", database);
 982   GSN_database = GNUNET_PLUGIN_load (db_lib_name, (void *) GSN_cfg);
 983   if (GSN_database == NULL)
 984     GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Could not load database backend `%s'\n",
 985         db_lib_name);
 986   GNUNET_free (database);
 987
 988   /* Configuring server handles */
 989   GNUNET_SERVER_add_handlers (server, handlers);
 990   snc = GNUNET_SERVER_notification_context_create (server, 16);
 991   GNUNET_SERVER_disconnect_notify (server,
 992                                    &client_disconnect_notification,
 993                                    NULL);
 994
 995   GNUNET_SCHEDULER_add_delayed (GNUNET_TIME_UNIT_FOREVER_REL, &cleanup_task,
 996                                 NULL);
 997
 998 }
 999
1000
1001 /**
1002  * The main function for the template service.
1003  *
1004  * @param argc number of arguments from the command line
1005  * @param argv command line arguments
1006  * @return 0 ok, 1 on error
1007  */
1008 int
1009 main (int argc, char *const *argv)
1010 {
1011   return (GNUNET_OK ==
1012           GNUNET_SERVICE_run (argc, argv, "namestore",
1013                               GNUNET_SERVICE_OPTION_NONE, &run, NULL)) ? 0 : 1;
1014 }
1015
1016 /* end of gnunet-service-namestore.c */