From 4f3b6f4283e0240060c04a137175cacfb385dc16 Mon Sep 17 00:00:00 2001
From: Dien-Nhung Nguyen-Phu <kein@kienlab.com>
Date: Thu, 11 Jul 2024 21:05:17 +0700
Subject: usb: handle device request and route to interfaces

---
 src/usb/ctrl.c | 151 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 151 insertions(+)
 create mode 100644 src/usb/ctrl.c

(limited to 'src/usb/ctrl.c')

diff --git a/src/usb/ctrl.c b/src/usb/ctrl.c
new file mode 100644
index 0000000..30aa7dd
--- /dev/null
+++ b/src/usb/ctrl.c
@@ -0,0 +1,151 @@
+
+#include "CH58x_common.h"
+
+#include "utils.h"
+#include "debug.h"
+
+extern void (*if_handlers[])(USB_SETUP_REQ *request);
+extern void (*ep_handlers[])();
+
+void handle_devreq(USB_SETUP_REQ *request);
+
+static uint8_t address;
+
+/* EP0 + EP4(IN + OUT) */
+__attribute__((aligned(4))) uint8_t ep0buf[64 + 64 + 64];
+
+void usb_set_address(uint8_t ad)
+{
+	address = ad;
+}
+
+void ctrl_start_load_block(void *buf, uint16_t len)
+{
+	usb_start_load_block(ep0buf, buf, len, 1);
+}
+
+static void route_interfaces(USB_SETUP_REQ *request)
+{
+	_TRACE();
+
+	uint8_t ifn = request->wIndex & 0xff;
+	PRINT("wInterfaceNumber: 0x%02x\n", ifn);
+
+	if (if_handlers[ifn])
+		if_handlers[ifn](request);
+}
+
+static void ep_handler()
+{
+	_TRACE();
+
+	USB_SETUP_REQ *request = (USB_SETUP_REQ *)ep0buf;
+	uint8_t req = request->bRequest;
+
+	/* Each interface will have their own request handler */
+	uint8_t recip = request->bRequestType & USB_REQ_RECIP_MASK;
+	if (recip == USB_REQ_RECIP_INTERF) {
+		route_interfaces(request);
+		return;
+	}
+
+	uint8_t token = R8_USB_INT_ST & MASK_UIS_TOKEN;
+	switch(token) {
+
+	case UIS_TOKEN_SETUP:
+		print_setuppk(req);
+
+		if (recip == USB_REQ_RECIP_DEVICE) {
+			handle_devreq(request);
+		}
+		break;
+
+	case UIS_TOKEN_OUT:
+		if (req == USB_CLEAR_FEATURE) {
+			ctrl_ack();
+		}
+
+	case UIS_TOKEN_IN:
+		R8_USB_DEV_AD = address;
+		usb_load_next_chunk();
+		break;
+	
+	default:
+		break;
+	}
+}
+
+void ctrl_init()
+{
+	ep_cb_register(0, ep_handler);
+	dma_register(0, ep0buf);
+}
+
+static void route_enpoints()
+{
+	_TRACE();
+
+	/* Workaround to solve a setup packet with EP 0x02 for unknown reason.
+	This happens when return from the bootloader or after a sotfware reset. */
+	uint8_t token = R8_USB_INT_ST & MASK_UIS_TOKEN;
+	if (token == UIS_TOKEN_SETUP) {
+		ep_handlers[0]();
+	}
+
+	uint8_t ep_num = R8_USB_INT_ST & MASK_UIS_ENDP;
+
+	if (ep_num < 8 && ep_handlers[ep_num])
+		ep_handlers[ep_num]();
+}
+
+static void handle_busReset()
+{
+	_TRACE();
+	R8_USB_DEV_AD = 0;
+	R8_UEP0_CTRL = UEP_R_RES_ACK | UEP_T_RES_NAK;
+	R8_UEP1_CTRL = UEP_R_RES_ACK | UEP_T_RES_NAK;
+	R8_UEP2_CTRL = UEP_R_RES_ACK | UEP_T_RES_NAK;
+	R8_UEP3_CTRL = UEP_R_RES_ACK | UEP_T_RES_NAK;
+	R8_UEP4_CTRL = UEP_R_RES_ACK | UEP_T_RES_NAK;
+	R8_UEP5_CTRL = UEP_R_RES_ACK | UEP_T_RES_NAK;
+	R8_UEP6_CTRL = UEP_R_RES_ACK | UEP_T_RES_NAK;
+	R8_UEP7_CTRL = UEP_R_RES_ACK | UEP_T_RES_NAK;
+}
+
+static void handle_powerChange()
+{
+	_TRACE();
+	if (R8_USB_MIS_ST & RB_UMS_SUSPEND) {
+		;// suspend
+	}
+	else {
+		;// resume
+	}
+}
+
+__INTERRUPT
+__HIGH_CODE
+void USB_IRQHandler(void) {
+	uint8_t intflag = R8_USB_INT_FG;
+	clear_handshake_sent_flag();
+	PRINT("\nusb: new interrupt\n");
+	print_intflag_reg();
+
+	if (intflag & RB_UIF_TRANSFER) {
+		PRINT("usb: RX Length reg: %d\n", R8_USB_RX_LEN);
+		print_status_reg();
+
+		route_enpoints();
+	}
+	else if (intflag & RB_UIF_BUS_RST) {
+		handle_busReset();
+	}
+	else if (intflag & RB_UIF_SUSPEND) {
+		handle_powerChange();
+	}
+
+	if (handshake_sent() == 0) {
+		PRINT("WARN: This transaction is being IGNORED!\n");
+	}
+	R8_USB_INT_FG = intflag; // clear interrupt flags
+}
\ No newline at end of file
-- 
cgit v1.2.3