From 4e7bdf6c473297e0c1b844eb27c2ff4aa95399ca Mon Sep 17 00:00:00 2001
From: cfig <yuyezhong@gmail.com>
Date: Sun, 19 Jun 2022 15:53:39 +0800
Subject: [PATCH] dtbo: support 'pull' and 'flash'

---
 bbootimg/src/main/kotlin/packable/DtboParser.kt | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/bbootimg/src/main/kotlin/packable/DtboParser.kt b/bbootimg/src/main/kotlin/packable/DtboParser.kt
index 51594ec..49f0143 100644
--- a/bbootimg/src/main/kotlin/packable/DtboParser.kt
+++ b/bbootimg/src/main/kotlin/packable/DtboParser.kt
@@ -57,6 +57,19 @@ class DtboParser(val workDir: File) : IPackable {
             .printPackSummary()
     }
 
+    override fun pull(fileName: String, deviceName: String) {
+        super.pull(fileName, deviceName)
+    }
+
+    override fun flash(fileName: String, deviceName: String) {
+        val stem = fileName.substring(0, fileName.indexOf("."))
+        super.flash("$fileName.signed", stem)
+
+        if (File("vbmeta.img.signed").exists()) {
+            super.flash("vbmeta.img.signed", "vbmeta")
+        }
+    }
+
     override fun `@verify`(fileName: String) {
         super.`@verify`(fileName)
     }