mirror of
https://github.com/trezor/trezor-firmware.git
synced 2025-01-16 02:10:55 +00:00
revert bytes/str change in tools.py
This commit is contained in:
parent
8a37c28ed6
commit
c550e5c703
@ -22,6 +22,17 @@ import binascii
|
|||||||
import struct
|
import struct
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
|
if sys.version_info < (3,):
|
||||||
|
def byteindex(data, index):
|
||||||
|
return ord(data[index])
|
||||||
|
|
||||||
|
def iterbytes(data):
|
||||||
|
return (ord(char) for char in data)
|
||||||
|
else:
|
||||||
|
def byteindex(data, index):
|
||||||
|
return data[index]
|
||||||
|
iterbytes = iter
|
||||||
|
|
||||||
|
|
||||||
def Hash(data):
|
def Hash(data):
|
||||||
return hashlib.sha256(hashlib.sha256(data).digest()).digest()
|
return hashlib.sha256(hashlib.sha256(data).digest()).digest()
|
||||||
@ -41,8 +52,8 @@ def hash_160_to_bc_address(h160, address_type):
|
|||||||
|
|
||||||
|
|
||||||
def compress_pubkey(public_key):
|
def compress_pubkey(public_key):
|
||||||
if public_key[0] == 4:
|
if byteindex(public_key, 0) == 4:
|
||||||
return bytes(public_key[64] & 1 + 2) + public_key[1:33]
|
return bytes((byteindex(public_key, 64) & 1) + 2) + public_key[1:33]
|
||||||
raise ValueError("Pubkey is already compressed")
|
raise ValueError("Pubkey is already compressed")
|
||||||
|
|
||||||
|
|
||||||
@ -62,7 +73,7 @@ def b58encode(v):
|
|||||||
""" encode v, which is a string of bytes, to base58."""
|
""" encode v, which is a string of bytes, to base58."""
|
||||||
|
|
||||||
long_value = 0
|
long_value = 0
|
||||||
for c in v:
|
for c in iterbytes(v):
|
||||||
long_value = long_value * 256 + c
|
long_value = long_value * 256 + c
|
||||||
|
|
||||||
result = ''
|
result = ''
|
||||||
@ -75,7 +86,7 @@ def b58encode(v):
|
|||||||
# Bitcoin does a little leading-zero-compression:
|
# Bitcoin does a little leading-zero-compression:
|
||||||
# leading 0-bytes in the input become leading-1s
|
# leading 0-bytes in the input become leading-1s
|
||||||
nPad = 0
|
nPad = 0
|
||||||
for c in v:
|
for c in iterbytes(v):
|
||||||
if c == 0:
|
if c == 0:
|
||||||
nPad += 1
|
nPad += 1
|
||||||
else:
|
else:
|
||||||
|
Loading…
Reference in New Issue
Block a user