[GNUnet-SVN] r16665 - gnunet/src/mesh

[gnunet]

Author: bartpolot
Date: 2011-09-01 05:29:46 +0200 (Thu, 01 Sep 2011)
New Revision: 16665

Modified:
   gnunet/src/mesh/mesh.h
   gnunet/src/mesh/mesh_api_new.c
Log:
Added peer add timeout handling, tunnel connect/disconnect notifications

Modified: gnunet/src/mesh/mesh.h
===================================================================
--- gnunet/src/mesh/mesh.h      2011-08-31 22:21:25 UTC (rev 16664)
+++ gnunet/src/mesh/mesh.h      2011-09-01 03:29:46 UTC (rev 16665)
@@ -146,13 +146,6 @@
   MESH_TunnelNumber tunnel_id GNUNET_PACKED;
 
   /**
-   * When should this request time out (the service abort trying to find
-   * a matching peer)?  The request should naturally also be aborted
-   * if the corresponding client disconnects.
-   */
-  struct GNUNET_TIME_AbsoluteNBO timeout;
-
-  /**
    * Peer to connect/disconnect.
    */
   struct GNUNET_PeerIdentity peer;
@@ -165,7 +158,8 @@
 struct GNUNET_MESH_ConnectPeerByType
 {
     /**
-     * Type: GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT_PEER_BY_TYPE
+     * Type: GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT_PEER_BY_TYPE |
+     *       GNUNET_MESSAGE_TYPE_MESH_LOCAL_DISCONNECT_PEER_BY_TYPE
      */
   struct GNUNET_MessageHeader header;
 
@@ -175,13 +169,6 @@
   MESH_TunnelNumber tunnel_id GNUNET_PACKED;
 
   /**
-   * When should this request time out (the service abort trying to find
-   * a matching peer)?  The request should naturally also be aborted
-   * if the corresponding client disconnects.
-   */
-  struct GNUNET_TIME_AbsoluteNBO timeout;
-
-  /**
    * Type specification
    */
   GNUNET_MESH_ApplicationType type GNUNET_PACKED;

Modified: gnunet/src/mesh/mesh_api_new.c
===================================================================
--- gnunet/src/mesh/mesh_api_new.c      2011-08-31 22:21:25 UTC (rev 16664)
+++ gnunet/src/mesh/mesh_api_new.c      2011-09-01 03:29:46 UTC (rev 16665)
@@ -40,10 +40,8 @@
 #if 0                           /* keep Emacsens' auto-indent happy */
 }
 #endif
-       /*
-        */
-#endif                          /*
-                                 */
+#endif
+
 #include "platform.h"
 #include "gnunet_common.h"
 #include "gnunet_client_lib.h"
@@ -204,6 +202,33 @@
 
 
 /**
+ * Description of a peer
+ */
+struct GNUNET_MESH_Peer
+{
+    /**
+     * ID of the peer in short form
+     */
+  GNUNET_PEER_Id id;
+
+  /**
+   * Tunnel this peer belongs to
+   */
+  struct GNUNET_MESH_Tunnel *t;
+
+  /**
+   * Flag indicating whether service has informed about its connection
+   */
+  int connected;
+
+    /**
+     * Task to cancel the connection request for this peer
+     */
+  GNUNET_SCHEDULER_TaskIdentifier cancel;
+};
+
+
+/**
  * Opaque handle to a tunnel.
  */
 struct GNUNET_MESH_Tunnel
@@ -228,7 +253,7 @@
     /**
      * All peers added to the tunnel
      */
