commit 99da2214e46472d91e181962ce29e9e94a1ad814
author Kelvin Zhang <zhangkelvin@google.com> Fri Sep 27 17:24:19 2024 -0700
committer Treehugger Robot <android-test-infra-autosubmit@system.gserviceaccount.com> Thu Oct 03 20:07:06 2024 +0000
tree 0aebc783e2c94abace2e14eef95c324e6b40df1a
parent 6d8a94df0c64838510c224dae1744581761561fe

Fix partial extraction feature in ota_extractor

In aosp/3164519 , partial extraction was accidentally removed during
refactoring. Fix

Test: th
Change-Id: I6a0a316ea627ae5559a35e199c1853cff6a0d789

aosp/ota_extractor.cc

diff --git a/aosp/ota_extractor.cc b/aosp/ota_extractor.cc
index 42270f4..29ba44c 100644
--- a/aosp/ota_extractor.cc
+++ b/aosp/ota_extractor.cc
@@ -206,6 +206,10 @@
 
   if (FLAGS_single_thread) {
     for (const auto& partition : manifest.partitions()) {
+      if (!partitions.empty() &&
+          partitions.count(partition.partition_name()) == 0) {
+        continue;
+      }
       if (!ExtractImageFromPartition(manifest,
                                      partition,
                                      data_begin,
@@ -221,6 +225,10 @@
   } else {
     std::vector<std::pair<std::future<bool>, std::string>> futures;
     for (const auto& partition : manifest.partitions()) {
+      if (!partitions.empty() &&
+          partitions.count(partition.partition_name()) == 0) {
+        continue;
+      }
       futures.push_back(std::make_pair(std::async(std::launch::async,
                                                   ExtractImageFromPartition,
                                                   manifest,