commit 107dfac7c0d7fd7cad6e7d41368d8878ea9ba2b0
author LuK1337 <priv.luk@gmail.com> Sat Jul 29 11:16:41 2023 +0200
committer Łukasz Patron <priv.luk@gmail.com> Sat Jul 29 12:19:56 2023 +0200
tree ec09e2a1f97c3360b65933d97ffe329e4384c4fa
parent 3a726d506c3d5e1e77f9ff242428384f035887aa

recovery: Ask to reboot before installing anything after A/B update

Change-Id: Id3e44ddbe11119fcc66d379216f8f5e683979a9f

install/install.cpp

diff --git a/install/install.cpp b/install/install.cpp
index 61bab17..ff256a0 100644
--- a/install/install.cpp
+++ b/install/install.cpp
@@ -62,6 +62,7 @@
 
 using namespace std::chrono_literals;
 
+bool ask_to_ab_reboot(Device* device);
 bool ask_to_continue_unverified(Device* device);
 bool ask_to_continue_downgrade(Device* device);
 
@@ -383,6 +384,14 @@
     return INSTALL_ERROR;
   }
 
+  static bool ab_package_installed = false;
+  if (ab_package_installed) {
+    if (ask_to_ab_reboot(device)) {
+      Reboot("userrequested,recovery,ui");
+    }
+    return INSTALL_ERROR;
+  }
+
   if (package_is_ab) {
     CHECK(package->GetType() == PackageType::kFile);
   }
@@ -568,7 +577,11 @@
     LOG(FATAL) << "Invalid status code " << status;
   }
   if (package_is_ab) {
+    ab_package_installed = true;
     PerformPowerwashIfRequired(zip, device);
+    if (ask_to_ab_reboot(device)) {
+      Reboot("userrequested,recovery,ui");
+    }
   }
 
   return INSTALL_SUCCESS;

recovery.cpp

diff --git a/recovery.cpp b/recovery.cpp
index 75430a2..36cb78b 100644
--- a/recovery.cpp
+++ b/recovery.cpp
@@ -175,6 +175,12 @@
   return (chosen_item == 1);
 }
 
+bool ask_to_ab_reboot(Device* device) {
+  device->GetUI()->SetProgressType(RecoveryUI::EMPTY);
+  return yes_no(device, "To install additional packages, you need to reboot recovery first",
+                "Do you want to reboot to recovery now?");
+}
+
 bool ask_to_continue_unverified(Device* device) {
   if (get_build_type() == "user") {
     return false;