simmel-bootloader/src/usb/msc_uf2.c

173 lines
5.4 KiB
C
Raw Normal View History

/*
* 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
#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
#include "bootloader.h"
2018-04-20 06:42:51 +00:00
/*------------------------------------------------------------------*/
/* MACRO TYPEDEF CONSTANT ENUM
*------------------------------------------------------------------*/
2018-06-20 09:44:11 +00:00
/*------------------------------------------------------------------*/
/* UF2
*------------------------------------------------------------------*/
static WriteState _wr_state = { 0 };
2018-06-21 05:22:42 +00:00
void read_block(uint32_t block_no, uint8_t *data);
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-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
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
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
}
// 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);
}
}
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