实战DeviceIoControl之四：获取硬盘的详细信息

 用IOCTL_DISK_GET_DRIVE_GEOMETRY或IOCTL_STORAGE_GET_MEDIA_TYPES_EX只能得到很少的磁盘参数，我想获得包括 硬盘 序列号在内的更加详细的信息，有什么办法呀？

A 确实，用你所说的I/O控制码，只能得到最基本的磁盘参数。获取磁盘出厂信息的I/O控制码，微软在VC/MFC环境中没有开放，在DDK中可以发现一些线索。早先，Lynn McGuire写了一个很出名的获取IDE硬盘详细信息的程序DiskID32，下面的例子是在其基础上经过增删和改进而成的。
本例中，我们要用到ATA/APAPI的IDENTIFY DEVICE指令。ATA/APAPI是国际组织T13起草和发布的IDE/EIDE/UDMA硬盘及其它可移动存储设备与主机接口的标准，至今已经到了ATA/APAPI-7版本。该接口标准规定了ATA/ATAPI设备的输入输出寄存器和指令集。欲了解更详细的ATA/ATAPI技术资料，可访问T13的站点。
用到的常量及数据结构有以下一些：

// IOCTL控制码
// #define DFP_SEND_DRIVE_COMMAND   0x0007c084
#define  DFP_SEND_DRIVE_COMMAND   CTL_CODE(IOCTL_DISK_BASE, 0x0021, METHOD_BUFFERED, FILE_READ_ACCESS | FILE_WRITE_ACCESS)
// #define DFP_RECEIVE_DRIVE_DATA   0x0007c088
#define  DFP_RECEIVE_DRIVE_DATA   CTL_CODE(IOCTL_DISK_BASE, 0x0022, METHOD_BUFFERED, FILE_READ_ACCESS | FILE_WRITE_ACCESS)

#define  FILE_DEVICE_SCSI         0x0000001b
#define  IOCTL_SCSI_MINIPORT_IDENTIFY      ((FILE_DEVICE_SCSI << 16) + 0x0501)
#define  IOCTL_SCSI_MINIPORT 0x0004D008  //  see NTDDSCSI.H for definition

// ATA/ATAPI指令
#define  IDE_ATA_IDENTIFY    0xEC   //  ATA的ID指令(IDENTIFY DEVICE)

// IDE命令寄存器
typedef struct _IDEREGS
{
   BYTE bFeaturesReg;       // 特征寄存器(用于SMART命令)
   BYTE bSectorCountReg;    // 扇区数目寄存器
   BYTE bSectorNumberReg;   // 开始扇区寄存器
   BYTE bCylLowReg;         // 开始柱面低字节寄存器
   BYTE bCylHighReg;        // 开始柱面高字节寄存器
   BYTE bDriveHeadReg;      // 驱动器/磁头寄存器
   BYTE bCommandReg;        // 指令寄存器
   BYTE bReserved;          // 保留
} IDEREGS, *PIDEREGS, *LPIDEREGS;

// 从驱动程序返回的状态
typedef struct _DRIVERSTATUS
{
   BYTE  bDriverError;      // 错误码
   BYTE  bIDEStatus;        // IDE状态寄存器
   BYTE  bReserved[2];      // 保留
   DWORD  dwReserved[2];    // 保留
} DRIVERSTATUS, *PDRIVERSTATUS, *LPDRIVERSTATUS;

// IDE设备IOCTL输入数据结构
typedef struct _SENDCMDINPARAMS
{
   DWORD     cBufferSize;   // 缓冲区字节数
   IDEREGS   irDriveRegs;   // IDE寄存器组
   BYTE bDriveNumber;       // 驱动器号
   BYTE bReserved[3];       // 保留
   DWORD     dwReserved[4]; // 保留
   BYTE      bBuffer[1];    // 输入缓冲区(此处象征性地包含1字节)
} SENDCMDINPARAMS, *PSENDCMDINPARAMS, *LPSENDCMDINPARAMS;

