X-Git-Url: https://git.mxchange.org/?a=blobdiff_plain;f=knode.py;h=d2cea727766f3d9865febb7bdc46043a99307f5c;hb=f9aff197ceaac1f35a2c727e9c9d1f1964245f80;hp=ef8226034f993926411fe6427d0a29eb60c09e6a;hpb=2ce2ba6146457d6c321e8ece8116d422cda39a70;p=quix0rs-apt-p2p.git

diff --git a/knode.py b/knode.py
index ef82260..d2cea72 100644
--- a/knode.py
+++ b/knode.py
@@ -1,39 +1,53 @@
+## Copyright 2002-2004 Andrew Loewenstern, All Rights Reserved
+# see LICENSE.txt for license information
+
+from const import NULL_ID
 from node import Node
-from twisted.internet.defer import Deferred
-from xmlrpcclient import XMLRPCClientFactory as factory
-from const import reactor, NULL_ID
 
-class KNode(Node):
-	def makeResponse(self, df):
-		""" Make our callback cover that checks to make sure the id of the response is the same as what we are expecting """
-		def _callback(args, d=df):
-			try:
-				l, sender = args
-			except:
-				d.callback(args)
-			else:
-				if self.id != NULL_ID and sender['id'] != self._senderDict['id']:
-					d.errback()
-				else:
-					d.callback(args)
-		return _callback
-	def ping(self, sender):
-		df = Deferred()
-		f = factory('ping', (sender,), self.makeResponse(df), df.errback)
-		reactor.connectTCP(self.host, self.port, f)
-		return df
-	def findNode(self, target, sender):
-		df = Deferred()
-		f = factory('find_node', (target.encode('base64'), sender), self.makeResponse(df), df.errback)
-		reactor.connectTCP(self.host, self.port, f)
-		return df
-	def storeValue(self, key, value, sender):
-		df = Deferred()
-		f = factory('store_value', (key.encode('base64'), value.encode('base64'), sender), self.makeResponse(df), df.errback)
-		reactor.connectTCP(self.host, self.port, f)
-		return df
-	def findValue(self, key, sender):
-		df = Deferred()
-		f = factory('find_value', (key.encode('base64'), sender), self.makeResponse(df), df.errback)
-		reactor.connectTCP(self.host, self.port, f)
-		return df
+class KNodeBase(Node):
+    def checkSender(self, dict):
+        try:
+            senderid = dict['rsp']['id']
+        except KeyError:
+            print ">>>> No peer id in response"
+            raise Exception, "No peer id in response."
+        else:
+            if self.id != NULL_ID and senderid != self.id:
+                print "Got response from different node than expected."
+                self.table.invalidateNode(self)
+                
+        return dict
+
+    def errBack(self, err):
+        print ">>> ", err
+        return err
+        
+    def ping(self, id):
+        df = self.conn.sendRequest('ping', {"id":id})
+        df.addErrback(self.errBack)
+        df.addCallback(self.checkSender)
+        return df
+    def findNode(self, target, id):
+        df = self.conn.sendRequest('find_node', {"target" : target, "id": id})
+        df.addErrback(self.errBack)
+        df.addCallback(self.checkSender)
+        return df
+
+class KNodeRead(KNodeBase):
+    def findValue(self, key, id):
+        df =  self.conn.sendRequest('find_value', {"key" : key, "id" : id})
+        df.addErrback(self.errBack)
+        df.addCallback(self.checkSender)
+        return df
+
+class KNodeWrite(KNodeRead):
+    def storeValue(self, key, value, id):
+        df = self.conn.sendRequest('store_value', {"key" : key, "value" : value, "id": id})
+        df.addErrback(self.errBack)
+        df.addCallback(self.checkSender)
+        return df
+    def storeValues(self, key, value, id):
+        df = self.conn.sendRequest('store_values', {"key" : key, "values" : value, "id": id})
+        df.addErrback(self.errBack)
+        df.addCallback(self.checkSender)
+        return df