masscollaborationlabs/p2pfs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Move request chunk register message to download method.

Browse source
This commit is contained in:
Yuxin Wang 2018-09-26 23:11:02 -04:00
parent 99bd980f10
commit f0740851d5
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
p2pfs/core/peer.py
View file

@ -121,6 +121,12 @@ class Peer(MessageServer):
dest_file.seek(number * (512 * 1024), 0)
dest_file.write(pybase64.b64decode(raw_data.encode('utf-8'), validate=True))
dest_file.flush()
# send request chunk register to server
self._write_message(self._server_sock, {
'type': MessageType.REQUEST_CHUNK_REGISTER,
'filename': file,
'chunknum': number
})
progress(i + 1, totalchunknum)
logger.debug('Got {}\'s chunk # {}'.format(file, number))
@ -178,11 +184,6 @@ class Peer(MessageServer):
'chunknum': message['chunknum'],
'data': pybase64.b64encode(raw_data).decode('utf-8')
})
self._write_message(self._server_sock, {
'type': MessageType.REQUEST_CHUNK_REGISTER,
'filename': message['filename'],
'chunknum': message['chunknum']
})
elif message['type'] == MessageType.PEER_REPLY_CHUNK:
self._download_results[message['filename']].put((message['chunknum'], message['data']))
else: