Sending ATA commands directly to device in Windows?

前端 未结 3 1029

I’m trying to send ATA commands to a physical disk in Windows, and get the response from the device.

Note: In this case I want to sen

3条回答
  •  轮回少年
    2020-12-08 11:28

    You need to use IOCTL_ATA_PASS_THROUGH/IOCTL_ATA_PASS_THROUGH_DIRECT, these are quite well documented. Also, you need GENERIC_READ|GENERIC_WRITE access for CreateFile.

    Be aware that pre XP SP2 does not support these properly. Also, if you have a nForce based MB with nvidia drivers, your SATA drives will appear as SCSI and you can't use this passthrough.

    In some cases, the SMART IOCTL's (e.g. SMART_RCV_DRIVE_DATA) will work on nForce drivers. You can use these to get IDENTIFY and SMART data, but not much else.

    The open source smartmontools is a good place to start looking for sample code.

    EDIT: Sample from an app talking to ATA devices.

    EResult DeviceOperationManagerWin::executeATACommandIndirect(ATACommand & Cmd) {
        const uint32 FillerSize = 0;
        Utils::ByteBuffer B;
        B.reserve(sizeof(ATA_PASS_THROUGH_EX) + 4 + Cmd.bufferSize());
        ATA_PASS_THROUGH_EX & PTE = * (ATA_PASS_THROUGH_EX *) B.appendPointer(sizeof(ATA_PASS_THROUGH_EX) + FillerSize + Cmd.bufferSize());
        uint8 * DataPtr = ((uint8 *) &PTE) + sizeof(ATA_PASS_THROUGH_EX) + FillerSize;
    
        memset(&PTE, 0, sizeof(ATA_PASS_THROUGH_EX) + FillerSize);
        PTE.Length = sizeof(PTE);
        PTE.AtaFlags = 0;
        PTE.AtaFlags |= Cmd.requiresDRDY() ? ATA_FLAGS_DRDY_REQUIRED : 0;
        switch (Cmd.dataDirection()) {
        case ddFromDevice: 
            PTE.AtaFlags |= ATA_FLAGS_DATA_IN; 
            break;
        case ddToDevice:
            PTE.AtaFlags |= ATA_FLAGS_DATA_OUT;
            memcpy(DataPtr, Cmd.buffer(), Cmd.bufferSize());
            break;
        default:
            break;
        }
        PTE.AtaFlags |= Cmd.is48Bit() ? ATA_FLAGS_48BIT_COMMAND : 0;
        PTE.AtaFlags |= Cmd.isDMA() ? ATA_FLAGS_USE_DMA : 0;
        PTE.DataTransferLength = Cmd.bufferSize();
        PTE.TimeOutValue = Cmd.timeout();
        PTE.DataBufferOffset = sizeof(PTE) + FillerSize;
        PTE.DataTransferLength = Cmd.bufferSize();
        PTE.CurrentTaskFile[0] = Cmd.taskFileIn0().Features;
        PTE.CurrentTaskFile[1] = Cmd.taskFileIn0().Count;
        PTE.CurrentTaskFile[2] = Cmd.taskFileIn0().LBALow;
        PTE.CurrentTaskFile[3] = Cmd.taskFileIn0().LBAMid;
        PTE.CurrentTaskFile[4] = Cmd.taskFileIn0().LBAHigh;
        PTE.CurrentTaskFile[5] = Cmd.taskFileIn0().Device;
        PTE.CurrentTaskFile[6] = Cmd.taskFileIn0().Command;
        PTE.CurrentTaskFile[7] = 0;
        if (Cmd.is48Bit()) {
            PTE.PreviousTaskFile[0] = Cmd.taskFileIn1().Features;
            PTE.PreviousTaskFile[1] = Cmd.taskFileIn1().Count;
            PTE.PreviousTaskFile[2] = Cmd.taskFileIn1().LBALow;
            PTE.PreviousTaskFile[3] = Cmd.taskFileIn1().LBAMid;
            PTE.PreviousTaskFile[4] = Cmd.taskFileIn1().LBAHigh;
            PTE.PreviousTaskFile[5] = Cmd.taskFileIn1().Device;
            PTE.PreviousTaskFile[6] = 0;
            PTE.PreviousTaskFile[7] = 0;
        }
    
        DWORD BR; 
        if (!DeviceIoControl(FHandle, IOCTL_ATA_PASS_THROUGH, &PTE, B.size(), &PTE, B.size(), &BR, 0)) {
            FLastOSError = GetLastError();
            LOG_W << "ioctl ATA_PT failed for " << Cmd << ": " << FLastOSError << " (" << Utils::describeOSError(FLastOSError) << ")";
            return Utils::mapOSError(FLastOSError);
        }
        Cmd.taskFileOut0().Error = PTE.CurrentTaskFile[0];
        Cmd.taskFileOut0().Count = PTE.CurrentTaskFile[1];
        Cmd.taskFileOut0().LBALow = PTE.CurrentTaskFile[2];
        Cmd.taskFileOut0().LBAMid = PTE.CurrentTaskFile[3];
        Cmd.taskFileOut0().LBAHigh = PTE.CurrentTaskFile[4];
        Cmd.taskFileOut0().Device = PTE.CurrentTaskFile[5];
        Cmd.taskFileOut0().Status = PTE.CurrentTaskFile[6];
        Cmd.taskFileOut1().Error = PTE.PreviousTaskFile[0];
        Cmd.taskFileOut1().Count = PTE.PreviousTaskFile[1];
        Cmd.taskFileOut1().LBALow = PTE.PreviousTaskFile[2];
        Cmd.taskFileOut1().LBAMid = PTE.PreviousTaskFile[3];
        Cmd.taskFileOut1().LBAHigh = PTE.PreviousTaskFile[4];
        Cmd.taskFileOut1().Device = PTE.PreviousTaskFile[5];
        Cmd.taskFileOut1().Status = PTE.PreviousTaskFile[6];
        if (Cmd.dataDirection() == ddFromDevice) {
            memcpy(Cmd.buffer(), DataPtr, Cmd.bufferSize());
        }
        return resOK;
        }
    

    EDIT: Sample without external dependencies.

    IDENTIFY requires a 512 byte buffer for data:

    unsigned char Buffer[512 + sizeof(ATA_PASS_THROUGH_EX)] = { 0 };
    ATA_PASS_THROUGH_EX & PTE = *(ATA_PASS_THROUGH_EX *) Buffer;
    PTE.Length = sizeof(PTE);
    PTE.TimeOutValue = 10;
    PTE.DataTransferLength = 512;
    PTE.DataBufferOffset = sizeof(ATA_PASS_THROUGH_EX);
    

    Set up the IDE registers as specified in ATA spec.

    IDEREGS * ir = (IDEREGS *) PTE.CurrentTaskFile;
    ir->bCommandReg = 0xEC;
    ir->bSectorCountReg = 1;
    

    IDENTIFY is neither 48-bit nor DMA, it reads from the device:

    PTE.AtaFlags = ATA_FLAGS_DATA_IN | ATA_FLAGS_DRDY_REQUIRED;
    

    Do the ioctl:

    DeviceIOControl(Handle, IOCTL_ATA_PASS_THROUGH, &PTE, sizeof(Buffer), &PTE, sizeof(Buffer), &BR, 0);
    

    Here you should insert error checking, both from DeviceIOControl and by looking at IDEREGS for device reported errors.

    Get the IDENTIFY data, assuming you have defined a struct IdentifyData

    IdentifyData * IDData = (IdentifyData *) (Buffer + sizeof(ATA_PASS_THROUGH_EX));
    

提交回复
热议问题