📄 getdeceiveinfo.cpp
字号:
#include <windows.h>
#include <iostream.h>
#include <stdio.h>
#include "Setupapi.h"
#include <devguid.h>
#define DFP_GET_VERSION 0x00074080
#define DFP_SEND_DRIVE_COMMAND 0x0007c084
#define DFP_RECEIVE_DRIVE_DATA 0x0007c088
#pragma pack(1)
typedef struct _GETVERSIONOUTPARAMS {
BYTE bVersion; // Binary driver version.
BYTE bRevision; // Binary driver revision.
BYTE bReserved; // Not used.
BYTE bIDEDeviceMap; // Bit map of IDE devices.
DWORD fCapabilities; // Bit mask of driver capabilities.
DWORD dwReserved[4]; // For future use.
} GETVERSIONOUTPARAMS, *PGETVERSIONOUTPARAMS, *LPGETVERSIONOUTPARAMS;
typedef struct _IDEREGS {
BYTE bFeaturesReg; // Used for specifying SMART "commands".
BYTE bSectorCountReg; // IDE sector count register
BYTE bSectorNumberReg; // IDE sector number register
BYTE bCylLowReg; // IDE low order cylinder value
BYTE bCylHighReg; // IDE high order cylinder value
BYTE bDriveHeadReg; // IDE drive/head register
BYTE bCommandReg; // Actual IDE command.
BYTE bReserved; // reserved for future use. Must be zero.
} IDEREGS, *PIDEREGS, *LPIDEREGS;
typedef struct _SENDCMDINPARAMS {
DWORD cBufferSize; // Buffer size in bytes
IDEREGS irDriveRegs; // Structure with drive register values.
BYTE bDriveNumber; // Physical drive number to send
// command to (0,1,2,3).
BYTE bReserved[3]; // Reserved for future expansion.
DWORD dwReserved[4]; // For future use.
BYTE bBuffer[1]; // Input buffer.
} SENDCMDINPARAMS, *PSENDCMDINPARAMS, *LPSENDCMDINPARAMS;
typedef struct _DRIVERSTATUS {
BYTE bDriverError; // Error code from driver,
// or 0 if no error.
BYTE bIDEStatus; // Contents of IDE Error register.
// Only valid when bDriverError
// is SMART_IDE_ERROR.
BYTE bReserved[2]; // Reserved for future expansion.
DWORD dwReserved[2]; // Reserved for future expansion.
} DRIVERSTATUS, *PDRIVERSTATUS, *LPDRIVERSTATUS;
typedef struct _SENDCMDOUTPARAMS {
DWORD cBufferSize; // Size of bBuffer in bytes
DRIVERSTATUS DriverStatus;// Driver status structure.
BYTE bBuffer[512]; // Buffer of arbitrary length
// in which to store the data read from the drive.
} SENDCMDOUTPARAMS, *PSENDCMDOUTPARAMS, *LPSENDCMDOUTPARAMS;
typedef struct _IDSECTOR {
USHORT wGenConfig;
USHORT wNumCyls;
USHORT wReserved;
USHORT wNumHeads;
USHORT wBytesPerTrack;
USHORT wBytesPerSector;
USHORT wSectorsPerTrack;
USHORT wVendorUnique[3];
CHAR sSerialNumber[20];
USHORT wBufferType;
USHORT wBufferSize;
USHORT wECCSize;
CHAR sFirmwareRev[8];
CHAR sModelNumber[40];
USHORT wMoreVendorUnique;
USHORT wDoubleWordIO;
USHORT wCapabilities;
USHORT wReserved1;
USHORT wPIOTiming;
USHORT wDMATiming;
USHORT wBS;
USHORT wNumCurrentCyls;
USHORT wNumCurrentHeads;
USHORT wNumCurrentSectorsPerTrack;
ULONG ulCurrentSectorCapacity;
USHORT wMultSectorStuff;
ULONG ulTotalAddressableSectors;
USHORT wSingleWordDMA;
USHORT wMultiWordDMA;
BYTE bReserved[128];
} IDSECTOR, *PIDSECTOR;
/*+++ Global vars ---*/
GETVERSIONOUTPARAMS vers;
SENDCMDINPARAMS in;
SENDCMDOUTPARAMS out;
HANDLE h;
DWORD i;
BYTE j;
void CopyRight()
{
cerr<<endl<<"HDD identifier v1.0 for WIN95/98/Me/NT/2000. written by Lu Lin"<<endl;
cerr<<"For more information, please visit Inside Programming: http://lu0.126.com"<<endl;
cerr<<"2000.11.3"<<endl<<endl;
}
VOID ChangeByteOrder(PCHAR szString, USHORT uscStrSize)
{
USHORT i;
CHAR temp;
for(i=0;i< uscStrSize; i+=2)
{
temp = szString[i];
szString[i] = szString[i+1];
szString[i+1] = temp;
}
}
void DetectIDE(BYTE bIDEDeviceMap)
{
if(bIDEDeviceMap&1)
{
if(bIDEDeviceMap&16)
{
cout<<"ATAPI device is attached to primary controller, drive0."<<endl;
}
else
{
cout<<"IDE device is attached to primary controller, drive0."<<endl;
}
}
if(bIDEDeviceMap&2)
{
if(bIDEDeviceMap&32)
{
cout<<"ATAPI device is attached to primary controller, drive1."<<endl;
}
else
{
cout<<"IDE device is attached to primary controller, drive1."<<endl;
}
}
if(bIDEDeviceMap&4)
{
if(bIDEDeviceMap&64)
{
cout<<"ATAPI device is attached to secondary controller, drive0."<<endl;
}
else
{
cout<<"IDE device is attached to secondary controller, drive0."<<endl;
}
}
if(bIDEDeviceMap&8)
{
if(bIDEDeviceMap&128)
{
cout<<"ATAPI device is attached to secondary controller, drive1."<<endl;
}
else
{
cout<<"IDE device is attached to secondary controller, drive1."<<endl;
}
}
}
void hdid9x()
{
ZeroMemory(&vers,sizeof(vers));
//We start in 95/98/Me
h=CreateFile("\\\\.\\Smartvsd",0,0,0,CREATE_NEW,0,0);
if(!h)
{
cout<<"open smartvsd.vxd failed"<<endl;
exit(0);
}
if(!DeviceIoControl(h,DFP_GET_VERSION,0,0,&vers,sizeof(vers),&i,0))
{
cout<<"DeviceIoControl failed:DFP_GET_VERSION"<<endl;
CloseHandle(h);
return;
}
//If IDE identify command not supported, fails
if(!(vers.fCapabilities&1))
{
cout<<"Error: IDE identify command not supported.";
CloseHandle(h);
return;
}
//Display IDE drive number detected
DetectIDE(vers.bIDEDeviceMap);
//Identify the IDE drives
for(j=0;j<4;j++)
{
PIDSECTOR phdinfo;
char s[41];
ZeroMemory(&in,sizeof(in));
ZeroMemory(&out,sizeof(out));
if(j&1)
{
in.irDriveRegs.bDriveHeadReg=0xb0;
}
else
{
in.irDriveRegs.bDriveHeadReg=0xa0;
}
if(vers.fCapabilities&(16>>j))
{
//We don't detect a ATAPI device.
cout<<"Drive "<<(int)(j+1)<<" is a ATAPI device, we don't detect it"<<endl;
continue;
}
else
{
in.irDriveRegs.bCommandReg=0xec;
}
in.bDriveNumber=j;
in.irDriveRegs.bSectorCountReg=1;
in.irDriveRegs.bSectorNumberReg=1;
in.cBufferSize=512;
if(!DeviceIoControl(h,DFP_RECEIVE_DRIVE_DATA,&in,sizeof(in),&out,sizeof(out),&i,0))
{
cout<<"DeviceIoControl failed:DFP_RECEIVE_DRIVE_DATA"<<endl;
CloseHandle(h);
return;
}
phdinfo=(PIDSECTOR)out.bBuffer;
memcpy(s,phdinfo->sModelNumber,40);
s[40]=0;
ChangeByteOrder(s,40);
cout<<endl<<"Module Number:"<<s<<endl;
memcpy(s,phdinfo->sFirmwareRev,8);
s[8]=0;
ChangeByteOrder(s,8);
cout<<"\tFirmware rev:"<<s<<endl;
memcpy(s,phdinfo->sSerialNumber,20);
s[20]=0;
ChangeByteOrder(s,20);
cout<<"\tSerial Number:"<<s<<endl;
cout<<"\tCapacity:"<<phdinfo->ulTotalAddressableSectors/2/1024<<"M"<<endl<<endl;
}
//Close handle before quit
CloseHandle(h);
CopyRight();
}
void hdidnt()
{
PIDSECTOR phdinfo;
char hd[80];
char s[41];
int j = 0;
unsigned long i;
ZeroMemory(&vers,sizeof(vers));
//We start in NT/Win2000
for(j=0;j<4;j++)
{
sprintf(hd,"\\\\.\\PhysicalDrive%d",j);
h=CreateFile(hd,GENERIC_READ|GENERIC_WRITE,FILE_SHARE_READ|FILE_SHARE_WRITE,0,OPEN_EXISTING,0,0);
if(!h)
{
continue;
}
if(!DeviceIoControl(h,DFP_GET_VERSION,0,0,&vers,sizeof(vers),&i,0))
{
CloseHandle(h);
continue;
}
//If IDE identify command not supported, fails
if(!(vers.fCapabilities&1))
{
cout<<"Error:IDE identify command not supported.";
CloseHandle(h);
return;
}
//Identify the IDE drives
ZeroMemory(&in,sizeof(in));
ZeroMemory(&out,sizeof(out));
if(j&1)
{
in.irDriveRegs.bDriveHeadReg=0xb0;
}
else
{
in.irDriveRegs.bDriveHeadReg=0xa0;
}
if(vers.fCapabilities&(16>>j))
{
//We don't detect a ATAPI device.
cout<<"Drive"<<(int)(j+1)<<"is a ATAPI device,we don't detect it"<<endl;
continue;
}
else
{
in.irDriveRegs.bCommandReg=0xec;
}
in.bDriveNumber=j;
in.irDriveRegs.bSectorCountReg=1;
in.irDriveRegs.bSectorNumberReg=1;
in.cBufferSize=512;
if(!DeviceIoControl(h,DFP_RECEIVE_DRIVE_DATA,&in,sizeof(in),&out,sizeof(out),&i,0))
{
cout<<"DeviceIoControl failed: DFP_RECEIVE_DRIVE_DATA"<<endl;
CloseHandle(h);
continue;
}
phdinfo=(PIDSECTOR)out.bBuffer;
memcpy(s,phdinfo->sModelNumber,40);
s[40]=0;
ChangeByteOrder(s,40);
cout<<endl<<"Module Number:"<<s<<endl;
memcpy(s,phdinfo->sFirmwareRev,8);
s[8]=0;
ChangeByteOrder(s,8);
cout<<"\tFirmware rev:"<<s<<endl;
memcpy(s,phdinfo->sSerialNumber,20);
s[20]=0;
ChangeByteOrder(s,20);
cout<<"\tSerial Number:"<<s<<endl;
cout<<"\tCapacity:"<<phdinfo->ulTotalAddressableSectors/2/1024<<"M"<<endl<<endl;
CloseHandle(h);
}
}
//枚举当前USB设备
GUID guid = {0xA5DCBF10L,0x6530,0x11D2,0x90,0x1F,0x00,0xC0,0x4F,0xB9,0x51,0xED};
void GetDeceiveModelNumber(char* buffer,char* ModelNumber)
{
char* temp = buffer;
while(*temp != '_')
{
++ temp;
}
++ temp;
while(*temp != '&')
{
*ModelNumber = *temp;
ModelNumber ++;
temp ++;
}
temp += 5;
while(*temp != '&')
{
*ModelNumber = *temp;
ModelNumber ++;
temp ++;
}
}
void GetDeceiveSerialNumber(char* buffer, char* SerialNumber)
{
int i =0;
char* temp = buffer;
while(*temp != '\\')
{
++ temp;
}
++ temp;
while(*temp != '\\')
{
temp ++;
}
++ temp;
while(*temp != '&')
{
*SerialNumber = *temp;
SerialNumber ++;
temp ++;
}
}
void GetUsbSerialNum()
{
char USBSN[23];
TCHAR buf[MAX_PATH] = {0};
TCHAR VID[MAX_PATH] = {0};
TCHAR PID[MAX_PATH] = {0};
TCHAR SN[MAX_PATH] = {0};
TCHAR SerialNumber[MAX_PATH] = {0};
TCHAR ModelNumber[MAX_PATH] = {0};
SP_DEVINFO_DATA DeviceInfoData;
HDEVINFO hDevInfo;
DWORD nSize = 0;
int i,j,bRes;
//DIGCF_ALLCLASSES
//hDevInfo = SetupDiGetClassDevs(&guid, NULL, NULL,DIGCF_PRESENT|DIGCF_DEVICEINTERFACE);
hDevInfo =SetupDiGetClassDevs(NULL,NULL,NULL, DIGCF_ALLCLASSES | DIGCF_PRESENT /*| DIGCF_DEVICEINTERFACE*/);
if(hDevInfo == INVALID_HANDLE_VALUE)
{
//no found USB device
cout<<"No found USB device"<<endl;
return;
}
DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
#if 1
i=0;
while (i>=0)
{
//DeviceInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
bRes =SetupDiEnumDeviceInfo(hDevInfo,(DWORD)i, &DeviceInfoData);
//bRes =SetupDiEnumDeviceInterfaces(hDevInfo,(DWORD)i, &DeviceInfoData);
if(bRes == 0 && GetLastError() == ERROR_NO_MORE_ITEMS)
{
break;
}
if(!SetupDiGetDeviceInstanceId(hDevInfo, &DeviceInfoData, buf, sizeof(buf),&nSize))
{
i++;
continue;
}
if(strncmp("USBSTOR",buf,7) == 0)
{
//cout<<"buf = "<<buf<<endl;
memset(ModelNumber,0,MAX_PATH);
memset(SerialNumber,0,MAX_PATH);
GetDeceiveModelNumber(buf,ModelNumber);
cout<<"ModelNumber = "<<ModelNumber<<endl;
GetDeceiveSerialNumber(buf,SerialNumber);
cout<<"SerialNumber = "<<SerialNumber<<"\n"<<endl;
}
i++;
}
#endif
/*
for(j=0; SetupDiEnumDeviceInfo(hDevInfo, j, &DeviceInfoData); j++)
{
if(!SetupDiGetDeviceInstanceId(hDevInfo, &DeviceInfoData, buf, sizeof(buf),&nSize))
{
continue;
}
#if 0
for(i=0;i <23;i++)
{
VID[i] = buf[8+i];
}
VID[i] = '\0';
cout<<"Module Num = "<<VID<<"\n"<<endl;
for(i=0; i < 8; i++)
{
PID[i] = buf[48+i];
}
PID[i] = '\0';
cout<<"Revision Num = "<<PID<<"\n"<<endl;
for (i=0;i <9; i++)
{
USBSN[i] = buf[57+i];
}
USBSN[i] = '\0';
cout<<"Other = "<<USBSN<<"\n"<<endl;
#endif
if(strncmp("USBSTOR",buf,7) == 0)
{
cout<<"buf = "<<buf<<"\n"<<endl;
}
}
*/
SetupDiDestroyDeviceInfoList(hDevInfo);
}
void main()
{
OSVERSIONINFO VersionInfo;
ZeroMemory(&VersionInfo,sizeof(VersionInfo));
VersionInfo.dwOSVersionInfoSize=sizeof(VersionInfo);
GetVersionEx(&VersionInfo);
switch (VersionInfo.dwPlatformId)
{
case VER_PLATFORM_WIN32s:
cout<<"Win32s is not supported by this programm."<<endl;
return;
case VER_PLATFORM_WIN32_WINDOWS:
hdid9x();
return;
case VER_PLATFORM_WIN32_NT:
//hdidnt();
GetUsbSerialNum();
return;
}
}⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -