409 lines
11 KiB
C
409 lines
11 KiB
C
|
/*
|
||
|
|
* Copyright (C) 2017 XRADIO TECHNOLOGY CO., 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 XRADIO TECHNOLOGY CO., LTD. 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
|
||
|
|
* OWNER 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.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
|
||
|
|
#include "kernel/os/os.h"
|
||
|
|
#include "net/wlan/wlan.h"
|
||
|
|
#include "lwip/sockets.h"
|
||
|
|
#include "lwip/netif.h"
|
||
|
|
#include "lwip/sockets.h"
|
||
|
|
#include "serial.h"
|
||
|
|
#include "ap_socket.h"
|
||
|
|
#include "atcmd/at_command.h"
|
||
|
|
|
||
|
|
typedef struct Timer Timer;
|
||
|
|
|
||
|
|
struct Timer {
|
||
|
|
unsigned int end_time;
|
||
|
|
};
|
||
|
|
|
||
|
|
#define SOCKET_DATALEN_MAX 1460
|
||
|
|
#define SOCKET_LISTEN_MAX 8
|
||
|
|
#define SOCKET_TEST_PORT 8080
|
||
|
|
|
||
|
|
#define CONNECT_MAX 5
|
||
|
|
#define SERVER_THREAD_STACK_SIZE (2 * 1024)
|
||
|
|
|
||
|
|
typedef struct SocketSend_queueinf_def {
|
||
|
|
int datalen;
|
||
|
|
uint8_t *p_senddata;
|
||
|
|
} SocketSend_queueinf_t;
|
||
|
|
|
||
|
|
static int xr_listen_fd;
|
||
|
|
int xr_client_fd[CONNECT_MAX];
|
||
|
|
static uint8_t *socket_revbuffer[CONNECT_MAX];
|
||
|
|
static struct sockaddr_in saddr;
|
||
|
|
|
||
|
|
static OS_Thread_t g_server_thread[CONNECT_MAX];
|
||
|
|
static void ap_connect0_read(void *pvParameters);
|
||
|
|
static void ap_connect1_read(void *pvParameters);
|
||
|
|
static void ap_connect2_read(void *pvParameters);
|
||
|
|
static void ap_connect3_read(void *pvParameters);
|
||
|
|
static void ap_connect4_read(void *pvParameters);
|
||
|
|
void close_server_socket(int id);
|
||
|
|
static uint8_t cur_connect_max = CONNECT_MAX;
|
||
|
|
int current_connections;
|
||
|
|
|
||
|
|
static void xr_countdown_ms(Timer *timer, unsigned int timeout_ms)
|
||
|
|
{
|
||
|
|
timer->end_time = OS_TicksToMSecs(OS_GetTicks()) + timeout_ms;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int xr_left_ms(Timer *timer)
|
||
|
|
{
|
||
|
|
int diff = (int)(timer->end_time) - (int)(OS_TicksToMSecs(OS_GetTicks()));
|
||
|
|
return (diff < 0) ? 0 : diff;
|
||
|
|
}
|
||
|
|
|
||
|
|
static char xr_expired(Timer *timer)
|
||
|
|
{
|
||
|
|
return 0 <= (int)OS_TicksToMSecs(OS_GetTicks()) - (int)(timer->end_time); /* is time_now over than end time */
|
||
|
|
}
|
||
|
|
|
||
|
|
void set_maxconn(int maxconn)
|
||
|
|
{
|
||
|
|
cur_connect_max = maxconn;
|
||
|
|
}
|
||
|
|
|
||
|
|
int xr_socket_init(void)
|
||
|
|
{
|
||
|
|
int i;
|
||
|
|
|
||
|
|
xr_listen_fd = -1;
|
||
|
|
memset(xr_client_fd, -1, CONNECT_MAX * sizeof(int));
|
||
|
|
|
||
|
|
for (i = 0; i < CONNECT_MAX; i++) {
|
||
|
|
if (socket_revbuffer[i] == NULL) {
|
||
|
|
socket_revbuffer[i] = malloc(SOCKET_DATALEN_MAX);
|
||
|
|
memset(socket_revbuffer[i], 0, SOCKET_DATALEN_MAX);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
OS_ThreadCreate(&g_server_thread[0],
|
||
|
|
"ap_connect0_read",
|
||
|
|
ap_connect0_read,
|
||
|
|
NULL,
|
||
|
|
OS_PRIORITY_NORMAL,
|
||
|
|
SERVER_THREAD_STACK_SIZE);
|
||
|
|
OS_ThreadCreate(&g_server_thread[1],
|
||
|
|
"ap_connect1_read",
|
||
|
|
ap_connect1_read,
|
||
|
|
NULL,
|
||
|
|
OS_PRIORITY_NORMAL,
|
||
|
|
SERVER_THREAD_STACK_SIZE);
|
||
|
|
OS_ThreadCreate(&g_server_thread[2],
|
||
|
|
"ap_connect2_read",
|
||
|
|
ap_connect2_read,
|
||
|
|
NULL,
|
||
|
|
OS_PRIORITY_NORMAL,
|
||
|
|
SERVER_THREAD_STACK_SIZE);
|
||
|
|
OS_ThreadCreate(&g_server_thread[3],
|
||
|
|
"ap_connect3_read",
|
||
|
|
ap_connect3_read,
|
||
|
|
NULL,
|
||
|
|
OS_PRIORITY_NORMAL,
|
||
|
|
SERVER_THREAD_STACK_SIZE);
|
||
|
|
OS_ThreadCreate(&g_server_thread[4],
|
||
|
|
"ap_connect4_read",
|
||
|
|
ap_connect4_read,
|
||
|
|
NULL,
|
||
|
|
OS_PRIORITY_NORMAL,
|
||
|
|
SERVER_THREAD_STACK_SIZE);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static int xr_socket_read(int fd, unsigned char *buffer, int len,
|
||
|
|
int timeout_ms)
|
||
|
|
{
|
||
|
|
int recvLen = 0;
|
||
|
|
int leftms;
|
||
|
|
int rc = -1;
|
||
|
|
struct timeval tv;
|
||
|
|
Timer timer;
|
||
|
|
fd_set fdset;
|
||
|
|
|
||
|
|
xr_countdown_ms(&timer, timeout_ms);
|
||
|
|
|
||
|
|
do {
|
||
|
|
leftms = xr_left_ms(&timer);
|
||
|
|
tv.tv_sec = leftms / 1000;
|
||
|
|
tv.tv_usec = (leftms % 1000) * 1000;
|
||
|
|
|
||
|
|
FD_ZERO(&fdset);
|
||
|
|
FD_SET(fd, &fdset);
|
||
|
|
|
||
|
|
rc = select(fd + 1, &fdset, NULL, NULL, &tv);
|
||
|
|
if (rc > 0) {
|
||
|
|
rc = recv(fd, buffer + recvLen, len - recvLen, 0);
|
||
|
|
if (rc > 0) {
|
||
|
|
/* received normally */
|
||
|
|
recvLen += rc;
|
||
|
|
} else if (rc == 0) {
|
||
|
|
/* has disconnected with server */
|
||
|
|
recvLen = -1;
|
||
|
|
break;
|
||
|
|
} else {
|
||
|
|
/* network error */
|
||
|
|
AP_SOCKET_ERR("recv return %d, errno = %d\n", rc, errno);
|
||
|
|
recvLen = -2;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
} else if (rc == 0) {
|
||
|
|
if (recvLen != 0)
|
||
|
|
AP_SOCKET_ERR("received timeout and length had received is %d\n", recvLen);
|
||
|
|
/* timeouted and return the length received */
|
||
|
|
break;
|
||
|
|
} else {
|
||
|
|
/* network error */
|
||
|
|
AP_SOCKET_ERR("select return %d, errno = %d\n", rc, errno);
|
||
|
|
recvLen = -2;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
} while (recvLen < len && !xr_expired(&timer)); /* expired() is redundant? */
|
||
|
|
|
||
|
|
return recvLen;
|
||
|
|
}
|
||
|
|
|
||
|
|
static void ap_connect0_read(void *pvParameters)
|
||
|
|
{
|
||
|
|
int rec_len;
|
||
|
|
|
||
|
|
while (1) {
|
||
|
|
OS_MSleep(1);
|
||
|
|
if (xr_client_fd[0] >= 0) {
|
||
|
|
rec_len = xr_socket_read(xr_client_fd[0], socket_revbuffer[0], SOCKET_DATALEN_MAX, 10);
|
||
|
|
if (rec_len > 0) {
|
||
|
|
at_dump("+IPD,0,%d\r\n", rec_len);
|
||
|
|
at_data_output((char *)socket_revbuffer[0], rec_len);
|
||
|
|
memset(socket_revbuffer[0], 0, SOCKET_DATALEN_MAX);
|
||
|
|
} else if (rec_len < 0) {
|
||
|
|
closesocket(xr_client_fd[0]);
|
||
|
|
xr_client_fd[0] = -1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
static void ap_connect1_read(void *pvParameters)
|
||
|
|
{
|
||
|
|
int rec_len;
|
||
|
|
|
||
|
|
while (1) {
|
||
|
|
OS_MSleep(1);
|
||
|
|
if (xr_client_fd[1] >= 0) {
|
||
|
|
rec_len = xr_socket_read(xr_client_fd[1], socket_revbuffer[1], SOCKET_DATALEN_MAX, 10);
|
||
|
|
if (rec_len > 0) {
|
||
|
|
at_dump("+IPD,1,%d\r\n", rec_len);
|
||
|
|
at_data_output((char *)socket_revbuffer[1], rec_len);
|
||
|
|
memset(socket_revbuffer[1], 0, SOCKET_DATALEN_MAX);
|
||
|
|
} else if (rec_len < 0) {
|
||
|
|
closesocket(xr_client_fd[1]);
|
||
|
|
xr_client_fd[1] = -1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
void ap_connect2_read(void *pvParameters)
|
||
|
|
{
|
||
|
|
int rec_len;
|
||
|
|
|
||
|
|
while (1) {
|
||
|
|
OS_MSleep(1);
|
||
|
|
if (xr_client_fd[2] >= 0) {
|
||
|
|
rec_len = xr_socket_read(xr_client_fd[2], socket_revbuffer[2], SOCKET_DATALEN_MAX, 10);
|
||
|
|
if (rec_len > 0) {
|
||
|
|
at_dump("+IPD,2,%d\r\n", rec_len);
|
||
|
|
at_data_output((char *)socket_revbuffer[2], rec_len);
|
||
|
|
memset(socket_revbuffer[2], 0, SOCKET_DATALEN_MAX);
|
||
|
|
} else if (rec_len < 0) {
|
||
|
|
closesocket(xr_client_fd[2]);
|
||
|
|
xr_client_fd[2] = -1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
static void ap_connect3_read(void *pvParameters)
|
||
|
|
{
|
||
|
|
int rec_len;
|
||
|
|
|
||
|
|
while (1) {
|
||
|
|
OS_MSleep(1);
|
||
|
|
if (xr_client_fd[3] >= 0) {
|
||
|
|
rec_len = xr_socket_read(xr_client_fd[3], socket_revbuffer[3], SOCKET_DATALEN_MAX, 10);
|
||
|
|
if (rec_len > 0) {
|
||
|
|
at_dump("+IPD,3,%d\r\n", rec_len);
|
||
|
|
at_data_output((char *)socket_revbuffer[3], rec_len);
|
||
|
|
memset(socket_revbuffer[3], 0, SOCKET_DATALEN_MAX);
|
||
|
|
} else if (rec_len < 0) {
|
||
|
|
closesocket(xr_client_fd[3]);
|
||
|
|
xr_client_fd[3] = -1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
static void ap_connect4_read(void *pvParameters)
|
||
|
|
{
|
||
|
|
int rec_len;
|
||
|
|
|
||
|
|
while (1) {
|
||
|
|
OS_MSleep(1);
|
||
|
|
if (xr_client_fd[4] >= 0) {
|
||
|
|
rec_len = xr_socket_read(xr_client_fd[4], socket_revbuffer[4], SOCKET_DATALEN_MAX, 10);
|
||
|
|
if (rec_len > 0) {
|
||
|
|
at_dump("+IPD,4,%d\r\n", rec_len);
|
||
|
|
at_data_output((char *)socket_revbuffer[4], rec_len);
|
||
|
|
memset(socket_revbuffer[4], 0, SOCKET_DATALEN_MAX);
|
||
|
|
} else if (rec_len < 0) {
|
||
|
|
closesocket(xr_client_fd[4]);
|
||
|
|
xr_client_fd[4] = -1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
extern void ap_task_delete(void);
|
||
|
|
extern void close_server(void);
|
||
|
|
|
||
|
|
void ap_socket_task(int port)
|
||
|
|
{
|
||
|
|
int err, i;
|
||
|
|
int option;
|
||
|
|
int temp_fd = -1;
|
||
|
|
socklen_t socklen = sizeof(saddr);
|
||
|
|
|
||
|
|
xr_socket_init();
|
||
|
|
|
||
|
|
if (xr_listen_fd < 0) {
|
||
|
|
xr_listen_fd = socket(AF_INET, SOCK_STREAM, IPPROTO_IP);
|
||
|
|
if (xr_listen_fd < 0) {
|
||
|
|
printf("socket create err!\r\n");
|
||
|
|
goto socket_err;
|
||
|
|
}
|
||
|
|
|
||
|
|
AP_SOCKET_DBG("->>>>>>>>>>port=%d\r\n", port);
|
||
|
|
saddr.sin_family = AF_INET;
|
||
|
|
saddr.sin_port = htons(port);
|
||
|
|
saddr.sin_addr.s_addr = htonl(INADDR_ANY);
|
||
|
|
|
||
|
|
option = 1;
|
||
|
|
if (setsockopt(xr_listen_fd, SOL_SOCKET, SO_REUSEADDR, &option, sizeof(option)) < 0) {
|
||
|
|
AP_SOCKET_ERR("failed to setsockopt sock_fd!\n");
|
||
|
|
closesocket(xr_listen_fd);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
option = 1;
|
||
|
|
if (setsockopt(xr_listen_fd, SOL_SOCKET, SO_REUSEPORT, &option, sizeof(option)) < 0) {
|
||
|
|
AP_SOCKET_ERR("failed to setsockopt sock_fd!\n");
|
||
|
|
closesocket(xr_listen_fd);
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
err = bind(xr_listen_fd, (struct sockaddr *)&saddr, sizeof(struct sockaddr_in));
|
||
|
|
if (err < 0) {
|
||
|
|
AP_SOCKET_ERR("Failed to bind tcp socket. Error %d\n", errno);
|
||
|
|
goto socket_err;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (listen(xr_listen_fd, SOCKET_LISTEN_MAX) != 0) {
|
||
|
|
AP_SOCKET_ERR("listen failed\n");
|
||
|
|
goto socket_err;
|
||
|
|
}
|
||
|
|
AP_SOCKET_DBG("socket listen success!============\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
while (1) {
|
||
|
|
option = 1000;
|
||
|
|
setsockopt(xr_listen_fd, SOL_SOCKET, SO_RCVTIMEO, &option, sizeof(option));
|
||
|
|
|
||
|
|
temp_fd = accept(xr_listen_fd, (struct sockaddr *)&saddr, &socklen);
|
||
|
|
if (temp_fd > 0) {
|
||
|
|
if (current_connections > cur_connect_max) {
|
||
|
|
AP_SOCKET_DBG("Too many connections is %d\r\n", current_connections);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
for (i = 0; i < cur_connect_max; i++) {
|
||
|
|
if (xr_client_fd[i] < 0) {
|
||
|
|
current_connections++;
|
||
|
|
AP_SOCKET_DBG("current_connections=%d\r\n", current_connections);
|
||
|
|
xr_client_fd[i] = temp_fd;
|
||
|
|
AP_SOCKET_DBG("connect [%d] = %d:\r\n", i, temp_fd);
|
||
|
|
at_dump("%d,CONNECT", i);
|
||
|
|
temp_fd = -1;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
OS_MSleep(100);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
socket_err:
|
||
|
|
ap_task_delete();
|
||
|
|
close_server();
|
||
|
|
}
|
||
|
|
|
||
|
|
void close_server(void)
|
||
|
|
{
|
||
|
|
int i;
|
||
|
|
|
||
|
|
AP_SOCKET_DBG("colse all socket!!!\r\n");
|
||
|
|
closesocket(xr_listen_fd);
|
||
|
|
xr_listen_fd = -1;
|
||
|
|
current_connections = 0;
|
||
|
|
for (i = 0; i < CONNECT_MAX; i++) {
|
||
|
|
closesocket(xr_client_fd[i]);
|
||
|
|
xr_client_fd[i] = -1;
|
||
|
|
if (socket_revbuffer[i] != NULL)
|
||
|
|
free(socket_revbuffer[i]);
|
||
|
|
OS_ThreadDelete(&g_server_thread[i]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void close_server_socket(int id)
|
||
|
|
{
|
||
|
|
if (socket_revbuffer[id] != NULL)
|
||
|
|
free(socket_revbuffer[id]);
|
||
|
|
closesocket(xr_client_fd[id]);
|
||
|
|
xr_client_fd[id] = -1;
|
||
|
|
current_connections--;
|
||
|
|
at_dump("%d,CLOSED", id);
|
||
|
|
}
|