如何使用Delphi通过驱动器号获取物理磁盘号

时间:2015-01-13 03:12:34

标签: delphi hardware disk

抱歉,我对此主题一无所知,所以请求您的帮助。我在谷歌上面找到了这个代码,通过驱动器号获取物理磁盘号,尽管它有效,但需要大约4或5秒才能得到结果。我想知道是否有更快的方法以及如何做到这一点?谢谢!

function GetPhysicalDiskNumber(Drive: Char): Byte;

  function GetLD(Drive: Char): Cardinal;
  var
    Buffer : String;
  begin
    Buffer := Format('\\.\%s:',[Drive]);
    Result := CreateFile(PChar(Buffer),GENERIC_READ Or GENERIC_WRITE,FILE_SHARE_READ,nil,OPEN_EXISTING,0,0);
    If Result = INVALID_HANDLE_VALUE Then
      begin
      Result := CreateFile(PChar(Buffer),GENERIC_READ,FILE_SHARE_READ,nil,OPEN_EXISTING,0,0);
    end;
  end;

type
  PDiskInfo = ^TDiskInfo;
  TDiskInfo = record
    BootStatus,
    StartHead    : Byte;
    StartSecClu  : Array[0..1]  Of Byte;
    ParitionType,
    LastHead     : Byte;
    LastSecClu   : Array[0..1]  Of Byte;
    ABSSector,
    TTLSector    : Integer;
    Reserved     : Array[0..47] Of Byte;
    Signature    : Array[0..1]  Of Byte;
  end;
  TDiskExtent = record
    DiskNumber: Cardinal;
    StartingOffset: Int64;
    ExtentLength: Int64;
  end;
  DISK_EXTENT = TDiskExtent;
  PDiskExtent = ^TDiskExtent;
  TVolumeDiskExtents = record
    NumberOfDiskExtents: Cardinal;
    Extents: array[0..0] of TDiskExtent;
  end;
  VOLUME_DISK_EXTENTS = TVolumeDiskExtents;
  PVolumeDiskExtents = ^TVolumeDiskExtents;

const
  FILE_DEVICE_DISK                     = $00000007;
  METHOD_BUFFERED                      = 0;
  FILE_ANY_ACCESS                      = 0;
  IOCTL_DISK_BASE                      = FILE_DEVICE_DISK;
  IOCTL_VOLUME_BASE                    = DWORD('V');
  IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS = ((IOCTL_VOLUME_BASE shl 16) or (FILE_ANY_ACCESS shl 14) or (0 shl 2) or METHOD_BUFFERED);

var
  LD : DWORD;
  DiskExtents : PVolumeDiskExtents;
  DiskExtent : TDiskExtent;
  BytesReturned : Cardinal;
begin
  Result := 0;
  LD := GetLD(Drive);
  If LD = INVALID_HANDLE_VALUE Then Exit;
  Try
    DiskExtents := AllocMem(Max_Path);
    DeviceIOControl(LD,IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS,nil,0,DiskExtents,Max_Path,BytesReturned,nil);
    If DiskExtents^.NumberOfDiskExtents > 0 Then
      begin
      DiskExtent := DiskExtents^.Extents[0];
      Result := DiskExtent.DiskNumber;
    end;
  Finally
    CloseHandle(LD);
  end;
end;

1 个答案:

答案 0 :(得分:4)

Documentation for CreateFile州:

  

•打开卷或软盘时,dwShareMode参数必须包含 FILE_SHARE_WRITE 标记。

您正在使用的代码缺少标记。该代码还具有一个独特的特征,即它不会通知失败。当CreateFile失败时,您的GetPhysicalDiskNumber会返回' 0',建议结果是第一张磁盘。

这就是我认为正在发生的事情:您在卷上测试系统无法锁定写访问权限并在尝试这样做时超时(因此延迟)。但是你的功能仍然会返回' 0',所以你认为它正在发挥作用。

无论如何,你需要旗帜。另外,当CreateFile失败时,我会提出异常,这样你就可以知道发生了什么。

  function GetLD(Drive: Char): Cardinal;
  var
    Buffer : String;
  begin
    Buffer := Format('\\.\%s:',[Drive]);
    Result := CreateFile(PChar(Buffer), GENERIC_READ,
        FILE_SHARE_READ or FILE_SHARE_WRITE, nil, OPEN_EXISTING, 0, 0);
    Win32Check(Result <> INVALID_HANDLE_VALUE);
  end;

你可以选择沉默失败。在这种情况下,您最初可以将Result的{​​{1}}设置为&#39; -1&#39;例如,传递GetPhysicalDiskNumberCreateFile的异常。

代码也无法释放它分配的内存,这是一个泄漏:

DeviceIoControl