.

Author: fabohax

README.md

@@ -1,3 +1,5 @@
+[WORK IN PROGRESS]
+
 # 🔑 ZKK: Zero-Knowledge Key
 
 ZKK is a proof-of-concept project demonstrating the use of **Zero-Knowledge Proofs (ZKP)** to validate a Bitcoin private key without exposing its value. The project generates a ZKP proving that a given private key corresponds to a derived public key and encodes the proof in a QR code for easy sharing and validation.
@@ -6,7 +8,7 @@ ZKK is a proof-of-concept project demonstrating the use of **Zero-Knowledge Proo
 
 - Validate Bitcoin private keys.
 - Derive public keys and addresses from private keys.
-- Generate Zero-Knowledge Proofs (ZKP) using `binius`.
+- Generate Zero-Knowledge Proofs (ZKP) using `zk-SNARKs`.
 - Encode ZKP into a QR code for secure sharing.
 - Display and save QR codes as SVG images.
 
@@ -21,10 +23,11 @@ ZKK is a proof-of-concept project demonstrating the use of **Zero-Knowledge Proo
 3. **Dependencies**:
    - `bitcoinlib`: For Bitcoin key derivation.
    - `qrcode`: For QR code generation.
+   - `snarkjs`: For zk-SNARK proof generation.
 
 Install dependencies using:
 ```bash
-pip install bitcoinlib qrcode
+pip install bitcoinlib qrcode snarkjs
 ```
 
 ## Installation
@@ -42,9 +45,9 @@ zkk/
 ├── bin/
 │   └── cli.py                 # CLI script for ZKP generation and verification
 ├── src/
-│   ├── binius/                # Binius proof-related code
+│   ├── snarks/                # zk-SNARK proof-related code
 │   │   ├── proof.py           # Proof generation and verification logic
-│   │   ├── binary_fields.py   # Binary field operations
+│   │   ├── constraints.circom # Circom circuit for zk-SNARKs
 │   │   ├── utils.py           # Utility functions for evaluation
 │   │   └── merkle.py          # Merkle tree implementation
 │   ├── bitcoin/
@@ -64,8 +67,8 @@ zkk/
    - It derives the corresponding public key and Bitcoin address.
 
 2. **Generate ZKP**:
-   - Using `binius`, the private key is input as a binary evaluation.
-   - The proof is generated based on a securely chosen evaluation point.
+   - Using `zk-SNARKs`, the private key is input into a Circom circuit.
+   - The proof is generated via `snarkjs` based on the circuit constraints.
    - The output is a ZKP that can be validated without exposing the private key.
 
 3. **Encode in QR Code**:

bin/cli.py

@@ -1,65 +1,71 @@
 import sys
 import os
 import argparse
-from bitcoin.key_utils import derive_public_key, derive_address
-from binius.proof import packed_binius_proof, verify_packed_binius_proof
-from binius.binary_fields import BinaryFieldElement
-from qr.generate_qr import generate_qr_code
+import json
+import qrcode
 
-sys.path.insert(0, os.path.join(os.path.dirname(__file__), "../src"))
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
-# Function to generate a proof and QR code
-def generate_proof(private_key_raw):
-    print("Deriving public key and address...")
-    public_key = derive_public_key(private_key_raw)
-    address = derive_address(private_key_raw)
+from src.bitcoin.key_utils import validate_private_key_raw, derive_public_key, derive_address
+from src.snarks.proof import generate_zk_proof, verify_zk_proof
 
+def generate_proof(private_key):
+    print("Validating and deriving public key and address...")
+    public_key = derive_public_key(private_key)
+    address = derive_address(private_key)
     print(f"Public Key: {public_key}")
     print(f"Address: {address}")
 
-    # Convert the raw private key to binary evaluations
-    evaluations = [BinaryFieldElement(int(bit)) for bit in bin(int(private_key_raw, 16))[2:]]
+    print("Generating ZKP...")
+    proof = generate_zk_proof(private_key, public_key)
+    proof_data = {
+        "proof": proof,
+        "public_key": public_key,
+        "address": address
+    }
+    
+    qr = qrcode.QRCode()
+    qr.add_data(json.dumps(proof_data))
+    qr.make(fit=True)
+    img = qr.make_image(fill_color="black", back_color="white")
+    img.save(f"zkp_qr_{public_key[:8]}.svg")
+    print("ZKP Generated Successfully.")
+    print(f"QR Code saved as zkp_qr_{public_key[:8]}.svg")
 
