- use proper signedness
[oweals/gnunet.git] / src / transport / gnunet-service-transport.c
index 50faeae89d1b4e4cf0150f8616533d8fb3f1ccdb..f54d30e21cd35caa1edc9b9d65a00348a86c3cab 100644 (file)
 #include "gnunet-service-transport_manipulation.h"
 #include "transport.h"
 
+
+/**
+ * Information we need for an asynchronous session kill.
+ */
+struct SessionKiller
+{
+  /**
+   * Kept in a DLL.
+   */
+  struct SessionKiller *next;
+
+  /**
+   * Kept in a DLL.
+   */
+  struct SessionKiller *prev;
+
+  /**
+   * Session to kill.
+   */
+  struct Session *session;
+
+  /**
+   * Plugin for the session.
+   */
+  struct GNUNET_TRANSPORT_PluginFunctions *plugin;
+
+  /**
+   * The kill task.
+   */
+  GNUNET_SCHEDULER_TaskIdentifier task;
+};
+
+
 /* globals */
 
 /**
@@ -77,15 +110,25 @@ struct GNUNET_CRYPTO_EddsaPrivateKey *GST_my_private_key;
  */
 struct GNUNET_ATS_SchedulingHandle *GST_ats;
 
+/**
+ * Hello address expiration
+ */
+struct GNUNET_TIME_Relative hello_expiration;
+
 /**
  * DEBUGGING connection counter
  */
 static int connections;
 
 /**
- * Hello address expiration
+ * Head of DLL of asynchronous tasks to kill sessions.
  */
-struct GNUNET_TIME_Relative hello_expiration;
+static struct SessionKiller *sk_head;
+
+/**
+ * Tail of DLL of asynchronous tasks to kill sessions.
+ */
+static struct SessionKiller *sk_tail;
 
 
 /**
@@ -124,7 +167,6 @@ process_hello_update (void *cls, const struct GNUNET_MessageHeader *hello)
 }
 
 
-
 /**
  * We received some payload.  Prepare to pass it on to our clients.
  *
@@ -163,7 +205,7 @@ process_payload (const struct GNUNET_PeerIdentity *peer,
     return ret;
   }
 
-  GST_ats_add_address ((struct GNUNET_HELLO_Address *) address, session);
+  GST_ats_add_address (address, session, NULL, 0);
 
   if (GNUNET_YES != do_forward)
     return ret;
@@ -177,15 +219,66 @@ process_payload (const struct GNUNET_PeerIdentity *peer,
 }
 
 
+/**
+ * Task to asynchronously terminate a session.
+ *
+ * @param cls the `struct SessionKiller` with the information for the kill
+ * @param tc scheduler context
+ */
+static void
+kill_session_task (void *cls,
+                   const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  struct SessionKiller *sk = cls;
+
+  sk->task = GNUNET_SCHEDULER_NO_TASK;
+  GNUNET_CONTAINER_DLL_remove (sk_head, sk_tail, sk);
+  sk->plugin->disconnect_session (sk->plugin->cls,
+                                  sk->session);
+  GNUNET_free (sk);
+}
+
+
+/**
+ * Force plugin to terminate session due to communication
+ * issue.
+ *
+ * @param plugin_name name of the plugin
+ * @param session session to termiante
+ */
+static void
+kill_session (const char *plugin_name,
+              struct Session *session)
+{
+  struct GNUNET_TRANSPORT_PluginFunctions *plugin;
+  struct SessionKiller *sk;
+
+  for (sk = sk_head; NULL != sk; sk = sk->next)
+    if (sk->session == session)
+      return;
+  plugin = GST_plugins_find (plugin_name);
+  if (NULL == plugin)
+  {
+    GNUNET_break (0);
+    return;
+  }
+  /* need to issue disconnect asynchronously */
+  sk = GNUNET_new (struct SessionKiller);
+  sk->session = session;
+  sk->plugin = plugin;
+  sk->task = GNUNET_SCHEDULER_add_now (&kill_session_task,
+                                       sk);
+  GNUNET_CONTAINER_DLL_insert (sk_head, sk_tail, sk);
+}
+
+
 /**
  * Function called by the transport for each received message.
- * This function should also be called with "NULL" for the
- * message to signal that the other peer disconnected.
  *
  * @param cls closure, const char* with the name of the plugin we received the message from
  * @param peer (claimed) identity of the other peer
  * @param message the message, NULL if we only care about
- *                learning about the delay until we should receive again -- FIXME!
+ *                learning about the delay until we should receive again
  * @param session identifier used for this session (NULL for plugins
  *                that do not offer bi-directional communication to the sender
  *                using the same "connection")
@@ -194,7 +287,7 @@ process_payload (const struct GNUNET_PeerIdentity *peer,
  *                for inbound TCP/UDP connections since it it not clear
  *                that we could establish ourselves a connection to that
  *                IP address and get the same system)
- * @param sender_address_len number of bytes in sender_address
+ * @param sender_address_len number of bytes in @a sender_address
  * @return how long the plugin should wait until receiving more data
  *         (plugins that do not support this, can ignore the return value)
  */
