首页
社区
课程
招聘
[旧帖] 驱动相关:IDE控制器模式如何用程序获取 0.00雪花
发表于: 2009-12-10 17:07 5027

[旧帖] 驱动相关:IDE控制器模式如何用程序获取 0.00雪花

2009-12-10 17:07
5027
借帐号发帖:
    如何 用程序 获取IDE控制器的驱动是什么模式(标准双通道PCIDE模式还是AHCI模式)?

给个提示或者方法,有函数名最好。

[课程]Linux pwn 探索篇!

收藏
免费 0
支持
分享
最新回复 (5)
雪    币: 41
活跃值: (10)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
2
h文件:

------------------
// MyDiskInfo.h: interface for the CMyDiskInfo class.
//
//////////////////////////////////////////////////////////////////////

#if !defined(AFX_MYDISKINFO_H__FF310990_FA82_421A_94CD_1F696F3AB948__INCLUDED_)
#define AFX_MYDISKINFO_H__FF310990_FA82_421A_94CD_1F696F3AB948__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000

class CMyDiskInfo  
{
public:
        int GetDiskInfo(int driver=0);
        char szModelNumber[64];
        char szSerialNumber[64];
        CMyDiskInfo();
        virtual ~CMyDiskInfo();

};

#endif // !defined(AFX_MYDISKINFO_H__FF310990_FA82_421A_94CD_1F696F3AB948__INCLUDED_)
2009-12-10 17:43
0
雪    币: 41
活跃值: (10)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
3
cpp文件

-----------------

// MyDiskInfo.cpp: implementation of the CMyDiskInfo class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "DiskInfo.h"
#include "MyDiskInfo.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

const WORD IDE_ATAPI_IDENTIFY = 0xA1;   // 读取ATAPI设备的命令
const WORD IDE_ATA_IDENTIFY   = 0xEC;   // 读取ATA设备的命令

#define _WIN32_WINNT 0x0400
#include "winioctl.h"

//#include "NTDDSCSI.h"
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CMyDiskInfo::CMyDiskInfo()
{

}

CMyDiskInfo::~CMyDiskInfo()
{

}

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;
}

int CMyDiskInfo::GetDiskInfo(int driver)
{
        CString sFilePath;
        sFilePath.Format("\\\\.\\PHYSICALDRIVE%d", driver);

        HANDLE hFile = INVALID_HANDLE_VALUE;
        hFile = ::CreateFile(sFilePath,
                                                GENERIC_READ | GENERIC_WRITE,
                                                FILE_SHARE_READ | FILE_SHARE_WRITE,
                                                NULL, OPEN_EXISTING,
                                                0, NULL);
        if (hFile == INVALID_HANDLE_VALUE)        return -1;

        DWORD dwBytesReturned;
        GETVERSIONINPARAMS gvopVersionParams;
        DeviceIoControl(hFile,
                                        SMART_GET_VERSION,
                                        NULL,
                                        0,
                                        &gvopVersionParams,
                                        sizeof(gvopVersionParams),
                                        &dwBytesReturned, NULL);

        if(gvopVersionParams.bIDEDeviceMap <= 0)        return -2;

        // IDE or ATAPI IDENTIFY cmd
        int btIDCmd = 0;
        SENDCMDINPARAMS InParams;
        int nDrive =0;
        btIDCmd = (gvopVersionParams.bIDEDeviceMap >> nDrive & 0x10) ? IDE_ATAPI_IDENTIFY : IDE_ATA_IDENTIFY;

        // 输出参数
        BYTE btIDOutCmd[sizeof(SENDCMDOUTPARAMS) + IDENTIFY_BUFFER_SIZE - 1];

        if(DoIdentify(hFile,
                                        &InParams,
                                        (PSENDCMDOUTPARAMS)btIDOutCmd,
                                        (BYTE)btIDCmd,
                                        (BYTE)nDrive, &dwBytesReturned) == FALSE)        return -3;
        ::CloseHandle(hFile);

        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(szSerialNumber, ConvertToString(dwDiskData, 10, 19));

        // 取模型号
        ZeroMemory(szModelNumber, sizeof(szModelNumber));
        strcpy(szModelNumber, ConvertToString(dwDiskData, 27, 46));

        return 0;
}
2009-12-10 17:43
0
雪    币: 41
活跃值: (10)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
4
看看是否有帮助呀
2009-12-10 17:44
0
雪    币: 444
活跃值: (10)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
5
谢谢楼上,刚开始接触这个东西,得慢慢看。

不过,看样子应该可以,Thanks。
2009-12-10 17:48
0
雪    币: 444
活跃值: (10)
能力值: ( LV2,RANK:10 )
在线值:
发帖
回帖
粉丝
6
据他说:
    “这个是磁盘的,我想要的是驱动上如何做。上面的代码跟我要做的事情关系不大。”

这个东西接触的人可能不太多,好像跟驱动下注册表信息相关,知道的给个提示就好。

也不抱多大的希望,3天内结贴。再次感谢2楼。
2009-12-11 09:18
0
游客
登录 | 注册 方可回帖
返回
//