mirror of
https://github.com/rbmm/SC
synced 2026-06-08 17:03:41 +00:00
rbmm
74 lines
1.2 KiB
C++
74 lines
1.2 KiB
C++
#pragma once
|
|
|
|
struct PrintInfo : TEB_ACTIVE_FRAME, TEB_ACTIVE_FRAME_CONTEXT
|
|
{
|
|
HANDLE _G_hFile;
|
|
UINT _G_CodePage;
|
|
BOOLEAN _G_bConsole;
|
|
|
|
PrintInfo()
|
|
{
|
|
Context = this;
|
|
TEB_ACTIVE_FRAME_CONTEXT::Flags = 0;
|
|
FrameName = name();
|
|
RtlPushFrame(this);
|
|
}
|
|
|
|
~PrintInfo()
|
|
{
|
|
RtlPopFrame(this);
|
|
}
|
|
|
|
static PrintInfo* get()
|
|
{
|
|
if (TEB_ACTIVE_FRAME* prf = RtlGetFrame())
|
|
{
|
|
do
|
|
{
|
|
if (name() == prf->Context->FrameName) return static_cast<PrintInfo*>(prf);
|
|
} while (prf = prf->Previous);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static PCSTR name()
|
|
{
|
|
return __FUNCDNAME__;
|
|
}
|
|
};
|
|
|
|
void PutChars(PCWSTR pwz, ULONG cch);
|
|
|
|
inline void PutChars(PCWSTR pwz)
|
|
{
|
|
PutChars(pwz, (ULONG)wcslen(pwz));
|
|
}
|
|
|
|
void PrintWA_v(PCWSTR format, ...);
|
|
|
|
#define DbgPrint(fmt, ...) PrintWA_v(_CRT_WIDE(fmt), __VA_ARGS__ )
|
|
|
|
template <typename T>
|
|
T HR(HRESULT& hr, T t)
|
|
{
|
|
hr = t ? NOERROR : GetLastError();
|
|
return t;
|
|
}
|
|
|
|
HRESULT PrintError(HRESULT dwError);
|
|
|
|
void InitPrintf();
|
|
|
|
#define echo(x) x
|
|
#define label(x) echo(x)##__LINE__
|
|
|
|
#define BEGIN_PRIVILEGES(name, n) static const union { TOKEN_PRIVILEGES name;\
|
|
struct { ULONG PrivilegeCount; LUID_AND_ATTRIBUTES Privileges[n];} label(_) = { n, {
|
|
|
|
#define LAA(se) {{se}, SE_PRIVILEGE_ENABLED }
|
|
#define LAA_D(se) {{se} }
|
|
|
|
#define END_PRIVILEGES }};};
|
|
|