Add 2044-get-stacktraces

Stress test getStackTrace() from multiple threads to make sure it
doesn't deadlock.

Bug: 234542166
Test: Treehugger
Change-Id: Ia96fa7b4ad29ab0464759cb80db4d2c9721e2eae

6 files changed, 150 insertions, 0 deletions

diff --git a/TEST_MAPPING b/TEST_MAPPING
index 91a19d2d08..002d6419db 100644
--- a/TEST_MAPPING
+++ b/TEST_MAPPING
@@ -407,6 +407,9 @@
       "name": "art-run-test-2043-reference-pauses[com.google.android.art.apex]"
     },
     {
+      "name": "art-run-test-2044-get-stack-traces[com.google.android.art.apex]"
+    },
+    {
       "name": "art-run-test-2231-checker-heap-poisoning[com.google.android.art.apex]"
     },
     {
@@ -1711,6 +1714,9 @@
       "name": "art-run-test-2043-reference-pauses"
     },
     {
+      "name": "art-run-test-2044-get-stack-traces"
+    },
+    {
       "name": "art-run-test-2231-checker-heap-poisoning"
     },
     {
@@ -3006,6 +3012,15 @@
       "name": "art-run-test-2042-checker-dce-always-throw"
     },
     {
+      "name": "art-run-test-2042-reference-processing"
+    },
+    {
+      "name": "art-run-test-2043-reference-pauses"
+    },
+    {
+      "name": "art-run-test-2044-get-stack-traces"
+    },
+    {
       "name": "art-run-test-2231-checker-heap-poisoning"
     },
     {
diff --git a/test/2044-get-stack-traces/Android.bp b/test/2044-get-stack-traces/Android.bp
new file mode 100644
index 0000000000..79aea7c7ff
--- /dev/null
+++ b/test/2044-get-stack-traces/Android.bp
@@ -0,0 +1,40 @@
+// Generated by `regen-test-files`. Do not edit manually.
+
+// Build rules for ART run-test `2044-get-stack-traces`.
+
+package {
+    // See: http://go/android-license-faq
+    // A large-scale-change added 'default_applicable_licenses' to import
+    // all of the 'license_kinds' from "art_license"
+    // to get the below license kinds:
+    //   SPDX-license-identifier-Apache-2.0
+    default_applicable_licenses: ["art_license"],
+}
+
+// Test's Dex code.
+java_test {
+    name: "art-run-test-2044-get-stack-traces",
+    defaults: ["art-run-test-defaults"],
+    test_config_template: ":art-run-test-target-template",
+    srcs: ["src/**/*.java"],
+    data: [
+        ":art-run-test-2044-get-stack-traces-expected-stdout",
+        ":art-run-test-2044-get-stack-traces-expected-stderr",
+    ],
+}
+
+// Test's expected standard output.
+genrule {
+    name: "art-run-test-2044-get-stack-traces-expected-stdout",
+    out: ["art-run-test-2044-get-stack-traces-expected-stdout.txt"],
+    srcs: ["expected-stdout.txt"],
+    cmd: "cp -f $(in) $(out)",
+}
+
+// Test's expected standard error.
+genrule {
+    name: "art-run-test-2044-get-stack-traces-expected-stderr",
+    out: ["art-run-test-2044-get-stack-traces-expected-stderr.txt"],
+    srcs: ["expected-stderr.txt"],
+    cmd: "cp -f $(in) $(out)",
+}
diff --git a/test/2044-get-stack-traces/expected-stderr.txt b/test/2044-get-stack-traces/expected-stderr.txt
new file mode 100644
index 0000000000..e69de29bb2
--- /dev/null
+++ b/test/2044-get-stack-traces/expected-stderr.txt
diff --git a/test/2044-get-stack-traces/expected-stdout.txt b/test/2044-get-stack-traces/expected-stdout.txt
new file mode 100644
index 0000000000..d396083041
--- /dev/null
+++ b/test/2044-get-stack-traces/expected-stdout.txt
@@ -0,0 +1,8 @@
+Starting
+Starting helper
+Starting helper
+Starting helper
+Starting helper
+Starting helper
+Finished worker stack traces
+Finished
diff --git a/test/2044-get-stack-traces/info.txt b/test/2044-get-stack-traces/info.txt
new file mode 100644
index 0000000000..f9cac7aba9
--- /dev/null
+++ b/test/2044-get-stack-traces/info.txt
@@ -0,0 +1,4 @@
+Tests multiple simultaneous calls to Thread.getStackTrace()
+
+This is a stress test for code under suspicion in the context of b/234542166
+and others.
diff --git a/test/2044-get-stack-traces/src/Main.java b/test/2044-get-stack-traces/src/Main.java
new file mode 100644
index 0000000000..d17e7c18f4
--- /dev/null
+++ b/test/2044-get-stack-traces/src/Main.java
@@ -0,0 +1,83 @@
+/*
+ * Copyright (C) 2022 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.
+ */
+
+import java.lang.ref.Reference;
+import java.lang.ref.WeakReference;
+import java.lang.ref.SoftReference;
+import java.math.BigInteger;
+import java.util.ArrayList;
+
+/**
+ * We construct a main thread and worker threads, each retrieving stack traces
+ * from the other. Since there are multiple workers, we may get a large number
+ * of simultaneous stack trace attempts.
+ */
+public class Main {
+    static final int NUM_THREADS = 5;
+    static Thread mainThread;
+    static volatile boolean pleaseStop = false;
+
+    private static void getTrace(Thread t) {
+      StackTraceElement trace[] = t.getStackTrace();
+      if (trace.length < 1 || trace.length > 20) {
+        System.out.println("Stack trace for " + t.getName() + " has size " + trace.length);
+        for (StackTraceElement e : trace) {
+          System.out.println(e.toString());
+        }
+      }
+    }
+
+    /**
+     * Repeatedly get and minimally check stack trace of main thread.
+     */
+    static Runnable traceGetter = new Runnable() {
+        public void run() {
+          System.out.println("Starting helper");
+          while (!pleaseStop) {
+            getTrace(mainThread);
+          }
+        }
+    };
+
+    public static void main(String[] args) throws Exception {
+        System.out.println("Starting");
+        Thread[] t = new Thread[NUM_THREADS];
+        mainThread = Thread.currentThread();
+        for (int i = 0; i < NUM_THREADS; ++i) {
+          t[i] = new Thread(traceGetter);
+          t[i].start();
+        }
+        try {
+          Thread.sleep(1000);
+        } catch (InterruptedException e) {
+            System.out.println("Unexpectedly interrupted");
+        }
+        for (int i = 0; i < NUM_THREADS; ++i) {
+          getTrace(t[i]);
+        }
+        System.out.println("Finished worker stack traces");
+        long now = System.currentTimeMillis();
+        while (System.currentTimeMillis() - now < 2000) {
+          try {
+            Thread.sleep(1);
+          } catch (InterruptedException e) {
+            System.out.println("Unexpectedly interrupted");
+          }
+        }
+        pleaseStop = true;
+        System.out.println("Finished");
+    }
+}