Add file_size to request to avoid unnecessary download

This commit is contained in:
shortcutme 2017-06-19 16:10:18 +02:00
parent 9a9bd71634
commit f773bf3336
No known key found for this signature in database
GPG key ID: 5B63BAE6CB9613AE
3 changed files with 7 additions and 3 deletions

View file

@ -198,6 +198,10 @@ class FileRequest(object):
file.seek(params["location"])
file.read_bytes = FILE_BUFF
file_size = os.fstat(file.fileno()).st_size
if params.get("file_size") and params["file_size"] != file_size:
self.connection.badAction(5)
raise RequestError("File size does not match")
if params["location"] > file_size:
self.connection.badAction(5)
raise RequestError("Bad file location")