1 //===-- LLDBUtils.cpp -------------------------------------------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 
9 #include "LLDBUtils.h"
10 #include "VSCode.h"
11 
12 namespace lldb_vscode {
13 
RunLLDBCommands(llvm::StringRef prefix,const llvm::ArrayRef<std::string> & commands,llvm::raw_ostream & strm)14 void RunLLDBCommands(llvm::StringRef prefix,
15                      const llvm::ArrayRef<std::string> &commands,
16                      llvm::raw_ostream &strm) {
17   if (commands.empty())
18     return;
19   lldb::SBCommandInterpreter interp = g_vsc.debugger.GetCommandInterpreter();
20   if (!prefix.empty())
21     strm << prefix << "\n";
22   for (const auto &command : commands) {
23     lldb::SBCommandReturnObject result;
24     strm << "(lldb) " << command << "\n";
25     interp.HandleCommand(command.c_str(), result);
26     auto output_len = result.GetOutputSize();
27     if (output_len) {
28       const char *output = result.GetOutput();
29       strm << output;
30     }
31     auto error_len = result.GetErrorSize();
32     if (error_len) {
33       const char *error = result.GetError();
34       strm << error;
35     }
36   }
37 }
38 
RunLLDBCommands(llvm::StringRef prefix,const llvm::ArrayRef<std::string> & commands)39 std::string RunLLDBCommands(llvm::StringRef prefix,
40                             const llvm::ArrayRef<std::string> &commands) {
41   std::string s;
42   llvm::raw_string_ostream strm(s);
43   RunLLDBCommands(prefix, commands, strm);
44   strm.flush();
45   return s;
46 }
47 
ThreadHasStopReason(lldb::SBThread & thread)48 bool ThreadHasStopReason(lldb::SBThread &thread) {
49   switch (thread.GetStopReason()) {
50   case lldb::eStopReasonTrace:
51   case lldb::eStopReasonPlanComplete:
52   case lldb::eStopReasonBreakpoint:
53   case lldb::eStopReasonWatchpoint:
54   case lldb::eStopReasonInstrumentation:
55   case lldb::eStopReasonSignal:
56   case lldb::eStopReasonException:
57   case lldb::eStopReasonExec:
58     return true;
59   case lldb::eStopReasonThreadExiting:
60   case lldb::eStopReasonInvalid:
61   case lldb::eStopReasonNone:
62     break;
63   }
64   return false;
65 }
66 
67 static uint32_t constexpr THREAD_INDEX_SHIFT = 19;
68 
GetLLDBThreadIndexID(uint64_t dap_frame_id)69 uint32_t GetLLDBThreadIndexID(uint64_t dap_frame_id) {
70   return dap_frame_id >> THREAD_INDEX_SHIFT;
71 }
72 
GetLLDBFrameID(uint64_t dap_frame_id)73 uint32_t GetLLDBFrameID(uint64_t dap_frame_id) {
74   return dap_frame_id & ((1u << THREAD_INDEX_SHIFT) - 1);
75 }
76 
MakeVSCodeFrameID(lldb::SBFrame & frame)77 int64_t MakeVSCodeFrameID(lldb::SBFrame &frame) {
78   return (int64_t)(frame.GetThread().GetIndexID() << THREAD_INDEX_SHIFT |
79                    frame.GetFrameID());
80 }
81 
82 static uint32_t constexpr BREAKPOINT_ID_SHIFT = 22;
83 
GetLLDBBreakpointID(uint64_t dap_breakpoint_id)84 uint32_t GetLLDBBreakpointID(uint64_t dap_breakpoint_id) {
85   return dap_breakpoint_id >> BREAKPOINT_ID_SHIFT;
86 }
87 
GetLLDBBreakpointLocationID(uint64_t dap_breakpoint_id)88 uint32_t GetLLDBBreakpointLocationID(uint64_t dap_breakpoint_id) {
89   return dap_breakpoint_id & ((1u << BREAKPOINT_ID_SHIFT) - 1);
90 }
91 
MakeVSCodeBreakpointID(lldb::SBBreakpointLocation & bp_loc)92 int64_t MakeVSCodeBreakpointID(lldb::SBBreakpointLocation &bp_loc) {
93   return (int64_t)(bp_loc.GetBreakpoint().GetID() << BREAKPOINT_ID_SHIFT |
94                    bp_loc.GetID());
95 }
96 
97 } // namespace lldb_vscode
98