gnunet-svn
[Top][All Lists]
Advanced

[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]

[GNUnet-SVN] r12504 - gnunet/src/dht


From: gnunet
Subject: [GNUnet-SVN] r12504 - gnunet/src/dht
Date: Mon, 9 Aug 2010 12:42:26 +0200

Author: nevans
Date: 2010-08-09 12:42:26 +0200 (Mon, 09 Aug 2010)
New Revision: 12504

Modified:
   gnunet/src/dht/dht_api.c
   gnunet/src/dht/dhtlog.c
   gnunet/src/dht/gnunet-dht-driver.c
   gnunet/src/dht/gnunet-service-dht.c
   gnunet/src/dht/plugin_dhtlog_mysql.c
Log:
mosgly fixes for memory leaks, misunderstanding of some mysql library stuff

Modified: gnunet/src/dht/dht_api.c
===================================================================
--- gnunet/src/dht/dht_api.c    2010-08-08 20:35:49 UTC (rev 12503)
+++ gnunet/src/dht/dht_api.c    2010-08-09 10:42:26 UTC (rev 12504)
@@ -74,6 +74,13 @@
    */
   uint64_t unique_id;
 
+  /**
+   * Free the saved message once sent, set
+   * to GNUNET_YES for messages that don't
+   * receive responses!
+   */
+  int free_on_send;
+
 };
 
 struct PendingMessageList
@@ -366,7 +373,7 @@
     }
 
   GNUNET_assert(handle->th == NULL);
-  if (pos->unique_id == 0)
+  if (pos->free_on_send == GNUNET_YES)
     GNUNET_free(pos->msg);
   GNUNET_free (pos);
   handle->current = NULL;
@@ -896,6 +903,8 @@
   pending = GNUNET_malloc (sizeof (struct PendingMessage));
   pending->msg = &message->header;
   pending->timeout = timeout;
+  if (iter == NULL)
+    pending->free_on_send = GNUNET_YES;
   pending->cont = cont;
   pending->cont_cls = cont_cls;
   pending->unique_id = route_handle->uid;
@@ -1215,7 +1224,9 @@
         }
     }
   else
-    GNUNET_free(put_route);
+    {
+      GNUNET_free(put_route);
+    }
 
   GNUNET_free (put_msg);
 }

Modified: gnunet/src/dht/dhtlog.c
===================================================================
--- gnunet/src/dht/dhtlog.c     2010-08-08 20:35:49 UTC (rev 12503)
+++ gnunet/src/dht/dhtlog.c     2010-08-09 10:42:26 UTC (rev 12504)
@@ -69,6 +69,7 @@
     }
 
   api = plugin->dhtlog_api;
+  GNUNET_free (plugin_name);
   GNUNET_free(plugin);
   return api;
 }
@@ -87,6 +88,7 @@
     {
       GNUNET_PLUGIN_unload(libname, api);
     }
+  GNUNET_free_non_null(libname);
 }
 
 /* end of dhtlog.c */

Modified: gnunet/src/dht/gnunet-dht-driver.c
===================================================================
--- gnunet/src/dht/gnunet-dht-driver.c  2010-08-08 20:35:49 UTC (rev 12503)
+++ gnunet/src/dht/gnunet-dht-driver.c  2010-08-09 10:42:26 UTC (rev 12504)
@@ -489,7 +489,8 @@
       GNUNET_assert(dhtlog_handle != NULL);
       fprintf(stderr, "topology iteration finished (%u connections), 
scheduling continuation\n", topo_ctx->total_connections);
       dhtlog_handle->update_topology(topo_ctx->total_connections);
-      GNUNET_SCHEDULER_add_now (sched, topo_ctx->cont, topo_ctx->cls);
+      if (topo_ctx->cont != NULL)
+        GNUNET_SCHEDULER_add_now (sched, topo_ctx->cont, topo_ctx->cls);
       GNUNET_free(topo_ctx);
     }
 }
@@ -812,6 +813,7 @@
   struct TestGetContext *test_get;
   int remember[num_puts][num_peers];
 
