From f2066c4629a9df68193762d4fe814bee43b30b88 Mon Sep 17 00:00:00 2001 From: Janus Date: Wed, 11 Apr 2018 13:10:02 +0200 Subject: [PATCH] lnbase: avoid reimplementing int.from_bytes --- lib/lnbase.py | 18 +----------------- 1 file changed, 1 insertion(+), 17 deletions(-) diff --git a/lib/lnbase.py b/lib/lnbase.py index d70fee6ee..21edc22da 100644 --- 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