659 lines
33 KiB
C
659 lines
33 KiB
C
#include <stdint.h>
|
|
|
|
typedef struct USBD_descriptor {
|
|
uint32_t addr_tx;
|
|
uint32_t count_tx;
|
|
uint32_t addr_rx;
|
|
uint32_t count_rx;
|
|
} USBD_descriptor;
|
|
|
|
// USBD memory allocation:
|
|
// WARNING THIS IS A 16-bit MEMORY LMAO
|
|
// +0x00
|
|
// descriptors (two sets of two endpoints)
|
|
// +0x20
|
|
// EP 0 OUT (8 bytes)
|
|
// +0x30
|
|
// EP 0 IN (8 bytes)
|
|
// +0x40
|
|
// EP 1 OUT (64 bytes)
|
|
// +0xc0
|
|
// EP 1 IN (64 bytes)
|
|
// +0x140
|
|
// <unused>
|
|
// +0x200
|
|
// <256-byte flash page buffer>
|
|
|
|
const uint8_t USB_DEV_DESC[18] __attribute__((aligned(16))) = {
|
|
18, // bLength
|
|
1, // bDescriptorType
|
|
0x00, 0x02, // bcdUSB
|
|
0, // bDeviceClass
|
|
0, // bDeviceSubClass
|
|
0, // bDeviceProtocol
|
|
8, // bMaxPacketSize0
|
|
0x55, 0xf0, // idVendor
|
|
0x00, 0x00, // idProduct
|
|
0x00, 0x00, // bcdDevice
|
|
1, // iManufacturer
|
|
2, // iProduct
|
|
3, // iSerialNumber
|
|
1, // bNumConfigurations
|
|
};
|
|
|
|
const uint8_t USB_CONF_DESC[0x20] __attribute__((aligned(16))) = {
|
|
9, // bLength
|
|
2, // bDescriptorType
|
|
0x20, 0x00, // wTotalLength
|
|
1, // bNumInterfaces
|
|
1, // bConfigurationValue
|
|
0, // iConfiguration
|
|
0x80, // bmAttributes
|
|
250, // bMaxPower
|
|
|
|
9, // bLength
|
|
4, // bDescriptorType
|
|
0, // bInterfaceNumber
|
|
0, // bAlternateSetting
|
|
2, // bNumEndpoints
|
|
0x08, // bInterfaceClass
|
|
0x05, // bInterfaceSubClass
|
|
0x50, // bInterfaceProtocol
|
|
0, // iInterface
|
|
|
|
7, // bLength
|
|
5, // bDescriptorType
|
|
0x81, // bEndpointAddress
|
|
2, // bmAttributes
|
|
0x40, 0x00, // wMaxPacketSize
|
|
0, // bInterval
|
|
|
|
7, // bLength
|
|
5, // bDescriptorType
|
|
0x01, // bEndpointAddress
|
|
2, // bmAttributes
|
|
0x40, 0x00, // wMaxPacketSize
|
|
0, // bInterval
|
|
};
|
|
|
|
const uint16_t USB_MANUF[13] = u"\u031aArcaneNibble";
|
|
const uint16_t USB_PRODUCT[15] = u"\u031eCH32V UF2 Boot";
|
|
const uint8_t HEXLUT[16] = "0123456789ABCDEF";
|
|
|
|
const uint8_t INQUIRY_RESPONSE[36] __attribute__((aligned(16))) = {
|
|
0x00,
|
|
0x00,
|
|
0x04,
|
|
0x02,
|
|
0x1F,
|
|
0x00,
|
|
0x00,
|
|
0x00,
|
|
'A', 'r', 'c', 'a', 'n', 'e', 'N', 'b',
|
|
'C', 'H', '3', '2', 'V', ' ', 'U', 'F', '2', ' ', 'B', 'o', 'o', 't', ' ', ' ',
|
|
' ', ' ', ' ', ' ',
|
|
};
|
|
|
|
const uint8_t BOOT_SECTOR[0x3e] __attribute__((aligned(16))) = {
|
|
0x00, 0x00, 0x00, // jump
|
|
'A', 'r', 'c', 'a', 'n', 'e', 'N', 'b', // oem name
|
|
0x00, 0x02, // 512 bytes/sector
|
|
0x01, // 1 sector/cluster
|
|
0x01, 0x00, // 1 reserved sector
|
|
0x01, // 1 fat
|
|
0x10, 0x00, // 16 root dir entries
|
|
0x00, 0x40, // num sectors
|
|
0xf8, // media descriptor
|
|
0x40, 0x00, // 16384 sectors/clusters -> 64 FAT sectors
|
|
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, // useless stuff
|
|
0x80, // "drive number"
|
|
0x00, // reserved
|
|
0x29, // magic
|
|
0xde, 0xad, 0xbe, 0xef, // serial number
|
|
'C', 'H', '3', '2', 'V', ' ', 'U', 'F', '2', ' ', ' ', // volume label
|
|
'F', 'A', 'T', '1', '6', ' ', ' ', ' ', // fs type
|
|
};
|
|
|
|
const uint8_t ROOT_DIR[32 * 3] __attribute__((aligned(16))) = {
|
|
'C', 'H', '3', '2', 'V', ' ', 'U', 'F', '2', ' ', ' ', // name
|
|
0x08, // attributes (volume label)
|
|
0x00, 0x00, // reserved
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, // timestamps
|
|
0x00, 0x00, // reserved
|
|
0x00, 0x00, 0x00, 0x00, // timestamps
|
|
0x00, 0x00, // start cluster
|
|
0x00, 0x00, 0x00, 0x00, // file size]
|
|
|
|
'I', 'N', 'F', 'O', '_', 'U', 'F', '2', 'T', 'X', 'T', // name
|
|
0x01, // attributes (RO)
|
|
0x00, 0x00, // reserved
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, // timestamps
|
|
0x00, 0x00, // reserved
|
|
0x00, 0x00, 0x00, 0x00, // timestamps
|
|
0x00, 0x00, // start cluster
|
|
0x00, 0x00, 0x00, 0x00, // file size
|
|
|
|
'I', 'N', 'D', 'E', 'X', ' ', ' ', ' ', 'H', 'T', 'M', // name
|
|
0x01, // attributes (RO)
|
|
0x00, 0x00, // reserved
|
|
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, // timestamps
|
|
0x00, 0x00, // reserved
|
|
0x00, 0x00, 0x00, 0x00, // timestamps
|
|
0x00, 0x00, // start cluster
|
|
0x00, 0x00, 0x00, 0x00, // file size
|
|
};
|
|
|
|
#define ESIG_UNIID(x) (*(volatile uint8_t*)(0x1FFFF7E8 + (x)))
|
|
// XXX manual claims 96 bits but only 64 bits seem to actually be programmed?
|
|
|
|
#define USB_DESCS ((volatile USBD_descriptor*)0x40006000)
|
|
#define USB_EP0_OUT(offs) (*(volatile uint32_t *)(0x40006020 + 2 * (offs)))
|
|
#define USB_EP0_IN(offs) (*(volatile uint32_t *)(0x40006030 + 2 * (offs)))
|
|
#define USB_EP1_OUT(offs) (*(volatile uint32_t *)(0x40006040 + 2 * (offs)))
|
|
#define USB_EP1_IN(offs) (*(volatile uint32_t *)(0x400060C0 + 2 * (offs)))
|
|
|
|
#define USB_EPTYPE_BULK 0b00
|
|
#define USB_EPTYPE_CONTROL 0b01
|
|
|
|
#define USB_STAT_DISABLED 0b00
|
|
#define USB_STAT_STALL 0b01
|
|
#define USB_STAT_ACK 0b11
|
|
|
|
#define R32_RCC_CTLR (*(volatile uint32_t*)0x40021000)
|
|
#define R32_RCC_CFGR0 (*(volatile uint32_t*)0x40021004)
|
|
#define R32_RCC_APB2PCENR (*(volatile uint32_t*)0x40021018)
|
|
#define R32_RCC_APB1PCENR (*(volatile uint32_t*)0x4002101C)
|
|
#define R32_GPIOA_CFGLR (*(volatile uint32_t*)0x40010800)
|
|
#define R32_GPIOA_CFGHR (*(volatile uint32_t*)0x40010804)
|
|
#define R32_GPIOA_BSHR (*(volatile uint32_t*)0x40010810)
|
|
|
|
#define R16_USBD_EPR(n) (*(volatile uint16_t*)(0x40005C00 + 4 * (n)))
|
|
#define R16_USBD_CNTR (*(volatile uint16_t*)0x40005C40)
|
|
#define R16_USBD_ISTR (*(volatile uint16_t*)0x40005C44)
|
|
#define R16_USBD_DADDR (*(volatile uint16_t*)0x40005C4C)
|
|
|
|
#define R32_EXTEN_CTR (*(volatile uint32_t*)0x40023800)
|
|
|
|
// state[7:0] = addr
|
|
#define STATE_SET_ADDR 0x00
|
|
// state[15:8] = byte pos
|
|
// state[7:0] = bytes left
|
|
#define STATE_GET_DESC 0x01
|
|
#define STATE_GET_STR_SERIAL 0x02
|
|
#define STATE_CTRL_SIMPLE_IN 0x03
|
|
|
|
#define STATE_WANT_CBW 0x00
|
|
#define STATE_SENT_CSW 0x01
|
|
#define STATE_SENT_DATA_IN 0x02
|
|
// state[10:8] = sector fragment
|
|
#define STATE_SEND_MORE_READ 0x03
|
|
|
|
__attribute__((always_inline)) static inline void set_ep_mode(uint32_t epidx, uint32_t epaddr, uint32_t eptype, uint32_t stat_rx, uint32_t stat_tx, uint32_t xtra, int clear_dtog) {
|
|
uint32_t val = R16_USBD_EPR(epidx);
|
|
uint32_t cur_stats;
|
|
if (clear_dtog)
|
|
cur_stats = val & 0x7070;
|
|
else
|
|
cur_stats = val & 0x3030;
|
|
uint32_t want_stats = (stat_rx << 12) | (stat_tx) << 4;
|
|
R16_USBD_EPR(epidx) = epaddr | (eptype << 9) | xtra | (cur_stats ^ want_stats);
|
|
}
|
|
|
|
__attribute__((always_inline)) static inline uint32_t min(uint32_t a, uint32_t b) {
|
|
if (a <= b)
|
|
return a;
|
|
return b;
|
|
}
|
|
|
|
__attribute__((always_inline)) static inline void synthesize_block(uint32_t block, uint32_t piece) {
|
|
if (block == 0) {
|
|
if (piece == 0) {
|
|
for (int i = 0; i < sizeof(BOOT_SECTOR) / 2; i++)
|
|
USB_EP1_IN(i * 2) = ((uint16_t*)BOOT_SECTOR)[i];
|
|
for (int i = sizeof(BOOT_SECTOR) / 2; i < 32; i++)
|
|
USB_EP1_IN(i * 2) = 0;
|
|
} else {
|
|
for (int i = 0; i < 32; i++)
|
|
USB_EP1_IN(i * 2) = 0;
|
|
}
|
|
} else if (block == 1 && piece == 0) {
|
|
USB_EP1_IN(0) = 0xfff8;
|
|
USB_EP1_IN(2) = 0xffff;
|
|
for (int i = 2; i < 32; i++)
|
|
USB_EP1_IN(i * 2) = 0;
|
|
} else if (block == 65 && piece == 0) {
|
|
for (int i = 0; i < 32; i++)
|
|
USB_EP1_IN(i * 2) = ((uint16_t*)ROOT_DIR)[i];
|
|
} else if (block == 65 && piece == 1) {
|
|
for (int i = 0; i < 16; i++)
|
|
USB_EP1_IN(i * 2) = ((uint16_t*)ROOT_DIR)[32+ i];
|
|
for (int i = 16; i < 32; i++)
|
|
USB_EP1_IN(i * 2) = 0;
|
|
} else {
|
|
for (int i = 0; i < 32; i++)
|
|
USB_EP1_IN(i * 2) = 0;
|
|
}
|
|
|
|
USB_DESCS[1].count_tx = 64;
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
}
|
|
|
|
__attribute__((naked)) int main(void) {
|
|
// PLL setup: system clock 96 MHz
|
|
R32_EXTEN_CTR |= (1 << 4); // sneaky div2
|
|
R32_RCC_CFGR0 = (R32_RCC_CFGR0 & ~0xff0000) | (0b01101000 << 16);
|
|
R32_RCC_CTLR |= (1 << 24);
|
|
while (!(R32_RCC_CTLR & (1 << 25))) {}
|
|
R32_RCC_CFGR0 = (R32_RCC_CFGR0 & ~0b11) | 0b10;
|
|
while ((R32_RCC_CFGR0 & 0b1100) != 0b1000) {}
|
|
|
|
// As recommended in the manual, output low on these pins before enabling USB
|
|
R32_RCC_APB2PCENR |= (1 << 2);
|
|
R32_GPIOA_CFGHR = (R32_GPIOA_CFGHR & ~(0xff << 12)) | (0b00100010 << 12);
|
|
R32_GPIOA_BSHR = (1 << 27) | (1 << 28);
|
|
|
|
// Enable USB
|
|
R32_RCC_APB1PCENR |= (1 << 23);
|
|
R16_USBD_CNTR &= ~(1 << 1);
|
|
USB_DESCS[0].addr_tx = 0x30 / 2;
|
|
USB_DESCS[0].addr_rx = 0x20 / 2;
|
|
USB_DESCS[0].count_rx = (8 << 10);
|
|
USB_DESCS[1].addr_tx = 0xc0 / 2;
|
|
USB_DESCS[1].addr_rx = 0x40 / 2;
|
|
USB_DESCS[1].count_rx = (2 << 10) | (1 << 15);
|
|
// XXX because the table is at offset 0 don't bother writing this
|
|
|
|
// Attach USB
|
|
R16_USBD_CNTR &= ~(1 << 0);
|
|
R32_EXTEN_CTR |= (1 << 1);
|
|
|
|
uint32_t ctrl_xfer_state = 0;
|
|
uint32_t active_config = 0;
|
|
const uint8_t *outputting_desc = 0;
|
|
uint32_t outputting_desc_sz = 0;
|
|
|
|
// bits [31:24] = additional sense code
|
|
// bits [23:20] = sense key
|
|
// bits [7:0] = state
|
|
uint32_t msc_state = STATE_WANT_CBW;
|
|
uint32_t dCSWTag = 0;
|
|
// [31:0] = current lba
|
|
// [15:0] = blocks left
|
|
uint32_t scsi_xfer_lba_blocks = 0;
|
|
|
|
while (1) {
|
|
uint32_t usb_int_status = R16_USBD_ISTR;
|
|
if (usb_int_status & (1 << 10)) {
|
|
// RESET
|
|
// set all to STALL, SETUP will come in nonetheless
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_STALL, 0, 0);
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_DISABLED, USB_STAT_DISABLED, 0, 0);
|
|
R16_USBD_DADDR = 0x80;
|
|
} else if (usb_int_status & (1 << 15)) {
|
|
uint32_t epidx = usb_int_status & 0xf;
|
|
uint32_t ep_status = R16_USBD_EPR(epidx);
|
|
if ((ep_status & (1 << 15)) && (epidx == 0)) {
|
|
if (ep_status & (1 << 11)) {
|
|
// setup, STAT_RX and STAT_TX both now NAK
|
|
uint16_t bRequest_bmRequestType = USB_EP0_OUT(0);
|
|
uint32_t wLength = USB_EP0_OUT(6);
|
|
if (bRequest_bmRequestType == 0x0080 || bRequest_bmRequestType == 0x0081) {
|
|
// GET_STATUS
|
|
USB_EP0_IN(0) = 0;
|
|
USB_DESCS[0].count_tx = min(2, wLength);
|
|
ctrl_xfer_state = (STATE_CTRL_SIMPLE_IN << 24);
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
} else if (bRequest_bmRequestType == 0x0500) {
|
|
// SET_ADDRESS
|
|
ctrl_xfer_state = (STATE_SET_ADDR << 24) | USB_EP0_OUT(2);
|
|
USB_DESCS[0].count_tx = 0;
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
} else if (bRequest_bmRequestType == 0x0680) {
|
|
// GET_DESCRIPTOR
|
|
uint32_t wValue = USB_EP0_OUT(2);
|
|
if (wValue == 0x0100 || wValue == 0x0200 || wValue == 0x0301 || wValue == 0x0302) {
|
|
if (wValue == 0x0100) {
|
|
outputting_desc = USB_DEV_DESC;
|
|
outputting_desc_sz = sizeof(USB_DEV_DESC);
|
|
} else if (wValue == 0x0200) {
|
|
outputting_desc = USB_CONF_DESC;
|
|
outputting_desc_sz = sizeof(USB_CONF_DESC);
|
|
} else if (wValue == 0x0301) {
|
|
outputting_desc = (uint8_t*)USB_MANUF;
|
|
outputting_desc_sz = sizeof(USB_MANUF);
|
|
} else if (wValue == 0x0302) {
|
|
outputting_desc = (uint8_t*)USB_PRODUCT;
|
|
outputting_desc_sz = sizeof(USB_PRODUCT);
|
|
}
|
|
|
|
for (int i = 0; i < 8; i+=2)
|
|
USB_EP0_IN(i) = *((uint16_t*)(outputting_desc + i));
|
|
if (wLength < 8) {
|
|
USB_DESCS[0].count_tx = wLength;
|
|
ctrl_xfer_state = (STATE_GET_DESC << 24);
|
|
} else {
|
|
USB_DESCS[0].count_tx = 8;
|
|
ctrl_xfer_state = (STATE_GET_DESC << 24) | (8 << 8) | (wLength - 8);
|
|
}
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
} else if (wValue == 0x0300) {
|
|
// string LANGIDs
|
|
USB_EP0_IN(0) = 0x0304;
|
|
USB_EP0_IN(2) = 0x0409;
|
|
USB_DESCS[0].count_tx = min(4, wLength);
|
|
ctrl_xfer_state = (STATE_CTRL_SIMPLE_IN << 24);
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
} else if (wValue == 0x0303) {
|
|
USB_EP0_IN(0) = 0x0300 | (24 * 2 + 2);
|
|
USB_EP0_IN(2) = HEXLUT[ESIG_UNIID(0) >> 4];
|
|
USB_EP0_IN(4) = HEXLUT[ESIG_UNIID(0) & 0xf];
|
|
USB_EP0_IN(6) = HEXLUT[ESIG_UNIID(1) >> 4];
|
|
if (wLength < 8) {
|
|
USB_DESCS[0].count_tx = wLength;
|
|
ctrl_xfer_state = (STATE_GET_STR_SERIAL << 24);
|
|
} else {
|
|
USB_DESCS[0].count_tx = 8;
|
|
ctrl_xfer_state = (STATE_GET_STR_SERIAL << 24) | (3 << 8) | (wLength - 8);
|
|
}
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
} else {
|
|
// bad descriptor
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_STALL, 0, 0);
|
|
}
|
|
} else if (bRequest_bmRequestType == 0x0880) {
|
|
// GET_CONFIGURATION
|
|
USB_EP0_IN(0) = active_config;
|
|
USB_DESCS[0].count_tx = min(1, wLength);
|
|
ctrl_xfer_state = (STATE_CTRL_SIMPLE_IN << 24);
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
} else if (bRequest_bmRequestType == 0x0900) {
|
|
// SET_CONFIGURATION
|
|
uint32_t wValue = USB_EP0_OUT(2);
|
|
if (wValue == 0 || wValue == 1) {
|
|
active_config = wValue;
|
|
USB_DESCS[0].count_tx = 0;
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
if (wValue) {
|
|
// activate, allow OUT
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_ACK, USB_STAT_STALL, 0, 1);
|
|
msc_state = STATE_WANT_CBW;
|
|
} else {
|
|
// deactivate
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_DISABLED, USB_STAT_DISABLED, 0, 0);
|
|
}
|
|
} else {
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_STALL, 0, 0);
|
|
}
|
|
} else if (bRequest_bmRequestType == 0x0A81) {
|
|
// GET_INTERFACE
|
|
USB_EP0_IN(0) = 0;
|
|
USB_DESCS[0].count_tx = min(1, wLength);
|
|
ctrl_xfer_state = (STATE_CTRL_SIMPLE_IN << 24);
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
} else {
|
|
// unknown
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_STALL, 0, 0);
|
|
}
|
|
} else {
|
|
// back to stall for everything, expect SETUP
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_STALL, 0, 0);
|
|
}
|
|
} else if ((ep_status & (1 << 7)) && (epidx == 0)) {
|
|
switch (ctrl_xfer_state >> 24) {
|
|
case STATE_SET_ADDR:
|
|
R16_USBD_DADDR = 0x80 | (ctrl_xfer_state & 0x7f);
|
|
ctrl_xfer_state = 0;
|
|
// back to stall for everything, expect SETUP
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_STALL, 0, 0);
|
|
break;
|
|
case STATE_GET_DESC:
|
|
{
|
|
uint32_t byte_pos = (ctrl_xfer_state >> 8) & 0xff;
|
|
uint32_t bytes_left = ctrl_xfer_state & 0xff;
|
|
if (bytes_left == 0) {
|
|
// ACK for OUT ZLP, STALL for IN
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_ACK, USB_STAT_STALL, 1 << 8, 0);
|
|
} else {
|
|
uint32_t bytes = bytes_left;
|
|
if (bytes > 8) bytes = 8;
|
|
if (byte_pos + bytes > outputting_desc_sz)
|
|
bytes = outputting_desc_sz - byte_pos;
|
|
for (int i = 0; i < bytes; i+=2)
|
|
USB_EP0_IN(i) = *((uint16_t*)(outputting_desc + byte_pos + i));
|
|
if (bytes < 8) {
|
|
USB_DESCS[0].count_tx = bytes;
|
|
ctrl_xfer_state = (STATE_GET_DESC << 24);
|
|
} else {
|
|
USB_DESCS[0].count_tx = 8;
|
|
ctrl_xfer_state = (STATE_GET_DESC << 24) | ((byte_pos + 8) << 8) | (bytes_left - 8);
|
|
}
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
}
|
|
}
|
|
break;
|
|
case STATE_GET_STR_SERIAL:
|
|
{
|
|
uint32_t str_pos = (ctrl_xfer_state >> 8) & 0xff;
|
|
uint32_t bytes_left = ctrl_xfer_state & 0xff;
|
|
if (bytes_left == 0) {
|
|
// ACK for OUT ZLP, STALL for IN
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_ACK, USB_STAT_STALL, 1 << 8, 0);
|
|
} else {
|
|
// XXX this is confusing
|
|
uint32_t ascii_bytes = (bytes_left + 1) / 2;
|
|
if (ascii_bytes > 4) ascii_bytes = 4;
|
|
if (str_pos + ascii_bytes > 24)
|
|
ascii_bytes = 24 - str_pos;
|
|
for (int i = 0; i < ascii_bytes; i++)
|
|
USB_EP0_IN(i * 2) = HEXLUT[(ESIG_UNIID((str_pos + i) / 2) >> ((1 - ((str_pos + i) % 2)) * 4)) & 0xf];
|
|
if (ascii_bytes < 4) {
|
|
USB_DESCS[0].count_tx = bytes_left;
|
|
ctrl_xfer_state = (STATE_GET_STR_SERIAL << 24);
|
|
} else {
|
|
USB_DESCS[0].count_tx = 8;
|
|
ctrl_xfer_state = (STATE_GET_STR_SERIAL << 24) | ((str_pos + 4) << 8) | (bytes_left - 8);
|
|
}
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
}
|
|
}
|
|
break;
|
|
case STATE_CTRL_SIMPLE_IN:
|
|
// ACK for OUT ZLP, STALL for IN
|
|
set_ep_mode(0, 0, USB_EPTYPE_CONTROL, USB_STAT_ACK, USB_STAT_STALL, 1 << 8, 0);
|
|
break;
|
|
}
|
|
} else if ((ep_status & (1 << 15)) && (epidx == 1)) {
|
|
// ep 1 out
|
|
switch (msc_state & 0xff) {
|
|
case STATE_WANT_CBW:
|
|
if ((USB_DESCS[1].count_rx & 0x3f) != 0x1f || USB_EP1_OUT(0) != 0x5355 || USB_EP1_OUT(2) != 0x4342) {
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_ACK, USB_STAT_STALL, 0, 0);
|
|
} else {
|
|
dCSWTag = USB_EP1_OUT(4) | (USB_EP1_OUT(6) << 16);
|
|
uint32_t dCBWDataTransferLength = USB_EP1_OUT(8) | (USB_EP1_OUT(10) << 16);
|
|
uint32_t bmCBWFlags = USB_EP1_OUT(12) & 0xff;
|
|
uint32_t operation_code = USB_EP1_OUT(14) >> 8;
|
|
|
|
switch (operation_code) {
|
|
case 0x00:
|
|
// test unit ready
|
|
USB_EP1_IN(0) = 0x5355;
|
|
USB_EP1_IN(2) = 0x5342;
|
|
USB_EP1_IN(4) = dCSWTag;
|
|
USB_EP1_IN(6) = dCSWTag >> 16;
|
|
USB_EP1_IN(8) = 0;
|
|
USB_EP1_IN(10) = 0;
|
|
USB_EP1_IN(12) = 0;
|
|
USB_DESCS[1].count_tx = 13;
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
msc_state = STATE_SENT_CSW;
|
|
break;
|
|
case 0x03:
|
|
// request sense
|
|
USB_EP1_IN(0) = 0x0070;
|
|
USB_EP1_IN(2) = (msc_state >> 20) & 0xf;
|
|
USB_EP1_IN(4) = 0;
|
|
USB_EP1_IN(6) = 0;
|
|
USB_EP1_IN(8) = 0;
|
|
USB_EP1_IN(10) = 0;
|
|
USB_EP1_IN(12) = (msc_state >> 24) & 0xff;
|
|
USB_EP1_IN(14) = 0;
|
|
USB_EP1_IN(16) = 0;
|
|
USB_DESCS[1].count_tx = 18;
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
msc_state = STATE_SENT_DATA_IN;
|
|
break;
|
|
case 0x12:
|
|
// inquiry
|
|
if (USB_EP1_OUT(16) == 0) {
|
|
for (int i = 0; i < 36 / 2; i++) {
|
|
USB_EP1_IN(i * 2) = ((uint16_t*)INQUIRY_RESPONSE)[i];
|
|
}
|
|
USB_DESCS[1].count_tx = 36;
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
msc_state = STATE_SENT_DATA_IN;
|
|
break;
|
|
}
|
|
msc_state = STATE_SENT_CSW | (5 << 20) | (0x24 << 24);
|
|
goto error_csw;
|
|
case 0x1a:
|
|
// mode sense (6)
|
|
USB_EP1_IN(0) = 0x0003;
|
|
USB_EP1_IN(2) = 0x0000;
|
|
USB_DESCS[1].count_tx = 4;
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
msc_state = STATE_SENT_DATA_IN;
|
|
break;
|
|
case 0x5a:
|
|
// mode sense (10)
|
|
USB_EP1_IN(0) = 0x0800;
|
|
USB_EP1_IN(2) = 0x0000;
|
|
USB_EP1_IN(4) = 0x0000;
|
|
USB_EP1_IN(6) = 0x0000;
|
|
USB_DESCS[1].count_tx = 8;
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
msc_state = STATE_SENT_DATA_IN;
|
|
break;
|
|
case 0x25:
|
|
// READ CAPACITY (10)
|
|
// xxx don't bother checking the silly fields
|
|
USB_EP1_IN(0) = 0x0000; // 8 MiB
|
|
USB_EP1_IN(2) = 0xff3f;
|
|
USB_EP1_IN(4) = 0x0000; // 512 byte sectors
|
|
USB_EP1_IN(6) = 0x0002;
|
|
USB_DESCS[1].count_tx = 8;
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
msc_state = STATE_SENT_DATA_IN;
|
|
break;
|
|
case 0x28:
|
|
{
|
|
// READ (10)
|
|
// xxx also don't bother checking the flags
|
|
// @ 16: flags lba3
|
|
// @ 18: lba2 lba1
|
|
// @ 20: lba0 group
|
|
// @ 22: len1 len0
|
|
uint32_t lba = (USB_EP1_OUT(16) << 16) & 0xff000000;
|
|
uint32_t tmp = USB_EP1_OUT(18);
|
|
lba |= (tmp & 0xff) << 24;
|
|
lba |= (tmp & 0xff00);
|
|
tmp = USB_EP1_OUT(20);
|
|
lba |= (tmp & 0xff);
|
|
tmp = USB_EP1_OUT(22);
|
|
uint32_t blocks = (tmp >> 8) | ((tmp & 0xff) << 8);
|
|
|
|
if (blocks > 0x4000 || lba >= 0x4000 || (blocks + lba) > 0x4000) {
|
|
msc_state = STATE_SENT_CSW | (5 << 20) | (0x24 << 24);
|
|
goto error_csw;
|
|
}
|
|
|
|
if (blocks == 0) {
|
|
USB_EP1_IN(0) = 0x5355;
|
|
USB_EP1_IN(2) = 0x5342;
|
|
USB_EP1_IN(4) = dCSWTag;
|
|
USB_EP1_IN(6) = dCSWTag >> 16;
|
|
USB_EP1_IN(8) = 0;
|
|
USB_EP1_IN(10) = 0;
|
|
USB_EP1_IN(12) = 0;
|
|
USB_DESCS[1].count_tx = 13;
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
msc_state = STATE_SENT_CSW;
|
|
}
|
|
|
|
synthesize_block(lba, 0);
|
|
scsi_xfer_lba_blocks = (lba << 16) | blocks;
|
|
msc_state = STATE_SEND_MORE_READ;
|
|
break;
|
|
}
|
|
default:
|
|
// illegal command
|
|
msc_state = STATE_SENT_CSW | (5 << 20);
|
|
error_csw:
|
|
USB_EP1_IN(0) = 0x5355;
|
|
USB_EP1_IN(2) = 0x5342;
|
|
USB_EP1_IN(4) = dCSWTag;
|
|
USB_EP1_IN(6) = dCSWTag >> 16;
|
|
USB_EP1_IN(8) = dCBWDataTransferLength;
|
|
USB_EP1_IN(10) = dCBWDataTransferLength >> 16;
|
|
USB_EP1_IN(12) = 1;
|
|
USB_DESCS[1].count_tx = 13;
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
break;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
} else if ((ep_status & (1 << 7)) && (epidx == 1)) {
|
|
// ep 1 in
|
|
switch (msc_state & 0xff) {
|
|
case STATE_SENT_CSW:
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_ACK, USB_STAT_STALL, 0, 0);
|
|
msc_state = (msc_state & 0xffffff00) | STATE_WANT_CBW;
|
|
break;
|
|
case STATE_SENT_DATA_IN:
|
|
USB_EP1_IN(0) = 0x5355;
|
|
USB_EP1_IN(2) = 0x5342;
|
|
USB_EP1_IN(4) = dCSWTag;
|
|
USB_EP1_IN(6) = dCSWTag >> 16;
|
|
USB_EP1_IN(8) = 0;
|
|
USB_EP1_IN(10) = 0;
|
|
USB_EP1_IN(12) = 0;
|
|
USB_DESCS[1].count_tx = 13;
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
msc_state = STATE_SENT_CSW;
|
|
break;
|
|
case STATE_SEND_MORE_READ:
|
|
uint32_t piece = (msc_state >> 8) & 0b111;
|
|
if (piece != 7) {
|
|
synthesize_block(scsi_xfer_lba_blocks >> 16, piece + 1);
|
|
msc_state += 0x100;
|
|
} else {
|
|
uint32_t lba = scsi_xfer_lba_blocks >> 16;
|
|
uint32_t blocks_left = scsi_xfer_lba_blocks & 0xffff;
|
|
blocks_left--;
|
|
if (blocks_left == 0) {
|
|
USB_EP1_IN(0) = 0x5355;
|
|
USB_EP1_IN(2) = 0x5342;
|
|
USB_EP1_IN(4) = dCSWTag;
|
|
USB_EP1_IN(6) = dCSWTag >> 16;
|
|
USB_EP1_IN(8) = 0;
|
|
USB_EP1_IN(10) = 0;
|
|
USB_EP1_IN(12) = 0;
|
|
USB_DESCS[1].count_tx = 13;
|
|
set_ep_mode(1, 1, USB_EPTYPE_BULK, USB_STAT_STALL, USB_STAT_ACK, 0, 0);
|
|
msc_state = STATE_SENT_CSW;
|
|
} else {
|
|
lba++;
|
|
synthesize_block(lba, 0);
|
|
scsi_xfer_lba_blocks = (lba << 16) | blocks_left;
|
|
msc_state = STATE_SEND_MORE_READ;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
R16_USBD_ISTR = 0;
|
|
}
|
|
}
|