// IDE设备IOCTL输出数据结构
typedef struct _SENDCMDOUTPARAMS
{
   DWORD         cBufferSize;   // 缓冲区字节数
   DRIVERSTATUS  DriverStatus;  // 驱动程序返回状态
   BYTE          bBuffer[1];    // 输入缓冲区(此处象征性地包含1字节)
} SENDCMDOUTPARAMS, *PSENDCMDOUTPARAMS, *LPSENDCMDOUTPARAMS;

// IDE的ID命令返回的数据
// 共512字节(256个WORD)，这里仅定义了一些感兴趣的项(基本上依据ATA/ATAPI-4)
typedef struct _IDINFO
{
 USHORT  wGenConfig;     // WORD 0: 基本信息字
 USHORT  wNumCyls;     // WORD 1: 柱面数
 USHORT  wReserved2;     // WORD 2: 保留
 USHORT  wNumHeads;     // WORD 3: 磁头数
 USHORT  wReserved4;        // WORD 4: 保留
 USHORT  wReserved5;        // WORD 5: 保留
 USHORT  wNumSectorsPerTrack;  // WORD 6: 每磁道扇区数
 USHORT  wVendorUnique[3];   // WORD 7-9: 厂家设定值
 CHAR    sSerialNumber[20];   // WORD 10-19:序列号
 USHORT  wBufferType;    // WORD 20: 缓冲类型
 USHORT  wBufferSize;    // WORD 21: 缓冲大小
 USHORT  wECCSize;     // WORD 22: ECC校验大小
 CHAR    sFirmwareRev[8];   // WORD 23-26: 固件版本
 CHAR    sModelNumber[40];   // WORD 27-46: 内部型号
 USHORT  wMoreVendorUnique;   // WORD 47: 厂家设定值
 USHORT  wReserved48;    // WORD 48: 保留
 struct {
  USHORT  reserved1:8;
  USHORT  DMA:1;     // 1=支持DMA
  USHORT  LBA:1;     // 1=支持LBA
  USHORT  DisIORDY:1;    // 1=可不使用IORDY
  USHORT  IORDY:1;    // 1=支持IORDY
  USHORT  SoftReset:1;   // 1=需要ATA软启动
  USHORT  Overlap:1;    // 1=支持重叠操作
  USHORT  Queue:1;    // 1=支持命令队列
  USHORT  InlDMA:1;    // 1=支持交叉存取DMA
 } wCapabilities;     // WORD 49: 一般能力
 USHORT  wReserved1;     // WORD 50: 保留
 USHORT  wPIOTiming;     // WORD 51: PIO时序
 USHORT  wDMATiming;     // WORD 52: DMA时序
 struct {
  USHORT  CHSNumber:1;   // 1=WORD 54-58有效
  USHORT  CycleNumber:1;   // 1=WORD 64-70有效
  USHORT  UnltraDMA:1;   // 1=WORD 88有效
  USHORT  reserved:13;
 } wFieldValidity;     // WORD 53: 后续字段有效性标志
 USHORT  wNumCurCyls;    // WORD 54: CHS可寻址的柱面数
 USHORT  wNumCurHeads;    // WORD 55: CHS可寻址的磁头数
 USHORT  wNumCurSectorsPerTrack;  // WORD 56: CHS可寻址每磁道扇区数
 USHORT  wCurSectorsLow;    // WORD 57: CHS可寻址的扇区数低位字
 USHORT  wCurSectorsHigh;   // WORD 58: CHS可寻址的扇区数高位字
 struct {
  USHORT  CurNumber:8;   // 当前一次性可读写扇区数
  USHORT  Multi:1;    // 1=已选择多扇区读写
  USHORT  reserved1:7;
 } wMultSectorStuff;     // WORD 59: 多扇区读写设定
 ULONG  dwTotalSectors;    // WORD 60-61: LBA可寻址的扇区数
 USHORT  wSingleWordDMA;    // WORD 62: 单字节DMA支持能力
 struct {
  USHORT  Mode0:1;    // 1=支持模式0 (4.17Mb/s)
  USHORT  Mode1:1;    // 1=支持模式1 (13.3Mb/s)
  USHORT  Mode2:1;    // 1=支持模式2 (16.7Mb/s)
  USHORT  Reserved1:5;
  USHORT  Mode0Sel:1;    // 1=已选择模式0
  USHORT  Mode1Sel:1;    // 1=已选择模式1
  USHORT  Mode2Sel:1;    // 1=已选择模式2
  USHORT  Reserved2:5;
 } wMultiWordDMA;     // WORD 63: 多字节DMA支持能力
 struct {
  USHORT  AdvPOIModes:8;   // 支持高级POI模式数
  USHORT  reserved:8;
 } wPIOCapacity;      // WORD 64: 高级PIO支持能力
 USHORT  wMinMultiWordDMACycle;  // WORD 65: 多字节DMA传输周期的最小值
 USHORT  wRecMultiWordDMACycle;  // WORD 66: 多字节DMA传输周期的建议值
 USHORT  wMinPIONoFlowCycle;   // WORD 67: 无流控制时PIO传输周期的最小值
 USHORT  wMinPOIFlowCycle;   // WORD 68: 有流控制时PIO传输周期的最小值
 USHORT  wReserved69[11];   // WORD 69-79: 保留
 struct {
  USHORT  Reserved1:1;
  USHORT  ATA1:1;     // 1=支持ATA-1
  USHORT  ATA2:1;     // 1=支持ATA-2
  USHORT  ATA3:1;     // 1=支持ATA-3
  USHORT  ATA4:1;     // 1=支持ATA/ATAPI-4
  USHORT  ATA5:1;     // 1=支持ATA/ATAPI-5
  USHORT  ATA6:1;     // 1=支持ATA/ATAPI-6
  USHORT  ATA7:1;     // 1=支持ATA/ATAPI-7
  USHORT  ATA8:1;     // 1=支持ATA/ATAPI-8
  USHORT  ATA9:1;     // 1=支持ATA/ATAPI-9
  USHORT  ATA10:1;    // 1=支持ATA/ATAPI-10
  USHORT  ATA11:1;    // 1=支持ATA/ATAPI-11
  USHORT  ATA12:1;    // 1=支持ATA/ATAPI-12
  USHORT  ATA13:1;    // 1=支持ATA/ATAPI-13
  USHORT  ATA14:1;    // 1=支持ATA/ATAPI-14
  USHORT  Reserved2:1;
 } wMajorVersion;     // WORD 80: 主版本
 USHORT  wMinorVersion;    // WORD 81: 副版本
 USHORT  wReserved82[6];    // WORD 82-87: 保留
 struct {
  USHORT  Mode0:1;    // 1=支持模式0 (16.7Mb/s)
  USHORT  Mode1:1;    // 1=支持模式1 (25Mb/s)
  USHORT  Mode2:1;    // 1=支持模式2 (33Mb/s)
  USHORT  Mode3:1;    // 1=支持模式3 (44Mb/s)
  USHORT  Mode4:1;    // 1=支持模式4 (66Mb/s)
  USHORT  Mode5:1;    // 1=支持模式5 (100Mb/s)
  USHORT  Mode6:1;    // 1=支持模式6 (133Mb/s)
  USHORT  Mode7:1;    // 1=支持模式7 (166Mb/s) ???
  USHORT  Mode0Sel:1;    // 1=已选择模式0
  USHORT  Mode1Sel:1;    // 1=已选择模式1
  USHORT  Mode2Sel:1;    // 1=已选择模式2
  USHORT  Mode3Sel:1;    // 1=已选择模式3
  USHORT  Mode4Sel:1;    // 1=已选择模式4
  USHORT  Mode5Sel:1;    // 1=已选择模式5
  USHORT  Mode6Sel:1;    // 1=已选择模式6
  USHORT  Mode7Sel:1;    // 1=已选择模式7
 } wUltraDMA;      // WORD 88:  Ultra DMA支持能力
 USHORT    wReserved89[167];   // WORD 89-255
} IDINFO, *PIDINFO;

