/*++
wstdump.c
Dump routines for WST
History:
10-26-92 RezaB - created.
--*/
#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <windows.h>
#include <wst.h>
#include "wstdump.h"
//
// dump clear switches and defaults
//
WORD fDump = FALSE;
WORD fClear = FALSE;
WORD fPause = TRUE;
HANDLE hDoneEvent;
HANDLE hDumpEvent;
HANDLE hClearEvent;
HANDLE hPauseEvent;
HANDLE hdll;
SECURITY_DESCRIPTOR SecDescriptor;
//
// error handling
//
#define LOG_FILE "wstdump.log"
FILE *pfLog;
void ClearDumpInfo (void);
INT_PTR APIENTRY DialogProc(HWND, UINT, WPARAM, LPARAM);
/*++
Main Routine
--*/
int WinMain(HINSTANCE hInstance,HINSTANCE hPrevInst,LPSTR lpCmdLine,int nCmdShow)
{
NTSTATUS Status;
STRING EventName;
UNICODE_STRING EventUnicodeName;
OBJECT_ATTRIBUTES EventAttributes;
// Prevent compiler from complaining..
//
hPrevInst;
lpCmdLine;
nCmdShow;
// Open the log file for logging possible errors
//
pfLog = fopen (LOG_FILE, "w");
if (!pfLog) {
exit (1);
}
// Create public share security descriptor for all the named objects
//
Status = RtlCreateSecurityDescriptor (
&SecDescriptor,
SECURITY_DESCRIPTOR_REVISION1
);
if (!NT_SUCCESS(Status)) {
fprintf (pfLog, "WSTDUMP: main () - RtlCreateSecurityDescriptor() "
"failed - %lx\n", Status);
exit (1);
}
Status = RtlSetDaclSecurityDescriptor (
&SecDescriptor, // SecurityDescriptor
TRUE, // DaclPresent
NULL, // Dacl
FALSE // DaclDefaulted
);
if (!NT_SUCCESS(Status)) {
fprintf (pfLog, "WSTDUMP: main () - RtlSetDaclSecurityDescriptor() "
"failed - %lx\n", Status);
exit (1);
}
// Initialization for DONE event creation
//
RtlInitString (&EventName, DONEEVENTNAME);
Status = RtlAnsiStringToUnicodeString (&EventUnicodeName,
&EventName,
TRUE);
if (NT_SUCCESS(Status)) {
InitializeObjectAttributes (&EventAttributes,
&EventUnicodeName,
OBJ_OPENIF | OBJ_CASE_INSENSITIVE,
NULL,
&SecDescriptor);
}
else {
fprintf (pfLog, "WSTDUMP: main () - RtlAnsiStringToUnicodeString() "
"failed for DUMP event name - %lx\n", Status);
exit (1);
}
//
// Create DONE event
//
Status = NtCreateEvent (&hDoneEvent,
EVENT_QUERY_STATE |
EVENT_MODIFY_STATE |
SYNCHRONIZE,
&EventAttributes,
NotificationEvent,
TRUE);
if (!NT_SUCCESS(Status)) {
fprintf (pfLog, "WSTDUMP: main () - NtCreateEvent() "
"failed to create DUMP event - %lx\n", Status);
exit (1);
}
// Initialization for DUMP event creation
//
RtlInitString (&EventName, DUMPEVENTNAME);
Status = RtlAnsiStringToUnicodeString (&EventUnicodeName,
&EventName,
TRUE);
if (NT_SUCCESS(Status)) {
InitializeObjectAttributes (&EventAttributes,
&EventUnicodeName,
OBJ_OPENIF | OBJ_CASE_INSENSITIVE,
NULL,
&SecDescriptor);
}
else {
fprintf (pfLog, "WSTDUMP: main () - RtlAnsiStringToUnicodeString() "
"failed for DUMP event name - %lx\n", Status);
exit (1);
}
//
// Create DUMP event
//
Status = NtCreateEvent (&hDumpEvent,
EVENT_QUERY_STATE |
EVENT_MODIFY_STATE |
SYNCHRONIZE,
&EventAttributes,
NotificationEvent,
FALSE);
if (!NT_SUCCESS(Status)) {
fprintf (pfLog, "WSTDUMP: main () - NtCreateEvent() "
"failed to create DUMP event - %lx\n", Status);
exit (1);
}
// Initialization for CLEAR event creation
//
RtlInitString (&EventName, CLEAREVENTNAME);
Status = RtlAnsiStringToUnicodeString (&EventUnicodeName,
&EventName,
TRUE);
if (NT_SUCCESS(Status)) {
InitializeObjectAttributes (&EventAttributes,
&EventUnicodeName,
OBJ_OPENIF | OBJ_CASE_INSENSITIVE,
NULL,
&SecDescriptor);
}
else {
fprintf (pfLog, "WSTDUMP: main () - RtlAnsiStringToUnicodeString() "
"failed for CLEAR event name - %lx\n", Status);
exit (1);
}
//
// Create CLEAR event
//
Status = NtCreateEvent (&hClearEvent,
EVENT_QUERY_STATE |
EVENT_MODIFY_STATE |
SYNCHRONIZE,
&EventAttributes,
NotificationEvent,
FALSE);
if (!NT_SUCCESS(Status)) {
fprintf (pfLog, "WSTDUMP: main () - NtCreateEvent() "
"failed to create CLEAR event - %lx\n", Status);
exit (1);
}
// Initialization for PAUSE event creation
//
RtlInitString (&EventName, PAUSEEVENTNAME);
Status = RtlAnsiStringToUnicodeString (&EventUnicodeName,
&EventName,
TRUE);
if (NT_SUCCESS(Status)) {
InitializeObjectAttributes (&EventAttributes,
&EventUnicodeName,
OBJ_OPENIF | OBJ_CASE_INSENSITIVE,
NULL,
&SecDescriptor);
}
else {
fprintf (pfLog, "WSTDUMP: main () - RtlAnsiStringToUnicodeString() "
"failed for PAUSE event name - %lx\n", Status);
exit (1);
}
//
// Create PAUSE event
//
Status = NtCreateEvent (&hPauseEvent,
EVENT_QUERY_STATE |
EVENT_MODIFY_STATE |
SYNCHRONIZE,
&EventAttributes,
NotificationEvent,
FALSE);
if (!NT_SUCCESS(Status)) {
fprintf (pfLog, "WSTDUMP: main () - NtCreateEvent() "
"failed to create PAUSE event - %lx\n", Status);
exit (1);
}
//
// show dialog box
//
DialogBox(hInstance, "DumpDialog", NULL, DialogProc);
return (0);
} /* main */
/*++
Clears and/or dump profiling info to the dump file.
Input:
-none-
Output:
-none-
--*/
void ClearDumpInfo (void)
{
NTSTATUS Status;
//
// Pause profiling?
//
if (fPause) {
Status = NtPulseEvent (hPauseEvent, NULL);
if (!NT_SUCCESS(Status)) {
fprintf (pfLog, "WSTDUMP: ClearDumpInfo () - NtPulseEvent() "
"failed for PAUSE event - %lx\n", Status);
exit (1);
}
}
//
// Dump data?
//
else if (fDump) {
Status = NtPulseEvent (hDumpEvent, NULL);
if (!NT_SUCCESS(Status)) {
fprintf (pfLog, "WSTDUMP: ClearDumpInfo () - NtPulseEvent() "
"failed for DUMP event - %lx\n", Status);
exit (1);
}
}
//
// Clear data?
//
else if (fClear) {
Status = NtPulseEvent (hClearEvent, NULL);
if (!NT_SUCCESS(Status)) {
fprintf (pfLog, "WSTDUMP: ClearDumpInfo () - NtPulseEvent() "
"failed for CLEAR event - %lx\n", Status);
exit (1);
}
}
//
// Wait for the DONE event..
//
Sleep (500);
Status = NtWaitForSingleObject (hDoneEvent, FALSE, NULL);
if (!NT_SUCCESS(Status)) {
fprintf (pfLog, "WSTDUMP: ClearDumpInfo () - NtWaitForSingleObject() "
"failed for DONE event - %lx\n", Status);
exit (1);
}
} /* ClearDumpInfo() */
/*++
Dump dialog procedure -- exported to windows.
Allows user to change defaults: dump, clear, and ".dmp" as dump
file extension.
Input:
Messages from windows:
- WM_INITDIALOG - initialize dialog box
- WM_COMMAND - user input received
Output:
returns TRUE if message processed, false otherwise
SideEffects:
global flags fDump and fClear may be altered
--*/
INT_PTR APIENTRY DialogProc(HWND hDlg, UINT wMesg, WPARAM wParam, LPARAM lParam)
{
HICON hIcon;
lParam; //Avoid Compiler warnings
switch (wMesg) {
case WM_CREATE:
hIcon = LoadIcon ((HINSTANCE)hDlg, "WSTDUMP.ICO");
SetClassLongPtr (hDlg, GCLP_HICON, (LONG_PTR)hIcon);
return TRUE;
case WM_INITDIALOG:
CheckDlgButton(hDlg, ID_DUMP, fDump);
CheckDlgButton(hDlg, ID_CLEAR, fClear);
CheckDlgButton(hDlg, ID_PAUSE, fPause);
return TRUE;
case WM_COMMAND:
switch (wParam) {
case IDOK:
if (fDump) {
SetWindowText(hDlg, "Dumping Data..");
}
else if (fClear) {
SetWindowText(hDlg, "Clearing Data..");
}
else if (fPause) {
SetWindowText(hDlg, "Stopping WST..");
}
ClearDumpInfo();
SetWindowText(hDlg, "WST Dump Utility");
return (TRUE);
case IDEXIT:
EndDialog(hDlg, IDEXIT);
return (TRUE);
case ID_DUMP:
fDump = TRUE;
fPause = FALSE;
fClear = FALSE;
CheckDlgButton(hDlg, ID_DUMP, fDump);
CheckDlgButton(hDlg, ID_PAUSE, fPause);
CheckDlgButton(hDlg, ID_CLEAR, fClear);
return (TRUE);
case ID_CLEAR:
fClear = TRUE;
fPause = FALSE;
fDump = FALSE;
CheckDlgButton(hDlg, ID_CLEAR, fClear);
CheckDlgButton(hDlg, ID_PAUSE, fPause);
CheckDlgButton(hDlg, ID_DUMP, fDump);
return (TRUE);
case ID_PAUSE:
fPause = TRUE;
fClear = FALSE;
fDump = FALSE;
CheckDlgButton(hDlg, ID_PAUSE, fPause);
CheckDlgButton(hDlg, ID_CLEAR, fClear);
CheckDlgButton(hDlg, ID_DUMP, fDump);
return (TRUE);
}
}
return (FALSE); /* did not process a message */
} /* DialogProc() */