[GNUnet-SVN] r34657 - gnunet/src/hostlist

[gnunet]

Author: grothoff
Date: 2014-12-16 21:31:20 +0100 (Tue, 16 Dec 2014)
New Revision: 34657

Modified:
   gnunet/src/hostlist/gnunet-daemon-hostlist_server.c
Log:
-remove TH from map upon completion of transmission

Modified: gnunet/src/hostlist/gnunet-daemon-hostlist_server.c
===================================================================
--- gnunet/src/hostlist/gnunet-daemon-hostlist_server.c 2014-12-16 20:30:52 UTC 
(rev 34656)
+++ gnunet/src/hostlist/gnunet-daemon-hostlist_server.c 2014-12-16 20:31:20 UTC 
(rev 34657)
@@ -430,7 +430,8 @@
 /**
  * Handler called by CORE when CORE is ready to transmit message
  *
- * @param cls closure
+ * @param cls closure with the `const struct GNUNET_PeerIdentity *` of
+ *            the peer we are sending to
  * @param size size of buffer to copy message to
  * @param buf buffer to copy message to
  * @return number of bytes copied to @a buf
@@ -440,12 +441,21 @@
                     size_t size,
                     void *buf)
 {
+  const struct GNUNET_PeerIdentity *peer = cls;
   static uint64_t hostlist_adv_count;
   size_t transmission_size;
   size_t uri_size;              /* Including \0 termination! */
   struct GNUNET_MessageHeader header;
   char *cbuf;
+  struct GNUNET_CORE_TransmitHandle *th;
 
+  th = GNUNET_CONTAINER_multipeermap_get (advertisements,
+                                          peer);
+  GNUNET_assert (NULL != th);
+  GNUNET_assert (GNUNET_YES ==
+                 GNUNET_CONTAINER_multipeermap_remove (advertisements,
+                                                       peer,
+                                                       th));
   if (NULL == buf)
   {
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
@@ -518,7 +528,8 @@
                                                GNUNET_ADV_TIMEOUT,
                                                peer,
                                                size,
-                                               &adv_transmit_ready, NULL)) )
+                                               &adv_transmit_ready,
+                                               (void *) peer)) )
   {
     GNUNET_log (GNUNET_ERROR_TYPE_WARNING,
                 _("Advertisement message could not be queued by core\n"));