def _testFirewall( tcpListener, rpcSock, nodeAddrList, localIP,
reactor, callback=None ) :
def doCancel() :
for checkOp in checkOps :
checkOp.cancel()
checkOps.clear()
testServer.close()
def onCheck( checkOp, err, payload ) :
checkOps.remove( checkOp )
if err >= 0 :
fwResult,fwToken = payload
if (fwResult >= 0) and (fwToken == token) :
successList.append( 1 )
if not checkOps :
testServer.close()
# n(success)/n(nodes) >= 1/2
op.notify( 2*len(successList) >= len(nodeAddrList) )
def doCheck( nodeAddr ) :
checkOp = dhtClient.callFirewallCheck( localIP, nodeAddr,
lambda e,p : onCheck(checkOp,e,p) )
checkOps.add( checkOp )
assert nodeAddrList
successList = []
checkOps = set()
token = rand_bytes( 20 )
dhtClient = DHTClient( rpcSock )
testServer = FirewallTestServer( tcpListener, token, reactor )
for nodeAddr in nodeAddrList :
doCheck( nodeAddr )
op = AsyncOp( callback, doCancel )
return op
def addIncoming( self, sslConn, peerKey ) :
while True :
connectionId = hexEncode( rand_bytes(8) )
if connectionId not in self.connections : break
def onTimeout() : self._onTimeout( connectionId )
timerOp = self.reactor.callLater( 30, onTimeout )
self.connections[connectionId] = (sslConn,peerKey,timerOp)
return connectionId
def request( self, data, destaddr, callback=None ) :
while 1 :
id = rand_bytes( 20 )
if not self.pending.has_key(id) : break
msg = (0, id, data)
s = encode( msg )
def cancelRequest() :
del self.pending[id]
op = RPCOp( id, callback, cancelRequest )
self.pending[id] = op
self._sendData( s, destaddr )
return op
def request(self, data, destaddr, callback=None):
while 1:
id = rand_bytes(20)
if not self.pending.has_key(id): break
msg = (0, id, data)
s = encode(msg)
def cancelRequest():
del self.pending[id]
op = RPCOp(id, callback, cancelRequest)
self.pending[id] = op
self._sendData(s, destaddr)
return op
def _doRegister( self, args, ctx ) :
try :
assert self.state == self.INITIAL
assert len(args) == 0
except :
ctx.response( (-1,'') )
return
while True :
routerId = rand_bytes( 20 )
if routerId not in self.routerState.registered : break
self.routerId = routerId
self.state = self.REGISTERED
self.routerState.registered[routerId] = self
ctx.response( (0,self.routerId) )
def _doConnect( self, args, ctx ) :
try :
assert self.state == self.INITIAL
assert len(args) == 1
routerId = args[0]
assert type(routerId) is str
handler = self.routerState.registered.get( routerId )
assert handler is not None
except :
ctx.response( [-1] )
return
while True :
connectionId = rand_bytes( 20 )
if connectionId not in self.routerState.connecting : break
self.connectionId = connectionId
self.connectCtx = ctx
self.state = self.CONNECTING
self.routerState.connecting[connectionId] = self
handler.sendIncoming( connectionId )
def _testFirewall(tcpListener,
rpcSock,
nodeAddrList,
localIP,
reactor,
callback=None):
def doCancel():
for checkOp in checkOps:
checkOp.cancel()
checkOps.clear()
testServer.close()
def onCheck(checkOp, err, payload):
checkOps.remove(checkOp)
if err >= 0:
fwResult, fwToken = payload
if (fwResult >= 0) and (fwToken == token):
successList.append(1)
if not checkOps:
testServer.close()
# n(success)/n(nodes) >= 1/2
op.notify(2 * len(successList) >= len(nodeAddrList))
def doCheck(nodeAddr):
checkOp = dhtClient.callFirewallCheck(
localIP, nodeAddr, lambda e, p: onCheck(checkOp, e, p))
checkOps.add(checkOp)
assert nodeAddrList
successList = []
checkOps = set()
token = rand_bytes(20)
dhtClient = DHTClient(rpcSock)
testServer = FirewallTestServer(tcpListener, token, reactor)
for nodeAddr in nodeAddrList:
doCheck(nodeAddr)
op = AsyncOp(callback, doCancel)
return op