[Lldb-commits] [lldb] [lldb-mcp] Launch lldb on demand, if needed. (PR #158701)
via lldb-commits
lldb-commits at lists.llvm.org
Mon Sep 15 10:57:16 PDT 2025
llvmbot wrote:
<!--LLVM PR SUMMARY COMMENT-->
@llvm/pr-subscribers-lldb
Author: John Harrison (ashgti)
<details>
<summary>Changes</summary>
Adding support for launching lldb with `-O protocol start MCP` if a valid ~/.lldb/lldb-mcp-*.json` file is not found.
---
Full diff: https://github.com/llvm/llvm-project/pull/158701.diff
2 Files Affected:
- (modified) lldb/source/Host/common/Socket.cpp (+1-1)
- (modified) lldb/tools/lldb-mcp/lldb-mcp.cpp (+91-31)
``````````diff
diff --git a/lldb/source/Host/common/Socket.cpp b/lldb/source/Host/common/Socket.cpp
index 3511cde8bb36f..bc3d849c5c6c6 100644
--- a/lldb/source/Host/common/Socket.cpp
+++ b/lldb/source/Host/common/Socket.cpp
@@ -506,7 +506,7 @@ Socket::GetProtocolAndMode(llvm::StringRef scheme) {
.Case("unix-abstract-accept",
ProtocolModePair{SocketProtocol::ProtocolUnixAbstract,
SocketMode::ModeAccept})
- .Cases("connect", "tcp-connect",
+ .Cases("connect", "tcp-connect", "connection",
ProtocolModePair{SocketProtocol::ProtocolTcp,
SocketMode::ModeConnect})
.Case("udp", ProtocolModePair{SocketProtocol::ProtocolTcp,
diff --git a/lldb/tools/lldb-mcp/lldb-mcp.cpp b/lldb/tools/lldb-mcp/lldb-mcp.cpp
index 12545dcf3a3cc..42e82709dd9df 100644
--- a/lldb/tools/lldb-mcp/lldb-mcp.cpp
+++ b/lldb/tools/lldb-mcp/lldb-mcp.cpp
@@ -8,12 +8,16 @@
#include "lldb/Host/Config.h"
#include "lldb/Host/File.h"
+#include "lldb/Host/FileSystem.h"
+#include "lldb/Host/Host.h"
#include "lldb/Host/MainLoop.h"
#include "lldb/Host/MainLoopBase.h"
+#include "lldb/Host/ProcessLaunchInfo.h"
#include "lldb/Host/Socket.h"
#include "lldb/Initialization/SystemInitializerCommon.h"
#include "lldb/Initialization/SystemLifetimeManager.h"
#include "lldb/Protocol/MCP/Server.h"
+#include "lldb/Utility/FileSpec.h"
#include "lldb/Utility/Status.h"
#include "lldb/Utility/UriParser.h"
#include "lldb/lldb-forward.h"
@@ -24,7 +28,9 @@
#include "llvm/Support/ManagedStatic.h"
#include "llvm/Support/Signals.h"
#include "llvm/Support/WithColor.h"
+#include <chrono>
#include <cstdlib>
+#include <future>
#include <memory>
#if defined(_WIN32)
@@ -35,13 +41,19 @@ using namespace llvm;
using namespace lldb;
using namespace lldb_protocol::mcp;
+using lldb_private::Environment;
using lldb_private::File;
+using lldb_private::FileSpec;
+using lldb_private::FileSystem;
+using lldb_private::Host;
using lldb_private::MainLoop;
using lldb_private::MainLoopBase;
using lldb_private::NativeFile;
namespace {
+constexpr size_t kForwardIOBufferSize = 1024;
+
inline void exitWithError(llvm::Error Err, StringRef Prefix = "") {
handleAllErrors(std::move(Err), [&](ErrorInfoBase &Info) {
WithColor::error(errs(), Prefix) << Info.message() << '\n';
@@ -49,10 +61,68 @@ inline void exitWithError(llvm::Error Err, StringRef Prefix = "") {
std::exit(EXIT_FAILURE);
}
-constexpr size_t kForwardIOBufferSize = 1024;
+FileSpec driverPath() {
+ Environment host_env = Host::GetEnvironment();
+
+ // Check if an override for which lldb we're using exists, otherwise look next
+ // to the current binary.
+ std::string lldb_exe_path = host_env.lookup("LLDB_EXE_PATH");
+ auto &fs = FileSystem::Instance();
+ if (fs.Exists(lldb_exe_path)) {
+ return FileSpec(lldb_exe_path);
+ }
+ FileSpec lldb_exec_spec = lldb_private::HostInfo::GetProgramFileSpec();
+ lldb_exec_spec.SetFilename("lldb");
+ return lldb_exec_spec;
+}
+
+llvm::Error launch() {
+ FileSpec lldb_exec = driverPath();
+ lldb_private::ProcessLaunchInfo info;
+ info.SetExecutableFile(lldb_exec,
+ /*add_exe_file_as_first_arg=*/true);
+ info.GetArguments().AppendArgument("-O");
+ info.GetArguments().AppendArgument("protocol start MCP");
+ std::promise<int> exit_status;
+ info.SetMonitorProcessCallback([&](lldb::pid_t pid, int signal, int status) {
+ exit_status.set_value(status);
+ });
+
+ return Host::LaunchProcess(info).takeError();
+}
+
+Expected<ServerInfo> loadOrStart(
+ lldb_private::Timeout<std::micro> timeout = std::chrono::seconds(30)) {
+ using namespace std::chrono;
+ bool started = false;
+
+ auto deadline = steady_clock::now() + *timeout;
+ while (steady_clock::now() < deadline) {
+ auto servers = ServerInfo::Load();
+ if (!servers)
+ return servers.takeError();
+
+ if (servers->empty()) {
+ if (!started) {
+ started = true;
+ if (llvm::Error err = launch())
+ return std::move(err);
+ }
+ std::this_thread::sleep_for(std::chrono::microseconds(250));
+ continue;
+ }
+
+ if (servers->size() > 1)
+ return createStringError("To many MCP servers running, picking a "
+ "specific one is not yet implemented");
+
+ return servers->front();
+ }
+
+ return createStringError("timed out waiting for MCP server to start");
+}
-void forwardIO(lldb_private::MainLoopBase &loop, lldb::IOObjectSP &from,
- lldb::IOObjectSP &to) {
+void forwardIO(MainLoopBase &loop, IOObjectSP &from, IOObjectSP &to) {
char buf[kForwardIOBufferSize];
size_t num_bytes = sizeof(buf);
@@ -67,21 +137,24 @@ void forwardIO(lldb_private::MainLoopBase &loop, lldb::IOObjectSP &from,
exitWithError(std::move(err));
}
-void connectAndForwardIO(lldb_private::MainLoop &loop, ServerInfo &info,
- IOObjectSP &input_sp, IOObjectSP &output_sp) {
+llvm::Error connectAndForwardIO(lldb_private::MainLoop &loop, ServerInfo &info,
+ IOObjectSP &input_sp, IOObjectSP &output_sp) {
auto uri = lldb_private::URI::Parse(info.connection_uri);
if (!uri)
- exitWithError(createStringError("invalid connection_uri"));
+ return createStringError("invalid connection_uri");
std::optional<lldb_private::Socket::ProtocolModePair> protocol_and_mode =
lldb_private::Socket::GetProtocolAndMode(uri->scheme);
+ if (!protocol_and_mode)
+ return createStringError("unknown protocol scheme");
+
lldb_private::Status status;
std::unique_ptr<lldb_private::Socket> sock =
lldb_private::Socket::Create(protocol_and_mode->first, status);
if (status.Fail())
- exitWithError(status.takeError());
+ return status.takeError();
if (uri->port && !uri->hostname.empty())
status = sock->Connect(
@@ -89,24 +162,22 @@ void connectAndForwardIO(lldb_private::MainLoop &loop, ServerInfo &info,
else
status = sock->Connect(uri->path);
if (status.Fail())
- exitWithError(status.takeError());
+ return status.takeError();
IOObjectSP sock_sp = std::move(sock);
auto input_handle = loop.RegisterReadObject(
input_sp, std::bind(forwardIO, std::placeholders::_1, input_sp, sock_sp),
status);
if (status.Fail())
- exitWithError(status.takeError());
+ return status.takeError();
auto socket_handle = loop.RegisterReadObject(
sock_sp, std::bind(forwardIO, std::placeholders::_1, sock_sp, output_sp),
status);
if (status.Fail())
- exitWithError(status.takeError());
+ return status.takeError();
- status = loop.Run();
- if (status.Fail())
- exitWithError(status.takeError());
+ return loop.Run().takeError();
}
llvm::ManagedStatic<lldb_private::SystemLifetimeManager> g_debugger_lifetime;
@@ -147,30 +218,19 @@ int main(int argc, char *argv[]) {
IOObjectSP output_sp = std::make_shared<NativeFile>(
fileno(stdout), File::eOpenOptionWriteOnly, NativeFile::Unowned);
- static MainLoop loop;
+ Expected<ServerInfo> server_info = loadOrStart();
+ if (!server_info)
+ exitWithError(server_info.takeError());
+ static MainLoop loop;
sys::SetInterruptFunction([]() {
loop.AddPendingCallback(
[](MainLoopBase &loop) { loop.RequestTermination(); });
});
- auto existing_servers = ServerInfo::Load();
-
- if (!existing_servers)
- exitWithError(existing_servers.takeError());
-
- // FIXME: Launch `lldb -o 'protocol start MCP'`.
- if (existing_servers->empty())
- exitWithError(createStringError("No MCP servers running"));
-
- // FIXME: Support selecting a specific server.
- if (existing_servers->size() != 1)
- exitWithError(
- createStringError("To many MCP servers running, picking a specific "
- "one is not yet implemented."));
-
- ServerInfo &info = existing_servers->front();
- connectAndForwardIO(loop, info, input_sp, output_sp);
+ if (llvm::Error error =
+ connectAndForwardIO(loop, *server_info, input_sp, output_sp))
+ exitWithError(std::move(error));
return EXIT_SUCCESS;
}
``````````
</details>
https://github.com/llvm/llvm-project/pull/158701
More information about the lldb-commits
mailing list