terminal/src/server/ObjectHandle.h
Michael Niksa 52d7de38b1
Prevent cleanup of object given to handle that failed access check (#3414)
* Ensure rights check and increments pass before assigning an object to a handle (since deallocation of handles will automatically attempt to cleanup).
2019-11-05 14:22:55 -08:00

86 lines
2.4 KiB
C++

/*++
Copyright (c) Microsoft Corporation
Licensed under the MIT license.
Module Name:
- ObjectHandle.h
Abstract:
- This file defines a handle associated with a console input or output buffer object.
- This is used to expose a handle to a client application via the API.
Author:
- Michael Niksa (miniksa) 12-Oct-2016
Revision History:
- Adapted from original items in handle.h
--*/
#pragma once
class INPUT_READ_HANDLE_DATA;
class InputBuffer;
class SCREEN_INFORMATION;
#include "WaitQueue.h"
class ConsoleHandleData final
{
public:
ConsoleHandleData(const ACCESS_MASK amAccess,
const ULONG ulShareAccess);
void Initialize(const ULONG ulHandleType,
PVOID const pvClientPointer);
~ConsoleHandleData();
ConsoleHandleData(const ConsoleHandleData&) = delete;
ConsoleHandleData(ConsoleHandleData&&) = delete;
ConsoleHandleData& operator=(const ConsoleHandleData&) & = delete;
ConsoleHandleData& operator=(ConsoleHandleData&&) & = delete;
[[nodiscard]] HRESULT GetInputBuffer(const ACCESS_MASK amRequested,
_Outptr_ InputBuffer** const ppInputBuffer) const;
[[nodiscard]] HRESULT GetScreenBuffer(const ACCESS_MASK amRequested,
_Outptr_ SCREEN_INFORMATION** const ppScreenInfo) const;
[[nodiscard]] HRESULT GetWaitQueue(_Outptr_ ConsoleWaitQueue** const ppWaitQueue) const;
INPUT_READ_HANDLE_DATA* GetClientInput() const;
bool IsReadAllowed() const;
bool IsReadShared() const;
bool IsWriteAllowed() const;
bool IsWriteShared() const;
// TODO: MSFT 9355178 Temporary public access to types... http://osgvsowi/9355178
bool IsInputHandle() const
{
return _IsInput();
}
enum HandleType
{
NotReady = 0x0,
Input = 0x1,
Output = 0x2
};
private:
bool _IsInput() const;
bool _IsOutput() const;
[[nodiscard]] HRESULT _CloseInputHandle();
[[nodiscard]] HRESULT _CloseOutputHandle();
ACCESS_MASK const _amAccess;
ULONG const _ulShareAccess;
ULONG _ulHandleType;
PVOID _pvClientPointer; // This will be a pointer to a SCREEN_INFORMATION or INPUT_INFORMATION object.
std::unique_ptr<INPUT_READ_HANDLE_DATA> _pClientInput;
};
DEFINE_ENUM_FLAG_OPERATORS(ConsoleHandleData::HandleType);