+  memset(&remember, 0, sizeof(int) * num_puts * num_peers);
   for (i = 0; i < num_puts; i++)
     {
       test_put = GNUNET_malloc(sizeof(struct TestPutContext));
@@ -849,9 +851,16 @@
 static void
 continue_puts_and_gets (void *cls, const struct GNUNET_SCHEDULER_TaskContext * 
tc)
 {
+  int i;
   struct TopologyIteratorContext *topo_ctx;
   if (dhtlog_handle != NULL)
     {
+      for (i = 1; i < (settle_time / 60)  - 2; i++)
+        {
+          topo_ctx = GNUNET_malloc(sizeof(struct TopologyIteratorContext));
+          fprintf(stderr, "scheduled topology iteration in %d minutes\n", i);
+          GNUNET_SCHEDULER_add_delayed(sched, 
GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_MINUTES, i), 
&capture_current_topology, topo_ctx);
+        }
       topo_ctx = GNUNET_malloc(sizeof(struct TopologyIteratorContext));
       topo_ctx->cont = &setup_puts_and_gets;
       GNUNET_SCHEDULER_add_delayed(sched, 
GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, settle_time), 
&capture_current_topology, topo_ctx);
@@ -1008,6 +1017,7 @@
       die_task = GNUNET_SCHEDULER_add_now (sched,
                                            &end_badly, "from create topology 
(bad return)");
     }
+  GNUNET_free_non_null(blacklist_transports);
   GNUNET_SCHEDULER_cancel (sched, die_task);
   die_task = GNUNET_SCHEDULER_add_delayed (sched,
                                            
GNUNET_TIME_relative_multiply(seconds_per_peer_start, num_peers),
@@ -1146,6 +1156,8 @@
         return;
       }
 
+    GNUNET_free_non_null(hostfile);
+
     buf = data;
     count = 0;
     while (count < frstat.st_size)
@@ -1254,6 +1266,7 @@
   /**
    * Get testing related options.
    */
+  topology_str = NULL;
   if ((GNUNET_YES ==
       GNUNET_CONFIGURATION_get_value_string(cfg, "testing", "topology",
                                             &topology_str)) && (GNUNET_NO == 
GNUNET_TESTING_topology_get(&topology, topology_str)))
@@ -1270,6 +1283,7 @@
   else
     {
       topology_percentage = atof (topology_percentage_str);
+      GNUNET_free(topology_percentage_str);
     }
 
   if (GNUNET_OK !=
@@ -1279,6 +1293,7 @@
   else
     {
      topology_probability = atof (topology_probability_str);
+     GNUNET_free(topology_probability_str);
     }
 
   if ((GNUNET_YES ==
@@ -1289,6 +1304,7 @@
                   "Invalid connect topology `%s' given for section %s option 
%s\n", connect_topology_str, "TESTING", "CONNECT_TOPOLOGY");
     }
   GNUNET_free_non_null(connect_topology_str);
+
   if ((GNUNET_YES ==
       GNUNET_CONFIGURATION_get_value_string(cfg, "testing", 
"connect_topology_option",
                                             &connect_topology_option_str)) && 
(GNUNET_NO == GNUNET_TESTING_topology_option_get(&connect_topology_option, 
connect_topology_option_str)))
@@ -1298,6 +1314,7 @@
       connect_topology_option = GNUNET_TESTING_TOPOLOGY_OPTION_ALL; /* 
Defaults to NONE, set to ALL */
     }
   GNUNET_free_non_null(connect_topology_option_str);
+
   if (GNUNET_YES ==
         GNUNET_CONFIGURATION_get_value_string (cfg, "testing", 
"connect_topology_option_modifier",
                                                
&connect_topology_option_modifier_string))
@@ -1372,6 +1389,8 @@
                                     malicious_droppers, "");
     }
 
+  GNUNET_free_non_null(trialmessage);
+
   hostkey_meter = create_meter(peers_left, "Hostkeys created ", GNUNET_YES);
   peer_start_meter = create_meter(peers_left, "Peers started ", GNUNET_YES);
 

Modified: gnunet/src/dht/gnunet-service-dht.c
===================================================================
--- gnunet/src/dht/gnunet-service-dht.c 2010-08-08 20:35:49 UTC (rev 12503)
+++ gnunet/src/dht/gnunet-service-dht.c 2010-08-09 10:42:26 UTC (rev 12504)
@@ -72,6 +72,8 @@
 
 #define DHT_DEFAULT_FIND_PEER_INTERVAL 
GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_MINUTES, 1)
 
