void ep_send_zlp(U8 ep_num)
{
if( ep_num == 0 )
{
SI32_USB_A_set_data_end_ep0( SI32_USB_0 );
SI32_USB_A_clear_out_packet_ready_ep0( SI32_USB_0 );
}
}
// Finish handling the setup packet and move on to the IN/OUT data phase
// of the control request
void Finish_Setup(void)
{
// Copy selected bytes from the Setup packet
for (int i = 0; i < DFU_SETUP_PACKET_COPY_SIZE; i++)
{
RxBuffer[i] = ((uint8_t*)&Setup)[i + DFU_SETUP_PACKET_COPY_OFFSET];
}
// Set Serviced Out packet ready
SI32_USB_A_clear_out_packet_ready_ep0(SI32_USB_0);
// Handle out class requests
if ((Setup.bmRequestType & DIR_BITMASK) == DIR_OUT)
{
// No data phase, end the control transfer
if (Setup.wLength == 0)
{
// Set Serviced Out packet ready and
// data end to indicate transaction
// is over
SI32_USB_A_set_data_end_ep0(SI32_USB_0);
// Return to IDLE state
EP_Status = EP_IDLE;
// Control transfer complete
USB0_RX_Complete(DFU_SETUP_PACKET_COPY_SIZE);
}
else
{
// Set the DataPtr to store data in the ClassOutBuffer (after the Setup packet data)
ReadDataPtr = RxBuffer + DFU_SETUP_PACKET_COPY_SIZE;
ReadDataSize = RxBufferSize - DFU_SETUP_PACKET_COPY_SIZE;
ReadDataReceived = 0;
// Setup the RX phase of the control transfer
EP_Status = EP_RX;
}
}
// Handle in class requests
else
{
// Wait until TX data is ready from the App
EP_Status = EP_WAIT_TX;
// Control transfer setup complete
USB0_RX_Complete(DFU_SETUP_PACKET_COPY_SIZE);
}
}
// Pass in the TX buffer used to send the IN data phase
void USB0_TX_Start (uint8_t* buffer, uint16_t size)
{
__disable_irq();
WriteDataPtr = buffer;
WriteDataSize = size;
WriteDataSent = 0; // Reset Data Sent counter
TxComplete = false;
// Set Serviced Out packet ready
SI32_USB_A_clear_out_packet_ready_ep0(SI32_USB_0);
// Setup the TX phase of the control transfer
EP_Status = EP_TX;
__enable_irq();
}
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;
}
}
