commit 576efc53d5499558c911dbe476487016f9490592
author Kelvin Zhang <zhangkelvin@google.com> Tue Dec 01 12:06:40 2020 -0500
committer Treehugger Robot <treehugger-gerrit@google.com> Mon Dec 14 19:28:58 2020 +0000
tree 3a0174c617b10f0dfe70f336ebe2878b672819aa
parent eb8703b9f6db3fb386507b71f180c98ca3137638

Add host side simulation python script

This enables host side simulation of partial updates with the original
complete target file

Bug: 171519321
Test: run this script with a partial update

     python3 simulate_ota.py --source
     ~/aosp/aosp_cf_x86_phone-target_files-6949164.zip --target
     ~/aosp/aosp_cf_x86_phone-target_files-6937553.zip
     ~/aosp/cf_x86_downgrade.zip

Change-Id: Iec52d8ac585b499d50bdb0c89d203ea353f2e94f

scripts/update_payload/payload.py

diff --git a/scripts/update_payload/payload.py b/scripts/update_payload/payload.py
index fe3a450..6b28046 100644
--- a/scripts/update_payload/payload.py
+++ b/scripts/update_payload/payload.py
@@ -124,8 +124,11 @@
     if zipfile.is_zipfile(payload_file):
       with zipfile.ZipFile(payload_file) as zfp:
         with zfp.open("payload.bin") as payload_fp:
-          payload_file = io.BytesIO(payload_fp.read())
-    self.payload_file = payload_file
+          self.payload_file = io.BytesIO(payload_fp.read())
+    elif isinstance(payload_file, str):
+      self.payload_file = open(payload_file, "rb")
+    else:
+      self.payload_file = payload_file
     self.payload_file_offset = payload_file_offset
     self.manifest_hasher = None
     self.is_init = False