Routine Control (0x31) - Erase, Verify, Check

#!/usr/bin/env python3
# 0x31 0x01 0xFF 0x00: Erase Memory routine
import udsoncan, time
from udsoncan.client import Client

def erase_flash_block(client, start_addr: int, length: int):
    # Routine 0xFF00: EraseMemory (OEM-standard; address/length as parameters)
    routine_params = bytes([
        (start_addr >> 24) & 0xFF,
        (start_addr >> 16) & 0xFF,
        (start_addr >>  8) & 0xFF,
        (start_addr >>  0) & 0xFF,
        (length >> 24) & 0xFF,
        (length >> 16) & 0xFF,
        (length >>  8) & 0xFF,
        (length >>  0) & 0xFF,
    ])

    print(f"Erasing 0x{start_addr:08X}..0x{start_addr+length-1:08X} ({length//1024} kB)...")
    resp = client.routine_control(0x01, 0xFF00, routine_params)

    # Expect NRC 0x78 while erasing (can take 1–30s per block)
    # udsoncan library handles 0x78 polling automatically via response_pending config
    if resp.positive:
        print(f"Erase complete: result={resp.service_data.routine_status_record.hex()}")
        return True
    else:
        print(f"Erase FAILED: NRC 0x{resp.code:02X}")
        return False

# Usage:
# erase_flash_block(client, 0x80080000, 0x00780000)  # application block