mass_storage_device.h

Go to the documentation of this file.

/**
 * @file    mass_storage_device.h
 *
 * @date April 9, 2021
 * @author Matthew Kennedy, (c) 2021
 *
 * This file implements a USB mass storage device with multiple LUNs, so multiple drives can be mounted at once.
 */
 
#include "hal_usb_msd.h"
#include "thread_controller.h"
 
#if HAL_USE_USB_MSD
 
class MassStorageController : public ThreadController<USB_MSD_THREAD_WA_SIZE> {
public:
    MassStorageController(USBDriver* usb);
 
    void attachLun(uint8_t lunIndex, BaseBlockDevice *blkdev, uint8_t *blkbuf, size_t blkbufsize,
                    const scsi_inquiry_response_t *inquiry,
                    const scsi_unit_serial_number_inquiry_response_t *serialInquiry);
protected:
    void ThreadTask() override;
 
private:
    static bool cbwValid(const msd_cbw_t& cbw, msg_t status);
    static bool cbwMeaningful(const msd_cbw_t& cbw);
 
    void sendCsw(uint8_t status, uint32_t residue);
 
    usbmsdstate_t m_state;
 
    USBDriver* const m_usb;
 
    // temporary storage
    msd_cbw_t m_cbw;
    msd_csw_t m_csw;
 
    SCSITransport m_scsiTransport;
    usb_scsi_transport_handler_t m_scsiTransportHandler;
 
    chibios_rt::Mutex m_lunMutex;
 
    struct LunEntry {
        SCSITarget target;
        SCSITargetConfig config;
    };
 
    LunEntry m_luns[USB_MSD_LUN_COUNT];
};
 
#endif // HAL_USE_USB_MSD