Update app.py

app.py

@@ -1,198 +1,20 @@
-from itertools import permutations, combinations
-import hashlib, hmac, os, time, math, gc
+import time
 from multiprocessing import Pool
-from eth_account import Account
-import coincurve
-from eth_utils import keccak, to_checksum_address
 
-Account.enable_unaudited_hdwallet_features()
-# ============================================================
-words = [
-    "lesson", "cushion", "crane", "affair", "federal",
-    "decline", "limb", "spend", "reunion", "chief", "cart", "zebra",
-]
-
-PHRASE_MODE = "order"
-USE_BIP39   = False
-BIP39_PASSPHRASE = ""
-HD_DERIVATION = True
-ADDRESS_INDEX = 0
-CHECK_BIP39_CHECKSUM = False
-ADDRESSES_FILE = "12730802_addresses.csv"
-BATCH_SIZE  = 5000
-NUM_WORKERS = 96
-# ============================================================
-
-# ---------- BIP32 (same as before) ----------
-_SECP256K1_N = 0xFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFEBAAEDCE6AF48A03BBFD25E8CD0364141
-_HARDENED    = 0x80000000
-
-def _bip32_derive_child_priv(parent_priv, parent_chain, index):
-    if index >= _HARDENED:
-        data = b'\x00' + parent_priv + index.to_bytes(4, 'big')
-    else:
-        pub = coincurve.PrivateKey(parent_priv).public_key.format(compressed=True)
-        data = pub + index.to_bytes(4, 'big')
-    I = hmac.new(parent_chain, data, hashlib.sha512).digest()
-    child_int = (int.from_bytes(I[:32], 'big') + int.from_bytes(parent_priv, 'big')) % _SECP256K1_N
-    if child_int == 0:
-        raise ValueError("Invalid child key")
-    return child_int.to_bytes(32, 'big'), I[32:]
-
-def seed_to_privkey_hd(seed, address_index=0):
-    I = hmac.new(b"Bitcoin seed", seed, hashlib.sha512).digest()
-    key, chain = I[:32], I[32:]
-    path = [44 + _HARDENED, 60 + _HARDENED, _HARDENED, 0, address_index]
-    for idx in path:
-        key, chain = _bip32_derive_child_priv(key, chain, idx)
-    return key
-
-# ---------- BIP39 ----------
-try:
-    import bip39 as _bip39_rust
-    _USE_RUST_BIP39 = True
-except ImportError:
-    _bip39_rust = None
-    _USE_RUST_BIP39 = False
-
-_MNEMONIC = None
-
-def _get_mnemonic():
-    global _MNEMONIC
-    if _MNEMONIC is None:
-        from mnemonic import Mnemonic
-        _MNEMONIC = Mnemonic("english")
-    return _MNEMONIC
-
-def is_valid_bip39_fast(phrase):
-    return _get_mnemonic().check(phrase)
-
-def phrase_to_seed(phrase):
-    if _USE_RUST_BIP39:
-        return _bip39_rust.bip39_mnemonic_to_seed(phrase, BIP39_PASSPHRASE)
-    return _get_mnemonic().to_seed(phrase, passphrase=BIP39_PASSPHRASE)
-
-# ---------- Address file ----------
-def load_addresses(path):
-    s = set()
-    with open(path, "r", encoding="utf-8") as f:
-        for line in f:
-            a = line.strip().strip('"').strip("'").lstrip("\ufeff").lower()
-            if a.startswith("0x"):
-                a = a[2:]
-            if a and len(a) == 40:
-                s.add(a)
-    return frozenset(s)
-
-def normalize(addr):
-    a = addr.strip().lower()
-    if a.startswith("0x"):
-        a = a[2:]
-    return a
-
-# ---------- Generator ----------
-def generate_phrases():
-    n = len(words)
-    if PHRASE_MODE == "order":
-        for perm in permutations(words):
-            yield " ".join(perm)
-    elif PHRASE_MODE == "replace":
-        for i in range(n):
-            for w in words:
-                new = words.copy()
-                new[i] = w
-                yield " ".join(new)
-    elif PHRASE_MODE == "subset":
-        for r in range(1, n + 1):
-            for combo in combinations(words, r):
-                yield " ".join(combo)
-
-def count_total():
-    n = len(words)
-    if PHRASE_MODE == "order":   return math.factorial(n)
-    if PHRASE_MODE == "replace": return n * n
-    if PHRASE_MODE == "subset":  return 2**n - 1
-
-# ---------- Worker ----------
-_KNOWN = None
-
-def init_worker(known):
-    global _KNOWN
-    _KNOWN = known
-    _get_mnemonic()
-
-def process_batch(batch):
-    hits = []
-    known = _KNOWN
-    for phrase in batch:
-        try:
-            acct = Account.from_mnemonic(
-                phrase,
-                passphrase=BIP39_PASSPHRASE,
-                account_path=f"m/44'/60'/0'/0/{ADDRESS_INDEX}"
-            )
-            addr = acct.address
-            priv = acct.key
-        except Exception as e:
-            # print(f"❌ Exception: {e}")
-            continue
-        
-        normalized = normalize(addr)
-        if normalized in known:
-            hits.append((phrase, priv.hex(), addr))
-    return hits
-
-def batch_generator():
-    batch = []
-    for phrase in generate_phrases():
-        batch.append(phrase)
-        if len(batch) >= BATCH_SIZE:
-            yield batch
-            batch = []
-    if batch:
-        yield batch
-
-# ---------- Main ----------
-def main():
-    if not os.path.exists(ADDRESSES_FILE):
-        raise SystemExit(f"File nahi mili: {ADDRESSES_FILE}")
-
-    total = count_total()
-    print(f"\nMode: {PHRASE_MODE} | Total: {total:,} | Workers: {NUM_WORKERS}")
-    print(f"HD: {HD_DERIVATION} (index {ADDRESS_INDEX})")
-    print(f"Checksum prefilter: {CHECK_BIP39_CHECKSUM}\n")
-
-    known = load_addresses(ADDRESSES_FILE)
-    print(f"Loaded {len(known):,} addresses\n")
-
-    matches = []
-    done = 0
-    start = time.time()
-    last_log = start
-
-    gc.collect()
-    gc.freeze()
-
-    with Pool(processes=NUM_WORKERS,
-              initializer=init_worker,
-              initargs=(known,),
-              maxtasksperchild=1000) as pool:
-        for hits in pool.imap_unordered(process_batch, batch_generator(), chunksize=10):
-            done += BATCH_SIZE
-            for phrase, key_hex, addr in hits:
-                print(f"\n✅ MATCH: {phrase}\n   priv: {key_hex}\n   addr: {addr}\n")
-                matches.append((phrase, key_hex, addr))
-
-            now = time.time()
-            if now - last_log >= 2:
-                elapsed = now - start
-                rate = done / elapsed if elapsed else 0
-                eta = (total - done) / rate if rate else 0
-                print(f"  ...{done:,}/{total:,} ({100*done/total:.1f}%) ETA {eta:.0f}s")
-                last_log = now
-
-    elapsed = time.time() - start
-    print(f"\nDone {elapsed:.1f}s. {len(matches)} match(es).")
+def work(n):
+    # BIP39 PBKDF2 simulation
+    from mnemonic import Mnemonic
+    m = Mnemonic("english")
+    phrase = "lesson cushion crane affair federal decline limb spend reunion chief cart zebra"
+    for _ in range(n):
+        m.to_seed(phrase)
+    return n
 
 if __name__ == "__main__":
-    main()
+    for workers in [48, 96, 192]:
+        with Pool(workers) as p:
+            start = time.time()
+            p.map(work, [10] * workers)
+            elapsed = time.time() - start
+            rate = (workers * 10) / elapsed
+            print(f"{workers} workers: {rate:,.0f} phrases/sec")