获取物理磁盘和移动磁盘信息

class CPhysicalDriver
{
public:
void mfGetSerialNumber(int); //序列号 serial number
BOOL mfGetCapacity(int);     //容量 capacity
void mfGetLogicalNum();
void mfSetLogicalNum();
BOOL mfGetDerverDeteilInfo(int);

public:
CPhysicalDriver();
CPhysicalDriver(int miNumber);
void mfGetPhysicalInfo(int);
private:
void mfSetPhysicalBaseInfo(int);
public:
double mdPhysicalCapacity;
int miPhysicalNumber;
int miLogicalNum;
char macPhysicalSerial[50]; //用于磁盘信息 显示 硬盘显示序列号 移动磁盘显示 驱动名称
int  miPhysicalTypeFlag;    //0 硬盘 1 移动磁盘
char macPhysicalName[20];
char macPhysicalLogicalTable[26];
};CPhysicalDriver::CPhysicalDriver()
{
mdPhysicalCapacity =0;
miPhysicalNumber =0;
miLogicalNum =0;
miPhysicalTypeFlag =-1;
memset(macPhysicalName,0,sizeof(macPhysicalName));
memset(macPhysicalSerial,0,sizeof(macPhysicalSerial));
}

void CPhysicalDriver::mfSetPhysicalBaseInfo(int miNumber)
{
this->miPhysicalNumber =miNumber;
sprintf(this->macPhysicalName,"\\\\.\\physicaldrive%d",miNumber);
}
void  CPhysicalDriver::mfGetSerialNumber(int miNumber)
{
miPhysicalNumber =miNumber;
char mcPhysicalTempName[20]={0};
sprintf(mcPhysicalTempName,"\\\\.\\physicaldrive%d",miNumber);
HANDLE mHandler =CreateFileA(mcPhysicalTempName, GENERIC_READ|GENERIC_WRITE,FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_SYSTEM, NULL);
if (mHandler == INVALID_HANDLE_VALUE)
{
printf(" open error");
return ;
}

SENDCMDINPARAMS inParam = { 0 };
char szOutput[sizeof(SENDCMDOUTPARAMS)+IDENTIFY_BUFFER_SIZE-1] = { 0 };
DWORD dwRet = 0;

inParam.cBufferSize = IDENTIFY_BUFFER_SIZE;
inParam.bDriveNumber = miPhysicalNumber;
inParam.irDriveRegs.bFeaturesReg = 0;
inParam.irDriveRegs.bSectorCountReg = 1;
inParam.irDriveRegs.bSectorNumberReg = 1;
inParam.irDriveRegs.bCylLowReg = 0;
inParam.irDriveRegs.bCylHighReg = 0;
inParam.irDriveRegs.bDriveHeadReg = (miPhysicalNumber & 1) ? 0xB0 : 0xA0;    // 主盘和从盘所对应的值是不一样的
inParam.irDriveRegs.bCommandReg = ID_CMD;
if(::DeviceIoControl(mHandler, SMART_RCV_DRIVE_DATA, &inParam, sizeof(inParam)-1, szOutput, sizeof(SENDCMDOUTPARAMS)+IDENTIFY_BUFFER_SIZE-1, &dwRet, NULL))
{
SIDSECTOR msDrivesInfo;
memset(&msDrivesInfo, 0, sizeof(msDrivesInfo));
memcpy(&msDrivesInfo, ((SENDCMDOUTPARAMS*)szOutput)->bBuffer, sizeof(msDrivesInfo));
char szSerialNumber[SERIALNUMBER_LEN] = { 0 }; // 序列号
//获取序列号
{
char szTemp[SERIALNUMBER_LEN] = { 0 };

for (int i=0, len=SERIALNUMBER_LEN-1; i<len; i+=2)
{
szTemp[i+1] = msDrivesInfo.sSerialNumber[i];
szTemp[i] = msDrivesInfo.sSerialNumber[i+1];
}
for (int i=0, i_Dst=0; i<SERIALNUMBER_LEN; i++)
{
if (szTemp[i]>32 && szTemp[i]<127)
szSerialNumber[i_Dst++] = szTemp[i];
}
}
strncpy(macPhysicalSerial, szSerialNumber,strlen(szSerialNumber));
}

}

