commit 47d70a55598998005b7c243c92d0ea405eb184c3
author Daniel Zheng <zhengdaniel@google.com> Tue Feb 14 17:44:40 2023 +0000
committer Daniel Zheng <zhengdaniel@google.com> Wed Mar 08 23:07:48 2023 +0000
tree 270a3a63ca35d8bf7cbed78bad026347f0567784
parent 42908c45e56fa31b2de8404ca7eb1b34d60156db

Added support for FlashSuperLayoutTask

Test: testing flashall code with this change
Bug: 194686221
Change-Id: Ibb96980402db51b48c9a296338f29195f68c65bf

fastboot/task.h

diff --git a/fastboot/task.h b/fastboot/task.h
index d8b9e21..7acb3d5 100644
--- a/fastboot/task.h
+++ b/fastboot/task.h
@@ -20,6 +20,8 @@
 
 #include "fastboot.h"
 #include "fastboot_driver.h"
+#include "super_flash_helper.h"
+#include "util.h"
 
 class Task {
   public:
@@ -52,4 +54,19 @@
   private:
     const std::string reboot_target_ = "";
     FlashingPlan* fp_;
-};
\ No newline at end of file
+};
+
+class FlashSuperLayoutTask : public Task {
+  public:
+    FlashSuperLayoutTask(const std::string& super_name, std::unique_ptr<SuperFlashHelper> helper,
+                         SparsePtr sparse_layout);
+    static std::unique_ptr<FlashSuperLayoutTask> Initialize(FlashingPlan* fp,
+                                                            std::vector<ImageEntry>& os_images);
+    using ImageEntry = std::pair<const Image*, std::string>;
+    void Run() override;
+
+  private:
+    const std::string super_name_;
+    std::unique_ptr<SuperFlashHelper> helper_;
+    SparsePtr sparse_layout_;
+};