@@ -236,39 +329,65 @@ GST_receive_callback (void *cls,
     /* Legacy HELLO message, discard  */
     return ret;
   case GNUNET_MESSAGE_TYPE_HELLO:
-    GST_validation_handle_hello (message);
+    if (GNUNET_OK !=
+        GST_validation_handle_hello (message))
+    {
+      GNUNET_break_op (0);
+      kill_session (plugin_name, session);
+    }
     return ret;
   case GNUNET_MESSAGE_TYPE_TRANSPORT_PING:
     GNUNET_log (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK,
                 "Processing `%s' from `%s'\n", "PING",
                 (sender_address !=
                  NULL) ? GST_plugins_a2s (&address) : TRANSPORT_SESSION_INBOUND_STRING);
-    GST_validation_handle_ping (peer, message, &address, session);
+    if (GNUNET_OK !=
+        GST_validation_handle_ping (peer, message, &address, session))
+      kill_session (plugin_name, 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) : TRANSPORT_SESSION_INBOUND_STRING);
-    GST_validation_handle_pong (peer, message);
+    if (GNUNET_OK !=
+        GST_validation_handle_pong (peer, message))
+    {
+      GNUNET_break_op (0);
+      kill_session (plugin_name, session);
+    }
     break;
   case GNUNET_MESSAGE_TYPE_TRANSPORT_SESSION_CONNECT:
-    GST_neighbours_handle_connect (message, peer, &address, session);
+    if (GNUNET_OK !=
+        GST_neighbours_handle_connect (message, peer, &address, session))
+    {
+      GNUNET_break_op (0);
+      kill_session (plugin_name, session);
+    }
     break;
   case GNUNET_MESSAGE_TYPE_TRANSPORT_SESSION_CONNECT_ACK:
-    GST_neighbours_handle_connect_ack (message, peer, &address, session);
+    if (GNUNET_OK !=
+        GST_neighbours_handle_connect_ack (message, peer, &address, session))
+    {
+      kill_session (plugin_name, session);
+    }
     break;
   case GNUNET_MESSAGE_TYPE_TRANSPORT_SESSION_ACK:
-    GST_neighbours_handle_session_ack (message, peer, &address, session);
+    if (GNUNET_OK !=
+        GST_neighbours_handle_session_ack (message, peer, &address, session))
+    {
+      GNUNET_break_op (0);
+      kill_session (plugin_name, session);
+    }
     break;
   case GNUNET_MESSAGE_TYPE_TRANSPORT_SESSION_DISCONNECT:
     GST_neighbours_handle_disconnect_message (peer, message);
     break;
   case GNUNET_MESSAGE_TYPE_TRANSPORT_SESSION_KEEPALIVE:
-    GST_neighbours_keepalive (peer);
+    GST_neighbours_keepalive (peer, message);
     break;
   case GNUNET_MESSAGE_TYPE_TRANSPORT_SESSION_KEEPALIVE_RESPONSE:
-    GST_neighbours_keepalive_response (peer);
+    GST_neighbours_keepalive_response (peer, message);
     break;
   default:
     /* should be payload */
