iniwf

风是温柔的,雨是伤心的,云是快乐的,月是多情的,爱是迷失的,恋是醉人的,情是难忘的,天是长久的,地是永恒的

实战DeviceIoControl 之二:获取软盘/硬盘/光盘的参数

转自http://www.cnblogs.com/henryzc/archive/2005/11/08/271909.html

Q 在MSDN的那个demo中,将设备名换成“A:”取A盘参数,先用资源管理器读一下盘,再运行这个程序可以成功,但换一张盘后就失败;换成“CDROM0”取CDROM参数,无论如何都不行。这个问题如何解决呢?

A 取软盘参数是从软盘上读取格式化后的信息,也就是必须执行读操作,这一点与硬盘不同。将CreateFile中的访问方式改为GENERIC_READ就行了。

IOCTL_DISK_GET_DRIVE_GEOMETRY这个I/O控制码,对软盘和硬盘有效,但对一些可移动媒介如CD/DVD-ROM、TAPE等就不管用了。要取CDROM参数,还得另辟蹊径。IOCTL_STORAGE_GET_MEDIA_TYPES_EX能够帮我们解决问题。

Q 使用这些I/O控制码,需要什么样的输入输出数据格式呢?

A DeviceIoControl使用这两个控制码时,都不需要输入数据。

IOCTL_DISK_GET_DRIVE_GEOMETRY直接输出一个DISK_GEOMETRY结构:

typedef struct _DISK_GEOMETRY {
LARGE_INTEGER Cylinders;   // 柱面数
MEDIA_TYPE MediaType;      // 介质类型
DWORD TracksPerCylinder;   // 每柱面的磁道数
DWORD SectorsPerTrack;     // 每磁道的扇区数
DWORD BytesPerSector;      // 每扇区的字节数
} DISK_GEOMETRY;

IOCTL_STORAGE_GET_MEDIA_TYPES_EX输出一个GET_MEDIA_TYPES结构:

typedef struct _GET_MEDIA_TYPES {
DWORD DeviceType;               // 设备类型
DWORD MediaInfoCount;           // 介质信息条数
DEVICE_MEDIA_INFO MediaInfo[1]; // 介质信息
} GET_MEDIA_TYPES;

让我们来看一下DEVICE_MEDIA_INFO结构的定义:

typedef struct _DEVICE_MEDIA_INFO {
union {
struct {
LARGE_INTEGER Cylinders;       // 柱面数
STORAGE_MEDIA_TYPE MediaType;  // 介质类型
DWORD TracksPerCylinder;       // 每柱面的磁道数
DWORD SectorsPerTrack;         // 每磁道的扇区数
DWORD BytesPerSector;          // 每扇区的字节数
DWORD NumberMediaSides;        // 介质面数
DWORD MediaCharacteristics;    // 介质特性
} DiskInfo;            // 硬盘信息
struct {
LARGE_INTEGER Cylinders;       // 柱面数
STORAGE_MEDIA_TYPE MediaType;  // 介质类型
DWORD TracksPerCylinder;       // 每柱面的磁道数
DWORD SectorsPerTrack;         // 每磁道的扇区数
DWORD BytesPerSector;          // 每扇区的字节数
DWORD NumberMediaSides;        // 介质面数
DWORD MediaCharacteristics;    // 介质特性
} RemovableDiskInfo;   // “可移动盘”信息
struct {
STORAGE_MEDIA_TYPE MediaType;  // 介质类型
DWORD   MediaCharacteristics;  // 介质特性
DWORD   CurrentBlockSize;      // 块的大小
} TapeInfo;           // 磁带信息
} DeviceSpecific;
} DEVICE_MEDIA_INFO;

其中CD-ROM属于“可移动盘”的范围。请注意,GET_MEDIA_TYPES结构本身只定义了一条DEVICE_MEDIA_INFO,额外的DEVICE_MEDIA_INFO需要紧接此结构的另外的空间。

Q 调用方法我了解了,请用VC举个例子来实现我所期待已久的功能吧?

A 好,现在就演示一下如何取软盘/硬盘/光盘的参数。测试时,记得要有软盘/光盘插在驱动器里喔!

首先,用MFC AppWizard生成一个单文档的应用程序,取名为DiskGeometry,让它的View基于CEditView。

然后,添加以下的.h和.cpp文件。

