diff --git a/init/reboot.cpp b/init/reboot.cpp index df7912f96..be1e7607f 100644 --- a/init/reboot.cpp +++ b/init/reboot.cpp @@ -21,6 +21,7 @@ #include #include #include +#include #include #include #include @@ -39,6 +40,7 @@ #include #include #include +#include #include #include #include @@ -162,9 +164,30 @@ static void LogShutdownTime(UmountStat stat, Timer* t) { LOG(WARNING) << "powerctl_shutdown_time_ms:" << std::to_string(t->duration_ms()) << ":" << stat; } +static bool IsRebootCapable() { + cap_t caps = cap_init(); + auto scopeguard = android::base::make_scope_guard([&caps] { cap_free(caps); }); + + // Assume we have the capability. + cap_flag_value_t value = CAP_SET; + if (!CAP_IS_SUPPORTED(CAP_SYS_BOOT) || + cap_get_flag(caps, CAP_SYS_BOOT, CAP_EFFECTIVE, &value) != 0) { + PLOG(ERROR) << "cap_get_flag(CAP_BOOT, EFFECTIVE) failed"; + } + + return value == CAP_SET; +} + static void __attribute__((noreturn)) RebootSystem(unsigned int cmd, const std::string& rebootTarget) { LOG(INFO) << "Reboot ending, jumping to kernel"; + + if (!IsRebootCapable()) { + // On systems where init does not have the capability of rebooting the + // device, just exit cleanly. + exit(0); + } + switch (cmd) { case ANDROID_RB_POWEROFF: reboot(RB_POWER_OFF);