https://github.com/charles-zablit updated https://github.com/llvm/llvm-project/pull/181811
>From fb7839473960ab927e50d3807cf828a635da4ac5 Mon Sep 17 00:00:00 2001 From: Charles Zablit <[email protected]> Date: Tue, 17 Feb 2026 12:10:28 +0000 Subject: [PATCH 1/3] [lldb][windows] release the ConPTY instead of copying its reference --- lldb/include/lldb/Host/ProcessLaunchInfo.h | 2 +- lldb/include/lldb/Target/Process.h | 7 +------ lldb/source/Host/common/ProcessLaunchInfo.cpp | 3 +++ lldb/source/Host/windows/PseudoConsole.cpp | 4 ++-- .../Plugins/Platform/Windows/PlatformWindows.cpp | 3 +-- .../Plugins/Process/Windows/Common/ProcessWindows.cpp | 11 ++++++----- .../Plugins/Process/Windows/Common/ProcessWindows.h | 3 +-- 7 files changed, 15 insertions(+), 18 deletions(-) diff --git a/lldb/include/lldb/Host/ProcessLaunchInfo.h b/lldb/include/lldb/Host/ProcessLaunchInfo.h index 023d150503da3..a07f35ac472e9 100644 --- a/lldb/include/lldb/Host/ProcessLaunchInfo.h +++ b/lldb/include/lldb/Host/ProcessLaunchInfo.h @@ -130,7 +130,7 @@ class ProcessLaunchInfo : public ProcessInfo { PTY &GetPTY() const { return *m_pty; } - std::shared_ptr<PTY> GetPTYSP() const { return m_pty; } + std::shared_ptr<PTY> ReleasePTY() { return std::move(m_pty); } /// Returns whether if lldb should read information from the PTY. This is /// always true on non Windows. diff --git a/lldb/include/lldb/Target/Process.h b/lldb/include/lldb/Target/Process.h index 7a15adebc63ee..c4eb8cf3694b1 100644 --- a/lldb/include/lldb/Target/Process.h +++ b/lldb/include/lldb/Target/Process.h @@ -2561,15 +2561,10 @@ void PruneThreadPlans(); /// When data is successfully read from the ConPTY, it is stored in /// m_stdout_data. There is no differentiation between stdout and stderr. /// - /// \param[in] pty - /// The ConPTY to use for process STDIO communication. It's - /// assumed to be valid. - /// /// \see lldb_private::Process::STDIOReadThreadBytesReceived() /// \see lldb_private::IOHandlerProcessSTDIOWindows /// \see lldb_private::PseudoConsole - virtual void - SetPseudoConsoleHandle(const std::shared_ptr<PseudoConsole> &pty) {}; + virtual void SetPseudoConsoleHandle() {}; #endif /// Associates a file descriptor with the process' STDIO handling diff --git a/lldb/source/Host/common/ProcessLaunchInfo.cpp b/lldb/source/Host/common/ProcessLaunchInfo.cpp index e82cc11187fe5..3c354c9164a50 100644 --- a/lldb/source/Host/common/ProcessLaunchInfo.cpp +++ b/lldb/source/Host/common/ProcessLaunchInfo.cpp @@ -200,6 +200,9 @@ void ProcessLaunchInfo::SetDetachOnError(bool enable) { llvm::Error ProcessLaunchInfo::SetUpPtyRedirection() { Log *log = GetLog(LLDBLog::Process); + if (m_pty == nullptr) + m_pty = std::make_shared<PTY>(); + bool stdin_free = GetFileActionForFD(STDIN_FILENO) == nullptr; bool stdout_free = GetFileActionForFD(STDOUT_FILENO) == nullptr; bool stderr_free = GetFileActionForFD(STDERR_FILENO) == nullptr; diff --git a/lldb/source/Host/windows/PseudoConsole.cpp b/lldb/source/Host/windows/PseudoConsole.cpp index 413c8a3328445..2228c315665f4 100644 --- a/lldb/source/Host/windows/PseudoConsole.cpp +++ b/lldb/source/Host/windows/PseudoConsole.cpp @@ -72,8 +72,8 @@ llvm::Error PseudoConsole::OpenPseudoConsole() { return llvm::make_error<llvm::StringError>("ConPTY is not available", llvm::errc::io_error); - // close any previously opened handles - Close(); + assert(m_conpty_handle == INVALID_HANDLE_VALUE && + "ConPTY has already been opened"); HRESULT hr; HANDLE hInputRead = INVALID_HANDLE_VALUE; diff --git a/lldb/source/Plugins/Platform/Windows/PlatformWindows.cpp b/lldb/source/Plugins/Platform/Windows/PlatformWindows.cpp index 0e74cab23c99e..1bc9d3fb9978d 100644 --- a/lldb/source/Plugins/Platform/Windows/PlatformWindows.cpp +++ b/lldb/source/Plugins/Platform/Windows/PlatformWindows.cpp @@ -523,8 +523,7 @@ ProcessSP PlatformWindows::DebugProcess(ProcessLaunchInfo &launch_info, error = process_sp->Launch(launch_info); #ifdef _WIN32 if (error.Success()) { - if (launch_info.ShouldUsePTY()) - process_sp->SetPseudoConsoleHandle(launch_info.GetPTYSP()); + process_sp->SetPseudoConsoleHandle(); } else { Log *log = GetLog(LLDBLog::Platform); LLDB_LOGF(log, "Platform::%s LaunchProcess() failed: %s", __FUNCTION__, diff --git a/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.cpp b/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.cpp index 373729c952071..ce969e2ff8454 100644 --- a/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.cpp +++ b/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.cpp @@ -214,7 +214,7 @@ Status ProcessWindows::DoLaunch(Module *exe_module, error = LaunchProcess(launch_info, delegate); if (error.Success()) SetID(launch_info.GetProcessID()); - m_pty = launch_info.GetPTYSP(); + m_pty = launch_info.ReleasePTY(); return error; } @@ -1130,10 +1130,11 @@ class IOHandlerProcessSTDIOWindows : public IOHandler { bool m_is_running = false; }; -void ProcessWindows::SetPseudoConsoleHandle( - const std::shared_ptr<PseudoConsole> &pty) { +void ProcessWindows::SetPseudoConsoleHandle() { + if (m_pty == nullptr) + return; m_stdio_communication.SetConnection( - std::make_unique<ConnectionGenericFile>(pty->GetSTDOUTHandle(), false)); + std::make_unique<ConnectionGenericFile>(m_pty->GetSTDOUTHandle(), false)); if (m_stdio_communication.IsConnected()) { m_stdio_communication.SetReadThreadBytesReceivedCallback( STDIOReadThreadBytesReceived, this); @@ -1144,7 +1145,7 @@ void ProcessWindows::SetPseudoConsoleHandle( std::lock_guard<std::mutex> guard(m_process_input_reader_mutex); if (!m_process_input_reader) m_process_input_reader = std::make_shared<IOHandlerProcessSTDIOWindows>( - this, pty->GetSTDINHandle()); + this, m_pty->GetSTDINHandle()); } } } diff --git a/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.h b/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.h index becab6964a4aa..d3e3f9a5ed0ce 100644 --- a/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.h +++ b/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.h @@ -98,8 +98,7 @@ class ProcessWindows : public Process, public ProcessDebugger { Status DisableWatchpoint(lldb::WatchpointSP wp_sp, bool notify = true) override; - void - SetPseudoConsoleHandle(const std::shared_ptr<PseudoConsole> &pty) override; + void SetPseudoConsoleHandle() override; protected: ProcessWindows(lldb::TargetSP target_sp, lldb::ListenerSP listener_sp); >From c7fa263a2ef99ae4ebf7bf977534626813b78425 Mon Sep 17 00:00:00 2001 From: Charles Zablit <[email protected]> Date: Tue, 17 Feb 2026 15:09:10 +0000 Subject: [PATCH 2/3] assert that use_count == 1 after ReleasePTY --- lldb/source/Plugins/Process/Windows/Common/ProcessWindows.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.cpp b/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.cpp index ce969e2ff8454..19d30c9a49a0a 100644 --- a/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.cpp +++ b/lldb/source/Plugins/Process/Windows/Common/ProcessWindows.cpp @@ -215,6 +215,10 @@ Status ProcessWindows::DoLaunch(Module *exe_module, if (error.Success()) SetID(launch_info.GetProcessID()); m_pty = launch_info.ReleasePTY(); + // At this point, Process owns the ConPTY. If ProcessLaunchInfo still has a + // reference to it, it might get closed prematurely if another target is + // created. + assert(m_pty.use_count() == 1 && "More than one reference to the ConPTY"); return error; } >From 22df0f602ce3d4124d355ee778360e832ebcdd48 Mon Sep 17 00:00:00 2001 From: Charles Zablit <[email protected]> Date: Tue, 17 Feb 2026 22:01:49 +0000 Subject: [PATCH 3/3] Update lldb/source/Host/common/ProcessLaunchInfo.cpp Co-authored-by: Adrian Prantl <[email protected]> --- lldb/source/Host/common/ProcessLaunchInfo.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lldb/source/Host/common/ProcessLaunchInfo.cpp b/lldb/source/Host/common/ProcessLaunchInfo.cpp index 3c354c9164a50..ea036d45e7c80 100644 --- a/lldb/source/Host/common/ProcessLaunchInfo.cpp +++ b/lldb/source/Host/common/ProcessLaunchInfo.cpp @@ -200,7 +200,7 @@ void ProcessLaunchInfo::SetDetachOnError(bool enable) { llvm::Error ProcessLaunchInfo::SetUpPtyRedirection() { Log *log = GetLog(LLDBLog::Process); - if (m_pty == nullptr) + if (!m_pty) m_pty = std::make_shared<PTY>(); bool stdin_free = GetFileActionForFD(STDIN_FILENO) == nullptr; _______________________________________________ lldb-commits mailing list [email protected] https://lists.llvm.org/cgi-bin/mailman/listinfo/lldb-commits
