X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=src%2Fats%2Fgnunet-service-ats_addresses.c;h=b55703caef15bf6db0544fb0c8879299128f0550;hb=ddd4931558eac0b12cf977a5d3c14e3666094b94;hp=ff83f4fd802c744916fec1dc2b36e382b9e5afb4;hpb=42f0a310ce2d0c9c2cd4b62e8da617d8bc9b8290;p=oweals%2Fgnunet.git

diff --git a/src/ats/gnunet-service-ats_addresses.c b/src/ats/gnunet-service-ats_addresses.c
index ff83f4fd8..b55703cae 100644
--- a/src/ats/gnunet-service-ats_addresses.c
+++ b/src/ats/gnunet-service-ats_addresses.c
@@ -22,10 +22,12 @@
  * @file ats/gnunet-service-ats_addresses.c
  * @brief ats service address management
  * @author Matthias Wachs
+ * @author Christian Grothoff
  */
 #include "platform.h"
 #include "gnunet-service-ats_addresses.h"
-
+#include "gnunet-service-ats_scheduling.h"
+#include "gnunet-service-ats_reservations.h"
 
 struct ATS_Address
 {
@@ -33,19 +35,33 @@ struct ATS_Address
 
   size_t addr_len;
 
+  struct GNUNET_SERVER_Client *session_client;
+		   
   uint32_t session_id;
 
   uint32_t ats_count;
 
-  void * addr;
+  const void * addr;
 
   char * plugin;
 
   struct GNUNET_TRANSPORT_ATS_Information * ats;
+
+  struct GNUNET_BANDWIDTH_Value32NBO bw_in;
+
+  struct GNUNET_BANDWIDTH_Value32NBO bw_out;
+
 };
 
+
 static struct GNUNET_CONTAINER_MultiHashMap * addresses;
 
+static unsigned long long total_quota_in;
+
+static unsigned long long total_quota_out;
+
+static unsigned int active_addr_count;
+
 
 struct CompareAddressContext
 {
@@ -53,50 +69,267 @@ struct CompareAddressContext
   struct ATS_Address * result;
 };
 
-int compare_address_it (void *cls,
-               const GNUNET_HashCode * key,
-               void *value)
+
+static void
+destroy_address (struct ATS_Address *addr)
+{
+  GNUNET_assert (GNUNET_YES == 
+		 GNUNET_CONTAINER_multihashmap_remove(addresses, 
+						      &addr->peer.hashPubKey, 
+						      addr));
+  if (ntohl (addr->bw_in.value__) > 0)
+  {
+    active_addr_count--;
+    // FIXME: update address assignment for other peers...
+  }
+  GNUNET_free_non_null (addr->ats);
+  GNUNET_free (addr->plugin);
+  GNUNET_free (addr);
+}
+
+
+static int 
+compare_address_it (void *cls,
+		    const GNUNET_HashCode * key,
+		    void *value)
 {
   struct CompareAddressContext * cac = cls;
   struct ATS_Address * aa = (struct ATS_Address *) value;
+
+  /* compare sessions */
+  if ((aa->session_client != cac->search->session_client) ||
+      (aa->session_id != cac->search->session_id))
+    return GNUNET_YES;
+
+  if (aa->addr_len != cac->search->addr_len)
+  {
+    return GNUNET_YES;
+  }
+
   if (0 == strcmp(aa->plugin, cac->search->plugin))
   {
-    if ((aa->addr_len == cac->search->addr_len) &&
-        (0 == memcmp (aa->addr, cac->search->addr, aa->addr_len)))
-      cac->result = aa;
+    return GNUNET_YES;
+  }
+
+  if (0 == memcmp (aa->addr, cac->search->addr, aa->addr_len))
+  {
+    cac->result = aa;
     return GNUNET_NO;
   }
   return GNUNET_YES;
 }
 
 
-static int 
-free_address_it (void *cls,
-		 const GNUNET_HashCode * key,
-		 void *value)
+struct ATS_Address *
+find_address (const struct GNUNET_PeerIdentity *peer,
+              struct ATS_Address * addr)
+{
+  struct CompareAddressContext cac;
+  cac.result = NULL;
+  cac.search = addr;
+
+  GNUNET_CONTAINER_multihashmap_get_multiple(addresses,
+         &peer->hashPubKey,
+         compare_address_it,
+         &cac);
+
+  return cac.result;
+}
+
+
+void
+GAS_address_update (const struct GNUNET_PeerIdentity *peer,
+		    const char *plugin_name,
+		    const void *plugin_addr, size_t plugin_addr_len,
+		    struct GNUNET_SERVER_Client *session_client,
+		    uint32_t session_id,
+		    const struct GNUNET_TRANSPORT_ATS_Information *atsi,
+		    uint32_t atsi_count)
 {
-  struct ATS_Address * aa = cls;
+  struct ATS_Address * aa;
+  struct ATS_Address * old;
+
+  aa = GNUNET_malloc (sizeof (struct ATS_Address) + plugin_addr_len);
+  aa->ats = GNUNET_malloc(atsi_count * sizeof (struct GNUNET_TRANSPORT_ATS_Information));
+  aa->peer = *peer;
+  aa->addr_len = plugin_addr_len;
+  aa->ats_count = atsi_count;
+  memcpy (aa->ats, atsi, atsi_count * sizeof (struct GNUNET_TRANSPORT_ATS_Information));
+  aa->addr = &aa[1];
+  memcpy (&aa[1], plugin_addr, plugin_addr_len);
+  aa->plugin = GNUNET_strdup (plugin_name);
+  aa->session_client = session_client;
+  aa->session_id = session_id;
+  old = find_address (peer, aa);
+  if (old == NULL)
+  {
+    GNUNET_assert (GNUNET_OK ==
+                   GNUNET_CONTAINER_multihashmap_put (addresses,
+						      &peer->hashPubKey,
+						      aa,
+						      GNUNET_CONTAINER_MULTIHASHMAPOPTION_MULTIPLE));
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+		"Added new address for peer `%s' %X\n",
+		GNUNET_i2s (peer), aa);
+    return;
+  }
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+	      "Updated existing address for peer `%s' %X \n",
+	      GNUNET_i2s (peer), old);
+  GNUNET_free_non_null (old->ats);
+  old->ats = NULL;
+  old->ats_count = 0;
+  old->ats = aa->ats;
+  old->ats_count = aa->ats_count;
+  GNUNET_free (aa->plugin);
   GNUNET_free (aa);
