| import common |
| import struct |
| |
| def FindRadio(zipfile): |
| try: |
| return zipfile.read("RADIO/radio.img") |
| except KeyError: |
| return None |
| |
| |
| def FullOTA_InstallEnd(info): |
| try: |
| bootloader_img = info.input_zip.read("RADIO/bootloader.img") |
| except KeyError: |
| print "no bootloader.img in target_files; skipping install" |
| else: |
| WriteBootloader(info, bootloader_img) |
| |
| radio_img = FindRadio(info.input_zip) |
| if radio_img: |
| WriteRadio(info, radio_img) |
| else: |
| print "no radio.img in target_files; skipping install" |
| |
| |
| def IncrementalOTA_VerifyEnd(info): |
| target_radio_img = FindRadio(info.target_zip) |
| source_radio_img = FindRadio(info.source_zip) |
| if not target_radio_img or not source_radio_img: return |
| if source_radio_img != target_radio_img: |
| info.script.CacheFreeSpaceCheck(len(source_radio_img)) |
| radio_type, radio_device = common.GetTypeAndDevice("/radio", info.info_dict) |
| info.script.PatchCheck("%s:%s:%d:%s:%d:%s" % ( |
| radio_type, radio_device, |
| len(source_radio_img), common.sha1(source_radio_img).hexdigest(), |
| len(target_radio_img), common.sha1(target_radio_img).hexdigest())) |
| |
| |
| def IncrementalOTA_InstallEnd(info): |
| try: |
| target_bootloader_img = info.target_zip.read("RADIO/bootloader.img") |
| try: |
| source_bootloader_img = info.source_zip.read("RADIO/bootloader.img") |
| except KeyError: |
| source_bootloader_img = None |
| |
| if source_bootloader_img == target_bootloader_img: |
| print "bootloader unchanged; skipping" |
| else: |
| WriteBootloader(info, target_bootloader_img) |
| except KeyError: |
| print "no bootloader.img in target target_files; skipping install" |
| |
| tf = FindRadio(info.target_zip) |
| if not tf: |
| # failed to read TARGET radio image: don't include any radio in update. |
| print "no radio.img in target target_files; skipping install" |
| else: |
| tf = common.File("radio.img", tf) |
| |
| sf = FindRadio(info.source_zip) |
| if not sf: |
| # failed to read SOURCE radio image: include the whole target |
| # radio image. |
| WriteRadio(info, tf.data) |
| else: |
| sf = common.File("radio.img", sf) |
| |
| if tf.sha1 == sf.sha1: |
| print "radio image unchanged; skipping" |
| else: |
| diff = common.Difference(tf, sf, diff_program="bsdiff") |
| common.ComputeDifferences([diff]) |
| _, _, d = diff.GetPatch() |
| if d is None or len(d) > tf.size * common.OPTIONS.patch_threshold: |
| # computing difference failed, or difference is nearly as |
| # big as the target: simply send the target. |
| WriteRadio(info, tf.data) |
| else: |
| common.ZipWriteStr(info.output_zip, "radio.img.p", d) |
| info.script.Print("Patching radio...") |
| radio_type, radio_device = common.GetTypeAndDevice( |
| "/radio", info.info_dict) |
| info.script.ApplyPatch( |
| "%s:%s:%d:%s:%d:%s" % (radio_type, radio_device, |
| sf.size, sf.sha1, tf.size, tf.sha1), |
| "-", tf.size, tf.sha1, sf.sha1, "radio.img.p") |
| |
| |
| def WriteRadio(info, radio_img): |
| info.script.Print("Writing radio...") |
| common.ZipWriteStr(info.output_zip, "radio.img", radio_img) |
| _, device = common.GetTypeAndDevice("/radio", info.info_dict) |
| info.script.AppendExtra( |
| 'package_extract_file("radio.img", "%s");' % (device,)) |
| |
| |
| # /* msm8960 bootloader.img format */ |
| # |
| # #define BOOTLDR_MAGIC "BOOTLDR!" |
| # #define BOOTLDR_MAGIC_SIZE 8 |
| # |
| # struct bootloader_images_header { |
| # char magic[BOOTLDR_MAGIC_SIZE]; |
| # unsigned int num_images; |
| # unsigned int start_offset; |
| # unsigned int bootldr_size; |
| # struct { |
| # char name[64]; |
| # unsigned int size; |
| # } img_info[]; |
| # }; |
| |
| def WriteBootloader(info, bootloader): |
| info.script.Print("Writing bootloader...") |
| |
| # bootloader.img contains 6 separate images. Each goes to its own |
| # partition; we write all 6 for development devices but skip one for |
| # release devices.. There are backup partitions of all but the |
| # special one that we also write. The special one is "sbl1", which |
| # does not have a backup, so we don't update it on release devices.. |
| |
| |
| header_fmt = "<8sIII" |
| header_size = struct.calcsize(header_fmt) |
| magic, num_images, start_offset, bootloader_size = struct.unpack( |
| header_fmt, bootloader[:header_size]) |
| assert magic == "BOOTLDR!", "bootloader.img bad magic value" |
| |
| img_info_fmt = "<64sI" |
| img_info_size = struct.calcsize(img_info_fmt) |
| |
| imgs = [struct.unpack(img_info_fmt, |
| bootloader[header_size+i*img_info_size: |
| header_size+(i+1)*img_info_size]) |
| for i in range(num_images)] |
| |
| total = 0 |
| p = start_offset |
| img_dict = {} |
| for name, size in imgs: |
| img_dict[trunc_to_null(name)] = p, size |
| p += size |
| assert p - start_offset == bootloader_size, "bootloader.img corrupted" |
| imgs = img_dict |
| |
| common.ZipWriteStr(info.output_zip, "bootloader-flag.txt", |
| "updating-bootloader" + "\0" * 13) |
| common.ZipWriteStr(info.output_zip, "bootloader-flag-clear.txt", "\0" * 32) |
| |
| _, misc_device = common.GetTypeAndDevice("/misc", info.info_dict) |
| |
| info.script.AppendExtra( |
| 'package_extract_file("bootloader-flag.txt", "%s");' % |
| (misc_device,)) |
| |
| # flashing sbl1 is somewhat dangerous because if we die while doing |
| # it the device can't boot. Do it for development devices but not |
| # release devices. |
| fp = info.info_dict["build.prop"]["ro.build.fingerprint"] |
| if "release-keys" in fp: |
| to_flash = "sbl2 sbl3 tz rpm aboot".split() |
| else: |
| to_flash = "sbl1 sbl2 sbl3 tz rpm aboot".split() |
| |
| # Write the images to separate files in the OTA package |
| for i in to_flash: |
| try: |
| _, device = common.GetTypeAndDevice("/"+i, info.info_dict) |
| except KeyError: |
| print "skipping flash of %s; not in recovery.fstab" % (i,) |
| continue |
| common.ZipWriteStr(info.output_zip, "bootloader.%s.img" % (i,), |
| bootloader[imgs[i][0]:imgs[i][0]+imgs[i][1]]) |
| |
| info.script.AppendExtra('package_extract_file("bootloader.%s.img", "%s");' % |
| (i, device)) |
| |
| info.script.AppendExtra( |
| 'package_extract_file("bootloader-flag-clear.txt", "%s");' % |
| (misc_device,)) |
| |
| try: |
| # there is no "sbl1b" partition |
| for i in "sbl2 sbl3 tz rpm aboot".split(): |
| _, device = common.GetTypeAndDevice("/"+i+"b", info.info_dict) |
| info.script.AppendExtra( |
| 'package_extract_file("bootloader.%s.img", "%s");' % (i, device)) |
| except KeyError: |
| pass |
| |
| |
| def trunc_to_null(s): |
| if '\0' in s: |
| return s[:s.index('\0')] |
| else: |
| return s |
