336 lines
11 KiB
C++
336 lines
11 KiB
C++
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
// Copyright (c) 2002 Microsoft Corporation. All rights reserved.
|
||
|
// Copyright (c) 2002 OSR Open Systems Resources, Inc.
|
||
|
//
|
||
|
// ProviderControlGUIDDlg.cpp : implementation file
|
||
|
//////////////////////////////////////////////////////////////////////////////
|
||
|
|
||
|
#include "stdafx.h"
|
||
|
#include <tchar.h>
|
||
|
#include <wmistr.h>
|
||
|
#include <initguid.h>
|
||
|
extern "C" {
|
||
|
#include <evntrace.h>
|
||
|
}
|
||
|
#include <traceprt.h>
|
||
|
#include "TraceView.h"
|
||
|
#include "LogSession.h"
|
||
|
#include "DisplayDlg.h"
|
||
|
#include "utils.h"
|
||
|
#include "ProviderControlGUIDDlg.h"
|
||
|
|
||
|
|
||
|
// CProviderControlGuidDlg dialog
|
||
|
|
||
|
IMPLEMENT_DYNAMIC(CProviderControlGuidDlg, CDialog)
|
||
|
CProviderControlGuidDlg::CProviderControlGuidDlg(CWnd* pParent, CTraceSession *pTraceSession)
|
||
|
: CDialog(CProviderControlGuidDlg::IDD, pParent)
|
||
|
{
|
||
|
m_pTraceSession = pTraceSession;
|
||
|
}
|
||
|
|
||
|
CProviderControlGuidDlg::~CProviderControlGuidDlg()
|
||
|
{
|
||
|
}
|
||
|
|
||
|
int CProviderControlGuidDlg::OnInitDialog()
|
||
|
{
|
||
|
int ret;
|
||
|
|
||
|
ret = CDialog::OnInitDialog();
|
||
|
|
||
|
//
|
||
|
// Default to the PDB radio button being selected
|
||
|
//
|
||
|
CheckRadioButton(IDC_PDB_SELECT_RADIO,
|
||
|
IDC_MANUAL_SELECT_RADIO,
|
||
|
IDC_PDB_SELECT_RADIO);
|
||
|
|
||
|
//
|
||
|
// Enable PDB edit box and browse button
|
||
|
//
|
||
|
m_pdbFileName.EnableWindow(TRUE);
|
||
|
GetDlgItem(IDC_PDB_BROWSE_BUTTON)->EnableWindow(TRUE);
|
||
|
|
||
|
//
|
||
|
// Disable all controls associated with other radio buttons
|
||
|
//
|
||
|
m_ctlFileName.EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_CTL_BROWSE_BUTTON)->EnableWindow(FALSE);
|
||
|
m_controlGuidName.EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_PROCESS_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_THREAD_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_NET_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_DISK_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_PAGEFAULT_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_HARDFAULT_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_IMAGELOAD_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_REGISTRY_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_FILEIO_CHECK)->EnableWindow(FALSE);
|
||
|
|
||
|
return ret;
|
||
|
}
|
||
|
|
||
|
void CProviderControlGuidDlg::DoDataExchange(CDataExchange* pDX)
|
||
|
{
|
||
|
CDialog::DoDataExchange(pDX);
|
||
|
DDX_Control(pDX, IDC_PDB_FILE_EDIT, m_pdbFileName);
|
||
|
DDX_Control(pDX, IDC_CTL_FILE_EDIT, m_ctlFileName);
|
||
|
DDX_Control(pDX, IDC_MANUAL_GUID_EDIT, m_controlGuidName);
|
||
|
}
|
||
|
|
||
|
|
||
|
BEGIN_MESSAGE_MAP(CProviderControlGuidDlg, CDialog)
|
||
|
ON_BN_CLICKED(IDC_PDB_BROWSE_BUTTON, OnBnClickedPdbBrowseButton)
|
||
|
ON_BN_CLICKED(IDC_CTL_BROWSE_BUTTON, OnBnClickedCtlBrowseButton)
|
||
|
ON_BN_CLICKED(IDC_PDB_SELECT_RADIO, OnBnClickedPdbSelectRadio)
|
||
|
ON_BN_CLICKED(IDC_CTL_SELECT_RADIO, OnBnClickedCtlSelectRadio)
|
||
|
ON_BN_CLICKED(IDC_MANUAL_SELECT_RADIO, OnBnClickedManualSelectRadio)
|
||
|
ON_BN_CLICKED(IDOK, OnBnClickedOk)
|
||
|
ON_BN_CLICKED(IDC_KERNEL_LOGGER_SELECT_RADIO, OnBnClickedKernelLoggerSelectRadio)
|
||
|
END_MESSAGE_MAP()
|
||
|
|
||
|
|
||
|
// CProviderControlGuidDlg message handlers
|
||
|
|
||
|
void CProviderControlGuidDlg::OnBnClickedPdbBrowseButton()
|
||
|
{
|
||
|
//
|
||
|
// Use the common controls file open dialog
|
||
|
//
|
||
|
CFileDialog fileDlg(TRUE,
|
||
|
_T("pdb"),_T("*.pdb"),
|
||
|
OFN_PATHMUSTEXIST | OFN_FILEMUSTEXIST | OFN_READONLY |
|
||
|
OFN_HIDEREADONLY | OFN_EXPLORER | OFN_NOCHANGEDIR,
|
||
|
_T("Program Database Files (*.pdb)|*.pdb||"),
|
||
|
this);
|
||
|
|
||
|
//
|
||
|
// Pop the dialog... Any error, just return
|
||
|
//
|
||
|
if( fileDlg.DoModal()!=IDOK ) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Get the file name and display it
|
||
|
//
|
||
|
if(!fileDlg.GetPathName().IsEmpty()) {
|
||
|
m_pdbFileName.SetWindowText(fileDlg.GetPathName());
|
||
|
m_pdbFileName.SetFocus();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void CProviderControlGuidDlg::OnBnClickedCtlBrowseButton()
|
||
|
{
|
||
|
//
|
||
|
// Use the common controls file open dialog
|
||
|
//
|
||
|
CFileDialog fileDlg(TRUE,
|
||
|
_T("ctl"),_T("*.ctl"),
|
||
|
OFN_PATHMUSTEXIST | OFN_FILEMUSTEXIST | OFN_READONLY |
|
||
|
OFN_HIDEREADONLY | OFN_EXPLORER | OFN_NOCHANGEDIR,
|
||
|
_T("Control GUID Files (*.ctl)|*.ctl|All Files (*.*)|*.*||"),
|
||
|
this);
|
||
|
|
||
|
//
|
||
|
// Pop the dialog... Any error, just return
|
||
|
//
|
||
|
if( fileDlg.DoModal()!=IDOK ) {
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
//
|
||
|
// Get the file name and display it
|
||
|
//
|
||
|
if(!fileDlg.GetPathName().IsEmpty()) {
|
||
|
m_ctlFileName.SetWindowText(fileDlg.GetPathName());
|
||
|
m_ctlFileName.SetFocus();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
void CProviderControlGuidDlg::OnBnClickedPdbSelectRadio()
|
||
|
{
|
||
|
//
|
||
|
// Set the PDB radio button and unset the rest
|
||
|
//
|
||
|
CheckRadioButton(IDC_PDB_SELECT_RADIO,
|
||
|
IDC_KERNEL_LOGGER_SELECT_RADIO,
|
||
|
IDC_PDB_SELECT_RADIO);
|
||
|
|
||
|
//
|
||
|
// Enable the PDB filename edit box and browse button
|
||
|
//
|
||
|
m_pdbFileName.EnableWindow(TRUE);
|
||
|
GetDlgItem(IDC_PDB_BROWSE_BUTTON)->EnableWindow(TRUE);
|
||
|
|
||
|
//
|
||
|
// Disable all controls associated with other radio buttons
|
||
|
//
|
||
|
m_ctlFileName.EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_CTL_BROWSE_BUTTON)->EnableWindow(FALSE);
|
||
|
m_controlGuidName.EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_PROCESS_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_THREAD_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_NET_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_DISK_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_PAGEFAULT_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_HARDFAULT_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_IMAGELOAD_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_REGISTRY_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_FILEIO_CHECK)->EnableWindow(FALSE);
|
||
|
}
|
||
|
|
||
|
void CProviderControlGuidDlg::OnBnClickedCtlSelectRadio()
|
||
|
{
|
||
|
//
|
||
|
// Set the CTL radio button and unset the rest
|
||
|
//
|
||
|
CheckRadioButton(IDC_PDB_SELECT_RADIO,
|
||
|
IDC_KERNEL_LOGGER_SELECT_RADIO,
|
||
|
IDC_CTL_SELECT_RADIO);
|
||
|
|
||
|
//
|
||
|
// Enable the CTL name edit box and browse button
|
||
|
//
|
||
|
m_ctlFileName.EnableWindow(TRUE);
|
||
|
GetDlgItem(IDC_CTL_BROWSE_BUTTON)->EnableWindow(TRUE);
|
||
|
|
||
|
//
|
||
|
// Disable all controls associated with other radio buttons
|
||
|
//
|
||
|
m_pdbFileName.EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_PDB_BROWSE_BUTTON)->EnableWindow(FALSE);
|
||
|
m_controlGuidName.EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_PROCESS_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_THREAD_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_NET_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_DISK_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_PAGEFAULT_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_HARDFAULT_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_IMAGELOAD_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_REGISTRY_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_FILEIO_CHECK)->EnableWindow(FALSE);
|
||
|
}
|
||
|
|
||
|
void CProviderControlGuidDlg::OnBnClickedManualSelectRadio()
|
||
|
{
|
||
|
//
|
||
|
// Set the Manual radio button and unset the rest
|
||
|
//
|
||
|
CheckRadioButton(IDC_PDB_SELECT_RADIO,
|
||
|
IDC_KERNEL_LOGGER_SELECT_RADIO,
|
||
|
IDC_MANUAL_SELECT_RADIO);
|
||
|
|
||
|
//
|
||
|
// Enable the GUID name edit box
|
||
|
//
|
||
|
m_controlGuidName.EnableWindow(TRUE);
|
||
|
|
||
|
//
|
||
|
// Disable all controls associated with other radio buttons
|
||
|
//
|
||
|
m_pdbFileName.EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_PDB_BROWSE_BUTTON)->EnableWindow(FALSE);
|
||
|
m_ctlFileName.EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_CTL_BROWSE_BUTTON)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_PROCESS_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_THREAD_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_NET_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_DISK_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_PAGEFAULT_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_HARDFAULT_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_IMAGELOAD_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_REGISTRY_CHECK)->EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_FILEIO_CHECK)->EnableWindow(FALSE);
|
||
|
}
|
||
|
|
||
|
void CProviderControlGuidDlg::OnBnClickedKernelLoggerSelectRadio()
|
||
|
{
|
||
|
//
|
||
|
// Set the Kernel Logger radio button and unset the rest
|
||
|
//
|
||
|
CheckRadioButton(IDC_PDB_SELECT_RADIO,
|
||
|
IDC_KERNEL_LOGGER_SELECT_RADIO,
|
||
|
IDC_KERNEL_LOGGER_SELECT_RADIO);
|
||
|
|
||
|
|
||
|
//
|
||
|
// Enable the kernel logger check boxes
|
||
|
//
|
||
|
GetDlgItem(IDC_PROCESS_CHECK)->EnableWindow(TRUE);
|
||
|
GetDlgItem(IDC_THREAD_CHECK)->EnableWindow(TRUE);
|
||
|
GetDlgItem(IDC_NET_CHECK)->EnableWindow(TRUE);
|
||
|
GetDlgItem(IDC_DISK_CHECK)->EnableWindow(TRUE);
|
||
|
GetDlgItem(IDC_PAGEFAULT_CHECK)->EnableWindow(TRUE);
|
||
|
GetDlgItem(IDC_HARDFAULT_CHECK)->EnableWindow(TRUE);
|
||
|
GetDlgItem(IDC_IMAGELOAD_CHECK)->EnableWindow(TRUE);
|
||
|
GetDlgItem(IDC_REGISTRY_CHECK)->EnableWindow(TRUE);
|
||
|
GetDlgItem(IDC_FILEIO_CHECK)->EnableWindow(TRUE);
|
||
|
|
||
|
//
|
||
|
// Disable all controls associated with other radio buttons
|
||
|
//
|
||
|
m_pdbFileName.EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_PDB_BROWSE_BUTTON)->EnableWindow(FALSE);
|
||
|
m_ctlFileName.EnableWindow(FALSE);
|
||
|
GetDlgItem(IDC_CTL_BROWSE_BUTTON)->EnableWindow(FALSE);
|
||
|
m_controlGuidName.EnableWindow(FALSE);
|
||
|
}
|
||
|
|
||
|
void CProviderControlGuidDlg::OnBnClickedOk()
|
||
|
{
|
||
|
CString str;
|
||
|
|
||
|
if(BST_CHECKED == IsDlgButtonChecked(IDC_PDB_SELECT_RADIO)) {
|
||
|
m_pdbFileName.GetWindowText(m_pTraceSession->m_pdbFile);
|
||
|
|
||
|
EndDialog(1);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if(BST_CHECKED == IsDlgButtonChecked(IDC_CTL_SELECT_RADIO)) {
|
||
|
m_ctlFileName.GetWindowText(m_pTraceSession->m_ctlFile);
|
||
|
|
||
|
EndDialog(1);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if(BST_CHECKED == IsDlgButtonChecked(IDC_MANUAL_SELECT_RADIO)) {
|
||
|
m_controlGuidName.GetWindowText(str);
|
||
|
m_pTraceSession->m_controlGuid.Add(str);
|
||
|
|
||
|
EndDialog(1);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
if(BST_CHECKED == IsDlgButtonChecked(IDC_KERNEL_LOGGER_SELECT_RADIO)) {
|
||
|
//
|
||
|
// Convert the system kernel logger control GUID to a string
|
||
|
//
|
||
|
GuidToString(SystemTraceControlGuid, str);
|
||
|
|
||
|
m_pTraceSession->m_controlGuid.Add(str);
|
||
|
|
||
|
m_pTraceSession->m_bKernelLogger = TRUE;
|
||
|
|
||
|
//
|
||
|
// Get the kernel logger options
|
||
|
//
|
||
|
m_bProcess = ((CButton *)GetDlgItem(IDC_PROCESS_CHECK))->GetCheck();
|
||
|
m_bThread = ((CButton *)GetDlgItem(IDC_THREAD_CHECK))->GetCheck();
|
||
|
m_bDisk = ((CButton *)GetDlgItem(IDC_DISK_CHECK))->GetCheck();
|
||
|
m_bNet = ((CButton *)GetDlgItem(IDC_NET_CHECK))->GetCheck();
|
||
|
m_bFileIO = ((CButton *)GetDlgItem(IDC_FILEIO_CHECK))->GetCheck();
|
||
|
m_bPageFault = ((CButton *)GetDlgItem(IDC_PAGEFAULT_CHECK))->GetCheck();
|
||
|
m_bHardFault = ((CButton *)GetDlgItem(IDC_HARDFAULT_CHECK))->GetCheck();
|
||
|
m_bImageLoad = ((CButton *)GetDlgItem(IDC_IMAGELOAD_CHECK))->GetCheck();
|
||
|
m_bRegistry = ((CButton *)GetDlgItem(IDC_REGISTRY_CHECK))->GetCheck();
|
||
|
|
||
|
EndDialog(1);
|
||
|
return;
|
||
|
}
|
||
|
|
||
|
EndDialog(2);
|
||
|
}
|