tequilaOS/platform_bootable_recovery
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge "Add --keep_memtag_mode for --wipe_data" into main

Browse source
This commit is contained in:
Treehugger Robot 2023-10-31 10:13:30 +00:00 committed by Gerrit Code Review
commit 0d4e8c333c
3 changed files with 16 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
install/include/install/wipe_data.h
View file

@ -27,4 +27,4 @@ struct selabel_handle;
bool WipeCache(RecoveryUI* ui, const std::function<bool()>& confirm);
// Returns true on success.
bool WipeData(Device* device);
bool WipeData(Device* device, bool keep_memtag_mode = false);

16
install/wipe_data.cpp
View file

@ -79,7 +79,7 @@ bool WipeCache(RecoveryUI* ui, const std::function<bool()>& confirm_func) {
return success;
}
bool WipeData(Device* device) {
bool WipeData(Device* device, bool keep_memtag_mode) {
RecoveryUI* ui = device->GetUI();
ui->Print("\n-- Wiping data...\n");
ui->SetBackground(RecoveryUI::ERASING);
@ -101,11 +101,15 @@ bool WipeData(Device* device) {
success &= EraseVolume(METADATA_ROOT, ui);
}
}
ui->Print("Resetting memtag message...\n");
std::string err;
if (!WriteMiscMemtagMessage({}, &err)) {
ui->Print("Failed to reset memtag message: %s\n", err.c_str());
success = false;
if (keep_memtag_mode) {
ui->Print("NOT resetting memtag message as per request...\n");
} else {
ui->Print("Resetting memtag message...\n");
std::string err;
if (!WriteMiscMemtagMessage({}, &err)) {
ui->Print("Failed to reset memtag message: %s\n", err.c_str());
success = false;
}
}
if (success) {
success &= device->PostWipeData();

6
recovery.cpp
View file

@ -608,6 +608,7 @@ Device::BuiltinAction start_recovery(Device* device, const std::vector<std::stri
{ "wipe_ab", no_argument, nullptr, 0 },
{ "wipe_cache", no_argument, nullptr, 0 },
{ "wipe_data", no_argument, nullptr, 0 },
{ "keep_memtag_mode", no_argument, nullptr, 0 },
{ "wipe_package_size", required_argument, nullptr, 0 },
{ nullptr, 0, nullptr, 0 },
};
@ -616,6 +617,7 @@ Device::BuiltinAction start_recovery(Device* device, const std::vector<std::stri
bool install_with_fuse = false; // memory map the update package by default.
bool should_wipe_data = false;
bool should_prompt_and_wipe_data = false;
bool should_keep_memtag_mode = false;
bool should_wipe_cache = false;
bool should_wipe_ab = false;
size_t wipe_package_size = 0;
@ -675,6 +677,8 @@ Device::BuiltinAction start_recovery(Device* device, const std::vector<std::stri
should_wipe_data = true;
} else if (option == "wipe_package_size") {
android::base::ParseUint(optarg, &wipe_package_size);
} else if (option == "keep_memtag_mode") {
should_keep_memtag_mode = true;
}
break;
}
@ -793,7 +797,7 @@ Device::BuiltinAction start_recovery(Device* device, const std::vector<std::stri
} else if (should_wipe_data) {
save_current_log = true;
CHECK(device->GetReason().has_value());
if (!WipeData(device)) {
if (!WipeData(device, should_keep_memtag_mode)) {
status = INSTALL_ERROR;
}
} else if (should_prompt_and_wipe_data) {