diff --git a/battery_mitigation/battery_mitigation.cpp b/battery_mitigation/battery_mitigation.cpp index 230f297..2443226 100644 --- a/battery_mitigation/battery_mitigation.cpp +++ b/battery_mitigation/battery_mitigation.cpp @@ -19,6 +19,8 @@ #include #include +#define COUNT_LIMIT 10 + using android::hardware::google::pixel::BatteryMitigation; using android::hardware::google::pixel::MitigationConfig; @@ -94,7 +96,7 @@ int main(int /*argc*/, char ** /*argv*/) { bool isBatteryMitigationReady = false; std::string ready_str; int val = 0; - while (!isBatteryMitigationReady) { + for (int i = 0; i < COUNT_LIMIT; i++) { if (!android::base::ReadFileToString(kReadyFilePath, &ready_str)) { continue; } @@ -104,9 +106,12 @@ int main(int /*argc*/, char ** /*argv*/) { } if (val == 1) { isBatteryMitigationReady = true; + break; } } - android::base::SetProperty(kReadyProperty, "1"); + if (isBatteryMitigationReady) { + android::base::SetProperty(kReadyProperty, "1"); + } while (true) { pause(); }