commit c6a3c3e80ee1cb450547d288dbea9ec42f1591e3
author Louis Popi <theh2o64@gmail.com> Sat Jul 16 16:19:13 2016 +0200
committer Nathan Prevette <nprevette420@gmail.com> Wed Jul 20 16:07:29 2016 -0700
tree e7a520c0f9825eeac2bd9634428869cd53c922f2
parent 3ac88f02f7460b8f371b21baca6d213f11105fd3

releasetool: Add support for partition specific *.img

Change-Id: Icaabb51d3d72ede9a77f96bbc0fe493aa979e483

releasetools.py

diff --git a/releasetools.py b/releasetools.py
index f50b751..4dc95d8 100755
--- a/releasetools.py
+++ b/releasetools.py
@@ -23,7 +23,7 @@
 bootImages = {}
 binImages = {}
 fwImages = {}
-
+imgImages = {}
 
 # Parse filesmap file containing firmware residing places
 def LoadFilesMap(zip, name="RADIO/filesmap"):
@@ -84,6 +84,7 @@
   boot = {}
   bin = {}
   fw = {}
+  img = {}
 
   for f in files:
     extIdx = -1
@@ -97,9 +98,11 @@
       boot[f] = files[f]
     elif dotSeparated[extIdx] == 'bin':
       bin[f] = files[f]
+    elif dotSeparated[extIdx] == 'img':
+      img[f] = files[f]
     else:
       fw[f] = files[f]
-  return boot, bin, fw
+  return boot, bin, fw, img
 
 
 # Prepare radio-update files and verify them
@@ -164,7 +167,8 @@
   global bootImages
   global binImages
   global fwImages
-  bootImages, binImages, fwImages = SplitFwTypes(update_list)
+  global imgImages
+  bootImages, binImages, fwImages, imgImages = SplitFwTypes(update_list)
 
   # If there are incremental patches verify them
   if largest_source_size != 0:
@@ -188,7 +192,6 @@
         continue
       info.script.PatchCheck("EMMC:%s:%d:%s:%d:%s" %
               (dest, sf.size, sf.sha1, tf.size, tf.sha1))
-
     last_mounted = ""
     for f in fwImages:
       dest, destBak, tf, sf = fwImages[f]
@@ -210,6 +213,14 @@
       else:
         dest = dest + "/" + f
       info.script.PatchCheck(dest, tf.sha1, sf.sha1)
+    for f in imgImages:
+      dest, tf, sf = imgImages[f]
+      # Not incremental
+      if sf is None:
+        continue
+      info.script.PatchCheck("EMMC:%s:%d:%s:%d:%s" %
+              (dest, sf.size, sf.sha1, tf.size, tf.sha1))
+    last_mounted = ""
 
     info.script.CacheFreeSpaceCheck(largest_source_size)
   return True
@@ -318,6 +329,13 @@
     script.AppendExtra('unmount("/firmware");')
   return
 
+# This function handles *.img images
+def InstallImgImages(script, files):
+  for f in files:
+    dest, _, tf, sf = files[f]
+    InstallRawImage(script, f, dest, tf, sf)
+  return
+
 
 def OTA_InstallEnd(info):
   print "Applying radio-update script modifications..."
@@ -330,6 +348,8 @@
     InstallBinImages(info.script, binImages)
   if fwImages != {}:
     InstallFwImages(info.script, fwImages)
+  if imgImages != {}:
+    InstallImgImages(info.script, imgImages)
   return