/** Event handler for the USB_ControlRequest event. This is used to catch and process control requests sent to
* the device from the USB host before passing along unhandled control requests to the library for processing
* internally.
*/
void EVENT_USB_Device_ControlRequest(void)
{
static unsigned char buffer[MAX_CONTROL_TRANSFER_SIZE];
/* Handle HID Class specific requests */
switch (USB_ControlRequest.bRequest)
{
case REQ_GetReport:
if (USB_ControlRequest.bmRequestType == (REQDIR_DEVICETOHOST | REQTYPE_CLASS | REQREC_INTERFACE))
{
uint8_t reportType = USB_ControlRequest.wValue >> 8;
uint8_t reportId = USB_ControlRequest.wValue & 0xff;
if(reportType == REPORT_TYPE_FEATURE)
{
switch(reportId)
{
default:
Serial_SendByte(BYTE_DEBUG);
Serial_SendByte(BYTE_LEN_1_BYTE);
Serial_SendByte(reportId);
break;
}
}
}
break;
case REQ_SetReport:
if (USB_ControlRequest.bmRequestType == (REQDIR_HOSTTODEVICE | REQTYPE_CLASS | REQREC_INTERFACE))
{
Endpoint_ClearSETUP();
Endpoint_Read_Control_Stream_LE(buffer, USB_ControlRequest.wLength);
Endpoint_ClearIN();
uint8_t reportType = USB_ControlRequest.wValue >> 8;
uint8_t reportId = USB_ControlRequest.wValue & 0xff;
if(reportType == REPORT_TYPE_FEATURE)
{
switch(reportId)
{
default:
Serial_SendByte(BYTE_DEBUG);
Serial_SendByte(sizeof(USB_ControlRequest) + (USB_ControlRequest.wLength & 0xFF));
Serial_SendData(&USB_ControlRequest, sizeof(USB_ControlRequest));
Serial_SendData(buffer, USB_ControlRequest.wLength);
break;
}
}
}
/** Sends the next HID report to the host, via the IN endpoint. */
void SendNextReport(void)
{
/* Select the IN Report Endpoint */
Endpoint_SelectEndpoint(IN_EPNUM);
if (ready && sendReport)
{
/* Wait until the host is ready to accept another packet */
while (!Endpoint_IsINReady()) {}
/* Write IN Report Data */
Endpoint_Write_Stream_LE(report, sizeof(report), NULL);
/* Finalize the stream transfer to send the last packet */
Endpoint_ClearIN();
sendReport = 0;
#ifdef REPORT_NB_INFO
nbReports++;
if(!nbReports) {
Serial_SendData(info, sizeof(info));
}
#endif
}
}
/** Event handler for the USB_ControlRequest event. This is used to catch and process control requests sent to
* the device from the USB host before passing along unhandled control requests to the library for processing
* internally.
*/
void EVENT_USB_Device_ControlRequest(void)
{
static unsigned char buffer[MAX_CONTROL_TRANSFER_SIZE];
unsigned char len = 0;
/* Handle HID Class specific requests */
switch (USB_ControlRequest.bRequest)
{
case REQ_GetReport:
if (USB_ControlRequest.bmRequestType == (REQDIR_DEVICETOHOST | REQTYPE_CLASS | REQREC_INTERFACE))
{
if(USB_ControlRequest.wValue == 0x0303)
{
memcpy_P(buffer, buf03, sizeof(buf03));
len = sizeof(buf03);
Endpoint_ClearSETUP();
Endpoint_Write_Control_Stream_LE(buffer, len);
Endpoint_ClearOUT();
ready = 1;
}
else if(USB_ControlRequest.wValue == 0x03f3)
{
memcpy_P(buffer, buff3, sizeof(buff3));
len = sizeof(buff3);
Endpoint_ClearSETUP();
Endpoint_Write_Control_Stream_LE(buffer, len);
Endpoint_ClearOUT();
}
else if(USB_ControlRequest.wValue == 0x03f1
|| USB_ControlRequest.wValue == 0x03f2)
{
spoofReply = 0;
send_spoof_header();
while(!spoofReply);
Endpoint_ClearSETUP();
Endpoint_Write_Control_Stream_LE(buf, spoofReplyLen);
Endpoint_ClearOUT();
}
}
break;
case REQ_SetReport:
if (USB_ControlRequest.bmRequestType == (REQDIR_HOSTTODEVICE | REQTYPE_CLASS | REQREC_INTERFACE))
{
Endpoint_ClearSETUP();
Endpoint_Read_Control_Stream_LE(buffer, USB_ControlRequest.wLength);
Endpoint_ClearIN();
if(USB_ControlRequest.wValue == 0x03f0)
{
send_spoof_header();
Serial_SendData(buffer, USB_ControlRequest.wLength);
}
}
break;
}
}
/** Reads the next OUT report from the host from the OUT endpoint, if one has been sent. */
void ReceiveNextReport(void)
{
static struct
{
struct
{
unsigned char type;
unsigned char length;
} header;
unsigned char buffer[EPSIZE];
} packet = { .header.type = BYTE_OUT_REPORT };
uint16_t length = 0;
/* Select the OUT Report Endpoint */
Endpoint_SelectEndpoint(OUT_EPNUM);
/* Check if OUT Endpoint contains a packet */
if (Endpoint_IsOUTReceived())
{
/* Check to see if the packet contains data */
if (Endpoint_IsReadWriteAllowed())
{
/* Read OUT Report Data */
uint8_t ErrorCode = Endpoint_Read_Stream_LE(packet.buffer, sizeof(packet.buffer), &length);
if(ErrorCode == ENDPOINT_RWSTREAM_NoError)
{
length = sizeof(packet.buffer);
}
}
/* Handshake the OUT Endpoint - clear endpoint and ready for next report */
Endpoint_ClearOUT();
if(length)
{
packet.header.length = length & 0xFF;
Serial_SendData(&packet, sizeof(packet.header) + packet.header.length);
}
}
}
/** Function to manage HID report generation and transmission to the host, when in report mode. */
void HID_Task(void)
{
/* Device must be connected and configured for the task to run */
if (USB_DeviceState != DEVICE_STATE_Configured)
return;
/* Send the next keypress report to the host */
SendNextReport();
/* Process the LED report sent from the host */
ReceiveNextReport();
}
static inline void send_spoof_header(void)
{
Serial_SendByte(BYTE_CONTROL_DATA);
if( USB_ControlRequest.bmRequestType & REQDIR_DEVICETOHOST )
{
Serial_SendByte(sizeof(USB_ControlRequest));
}
else
{
Serial_SendByte(sizeof(USB_ControlRequest) + (USB_ControlRequest.wLength & 0xFF));
}
Serial_SendData(&USB_ControlRequest, sizeof(USB_ControlRequest));
}
int main( void )
{
static uint8_t i = 0;
static uint8_t recvData = 0;
static int8_t state = ERROR;
System_Init();
while(1) {
state = Serial_RecvDataWTO(&recvData, 1, 500);
if(state == ERROR) {
printf("timeout ... %d\r\n", i);
i = (i == 255) ? 0 : i + 1;
}
else if(recvData == 0x0D) // if press enter
printf("\r\n");
else
Serial_SendData(&recvData, 1);
}
}