BOOL CPhysicalDriver::mfGetCapacity(int miNumber)
{
DISK_GEOMETRY pdg;
HANDLE hDevice;               // handle to the drive to be examined
BOOL bResult;                 // results flag
DWORD junk;                   // discard results
char mcPhysicalTempBuffer[30]={0};
sprintf(mcPhysicalTempBuffer,"\\\\.\\physicaldrive%d",miNumber);
hDevice = CreateFileA(mcPhysicalTempBuffer,  // drive to open
0,                // no access to the drive
FILE_SHARE_READ | // share mode
FILE_SHARE_WRITE,
NULL,             // default security attributes
OPEN_EXISTING,    // disposition
0,                // file attributes
NULL);            // do not copy file attributes

if (hDevice == INVALID_HANDLE_VALUE) // cannot open the drive
{
return (FALSE);
}

bResult = DeviceIoControl(hDevice,  // device to be queried
IOCTL_DISK_GET_DRIVE_GEOMETRY,  // operation to perform
NULL, 0, // no input buffer
&pdg, sizeof(pdg),     // output buffer
&junk,                 // # bytes returned
(LPOVERLAPPED) NULL);  // synchronous I/O

CloseHandle(hDevice);

// 设置磁盘的类型
if(pdg.MediaType == RemovableMedia)
miPhysicalTypeFlag =1;
else if(pdg.MediaType == FixedMedia)
miPhysicalTypeFlag =0;

//容量
if(bResult)
{
mdPhysicalCapacity=pdg.Cylinders.QuadPart * (ULONG)pdg.TracksPerCylinder *
(ULONG)pdg.SectorsPerTrack * (ULONG)pdg.BytesPerSector;
}
else
{
printf(" bResult error ");
}
return (bResult);
}

void CPhysicalDriver::mfGetPhysicalInfo(int miNumber)
{
mfSetPhysicalBaseInfo(miNumber);
if(!this->mfGetCapacity(miNumber))
printf("get capacity error \n");
if(!this->mfGetDerverDeteilInfo(miNumber))
printf("get dervershow info error \n");
}

BOOL CPhysicalDriver::mfGetDerverDeteilInfo(int miNumber)
{
PSTORAGE_DEVICE_DESCRIPTOR msDeverInfo;
msDeverInfo=(PSTORAGE_DEVICE_DESCRIPTOR)new  BYTE[sizeof(STORAGE_DEVICE_DESCRIPTOR) + 512 - 1];
msDeverInfo->Size =  sizeof(STORAGE_DEVICE_DESCRIPTOR) + 512 - 1;

STORAGE_PROPERTY_QUERY  Query;    // 查询输入参数
DWORD  dwOutBytes;                // IOCTL输出数据长度
BOOL  bResult;
// IOCTL返回值
char *mpUsbInfo;
char mcPhysicalTempBuffer[30]={0};
sprintf(mcPhysicalTempBuffer,"\\\\.\\physicaldrive%d",miNumber);
HANDLE hDevice = CreateFileA(mcPhysicalTempBuffer,  // drive to open
0,                // no access to the drive
FILE_SHARE_READ | // share mode
FILE_SHARE_WRITE,
NULL,             // default security attributes
OPEN_EXISTING,    // disposition
0,                // file attributes
NULL);            // do not copy file attributes

if (hDevice == INVALID_HANDLE_VALUE) // cannot open the drive
{
return (FALSE);
}
// 指定查询方式
Query.PropertyId =  StorageDeviceProperty;
Query.QueryType =  PropertyStandardQuery;

// 用IOCTL_STORAGE_QUERY_PROPERTY取设备属性信息
bResult =  ::DeviceIoControl(hDevice, // 设备句柄
IOCTL_STORAGE_QUERY_PROPERTY,    // 取设备属性信息
&Query,  sizeof(STORAGE_PROPERTY_QUERY),    // 输入数据缓冲区
msDeverInfo,  msDeverInfo->Size,        // 输出数据缓冲区
&dwOutBytes,                     // 输出数据长度
(LPOVERLAPPED)NULL);             // 用同步I/O
if(bResult)
{
mpUsbInfo =(char*)msDeverInfo;
if(msDeverInfo->RemovableMedia)
{
if (!msDeverInfo->VendorIdOffset) //厂家设定值 偏移
{
printf(" get VendorId error in the GetUsbInfo\n");
}
if (!msDeverInfo->ProductIdOffset) //产品ID的偏移
{
printf(" get ProductId error in the GetUsbInfo");
}
sprintf(macPhysicalSerial,"%s %s %s",&mpUsbInfo[msDeverInfo->VendorIdOffset],&mpUsbInfo[msDeverInfo->ProductIdOffset],&mpUsbInfo[msDeverInfo->SerialNumberOffset]);
}
else
{
if(!msDeverInfo->SerialNumberOffset)
{
printf("get SerialNumber error in the GetUsbInfo");
}
sprintf(macPhysicalSerial,"%s",&mpUsbInfo[msDeverInfo->ProductIdOffset]);
}
printf("%s ",macPhysicalSerial);
return true;
}
else
{
return false;
}

}
CPhysicalDriver::CPhysicalDriver(int miNumber)
{
mdPhysicalCapacity =0;
miPhysicalNumber =miNumber;
miLogicalNum =0;
miPhysicalTypeFlag =-1;
memset(macPhysicalName,0,sizeof(macPhysicalName));
memset(macPhysicalSerial,0,sizeof(macPhysicalSerial));
sprintf(macPhysicalName,"\\\\.\\physicaldrive%d",miNumber);
}

void CPhysicalDriver::mfSetLogicalNum()
{
miLogicalNum++;
}