46 lines
2.0 KiB
Python
46 lines
2.0 KiB
Python
import hashlib
|
|
from pyelliptic import arithmetic
|
|
|
|
# This function expects that pubkey begin with \x04
|
|
def calculateBitcoinAddressFromPubkey(pubkey):
|
|
"""This function expects that pubkey begin's with the bitcoin prefix"""
|
|
if len(pubkey) != 65:
|
|
print 'Could not calculate Bitcoin address from pubkey because function was passed a pubkey that was', len(pubkey), 'bytes long rather than 65.'
|
|
return "error"
|
|
ripe = hashlib.new('ripemd160')
|
|
sha = hashlib.new('sha256')
|
|
sha.update(pubkey)
|
|
ripe.update(sha.digest())
|
|
ripeWithProdnetPrefix = '\x00' + ripe.digest()
|
|
|
|
checksum = hashlib.sha256(hashlib.sha256(
|
|
ripeWithProdnetPrefix).digest()).digest()[:4]
|
|
binaryBitcoinAddress = ripeWithProdnetPrefix + checksum
|
|
numberOfZeroBytesOnBinaryBitcoinAddress = 0
|
|
while binaryBitcoinAddress[0] == '\x00':
|
|
numberOfZeroBytesOnBinaryBitcoinAddress += 1
|
|
binaryBitcoinAddress = binaryBitcoinAddress[1:]
|
|
base58encoded = arithmetic.changebase(binaryBitcoinAddress, 256, 58)
|
|
return "1" * numberOfZeroBytesOnBinaryBitcoinAddress + base58encoded
|
|
|
|
|
|
def calculateTestnetAddressFromPubkey(pubkey):
|
|
if len(pubkey) != 65:
|
|
print 'Could not calculate Bitcoin address from pubkey because function was passed a pubkey that was', len(pubkey), 'bytes long rather than 65.'
|
|
return "error"
|
|
ripe = hashlib.new('ripemd160')
|
|
sha = hashlib.new('sha256')
|
|
sha.update(pubkey)
|
|
ripe.update(sha.digest())
|
|
ripeWithProdnetPrefix = '\x6F' + ripe.digest()
|
|
|
|
checksum = hashlib.sha256(hashlib.sha256(
|
|
ripeWithProdnetPrefix).digest()).digest()[:4]
|
|
binaryBitcoinAddress = ripeWithProdnetPrefix + checksum
|
|
numberOfZeroBytesOnBinaryBitcoinAddress = 0
|
|
while binaryBitcoinAddress[0] == '\x00':
|
|
numberOfZeroBytesOnBinaryBitcoinAddress += 1
|
|
binaryBitcoinAddress = binaryBitcoinAddress[1:]
|
|
base58encoded = arithmetic.changebase(binaryBitcoinAddress, 256, 58)
|
|
return "1" * numberOfZeroBytesOnBinaryBitcoinAddress + base58encoded
|