From 3a1ba2c713e7e939a3355e345ec5c1cec08e1873 Mon Sep 17 00:00:00 2001
From: Bjoern Kerler <info@revskills.de>
Date: Sat, 24 Aug 2024 10:59:27 +0200
Subject: [PATCH] Fix efuse support

---
 mtkclient/Library/DA/xflash/extension/xflash.py | 3 ++-
 mtkclient/Library/DA/xml/extension/v6.py        | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/mtkclient/Library/DA/xflash/extension/xflash.py b/mtkclient/Library/DA/xflash/extension/xflash.py
index f03eab4d..64a7565e 100755
--- a/mtkclient/Library/DA/xflash/extension/xflash.py
+++ b/mtkclient/Library/DA/xflash/extension/xflash.py
@@ -170,7 +170,8 @@ def patch(self):
                 daextdata[ufshcd_get_free_tag_ptr:ufshcd_get_free_tag_ptr + 4] = pack("<I", ufshcd_get_free_tag)
                 daextdata[ufshcd_queuecommand_ptr:ufshcd_queuecommand_ptr + 4] = pack("<I", ufshcd_queuecommand)
                 daextdata[ptr_g_ufs_hba_ptr:ptr_g_ufs_hba_ptr + 4] = pack("<I", g_ufs_hba)
-                daextdata[efuse_addr_ptr:efuse_addr_ptr+4]=pack("<I", efuse_addr)
+                if efuse_addr_ptr!=-1:
+                    daextdata[efuse_addr_ptr:efuse_addr_ptr+4]=pack("<I", efuse_addr)
 
                 # print(hexlify(daextdata).decode('utf-8'))
                 # open("daext.bin","wb").write(daextdata)
diff --git a/mtkclient/Library/DA/xml/extension/v6.py b/mtkclient/Library/DA/xml/extension/v6.py
index b7a28d32..efb558e5 100755
--- a/mtkclient/Library/DA/xml/extension/v6.py
+++ b/mtkclient/Library/DA/xml/extension/v6.py
@@ -199,7 +199,8 @@ def patch(self):
                 daextdata[ufshcd_get_free_tag_ptr:ufshcd_get_free_tag_ptr + 4] = pack("<I", ufshcd_get_free_tag)
                 daextdata[ufshcd_queuecommand_ptr:ufshcd_queuecommand_ptr + 4] = pack("<I", ufshcd_queuecommand)
                 daextdata[ptr_g_ufs_hba_ptr:ptr_g_ufs_hba_ptr + 4] = pack("<I", g_ufs_hba)
-                daextdata[efuse_addr_ptr:efuse_addr_ptr + 4] = pack("<I", efuse_addr)
+                if efuse_addr_ptr != -1:
+                    daextdata[efuse_addr_ptr:efuse_addr_ptr + 4] = pack("<I", efuse_addr)
 
                 # print(hexlify(daextdata).decode('utf-8'))
                 # open("daext.bin","wb").write(daextdata)