Files
checker_slave/source/task/tcp.c
2023-06-14 22:15:00 +08:00

153 lines
3.3 KiB
C

#include "lwip/opt.h"
#include "lwip/sys.h"
#include "lwip/api.h"
#include "debug.h"
#include "lwip/tcp.h"
#include "board.h"
#include "dev_flash.h"
#include "tcp.h"
typedef struct{
int inited;
int run;
struct netconn *conn;
struct netbuf *buf;
int if_connect;
void (*irq_fun)(void *t,uint8_t d);
void (*irq_fun_end)(void *t,uint32_t len);
void *ptr;
uint8_t *rx_buff;
uint32_t rx_buff_size;
}self_def;
void *tappend(void *p,void *del);
static void tcp_thread(void *arg)
{
self_def *s=arg;
err_t err;
LWIP_UNUSED_ARG(arg);
ip_addr_t dst;
const sys_param_def *par=sys_param();
IP4_ADDR(&dst,par->host_ip[0],par->host_ip[1],par->host_ip[2],par->host_ip[3]);
DBG_LOG("tcp_thread init.");
while (s->run) {
s->conn = netconn_new(NETCONN_TCP);
s->conn->pcb.tcp->so_options |= SOF_KEEPALIVE;
netconn_bind(s->conn, IP_ADDR_ANY, 1234);
err=netconn_connect(s->conn,&dst,par->host_port);
if (err == ERR_OK) {
void *data;
u16_t len;
s->if_connect=1;
emit tcp_connect_signal(s);
while ((err = netconn_recv(s->conn, &s->buf)) == ERR_OK) {
do {
netbuf_data(s->buf, &data, &len);
if(s->irq_fun){
for(int i=0;i<len;i++)
s->irq_fun(s->ptr,((uint8_t *)data)[i]);
}
if(s->irq_fun_end){
if(len>s->rx_buff_size) len=s->rx_buff_size;
memcpy(s->rx_buff,data,len);
s->irq_fun_end(s->ptr,len);
}
array_def *arr=arr_creat();
arr_appends(arr,data,len);
emit tcp_recv_signal(s,arr_temp(arr));
} while (netbuf_next(s->buf) >= 0);
netbuf_delete(s->buf);
}
}
else
{
//DBG_LOG("tcp:err=%d",err);
}
s->if_connect=0;
s->buf=0;
netconn_close(s->conn);
netconn_delete(s->conn);
sys_msleep(1000);
}
}
static self_def *mytcp_init(void)
{
self_def *s=calloc(1,sizeof(self_def));
s->run=1;
sys_thread_new("tcp_thread", tcp_thread, s, 2048, 22);
s->inited=1;
return s;
}
static int init(uart_def *u,int bsp){
self_def *s=mytcp_init();
u->private_data=s;
app_variable("tcp",s,0);
return 0;
}
static int deinit(uart_def *u){
return 0;
}
static int set_irq(uart_def *u,void (*irq)(void *t,uint8_t d),void *t){
param_check(u);
param_check(u->private_data);
self_def *self=u->private_data;
irq_disable();
self->irq_fun=irq;
self->ptr=t;
irq_enable();
return 0;
}
static int set_end_irq(uart_def *u,uint8_t *rx_buff,int rx_buff_size,
void (*irq)(void *t,uint32_t len),void *t)
{
param_check(u);
param_check(u->private_data);
self_def *self=u->private_data;
irq_disable();
self->irq_fun_end=irq;
self->ptr=t;
self->rx_buff=rx_buff;
self->rx_buff_size=rx_buff_size;
irq_enable();
return 0;
}
static int read(uart_def *u,uint8_t *d,int len){
return 0;
}
static int write(uart_def *u,const uint8_t *b,int len){
param_check(u);
param_check(u->private_data);
self_def *s=u->private_data;
err_t err;
DBG_LOG("tcp:write");
if(s->if_connect){
err = netconn_write(s->conn, b, len, NETCONN_COPY);
}
else err=ERR_CONN;
if(err==ERR_OK)
return len;
else
return 0;
}
uart_init_export(utcp,init,deinit,set_irq,set_end_irq,read,write,0)