1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
|
# The contents of this file are subject to the BitTorrent Open Source License
# Version 1.1 (the License). You may not copy or use this file, in either
# source code or executable form, except in compliance with the License. You
# may obtain a copy of the License at http://www.bittorrent.com/license/.
#
# Software distributed under the License is distributed on an AS IS basis,
# WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
# for the specific language governing rights and limitations under the
# License.
from node import Node
from BitTorrent.defer import Deferred
from const import NULL_ID
from krpc import KRPCProtocolError
class IDChecker:
def __init__(self, id):
self.id = id
class KNodeBase(Node):
def __init__(self, cfa):
Node.__init__(self)
self.cfa = cfa
def conn(self):
return self.cfa((self.host, self.port))
def checkSender(self, dict):
try:
senderid = dict['rsp']['id']
except KeyError:
raise KRPCProtocolError, "No peer id in response."
else:
if self.id != NULL_ID and senderid != self.id:
self.table.table.invalidateNode(self)
else:
if self.id == NULL_ID:
self.id = senderid
self.table.insertNode(self, contacted=1)
return dict
def errBack(self, err):
self.table.table.nodeFailed(self)
return err[0]
def ping(self, id):
df = self.conn().sendRequest('ping', {"id":id})
self.conn().pinging = True
def endping(x):
self.conn().pinging = False
return x
df.addCallbacks(endping, endping)
df.addCallbacks(self.checkSender, self.errBack)
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
def inPing(self):
return self.conn().pinging
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
|