// SCSI驱动所需的输入输出共用的结构
typedef struct _SRB_IO_CONTROL
{
   ULONG HeaderLength;  // 头长度
   UCHAR Signature[8];  // 特征名称
   ULONG Timeout;   // 超时时间
   ULONG ControlCode;  // 控制码
   ULONG ReturnCode;  // 返回码
   ULONG Length;   // 缓冲区长度
} SRB_IO_CONTROL, *PSRB_IO_CONTROL;

需要引起注意的是IDINFO第57-58 WORD (CHS可寻址的扇区数)，因为不满足32位对齐的要求，不可定义为一个ULONG字段。Lynn McGuire的程序里正是由于定义为一个ULONG字段，导致该结构不可用。

以下是核心代码：

// 打开设备
// filename: 设备的“文件名”
HANDLE OpenDevice(LPCTSTR filename)
{
 HANDLE hDevice;

 // 打开设备
 hDevice= ::CreateFile(filename,   // 文件名
  GENERIC_READ | GENERIC_WRITE,  // 读写方式
  FILE_SHARE_READ | FILE_SHARE_WRITE, // 共享方式
  NULL,     // 默认的安全描述符
  OPEN_EXISTING,    // 创建方式
  0,     // 不需设置文件属性
  NULL);     // 不需参照模板文件

 return hDevice;
}

