From 032365882b21ab8d621cbdc19779458592bf3d53 Mon Sep 17 00:00:00 2001
From: Tri Vo <trong@google.com>
Date: Sat, 9 Nov 2019 16:26:04 -0800
Subject: [PATCH] block_suspend: test util to block suspend from shell

Bug: 143284916
Test: ./block_suspend; dumpsys suspend_control shows that wakelocks are
created as expected
Change-Id: Ifacb4f476632e0c36f7ec4f59164f91e89faf1f1
---
 Android.bp        | 11 +++++++++++
 block_suspend.cpp | 39 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 50 insertions(+)
 create mode 100644 block_suspend.cpp

diff --git a/Android.bp b/Android.bp
index bc453fe..d196ee6 100644
--- a/Android.bp
+++ b/Android.bp
@@ -74,3 +74,14 @@ cc_library_shared {
         "uevent.c",
     ],
 }
+
+cc_test {
+    name: "block_suspend",
+    defaults: ["libpower_defaults"],
+    srcs: ["block_suspend.cpp"],
+    static_libs: ["libpower"],
+    shared_libs: [
+        "android.system.suspend@1.0",
+    ],
+    gtest: false,
+}
diff --git a/block_suspend.cpp b/block_suspend.cpp
new file mode 100644
index 0000000..99341b8
--- /dev/null
+++ b/block_suspend.cpp
@@ -0,0 +1,39 @@
+/*
+ * Copyright 2019 The Android Open Source Project
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <iostream>
+
+#include <wakelock/wakelock.h>
+
+static constexpr const char *gWakeLockName = "block_suspend";
+
+static void usage() {
+    std::cout << "Usage: block_suspend\n"
+              << "Prevent device from suspending indefinitely. "
+              << "Process must be killed to unblock suspend.\n";
+}
+
+int main(int argc, char ** /* argv */) {
+    if (argc > 1) {
+        usage();
+        return 0;
+    }
+
+    android::wakelock::WakeLock wl{gWakeLockName};  // RAII object
+    while (true) {};
+    std::abort();  // should never reach here
+    return 0;
+}