/*
 * Copyright 2008, Marcus Overhagen. All rights reserved.
 * Distributed under the terms of the MIT License.
 */
#ifndef _SATA_REQUEST_H
#define _SATA_REQUEST_H


#include "ahci_defs.h"
#include "scsi_cmds.h"


class sata_request {
public:
                                                                sata_request();
                                                                sata_request(scsi_ccb* ccb);
                                                                ~sata_request();

                        void                            SetData(void* data, size_t dataSize);

                        void                            SetATACommand(uint8 command);
                        void                            SetATA28Command(uint8 command, uint32 lba,
                                                                        uint8 sectorCount);
                        void                            SetATA48Command(uint8 command, uint64 lba,
                                                                        uint16 sectorCount);
                        void                            SetFeature(uint16 feature);

                        void                            SetATAPICommand(size_t transferLength);
                        bool                            IsATAPI();
                        bool                            IsTestUnitReady();

                        scsi_ccb*                       CCB();
                        const void*                     FIS();
                        void*                           Data();
                        int                                     Size();
                        void                            Finish(int tfd, size_t bytesTransfered);
                        void                            Abort();

                        void                            WaitForCompletion();
                        int                                     CompletionStatus();

private:
                        scsi_ccb*                       fCcb;
                        uint8                           fFis[20];
                        bool                            fIsATAPI;
                        sem_id                          fCompletionSem;
                        int                                     fCompletionStatus;
                        void*                           fData;
                        size_t                          fDataSize;
};


inline scsi_ccb*
sata_request::CCB()
{
        return fCcb;
}


inline const void*
sata_request::FIS()
{
        return fFis;
}


inline bool
sata_request::IsATAPI()
{
        return fIsATAPI;
}


inline bool
sata_request::IsTestUnitReady()
{
        return fIsATAPI && fCcb != NULL && fCcb->cdb[0] == SCSI_OP_TEST_UNIT_READY;
}


inline void*
sata_request::Data()
{
        return fData;
}


inline int
sata_request::Size()
{
        return fDataSize;
}


#endif  /* _SATA_REQUEST_H */