1 //===-- HostProcessWindows.cpp --------------------------------------------===//
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 "lldb/Host/windows/HostProcessWindows.h"
10 #include "lldb/Host/HostThread.h"
11 #include "lldb/Host/ThreadLauncher.h"
12 #include "lldb/Host/windows/windows.h"
13 #include "lldb/Utility/FileSpec.h"
14 
15 #include "llvm/ADT/STLExtras.h"
16 #include "llvm/Support/ConvertUTF.h"
17 #include "llvm/Support/WindowsError.h"
18 
19 #include <psapi.h>
20 
21 using namespace lldb_private;
22 
23 namespace {
24 struct MonitorInfo {
25   Host::MonitorChildProcessCallback callback;
26   HANDLE process_handle;
27 };
28 }
29 
30 HostProcessWindows::HostProcessWindows()
31     : HostNativeProcessBase(), m_owns_handle(true) {}
32 
33 HostProcessWindows::HostProcessWindows(lldb::process_t process)
34     : HostNativeProcessBase(process), m_owns_handle(true) {}
35 
36 HostProcessWindows::~HostProcessWindows() { Close(); }
37 
38 void HostProcessWindows::SetOwnsHandle(bool owns) { m_owns_handle = owns; }
39 
40 Status HostProcessWindows::Terminate() {
41   Status error;
42   if (m_process == nullptr)
43     error.SetError(ERROR_INVALID_HANDLE, lldb::eErrorTypeWin32);
44 
45   if (!::TerminateProcess(m_process, 0))
46     error.SetError(::GetLastError(), lldb::eErrorTypeWin32);
47 
48   return error;
49 }
50 
51 lldb::pid_t HostProcessWindows::GetProcessId() const {
52   return (m_process == LLDB_INVALID_PROCESS) ? -1 : ::GetProcessId(m_process);
53 }
54 
55 bool HostProcessWindows::IsRunning() const {
56   if (m_process == nullptr)
57     return false;
58 
59   DWORD code = 0;
60   if (!::GetExitCodeProcess(m_process, &code))
61     return false;
62 
63   return (code == STILL_ACTIVE);
64 }
65 
66 llvm::Expected<HostThread> HostProcessWindows::StartMonitoring(
67     const Host::MonitorChildProcessCallback &callback, bool monitor_signals) {
68   MonitorInfo *info = new MonitorInfo;
69   info->callback = callback;
70 
71   // Since the life of this HostProcessWindows instance and the life of the
72   // process may be different, duplicate the handle so that the monitor thread
73   // can have ownership over its own copy of the handle.
74   if (::DuplicateHandle(GetCurrentProcess(), m_process, GetCurrentProcess(),
75                         &info->process_handle, 0, FALSE, DUPLICATE_SAME_ACCESS)) {
76     return ThreadLauncher::LaunchThread("ChildProcessMonitor",
77                                         HostProcessWindows::MonitorThread,
78                                         info);
79   } else {
80     return llvm::errorCodeToError(llvm::mapWindowsError(GetLastError()));
81   }
82 }
83 
84 lldb::thread_result_t HostProcessWindows::MonitorThread(void *thread_arg) {
85   DWORD exit_code;
86 
87   MonitorInfo *info = static_cast<MonitorInfo *>(thread_arg);
88   if (info) {
89     ::WaitForSingleObject(info->process_handle, INFINITE);
90     ::GetExitCodeProcess(info->process_handle, &exit_code);
91     info->callback(::GetProcessId(info->process_handle), true, 0, exit_code);
92     ::CloseHandle(info->process_handle);
93     delete (info);
94   }
95   return {};
96 }
97 
98 void HostProcessWindows::Close() {
99   if (m_owns_handle && m_process != LLDB_INVALID_PROCESS)
100     ::CloseHandle(m_process);
101   m_process = nullptr;
102 }
103