arno/trezor-firmware
1
0
Fork 0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-11-26 17:38:39 +00:00
Code Issues Releases Wiki Activity

feat(core): Add get_base_weight() to TxWeightCalculator.

Browse Source 
[no changelog]
This commit is contained in:
Andrew Kozlik 2022-07-19 17:27:16 +02:00 committed by Andrew Kozlik
parent 0ce2c079b0
commit 421eec1d3b
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
core/src/apps/bitcoin/sign_tx/tx_weight.py
View File

@ -45,7 +45,7 @@ class TxWeightCalculator:
def __init__(self) -> None: def __init__(self) -> None:
self.inputs_count = 0 self.inputs_count = 0
self.outputs_count = 0 self.outputs_count = 0
self.counter = 4 * (_TXSIZE_HEADER + _TXSIZE_FOOTER) self.counter = 0
self.segwit_inputs_count = 0 self.segwit_inputs_count = 0
def add_input(self, i: TxInput) -> None: def add_input(self, i: TxInput) -> None:
@ -131,12 +131,19 @@ class TxWeightCalculator:
script_size = self.compact_size_len(len(script)) + len(script) script_size = self.compact_size_len(len(script)) + len(script)
self.counter += 4 * (_TXSIZE_OUTPUT + script_size) self.counter += 4 * (_TXSIZE_OUTPUT + script_size)
def get_base_weight(self) -> int:
base_weight = 4 * (_TXSIZE_HEADER + _TXSIZE_FOOTER)
base_weight += 4 * self.compact_size_len(self.inputs_count)
base_weight += 4 * self.compact_size_len(self.outputs_count)
if self.segwit_inputs_count:
base_weight += _TXSIZE_SEGWIT_OVERHEAD
return base_weight
def get_total(self) -> int: def get_total(self) -> int:
total = self.counter total = self.counter
total += 4 * self.compact_size_len(self.inputs_count) total += self.get_base_weight()
total += 4 * self.compact_size_len(self.outputs_count)
if self.segwit_inputs_count: if self.segwit_inputs_count:
total += _TXSIZE_SEGWIT_OVERHEAD
# add one byte of witness stack item count per non-segwit input # add one byte of witness stack item count per non-segwit input
total += self.inputs_count - self.segwit_inputs_count total += self.inputs_count - self.segwit_inputs_count