LBRYCommunity/lbry-sdk
1
0
Fork 1
Code Issues Pull requests Projects Releases 282 Packages Wiki Activity

raise proper exception on undecode-able blobs

Browse source
This commit is contained in:
Victor Shyba 2019-02-21 20:42:51 -03:00
parent aa40337ded
commit 4e7d88311f
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
lbrynet/stream/descriptor.py
View file

@ -131,7 +131,10 @@ class StreamDescriptor:
assert os.path.isfile(blob.file_path) assert os.path.isfile(blob.file_path)
with open(blob.file_path, 'rb') as f: with open(blob.file_path, 'rb') as f:
json_bytes = f.read() json_bytes = f.read()
decoded = json.loads(json_bytes.decode()) try:
decoded = json.loads(json_bytes.decode())
except json.JSONDecodeError:
raise InvalidStreamDescriptorError("Does not decode as valid JSON")
if decoded['blobs'][-1]['length'] != 0: if decoded['blobs'][-1]['length'] != 0:
raise InvalidStreamDescriptorError("Does not end with a zero-length blob.") raise InvalidStreamDescriptorError("Does not end with a zero-length blob.")
if any([blob_info['length'] == 0 for blob_info in decoded['blobs'][:-1]]): if any([blob_info['length'] == 0 for blob_info in decoded['blobs'][:-1]]):