mirror of
https://github.com/BeneficialCode/WinArk.git
synced 2026-10-18 13:41:03 +08:00
BeneficialCode
114 lines
3.0 KiB
C++
114 lines
3.0 KiB
C++
#pragma once
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
namespace WinSys {
|
|
enum class ThreadPriority {
|
|
Idle = THREAD_PRIORITY_IDLE,
|
|
Lowest = THREAD_PRIORITY_LOWEST,
|
|
BelowNormal = THREAD_PRIORITY_BELOW_NORMAL,
|
|
Normal = THREAD_PRIORITY_NORMAL,
|
|
AboveNormal = THREAD_PRIORITY_ABOVE_NORMAL,
|
|
Highest = THREAD_PRIORITY_HIGHEST,
|
|
TimeCritical = THREAD_PRIORITY_TIME_CRITICAL
|
|
};
|
|
|
|
enum class ThreadAccessMask : uint32_t {
|
|
Synchronize = SYNCHRONIZE,
|
|
DirectImpersonation = THREAD_DIRECT_IMPERSONATION,
|
|
GetContext = THREAD_GET_CONTEXT,
|
|
SetContext = THREAD_SET_CONTEXT,
|
|
SetInformation = THREAD_SET_INFORMATION,
|
|
Impersonate = THREAD_IMPERSONATE,
|
|
QueryInformation = THREAD_QUERY_INFORMATION,
|
|
QueryLimitedInformation = THREAD_QUERY_LIMITED_INFORMATION,
|
|
SetLimitedInformation = THREAD_SET_LIMITED_INFORMATION,
|
|
SetThreadToken = THREAD_SET_THREAD_TOKEN,
|
|
SuspendResume = THREAD_SUSPEND_RESUME,
|
|
Terminate = THREAD_TERMINATE,
|
|
SystemSecurity = ACCESS_SYSTEM_SECURITY,
|
|
AllAccess = THREAD_ALL_ACCESS
|
|
};
|
|
|
|
enum class IoPriority {
|
|
Unknown = -1,
|
|
VeryLow = 0,
|
|
Low,
|
|
Normal,
|
|
High,
|
|
Critical
|
|
};
|
|
|
|
struct CpuNumber {
|
|
uint16_t Group;
|
|
uint8_t Number;
|
|
};
|
|
|
|
enum class ComFlags :uint32_t {
|
|
None = 0,
|
|
LocalTid = 0x01,
|
|
UuidInitialized = 0x02,
|
|
InThreadDetach = 0x04,
|
|
ChannelInitialized = 0x08,
|
|
WowThread = 0x10,
|
|
ThreadUninitializing = 0x20,
|
|
DisableOle1DDE = 0x40,
|
|
STA = 0x80,
|
|
MTA = 0x100,
|
|
Impersonating = 0x200,
|
|
DisableEventLogger = 0x400,
|
|
InNeutralApartment = 0x800,
|
|
DispatchThread = 0x1000,
|
|
HostThread = 0x2000,
|
|
AllowCoInit = 0x4000,
|
|
PendingUninit = 0x8000,
|
|
FirstMTAInit = 0x10000,
|
|
FirstNTAInit = 0x20000,
|
|
ApartmentInitializing = 0x40000,
|
|
UIMessageInModalLoop = 0x80000,
|
|
MarshallingErrorObject = 0x100000,
|
|
WinRTInitialize = 0x200000,
|
|
ASTA = 0x400000,
|
|
InShutdownCallbacks = 0x800000,
|
|
PointerInputBlocked = 0x1000000,
|
|
InActivationFilter = 0x2000000,
|
|
ASTAtoASTAExempQuirk = 0x4000000,
|
|
ASTAtoASTAExempProxy = 0x8000000,
|
|
ASTAtoASTAExempIndoubt = 0x10000000,
|
|
DetectedUserInit = 0x20000000,
|
|
BridgeSTA = 0x40000000,
|
|
MainInitializing = 0x80000000,
|
|
|
|
Error = 0xffffffff
|
|
};
|
|
DEFINE_ENUM_FLAG_OPERATORS(WinSys::ComFlags);
|
|
|
|
class Thread final {
|
|
public:
|
|
static std::unique_ptr<Thread> OpenById(uint32_t tid, ThreadAccessMask accessMask = ThreadAccessMask::QueryInformation);
|
|
explicit Thread(HANDLE handle, bool own = false); // 即被explicit关键字修饰的类构造函数，不能进行自动地隐式类型转换，只能显式地进行类型转换。
|
|
explicit Thread(uint32_t tid, ThreadAccessMask accessMask = ThreadAccessMask::QueryInformation);
|
|
~Thread();
|
|
|
|
HANDLE GetHandle() const {
|
|
return _handle;
|
|
}
|
|
|
|
bool IsValid() const;
|
|
|
|
ThreadPriority GetPriority() const;
|
|
bool SetPrioriry(ThreadPriority priority);
|
|
CpuNumber GetIdealProcessor() const;
|
|
bool Terminate(uint32_t exitCode = 0);
|
|
int GetMemoryPriority() const;
|
|
IoPriority GetIoPriority() const;
|
|
size_t GetSubProcessTag() const;
|
|
std::wstring GetServiceNameByTag(uint32_t pid);
|
|
ComFlags GetComFlags() const;
|
|
|
|
private:
|
|
HANDLE _handle;
|
|
bool _own;
|
|
};
|
|
} |