implemented seeding

3 files changed, 75 insertions, 22 deletions

diff --git a/src/rps/gnunet-service-rps.c b/src/rps/gnunet-service-rps.c
index 2a2af94f4..0896e126b 100644
--- a/src/rps/gnunet-service-rps.c
+++ b/src/rps/gnunet-service-rps.c
@@ -190,7 +190,7 @@ static unsigned int sampler_size;
  * The size of sampler we need to be able to satisfy the client's need of
  * random peers.
  */
-static unsigned int sampler_size_client_need;
+//static unsigned int sampler_size_client_need;
 
 
 /**
@@ -526,8 +526,7 @@ nse_callback(void *cls, struct GNUNET_TIME_Absolute timestamp, double logestimat
  * @param message the actual message
  */
 static void
-// TODO rename
-handle_cs_request (void *cls,
+handle_client_request (void *cls,
             struct GNUNET_SERVER_Client *client,
             const struct GNUNET_MessageHeader *message)
 {
@@ -595,6 +594,49 @@ handle_cs_request (void *cls,
                               GNUNET_OK);
 }
 
+
+/**
+ * Handle seed from the client.
+ *
+ * @param cls closure
+ * @param client identification of the client
+ * @param message the actual message
+ */
+  static void
+handle_client_seed (void *cls,
+            struct GNUNET_SERVER_Client *client,
+            const struct GNUNET_MessageHeader *message)
+{
+  struct GNUNET_RPS_CS_SeedMessage *in_msg;
+  struct GNUNET_PeerIdentity *peers;
+  uint64_t i;
+
+  if (sizeof (struct GNUNET_RPS_CS_SeedMessage) < ntohs (message->size))
+  {
+    GNUNET_break_op (0);
+    GNUNET_SERVER_receive_done (client,
+              GNUNET_SYSERR);
+  }
+  in_msg = (struct GNUNET_RPS_CS_SeedMessage *) message;
+  if (ntohs (message->size) - sizeof (struct GNUNET_RPS_CS_SeedMessage) /
+      sizeof (struct GNUNET_PeerIdentity) != GNUNET_ntohll (in_msg->num_peers))
+  {
+    GNUNET_break_op (0);
+    GNUNET_SERVER_receive_done (client,
+              GNUNET_SYSERR);
+  }
+
+  in_msg = (struct GNUNET_RPS_CS_SeedMessage *) message;
+  peers = (struct GNUNET_PeerIdentity *) &message[1];
+
+  for ( i = 0 ; i < GNUNET_ntohll (in_msg->num_peers) ; i++ )
+    RPS_sampler_update_list (&peers[i]);
+
+  GNUNET_SERVER_receive_done (client,
+                              GNUNET_OK);
+}
+
+
 /**
  * Handle a PUSH message from another peer.
  *
@@ -1100,8 +1142,9 @@ static void
 rps_start (struct GNUNET_SERVER_Handle *server)
 {
   static const struct GNUNET_SERVER_MessageHandler handlers[] = {
-    {&handle_cs_request, NULL, GNUNET_MESSAGE_TYPE_RPS_CS_REQUEST,
+    {&handle_client_request, NULL, GNUNET_MESSAGE_TYPE_RPS_CS_REQUEST,
       sizeof (struct GNUNET_RPS_CS_RequestMessage)},
+    {&handle_client_seed,    NULL, GNUNET_MESSAGE_TYPE_RPS_CS_SEED, 0},
     {NULL, NULL, 0, 0}
   };
 
diff --git a/src/rps/rps.h b/src/rps/rps.h
index 77ce2f18f..3ba0cffcf 100644
--- a/src/rps/rps.h
+++ b/src/rps/rps.h
@@ -150,12 +150,9 @@ struct GNUNET_RPS_CS_SeedMessage
   /**
    * Number of peers
    */
-  uint64_t n;
+  uint64_t num_peers;
 
-  /**
-   * Peers
-   */
-  struct GNUNET_PeerIdentity ids;
+  /* Followed by num_peers * GNUNET_PeerIdentity */
 };
 
 GNUNET_NETWORK_STRUCT_END
diff --git a/src/rps/rps_api.c b/src/rps/rps_api.c
index ee00db911..9d6183103 100644
--- a/src/rps/rps_api.c
+++ b/src/rps/rps_api.c
@@ -108,8 +108,6 @@ struct cb_cls_pack
 };
 
 
