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

[gnunet]

Author: grothoff
Date: 2009-06-05 12:27:45 -0600 (Fri, 05 Jun 2009)
New Revision: 8491

Modified:
   gnunet/src/transport/gnunet-service-transport.c
   gnunet/src/transport/plugin_transport_tcp.c
Log:
better

Modified: gnunet/src/transport/gnunet-service-transport.c
===================================================================
--- gnunet/src/transport/gnunet-service-transport.c     2009-06-05 17:28:13 UTC 
(rev 8490)
+++ gnunet/src/transport/gnunet-service-transport.c     2009-06-05 18:27:45 UTC 
(rev 8491)
@@ -915,6 +915,7 @@
 
   GNUNET_assert (mq != NULL);
   n = mq->neighbour;
+  GNUNET_assert (n != NULL);
   GNUNET_assert (0 ==
                  memcmp (&n->id, target,
                          sizeof (struct GNUNET_PeerIdentity)));
@@ -2299,6 +2300,7 @@
                   _
                   ("Dropping incoming message due to repeated bandwidth quota 
violations.\n"));
       /* TODO: call stats */
+      GNUNET_assert (NULL != service_context->neighbour);
       return service_context;
     }
   switch (ntohs (message->type))
@@ -2346,6 +2348,7 @@
         }
       GNUNET_free (im);
     }
+  GNUNET_assert (NULL != service_context->neighbour);
   return service_context;
 }
 

Modified: gnunet/src/transport/plugin_transport_tcp.c
===================================================================
--- gnunet/src/transport/plugin_transport_tcp.c 2009-06-05 17:28:13 UTC (rev 
8490)
+++ gnunet/src/transport/plugin_transport_tcp.c 2009-06-05 18:27:45 UTC (rev 
8491)
@@ -1004,7 +1004,7 @@
   struct PendingMessage *pme;
 
   if (session == NULL)
-    session = find_session_by_target (plugin, target);
+    session = find_session_by_target (plugin, target);  
   pm = GNUNET_malloc (sizeof (struct PendingMessage) + ntohs (msg->size));
   pm->msg = (struct GNUNET_MessageHeader *) &pm[1];
   memcpy (pm->msg, msg, ntohs (msg->size));
@@ -1029,6 +1029,7 @@
       session->expecting_welcome = GNUNET_YES;
       session->still_connecting = GNUNET_YES;
       session->pending_messages = pm;
+      session->service_context = service_context;
       GNUNET_PEERINFO_for_all (plugin->env->cfg,
                                plugin->env->sched,
                                target,
@@ -1037,6 +1038,7 @@
     }
   GNUNET_assert (session != NULL);
   GNUNET_assert (session->still_connecting == GNUNET_NO);
+  session->service_context = service_context;
   /* append pm to pending_messages list */
   pme = session->pending_messages;
   if (pme == NULL)