//////////////////////////////////////////////////////////////////////////////
// GetDiskGeometry.h
//////////////////////////////////////////////////////////////////////////////
#if !defined(GET_DISK_GEOMETRY_H__)
#define GET_DISK_GEOMETRY_H__
#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
#include <winioctl.h>
BOOL GetDriveGeometry(const char* filename, DISK_GEOMETRY *pdg);
#endif // !defined(GET_DISK_GEOMETRY_H__)
//////////////////////////////////////////////////////////////////////////////
// GetDiskGeometry.cpp
//////////////////////////////////////////////////////////////////////////////
#include "stdafx.h"
#include "GetDiskGeometry.h"
// IOCTL_STORAGE_GET_MEDIA_TYPES_EX可能返回不止一条DEVICE_MEDIA_INFO,故定义足够的空间
#define MEDIA_INFO_SIZE    sizeof(GET_MEDIA_TYPES)+15*sizeof(DEVICE_MEDIA_INFO)
// filename -- 用于设备的文件名
// pdg -- 参数缓冲区指针
BOOL GetDriveGeometry(const char* filename, DISK_GEOMETRY *pdg)
{
HANDLE hDevice;         // 设备句柄
BOOL bResult;           // DeviceIoControl的返回结果
GET_MEDIA_TYPES *pmt;   // 内部用的输出缓冲区
DWORD dwOutBytes;       // 输出数据长度
// 打开设备
hDevice = ::CreateFile(filename,           // 文件名
GENERIC_READ,                          // 软驱需要读盘
FILE_SHARE_READ | FILE_SHARE_WRITE,    // 共享方式
NULL,                                  // 默认的安全描述符
OPEN_EXISTING,                         // 创建方式
0,                                     // 不需设置文件属性
NULL);                                 // 不需参照模板文件
if (hDevice == INVALID_HANDLE_VALUE)
{
// 设备无法打开...
return FALSE;
}
// 用IOCTL_DISK_GET_DRIVE_GEOMETRY取磁盘参数
bResult = ::DeviceIoControl(hDevice,       // 设备句柄
IOCTL_DISK_GET_DRIVE_GEOMETRY,         // 取磁盘参数
NULL, 0,                               // 不需要输入数据
pdg, sizeof(DISK_GEOMETRY),            // 输出数据缓冲区
&dwOutBytes,                           // 输出数据长度
(LPOVERLAPPED)NULL);                   // 用同步I/O
// 如果失败,再用IOCTL_STORAGE_GET_MEDIA_TYPES_EX取介质类型参数
if (!bResult)
{
pmt = (GET_MEDIA_TYPES *)new BYTE[MEDIA_INFO_SIZE];
bResult = ::DeviceIoControl(hDevice,    // 设备句柄
IOCTL_STORAGE_GET_MEDIA_TYPES_EX,   // 取介质类型参数
NULL, 0,                            // 不需要输入数据
pmt, MEDIA_INFO_SIZE,               // 输出数据缓冲区
&dwOutBytes,                        // 输出数据长度
(LPOVERLAPPED)NULL);                // 用同步I/O
if (bResult)
{
// 注意到结构DEVICE_MEDIA_INFO是在结构DISK_GEOMETRY的基础上扩充的
// 为简化程序,用memcpy代替如下多条赋值语句:
// pdg->MediaType = (MEDIA_TYPE)pmt->MediaInfo[0].DeviceSpecific.DiskInfo.MediaType;
// pdg->Cylinders = pmt->MediaInfo[0].DeviceSpecific.DiskInfo.Cylinders;
// pdg->TracksPerCylinder = pmt->MediaInfo[0].DeviceSpecific.DiskInfo.TracksPerCylinder;
// ... ...
::memcpy(pdg, pmt->MediaInfo, sizeof(DISK_GEOMETRY));
}
delete pmt;
}
// 关闭设备句柄
::CloseHandle(hDevice);
return (bResult);
}

然后,在Toolbar的IDR_MAINFRAME上添加一个按钮,ID为ID_GET_DISK_GEOMETRY。打开ClassWizard,在DiskGeometryView中

添加ID_GET_DISK_GEOMETRY的映射函数OnGetDiskGeometry。打开DiskGeometryView.cpp,包含头文件GetDiskGeometry.h。

在OnGetDiskGeometry中,添加以下代码

    const char *szDevName[]=
{
"\\\\.\\A:",
"\\\\.\\B:",
"\\\\.\\PhysicalDrive0",
"\\\\.\\PhysicalDrive1",
"\\\\.\\PhysicalDrive2",
"\\\\.\\PhysicalDrive3",
"\\\\.\\Cdrom0",
"\\\\.\\Cdrom1",
};
DISK_GEOMETRY dg;
ULONGLONG DiskSize;
BOOL bResult;
CString strMsg;
CString strTmp;
for (int i = 0; i < sizeof(szDevName)/sizeof(char*); i++)
{
bResult = GetDriveGeometry(szDevName[i], &dg);
strTmp.Format("\r\n%s  result = %s\r\n", szDevName[i], bResult ? "success" : "failure");
strMsg+=strTmp;
if (!bResult) continue;
strTmp.Format("    Media Type = %d\r\n", dg.MediaType);
strMsg+=strTmp;
strTmp.Format("    Cylinders = %I64d\r\n", dg.Cylinders);
strMsg+=strTmp;
strTmp.Format("    Tracks per cylinder = %ld\r\n", (ULONG) dg.TracksPerCylinder);
strMsg+=strTmp;
strTmp.Format("    Sectors per track = %ld\r\n", (ULONG) dg.SectorsPerTrack);
strMsg+=strTmp;
strTmp.Format("    Bytes per sector = %ld\r\n", (ULONG) dg.BytesPerSector);
strMsg+=strTmp;
DiskSize = dg.Cylinders.QuadPart * (ULONG)dg.TracksPerCylinder *
(ULONG)dg.SectorsPerTrack * (ULONG)dg.BytesPerSector;
strTmp.Format("    Disk size = %I64d (Bytes) = %I64d (Mb)\r\n", DiskSize, DiskSize / (1024 * 1024));
strMsg+=strTmp;
}
CEdit& Edit = GetEditCtrl();
Edit.SetWindowText(strMsg);

最后,最后干什么呢?编译,运行......

[相关资源]

  • 本文Demo源码:DiskGeometry.zip (21KB)
  • posted on 2009-04-13 21:03 iniwf 阅读(500) 评论(0)  编辑 收藏 引用 所属分类: 驱动API应用


    只有注册用户登录后才能发表评论。
    网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理


    导航

    统计

    常用链接

    留言簿(2)

    随笔分类

    随笔档案

    收藏夹

    IT技术

    积分与排名

    最新评论

    阅读排行榜

    评论排行榜