Windows-Server-2003/termsrv/common/trace/ezippy/zippywindow.cpp

1815 lines
42 KiB
C++

/*++
Copyright (c) 2000 Microsoft Corporation
Module Name:
Zippy Main Window
Abstract:
This class implements the main window for zippy as well as controlling
its child windows.
Author:
Marc Reyhner 8/28/2000
--*/
#include "stdafx.h"
#include "ZippyWindow.h"
#include "resource.h"
#include "eZippy.h"
#include "richedit.h"
#include "ModalDialog.h"
#include "TraceManager.h"
#include "OptionsDialog.h"
BOOL CZippyWindow::gm_Inited = FALSE;
ATOM CZippyWindow::gm_Atom = NULL;
UINT CZippyWindow::gm_FindMessageStringMsg = 0;
static DWORD CALLBACK SaveCallback(DWORD_PTR dwCookie,LPBYTE pbBuff,LONG cb,LONG *pcb);
#define TRC_PROC_FMT _T("%04.4lx")
#define SAVE_FILE_TYPE _T("Text Files (*.txt)\0*.txt\0")
#define SAVE_FILE_EXTENSION _T("txt")
#define SAVE_CONF_FILE_TYPE _T("Trace Configuration Files (*.tcf)\0*.tcf\0")
#define SAVE_CONF_FILE_EXTENSION _T("tcf")
#define NUM_COLORS 15
#define APPENDMUTEXNAME _T("Local\\MicrosoftTerminalServerTraceViewerAppendMutex")
#define ZIPPY_WINDOW_POS_VALUE _T("WindowPosition")
#define WINDOW_DEF_TOP 50
#define WINDOW_DEF_BOTTOM 530
#define WINDOW_DEF_RIGHT 690
#define WINDOW_DEF_LEFT 50
// We use an 80 character buffer for find
// and replace operations.
#define FIND_REPLACE_BUFFER_SIZE 80
// This list of colors we cycle through. Note if you change this
// list you need to update NUM_COLORS to the new count.
static COLORREF colors[NUM_COLORS] = {
RGB(153,51,0), /* Brown */
RGB(0,51,102), /* Dark Teal */
RGB(51,51,153), /* Indigo */
RGB(128,0,0), /* Dark Red */
RGB(255,102,0), /* Orange */
RGB(0,128,0), /* Green */
RGB(0,0,255), /* Blue */
RGB(255,0,0), /* Red */
RGB(51,204,204),/* Acqua */
RGB(128,0,128), /* Violet */
RGB(255,0,255), /* Pink */
RGB(255,255,0), /* Yellow */
RGB(0,255,0), /* Bright Green */
RGB(0,255,255), /* Turquoise */
RGB(204,153,255)/* Lavender */
};
//
// *** Public Class Members ***
//
CZippyWindow::CZippyWindow(
)
/*++
Routine Description:
The constructor simply initializes the class variables.
Arguments:
None
Return value:
None
--*/
{
m_bIsTracing = TRUE;
m_bIsStoringTraceData = FALSE;
ZeroMemory(m_threadHistory,sizeof(m_threadHistory));
m_nextThreadIndex = 0;
m_nextThreadColor = 0;
m_lastProcessId = 0;
m_LastLogEndedInNewLine = TRUE;
m_hWnd = NULL;
m_hStatusWnd = NULL;
m_hControlWnd = NULL;
m_hWndFindReplace = NULL;
m_lpSavedOutputStart = NULL;
m_lpSavedOutputTail = NULL;
ZeroMemory(&m_FindReplace,sizeof(m_FindReplace));
ZeroMemory(m_SaveFile,sizeof(m_SaveFile));
ZeroMemory(m_SaveConfFile,sizeof(m_SaveConfFile));
ZeroMemory(m_LoadConfFile,sizeof(m_LoadConfFile));
}
CZippyWindow::~CZippyWindow(
)
/*++
Routine Description:
Cleans up any dynamicly allocated memory,
Arguments:
None
Return value:
None
--*/
{
if (m_FindReplace.lpstrFindWhat) {
HeapFree(GetProcessHeap(),0,m_FindReplace.lpstrFindWhat);
}
if (m_FindReplace.lpstrReplaceWith) {
HeapFree(GetProcessHeap(),0,m_FindReplace.lpstrReplaceWith);
}
}
DWORD CZippyWindow::Create(
IN CTraceManager *rTracer
)
/*++
Routine Description:
Actually creates the zippy window.
Arguments:
rTracer - A pointer to the trace manager
Return value:
0 - Success
Non zero - An error occurred creating the window
--*/
{
DWORD dwResult;
DWORD dwWindowStyleEx;
DWORD dwWindowStyle;
TCHAR wndTitle[MAX_STR_LEN];
RECT wndRect;
m_rTracer = rTracer;
if (!gm_Inited) {
dwResult = _InitClassStaticMembers();
if (dwResult != ERROR_SUCCESS) {
return dwResult;
}
}
m_hAppendMutex = CreateMutex(NULL,FALSE,APPENDMUTEXNAME);
m_FindReplace.lStructSize = sizeof(m_FindReplace);
m_FindReplace.lpstrFindWhat = (LPTSTR)HeapAlloc(GetProcessHeap(),
HEAP_ZERO_MEMORY,FIND_REPLACE_BUFFER_SIZE*sizeof(TCHAR));
if (!m_FindReplace.lpstrFindWhat) {
return GetLastError();
}
m_FindReplace.lpstrReplaceWith = (LPTSTR)HeapAlloc(GetProcessHeap(),
HEAP_ZERO_MEMORY,FIND_REPLACE_BUFFER_SIZE*sizeof(TCHAR));
if (!m_FindReplace.lpstrReplaceWith) {
return GetLastError();
}
LoadStringSimple(IDS_ZIPPYWINDOWTITLE,wndTitle);
GetSavedWindowPos(&wndRect);
dwWindowStyleEx = WS_EX_WINDOWEDGE;
dwWindowStyle = WS_OVERLAPPEDWINDOW|WS_CLIPSIBLINGS|WS_VISIBLE;
m_hWnd = CreateWindowEx(dwWindowStyleEx, (LPTSTR)gm_Atom, wndTitle,
dwWindowStyle, wndRect.left, wndRect.top, wndRect.right,
wndRect.bottom,NULL,NULL,g_hInstance,this);
if (!m_hWnd) {
return GetLastError();
}
return ERROR_SUCCESS;
}
VOID
CZippyWindow::AppendTextToWindow(
IN DWORD processID,
IN LPCTSTR text,
IN UINT len
)
/*++
Routine Description:
Appends new trace data to the end of the rich edit contrl
Arguments:
processID - Process ID of the process sending the debug string
text - The data sent via OutputDebugString
len - Length of the data
Return value:
None
--*/
{
UINT controlTextLength;
CHARRANGE newSel;
BOOL computeColor;
BOOL setNewColor;
CHARFORMAT newFormat;
LPSAVEDOUTPUT lpSave;
if (!m_bIsTracing) {
return;
}
WaitForSingleObject(m_hAppendMutex,INFINITE);
if (m_bIsStoringTraceData) {
// This is kinda sketchy but what we do is to allocate room for the string
// at the end of the structure. There shouldn't be any alignment problems
// since we need to align on a short and the structure has no items
// to get that off.
lpSave = (LPSAVEDOUTPUT)HeapAlloc(GetProcessHeap(),0,sizeof(SAVEDOUTPUT) +
(sizeof(TCHAR) * (len+1)));
if (!lpSave) {
// eom error?
goto CLEANUP_AND_EXIT;
}
lpSave->procID = processID;
lpSave->text = (LPTSTR)((BYTE)lpSave + sizeof(SAVEDOUTPUT));
_tcscpy(lpSave->text,text);
lpSave->len = len;
lpSave->next = NULL;
if (!m_lpSavedOutputTail) {
m_lpSavedOutputStart = lpSave;
} else {
m_lpSavedOutputTail->next = lpSave;
}
m_lpSavedOutputTail = lpSave;
goto CLEANUP_AND_EXIT;
}
if (m_lastProcessId != processID ||
m_LastLogEndedInNewLine) {
computeColor = TRUE;
} else {
computeColor = FALSE;
}
setNewColor = ComputeNewColor(processID,text,len,&newFormat);
m_LastLogEndedInNewLine = (text[len-1] == '\n') ? TRUE : FALSE;
m_lastProcessId = processID;
controlTextLength = (UINT)SendMessage(m_hControlWnd,WM_GETTEXTLENGTH,0,0);
newSel.cpMin = controlTextLength;
newSel.cpMax = controlTextLength+1;
// set the new text
SendMessage(m_hControlWnd,EM_EXSETSEL,0,(LPARAM)&newSel);
if (setNewColor) {
SendMessage(m_hControlWnd,EM_SETCHARFORMAT,SCF_SELECTION,(LPARAM)&newFormat);
}
SendMessage(m_hControlWnd,EM_REPLACESEL,0,(LPARAM)text);
CLEANUP_AND_EXIT:
ReleaseMutex(m_hAppendMutex);
}
VOID
CZippyWindow::LoadConfFile(
IN LPTSTR confFile
)
/*++
Routine Description:
This sets the tracing configuration using the given file
Arguments:
confFile - File containing the tracing configuration
Return value:
None
--*/
{
_tcscpy(m_LoadConfFile,confFile);
DoLoadConfInternal();
}
BOOL
CZippyWindow::IsDialogMessage(
IN LPMSG lpMsg
)
/*++
Routine Description:
This calls IsDialogMessage on any non modal dialogs that this window
is hosting to see if the message is for them
Arguments:
lpMsg - Message to check if it is a dialog message
Return value:
TRUE - The message did belong to a dialog
FALSE - The message did not belong to a dialog
--*/
{
if (IsWindow(m_hWndFindReplace)) {
// The :: below is necessary to make it use the Win32 function
// not our method
return ::IsDialogMessage(m_hWndFindReplace,lpMsg);
}
return FALSE;
}
INT WINAPI
CZippyWindow::TranslateAccelerator(
IN HACCEL hAccTable,
IN LPMSG lpMsg
)
/*++
Routine Description:
This calls the win32 TranslateAccelerator to determine
if the given message is an accelerator for this window
Arguments:
hAccTable - Accelerator table to use
lpMsg - Message to check
Return value:
See Win32 TranslateAccelerator documentation
--*/
{
// :: Necessary to get the win32 call.
return ::TranslateAccelerator(m_hWnd,hAccTable,lpMsg);
}
//
// *** Private Class Members ***
//
// static members
DWORD
CZippyWindow::_InitClassStaticMembers(
)
/*++
Routine Description:
Creates the window class for zippy and registers
for the FINDMSGSTRING windows message
Arguments:
None
Return value:
0 - Success
Non zero - Win32 error code
--*/
{
WNDCLASS wndClass;
HMODULE hLibrary;
// We want to load RichEdit for the lifetime of our app.
hLibrary = LoadLibrary(_T("Riched20.dll"));
if (!hLibrary) {
return GetLastError();
}
ZeroMemory(&wndClass,sizeof(wndClass));
wndClass.style = CS_PARENTDC;
wndClass.lpfnWndProc = _WindowProc;
wndClass.hInstance = g_hInstance;
wndClass.hIcon = (HICON)LoadImage(g_hInstance,MAKEINTRESOURCE(IDI_MAINFRAME),
IMAGE_ICON,0,0,LR_SHARED);
wndClass.hbrBackground = (HBRUSH)COLOR_WINDOWFRAME;
wndClass.lpszMenuName = MAKEINTRESOURCE(IDR_MAINMENU);
wndClass.lpszClassName = _T("ZippyWindowClass");
gm_Atom = RegisterClass(&wndClass);
if (!gm_Atom) {
return GetLastError();
}
gm_FindMessageStringMsg = RegisterWindowMessage(FINDMSGSTRING);
if (!gm_FindMessageStringMsg) {
return GetLastError();
}
gm_Inited = TRUE;
return ERROR_SUCCESS;
}
LRESULT CALLBACK
CZippyWindow::_WindowProc(
IN HWND hWnd,
IN UINT uMsg,
IN WPARAM wParam,
IN LPARAM lParam)
/*++
Routine Description:
Static version of the window proc. On WM_CREATE it calls OnCreate,
otherwise it calls the non-static window proc
Arguments:
See Win32 Window Proc docs
Return value:
Message specific. See individual handlers for detail.
--*/
{
CZippyWindow *theClass;
if (uMsg == WM_CREATE) {
SetLastError(0);
theClass = (CZippyWindow *)((LPCREATESTRUCT)lParam)->lpCreateParams;
SetWindowLongPtr(hWnd,GWLP_USERDATA,(LONG_PTR)theClass);
if (GetLastError()) {
return -1;
}
return theClass->OnCreate(hWnd);
}
theClass = (CZippyWindow*)GetWindowLongPtr(hWnd,GWLP_USERDATA);
if (theClass) {
return theClass->WindowProc(hWnd,uMsg,wParam,lParam);
} else {
return DefWindowProc(hWnd,uMsg,wParam,lParam);
}
}
LRESULT CALLBACK
CZippyWindow::WindowProc(
IN HWND hWnd,
IN UINT uMsg,
IN WPARAM wParam,
IN LPARAM lParam)
/*++
Routine Description:
Non-static window proc. Either calls the default window proc or
refers to the individual message handlers
Arguments:
See Win32 Window Proc docs
Return value:
Message specific. See individual handlers for detail.
--*/
{
LRESULT retCode = 0;
switch (uMsg) {
case WM_COMMAND:
OnCommand(wParam,lParam);
break;
case WM_SETFOCUS:
OnSetFocus();
break;
case WM_SIZE:
OnSize(LOWORD(lParam),HIWORD(lParam));
break;
case WM_INITMENUPOPUP:
OnInitMenuPopup(wParam,lParam);
break;
case WM_MENUSELECT:
OnMenuSelect(wParam,lParam);
break;
case WM_CLOSE:
OnClose();
break;
case WM_DESTROY:
OnDestroy();
break;
default:
if (uMsg == gm_FindMessageStringMsg) {
OnFindMessageString(lParam);
} else {
retCode = DefWindowProc(hWnd,uMsg,wParam,lParam);
}
break;
}
return retCode;
}
LRESULT
CZippyWindow::OnCreate(
IN HWND hWnd
)
/*++
Routine Description:
Creates the child windows and sets their initial parameters
Arguments:
hWnd - Pointer to the new main window
Return value:
0 - Window was created
-1 - Error occurred
--*/
{
DWORD dwStyle;
CHARFORMAT charFormat;
TCHAR readyString[MAX_STR_LEN];
dwStyle = WS_CHILD|WS_VISIBLE|WS_VSCROLL|WS_HSCROLL|ES_SUNKEN|
ES_MULTILINE|ES_LEFT|ES_AUTOHSCROLL|ES_AUTOVSCROLL|ES_NOHIDESEL;
m_hControlWnd = CreateWindow(RICHEDIT_CLASS,_T(""),
dwStyle,0,0,0,0,hWnd,NULL,g_hInstance,NULL);
if (!m_hControlWnd) {
return -1;
}
dwStyle = SBARS_SIZEGRIP|WS_CHILD|WS_VISIBLE;
m_hStatusWnd = CreateWindow(STATUSCLASSNAME,NULL,dwStyle,0,0,0,0,hWnd,NULL,
g_hInstance,NULL);
if (!m_hStatusWnd) {
return -1;
}
LoadStringSimple(IDS_STATUSBARREADY,readyString);
SendMessage(m_hStatusWnd,SB_SETTEXT,0|SBT_NOBORDERS,(LPARAM)readyString);
charFormat.cbSize = sizeof(charFormat);
charFormat.dwMask = CFM_FACE|CFM_SIZE;
charFormat.yHeight = ZIPPY_FONT_SIZE*20;
_tcscpy(charFormat.szFaceName,ZIPPY_FONT);
// 4 billion characters should be a large enough limit...
SendMessage(m_hControlWnd,EM_EXLIMITTEXT,0,0xFFFFFFFF);
SendMessage(m_hControlWnd,EM_SETCHARFORMAT,SCF_ALL,(LPARAM)&charFormat);
SendMessage(m_hControlWnd,EM_SETMODIFY,FALSE,0);
SendMessage(m_hControlWnd,EM_EMPTYUNDOBUFFER,0,0);
return 0;
}
VOID
CZippyWindow::OnMenuSelect(
IN WPARAM wParam,
IN LPARAM lParam
)
/*++
Routine Description:
Sets the help string in the status bar for the selected menu.
Arguments:
wParam - menu itemid (LOWORD) and flags (HIWORD)
lParam - menu handle
Return value:
None
--*/
{
UINT item;
UINT flags;
HMENU hMenu;
TCHAR statusMessage[MAX_STR_LEN];
item = LOWORD(wParam);
flags = HIWORD(wParam);
hMenu = (HMENU)lParam;
if (!item && flags == 0xFFFF) {
// the menu was closed. Go back to the ready string.
LoadStringSimple(IDS_STATUSBARREADY,statusMessage);
SendMessage(m_hStatusWnd,SB_SETTEXT,0|SBT_NOBORDERS,(LPARAM)statusMessage);
return;
}
if (flags & MF_POPUP) {
statusMessage[0] = 0;
} else if (!LoadStringSimple(item,statusMessage)) {
// if we can't find the help string use the empty string.
statusMessage[0] = 0;
}
SendMessage(m_hStatusWnd,SB_SETTEXT,0|SBT_NOBORDERS,(LPARAM)statusMessage);
}
VOID
CZippyWindow::OnSize(
IN INT width,
IN INT height
)
/*++
Routine Description:
Resizes client windows to reflect the new size of the main window
Arguments:
width - New width of the client area
height - New height of the client area
Return value:
None
--*/
{
RECT statusBarArea;
UINT statusBarHeight;
RECT wndRect;
if (!(width==0&&height==0)) {
if (GetWindowRect(m_hWnd,&wndRect)) {
SaveWindowPos(&wndRect);
}
}
if (IsWindowVisible(m_hStatusWnd)) {
GetWindowRect(m_hStatusWnd,&statusBarArea);
statusBarHeight = statusBarArea.bottom - statusBarArea.top;
SetWindowPos(m_hControlWnd,NULL,0,0,width,height-statusBarHeight,SWP_NOZORDER);
// the status bar autosizes. We just need to tell it that it should
SetWindowPos(m_hStatusWnd,NULL,0,0,0,0,SWP_NOZORDER);
} else {
SetWindowPos(m_hControlWnd,NULL,0,0,width,height,SWP_NOZORDER);
}
}
VOID
CZippyWindow::OnSetFocus(
)
/*++
Routine Description:
When we get focus we kick it down to the rich edit control
Arguments:
None
Return value:
None
--*/
{
SetFocus(m_hControlWnd);
}
VOID
CZippyWindow::OnInitMenuPopup(
IN WPARAM wParam,
IN LPARAM lParam
)
/*++
Routine Description:
When the user opens the menus we need to specify
which are disabled and which are checked. Note that
the menu id's are hard coded. I've commented which
corespond to which for the switch statement
Arguments:
wParam - The menu handle
lParam - (loword) the menu item id
Return value:
None
--*/
{
HMENU hMenu;
WORD item;
item = LOWORD(lParam);
hMenu = (HMENU)wParam;
switch (item) {
case 1: // Edit Menu
UINT canUndo;
UINT canRedo;
UINT cutCopyEnabled;
UINT pasteEnabled;
UINT selectAllEnabled;
UINT findEnabled;
UINT findNextEnabled;
UINT replaceEnabled;
LRESULT textLength;
CHARRANGE selRegion;
if (SendMessage(m_hControlWnd,EM_CANUNDO,0,0)) {
canUndo = MF_ENABLED;
} else {
canUndo = MF_GRAYED;
}
if (SendMessage(m_hControlWnd,EM_CANREDO,0,0)) {
canRedo = MF_ENABLED;
} else {
canRedo = MF_GRAYED;
}
textLength = SendMessage(m_hControlWnd,WM_GETTEXTLENGTH,0,0);
if (textLength == 0) {
selectAllEnabled = MF_GRAYED;
findEnabled = MF_GRAYED;
findNextEnabled = MF_GRAYED;
replaceEnabled = MF_GRAYED;
} else {
selectAllEnabled = MF_ENABLED;
findEnabled = MF_ENABLED;
replaceEnabled = MF_ENABLED;
if (m_FindReplace.lpstrFindWhat[0] != 0) {
findNextEnabled = MF_ENABLED;
} else {
findNextEnabled = MF_GRAYED;
}
}
SendMessage(m_hControlWnd,EM_EXGETSEL,0,(LPARAM)&selRegion);
if (selRegion.cpMax == selRegion.cpMin) {
cutCopyEnabled = MF_GRAYED;
} else {
cutCopyEnabled = MF_ENABLED;
// override select all since they selected the next character
// to be typed
selectAllEnabled = MF_ENABLED;
}
if (SendMessage(m_hControlWnd,EM_CANPASTE,0,0)) {
pasteEnabled = MF_ENABLED;
} else {
pasteEnabled = MF_GRAYED;
}
EnableMenuItem(hMenu,ID_EDIT_UNDO,MF_BYCOMMAND|canUndo);
EnableMenuItem(hMenu,ID_EDIT_REDO,MF_BYCOMMAND|canRedo);
EnableMenuItem(hMenu,ID_EDIT_CUT,MF_BYCOMMAND|cutCopyEnabled);
EnableMenuItem(hMenu,ID_EDIT_COPY,MF_BYCOMMAND|cutCopyEnabled);
EnableMenuItem(hMenu,ID_EDIT_PASTE,MF_BYCOMMAND|pasteEnabled);
EnableMenuItem(hMenu,ID_EDIT_SELECTALL,MF_BYCOMMAND|selectAllEnabled);
EnableMenuItem(hMenu,ID_EDIT_FIND,MF_BYCOMMAND|findEnabled);
EnableMenuItem(hMenu,ID_EDIT_FINDNEXT,MF_BYCOMMAND|findNextEnabled);
EnableMenuItem(hMenu,ID_EDIT_REPLACE,MF_BYCOMMAND|replaceEnabled);
break;
case 2: // View Menu
UINT statusBarChecked;
if (IsWindowVisible(m_hStatusWnd)) {
statusBarChecked = MF_CHECKED;
} else {
statusBarChecked = MF_UNCHECKED;
}
CheckMenuItem(hMenu,ID_VIEW_STATUSBAR,MF_BYCOMMAND|statusBarChecked);
break;
case 3: // Monitoring Menu
UINT startActivated;
UINT stopActivated;
if (m_bIsTracing) {
startActivated = MF_GRAYED;
stopActivated = MF_ENABLED;
} else {
startActivated = MF_ENABLED;
stopActivated = MF_GRAYED;
}
EnableMenuItem(hMenu,ID_MONITORING_START,MF_BYCOMMAND|startActivated);
EnableMenuItem(hMenu,ID_MONITORING_STOP,MF_BYCOMMAND|stopActivated);
// record is activated when stop is.
EnableMenuItem(hMenu,ID_MONITORING_RECORD,MF_BYCOMMAND|stopActivated);
break;
}
}
VOID
CZippyWindow::OnFindMessageString(
IN LPARAM lParam
)
/*++
Routine Description:
This handles a message from the find/replace
dialog when a user hits a button
Arguments:
lParam - LPFINDREPLACE struct for the dialog
Return value:
None
--*/
{
LPFINDREPLACE lpFindReplace;
lpFindReplace = (LPFINDREPLACE)lParam;
if (lpFindReplace->Flags & FR_DIALOGTERM) {
// the dialog is closing
m_hWndFindReplace = NULL;
} else if (lpFindReplace->Flags & FR_FINDNEXT) {
// the user selected find
DoFindNext(lpFindReplace);
} else if (lpFindReplace->Flags & FR_REPLACE) {
DoReplace(lpFindReplace);
} else if (lpFindReplace->Flags & FR_REPLACEALL) {
DoReplaceAll(lpFindReplace);
}
}
VOID
CZippyWindow::OnClose(
)
/*++
Routine Description:
When we receive a close window request we prompt
the user to sace the trace if they have changed it.
Arguments:
None
Return value:
None
--*/
{
INT result;
TCHAR dlgMessage[MAX_STR_LEN];
TCHAR dlgTitle[MAX_STR_LEN];
if (SendMessage(m_hControlWnd,EM_GETMODIFY,0,0)) {
LoadStringSimple(IDS_SAVEFILEPROMPT,dlgMessage);
LoadStringSimple(IDS_ZIPPYWINDOWTITLE,dlgTitle);
result = MessageBox(m_hWnd,dlgMessage,dlgTitle,MB_YESNOCANCEL|MB_ICONQUESTION);
switch (result) {
case IDYES:
OnSave();
if (SendMessage(m_hControlWnd,EM_GETMODIFY,0,0)) {
// if there was an error saving we will try again.
PostMessage(m_hWnd,WM_CLOSE,0,0);
return;
}
case IDNO:
DestroyWindow(m_hWnd);
break;
}
} else {
DestroyWindow(m_hWnd);
}
}
VOID
CZippyWindow::OnDestroy(
)
/*++
Routine Description:
When the main window exits we halt the message loop
Arguments:
None
Return value:
None
--*/
{
// If we don't clean up the tracing stuff here. There is a long
// delay exiting for some reason.
CTraceManager::_CleanupTraceManager();
PostQuitMessage(0);
}
VOID
CZippyWindow::OnCommand(
IN WPARAM wParam,
IN LPARAM lParam
)
/*++
Routine Description:
Below is WM_COMMAND and all the handler functions.
The individual handler funcitons are not
that interesting so I didn't individually comment them.
Arguments:
wParam - (loword) command the user selected
lParam - not used but it is the control
Return value:
None
--*/
{
WORD command;
command = LOWORD(wParam);
switch (command) {
case ID_FILE_SAVE:
OnSave();
break;
case ID_FILE_SAVEAS:
OnSaveAs();
break;
case ID_FILE_LOADCONF:
OnLoadConfiguration();
break;
case ID_FILE_SAVECONF:
OnSaveConfiguration();
break;
case ID_FILE_SAVECONFAS:
OnSaveConfigurationAs();
break;
case ID_FILE_EXIT:
OnExit();
break;
case ID_EDIT_UNDO:
OnUndo();
break;
case ID_EDIT_REDO:
OnRedo();
break;
case ID_EDIT_CUT:
OnCut();
break;
case ID_EDIT_COPY:
OnCopy();
break;
case ID_EDIT_PASTE:
OnPaste();
break;
case ID_EDIT_FIND:
OnFind();
break;
case ID_EDIT_FINDNEXT:
OnFindNext();
break;
case ID_EDIT_REPLACE:
OnReplace();
break;
case ID_EDIT_SELECTALL:
OnSelectAll();
break;
case ID_VIEW_STATUSBAR:
OnChangeStatusBar();
break;
case ID_MONITORING_START:
OnStartTracing();
break;
case ID_MONITORING_STOP:
OnStopTracing();
break;
case ID_MONITORING_RECORD:
OnRecordTracing();
break;
case ID_MONITORING_CLEARSCREEN:
OnClearScreen();
break;
case ID_MONITORING_RESETTRACEFILES:
OnResetTraceFiles();
break;
case ID_MONITORING_PREFERENCES:
OnPreferences();
break;
case ID_HELP_ABOUTEZIPPY:
OnAbout();
break;
}
}
VOID CZippyWindow::OnSave()
{
if (m_SaveFile[0] == 0) {
// if we don't have a file name do the
// Save As version
OnSaveAs();
} else {
DoSaveInternal();
}
}
VOID CZippyWindow::OnSaveAs()
{
OPENFILENAME fileInfo;
BOOL bResult;
ZeroMemory(&fileInfo,sizeof(fileInfo));
fileInfo.lStructSize = sizeof(fileInfo);
fileInfo.hwndOwner = m_hWnd;
fileInfo.hInstance = g_hInstance;
fileInfo.lpstrFilter = SAVE_FILE_TYPE;
fileInfo.lpstrFile = m_SaveFile;
fileInfo.nMaxFile = MAX_STR_LEN;
fileInfo.Flags = OFN_OVERWRITEPROMPT;
fileInfo.lpstrDefExt = SAVE_FILE_EXTENSION;
bResult = GetSaveFileName(&fileInfo);
if (!bResult) {
return;
}
DoSaveInternal();
}
VOID CZippyWindow::OnLoadConfiguration()
{
OPENFILENAME fileInfo;
BOOL bResult;
ZeroMemory(&fileInfo,sizeof(fileInfo));
fileInfo.lStructSize = sizeof(fileInfo);
fileInfo.hwndOwner = m_hWnd;
fileInfo.hInstance = g_hInstance;
fileInfo.lpstrFilter = SAVE_CONF_FILE_TYPE;
fileInfo.lpstrFile = m_LoadConfFile;
fileInfo.nMaxFile = MAX_STR_LEN;
fileInfo.Flags = OFN_FILEMUSTEXIST;
fileInfo.lpstrDefExt = SAVE_CONF_FILE_EXTENSION;
bResult = GetOpenFileName(&fileInfo);
if (!bResult) {
return;
}
DoLoadConfInternal();
}
VOID CZippyWindow::OnSaveConfiguration()
{
if (m_SaveConfFile[0] == 0) {
// if we don't have a file name do the
// Save As version
OnSaveConfigurationAs();
} else {
DoSaveConfInternal();
}
}
VOID CZippyWindow::OnSaveConfigurationAs()
{
OPENFILENAME fileInfo;
BOOL bResult;
ZeroMemory(&fileInfo,sizeof(fileInfo));
fileInfo.lStructSize = sizeof(fileInfo);
fileInfo.hwndOwner = m_hWnd;
fileInfo.hInstance = g_hInstance;
fileInfo.lpstrFilter = SAVE_CONF_FILE_TYPE;
fileInfo.lpstrFile = m_SaveConfFile;
fileInfo.nMaxFile = MAX_STR_LEN;
fileInfo.Flags = OFN_OVERWRITEPROMPT;
fileInfo.lpstrDefExt = SAVE_CONF_FILE_EXTENSION;
bResult = GetSaveFileName(&fileInfo);
if (!bResult) {
return;
}
DoSaveConfInternal();
}
VOID CZippyWindow::OnExit()
{
PostMessage(m_hWnd,WM_CLOSE,0,0);
}
// All the edit menu commands. Except for select all they just call the
// corresponding message in the rich edit control. Select all has to
// manually set the selection
VOID CZippyWindow::OnUndo()
{
SendMessage(m_hControlWnd,WM_UNDO,0,0);
}
VOID CZippyWindow::OnRedo()
{
SendMessage(m_hControlWnd,EM_REDO,0,0);
}
VOID CZippyWindow::OnCut()
{
SendMessage(m_hControlWnd,WM_CUT,0,0);
}
VOID CZippyWindow::OnCopy()
{
SendMessage(m_hControlWnd,WM_COPY,0,0);
}
VOID CZippyWindow::OnPaste()
{
SendMessage(m_hControlWnd,WM_PASTE,0,0);
}
VOID CZippyWindow::OnSelectAll()
{
CHARRANGE selection;
selection.cpMin = 0;
selection.cpMax = -1;
SendMessage(m_hControlWnd,EM_EXSETSEL,0,(LPARAM)&selection);
}
VOID CZippyWindow::OnFind()
{
CHARRANGE currentSel;
TEXTRANGE textRange;
if (IsWindow(m_hWndFindReplace) && !m_bIsFindNotReplace) {
// If they were in a replace dialog we destroy it and then
// start over with a find dialog
DestroyWindow(m_hWndFindReplace);
m_hWndFindReplace = NULL;
}
if (!IsWindow(m_hWndFindReplace)) {
SendMessage(m_hControlWnd,EM_EXGETSEL,0,(LPARAM)&currentSel);
textRange.chrg.cpMin = currentSel.cpMin;
if (currentSel.cpMax - currentSel.cpMin >= FIND_REPLACE_BUFFER_SIZE) {
textRange.chrg.cpMax = currentSel.cpMin + FIND_REPLACE_BUFFER_SIZE-1;
} else {
textRange.chrg.cpMax = currentSel.cpMax;
}
textRange.lpstrText = m_FindReplace.lpstrFindWhat;
SendMessage(m_hControlWnd,EM_GETTEXTRANGE,0,(LPARAM)&textRange);
m_bIsFindNotReplace = TRUE;
m_FindReplace.hwndOwner = m_hWnd;
m_FindReplace.hInstance = g_hInstance;
m_FindReplace.Flags = FR_DOWN|FR_HIDEUPDOWN;
m_FindReplace.wFindWhatLen = FIND_REPLACE_BUFFER_SIZE;
m_hWndFindReplace = FindText(&m_FindReplace);
} else {
SetActiveWindow(m_hWndFindReplace);
}
}
VOID CZippyWindow::OnFindNext()
{
DoFindNext(&m_FindReplace);
}
VOID CZippyWindow::OnReplace()
{
CHARRANGE currentSel;
TEXTRANGE textRange;
if (IsWindow(m_hWndFindReplace) && m_bIsFindNotReplace) {
// If they were in a replace dialog we destroy it and then
// start over with a find dialog
DestroyWindow(m_hWndFindReplace);
m_hWndFindReplace = NULL;
}
if (!IsWindow(m_hWndFindReplace)) {
SendMessage(m_hControlWnd,EM_EXGETSEL,0,(LPARAM)&currentSel);
textRange.chrg.cpMin = currentSel.cpMin;
if (currentSel.cpMax - currentSel.cpMin >= FIND_REPLACE_BUFFER_SIZE) {
textRange.chrg.cpMax = currentSel.cpMin + FIND_REPLACE_BUFFER_SIZE-1;
} else {
textRange.chrg.cpMax = currentSel.cpMax;
}
textRange.lpstrText = m_FindReplace.lpstrFindWhat;
SendMessage(m_hControlWnd,EM_GETTEXTRANGE,0,(LPARAM)&textRange);
m_bIsFindNotReplace = FALSE;
m_FindReplace.hwndOwner = m_hWnd;
m_FindReplace.hInstance = g_hInstance;
m_FindReplace.Flags = FR_DOWN;
m_FindReplace.wFindWhatLen = FIND_REPLACE_BUFFER_SIZE;
m_FindReplace.wReplaceWithLen = FIND_REPLACE_BUFFER_SIZE;
m_hWndFindReplace = ReplaceText(&m_FindReplace);
} else {
SetActiveWindow(m_hWndFindReplace);
}
}
VOID CZippyWindow::OnChangeStatusBar()
{
RECT clientRect;
if (IsWindowVisible(m_hStatusWnd)) {
ShowWindow(m_hStatusWnd,SW_HIDE);
} else {
ShowWindow(m_hStatusWnd,SW_SHOW);
}
// we do this to make the client windows resize themselves
// around the status bar
GetClientRect(m_hWnd,&clientRect);
OnSize(clientRect.right,clientRect.bottom);
}
VOID CZippyWindow::OnStartTracing()
{
m_bIsTracing = TRUE;
}
VOID CZippyWindow::OnStopTracing()
{
m_bIsTracing = FALSE;
}
VOID CZippyWindow::OnRecordTracing()
{
CModalOkDialog recordDialog;
LPSAVEDOUTPUT lpTemp;
m_bIsStoringTraceData = TRUE;
recordDialog.DoModal(MAKEINTRESOURCE(IDD_RECORDTRACE),m_hWnd);
WaitForSingleObject(m_hAppendMutex,INFINITE);
m_bIsStoringTraceData = FALSE;
while (m_lpSavedOutputStart) {
AppendTextToWindow(m_lpSavedOutputStart->procID,
m_lpSavedOutputStart->text,m_lpSavedOutputStart->len);
lpTemp = m_lpSavedOutputStart;
m_lpSavedOutputStart = m_lpSavedOutputStart->next;
HeapFree(GetProcessHeap(),0,lpTemp);
}
m_lpSavedOutputTail = NULL;
ReleaseMutex(m_hAppendMutex);
}
VOID CZippyWindow::OnClearScreen()
{
TCHAR dlgTitle[MAX_STR_LEN];
TCHAR dlgMessage[MAX_STR_LEN];
LoadStringSimple(IDS_CLEARCONFIRMTITLE,dlgTitle);
LoadStringSimple(IDS_CLEARCONFIRMMESSAGE,dlgMessage);
if (IDYES != MessageBox(m_hWnd,dlgMessage,dlgTitle,MB_YESNO)) {
return;
}
OnSelectAll();
SendMessage(m_hControlWnd, EM_REPLACESEL,FALSE,(LPARAM)_T(""));
}
VOID CZippyWindow::OnResetTraceFiles()
{
TCHAR dlgTitle[MAX_STR_LEN];
TCHAR dlgMessage[MAX_STR_LEN];
LoadStringSimple(IDS_CONFIRMRESETTRACETITLE,dlgTitle);
LoadStringSimple(IDS_CONFIRMRESETTRACEMESSAGE,dlgMessage);
if (IDYES != MessageBox(m_hWnd,dlgMessage,dlgTitle,MB_YESNO)) {
return;
}
m_rTracer->TRC_ResetTraceFiles();
}
VOID CZippyWindow::OnPreferences()
{
COptionsDialog optionsDialog(m_rTracer);
optionsDialog.DoDialog(m_hWnd);
}
VOID CZippyWindow::OnAbout()
{
HICON appIcon;
TCHAR appTitle[MAX_STR_LEN];
TCHAR appOtherStuff[MAX_STR_LEN];
LoadStringSimple(IDS_ABOUTAPPTITLE,appTitle);
LoadStringSimple(IDS_ABOUTOTHERSTUFF,appOtherStuff);
appIcon = (HICON)LoadImage(g_hInstance,MAKEINTRESOURCE(IDI_MAINFRAME),
IMAGE_ICON,0,0,LR_SHARED);
if( NULL != appIcon )
{
ShellAbout(m_hWnd,appTitle,appOtherStuff,appIcon);
// even though the icon is shared we should destroy it to keep the reference
// count somewhat sane.
DestroyIcon(appIcon);
}
}
//
// *** Private Helper Functions ***
//
//
// Computes the color for the given debut output. It parses the text to dtermine
// what the thread id is and then either retrieves the color for that thread or
// picks a new color
//
BOOL CZippyWindow::ComputeNewColor(DWORD processID, LPCTSTR text, UINT len, CHARFORMAT *lpFormat)
{
LPTSTR procIdStr;
DWORD threadId;
LPCTSTR procBase;
UINT maxStrLen;
BOOL bSuccess;
UINT threadLen;
LPTHREADCOLOR newColor;
procIdStr = NULL;
bSuccess = TRUE;
// first we will just make sure the format struct is in a safe state.
lpFormat->cbSize = sizeof(CHARFORMAT);
lpFormat->dwMask = 0;
maxStrLen = sizeof(DWORD) * 2;
procIdStr = (LPTSTR)HeapAlloc(GetProcessHeap(),0,sizeof(TCHAR) * (maxStrLen+1));
if (!procIdStr) {
bSuccess = FALSE;
goto CLEANUP_AND_EXIT;
}
wsprintf(procIdStr,TRC_PROC_FMT,processID);
procBase = _tcsstr(text,procIdStr);
if (!procBase) {
bSuccess = FALSE;
goto CLEANUP_AND_EXIT;
}
procBase += _tcslen(procIdStr);
if (*procBase != ':') {
bSuccess = FALSE;
goto CLEANUP_AND_EXIT;
}
procBase++;
threadLen = 0;
while (_istxdigit(*(procBase + threadLen))) {
threadLen++;
}
if (!threadLen) {
bSuccess = FALSE;
goto CLEANUP_AND_EXIT;
}
threadId = ConvertHexStrToDword(procBase,threadLen);
newColor = FindColorForThread(processID,threadId);
lpFormat->crTextColor = newColor->color;
lpFormat->dwEffects = 0;
lpFormat->dwMask = CFM_COLOR;
CLEANUP_AND_EXIT:
if (procIdStr) {
HeapFree(GetProcessHeap(),0,procIdStr);
}
return bSuccess;
}
//
// This converts a hex string to the equivalent DWORD value for example
// the string "FF" would cause the function to return 0xFF (255)
//
DWORD CZippyWindow::ConvertHexStrToDword(LPCTSTR str, UINT strLen)
{
DWORD total;
TCHAR current;
INT currentValue;
total = 0;
if (strLen == 0) {
strLen = _tcslen(str);
}
while (strLen-- > 0) {
current = *(str++);
if (_istdigit(current)) {
currentValue = current - '0';
} else {
current = (TCHAR)tolower((INT)current);
currentValue = 10 + (current - 'a');
}
total = (total * 16) + currentValue;
}
return total;
}
// This looks up the color for the given thread. If the thread has not been
// seen before a new color is picked and the color for the thread is saved.
LPTHREADCOLOR CZippyWindow::FindColorForThread(DWORD processId, DWORD threadId)
{
int i = 0;
LPTHREADCOLOR lpThreadColor;
for (i=0;i<COLOR_HISTORY_COUNT;i++) {
if (m_threadHistory[i].threadId == threadId &&
m_threadHistory[i].processId == processId) {
return &m_threadHistory[i];
}
}
// else this is the first time we saw the thread
lpThreadColor = &m_threadHistory[m_nextThreadIndex++];
if (m_nextThreadIndex == COLOR_HISTORY_COUNT) {
m_nextThreadIndex = 0;
}
lpThreadColor->processId = processId;
lpThreadColor->threadId = threadId;
lpThreadColor->color = colors[m_nextThreadColor++];
if (m_nextThreadColor == NUM_COLORS) {
m_nextThreadColor = 0;
}
return lpThreadColor;
}
// This handles actually saving the document.
VOID CZippyWindow::DoSaveInternal()
{
HANDLE saveFile;
EDITSTREAM saveStream;
LRESULT bytesSaved;
TCHAR dlgTitle[MAX_STR_LEN];
TCHAR dlgMessage[MAX_STR_LEN];
saveFile = CreateFile(m_SaveFile,GENERIC_WRITE,0,NULL,CREATE_ALWAYS,
0,NULL);
if (saveFile==INVALID_HANDLE_VALUE) {
LoadStringSimple(IDS_FILEOPENERROR,dlgMessage);
LoadStringSimple(IDS_ZIPPYWINDOWTITLE,dlgTitle);
MessageBox(m_hWnd,dlgMessage,dlgTitle,MB_OK|MB_ICONERROR);
return;
}
saveStream.dwCookie = (DWORD_PTR)saveFile;
saveStream.dwError = 0;
saveStream.pfnCallback = SaveCallback;
bytesSaved = SendMessage(m_hControlWnd,EM_STREAMOUT,SF_TEXT,(LPARAM)&saveStream);
CloseHandle(saveFile);
if (saveStream.dwError != 0) {
LoadStringSimple(IDS_FILESAVEERROR,dlgMessage);
LoadStringSimple(IDS_ZIPPYWINDOWTITLE,dlgTitle);
MessageBox(m_hWnd,dlgMessage,dlgTitle,MB_OK|MB_ICONERROR);
} else {
SendMessage(m_hControlWnd,EM_SETMODIFY,FALSE,0);
}
}
// This is a private callback function which rich edit calls when
// saving out the document
static DWORD CALLBACK
SaveCallback(DWORD_PTR dwCookie,LPBYTE pbBuff,LONG cb,LONG *pcb)
{
HANDLE fileHandle;
fileHandle = (HANDLE)dwCookie;
if (!WriteFile(fileHandle,pbBuff,cb,(PULONG)pcb,NULL)) {
return GetLastError();
}
return 0;
}
// As the function name says this does a find next operation
// on the rich edit control
BOOL CZippyWindow::DoFindNext(LPFINDREPLACE lpFindReplace)
{
FINDTEXTEX findText;
WPARAM searchOptions;
CHARRANGE currentSel;
TCHAR dlgTitle[MAX_STR_LEN];
TCHAR dlgMessage[MAX_STR_LEN];
SendMessage(m_hControlWnd,EM_EXGETSEL,0,(LPARAM)&currentSel);
findText.chrg.cpMin = currentSel.cpMax;
findText.chrg.cpMax = -1;
findText.lpstrText = lpFindReplace->lpstrFindWhat;
searchOptions = FR_DOWN;
if (lpFindReplace->Flags & FR_MATCHCASE) {
searchOptions |= FR_MATCHCASE;
}
if (lpFindReplace->Flags & FR_WHOLEWORD) {
searchOptions |= FR_WHOLEWORD;
}
if (0 <= SendMessage(m_hControlWnd, EM_FINDTEXTEX,searchOptions,
(LPARAM)&findText)) {
SendMessage(m_hControlWnd, EM_EXSETSEL,0,(LPARAM)&findText.chrgText);
} else {
LoadStringSimple(IDS_SEARCHFAILURE,dlgMessage);
LoadStringSimple(IDS_ZIPPYWINDOWTITLE,dlgTitle);
MessageBox(m_hWndFindReplace,dlgMessage,dlgTitle,MB_OK);
return FALSE;
}
return TRUE;
}
// This does a replace operation on the control
BOOL CZippyWindow::DoReplace(LPFINDREPLACE lpFindReplace)
{
FINDTEXTEX findText;
WPARAM searchOptions;
CHARRANGE currentSel;
TCHAR dlgTitle[MAX_STR_LEN];
TCHAR dlgMessage[MAX_STR_LEN];
SendMessage(m_hControlWnd,EM_EXGETSEL,0,(LPARAM)&currentSel);
findText.chrg.cpMin = currentSel.cpMin;
findText.chrg.cpMax = -1;
findText.lpstrText = lpFindReplace->lpstrFindWhat;
searchOptions = FR_DOWN;
if (lpFindReplace->Flags & FR_MATCHCASE) {
searchOptions |= FR_MATCHCASE;
}
if (lpFindReplace->Flags & FR_WHOLEWORD) {
searchOptions |= FR_WHOLEWORD;
}
if (-1 == SendMessage(m_hControlWnd, EM_FINDTEXTEX,searchOptions,
(LPARAM)&findText)) {
// if we can't find what they were looking for just give up.
LoadStringSimple(IDS_SEARCHFAILURE,dlgMessage);
LoadStringSimple(IDS_ZIPPYWINDOWTITLE,dlgTitle);
MessageBox(m_hWndFindReplace,dlgMessage,dlgTitle,MB_OK);
return FALSE;
}
if (currentSel.cpMin == findText.chrgText.cpMin &&
currentSel.cpMax == findText.chrgText.cpMax) {
SendMessage(m_hControlWnd,EM_REPLACESEL,0,(LPARAM)lpFindReplace->lpstrReplaceWith);
// Now select the next occurrence
return DoFindNext(lpFindReplace);
} else {
// They weren't on what they were searching for so select it.
SendMessage(m_hControlWnd, EM_EXSETSEL,0,(LPARAM)&findText.chrgText);
}
return TRUE;
}
// This loops on DoReplace until DoReplace returns FALSE
VOID CZippyWindow::DoReplaceAll(LPFINDREPLACE lpFindReplace)
{
while (DoReplace(lpFindReplace));
}
// This actually saves the traceconfiguration. We just write
// out the binary config structure to the file
VOID CZippyWindow::DoSaveConfInternal()
{
HANDLE saveFile;
TRC_CONFIG trcConfig;
TCHAR dlgTitle[MAX_STR_LEN];
TCHAR dlgMessage[MAX_STR_LEN];
DWORD bytesWritten;
saveFile = CreateFile(m_SaveConfFile,GENERIC_WRITE,0,NULL,CREATE_ALWAYS,
0,NULL);
if (saveFile==INVALID_HANDLE_VALUE) {
LoadStringSimple(IDS_FILEOPENERROR,dlgMessage);
LoadStringSimple(IDS_ZIPPYWINDOWTITLE,dlgTitle);
MessageBox(m_hWnd,dlgMessage,dlgTitle,MB_OK|MB_ICONERROR);
return;
}
m_rTracer->GetCurrentConfig(&trcConfig);
if (!WriteFile(saveFile,&trcConfig,sizeof(trcConfig),&bytesWritten,NULL) ||
bytesWritten != sizeof(trcConfig)) {
LoadStringSimple(IDS_FILESAVEERROR,dlgMessage);
LoadStringSimple(IDS_ZIPPYWINDOWTITLE,dlgTitle);
MessageBox(m_hWnd,dlgMessage,dlgTitle,MB_OK|MB_ICONERROR);
}
CloseHandle(saveFile);
}
// This reads in the binary configuration structure and then
// sets it as the current tracing config
VOID CZippyWindow::DoLoadConfInternal()
{
HANDLE openFile;
DWORD bytesRead;
TRC_CONFIG trcConfig;
TCHAR dlgTitle[MAX_STR_LEN];
TCHAR dlgMessage[MAX_STR_LEN];
openFile = CreateFile(m_LoadConfFile,GENERIC_READ,FILE_SHARE_READ,NULL,OPEN_EXISTING,
0,NULL);
if (openFile==INVALID_HANDLE_VALUE) {
LoadStringSimple(IDS_FILELOADOPENERROR,dlgMessage);
LoadStringSimple(IDS_ZIPPYWINDOWTITLE,dlgTitle);
MessageBox(m_hWnd,dlgMessage,dlgTitle,MB_OK|MB_ICONERROR);
return;
}
if (!ReadFile(openFile,&trcConfig,sizeof(trcConfig),&bytesRead,NULL)||
bytesRead != sizeof(trcConfig)) {
LoadStringSimple(IDS_FILELOADERROR,dlgMessage);
LoadStringSimple(IDS_ZIPPYWINDOWTITLE,dlgTitle);
MessageBox(m_hWnd,dlgMessage,dlgTitle,MB_OK|MB_ICONERROR);
}
m_rTracer->SetCurrentConfig(&trcConfig);
}
// Reads the saved window position in from the registry.
VOID CZippyWindow::GetSavedWindowPos(LPRECT savedPos)
{
DWORD dwResult;
DWORD dwSize;
DWORD dwType;
RECT rect;
HKEY hKey;
savedPos->top = WINDOW_DEF_TOP;
savedPos->bottom = WINDOW_DEF_BOTTOM;
savedPos->left = WINDOW_DEF_LEFT;
savedPos->right = WINDOW_DEF_RIGHT;
dwResult = RegOpenKeyEx(HKEY_CURRENT_USER,ZIPPY_REG_KEY,0,
KEY_QUERY_VALUE,&hKey);
if (dwResult) {
return;
}
dwSize = sizeof(RECT);
dwResult = RegQueryValueEx(hKey,ZIPPY_WINDOW_POS_VALUE,NULL,&dwType,
(LPBYTE)&rect,&dwSize);
RegCloseKey(hKey);
if (dwResult||dwSize != sizeof(RECT)||dwType!=REG_BINARY) {
return;
}
*savedPos = rect;
}
// Saves the window position out to the registry
VOID CZippyWindow::SaveWindowPos(LPRECT newPos)
{
DWORD dwResult;
HKEY hKey;
dwResult = RegCreateKeyEx(HKEY_CURRENT_USER,ZIPPY_REG_KEY,0,_T(""),0,
KEY_SET_VALUE,NULL,&hKey,NULL);
if (dwResult) {
return;
}
RegSetValueEx(hKey,ZIPPY_WINDOW_POS_VALUE,0,REG_BINARY,
(LPBYTE)newPos,sizeof(RECT));
RegCloseKey(hKey);
}