+#define DHT_DEFAULT_PING_DELAY 
GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_MINUTES, 1)
+
 /**
  * Real maximum number of hops, at which point we refuse
  * to forward the message.
@@ -170,6 +172,11 @@
    */
   unsigned int distance;
 
+  /**
+   * Task for scheduling periodic ping messages for this peer.
+   */
+  GNUNET_SCHEDULER_TaskIdentifier ping_task;
+
 };
 
 /**
@@ -963,8 +970,6 @@
 {
   struct P2PPendingMessage *pos;
   struct P2PPendingMessage *next;
-  //fprintf(stderr, "BEFORE REMOVAL\n");
-  //print_routing_table();
 #if EXTRA_CHECKS
   struct PeerInfo *peer_pos;
 
@@ -994,10 +999,8 @@
     }
 
   GNUNET_assert(GNUNET_CONTAINER_multihashmap_contains(all_known_peers, 
&peer->id.hashPubKey));
-  GNUNET_assert(GNUNET_YES == GNUNET_CONTAINER_multihashmap_remove 
(all_known_peers, &peer->id.hashPubKey, peer));
+  GNUNET_CONTAINER_multihashmap_remove (all_known_peers, &peer->id.hashPubKey, 
peer);
   GNUNET_free(peer);
-  //fprintf(stderr, "AFTER REMOVAL\n");
-  //print_routing_table();
 }
 
 
@@ -1056,6 +1059,7 @@
 
   /* Remove peers from lowest bucket, insert into next lowest bucket */
   GNUNET_CONTAINER_multihashmap_iterate(to_remove, &move_lowest_bucket, NULL);
+  GNUNET_CONTAINER_multihashmap_destroy(to_remove);
   lowest_bucket = lowest_bucket - 1;
 #if PRINT_TABLES
   fprintf(stderr, "Printing RT after new bucket\n");
@@ -1063,8 +1067,113 @@
 #endif
 }
 
