From 69d9a7e7723768f3480d578af883c4d76f8a5765 Mon Sep 17 00:00:00 2001
From: bongbui321 <hoang.bui.vh@gmail.com>
Date: Wed, 6 Mar 2024 00:04:47 -0500
Subject: [PATCH] fix setactiveslot

---
 edlclient/Library/firehose.py | 27 ++++++++++++++++-----------
 edlclient/Library/gpt.py      |  4 ++--
 2 files changed, 18 insertions(+), 13 deletions(-)

diff --git a/edlclient/Library/firehose.py b/edlclient/Library/firehose.py
index 3022e53..724983e 100755
--- a/edlclient/Library/firehose.py
+++ b/edlclient/Library/firehose.py
@@ -1303,19 +1303,24 @@ def cmd_getstorageinfo(self):
             return None
 
     def cmd_setactiveslot(self, slot: str):
-        def cmd_patch_multiple(lun, start_sector_patch, byte_offset_patch, headeroffset, pdata, header):
-            offset = 0 
+        def cmd_patch_multiple(lun, start_sector_patch, byte_offset_patch, headeroffset, pdata,  header):
+            offset = 0
             header_size = len(header)
-            pdata_size = len(pdata)
-            write_size = pdata_size # with assumption pdata_size > header_size
-            patch_subset_size = 4
-            for i in range(0, write_size, patch_subset_size):
-                pdata_subset = int(unpack("<I", pdata[offset:offset+patch_subset_size])[0])
-                self.cmd_patch(lun, start_sector_patch, byte_offset_patch, pdata_subset, patch_subset_size, True)
+            size_each_patch = 4
+            write_size = len(pdata)
+            for i in range(0, write_size, size_each_patch):
+                pdata_subset = int(unpack("<I", pdata[offset:offset+size_each_patch])[0])
+                self.cmd_patch( lun, start_sector_patch, \
+                                byte_offset_patch + offset, \
+                                pdata_subset, \
+                                size_each_patch, True)
                 if i < header_size:
-                    header_subset = int(unpack("<I", header[offset:offset+patch_subset_size])[0])
-                    self.cmd_patch(lun, headeroffset, 0, header_subset, patch_subset_size, True)
-                offset += patch_subset_size
+                    header_subset = int(unpack("<I", header[offset:offset+size_each_patch])[0])
+                    self.cmd_patch(lun, headeroffset, \
+                                offset, \
+                                    header_subset, \
+                                    size_each_patch, True)
+                offset += size_each_patch
             return True
 
         if slot.lower() not in ["a", "b"]:
diff --git a/edlclient/Library/gpt.py b/edlclient/Library/gpt.py
index 50ac799..0ebb6f2 100755
--- a/edlclient/Library/gpt.py
+++ b/edlclient/Library/gpt.py
@@ -224,7 +224,7 @@ def __init__(self, data):
             self.name = sh.string(72)
 
         def create(self):
-            val = pack("16s16sQQQ72s", self.type, self.unique, self.first_lba, self.last_lba, self.flags, self.name)
+            val = pack("<16s16sQQQ72s", self.type, self.unique, self.first_lba, self.last_lba, self.flags, self.name)
             return val
 
     class efi_type(Enum):
@@ -498,7 +498,7 @@ def patch(self, data:bytes, partitionname="boot", active: bool = True):
                             if active:
                                 flags |= AB_PARTITION_ATTR_SLOT_ACTIVE << (AB_FLAG_OFFSET*8)
                             else:
-                                flags |= AB_PARTITION_ATTR_UNBOOTABLE << (AB_FLAG_OFFSET*8)
+                                flags &= AB_PARTITION_ATTR_UNBOOTABLE << (AB_FLAG_OFFSET*8)
                             partentry.flags = flags
                             pdata = partentry.create()
                             return pdata, partition.entryoffset