Many include statements use forward slashes, while others use backwards slashes. This is inconsistent formatting. For this reason, I changed the backward slashes to forward slashes since that is the standard.
62 lines
2.3 KiB
C++
62 lines
2.3 KiB
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT license.
|
|
|
|
#include "precomp.h"
|
|
|
|
#include "ProcessHandle.h"
|
|
|
|
#include "../host/globals.h"
|
|
#include "../host/telemetry.hpp"
|
|
|
|
// Routine Description:
|
|
// - Constructs an instance of the ConsoleProcessHandle Class
|
|
// - NOTE: Can throw if allocation fails or if there is a console policy we do not understand.
|
|
// - NOTE: Not being able to open the process by ID isn't a failure. It will be logged and continued.
|
|
ConsoleProcessHandle::ConsoleProcessHandle(const DWORD dwProcessId,
|
|
const DWORD dwThreadId,
|
|
const ULONG ulProcessGroupId) :
|
|
pWaitBlockQueue(std::make_unique<ConsoleWaitQueue>()),
|
|
pInputHandle(nullptr),
|
|
pOutputHandle(nullptr),
|
|
fRootProcess(false),
|
|
dwProcessId(dwProcessId),
|
|
dwThreadId(dwThreadId),
|
|
_ulTerminateCount(0),
|
|
_ulProcessGroupId(ulProcessGroupId),
|
|
_hProcess(LOG_LAST_ERROR_IF_NULL(OpenProcess(MAXIMUM_ALLOWED,
|
|
FALSE,
|
|
dwProcessId))),
|
|
_policy(ConsoleProcessPolicy::s_CreateInstance(_hProcess.get())),
|
|
_shimPolicy(ConsoleShimPolicy::s_CreateInstance(_hProcess.get()))
|
|
{
|
|
if (nullptr != _hProcess.get())
|
|
{
|
|
Telemetry::Instance().LogProcessConnected(_hProcess.get());
|
|
}
|
|
}
|
|
|
|
CD_CONNECTION_INFORMATION ConsoleProcessHandle::GetConnectionInformation() const
|
|
{
|
|
CD_CONNECTION_INFORMATION result = { 0 };
|
|
result.Process = reinterpret_cast<ULONG_PTR>(this);
|
|
result.Input = reinterpret_cast<ULONG_PTR>(pInputHandle.get());
|
|
result.Output = reinterpret_cast<ULONG_PTR>(pOutputHandle.get());
|
|
return result;
|
|
}
|
|
|
|
// Routine Description:
|
|
// - Retrieves the policies set on this particular process handle
|
|
// - This specifies restrictions that may apply to the calling console client application
|
|
const ConsoleProcessPolicy ConsoleProcessHandle::GetPolicy() const
|
|
{
|
|
return _policy;
|
|
}
|
|
|
|
// Routine Description:
|
|
// - Retrieves the policies set on this particular process handle
|
|
// - This specifies compatibility shims that we might need to make for certain applications.
|
|
const ConsoleShimPolicy ConsoleProcessHandle::GetShimPolicy() const
|
|
{
|
|
return _shimPolicy;
|
|
}
|