Return a token in find_node responses, use it in store_value requests.

[quix0rs-apt-p2p.git] / apt_dht_Khashmir / node.py

diff --git a/apt_dht_Khashmir/node.py b/apt_dht_Khashmir/node.py
index 580ad9c84dc5226f04b222b21ea6ee30a7a66f23..3f04ef2724b8466e14fed017e8618e02878aa29a 100644 (file)
--- a/apt_dht_Khashmir/node.py
+++ b/apt_dht_Khashmir/node.py
@@ -30,11 +30,15 @@ class Node:
         self.num = khash.intify(id)
         self.host = host
         self.port = int(port)
+        self.token = ''
         self._contactInfo = None
     
     def updateLastSeen(self):
         self.lastSeen = datetime.now()
         self.fails = 0
+        
+    def updateToken(self, token):
+        self.token = token
     
     def msgFailed(self):
         self.fails = self.fails + 1