X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Fregex%2Fgnunet-service-regex.c;h=96a6b82b1e39ccabbc4a62a7dfb0a34a6f3834fc;hb=e6f363e48edb537b36955cb31b42ab71eeba486c;hp=28f3a20b8ab894721f481c470924c9d496cc5a98;hpb=a310c0beee76349ea55f314a7b9736ed9a7a1448;p=oweals%2Fgnunet.git

diff --git a/src/regex/gnunet-service-regex.c b/src/regex/gnunet-service-regex.c
index 28f3a20b8..96a6b82b1 100644
--- a/src/regex/gnunet-service-regex.c
+++ b/src/regex/gnunet-service-regex.c
@@ -26,7 +26,7 @@
  */
 #include "platform.h"
 #include "gnunet_util_lib.h"
-#include "gnunet_regex_lib.h"
+#include "regex_internal_lib.h"
 #include "regex_ipc.h"
 
 
@@ -54,12 +54,12 @@ struct ClientEntry
   /**
    * Search handle (if this client is searching).
    */
-  struct GNUNET_REGEX_search_handle *sh;
+  struct REGEX_INTERNAL_Search *sh;
 
   /**
    * Announcement handle (if this client is announcing).
    */
-  struct GNUNET_REGEX_announce_handle *ah;
+  struct REGEX_INTERNAL_Announcement *ah;
 
   /**
    * Refresh frequency for announcements.
@@ -99,6 +99,11 @@ static struct ClientEntry *client_tail;
  */
 static struct GNUNET_SERVER_NotificationContext *nc;
 
+/**
+ * Private key for this peer.
+ */
+static struct GNUNET_CRYPTO_EccPrivateKey *my_private_key;
+
 
 /**
  * Task run during shutdown.
@@ -115,6 +120,8 @@ cleanup_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
   stats = NULL;
   GNUNET_SERVER_notification_context_destroy (nc);
   nc = NULL;
+  GNUNET_CRYPTO_ecc_key_free (my_private_key);
+  my_private_key = NULL;
 }
 
 
@@ -143,12 +150,12 @@ handle_client_disconnect (void *cls, struct GNUNET_SERVER_Client *client)
       }
       if (NULL != ce->ah)
       {
-	GNUNET_REGEX_announce_cancel (ce->ah);
+	REGEX_INTERNAL_announce_cancel (ce->ah);
 	ce->ah = NULL;
       }
       if (NULL != ce->sh)
       {
-	GNUNET_REGEX_search_cancel (ce->sh);
+	REGEX_INTERNAL_search_cancel (ce->sh);
 	ce->sh = NULL;
       }
       GNUNET_CONTAINER_DLL_remove (client_head, client_tail, ce);
@@ -171,7 +178,7 @@ reannounce (void *cls,
 {
   struct ClientEntry *ce = cls;
 
-  GNUNET_REGEX_reannounce (ce->ah);
+  REGEX_INTERNAL_reannounce (ce->ah);
   ce->refresh_task = GNUNET_SCHEDULER_add_delayed (ce->frequency,
 						   &reannounce,
 						   ce);
@@ -207,11 +214,11 @@ handle_announce (void *cls,
   }
   ce = GNUNET_new (struct ClientEntry);
   ce->client = client;
-  ce->ah = GNUNET_REGEX_announce (dht,
-				  &am->pid,
-				  regex,
-				  ntohs (am->compression),
-				  stats);
+  ce->ah = REGEX_INTERNAL_announce (dht,
+				    my_private_key,
+				    regex,
+				    ntohs (am->compression),
+				    stats);
   if (NULL == ce->ah)
   {
     GNUNET_break (0);
@@ -311,7 +318,7 @@ handle_search (void *cls,
   }
   ce = GNUNET_new (struct ClientEntry);
   ce->client = client;
-  ce->sh = GNUNET_REGEX_search (dht,
+  ce->sh = REGEX_INTERNAL_search (dht,
 				string,
 				&handle_search_result,
 				ce,
@@ -347,9 +354,18 @@ run (void *cls, struct GNUNET_SERVER_Handle *server,
     {&handle_search, NULL, GNUNET_MESSAGE_TYPE_REGEX_SEARCH, 0},
     {NULL, NULL, 0, 0}
   };
+
+  my_private_key = GNUNET_CRYPTO_ecc_key_create_from_configuration (cfg);
+  if (NULL == my_private_key)
+  {   
+    GNUNET_SCHEDULER_shutdown ();
+    return;
+  }
   dht = GNUNET_DHT_connect (cfg, 1024);
   if (NULL == dht)
   {
+    GNUNET_CRYPTO_ecc_key_free (my_private_key);
+    my_private_key = NULL;
     GNUNET_SCHEDULER_shutdown ();
     return;
   }