// 向驱动发“IDENTIFY DEVICE”命令，获得设备信息
// hDevice: 设备句柄
// pIdInfo:  设备信息结构指针
BOOL IdentifyDevice(HANDLE hDevice, PIDINFO pIdInfo)
{
 PSENDCMDINPARAMS pSCIP;  // 输入数据结构指针
 PSENDCMDOUTPARAMS pSCOP; // 输出数据结构指针
 DWORD dwOutBytes;   // IOCTL输出数据长度
 BOOL bResult;    // IOCTL返回值

 // 申请输入/输出数据结构空间
    pSCIP = (PSENDCMDINPARAMS)::GlobalAlloc(LMEM_ZEROINIT, sizeof(SENDCMDINPARAMS)-1);
    pSCOP = (PSENDCMDOUTPARAMS)::GlobalAlloc(LMEM_ZEROINIT, sizeof(SENDCMDOUTPARAMS)+sizeof(IDINFO)-1);

 // 指定ATA/ATAPI命令的寄存器值
// pSCIP->irDriveRegs.bFeaturesReg = 0;
// pSCIP->irDriveRegs.bSectorCountReg = 0;
// pSCIP->irDriveRegs.bSectorNumberReg = 0;
// pSCIP->irDriveRegs.bCylLowReg = 0;
// pSCIP->irDriveRegs.bCylHighReg = 0;
// pSCIP->irDriveRegs.bDriveHeadReg = 0;
 pSCIP->irDriveRegs.bCommandReg = IDE_ATA_IDENTIFY;

 // 指定输入/输出数据缓冲区大小
 pSCIP->cBufferSize = 0;
 pSCOP->cBufferSize = sizeof(IDINFO);

 // IDENTIFY DEVICE
 bResult = ::DeviceIoControl(hDevice,  // 设备句柄
  DFP_RECEIVE_DRIVE_DATA,   // 指定IOCTL
  pSCIP, sizeof(SENDCMDINPARAMS) - 1, // 输入数据缓冲区
  pSCOP, sizeof(SENDCMDOUTPARAMS) + sizeof(IDINFO) - 1, // 输出数据缓冲区
  &dwOutBytes,    // 输出数据长度
  (LPOVERLAPPED)NULL);   // 用同步I/O

 // 复制设备参数结构
 ::memcpy(pIdInfo, pSCOP->bBuffer, sizeof(IDINFO));

 // 释放输入/输出数据空间
 ::GlobalFree(pSCOP);
 ::GlobalFree(pSCIP);

 return bResult;
}

