HanDs
管理员

[Delphi文章] 获取优盘序列号的实现方法 





学习中请遵循国家相关法律法规,黑客不作恶。没有网络安全就没有国家安全

本站需要登陆后才能查看

Uses Registry;

Const
  DIGCF_PRESENT = $00000002;
  DIGCF_DEVICEINTERFACE = $00000010;
  ANYSIZE_ARRAY = 1;
  SetupAPI = 'SetupAPI.DLL';

Type
  HDEVINFO = Pointer;
  ULONG_PTR = DWORD;

  PSPDevInfoData = ^TSPDevInfoData;
  SP_DEVINFO_DATA = Packed Record
    cbSize: DWORD;
    ClassGuid: TGUID;
    DevInst: DWORD;
    Reserved: ULONG_PTR;
  End;
  {$EXTERNALSYM SP_DEVINFO_DATA}
  TSPDevInfoData = SP_DEVINFO_DATA;

  PSPDeviceInterfaceData = ^TSPDeviceInterfaceData;
  SP_DEVICE_INTERFACE_DATA = Packed Record
    cbSize: DWORD;
    InterfaceClassGuid: TGUID;
    Flags: DWORD;
    Reserved: ULONG_PTR;
  End;
  {$EXTERNALSYM SP_DEVICE_INTERFACE_DATA}
  TSPDeviceInterfaceData = SP_DEVICE_INTERFACE_DATA;

  PSPDeviceInterfaceDetailDataA = ^TSPDeviceInterfaceDetailDataA;
  SP_DEVICE_INTERFACE_DETAIL_DATA_A = Packed Record
    cbSize: DWORD;
    DevicePath: Array[0..ANYSIZE_ARRAY - 1] Of AnsiChar;
  End;
  {$EXTERNALSYM SP_DEVICE_INTERFACE_DETAIL_DATA_A}
  TSPDeviceInterfaceDetailDataA = SP_DEVICE_INTERFACE_DETAIL_DATA_A;

Function SetupDiGetClassDevsA(ClassGuid: PGUID; Const Enumerator: PAnsiChar;
  hwndParent: HWND; Flags: DWORD): HDEVINFO; Stdcall; External SetupAPI;

Function SetupDiEnumDeviceInterfaces(DeviceInfoSet: HDEVINFO;
  DeviceInfoData: PSPDevInfoData; Const InterfaceClassGuid: TGUID;
  MemberIndex: DWORD; Var DeviceInterfaceData: TSPDeviceInterfaceData): BOOL; Stdcall; External SetupAPI;
{$EXTERNALSYM SetupDiEnumDeviceInterfaces}

Function SetupDiGetDeviceInterfaceDetailA(DeviceInfoSet: HDEVINFO;
  DeviceInterfaceData: PSPDeviceInterfaceData;
  DeviceInterfaceDetailData: PSPDeviceInterfaceDetailDataA;
  DeviceInterfaceDetailDataSize: DWORD; Var RequiredSize: DWORD;
  Device: PSPDevInfoData): BOOL; Stdcall; External SetupAPI;

Function SetupDiDestroyDeviceInfoList(DeviceInfoSet: HDEVINFO): BOOL; Stdcall; External SetupAPI;

Function GetUSBDiskID(DiskID: String; Var PID: String): Boolean;
Var
  USBGuid: TGUID;
  USBHandle: HDEVINFO;
  Success: LongBool;
  Devn: Integer;
  DevData: TSPDevInfoData;
  DeviceInterfaceData: TSPDeviceInterfaceData;
  FunctionClassDeviceData: PSPDeviceInterfaceDetailDataA;
  BytesReturned: DWORD;
  Reg: TRegistry;
  RegData: Array Of Char;
  i, RegSize: Integer;
  Str, USBPath: String;
Begin
  Result := false;
  Pid := '';
  Reg := TRegistry.Create;
  Reg.RootKey := HKEY_LOCAL_MACHINE;
  Reg.OpenKey('SYSTEM\MountedDevices', false);
  RegSize := Reg.GetDataSize(Format('\DosDevices\%s', [DiskID]));
  SetLength(RegData, RegSize + 1);
  Reg.ReadBinaryData(Format('\DosDevices\%s', [DiskID]), RegData[0], RegSize + 1);
  For i := 0 To RegSize - 1 Do
    If RegData[i] <> #0 Then Str := Str + RegData[i];
  Str := Copy(Str, Pos('#RemovableMedia#', Str) + 16, Length(Str));
  Str := Copy(Str, 1, Pos('RM', Str) - 2);
  Str := UpperCase(Str);
  Reg.CloseKey;

  USBGuid := StringToGUID('{53f56307-b6bf-11d0-94f2-00a0c91efb8b}');
  USBHandle := SetupDiGetClassDevsA(@USBGuid, Nil, 0, DIGCF_PRESENT Or DIGCF_DEVICEINTERFACE);
  If USBHandle = Pointer(INVALID_HANDLE_VALUE) Then Exit;
  Devn := 0;
  Repeat
    DeviceInterfaceData.cbSize := SizeOf(TSPDeviceInterfaceData);
    Success := SetupDiEnumDeviceInterfaces(USBHandle, Nil, USBGuid, Devn, DeviceInterfaceData);
    If Success Then
    Begin
      DevData.cbSize := SizeOf(DevData);
      BytesReturned := 0;
      SetupDiGetDeviceInterfaceDetailA(USBHandle, @DeviceInterfaceData, Nil, 0, BytesReturned, @DevData);
      If (BytesReturned <> 0) And (GetLastError = ERROR_INSUFFICIENT_BUFFER) Then
      Begin
        FunctionClassDeviceData := AllocMem(BytesReturned);
        FunctionClassDeviceData^.cbSize := SizeOf(TSPDeviceInterfaceDetailDataA);
        If SetupDiGetDeviceInterfaceDetailA(USBHandle, @DeviceInterfaceData,
          FunctionClassDeviceData, BytesReturned, BytesReturned, @DevData) Then
        Begin
          USBPath := StrPas(PChar(@FunctionClassDeviceData.DevicePath));
          If Reg.OpenKeyReadOnly(Format('SYSTEM\CurrentControlSet\Enum%s', [StringReplace(Copy(USBPath, 4, Pos('{', USBPath) - 5), '#', '\', [rfReplaceAll])])) Then
          If UpperCase(Reg.ReadString('ParentIdPrefix')) = Str Then
          Begin
          PID := StringReplace(Str, '&', '', [rfReplaceAll]);
          Result := True;
          Break;
          End;
          Reg.CloseKey;
          Inc(Devn);
        End;
        FreeMem(FunctionClassDeviceData);
      End;
    End;
  Until Not Success;
  SetupDiDestroyDeviceInfoList(USBHandle);
  Reg.Free;
End;

//使用方法
Procedure TForm1.Button1Click(Sender: TObject);
Var
  Drv, Pid: String;
Begin
  Drv := ExtractFileDrive(ParamStr(0));
  If GetDriveType(PChar(Drv + '\')) <> DRIVE_REMOVABLE Then
    Application.MessageBox('对不起,请把本程序放至到优盘上使用!', 'Error', MB_ICONHAND)
  Else
    If GetUSBDiskID(Drv, Pid) Then Memo1.Lines.Add(Pid);
End;


学习中请遵守法律法规,本网站内容均来自于互联网,本网站不负担法律责任
获取 优盘 序列号 实现 方法
#1楼
发帖时间:2016-7-9   |   查看数:0   |   回复数:0
游客组
快速回复