bootsignature: includes dt image size for signed boot image

[android-x86/system-extras.git] / verity / BootSignature.java

diff --git a/verity/BootSignature.java b/verity/BootSignature.java
index 3cf9499..51b1558 100644 (file)
--- a/verity/BootSignature.java
+++ b/verity/BootSignature.java
@@ -203,11 +203,13 @@ public class BootSignature extends ASN1Object
         int secondSize = image.getInt();
         image.getLong(); // second_addr + tags_addr
         int pageSize = image.getInt();
+        int dtSize   = image.getInt(); //device tree size
 
         int length = pageSize // include the page aligned image header
                 + ((kernelSize + pageSize - 1) / pageSize) * pageSize
                 + ((ramdskSize + pageSize - 1) / pageSize) * pageSize
-                + ((secondSize + pageSize - 1) / pageSize) * pageSize;
+                + ((secondSize + pageSize - 1) / pageSize) * pageSize
+                + ((dtSize     + pageSize - 1) / pageSize) * pageSize;
 
         length = ((length + pageSize - 1) / pageSize) * pageSize;