// 向SCSI MINI-PORT驱动发“IDENTIFY DEVICE”命令，获得设备信息
// hDevice: 设备句柄
// pIdInfo:  设备信息结构指针
BOOL IdentifyDeviceAsScsi(HANDLE hDevice, int nDrive, PIDINFO pIdInfo)
{
 PSENDCMDINPARAMS pSCIP;  // 输入数据结构指针
 PSENDCMDOUTPARAMS pSCOP; // 输出数据结构指针
 PSRB_IO_CONTROL pSRBIO;  // SCSI输入输出数据结构指针
 DWORD dwOutBytes;   // IOCTL输出数据长度
 BOOL bResult;    // IOCTL返回值

 // 申请输入/输出数据结构空间
    pSRBIO = (PSRB_IO_CONTROL)::GlobalAlloc(LMEM_ZEROINIT, sizeof(SRB_IO_CONTROL)+sizeof(SENDCMDOUTPARAMS)+sizeof(IDINFO)-1);
    pSCIP = (PSENDCMDINPARAMS)((char *)pSRBIO+sizeof(SRB_IO_CONTROL));
    pSCOP = (PSENDCMDOUTPARAMS)((char *)pSRBIO+sizeof(SRB_IO_CONTROL));

 // 填充输入/输出数据
 pSRBIO->HeaderLength = sizeof(SRB_IO_CONTROL);
 pSRBIO->Timeout = 10000;
 pSRBIO->Length = sizeof(SENDCMDOUTPARAMS)+sizeof(IDINFO)-1;
 pSRBIO->ControlCode = IOCTL_SCSI_MINIPORT_IDENTIFY;
 ::strncpy ((char *)pSRBIO->Signature, "SCSIDISK", 8);

 // 指定ATA/ATAPI命令的寄存器值
// pSCIP->irDriveRegs.bFeaturesReg = 0;
// pSCIP->irDriveRegs.bSectorCountReg = 0;
// pSCIP->irDriveRegs.bSectorNumberReg = 0;
// pSCIP->irDriveRegs.bCylLowReg = 0;
// pSCIP->irDriveRegs.bCylHighReg = 0;
// pSCIP->irDriveRegs.bDriveHeadReg = 0;
 pSCIP->irDriveRegs.bCommandReg = IDE_ATA_IDENTIFY;
 pSCIP->bDriveNumber = nDrive;

 // IDENTIFY DEVICE
 bResult = ::DeviceIoControl(hDevice,  // 设备句柄
  IOCTL_SCSI_MINIPORT,   // 指定IOCTL
  pSRBIO, sizeof(SRB_IO_CONTROL) +sizeof(SENDCMDINPARAMS) - 1, // 输入数据缓冲区
  pSRBIO, sizeof(SRB_IO_CONTROL) +sizeof(SENDCMDOUTPARAMS) + sizeof(IDINFO) - 1, // 输出数据缓冲区
  &dwOutBytes,  // 输出数据长度
  (LPOVERLAPPED)NULL); // 用同步I/O
 
 // 复制设备参数结构
 ::memcpy(pIdInfo, pSCOP->bBuffer, sizeof(IDINFO));

 // 释放输入/输出数据空间
 ::GlobalFree(pSRBIO);

 return bResult;
}

// 将串中的字符两两颠倒
// 原因是ATA/ATAPI中的WORD，与Windows采用的字节顺序相反
// 驱动程序中已经将收到的数据全部反过来，我们来个负负得正
void AdjustString(char* str, int len)
{
 char ch;
 int i;

 // 两两颠倒
 for(i=0;i<len;i+=2)
 {
  ch = str[i];
  str[i] = str[i+1];
  str[i+1] = ch;
 }

 // 若是右对齐的，调整为左对齐 (去掉左边的空格)
 i=0;
 while(i<len && str[i]==' ') i++;

 ::memmove(str, &str[i], len-i);

 // 去掉右边的空格
 i = len - 1;
 while(i>=0 && str[i]==' ')
 {
  str[i] = '/0';
  i--;
 }
}

