commit 07dbdf4b744b8921acd885db7c8bbda5855a9a3d
author Andres Celis <celisa@google.com> Mon Sep 11 17:31:01 2017 -0700
committer Andres Celis <celisa@google.com> Mon Sep 18 11:30:23 2017 -0700
tree a0e5a05b50c4553ddbd509d39f4279ec5a40b3fa
parent 336c7eb32b8cbe1c8e4c7f65f00f3324c1122e88

Add Malloc Debug feature for CTS/STS tests.

Bug:65559061
Change-Id: I0e632404b3042c25f66dfab8b64849d17ef6e1ab

hostsidetests/security/src/android/security/cts/AdbUtils.java

diff --git a/hostsidetests/security/src/android/security/cts/AdbUtils.java b/hostsidetests/security/src/android/security/cts/AdbUtils.java
index f44870a..77a8f7b 100644
--- a/hostsidetests/security/src/android/security/cts/AdbUtils.java
+++ b/hostsidetests/security/src/android/security/cts/AdbUtils.java
@@ -21,6 +21,7 @@
 import com.android.tradefed.device.DeviceNotAvailableException;
 import com.android.tradefed.device.ITestDevice;
 import com.android.tradefed.testtype.DeviceTestCase;
+import com.android.tradefed.log.LogUtil.CLog;
 
 import android.platform.test.annotations.RootPermissionTest;
 
@@ -89,6 +90,33 @@
     }
 
     /**
+     * Enables malloc debug on a given process.
+     *
+     * @param processName the name of the process to run with libc malloc debug
+     * @param device the device to use
+     * @return true if enabling malloc debug succeeded
+     */
+    public static boolean enableLibcMallocDebug(String processName, ITestDevice device) throws Exception {
+        device.executeShellCommand("setprop libc.debug.malloc.program " + processName);
+        device.executeShellCommand("setprop libc.debug.malloc.options \"backtrace guard\"");
+        String cmdOut = device.executeShellCommand("ps | fgrep " + processName);
+        Scanner s = new Scanner(cmdOut);
+        if(!s.hasNextInt()) {
+            CLog.w("Could not find pid for process: " + processName);
+            return false;
+        }
+
+        String result = device.executeShellCommand("kill -9 " + s.nextInt());
+        if(!result.equals("")) {
+            CLog.w("Could not restart process: " + processName);
+            return false;
+        }
+
+        TimeUnit.SECONDS.sleep(1);
+        return true;
+    }
+
+    /**
      * Pushes and installs an apk to the selected device
      *
      * @param pathToApk a string path to apk from the /res folder