mirror of
https://github.com/pvvx/RTL00MP3.git
synced 2025-07-31 12:41:06 +00:00
add example
This commit is contained in:
pvvx
parent
0cd01e4dc1
commit
5cd34b0c9f
75 changed files with 6023 additions and 217 deletions
|
|
@ -0,0 +1,170 @@
|
|||
#include "FreeRTOS.h"
|
||||
#include "task.h"
|
||||
#include <platform/platform_stdlib.h>
|
||||
#include <lwip/sockets.h>
|
||||
#include <lwip_netconf.h>
|
||||
|
||||
#define TEST_MODE 0 // 0 to test client keepalive, 1 to test server keepalive
|
||||
#define SERVER_IP 192.168.1.1
|
||||
#define SERVER_PORT 80
|
||||
#define LISTEN_QLEN 2
|
||||
#define MAX_SOCKETS 10
|
||||
#define SELECT_TIMEOUT 10
|
||||
|
||||
extern struct netif xnetif[];
|
||||
|
||||
static void example_tcp_keepalive_thread(void *param)
|
||||
{
|
||||
// Delay to wait for IP by DHCP
|
||||
vTaskDelay(10000);
|
||||
printf("\nExample: TCP Keepalive\n");
|
||||
#if !LWIP_TCP_KEEPALIVE
|
||||
printf("\nPlease enable LWIP_TCP_KEEPALIVE\n");
|
||||
#else
|
||||
|
||||
#if (TEST_MODE == 0)
|
||||
int server_socket, keepalive = 1, keepalive_idle = 3, keepalive_interval = 5, keepalive_count = 3;
|
||||
struct sockaddr_in server_addr;
|
||||
unsigned char *server_ip = LwIP_GetGW(&xnetif[0]);;
|
||||
server_socket = socket(AF_INET, SOCK_STREAM, 0);
|
||||
|
||||
// enable socket keepalive with keepalive timeout = idle(3) + interval(5) * count(3) = 18 seconds
|
||||
if(setsockopt(server_socket, SOL_SOCKET, SO_KEEPALIVE, &keepalive, sizeof(keepalive)) != 0)
|
||||
printf("ERROR: SO_KEEPALIVE\n");
|
||||
if(setsockopt(server_socket, IPPROTO_TCP, TCP_KEEPIDLE, &keepalive_idle, sizeof(keepalive_idle)) != 0)
|
||||
printf("ERROR: TCP_KEEPIDLE\n");
|
||||
if(setsockopt(server_socket, IPPROTO_TCP, TCP_KEEPINTVL, &keepalive_interval, sizeof(keepalive_interval)) != 0)
|
||||
printf("ERROR: TCP_KEEPINTVL\n");
|
||||
if(setsockopt(server_socket, IPPROTO_TCP, TCP_KEEPCNT, &keepalive_count, sizeof(keepalive_count)) != 0)
|
||||
printf("ERROR: TCP_KEEPCNT\n");
|
||||
|
||||
server_addr.sin_family = AF_INET;
|
||||
server_addr.sin_addr.s_addr = *((unsigned int *) server_ip);
|
||||
server_addr.sin_port = htons(SERVER_PORT);
|
||||
|
||||
if(connect(server_socket, (struct sockaddr *)&server_addr, sizeof(server_addr)) == 0) {
|
||||
unsigned char response_buf[100];
|
||||
int read_size;
|
||||
printf("connect OK\n");
|
||||
|
||||
while((read_size = read(server_socket, response_buf, sizeof(response_buf))) > 0)
|
||||
printf("read %d bytes\n", read_size);
|
||||
|
||||
printf("ERROR: read %d\n", read_size);
|
||||
close(server_socket);
|
||||
}
|
||||
else {
|
||||
printf("ERROR: connect\n");
|
||||
close(server_socket);
|
||||
}
|
||||
|
||||
#elif (TEST_MODE == 1)
|
||||
int max_socket_fd = -1;
|
||||
struct sockaddr_in server_addr;
|
||||
int server_fd = -1;
|
||||
int socket_used[MAX_SOCKETS];
|
||||
|
||||
memset(socket_used, 0, sizeof(socket_used));
|
||||
|
||||
if((server_fd = socket(AF_INET, SOCK_STREAM, 0)) >= 0) {
|
||||
server_addr.sin_family = AF_INET;
|
||||
server_addr.sin_port = htons(SERVER_PORT);
|
||||
server_addr.sin_addr.s_addr = INADDR_ANY;
|
||||
|
||||
if(bind(server_fd, (struct sockaddr *) &server_addr, sizeof(server_addr)) != 0) {
|
||||
printf("bind error\n");
|
||||
goto exit;
|
||||
}
|
||||
|
||||
if(listen(server_fd, LISTEN_QLEN) != 0) {
|
||||
printf("listen error\n");
|
||||
goto exit;
|
||||
}
|
||||
|
||||
socket_used[server_fd] = 1;
|
||||
|
||||
if(server_fd > max_socket_fd)
|
||||
max_socket_fd = server_fd;
|
||||
}
|
||||
else {
|
||||
printf("socket error\n");
|
||||
goto exit;
|
||||
}
|
||||
|
||||
while(1) {
|
||||
int socket_fd;
|
||||
unsigned char buf[512];
|
||||
fd_set read_fds;
|
||||
struct timeval timeout;
|
||||
|
||||
FD_ZERO(&read_fds);
|
||||
timeout.tv_sec = SELECT_TIMEOUT;
|
||||
timeout.tv_usec = 0;
|
||||
|
||||
for(socket_fd = 0; socket_fd < MAX_SOCKETS; socket_fd ++)
|
||||
if(socket_used[socket_fd])
|
||||
FD_SET(socket_fd, &read_fds);
|
||||
|
||||
if(select(max_socket_fd + 1, &read_fds, NULL, NULL, &timeout)) {
|
||||
for(socket_fd = 0; socket_fd < MAX_SOCKETS; socket_fd ++) {
|
||||
if(socket_used[socket_fd] && FD_ISSET(socket_fd, &read_fds)) {
|
||||
if(socket_fd == server_fd) {
|
||||
struct sockaddr_in client_addr;
|
||||
unsigned int client_addr_size = sizeof(client_addr);
|
||||
int fd = accept(server_fd, (struct sockaddr *) &client_addr, &client_addr_size);
|
||||
|
||||
if(fd >= 0) {
|
||||
int keepalive = 1, keepalive_idle = 3, keepalive_interval = 5, keepalive_count = 3;
|
||||
printf("accept socket fd(%d)\n", fd);
|
||||
socket_used[fd] = 1;
|
||||
|
||||
// enable socket keepalive with keepalive timeout = idle(3) + interval(5) * count(3) = 18 seconds
|
||||
if(setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &keepalive, sizeof(keepalive)) != 0)
|
||||
printf("ERROR: SO_KEEPALIVE\n");
|
||||
if(setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, &keepalive_idle, sizeof(keepalive_idle)) != 0)
|
||||
printf("ERROR: TCP_KEEPIDLE\n");
|
||||
if(setsockopt(fd, IPPROTO_TCP, TCP_KEEPINTVL, &keepalive_interval, sizeof(keepalive_interval)) != 0)
|
||||
printf("ERROR: TCP_KEEPINTVL\n");
|
||||
if(setsockopt(fd, IPPROTO_TCP, TCP_KEEPCNT, &keepalive_count, sizeof(keepalive_count)) != 0)
|
||||
printf("ERROR: TCP_KEEPCNT\n");
|
||||
|
||||
if(fd > max_socket_fd)
|
||||
max_socket_fd = fd;
|
||||
}
|
||||
else {
|
||||
printf("accept error\n");
|
||||
}
|
||||
}
|
||||
else {
|
||||
int read_size = read(socket_fd, buf, sizeof(buf));
|
||||
|
||||
if(read_size > 0) {
|
||||
write(socket_fd, buf, read_size);
|
||||
}
|
||||
else {
|
||||
printf("socket fd(%d) disconnected\n", socket_fd);
|
||||
socket_used[socket_fd] = 0;
|
||||
close(socket_fd);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
vTaskDelay(10);
|
||||
}
|
||||
|
||||
exit:
|
||||
if(server_fd >= 0)
|
||||
close(server_fd);
|
||||
#endif /* TEST_MODE */
|
||||
#endif /* LWIP_TCP_KEEPALIVE */
|
||||
|
||||
vTaskDelete(NULL);
|
||||
}
|
||||
|
||||
void example_tcp_keepalive(void)
|
||||
{
|
||||
if(xTaskCreate(example_tcp_keepalive_thread, ((const char*)"example_tcp_keepalive_thread"), 1024, NULL, tskIDLE_PRIORITY + 1, NULL) != pdPASS)
|
||||
printf("\n\r%s xTaskCreate(init_thread) failed", __FUNCTION__);
|
||||
}
|
||||
|
|
@ -0,0 +1,6 @@
|
|||
#ifndef EXAMPLE_TCP_KEEPALIVE_H
|
||||
#define EXAMPLE_TCP_KEEPALIVE_H
|
||||
|
||||
void example_tcp_keepalive(void);
|
||||
|
||||
#endif /* EXAMPLE_TCP_KEEPALIVE_H */
|
||||
|
|
@ -0,0 +1,26 @@
|
|||
LWIP TCP KEEPALIVE EXAMPLE
|
||||
|
||||
Description:
|
||||
Example for socket client and server implementation to break out of blocking read by TCP keepalive timeout.
|
||||
|
||||
Configuration:
|
||||
[lwipopts.h]
|
||||
#define LWIP_TCP_KEEPALIVE 1
|
||||
[platform_opts.h]
|
||||
#define CONFIG_EXAMPLE_TCP_KEEPALIVE 1
|
||||
|
||||
Execution:
|
||||
Can make automatical Wi-Fi connection when booting by using wlan fast connect example.
|
||||
A TCP keepalive example thread will be started automatically when booting.
|
||||
|
||||
Setup TEST_MODE in example_tcp_keepalive.c to verify client or server examples. Modify SERVER_IP and SERVER_PORT for client keepalive test. Modify SERVER_PORT for server keepalive test. Modify keepalive timeout by setting keepalive idle, interval and count value to socket options based on requirement.
|
||||
|
||||
1. For client TCP keepalive example, example thread connects to indicated server and continue to perform blocking read.
|
||||
When using ATWD command to disconnect wifi, client thread will break out of blocking read after keepalive timeout
|
||||
|
||||
2. For server TCP keepalive example, example thread listens on indicated port and read data from the accepted connection.
|
||||
Please use a client program to connect to server port of example thread
|
||||
When using ATWD command to disconnect wifi, server thread will get select read event after keepalive timeout
|
||||
|
||||
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue