|
|
//
// Driver Verifier UI
// Copyright (c) Microsoft Corporation, 1999
//
//
//
// module: SDrvPage.cpp
// author: DMihai
// created: 11/1/00
//
// Description:
//
#include "stdafx.h"
#include <Cderr.h>
#include "verifier.h"
#include "SDrvPage.h"
#include "VrfUtil.h"
#include "VGlobal.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__; #endif
//
// Help IDs
//
static DWORD MyHelpIds[] = { IDC_SELDRV_LIST, IDH_DV_SelectDriversToVerify, IDC_SELDRV_ADD_BUTTON, IDH_DV_Addbut_UnloadedDrivers, 0, 0 };
/////////////////////////////////////////////////////////////////////////////
// CSelectDriversPage property page
IMPLEMENT_DYNCREATE(CSelectDriversPage, CVerifierPropertyPage)
CSelectDriversPage::CSelectDriversPage() : CVerifierPropertyPage(CSelectDriversPage::IDD) { //{{AFX_DATA_INIT(CSelectDriversPage)
// NOTE: the ClassWizard will add member initialization here
//}}AFX_DATA_INIT
m_pParentSheet = NULL;
m_nSortColumnIndex = 1; m_bAscendSortVerified = FALSE; m_bAscendSortDrvName = FALSE; m_bAscendSortProvName = FALSE; m_bAscendSortVersion = FALSE; }
CSelectDriversPage::~CSelectDriversPage() { }
void CSelectDriversPage::DoDataExchange(CDataExchange* pDX) { CVerifierPropertyPage::DoDataExchange(pDX); //{{AFX_DATA_MAP(CSelectDriversPage)
DDX_Control(pDX, IDC_SELDRV_NEXT_DESCR_STATIC, m_NextDescription); DDX_Control(pDX, IDC_SELDRV_LIST, m_DriversList); //}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CSelectDriversPage, CVerifierPropertyPage) //{{AFX_MSG_MAP(CSelectDriversPage)
ON_BN_CLICKED(IDC_SELDRV_ADD_BUTTON, OnAddButton) ON_NOTIFY(LVN_COLUMNCLICK, IDC_SELDRV_LIST, OnColumnclickSeldrvList) ON_WM_CONTEXTMENU() ON_MESSAGE( WM_HELP, OnHelp ) //}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
VOID CSelectDriversPage::SetupListHeader() { CString strTitle; CRect rectWnd; LVCOLUMN lvColumn;
//
// The list's rectangle
//
m_DriversList.GetClientRect( &rectWnd );
ZeroMemory( &lvColumn, sizeof( lvColumn ) );
lvColumn.mask = LVCF_FMT | LVCF_SUBITEM | LVCF_TEXT | LVCF_WIDTH; lvColumn.fmt = LVCFMT_LEFT;
//
// Column 0
//
VERIFY( strTitle.LoadString( IDS_VERIFICATION_STATUS ) );
lvColumn.iSubItem = 0; lvColumn.cx = (int)( rectWnd.Width() * 0.08 ); lvColumn.pszText = strTitle.GetBuffer( strTitle.GetLength() + 1 );
if (NULL != lvColumn.pszText) { VERIFY( m_DriversList.InsertColumn( 0, &lvColumn ) != -1 ); strTitle.ReleaseBuffer(); } else { lvColumn.pszText = g_szVoidText; VERIFY( m_DriversList.InsertColumn( 0, &lvColumn ) != -1 ); }
//
// Column 1
//
VERIFY( strTitle.LoadString( IDS_DRIVERS ) );
lvColumn.iSubItem = 1; lvColumn.cx = (int)( rectWnd.Width() * 0.20 ); lvColumn.pszText = strTitle.GetBuffer( strTitle.GetLength() + 1 );
if (NULL != lvColumn.pszText) { VERIFY( m_DriversList.InsertColumn( 1, &lvColumn ) != -1 ); strTitle.ReleaseBuffer(); } else { lvColumn.pszText = g_szVoidText; VERIFY( m_DriversList.InsertColumn( 1, &lvColumn ) != -1 ); }
//
// Column 2
//
VERIFY( strTitle.LoadString( IDS_PROVIDER ) );
lvColumn.iSubItem = 2; lvColumn.cx = (int)( rectWnd.Width() * 0.47 ); lvColumn.pszText = strTitle.GetBuffer( strTitle.GetLength() + 1 );
if (NULL != lvColumn.pszText) { VERIFY( m_DriversList.InsertColumn( 2, &lvColumn ) != -1 ); strTitle.ReleaseBuffer(); } else { lvColumn.pszText = g_szVoidText; VERIFY( m_DriversList.InsertColumn( 2, &lvColumn ) != -1 ); }
//
// Column 3
//
VERIFY( strTitle.LoadString( IDS_VERSION ) );
lvColumn.iSubItem = 3; lvColumn.cx = (int)( rectWnd.Width() * 0.22 ); lvColumn.pszText = strTitle.GetBuffer( strTitle.GetLength() + 1 );
if (NULL != lvColumn.pszText) { VERIFY( m_DriversList.InsertColumn( 3, &lvColumn ) != -1 ); strTitle.ReleaseBuffer(); } else { VERIFY( m_DriversList.InsertColumn( 3, &lvColumn ) != -1 ); lvColumn.pszText = g_szVoidText; } }
/////////////////////////////////////////////////////////////////////////////
VOID CSelectDriversPage::FillTheList() { INT_PTR nDriversNo; INT_PTR nCrtDriverIndex; CDriverData *pCrtDrvData; const CDriverDataArray &DrvDataArray = g_NewVerifierSettings.m_DriversSet.m_aDriverData;
m_DriversList.DeleteAllItems();
//
// Parse the driver data array
//
nDriversNo = DrvDataArray.GetSize();
for( nCrtDriverIndex = 0; nCrtDriverIndex < nDriversNo; nCrtDriverIndex += 1) { pCrtDrvData = DrvDataArray.GetAt( nCrtDriverIndex );
ASSERT_VALID( pCrtDrvData );
AddListItem( nCrtDriverIndex, pCrtDrvData ); } }
/////////////////////////////////////////////////////////////////////////////
INT CSelectDriversPage::AddListItem( INT_PTR nIndexInArray, CDriverData *pCrtDrvData ) { INT nActualIndex; LVITEM lvItem;
ASSERT_VALID( pCrtDrvData );
nActualIndex = -1;
ZeroMemory( &lvItem, sizeof( lvItem ) );
//
// LVITEM's member pszText is not a const pointer
// so we need to GetBuffer here :-(
//
//
// Sub-item 0 - verification status - empty text and a checkbox
//
lvItem.pszText = g_szVoidText; lvItem.mask = LVIF_TEXT | LVIF_PARAM; lvItem.lParam = nIndexInArray; lvItem.iItem = m_DriversList.GetItemCount();
nActualIndex = m_DriversList.InsertItem( &lvItem );
if( nActualIndex < 0 ) { //
// Could not add an item in the list - give up
//
goto Done; }
m_DriversList.SetCheck( nActualIndex, FALSE );
//
// Sub-item 1 - driver name
//
lvItem.pszText = pCrtDrvData->m_strName.GetBuffer( pCrtDrvData->m_strName.GetLength() + 1 ); if( NULL == lvItem.pszText ) { goto Done; }
lvItem.mask = LVIF_TEXT; lvItem.iItem = nActualIndex; lvItem.iSubItem = 1; VERIFY( m_DriversList.SetItem( &lvItem ) );
pCrtDrvData->m_strName.ReleaseBuffer();
//
// Sub-item 2 - provider
//
lvItem.pszText = pCrtDrvData->m_strCompanyName.GetBuffer( pCrtDrvData->m_strCompanyName.GetLength() + 1 ); if( NULL == lvItem.pszText ) { goto Done; }
lvItem.mask = LVIF_TEXT; lvItem.iItem = nActualIndex; lvItem.iSubItem = 2;
VERIFY( m_DriversList.SetItem( &lvItem ) ); pCrtDrvData->m_strCompanyName.ReleaseBuffer();
//
// Sub-item 3 - version
//
lvItem.pszText = pCrtDrvData->m_strFileVersion.GetBuffer( pCrtDrvData->m_strFileVersion.GetLength() + 1 ); if( NULL == lvItem.pszText ) { goto Done; }
lvItem.mask = LVIF_TEXT; lvItem.iItem = nActualIndex; lvItem.iSubItem = 3;
VERIFY( m_DriversList.SetItem( &lvItem ) ); pCrtDrvData->m_strFileVersion.ReleaseBuffer();
Done: //
// All done
//
return nActualIndex; }
/////////////////////////////////////////////////////////////////////////////
BOOL CSelectDriversPage::GetNewVerifiedDriversList() { INT nListItemCount; INT nCrtListItem; INT_PTR nCrtDriversArrayIndex; BOOL bVerified; CDriverData *pCrtDrvData; const CDriverDataArray &DrvDataArray = g_NewVerifierSettings.m_DriversSet.m_aDriverData; CDriverData::VerifyDriverTypeEnum VerifyStatus; nListItemCount = m_DriversList.GetItemCount();
for( nCrtListItem = 0; nCrtListItem < nListItemCount; nCrtListItem += 1 ) { //
// Verification status for the current list item
//
bVerified = m_DriversList.GetCheck( nCrtListItem );
if( bVerified ) { VerifyStatus = CDriverData::VerifyDriverYes; } else { VerifyStatus = CDriverData::VerifyDriverNo; }
//
// Set the right verify state in our driver array
//
nCrtDriversArrayIndex = m_DriversList.GetItemData( nCrtListItem );
pCrtDrvData = DrvDataArray.GetAt( nCrtDriversArrayIndex );
ASSERT_VALID( pCrtDrvData );
pCrtDrvData->m_VerifyDriverStatus = VerifyStatus; }
return TRUE; }
/////////////////////////////////////////////////////////////
VOID CSelectDriversPage::SortTheList() { if( 0 != m_nSortColumnIndex ) { //
// Sort by driver name, provider or version
//
m_DriversList.SortItems( StringCmpFunc, (LPARAM)this ); } else { //
// Sort by verified status
//
m_DriversList.SortItems( CheckedStatusCmpFunc, (LPARAM)this ); } }
/////////////////////////////////////////////////////////////
int CALLBACK CSelectDriversPage::StringCmpFunc( LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort) { int nCmpRez = 0; BOOL bSuccess; CString strName1; CString strName2;
CSelectDriversPage *pThis = (CSelectDriversPage *)lParamSort; ASSERT_VALID( pThis );
ASSERT( 0 != pThis->m_nSortColumnIndex );
//
// Get the first name
//
bSuccess = pThis->GetColumnStrValue( lParam1, strName1 );
if( FALSE == bSuccess ) { goto Done; }
//
// Get the second name
//
bSuccess = pThis->GetColumnStrValue( lParam2, strName2 );
if( FALSE == bSuccess ) { goto Done; }
//
// Compare the names
//
nCmpRez = strName1.CompareNoCase( strName2 ); switch( pThis->m_nSortColumnIndex ) { case 1: //
// Sort by driver name
//
if( FALSE != pThis->m_bAscendSortDrvName ) { nCmpRez *= -1; }
break;
case 2: //
// Sort by provider name
//
if( FALSE != pThis->m_bAscendSortProvName ) { nCmpRez *= -1; }
break;
case 3: //
// Sort by version
//
if( FALSE != pThis->m_bAscendSortVersion ) { nCmpRez *= -1; }
break;
default: //
// Oops - how did we get here ?!?
//
ASSERT( FALSE ); break; }
Done:
return nCmpRez;
}
/////////////////////////////////////////////////////////////
int CALLBACK CSelectDriversPage::CheckedStatusCmpFunc( LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort) { int nCmpRez = 0; INT nItemIndex; BOOL bVerified1; BOOL bVerified2; LVFINDINFO FindInfo;
CSelectDriversPage *pThis = (CSelectDriversPage *)lParamSort; ASSERT_VALID( pThis );
ASSERT( 0 == pThis->m_nSortColumnIndex );
//
// Find the first item
//
ZeroMemory( &FindInfo, sizeof( FindInfo ) ); FindInfo.flags = LVFI_PARAM; FindInfo.lParam = lParam1;
nItemIndex = pThis->m_DriversList.FindItem( &FindInfo );
if( nItemIndex < 0 ) { ASSERT( FALSE );
goto Done; }
bVerified1 = pThis->m_DriversList.GetCheck( nItemIndex );
//
// Find the second item
//
FindInfo.flags = LVFI_PARAM; FindInfo.lParam = lParam2;
nItemIndex = pThis->m_DriversList.FindItem( &FindInfo );
if( nItemIndex < 0 ) { ASSERT( FALSE );
goto Done; }
bVerified2 = pThis->m_DriversList.GetCheck( nItemIndex );
//
// Compare them
//
if( bVerified1 != bVerified2 ) { if( FALSE != bVerified1 ) { nCmpRez = 1; } else { nCmpRez = -1; }
if( FALSE != pThis->m_bAscendSortVerified ) { nCmpRez *= -1; } }
Done:
return nCmpRez;
}
/////////////////////////////////////////////////////////////
BOOL CSelectDriversPage::GetColumnStrValue( LPARAM lItemData, CString &strName ) { CDriverData *pCrtDrvData;
pCrtDrvData = g_NewVerifierSettings.m_DriversSet.m_aDriverData.GetAt( (INT_PTR) lItemData );
ASSERT_VALID( pCrtDrvData ); switch( m_nSortColumnIndex ) { case 1: //
// Sort by driver name
//
strName = pCrtDrvData->m_strName; break;
case 2: //
// Sort by provider name
//
strName = pCrtDrvData->m_strCompanyName;
break;
case 3: //
// Sort by version
//
strName = pCrtDrvData->m_strFileVersion;
break;
default: //
// Oops - how did we get here ?!?
//
ASSERT( FALSE ); break;
} return TRUE; }
/////////////////////////////////////////////////////////////////////////////
BOOL CSelectDriversPage::OnSetActive() { //
// The wizard has at least one more step (display the summarry)
//
if( (FALSE == g_bShowDiskPropertyPage) && (FALSE == g_NewVerifierSettings.m_aDiskData.VerifyAnyDisk()) ) { //
// Disk verifier is disabled.
//
m_pParentSheet->SetWizardButtons( PSWIZB_BACK | PSWIZB_FINISH );
VrfSetWindowText( m_NextDescription, IDS_SELDRV_PAGE_NEXT_DESCR_FINISH ); } else { //
// Disk verifier is enabled.
//
m_pParentSheet->SetWizardButtons( PSWIZB_BACK | PSWIZB_NEXT );
VrfSetWindowText( m_NextDescription, IDS_SELDRV_PAGE_NEXT_DESCR_NEXT ); }
return CVerifierPropertyPage::OnSetActive(); }
/////////////////////////////////////////////////////////////////////////////
// CSelectDriversPage message handlers
BOOL CSelectDriversPage::OnInitDialog() { CVerifierPropertyPage::OnInitDialog();
//
// setup the list
//
m_DriversList.SetExtendedStyle( LVS_EX_FULLROWSELECT | LVS_EX_CHECKBOXES | m_DriversList.GetExtendedStyle() );
SetupListHeader(); FillTheList(); SortTheList();
return TRUE; // return TRUE unless you set the focus to a control
// EXCEPTION: OCX Property Pages should return FALSE
}
/////////////////////////////////////////////////////////////////////////////
BOOL CSelectDriversPage::OnWizardFinish() { BOOL bExitTheApp;
bExitTheApp = FALSE;
if( GetNewVerifiedDriversList() ) { if( FALSE == g_NewVerifierSettings.m_DriversSet.ShouldVerifySomeDrivers() ) { VrfErrorResourceFormat( IDS_SELECT_AT_LEAST_ONE_DRIVER );
goto Done; }
g_NewVerifierSettings.SaveToRegistry(); //
// Exit the app
//
bExitTheApp = CVerifierPropertyPage::OnWizardFinish(); }
Done:
return bExitTheApp; }
/////////////////////////////////////////////////////////////////////////////
LRESULT CSelectDriversPage::OnWizardNext() { LRESULT lNextPageId;
lNextPageId = -1;
//
// We can get here only if the disk integrity checking is enabled.
//
ASSERT( FALSE != g_bShowDiskPropertyPage || FALSE != g_NewVerifierSettings.m_aDiskData.VerifyAnyDisk() );
if( GetNewVerifiedDriversList() ) { if( FALSE == g_NewVerifierSettings.m_DriversSet.ShouldVerifySomeDrivers() ) { VrfErrorResourceFormat( IDS_SELECT_AT_LEAST_ONE_DRIVER );
goto Done; }
lNextPageId = IDD_DISK_LIST_PAGE;
GoingToNextPageNotify( lNextPageId ); }
Done:
return lNextPageId; }
/////////////////////////////////////////////////////////////////////////////
#define VRF_MAX_CHARS_FOR_OPEN 4096
void CSelectDriversPage::OnAddButton() { POSITION pos; DWORD dwRetValue; DWORD dwOldMaxFileName = 0; DWORD dwErrorCode; INT nFileNameStartIndex; INT nNewListItemIndex; INT_PTR nResult; INT_PTR nNewDriverDataIndex; CDriverData *pNewDrvData; TCHAR szDriversDir[ _MAX_PATH ]; TCHAR szAppTitle[ _MAX_PATH ]; TCHAR *szFilesBuffer = NULL; TCHAR *szOldFilesBuffer = NULL; CString strPathName; CString strFileName;
CFileDialog fileDlg( TRUE, // open file
_T( "sys" ), // default extension
NULL, // no initial file name
OFN_ALLOWMULTISELECT | // multiple selection
OFN_HIDEREADONLY | // hide the "open read-only" checkbox
OFN_NONETWORKBUTTON | // no network button
OFN_NOTESTFILECREATE | // don't test for write protection, a full disk, etc.
OFN_SHAREAWARE, // don't check the existance of file with OpenFile
_T( "Drivers (*.sys)|*.sys||" ) ); // only one filter
//
// check the max length for the returned string
//
if( fileDlg.m_ofn.nMaxFile < VRF_MAX_CHARS_FOR_OPEN ) { //
// allocate a new buffer for the file names
//
szFilesBuffer = new TCHAR[ VRF_MAX_CHARS_FOR_OPEN ]; if (NULL == szFilesBuffer) { VrfErrorResourceFormat( IDS_NOT_ENOUGH_MEMORY ); goto Done; }
szFilesBuffer[ 0 ] = (TCHAR)0;
if( szFilesBuffer != NULL ) { //
// Save the old buffer address and length
//
dwOldMaxFileName = fileDlg.m_ofn.nMaxFile; szOldFilesBuffer = fileDlg.m_ofn.lpstrFile; //
// Set the new buffer address and length
//
fileDlg.m_ofn.lpstrFile = szFilesBuffer; fileDlg.m_ofn.nMaxFile = VRF_MAX_CHARS_FOR_OPEN; } }
//
// Dialog title
//
if( VrfLoadString( IDS_APPTITLE, szAppTitle, ARRAY_LENGTH( szAppTitle ) ) ) { fileDlg.m_ofn.lpstrTitle = szAppTitle; }
//
// We change directory first time we try this to %windir%\system32\drivers
//
dwRetValue = ExpandEnvironmentStrings( _T( "%windir%\\system32\\drivers" ), szDriversDir, ARRAY_LENGTH( szDriversDir ) );
if( dwRetValue > 0 && dwRetValue <= ARRAY_LENGTH( szDriversDir ) ) { fileDlg.m_ofn.lpstrInitialDir = szDriversDir; }
//
// Show the file selection dialog
//
nResult = fileDlg.DoModal();
switch( nResult ) { case IDOK: break;
case IDCANCEL: goto cleanup;
default: dwErrorCode = CommDlgExtendedError();
if( dwErrorCode == FNERR_BUFFERTOOSMALL ) { VrfErrorResourceFormat( IDS_TOO_MANY_FILES_SELECTED ); } else { VrfErrorResourceFormat( IDS_CANNOT_OPEN_FILES, dwErrorCode ); }
goto cleanup; }
//
// Parse all the selected files and try to enable them for verification
//
pos = fileDlg.GetStartPosition();
while( pos != NULL ) { //
// Get the full path for the next file
//
strPathName = fileDlg.GetNextPathName( pos );
//
// Split only the file name, without the directory
//
nFileNameStartIndex = strPathName.ReverseFind( _T( '\\' ) ); if( nFileNameStartIndex < 0 ) { //
// This shoudn't happen but you never know :-)
//
nFileNameStartIndex = 0; } else { //
// skip the backslash
//
nFileNameStartIndex += 1; }
strFileName = strPathName.Right( strPathName.GetLength() - nFileNameStartIndex );
//
// Try to add this driver to our global driver list
//
if( g_NewVerifierSettings.m_DriversSet.IsDriverNameInList( strFileName ) ) { VrfErrorResourceFormat( IDS_DRIVER_IS_ALREADY_IN_LIST, (LPCTSTR) strFileName ); } else { nNewDriverDataIndex = g_NewVerifierSettings.m_DriversSet.AddNewDriverData( strFileName );
if( nNewDriverDataIndex >= 0 ) { //
// Force refreshing the unsigned driver data
//
g_NewVerifierSettings.m_DriversSet.m_bUnsignedDriverDataInitialized = FALSE;
//
// Add a new item to our list, for the new driver
//
pNewDrvData = g_NewVerifierSettings.m_DriversSet.m_aDriverData.GetAt( nNewDriverDataIndex ); ASSERT_VALID( pNewDrvData );
nNewListItemIndex = AddListItem( nNewDriverDataIndex, pNewDrvData );
if( nNewListItemIndex >= 0 ) { m_DriversList.EnsureVisible( nNewListItemIndex, TRUE ); } } } }
cleanup: if( szFilesBuffer != NULL ) { fileDlg.m_ofn.nMaxFile = dwOldMaxFileName; fileDlg.m_ofn.lpstrFile = szOldFilesBuffer;
delete [] szFilesBuffer; }
Done:
NOTHING; }
/////////////////////////////////////////////////////////////////////////////
void CSelectDriversPage::OnColumnclickSeldrvList(NMHDR* pNMHDR, LRESULT* pResult) { NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
switch( pNMListView->iSubItem ) { case 0: //
// Clicked on the "verified" column
//
if( m_nSortColumnIndex == pNMListView->iSubItem ) { //
// Change the current ascend/descend order for this column
//
m_bAscendSortVerified = !m_bAscendSortVerified; }
break;
case 1: //
// Clicked on the driver name column
//
if( m_nSortColumnIndex == pNMListView->iSubItem ) { //
// Change the current ascend/descend order for this column
//
m_bAscendSortDrvName = !m_bAscendSortDrvName; }
break;
case 2: //
// Clicked on the provider column
//
if( m_nSortColumnIndex == pNMListView->iSubItem ) { //
// Change the current ascend/descend order for this column
//
m_bAscendSortProvName = !m_bAscendSortProvName; }
break;
case 3: //
// Clicked on the version column
//
if( m_nSortColumnIndex == pNMListView->iSubItem ) { //
// Change the current ascend/descend order for this column
//
m_bAscendSortVersion = !m_bAscendSortVersion; }
break;
default: //
// Oops - how did we get here ?!?
//
ASSERT( FALSE ); goto Done; }
m_nSortColumnIndex = pNMListView->iSubItem;
SortTheList();
Done:
*pResult = 0; }
/////////////////////////////////////////////////////////////
LONG CSelectDriversPage::OnHelp( WPARAM wParam, LPARAM lParam ) { LONG lResult = 0; LPHELPINFO lpHelpInfo = (LPHELPINFO)lParam;
::WinHelp( (HWND) lpHelpInfo->hItemHandle, g_szVerifierHelpFile, HELP_WM_HELP, (DWORD_PTR) MyHelpIds );
return lResult; }
/////////////////////////////////////////////////////////////////////////////
void CSelectDriversPage::OnContextMenu(CWnd* pWnd, CPoint point) { ::WinHelp( pWnd->m_hWnd, g_szVerifierHelpFile, HELP_CONTEXTMENU, (DWORD_PTR) MyHelpIds ); }
|