1 //===-- HostProcessWindows.cpp ----------------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "lldb/Host/windows/HostProcessWindows.h"
11 #include "lldb/Host/FileSpec.h"
12 #include "lldb/Host/HostThread.h"
13 #include "lldb/Host/ThreadLauncher.h"
14 #include "lldb/Host/windows/windows.h"
15 
16 #include "llvm/ADT/STLExtras.h"
17 #include "llvm/Support/ConvertUTF.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 Error HostProcessWindows::Terminate() {
41   Error 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 Error HostProcessWindows::GetMainModule(FileSpec &file_spec) const {
52   Error error;
53   if (m_process == nullptr)
54     error.SetError(ERROR_INVALID_HANDLE, lldb::eErrorTypeWin32);
55 
56   std::vector<wchar_t> wpath(PATH_MAX);
57   if (::GetProcessImageFileNameW(m_process, wpath.data(), wpath.size())) {
58     std::string path;
59     if (llvm::convertWideToUTF8(wpath.data(), path))
60       file_spec.SetFile(path, false);
61     else
62       error.SetErrorString("Error converting path to UTF-8");
63   } else
64     error.SetError(::GetLastError(), lldb::eErrorTypeWin32);
65 
66   return error;
67 }
68 
69 lldb::pid_t HostProcessWindows::GetProcessId() const {
70   return (m_process == LLDB_INVALID_PROCESS) ? -1 : ::GetProcessId(m_process);
71 }
72 
73 bool HostProcessWindows::IsRunning() const {
74   if (m_process == nullptr)
75     return false;
76 
77   DWORD code = 0;
78   if (!::GetExitCodeProcess(m_process, &code))
79     return false;
80 
81   return (code == STILL_ACTIVE);
82 }
83 
84 HostThread HostProcessWindows::StartMonitoring(
85     const Host::MonitorChildProcessCallback &callback, bool monitor_signals) {
86   HostThread monitor_thread;
87   MonitorInfo *info = new MonitorInfo;
88   info->callback = callback;
89 
90   // Since the life of this HostProcessWindows instance and the life of the
91   // process may be different, duplicate the handle so that
92   // the monitor thread can have ownership over its own copy of the handle.
93   HostThread result;
94   if (::DuplicateHandle(GetCurrentProcess(), m_process, GetCurrentProcess(),
95                         &info->process_handle, 0, FALSE, DUPLICATE_SAME_ACCESS))
96     result = ThreadLauncher::LaunchThread("ChildProcessMonitor",
97                                           HostProcessWindows::MonitorThread,
98                                           info, nullptr);
99   return result;
100 }
101 
102 lldb::thread_result_t HostProcessWindows::MonitorThread(void *thread_arg) {
103   DWORD exit_code;
104 
105   MonitorInfo *info = static_cast<MonitorInfo *>(thread_arg);
106   if (info) {
107     ::WaitForSingleObject(info->process_handle, INFINITE);
108     ::GetExitCodeProcess(info->process_handle, &exit_code);
109     info->callback(::GetProcessId(info->process_handle), true, 0, exit_code);
110     ::CloseHandle(info->process_handle);
111     delete (info);
112   }
113   return 0;
114 }
115 
116 void HostProcessWindows::Close() {
117   if (m_owns_handle && m_process != LLDB_INVALID_PROCESS)
118     ::CloseHandle(m_process);
119   m_process = nullptr;
120 }
121