-  GNUNET_PEER_Id *peers;
+  struct GNUNET_MESH_Peer **peers;
 
     /**
      * Closure for the connect/disconnect handlers
@@ -314,6 +339,39 @@
 
 
 /**
+ * Notify client that the transmission has timed out
+ * @param cls closure
+ * @param tc task context
+ */
+static void
+timeout_peer_request (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  struct GNUNET_MESH_Peer *p = cls;
+  struct GNUNET_PeerIdentity id;
+  unsigned int i;
+
+  GNUNET_assert (0 == p->connected);
+  for (i = 0; i < p->t->npeers; i++)
+  {
+    if (p->t->peers[i] == p)
+      break;
+  }
+  if (i == p->t->npeers)
+  {
+    GNUNET_break (0);
+    return;
+  }
+  p->t->peers[i] = p->t->peers[p->t->npeers - 1];
+  GNUNET_array_grow (p->t->peers, p->t->npeers, p->t->npeers - 1);
+
+  if (NULL != p->t->connect_handler)
+    p->t->connect_handler (p->t->cls, 0, NULL);
+
+  GNUNET_PEER_resolve (p->id, &id);
+  GNUNET_MESH_peer_request_connect_del (p->t, &id);
+}
+
+/**
  * Add a transmit handle to the transmission queue by priority and set the
  * timeout if needed.
  *
@@ -844,6 +902,8 @@
 /**
  * Request that a peer should be added to the tunnel.  The existing
  * connect handler will be called ONCE with either success or failure.
+ * This function should NOT be called again with the same peer before the
+ * connect handler is called
  *
  * @param tunnel handle to existing tunnel
  * @param timeout how long to try to establish a connection
@@ -855,29 +915,29 @@
                                       const struct GNUNET_PeerIdentity *peer)
 {
   struct GNUNET_MESH_PeerControl msg;
+  struct GNUNET_MESH_Peer *p;
   GNUNET_PEER_Id peer_id;
   unsigned int i;
 
   peer_id = GNUNET_PEER_intern (peer);
   for (i = 0; i < tunnel->npeers; i++)
   {
-    if (tunnel->peers[i] == peer_id)
+    if (tunnel->peers[i]->id == peer_id)
     {
       GNUNET_PEER_change_rc (peer_id, -1);
       /* FIXME: peer was already in the tunnel */
       return;
     }
   }
-  tunnel->npeers++;
-  tunnel->peers =
-      GNUNET_realloc (tunnel->peers, tunnel->npeers * sizeof (GNUNET_PEER_Id));
-  tunnel->peers[tunnel->npeers - 1] = peer_id;
+  p = GNUNET_malloc (sizeof (struct GNUNET_MESH_Peer));
+  p->id = peer_id;
+  p->t = tunnel;
+  p->cancel = GNUNET_SCHEDULER_add_delayed (timeout, &timeout_peer_request, p);
+  GNUNET_array_append (tunnel->peers, tunnel->npeers, p);
 
   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT_PEER_ADD);
   msg.tunnel_id = htonl (tunnel->tid);
-  msg.timeout =
-      GNUNET_TIME_absolute_hton (GNUNET_TIME_relative_to_absolute (timeout));
   msg.peer = *peer;
   send_packet (tunnel->mesh, &msg.header);
 //   tunnel->connect_handler (tunnel->cls, peer, NULL); FIXME call this later
@@ -908,20 +968,23 @@
     return;
   }
   for (i = 0; i < tunnel->npeers; i++)
-    if (tunnel->peers[i] == peer_id)
+    if (tunnel->peers[i]->id == peer_id)
       break;
   if (i == tunnel->npeers)
   {
     GNUNET_break (0);
     return;
   }
+  if (NULL != tunnel->disconnect_handler && tunnel->peers[i]->connected == 1)
+      tunnel->disconnect_handler (tunnel->cls, peer);
   GNUNET_PEER_change_rc (peer_id, -1);
+  GNUNET_free (tunnel->peers[i]);
   tunnel->peers[i] = tunnel->peers[tunnel->npeers - 1];
   GNUNET_array_grow (tunnel->peers, tunnel->npeers, tunnel->npeers - 1);
+
   msg.header.size = htons (sizeof (struct GNUNET_MESH_PeerControl));
   msg.header.type = htons (GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT_PEER_DEL);
   msg.tunnel_id = htonl (tunnel->tid);
-  msg.timeout = GNUNET_TIME_absolute_hton (GNUNET_TIME_UNIT_FOREVER_ABS);
   memcpy (&msg.peer, peer, sizeof (struct GNUNET_PeerIdentity));
   send_packet (tunnel->mesh, &msg.header);
 }
@@ -947,8 +1010,6 @@
   msg.header.size = htons (sizeof (struct GNUNET_MESH_ConnectPeerByType));
   msg.header.type = htons 
(GNUNET_MESSAGE_TYPE_MESH_LOCAL_CONNECT_PEER_BY_TYPE);
   msg.tunnel_id = htonl (tunnel->tid);
-  msg.timeout =
-      GNUNET_TIME_absolute_hton (GNUNET_TIME_relative_to_absolute (timeout));
   msg.type = htonl (app_type);
   send_packet (tunnel->mesh, &msg.header);
 }