From 0b8a656ba110884e40e9ec79c936139eb6fce0b6 Mon Sep 17 00:00:00 2001 From: Ebuka Ezike Date: Thu, 26 Jun 2025 18:22:47 +0100 Subject: [PATCH] [lldb-dap] Fix source references (#144364) The [protocol](https://microsoft.github.io/debug-adapter-protocol//specification.html#Types_Source) expects that `sourceReference` be less than `(2^31)-1`, but we currently represent memory address as source reference, this can be truncated either when sending through json or by the client. Instead, generate new source references based on the memory address. Make the `ResolveSource` function return an optional source. --- .../TestDAP_breakpointAssembly.py | 14 ++-- lldb/tools/lldb-dap/Breakpoint.cpp | 4 +- lldb/tools/lldb-dap/DAP.cpp | 70 +++++++++++++++++ lldb/tools/lldb-dap/DAP.h | 31 +++++++- .../Handler/DisassembleRequestHandler.cpp | 13 ++-- .../Handler/LocationsRequestHandler.cpp | 22 +++++- .../lldb-dap/Handler/SourceRequestHandler.cpp | 40 ++++++---- .../Handler/StackTraceRequestHandler.cpp | 2 +- lldb/tools/lldb-dap/JSONUtils.cpp | 11 ++- lldb/tools/lldb-dap/JSONUtils.h | 5 +- .../lldb-dap/Protocol/ProtocolRequests.h | 2 +- .../tools/lldb-dap/Protocol/ProtocolTypes.cpp | 2 +- lldb/tools/lldb-dap/Protocol/ProtocolTypes.h | 3 +- lldb/tools/lldb-dap/ProtocolUtils.cpp | 75 +++++-------------- lldb/tools/lldb-dap/ProtocolUtils.h | 19 +---- lldb/tools/lldb-dap/SourceBreakpoint.cpp | 8 +- 16 files changed, 206 insertions(+), 115 deletions(-) diff --git a/lldb/test/API/tools/lldb-dap/breakpoint-assembly/TestDAP_breakpointAssembly.py b/lldb/test/API/tools/lldb-dap/breakpoint-assembly/TestDAP_breakpointAssembly.py index 8bccc2bcf415..674bfe4199b4 100644 --- a/lldb/test/API/tools/lldb-dap/breakpoint-assembly/TestDAP_breakpointAssembly.py +++ b/lldb/test/API/tools/lldb-dap/breakpoint-assembly/TestDAP_breakpointAssembly.py @@ -67,19 +67,19 @@ class TestDAP_setBreakpointsAssembly(lldbdap_testcase.DAPTestCaseBase): "Invalid sourceReference.", ) - # Verify that setting a breakpoint on a source reference without a symbol also fails + # Verify that setting a breakpoint on a source reference that is not created fails response = self.dap_server.request_setBreakpoints( - Source(source_reference=0), [1] + Source(source_reference=200), [1] ) self.assertIsNotNone(response) breakpoints = response["body"]["breakpoints"] self.assertEqual(len(breakpoints), 1) - breakpoint = breakpoints[0] + break_point = breakpoints[0] self.assertFalse( - breakpoint["verified"], "Expected breakpoint to not be verified" + break_point["verified"], "Expected breakpoint to not be verified" ) - self.assertIn("message", breakpoint, "Expected message to be present") + self.assertIn("message", break_point, "Expected message to be present") self.assertEqual( - breakpoint["message"], - "Breakpoints in assembly without a valid symbol are not supported yet.", + break_point["message"], + "Invalid sourceReference.", ) diff --git a/lldb/tools/lldb-dap/Breakpoint.cpp b/lldb/tools/lldb-dap/Breakpoint.cpp index ef5646c4c3d1..b4e593eb83d2 100644 --- a/lldb/tools/lldb-dap/Breakpoint.cpp +++ b/lldb/tools/lldb-dap/Breakpoint.cpp @@ -64,8 +64,8 @@ protocol::Breakpoint Breakpoint::ToProtocolBreakpoint() { "0x" + llvm::utohexstr(bp_addr.GetLoadAddress(m_bp.GetTarget())); breakpoint.instructionReference = formatted_addr; - auto source = CreateSource(bp_addr, m_dap.target); - if (!IsAssemblySource(source)) { + std::optional source = m_dap.ResolveSource(bp_addr); + if (source && !IsAssemblySource(*source)) { auto line_entry = bp_addr.GetLineEntry(); const auto line = line_entry.GetLine(); if (line != LLDB_INVALID_LINE_NUMBER) diff --git a/lldb/tools/lldb-dap/DAP.cpp b/lldb/tools/lldb-dap/DAP.cpp index c171b55951cb..cd97458bd4aa 100644 --- a/lldb/tools/lldb-dap/DAP.cpp +++ b/lldb/tools/lldb-dap/DAP.cpp @@ -497,6 +497,27 @@ DAP::SendFormattedOutput(OutputType o, const char *format, ...) { o, llvm::StringRef(buffer, std::min(actual_length, sizeof(buffer)))); } +int32_t DAP::CreateSourceReference(lldb::addr_t address) { + std::lock_guard guard(m_source_references_mutex); + auto iter = llvm::find(m_source_references, address); + if (iter != m_source_references.end()) + return std::distance(m_source_references.begin(), iter) + 1; + + m_source_references.emplace_back(address); + return static_cast(m_source_references.size()); +} + +std::optional DAP::GetSourceReferenceAddress(int32_t reference) { + std::lock_guard guard(m_source_references_mutex); + if (reference <= LLDB_DAP_INVALID_SRC_REF) + return std::nullopt; + + if (static_cast(reference) > m_source_references.size()) + return std::nullopt; + + return m_source_references[reference - 1]; +} + ExceptionBreakpoint *DAP::GetExceptionBPFromStopReason(lldb::SBThread &thread) { const auto num = thread.GetStopReasonDataCount(); // Check to see if have hit an exception breakpoint and change the @@ -602,6 +623,55 @@ ReplMode DAP::DetectReplMode(lldb::SBFrame frame, std::string &expression, llvm_unreachable("enum cases exhausted."); } +std::optional DAP::ResolveSource(lldb::SBAddress address) { + if (DisplayAssemblySource(debugger, address)) + return ResolveAssemblySource(address); + + lldb::SBLineEntry line_entry = GetLineEntryForAddress(target, address); + if (!line_entry.IsValid()) + return std::nullopt; + + return CreateSource(line_entry.GetFileSpec()); +} + +std::optional +DAP::ResolveAssemblySource(lldb::SBAddress address) { + lldb::SBSymbol symbol = address.GetSymbol(); + lldb::addr_t load_addr = LLDB_INVALID_ADDRESS; + std::string name; + if (symbol.IsValid()) { + load_addr = symbol.GetStartAddress().GetLoadAddress(target); + name = symbol.GetName(); + } else { + load_addr = address.GetLoadAddress(target); + name = GetLoadAddressString(load_addr); + } + + if (load_addr == LLDB_INVALID_ADDRESS) + return std::nullopt; + + protocol::Source source; + source.sourceReference = CreateSourceReference(load_addr); + lldb::SBModule module = address.GetModule(); + if (module.IsValid()) { + lldb::SBFileSpec file_spec = module.GetFileSpec(); + if (file_spec.IsValid()) { + std::string path = GetSBFileSpecPath(file_spec); + if (!path.empty()) + source.path = path + '`' + name; + } + } + + source.name = std::move(name); + + // Mark the source as deemphasized since users will only be able to view + // assembly for these frames. + source.presentationHint = + protocol::Source::eSourcePresentationHintDeemphasize; + + return source; +} + bool DAP::RunLLDBCommands(llvm::StringRef prefix, llvm::ArrayRef commands) { bool required_command_failed = false; diff --git a/lldb/tools/lldb-dap/DAP.h b/lldb/tools/lldb-dap/DAP.h index 5ca5822f9bce..0e9a9e0eb674 100644 --- a/lldb/tools/lldb-dap/DAP.h +++ b/lldb/tools/lldb-dap/DAP.h @@ -219,7 +219,9 @@ struct DAP { void __attribute__((format(printf, 3, 4))) SendFormattedOutput(OutputType o, const char *format, ...); - static int64_t GetNextSourceReference(); + int32_t CreateSourceReference(lldb::addr_t address); + + std::optional GetSourceReferenceAddress(int32_t reference); ExceptionBreakpoint *GetExceptionBPFromStopReason(lldb::SBThread &thread); @@ -252,6 +254,29 @@ struct DAP { ReplMode DetectReplMode(lldb::SBFrame frame, std::string &expression, bool partial_expression); + /// Create a "Source" JSON object as described in the debug adapter + /// definition. + /// + /// \param[in] address + /// The address to use when populating out the "Source" object. + /// + /// \return + /// An optional "Source" JSON object that follows the formal JSON + /// definition outlined by Microsoft. + std::optional ResolveSource(lldb::SBAddress address); + + /// Create a "Source" JSON object as described in the debug adapter + /// definition. + /// + /// \param[in] address + /// The address to use when populating out the "Source" object. + /// + /// \return + /// An optional "Source" JSON object that follows the formal JSON + /// definition outlined by Microsoft. + std::optional + ResolveAssemblySource(lldb::SBAddress address); + /// \return /// \b false if a fatal error was found while executing these commands, /// according to the rules of \a LLDBUtils::RunLLDBCommands. @@ -406,6 +431,10 @@ private: std::thread progress_event_thread; /// @} + /// List of addresses mapped by sourceReference. + std::vector m_source_references; + std::mutex m_source_references_mutex; + /// Queue for all incoming messages. std::deque m_queue; std::mutex m_queue_mutex; diff --git a/lldb/tools/lldb-dap/Handler/DisassembleRequestHandler.cpp b/lldb/tools/lldb-dap/Handler/DisassembleRequestHandler.cpp index 85214b84b5c9..f66c87fa9893 100644 --- a/lldb/tools/lldb-dap/Handler/DisassembleRequestHandler.cpp +++ b/lldb/tools/lldb-dap/Handler/DisassembleRequestHandler.cpp @@ -85,7 +85,8 @@ static lldb::SBAddress GetDisassembleStartAddress(lldb::SBTarget target, } static DisassembledInstruction ConvertSBInstructionToDisassembledInstruction( - lldb::SBTarget &target, lldb::SBInstruction &inst, bool resolve_symbols) { + DAP &dap, lldb::SBInstruction &inst, bool resolve_symbols) { + lldb::SBTarget target = dap.target; if (!inst.IsValid()) return GetInvalidInstruction(); @@ -138,14 +139,14 @@ static DisassembledInstruction ConvertSBInstructionToDisassembledInstruction( si << " ; " << c; } - protocol::Source source = CreateSource(addr, target); + std::optional source = dap.ResolveSource(addr); lldb::SBLineEntry line_entry = GetLineEntryForAddress(target, addr); // If the line number is 0 then the entry represents a compiler generated // location. - if (!IsAssemblySource(source) && line_entry.GetStartAddress() == addr && - line_entry.IsValid() && line_entry.GetFileSpec().IsValid() && - line_entry.GetLine() != 0) { + if (source && !IsAssemblySource(*source) && + line_entry.GetStartAddress() == addr && line_entry.IsValid() && + line_entry.GetFileSpec().IsValid() && line_entry.GetLine() != 0) { disassembled_inst.location = std::move(source); const auto line = line_entry.GetLine(); @@ -221,7 +222,7 @@ DisassembleRequestHandler::Run(const DisassembleArguments &args) const { original_address_index = i; instructions.push_back(ConvertSBInstructionToDisassembledInstruction( - dap.target, inst, resolve_symbols)); + dap, inst, resolve_symbols)); } // Check if we miss instructions at the beginning. diff --git a/lldb/tools/lldb-dap/Handler/LocationsRequestHandler.cpp b/lldb/tools/lldb-dap/Handler/LocationsRequestHandler.cpp index 804bdcd622a2..cf9b5a3dbd06 100644 --- a/lldb/tools/lldb-dap/Handler/LocationsRequestHandler.cpp +++ b/lldb/tools/lldb-dap/Handler/LocationsRequestHandler.cpp @@ -137,7 +137,16 @@ void LocationsRequestHandler::operator()( return; } - body.try_emplace("source", CreateSource(line_entry.GetFileSpec())); + const std::optional source = + CreateSource(line_entry.GetFileSpec()); + if (!source) { + response["success"] = false; + response["message"] = "Failed to resolve file path for location"; + dap.SendJSON(llvm::json::Value(std::move(response))); + return; + } + + body.try_emplace("source", *source); if (int line = line_entry.GetLine()) body.try_emplace("line", line); if (int column = line_entry.GetColumn()) @@ -152,7 +161,16 @@ void LocationsRequestHandler::operator()( return; } - body.try_emplace("source", CreateSource(decl.GetFileSpec())); + const std::optional source = + CreateSource(decl.GetFileSpec()); + if (!source) { + response["success"] = false; + response["message"] = "Failed to resolve file path for location"; + dap.SendJSON(llvm::json::Value(std::move(response))); + return; + } + + body.try_emplace("source", *source); if (int line = decl.GetLine()) body.try_emplace("line", line); if (int column = decl.GetColumn()) diff --git a/lldb/tools/lldb-dap/Handler/SourceRequestHandler.cpp b/lldb/tools/lldb-dap/Handler/SourceRequestHandler.cpp index 353d3365564f..755ad206abe2 100644 --- a/lldb/tools/lldb-dap/Handler/SourceRequestHandler.cpp +++ b/lldb/tools/lldb-dap/Handler/SourceRequestHandler.cpp @@ -29,34 +29,42 @@ namespace lldb_dap { /// the source code for a given source reference. llvm::Expected SourceRequestHandler::Run(const protocol::SourceArguments &args) const { - const auto source = + + uint32_t source_ref = args.source->sourceReference.value_or(args.sourceReference); + const std::optional source_addr_opt = + dap.GetSourceReferenceAddress(source_ref); - if (!source) + if (!source_addr_opt) return llvm::make_error( - "invalid arguments, expected source.sourceReference to be set"); + llvm::formatv("Unknown source reference {}", source_ref)); - lldb::SBAddress address(source, dap.target); + lldb::SBAddress address(*source_addr_opt, dap.target); if (!address.IsValid()) return llvm::make_error("source not found"); lldb::SBSymbol symbol = address.GetSymbol(); + lldb::SBInstructionList insts; + + if (symbol.IsValid()) { + insts = symbol.GetInstructions(dap.target); + } else { + // No valid symbol, just return the disassembly. + insts = dap.target.ReadInstructions( + address, dap.k_number_of_assembly_lines_for_nodebug); + } + + if (!insts || insts.GetSize() == 0) + return llvm::make_error( + llvm::formatv("no instruction source for address {}", + address.GetLoadAddress(dap.target))); lldb::SBStream stream; lldb::SBExecutionContext exe_ctx(dap.target); - - if (symbol.IsValid()) { - lldb::SBInstructionList insts = symbol.GetInstructions(dap.target); - insts.GetDescription(stream, exe_ctx); - } else { - // No valid symbol, just return the disassembly. - lldb::SBInstructionList insts = dap.target.ReadInstructions( - address, dap.k_number_of_assembly_lines_for_nodebug); - insts.GetDescription(stream, exe_ctx); - } - + insts.GetDescription(stream, exe_ctx); return protocol::SourceResponseBody{/*content=*/stream.GetData(), - /*mimeType=*/"text/x-lldb.disassembly"}; + /*mimeType=*/ + "text/x-lldb.disassembly"}; } } // namespace lldb_dap diff --git a/lldb/tools/lldb-dap/Handler/StackTraceRequestHandler.cpp b/lldb/tools/lldb-dap/Handler/StackTraceRequestHandler.cpp index 4ea4cd1e517d..77ef952a1e34 100644 --- a/lldb/tools/lldb-dap/Handler/StackTraceRequestHandler.cpp +++ b/lldb/tools/lldb-dap/Handler/StackTraceRequestHandler.cpp @@ -69,7 +69,7 @@ static bool FillStackFrames(DAP &dap, lldb::SBThread &thread, break; } - stack_frames.emplace_back(CreateStackFrame(frame, frame_format)); + stack_frames.emplace_back(CreateStackFrame(dap, frame, frame_format)); } if (include_all && reached_end_of_stack) { diff --git a/lldb/tools/lldb-dap/JSONUtils.cpp b/lldb/tools/lldb-dap/JSONUtils.cpp index cf7db41559b8..08e65ab835a5 100644 --- a/lldb/tools/lldb-dap/JSONUtils.cpp +++ b/lldb/tools/lldb-dap/JSONUtils.cpp @@ -543,7 +543,7 @@ llvm::json::Object CreateEventObject(const llvm::StringRef event_name) { // }, // "required": [ "id", "name", "line", "column" ] // } -llvm::json::Value CreateStackFrame(lldb::SBFrame &frame, +llvm::json::Value CreateStackFrame(DAP &dap, lldb::SBFrame &frame, lldb::SBFormat &format) { llvm::json::Object object; int64_t frame_id = MakeDAPFrameID(frame); @@ -573,8 +573,10 @@ llvm::json::Value CreateStackFrame(lldb::SBFrame &frame, EmplaceSafeString(object, "name", frame_name); auto target = frame.GetThread().GetProcess().GetTarget(); - auto source = CreateSource(frame.GetPCAddress(), target); - if (!IsAssemblySource(source)) { + std::optional source = + dap.ResolveSource(frame.GetPCAddress()); + + if (source && !IsAssemblySource(*source)) { // This is a normal source with a valid line entry. auto line_entry = frame.GetLineEntry(); object.try_emplace("line", line_entry.GetLine()); @@ -598,7 +600,8 @@ llvm::json::Value CreateStackFrame(lldb::SBFrame &frame, object.try_emplace("column", 1); } - object.try_emplace("source", std::move(source)); + if (source) + object.try_emplace("source", std::move(source).value()); const auto pc = frame.GetPC(); if (pc != LLDB_INVALID_ADDRESS) { diff --git a/lldb/tools/lldb-dap/JSONUtils.h b/lldb/tools/lldb-dap/JSONUtils.h index 69da0725bd05..fd9a06931ebf 100644 --- a/lldb/tools/lldb-dap/JSONUtils.h +++ b/lldb/tools/lldb-dap/JSONUtils.h @@ -234,6 +234,9 @@ llvm::json::Object CreateEventObject(const llvm::StringRef event_name); /// "line" - the source file line number as an integer /// "column" - the source file column number as an integer /// +/// \param[in] dap +/// The DAP session associated with the stopped thread. +/// /// \param[in] frame /// The LLDB stack frame to use when populating out the "StackFrame" /// object. @@ -245,7 +248,7 @@ llvm::json::Object CreateEventObject(const llvm::StringRef event_name); /// \return /// A "StackFrame" JSON object with that follows the formal JSON /// definition outlined by Microsoft. -llvm::json::Value CreateStackFrame(lldb::SBFrame &frame, +llvm::json::Value CreateStackFrame(DAP &dap, lldb::SBFrame &frame, lldb::SBFormat &format); /// Create a "StackFrame" label object for a LLDB thread. diff --git a/lldb/tools/lldb-dap/Protocol/ProtocolRequests.h b/lldb/tools/lldb-dap/Protocol/ProtocolRequests.h index 7d9a99fdacce..d4b816c72679 100644 --- a/lldb/tools/lldb-dap/Protocol/ProtocolRequests.h +++ b/lldb/tools/lldb-dap/Protocol/ProtocolRequests.h @@ -468,7 +468,7 @@ struct SourceArguments { /// The reference to the source. This is the same as `source.sourceReference`. /// This is provided for backward compatibility since old clients do not /// understand the `source` attribute. - int64_t sourceReference; + int64_t sourceReference = LLDB_DAP_INVALID_SRC_REF; }; bool fromJSON(const llvm::json::Value &, SourceArguments &, llvm::json::Path); diff --git a/lldb/tools/lldb-dap/Protocol/ProtocolTypes.cpp b/lldb/tools/lldb-dap/Protocol/ProtocolTypes.cpp index 7f96c07faae1..f3635202175a 100644 --- a/lldb/tools/lldb-dap/Protocol/ProtocolTypes.cpp +++ b/lldb/tools/lldb-dap/Protocol/ProtocolTypes.cpp @@ -65,7 +65,7 @@ llvm::json::Value toJSON(const Source &S) { result.insert({"name", *S.name}); if (S.path) result.insert({"path", *S.path}); - if (S.sourceReference) + if (S.sourceReference && (*S.sourceReference > LLDB_DAP_INVALID_SRC_REF)) result.insert({"sourceReference", *S.sourceReference}); if (S.presentationHint) result.insert({"presentationHint", *S.presentationHint}); diff --git a/lldb/tools/lldb-dap/Protocol/ProtocolTypes.h b/lldb/tools/lldb-dap/Protocol/ProtocolTypes.h index 7fe745411399..6adfe3b7211b 100644 --- a/lldb/tools/lldb-dap/Protocol/ProtocolTypes.h +++ b/lldb/tools/lldb-dap/Protocol/ProtocolTypes.h @@ -28,6 +28,7 @@ #include #define LLDB_DAP_INVALID_VARRERF UINT64_MAX +#define LLDB_DAP_INVALID_SRC_REF 0 namespace lldb_dap::protocol { @@ -328,7 +329,7 @@ struct Source { /// `source` request (even if a path is specified). Since a `sourceReference` /// is only valid for a session, it can not be used to persist a source. The /// value should be less than or equal to 2147483647 (2^31-1). - std::optional sourceReference; + std::optional sourceReference; /// A hint for how to present the source in the UI. A value of `deemphasize` /// can be used to indicate that the source is not available or that it is diff --git a/lldb/tools/lldb-dap/ProtocolUtils.cpp b/lldb/tools/lldb-dap/ProtocolUtils.cpp index cb1ee6a42400..724d85110792 100644 --- a/lldb/tools/lldb-dap/ProtocolUtils.cpp +++ b/lldb/tools/lldb-dap/ProtocolUtils.cpp @@ -46,70 +46,33 @@ static bool ShouldDisplayAssemblySource( return false; } -static protocol::Source CreateAssemblySource(const lldb::SBTarget &target, - lldb::SBAddress address) { +std::optional CreateSource(const lldb::SBFileSpec &file) { + if (!file.IsValid()) + return std::nullopt; + protocol::Source source; - - auto symbol = address.GetSymbol(); - std::string name; - if (symbol.IsValid()) { - source.sourceReference = symbol.GetStartAddress().GetLoadAddress(target); - name = symbol.GetName(); - } else { - const auto load_addr = address.GetLoadAddress(target); - source.sourceReference = load_addr; - name = GetLoadAddressString(load_addr); - } - - lldb::SBModule module = address.GetModule(); - if (module.IsValid()) { - lldb::SBFileSpec file_spec = module.GetFileSpec(); - if (file_spec.IsValid()) { - std::string path = GetSBFileSpecPath(file_spec); - if (!path.empty()) - source.path = path + '`' + name; - } - } - - source.name = std::move(name); - - // Mark the source as deemphasized since users will only be able to view - // assembly for these frames. - source.presentationHint = - protocol::Source::PresentationHint::eSourcePresentationHintDeemphasize; - + if (const char *name = file.GetFilename()) + source.name = name; + char path[PATH_MAX] = ""; + if (file.GetPath(path, sizeof(path)) && + lldb::SBFileSpec::ResolvePath(path, path, PATH_MAX)) + source.path = path; return source; } -protocol::Source CreateSource(const lldb::SBFileSpec &file) { - protocol::Source source; - if (file.IsValid()) { - if (const char *name = file.GetFilename()) - source.name = name; - char path[PATH_MAX] = ""; - if (file.GetPath(path, sizeof(path)) && - lldb::SBFileSpec::ResolvePath(path, path, PATH_MAX)) - source.path = path; - } - return source; -} - -protocol::Source CreateSource(lldb::SBAddress address, lldb::SBTarget &target) { - lldb::SBDebugger debugger = target.GetDebugger(); - lldb::StopDisassemblyType stop_disassembly_display = - GetStopDisassemblyDisplay(debugger); - if (ShouldDisplayAssemblySource(address, stop_disassembly_display)) - return CreateAssemblySource(target, address); - - lldb::SBLineEntry line_entry = GetLineEntryForAddress(target, address); - return CreateSource(line_entry.GetFileSpec()); -} - bool IsAssemblySource(const protocol::Source &source) { // According to the specification, a source must have either `path` or // `sourceReference` specified. We use `path` for sources with known source // code, and `sourceReferences` when falling back to assembly. - return source.sourceReference.value_or(0) != 0; + return source.sourceReference.value_or(LLDB_DAP_INVALID_SRC_REF) > + LLDB_DAP_INVALID_SRC_REF; +} + +bool DisplayAssemblySource(lldb::SBDebugger &debugger, + lldb::SBAddress address) { + const lldb::StopDisassemblyType stop_disassembly_display = + GetStopDisassemblyDisplay(debugger); + return ShouldDisplayAssemblySource(address, stop_disassembly_display); } std::string GetLoadAddressString(const lldb::addr_t addr) { diff --git a/lldb/tools/lldb-dap/ProtocolUtils.h b/lldb/tools/lldb-dap/ProtocolUtils.h index 788d2fd054e2..f36bf0fb60a8 100644 --- a/lldb/tools/lldb-dap/ProtocolUtils.h +++ b/lldb/tools/lldb-dap/ProtocolUtils.h @@ -26,26 +26,15 @@ namespace lldb_dap { /// The SBFileSpec to use when populating out the "Source" object /// /// \return -/// A "Source" JSON object that follows the formal JSON +/// An optional "Source" JSON object that follows the formal JSON /// definition outlined by Microsoft. -protocol::Source CreateSource(const lldb::SBFileSpec &file); - -/// Create a "Source" JSON object as described in the debug adapter definition. -/// -/// \param[in] address -/// The address to use when populating out the "Source" object. -/// -/// \param[in] target -/// The target that has the address. -/// -/// \return -/// A "Source" JSON object that follows the formal JSON -/// definition outlined by Microsoft. -protocol::Source CreateSource(lldb::SBAddress address, lldb::SBTarget &target); +std::optional CreateSource(const lldb::SBFileSpec &file); /// Checks if the given source is for assembly code. bool IsAssemblySource(const protocol::Source &source); +bool DisplayAssemblySource(lldb::SBDebugger &debugger, lldb::SBAddress address); + /// Get the address as a 16-digit hex string, e.g. "0x0000000000012345" std::string GetLoadAddressString(const lldb::addr_t addr); diff --git a/lldb/tools/lldb-dap/SourceBreakpoint.cpp b/lldb/tools/lldb-dap/SourceBreakpoint.cpp index a4327ae18cf6..5fce9fe0ddbb 100644 --- a/lldb/tools/lldb-dap/SourceBreakpoint.cpp +++ b/lldb/tools/lldb-dap/SourceBreakpoint.cpp @@ -46,7 +46,13 @@ llvm::Error SourceBreakpoint::SetBreakpoint(const protocol::Source &source) { if (source.sourceReference) { // Breakpoint set by assembly source. - lldb::SBAddress source_address(*source.sourceReference, m_dap.target); + std::optional raw_addr = + m_dap.GetSourceReferenceAddress(*source.sourceReference); + if (!raw_addr) + return llvm::createStringError(llvm::inconvertibleErrorCode(), + "Invalid sourceReference."); + + lldb::SBAddress source_address(*raw_addr, m_dap.target); if (!source_address.IsValid()) return llvm::createStringError(llvm::inconvertibleErrorCode(), "Invalid sourceReference.");