[GNUnet-SVN] r18104 - gnunet/src/transport

[gnunet]

Author: wachs
Date: 2011-11-11 13:06:05 +0100 (Fri, 11 Nov 2011)
New Revision: 18104

Modified:
   gnunet/src/transport/gnunet-service-transport_neighbours.c
Log:


Modified: gnunet/src/transport/gnunet-service-transport_neighbours.c
===================================================================
--- gnunet/src/transport/gnunet-service-transport_neighbours.c  2011-11-11 
10:53:44 UTC (rev 18103)
+++ gnunet/src/transport/gnunet-service-transport_neighbours.c  2011-11-11 
12:06:05 UTC (rev 18104)
@@ -1239,7 +1239,15 @@
   /* Tell ATS that switching addresses was successful */
   switch (n->state) {
     case S_CONNECTED:
-      GNUNET_ATS_address_in_use (GST_ats, n->address, n->session, GNUNET_YES);
+      if (n->address_state == FRESH)
+      {
+          GST_validation_set_address_use (&n->id,
+                    cc->address,
+                    cc->session,
+                    GNUNET_YES);
+          GNUNET_ATS_address_in_use (GST_ats, cc->address, cc->session, 
GNUNET_YES);
+          n->address_state = USED;
+      }
       break;
     case S_FAST_RECONNECT:
 #if DEBUG_TRANSPORT
@@ -1257,9 +1265,9 @@
                     cc->address,
                     cc->session,
                     GNUNET_YES);
+          GNUNET_ATS_address_in_use (GST_ats, cc->address, cc->session, 
GNUNET_YES);
           n->address_state = USED;
       }
-      GNUNET_ATS_address_in_use (GST_ats, cc->address, cc->session, 
GNUNET_YES);
 
       if (n->keepalive_task == GNUNET_SCHEDULER_NO_TASK)
             n->keepalive_task = GNUNET_SCHEDULER_add_now 
(&neighbour_keepalive_task, n);
@@ -2304,9 +2312,9 @@
             n->address,
             n->session,
             GNUNET_YES);
+    GNUNET_ATS_address_in_use (GST_ats, n->address, n->session, GNUNET_YES);
     n->address_state = USED;
   }
-  GNUNET_ATS_address_in_use (GST_ats, n->address, n->session, GNUNET_YES);
 
   GST_neighbours_set_incoming_quota (&n->id, n->bandwidth_in);
 
@@ -2392,10 +2400,8 @@
   GNUNET_ATS_address_update (GST_ats, address, session, ats, ats_count);
   GNUNET_assert (n->address != NULL);
   change_state (n, S_CONNECTED);
-  GNUNET_ATS_address_in_use (GST_ats, n->address, n->session, GNUNET_YES);
 
   GST_neighbours_set_incoming_quota (&n->id, n->bandwidth_in);
-
   if (n->keepalive_task == GNUNET_SCHEDULER_NO_TASK)
         n->keepalive_task = GNUNET_SCHEDULER_add_now 
(&neighbour_keepalive_task, n);
   if (n->address_state == FRESH)
@@ -2404,6 +2410,7 @@
                                  n->address,
                                  n->session,
                                  GNUNET_YES);
+    GNUNET_ATS_address_in_use (GST_ats, n->address, n->session, GNUNET_YES);
     n->address_state = USED;
   }
   neighbours_connected++;