releasetools.py

import common
import hashlib
import struct

def GetImageEdifyScript(zip, partition, filename):
  file_size = zip.getinfo("RADIO/%s" % filename).file_size
  sha1 = hashlib.sha1()
  sha1.update(zip.read("RADIO/%s" % filename))
  return ('ifelse((sha1_check(read_file("EMMC:%s:%d:%s")) != ""),'
          '(ui_print("%s already up to date")),'
          '(package_extract_file("%s", "%s")));'
          % (partition, file_size, sha1.hexdigest(), partition, filename, partition))

def FullOTA_InstallEnd(info):
  try:
    firmware_img = info.input_zip.read("RADIO/bootloader.img")
  except KeyError:
    print "no bootloader.img in target_files; skipping install"
  else:
    info.script.Print("Writing bootloader.img...")
    common.ZipWriteStr(info.output_zip, "bootloader.img", firmware_img)
    info.script.AppendExtra(GetImageEdifyScript(info.input_zip,
        "/dev/block/platform/sdhci-tegra.3/by-name/OTA", "bootloader.img"));

  try:
    vendor_img = info.input_zip.read("RADIO/vendor.img")
  except KeyError:
    print "no vendor.img in target_files; skipping install"
  else:
    info.script.Print("Writing vendor.img...")
    common.ZipWriteStr(info.output_zip, "vendor.img", vendor_img)
    info.script.AppendExtra(GetImageEdifyScript(info.input_zip,
        "/dev/block/platform/sdhci-tegra.3/by-name/VNR", "vendor.img"));

def IncrementalOTA_InstallEnd(info):
  try:
    source_firmware_img = info.source_zip.read("RADIO/bootloader.img")
  except KeyError:
    print "no bootloader.img in source_files; skipping install"
  else:
    try:
      target_firmware_img = info.target_zip.read("RADIO/bootloader.img")
    except KeyError:
      print "no bootloader.img in target_files; skipping install"
    else:
      if source_firmware_img == target_firmware_img:
        return
      info.script.Print("Writing bootloader.img...")
      common.ZipWriteStr(info.output_zip, "bootloader.img", target_firmware_img)
      info.script.AppendExtra(
          'package_extract_file("bootloader.img", "/dev/block/platform/sdhci-tegra.3/by-name/OTA");')