def run(self):
self.print_welcome()
self.discoverer.start()
#Send something to the discover sock so we can bind it to a port
self.discoverer.sock.sendto("HURP DURP", ("<broadcast>", self.broadcast_port))
own_address = self.discoverer.sock.getsockname()
zap_debug_print("my udp discovery port is", own_address[1])
ignore_port = own_address[1]
b = ZapBroadcast(self.broadcast_port, self.local_files,
self.remote_files, self.ip, ignore_port)
b.start()
tcp_server = ZapTCPServer(port=self.tcp_port, host=self.ip, local_files=self.local_files)
tcp_server.start()
while True:
line = raw_input(self.prompt + " ")
if re.match('^quit$', line):
self.quit()
elif re.match('^name (\w+)$', line):
ZapConfig.name = re.match(r"^name (\w+)$", line).group(1)
print("Name is now %s" % ZapConfig.name)
elif re.match('^list$', line):
self.remote_files.clear()
query = ZapTorrentProtocolResponse(response_type="files?").as_response()
s = self.discoverer.sock
length = 0
while length < len(query):
sent_length = s.sendto(query, ("<broadcast>", self.broadcast_port))
length += sent_length
#now wait for the filesLister to get more info
print("Waiting for response from peers. . .")
time.sleep(3)
for k in self.remote_files.get_files():
print("File: %s" % k)
elif re.match('^load ([\w\._\-/]+)$', line):
path = re.match('^load ([\w\._\-/]+)$', line).group(1)
f = ZapFile()
if f.set_path(path):
zap_debug_print("Loaded a local file and it is ", f)
self.local_files.add(f)
print("File at %s loaded for sharing." % path)
else:
print("File at %s doesn't exist, or it is a directory. Try a different path." % path)
elif re.match('^get ([\w\.\-]+)$', line):
filename = re.match('^get ([\w\.\-]+)$', line).group(1)
remote_files = self.remote_files.get(filename)
if remote_files is not None:
downloader = ZapDownloader(remote_files, self.local_files)
downloader.start()
print("Starting to download %s." % filename)
else:
print("No files by that name. Sorry. Try 'list' to update your list of files.")
else:
self.print_usage()
continue
def run(self):
size = 55000
while True:
data, address = self.sock.recvfrom(size)
#ignore stuff sent from our own socket
if address[0] == ZapConfig.ip and address[1] == self.ignore_port:
continue
query = ZapTorrentProtocolParser(data)
zap_debug_print("Got some data! ", data)
zap_debug_print("my address is", (ZapConfig.ip, self.port))
zap_debug_print("other address is", address)
zap_debug_print("about to parse the query in FilesLister")
query.parse()
if query.message_type == "files":
# Parse the files out of the query, and store them in the remote files
zap_debug_print("got a files reponse: ", data)
ip = query.get_field('ip')
port = query.get_field('port')
name = query.get_field('name')
for f in query.get_files():
zf = ZapFile(status="not-present")
zf.ip = ip
zf.port = port
zf.name = name
zf.number_of_blocks = f['blocks']
zf.digest = f['digest']
zf.filename = f['filename']
zap_debug_print("Someone told me about a file named", zf.filename)
self.remote_files.add(zf)
def run(self):
"""Spawn off threads to download each block. When all
threads return, check if the file is completely
downloaded. If so, determine its digest and make
sure it matches, and save it to disk."""
start_time = time.time()
file_info = self.remote_files[0]
remote_file = ZapFile()
remote_file.filename = file_info.filename
remote_file.number_of_blocks = file_info.number_of_blocks
remote_file.mark_as_remote()
self.local_files.add(remote_file)
child_threads = []
for f in self.remote_files:
remote_location = {}
remote_location['ip'] = f.ip
remote_location['port'] = f.port
remote_location['name'] = f.name
child_thread = self.remote_file_downloader(remote_file, f)
child_thread.start()
child_threads.append(child_thread)
# How do we wait for them to finish?
# TODO: what if I can't download the whole file?
while not remote_file.is_downloaded():
time.sleep(4)
# Now all child threads are gone, I hope.
remote_file.save_to_disk()
zap_debug_print("remote file digest is ", remote_file.digest, "file_info.digest is ", file_info.digest)
if remote_file.digest != file_info.digest:
# Our file does not match. Quit this thread and return an error
zap_debug_print("Digest does not match! I should delete downloaded file!")
self.local_files.remove(remote_file)
os.remove(remote_file.path)
return False
else:
stop_time = time.time()
log_string = "file %s %s %s" % (remote_file.filename, os.path.getsize(remote_file.path),
stop_time - start_time)
zap_log(log_string)
print("Finished downloading %s. Find it at %s." % (remote_file.filename, remote_file.path))
return True
