#!/usr/bin/env python3
import os
import sys

def check_binary(file_path, patches):
    if not os.path.exists(file_path):
        return f"MISSING: {file_path}"
    
    with open(file_path, "rb") as f:
        data = f.read()
    
    results = []
    for name, offset, expected, patched in patches:
        if data[offset:offset+len(expected)] == expected:
            results.append(f"  [FAIL] {name}: UNPATCHED (found {expected.hex()})")
        elif data[offset:offset+len(patched)] == patched:
            results.append(f"  [PASS] {name}: PATCHED (found {patched.hex()})")
        else:
            found = data[offset:offset+max(len(expected), len(patched))].hex()
            results.append(f"  [ERR] {name}: UNKNOWN (expected {patched.hex()}, found {found})")
            
    return "\n".join(results)

if __name__ == "__main__":
    PATCHED_DIR = "/home/sapient/Public/esp32-analysis/patched_binaries"
    
    ultra_patches = [
        ("UI Branch", 0xBA62D, bytes.fromhex("26153c"), bytes.fromhex("060f00")),
        ("Global Status (Round 2)", 0xBA6AD, bytes.fromhex("040242"), bytes.fromhex("221000ec"))
    ]
    
    meshos_patches = [
        ("UI Branch", 0xB99ED, bytes.fromhex("26193c"), bytes.fromhex("060f00")),
        ("Global Status (Round 2)", 0xB9A54, bytes.fromhex("84210040"), bytes.fromhex("22100040"))
    ]
    
    targets = [
        ("Ultra Standalone", os.path.join(PATCHED_DIR, "Ultra-TDeck-v9.2.patched.bin"), ultra_patches),
        ("Ultra Merged", os.path.join(PATCHED_DIR, "Ultra-TDeck-v9.2-merged.patched.bin"), 
            [(n, o+0x10000, e, p) for n, o, e, p in ultra_patches]),
        ("MeshOS Standalone", os.path.join(PATCHED_DIR, "MeshOS-TDeck-1.1.8.patched.bin"), meshos_patches),
        ("MeshOS Merged", os.path.join(PATCHED_DIR, "MeshOS-TDeck-1.1.8-merged.patched.bin"),
            [(n, o+0x10000, e, p) for n, o, e, p in meshos_patches])
    ]
    
    all_passed = True
    print("=== Firmware Patch Verification Harness ===")
    for label, path, p in targets:
        print(f"\n{label}:")
        report = check_binary(path, p)
        print(report)
        if "[FAIL]" in report or "[ERR]" in report or "MISSING" in report:
            all_passed = False
    
    if all_passed:
        print("\n[COMPLETE] All binaries are fully patched.")
        sys.exit(0)
    else:
        print("\n[INCOMPLETE] Some patches are missing or invalid.")
        sys.exit(1)