From f794df7d9bc052dd44387afeccdc8c4f6d87b2e0 Mon Sep 17 00:00:00 2001
From: Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>
Date: Mon, 2 Nov 2009 13:16:22 +0000
Subject: 2009-11-02  Tatsuhiro Tsujikawa  <t-tujikawa@users.sourceforge.net>

	Logged received nodes in get_peers reply message.
	* src/DHTAbstractNodeLookupTask.cc
	* src/DHTAbstractNodeLookupTask.h
---
 src/DHTAbstractNodeLookupTask.cc | 29 ++++++++++++++++-------------
 1 file changed, 16 insertions(+), 13 deletions(-)

(limited to 'src/DHTAbstractNodeLookupTask.cc')

diff --git a/src/DHTAbstractNodeLookupTask.cc b/src/DHTAbstractNodeLookupTask.cc
index f4752b0c..b38f2dbd 100644
--- a/src/DHTAbstractNodeLookupTask.cc
+++ b/src/DHTAbstractNodeLookupTask.cc
@@ -66,11 +66,14 @@ void DHTAbstractNodeLookupTask::onReceived(const SharedHandle<DHTMessage>& messa
   toEntries(newEntries, nodes);
 
   size_t count = 0;
-  for(std::deque<SharedHandle<DHTNodeLookupEntry> >::const_iterator i = newEntries.begin();
-      i != newEntries.end(); ++i) {
+  for(std::deque<SharedHandle<DHTNodeLookupEntry> >::const_iterator i =
+	newEntries.begin(); i != newEntries.end(); ++i) {
     if(memcmp(_localNode->getID(), (*i)->_node->getID(), DHT_ID_LENGTH) != 0) {
       _entries.push_front(*i);
       ++count;
+      _logger->debug("Received nodes: id=%s, ip=%s",
+		     util::toHex((*i)->_node->getID(), DHT_ID_LENGTH).c_str(),
+		     (*i)->_node->getIPAddress().c_str());
     }
   }
 
@@ -81,16 +84,7 @@ void DHTAbstractNodeLookupTask::onReceived(const SharedHandle<DHTMessage>& messa
   if(_entries.size() > DHTBucket::K) {
     _entries.erase(_entries.begin()+DHTBucket::K, _entries.end());
   }
-  if(needsAdditionalOutgoingMessage()) {
-    sendMessage();
-  }
-  if(_inFlightMessage == 0) {
-    _logger->debug("Finished node_lookup for node ID %s",
-		   util::toHex(_targetID, DHT_ID_LENGTH).c_str());
-    onFinish();
-    updateBucket();
-    _finished = true;
-  }
+  sendMessageAndCheckFinish();
 }
 
 void DHTAbstractNodeLookupTask::onTimeout(const SharedHandle<DHTNode>& node)
@@ -104,6 +98,11 @@ void DHTAbstractNodeLookupTask::onTimeout(const SharedHandle<DHTNode>& node)
       break;
     }
   }
+  sendMessageAndCheckFinish();
+}
+
+void DHTAbstractNodeLookupTask::sendMessageAndCheckFinish()
+{
   if(needsAdditionalOutgoingMessage()) {
     sendMessage();
   }
@@ -113,7 +112,11 @@ void DHTAbstractNodeLookupTask::onTimeout(const SharedHandle<DHTNode>& node)
     onFinish();
     updateBucket();
     _finished = true;
-  }  
+  } else {
+    _logger->debug("%d in flight message for node ID %s",
+		   _inFlightMessage,
+		   util::toHex(_targetID, DHT_ID_LENGTH).c_str());
+  }
 }
 
 void DHTAbstractNodeLookupTask::sendMessage()
-- 
cgit