summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--twrp-functions.cpp6
1 files changed, 5 insertions, 1 deletions
diff --git a/twrp-functions.cpp b/twrp-functions.cpp
index 3e06f8770..d1f9e3dbd 100644
--- a/twrp-functions.cpp
+++ b/twrp-functions.cpp
@@ -12,7 +12,9 @@
#include <sys/sendfile.h>
#include <sys/stat.h>
#include <sys/vfs.h>
-#include "cutils/android_reboot.h"
+#ifdef ANDROID_RB_POWEROFF
+ #include "cutils/android_reboot.h"
+#endif
#include <iostream>
#include <fstream>
#include "twrp-functions.hpp"
@@ -375,7 +377,9 @@ int TWFunc::tw_reboot(RebootCommand command)
return __reboot(LINUX_REBOOT_MAGIC1, LINUX_REBOOT_MAGIC2, LINUX_REBOOT_CMD_RESTART2, (void*) "bootloader");
case rb_poweroff:
check_and_run_script("/sbin/poweroff.sh", "power off");
+#ifdef ANDROID_RB_POWEROFF
android_reboot(ANDROID_RB_POWEROFF, 0, 0);
+#endif
return reboot(RB_POWER_OFF);
case rb_download:
check_and_run_script("/sbin/rebootdownload.sh", "reboot download");