win32 文件过滤驱动中如何获得硬盘的大小

2015-05-09 03:48:07 423 阅读 AI 写同款 · GEO 优化AI 选题数据监测一键分发 ›

[学习笔记]Esp32 + 4G模块，随身wifi，ppp拨号上网 Esp32S3 + 4G模块制作随身WIFI教程阅读详情

Hi，各位大侠，
我最近遇到一个比较头痛的问题，我在一个文件驱动中在设备加载的时候想获得硬盘的大小
我往下传IOCTL_DISK_GET_DRIVE_GEOMETRY_EX控制命令，成功了，但是返回值都是0，大家有什么好的解决方法不？

NTSTATUS GetDiskGeometry(IN PDEVICE_OBJECT pDiskDevObj, 

						 OUT LARGE_INTEGER * pDiskSize)

{

    IO_STATUS_BLOCK IoStatusBlock;

    KEVENT Event;

    PIRP pIrp;

    NTSTATUS status;

	DISK_GEOMETRY_EX       diskGeo;

	PDISK_GEOMETRY_EX      diskGeometry;

	ULONG size ;





	/* allocate memory

   */

    size = sizeof (DISK_GEOMETRY_EX);

    diskGeometry =&diskGeo; 



    RtlZeroMemory(diskGeometry, sizeof(DISK_GEOMETRY_EX));

    KeInitializeEvent(&Event, NotificationEvent, FALSE);



	while(true)

	{

		/* Request for the device geometry

		   */

	  pIrp = IoBuildDeviceIoControlRequest (IOCTL_DISK_GET_DRIVE_GEOMETRY_EX,

										   pDiskDevObj,

										   NULL,

										   0,

										   (PVOID)diskGeometry,

										   size,

										   FALSE,

										   &Event,

										   &IoStatusBlock);

	

		if (!pIrp) 

		{

			//return STATUS_INSUFFICIENT_RESOURCES;

			KeResetEvent(&Event);

			break;

		}



		status = IoCallDriver(pDiskDevObj, pIrp);

		if (status == STATUS_PENDING) 

		{

			KeWaitForSingleObject(&Event, Executive, KernelMode, FALSE, NULL);

			status = IoStatusBlock.Status;

		}



		if (NT_SUCCESS(status))

		{

			pDiskSize->QuadPart = diskGeometry->DiskSize.QuadPart;

			if(pDiskSize->QuadPart!=0)

			{

				return status;

			}

		}



		break;

	}



    pIrp = IoBuildDeviceIoControlRequest(

        IOCTL_DISK_GET_DRIVE_GEOMETRY,

        pDiskDevObj,

        NULL,

        0, 

        &diskGeometry->Geometry,

        sizeof(DISK_GEOMETRY),

        FALSE, 

        &Event,

        &IoStatusBlock);



    if (!pIrp) 

    {

        return STATUS_INSUFFICIENT_RESOURCES;

    }

    status = IoCallDriver(pDiskDevObj, pIrp);

    if (status == STATUS_PENDING) 

    {

        KeWaitForSingleObject(&Event, Executive, KernelMode, FALSE, NULL);

        status = IoStatusBlock.Status;

    }



	if (NT_SUCCESS(status))

	{

		pDiskSize->QuadPart = diskGeometry->Geometry.Cylinders.QuadPart*diskGeometry->Geometry.SectorsPerTrack*diskGeometry->Geometry.TracksPerCylinder;

	}

    return status;

}