-    # Generate a dummy evaluation point (replace with a secure point in production)
-    evaluation_point = [BinaryFieldElement(0), BinaryFieldElement(1)]
-
-    print("Generating Zero-Knowledge Proof...")
-    proof = packed_binius_proof(evaluations, evaluation_point)
-
-    print("Proof generated successfully.")
-    generate_qr_code(str(proof), "zkp_qr.svg")
-    print("QR Code saved as zkp_qr.svg")
-
-# Function to verify a proof
-def verify_proof(proof):
-    print("Verifying proof...")
-    is_valid = verify_packed_binius_proof(proof)
+def verify_proof(proof_path):
+    with open(proof_path, "r") as f:
+        proof_data = json.load(f)
+    
+    print("Verifying ZKP...")
+    is_valid = verify_zk_proof(proof_data["proof"], proof_data["public_key"])
+    
     if is_valid:
-        print("Proof verification successful.")
+        print("Proof is valid!")
     else:
         print("Proof verification failed.")
 
-# Main CLI function
 def main():
-    parser = argparse.ArgumentParser(description="ZKK: Zero-Knowledge Key CLI")
-    subparsers = parser.add_subparsers(dest="command", required=True)
+    parser = argparse.ArgumentParser(description="ZKK CLI Tool")
+    subparsers = parser.add_subparsers(dest="command")
 
-    # Subcommand: Generate Proof
-    generate_parser = subparsers.add_parser("generate", help="Generate a Zero-Knowledge Proof and QR code")
-    generate_parser.add_argument("private_key_raw", help="Raw Bitcoin private key in hexadecimal format")
+    gen_parser = subparsers.add_parser("generate", help="Generate a zk-SNARK proof and QR code")
+    gen_parser.add_argument("private_key", type=str, help="Bitcoin private key")
 
-    # Subcommand: Verify Proof
-    verify_parser = subparsers.add_parser("verify", help="Verify a Zero-Knowledge Proof")
-    verify_parser.add_argument("proof_file", help="Path to the proof file to verify")
+    verify_parser = subparsers.add_parser("verify", help="Verify a zk-SNARK proof")
+    verify_parser.add_argument("proof_path", type=str, help="Path to the proof JSON file")
 
     args = parser.parse_args()
 
     if args.command == "generate":
-        generate_proof(args.private_key_raw)
+        try:
+            validate_private_key_raw(args.private_key)
+            generate_proof(args.private_key)
+        except ValueError as e:
+            print(f"Error: {e}")
     elif args.command == "verify":
-        with open(args.proof_file, "r") as f:
-            proof = eval(f.read())  # Replace eval with safer deserialization in production
-        verify_proof(proof)
+        verify_proof(args.proof_path)
+    else:
+        parser.print_help()
 
 if __name__ == "__main__":
     main()

circuit.r1cs

@@ -1,2 +1,3 @@
 bitcoinlib==0.6.3
+python-snarks==0.0.3
 qrcode==7.4

requirements.txt

@@ -3,20 +3,22 @@
 setup(
     name="zkk",
     version="1.0.0",
-    description="Zero-Knowledge Key (ZKK): A Python-based implementation of zero-knowledge proofs for Bitcoin private key validation using Binius.",
+    description="Zero-Knowledge Key (ZKK): A Python-based implementation of zero-knowledge proofs for Bitcoin private key validation using PySNARK.",
     author="fabohax",
     author_email="[email protected]",
     url="https://github.com/fabohax/zkk",
     packages=find_packages("src"),
     package_dir={"": "src"},
     install_requires=[
-        "bitcoinlib==0.6.3",
+        "bit", 
         "qrcode==7.4",
+        "pysnark",  
+        "numpy>=1.21.4",  
     ],
     classifiers=[
         "Programming Language :: Python :: 3",
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
     ],
     python_requires=">=3.8",
-)
+)