From c07204fc9531940ad9d854ab3b91d883c0096eea Mon Sep 17 00:00:00 2001
From: Steve Kondik <shade@chemlab.org>
Date: Thu, 3 Apr 2014 22:50:38 -0700
Subject: [PATCH] sr: Add performance control

 * Crank it up when installing

Change-Id: I997d937901ff446834e6c479aaf629bee51de388
---
 adb_install.cpp |  4 ++++
 install.cpp     | 25 ++++++++++++++++++++-----
 install.h       |  2 ++
 3 files changed, 26 insertions(+), 5 deletions(-)

diff --git a/adb_install.cpp b/adb_install.cpp
index 246ab462d..698589602 100644
--- a/adb_install.cpp
+++ b/adb_install.cpp
@@ -178,11 +178,15 @@ void stop_sideload() {
 }
 
 int wait_sideload() {
+    set_perf_mode(true);
+
     pthread_join(sideload_thread, NULL);
 
     ui->FlushKeys();
 
     maybe_restart_adbd();
 
+    set_perf_mode(false);
+
     return sideload_data.result;
 }
diff --git a/install.cpp b/install.cpp
index 5acbcf158..f1b90545d 100644
--- a/install.cpp
+++ b/install.cpp
@@ -45,6 +45,8 @@
 #include "ui.h"
 #include "verifier.h"
 
+#include "cutils/properties.h"
+
 extern RecoveryUI* ui;
 
 #define ASSUMED_UPDATE_BINARY_NAME  "META-INF/com/google/android/update-binary"
@@ -341,6 +343,8 @@ static int
 really_install_package(const char *path, bool* wipe_cache, bool needs_mount,
                        std::vector<std::string>& log_buffer, int retry_count)
 {
+    int ret = 0;
+
     ui->SetBackground(RecoveryUI::INSTALLING_UPDATE);
     ui->Print("Finding update package...\n");
     // Give verification half the progress bar...
@@ -395,6 +399,8 @@ really_install_package(const char *path, bool* wipe_cache, bool needs_mount,
     }
     LOGI("%zu key(s) loaded from %s\n", loadedKeys.size(), PUBLIC_KEYS_FILE);
 
+    set_perf_mode(true);
+
     // Verify package.
     ui->Print("Verifying update package...\n");
     auto t0 = std::chrono::system_clock::now();
@@ -419,7 +425,8 @@ really_install_package(const char *path, bool* wipe_cache, bool needs_mount,
         log_buffer.push_back(android::base::StringPrintf("error: %d", kZipVerificationFailure));
 
         sysReleaseMap(&map);
-        return INSTALL_CORRUPT;
+        ret = INSTALL_CORRUPT;
+        goto out;
     }
 
     // Try to open the package.
@@ -430,7 +437,8 @@ really_install_package(const char *path, bool* wipe_cache, bool needs_mount,
         log_buffer.push_back(android::base::StringPrintf("error: %d", kZipOpenFailure));
 
         sysReleaseMap(&map);
-        return INSTALL_CORRUPT;
+        ret = INSTALL_CORRUPT;
+        goto out;
     }
 
     // Verify and install the contents of the package.
@@ -439,7 +447,7 @@ really_install_package(const char *path, bool* wipe_cache, bool needs_mount,
         ui->Print("Retry attempt: %d\n", retry_count);
     }
     ui->SetEnableReboot(false);
-    int result = try_update_binary(path, &zip, wipe_cache, log_buffer, retry_count);
+    ret = try_update_binary(path, &zip, wipe_cache, log_buffer, retry_count);
     ui->SetEnableReboot(true);
     ui->Print("\n");
 
@@ -447,7 +455,7 @@ really_install_package(const char *path, bool* wipe_cache, bool needs_mount,
 
 #ifdef USE_MDTP
     /* If MDTP update failed, return an error such that recovery will not finish. */
-    if (result == INSTALL_SUCCESS) {
+    if (ret == INSTALL_SUCCESS) {
         if (!mdtp_update()) {
             ui->Print("Unable to verify integrity of /system for MDTP, update aborted.\n");
             return INSTALL_ERROR;
@@ -456,7 +464,9 @@ really_install_package(const char *path, bool* wipe_cache, bool needs_mount,
     }
 #endif /* USE_MDTP */
 
-    return result;
+out:
+    set_perf_mode(false);
+    return ret;
 }
 
 int
@@ -498,3 +508,8 @@ install_package(const char* path, bool* wipe_cache, const char* install_file,
     }
     return result;
 }
+
+void
+set_perf_mode(bool enable) {
+    property_set("recovery.perf.mode", enable ? "1" : "0");
+}
diff --git a/install.h b/install.h
index 66764f542..12cadcd51 100644
--- a/install.h
+++ b/install.h
@@ -31,6 +31,8 @@ enum { INSTALL_SUCCESS, INSTALL_ERROR, INSTALL_CORRUPT, INSTALL_NONE, INSTALL_SK
 int install_package(const char* root_path, bool* wipe_cache, const char* install_file,
                     bool needs_mount, int retry_count);
 
+void set_perf_mode(bool enable);
+
 #ifdef __cplusplus
 }
 #endif