Leaked source code of windows server 2003
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.
 
 
 
 
 
 

578 lines
17 KiB

// WiaLogCFGDlg.cpp : implementation file
//
#include "stdafx.h"
#include "WiaLogCFG.h"
#include "WiaLogCFGDlg.h"
#include "AddRemove.h"
#include "LogViewer.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// CWiaLogCFGDlg dialog
CWiaLogCFGDlg::CWiaLogCFGDlg(CWnd* pParent /*=NULL*/)
: CDialog(CWiaLogCFGDlg::IDD, pParent)
{
//{{AFX_DATA_INIT(CWiaLogCFGDlg)
m_dwCustomLevel = 0;
//}}AFX_DATA_INIT
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
m_bColorCodeLogViewerText = FALSE;
}
void CWiaLogCFGDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CWiaLogCFGDlg)
DDX_Control(pDX, IDC_COLOR_CODE_LOGVIEWER_TEXT, m_ColorCodeLogViewerTextCheckBox);
DDX_Control(pDX, IDC_LOG_TO_DEBUGGER, m_LogToDebuggerCheckBox);
DDX_Control(pDX, IDC_CLEARLOG_ON_BOOT, m_ClearLogOnBootCheckBox);
DDX_Control(pDX, IDC_PARSE_PROGRESS, m_ProgressCtrl);
DDX_Control(pDX, IDC_ADD_TIME, m_AddTimeCheckBox);
DDX_Control(pDX, IDC_ADD_THREADID, m_AddThreadIDCheckBox);
DDX_Control(pDX, IDC_ADD_MODULENAME, m_AddModuleCheckBox);
DDX_Control(pDX, IDC_TRUNCATE_ON_BOOT, m_TruncateOnBootCheckBox);
DDX_Control(pDX, IDC_SELECT_MODULE_COMBOBOX, m_ModuleComboBox);
DDX_Control(pDX, IDC_LOG_LEVEL_WARNING, m_WarningCheckBox);
DDX_Control(pDX, IDC_LOG_LEVEL_ERROR, m_ErrorCheckBox);
DDX_Control(pDX, IDC_LOG_LEVEL_TRACE, m_TraceCheckBox);
DDX_Control(pDX, IDC_FILTER_OFF, m_FilterOff);
DDX_Control(pDX, IDC_FILTER_1, m_Filter1);
DDX_Control(pDX, IDC_FILTER_2, m_Filter2);
DDX_Control(pDX, IDC_FILTER_3, m_Filter3);
DDX_Control(pDX, IDC_FILTER_CUSTOM, m_FilterCustom);
DDX_Text(pDX, IDC_EDIT_CUSTOM_LEVEL, m_dwCustomLevel);
DDV_MinMaxDWord(pDX, m_dwCustomLevel, 0, 9999);
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CWiaLogCFGDlg, CDialog)
//{{AFX_MSG_MAP(CWiaLogCFGDlg)
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
ON_BN_CLICKED(IDC_ADD_MODULE_BUTTON, OnAddModuleButton)
ON_BN_CLICKED(IDC_DELETE_MODULE_BUTTON, OnDeleteModuleButton)
ON_BN_CLICKED(IDC_WRITE_SETTINGS_BUTTON, OnWriteSettingsButton)
ON_CBN_SELCHANGE(IDC_SELECT_MODULE_COMBOBOX, OnSelchangeSelectModuleCombobox)
ON_BN_CLICKED(IDC_CLEARLOG_BUTTON, OnClearlogButton)
ON_BN_CLICKED(IDC_VIEW_LOG_BUTTON, OnViewLogButton)
ON_CBN_SETFOCUS(IDC_SELECT_MODULE_COMBOBOX, OnSetfocusSelectModuleCombobox)
ON_CBN_DROPDOWN(IDC_SELECT_MODULE_COMBOBOX, OnDropdownSelectModuleCombobox)
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CWiaLogCFGDlg message handlers
BOOL CWiaLogCFGDlg::OnInitDialog()
{
m_hInstance = NULL;
m_hInstance = AfxGetInstanceHandle();
CDialog::OnInitDialog();
ShowProgress(FALSE);
SetIcon(m_hIcon, TRUE); // Set big icon
SetIcon(m_hIcon, FALSE); // Set small icon
m_LogInfo.dwLevel = 0;
m_LogInfo.dwMaxSize = 100000;
m_LogInfo.dwMode = 0;
m_LogInfo.dwTruncateOnBoot = 0;
memset(m_LogInfo.szKeyName,0,sizeof(m_LogInfo.szKeyName));
m_CurrentSelection = 0;
InitializeDialogSettings(SETTINGS_RESET_DIALOG);
RegistryOperation(REG_READ);
InitializeDialogSettings(SETTINGS_TO_DIALOG);
CheckGlobalServiceSettings();
return TRUE; // return TRUE unless you set the focus to a control
}
// If you add a minimize button to your dialog, you will need the code below
// to draw the icon. For MFC applications using the document/view model,
// this is automatically done for you by the framework.
void CWiaLogCFGDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // device context for painting
SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
// Center icon in client rectangle
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// Draw the icon
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
HCURSOR CWiaLogCFGDlg::OnQueryDragIcon()
{
return (HCURSOR) m_hIcon;
}
void CWiaLogCFGDlg::OnAddModuleButton()
{
CAddRemove AddRemoveDlg;
AddRemoveDlg.SetTitle(TEXT("Add a Module"));
AddRemoveDlg.SetStatusText(TEXT("Enter a Module Name:"));
if(AddRemoveDlg.DoModal() == IDOK) {
m_TruncateOnBootCheckBox.SetCheck(0);
m_ClearLogOnBootCheckBox.SetCheck(0);
AddRemoveDlg.GetNewKeyName(m_LogInfo.szKeyName);
RegistryOperation(REG_ADD_KEY);
RegistryOperation(REG_READ);
CheckGlobalServiceSettings();
}
}
void CWiaLogCFGDlg::OnDeleteModuleButton()
{
if(MessageBox(TEXT("Are you sure you want to DELETE this module?"),
TEXT("Delete Module"),
MB_YESNO|MB_ICONQUESTION) == IDYES) {
//
// Delete this module
//
RegistryOperation(REG_DELETE_KEY);
RegistryOperation(REG_READ);
}
}
void CWiaLogCFGDlg::RegistryOperation(ULONG ulFlags)
{
TCHAR szAppRegistryKey[MAX_PATH];
TCHAR szValueName[MAX_PATH];
LoadString(m_hInstance, REGSTR_PATH_STICONTROL, szAppRegistryKey, MAX_PATH);
CRegistry Registry(szAppRegistryKey,HKEY_LOCAL_MACHINE);
//
// move to logging
//
LoadString(m_hInstance,REGSTR_VAL_LOGGING , szValueName, MAX_PATH);
Registry.MoveToSubKey(szValueName);
if(ulFlags == REG_ADD_KEY) {
//
// Add a new key
//
Registry.CreateKey(m_LogInfo.szKeyName);
//
// change current selection to an invalid selection
//
m_CurrentSelection = -99;
return;
}
if( ulFlags == REG_DELETE_KEY) {
//
// delete a Key
//
Registry.DeleteKey(m_LogInfo.szKeyName);
return;
}
//
// enumerate keys
//
DWORD dwIndex = 0;
TCHAR pszKeyName[64];
m_ModuleComboBox.ResetContent();
while(Registry.EnumerateKeys(dwIndex++,pszKeyName, sizeof(pszKeyName)) != ERROR_NO_MORE_ITEMS) {
m_ModuleComboBox.AddString(pszKeyName);
}
if(m_CurrentSelection == -99){
INT nIndex = m_ModuleComboBox.FindString(-1, m_LogInfo.szKeyName);
m_CurrentSelection = nIndex;
m_ModuleComboBox.SetCurSel(nIndex);
} else {
m_ModuleComboBox.GetLBText(m_CurrentSelection,m_LogInfo.szKeyName);
}
m_ModuleComboBox.SetCurSel(m_CurrentSelection);
//
// move to DLL specifc subkey
//
Registry.MoveToSubKey(m_LogInfo.szKeyName);
switch(ulFlags) {
case REG_WRITE:
LoadString(m_hInstance,REGSTR_VAL_LOG_LEVEL , szValueName, MAX_PATH);
Registry.SetValue(szValueName,m_LogInfo.dwLevel);
LoadString(m_hInstance,REGSTR_VAL_MODE , szValueName, MAX_PATH);
Registry.SetValue(szValueName,m_LogInfo.dwMode);
LoadString(m_hInstance,REGSTR_VAL_MAXSIZE , szValueName, MAX_PATH);
Registry.SetValue(szValueName,m_LogInfo.dwMaxSize);
LoadString(m_hInstance,REGSTR_VAL_TRUNCATE_ON_BOOT , szValueName, MAX_PATH);
Registry.SetValue(szValueName,m_LogInfo.dwTruncateOnBoot);
LoadString(m_hInstance,REGSTR_VAL_CLEARLOG_ON_BOOT , szValueName, MAX_PATH);
Registry.SetValue(szValueName,m_LogInfo.dwClearLogOnBoot);
LoadString(m_hInstance,REGSTR_VAL_DETAIL , szValueName, MAX_PATH);
Registry.SetValue(szValueName,m_LogInfo.dwDetail);
LoadString(m_hInstance,REGSTR_VAL_LOG_TO_DEBUGGER, szValueName, MAX_PATH);
Registry.SetValue(szValueName,m_LogInfo.dwLogToDebugger);
break;
case REG_READ:
default:
LoadString(m_hInstance,REGSTR_VAL_LOG_LEVEL , szValueName, MAX_PATH);
m_LogInfo.dwLevel = Registry.GetValue(szValueName,WIALOG_NO_LEVEL);
LoadString(m_hInstance,REGSTR_VAL_MODE , szValueName, MAX_PATH);
m_LogInfo.dwMode = Registry.GetValue(szValueName,WIALOG_ADD_MODULE|WIALOG_ADD_THREAD);
LoadString(m_hInstance,REGSTR_VAL_MAXSIZE , szValueName, MAX_PATH);
m_LogInfo.dwMaxSize = Registry.GetValue(szValueName,100000);
LoadString(m_hInstance,REGSTR_VAL_TRUNCATE_ON_BOOT, szValueName, MAX_PATH);
m_LogInfo.dwTruncateOnBoot = Registry.GetValue(szValueName,0);
LoadString(m_hInstance,REGSTR_VAL_CLEARLOG_ON_BOOT, szValueName, MAX_PATH);
m_LogInfo.dwClearLogOnBoot = Registry.GetValue(szValueName,0);
LoadString(m_hInstance,REGSTR_VAL_DETAIL , szValueName, MAX_PATH);
m_LogInfo.dwDetail = Registry.GetValue(szValueName,0);
LoadString(m_hInstance,REGSTR_VAL_LOG_TO_DEBUGGER, szValueName, MAX_PATH);
m_LogInfo.dwLogToDebugger = Registry.GetValue(szValueName,0);
break;
}
}
void CWiaLogCFGDlg::InitializeDialogSettings(ULONG ulFlags)
{
switch (ulFlags) {
case SETTINGS_TO_DIALOG:
//
// set level of detail
//
switch (m_LogInfo.dwDetail) {
case WIALOG_NO_LEVEL:
m_FilterOff.SetCheck(1);
break;
case WIALOG_LEVEL1 :
m_Filter1.SetCheck(1);
break;
case WIALOG_LEVEL2:
m_Filter2.SetCheck(1);
break;
case WIALOG_LEVEL3:
m_Filter3.SetCheck(1);
break;
default:
m_FilterCustom.SetCheck(1);
m_dwCustomLevel = m_LogInfo.dwDetail;
UpdateData(FALSE);
break;
}
//
// set truncate on boot check box
//
if (m_LogInfo.dwTruncateOnBoot != 0)
m_TruncateOnBootCheckBox.SetCheck(1);
else
m_TruncateOnBootCheckBox.SetCheck(0);
//
// set clear log on boot check box
//
if (m_LogInfo.dwClearLogOnBoot != 0)
m_ClearLogOnBootCheckBox.SetCheck(1);
else
m_ClearLogOnBootCheckBox.SetCheck(0);
//
// set log to debugger check box
//
if (m_LogInfo.dwLogToDebugger != 0)
m_LogToDebuggerCheckBox.SetCheck(1);
else
m_LogToDebuggerCheckBox.SetCheck(0);
//
// set trace level check boxes
//
if (m_LogInfo.dwLevel & WIALOG_TRACE)
m_TraceCheckBox.SetCheck(1);
if (m_LogInfo.dwLevel & WIALOG_ERROR)
m_ErrorCheckBox.SetCheck(1);
if (m_LogInfo.dwLevel & WIALOG_WARNING)
m_WarningCheckBox.SetCheck(1);
//
// set additional details check boxes
//
if (m_LogInfo.dwMode & WIALOG_ADD_TIME)
m_AddTimeCheckBox.SetCheck(1);
if (m_LogInfo.dwMode & WIALOG_ADD_MODULE)
m_AddModuleCheckBox.SetCheck(1);
if (m_LogInfo.dwMode & WIALOG_ADD_THREAD)
m_AddThreadIDCheckBox.SetCheck(1);
break;
case SETTINGS_FROM_DIALOG:
//
// get level of detail
//
if (m_FilterOff.GetCheck() == 1)
m_LogInfo.dwDetail = WIALOG_NO_LEVEL;
if (m_Filter1.GetCheck() == 1)
m_LogInfo.dwDetail = WIALOG_LEVEL1;
if (m_Filter2.GetCheck() == 1)
m_LogInfo.dwDetail = WIALOG_LEVEL2;
if (m_Filter3.GetCheck() == 1)
m_LogInfo.dwDetail = WIALOG_LEVEL3;
if (m_FilterCustom.GetCheck() == 1) {
UpdateData(TRUE);
m_LogInfo.dwDetail = m_dwCustomLevel;
}
//
// get truncate on boot check box
//
if (m_TruncateOnBootCheckBox.GetCheck() == 1)
m_LogInfo.dwTruncateOnBoot = 1;
else
m_LogInfo.dwTruncateOnBoot = 0;
//
// get clear log on boot check box
//
if (m_ClearLogOnBootCheckBox.GetCheck() == 1)
m_LogInfo.dwClearLogOnBoot = 1;
else
m_LogInfo.dwClearLogOnBoot = 0;
//
// get log to debugger check box
//
if(m_LogToDebuggerCheckBox.GetCheck() == 1)
m_LogInfo.dwLogToDebugger = 1;
else
m_LogInfo.dwLogToDebugger = 0;
//
// get trace level check boxes
//
m_LogInfo.dwLevel = 0;
if (m_TraceCheckBox.GetCheck() == 1)
m_LogInfo.dwLevel = m_LogInfo.dwLevel | WIALOG_TRACE;
if (m_ErrorCheckBox.GetCheck() == 1)
m_LogInfo.dwLevel = m_LogInfo.dwLevel | WIALOG_ERROR;
if (m_WarningCheckBox.GetCheck() == 1)
m_LogInfo.dwLevel = m_LogInfo.dwLevel | WIALOG_WARNING;
//
// set additional details check boxes
//
m_LogInfo.dwMode = 0;
if (m_AddTimeCheckBox.GetCheck() == 1)
m_LogInfo.dwMode = m_LogInfo.dwMode | WIALOG_ADD_TIME;
if (m_AddModuleCheckBox.GetCheck() == 1)
m_LogInfo.dwMode = m_LogInfo.dwMode | WIALOG_ADD_MODULE;
if (m_AddThreadIDCheckBox.GetCheck() == 1)
m_LogInfo.dwMode = m_LogInfo.dwMode | WIALOG_ADD_THREAD;
break;
default:
m_FilterOff.SetCheck(0);
m_Filter1.SetCheck(0);
m_Filter2.SetCheck(0);
m_Filter3.SetCheck(0);
m_FilterCustom.SetCheck(0);
m_TruncateOnBootCheckBox.SetCheck(0);
m_TraceCheckBox.SetCheck(0);
m_ErrorCheckBox.SetCheck(0);
m_WarningCheckBox.SetCheck(0);
m_AddTimeCheckBox.SetCheck(0);
m_AddModuleCheckBox.SetCheck(0);
m_AddThreadIDCheckBox.SetCheck(0);
m_dwCustomLevel = 0;
UpdateData(FALSE);
break;
}
}
void CWiaLogCFGDlg::OnOK()
{
InitializeDialogSettings(SETTINGS_FROM_DIALOG);
RegistryOperation(REG_WRITE);
CDialog::OnOK();
}
void CWiaLogCFGDlg::OnWriteSettingsButton()
{
InitializeDialogSettings(SETTINGS_FROM_DIALOG);
RegistryOperation(REG_WRITE);
}
void CWiaLogCFGDlg::OnSelchangeSelectModuleCombobox()
{
m_CurrentSelection = m_ModuleComboBox.GetCurSel();
if(m_CurrentSelection < 0)
return;
CheckGlobalServiceSettings();
InitializeDialogSettings(SETTINGS_RESET_DIALOG);
RegistryOperation(REG_READ);
InitializeDialogSettings(SETTINGS_TO_DIALOG);
}
void CWiaLogCFGDlg::OnClearlogButton()
{
//
// Get Windows Directory
//
TCHAR szLogFilePath[MAX_PATH];
DWORD dwLength = 0;
dwLength = ::GetWindowsDirectory(szLogFilePath,sizeof(szLogFilePath));
if (( dwLength == 0) || !*szLogFilePath ) {
OutputDebugString(TEXT("Could not GetWindowsDirectory()"));
return;
}
//
// Add log file name to Windows Directory
//
lstrcat(lstrcat(szLogFilePath,TEXT("\\")),TEXT("wiaservc.log"));
//
// Create / open Log file
//
HANDLE hLogFile = ::CreateFile(szLogFilePath,
GENERIC_WRITE,
FILE_SHARE_WRITE | FILE_SHARE_READ,
NULL, // security attributes
CREATE_ALWAYS,
FILE_ATTRIBUTE_NORMAL,
NULL); // template file handle
if(hLogFile != NULL)
CloseHandle(hLogFile);
}
void CWiaLogCFGDlg::OnViewLogButton()
{
CLogViewer LogViewer;
if (m_ColorCodeLogViewerTextCheckBox.GetCheck() == 1)
m_bColorCodeLogViewerText = TRUE;
else
m_bColorCodeLogViewerText = FALSE;
LogViewer.ColorizeText(m_bColorCodeLogViewerText);
//
// initialize progress
//
m_ProgCtrl.SetControl(&m_ProgressCtrl);
LogViewer.SetProgressCtrl(&m_ProgCtrl);
ShowProgress(TRUE);
LogViewer.DoModal();
ShowProgress(FALSE);
}
void CWiaLogCFGDlg::ShowProgress(BOOL bShow)
{
if(bShow) {
m_ProgressCtrl.ShowWindow(SW_SHOW);
} else {
m_ProgressCtrl.ShowWindow(SW_HIDE);
}
}
void CWiaLogCFGDlg::OnSetfocusSelectModuleCombobox()
{
OnWriteSettingsButton();
}
void CWiaLogCFGDlg::OnDropdownSelectModuleCombobox()
{
OnWriteSettingsButton();
}
void CWiaLogCFGDlg::CheckGlobalServiceSettings()
{
TCHAR szKeyName[MAX_PATH];
m_ModuleComboBox.GetLBText(m_CurrentSelection,szKeyName);
if(lstrcmp(szKeyName,TEXT("WIASERVC")) == 0) {
m_TruncateOnBootCheckBox.EnableWindow(TRUE);
m_ClearLogOnBootCheckBox.EnableWindow(TRUE);
} else {
m_TruncateOnBootCheckBox.EnableWindow(FALSE);
m_ClearLogOnBootCheckBox.EnableWindow(FALSE);
}
}