From 6d2172333010ed0371afacffd24dd3b722afae41 Mon Sep 17 00:00:00 2001 From: Pavol Rusnak Date: Sat, 2 Jun 2018 15:49:55 +0200 Subject: [PATCH] src/apps/wallet: add asserts to sign_tx.writers --- src/apps/wallet/sign_tx/writers.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/apps/wallet/sign_tx/writers.py b/src/apps/wallet/sign_tx/writers.py index ac181934f..ce8127c8e 100644 --- a/src/apps/wallet/sign_tx/writers.py +++ b/src/apps/wallet/sign_tx/writers.py @@ -33,6 +33,7 @@ def write_tx_output(w, o: TxOutputBinType): def write_op_push(w, n: int): + assert n >= 0 and n <= 4294967295 if n < 0x4C: w.append(n & 0xFF) elif n < 0xFF: @@ -55,6 +56,7 @@ def write_op_push(w, n: int): def write_varint(w, n: int): + assert n >= 0 and n <= 4294967295 if n < 253: w.append(n & 0xFF) elif n < 65536: @@ -70,6 +72,7 @@ def write_varint(w, n: int): def write_uint32(w, n: int): + assert n >= 0 and n <= 4294967295 w.append(n & 0xFF) w.append((n >> 8) & 0xFF) w.append((n >> 16) & 0xFF) @@ -77,6 +80,7 @@ def write_uint32(w, n: int): def write_uint64(w, n: int): + assert n >= 0 and n <= 18446744073709551615 w.append(n & 0xFF) w.append((n >> 8) & 0xFF) w.append((n >> 16) & 0xFF)