1 files changed, 67 insertions, 0 deletions
diff --git a/contrib/llvm-project/lldb/source/Plugins/Trace/intel-pt/ThreadDecoder.cpp b/contrib/llvm-project/lldb/source/Plugins/Trace/intel-pt/ThreadDecoder.cpp
new file mode 100644
index 000000000000..75d74edd44a1
--- /dev/null
+++ b/contrib/llvm-project/lldb/source/Plugins/Trace/intel-pt/ThreadDecoder.cpp
@@ -0,0 +1,67 @@
+//===-- ThreadDecoder.cpp --======-----------------------------------------===//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "ThreadDecoder.h"
+#include "../common/ThreadPostMortemTrace.h"
+#include "LibiptDecoder.h"
+#include "TraceIntelPT.h"
+#include "llvm/Support/MemoryBuffer.h"
+#include <optional>
+#include <utility>
+
+using namespace lldb;
+using namespace lldb_private;
+using namespace lldb_private::trace_intel_pt;
+using namespace llvm;
+
+ThreadDecoder::ThreadDecoder(const ThreadSP &thread_sp, TraceIntelPT &trace)
+    : m_thread_sp(thread_sp), m_trace(trace) {}
+
+Expected<std::optional<uint64_t>> ThreadDecoder::FindLowestTSC() {
+  std::optional<uint64_t> lowest_tsc;
+  Error err = m_trace.OnThreadBufferRead(
+      m_thread_sp->GetID(), [&](llvm::ArrayRef<uint8_t> data) -> llvm::Error {
+        Expected<std::optional<uint64_t>> tsc =
+            FindLowestTSCInTrace(m_trace, data);
+        if (!tsc)
+          return tsc.takeError();
+        lowest_tsc = *tsc;
+        return Error::success();
+      });
+  if (err)
+    return std::move(err);
+  return lowest_tsc;
+}
+
+Expected<DecodedThreadSP> ThreadDecoder::Decode() {
+  if (!m_decoded_thread.has_value()) {
+    if (Expected<DecodedThreadSP> decoded_thread = DoDecode()) {
+      m_decoded_thread = *decoded_thread;
+    } else {
+      return decoded_thread.takeError();
+    }
+  }
+  return *m_decoded_thread;
+}
+
+llvm::Expected<DecodedThreadSP> ThreadDecoder::DoDecode() {
+  return m_trace.GetThreadTimer(m_thread_sp->GetID())
+      .TimeTask("Decoding instructions", [&]() -> Expected<DecodedThreadSP> {
+        DecodedThreadSP decoded_thread_sp = std::make_shared<DecodedThread>(
+            m_thread_sp, m_trace.GetPerfZeroTscConversion());
+
+        Error err = m_trace.OnThreadBufferRead(
+            m_thread_sp->GetID(), [&](llvm::ArrayRef<uint8_t> data) {
+              return DecodeSingleTraceForThread(*decoded_thread_sp, m_trace,
+                                                data);
+            });
+
+        if (err)
+          return std::move(err);
+        return decoded_thread_sp;
+      });
+}