Выгрузка драйвера. Страница 1. Раздел «Drivers». Форум программистов «Весельчак У»

MherX

Гость

« : 20-01-2010 15:28 »

Привет всем, я новичок в этой области, и вот у меня возникли такие вопросы:
Я должен симулировать виртуальный HDD типа TrueCrypt : вот фрагмент из кода

Код:

#include <ntddk.h>

NTSTATUS DriverEntry  (IN PDRIVER_OBJECT DriverObject,IN PUNICODE_STRING UniRegistryPath);
NTSTATUS VDskAddDevice(IN PDRIVER_OBJECT DriverObject,IN PDEVICE_OBJECT PhysicalDeviceObject);
VOID     VDskUnload   (IN PDRIVER_OBJECT DriverObject);

NTSTATUS VDskCreateCreate		  (IN PDEVICE_OBJECT pDeviceObject, IN PIRP pIrp);
NTSTATUS VDskCreateClose		  (IN PDEVICE_OBJECT pDeviceObject, IN PIRP pIrp);
NTSTATUS VDskReadWrite		  (IN PDEVICE_OBJECT pDeviceObject, IN PIRP pIrp);
NTSTATUS VDskDispatchPnp		  (IN PDEVICE_OBJECT pDeviceObject, IN PIRP pIrp);
NTSTATUS VDskDispatchPower		  (IN PDEVICE_OBJECT pDeviceObject, IN PIRP pIrp);
NTSTATUS VDskIOCtl			  (IN PDEVICE_OBJECT pDeviceObject, IN PIRP pIrp);
NTSTATUS VDskDispatchSystemControl  (IN PDEVICE_OBJECT pDeviceObject, IN PIRP pIrp);

NTSTATUS CreateMainDeviceObject (PDRIVER_OBJECT DriverObject);

const WCHAR NT_DEVICE_NAME   [] = L"\\Device\\VDevice";
const WCHAR DOS_DEVICE_NAME[] = L"\\DosDevices\\VDeviceLink";

PDRIVER_OBJECT gDriverObject;

NTSTATUS DriverEntry(IN PDRIVER_OBJECT DriverObject,IN PUNICODE_STRING UniRegistryPath)
{
	NTSTATUS  Status = STATUS_SUCCESS;

	DbgPrint("Driver is loaded\nDriverEntry\n");

	DriverObject->MajorFunction[IRP_MJ_CREATE] = VDskCreateCreate;
	DriverObject->MajorFunction[IRP_MJ_CLOSE  ] = VDskCreateClose;
	DriverObject->MajorFunction[IRP_MJ_READ    ] = VDskReadWrite;
	DriverObject->MajorFunction[IRP_MJ_WRITE  ] = VDskReadWrite;	
	DriverObject->MajorFunction[IRP_MJ_PNP	] = VDskDispatchPnp; 
	DriverObject->MajorFunction[IRP_MJ_POWER ] = VDskDispatchPower;
	DriverObject->MajorFunction[IRP_MJ_DEVICE_CONTROL] = VDskIOCtl;
	DriverObject->MajorFunction[IRP_MJ_SYSTEM_CONTROL] = VDskDispatchSystemControl;

	DriverObject->DriverExtension->AddDevice = VDskAddDevice; 
	DriverObject->DriverUnload = VDskUnload;	

	return CreateMainDeviceObject(DriverObject);
}


NTSTATUS VDskAddDevice(IN PDRIVER_OBJECT DriverObject,IN PDEVICE_OBJECT PhysicalDeviceObject)
{
	NTSTATUS	   Status = STATUS_SUCCESS;
	UNICODE_STRING uniNtNameString, uniWin32NameString;

	DbgPrint("VDskAddDevice\n");

	RtlInitUnicodeString(&uniNtNameString, NT_DEVICE_NAME );

	Status = IoCreateDevice (
		DriverObject,
		0,	
		&uniNtNameString,
		FILE_DEVICE_UNKNOWN,	
		0,				
		FALSE,			
		&PhysicalDeviceObject
		);
	if( !NT_SUCCESS(Status) ){
		DbgPrint("Can't create device\n");
		return Status;
	}

	RtlInitUnicodeString(&uniWin32NameString, DOS_DEVICE_NAME);
	Status = IoCreateSymbolicLink(&uniWin32NameString, &uniNtNameString);
	if (!NT_SUCCESS(Status))
	{
		DbgPrint("Can't create symbol link\n");
		IoDeleteDevice( DriverObject->DeviceObject );
	}

	DbgPrint("Device created success\n");
	return Status;
}

NTSTATUS CreateMainDeviceObject (PDRIVER_OBJECT DriverObject)
{
	UNICODE_STRING Win32NameString, ntUnicodeString;

	PDEVICE_OBJECT DeviceObject;
	NTSTATUS ntStatus;
	
	RtlInitUnicodeString (&ntUnicodeString, NT_DEVICE_NAME);
	RtlInitUnicodeString (&Win32NameString, DOS_DEVICE_NAME);

	ntStatus = IoCreateDevice (
		DriverObject,
		0,
		&ntUnicodeString,
		FILE_DEVICE_UNKNOWN,
		0,
		FALSE,
		&DeviceObject);

	if (!NT_SUCCESS (ntStatus))
	{
		
		return ntStatus;/* Failed to create DeviceObject */
	}

	ntStatus = IoCreateSymbolicLink (&Win32NameString, &ntUnicodeString);

	if (!NT_SUCCESS (ntStatus))
	{
		IoDeleteDevice (DeviceObject);
		return ntStatus;
	}

	IoRegisterShutdownNotification (DeviceObject);

	return STATUS_SUCCESS;
}

VOID VDskUnload(IN PDRIVER_OBJECT DriverObject)
{
            PDEVICE_OBJECT pDeviceObject;
	UNICODE_STRING uniWin32NameString;

	DbgPrint("VDskUnload\n");

	pDeviceObject = pDriverObject->DeviceObject;

	RtlInitUnicodeString( &uniWin32NameString, DOS_DEVICE_NAME );
	IoDeleteDevice( pDriverObject->DeviceObject );

	DbgPrint("Unload is success\n");
}

Драйвер динамически не выгружается.
Заранее благодарен.


« Последнее редактирование: 20-01-2010 17:53 от Sel »	Записан

Ochkarik

Модератор

Offline
Пол: Мужской

	Автор	Тема: Выгрузка драйвера (Прочитано 24540 раз)
0 Пользователей и 12 Гостей смотрят эту тему.