+/**
+ * Function called to send a request out to another peer.
+ * Called both for locally initiated requests and those
+ * received from other peers.
+ *
+ * @param cls DHT service closure argument (unused)
+ * @param msg the encapsulated message
+ * @param peer the peer to forward the message to
+ * @param msg_ctx the context of the message (hop count, bloom, etc.)
+ */
+static void forward_message (void *cls,
+                             const struct GNUNET_MessageHeader *msg,
+                             struct PeerInfo *peer,
+                             struct DHT_MessageContext *msg_ctx)
+{
+  struct GNUNET_DHT_P2PRouteMessage *route_message;
+  struct P2PPendingMessage *pending;
+  size_t msize;
+  size_t psize;
 
+  msize = sizeof (struct GNUNET_DHT_P2PRouteMessage) + ntohs(msg->size);
+  GNUNET_assert(msize <= GNUNET_SERVER_MAX_MESSAGE_SIZE);
+  psize = sizeof(struct P2PPendingMessage) + msize;
+  pending = GNUNET_malloc(psize);
+  pending->msg = (struct GNUNET_MessageHeader *)&pending[1];
+  pending->importance = DHT_SEND_PRIORITY;
+  pending->timeout = GNUNET_TIME_relative_get_forever();
+  route_message = (struct GNUNET_DHT_P2PRouteMessage *)pending->msg;
+  route_message->header.size = htons(msize);
+  route_message->header.type = htons(GNUNET_MESSAGE_TYPE_DHT_P2P_ROUTE);
+  route_message->options = htonl(msg_ctx->msg_options);
+  route_message->hop_count = htonl(msg_ctx->hop_count + 1);
+  route_message->network_size = htonl(msg_ctx->network_size);
+  route_message->desired_replication_level = htonl(msg_ctx->replication);
+  route_message->unique_id = GNUNET_htonll(msg_ctx->unique_id);
+  if (msg_ctx->bloom != NULL)
+    GNUNET_assert(GNUNET_OK == 
GNUNET_CONTAINER_bloomfilter_get_raw_data(msg_ctx->bloom, 
route_message->bloomfilter, DHT_BLOOM_SIZE));
+  if (msg_ctx->key != NULL)
+    memcpy(&route_message->key, msg_ctx->key, sizeof(GNUNET_HashCode));
+  memcpy(&route_message[1], msg, ntohs(msg->size));
+#if DEBUG_DHT > 1
+  GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "%s:%s Adding pending message size %d 
for peer %s\n", my_short_id, "DHT", msize, GNUNET_i2s(&peer->id));
+#endif
+  GNUNET_CONTAINER_DLL_insert_after(peer->head, peer->tail, peer->tail, 
pending);
+  if (peer->send_task == GNUNET_SCHEDULER_NO_TASK)
+    peer->send_task = GNUNET_SCHEDULER_add_now(sched, &try_core_send, peer);
+}
+
+#if DO_PING
 /**
+ * Task used to send ping messages to peers so that
+ * they don't get disconnected.
+ *
+ * @param cls the peer to send a ping message to
+ * @param tc context, reason, etc.
+ */
+static void
+periodic_ping_task (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  struct PeerInfo *peer = cls;
+  struct GNUNET_MessageHeader ping_message;
+  struct DHT_MessageContext message_context;
+
+  if (tc->reason == GNUNET_SCHEDULER_REASON_SHUTDOWN)
+    return;
+
+  ping_message.size = htons(sizeof(struct GNUNET_MessageHeader));
+  ping_message.type = htons(GNUNET_MESSAGE_TYPE_DHT_P2P_PING);
+
+  memset(&message_context, 0, sizeof(struct DHT_MessageContext));
+#if DEBUG_PING
+  GNUNET_log(GNUNET_ERROR_TYPE_WARNING, "%s:%s Sending periodic ping to %s\n", 
my_short_id, "DHT", GNUNET_i2s(&peer->id));
+#endif
+  forward_message(NULL, &ping_message, peer, &message_context);
+  peer->ping_task = GNUNET_SCHEDULER_add_delayed(sched, 
DHT_DEFAULT_PING_DELAY, &periodic_ping_task, peer);
+}
+
+/**
+ * Schedule PING messages for the top X peers in each
+ * bucket of the routing table (so core won't disconnect them!)
+ */
+void schedule_ping_messages()
+{
+  unsigned int bucket;
+  unsigned int count;
+  struct PeerInfo *pos;
+  for (bucket = lowest_bucket; bucket < MAX_BUCKETS; bucket++)
+    {
+      pos = k_buckets[bucket].head;
+      count = 0;
+      while (pos != NULL)
+        {
+          if ((count < bucket_size) && (pos->ping_task == 
GNUNET_SCHEDULER_NO_TASK))
+            GNUNET_SCHEDULER_add_now(sched, &periodic_ping_task, pos);
+          else if ((count >= bucket_size) && (pos->ping_task != 
GNUNET_SCHEDULER_NO_TASK))
+            {
+              GNUNET_SCHEDULER_cancel(sched, pos->ping_task);
+              pos->ping_task = GNUNET_SCHEDULER_NO_TASK;
+            }
+          pos = pos->next;
+          count++;
+        }
+    }
+}
+#endif
+
+/**
  * Attempt to add a peer to our k-buckets.
  *
  * @param peer, the peer identity of the peer being added
@@ -1089,7 +1198,9 @@
 
   if ((k_buckets[lowest_bucket].peers_size) >= bucket_size)
     enable_next_bucket();
-
+#if DO_PING
+  schedule_ping_messages();
+#endif
   return new_peer;
 }
 
@@ -1265,11 +1376,6 @@
   /**
    * If a find peer result message is received and contains a valid
    * HELLO for another peer, offer it to the transport service.
-   *
-   * FIXME: Check whether we need this peer (based on routing table
-   * fullness) and only try to connect to it conditionally.  This should
-   * reduce trying to connect to say (500) peers when the bucket size will
-   * discard most of them.
    */
   if (ntohs(msg->type) == GNUNET_MESSAGE_TYPE_DHT_FIND_PEER_RESULT)
     {
@@ -1580,6 +1686,7 @@
   new_msg_ctx->bloom = GNUNET_CONTAINER_bloomfilter_init (NULL, 
DHT_BLOOM_SIZE, DHT_BLOOM_K);
   new_msg_ctx->hop_count = 0;
   route_result_message(cls, find_peer_result, new_msg_ctx);
+  GNUNET_free(new_msg_ctx);
 #if DEBUG_DHT_ROUTING
   if ((debug_routes) && (dhtlog_handle != NULL))
     {
@@ -1789,9 +1896,10 @@
         return GNUNET_NO;
       else if (other_bits == bits) /* We match the same number of bits, do 
distance comparison */
         {
+          return GNUNET_YES;
           /* FIXME: why not just return GNUNET_YES here?  We are certainly 
close. */
-          if (distance(&pos->id.hashPubKey, target) < my_distance)
-            return GNUNET_NO;
+          /*if (distance(&pos->id.hashPubKey, target) < my_distance)
+            return GNUNET_NO;*/
         }
       pos = pos->next;
     }
@@ -1944,52 +2052,7 @@
 #endif
 }
 
