#include "usb_common.h"
#include "usb_private.h"
static const uint8_t PROGMEM endpoint_config_table[] = {
EP_TYPE_INTERRUPT_IN, EP_SIZE(KEYBOARD_SIZE) | KEYBOARD_BUFFER,
EP_TYPE_INTERRUPT_IN, EP_SIZE(CDC_ACM_SIZE) | CDC_ACM_BUFFER,
EP_TYPE_BULK_OUT, EP_SIZE(CDC_RX_SIZE) | CDC_RX_BUFFER,
EP_TYPE_BULK_IN, EP_SIZE(CDC_TX_SIZE) | CDC_TX_BUFFER,
EP_TYPE_INTERRUPT_IN, EP_SIZE(MOUSE_SIZE) | MOUSE_BUFFER,
EP_TYPE_INTERRUPT_IN, EP_SIZE(JOYSTICK_SIZE) | JOYSTICK_BUFFER,
};
static const uint8_t PROGMEM device_descriptor[] = {
18, 1, 0x00, 0x02, 0xEF, 0x02, 0x01, ENDPOINT0_SIZE, LSB(VENDOR_ID), MSB(VENDOR_ID), LSB(PRODUCT_ID), MSB(PRODUCT_ID), 0x02, 0x01, 1, 2, 3, 1 };
static const uint8_t PROGMEM keyboard_hid_report_desc[] = {
0x05, 0x01, 0x09, 0x06, 0xA1, 0x01, 0x75, 0x01, 0x95, 0x08, 0x05, 0x07, 0x19, 0xE0, 0x29, 0xE7, 0x15, 0x00, 0x25, 0x01, 0x81, 0x02, 0x95, 0x08, 0x75, 0x01, 0x15, 0x00, 0x25, 0x01, 0x05, 0x0C, 0x09, 0xE9, 0x09, 0xEA, 0x09, 0xE2, 0x09, 0xCD, 0x09, 0xB5, 0x09, 0xB6, 0x09, 0xB7, 0x09, 0xB8, 0x81, 0x02, 0x95, 0x05, 0x75, 0x01, 0x05, 0x08, 0x19, 0x01, 0x29, 0x05, 0x91, 0x02, 0x95, 0x01, 0x75, 0x03, 0x91, 0x03, 0x95, 0x06, 0x75, 0x08, 0x15, 0x00, 0x25, 0x7F, 0x05, 0x07, 0x19, 0x00, 0x29, 0x7F, 0x81, 0x00, 0xc0 };
static const uint8_t PROGMEM mouse_hid_report_desc[] = {
0x05, 0x01, 0x09, 0x02, 0xA1, 0x01, 0x05, 0x09, 0x19, 0x01, 0x29, 0x08, 0x15, 0x00, 0x25, 0x01, 0x95, 0x08, 0x75, 0x01, 0x81, 0x02, 0x05, 0x01, 0x09, 0x30, 0x09, 0x31, 0x09, 0x38, 0x15, 0x81, 0x25, 0x7F, 0x75, 0x08, 0x95, 0x03, 0x81, 0x06, 0x05, 0x0C, 0x0A, 0x38, 0x02, 0x15, 0x81, 0x25, 0x7F, 0x75, 0x08, 0x95, 0x01, 0x81, 0x06, 0xC0 };
static const uint8_t PROGMEM joystick_hid_report_desc[] = {
0x05, 0x01, 0x09, 0x04, 0xA1, 0x01, 0x15, 0x00, 0x25, 0x01, 0x75, 0x01, 0x95, 0x20, 0x05, 0x09, 0x19, 0x01, 0x29, 0x20, 0x81, 0x02, 0x15, 0x00, 0x25, 0x07, 0x35, 0x00, 0x46, 0x3B, 0x01, 0x75, 0x04, 0x95, 0x01, 0x65, 0x14, 0x05, 0x01, 0x09, 0x39, 0x81, 0x42, 0x05, 0x01, 0x09, 0x01, 0xA1, 0x00, 0x15, 0x00, 0x26, 0xFF, 0x03, 0x75, 0x0A, 0x95, 0x04, 0x09, 0x30, 0x09, 0x31, 0x09, 0x32, 0x09, 0x35, 0x81, 0x02, 0xC0, 0x15, 0x00, 0x26, 0xFF, 0x03, 0x75, 0x0A, 0x95, 0x02, 0x09, 0x36, 0x09, 0x36, 0x81, 0x02, 0xC0 };
#define KEYBOARD_HID_DESC_OFFSET ( 9+8 + 9+5+5+4+5+7+9+7+7 + 9 )
#define MOUSE_HID_DESC_OFFSET ( 9+8 + 9+5+5+4+5+7+9+7+7 + 9+9+7 + 9 )
#define JOYSTICK_HID_DESC_OFFSET ( 9+8 + 9+5+5+4+5+7+9+7+7 + 9+9+7 + 9+9+7 + 9 )
#define CONFIG1_DESC_SIZE ( 9+8 + 9+5+5+4+5+7+9+7+7 + 9+9+7 + 9+9+7 + 9+9+7 )
static const uint8_t PROGMEM config1_descriptor[CONFIG1_DESC_SIZE] = {
9, 2, LSB(CONFIG1_DESC_SIZE), MSB(CONFIG1_DESC_SIZE),
5, 1, 0, 0xC0, 50, 8, 11, 0, 2, 0x02, 0x02, 0x01, 4,
9, 4, 0, 0, 1, 0x02, 0x02, 0x01, 0, 5, 0x24, 0x00, 0x10, 0x01, 5, 0x24, 0x01, 0x00, 1, 4, 0x24, 0x02, 0x06, 5, 0x24, 0x06, 0, 1, 7, 5, CDC_ACM_ENDPOINT | 0x80, 0x03, CDC_ACM_SIZE, 0, 64, 9, 4, 1, 0, 2, 0x0A, 0x00, 0x00, 0, 7, 5, CDC_RX_ENDPOINT, 0x02, CDC_RX_SIZE, 0, 0, 7, 5, CDC_TX_ENDPOINT | 0x80, 0x02, CDC_TX_SIZE, 0, 0,
9, 4, KEYBOARD_INTERFACE, 0, 1, 0x03, 0x01, 0x01, 0, 9, 0x21, 0x11, 0x01, 0, 1, 0x22, sizeof(keyboard_hid_report_desc), 0,
7, 5, KEYBOARD_ENDPOINT | 0x80, 0x03, KEYBOARD_SIZE, 0, KEYBOARD_INTERVAL,
9, 4, MOUSE_INTERFACE, 0, 1, 0x03, 0x01, 0x02, 0, 9, 0x21, 0x11, 0x01, 0, 1, 0x22, sizeof(mouse_hid_report_desc), 0,
7, 5, MOUSE_ENDPOINT | 0x80, 0x03, MOUSE_SIZE, 0, MOUSE_INTERVAL,
9, 4, JOYSTICK_INTERFACE, 0, 1, 0x03, 0x00, 0x00, 0, 9, 0x21, 0x11, 0x01, 0, 1, 0x22, sizeof(joystick_hid_report_desc), 0,
7, 5, JOYSTICK_ENDPOINT | 0x80, 0x03, 12, 0, JOYSTICK_INTERVAL };
struct usb_string_descriptor_struct {
uint8_t bLength;
uint8_t bDescriptorType;
int16_t wString[];
};
static const struct usb_string_descriptor_struct PROGMEM string0 = {
4,
3,
{0x0409}
};
static const struct usb_string_descriptor_struct PROGMEM string1 = {
sizeof(STR_MANUFACTURER),
3,
STR_MANUFACTURER
};
static const struct usb_string_descriptor_struct PROGMEM string2 = {
sizeof(STR_PRODUCT),
3,
STR_PRODUCT
};
static const struct usb_string_descriptor_struct PROGMEM string3 = {
sizeof(STR_SERIAL_NUMBER),
3,
STR_SERIAL_NUMBER
};
static const struct usb_string_descriptor_struct PROGMEM string4 = {
sizeof(STR_SERIAL),
3,
STR_SERIAL
};
static const struct descriptor_list_struct {
uint16_t wValue;
uint16_t wIndex;
const uint8_t *addr;
uint8_t length;
} PROGMEM descriptor_list[] = {
{0x0100, 0x0000, device_descriptor, sizeof(device_descriptor)},
{0x0200, 0x0000, config1_descriptor, sizeof(config1_descriptor)},
{0x2200, KEYBOARD_INTERFACE, keyboard_hid_report_desc, sizeof(keyboard_hid_report_desc)},
{0x2100, KEYBOARD_INTERFACE, config1_descriptor+KEYBOARD_HID_DESC_OFFSET, 9},
{0x2200, MOUSE_INTERFACE, mouse_hid_report_desc, sizeof(mouse_hid_report_desc)},
{0x2100, MOUSE_INTERFACE, config1_descriptor+MOUSE_HID_DESC_OFFSET, 9},
{0x2200, JOYSTICK_INTERFACE, joystick_hid_report_desc, sizeof(joystick_hid_report_desc)},
{0x2100, JOYSTICK_INTERFACE, config1_descriptor+JOYSTICK_HID_DESC_OFFSET, 9},
{0x0300, 0x0000, (const uint8_t *)&string0, 4},
{0x0301, 0x0409, (const uint8_t *)&string1, sizeof(STR_MANUFACTURER)},
{0x0302, 0x0409, (const uint8_t *)&string2, sizeof(STR_PRODUCT)},
{0x0303, 0x0409, (const uint8_t *)&string3, sizeof(STR_SERIAL_NUMBER)},
{0x0304, 0x0409, (const uint8_t *)&string4, sizeof(STR_SERIAL)},
};
#define NUM_DESC_LIST (sizeof(descriptor_list)/sizeof(struct descriptor_list_struct))
volatile uint8_t usb_configuration USBSTATE;
volatile uint8_t usb_suspended USBSTATE;
volatile uint8_t transmit_flush_timer=0;
volatile uint8_t reboot_timer=0;
uint8_t transmit_previous_timeout=0;
volatile uint8_t cdc_line_coding[7]={0x00, 0xE1, 0x00, 0x00, 0x00, 0x00, 0x08};
volatile uint8_t cdc_line_rtsdtr USBSTATE;
uint8_t keyboard_report_data[8] USBSTATE;
static uint8_t keyboard_protocol USBSTATE;
static uint8_t keyboard_idle_config USBSTATE;
uint8_t keyboard_idle_count USBSTATE;
volatile uint8_t keyboard_leds USBSTATE;
uint8_t mouse_buttons USBSTATE;
static uint8_t mouse_protocol USBSTATE;
uint8_t joystick_report_data[12] USBSTATE;
void usb_init(void)
{
uint8_t u;
u = USBCON;
if ((u & (1<<USBE)) && !(u & (1<<FRZCLK))) return;
HW_CONFIG();
USB_FREEZE(); PLL_CONFIG(); while (!(PLLCSR & (1<<PLOCK))) ; USB_CONFIG(); UDCON = 0; usb_configuration = 0;
usb_suspended = 0;
cdc_line_rtsdtr = 0;
keyboard_report_data[0] = 0;
keyboard_report_data[1] = 0;
keyboard_report_data[2] = 0;
keyboard_report_data[3] = 0;
keyboard_report_data[4] = 0;
keyboard_report_data[5] = 0;
keyboard_report_data[6] = 0;
keyboard_report_data[7] = 0;
keyboard_protocol = 1;
keyboard_idle_config = 125;
keyboard_idle_count = 0;
keyboard_leds = 0;
mouse_buttons = 0;
mouse_protocol = 1;
joystick_report_data[0] = 0;
joystick_report_data[1] = 0;
joystick_report_data[2] = 0;
joystick_report_data[3] = 0;
joystick_report_data[4] = 0x0F;
joystick_report_data[5] = 0x20;
joystick_report_data[6] = 0x80;
joystick_report_data[7] = 0x00;
joystick_report_data[8] = 0x02;
joystick_report_data[9] = 0x08;
joystick_report_data[10] = 0x20;
joystick_report_data[11] = 0x80;
UDINT = 0;
UDIEN = (1<<EORSTE)|(1<<SOFE)|(1<<SUSPE);
}
void usb_shutdown(void)
{
UDIEN = 0; UDCON = 1; USBCON = 0; PLLCSR = 0; usb_configuration = 0;
usb_suspended = 1;
}
ISR(USB_GEN_vect)
{
uint8_t intbits, t, i;
static uint8_t div4=0;
intbits = UDINT;
UDINT = 0;
if (intbits & (1<<EORSTI)) {
UENUM = 0;
UECONX = 1;
UECFG0X = EP_TYPE_CONTROL;
UECFG1X = EP_SIZE(ENDPOINT0_SIZE) | EP_SINGLE_BUFFER;
UEIENX = (1<<RXSTPE);
usb_configuration = 0;
cdc_line_rtsdtr = 0;
}
if (intbits & (1<<SOFI)) {
if (usb_configuration) {
t = transmit_flush_timer;
if (t) {
transmit_flush_timer = --t;
if (!t) {
UENUM = CDC_TX_ENDPOINT;
UEINTX = 0x3A;
}
}
t = reboot_timer;
if (t) {
reboot_timer = --t;
if (!t) _reboot_Teensyduino_();
}
if (keyboard_idle_config && (++div4 & 3) == 0) {
UENUM = KEYBOARD_ENDPOINT;
if (UEINTX & (1<<RWAL)) {
keyboard_idle_count++;
if (keyboard_idle_count == keyboard_idle_config) {
keyboard_idle_count = 0;
for (i=0; i < 8; i++) {
UEDATX = keyboard_report_data[i];
}
UEINTX = 0x3A;
}
}
}
}
}
if (intbits & (1<<SUSPI)) {
UDIEN = (1<<WAKEUPE);
usb_configuration = 0;
usb_suspended = 1;
#if (F_CPU >= 8000000L)
USB_FREEZE(); PLLCSR = 0; #endif
}
if (usb_suspended && (intbits & (1<<WAKEUPI))) {
#if (F_CPU >= 8000000L)
PLL_CONFIG();
while (!(PLLCSR & (1<<PLOCK))) ;
USB_CONFIG();
#endif
UDIEN = (1<<EORSTE)|(1<<SOFE)|(1<<SUSPE);
usb_suspended = 0;
return;
}
}
static inline void usb_wait_in_ready(void)
{
while (!(UEINTX & (1<<TXINI))) ;
}
static inline void usb_send_in(void)
{
UEINTX = ~(1<<TXINI);
}
static inline void usb_wait_receive_out(void)
{
while (!(UEINTX & (1<<RXOUTI))) ;
}
static inline void usb_ack_out(void)
{
UEINTX = ~(1<<RXOUTI);
}
ISR(USB_COM_vect)
{
uint8_t intbits;
const uint8_t *list;
const uint8_t *cfg;
uint8_t i, n, len, en;
volatile uint8_t *p;
uint8_t bmRequestType;
uint8_t bRequest;
uint16_t wValue;
uint16_t wIndex;
uint16_t wLength;
uint16_t desc_val;
const uint8_t *desc_addr;
uint8_t desc_length;
UENUM = 0;
intbits = UEINTX;
if (intbits & (1<<RXSTPI)) {
bmRequestType = UEDATX;
bRequest = UEDATX;
read_word_lsbfirst(wValue, UEDATX);
read_word_lsbfirst(wIndex, UEDATX);
read_word_lsbfirst(wLength, UEDATX);
UEINTX = ~((1<<RXSTPI) | (1<<RXOUTI) | (1<<TXINI));
if (bRequest == GET_DESCRIPTOR) {
list = (const uint8_t *)descriptor_list;
for (i=0; ; i++) {
if (i >= NUM_DESC_LIST) {
UECONX = (1<<STALLRQ)|(1<<EPEN); return;
}
pgm_read_word_postinc(desc_val, list);
if (desc_val != wValue) {
list += sizeof(struct descriptor_list_struct)-2;
continue;
}
pgm_read_word_postinc(desc_val, list);
if (desc_val != wIndex) {
list += sizeof(struct descriptor_list_struct)-4;
continue;
}
pgm_read_word_postinc(desc_addr, list);
desc_length = pgm_read_byte(list);
break;
}
len = (wLength < 256) ? wLength : 255;
if (len > desc_length) len = desc_length;
list = desc_addr;
do {
do {
i = UEINTX;
} while (!(i & ((1<<TXINI)|(1<<RXOUTI))));
if (i & (1<<RXOUTI)) return; n = len < ENDPOINT0_SIZE ? len : ENDPOINT0_SIZE;
for (i = n; i; i--) {
pgm_read_byte_postinc(UEDATX, list);
}
len -= n;
usb_send_in();
} while (len || n == ENDPOINT0_SIZE);
return;
}
if (bRequest == SET_ADDRESS) {
usb_send_in();
usb_wait_in_ready();
UDADDR = wValue | (1<<ADDEN);
return;
}
if (bRequest == SET_CONFIGURATION && bmRequestType == 0) {
usb_configuration = wValue;
cdc_line_rtsdtr = 0;
transmit_flush_timer = 0;
usb_send_in();
cfg = endpoint_config_table;
for (i=1; i<7; i++) {
UENUM = i;
UECONX = 1;
pgm_read_byte_postinc(UECFG0X, cfg);
pgm_read_byte_postinc(UECFG1X, cfg);
}
UERST = 0x1E;
UERST = 0;
return;
}
if (bRequest == GET_CONFIGURATION && bmRequestType == 0x80) {
usb_wait_in_ready();
UEDATX = usb_configuration;
usb_send_in();
return;
}
if (bRequest == CDC_GET_LINE_CODING && bmRequestType == 0xA1) {
usb_wait_in_ready();
p = cdc_line_coding;
for (i=0; i<7; i++) {
UEDATX = *p++;
}
usb_send_in();
return;
}
if (bRequest == CDC_SET_LINE_CODING && bmRequestType == 0x21) {
usb_wait_receive_out();
p = cdc_line_coding;
for (i=0; i<7; i++) {
*p++ = UEDATX;
}
usb_ack_out();
usb_send_in();
if (*(long *)cdc_line_coding == 134L) reboot_timer = 15;
if (*(long *)cdc_line_coding == 150L) {
UENUM = CDC_TX_ENDPOINT;
while (UESTA0X & 0x03) {
UEINTX = 0xFF;
while (UEINTX & 0x04) ;
}
_restart_Teensyduino_();
}
return;
}
if (bRequest == CDC_SET_CONTROL_LINE_STATE && bmRequestType == 0x21) {
cdc_line_rtsdtr = wValue;
usb_wait_in_ready();
usb_send_in();
return;
}
if (bRequest == CDC_SEND_BREAK && bmRequestType == 0x21) {
usb_wait_in_ready();
usb_send_in();
return;
}
if (bRequest == GET_STATUS) {
usb_wait_in_ready();
i = 0;
if (bmRequestType == 0x82) {
UENUM = wIndex;
if (UECONX & (1<<STALLRQ)) i = 1;
UENUM = 0;
}
UEDATX = i;
UEDATX = 0;
usb_send_in();
return;
}
if ((bRequest == CLEAR_FEATURE || bRequest == SET_FEATURE)
&& bmRequestType == 0x02 && wValue == 0) {
i = wIndex & 0x7F;
if (i >= 1 && i <= MAX_ENDPOINT) {
usb_send_in();
UENUM = i;
if (bRequest == SET_FEATURE) {
UECONX = (1<<STALLRQ)|(1<<EPEN);
} else {
UECONX = (1<<STALLRQC)|(1<<RSTDT)|(1<<EPEN);
UERST = (1 << i);
UERST = 0;
}
return;
}
}
if (wIndex == KEYBOARD_INTERFACE) {
if (bmRequestType == 0xA1) {
if (bRequest == HID_GET_REPORT) {
usb_wait_in_ready();
for (i=0; i < 8; i++) {
UEDATX = keyboard_report_data[i];
}
usb_send_in();
return;
}
if (bRequest == HID_GET_IDLE) {
usb_wait_in_ready();
UEDATX = keyboard_idle_config;
usb_send_in();
return;
}
if (bRequest == HID_GET_PROTOCOL) {
usb_wait_in_ready();
UEDATX = keyboard_protocol;
usb_send_in();
return;
}
}
if (bmRequestType == 0x21) {
if (bRequest == HID_SET_REPORT) {
usb_wait_receive_out();
keyboard_leds = UEDATX;
usb_ack_out();
usb_send_in();
return;
}
if (bRequest == HID_SET_IDLE) {
keyboard_idle_config = (wValue >> 8);
keyboard_idle_count = 0;
usb_send_in();
return;
}
if (bRequest == HID_SET_PROTOCOL) {
keyboard_protocol = wValue;
usb_send_in();
return;
}
}
}
if (wIndex == MOUSE_INTERFACE) {
if (bmRequestType == 0xA1) {
if (bRequest == HID_GET_REPORT) {
usb_wait_in_ready();
UEDATX = mouse_buttons;
UEDATX = 0;
UEDATX = 0;
UEDATX = 0;
UEDATX = 0;
usb_send_in();
return;
}
if (bRequest == HID_GET_PROTOCOL) {
usb_wait_in_ready();
UEDATX = mouse_protocol;
usb_send_in();
return;
}
}
if (bmRequestType == 0x21) {
if (bRequest == HID_SET_PROTOCOL) {
mouse_protocol = wValue;
usb_send_in();
return;
}
}
}
if (wIndex == JOYSTICK_INTERFACE) {
if (bmRequestType == 0xA1) {
if (bRequest == HID_GET_REPORT) {
usb_wait_in_ready();
for (i=0; i<12; i++) {
UEDATX = joystick_report_data[i];
}
usb_send_in();
return;
}
}
}
}
UECONX = (1<<STALLRQ) | (1<<EPEN); }