1 ## Copyright 2002-2003 Andrew Loewenstern, All Rights Reserved
2 # see LICENSE.txt for license information
5 from twisted.internet.defer import Deferred
6 from twisted.protocols import basic
7 from bencode import bencode, bdecode
8 from twisted.internet import protocol
10 from twisted.internet import reactor
14 from traceback import format_exception
21 KRPC_ERROR_METHOD_UNKNOWN = 2
22 KRPC_ERROR_RECEIVED_UNKNOWN = 3
23 KRPC_ERROR_TIMEOUT = 4
33 class hostbroker(protocol.DatagramProtocol):
34 def __init__(self, server):
36 # this should be changed to storage that drops old entries
39 def datagramReceived(self, datagram, addr):
40 #print `addr`, `datagram`
41 #if addr != self.addr:
42 c = self.connectionForAddr(addr)
43 c.datagramReceived(datagram, addr)
45 # del self.connections[addr]
47 def connectionForAddr(self, addr):
50 if not self.connections.has_key(addr):
51 conn = self.protocol(addr, self.server, self.transport)
52 self.connections[addr] = conn
54 conn = self.connections[addr]
57 def makeConnection(self, transport):
58 protocol.DatagramProtocol.makeConnection(self, transport)
59 tup = transport.getHost()
60 self.addr = (tup.host, tup.port)
65 def __init__(self, addr, server, transport):
66 self.transport = transport
72 def datagramReceived(self, str, addr):
78 print "response decode error: " + `e`
87 # tell factory to handle
88 f = getattr(self.factory ,"krpc_" + msg[REQ], None)
89 msg[ARG]['_krpc_sender'] = self.addr
92 ret = apply(f, (), msg[ARG])
95 out = bencode({TID:msg[TID], TYP:ERR, ERR :`format_exception(type(e), e, sys.exc_info()[2])`})
97 self.transport.write(out, addr)
101 out = bencode({TID : msg[TID], TYP : RSP, RSP : ret})
103 out = bencode({TID : msg[TID], TYP : RSP, RSP : {}})
106 self.transport.write(out, addr)
110 print "don't know about method %s" % msg[REQ]
112 out = bencode({TID:msg[TID], TYP:ERR, ERR : KRPC_ERROR_METHOD_UNKNOWN})
114 self.transport.write(out, addr)
116 print "%s %s >>> %s - %s %s %s" % (time.asctime(), addr, self.factory.node.port,
117 ilen, msg[REQ], olen)
118 elif msg[TYP] == RSP:
121 if self.tids.has_key(msg[TID]):
122 df = self.tids[msg[TID]]
124 del(self.tids[msg[TID]])
125 df.callback({'rsp' : msg[RSP], '_krpc_sender': addr})
127 print 'timeout ' + `msg[RSP]['id']`
128 # no tid, this transaction timed out already...
129 elif msg[TYP] == ERR:
132 if self.tids.has_key(msg[TID]):
133 df = self.tids[msg[TID]]
136 del(self.tids[msg[TID]])
138 # day late and dollar short
141 print "unknown message type " + `msg`
142 # unknown message type
143 df = self.tids[msg[TID]]
145 df.errback(KRPC_ERROR_RECEIVED_UNKNOWN)
146 del(self.tids[msg[TID]])
148 def sendRequest(self, method, args):
151 msg = {TID : chr(self.mtid), TYP : REQ, REQ : method, ARG : args}
152 self.mtid = (self.mtid + 1) % 256
155 self.tids[msg[TID]] = d
156 def timeOut(tids = self.tids, id = msg[TID]):
160 print ">>>>>> KRPC_ERROR_TIMEOUT"
161 df.errback(KRPC_ERROR_TIMEOUT)
162 reactor.callLater(KRPC_TIMEOUT, timeOut)
163 self.transport.write(str, self.addr)