-/**
- * Function called to send a request out to another peer.
- * Called both for locally initiated requests and those
- * received from other peers.
- *
- * @param cls DHT service closure argument
- * @param msg the encapsulated message
- * @param peer the peer to forward the message to
- * @param msg_ctx the context of the message (hop count, bloom, etc.)
- */
-static void forward_message (void *cls,
-                             const struct GNUNET_MessageHeader *msg,
-                             struct PeerInfo *peer,
-                             struct DHT_MessageContext *msg_ctx)
-{
-  struct GNUNET_DHT_P2PRouteMessage *route_message;
-  struct P2PPendingMessage *pending;
-  size_t msize;
-  size_t psize;
 
-  msize = sizeof (struct GNUNET_DHT_P2PRouteMessage) + ntohs(msg->size);
-  GNUNET_assert(msize <= GNUNET_SERVER_MAX_MESSAGE_SIZE);
-  psize = sizeof(struct P2PPendingMessage) + msize;
-  pending = GNUNET_malloc(psize);
-  pending->msg = (struct GNUNET_MessageHeader *)&pending[1];
-  pending->importance = DHT_SEND_PRIORITY;
-  pending->timeout = GNUNET_TIME_relative_get_forever();
-  route_message = (struct GNUNET_DHT_P2PRouteMessage *)pending->msg;
-  route_message->header.size = htons(msize);
-  route_message->header.type = htons(GNUNET_MESSAGE_TYPE_DHT_P2P_ROUTE);
-  route_message->options = htonl(msg_ctx->msg_options);
-  route_message->hop_count = htonl(msg_ctx->hop_count + 1);
-  route_message->network_size = htonl(msg_ctx->network_size);
-  route_message->desired_replication_level = htonl(msg_ctx->replication);
-  route_message->unique_id = GNUNET_htonll(msg_ctx->unique_id);
-  GNUNET_assert(GNUNET_OK == 
GNUNET_CONTAINER_bloomfilter_get_raw_data(msg_ctx->bloom, 
route_message->bloomfilter, DHT_BLOOM_SIZE));
-  memcpy(&route_message->key, msg_ctx->key, sizeof(GNUNET_HashCode));
-  memcpy(&route_message[1], msg, ntohs(msg->size));
-#if DEBUG_DHT > 1
-  GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "%s:%s Adding pending message size %d 
for peer %s\n", my_short_id, "DHT", msize, GNUNET_i2s(&peer->id));
-#endif
-  GNUNET_CONTAINER_DLL_insert_after(peer->head, peer->tail, peer->tail, 
pending);
-  if (peer->send_task == GNUNET_SCHEDULER_NO_TASK)
-    peer->send_task = GNUNET_SCHEDULER_add_now(sched, &try_core_send, peer);
-}
-
 /**
  * Task used to remove forwarding entries, either
  * after timeout, when full, or on shutdown.
@@ -2288,6 +2351,7 @@
   struct GNUNET_MessageHeader *find_peer_msg;
   struct DHT_MessageContext message_context;
   int ret;
+  struct GNUNET_TIME_Relative next_send_time;
 
   if (tc->reason == GNUNET_SCHEDULER_REASON_SHUTDOWN)
     return;
@@ -2304,12 +2368,14 @@
   message_context.peer = &my_identity;
 
   ret = route_message(NULL, find_peer_msg, &message_context);
-
+  GNUNET_free(find_peer_msg);
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
               "`%s:%s': Sent `%s' request to %d peers\n", my_short_id, "DHT",
               "FIND PEER", ret);
+  next_send_time = DHT_DEFAULT_FIND_PEER_INTERVAL;
+  next_send_time.value = 
GNUNET_CRYPTO_random_u64(GNUNET_CRYPTO_QUALITY_STRONG, next_send_time.value * 
3);
   GNUNET_SCHEDULER_add_delayed (sched,
-                                DHT_DEFAULT_FIND_PEER_INTERVAL,
+                                next_send_time,
                                 &send_find_peer_message, NULL);
 }
 
@@ -2423,6 +2489,14 @@
   struct GNUNET_MessageHeader *enc_msg = (struct GNUNET_MessageHeader 
*)&incoming[1];
   struct DHT_MessageContext *message_context;
 
+  if (ntohs(enc_msg->type) == GNUNET_MESSAGE_TYPE_DHT_P2P_PING) /* Throw these 
away. FIXME: Don't throw these away? (reply)*/
+    {
+#if DEBUG_PING
+      GNUNET_log(GNUNET_ERROR_TYPE_DEBUG, "%s:%s Received P2P Ping 
message.\n", my_short_id, "DHT");
+#endif
+      return GNUNET_YES;
+    }
+
   if (ntohs(enc_msg->size) > GNUNET_SERVER_MAX_MESSAGE_SIZE)
     {
       GNUNET_break_op(0);
@@ -2638,7 +2712,9 @@
                      latency,
                      distance);
   if (ret != NULL)
-    GNUNET_CONTAINER_multihashmap_put(all_known_peers, &peer->hashPubKey, ret, 
GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
+    {
+      GNUNET_CONTAINER_multihashmap_put(all_known_peers, &peer->hashPubKey, 
ret, GNUNET_CONTAINER_MULTIHASHMAPOPTION_UNIQUE_ONLY);
+    }
 #if DEBUG_DHT
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
                 "%s:%s Adding peer to routing list: %s\n", my_short_id, "DHT", 
ret == NULL ? "NOT ADDED" : "PEER ADDED");
@@ -2681,6 +2757,7 @@
      struct GNUNET_SERVER_Handle *server,
      const struct GNUNET_CONFIGURATION_Handle *c)
 {
+  int random_seconds;
   sched = scheduler;
   cfg = c;
   datacache = GNUNET_DATACACHE_create (sched, cfg, "dhtcache");
@@ -2751,6 +2828,7 @@
     }
 
 #if DO_FIND_PEER
