1 files changed, 46 insertions, 3 deletions

diff --git a/simpleperf/test_util.cpp b/simpleperf/test_util.cpp
index 60d8ea35..7e99b5d5 100644
--- a/simpleperf/test_util.cpp
+++ b/simpleperf/test_util.cpp
@@ -17,15 +17,46 @@
 
 #include <stdio.h>
 
+#include <optional>
+
 #include <android-base/logging.h>
 #include <android-base/properties.h>
 
+#include "command.h"
 #include "event_attr.h"
 #include "event_fd.h"
 #include "event_type.h"
+#include "record_file.h"
 #include "test_util.h"
 
-bool IsInNativeAbi() {
+#if defined(__linux__)
+
+static std::optional<bool> CanSampleRegsFor32BitABI() {
+  std::vector<std::unique_ptr<Workload>> workloads;
+  CreateProcesses(1, &workloads);
+  std::string pid = std::to_string(workloads[0]->GetPid());
+  std::unique_ptr<Command> cmd = CreateCommandInstance("record");
+  TemporaryFile tmpfile;
+  if (!cmd->Run({"-p", pid, "--call-graph", "dwarf,8", "--no-unwind", "-e", "cpu-clock:u",
+                 "--duration", "3", "-o", tmpfile.path})) {
+    return std::nullopt;
+  }
+  auto reader = RecordFileReader::CreateInstance(tmpfile.path);
+  if (!reader) {
+    return std::nullopt;
+  }
+  for (const std::unique_ptr<Record>& record : reader->DataSection()) {
+    if (record->type() == PERF_RECORD_SAMPLE) {
+      auto sample = static_cast<const SampleRecord*>(record.get());
+      if (sample->regs_user_data.abi == PERF_SAMPLE_REGS_ABI_32) {
+        return true;
+      }
+    }
+  }
+  return false;
+}
+
+std::optional<bool> IsInNativeAbi() {
   static int in_native_abi = -1;
   if (in_native_abi == -1) {
     FILE* fp = popen("uname -m", "re");
@@ -43,16 +74,27 @@ bool IsInNativeAbi() {
       if (s.find("arm") == std::string::npos && s.find("aarch64") == std::string::npos) {
         in_native_abi = 0;
       }
+      if (GetTargetArch() == ARCH_ARM) {
+        // If we can't get ARM registers in samples, probably we are running with a 32-bit
+        // translator on 64-bit only CPUs. Then we should make in_native_abi = 0.
+        if (auto result = CanSampleRegsFor32BitABI(); result.has_value()) {
+          in_native_abi = result.value() ? 1 : 0;
+        } else {
+          in_native_abi = 2;
+        }
+      }
     } else if (GetTargetArch() == ARCH_RISCV64) {
       if (s.find("riscv") == std::string::npos) {
         in_native_abi = 0;
       }
     }
   }
+  if (in_native_abi == 2) {
+    return std::nullopt;
+  }
   return in_native_abi == 1;
 }
 
-#if defined(__linux__)
 // Check if we can get a non-zero instruction event count by monitoring current thread.
 static bool HasNonZeroInstructionEventCount() {
   const simpleperf::EventType* type = simpleperf::FindEventTypeByName("instructions", false);
@@ -84,8 +126,9 @@ bool HasHardwareCounter() {
     bool is_emulator = android::base::StartsWith(fingerprint, "google/sdk_gphone") ||
                        android::base::StartsWith(fingerprint, "google/sdk_gpc") ||
                        android::base::StartsWith(fingerprint, "generic/cf");
+    bool in_native_abi = IsInNativeAbi() == std::optional(true);
 
-    if (arch == ARCH_X86_64 || arch == ARCH_X86_32 || !IsInNativeAbi() || is_emulator) {
+    if (arch == ARCH_X86_64 || arch == ARCH_X86_32 || !in_native_abi || is_emulator) {
       // On x86 and x86_64, or when we are not in native abi, it's likely to run on an emulator or
       // vm without hardware perf counters. It's hard to enumerate them all. So check the support
       // at runtime.