btc-heist.py

#!/usr/bin/env python3

import argparse
from datetime import datetime
from itertools import count
import multiprocessing
from bitcoin import sha256, privtopub, pubtoaddr
from mnemonic import Mnemonic


def seek(core, btc_address_queue):
    print(f"Core {core}: Searching for Private Key...")
    mnemo = Mnemonic("english")
    log_rate_iterations = 10000
    start_time = datetime.today().timestamp()

    for iteration in count(1):
        # Generate private + public keys and btc address
        private_key = sha256(mnemo.generate(strength=256))
        public_key = privtopub(private_key)
        btc_address = pubtoaddr(public_key)
        btc_address_queue.put((private_key, public_key, btc_address))

        # log rate
        if (iteration % log_rate_iterations) == 0:
            time_diff = datetime.today().timestamp() - start_time
            print(f"Core {core}: {iteration / time_diff} Key/s")


if __name__ == "__main__":

    parser = argparse.ArgumentParser()
    parser.add_argument(
        "-c",
        "--cores",
        default=4,
        type=int,
        help="Number of CPU cores to use (default: 4)",
    )
    parser.add_argument(
        "-f",
        "--addresses",
        default="Bitcoin_addresses_LATEST.txt",
        type=str,
        help="File containing BTC addresses (default: Bitcoin_addresses_LATEST.txt),",
    )
    parser.add_argument(
        "-o",
        "--keyfile",
        default="found_keys.txt",
        type=str,
        help="File to output found keys (default: found_keys.txt)",
    )

    args = parser.parse_args()
    cores = args.cores
    addresses_filename = args.addresses
    keyfile = args.keyfile

    # generate list of pubkey with BTC
    print(f'Loading "{addresses_filename}"...')
    with open(addresses_filename) as f:
        publist = frozenset(f)  # set() used for O(1) search
    print("Loaded.")

    btc_address_queue = multiprocessing.Queue()
    for core in range(cores):
        process = multiprocessing.Process(target=seek, args=(core, btc_address_queue))
        process.start()

    while True:
        private_key, public_key, btc_address = btc_address_queue.get()
        if f"{btc_address}\n" in publist:
            found_key = f"\nPublic: {public_key} | Private: {private_key} | Address: {btc_address}\n"
            print(found_key)
            with open(keyfile, "a") as f:
                f.write(found_key)