def __init__(self, port):
# Raul, 2012-03-09: Do not create a thread
# threading.Thread.__init__(self)
my_node = Node(('127.0.0.1', 7000), RandomId())
self.dht = Pymdht(my_node, '/sdcard/swift/', routing_m_mod,
lookup_m_mod, experimental_m_mod, None,
logging.ERROR, False)
self.rand_num = random.randint(0, 999)
self.port = port
self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.socket.settimeout(
TOAST_EACH) # This is to show that the thread is running
try:
self.socket.bind(('', port))
except (socket.error):
droid.log('EXCEP: port in use')
raise
self.channel_m = ChannelManager()
def __init__(self, port):
# Raul, 2012-03-09: Do not create a thread
# threading.Thread.__init__(self)
my_node = Node(('127.0.0.1', 7000), RandomId())
self.dht = Pymdht(my_node, '/sdcard/swift/',
routing_m_mod,
lookup_m_mod,
experimental_m_mod,
None,
logging.ERROR,
False)
self.rand_num = random.randint(0, 999)
self.port = port
self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.socket.settimeout(TOAST_EACH) # This is to show that the thread is running
try:
self.socket.bind(('', port))
except (socket.error):
droid.log('EXCEP: port in use')
raise
self.channel_m = ChannelManager()
class SwiftTraker(object):
def __init__(self, port):
# Raul, 2012-03-09: Do not create a thread
# threading.Thread.__init__(self)
my_node = Node(('127.0.0.1', 7000), RandomId())
self.dht = Pymdht(my_node, '/sdcard/swift/',
routing_m_mod,
lookup_m_mod,
experimental_m_mod,
None,
logging.ERROR,
False)
self.rand_num = random.randint(0, 999)
self.port = port
self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.socket.settimeout(TOAST_EACH) # This is to show that the thread is running
try:
self.socket.bind(('', port))
except (socket.error):
droid.log('EXCEP: port in use')
raise
self.channel_m = ChannelManager()
def start(self):#run(self):
stop_dht = False
while not stop_dht:
try:
data, addr = self.socket.recvfrom(1024)
except (socket.timeout):
droid.log('DHT alive %d' % self.rand_num)
except:
droid.log('EXCEPTION in recvfrom')
else:
stop_dht = self.handle(data, addr)
self.dht.stop()
def _on_peers_found(self, channel, peers, node):
#current_time = time.time()
# if not channel.open:
# print 'Got peers but channel is CLOSED'
# return
# if not peers:
# droid.log("DHT: end of lookup")
# print "end of lookup"
# self.channel_m.remove(channel)
# self.wfile.write('%d CLOSE\r\n' % (channel.send))
# return
new_peers = []
if peers is None:
droid.log('End of lookup')
return
for peer in peers:
if peer not in channel.peers:
channel.peers.add(peer)
new_peers.append(peer)
droid.log('DHT got %d peers' % len(channel.peers))
print 'got %d peer' % len(peers)
# for peer in new_peers:
# msg = '%d PEER %s:%d\r\n' % (channel.send,
# peer[0], peer[1])
# self.wfile.write(msg)
# return
def handle(self, data , addr):
droid.log("New connection")
if data == "KILL_DHT":
return True # stop DHT
data_len = len(data)
i = 0
print 'in: ',
remote_cid = data[:CHANNEL_SIZE]
i += CHANNEL_SIZE
print '%r' % remote_cid,
channel = self.channel_m.get(remote_cid, addr)
if not channel:
print 'Invalid channel id'
return
while i < data_len:
msg_type = ord(data[i])
i += 1
if msg_type == HANDSHAKE:
print 'HAND',
channel.remote_cid = data[i:i+CHANNEL_SIZE]
i += CHANNEL_SIZE
elif msg_type == DATA:
print 'DATA',
i = data_len # DATA always ends a datagram
elif msg_type == ACK:
print 'ACK',
i += TS_SIZE + BIN_SIZE
elif msg_type == HAVE:
print 'HAVE',
i += BIN_SIZE
elif msg_type == HASH:
print 'HASH',
i += BIN_SIZE
channel.rhash = Id(data[i:i+HASH_SIZE])
i += HASH_SIZE
elif msg_type == PEX_RES:
print 'PRES',
i += 0 #no arguments
elif msg_type == PEX_REQ:
print 'PREQ',
i += PEER_SIZE
elif msg_type == SIGNED_HASH:
print 'SHASH',
print `data`
raise NotImplemented
elif msg_type == HINT:
print 'HINT',
i += BIN_SIZE
elif msg_type == MSGTYPE_RCVD:
print 'MSGTYPE > not implemented',
print `data`
raise NotImplemented
elif msg_type == VERSION:
print 'VERSION',
i += VERSION_SIZE
else:
print 'UNKNOWN: ', msg_type,
print `data`
raise NotImplemented
print
if remote_cid == CHANNEL_ZERO and channel.rhash:
droid.log(">>>>>>> DHT: got HANDSHAKE from swift <<<<<<<")
self.dht.get_peers(channel, channel.rhash, self._on_peers_found, 0)
# need to complete handshake
reply = ''.join((channel.remote_cid,
chr(HANDSHAKE),
channel.local_cid,
))
self.socket.sendto(reply, addr)
droid.log('>>>>>>>>>>>>> GETTING PEERS <<<<<<<<<<<<<<')
reply = ''.join((channel.remote_cid,
chr(PEX_RES),
socket.inet_aton('130.161.211.194'), #Delft
chr(20050>>8),
chr(20050%256),
chr(PEX_RES),
socket.inet_aton('192.16.125.242'), #KTH
chr(20050>>8),
chr(20050%256),
))
self.socket.sendto(reply, addr)
class SwiftTraker(object):
def __init__(self, port):
# Raul, 2012-03-09: Do not create a thread
# threading.Thread.__init__(self)
my_node = Node(('127.0.0.1', 7000), RandomId())
self.dht = Pymdht(my_node, '/sdcard/swift/', routing_m_mod,
lookup_m_mod, experimental_m_mod, None,
logging.ERROR, False)
self.rand_num = random.randint(0, 999)
self.port = port
self.socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
self.socket.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
self.socket.settimeout(
TOAST_EACH) # This is to show that the thread is running
try:
self.socket.bind(('', port))
except (socket.error):
droid.log('EXCEP: port in use')
raise
self.channel_m = ChannelManager()
def start(self): #run(self):
stop_dht = False
while not stop_dht:
try:
data, addr = self.socket.recvfrom(1024)
except (socket.timeout):
droid.log('DHT alive %d' % self.rand_num)
except:
droid.log('EXCEPTION in recvfrom')
else:
stop_dht = self.handle(data, addr)
self.dht.stop()
def _on_peers_found(self, channel, peers, node):
#current_time = time.time()
# if not channel.open:
# print 'Got peers but channel is CLOSED'
# return
# if not peers:
# droid.log("DHT: end of lookup")
# print "end of lookup"
# self.channel_m.remove(channel)
# self.wfile.write('%d CLOSE\r\n' % (channel.send))
# return
new_peers = []
if peers is None:
droid.log('End of lookup')
return
for peer in peers:
if peer not in channel.peers:
channel.peers.add(peer)
new_peers.append(peer)
droid.log('DHT got %d peers' % len(channel.peers))
print 'got %d peer' % len(peers)
# for peer in new_peers:
# msg = '%d PEER %s:%d\r\n' % (channel.send,
# peer[0], peer[1])
# self.wfile.write(msg)
# return
def handle(self, data, addr):
droid.log("New connection")
if data == "KILL_DHT":
return True # stop DHT
data_len = len(data)
i = 0
print 'in: ',
remote_cid = data[:CHANNEL_SIZE]
i += CHANNEL_SIZE
print '%r' % remote_cid,
channel = self.channel_m.get(remote_cid, addr)
if not channel:
print 'Invalid channel id'
return
while i < data_len:
msg_type = ord(data[i])
i += 1
if msg_type == HANDSHAKE:
print 'HAND',
channel.remote_cid = data[i:i + CHANNEL_SIZE]
i += CHANNEL_SIZE
elif msg_type == DATA:
print 'DATA',
i = data_len # DATA always ends a datagram
elif msg_type == ACK:
print 'ACK',
i += TS_SIZE + BIN_SIZE
elif msg_type == HAVE:
print 'HAVE',
i += BIN_SIZE
elif msg_type == HASH:
print 'HASH',
i += BIN_SIZE
channel.rhash = Id(data[i:i + HASH_SIZE])
i += HASH_SIZE
elif msg_type == PEX_RES:
print 'PRES',
i += 0 #no arguments
elif msg_type == PEX_REQ:
print 'PREQ',
i += PEER_SIZE
elif msg_type == SIGNED_HASH:
print 'SHASH',
print ` data `
raise NotImplemented
elif msg_type == HINT:
print 'HINT',
i += BIN_SIZE
elif msg_type == MSGTYPE_RCVD:
print 'MSGTYPE > not implemented',
print ` data `
raise NotImplemented
elif msg_type == VERSION:
print 'VERSION',
i += VERSION_SIZE
else:
print 'UNKNOWN: ', msg_type,
print ` data `
raise NotImplemented
print
if remote_cid == CHANNEL_ZERO and channel.rhash:
droid.log(">>>>>>> DHT: got HANDSHAKE from swift <<<<<<<")
self.dht.get_peers(channel, channel.rhash, self._on_peers_found, 0)
# need to complete handshake
reply = ''.join((
channel.remote_cid,
chr(HANDSHAKE),
channel.local_cid,
))
self.socket.sendto(reply, addr)
droid.log('>>>>>>>>>>>>> GETTING PEERS <<<<<<<<<<<<<<')
reply = ''.join((
channel.remote_cid,
chr(PEX_RES),
socket.inet_aton('130.161.211.194'), #Delft
chr(20050 >> 8),
chr(20050 % 256),
chr(PEX_RES),
socket.inet_aton('192.16.127.98'), #KTH
chr(20050 >> 8),
chr(20050 % 256),
))
self.socket.sendto(reply, addr)