@@ -335,10 +454,13 @@ plugin_env_session_end (void *cls, const struct GNUNET_PeerIdentity *peer,
 {
   const char *transport_name = cls;
   struct GNUNET_HELLO_Address address;
+  struct SessionKiller *sk;
 
   GNUNET_assert (strlen (transport_name) > 0);
-  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Session %p to peer `%s' ended \n",
-              session, GNUNET_i2s (peer));
+  GNUNET_log (GNUNET_ERROR_TYPE_DEBUG,
+              "Session %p to peer `%s' ended \n",
+              session,
+              GNUNET_i2s (peer));
   if (NULL != session)
     GNUNET_log_from (GNUNET_ERROR_TYPE_DEBUG | GNUNET_ERROR_TYPE_BULK,
                      "transport-ats",
@@ -352,6 +474,16 @@ plugin_env_session_end (void *cls, const struct GNUNET_PeerIdentity *peer,
 
   /* Tell ATS that session has ended */
   GNUNET_ATS_address_destroyed (GST_ats, &address, session);
+  for (sk = sk_head; NULL != sk; sk = sk->next)
+  {
+    if (sk->session == session)
+    {
+      GNUNET_CONTAINER_DLL_remove (sk_head, sk_tail, sk);
+      GNUNET_SCHEDULER_cancel (sk->task);
+      GNUNET_free (sk);
+      break;
+    }
+  }
 }
 
 
@@ -370,9 +502,10 @@ plugin_env_address_to_type (void *cls,
                             size_t addrlen)
 {
   struct GNUNET_ATS_Information ats;
+
   ats.type = htonl (GNUNET_ATS_NETWORK_TYPE);
   ats.value = htonl (GNUNET_ATS_NET_UNSPECIFIED);
-  if (GST_ats == NULL)
+  if (NULL == GST_ats)
   {
     GNUNET_break (0);
     return ats;
@@ -381,13 +514,14 @@ plugin_env_address_to_type (void *cls,
       ((addr->sa_family != AF_INET6) && (addrlen != sizeof (struct sockaddr_in6))) &&
       (addr->sa_family != AF_UNIX))
   {
-    GNUNET_log (GNUNET_ERROR_TYPE_ERROR, "Malformed address with length %u `%s'\n",
+    GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
+                "Malformed address with length %u `%s'\n",
                 addrlen,
-                GNUNET_a2s(addr, addrlen));
+                GNUNET_a2s (addr, addrlen));
     GNUNET_break (0);
     return ats;
   }
-  return GNUNET_ATS_address_get_type(GST_ats, addr, addrlen);
+  return GNUNET_ATS_address_get_type (GST_ats, addr, addrlen);
 }
 
 
@@ -397,13 +531,17 @@ plugin_env_address_to_type (void *cls,
  *
  * @param address the address
  * @param session the session
+ * @param ats ats information
+ * @param ats_count number of @a ats information
  */
 void
 GST_ats_add_address (const struct GNUNET_HELLO_Address *address,
-                     struct Session *session)
+                     struct Session *session,
+                     const struct GNUNET_ATS_Information *ats,
+                     uint32_t ats_count)
 {
   struct GNUNET_TRANSPORT_PluginFunctions *papi;
-  struct GNUNET_ATS_Information ats;
+  struct GNUNET_ATS_Information ats2[ats_count + 1];
   uint32_t net;
 
   /* valid new address, let ATS know! */
@@ -422,16 +560,19 @@ GST_ats_add_address (const struct GNUNET_HELLO_Address *address,
   if (GNUNET_YES == GNUNET_ATS_session_known (GST_ats, address, session))
     return;
 
-  net = papi->get_network (NULL, (void *) session);
+  net = papi->get_network (NULL, 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));
+                _("Could not obtain a valid network for `%s' %s (%s)\n"),
+                GNUNET_i2s (&address->peer),
+                GST_plugins_a2s (address),
+                address->transport_name);
     GNUNET_break (0);
   }
-  ats.type = htonl (GNUNET_ATS_NETWORK_TYPE);
-  ats.value = htonl(net);
+  ats2[0].type = htonl (GNUNET_ATS_NETWORK_TYPE);
+  ats2[0].value = htonl(net);
+  memcpy (&ats2[1], ats, sizeof (struct GNUNET_ATS_Information) * ats_count);
   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),
@@ -439,7 +580,8 @@ GST_ats_add_address (const struct GNUNET_HELLO_Address *address,
               session,
               GNUNET_ATS_print_network_type(net));
   GNUNET_ATS_address_add (GST_ats,
-                          address, session, &ats, 1);
+                          address, session,
+                          ats2, ats_count + 1);
 }
 
 
@@ -568,16 +710,13 @@ plugin_env_session_start (void *cls,
   }
 
   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_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, ats, ats_count);
   GNUNET_free(addr);
 }
 
@@ -621,8 +760,8 @@ ats_request_address_change (void *cls,
     return;
   }
   GST_neighbours_switch_to_address (&address->peer, address, session, ats,
-                                         ats_count, bandwidth_in,
-                                         bandwidth_out);
+                                    ats_count, bandwidth_in,
+                                    bandwidth_out);
 }