2019-03-27 06:38:36 +00:00
|
|
|
/*
|
|
|
|
* The MIT License (MIT)
|
|
|
|
*
|
|
|
|
* Copyright (c) 2018 Ha Thach for Adafruit Industries
|
|
|
|
*
|
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
|
|
* in the Software without restriction, including without limitation the rights
|
|
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
|
|
* furnished to do so, subject to the following conditions:
|
|
|
|
*
|
|
|
|
* The above copyright notice and this permission notice shall be included in
|
|
|
|
* all copies or substantial portions of the Software.
|
|
|
|
*
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
|
|
* THE SOFTWARE.
|
|
|
|
*/
|
2018-04-05 11:35:30 +00:00
|
|
|
|
2018-08-14 08:56:24 +00:00
|
|
|
#include "tusb.h"
|
|
|
|
#include "uf2/uf2.h"
|
2018-04-05 11:35:30 +00:00
|
|
|
|
2018-04-12 11:11:45 +00:00
|
|
|
#if CFG_TUD_MSC
|
2018-04-05 11:35:30 +00:00
|
|
|
|
2019-02-21 17:21:34 +00:00
|
|
|
#include "bootloader.h"
|
|
|
|
|
2018-04-20 06:42:51 +00:00
|
|
|
/*------------------------------------------------------------------*/
|
|
|
|
/* MACRO TYPEDEF CONSTANT ENUM
|
|
|
|
*------------------------------------------------------------------*/
|
2018-05-11 13:34:17 +00:00
|
|
|
|
2018-06-20 09:44:11 +00:00
|
|
|
/*------------------------------------------------------------------*/
|
|
|
|
/* UF2
|
|
|
|
*------------------------------------------------------------------*/
|
2019-02-21 17:21:34 +00:00
|
|
|
static WriteState _wr_state = { 0 };
|
|
|
|
|
2018-06-21 05:22:42 +00:00
|
|
|
void read_block(uint32_t block_no, uint8_t *data);
|
2019-02-21 17:21:34 +00:00
|
|
|
int write_block(uint32_t block_no, uint8_t *data, bool quiet, WriteState *state);
|
2018-06-20 09:44:11 +00:00
|
|
|
|
2018-04-25 08:57:31 +00:00
|
|
|
//--------------------------------------------------------------------+
|
|
|
|
// tinyusb callbacks
|
|
|
|
//--------------------------------------------------------------------+
|
2018-07-31 06:53:55 +00:00
|
|
|
|
|
|
|
// Callback invoked when received an SCSI command not in built-in list below
|
|
|
|
// - READ_CAPACITY10, READ_FORMAT_CAPACITY, INQUIRY, MODE_SENSE6, REQUEST_SENSE
|
|
|
|
// - READ10 and WRITE10 has their own callbacks
|
2018-07-26 09:22:21 +00:00
|
|
|
int32_t tud_msc_scsi_cb (uint8_t lun, uint8_t const scsi_cmd[16], void* buffer, uint16_t bufsize)
|
2018-04-05 11:35:30 +00:00
|
|
|
{
|
2018-07-31 06:53:55 +00:00
|
|
|
void const* response = NULL;
|
2019-03-12 20:05:21 +00:00
|
|
|
int32_t resplen = 0;
|
2019-03-12 19:57:51 +00:00
|
|
|
memset(buffer, 0, bufsize);
|
2018-04-19 09:56:25 +00:00
|
|
|
|
2018-07-31 06:53:55 +00:00
|
|
|
switch ( scsi_cmd[0] )
|
2018-04-05 11:35:30 +00:00
|
|
|
{
|
|
|
|
case SCSI_CMD_TEST_UNIT_READY:
|
2018-07-26 09:22:21 +00:00
|
|
|
// Command that host uses to check our readiness before sending other commands
|
2018-07-31 06:53:55 +00:00
|
|
|
resplen = 0;
|
2018-04-05 11:35:30 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
case SCSI_CMD_PREVENT_ALLOW_MEDIUM_REMOVAL:
|
2018-07-26 09:22:21 +00:00
|
|
|
// Host is about to read/write etc ... better not to disconnect disk
|
2018-07-31 06:53:55 +00:00
|
|
|
resplen = 0;
|
2018-07-26 09:22:21 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
case SCSI_CMD_START_STOP_UNIT:
|
2018-07-31 06:53:55 +00:00
|
|
|
// Host try to eject/safe remove/poweroff us. We could safely disconnect with disk storage, or go into lower power
|
|
|
|
/* scsi_start_stop_unit_t const * start_stop = (scsi_start_stop_unit_t const *) scsi_cmd;
|
|
|
|
// Start bit = 0 : low power mode, if load_eject = 1 : unmount disk storage as well
|
|
|
|
// Start bit = 1 : Ready mode, if load_eject = 1 : mount disk storage
|
|
|
|
start_stop->start;
|
|
|
|
start_stop->load_eject;
|
|
|
|
*/
|
|
|
|
resplen = 0;
|
2018-04-05 11:35:30 +00:00
|
|
|
break;
|
|
|
|
|
2018-04-19 09:56:25 +00:00
|
|
|
default:
|
2018-07-31 06:53:55 +00:00
|
|
|
// Set Sense = Invalid Command Operation
|
|
|
|
tud_msc_set_sense(lun, SCSI_SENSE_ILLEGAL_REQUEST, 0x20, 0x00);
|
|
|
|
|
|
|
|
// negative means error -> tinyusb could stall and/or response with failed status
|
|
|
|
resplen = -1;
|
|
|
|
break;
|
2018-04-19 09:56:25 +00:00
|
|
|
}
|
|
|
|
|
2018-04-25 08:57:31 +00:00
|
|
|
// return len must not larger than bufsize
|
2019-03-12 20:05:21 +00:00
|
|
|
if ( resplen > (int32_t)bufsize ) resplen = bufsize;
|
2018-04-19 09:56:25 +00:00
|
|
|
|
2018-07-31 06:53:55 +00:00
|
|
|
// copy response to stack's buffer if any
|
|
|
|
if ( response && resplen )
|
2018-04-25 08:57:31 +00:00
|
|
|
{
|
2018-07-31 06:53:55 +00:00
|
|
|
memcpy(buffer, response, resplen);
|
2018-04-05 11:35:30 +00:00
|
|
|
}
|
|
|
|
|
2018-07-31 06:53:55 +00:00
|
|
|
return resplen;
|
2018-04-05 11:35:30 +00:00
|
|
|
}
|
|
|
|
|
2018-07-31 06:53:55 +00:00
|
|
|
// Callback invoked when received READ10 command.
|
|
|
|
// Copy disk's data to buffer (up to bufsize) and return number of copied bytes.
|
2018-07-26 09:22:21 +00:00
|
|
|
int32_t tud_msc_read10_cb (uint8_t lun, uint32_t lba, uint32_t offset, void* buffer, uint32_t bufsize)
|
2018-04-20 06:42:51 +00:00
|
|
|
{
|
2018-07-26 09:22:21 +00:00
|
|
|
(void) lun;
|
2019-03-12 19:57:51 +00:00
|
|
|
memset(buffer, 0, bufsize);
|
2018-04-25 08:57:31 +00:00
|
|
|
|
2018-06-21 05:22:42 +00:00
|
|
|
// since we return block size each, offset should always be zero
|
|
|
|
TU_ASSERT(offset == 0, -1);
|
2018-06-20 09:44:11 +00:00
|
|
|
|
2018-06-21 05:22:42 +00:00
|
|
|
uint32_t count = 0;
|
2018-06-20 10:03:38 +00:00
|
|
|
|
2018-06-21 05:22:42 +00:00
|
|
|
while ( count < bufsize )
|
|
|
|
{
|
|
|
|
read_block(lba, buffer);
|
2018-06-20 10:03:38 +00:00
|
|
|
|
2018-06-21 05:22:42 +00:00
|
|
|
lba++;
|
|
|
|
buffer += 512;
|
|
|
|
count += 512;
|
2018-06-20 09:18:55 +00:00
|
|
|
}
|
2018-04-05 11:35:30 +00:00
|
|
|
|
2018-06-21 05:22:42 +00:00
|
|
|
return count;
|
2018-04-05 11:35:30 +00:00
|
|
|
}
|
2018-04-25 08:57:31 +00:00
|
|
|
|
2018-07-31 06:53:55 +00:00
|
|
|
// Callback invoked when received WRITE10 command.
|
|
|
|
// Process data in buffer to disk's storage and return number of written bytes
|
2018-11-23 08:54:02 +00:00
|
|
|
int32_t tud_msc_write10_cb (uint8_t lun, uint32_t lba, uint32_t offset, uint8_t* buffer, uint32_t bufsize)
|
2018-04-05 11:35:30 +00:00
|
|
|
{
|
2018-07-26 09:22:21 +00:00
|
|
|
(void) lun;
|
2018-04-05 11:35:30 +00:00
|
|
|
|
2018-06-25 09:00:42 +00:00
|
|
|
uint32_t count = 0;
|
|
|
|
int wr_ret;
|
2018-04-25 08:57:31 +00:00
|
|
|
|
2019-02-21 17:21:34 +00:00
|
|
|
while ( (count < bufsize) && ((wr_ret = write_block(lba, buffer, false, &_wr_state)) > 0) )
|
2018-04-25 08:57:31 +00:00
|
|
|
{
|
2018-06-25 09:00:42 +00:00
|
|
|
lba++;
|
|
|
|
buffer += 512;
|
|
|
|
count += 512;
|
2018-04-25 08:57:31 +00:00
|
|
|
}
|
2018-06-25 09:00:42 +00:00
|
|
|
|
|
|
|
// Consider non-uf2 block write as successful
|
|
|
|
return (wr_ret < 0) ? bufsize : count;
|
2018-04-05 11:35:30 +00:00
|
|
|
}
|
|
|
|
|
2019-02-21 17:21:34 +00:00
|
|
|
// Callback invoked when WRITE10 command is completed (status received and accepted by host).
|
|
|
|
void tud_msc_write10_complete_cb(uint8_t lun)
|
|
|
|
{
|
|
|
|
// uf2 file writing is complete --> complete DFU process
|
|
|
|
if ( _wr_state.numBlocks && (_wr_state.numWritten >= _wr_state.numBlocks) )
|
|
|
|
{
|
|
|
|
led_state(STATE_WRITING_FINISHED);
|
|
|
|
|
|
|
|
dfu_update_status_t update_status;
|
|
|
|
|
|
|
|
memset(&update_status, 0, sizeof(dfu_update_status_t ));
|
|
|
|
update_status.status_code = DFU_UPDATE_APP_COMPLETE;
|
|
|
|
update_status.app_crc = 0; // skip CRC checking with uf2 upgrade
|
|
|
|
update_status.app_size = _wr_state.numBlocks*256;
|
|
|
|
|
|
|
|
bootloader_dfu_update_process(update_status);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-11-23 08:54:02 +00:00
|
|
|
void tud_msc_capacity_cb(uint8_t lun, uint32_t* block_count, uint16_t* block_size)
|
|
|
|
{
|
|
|
|
(void) lun;
|
|
|
|
|
|
|
|
*block_count = UF2_NUM_BLOCKS;
|
|
|
|
*block_size = 512;
|
|
|
|
}
|
|
|
|
|
2018-04-05 11:35:30 +00:00
|
|
|
#endif
|