kerms
/
wireless-proxy-esp32
0
0
Fork 0
Code
wireless-proxy-esp32/components/uart_tcp_bridge/uart_tcp_bridge.c

326 lines
11 KiB
C
Raw Permalink Blame History

/*
Copyright (c) 2015, SuperHouse Automation Pty Ltd
All rights reserved.
Redistribution and use in source and binary forms, with or without modification, are permitted provided that the following conditions are met:
1. Redistributions of source code must retain the above copyright notice, this list of conditions and the following disclaimer.
2. Redistributions in binary form must reproduce the above copyright notice, this list of conditions and the following disclaimer in the documentation and/or other materials provided with the distribution.
3. Neither the name of the copyright holder nor the names of its contributors may be used to endorse or promote products derived from this software without specific prior written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
/**
* @file uart_bridge.c
* @author windowsair
* @brief UART TCP bridge
* @version 0.1
* @date 2021-11-16
*
* @copyright Copyright (c) 2021
*
*/
#include <string.h>
#include <stdint.h>
#include <sys/param.h>
#include "freertos/FreeRTOS.h"
#include "freertos/queue.h"
#include "esp_system.h"
#include "esp_log.h"
#include "driver/uart.h"
#include "lwip/err.h"
#include "lwip/netbuf.h"
#include "lwip/api.h"
#include "lwip/tcp.h"
#include "uart_tcp_bridge.h"
#if defined CONFIG_IDF_TARGET_ESP32S3
#define UART_PORT UART_NUM_1
#define UART_HW (&UART1)
#define PORT_RX_PIN 18
#define PORT_TX_PIN 17
#elif defined CONFIG_IDF_TARGET_ESP32C3
#define UART_PORT UART_NUM_0
#define UART_HW (&UART0)
#define PORT_RX_PIN 20
#define PORT_TX_PIN 21
#elif defined CONFIG_IDF_TARGET_ESP32
#define UART_PORT UART_NUM_2
#define UART_HW (&UART2)
#define PORT_RX_PIN 16
#define PORT_TX_PIN 17
#else
#error "Unknown target"
#endif
#ifdef CONFIG_IDF_TARGET_ESP8266
#define UART_BRIDGE_TX UART_NUM_0
#define UART_BRIDGE_RX UART_NUM_1
#elif defined CONFIG_IDF_TARGET_ESP32
#define UART_BRIDGE_TX UART_NUM_2
#define UART_BRIDGE_RX UART_NUM_2
#define UART_BRIDGE_TX_PIN 23
#define UART_BRIDGE_RX_PIN 22
#elif defined CONFIG_IDF_TARGET_ESP32C3
#define UART_BRIDGE_TX UART_NUM_1
#define UART_BRIDGE_RX UART_NUM_1
#define UART_BRIDGE_TX_PIN 21
#define UART_BRIDGE_RX_PIN 20
#elif defined CONFIG_IDF_TARGET_ESP32S3
#define UART_BRIDGE_TX UART_NUM_1
#define UART_BRIDGE_RX UART_NUM_1
#define UART_BRIDGE_TX_PIN 17
#define UART_BRIDGE_RX_PIN 18
#else
#error unknown hardware
#endif
#define EVENTS_QUEUE_SIZE 10
#define UART_BUF_SIZE 512
#ifdef CALLBACK_DEBUG
#define debug(s, ...) printf("%s: " s "\n", "Cb:", ##__VA_ARGS__)
#else
#define debug(s, ...)
#endif
static const char *UART_TAG = "UART";
#define NETCONN_EVT_WIFI_DISCONNECTED (NETCONN_EVT_ERROR + 1)
static QueueHandle_t uart_server_events = NULL;
typedef struct
{
struct netconn *nc;
uint8_t type;
} netconn_events;
static uint8_t uart_read_buffer[UART_BUF_SIZE];
// use lwip buffer to write back
static struct netconn *uart_netconn = NULL;
static bool is_conn_valid = false; // lock free
static bool is_first_time_recv = false;
void uart_bridge_close() {
netconn_events events;
events.type = NETCONN_EVT_WIFI_DISCONNECTED;
xQueueSend(uart_server_events, &events, 1000);
}
static void uart_bridge_reset() {
uart_netconn = NULL;
is_conn_valid = false;
}
static int num_digits(int n) {
if (n < 10)
return 1;
if (n < 100)
return 2;
if (n < 1000)
return 3;
if (n < 10000)
return 4;
if (n < 100000)
return 5;
if (n < 1000000)
return 6;
return 7;
}
/*
* This function will be call in Lwip in each event on netconn
*/
static void netCallback(struct netconn *conn, enum netconn_evt evt, uint16_t length) {
// Show some callback information (debug)
debug("sock:%lu\tsta:%u\tevt:%u\tlen:%u\ttyp:%u\tfla:%02x\terr:%d",
(uint32_t)conn, conn->state, evt, length, conn->type, conn->flags, conn->pending_err);
netconn_events events;
// If netconn got error, it is close or deleted, dont do treatments on it.
if (conn->pending_err) {
return;
}
// Treatments only on rcv events.
switch (evt) {
case NETCONN_EVT_RCVPLUS:
events.nc = conn;
events.type = evt;
break;
default:
return;
break;
}
// Send the event to the queue
xQueueSend(uart_server_events, &events, pdMS_TO_TICKS(1000));
}
/*
* Initialize a server netconn and listen port
*/
static void set_tcp_server_netconn(struct netconn **nc, uint16_t port, netconn_callback callback) {
if (nc == NULL) {
ESP_LOGE(UART_TAG, "%s: netconn missing .\n", __FUNCTION__);
return;
}
*nc = netconn_new_with_callback(NETCONN_TCP, netCallback);
if (!*nc) {
ESP_LOGE(UART_TAG, "Status monitor: Failed to allocate netconn.\n");
return;
}
netconn_set_nonblocking(*nc, NETCONN_FLAG_NON_BLOCKING);
// netconn_set_recvtimeout(*nc, 10);
netconn_bind(*nc, IP_ADDR_ANY, port);
netconn_listen(*nc);
}
/*
* Close and delete a socket properly
*/
static void close_tcp_netconn(struct netconn *nc) {
nc->pending_err = ERR_CLSD; // It is hacky way to be sure than callback will don't do treatment on a netconn closed and deleted
netconn_close(nc);
netconn_delete(nc);
}
static void uart_bridge_setup() {
uart_config_t uart_config = {
.baud_rate = UART_BRIDGE_BAUDRATE,
.data_bits = UART_DATA_8_BITS,
.parity = UART_PARITY_DISABLE,
.stop_bits = UART_STOP_BITS_1,
.flow_ctrl = UART_HW_FLOWCTRL_DISABLE};
if (UART_BRIDGE_TX == UART_BRIDGE_RX) {
ESP_ERROR_CHECK(uart_param_config(UART_BRIDGE_RX, &uart_config));
ESP_ERROR_CHECK(uart_driver_install(UART_BRIDGE_RX, UART_BUF_SIZE, UART_BUF_SIZE, 0, NULL, 0));
} else {
uart_param_config(UART_BRIDGE_RX, &uart_config);
uart_param_config(UART_BRIDGE_TX, &uart_config);
uart_driver_install(UART_BRIDGE_RX, UART_BUF_SIZE, 0, 0, NULL, 0); // RX only
uart_driver_install(UART_BRIDGE_TX, 0, UART_BUF_SIZE, 0, NULL, 0); // TX only
}
#if defined CONFIG_IDF_TARGET_ESP32 || defined CONFIG_IDF_TARGET_ESP32C3 || defined CONFIG_IDF_TARGET_ESP32S3
ESP_ERROR_CHECK(uart_set_pin(UART_BRIDGE_TX, UART_BRIDGE_TX_PIN, UART_BRIDGE_RX_PIN, -1, -1));
#endif
}
void uart_bridge_init() {
uart_server_events = xQueueCreate(EVENTS_QUEUE_SIZE, sizeof(netconn_events));
}
void uart_bridge_task() {
uart_bridge_setup();
uart_server_events = xQueueCreate(EVENTS_QUEUE_SIZE, sizeof(netconn_events));
struct netconn *nc = NULL; // To create servers
set_tcp_server_netconn(&nc, UART_BRIDGE_PORT, netCallback);
struct netbuf *netbuf = NULL; // To store incoming Data
struct netconn *nc_in = NULL; // To accept incoming netconn
//
char *buffer;
uint16_t len_buf;
size_t uart_buf_len;
while (1) {
netconn_events events;
int ret = xQueueReceive(uart_server_events, &events, pdMS_TO_TICKS(100));
if (ret != pdTRUE) {
// timeout
if (is_conn_valid) {
ESP_ERROR_CHECK(uart_get_buffered_data_len(UART_BRIDGE_RX, &uart_buf_len));
uart_buf_len = uart_buf_len > UART_BUF_SIZE ? UART_BUF_SIZE : uart_buf_len;
uart_buf_len = uart_read_bytes(UART_BRIDGE_RX, uart_read_buffer, uart_buf_len, pdMS_TO_TICKS(5));
// then send data
netconn_write(uart_netconn, uart_read_buffer, uart_buf_len, NETCONN_COPY);
}
} else if (events.type == NETCONN_EVT_WIFI_DISCONNECTED) { // WIFI disconnected
if (is_conn_valid) {
close_tcp_netconn(uart_netconn);
uart_bridge_reset();
}
} else if (events.nc->state == NETCONN_LISTEN) {
if (is_conn_valid) {
netconn_accept(events.nc, &nc_in);
if (nc_in)
close_tcp_netconn(nc_in);
continue;
}
int err = netconn_accept(events.nc, &nc_in);
if (err != ERR_OK) {
if (nc_in)
netconn_delete(nc_in);
continue;
}
printf("uart bridge accepted\n");
ip_addr_t client_addr; // Address port
uint16_t client_port; // Client port
netconn_peer(nc_in, &client_addr, &client_port);
uart_netconn = nc_in;
is_conn_valid = true;
is_first_time_recv = true;
} else if (events.nc->state != NETCONN_LISTEN) {
// if (events.nc && events.nc->pcb.tcp)
// tcp_nagle_disable(events.nc->pcb.tcp);
uart_netconn = events.nc;
// read data from UART
ESP_ERROR_CHECK(uart_get_buffered_data_len(UART_BRIDGE_RX, &uart_buf_len));
uart_buf_len = uart_buf_len > UART_BUF_SIZE ? UART_BUF_SIZE : uart_buf_len;
uart_buf_len = uart_read_bytes(UART_BRIDGE_RX, uart_read_buffer, uart_buf_len, pdMS_TO_TICKS(5));
// then send data
netconn_write(events.nc, uart_read_buffer, uart_buf_len, NETCONN_COPY);
// try to get data
if ((netconn_recv(events.nc, &netbuf)) == ERR_OK) // data incoming ?
{
do {
netbuf_data(netbuf, (void *)&buffer, &len_buf);
// write to uart
if (is_first_time_recv) { // change bard rate
if (len_buf > 1 && len_buf < 8) {
char tmp_buff[8];
memcpy(tmp_buff, buffer, len_buf);
tmp_buff[len_buf] = '\0';
int baudrate = atoi(tmp_buff);
if (baudrate > 0 && baudrate < 2000000 && num_digits(baudrate) == len_buf) {
ESP_LOGI(UART_TAG, "change baud:%d", baudrate);
uart_set_baudrate(UART_BRIDGE_RX, baudrate);
uart_set_baudrate(UART_BRIDGE_TX, baudrate);
is_first_time_recv = false;
continue;
}
}
is_first_time_recv = false;
}
uart_write_bytes(UART_BRIDGE_TX, (const char *)buffer, len_buf);
} while (netbuf_next(netbuf) >= 0);
netbuf_delete(netbuf);
} else {
if (events.nc->pending_err == ERR_CLSD) {
continue; // The same hacky way to treat a closed connection
}
close_tcp_netconn(events.nc);
uart_bridge_reset();
}
}
}
}
View Git Blame Copy Permalink