feat(websocket): Add websocket main worker

This patch adds support for websocket servers, the websocket code is
copied from corsacOTA.

components/dap_proxy/tcp_server.c

@@ -28,7 +28,7 @@ int kSock = -1;
 
 void tcp_server_task(void *pvParameters)
 {
-    uint8_t tcp_rx_buffer[1500];
+    uint8_t tcp_rx_buffer[1500] = {0};
     char addr_str[128];
     enum usbip_server_state_t usbip_state = WAIT_DEVLIST;
     uint8_t *data;
@@ -41,15 +41,7 @@ void tcp_server_task(void *pvParameters)
     while (1)
     {
 
-#ifdef CONFIG_EXAMPLE_IPV6
+#ifdef CONFIG_EXAMPLE_IPV4
-	    struct sockaddr_in6 destAddr;
-        bzero(&destAddr.sin6_addr.un, sizeof(destAddr.sin6_addr.un));
-        destAddr.sin6_family = AF_INET6;
-        destAddr.sin6_port = htons(DAP_PROXY_PORT);
-        addr_family = AF_INET6;
-        ip_protocol = IPPROTO_IPV6;
-        inet6_ntoa_r(destAddr.sin6_addr, addr_str, sizeof(addr_str) - 1);
-#else // IPV6
         struct sockaddr_in destAddr;
         destAddr.sin_addr.s_addr = htonl(INADDR_ANY);
         destAddr.sin_family = AF_INET;
@@ -57,6 +49,14 @@ void tcp_server_task(void *pvParameters)
         addr_family = AF_INET;
         ip_protocol = IPPROTO_IP;
         inet_ntoa_r(destAddr.sin_addr, addr_str, sizeof(addr_str) - 1);
+#else // IPV6
+        struct sockaddr_in6 destAddr;
+        bzero(&destAddr.sin6_addr.un, sizeof(destAddr.sin6_addr.un));
+        destAddr.sin6_family = AF_INET6;
+        destAddr.sin6_port = htons(DAP_PROXY_PORT);
+        addr_family = AF_INET6;
+        ip_protocol = IPPROTO_IPV6;
+        inet6_ntoa_r(destAddr.sin6_addr, addr_str, sizeof(addr_str) - 1);
 #endif
 
         int listen_sock = socket(addr_family, SOCK_STREAM, ip_protocol);
@@ -127,6 +127,10 @@ void tcp_server_task(void *pvParameters)
                 else
                     usbip_state = WAIT_IMPORT;
                 usbip_worker(tcp_rx_buffer, sizeof(tcp_rx_buffer), &usbip_state);
+            } else if (header == 0x47455420) { // string "GET "
+#ifdef CONFIG_USE_WEBSOCKET_DAP
+                websocket_worker(kSock, tcp_rx_buffer, sizeof(tcp_rx_buffer));
+#endif
             } else {
                 printf("Unknown protocol\n");
             }

components/dap_proxy/websocket_server.c

@@ -867,24 +867,37 @@ static esp_err_t co_websocket_handshake_process(co_cb_t *cb, co_socket_cb_t *scb
     return ESP_OK;
 }
 
-static esp_err_t co_socket_data_process(co_cb_t *cb, co_socket_cb_t *scb) {
+int websocket_worker(int fd, uint8_t *base, uint32_t length) {
-    if (cb == NULL || scb == NULL) {
+    co_cb_t cb;
-        return ESP_ERR_INVALID_ARG;
+    co_socket_cb_t scb;
-    }
+    esp_err_t ret;
 
-    switch (scb->status) {
+    memset(&cb, 0, sizeof(co_cb_t));
-    case CO_SOCKET_ACCEPT:
+    memset(&scb, 0, sizeof(co_socket_cb_t));
-        ESP_LOGW(CO_TAG, LOG_FMT("This state should not occur"));
+
-        return ESP_FAIL; //// TODO: remove this?
+    cb.recv_data = NULL; // used in websocket text mode
-    case CO_SOCKET_HANDSHAKE:
+    cb.websocket = &scb;
-        return co_websocket_handshake_process(cb, scb);
+
-    case CO_SOCKET_WEBSOCKET_HEADER:
+    scb.fd = fd;
-    case CO_SOCKET_WEBSOCKET_EXTEND_LENGTH:
+    scb.status = CO_SOCKET_HANDSHAKE;
-    case CO_SOCKET_WEBSOCKET_MASK:
+    scb.buf = (char *)base;
-    case CO_SOCKET_WEBSOCKET_PAYLOAD:
+    scb.remaining_len = 4; // already read 4 byte
-        return co_websocket_process(cb, scb);
+
-    default:
+    global_cb = &cb;
-        ESP_LOGW(CO_TAG, LOG_FMT("This state should not occur"));
+
-        return ESP_OK;
+    // handshake
-    }
+    do {
+        ret = co_websocket_handshake_process(&cb, &scb);
+        if (ret != ESP_OK)
+            return ret;
+    } while (scb.status == CO_SOCKET_HANDSHAKE);
+
+    // websocket data process
+    do {
+        ret = co_websocket_process(&cb, &scb);
+        if (ret != ESP_OK)
+            return ret;
+    } while (1);
+
+    return 0;
 }

components/dap_proxy/websocket_server.h

@@ -0,0 +1,8 @@
+#ifndef __WEBSOCKET_SERVER_H__
+#define __WEBSOCKET_SERVER_H__
+
+#include <stdint.h>
+
+int websocket_worker(int fd, uint8_t *base, uint32_t length);
+
+#endif