commit f2066c4629a9df68193762d4fe814bee43b30b88
parent 94b87ac7d1a61648275260f7fd2aa440ac880e8e
Author: Janus <ysangkok@gmail.com>
Date: Wed, 11 Apr 2018 13:10:02 +0200
lnbase: avoid reimplementing int.from_bytes
Diffstat:
1 file changed, 1 insertion(+), 17 deletions(-)
diff --git a/lib/lnbase.py b/lib/lnbase.py
@@ -114,22 +114,6 @@ def gen_msg(msg_type, **kwargs):
data += param
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):
"""Return a bytestring version of the integer
value n, with a string length of s
@@ -260,7 +244,7 @@ class Peer(PrintError):
self.read_buffer += await self.reader.read(2**10)
lc = self.read_buffer[:18]
l = aead_decrypt(self.rk, self.rn, b'', lc)
- length = decode(l)
+ length = int.from_bytes(l, byteorder="big")
offset = 18 + length + 16
if len(self.read_buffer) < offset:
continue