LBRYFoundation/LBRY-Vault
1
0
Fork 0
mirror of https://github.com/LBRYFoundation/LBRY-Vault.git synced 2025-08-26 23:13:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

lnbase: avoid reimplementing int.from_bytes

Browse source
This commit is contained in:
Janus 2018-04-11 13:10:02 +02:00 • committed by ThomasV
parent 94b87ac7d1
commit f2066c4629
1 changed files with 1 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
lib/lnbase.py
View file

@ -114,22 +114,6 @@ def gen_msg(msg_type, **kwargs):
data += param data += param
return data return data
def decode(string):
"""Return the integer value of the
bytestring b
"""
if isinstance(string, str):
string = bytes(bytearray.fromhex(string))
result = 0
while len(string) > 0:
result *= 256
result += string[0]
string = string[1:]
return result
def encode(n, s): def encode(n, s):
"""Return a bytestring version of the integer """Return a bytestring version of the integer
value n, with a string length of s value n, with a string length of s
@ -260,7 +244,7 @@ class Peer(PrintError):
self.read_buffer += await self.reader.read(2**10) self.read_buffer += await self.reader.read(2**10)
lc = self.read_buffer[:18] lc = self.read_buffer[:18]
l = aead_decrypt(self.rk, self.rn, b'', lc) l = aead_decrypt(self.rk, self.rn, b'', lc)
length = decode(l) length = int.from_bytes(l, byteorder="big")
offset = 18 + length + 16 offset = 18 + length + 16
if len(self.read_buffer) < offset: if len(self.read_buffer) < offset:
continue continue