1 files changed, 20 insertions, 0 deletions
diff --git a/src/DHTPeerLookupTaskCallback.cc b/src/DHTPeerLookupTaskCallback.cc
index 5343dc70..49a9827a 100644
--- a/src/DHTPeerLookupTaskCallback.cc
+++ b/src/DHTPeerLookupTaskCallback.cc
@@ -34,17 +34,37 @@
 /* copyright --> */
 #include "DHTPeerLookupTaskCallback.h"
 #include "DHTPeerLookupTask.h"
+#include "DHTAnnouncePeerReplyMessage.h"
+#include "DHTFindNodeReplyMessage.h"
+#include "DHTPingReplyMessage.h"
+#include "Peer.h"
 
 namespace aria2 {
 
 DHTPeerLookupTaskCallback::DHTPeerLookupTaskCallback(DHTPeerLookupTask* task):
   _task(task) {}
 
+void DHTPeerLookupTaskCallback::visit
+(const DHTAnnouncePeerReplyMessage* message)
+{
+  onTimeout(message->getRemoteNode());
+}
+
+void DHTPeerLookupTaskCallback::visit(const DHTFindNodeReplyMessage* message)
+{
+  onTimeout(message->getRemoteNode());
+}
+
 void DHTPeerLookupTaskCallback::visit(const DHTGetPeersReplyMessage* message)
 {
   _task->onReceived(message);
 }
 
+void DHTPeerLookupTaskCallback::visit(const DHTPingReplyMessage* message)
+{
+  onTimeout(message->getRemoteNode());
+}
+
 void DHTPeerLookupTaskCallback::onTimeout
 (const SharedHandle<DHTNode>& remoteNode)
 {