#include <ntddk.h>
//#include <winuser.h>
#include <wchar.h>
#define AllDeviceNum 10

typedef struct _MY_DEVICEEXTEN
{
PDEVICE_OBJECT MyDevice;
UNICODE_STRING MyDeviceName;
UNICODE_STRING MySybolic;
ULONG MyDeviceNum;
}MY_DEVICEEXTEN,*PMY_DEVICEEXTEN;

VOID MyUnload(IN PDRIVER_OBJECT mydriver);
NTSTATUS CrDevice(IN PDRIVER_OBJECT mydriver);

#include "head.h"

NTSTATUS DriverEntry(IN PDRIVER_OBJECT mydriver,IN PUNICODE_STRING myregpath)
{
mydriver->DriverUnload=MyUnload;
CrDevice(mydriver);
return STATUS_SUCCESS;

}
NTSTATUS CrDevice(IN PDRIVER_OBJECT mydriver)
{
NTSTATUS status;
ULONG i;
PDEVICE_OBJECT PNextDevice;
PDEVICE_OBJECT PDevice;
UNICODE_STRING DeviceName[AllDeviceNum];//array
UNICODE_STRING DeviceSymbolic[AllDeviceNum];
LPWSTR Buffer;
LPWSTR SBuffer;
PMY_DEVICEEXTEN PDevExten;
PDevice=NULL;//point device of createdevice
PDevExten=NULL;//point exent
Buffer=NULL;//point devicename
SBuffer=NULL; //point symbolic
PNextDevice=NULL;//point next device

for (i=0;i<AllDeviceNum;i++)
{//

Buffer=(PWSTR)ExAllocatePool(PagedPool,512);
memset(Buffer,0,512);
// wsprintfW(Buffer,L"\\Device\\MyDevice%d",i);
swprintf(Buffer,L"\\Device\\MyDevice%d",i);
RtlInitUnicodeString(&DeviceName,Buffer);
//------------------------------------------------------------
SBuffer=(PWSTR)ExAllocatePool(PagedPool,512);
memset(SBuffer,0,512);
swprintf(SBuffer,L"\\??\\MyDevice%d",i);
RtlInitUnicodeString(&DeviceSymbolic,SBuffer);
//------------------------------------------------------------
status=IoCreateDevice(mydriver,sizeof(MY_DEVICEEXTEN),&DeviceName,FILE_DEVICE_UNKNOWN,0,TRUE,&PDevice);
if (!NT_SUCCESS(status))
{
KdPrint(("Error\n"));
return status;
}
KdPrint(("%wZ\n",&DeviceName));

PDevExten=(PMY_DEVICEEXTEN)PDevice->DeviceExtension;
memset(PDevExten,0,sizeof(MY_DEVICEEXTEN));
PDevExten->MyDevice=PDevice; //save deviceobject
PDevExten->MyDeviceName=DeviceName;//save devicename
PDevExten->MySybolic=DeviceSymbolic;//save devicesymbolic
PDevExten->MyDeviceNum=i;//save num
status=IoCreateSymbolicLink(&DeviceSymbolic,&DeviceName);
if (!NT_SUCCESS(status))
{
IoDeleteDevice(PDevice);
return status;
}
KdPrint(("%wZ\n",&DeviceSymbolic));
ExFreePool(Buffer);
ExFreePool(SBuffer);
Buffer=NULL;
SBuffer=NULL;
PDevice->NextDevice=PNextDevice;
PNextDevice=PDevice;
}//
mydriver->DeviceObject=PNextDevice; //point first device
return STATUS_SUCCESS;
}

VOID MyUnload(IN PDRIVER_OBJECT mydriver)
{
PDEVICE_OBJECT UnDevice;
PMY_DEVICEEXTEN UnDevExten;

UnDevice=mydriver->DeviceObject;//first device
KdPrint(("Start Unload\n"));
while (UnDevice!=NULL)
{
PDEVICE_OBJECT UnLoadDevice;
UnDevExten=(PMY_DEVICEEXTEN)UnDevice->DeviceExtension;
UnLoadDevice=UnDevExten->MyDevice;
KdPrint(("%wZ\n",&UnDevExten->MyDeviceName)); //devicename
KdPrint(("第%d个设备!\n",UnDevExten->MyDeviceNum)); //devicenum
IoDeleteSymbolicLink(&UnDevExten->MySybolic);//delete symbolic
IoDeleteDevice(UnLoadDevice); //delete device
UnDevice=UnDevice->NextDevice;
}

KdPrint(("End Unload\n"));
}