control.cpp

配置PCI配置空间的驱动程序源代码

// Control.cpp -- IOCTL handlers for PCIConf driver
// Copyright (C) 1999 by Walter Oney
// All rights reserved

#include "stddcls.h"
#include "driver.h"
#include "ioctls.h"

///////////////////////////////////////////////////////////////////////////////

#pragma PAGEDCODE

NTSTATUS DispatchControl(PDEVICE_OBJECT fdo, PIRP Irp)
	{							// DispatchControl
	PAGED_CODE();
	PDEVICE_EXTENSION pdx = (PDEVICE_EXTENSION) fdo->DeviceExtension;

	NTSTATUS status = IoAcquireRemoveLock(&pdx->RemoveLock, Irp);
	if (!NT_SUCCESS(status))
		return CompleteRequest(Irp, status, 0);
	ULONG info = 0;

	PIO_STACK_LOCATION stack = IoGetCurrentIrpStackLocation(Irp);
	ULONG cbin = stack->Parameters.DeviceIoControl.InputBufferLength;
	ULONG cbout = stack->Parameters.DeviceIoControl.OutputBufferLength;
	ULONG code = stack->Parameters.DeviceIoControl.IoControlCode;

	switch (code)
		{						// process request

	case IOCTL_GET_PCI_BUF:				// code == 0x800
		{						// IOCTL_GET_VERSION_BUFFERED
			PULONG pData = (PULONG) Irp->AssociatedIrp.SystemBuffer;
			
			unsigned long iobase, ioa0, iod;
			int bus, device, offset;
			bus=*pData;
			device=*(pData+1);
			iobase = 0x80000000 + bus * 0x10000+ (device * 8 ) * 0x100;
			ioa0 = iobase + 0;
			WRITE_PORT_ULONG((PULONG)0xcf8, ioa0);
			iod = READ_PORT_ULONG((PULONG)0xcfc);
			if(iod != 0xffffffff)
			{
				for(offset=0; offset<16; offset++)
				{
					ioa0 = iobase + offset*4;
					WRITE_PORT_ULONG((PULONG)0xcf8, ioa0);
					iod = READ_PORT_ULONG((PULONG)0xcfc);
					*(pData+offset)=iod;
				}
				info=offset*4;
			}
			else
			{
				*pData=0xffffffff;
				info=4;
			}
			break;
		}						// IOCTL_GET_VERSION_BUFFERED

	case IOCTL_GET_PCI_INFO:				// code == 0x801
		{						// IOCTL_GET_PCI_INFO CTL_CODE
			unsigned long iobase, ioa0, iod;
			int bus, device, offset;
			
			KdPrint(("--- Now in IOCTL_GET_PORT section! ---"));
			for(bus=0; bus<5; bus++)
			{
				for(device=0; device<32; device++)
				{
					iobase = 0x80000000 + bus * 0x10000+ (device * 8 ) * 0x100;
					ioa0 = iobase + 0;
					WRITE_PORT_ULONG((PULONG)0xcf8, ioa0);
					iod = READ_PORT_ULONG((PULONG)0xcfc);
					if(iod != 0xffffffff)
					{
						KdPrint(("        "));
						KdPrint(("------------------------"));
						KdPrint(("Bus = %x\tDevice = %x", bus, device));
						KdPrint(("ID# = %lx ", iod));
						for(offset=0; offset<16; offset++)
						{
							ioa0 = iobase + offset*4;
							WRITE_PORT_ULONG((PULONG)0xcf8, ioa0);
							iod = READ_PORT_ULONG((PULONG)0xcfc);
							KdPrint(("%x : %8lx", offset, iod));
						}
					}
				}
			}
			break;
		}						// IOCTL_GET_PCI_INFO CTL_CODE

	case IOCTL_GET_VERSION_BUFFERED:				// code == 0x802
		{						// IOCTL_GET_PCI_BUF
			if (cbout < sizeof(ULONG))
			{					// not enough output data
				status = STATUS_INVALID_BUFFER_SIZE;
				break;
			}					// not enough output data
			
			PULONG pversion = (PULONG) Irp->AssociatedIrp.SystemBuffer;
			info = sizeof(ULONG);
			*pversion = 0x04000A;
			break;
		}						// IOCTL_GET_PCI_BUF

	default:
		status = STATUS_INVALID_DEVICE_REQUEST;
		break;

		}						// process request

	IoReleaseRemoveLock(&pdx->RemoveLock, Irp);
	return CompleteRequest(Irp, status, info);
	}							// DispatchControl