void ep_read(U8 ep_num)
{
U8 i, len;
usb_pcb_t *pcb = usb_pcb_get();
if( ep_num == 0 )
{
len = SI32_USB_A_read_ep0_count( SI32_USB_0 );
for (i=0; i<len; i++)
{
usb_buf_write( ep_num, SI32_USB_A_read_ep0_fifo_u8( SI32_USB_0 ) );
}
//if( len == 8 )
//
if( 0==SI32_USB_A_read_ep0_count( SI32_USB_0 ) )
{
//SI32_USB_A_set_data_end_ep0(SI32_USB_0);
}
}
else if( ep_num > 0 )
{
if( ep_dir_get( ep_num ) == DIR_IN )
return;
len = SI32_USBEP_A_read_data_count( usb_ep[ ep_num - 1 ] );
if( pcb->fifo[ ep_num ].len + len > MAX_BUF_SZ )
{
pcb->pending_data |= ( 1 << ep_num );
return;
}
for (i=0; i<len; i++)
{
usb_buf_write(ep_num, SI32_USBEP_A_read_fifo_u8( usb_ep[ ep_num - 1 ] ));
}
//if ( 0==SI32_USBEP_A_read_data_count( usb_ep[ ep_num - 1 ] ))
//{
// Clear overrun out overrun if it has occured
if ( SI32_USBEP_A_has_out_data_overrun_occurred( usb_ep[ ep_num - 1 ] ) )
SI32_USBEP_A_clear_out_data_overrun( usb_ep[ ep_num - 1 ] );
SI32_USBEP_A_clear_outpacket_ready( usb_ep[ ep_num - 1 ] );
pcb->pending_data &= ~(1<<ep_num);
//}
}
if (len > 0)
{
pcb->flags |= ( ep_num == 0 ) ? ( 1 << SETUP_DATA_AVAIL ) : ( 1 << RX_DATA_AVAIL );
}
//cdc_demo_putchar("a", NULL);
}
void cdc_req_handler(req_t *req)
{
U8 i;
usb_pcb_t *pcb = usb_pcb_get();
switch (req->req)
{
case GET_LINE_CODING:
if (req->type & (DEVICE_TO_HOST | TYPE_CLASS | RECIPIENT_INTF))
{
// send the line coding to the host
for (i=0; i<LINE_CODE_SZ; i++)
{
usb_buf_write(EP_CTRL, line_code[i]);
}
ep_write(EP_CTRL);
}
break;
case SET_LINE_CODING:
if (req->type & (HOST_TO_DEVICE | TYPE_CLASS | RECIPIENT_INTF))
{
// wait for the setup data to be sent to the control endpoint
while (pcb->fifo[EP_CTRL].len == 0)
{
// keep the nop for a place to set a breakpoint on and to make it obvious we're
// waiting for something.
asm("nop");
}
// clear the setup flag if needed
pcb->flags &= ~(1<<SETUP_DATA_AVAIL);
// send out a zero-length packet to ack to the host that we received
// the new line coding
ep_send_zlp(EP_CTRL);
// set the new line code. the first 8 bytes in the fifo are just
// for the setup packet so we want to write the next 7 bytes for the
// line code.
for (i=0; i<LINE_CODE_SZ; i++)
{
line_code[i] = usb_buf_read(EP_CTRL);
}
}
break;
case SET_CTRL_LINE_STATE:
if (req->type & (HOST_TO_DEVICE | TYPE_CLASS | RECIPIENT_INTF))
{
ep_send_zlp(EP_CTRL);
}
break;
default:
ep_set_stall(EP_CTRL);
break;
}
}
/*
* This is the putchar function that is used by avr-libc's printf. We need
* to hook this function into the stdout file stream using the FDEV_SETUP_STREAM
* macro in avr-libc. Once the stream is set up, we hook the stream to stdout
* and we can do printfs via USB.
*/
int freakusb_putchar(char c, FILE *unused)
{
usb_pcb_t *pcb = usb_pcb_get();
if (!(pcb->flags & (1 << ENUMERATED)))
return 0;
if (c == '\n') {
usb_buf_write(EP_1, '\n');
usb_buf_write(EP_1, '\r');
} else {
usb_buf_write(EP_1, (U8)c);
}
ep_write(EP_1);
return 0;
}
void rx()
{
U8 i, c, ep_num, len;
usb_pcb_t *pcb = usb_pcb_get();
// get the ep number of any endpoint with pending rx data
if ((ep_num = usb_buf_data_pending(DIR_OUT)) != 0xFF)
{
// get the length of data in the OUT buffer
len = pcb->fifo[ep_num].len;
// read out the data in the buffer and echo it back to the host.
for (i=0; i<len; i++)
{
c = usb_buf_read(ep_num);
switch (c)
{
case '\r':
// terminate the msg and reset the msg ptr. then send
// it to the handler for processing.
*msg_ptr = '\0';
printf_P(PSTR("\n\r"));
cmd_parse((char *)msg);
msg_ptr = msg;
break;
case '\b':
usb_buf_write(EP_1, c);
if (msg_ptr > msg)
{
msg_ptr--;
}
break;
default:
usb_buf_write(EP_1, c);
*msg_ptr++ = c;
break;
}
}
pcb->flags |= (1 << TX_DATA_AVAIL);
}
}
void ep_read(U8 ep_num)
{
U8 i, len;
usb_pcb_t *pcb = usb_pcb_get();
len = FIFO_BYTE_CNT;
for (i=0; i<len; i++)
{
usb_buf_write(ep_num, UEDATX);
}
if (len > 0)
{
pcb->flags |= (ep_num == 0) ? (1<<SETUP_DATA_AVAIL) : (1<<RX_DATA_AVAIL);
}
}
void dfu_req_handler(req_t *req)
{
U8 i;
usb_pcb_t *pcb = usb_pcb_get();
switch (req->req)
{
case DFU_DETACH:
if (req->type & (HOST_TO_DEVICE | TYPE_CLASS | RECIPIENT_INTF))
{
// wvalue is wTimeout
// wLength is zero
// data is none
dfu_status.bState = appDETACH;
dfu_status.bStatus = OK;
}
break;
case DFU_DNLOAD:
if (req->type & (HOST_TO_DEVICE | TYPE_CLASS | RECIPIENT_INTF))
{
// wvalue is wBlockNum
// wlength is Length
// data is firmware
if( dfu_status.bState == dfuIDLE )
{
if( req->len > 0 )
{
hw_state_indicator( HW_STATE_TRANSFER );
dfu_status.bState = dfuDNLOAD_SYNC;
}
else
{
dfu_status.bState = dfuERROR;
dfu_status.bStatus = errNOTDONE;
hw_state_indicator( HW_STATE_ERROR );
ep_send_zlp(EP_CTRL);
return;
}
}
i = req->val;
if( dfu_status.bState == dfuDNLOAD_IDLE )
{
if( req->len > 0 )
{
dfu_status.bState = dfuDNLOAD_SYNC;
}
else
{
if( flash_buffer_ptr > flash_buffer )
{
need_to_write = 1;
//flash_buffer_ptr = flash_buffer;
}
dfu_status.bState = dfuMANIFEST_SYNC;
ep_send_zlp(EP_CTRL);
return;
}
}
SI32_USB_A_clear_out_packet_ready_ep0(SI32_USB_0);
while(pcb->fifo[EP_CTRL].len < req->len)
{
//ep_read(EP_CTRL);
i = pcb->fifo[EP_CTRL].len;
}
// clear the setup flag if needed
pcb->flags &= ~(1<<SETUP_DATA_AVAIL);
// send out a zero-length packet to ack to the host that we received
// the new line coding
U8* byte_buf_ptr = ( U8* )flash_buffer_ptr;
U8 tmp_len = pcb->fifo[EP_CTRL].len;
for(i = 0; i < tmp_len; i++)
{
*byte_buf_ptr = usb_buf_read(EP_CTRL);
byte_buf_ptr++;
}
flash_buffer_ptr += i/4;
if( flash_buffer_ptr == flash_buffer + BLOCK_SIZE_U32 )
{
// Reset buffer pointer
//flash_buffer_ptr = flash_buffer;
need_to_write = 1;
}
if( flash_buffer_ptr > flash_buffer + BLOCK_SIZE_U32)
{
dfu_status.bState = dfuERROR;
hw_state_indicator( HW_STATE_ERROR );
}
ep_send_zlp(EP_CTRL);
}
break;
case DFU_UPLOAD:
if (req->type & (DEVICE_TO_HOST | TYPE_CLASS | RECIPIENT_INTF))
{
// wvalue is zero
// wlength is length
// data is firmware
// NOT SUPPORTED
ep_set_stall(EP_CTRL);
}
break;
case DFU_GETSTATUS:
if (req->type & (DEVICE_TO_HOST | TYPE_CLASS | RECIPIENT_INTF))
{
if( dfu_communication_started == 0 )
hw_state_indicator( HW_STATE_CONNECTED );
dfu_communication_started = 1;
// If we're still transmitting blocks
if( dfu_status.bState == dfuDNLOAD_SYNC )
{
if( need_to_write == 0 )
{
dfu_status.bState=dfuDNLOAD_IDLE;
dfu_status.bwPollTimeout0 = 0x00;
}
else
{
dfu_status.bState=dfuDNBUSY;
dfu_status.bwPollTimeout0 = 0x3F;
}
}
else if( dfu_status.bState == dfuDNBUSY )
{
if( need_to_write == 0)
dfu_status.bState=dfuDNLOAD_SYNC;
}
else if( dfu_status.bState == dfuMANIFEST_SYNC)
{
dfu_status.bState=dfuMANIFEST;
dfu_status.bwPollTimeout0 = 0xFF;
hw_state_indicator( HW_STATE_DONE );
}
else if( dfu_status.bState == dfuMANIFEST &&
need_to_write == 0)
{
// Finish erasing flash
while( flash_target < SI32_MCU_FLASH_SIZE)
{
if( 0 != hw_flash_erase( flash_target, 1 ) )
{
dfu_status.bState = dfuERROR;
dfu_status.bStatus = errERASE;
hw_state_indicator( HW_STATE_ERROR );
}
flash_target += BLOCK_SIZE_U8;
}
dfu_status.bState=dfuMANIFEST_WAIT_RESET;
}
for (i=0; i<STATUS_SZ; i++)
{
usb_buf_write(EP_CTRL, *((U8 *)&dfu_status + i));
}
ep_write(EP_CTRL);
if( dfu_status.bState == dfuMANIFEST_WAIT_RESET )
{
hw_wait_ms(200);
hw_boot_image( 1 );
}
if( need_to_write )
{
if( 0 != hw_flash_erase( flash_target, 1 ) )
{
dfu_status.bState = dfuERROR;
dfu_status.bStatus = errERASE;
hw_state_indicator( HW_STATE_ERROR );
}
if( 0 != hw_flash_write( flash_target, ( U32* )flash_buffer, flash_buffer_ptr - flash_buffer, 1 ) )
{
dfu_status.bState = dfuERROR;
dfu_status.bStatus = errVERIFY;
hw_state_indicator( HW_STATE_ERROR );
}
flash_buffer_ptr = flash_buffer;
flash_target += BLOCK_SIZE_U8;
need_to_write = 0;
if( dfu_status.bState != dfuMANIFEST )
dfu_status.bState=dfuDNLOAD_SYNC;
}
}
break;
case DFU_CLRSTATUS:
if (req->type & (HOST_TO_DEVICE | TYPE_CLASS | RECIPIENT_INTF))
{
// wvalue is zero
// wlength is 0
// data is none
if( dfu_status.bState == dfuERROR )
{
dfu_status.bStatus = OK;
dfu_status.bState = dfuIDLE;
hw_state_indicator( HW_STATE_ERROR_CLR );
}
}
break;
case DFU_GETSTATE:
if (req->type & (DEVICE_TO_HOST | TYPE_CLASS | RECIPIENT_INTF))
{
// wvalue is zero
// wlength is 1
// data is state
// Transition?: No State Transition
usb_buf_write( EP_CTRL, dfu_status.bState );
ep_write(EP_CTRL);
}
break;
case DFU_ABORT:
if (req->type & (HOST_TO_DEVICE | TYPE_CLASS | RECIPIENT_INTF))
{
// wvalue is zero
// wlength is 0
// data is none
if( dfu_status.bState == dfuIDLE )
{
dfu_status.bStatus = OK;
dfu_status.bState = dfuIDLE;
}
else if ( dfu_status.bState == dfuDNLOAD_IDLE )
{
flash_target = FLASH_TARGET;
if( 0 != hw_flash_erase( flash_target, 1 ) )
{
dfu_status.bState = dfuERROR;
dfu_status.bStatus = errERASE;
hw_state_indicator( HW_STATE_ERROR );
}
dfu_status.bStatus = OK;
dfu_status.bState = dfuIDLE;
ep_send_zlp(EP_CTRL);
}
}
break;
default:
ep_set_stall(EP_CTRL);
break;
}
}
