ورک فلو پاک کردن فلش

scripts/serial_command.py

@@ -141,9 +141,31 @@ def transfer(cli, firmware, *, dtb=False, kernel=False, rootfs=False, rootfs_siz
     print("\nFirmware flash write and full read-back comparison verified.")
 
 
+def erase_flash(cli):
+    cli.command("sf probe 0:0", r"^SF: Detected .+total 32 MiB\s*$", timeout=20)
+    cli.command(
+        "sf erase 0x0 0x2000000",
+        r"^SF: 33554432 bytes @ 0x0 Erased: OK\s*$", timeout=600,
+    )
+    cli.command(
+        "sf read 0x42800000 0x0 0x100",
+        r"^SF: 256 bytes @ 0x0 Read: OK\s*$", timeout=20,
+    )
+    output = cli.command("md.b 0x42800000 0x100", r"^42800000:")
+    for offset in range(0, 0x100, 16):
+        address = f"{0x42800000 + offset:08x}"
+        rows = re.findall(
+            r"^" + address + r":((?: [0-9a-fA-F]{2}){16})(?:\s|$)",
+            output, re.MULTILINE,
+        )
+        if len(rows) != 1 or bytes.fromhex(rows[0]) != b"\xff" * 16:
+            raise RuntimeError("Erase verification failed at " + address)
+    print("\nFull-flash erase command succeeded; first 256 bytes verified as FF.")
+
+
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument("mode", choices=("check", "configure", "transfer", "transfer-dtb", "transfer-kernel", "transfer-rootfs"), default="check", nargs="?")
+    parser.add_argument("mode", choices=("check", "configure", "transfer", "transfer-dtb", "transfer-kernel", "transfer-rootfs", "erase"), default="check", nargs="?")
     parser.add_argument("--port", default="/dev/ttyUSB0")
     parser.add_argument("--firmware", type=Path, default=None)
     parser.add_argument("--rootfs-size", type=lambda value: int(value, 0), help="Actual RootFS partition size in bytes (decimal or 0x-prefixed hex)")
@@ -163,6 +185,9 @@ def main():
             cli.detect()
             if args.mode == "check":
                 return
+            if args.mode == "erase":
+                erase_flash(cli)
+                return
             if args.mode in ("transfer", "transfer-dtb", "transfer-kernel", "transfer-rootfs"):
                 transfer(cli, args.firmware, dtb=args.mode == "transfer-dtb", kernel=args.mode == "transfer-kernel", rootfs=args.mode == "transfer-rootfs", rootfs_size=args.rootfs_size)
                 return