charfilterinst.cpp

windows 2000/XP WDM设备驱动程序开发 附书光盘 武安河著

// CharFilterInst.cpp : Class CharFilterInstaller implementation.
//

#include "stdafx.h"
#include "devinfo.h"
#include "filterinstall.h"
#include "CharFilterInst.h"


//////////////////////////////////////////////////////////////////////////////
// static initialization

DECLARE_DEVICE_INFO

LPCTSTR CharFilterInstaller::m_szFilterName	 = _T("CharFilter");
//注册表CharSample中的UpperFilters信息，见图11-10
LPCTSTR CharFilterInstaller::m_szFilterBinary	 = _T("System32\\Drivers\\CharFilter.sys");
LPCTSTR CharFilterInstaller::m_szFilterSys	 = _T("..\\sys\\objchk\\i386\\CharFilter.sys");	
//安装文件及路径，默认是当前路径
LPCTSTR CharFilterInstaller::m_szDeviceDesc	 = _T("CharSample Driver");
	// TODO:  Customize which class GUID for use in selecting a set of devices
	// The GUID represents an interface class if the DIGCF_DEVICEINTERFACE flag is set; 
	// otherwise, it represents a setup class 
GUID    CharFilterInstaller::m_Guid			 = {0};
	// TODO:  Select Flags to use:  
	// DIGCF_ALLCLASSES, DIGCF_DEVICEINTERFACE, DIGCF_PRESENT, DIGCF_PROFILE  
DWORD   CharFilterInstaller::m_Flags			 = DIGCF_ALLCLASSES;  // This will select all devices in all classes


//////////////////////////////////////////////////////////////////////////////
// CharFilterInstaller implementation

/////////////////////////////////////////////////////////////////////
//  CharFilterInstaller::doUpper
// 
//	Routine Description:
//		This routine determines if the given device should be 
//		filtered by the filter driver. 
//
//	Parameters:
//		pDevInfo	- pointer to a CDeviceInfo object representing 
//			an instance of a device in the system.  Use convenience  
//			methods to access information about the device.
//
//	Return Value:
//		TRUE if the device should be filtered (i.e. install the filter
//			driver for this device)
//
//	Comments:
//		The job of this method is to test if we want to install our filter
//	for the given device.  To install the filter for a specific device,
//	use a device property to identify the device to be filtered (i.e. Hardware 
//	ID, Friendly Name, Device Description, etc.)  Use the convenience
//	methods of CDeviceInfo to obtain the device property.  Or, to filter
//	all devices of a class, return TRUE;
//
BOOL CharFilterInstaller::doUpper(CDeviceInfo* pDevInfo)
{ 
	// TODO:  Use a device property to identify the device to 
	// be filtered.  i.e. Hardware ID, Friendly Name, Device Description, etc.

	// For now, use the device description
	BOOL bRet = FALSE;
	try {
		CDeviceRegProp DeviceDesc;
		DWORD Status = pDevInfo->GetDeviceDescription(DeviceDesc);

		if (ERROR_SUCCESS == Status) {

			// Compare the Device Description of the given device with the Device
			// Description of the device to be filtered
			bRet = _tcsicmp((LPCTSTR)(PBYTE) DeviceDesc, m_szDeviceDesc) == 0;
		}
	}
	catch(...) {
		bRet = FALSE;
	}

	return bRet;
}


/////////////////////////////////////////////////////////////////////
//  CharFilterInstaller::AddUpperFilter
//
//	Routine Description:
//		This routine is called to add a upper filter to the device.
//
//	Parameters:
//		pDevInfo	- pointer to a CDeviceInfo object representing
//			an instance of a device in the system.  Use convenience
//			methods to access information about the device.
//
//	Return Value:
//		ERROR_CONTINUE - don't restart the device
//		ERROR_SUCCESS - restart the device if the filter is installed
//
//	Comments:
//		The job of this method is to Add the Filter Name to the 
//	UpperFilters Registry Property of the device.  The base class
//  method implementation does just that.  This override will check
//	if the filter is already installed.  If it is already installed,
//	it will just return.  Otherwise, it will call the base class 
//	method to do the work.
//
DWORD CharFilterInstaller::AddUpperFilter(CDeviceInfo* pDevInfo)
{
	// Guard against installing 2 instances of the filter
	DWORD dwStatus = ERROR_CONTINUE;

	BOOLEAN bInstallFilter = TRUE;

	CDeviceRegProp DevLowerFilters;
	DWORD Status = pDevInfo->GetLowerFilters(DevLowerFilters);

	if (ERROR_SUCCESS == Status) {

		bInstallFilter = _tcsstr((LPCTSTR)(PBYTE) DevLowerFilters, m_szFilterName) == NULL;
	}

	if (bInstallFilter) {
		dwStatus = CDeviceFilterInstaller::AddUpperFilter(pDevInfo);
	}
	
	return dwStatus;
}


//////////////////////////////////////////////////////////////////////////////
// RemoveService
//
// This function removes a service from the registry
//
DWORD RemoveService(LPCTSTR pszFilterName)
{
 	DWORD dwRet = ERROR_SUCCESS;

	SC_HANDLE schSCManager = OpenSCManager( 
									NULL,					// computer name
									NULL,					// SCM database name
									SC_MANAGER_ALL_ACCESS	// access type
									);

	if (schSCManager) {
		SC_HANDLE schService = OpenService(  
									schSCManager,		// handle to SCM database
									pszFilterName,		// service name
									SERVICE_ALL_ACCESS  // access
									);

		if (schService) {
			dwRet = DeleteService(schService);
			CloseServiceHandle(schService);
		}
		else {
			dwRet = GetLastError();
		}

 		CloseServiceHandle(schSCManager);
	}
	else {
		dwRet = GetLastError();
	}

    return dwRet;
}


//////////////////////////////////////////////////////////////////////////////
// AddService
//
// This function adds a service to the registry 
//
DWORD AddService(LPCTSTR pszFilterName, LPCTSTR pszServiceBinary)
{
	DWORD dwRet = ERROR_SUCCESS;

	SC_HANDLE schSCManager = OpenSCManager( 
								NULL,				  // computer name
								NULL,				  // SCM database name
								SC_MANAGER_ALL_ACCESS // access type
								);

	if (schSCManager) {

		SC_HANDLE schService = CreateService( 
									schSCManager,          // SCManager database
									pszFilterName,         // name of service
									pszFilterName,         // name to display
									SERVICE_ALL_ACCESS,    // desired access
									SERVICE_KERNEL_DRIVER, // service type
									SERVICE_DEMAND_START,  // start type
									SERVICE_ERROR_NORMAL,  // error control type
									pszServiceBinary,      // service's binary
									NULL,                  // no load ordering group
									NULL,                  // no tag identifier
									NULL,                  // no dependencies
									NULL,                  // LocalSystem account
									NULL                   // no password
									);
		if (schService) {
			CloseServiceHandle(schService);
		}
		else {
			dwRet = GetLastError();
		}

 		CloseServiceHandle(schSCManager);
	}
	else {
		dwRet = GetLastError();
	}

	return dwRet;
}