Extending the testcases to use bluetooth
[oweals/gnunet.git] / src / transport / gnunet-service-transport.c
index 924c7d9b31c982f6e61758ec69ed84a8739fbfb5..0039b9a4c61b8991cf4acd0268c75231e381c95c 100644 (file)
@@ -171,6 +171,8 @@ process_payload (const struct GNUNET_PeerIdentity *peer,
     return ret;
   }
 
+  GST_ats_add_address ((struct GNUNET_HELLO_Address *) address, session);
+
   if (do_forward != GNUNET_YES)
     return ret;
   im = (struct InboundMessage *) buf;
@@ -179,7 +181,6 @@ process_payload (const struct GNUNET_PeerIdentity *peer,
   im->peer = *peer;
   memcpy (&im[1], message, ntohs (message->size));
 
-  GNUNET_ATS_address_add (GST_ats, address, session, NULL, 0);
   GST_clients_broadcast (&im->header, GNUNET_YES);
 
   return ret;
@@ -246,14 +247,14 @@ GST_receive_callback (void *cls, const struct GNUNET_PeerIdentity *peer,
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK,
                 "Processing `%s' from `%s'\n", "PING",
                 (sender_address !=
-                 NULL) ? GST_plugins_a2s (&address) : "<inbound>");
+                 NULL) ? GST_plugins_a2s (&address) : TRANSPORT_SESSION_INBOUND_STRING);
     GST_validation_handle_ping (peer, message, &address, session);
     break;
   case GNUNET_MESSAGE_TYPE_TRANSPORT_PONG:
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK,
                 "Processing `%s' from `%s'\n", "PONG",
                 (sender_address !=
-                 NULL) ? GST_plugins_a2s (&address) : "<inbound>");
+                 NULL) ? GST_plugins_a2s (&address) : TRANSPORT_SESSION_INBOUND_STRING);
     GST_validation_handle_pong (peer, message);
     break;
   case GNUNET_MESSAGE_TYPE_TRANSPORT_SESSION_CONNECT:
@@ -393,14 +394,79 @@ plugin_env_address_to_type (void *cls,
 }
 
 
+/**
+ * Notify ATS about the new address including the network this address is
+ * located in.
+ *
+ * @param address the address
+ * @param session the session
+ */
 void
-GST_update_ats_metrics (const struct GNUNET_PeerIdentity *peer,
+GST_ats_add_address (const struct GNUNET_HELLO_Address *address,
+                                                                                struct Session *session)
+{
+  struct GNUNET_TRANSPORT_PluginFunctions *papi;
+       struct GNUNET_ATS_Information ats;
+       uint32_t net;
+
+  /* valid new address, let ATS know! */
+  if (NULL == address->transport_name)
+  {
+       GNUNET_break (0);
+       return;
+  }
+  if (NULL == (papi = GST_plugins_find (address->transport_name)))
+  {
+    /* we don't have the plugin for this address */
+       GNUNET_break (0);
+       return;
+  }
+
+  if (GNUNET_YES == GNUNET_ATS_session_known (GST_ats, address, session))
+       return;
+
+       net = papi->get_network (NULL, (void *) session);
+  if (GNUNET_ATS_NET_UNSPECIFIED == net)
+  {
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                                               _("Could not obtain a valid network for `%s' %s\n"),
+                GNUNET_i2s (&address->peer), GST_plugins_a2s (address));
+       GNUNET_break (0);
+  }
+       ats.type = htonl (GNUNET_ATS_NETWORK_TYPE);
+       ats.value = htonl(net);
+       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                       "Notifying ATS about peer `%s''s new address `%s' session %p in network %s\n",
+                       GNUNET_i2s (&address->peer),
+                       (0 == address->address_length) ? "<inbound>" : GST_plugins_a2s (address),
+                       session,
+                       GNUNET_ATS_print_network_type(net));
+       GNUNET_ATS_address_add (GST_ats,
+                       address, session, &ats, 1);
+}
+
+
+/**
+ * Notify ATS about property changes to an address
+ *
+ * @param peer the peer
+ * @param address the address
+ * @param session the session
+ * @param ats performance information
+ * @param ats_count number of elements in ats
+ */
+void
+GST_ats_update_metrics (const struct GNUNET_PeerIdentity *peer,
                        const struct GNUNET_HELLO_Address *address,
                        struct Session *session,
                        const struct GNUNET_ATS_Information *ats,
                        uint32_t ats_count)
 {
        struct GNUNET_ATS_Information *ats_new;
+
+  if (GNUNET_NO == GNUNET_ATS_session_known (GST_ats, address, session))
+       return;
+
        /* Call to manipulation to manipulate ATS information */
        ats_new = GST_manipulation_manipulate_metrics (peer, address, session, ats, ats_count);
        if (NULL == ats_new)
@@ -408,7 +474,12 @@ GST_update_ats_metrics (const struct GNUNET_PeerIdentity *peer,
                        GNUNET_break (0);
                        return;
        }
-  GNUNET_ATS_address_update (GST_ats, address, session, ats_new, ats_count);
+  if (GNUNET_NO == GNUNET_ATS_address_update (GST_ats, address, session, ats_new, ats_count))
+  {
+       GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                       _("Address or session unknown: failed to update properties for peer `%s' plugin `%s' address `%s' session %p\n"),
+                       GNUNET_i2s (peer), address->transport_name, GST_plugins_a2s (address), session);
+  }
   GNUNET_free (ats_new);
 }
 
