forked from OSchip/llvm-project
101 lines
3.0 KiB
C++
101 lines
3.0 KiB
C++
//===-- HostProcessWindows.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 "lldb/Host/windows/HostProcessWindows.h"
|
|
#include "lldb/Host/HostThread.h"
|
|
#include "lldb/Host/ThreadLauncher.h"
|
|
#include "lldb/Host/windows/windows.h"
|
|
#include "lldb/Utility/FileSpec.h"
|
|
|
|
#include "llvm/ADT/STLExtras.h"
|
|
#include "llvm/Support/ConvertUTF.h"
|
|
#include "llvm/Support/WindowsError.h"
|
|
|
|
#include <psapi.h>
|
|
|
|
using namespace lldb_private;
|
|
|
|
namespace {
|
|
struct MonitorInfo {
|
|
Host::MonitorChildProcessCallback callback;
|
|
HANDLE process_handle;
|
|
};
|
|
}
|
|
|
|
HostProcessWindows::HostProcessWindows()
|
|
: HostNativeProcessBase(), m_owns_handle(true) {}
|
|
|
|
HostProcessWindows::HostProcessWindows(lldb::process_t process)
|
|
: HostNativeProcessBase(process), m_owns_handle(true) {}
|
|
|
|
HostProcessWindows::~HostProcessWindows() { Close(); }
|
|
|
|
void HostProcessWindows::SetOwnsHandle(bool owns) { m_owns_handle = owns; }
|
|
|
|
Status HostProcessWindows::Terminate() {
|
|
Status error;
|
|
if (m_process == nullptr)
|
|
error.SetError(ERROR_INVALID_HANDLE, lldb::eErrorTypeWin32);
|
|
|
|
if (!::TerminateProcess(m_process, 0))
|
|
error.SetError(::GetLastError(), lldb::eErrorTypeWin32);
|
|
|
|
return error;
|
|
}
|
|
|
|
lldb::pid_t HostProcessWindows::GetProcessId() const {
|
|
return (m_process == LLDB_INVALID_PROCESS) ? -1 : ::GetProcessId(m_process);
|
|
}
|
|
|
|
bool HostProcessWindows::IsRunning() const {
|
|
if (m_process == nullptr)
|
|
return false;
|
|
|
|
DWORD code = 0;
|
|
if (!::GetExitCodeProcess(m_process, &code))
|
|
return false;
|
|
|
|
return (code == STILL_ACTIVE);
|
|
}
|
|
|
|
static lldb::thread_result_t
|
|
MonitorThread(const Host::MonitorChildProcessCallback &callback,
|
|
HANDLE process_handle) {
|
|
DWORD exit_code;
|
|
|
|
::WaitForSingleObject(process_handle, INFINITE);
|
|
::GetExitCodeProcess(process_handle, &exit_code);
|
|
callback(::GetProcessId(process_handle), 0, exit_code);
|
|
::CloseHandle(process_handle);
|
|
return {};
|
|
}
|
|
|
|
llvm::Expected<HostThread> HostProcessWindows::StartMonitoring(
|
|
const Host::MonitorChildProcessCallback &callback) {
|
|
HANDLE process_handle;
|
|
|
|
// Since the life of this HostProcessWindows instance and the life of the
|
|
// process may be different, duplicate the handle so that the monitor thread
|
|
// can have ownership over its own copy of the handle.
|
|
if (::DuplicateHandle(GetCurrentProcess(), m_process, GetCurrentProcess(),
|
|
&process_handle, 0, FALSE, DUPLICATE_SAME_ACCESS)) {
|
|
return ThreadLauncher::LaunchThread(
|
|
"ChildProcessMonitor", [callback, process_handle] {
|
|
return MonitorThread(callback, process_handle);
|
|
});
|
|
} else {
|
|
return llvm::errorCodeToError(llvm::mapWindowsError(GetLastError()));
|
|
}
|
|
}
|
|
|
|
void HostProcessWindows::Close() {
|
|
if (m_owns_handle && m_process != LLDB_INVALID_PROCESS)
|
|
::CloseHandle(m_process);
|
|
m_process = nullptr;
|
|
}
|