7cf9572c26
* Merge #11881: Remove Python2 support 1874058 Make base58 python contrib code work with python3 (Evan Klitzke) bc6fdf2 Change all python files to use Python3 (John Newbery) Pull request description: Following discussion here: https://github.com/bitcoin/bitcoin/pull/11843#issuecomment-351033742 It's easier for maintainers if all python tools/scripts support only a single version of Python. There are only a few scripts that aren't explicitly python3 at this point, so this PR changes those remaining scripts to explicitly require python3. Tree-SHA512: 5d38eef6e0fc7d8515e23a1f4c75e8b4160fd0fe23cba52a1f41689b114e54a9e503e0724829e8b41982ef98f2d113df80d9e238213b74f09ceaed0344a19e24 * Merge #12829: Python3 fixup f50975b [contrib] fixup symbol-check.py Python3 support (John Newbery) 5de2b18 [contrib] fixup security-check.py Python3 support (John Newbery) Pull request description: security-check.py and symbol-check.py were broken by #11881. Fix them. Tree-SHA512: 86de3d6dc3292b1ae4cc04c2d7d7dbbf39c9270551d7b224b8d8b19e3184c30c897dbf823200403706d06bb405c0decad5cfd690cb2c0312992a235a4ffcf6bf
113 lines
3.0 KiB
Python
113 lines
3.0 KiB
Python
# Copyright (c) 2012 The Bitcoin Core developers
|
|
# Distributed under the MIT software license, see the accompanying
|
|
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
|
|
'''
|
|
Bitcoin base58 encoding and decoding.
|
|
|
|
Based on https://bitcointalk.org/index.php?topic=1026.0 (public domain)
|
|
'''
|
|
import hashlib
|
|
|
|
# for compatibility with following code...
|
|
class SHA256:
|
|
new = hashlib.sha256
|
|
|
|
if str != bytes:
|
|
# Python 3.x
|
|
def ord(c):
|
|
return c
|
|
def chr(n):
|
|
return bytes( (n,) )
|
|
|
|
__b58chars = '123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz'
|
|
__b58base = len(__b58chars)
|
|
b58chars = __b58chars
|
|
|
|
def b58encode(v):
|
|
""" encode v, which is a string of bytes, to base58.
|
|
"""
|
|
long_value = 0
|
|
for (i, c) in enumerate(v[::-1]):
|
|
if isinstance(c, str):
|
|
c = ord(c)
|
|
long_value += (256**i) * c
|
|
|
|
result = ''
|
|
while long_value >= __b58base:
|
|
div, mod = divmod(long_value, __b58base)
|
|
result = __b58chars[mod] + result
|
|
long_value = div
|
|
result = __b58chars[long_value] + result
|
|
|
|
# Bitcoin does a little leading-zero-compression:
|
|
# leading 0-bytes in the input become leading-1s
|
|
nPad = 0
|
|
for c in v:
|
|
if c == 0: nPad += 1
|
|
else: break
|
|
|
|
return (__b58chars[0]*nPad) + result
|
|
|
|
def b58decode(v, length = None):
|
|
""" decode v into a string of len bytes
|
|
"""
|
|
long_value = 0
|
|
for i, c in enumerate(v[::-1]):
|
|
pos = __b58chars.find(c)
|
|
assert pos != -1
|
|
long_value += pos * (__b58base**i)
|
|
|
|
result = bytes()
|
|
while long_value >= 256:
|
|
div, mod = divmod(long_value, 256)
|
|
result = chr(mod) + result
|
|
long_value = div
|
|
result = chr(long_value) + result
|
|
|
|
nPad = 0
|
|
for c in v:
|
|
if c == __b58chars[0]:
|
|
nPad += 1
|
|
continue
|
|
break
|
|
|
|
result = bytes(nPad) + result
|
|
if length is not None and len(result) != length:
|
|
return None
|
|
|
|
return result
|
|
|
|
def checksum(v):
|
|
"""Return 32-bit checksum based on SHA256"""
|
|
return SHA256.new(SHA256.new(v).digest()).digest()[0:4]
|
|
|
|
def b58encode_chk(v):
|
|
"""b58encode a string, with 32-bit checksum"""
|
|
return b58encode(v + checksum(v))
|
|
|
|
def b58decode_chk(v):
|
|
"""decode a base58 string, check and remove checksum"""
|
|
result = b58decode(v)
|
|
if result is None:
|
|
return None
|
|
if result[-4:] == checksum(result[:-4]):
|
|
return result[:-4]
|
|
else:
|
|
return None
|
|
|
|
def get_bcaddress_version(strAddress):
|
|
""" Returns None if strAddress is invalid. Otherwise returns integer version of address. """
|
|
addr = b58decode_chk(strAddress)
|
|
if addr is None or len(addr)!=21: return None
|
|
version = addr[0]
|
|
return ord(version)
|
|
|
|
if __name__ == '__main__':
|
|
# Test case (from http://gitorious.org/bitcoin/python-base58.git)
|
|
assert get_bcaddress_version('15VjRaDX9zpbA8LVnbrCAFzrVzN7ixHNsC') is 0
|
|
_ohai = 'o hai'.encode('ascii')
|
|
_tmp = b58encode(_ohai)
|
|
assert _tmp == 'DYB3oMS'
|
|
assert b58decode(_tmp, 5) == _ohai
|
|
print("Tests passed")
|