send radio as patch in incremental updates

Change-Id: Id7ca5f2499a4dad650947a6014d7fdb6d0a08e10

1 files changed, 43 insertions, 10 deletions

diff --git a/releasetools.py b/releasetools.py
index 493f94f..aa3786a 100644
--- a/releasetools.py
+++ b/releasetools.py
@@ -12,12 +12,15 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import sha
+
 import common
 
 def FullOTA_InstallEnd(info):
   try:
     bootloader_img = info.input_zip.read("RADIO/bootloader.img")
     common.ZipWriteStr(info.output_zip, "bootloader.img", bootloader_img)
+    info.script.Print("Writing bootloader...")
     info.script.WriteRawImage("bootloader", "bootloader.img")
   except KeyError:
     print "no bootloader.img in target_files; skipping install"
@@ -25,10 +28,23 @@ def FullOTA_InstallEnd(info):
   try:
     radio_img = info.input_zip.read("RADIO/radio.img")
     common.ZipWriteStr(info.output_zip, "radio.img", radio_img)
+    info.script.Print("Writing radio...")
     info.script.WriteRawImage("radio", "radio.img")
   except KeyError:
     print "no radio.img in target_files; skipping install"
 
+def IncrementalOTA_VerifyEnd(info):
+  try:
+    target_radio_img = info.target_zip.read("RADIO/radio.img")
+    source_radio_img = info.source_zip.read("RADIO/radio.img")
+    if source_radio_img != target_radio_img:
+      info.script.CacheFreeSpaceCheck(len(source_radio_img))
+      info.script.PatchCheck("MTD:radio:%d:%s:%d:%s" % (
+          len(source_radio_img), sha.sha(source_radio_img).hexdigest(),
+          len(target_radio_img), sha.sha(target_radio_img).hexdigest()))
+  except KeyError:
+    pass
+
 def IncrementalOTA_InstallEnd(info):
   try:
     target_bootloader_img = info.target_zip.read("RADIO/bootloader.img")
@@ -41,24 +57,41 @@ def IncrementalOTA_InstallEnd(info):
       print "bootloader unchanged; skipping"
     else:
       common.ZipWriteStr(info.output_zip, "bootloader.img", bootloader_img)
+      info.script.Print("Writing bootloader...")
       info.script.WriteRawImage("bootloader", "bootloader.img")
 
   except KeyError:
     print "no bootloader.img in target target_files; skipping install"
 
+
   try:
-    target_radio_img = info.target_zip.read("RADIO/radio.img")
+    tf = common.File("radio.img", info.target_zip.read("RADIO/radio.img"))
     try:
-      source_radio_img = info.source_zip.read("RADIO/radio.img")
-    except KeyError:
-      source_radio_img = None
+      sf = common.File("radio.img", info.source_zip.read("RADIO/radio.img"))
 
-    if source_radio_img == target_radio_img:
-      print "radio unchanged; skipping"
-    else:
-      # TODO: send radio image as binary patch
+      diff = common.Difference(tf, sf)
+      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.
+        tf.AddToZip(info.output_zip)
+        info.script.Print("Writing radio...")
+        info.script.WriteRawImage("radio", tf.name)
+      else:
+        common.ZipWriteStr(info.output_zip, "radio.img.p", d)
+        info.script.Print("Patching radio...")
+        info.script.ApplyPatch(
+            "MTD:radio:%d:%s:%d:%s" % (sf.size, sf.sha1, tf.size, tf.sha1),
+            "-", tf.size, tf.sha1, sf.sha1, "radio.img.p")
+
+    except KeyError:
+      # failed to read SOURCE radio image: include the whole target
+      # radio image.
+      tf.AddToZip(info.output_zip)
+      info.script.Print("Writing radio...")
+      info.script.WriteRawImage("radio", tf.name)
 
-      common.ZipWriteStr(info.output_zip, "radio.img", radio_img)
-      info.script.WriteRawImage("radio", "radio.img")
   except KeyError:
+    # failed to read TARGET radio image: don't include any radio in update.
     print "no radio.img in target target_files; skipping install"