soc: realtek: rts5912 image tool add SHA2 tag

Let rts5912 image tool to add SHA2 tag at the tail end of image.

Signed-off-by: Dylan Hsieh <[email protected]>

Author: dylanHsieh4963

soc/realtek/ec/common/rts5912_imgtool/img_gen.py

@@ -10,6 +10,7 @@
 """
 
 import argparse
+import hashlib
 import os
 import struct
 
@@ -24,6 +25,11 @@
 DUAL_read = "0x3B"
 QUAD_read = "0x6B"
 
+TAG_MAGIC = 0x6907
+TAG_LEN = 0x0028
+TAG_TYPE_SHA = 0x0010
+TAG_SHA_LEN = 0x0020
+
 
 def parse_args():
     """
@@ -88,10 +94,10 @@ def img_gen(load_addr, spi_freq, spi_rdcmd, original_bin, signed_bin):
     To obtain the RTS5912 image header and output a new BIN file
     """
     img_size = os.path.getsize(original_bin)
-    payload = bytearray(0)
-    img_size = os.path.getsize(original_bin)
+    hdr_payload = bytearray(0)
+    tag_payload = bytearray(0)
 
-    fmt = (
+    hdr_fmt = (
         "<"
         +
         # type ImageHdr struct {
@@ -108,8 +114,8 @@ def img_gen(load_addr, spi_freq, spi_rdcmd, original_bin, signed_bin):
         + "H"  # reserved uint16
     )  # }
 
-    header = struct.pack(
-        fmt,
+    hdr = struct.pack(
+        hdr_fmt,
         IMAGE_MAGIC,
         load_addr,
         IMAGE_HDR_SIZE,
@@ -123,15 +129,37 @@ def img_gen(load_addr, spi_freq, spi_rdcmd, original_bin, signed_bin):
         0,
     )
 
-    payload[: len(header)] = header
+    hdr_payload[: len(hdr)] = hdr
+
+    tag_fmt = (
+        "<"
+        +
+        # type ImageTag struct {
+        "H"  # Magic      uint16
+        + "H"  # TagSz    uint16
+        + "H"  # TagHash  uint16
+        + "H"  # HashSz   uint16
+    )  # }
+
+    tag = struct.pack(tag_fmt, TAG_MAGIC, TAG_LEN, TAG_TYPE_SHA, TAG_SHA_LEN)
+
+    tag_payload[: len(tag)] = tag
+
+    sha = hashlib.sha256()
+    sha.update(hdr_payload)
+    with open(original_bin, "rb") as original:
+        sha.update(original.read())
+    digest = sha.digest()
 
     with open(signed_bin, "wb") as signed:
-        signed.write(payload)
+        signed.write(hdr_payload)
         signed.flush()
         signed.close()
 
     with open(signed_bin, "ab") as signed, open(original_bin, "rb") as original:
         signed.write(original.read())
+        signed.write(tag_payload)
+        signed.write(digest)
         signed.flush()
         signed.close()
         original.close()