// 读取IDE硬盘的设备信息，必须有足够权限
// nDrive: 驱动器号(0=第一个硬盘，1=0=第二个硬盘，......)
// pIdInfo: 设备信息结构指针
BOOL GetPhysicalDriveInfoInNT(int nDrive, PIDINFO pIdInfo)
{
 HANDLE hDevice;   // 设备句柄
 BOOL bResult;   // 返回结果
 char szFileName[20]; // 文件名

 ::sprintf(szFileName,".//PhysicalDrive%d", nDrive);

 hDevice = ::OpenDevice(szFileName);

 if(hDevice == INVALID_HANDLE_VALUE)
 {
  return FALSE;
 }

 // IDENTIFY DEVICE
 bResult = ::IdentifyDevice(hDevice, pIdInfo);

 if(bResult)
 {
  // 调整字符串
  ::AdjustString(pIdInfo->sSerialNumber, 20);
  ::AdjustString(pIdInfo->sModelNumber, 40);
  ::AdjustString(pIdInfo->sFirmwareRev, 8);
 }

 ::CloseHandle (hDevice);

 return bResult;
}

// 用SCSI驱动读取IDE硬盘的设备信息，不受权限制约
// nDrive: 驱动器号(0=Primary Master, 1=Promary Slave, 2=Secondary master, 3=Secondary slave)
// pIdInfo: 设备信息结构指针
BOOL GetIdeDriveAsScsiInfoInNT(int nDrive, PIDINFO pIdInfo)
{
 HANDLE hDevice;   // 设备句柄
 BOOL bResult;   // 返回结果
 char szFileName[20]; // 文件名

 ::sprintf(szFileName,".//Scsi%d:", nDrive/2);

 hDevice = ::OpenDevice(szFileName);
     
 if(hDevice == INVALID_HANDLE_VALUE)
 {
  return FALSE;
 }

 // IDENTIFY DEVICE
 bResult = ::IdentifyDeviceAsScsi(hDevice, nDrive%2, pIdInfo);

 // 检查是不是空串
 if(pIdInfo->sModelNumber[0]=='/0')
 {
  bResult = FALSE;
 }

 if(bResult)
 {
  // 调整字符串
  ::AdjustString(pIdInfo->sSerialNumber, 20);
  ::AdjustString(pIdInfo->sModelNumber, 40);
  ::AdjustString(pIdInfo->sFirmwareRev, 8);
 }

 return bResult;
}

Q 我注意到ATA/ATAPI里，以及DiskID32里，有一个“IDENTIFY PACKET DEVICE”指令，与“IDENTIFY DEVICE”有什么区别？

A IDENTIFY DEVICE专门用于固定硬盘，而IDENTIFY PACKET DEVICE用于可移动存储设备如CDROM、CF、MO、ZIP、TAPE等。因为驱动程序的原因，实际上用本例的方法，不管是IDENTIFY DEVICE也好，IDENTIFY PACKET DEVICE也好，获取可移动存储设备的详细信息，一般是做不到的。而且除了IDE硬盘，对SCSI、USB等接口的硬盘也不起作用。除非厂商提供的驱动支持这样的功能。

Q ATA/ATAPI有很多指令，如READ SECTORS, WRITE SECTORS, SECURITY, SLEEP, STANDBY等，利用上述方法，是否可进行相应操作？

A 应该没问题。但切记，要慎重慎重再慎重！

Q 关于权限问题，请解释一下好吗？

A 在NT/2000/XP下，administrator可以管理设备，上述两种访问驱动的方法都行。但在user身份下，或者登录到域后，用户无法访问PhysicalDrive驱动的核心层，但SCSI MINI-PORT驱动却可以。目前是可以，不知道Windows以后的版本是否支持。因为这肯定是一个安全隐患。
另外，我们着重讨论NT/2000/XP中DeviceIoControl的应用，如果需要在98/ME中得到包括硬盘序列号在内的更加详细的信息，请参考DiskID32。