mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
371 lines
12 KiB
371 lines
12 KiB
|
|
#ifndef WPP_MOF_RESOURCENAME
|
|
#define WPP_REG_TRACE_REGKEY L"SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\Tracing"
|
|
|
|
#define WPP_REG_TRACE_ENABLED L"EnableTracing"
|
|
|
|
#define WPP_REG_TRACE_LOG_FILE_NAME L"LogFileName"
|
|
#define WPP_REG_TRACE_LOG_SESSION_NAME L"LogSessionName"
|
|
#define WPP_REG_TRACE_LOG_BUFFER_SIZE L"BufferSize"
|
|
#define WPP_REG_TRACE_LOG_MIN_BUFFERS L"MinBuffers"
|
|
#define WPP_REG_TRACE_LOG_MAX_BUFFERS L"MaxBuffers"
|
|
#define WPP_REG_TRACE_LOG_MAX_FILESIZE L"MaxFileSize"
|
|
#define WPP_REG_TRACE_LOG_MAX_HISTORY L"MaxHistorySize"
|
|
#define WPP_REG_TRACE_LOG_MAX_BACKUPS L"MaxBackups"
|
|
|
|
#define WPP_REG_TRACE_ACTIVE L"Active"
|
|
#define WPP_REG_TRACE_CONTROL L"ControlFlags"
|
|
#define WPP_REG_TRACE_LEVEL L"Level"
|
|
#define WPP_REG_TRACE_GUID L"Guid"
|
|
#define WPP_REG_TRACE_BITNAMES L"BitNames"
|
|
#endif // #ifndef WPP_MOF_RESOURCENAME
|
|
|
|
#define DEFAULT_LOGGER_NAME L"stdout"
|
|
|
|
#if !defined(WppDebug)
|
|
# define WppDebug(a,b)
|
|
#endif
|
|
|
|
#if !defined(WPPINIT_STATIC)
|
|
# define WPPINIT_STATIC
|
|
#endif
|
|
|
|
#if !defined(WPPINIT_EXPORT)
|
|
# define WPPINIT_EXPORT
|
|
#endif
|
|
|
|
#define WPP_GUID_FORMAT "%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x"
|
|
#define WPP_GUID_ELEMENTS(p) \
|
|
p->Data1, p->Data2, p->Data3,\
|
|
p->Data4[0], p->Data4[1], p->Data4[2], p->Data4[3],\
|
|
p->Data4[4], p->Data4[5], p->Data4[6], p->Data4[7]
|
|
|
|
|
|
WPPINIT_STATIC
|
|
void WppIntToHex(
|
|
LPWSTR Buf,
|
|
unsigned int value,
|
|
int digits
|
|
)
|
|
{
|
|
static LPCWSTR hexDigit = L"0123456789abcdef";
|
|
while (--digits >= 0) {
|
|
Buf[digits] = hexDigit[ value & 15 ];
|
|
value /= 16; // compiler is smart enough to change it to bitshift
|
|
}
|
|
}
|
|
|
|
#define WPP_TEXTGUID_LEN 37
|
|
|
|
// b1e5deaf-1524-4a04-82c4-c9dfbce6cf97<NULL>
|
|
// 0 1 2 3
|
|
// 0123456789012345678901234567890123456
|
|
WPPINIT_STATIC
|
|
void WppGuidToStr(LPWSTR buf, LPCGUID guid) {
|
|
WppIntToHex(buf + 0, guid->Data1, 8);
|
|
buf[8] = '-';
|
|
WppIntToHex(buf + 9, guid->Data2, 4);
|
|
buf[13] = '-';
|
|
WppIntToHex(buf + 14, guid->Data3, 4);
|
|
buf[18] = '-';
|
|
WppIntToHex(buf + 19, guid->Data4[0], 2);
|
|
WppIntToHex(buf + 21, guid->Data4[1], 2);
|
|
buf[23] = '-';
|
|
WppIntToHex(buf + 24, guid->Data4[2], 2);
|
|
WppIntToHex(buf + 26, guid->Data4[3], 2);
|
|
WppIntToHex(buf + 28, guid->Data4[4], 2);
|
|
WppIntToHex(buf + 30, guid->Data4[5], 2);
|
|
WppIntToHex(buf + 32, guid->Data4[6], 2);
|
|
WppIntToHex(buf + 34, guid->Data4[7], 2);
|
|
buf[36] = 0;
|
|
}
|
|
|
|
#ifndef WPP_MOF_RESOURCENAME
|
|
WPPINIT_STATIC
|
|
ULONG
|
|
WppPublishTraceInfo(
|
|
PWPP_REGISTRATION_BLOCK Head,
|
|
LPCWSTR ProductName)
|
|
{
|
|
DWORD status;
|
|
DWORD disposition;
|
|
HKEY TracingKey = 0;
|
|
HKEY ProductKey = 0;
|
|
DWORD dwValue;
|
|
|
|
status = RegCreateKeyExW(HKEY_LOCAL_MACHINE,
|
|
WPP_REG_TRACE_REGKEY,
|
|
0,
|
|
NULL, // Class
|
|
REG_OPTION_NON_VOLATILE,
|
|
KEY_READ | KEY_CREATE_SUB_KEY,
|
|
NULL, // Sec Attributes
|
|
&TracingKey,
|
|
&disposition
|
|
);
|
|
if (status != ERROR_SUCCESS) {
|
|
WppDebug(1,("[WppInit] Failed to create Trace Key, %1!d!", status) );
|
|
goto exit_gracefully;
|
|
}
|
|
|
|
status = RegCreateKeyExW(TracingKey,
|
|
ProductName,
|
|
0,
|
|
NULL, // Class
|
|
REG_OPTION_NON_VOLATILE,
|
|
KEY_WRITE,
|
|
NULL, // Sec Attributes
|
|
&ProductKey,
|
|
&disposition
|
|
);
|
|
if (status != ERROR_SUCCESS) {
|
|
WppDebug(1,("[WppInit] Failed to create Trace Key, %1!d!", status) );
|
|
goto exit_gracefully;
|
|
}
|
|
|
|
status = RegSetValueExW(ProductKey,
|
|
WPP_REG_TRACE_LOG_SESSION_NAME,
|
|
0, // Reserved
|
|
REG_EXPAND_SZ,
|
|
(const BYTE*)DEFAULT_LOGGER_NAME,
|
|
sizeof(DEFAULT_LOGGER_NAME) );
|
|
if (status != ERROR_SUCCESS) {
|
|
WppDebug(1,("[WppInit] Failed to create LogSession value, %1!d!", status) );
|
|
goto exit_gracefully;
|
|
}
|
|
|
|
|
|
dwValue = 1;
|
|
status = RegSetValueExW(ProductKey,
|
|
WPP_REG_TRACE_ACTIVE,
|
|
0, // Reserved
|
|
REG_DWORD,
|
|
(const BYTE*)&dwValue,
|
|
sizeof(dwValue) );
|
|
if (status != ERROR_SUCCESS) {
|
|
WppDebug(1, ("[WppInit] Failed to create Active value, %1!d!", status) );
|
|
goto exit_gracefully;
|
|
}
|
|
dwValue = 1;
|
|
status = RegSetValueExW(ProductKey,
|
|
WPP_REG_TRACE_CONTROL,
|
|
0, // Reserved
|
|
REG_DWORD,
|
|
(const BYTE*)&dwValue,
|
|
sizeof(dwValue) );
|
|
if (status != ERROR_SUCCESS) {
|
|
WppDebug(1,("[WppInit] Failed to create Control value, %1!d!", status) );
|
|
goto exit_gracefully;
|
|
}
|
|
|
|
for(;Head;Head = Head->Next) {
|
|
HKEY ComponentKey;
|
|
|
|
status = RegCreateKeyExW(ProductKey,
|
|
Head->FriendlyName,
|
|
0,
|
|
NULL, // Class
|
|
REG_OPTION_NON_VOLATILE,
|
|
KEY_WRITE,
|
|
NULL, // Sec Attributes
|
|
&ComponentKey,
|
|
&disposition
|
|
);
|
|
if (status != ERROR_SUCCESS) {
|
|
WppDebug(1,("[WppInit] Failed to create %ws Key, %d",
|
|
Head->FriendlyName, status) );
|
|
} else {
|
|
GUID guid;
|
|
WCHAR guidBuf[WPP_TEXTGUID_LEN];
|
|
|
|
guid = *Head->ControlGuid;
|
|
WppGuidToStr(guidBuf, &guid);
|
|
|
|
status = RegSetValueExW(ComponentKey,
|
|
WPP_REG_TRACE_GUID,
|
|
0, // Reserved
|
|
REG_SZ,
|
|
(const BYTE*)guidBuf,
|
|
sizeof(guidBuf) );
|
|
if (status != ERROR_SUCCESS) {
|
|
WppDebug(1,("[WppInit] Failed to create GUID value, %1!d!", status) );
|
|
}
|
|
status = RegSetValueExW(ComponentKey,
|
|
WPP_REG_TRACE_BITNAMES,
|
|
0, // Reserved
|
|
REG_SZ,
|
|
(const BYTE*)Head->BitNames,
|
|
(DWORD)(wcslen(Head->BitNames) * sizeof(WCHAR)) );
|
|
if (status != ERROR_SUCCESS) {
|
|
WppDebug(1,("[WppInit] Failed to create GUID value, %1!d!", status) );
|
|
}
|
|
RegCloseKey(ComponentKey);
|
|
}
|
|
}
|
|
|
|
exit_gracefully:
|
|
|
|
if (ProductKey) {
|
|
RegCloseKey(ProductKey);
|
|
}
|
|
if (TracingKey) {
|
|
RegCloseKey(TracingKey);
|
|
}
|
|
|
|
return status;
|
|
}
|
|
#endif // #ifndef WPP_MOF_RESOURCENAME
|
|
|
|
|
|
ULONG
|
|
WINAPI
|
|
WppControlCallback(
|
|
IN WMIDPREQUESTCODE RequestCode,
|
|
IN PVOID Context,
|
|
IN OUT ULONG *InOutBufferSize,
|
|
IN OUT PVOID Buffer
|
|
)
|
|
{
|
|
PWPP_TRACE_CONTROL_BLOCK Ctx = (PWPP_TRACE_CONTROL_BLOCK)Context;
|
|
TRACEHANDLE Logger;
|
|
UCHAR Level;
|
|
DWORD Flags;
|
|
|
|
*InOutBufferSize = 0;
|
|
|
|
switch (RequestCode)
|
|
{
|
|
case WMI_ENABLE_EVENTS:
|
|
{
|
|
Logger = GetTraceLoggerHandle( Buffer );
|
|
Level = GetTraceEnableLevel(Logger);
|
|
Flags = GetTraceEnableFlags(Logger);
|
|
WppDebug(1, ("[WppInit] WMI_ENABLE_EVENTS Ctx %p Flags %x"
|
|
" Lev %d Logger %I64x\n",
|
|
Ctx, Flags, Level, Logger) );
|
|
break;
|
|
}
|
|
case WMI_DISABLE_EVENTS:
|
|
{
|
|
Logger = 0;
|
|
Flags = 0;
|
|
Level = 0;
|
|
WppDebug(1, ("[WppInit] WMI_DISABLE_EVENTS Ctx 0x%08p\n", Ctx));
|
|
break;
|
|
}
|
|
default:
|
|
{
|
|
return(ERROR_INVALID_PARAMETER);
|
|
}
|
|
}
|
|
if (Ctx->Options & WPP_VER_WIN2K_CB_FORWARD_PTR && Ctx->Win2kCb) {
|
|
Ctx->Win2kCb->Logger = Logger;
|
|
Ctx->Win2kCb->Level = Level;
|
|
Ctx->Win2kCb->Flags = Flags;
|
|
} else {
|
|
if (Ctx->Options & WPP_VER_WHISTLER_CB_FORWARD_PTR && Ctx->Cb) {
|
|
Ctx = Ctx->Cb; // use forwarding address
|
|
}
|
|
Ctx->Logger = Logger;
|
|
Ctx->Level = Level;
|
|
Ctx->Flags[0] = Flags;
|
|
}
|
|
return(ERROR_SUCCESS);
|
|
}
|
|
|
|
|
|
WPPINIT_EXPORT
|
|
VOID WppInitUm(LPCWSTR AppName, PWPP_REGISTRATION_BLOCK Registration)
|
|
{
|
|
PWPP_REGISTRATION_BLOCK p = Registration;
|
|
#ifdef WPP_MOF_RESOURCENAME
|
|
HMODULE hModule = NULL ;
|
|
WCHAR ImagePath[4 * MAX_PATH] = {UNICODE_NULL} ;
|
|
WCHAR WppMofResourceName[] = WPP_MOF_RESOURCENAME ;
|
|
#endif //#ifdef WPP_MOF_RESOURCENAME
|
|
|
|
#ifndef WPP_MOF_RESOURCENAME
|
|
if (AppName) {
|
|
WppPublishTraceInfo(Registration, AppName);
|
|
}
|
|
#endif // #ifndef WPP_MOF_RESOURCENAME
|
|
|
|
WppDebug(1, ("Registering %ws\n", AppName) );
|
|
for(; p; p = p->Next) {
|
|
TRACE_GUID_REGISTRATION FakeTraceGuid;
|
|
WPP_REGISTRATION_BLOCK RegBlock = *p;
|
|
PWPP_TRACE_CONTROL_BLOCK cb = (PWPP_TRACE_CONTROL_BLOCK)p;
|
|
ULONG status;
|
|
|
|
WppDebug(1,(WPP_GUID_FORMAT " %ws : %d:%ws\n",
|
|
WPP_GUID_ELEMENTS(p->ControlGuid),
|
|
p->FriendlyName, p->FlagsLen, p->BitNames));
|
|
|
|
ZeroMemory(cb, sizeof(WPP_TRACE_CONTROL_BLOCK)
|
|
+ sizeof(ULONG) * (RegBlock.FlagsLen - 1) );
|
|
p->Next = RegBlock.Next;
|
|
cb->FlagsLen = RegBlock.FlagsLen;
|
|
cb->Options = RegBlock.Options;
|
|
cb->Ptr = RegBlock.Ptr;
|
|
|
|
// Jee, do we need this fake trace guid? //
|
|
FakeTraceGuid.Guid = RegBlock.ControlGuid;
|
|
FakeTraceGuid.RegHandle = 0;
|
|
#ifdef WPP_MOF_RESOURCENAME
|
|
if (AppName != NULL) {
|
|
DWORD Status ;
|
|
#ifdef WPP_DLL
|
|
if ((hModule = GetModuleHandleW(AppName)) != NULL) {
|
|
Status = GetModuleFileNameW(hModule, ImagePath, MAX_PATH) ;
|
|
if (Status == 0) {
|
|
WppDebug(1,("RegisterTraceGuids => GetModuleFileName(DLL) Failed 0x%08X\n",GetLastError()));
|
|
}
|
|
} else {
|
|
WppDebug(1,("RegisterTraceGuids => GetModuleHandleW failed for %ws (0x%08X)\n",AppName,GetLastError()));
|
|
}
|
|
#else // #ifdef WPP_DLL
|
|
Status = GetModuleFileNameW(NULL,ImagePath,MAX_PATH);
|
|
if (Status == 0) {
|
|
WppDebug(1,("GetModuleFileName(EXE) Failed 0x%08X\n",GetLastError()));
|
|
}
|
|
#endif // #ifdef WPP_DLL
|
|
}
|
|
WppDebug(1,("registerTraceGuids => registering with WMI, App=%ws, Mof=%ws, ImagePath=%ws\n",AppName,WppMofResourceName,ImagePath));
|
|
|
|
status = RegisterTraceGuidsW( // Always use Unicode
|
|
#else // ifndef WPP_MOF_RESOURCENAME
|
|
|
|
status = RegisterTraceGuids( // So no change for existing users
|
|
#endif // ifndef WPP_MOF_RESOURCENAME
|
|
|
|
WppControlCallback,
|
|
cb, // Context for the callback
|
|
RegBlock.ControlGuid, // Control Guid
|
|
1, &FakeTraceGuid, // #TraceGuids, TraceGuid
|
|
#ifndef WPP_MOF_RESOURCENAME
|
|
0, //ImagePath,
|
|
0, //ResourceName,
|
|
#else // #ifndef WPP_MOF_RESOURCENAME
|
|
ImagePath,
|
|
WppMofResourceName,
|
|
#endif // #ifndef WPP_MOF_RESOURCENAME
|
|
&cb->UmRegistrationHandle
|
|
);
|
|
|
|
WppDebug(1, ("RegisterTraceGuid => %d\n", status) );
|
|
}
|
|
}
|
|
|
|
WPPINIT_EXPORT
|
|
VOID WppCleanupUm(PWPP_REGISTRATION_BLOCK Registration)
|
|
{
|
|
PWPP_TRACE_CONTROL_BLOCK x = (PWPP_TRACE_CONTROL_BLOCK)Registration;
|
|
WppDebug(1, ("Cleanup\n") );
|
|
for(; x; x = x->Next) {
|
|
WppDebug(1,("UnRegistering %I64x\n", x->UmRegistrationHandle) );
|
|
if (x->UmRegistrationHandle) {
|
|
UnregisterTraceGuids(x->UmRegistrationHandle);
|
|
}
|
|
}
|
|
}
|
|
|