程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> 更多編程語言 >> 匯編語言 >> 實戰DeviceIoControl系列之二:獲取軟盤/硬盤/光盤的參數

實戰DeviceIoControl系列之二:獲取軟盤/硬盤/光盤的參數

編輯:匯編語言

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); 

最後,最後干什麼呢?編譯,運行......

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved