From 80b36acd463c94cc943e31d509f9496843842474 Mon Sep 17 00:00:00 2001
From: shgutte <shgutte@silabs.com>
Date: Mon, 24 Feb 2025 09:06:31 +0530
Subject: [PATCH] Adds changes for the script

---
 .../tools/silabs/ota/ota_multi_image_tool.py  | 32 +++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/scripts/tools/silabs/ota/ota_multi_image_tool.py b/scripts/tools/silabs/ota/ota_multi_image_tool.py
index 280c80ea516a5d..a674017118f48c 100755
--- a/scripts/tools/silabs/ota/ota_multi_image_tool.py
+++ b/scripts/tools/silabs/ota/ota_multi_image_tool.py
@@ -62,6 +62,7 @@ class TAG:
     APPLICATION = 1
     BOOTLOADER = 2
     FACTORY_DATA = 3
+    WIFI_TA_M4 = 4
 
 
 def set_logger():
@@ -201,6 +202,32 @@ def validate_json(data: str):
         sys.exit(1)
 
 
+def generate_wifi_image(args: object):
+    """
+    Generate app payload with descriptor. If a certain option is not specified, use the default values.
+    """
+    logging.info("App descriptor information:")
+
+    descriptor = generate_descriptor(args.app_version, args.app_version_str, args.app_build_date)
+    logging.info(f"App encryption enable: {args.enc_enable}")
+    if args.enc_enable:
+        inputFile = open(args.wifi_input_file, "rb")
+        enc_file = crypto_utils.encryptData(inputFile.read(), args.input_ota_key, INITIALIZATION_VECTOR)
+        enc_file1 = bytes([ord(x) for x in enc_file])
+        file_size = len(enc_file1)
+        payload = generate_header(TAG.WIFI_TA_M4, len(descriptor) + file_size) + descriptor + enc_file1
+    else:
+        file_size = os.path.getsize(args.wifi_input_file)
+        logging.info(f"file size: {file_size}")
+        payload = generate_header(TAG.WIFI_TA_M4, len(descriptor) + file_size) + descriptor
+
+    write_to_temp(OTA_APP_TLV_TEMP, payload)
+    if args.enc_enable:
+        return [OTA_APP_TLV_TEMP]
+    else:
+        return [OTA_APP_TLV_TEMP, args.wifi_input_file]
+
+
 def generate_custom_tlvs(data):
     """
     Generate custom OTA payload from a JSON object following a predefined schema.
@@ -259,6 +286,9 @@ def create_image(args: object):
     if args.app_input_file:
         input_files += generate_app(args)
 
+    if args.wifi_input_file:
+        input_files += generate_wifi_image(args)
+
     if len(input_files) == 0:
         print("Please specify an input option.")
         sys.exit(1)
@@ -327,6 +357,8 @@ def any_base_int(s): return int(s, 0)
                                help='Bootloader Software version (string)')
     create_parser.add_argument('--bl-build-date', type=str,
                                help='Bootloader build date (string)')
+    create_parser.add_argument('-wifi', "--wifi-input-file",
+                               help='Path to OTA image for SiWx917 (TA/M4/Combined file)')
 
     # Factory data specific arguments. Will be used to generate the TLV payload.
     create_parser.add_argument('-fd', '--factory-data', action='store_true',