From c086370440bb32501da427718a7313f12ec2c841 Mon Sep 17 00:00:00 2001 From: Tao Bao Date: Mon, 7 Mar 2016 13:03:19 -0800 Subject: [PATCH] releasetools: Support OTAs that have OEM properties changes. We need to handle a special case that an OTA goes from a source build without OEM properties to a target build with those properties (or vice versa). Add support in OTA scripts to deal the case properly, by a) using two oem_props variables to handle source and target builds respectively; b) adjusting the fingerprint/thumbprint assertions to allow a mix of both. Change-Id: Ib517c366730b967a974c89528df9d42887c92ec2 --- tools/releasetools/edify_generator.py | 11 +++ tools/releasetools/ota_from_target_files.py | 97 ++++++++++++--------- 2 files changed, 68 insertions(+), 40 deletions(-) diff --git a/tools/releasetools/edify_generator.py b/tools/releasetools/edify_generator.py index 57f8cda07c..ecdc16731f 100644 --- a/tools/releasetools/edify_generator.py +++ b/tools/releasetools/edify_generator.py @@ -118,6 +118,17 @@ class EdifyGenerator(object): " or ".join(fp)) self.script.append(cmd) + def AssertFingerprintOrThumbprint(self, fp, tp): + """Assert that the current recovery build fingerprint is fp, or thumbprint + is tp.""" + cmd = ('getprop("ro.build.fingerprint") == "{fp}" ||\n' + ' getprop("ro.build.thumbprint") == "{tp}" ||\n' + ' abort("Package expects build fingerprint of {fp} or ' + 'thumbprint of {tp}; this device has a fingerprint of " ' + '+ getprop("ro.build.fingerprint") and a thumbprint of " ' + '+ getprop("ro.build.thumbprint") + ".");').format(fp=fp, tp=tp) + self.script.append(cmd) + def AssertOlderBuild(self, timestamp, timestamp_text): """Assert that the build on the device is older (or the same as) the given timestamp.""" diff --git a/tools/releasetools/ota_from_target_files.py b/tools/releasetools/ota_from_target_files.py index 8923a0532c..9e9d480f6c 100755 --- a/tools/releasetools/ota_from_target_files.py +++ b/tools/releasetools/ota_from_target_files.py @@ -438,7 +438,7 @@ def SignOutput(temp_zip_name, output_zip_name): def AppendAssertions(script, info_dict, oem_dict=None): oem_props = info_dict.get("oem_fingerprint_properties") - if oem_props is None or len(oem_props) == 0: + if not oem_props: device = GetBuildProp("ro.product.device", info_dict) script.AssertDevice(device) else: @@ -527,10 +527,10 @@ def WriteFullOTAPackage(input_zip, output_zip): # in the target build. script = edify_generator.EdifyGenerator(3, OPTIONS.info_dict) - oem_props = OPTIONS.info_dict.get("oem_fingerprint_properties") recovery_mount_options = OPTIONS.info_dict.get("recovery_mount_options") + oem_props = OPTIONS.info_dict.get("oem_fingerprint_properties") oem_dict = None - if oem_props is not None and len(oem_props) > 0: + if oem_props: if OPTIONS.oem_source is None: raise common.ExternalError("OEM source required for this build") if not OPTIONS.oem_no_mount: @@ -538,9 +538,10 @@ def WriteFullOTAPackage(input_zip, output_zip): oem_dict = common.LoadDictionaryFromLines( open(OPTIONS.oem_source).readlines()) + target_fp = CalculateFingerprint(oem_props, oem_dict, + OPTIONS.target_info_dict) metadata = { - "post-build": CalculateFingerprint(oem_props, oem_dict, - OPTIONS.info_dict), + "post-build": target_fp, "pre-device": GetOemProperty("ro.product.device", oem_props, oem_dict, OPTIONS.info_dict), "post-timestamp": GetBuildProp("ro.build.date.utc", OPTIONS.info_dict), @@ -609,8 +610,7 @@ else if get_stage("%(bcb_dev)s") == "3/3" then """ % bcb_dev) # Dump fingerprints - script.Print("Target: %s" % CalculateFingerprint( - oem_props, oem_dict, OPTIONS.info_dict)) + script.Print("Target: %s" % target_fp) device_specific.FullOTA_InstallBegin() @@ -772,17 +772,18 @@ def WriteBlockIncrementalOTAPackage(target_zip, source_zip, output_zip): target_version = OPTIONS.target_info_dict["recovery_api_version"] if source_version == 0: - print ("WARNING: generating edify script for a source that " - "can't install it.") + print("WARNING: generating edify script for a source that " + "can't install it.") script = edify_generator.EdifyGenerator( source_version, OPTIONS.target_info_dict, fstab=OPTIONS.source_info_dict["fstab"]) - oem_props = OPTIONS.info_dict.get("oem_fingerprint_properties") recovery_mount_options = OPTIONS.source_info_dict.get( "recovery_mount_options") + source_oem_props = OPTIONS.source_info_dict.get("oem_fingerprint_properties") + target_oem_props = OPTIONS.target_info_dict.get("oem_fingerprint_properties") oem_dict = None - if oem_props is not None and len(oem_props) > 0: + if source_oem_props or target_oem_props: if OPTIONS.oem_source is None: raise common.ExternalError("OEM source required for this build") if not OPTIONS.oem_no_mount: @@ -791,8 +792,8 @@ def WriteBlockIncrementalOTAPackage(target_zip, source_zip, output_zip): open(OPTIONS.oem_source).readlines()) metadata = { - "pre-device": GetOemProperty("ro.product.device", oem_props, oem_dict, - OPTIONS.source_info_dict), + "pre-device": GetOemProperty("ro.product.device", source_oem_props, + oem_dict, OPTIONS.source_info_dict), "ota-type": "BLOCK", } @@ -827,9 +828,9 @@ def WriteBlockIncrementalOTAPackage(target_zip, source_zip, output_zip): metadata=metadata, info_dict=OPTIONS.source_info_dict) - source_fp = CalculateFingerprint(oem_props, oem_dict, + source_fp = CalculateFingerprint(source_oem_props, oem_dict, OPTIONS.source_info_dict) - target_fp = CalculateFingerprint(oem_props, oem_dict, + target_fp = CalculateFingerprint(target_oem_props, oem_dict, OPTIONS.target_info_dict) metadata["pre-build"] = source_fp metadata["post-build"] = target_fp @@ -925,32 +926,39 @@ else if get_stage("%(bcb_dev)s") != "3/3" then """ % bcb_dev) # Dump fingerprints - script.Print("Source: %s" % CalculateFingerprint( - oem_props, oem_dict, OPTIONS.source_info_dict)) - script.Print("Target: %s" % CalculateFingerprint( - oem_props, oem_dict, OPTIONS.target_info_dict)) + script.Print(source_fp) + script.Print(target_fp) script.Print("Verifying current system...") device_specific.IncrementalOTA_VerifyBegin() - if oem_props is None: - # When blockimgdiff version is less than 3 (non-resumable block-based OTA), - # patching on a device that's already on the target build will damage the - # system. Because operations like move don't check the block state, they - # always apply the changes unconditionally. - if blockimgdiff_version <= 2: + # When blockimgdiff version is less than 3 (non-resumable block-based OTA), + # patching on a device that's already on the target build will damage the + # system. Because operations like move don't check the block state, they + # always apply the changes unconditionally. + if blockimgdiff_version <= 2: + if source_oem_props is None: script.AssertSomeFingerprint(source_fp) else: - script.AssertSomeFingerprint(source_fp, target_fp) - else: - if blockimgdiff_version <= 2: script.AssertSomeThumbprint( GetBuildProp("ro.build.thumbprint", OPTIONS.source_info_dict)) - else: + + else: # blockimgdiff_version > 2 + if source_oem_props is None and target_oem_props is None: + script.AssertSomeFingerprint(source_fp, target_fp) + elif source_oem_props is not None and target_oem_props is not None: script.AssertSomeThumbprint( GetBuildProp("ro.build.thumbprint", OPTIONS.target_info_dict), GetBuildProp("ro.build.thumbprint", OPTIONS.source_info_dict)) + elif source_oem_props is None and target_oem_props is not None: + script.AssertFingerprintOrThumbprint( + source_fp, + GetBuildProp("ro.build.thumbprint", OPTIONS.target_info_dict)) + else: + script.AssertFingerprintOrThumbprint( + target_fp, + GetBuildProp("ro.build.thumbprint", OPTIONS.source_info_dict)) # Check the required cache size (i.e. stashed blocks). size = [] @@ -1068,7 +1076,7 @@ def WriteVerifyPackage(input_zip, output_zip): recovery_mount_options = OPTIONS.info_dict.get( "recovery_mount_options") oem_dict = None - if oem_props is not None and len(oem_props) > 0: + if oem_props: if OPTIONS.oem_source is None: raise common.ExternalError("OEM source required for this build") if not OPTIONS.oem_no_mount: @@ -1411,17 +1419,18 @@ def WriteIncrementalOTAPackage(target_zip, source_zip, output_zip): target_version = OPTIONS.target_info_dict["recovery_api_version"] if source_version == 0: - print ("WARNING: generating edify script for a source that " - "can't install it.") + print("WARNING: generating edify script for a source that " + "can't install it.") script = edify_generator.EdifyGenerator( source_version, OPTIONS.target_info_dict, fstab=OPTIONS.source_info_dict["fstab"]) - oem_props = OPTIONS.info_dict.get("oem_fingerprint_properties") recovery_mount_options = OPTIONS.source_info_dict.get( "recovery_mount_options") + source_oem_props = OPTIONS.source_info_dict.get("oem_fingerprint_properties") + target_oem_props = OPTIONS.target_info_dict.get("oem_fingerprint_properties") oem_dict = None - if oem_props is not None and len(oem_props) > 0: + if source_oem_props or target_oem_props: if OPTIONS.oem_source is None: raise common.ExternalError("OEM source required for this build") if not OPTIONS.oem_no_mount: @@ -1430,8 +1439,8 @@ def WriteIncrementalOTAPackage(target_zip, source_zip, output_zip): open(OPTIONS.oem_source).readlines()) metadata = { - "pre-device": GetOemProperty("ro.product.device", oem_props, oem_dict, - OPTIONS.source_info_dict), + "pre-device": GetOemProperty("ro.product.device", source_oem_props, + oem_dict, OPTIONS.source_info_dict), "ota-type": "FILE", } @@ -1474,17 +1483,25 @@ def WriteIncrementalOTAPackage(target_zip, source_zip, output_zip): else: vendor_diff = None - target_fp = CalculateFingerprint(oem_props, oem_dict, + target_fp = CalculateFingerprint(target_oem_props, oem_dict, OPTIONS.target_info_dict) - source_fp = CalculateFingerprint(oem_props, oem_dict, + source_fp = CalculateFingerprint(source_oem_props, oem_dict, OPTIONS.source_info_dict) - if oem_props is None: + if source_oem_props is None and target_oem_props is None: script.AssertSomeFingerprint(source_fp, target_fp) - else: + elif source_oem_props is not None and target_oem_props is not None: script.AssertSomeThumbprint( GetBuildProp("ro.build.thumbprint", OPTIONS.target_info_dict), GetBuildProp("ro.build.thumbprint", OPTIONS.source_info_dict)) + elif source_oem_props is None and target_oem_props is not None: + script.AssertFingerprintOrThumbprint( + source_fp, + GetBuildProp("ro.build.thumbprint", OPTIONS.target_info_dict)) + else: + script.AssertFingerprintOrThumbprint( + target_fp, + GetBuildProp("ro.build.thumbprint", OPTIONS.source_info_dict)) metadata["pre-build"] = source_fp metadata["post-build"] = target_fp