handle invalid data

This commit is contained in:
Victor Shyba 2019-03-28 21:57:03 -03:00 committed by Lex Berezhny
parent 89c80cedfd
commit d16e2716c8
2 changed files with 6 additions and 4 deletions

View file

@ -1,6 +1,8 @@
import json
from decimal import Decimal
from google.protobuf.message import DecodeError
from lbrynet.schema.types.v1.legacy_claim_pb2 import Claim as OldClaimMessage
from lbrynet.schema.types.v1.certificate_pb2 import KeyType
from lbrynet.schema.types.v1.fee_pb2 import Fee as FeeMessage
@ -37,7 +39,7 @@ def from_old_json_schema(claim, payload: bytes):
elif currency == 'BTC':
stream.fee.btc = Decimal(fee[currency]['amount'])
else:
raise ValueError(f'Unknown currency: {currency}')
raise DecodeError(f'Unknown currency: {currency}')
stream.fee.address = fee[currency]['address']
return claim
@ -70,7 +72,7 @@ def from_types_v1(claim, payload: bytes):
elif currency == 'BTC':
stream.fee.btc = Decimal(fee.amount)
else:
raise ValueError(f'Unsupported currency: {currency}')
raise DecodeError(f'Unsupported currency: {currency}')
if old.HasField('publisherSignature'):
sig = old.publisherSignature
claim.signature = sig.signature
@ -82,5 +84,5 @@ def from_types_v1(claim, payload: bytes):
channel = claim.channel
channel.public_key_bytes = old.certificate.publicKey
else:
raise ValueError('claimType must be 1 for Streams and 2 for Channel')
raise DecodeError('claimType must be 1 for Streams and 2 for Channel')
return claim

View file

@ -420,7 +420,7 @@ def _decode_claim_result(claim):
return claim
try:
if not isinstance(claim['value'], Claim):
claim['value'] = Claim.from_bytes(claim['value'])
claim['value'] = Claim.from_bytes(unhexlify(claim['value']))
claim['hex'] = hexlify(claim['value'].to_bytes())
except DecodeError:
claim['hex'] = claim['value']