From 920583502e654e8ae3e47ecb8b5df515b2d072f2 Mon Sep 17 00:00:00 2001
From: Tatsuhiro Tsujikawa <tatsuhiro.t@gmail.com>
Date: Fri, 7 Jan 2011 00:14:42 +0900
Subject: Consider IP address and port when finding DHT node.

---
 src/DHTBucket.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'src/DHTBucket.cc')

diff --git a/src/DHTBucket.cc b/src/DHTBucket.cc
index ee5d0cae..04d2bb30 100644
--- a/src/DHTBucket.cc
+++ b/src/DHTBucket.cc
@@ -232,7 +232,8 @@ SharedHandle<DHTNode> DHTBucket::getNode(const unsigned char* nodeID, const std:
   node->setPort(port);
   std::deque<SharedHandle<DHTNode> >::const_iterator itr =
     std::find_if(nodes_.begin(), nodes_.end(), derefEqual(node));
-  if(itr == nodes_.end()) {
+  if(itr == nodes_.end() ||
+     (*itr)->getIPAddress() != ipaddr || (*itr)->getPort() != port) {
     return SharedHandle<DHTNode>();
   } else {
     return *itr;
-- 
cgit