def __init__(self,
torrent_path,
directory='',
port=55308,
download_all=False,
visualizer=None):
torrent_dict = tparser.bdecode_file(torrent_path)
self.torrent_dict = torrent_dict
self.peer_dict = {}
self.peer_ips = []
self.port = port
self.download_all = download_all
self.r = None
self.tracker_response = None
self.peer_dict = {}
self.hash_string = None
self.queued_requests = []
self.vis_write_sock = ''
self.reactor = reactor.Reactor()
self.reactor.add_listeners([
PeerListener(torrent=self, port=7000),
VisListener(torrent=self, port=8035)
])
# Try to connect to visualization server
vis_socket = connect_vis(visualizer) if visualizer else None
if directory:
os.chdir(directory)
if 'files' in self.torrent_dict['info']:
dirname = self.torrent_dict['info']['name']
else:
dirname = None
file_list = []
# Multifile case
if 'files' in self.torrent_dict['info']:
file_list.extend(self.torrent_dict['info']['files'])
multifile = True
# Deal with single-file torrents by building up the kind of dict
# found in torrent_dict['info']['files']
elif 'name' in self.torrent_dict['info']:
info_dict = {}
info_dict['path'] = self.torrent_dict['info']['name']
info_dict['length'] = self.torrent_dict['info']['length']
file_list.append(info_dict)
multifile = False
else:
raise Exception('Invalid .torrent file')
self.switchboard = Switchboard(dirname=dirname,
file_list=file_list,
piece_length=self.piece_length,
num_pieces=self.num_pieces,
multifile=multifile,
download_all=download_all,
vis_socket=vis_socket)
def __init__(self, torrent_path, directory='', port=55308,
download_all=False, visualizer=None):
print("torrent path is...")
print(torrent_path)
torrent_dict = tparser.bdecode_file(torrent_path)
print(torrent_dict)
self.torrent_dict = torrent_dict
self.peer_dict = {}
self.peer_ips = []
self.port = port
self.download_all = download_all
self.r = None
self.is_udp = False
self.tracker_response = None
self.peer_dict = {}
self.hash_string = None
self.queued_requests = []
self.vis_write_sock = ''
self.reactor = reactor.Reactor()
self.reactor.add_listeners([PeerListener(torrent=self, port=7000),
VisListener(torrent=self, port=8035)])
# Try to connect to visualization server
vis_socket = connect_vis(visualizer) if visualizer else None
if directory:
os.chdir(directory)
if 'files' in self.torrent_dict['info']:
dirname = self.torrent_dict['info']['name']
else:
dirname = None
file_list = []
# Multifile case
if 'files' in self.torrent_dict['info']:
file_list.extend(self.torrent_dict['info']['files'])
multifile = True
# Deal with single-file torrents by building up the kind of dict
# found in torrent_dict['info']['files']
elif 'name' in self.torrent_dict['info']:
info_dict = {}
info_dict['path'] = self.torrent_dict['info']['name']
info_dict['length'] = self.torrent_dict['info']['length']
file_list.append(info_dict)
multifile = False
else:
raise Exception('Invalid .torrent file')
self.switchboard = Switchboard(dirname=dirname,
file_list=file_list, piece_length=
self.piece_length, num_pieces=
self.num_pieces, multifile=multifile,
download_all=download_all,
vis_socket=vis_socket)
class Torrent(object):
def __init__(self, torrent_path, directory='', port=55308,
download_all=False, visualizer=None):
print("torrent path is...")
print(torrent_path)
torrent_dict = tparser.bdecode_file(torrent_path)
print(torrent_dict)
self.torrent_dict = torrent_dict
self.peer_dict = {}
self.peer_ips = []
self.port = port
self.download_all = download_all
self.r = None
self.is_udp = False
self.tracker_response = None
self.peer_dict = {}
self.hash_string = None
self.queued_requests = []
self.vis_write_sock = ''
self.reactor = reactor.Reactor()
self.reactor.add_listeners([PeerListener(torrent=self, port=7000),
VisListener(torrent=self, port=8035)])
# Try to connect to visualization server
vis_socket = connect_vis(visualizer) if visualizer else None
if directory:
os.chdir(directory)
if 'files' in self.torrent_dict['info']:
dirname = self.torrent_dict['info']['name']
else:
dirname = None
file_list = []
# Multifile case
if 'files' in self.torrent_dict['info']:
file_list.extend(self.torrent_dict['info']['files'])
multifile = True
# Deal with single-file torrents by building up the kind of dict
# found in torrent_dict['info']['files']
elif 'name' in self.torrent_dict['info']:
info_dict = {}
info_dict['path'] = self.torrent_dict['info']['name']
info_dict['length'] = self.torrent_dict['info']['length']
file_list.append(info_dict)
multifile = False
else:
raise Exception('Invalid .torrent file')
self.switchboard = Switchboard(dirname=dirname,
file_list=file_list, piece_length=
self.piece_length, num_pieces=
self.num_pieces, multifile=multifile,
download_all=download_all,
vis_socket=vis_socket)
@property
def piece_length(self):
return self.torrent_dict['info']['piece length']
@property
def num_pieces(self):
num, rem = divmod(len(self.torrent_dict['info']['pieces']), 20)
if rem == 0:
return num
else:
raise Exception("Improperly formed 'pieces' entry in torrent_dict")
@property
def length(self):
if 'files' in self.torrent_dict['info']:
return sum([i['length'] for i in
self.torrent_dict['info']['files']])
else:
return self.torrent_dict['info']['length']
@property
def last_piece_length(self):
return self.length - (self.piece_length * (self.num_pieces - 1))
@property
def last_piece(self):
return self.num_pieces - 1
def build_payload(self):
'''
Builds the payload that will be sent in tracker_request
'''
payload = {}
hashed_info = hashlib.sha1(tparser.bencode(self.torrent_dict['info']))
self.hash_string = hashed_info.digest()
self.peer_id = ('-DR' + VERSION +
''.join(random.sample(ALPHANUM, 13)))
assert len(self.peer_id) == 20
payload['info_hash'] = self.hash_string
payload['peer_id'] = self.peer_id
payload['port'] = self.port
payload['uploaded'] = 0
payload['downloaded'] = 0
payload['left'] = self.length
payload['compact'] = 1
payload['supportcrypto'] = 1
payload['event'] = 'started'
return payload
def handle_udp(self,payload):
ip_to_socket = dict()
lol = self.torrent_dict['announce-list']
lol = lol[0:1]
for litem in lol:
print(litem)
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
last = litem.pop()
ip_to_socket[last] = sock
a = last.split("://")
protocol = a[0]
url = a[1]
url_po = url.split(":")
url = url_po[0]
port = url_po[1].split("/announce")[0]
#print(protocol, url, port)
i = (url,int(port))
sock.setblocking(False)
sock.settimeout(3)
sock.connect(i)
connection_id = 0x41727101980
action = 0
transaction_id = randrange(1,65535)
request = struct.pack('>QLL',connection_id, action, transaction_id)
sent = sock.send(request)
response = sock.recv(68)
action,transaction_id,connection_id=struct.unpack(">LLQ",response)
cid = connection_id
action = 1
tid = transaction_id
info_hash = payload["info_hash"]
peer_id = payload["peer_id"]
downloaded = payload["downloaded"]
left = payload["left"]
uploaded = payload["uploaded"]
event = payload["event"]
ip = 0
key = 0
num_want = 10
port = DEFAULT_PORT
event_hash = dict()
event_hash['none'] = 0
event_hash['completed'] = 1
event_hash['started'] = 2
event_hash['stopped'] = 3
lst = [cid,action,tid,info_hash,peer_id,downloaded,uploaded,left,event_hash[event],ip,key,num_want,port]
for x in lst:
print(x, type(x))
request = struct.pack('>QLL20s20sQQQLLLLH',cid,action,tid,info_hash,peer_id,downloaded,uploaded,left,event_hash[event],ip,key,num_want,port)
sent = sock.send(request)
print("Send!!!")
response = sock.recv(98)
#print(len(response))
# act, tr_id, interval, leechers, seeders = struct.unpack('>LLLLL', response)
action = struct.unpack('>LLLLL', response[:20])
(a,b,c,d,e) = action
lenbytes = len(bytes(response[20:]))
print(lenbytes)
num_peers = lenbytes/6
newstr = 'LH'*num_peers
rest_of_response = struct.unpack(">"+newstr,bytes(response[20:]))
print(rest_of_response)
ips = []
ports = []
for ind, item in enumerate(rest_of_response):
if ind % 2 == 0:
ips.append(item)
else:
ports.append(item)
couple = []
for ind, item in enumerate(ips):
couple.append((self.int_to_ip(ips[ind]), ports[ind]))
self.peer_ips.append((self.int_to_ip(ips[ind]), ports[ind]))
for x in couple: print(x)
# TODO -- refactor?
def tracker_request(self):
'''
Sends the initial request to the tracker, compiling list of all peers
announcing to the tracker
'''
assert self.torrent_dict['info']
payload = self.build_payload()
if self.torrent_dict['announce'].startswith('udp'):
# self.r = requests.get(self.torrent_dict['announce'],
# params=payload)
# print(self.torrent_dict.keys())
# print(self.torrent_dict["announce"])
self.is_udp = True
# print(payload)
self.handle_udp(payload)
# raise Exception('need to deal with UDP')
else:
self.r = requests.get(self.torrent_dict['announce'],
params=payload)
print(self.torrent_dict['announce'])
# Decoding response from tracker
print("Content")
print(self.r.content)
self.tracker_response = tparser.bdecode(self.r.content)
print("tracker response is ...")
print(self.tracker_response)
self.get_peer_ips()
def get_peer_ips(self):
'''
Generates list of peer IPs from tracker response. Note: not all of
these IPs might be good, which is why we only init peer objects for
the subset that respond to handshake
'''
presponse = [ord(i) for i in self.tracker_response['peers']]
while presponse:
peer_ip = (('.'.join(str(x) for x in presponse[0:4]),
256 * presponse[4] + presponse[5]))
print(peer_ip)
if peer_ip not in self.peer_ips:
self.peer_ips.append(peer_ip)
presponse = presponse[6:]
def int_to_ip(self, i):
return str((i >> 24) & 0xFF) + "." + str((i >> 16) & 0xFF) + "." + str((i >> 8) & 0xFF) + "." + str(i & 0xFF)
# TODO -- refactor this so it takes peer IPs or
# sockets (in the case of incoming connections)
def handshake_peers(self):
'''
pstrlen = length of pstr as one byte
pstr = BitTorrent protocol
reserved = chr(0)*8
info_hash = 20-byte hash above (aka self.hash_string)
peer_id = 20-byte string
'''
pstr = 'BitTorrent protocol'
pstrlen = len(pstr)
info_hash = self.hash_string
peer_id = self.peer_id
packet = ''.join([chr(pstrlen), pstr, chr(0) * 8, info_hash,
peer_id])
print("packet: ")
print(packet)
print "Here's my packet {}".format(repr(packet))
# TODO -- add some checks in here so that I'm talking
# to a maximum of 30 peers
# TODO -- think about why i'm deleting self.peer_ips.
# What was the point of it? Why won't I need it?
# Think about what we're doing -- using this list to create
# new peer objects. Should make this functional, that way I
# can also call when I get new peers.
for i in self.peer_ips:
if len(self.peer_dict) >= 30:
break
try:
s = socket.create_connection(l, 5, ('0.0.0.0',DEFAULT_PORT))
#s = socket.socket()
# connection_id = 0x41727101980
# action = 0
# transaction_id = randrange(1,65535)
# request = struct.pack('>QLL',connection_id, action, transaction_id)
s.send(packet)
try:
data = s.recv(68) # Peer's handshake - len from docs
if data:
print 'From {} received: {}'.format(s.fileno(), repr(data))
self.initpeer(s)
except:
print '{} timed out on recv'.format(s.fileno())
except socket.timeout:
print '{} timed out on connect'.format(s.fileno())
continue
except socket.error:
print '{} threw a socket error'.format(s.fileno())
continue
except Exception as ex:
template = "An exception of type {0} occured. Arguments:\n{1!r}"
message = template.format(type(ex).__name__, ex.args)
print message
else:
self.peer_ips = []
def initpeer(self, sock):
'''
Creates a new peer object for a valid socket and adds it to reactor's
listen list
'''
location_json = requests.request("GET", "http://freegeoip.net/json/"
+ sock.getpeername()[0]).content
print(location_json)
location = json.loads(location_json)
tpeer = peer.Peer(sock, self.reactor, self, location)
self.peer_dict[sock] = tpeer
self.reactor.select_list.append(tpeer)
def add_peer(self, sock):
print 'adding peer at', sock.getpeername()
time.sleep(3)
def kill_peer(self, tpeer):
thispeer = self.peer_dict.pop(tpeer.sock)
print 'peer with fileno {} killing itself'.format(thispeer.fileno())
self.reactor.select_list.remove(thispeer)
def set_sock(self, sock):
self.vis_write_sock = sock
self.switchboard.vis_write_sock = self.vis_write_sock
def __enter__(self):
pass
def __exit__(self, type, value, tb):
self.switchboard.close()
class Torrent(object):
def __init__(self,
torrent_path,
directory='',
port=55308,
download_all=False,
visualizer=None):
torrent_dict = tparser.bdecode_file(torrent_path)
self.torrent_dict = torrent_dict
self.peer_dict = {}
self.peer_ips = []
self.port = port
self.download_all = download_all
self.r = None
self.tracker_response = None
self.peer_dict = {}
self.hash_string = None
self.queued_requests = []
self.vis_write_sock = ''
self.reactor = reactor.Reactor()
self.reactor.add_listeners([
PeerListener(torrent=self, port=7000),
VisListener(torrent=self, port=8035)
])
# Try to connect to visualization server
vis_socket = connect_vis(visualizer) if visualizer else None
if directory:
os.chdir(directory)
if 'files' in self.torrent_dict['info']:
dirname = self.torrent_dict['info']['name']
else:
dirname = None
file_list = []
# Multifile case
if 'files' in self.torrent_dict['info']:
file_list.extend(self.torrent_dict['info']['files'])
multifile = True
# Deal with single-file torrents by building up the kind of dict
# found in torrent_dict['info']['files']
elif 'name' in self.torrent_dict['info']:
info_dict = {}
info_dict['path'] = self.torrent_dict['info']['name']
info_dict['length'] = self.torrent_dict['info']['length']
file_list.append(info_dict)
multifile = False
else:
raise Exception('Invalid .torrent file')
self.switchboard = Switchboard(dirname=dirname,
file_list=file_list,
piece_length=self.piece_length,
num_pieces=self.num_pieces,
multifile=multifile,
download_all=download_all,
vis_socket=vis_socket)
@property
def piece_length(self):
return self.torrent_dict['info']['piece length']
@property
def num_pieces(self):
num, rem = divmod(len(self.torrent_dict['info']['pieces']), 20)
if rem == 0:
return num
else:
raise Exception("Improperly formed 'pieces' entry in torrent_dict")
@property
def length(self):
if 'files' in self.torrent_dict['info']:
return sum(
[i['length'] for i in self.torrent_dict['info']['files']])
else:
return self.torrent_dict['info']['length']
@property
def last_piece_length(self):
return self.length - (self.piece_length * (self.num_pieces - 1))
@property
def last_piece(self):
return self.num_pieces - 1
def build_payload(self):
'''
Builds the payload that will be sent in tracker_request
'''
payload = {}
hashed_info = hashlib.sha1(tparser.bencode(self.torrent_dict['info']))
self.hash_string = hashed_info.digest()
self.peer_id = ('-DR' + VERSION + ''.join(random.sample(ALPHANUM, 13)))
assert len(self.peer_id) == 20
payload['info_hash'] = self.hash_string
payload['peer_id'] = self.peer_id
payload['port'] = self.port
payload['uploaded'] = 0
payload['downloaded'] = 0
payload['left'] = self.length
payload['compact'] = 1
payload['supportcrypto'] = 1
payload['event'] = 'started'
return payload
#ペイロードとは、IT用語としては、パケット通信においてパケットに含まれるヘッダやトレーラなどの付加的情報を除いた、データ本体のことである。
# TODO -- refactor?
def tracker_request(self):
'''
Sends the initial request to the tracker, compiling list of all peers
announcing to the tracker
最初のリクエストをトラッカーに送信し、トラッカーにアナウンスするすべてのピアのリストをコンパイルします
'''
assert self.torrent_dict['info']
payload = self.build_payload()
if self.torrent_dict['announce'].startswith('udp'):
raise Exception('need to deal with UDP')
else:
self.r = requests.get(self.torrent_dict['announce'],
params=payload)
# Decoding response from tracker
self.tracker_response = tparser.bdecode(self.r.content)
self.get_peer_ips()
def get_peer_ips(self):
'''
Generates list of peer IPs from tracker response. Note: not all of
these IPs might be good, which is why we only init peer objects for
the subset that respond to handshake
トラッカー応答からピアIPのリストを生成します。注：すべてではありません
これらのIPは良いかもしれません。そのため、ピアオブジェクトを初期化するのは
ハンドシェイクに応答するサブセット
'''
presponse = [ord(i) for i in self.tracker_response['peers']]
while presponse:
peer_ip = (('.'.join(str(x) for x in presponse[0:4]),
256 * presponse[4] + presponse[5]))
if peer_ip not in self.peer_ips:
self.peer_ips.append(peer_ip)
presponse = presponse[6:]
# TODO -- refactor this so it takes peer IPs or
# sockets (in the case of incoming connections)
def handshake_peers(self):
'''
pstrlen = length of pstr as one byte
pstr = BitTorrent protocol
reserved = chr(0)*8
info_hash = 20-byte hash above (aka self.hash_string)
peer_id = 20-byte string
-------------------------
pstrlen = 1バイトとしてのpstrの長さ
pstr = BitTorrentプロトコル
予約済み= chr（0）* 8
info_hash =上記の20バイトのハッシュ（別名self.hash_string）
peer_id = 20バイトの文字列
'''
pstr = 'BitTorrent protocol'
pstrlen = len(pstr)
info_hash = self.hash_string
peer_id = self.peer_id
packet = ''.join([chr(pstrlen), pstr, chr(0) * 8, info_hash, peer_id])
print "Here's my packet {}".format(repr(packet))
# TODO -- add some checks in here so that I'm talking
# to a maximum of 30 peers
# TODO -- think about why i'm deleting self.peer_ips.
# What was the point of it? Why won't I need it?
# Think about what we're doing -- using this list to create
# new peer objects. Should make this functional, that way I
# can also call when I get new peers.
for i in self.peer_ips:
if len(self.peer_dict) >= 30:
break
s = socket.socket()
s.setblocking(True)
s.settimeout(0.5)
try:
s.connect(i)
except socket.timeout:
print '{} timed out on connect'.format(s.fileno())
continue
except socket.error:
print '{} threw a socket error'.format(s.fileno())
continue
except:
raise Exception
s.send(packet)
try:
data = s.recv(68) # Peer's handshake - len from docs
if data:
print 'From {} received: {}'.format(s.fileno(), repr(data))
self.initpeer(s)
except:
print '{} timed out on recv'.format(s.fileno())
continue
else:
self.peer_ips = []
def initpeer(self, sock):
'''
Creates a new peer object for a nvalid socket and adds it to reactor's
listen list
nvalidソケットの新しいピアオブジェクトを作成し、reactorのに追加します
'''
location_json = requests.request(
"GET",
"http://freegeoip.net/json/" + sock.getpeername()[0]).content
location = json.loads(location_json)
tpeer = peer.Peer(sock, self.reactor, self, location)
self.peer_dict[sock] = tpeer
self.reactor.select_list.append(tpeer)
def add_peer(self, sock):
print 'adding peer at', sock.getpeername()
time.sleep(3)
def kill_peer(self, tpeer):
thispeer = self.peer_dict.pop(tpeer.sock)
print 'peer with fileno {} killing itself'.format(thispeer.fileno())
self.reactor.select_list.remove(thispeer)
def set_sock(self, sock):
self.vis_write_sock = sock
self.switchboard.vis_write_sock = self.vis_write_sock
def __enter__(self):
pass
def __exit__(self, type, value, tb):
self.switchboard.close()
from flask import Flask
from switchboard import Switchboard, Workflow
app = Flask(__name__)
class TestWorkflow(Workflow):
pid, form, status = 17921, 'demographics', 2
def execute(self, trigger):
print "ah ha"
app.config['SWITCHBOARD_WORKFLOWS'] = [TestWorkflow()]
Switchboard(app)
@app.route('/', methods=['GET'])
def index():
return "Hello World", 200
if __name__ == '__main__':
app.run(host='0.0.0.0', debug=True)
class Torrent(object):
def __init__(self, torrent_path, directory='', port=55308,
download_all=False, visualizer=None):
torrent_dict = tparser.bdecode_file(torrent_path)
self.torrent_dict = torrent_dict
self.peer_dict = {}
self.peer_ips = []
self.port = port
self.download_all = download_all
self.r = None
self.tracker_response = None
self.peer_dict = {}
self.hash_string = None
self.queued_requests = []
self.vis_write_sock = ''
self.reactor = reactor.Reactor()
self.reactor.add_listeners([PeerListener(torrent=self, port=7000),
VisListener(torrent=self, port=8035)])
# Try to connect to visualization server
vis_socket = connect_vis(visualizer) if visualizer else None
if directory:
os.chdir(directory)
if 'files' in self.torrent_dict['info']:
dirname = self.torrent_dict['info']['name']
else:
dirname = None
file_list = []
# Multifile case
if 'files' in self.torrent_dict['info']:
file_list.extend(self.torrent_dict['info']['files'])
multifile = True
# Deal with single-file torrents by building up the kind of dict
# found in torrent_dict['info']['files']
elif 'name' in self.torrent_dict['info']:
info_dict = {}
info_dict['path'] = self.torrent_dict['info']['name']
info_dict['length'] = self.torrent_dict['info']['length']
file_list.append(info_dict)
multifile = False
else:
raise Exception('Invalid .torrent file')
self.switchboard = Switchboard(dirname=dirname,
file_list=file_list, piece_length=
self.piece_length, num_pieces=
self.num_pieces, multifile=multifile,
download_all=download_all,
vis_socket=vis_socket)
@property
def piece_length(self):
return self.torrent_dict['info']['piece length']
@property
def num_pieces(self):
num, rem = divmod(len(self.torrent_dict['info']['pieces']), 20)
if rem == 0:
return num
else:
raise Exception("Improperly formed 'pieces' entry in torrent_dict")
@property
def length(self):
if 'files' in self.torrent_dict['info']:
return sum([i['length'] for i in
self.torrent_dict['info']['files']])
else:
return self.torrent_dict['info']['length']
@property
def last_piece_length(self):
return self.length - (self.piece_length * (self.num_pieces - 1))
@property
def last_piece(self):
return self.num_pieces - 1
def build_payload(self):
'''
Builds the payload that will be sent in tracker_request
'''
payload = {}
hashed_info = hashlib.sha1(tparser.bencode(self.torrent_dict['info']))
self.hash_string = hashed_info.digest()
self.peer_id = ('-DR' + VERSION +
''.join(random.sample(ALPHANUM, 13)))
assert len(self.peer_id) == 20
payload['info_hash'] = self.hash_string
payload['peer_id'] = self.peer_id
payload['port'] = self.port
payload['uploaded'] = 0
payload['downloaded'] = 0
payload['left'] = self.length
payload['compact'] = 1
payload['supportcrypto'] = 1
payload['event'] = 'started'
return payload
# TODO -- refactor?
def tracker_request(self):
'''
Sends the initial request to the tracker, compiling list of all peers
announcing to the tracker
'''
assert self.torrent_dict['info']
payload = self.build_payload()
if self.torrent_dict['announce'].startswith('udp'):
raise Exception('need to deal with UDP')
else:
self.r = requests.get(self.torrent_dict['announce'],
params=payload)
# Decoding response from tracker
self.tracker_response = tparser.bdecode(self.r.text.encode('latin-1'))
self.get_peer_ips()
def get_peer_ips(self):
'''
Generates list of peer IPs from tracker response. Note: not all of
these IPs might be good, which is why we only init peer objects for
the subset that respond to handshake
'''
presponse = [ord(i) for i in self.tracker_response['peers']]
while presponse:
peer_ip = (('.'.join(str(x) for x in presponse[0:4]),
256 * presponse[4] + presponse[5]))
if peer_ip not in self.peer_ips:
self.peer_ips.append(peer_ip)
presponse = presponse[6:]
# TODO -- refactor this so it takes peer IPs or
# sockets (in the case of incoming connections)
def handshake_peers(self):
'''
pstrlen = length of pstr as one byte
pstr = BitTorrent protocol
reserved = chr(0)*8
info_hash = 20-byte hash above (aka self.hash_string)
peer_id = 20-byte string
'''
pstr = 'BitTorrent protocol'
pstrlen = len(pstr)
info_hash = self.hash_string
peer_id = self.peer_id
packet = ''.join([chr(pstrlen), pstr, chr(0) * 8, info_hash,
peer_id])
print "Here's my packet {}".format(repr(packet))
# TODO -- add some checks in here so that I'm talking
# to a maximum of 30 peers
# TODO -- think about why i'm deleting self.peer_ips.
# What was the point of it? Why won't I need it?
# Think about what we're doing -- using this list to create
# new peer objects. Should make this functional, that way I
# can also call when I get new peers.
for i in self.peer_ips:
if len(self.peer_dict) >= 30:
break
s = socket.socket()
s.setblocking(True)
s.settimeout(0.5)
try:
s.connect(i)
except socket.timeout:
print '{} timed out on connect'.format(s.fileno())
continue
except socket.error:
print '{} threw a socket error'.format(s.fileno())
continue
except:
raise Exception
s.send(packet)
try:
data = s.recv(68) # Peer's handshake - len from docs
if data:
print 'From {} received: {}'.format(s.fileno(), repr(data))
self.initpeer(s)
except:
print '{} timed out on recv'.format(s.fileno())
continue
else:
self.peer_ips = []
def initpeer(self, sock):
'''
Creates a new peer object for a nvalid socket and adds it to reactor's
listen list
'''
location_json = requests.request("GET", "http://freegeoip.net/json/"
+ sock.getpeername()[0]).content
location = json.loads(location_json)
tpeer = peer.Peer(sock, self.reactor, self, location)
self.peer_dict[sock] = tpeer
self.reactor.select_list.append(tpeer)
def add_peer(self, sock):
print 'adding peer at', sock.getpeername()
time.sleep(3)
def kill_peer(self, tpeer):
thispeer = self.peer_dict.pop(tpeer.sock)
print 'peer with fileno {} killing itself'.format(thispeer.fileno())
self.reactor.select_list.remove(thispeer)
def set_sock(self, sock):
self.vis_write_sock = sock
self.switchboard.vis_write_sock = self.vis_write_sock
def __enter__(self):
pass
def __exit__(self, type, value, tb):
self.switchboard.close()