432 lines
10 KiB
C
432 lines
10 KiB
C
|
#include "sys_config.h"
|
||
|
|
#include "tx_platform.h"
|
||
|
|
#include <csi_kernel.h>
|
||
|
|
#include "socket_module.h"
|
||
|
|
#include "lib/net/eloop/eloop.h"
|
||
|
|
#include "typesdef.h"
|
||
|
|
#include "lwip/api.h"
|
||
|
|
#include "lwip/sockets.h"
|
||
|
|
#include "event.h"
|
||
|
|
|
||
|
|
|
||
|
|
#define UDP_RCV_SIZE 1460
|
||
|
|
#define UDP_SND_SIZE UDP_RCV_SIZE
|
||
|
|
|
||
|
|
//udp定时发送结构
|
||
|
|
struct UdpSpeed
|
||
|
|
{
|
||
|
|
struct sockaddr_in remote_addr;
|
||
|
|
struct event *timer_e;
|
||
|
|
struct event *udp_e;
|
||
|
|
int fd;
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
static void udp_speed_test(int fd)
|
||
|
|
{
|
||
|
|
int len;
|
||
|
|
unsigned char *udp_data;
|
||
|
|
struct sockaddr remote_addr;
|
||
|
|
int retval;
|
||
|
|
int ret;
|
||
|
|
uint64 time_if;
|
||
|
|
int len_total = 0;
|
||
|
|
uint8_t *pt;
|
||
|
|
uint8_t type_eloop = 0;
|
||
|
|
|
||
|
|
retval = 16;
|
||
|
|
|
||
|
|
udp_data = (unsigned char*)malloc(UDP_RCV_SIZE );
|
||
|
|
time_if = os_jiffies();
|
||
|
|
_os_printf("udp_test_bandwidth start...\n");
|
||
|
|
while(1){
|
||
|
|
//len = sendto(send_socket_fd, (unsigned char *)udp_data, UDP_XMIT_SIZE, 0, &udp_remote_client, sizeof(struct sockaddr));
|
||
|
|
type_eloop = 0;
|
||
|
|
_os_printf("test recv\r\n");
|
||
|
|
while(type_eloop < 10){
|
||
|
|
len = recvfrom(fd, udp_data, UDP_RCV_SIZE, MSG_WAITALL, &remote_addr, (socklen_t*)&retval);
|
||
|
|
len_total += len;
|
||
|
|
if((os_jiffies() - time_if) > 2000){
|
||
|
|
_os_printf("recvfrom loop:%d\r\n",type_eloop);
|
||
|
|
type_eloop++;
|
||
|
|
time_if = os_jiffies();
|
||
|
|
pt = (uint8_t*)&len_total;
|
||
|
|
udp_data[0] = pt[0]&0x0f;
|
||
|
|
udp_data[1] = (pt[0]>>4)&0x0f;
|
||
|
|
udp_data[2] = pt[1]&0x0f;
|
||
|
|
udp_data[3] = (pt[1]>>4)&0x0f;
|
||
|
|
udp_data[4] = pt[2]&0x0f;
|
||
|
|
udp_data[5] = (pt[2]>>4)&0x0f;
|
||
|
|
udp_data[6] = pt[3]&0x0f;
|
||
|
|
udp_data[7] = (pt[3]>>4)&0x0f;
|
||
|
|
udp_data[8] = 100;
|
||
|
|
_os_printf("len_total::%x %x%x%x%x\r\n",len_total,udp_data[0],udp_data[1],udp_data[2],udp_data[3]);
|
||
|
|
do{
|
||
|
|
ret = sendto(fd, (unsigned char *)udp_data, 12, 0, &remote_addr, sizeof(struct sockaddr));
|
||
|
|
}while(ret<0);
|
||
|
|
|
||
|
|
do{
|
||
|
|
ret = sendto(fd, (unsigned char *)udp_data, 12, 0, &remote_addr, sizeof(struct sockaddr));
|
||
|
|
}while(ret<0);
|
||
|
|
|
||
|
|
do{
|
||
|
|
ret = sendto(fd, (unsigned char *)udp_data, 12, 0, &remote_addr, sizeof(struct sockaddr));
|
||
|
|
}while(ret<0);
|
||
|
|
|
||
|
|
len_total = 0;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
_os_printf("test send\r\n");
|
||
|
|
|
||
|
|
time_if = os_jiffies();
|
||
|
|
//for(type_eloop = 0;type_eloop < 10;type_eloop++)
|
||
|
|
//{
|
||
|
|
type_eloop = 0;
|
||
|
|
while(type_eloop < 10){
|
||
|
|
udp_data[8] = 101;
|
||
|
|
sendto(fd, (unsigned char *)udp_data, UDP_RCV_SIZE, 0, &remote_addr, sizeof(struct sockaddr));
|
||
|
|
if((os_jiffies() - time_if) > 2000){
|
||
|
|
|
||
|
|
_os_printf("sendto loop:%d\r\n",type_eloop);
|
||
|
|
time_if = os_jiffies();
|
||
|
|
type_eloop++;
|
||
|
|
udp_data[8] = 102;
|
||
|
|
do{
|
||
|
|
ret = sendto(fd, (unsigned char *)udp_data, 12, 0, &remote_addr, sizeof(struct sockaddr));
|
||
|
|
}while(ret<0);
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
//}
|
||
|
|
udp_data[8] = 103;
|
||
|
|
|
||
|
|
do{
|
||
|
|
ret = sendto(fd, (unsigned char *)udp_data, 12, 0, &remote_addr, sizeof(struct sockaddr));
|
||
|
|
}while(ret<0);
|
||
|
|
do{
|
||
|
|
ret = sendto(fd, (unsigned char *)udp_data, 12, 0, &remote_addr, sizeof(struct sockaddr));
|
||
|
|
}while(ret<0);
|
||
|
|
do{
|
||
|
|
ret = sendto(fd, (unsigned char *)udp_data, 12, 0, &remote_addr, sizeof(struct sockaddr));
|
||
|
|
}while(ret<0);
|
||
|
|
}
|
||
|
|
|
||
|
|
_os_printf("\n\n\n [ENTER] %s, Line = %d \n\n\n",__func__, __LINE__);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
static void udp_server_speed_accept(struct event_info *e, void *d)
|
||
|
|
{
|
||
|
|
char *tcp_rcv_buff;
|
||
|
|
struct sockaddr remote_addr;
|
||
|
|
int retval, size;
|
||
|
|
retval = 16;
|
||
|
|
int udp_server_fd = (int)d;
|
||
|
|
|
||
|
|
tcp_rcv_buff = (char*)malloc(UDP_RCV_SIZE + 4);
|
||
|
|
|
||
|
|
_os_printf("udp server fd: %d\n", udp_server_fd);
|
||
|
|
size = recvfrom(udp_server_fd, tcp_rcv_buff, UDP_RCV_SIZE, MSG_DONTWAIT, &remote_addr, (socklen_t*)&retval);
|
||
|
|
_os_printf("size:%d %s\n",size,tcp_rcv_buff);
|
||
|
|
_os_printf("udp client ip: %x\n", ((struct sockaddr_in *)&remote_addr)->sin_addr.s_addr);
|
||
|
|
if(size > 0) {
|
||
|
|
udp_speed_test(udp_server_fd);
|
||
|
|
}
|
||
|
|
|
||
|
|
free(tcp_rcv_buff);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
void udp_client_speed_send(struct event_info *e, void *d)
|
||
|
|
{
|
||
|
|
static char *send_buf = NULL;
|
||
|
|
int size;
|
||
|
|
if(!send_buf)
|
||
|
|
{
|
||
|
|
send_buf = (char*)malloc(UDP_RCV_SIZE);
|
||
|
|
memset(send_buf,0,UDP_RCV_SIZE);
|
||
|
|
}
|
||
|
|
uint32_t sec, usec,spcount;
|
||
|
|
static uint32_t pcount = 0;
|
||
|
|
++pcount;
|
||
|
|
uint64 times = os_jiffies();
|
||
|
|
sec = times/1000;
|
||
|
|
|
||
|
|
sec = htonl(times/1000);
|
||
|
|
usec = htonl((times%1000)*1000);
|
||
|
|
spcount = htonl(pcount);
|
||
|
|
|
||
|
|
|
||
|
|
memcpy(send_buf, &spcount, sizeof(spcount));
|
||
|
|
memcpy(send_buf+4, &sec, sizeof(sec));
|
||
|
|
memcpy(send_buf+8, &usec, sizeof(usec));
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
struct UdpSpeed *udp_struct = (struct UdpSpeed *)d;
|
||
|
|
struct sockaddr *remote_addr = (struct sockaddr*)&udp_struct->remote_addr;
|
||
|
|
int udp_client_fd = (int)udp_struct->fd;
|
||
|
|
size = sendto(udp_client_fd, (unsigned char *)send_buf, UDP_RCV_SIZE, 0, remote_addr, sizeof(struct sockaddr));
|
||
|
|
if(size < 0)
|
||
|
|
{
|
||
|
|
pcount--;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void udp_client_speed_recv(struct event_info *e, void *d)
|
||
|
|
{
|
||
|
|
static char *send_buf = NULL;
|
||
|
|
int size;
|
||
|
|
if(!send_buf)
|
||
|
|
{
|
||
|
|
send_buf = (char*)malloc(UDP_RCV_SIZE);
|
||
|
|
memset(send_buf,0,UDP_RCV_SIZE);
|
||
|
|
}
|
||
|
|
struct sockaddr remote_addr;
|
||
|
|
int retval;
|
||
|
|
retval = 16;
|
||
|
|
|
||
|
|
struct UdpSpeed *udp_struct = (struct UdpSpeed *)d;
|
||
|
|
int udp_client_fd = (int)udp_struct->fd;
|
||
|
|
size = recvfrom(udp_client_fd, send_buf, UDP_RCV_SIZE, MSG_DONTWAIT, &remote_addr, (socklen_t*)&retval);
|
||
|
|
if(size > 0)
|
||
|
|
{
|
||
|
|
_os_printf("................................size:%d\tsendbuf[0]:%X\n",size,send_buf[0]);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void speedTest_udp_server(int port)
|
||
|
|
{
|
||
|
|
|
||
|
|
int udp_server_fd;
|
||
|
|
//uint16_t port = 8060;
|
||
|
|
|
||
|
|
_os_printf("> init udp server\nudp port:%d\n", port);
|
||
|
|
|
||
|
|
udp_server_fd = creat_udp_socket(port);
|
||
|
|
eloop_add_fd( udp_server_fd, EVENT_READ, EVENT_F_ENABLED, (void*)udp_server_speed_accept, (void*)udp_server_fd );
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
void speedTest_udp_client(const char *ip,int port)
|
||
|
|
{
|
||
|
|
int udp_client_fd;
|
||
|
|
udp_client_fd = creat_udp_socket(0);
|
||
|
|
struct UdpSpeed *udp_struct = (struct UdpSpeed*)malloc(sizeof(struct UdpSpeed));
|
||
|
|
struct sockaddr_in *addrServer = &udp_struct->remote_addr;
|
||
|
|
memset(addrServer,0,sizeof(struct sockaddr_in));
|
||
|
|
addrServer->sin_family=AF_INET;
|
||
|
|
addrServer->sin_addr.s_addr=inet_addr(ip);
|
||
|
|
addrServer->sin_port=htons(port);
|
||
|
|
udp_struct->fd = udp_client_fd;
|
||
|
|
//shutdown(udp_client_fd,SHUT_RD);
|
||
|
|
|
||
|
|
//csi_kernel_task_new((k_task_entry_t)udp_client_speed_send, "speed", udp_struct, 1, 0, NULL, 1024, &eloop_task_handle);
|
||
|
|
eloop_add_fd( udp_client_fd, EVENT_WRITE, EVENT_F_ENABLED, (void*)udp_client_speed_send, (void*)udp_struct );
|
||
|
|
eloop_add_fd( udp_client_fd, EVENT_READ, EVENT_F_ENABLED, (void*)udp_client_speed_recv, (void*)udp_struct );
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
/****************************************************************TCP*****************************************************/
|
||
|
|
|
||
|
|
void TcpTX_event(struct event *ei, void *d)
|
||
|
|
{
|
||
|
|
int fd = (int)d;
|
||
|
|
int res;
|
||
|
|
static uint8_t *send_buf = NULL;
|
||
|
|
|
||
|
|
|
||
|
|
if(!send_buf)
|
||
|
|
{
|
||
|
|
send_buf = (uint8_t*)malloc(UDP_RCV_SIZE*10);
|
||
|
|
|
||
|
|
if(!send_buf)
|
||
|
|
{
|
||
|
|
_os_printf("%s no enough space\n",__FUNCTION__);
|
||
|
|
eloop_remove_event(ei);
|
||
|
|
if(fd > 0)
|
||
|
|
{
|
||
|
|
closesocket(fd);
|
||
|
|
}
|
||
|
|
send_buf = NULL;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
res = send(fd,send_buf,UDP_RCV_SIZE*10,0);
|
||
|
|
|
||
|
|
if(res < 0)
|
||
|
|
{
|
||
|
|
_os_printf("%s res:%d\n",__FUNCTION__,res);
|
||
|
|
eloop_remove_event(ei);
|
||
|
|
if(fd > 0)
|
||
|
|
{
|
||
|
|
closesocket(fd);
|
||
|
|
}
|
||
|
|
if(send_buf)
|
||
|
|
{
|
||
|
|
free(send_buf);
|
||
|
|
send_buf = NULL;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void TcpTX_event2(void *d)
|
||
|
|
{
|
||
|
|
int fd = (int)d;
|
||
|
|
int res;
|
||
|
|
uint8_t *send_buf = (uint8_t*)malloc(UDP_RCV_SIZE);
|
||
|
|
|
||
|
|
|
||
|
|
if(!send_buf)
|
||
|
|
{
|
||
|
|
if(fd > 0)
|
||
|
|
{
|
||
|
|
closesocket(fd);
|
||
|
|
}
|
||
|
|
return;
|
||
|
|
|
||
|
|
}
|
||
|
|
while(1)
|
||
|
|
{
|
||
|
|
res = send(fd,send_buf,UDP_RCV_SIZE,0);
|
||
|
|
|
||
|
|
if(res < 0)
|
||
|
|
{
|
||
|
|
|
||
|
|
if(fd > 0)
|
||
|
|
{
|
||
|
|
closesocket(fd);
|
||
|
|
}
|
||
|
|
free(send_buf);
|
||
|
|
break;
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
_os_printf("%s end\n",__FUNCTION__);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
void tcp_server_accept(struct event_info *e, void *d)
|
||
|
|
{
|
||
|
|
struct sockaddr_in addr;
|
||
|
|
socklen_t namelen = sizeof(addr);
|
||
|
|
int server_fd = (int)d;
|
||
|
|
int client_fd;
|
||
|
|
_os_printf("%s:%d\tserver_fd:%d\n",__FUNCTION__,__LINE__,server_fd);
|
||
|
|
client_fd = accept(server_fd,(struct sockaddr *)&addr,&namelen);
|
||
|
|
if(client_fd > 0)
|
||
|
|
{
|
||
|
|
k_task_handle_t eloop_task_handle;
|
||
|
|
//添加读事件
|
||
|
|
//eloop_add_fd( client_fd , EVENT_WRITE, EVENT_F_ENABLED, (void*)TcpTX_event, (void*)client_fd);
|
||
|
|
csi_kernel_task_new((k_task_entry_t)TcpTX_event2, "tcpTX", (void*)client_fd, 10, 0, NULL, 2048, &eloop_task_handle);
|
||
|
|
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
void speedTest_tcp_server(int port)
|
||
|
|
{
|
||
|
|
|
||
|
|
int tcp_server_fd;
|
||
|
|
//uint16_t port = 8060;
|
||
|
|
|
||
|
|
_os_printf("%s port:%d\n",__FUNCTION__,port);
|
||
|
|
|
||
|
|
tcp_server_fd = creat_tcp_socket(port);
|
||
|
|
_os_printf("%s fd:%d\n",__FUNCTION__,tcp_server_fd);
|
||
|
|
if(tcp_server_fd>=0)
|
||
|
|
{
|
||
|
|
listen(tcp_server_fd,3);
|
||
|
|
eloop_add_fd( tcp_server_fd, EVENT_READ, EVENT_F_ENABLED, (void*)tcp_server_accept, (void*)tcp_server_fd );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
//udp定时发送结构
|
||
|
|
struct UdpToIp
|
||
|
|
{
|
||
|
|
struct sockaddr_in remote_addr;
|
||
|
|
struct event *timer_e;
|
||
|
|
struct event *udp_e;
|
||
|
|
int timer;
|
||
|
|
int fd;
|
||
|
|
};
|
||
|
|
/*********************************************************************************************
|
||
|
|
定时用udp发送数据
|
||
|
|
*********************************************************************************************/
|
||
|
|
|
||
|
|
|
||
|
|
static void Udp_to_ip_send(struct event *ei, void *d)
|
||
|
|
{
|
||
|
|
struct UdpToIp *udp_struct = (struct UdpToIp *)d;
|
||
|
|
int serverfd = udp_struct->fd;
|
||
|
|
struct sockaddr *remote_addr = (struct sockaddr *)&udp_struct->remote_addr;
|
||
|
|
int retval, size;
|
||
|
|
retval = 16;
|
||
|
|
|
||
|
|
size = sendto(serverfd,"12345678", 8, 0, remote_addr, sizeof(struct sockaddr));
|
||
|
|
_os_printf("%s:%d\tsize:%d\tEINPROGRESS:%d\n",__FUNCTION__,__LINE__,size,EINPROGRESS);
|
||
|
|
eloop_set_event_enabled(ei,EVENT_F_DISABLED);
|
||
|
|
}
|
||
|
|
static void Udp_to_ip_timer(void *ei, void *d)
|
||
|
|
{
|
||
|
|
struct UdpToIp *udp_struct = (struct UdpToIp *)d;
|
||
|
|
eloop_set_event_enabled(udp_struct->udp_e,EVENT_F_ENABLED);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//定时向udp发送数据
|
||
|
|
void Udp_to_ip(char *ip,int port,int timer)
|
||
|
|
{
|
||
|
|
int t_socket;
|
||
|
|
t_socket = creat_udp_socket(0);
|
||
|
|
struct UdpToIp *udp_struct = (struct UdpToIp*)malloc(sizeof(struct UdpToIp));
|
||
|
|
struct sockaddr_in *addrServer = &udp_struct->remote_addr;
|
||
|
|
memset(addrServer,0,sizeof(struct sockaddr_in));;
|
||
|
|
addrServer->sin_family=AF_INET;
|
||
|
|
addrServer->sin_addr.s_addr=inet_addr(ip);
|
||
|
|
addrServer->sin_port=htons(port);
|
||
|
|
udp_struct->timer = timer;
|
||
|
|
udp_struct->fd = t_socket;
|
||
|
|
|
||
|
|
|
||
|
|
//添加事件
|
||
|
|
udp_struct->udp_e = eloop_add_fd( t_socket, EVENT_WRITE, EVENT_F_NOACTION, (void*)Udp_to_ip_send, (void*)udp_struct );
|
||
|
|
udp_struct->timer_e = eloop_add_timer(timer,EVENT_F_ENABLED,Udp_to_ip_timer,(void*)udp_struct);
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
void Udp_rx_server_register(int port,void *arg,event_callback f)
|
||
|
|
{
|
||
|
|
int t_socket;
|
||
|
|
t_socket = creat_udp_socket(port);
|
||
|
|
//fcntl(t_socket, F_SETFL, O_NONBLOCK);
|
||
|
|
|
||
|
|
int timeout = 10; // 设置超时时间为100ms
|
||
|
|
//fcntl(t_socket, F_SETFL, O_NONBLOCK);
|
||
|
|
if (setsockopt(t_socket, SOL_SOCKET, SO_RCVTIMEO, &timeout, sizeof(timeout)) < 0) {
|
||
|
|
// 错误处理
|
||
|
|
}
|
||
|
|
//添加事件
|
||
|
|
eloop_add_fd( t_socket, EVENT_READ, EVENT_F_ENABLED, f, (void*)arg );
|
||
|
|
}
|