//
// RecognizerDetectFileSystem
//
// 用户需要根据自己文件系统的不同改变这些代码
// 输入参数:
// Irp – 请求装配卷的IRP.
//
// 输出参数:
// None.
//
// Returns:
// STATUS_SUCCESS – 这是我们的驱动可以识别的文件系统
// Other - I/O error
//
// Notes:
// None.
//

static NTSTATUS RecognizerDetectFileSystem(PIRP Irp)
{
	NTSTATUS code =STATUS_SUCCESS;
	DISK_GEOMETRY diskGeometry;
	PARTITION_INFORMATION partitionInfo;
	PIO_STACK_LOCATION irpSp = IoGetCurrentIrpStackLocation(Irp);
	PVPB vpb = irpSp->Parameters.MountVolume.Vpb;
	PDEVICE_OBJECT mediaDeviceObject = irpSp->Parameters.MountVolume.DeviceObject;
	unsigned char* pBuffer = NULL;

	//首先构造一个可以获得卷的分区表等信息的IRP，用RecognizerIoControl传递到下层驱动。
	// IF error{
	//   return error
	//}
	// else
	//   {
	//用RecognizerReadDiskSector 读卷参数块(VPB)}
	// If read error{
	//    return error
	//}
	// else
	// {检查文件系统的标识符，如果不匹配
	// return error
	//}
	// else
	// return code STATUS_SUCCESS;
	// endif
	// endif
	// endif
	//

	// 现在返回错误码
	code = STATUS_UNRECOGNIZED_VOLUME;
	return code;
}

//
// RecognizerIoControl
//
// 这个函数用于发送特定的IRP个下层驱动
//
// 输出参数:
// MediaHandle – 特定设备的卷句柄
// Offset – 读请求的逻辑偏移量
// Length – 读的长度
// MDL – 数据将要被拷贝的MDL链t
//
// 输出参数:
// None.
//
// Returns:
// STATUS_SUCCESS - I/O completed successfully
// Other - I/O error
//
// Notes:
// None.
//

static NTSTATUS RecognizerIoControl( IN PDEVICE_OBJECT deviceObject,
	IN ULONG IoctlCode,
	IN PVOID InputBuffer,
	IN ULONG InputBufferSize,
	OUT PVOID OutputBuffer,
	OUT ULONG OutputBufferSize)
{
	PIRP irp;
	NTSTATUS code;
	KEVENT event;
	IO_STATUS_BLOCK iosb;

	// 初始化用于等待操作成功完成的事件对象
	KeInitializeEvent(&event, SynchronizationEvent, FALSE);

	// 创建请求
	//
	irp = IoBuildDeviceIoControlRequest(IoctlCode,
		deviceObject,
		InputBuffer,
		InputBufferSize,
		OutputBuffer,
		OutputBufferSize,
		FALSE,
		&event,
		&iosb);

	// 向下层驱动发送IRP
	code = IoCallDriver(deviceObject, irp);

	//如果可能的话，我们必须等待。注意，我们不接受异步过程调用。在IO操作完成之前，我们不能返回。
	if (code == STATUS_PENDING) {
		(void) KeWaitForSingleObject(&event, Executive, KernelMode,
		TRUE, 0);
		code = iosb.Status;
	}

	//
	// 设置最终的输出缓冲区大小.
	//

	OutputBufferSize = iosb.Information;

	//
	// 完成
	//
	return(code);
}

//
//
// RecognizerReadDiskSector
//
// 从一个特定的卷读取一个扇区
//
// 输入参数:
// pDeviceObject –磁盘设备对象的指针.
// DiskSector – 需要读取的扇区数目.
//
// Outputs:
// Buffer – 用于接收读取内容的缓冲区.
//
// Returns:
// Returns TRUE if the disk sector was read.
//
// Notes:
// None.
//

static BOOLEAN RecognizerReadDiskSector(
	IN PDEVICE_OBJECT pDeviceObject,
	IN ULONG DiskSector,
	IN UCHAR* Buffer // must be DEVICE_LOGICAL_BLOCKSIZE bytes long.)
{
	LARGE_INTEGER sectorNumber;
	PIRP irp;
	IO_STATUS_BLOCK ioStatus;
	KEVENT event;
	NTSTATUS status;
	ULONG sectorSize;
	PULONG mbr;
	PAGED_CODE();
	sectorNumber.QuadPart = (LONGLONG) DiskSector *DEVICE_LOGICAL_BLOCKSIZE;

	//创建一个用于检查是否完成通知事件对象
	KeInitializeEvent(&event, NotificationEvent, FALSE);

	// 获得扇区大小
	sectorSize = DEVICE_LOGICAL_BLOCKSIZE;

	// 分配内存.
	mbr = ExAllocatePool(NonPagedPoolCacheAligned, sectorSize);
	if (!mbr) {
		return FALSE;
	}

	// 创建读MBR的IRP
	irp = IoBuildSynchronousFsdRequest(IRP_MJ_READ,
		pDeviceObject,
		mbr,
		sectorSize,
		§orNumber,
		&event,
		&ioStatus );
	if (!irp) {
		ExFreePool(mbr);
		return FALSE;
	}

	// 把IRP传送个端口驱动程序
	status = IoCallDriver(pDeviceObject, irp);

	if (status == STATUS_PENDING) {
		KeWaitForSingleObject(&event,
		Suspended,
		KernelMode,
		FALSE,
		NULL);

	status = ioStatus.Status;
}


if (!NT_SUCCESS(status)) {
ExFreePool(mbr);
return FALSE;
}


//
// 返回读取的扇区信息
//

RtlCopyMemory(Buffer,mbr,sectorSize);
ExFreePool(mbr);
return TRUE;
}