From 297e28384ae74a468895f950552ae4acfb7efb0d Mon Sep 17 00:00:00 2001
From: Christian Grothoff <christian@grothoff.org>
Date: Thu, 11 Aug 2011 12:46:39 +0000
Subject: [PATCH] stuff

---
 .../gnunet-service-transport_ats-new.c        | 85 ++++++++++++-------
 1 file changed, 56 insertions(+), 29 deletions(-)

diff --git a/src/transport/gnunet-service-transport_ats-new.c b/src/transport/gnunet-service-transport_ats-new.c
index 250ebc119..983674c07 100644
--- a/src/transport/gnunet-service-transport_ats-new.c
+++ b/src/transport/gnunet-service-transport_ats-new.c
@@ -102,9 +102,24 @@ struct GST_AtsHandle
    */
   struct GNUNET_CONTAINER_MultiHashMap *peers;
 
+  /**
+   * Task scheduled to update our bandwidth assignment.
+   */
+  GNUNET_SCHEDULER_TaskIdentifier ba_task;
 };
 
 
+static void
+update_bandwidth_task (void *cls,
+		       const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  struct GST_AtsHandle *atc = cls;
+
+  atc->ba_task = GNUNET_SCHEDULER_NO_TASK;
+  /* FIXME: update calculations! */
+
+}
+
 
 /**
  * Calculate an updated bandwidth assignment and notify.
@@ -116,7 +131,10 @@ static void
 update_bandwidth_assignment (struct GST_AtsHandle *atc,
 			     struct AllocationRecord *change)
 {
-  
+  /* FIXME: based on the 'change', update the problem... */
+  if (atc->ba_task == GNUNET_SCHEDULER_NO_TASK)
+    atc->ba_task = GNUNET_SCHEDULER_add_now (&update_bandwidth_task,
+					     atc);
 }
 
 
@@ -174,6 +192,11 @@ destroy_allocation_record (void *cls,
 void
 GST_ats_shutdown (struct GST_AtsHandle *atc)
 {
+  if (GNUNET_SCHEDULER_NO_TASK != atc->ba_task)
+    {
+      GNUNET_SCHEDULER_cancel (atc->ba_task);
+      atc->ba_task = GNUNET_SCHEDULER_NO_TASK;
+    }
   GNUNET_CONTAINER_multihashmap_iterate (atc->peers,
 					 &destroy_allocation_record,
 					 NULL);
@@ -286,6 +309,31 @@ create_allocation_record (const char *plugin_name,
 }
 
 
+/**
+ * Mark all matching allocation records as not connected.
+ *
+ * @param cls 'struct GTS_AtsHandle'
+ * @param key identity of the peer associated with the record
+ * @param value the 'struct AllocationRecord' to clear the 'connected' flag
+ * @return GNUNET_OK (continue to iterate)
+ */
+static int
+disconnect_peer (void *cls,
+		 const GNUNET_HashCode *key,
+		 void *value)
+{
+  struct GST_AtsHandle *atc = cls;
+  struct AllocationRecord *ar = value;
+
+  if (GNUNET_YES == ar->connected)
+    {
+      ar->connected = GNUNET_NO;     
+      update_bandwidth_assignment (atc, ar);
+    }
+  return GNUNET_OK;
+}
+
+
 /**
  * We established a new connection with a peer (for example, because
  * core asked for it or because the other peer connected to us).
@@ -313,6 +361,9 @@ GST_ats_peer_connect (struct GST_AtsHandle *atc,
   struct AllocationRecord *ar;
   struct UpdateSessionContext usc;
 
+  (void) GNUNET_CONTAINER_multihashmap_iterate (atc->peers,
+						&disconnect_peer,
+						atc);
   ar = create_allocation_record (plugin_name,
 				 session,
 				 plugin_addr,
@@ -338,31 +389,6 @@ GST_ats_peer_connect (struct GST_AtsHandle *atc,
 }
 
 
-/**
- * Mark all matching allocation records as not connected.
- *
- * @param cls 'struct GTS_AtsHandle'
- * @param key identity of the peer associated with the record
- * @param value the 'struct AllocationRecord' to clear the 'connected' flag
- * @return GNUNET_OK (continue to iterate)
- */
-static int
-disconnect_peer (void *cls,
-		 const GNUNET_HashCode *key,
-		 void *value)
-{
-  struct GST_AtsHandle *atc = cls;
-  struct AllocationRecord *ar = value;
-
-  if (GNUNET_YES == ar->connected)
-    {
-      ar->connected = GNUNET_NO;     
-      update_bandwidth_assignment (atc, ar);
-    }
-  return GNUNET_OK;
-}
-
-
 /**
  * We disconnected from the given peer (for example, because ats, core
  * or blacklist asked for it or because the other peer disconnected).
@@ -375,9 +401,10 @@ void
 GST_ats_peer_disconnect (struct GST_AtsHandle *atc,
 			 const struct GNUNET_PeerIdentity *peer)
 {
-  (void) GNUNET_CONTAINER_multihashmap_iterate (atc->peers,
-						&disconnect_peer,
-						atc);
+  (void) GNUNET_CONTAINER_multihashmap_get_multiple (atc->peers,
+						     &peer->hashPubKey,
+						     &disconnect_peer,
+						     atc);
 }
 
 
-- 
2.25.1