Create a HostProcess class.
This is a lightweight wrapper around a pid. It is intended to be lightweight enough to serve as a replacement anywhere we currently store a pid. It provides convenience methods and common process operations. This patch does not yet make use of HostProcess anywhere. llvm-svn: 216607
This commit is contained in:
112
lldb/source/Host/windows/HostProcessWindows.cpp
Normal file
112
lldb/source/Host/windows/HostProcessWindows.cpp
Normal file
@@ -0,0 +1,112 @@
|
||||
//===-- HostProcessWindows.cpp ----------------------------------*- C++ -*-===//
|
||||
//
|
||||
// The LLVM Compiler Infrastructure
|
||||
//
|
||||
// This file is distributed under the University of Illinois Open Source
|
||||
// License. See LICENSE.TXT for details.
|
||||
//
|
||||
//===----------------------------------------------------------------------===//
|
||||
|
||||
#include "lldb/Host/windows/windows.h"
|
||||
|
||||
#include <Psapi.h>
|
||||
|
||||
#include "lldb/Host/windows/HostProcessWindows.h"
|
||||
|
||||
#include "llvm/ADT/STLExtras.h"
|
||||
|
||||
using namespace lldb_private;
|
||||
|
||||
HostProcessWindows::HostProcessWindows()
|
||||
: m_process(NULL)
|
||||
, m_pid(0)
|
||||
{
|
||||
}
|
||||
|
||||
HostProcessWindows::~HostProcessWindows()
|
||||
{
|
||||
Close();
|
||||
}
|
||||
|
||||
Error HostProcessWindows::Create(lldb::pid_t pid)
|
||||
{
|
||||
Error error;
|
||||
if (pid == m_pid)
|
||||
return error;
|
||||
Close();
|
||||
|
||||
m_process = ::OpenProcess(PROCESS_TERMINATE | PROCESS_QUERY_LIMITED_INFORMATION, FALSE, pid);
|
||||
if (m_process == NULL)
|
||||
{
|
||||
error.SetError(::GetLastError(), lldb::eErrorTypeWin32);
|
||||
return error;
|
||||
}
|
||||
m_pid = pid;
|
||||
return error;
|
||||
}
|
||||
|
||||
Error HostProcessWindows::Create(lldb::process_t process)
|
||||
{
|
||||
Error error;
|
||||
if (process == m_process)
|
||||
return error;
|
||||
Close();
|
||||
|
||||
m_pid = ::GetProcessId(process);
|
||||
if (m_pid == 0)
|
||||
error.SetError(::GetLastError(), lldb::eErrorTypeWin32);
|
||||
m_process = process;
|
||||
return error;
|
||||
}
|
||||
|
||||
Error HostProcessWindows::Terminate()
|
||||
{
|
||||
Error error;
|
||||
if (m_process == NULL)
|
||||
error.SetError(ERROR_INVALID_HANDLE, lldb::eErrorTypeWin32);
|
||||
|
||||
if (!::TerminateProcess(m_process, 0))
|
||||
error.SetError(::GetLastError(), lldb::eErrorTypeWin32);
|
||||
|
||||
return error;
|
||||
}
|
||||
|
||||
Error HostProcessWindows::GetMainModule(FileSpec &file_spec) const
|
||||
{
|
||||
Error error;
|
||||
if (m_process == NULL)
|
||||
error.SetError(ERROR_INVALID_HANDLE, lldb::eErrorTypeWin32);
|
||||
|
||||
char path[MAX_PATH] = { 0 };
|
||||
if (::GetProcessImageFileName(m_process, path, llvm::array_lengthof(path)))
|
||||
file_spec.SetFile(path, false);
|
||||
else
|
||||
error.SetError(::GetLastError(), lldb::eErrorTypeWin32);
|
||||
|
||||
return error;
|
||||
}
|
||||
|
||||
lldb::pid_t HostProcessWindows::GetProcessId() const
|
||||
{
|
||||
return m_pid;
|
||||
}
|
||||
|
||||
bool HostProcessWindows::IsRunning() const
|
||||
{
|
||||
if (m_process == NULL)
|
||||
return false;
|
||||
|
||||
DWORD code = 0;
|
||||
if (!::GetExitCodeProcess(m_process, &code))
|
||||
return false;
|
||||
|
||||
return (code == STILL_ACTIVE);
|
||||
}
|
||||
|
||||
void HostProcessWindows::Close()
|
||||
{
|
||||
if (m_process != NULL)
|
||||
::CloseHandle(m_process);
|
||||
m_process = nullptr;
|
||||
m_pid = 0;
|
||||
}
|
||||
Reference in New Issue
Block a user