+}
+
+
+static int
+remove_address_by_client (void *cls,
+			  const GNUNET_HashCode * key,
+			  void *value)
+{
+  struct GNUNET_SERVER_Client *client = cls;
+  struct ATS_Address * aa = value;
+
+  if (aa->session_client == client)
+    destroy_address (aa);  
   return GNUNET_OK;
 }
 
 
+void
+GAS_address_client_disconnected (struct GNUNET_SERVER_Client *client)
+{
+  if (addresses != NULL)
+    GNUNET_CONTAINER_multihashmap_iterate(addresses, 
+					  &remove_address_by_client, client);
+}
+
+
+void
+GAS_address_destroyed (const struct GNUNET_PeerIdentity *peer,
+		       const char *plugin_name,
+		       const void *plugin_addr, size_t plugin_addr_len,
+		       struct GNUNET_SERVER_Client *session_client,
+		       uint32_t session_id)
+{
+
+  struct ATS_Address aa;
+  struct ATS_Address *res;
+
+  aa.peer = *peer;
+  aa.addr_len = plugin_addr_len;
+  aa.addr = plugin_addr;
+  aa.plugin = (char*) plugin_name;
+  aa.session_client = session_client;
+  aa.session_id = session_id;
+
+  res = find_address (peer, &aa);
+  if (res == NULL)
+  {
+    /* we don't even know this one, can this happen? */
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+		"Asked to delete unknown address for peer `%s'\n",
+		GNUNET_i2s (peer));
+    return; 
+  }
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+	      "Deleting address for peer `%s': `%s'\n",
+	      GNUNET_i2s (peer), plugin_name);
+  destroy_address (res);
+}
+
+
+void
+GAS_addresses_request_address (const struct GNUNET_PeerIdentity *peer)
+{
+  struct ATS_Address * aa;
+
+  aa = GNUNET_CONTAINER_multihashmap_get (addresses, &peer->hashPubKey);
+  if (aa == NULL)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+		"Cannot suggest address for peer `%s'\n",
+		GNUNET_i2s (peer));
+    return; 
+  }
+  /* FIXME: ensure that we don't do this multiple times per peer! */
+  if (ntohl (aa->bw_in.value__) == 0)
+  {
+    active_addr_count++;
+    aa->bw_in.value__ = htonl (total_quota_in / active_addr_count);
+    aa->bw_out.value__ = htonl (total_quota_out / active_addr_count);
+    /* FIXME: update bw assignments for other addresses... */
+  }
+  GAS_reservations_set_bandwidth (peer,
+				  aa->bw_in);
+  GAS_scheduling_transmit_address_suggestion (peer, aa->plugin, 
+					      aa->addr, aa->addr_len, 
+					      aa->session_client, aa->session_id, 
+					      aa->ats, aa->ats_count, 
+					      aa->bw_out, aa->bw_in);
+}
+
+
 /**
+ * Initialize address subsystem.
+ *
+ * @param cfg configuration to use
  */
 void
-GAS_addresses_done ()
+GAS_addresses_init (const struct GNUNET_CONFIGURATION_Handle *cfg)
 {
-  GNUNET_CONTAINER_multihashmap_iterate (addresses, &free_address_it, NULL);
-  GNUNET_CONTAINER_multihashmap_destroy (addresses);
+  GNUNET_assert (GNUNET_OK ==
+		 GNUNET_CONFIGURATION_get_value_number (cfg,
+							"core",
+							"TOTAL_QUOTA_IN",
+							&total_quota_in));
+  GNUNET_assert (GNUNET_OK ==
+		 GNUNET_CONFIGURATION_get_value_number (cfg,
+							"core",
+							"TOTAL_QUOTA_OUT",
+							&total_quota_out));
+  addresses = GNUNET_CONTAINER_multihashmap_create(128);
+}
+
+
+/**
+ * Free memory of address.
+ *
+ * @param cls NULL
+ * @param key peer identity (unused)
+ * @param value the 'struct ATS_Address' to free
+ * @return GNUNET_OK (continue to iterate)
+ */
+static int 
+free_address_it (void *cls,
+		 const GNUNET_HashCode * key,
+		 void *value)
+{
+  struct ATS_Address * aa = value;
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+    "Freeing address for peer `%s' %X\n",
+    GNUNET_i2s (&aa->peer), aa);
+  GNUNET_CONTAINER_multihashmap_remove (addresses, key, value);
+  destroy_address (aa);
+  return GNUNET_OK;
 }
 
 
+
 /**
+ * Shutdown address subsystem.
  */
 void
-GAS_addresses_init ()
+GAS_addresses_done ()
 {
-  addresses = GNUNET_CONTAINER_multihashmap_create(128);
+  GNUNET_CONTAINER_multihashmap_iterate (addresses, &free_address_it, NULL);
+  GNUNET_CONTAINER_multihashmap_destroy (addresses);
+  addresses = NULL;
 }
 
+
 /* end of gnunet-service-ats_addresses.c */