kerms
/
wireless-proxy-esp32
0
0
Fork 0
Code

feat(websocket): Add websocket main worker 

This patch adds support for websocket servers, the websocket code is
copied from corsacOTA.
This commit is contained in:
windowsair 2024-04-21 13:19:50 +08:00 committed by kerms
parent 1f31710f97
commit d63df67c14
3 changed files with 65 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
components/dap_proxy/tcp_server.c
View File

@ -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
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
#ifdef CONFIG_EXAMPLE_IPV4
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");
}

51
components/dap_proxy/websocket_server.c
View File

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

8
components/dap_proxy/websocket_server.h Normal file
View File

@ -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