@@ -441,14 +512,59 @@ plugin_env_update_metrics (void *cls,
                return;
        GNUNET_assert (NULL != GST_ats);
 
+
        haddress.peer = *peer;
        haddress.address = address;
   haddress.address_length = address_len;
   haddress.transport_name = plugin_name;
 
-  GST_update_ats_metrics (peer, &haddress, session, ats, ats_count);
+       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Updating metrics for peer `%s' address %s session %p\n",
+                       GNUNET_i2s (peer), GST_plugins_a2s(&haddress), session);
+  GST_ats_update_metrics (peer, &haddress, session, ats, ats_count);
 }
 
+static void
+plugin_env_session_start (void *cls,
+          const struct GNUNET_PeerIdentity *peer,
+          const char *plugin,
+          const void *address,
+          uint16_t address_len,
+          struct Session *session,
+          const struct GNUNET_ATS_Information *ats,
+          uint32_t ats_count)
+{
+       if (NULL == peer)
+       {
+               GNUNET_break (0);
+               return;
+       }
+       if (NULL == plugin)
+       {
+               GNUNET_break (0);
+               return;
+       }
+       if ((address_len != 0) && (NULL == address))
+       {
+               GNUNET_break (0);
+               return;
+       }
+       if (NULL == session)
+       {
+               GNUNET_break (0);
+               return;
+       }
+
+       struct GNUNET_HELLO_Address *addr;
+       addr = GNUNET_HELLO_address_allocate (peer, plugin, address, address_len);
+       GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+                       "Notification from plugin `%s' about new session %p from peer `%s' address `%s'\n",
+                       plugin, session, GNUNET_i2s (peer), GST_plugins_a2s(addr));
+       GST_ats_add_address (addr, session);
+
+       if (0 < ats_count)
+               GST_ats_update_metrics (peer, addr, session, ats, ats_count);
+       GNUNET_free (addr);
+}
 
 /**
  * Function called by ATS to notify the callee that the
@@ -707,6 +823,7 @@ run (void *cls, struct GNUNET_SERVER_Handle *server,
   GST_manipulation_init (GST_cfg);
   GST_plugins_load (&GST_manipulation_recv,
                     &plugin_env_address_change_notification,
+                    &plugin_env_session_start,
                     &plugin_env_session_end,
                     &plugin_env_address_to_type,
                     &plugin_env_update_metrics);