Actually override ASW & Version lengths

bri3d · Oct 2, 2024 · 3452000 · 3452000
1 parent 28e133a
commit 3452000
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/lib/haldex_flash_utils.py b/lib/haldex_flash_utils.py
@@ -181,7 +181,9 @@ def flash_bin(
     prepared_blocks = prepare_blocks(flash_info, input_blocks, callback)
 
     # Manually override flash_info
-    flash_info.block_lengths[2] = len(prepared_blocks['FD_1DATA'].block_encrypted_bytes)
+    flash_info.block_lengths[2] = len(prepared_blocks["FD_1DATA"].block_encrypted_bytes)
+    flash_info.block_lengths[3] = len(prepared_blocks["FD_2DATA"].block_encrypted_bytes)
+    flash_info.block_lengths[4] = len(prepared_blocks["FD_3DATA"].block_encrypted_bytes)
 
     flash_uds.flash_blocks(
         flash_info=flash_info,