charsample.cpp

来自「windows 2000/xpWDM设备驱动程序开发光盘代码」· C++ 代码 · 共 114 行

// CharSample.cpp
//
// Generated by DriverWizard version DriverStudio 3.1.0 (Build 1722)
// Requires Compuware's DriverWorks classes
//

#define VDW_MAIN

#include <vdw.h>
#include "CharSample.h"
#include "CharSampleDevice.h"

#pragma hdrstop("CharSample.pch")

// Generated by DriverWizard version DriverStudio 3.1.0 (Build 1722)

// Set a default 32-bit tag value to be stored with each heap block
// allocated by operator new. Use BoundsChecker to view the memory pool.
// This value can be overridden using the global function SetPoolTag().
POOLTAG DefaultPoolTag('rahC');

/////////////////////////////////////////////////////////////////////
// Begin INIT section
#pragma code_seg("INIT")

DECLARE_DRIVER_CLASS(CharSample, NULL)
/////////////////////////////////////////////////////////////////////
//  CharSample::DriverEntry
//
//	Routine Description:
//		This is the first entry point called by the system when the
//		driver is loaded.
// 
//	Parameters:
//		RegistryPath - String used to find driver parameters in the
//			registry.  To locate CharSample look for:
//			HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\CharSample
//
//	Return Value:
//		NTSTATUS - Return STATUS_SUCCESS if no errors are encountered.
//			Any other indicates to the system that an error has occured.
//
//	Comments:
//

NTSTATUS CharSample::DriverEntry(PUNICODE_STRING RegistryPath)
{

	// The following macro simply allows compilation at Warning Level 4
	// If you reference this parameter in the function simply remove the macro.
	UNREFERENCED_PARAMETER(RegistryPath);

	m_Unit = 0;
	return STATUS_SUCCESS;
}

// End INIT section
/////////////////////////////////////////////////////////////////////
#pragma code_seg()

/////////////////////////////////////////////////////////////////////
//  CharSample::AddDevice
//
//	Routine Description:
//		Called when the system detects a device for which this
//		driver is responsible.
//
//	Parameters:
//		Pdo - Physical Device Object. This is a pointer to a system device
//			object that represents the physical device.
//
//	Return Value:
//		NTSTATUS - Success or failure code.
//
//	Comments:
//		This function creates the Functional Device Object, or FDO. The FDO
//		enables this driver to handle requests for the physical device. 
//
NTSTATUS CharSample::AddDevice(PDEVICE_OBJECT Pdo)
{

    // Create the device object. Note that we used a form of "placement" new,
	// that is a member operator of KDevice.  This form will use storage
	// allocated by the system in the device object's device to store our
	// class instance.
	CharSampleDevice * pDevice = new (
			NULL,
			FILE_DEVICE_UNKNOWN,
			NULL,
			0,
			DO_DIRECT_IO
			)
		CharSampleDevice(Pdo, m_Unit);

	if (pDevice == NULL)
	{
	    return STATUS_INSUFFICIENT_RESOURCES;
	}

	NTSTATUS status = pDevice->ConstructorStatus();

	if ( !NT_SUCCESS(status) )
	{
		delete pDevice;
	}
	else
	{
		m_Unit++;
	}

	return status;
}

// EOF