+  random_seconds = GNUNET_CRYPTO_random_u32(GNUNET_CRYPTO_QUALITY_WEAK, 180);
   GNUNET_SCHEDULER_add_delayed (sched,
                                 
GNUNET_TIME_relative_multiply(GNUNET_TIME_UNIT_SECONDS, 30),
                                 &send_find_peer_message, NULL);

Modified: gnunet/src/dht/plugin_dhtlog_mysql.c
===================================================================
--- gnunet/src/dht/plugin_dhtlog_mysql.c        2010-08-08 20:35:49 UTC (rev 
12503)
+++ gnunet/src/dht/plugin_dhtlog_mysql.c        2010-08-09 10:42:26 UTC (rev 
12504)
@@ -86,16 +86,6 @@
 
 static unsigned long long current_trial = 0;    /* I like to assign 0, just to 
remember */
 
-static char *user;
-
-static char *password;
-
-static char *server;
-
-static char *database;
-
-static unsigned long long port;
-
 /**
  * Connection to the MySQL Server.
  */
@@ -298,7 +288,7 @@
 }
 
 /**
- * Create a prepared statement.
+ * Close a prepared statement.
  *
  * @return NULL on error
  */
@@ -308,8 +298,7 @@
   if (s == NULL)
     return;
 
-  if (s->query != NULL)
-    GNUNET_free(s->query);
+  GNUNET_free_non_null(s->query);
   if (s->valid == GNUNET_YES)
     mysql_stmt_close(s->statement);
   GNUNET_free(s);
@@ -319,20 +308,71 @@
  * Initialize the prepared statements for use with dht test logging
  */
 static int