-
-
 /**
  * This function is called, when the service replies to our request.
  * It calls the callback the caller gave us with the provided closure
@@ -141,9 +139,16 @@ handle_reply (void *cls,
 }
 
 /**
+ * Error handler for mq.
+ *
+ * This function is called whan mq encounters an error.
+ * Until now mq doesn't provide useful error messages.
+ *
+ * @param cls the closure
+ * @param error error code without specyfied meaning
  */
   static void
-mq_error_handler(void *cls, enum GNUNET_MQ_Error error)
+mq_error_handler (void *cls, enum GNUNET_MQ_Error error)
 {
   //TODO LOG
 }
@@ -155,7 +160,7 @@ mq_error_handler(void *cls, enum GNUNET_MQ_Error error)
  * @return a handle to the service
  */
   struct GNUNET_RPS_Handle *
-GNUNET_RPS_connect( const struct GNUNET_CONFIGURATION_Handle *cfg )
+GNUNET_RPS_connect (const struct GNUNET_CONFIGURATION_Handle *cfg)
 {
   struct GNUNET_RPS_Handle *h;
   //struct GNUNET_RPS_Request_Handle *rh;
@@ -197,19 +202,19 @@ GNUNET_RPS_request_peers (struct GNUNET_RPS_Handle *h, uint64_t n,
   struct GNUNET_RPS_CS_RequestMessage *msg;
 
   // assert func != NULL
-  rh = GNUNET_new(struct GNUNET_RPS_Request_Handle);
+  rh = GNUNET_new (struct GNUNET_RPS_Request_Handle);
   rh->h = h;
   rh->n = req_handlers_size; // TODO ntoh
   rh->ready_cb = ready_cb;
   rh->ready_cb_cls = cls;
 
-  GNUNET_array_append(req_handlers, req_handlers_size, *rh);
+  GNUNET_array_append (req_handlers, req_handlers_size, *rh);
   //memcpy(&req_handlers[req_handlers_size-1], rh, sizeof(struct GNUNET_RPS_Request_Handle));
 
-  ev = GNUNET_MQ_msg(msg, GNUNET_MESSAGE_TYPE_RPS_CS_REQUEST);
-  msg->num_peers = GNUNET_htonll(n);
+  ev = GNUNET_MQ_msg (msg, GNUNET_MESSAGE_TYPE_RPS_CS_REQUEST);
+  msg->num_peers = GNUNET_htonll (n);
   msg->n = rh->n;
-  GNUNET_MQ_send(h->mq, ev);
+  GNUNET_MQ_send (h->mq, ev);
   return rh;
 }
 
@@ -222,8 +227,16 @@ GNUNET_RPS_request_peers (struct GNUNET_RPS_Handle *h, uint64_t n,
  */
   void
 GNUNET_RPS_seed_ids (struct GNUNET_RPS_Handle *h, uint64_t n,
-                     struct GNUNET_PeerIdentity * ids)
+                     const struct GNUNET_PeerIdentity * ids)
 {
+  struct GNUNET_MQ_Envelope *ev;
+  struct GNUNET_RPS_CS_SeedMessage *msg;
+
+  ev = GNUNET_MQ_msg_extra (msg, n * sizeof (struct GNUNET_PeerIdentity),
+                            GNUNET_MESSAGE_TYPE_RPS_CS_SEED);
+  msg->num_peers = GNUNET_htonll (n);
+  memcpy (&msg[1], ids, n * sizeof (struct GNUNET_PeerIdentity));
+  GNUNET_MQ_send (h->mq, ev);
 }
 
 /**
@@ -232,7 +245,7 @@ GNUNET_RPS_seed_ids (struct GNUNET_RPS_Handle *h, uint64_t n,
  * @param rh request handle of request to cancle
  */
   void
-GNUNET_RPS_request_cancel ( struct GNUNET_RPS_Request_Handle *rh )
+GNUNET_RPS_request_cancel (struct GNUNET_RPS_Request_Handle *rh)
 {
   // TODO
 }
@@ -243,10 +256,10 @@ GNUNET_RPS_request_cancel ( struct GNUNET_RPS_Request_Handle *rh )
  * @param h the handle to the rps service
  */
   void
-GNUNET_RPS_disconnect ( struct GNUNET_RPS_Handle *h )
+GNUNET_RPS_disconnect (struct GNUNET_RPS_Handle *h)
 {
   if ( NULL != h->conn ) {
-    GNUNET_CLIENT_disconnect(h->conn);
+    GNUNET_CLIENT_disconnect (h->conn);
   }
 }