#include "stdafx.h"
|
#include "DiskFunc.h"
|
|
//´ò¿ªÉ豸 //sFileName É豸µÄ ÎļþÃû £¨¾ø¶Ô·¾¶£©
|
HANDLE OpenDevice(LPCTSTR sFileName)
|
{
|
HANDLE hDevice; //´ò¿ªÉ豸
|
hDevice =:: CreateFile(sFileName, //ÎļþÃû
|
GENERIC_READ|GENERIC_WRITE, //¶Áд·½Ê½
|
FILE_SHARE_READ|FILE_SHARE_WRITE,//¹²Ïí·½Ê½
|
NULL, //ĬÈϵݲȫÃèÊö·û
|
OPEN_EXISTING, //´´½¨·½Ê½
|
0, //²»ÐèÉèÖÃÎļþÊôÐÔ
|
NULL); //²»Ðè²ÎÕÕÄ£°æÎļþ
|
return hDevice;
|
}
|
char szSerialNumber[2048];
|
char szModelNumber[2048];
|
BOOL __fastcall DoIdentify( HANDLE hPhysicalDriveIOCTL,
|
PSENDCMDINPARAMS pSCIP,
|
PSENDCMDOUTPARAMS pSCOP,
|
BYTE btIDCmd,
|
BYTE btDriveNum,
|
PDWORD pdwBytesReturned)
|
{
|
pSCIP->cBufferSize = IDENTIFY_BUFFER_SIZE;
|
pSCIP->irDriveRegs.bFeaturesReg = 0;
|
pSCIP->irDriveRegs.bSectorCountReg = 1;
|
pSCIP->irDriveRegs.bSectorNumberReg = 1;
|
pSCIP->irDriveRegs.bCylLowReg = 0;
|
pSCIP->irDriveRegs.bCylHighReg = 0;
|
|
pSCIP->irDriveRegs.bDriveHeadReg = (btDriveNum & 1) ? 0xB0 : 0xA0;
|
pSCIP->irDriveRegs.bCommandReg = btIDCmd;
|
pSCIP->bDriveNumber = btDriveNum;
|
pSCIP->cBufferSize = IDENTIFY_BUFFER_SIZE;
|
|
return DeviceIoControl( hPhysicalDriveIOCTL,
|
SMART_RCV_DRIVE_DATA,
|
(LPVOID)pSCIP,
|
sizeof(SENDCMDINPARAMS) - 1,
|
(LPVOID)pSCOP,
|
sizeof(SENDCMDOUTPARAMS) + IDENTIFY_BUFFER_SIZE - 1,
|
pdwBytesReturned, NULL);
|
}
|
|
char *__fastcall ConvertToString(DWORD dwDiskData[256], int nFirstIndex, int nLastIndex)
|
{
|
static char szResBuf[1024];
|
char ss[256];
|
int nIndex = 0;
|
int nPosition = 0;
|
|
for(nIndex = nFirstIndex; nIndex <= nLastIndex; nIndex++)
|
{
|
ss[nPosition] = (char)(dwDiskData[nIndex] / 256);
|
nPosition++;
|
|
// Get low BYTE for 2nd character
|
ss[nPosition] = (char)(dwDiskData[nIndex] % 256);
|
nPosition++;
|
}
|
|
// End the string
|
ss[nPosition] = '\0';
|
|
int i, index=0;
|
for(i=0; i<nPosition; i++)
|
{
|
if(ss[i]==0 || ss[i]==32) continue;
|
szResBuf[index]=ss[i];
|
index++;
|
}
|
szResBuf[index]=0;
|
|
return szResBuf;
|
}
|
void AdjustString(char * str, int len)
|
{
|
char ch;int i;
|
//Á½Á½µßµ¹
|
for (i =0; i<len; i+=2)
|
{
|
ch=str[i];
|
str[i]=str[i+1];
|
str[i+1]=ch;
|
}
|
str[len-1]=0;
|
//ÈôÊÇÓÒ¶ÔÆ䣬µ÷ÕûΪ×ó¶ÔÆä
|
|
}
|
BOOL IDentifyDeviceAsScsi(HANDLE hDevice, int nDrive, PIDINFO pIdInfo)
|
{
|
PSENDCMDINPARAMS pSCIP;
|
PSENDCMDOUTPARAMS pSCOP;
|
PSRB_IO_CONTROL pSRBIO;
|
DWORD dwOutBytes;
|
BOOL bResult;
|
|
pSRBIO = (PSRB_IO_CONTROL)::GlobalAlloc(LMEM_ZEROINIT, sizeof(SRB_IO_CONTROL)+sizeof(SENDCMDOUTPARAMS)+sizeof(IDINFO) -1);
|
pSCIP = (PSENDCMDINPARAMS)((char *)pSRBIO+sizeof(SRB_IO_CONTROL));
|
pSCOP = (PSENDCMDOUTPARAMS)pSCIP;
|
//Ìî³äÊý¾Ý
|
pSRBIO->HeaderLength=sizeof(SRB_IO_CONTROL);
|
pSRBIO->Timeout=10000;
|
pSRBIO->Length=sizeof(SENDCMDOUTPARAMS)+sizeof(IDINFO)-1;
|
pSRBIO->ControlCode=IOCTL_SCSI_MINIPORT_IDENTIFY;
|
memcpy_s((char *)pSRBIO->Signature,8,"SCSIDISK",8);
|
//strncpy_s((char *)pSRBIO->Signature,8,"SCSIDISK",8);
|
//Ö¸¶¨ATA/ATAPI ÃüÁîµÄ¼Ä´æÆ÷Öµ
|
pSCIP->irDriveRegs.bFeaturesReg = 0;
|
pSCIP->irDriveRegs.bSectorCountReg = 0;
|
pSCIP->irDriveRegs.bSectorNumberReg = 0;
|
pSCIP->irDriveRegs.bCylLowReg = 0;
|
pSCIP->irDriveRegs.bCylHighReg = 0;
|
pSCIP->irDriveRegs.bDriveHeadReg =0;
|
pSCIP->irDriveRegs.bCommandReg = IDE_ATA_IDENTIFY;
|
pSCIP->bDriveNumber = nDrive;
|
//IDENTIFY DEVICE
|
bResult = ::DeviceIoControl(hDevice, IOCTL_SCSI_MINIPORT,
|
pSRBIO,sizeof(SRB_IO_CONTROL) + sizeof(SENDCMDINPARAMS) -1,
|
pSRBIO,sizeof(SRB_IO_CONTROL) + sizeof(SENDCMDOUTPARAMS) +sizeof(IDINFO)-1,
|
&dwOutBytes,
|
NULL);
|
::memcpy(pIdInfo,pSCOP->bBuffer,sizeof(IDINFO));
|
::GlobalFree(pSRBIO);
|
return bResult;
|
|
}
|
//ÓÃSCSIÇý¶¯¶ÁÈ¡IDEÓ²Å̵ÄÉ豸ÐÅÏ¢£¬²»ÊÜȨÏÞÖÆÔ¼
|
//nDriver Çý¶¯Æ÷ºÃ( 0 = Primary master, 1=Primary Slave, 2 = Second Master,
|
//pIdInfo É豸ÐÅÏ¢½á¹¹Ö¸Õë;
|
BOOL GetIdeDriveAsScsiInfoInNT(int nDrive, PIDINFO pIdInfo)
|
{
|
HANDLE hDevice;//
|
BOOL bResult; //
|
// char szFileName[20];
|
// ::sprintf(szFileName,"\\\\.\\Scsi%d:",nDrive/2);
|
CString sFileName;
|
sFileName.Format(_T("\\\\.\\Scsi%d:"),nDrive/2);
|
hDevice = OpenDevice(sFileName);
|
if (hDevice == INVALID_HANDLE_VALUE) {return false;}
|
//IDENTIFY DEVICE
|
bResult = IDentifyDeviceAsScsi(hDevice,nDrive%2,pIdInfo);
|
CloseHandle(hDevice);
|
//¼ì²éÊDz»ÊÇ¿Õ´®;
|
if (pIdInfo->sModelNumber[0] == '\0') {bResult = false;}
|
if (bResult)
|
{
|
//µ÷Õû×Ö·û´®
|
::AdjustString(pIdInfo->sSerialNumber,20);
|
::AdjustString(pIdInfo->sModelNumber,40);
|
::AdjustString(pIdInfo->sFirmwareRev,8);
|
}
|
return bResult;
|
|
}
|
int GetHDSerialID(int driver)
|
{
|
// Ó²ÅÌÎïÀíÐòÁкÅÊÇÓ²Å̵ijö³§ÐòÁкţ¬ËüÊÇÈ«Çò¶¼ÊÇΨһµÄ£¬²»»áËæ×ÅϵͳµÄ°²×°¡¢Ó²Å̵ĸñʽ»¯µÈ²Ù×÷¶ø¸Ä±ä£¬¸úmacµØÖ·Ò»Ñù¶¼¾ßÓÐΨһÐÔ¡£
|
CString s1;
|
// 1£¬µÚÒ»²½£º´´½¨É豸¶ÔÏ󣬵õ½É豸¾ä±ú£¬É豸ΪӲÅÌ¡£
|
{
|
// int driver=0;
|
CString sFilePath;
|
sFilePath.Format(_T("\\\\.\\PHYSICALDRIVE%d"), driver);
|
HANDLE hFile=::CreateFile(sFilePath,
|
GENERIC_READ , FILE_SHARE_READ ,
|
NULL, OPEN_EXISTING,0, NULL);
|
s1.Format(_T("´´½¨Îļþ %s ½á¹û ¾ä±ú %X \r\n"),sFilePath,hFile);
|
// mylogger1.LogTxt(s1);
|
if (hFile==INVALID_HANDLE_VALUE) {return -1;}
|
DWORD dwBytesReturned;
|
GETVERSIONINPARAMS gvopVersionParams;
|
int j = DeviceIoControl(hFile, //ÏòÉ豸¶ÔÏó·¢ËÍSMART_GET_VERSIONÉ豸ÇëÇ󣬻ñÈ¡Ó²ÅÌÊôÐÔ
|
SMART_GET_VERSION, NULL, 0,
|
&gvopVersionParams, sizeof(gvopVersionParams),
|
&dwBytesReturned, NULL);
|
s1.Format(_T("»ñÈ¡´ÅÅÌÊôÐÔ 0x%X Return %d Bytes %d bIDEDeviceMap %d \r\n"),SMART_GET_VERSION,j,dwBytesReturned,gvopVersionParams.bIDEDeviceMap);
|
// mylogger1.LogTxt(s1);
|
|
// if(gvopVersionParams.bIDEDeviceMap <= 0) return -2;
|
// 2¡£µÚ¶þ²½£¬·¢ËÍSMART_RCV_DRIVE_DATAÉ豸ÇëÇ󣬻ñÈ¡Ó²ÅÌÏêϸÐÅÏ¢¡£
|
// IDE or ATAPI IDENTIFY cmd
|
int btIDCmd = 0;
|
SENDCMDINPARAMS InParams;
|
int nDrive =0;
|
btIDCmd = (gvopVersionParams.bIDEDeviceMap >> nDrive & 0x10) ? IDE_ATAPI_IDENTIFY : IDE_ATA_IDENTIFY;
|
|
s1.Format(_T("btIDCmd %X \r\n "),btIDCmd);
|
// mylogger1.LogTxt(s1);
|
|
// Êä³ö²ÎÊý
|
BYTE btIDOutCmd[sizeof(SENDCMDOUTPARAMS) + IDENTIFY_BUFFER_SIZE - 1];
|
int bResult=DoIdentify(hFile,
|
&InParams,
|
(PSENDCMDOUTPARAMS)btIDOutCmd,
|
(BYTE)btIDCmd,
|
(BYTE)nDrive, &dwBytesReturned);
|
::CloseHandle(hFile);
|
if( bResult== FALSE) return -3;
|
|
DWORD dwDiskData[256];
|
USHORT *pIDSector; // ¶ÔÓ¦½á¹¹IDSECTOR£¬¼ûÍ·Îļþ
|
|
pIDSector = (USHORT*)((SENDCMDOUTPARAMS*)btIDOutCmd)->bBuffer;
|
for(int i=0; i < 256; i++) dwDiskData[i] = pIDSector[i];
|
|
// ȡϵÁкÅ
|
ZeroMemory(szSerialNumber, sizeof(szSerialNumber));
|
strcpy_s(szSerialNumber, 2048, ConvertToString(dwDiskData, 10, 19));
|
|
// È¡Ä£ÐͺÅ
|
ZeroMemory(szModelNumber, sizeof(szModelNumber));
|
strcpy_s(szModelNumber, 2048, ConvertToString(dwDiskData, 27, 46));
|
s1.Format(_T("\r\n Model %S \r\n Serial %S \r\n"),szModelNumber,szSerialNumber);
|
// mylogger1.LogTxt(s1);
|
int len1=strlen(szSerialNumber);
|
short int machinecode[15];
|
int len2=(len1+1)/2;
|
s1.Format(_T("MachineCode: "));
|
for (int i=0;i<len2;i++)
|
{
|
if (i!=0) s1.Append(_T("-"));
|
int j=((szSerialNumber[i*2]-48)+(szSerialNumber[i*2+1]-48)*64)%8192+1234;
|
machinecode[i]=j;
|
s1.AppendFormat(_T("%d"),j);
|
}
|
s1.Append(_T("\r\n"));
|
// mylogger1.LogTxt(s1);
|
s1.Format(_T("Back : "));
|
for (int i=0;i<len2;i++)
|
{
|
int j=machinecode[i];
|
int k,l;
|
k=((j-1234)%64)+48;l=((j-1234)/64)+48;
|
s1.AppendFormat(_T("%c%c"),k,l);
|
}
|
s1.AppendFormat(_T("\r\n"));
|
// mylogger1.LogTxt(s1);
|
return 0;
|
}
|
}
|