-iopen ()
+iopen (struct GNUNET_DHTLOG_Plugin *plugin)
 {
   int ret;
+  my_bool reconnect;
+  unsigned int timeout;
+  char *user;
+  char *password;
+  char *server;
+  char *database;
+  unsigned long long port;
 
   conn = mysql_init (NULL);
   if (conn == NULL)
     return GNUNET_SYSERR;
 
+  if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (plugin->cfg,
+                                                         "MYSQL", "DATABASE",
+                                                         &database))
+    {
+      database = GNUNET_strdup("gnunet");
+    }
+
+  if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (plugin->cfg,
+                                                          "MYSQL", "USER", 
&user))
+    {
+      user = GNUNET_strdup("dht");
+    }
+
+  if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (plugin->cfg,
+                                                          "MYSQL", "PASSWORD", 
&password))
+    {
+      password = GNUNET_strdup("dhttest**");
+    }
+
+  if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (plugin->cfg,
+                                                          "MYSQL", "SERVER", 
&server))
+    {
+      server = GNUNET_strdup("localhost");
+    }
+
+  if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_number (plugin->cfg,
+                                                          "MYSQL", 
"MYSQL_PORT", &port))
+    {
+      port = 0;
+    }
+
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Connecting to mysql with: user %s, 
pass %s, server %s, database %s, port %d\n",
               user, password, server, database, port);
 
+  reconnect = 0;
+  timeout = 60; /* in seconds */
+  mysql_options (conn, MYSQL_OPT_RECONNECT, &reconnect);
+  mysql_options (conn,
+                 MYSQL_OPT_CONNECT_TIMEOUT, (const void *) &timeout);
+  mysql_options(conn, MYSQL_SET_CHARSET_NAME, "UTF8");
+  mysql_options (conn, MYSQL_OPT_READ_TIMEOUT, (const void *) &timeout);
+  mysql_options (conn, MYSQL_OPT_WRITE_TIMEOUT, (const void *) &timeout);
   mysql_real_connect (conn, server, user, password,
-                      database, (unsigned int) port, NULL, 0);
+                      database, (unsigned int) port, NULL, 
CLIENT_IGNORE_SIGPIPE);
 
+  GNUNET_free_non_null(server);
+  GNUNET_free_non_null(password);
+  GNUNET_free_non_null(user);
+  GNUNET_free_non_null(database);
+
   if (mysql_error (conn)[0])
     {
       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
@@ -708,12 +748,6 @@
 
   va_start (ap, insert_id);
 
-  if (mysql_stmt_prepare (s->statement, s->query, strlen (s->query)))
-      {
-        GNUNET_log(GNUNET_ERROR_TYPE_ERROR, "mysql_stmt_prepare ERROR");
-        return GNUNET_SYSERR;
-      }
-
   if (GNUNET_OK != init_params (s, ap))
     {
       va_end (ap);
@@ -1345,43 +1379,13 @@
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "MySQL DHT Logger: initializing 
database\n");
 #endif
 
-  if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (plugin->cfg,
-                                                         "MYSQL", "DATABASE",
-                                                         &database))
+  if (iopen (plugin) != GNUNET_OK)
     {
-      database = GNUNET_strdup("gnunet");
-    }
-
-  if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (plugin->cfg,
-                                                          "MYSQL", "USER", 
&user))
-    {
-      user = GNUNET_strdup("dht");
-    }
-
-  if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (plugin->cfg,
-                                                          "MYSQL", "PASSWORD", 
&password))
-    {
-      password = GNUNET_strdup("dhttest**");
-    }
-
-  if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_string (plugin->cfg,
-                                                          "MYSQL", "SERVER", 
&server))
-    {
-      server = GNUNET_strdup("localhost");
-    }
-
-  if (GNUNET_OK != GNUNET_CONFIGURATION_get_value_number (plugin->cfg,
-                                                          "MYSQL", 
"MYSQL_PORT", &port))
-    {
-      port = 0;
-    }
-
-  if (iopen () != GNUNET_OK)
-    {
       GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
                   _("Failed to initialize MySQL database connection for 
dhtlog.\n"));
       return NULL;
     }
+
   GNUNET_assert(plugin->dhtlog_api == NULL);
   plugin->dhtlog_api = GNUNET_malloc(sizeof(struct GNUNET_DHTLOG_Handle));
   plugin->dhtlog_api->insert_trial = &add_trial;
@@ -1428,7 +1432,8 @@
 
   if (conn != NULL)
     mysql_close (conn);
-
+  conn = NULL;
+  mysql_library_end();
   GNUNET_free(dhtlog_api);
   return NULL;
 }




reply via email to

[Prev in Thread] Current Thread [Next in Thread]