Flash Checksum Hierarchy

#!/usr/bin/env python3
"""
Bosch ME7.5 1024 KB Flash Checksum Verification & Recalculation Engine
Target Platform: AMD AM29F800BB Parallel Flash (1048576 Bytes)
Compatible with: 06A906032LP, PL, SK, HS, CL, DL, JJ, HN
"""

import struct
import sys
import zlib

def verify_and_fix_me7_checksums(bin_path, output_path=None):
    with open(bin_path, 'rb') as f:
        data = bytearray(f.read())

    if len(data) != 1048576:
        print(f"Error: Invalid file size {len(data)} bytes. Must be exactly 1,048,576 bytes.")
        return False

    print(f"Loaded {bin_path} (1024 KB). Scanning for ME7.5 Checksum Descriptors...")

    # Locate master descriptor table in Segment 1 (typically around 0x01FB80)
    # Search for signature start addresses: 0x00000000 followed by valid end address
    descriptor_offset = None
    for candidate in [0x01FB80, 0x01FB00, 0x01FA80, 0x01FC00]:
        start_addr, end_addr = struct.unpack_from('<II', data, candidate)
        if start_addr == 0x00000000 and 0x00001000 <= end_addr <= 0x000FFFFF:
            descriptor_offset = candidate
            break

    if descriptor_offset is None:
        print("Warning: Could not auto-detect descriptor table. Using standard 24B offset 0x01FB80.")
        descriptor_offset = 0x01FB80

    print(f"Checksum table detected at 0x{descriptor_offset:06X}")

    fixed_count = 0
    # Process up to 16 descriptor blocks
    for block_idx in range(16):
        entry_offset = descriptor_offset + (block_idx * 12)
        start_addr, end_addr, stored_sum, flags = struct.unpack_from('<IIHH', data, entry_offset)

        # Terminating condition or invalid entry
        if start_addr >= end_addr or end_addr >= len(data):
            break

        # Calculate 16-bit word summation
        calc_sum = 0
        block_bytes = data[start_addr:end_addr + 1]

        # Unpack as little-endian 16-bit words
        word_count = len(block_bytes) // 2
        words = struct.unpack(f'<{word_count}H', block_bytes[:word_count * 2])
        calc_sum = sum(words) & 0xFFFF

        status = "OK" if calc_sum == stored_sum else "FAIL"
        print(f"Block {block_idx:02d} [0x{start_addr:06X}..0x{end_addr:06X}]: "
              f"Stored=0x{stored_sum:04X}, Calc=0x{calc_sum:04X} -> {status}")

        if calc_sum != stored_sum:
            # Fix checksum in descriptor table
            struct.pack_into('<H', data, entry_offset + 8, calc_sum)
            fixed_count += 1

    # Verify and fix complement pairs in Segment 1
    # Check complement pairs at 0x01FBEE and 0x01FBF0
    comp1, comp2 = struct.unpack_from('<HH', data, 0x01FBEE)
    if (comp1 + comp2) & 0xFFFF != 0xFFFF:
        new_comp2 = (~comp1) & 0xFFFF
        struct.pack_into('<H', data, 0x01FBF0, new_comp2)
        print(f"Fixed Complement Pair at 0x01FBEE: 0x{comp1:04X} + 0x{new_comp2:04X} = 0xFFFF")
        fixed_count += 1

    if fixed_count > 0 and output_path:
        with open(output_path, 'wb') as f:
            f.write(data)
        print(f"Successfully corrected {fixed_count} checksum(s). Written to {output_path}.")
    elif fixed_count == 0:
        print("All flash checksums are 100% valid. No corrections required.")

    return True

if __name__ == '__main__':
    if len(sys.argv) < 2:
        print("Usage: python3 me7_flash_checksum.py <input.bin> [output_fixed.bin]")
        sys.exit(1)
    out_file = sys.argv[2] if len(sys.argv) > 2 else sys.argv[1]
    verify_and_fix_me7_checksums(sys.argv[1], out_file)