from __future__ import print_function
import hashlib
import time
import binascii
import base64
import sys
import ahocorasick


ALPHABET = "123456789ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijkmnopqrstuvwxyz";

def base58encode(inp):
	BASE = 58;
	bi = int(binascii.hexlify(inp), 16)
	s = ""
	while bi > 0:
		s = ALPHABET[bi%BASE] + s
		bi = bi/BASE                                                                                                                   
	nPad = 0
	for c in inp:
		if c == '\0': nPad += 1
		else: break
	
	return (ALPHABET[0]*nPad) + s

def public2address(publickey):

	ripemd = hashlib.new('ripemd160')

	version = '\x00'

	ripemd.update(hashlib.sha256(publickey).digest())

	keyhash = version+binascii.unhexlify(ripemd.hexdigest())

	checksum = hashlib.sha256(hashlib.sha256(keyhash).digest()).digest()[0:4]

	bitcoinaddress = base58encode(keyhash+checksum)

	return bitcoinaddress

def main():
	outfile = sys.argv[1]
	target = sys.argv[2]
	
	from M2Crypto import EC
	curve   = 'secp256k1'
	ec_curve = eval('EC.NID_%s' % curve)
	
	tree = ahocorasick.KeywordTree()
	file = open(target)
	
	print("Reading file...")
	
	for line in file:
		l = line.rstrip()
		tree.add(l)
	
	print("Generating tree...")
	
	tree.make()
	
	ec = EC.gen_params(ec_curve)
	CONST1 = binascii.unhexlify("308201130201010420")
	CONST2 = binascii.unhexlify("a081a53081a2020101302c06072a8648ce3d0101022100fffffffffffffffffffffffffffffffffffffffffffffffffffffffefffffc2f300604010004010704410479be667ef9dcbbac55a06295ce870b07029bfcdb2dce28d959f2815b16f81798483ada7726a3c4655da4fbfc0e1108a8fd17b448a68554199c47d08ffb10d4b8022100fffffffffffffffffffffffffffffffebaaedce6af48a03bbfd25e8cd0364141020101a144034200")
	
	test = False
	
	count = 0
	t = time.time()
	interval = 500
	
	with open(outfile, 'a') as finds:
		while True:
			ec.gen_key()
		
			#M2Crypto adds some stuff to the public key		
			pubkey = ec.pub().get_der()[23:]
		
			address = public2address(pubkey)
			where = tree.search(address.lower()[:10])
			if test or where:
				print("Address: "+address)
				print("Public key: "+binascii.hexlify(pubkey))
				#For some reason it can only save the private key to a file -.-
				ec.save_key("key"+outfile, None)
				with open("key"+outfile) as keyfile:
					keyfile.readline()
					priv = keyfile.readline().rstrip()
					#M2Crypto adds some stuff to the private key.
					#Removing it, and re-encoding to base58
					shortpriv = base64.b64decode(priv)[7:-9]
					print("Private key base58: "+base58encode('\x80'+shortpriv+hashlib.sha256(hashlib.sha256('\x80'+shortpriv).digest()).digest()[0:4]))
					priv = CONST1 + shortpriv + CONST2 + pubkey
					print("Private key full hex: "+binascii.hexlify(priv))
				if not test:
					word = address[where[0]:where[1]]
					print(word)
					finds.write(word+" "+address+" "+binascii.hexlify(pubkey)+" "+binascii.hexlify(priv)+"\n")
				print("")
			count += 1
			if count%interval == 0:
				dt = time.time() - t
				print(str(int(interval/dt)) + " keys/sec    ",end='\r')
				t = time.time()

main()