Files
2019-05-23 17:10:44 +08:00

833 lines
19 KiB
C

// Copyright 2018-2019 Espressif Systems (Shanghai) PTE LTD
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include "lwip/opt.h"
#ifdef SOCKETS_MT
#define SOCKETS_MT_DISABLE_SHUTDOWN
#include "lwip/priv/api_msg.h"
/* disable all LWIP socket API when compiling LWIP raw socket */
#undef lwip_accept
#undef lwip_bind
#undef lwip_shutdown
#undef lwip_getpeername
#undef lwip_getsockname
#undef lwip_setsockopt
#undef lwip_getsockopt
#undef lwip_close
#undef lwip_connect
#undef lwip_listen
#undef lwip_recv
#undef lwip_recvfrom
#undef lwip_send
#undef lwip_sendmsg
#undef lwip_sendto
#undef lwip_socket
#undef lwip_select
#undef lwip_ioctlsocket
#define lwip_accept lwip_accept_esp
#define lwip_bind lwip_bind_esp
#define lwip_shutdown lwip_shutdown_esp
#define lwip_getpeername lwip_getpeername_esp
#define lwip_getsockname lwip_getsockname_esp
#define lwip_setsockopt lwip_setsockopt_esp
#define lwip_getsockopt lwip_getsockopt_esp
#define lwip_close lwip_closesocket_esp
#define lwip_connect lwip_connect_esp
#define lwip_listen lwip_listen_esp
#define lwip_recv lwip_recv_esp
#define lwip_recvfrom lwip_recvfrom_esp
#define lwip_send lwip_send_esp
#define lwip_sendmsg lwip_sendmsg_esp
#define lwip_sendto lwip_sendto_esp
#define lwip_socket lwip_socket_esp
#define lwip_select lwip_select_esp
#define lwip_ioctlsocket lwip_ioctl_esp
#undef lwip_read
#undef lwip_write
#undef lwip_writev
#undef lwip_close
#undef closesocket
#undef lwip_fcntl
#undef lwip_ioctl
#define lwip_read lwip_read_esp
#define lwip_write lwip_write_esp
#define lwip_writev lwip_writev_esp
#define lwip_close lwip_close_esp
#define lwip_fcntl lwip_fcntl_esp
#define lwip_ioctl lwip_ioctl_esp
#include "../../lwip/src/api/sockets.c"
/* disable macros to enable LWIP function */
#undef lwip_accept
#undef lwip_bind
#undef lwip_shutdown
#undef lwip_getpeername
#undef lwip_getsockname
#undef lwip_setsockopt
#undef lwip_getsockopt
#undef lwip_close
#undef lwip_connect
#undef lwip_listen
#undef lwip_recv
#undef lwip_recvfrom
#undef lwip_send
#undef lwip_sendmsg
#undef lwip_sendto
#undef lwip_socket
#undef lwip_select
#undef lwip_ioctlsocket
#undef lwip_read
#undef lwip_write
#undef lwip_writev
#undef lwip_close
#undef closesocket
#undef lwip_fcntl
#undef lwip_ioctl
/********************************************************************/
#define LWIP_SYNC_MT_SLEEP_MS 10
#define SOCK_MT_DEBUG_LEVEL 255
typedef struct socket_conn_sync {
struct tcpip_api_call_data call;
struct netconn *conn;
} socket_conn_sync_t;
typedef int (*lwip_io_mt_fn)(int, int );
/* Use about 2 bit */
#define SOCK_MT_STATE_SOCK 0
#define SOCK_MT_STATE_ACCEPT 1
#define SOCK_MT_STATE_CONNECT 2
#define SOCK_MT_STATE_SEND 3
#define SOCK_MT_STATE_ILL 4
#define SOCK_MT_LOCK_SEND (1 << 0)
#define SOCK_MT_LOCK_RECV (1 << 1)
#define SOCK_MT_LOCK_IOCTL (1 << 2)
#define SOCK_MT_LOCK_MIN SOCK_MT_LOCK_SEND
#define SOCK_MT_LOCK_MAX SOCK_MT_LOCK_IOCTL
#define SOCK_MT_SELECT_RECV (1 << 0)
#define SOCK_MT_SELECT_SEND (1 << 1)
typedef struct _sock_mt {
uint8_t opened : 1;
uint8_t state : 2;
uint8_t select : 2;
uint8_t lock : 3;
} sock_mt_t;
#if (SOCK_MT_DEBUG_LEVEL < 16)
#define SOCK_MT_DEBUG(level, ...) \
if (level >= SOCK_MT_DEBUG_LEVEL) \
printf(__VA_ARGS__);
#else
#define SOCK_MT_DEBUG(level, ...)
#endif
#define SOCK_MT_ENTER_CHECK(s, l, st) \
{ \
if (_sock_lock(s, l) != ERR_OK) \
return -1; \
if (st != SOCK_MT_STATE_ILL) \
_sock_set_state(s, st); \
}
#define SOCK_MT_EXIT_CHECK(s, l, st) \
{ \
if (st != SOCK_MT_STATE_ILL) \
_sock_set_state(s, SOCK_MT_STATE_SOCK); \
if (_sock_unlock(s, l) != ERR_OK) \
return -1; \
}
static volatile sock_mt_t DRAM_ATTR sockets_mt[NUM_SOCKETS];
static inline void _sock_mt_init(int s)
{
memset((void *)&sockets_mt[s - LWIP_SOCKET_OFFSET], 0, sizeof(sock_mt_t));
}
static inline int _sock_is_opened(int s)
{
return sockets_mt[s - LWIP_SOCKET_OFFSET].opened != 0;
}
static inline void _sock_set_open(int s, int opened)
{
SYS_ARCH_DECL_PROTECT(lev);
SYS_ARCH_PROTECT(lev);
sockets_mt[s - LWIP_SOCKET_OFFSET].opened = opened;
SYS_ARCH_UNPROTECT(lev);
}
static inline void _sock_set_state(int s, int state)
{
SYS_ARCH_DECL_PROTECT(lev);
SYS_ARCH_PROTECT(lev);
sockets_mt[s - LWIP_SOCKET_OFFSET].state = state;
SYS_ARCH_UNPROTECT(lev);
}
static inline int _sock_get_state(int s)
{
return sockets_mt[s - LWIP_SOCKET_OFFSET].state;
}
static inline int _sock_get_select(int s, int select)
{
return sockets_mt[s - LWIP_SOCKET_OFFSET].select & select;
}
static int inline _sock_is_lock(int s, int l)
{
return sockets_mt[s - LWIP_SOCKET_OFFSET].lock & l;
}
static int inline _sock_next_lock(int lock)
{
return lock << 1;
}
static void inline _sock_set_select(int s, int select)
{
SYS_ARCH_DECL_PROTECT(lev);
SYS_ARCH_PROTECT(lev);
sockets_mt[s - LWIP_SOCKET_OFFSET].select |= select;
SYS_ARCH_UNPROTECT(lev);
}
static void inline _sock_reset_select(int s, int select)
{
SYS_ARCH_DECL_PROTECT(lev);
SYS_ARCH_PROTECT(lev);
sockets_mt[s - LWIP_SOCKET_OFFSET].select &= ~select;
SYS_ARCH_UNPROTECT(lev);
}
static int _sock_try_lock(int s, int l)
{
int ret = ERR_OK;
SYS_ARCH_DECL_PROTECT(lev);
if (!_sock_is_opened(s)) {
ret = ERR_CLSD;
goto exit;
}
if (sockets_mt[s - LWIP_SOCKET_OFFSET].lock & l) {
ret = ERR_INPROGRESS;
goto exit;
}
SYS_ARCH_PROTECT(lev);
sockets_mt[s - LWIP_SOCKET_OFFSET].lock |= l;
SYS_ARCH_UNPROTECT(lev);
exit:
return ret;
}
static int _sock_lock(int s, int l)
{
int ret = ERR_OK;
if (tryget_socket(s) == NULL)
return -1;
SOCK_MT_DEBUG(1, "s %d l %d enter ", s, l);
while (1) {
ret = _sock_try_lock(s, l);
if (ret != ERR_INPROGRESS)
break;
vTaskDelay(1);
}
SOCK_MT_DEBUG(1, "OK %d\n", ret);
return ret;
}
static int _sock_unlock(int s, int l)
{
int ret = 0;
SYS_ARCH_DECL_PROTECT(lev);
SOCK_MT_DEBUG(1, "s %d l %d exit ", s, l);
SYS_ARCH_PROTECT(lev);
sockets_mt[s - LWIP_SOCKET_OFFSET].lock &= ~l;
SYS_ARCH_UNPROTECT(lev);
if (!_sock_is_opened(s)) {
ret = ERR_CLSD;
goto exit;
}
exit:
SOCK_MT_DEBUG(1, "OK %d\n", ret);
return ret;
}
static int lwip_enter_mt_select(int s, fd_set *read_set, fd_set *write_set)
{
int i;
if (s > NUM_SOCKETS + LWIP_SOCKET_OFFSET || s < LWIP_SOCKET_OFFSET)
return -1;
for (i = 0; i < s; i++) {
if(FD_ISSET(i, read_set) || FD_ISSET(i, write_set))
if (tryget_socket(i) == NULL)
goto failed1;
if (FD_ISSET(i, read_set)) {
err_t err;
_sock_set_select(i, SOCK_MT_SELECT_RECV);
err = _sock_lock(i, SOCK_MT_LOCK_RECV);
if (err != ERR_OK) {
goto failed2;
}
}
if (FD_ISSET(i, write_set)) {
err_t err;
_sock_set_select(i, SOCK_MT_SELECT_SEND);
err = _sock_lock(i, SOCK_MT_LOCK_SEND);
if (err != ERR_OK) {
goto failed3;
}
}
}
return 0;
failed3:
_sock_unlock(i, SOCK_MT_LOCK_SEND);
_sock_reset_select(i, SOCK_MT_SELECT_SEND);
failed2:
if (FD_ISSET(i, read_set)) {
_sock_unlock(i, SOCK_MT_LOCK_RECV);
_sock_reset_select(i, SOCK_MT_SELECT_RECV);
}
failed1:
for (i--; i >=0; i--) {
if (FD_ISSET(i, read_set) ) {
_sock_unlock(i, SOCK_MT_LOCK_RECV);
_sock_reset_select(i, SOCK_MT_SELECT_RECV);
}
if (FD_ISSET(i, write_set)) {
_sock_unlock(i, SOCK_MT_LOCK_SEND);
_sock_reset_select(i, SOCK_MT_SELECT_SEND);
}
}
return -1;
}
static void lwip_exit_mt_select(int s, fd_set *read_set, fd_set *write_set)
{
int i;
for (i = 0; i < s; i++) {
if (FD_ISSET(i, read_set)) {
_sock_unlock(i, SOCK_MT_LOCK_RECV);
_sock_reset_select(i, SOCK_MT_SELECT_RECV);
}
if (FD_ISSET(i, write_set)) {
_sock_unlock(i, SOCK_MT_LOCK_SEND);
_sock_reset_select(i, SOCK_MT_SELECT_SEND);
}
}
}
static err_t lwip_do_sync_accept(struct tcpip_api_call_data *call)
{
socket_conn_sync_t *sync = (socket_conn_sync_t *)call;
struct netconn *conn = sync->conn;
SYS_ARCH_DECL_PROTECT(lev);
SYS_ARCH_PROTECT(lev);
if (sys_mbox_valid(&conn->acceptmbox))
sys_mbox_trypost(&conn->acceptmbox, NULL);
conn->state = NETCONN_NONE;
SYS_ARCH_UNPROTECT(lev);
return ERR_OK;
}
static err_t lwip_do_sync_send(struct tcpip_api_call_data *call)
{
socket_conn_sync_t *sync = (socket_conn_sync_t *)call;
struct netconn *conn = sync->conn;
SYS_ARCH_DECL_PROTECT(lev);
SYS_ARCH_PROTECT(lev);
if (conn->current_msg) {
conn->current_msg->err = ERR_OK;
if (sys_sem_valid(conn->current_msg->op_completed_sem))
sys_sem_signal(conn->current_msg->op_completed_sem);
conn->current_msg = NULL;
}
conn->state = NETCONN_NONE;
SYS_ARCH_UNPROTECT(lev);
return ERR_OK;
}
static err_t lwip_do_sync_recv_state(struct tcpip_api_call_data *call)
{
socket_conn_sync_t *sync = (socket_conn_sync_t *)call;
struct netconn *conn = sync->conn;
SYS_ARCH_DECL_PROTECT(lev);
SYS_ARCH_PROTECT(lev);
SOCK_MT_DEBUG(1, "sync recv %d\n", conn->socket);
if (sys_mbox_valid(&conn->recvmbox))
sys_mbox_trypost(&conn->recvmbox, NULL);
conn->state = NETCONN_NONE;
SYS_ARCH_UNPROTECT(lev);
return ERR_OK;
}
static err_t lwip_do_sync_select_state(struct tcpip_api_call_data *call)
{
socket_conn_sync_t *sync = (socket_conn_sync_t *)call;
struct netconn *conn = sync->conn;
SYS_ARCH_DECL_PROTECT(lev);
SYS_ARCH_PROTECT(lev);
event_callback(conn, NETCONN_EVT_ERROR, 0);
conn->state = NETCONN_NONE;
SYS_ARCH_UNPROTECT(lev);
return ERR_OK;
}
static void lwip_sync_state_mt(int s)
{
struct lwip_sock *sock = tryget_socket(s);
int state = _sock_get_state(s);
socket_conn_sync_t sync = {
.conn = sock->conn,
};
SOCK_MT_DEBUG(1, "sync state %d\n", state);
switch (state) {
case SOCK_MT_STATE_ACCEPT :
tcpip_api_call(lwip_do_sync_accept, &sync.call);
break;
case SOCK_MT_STATE_CONNECT:
case SOCK_MT_STATE_SEND :
tcpip_api_call(lwip_do_sync_send, &sync.call);
break;
default :
break;
}
}
static void lwip_sync_recv_mt(int s)
{
struct lwip_sock *sock = tryget_socket(s);
socket_conn_sync_t sync = {
.conn = sock->conn,
};
tcpip_api_call(lwip_do_sync_recv_state, &sync.call);
}
static void lwip_sync_select_mt(int s)
{
struct lwip_sock *sock = tryget_socket(s);
socket_conn_sync_t sync = {
.conn = sock->conn,
};
tcpip_api_call(lwip_do_sync_select_state, &sync.call);
}
static void lwip_sync_mt(int s, int how)
{
int lock = SOCK_MT_LOCK_MIN;
do {
if (_sock_is_lock(s, lock)) {
int need_wait = 0;
extern void sys_arch_msleep(int ms);
if (!_sock_get_select(s, SOCK_MT_SELECT_RECV | SOCK_MT_SELECT_SEND)) {
switch (lock) {
case SOCK_MT_LOCK_SEND:
lwip_sync_state_mt(s);
need_wait = 1;
break;
case SOCK_MT_LOCK_RECV:
lwip_sync_recv_mt(s);
need_wait = 1;
break;
default :
break;
}
} else {
lwip_sync_select_mt(s);
need_wait = 1;
}
if (need_wait)
sys_arch_msleep(LWIP_SYNC_MT_SLEEP_MS);
} else
lock = _sock_next_lock(lock);
} while (lock < SOCK_MT_LOCK_MAX);
}
#if SET_SOLINGER_DEFAULT
#if LWIP_SO_LINGER
static void lwip_socket_set_so_link(int s, int linger)
{
struct lwip_sock *sock = get_socket(s);
if (sock) {
/*
* linker:
* -1: nothing
* 0: free sent_buf immediately
*/
sock->conn->linger = linger;
}
}
#else
#error "LWIP_SO_LINGER must be enable"
#endif /* LWIP_SO_LINGER */
#else /* SET_SOLINGER_DEFAULT */
#define lwip_socket_set_so_link(_s, _linger)
#endif /* SET_SOLINGER_DEFAULT */
int lwip_socket(int domain, int type, int protocol)
{
int s;
s = lwip_socket_esp(domain, type, protocol);
if (s < 0)
return -1;
lwip_socket_set_so_link(s, 0);
_sock_mt_init(s);
_sock_set_open(s, 1);
return s;
}
int lwip_bind(int s, const struct sockaddr *name, socklen_t namelen)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_ILL);
ret = lwip_bind_esp(s, name, namelen);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_ILL);
return ret;
}
int lwip_connect(int s, const struct sockaddr *name, socklen_t namelen)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_CONNECT);
ret = lwip_connect_esp(s, name, namelen);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_CONNECT);
return ret;
}
int lwip_listen(int s, int backlog)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_ILL);
lwip_socket_set_so_link(s, -1);
ret = lwip_listen_esp(s, backlog);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_ILL);
return ret;
}
int lwip_accept(int s, struct sockaddr *addr, socklen_t *addrlen)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_ACCEPT)
lwip_socket_set_so_link(s, -1);
ret = lwip_accept_esp(s, addr, addrlen);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_ACCEPT);
if (ret < 0)
return -1;
lwip_socket_set_so_link(ret, 0);
_sock_mt_init(ret);
_sock_set_open(ret, 1);
return ret;
}
int lwip_getpeername(int s, struct sockaddr *name, socklen_t *namelen)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL);
ret = lwip_getpeername_esp(s, name, namelen);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL);
return ret;
}
int lwip_getsockname(int s, struct sockaddr *name, socklen_t *namelen)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL);
ret = lwip_getsockname_esp(s, name, namelen);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL);
return ret;
}
int lwip_setsockopt(int s, int level, int optname, const void *optval, socklen_t optlen)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL);
ret = lwip_setsockopt_esp(s, level, optname, optval, optlen);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL);
return ret;
}
int lwip_getsockopt(int s, int level, int optname, void *optval, socklen_t *optlen)
{
int ret;
if (tryget_socket(s) == NULL)
return -1;
if (optname == SO_ERROR) {
int retval = 0;
if (!_sock_is_opened(s))
retval = ENOTCONN;
if (retval) {
*(int *)optval = retval;
return 0;
}
}
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL);
ret = lwip_getsockopt_esp(s, level, optname, optval, optlen);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL);
return ret;
}
int lwip_ioctl(int s, long cmd, void *argp)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL)
ret = lwip_ioctl_esp(s, cmd, argp);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL);
return ret;
}
int lwip_sendto(int s, const void *data, size_t size, int flags,
const struct sockaddr *to, socklen_t tolen)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_SEND);
ret = lwip_sendto_esp(s, data, size, flags, to, tolen);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_SEND);
return ret;
}
int lwip_send(int s, const void *data, size_t size, int flags)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_SEND);
ret = lwip_send_esp(s, data, size, flags);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_SEND, SOCK_MT_STATE_SEND);
return ret;
}
int lwip_recvfrom(int s, void *mem, size_t len, int flags,
struct sockaddr *from, socklen_t *fromlen)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_RECV, SOCK_MT_STATE_ILL);
ret = lwip_recvfrom_esp(s, mem, len, flags, from, fromlen);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_RECV, SOCK_MT_STATE_ILL);
return ret;
}
int lwip_recv(int s, void *mem, size_t len, int flags)
{
return lwip_recvfrom(s, mem, len, flags, NULL, NULL);
}
int lwip_read(int s, void *mem, size_t len)
{
return lwip_recvfrom(s, mem, len, 0, NULL, NULL);
}
int lwip_write(int s, const void *data, size_t size)
{
return lwip_send(s, data, size, 0);
}
int lwip_fcntl(int s, int cmd, int val)
{
int ret;
SOCK_MT_ENTER_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL);
ret = lwip_fcntl_esp(s, cmd, val);
SOCK_MT_EXIT_CHECK(s, SOCK_MT_LOCK_IOCTL, SOCK_MT_STATE_ILL);
return ret;
}
int lwip_shutdown(int s, int how)
{
return 0;
}
int lwip_close(int s)
{
int ret;
SYS_ARCH_DECL_PROTECT(lev);
if (tryget_socket(s) == NULL)
return -1;
SYS_ARCH_PROTECT(lev);
if (_sock_is_opened(s)) {
_sock_set_open(s, 0);
SYS_ARCH_UNPROTECT(lev);
} else {
SYS_ARCH_UNPROTECT(lev);
return -1;
}
#if ESP_UDP
struct lwip_sock *sock = get_socket(s);
if (sock)
udp_sync_close_netconn(sock->conn);
#endif
lwip_sync_mt(s, SHUT_RDWR);
ret = lwip_close_esp(s);
return ret;
}
int lwip_select(int maxfdp1, fd_set *readset, fd_set *writeset,
fd_set *exceptset, struct timeval *timeout)
{
int ret;
fd_set read_set, write_set;
if (readset)
MEMCPY(&read_set, readset, sizeof(fd_set));
else
FD_ZERO(&read_set);
if (writeset)
MEMCPY(&write_set, writeset, sizeof(fd_set));
else
FD_ZERO(&write_set);
ret = lwip_enter_mt_select(maxfdp1, &read_set, &write_set);
if (ret)
return ret;
ret = lwip_select_esp(maxfdp1, readset, writeset, exceptset, timeout);
lwip_exit_mt_select(maxfdp1, &read_set, &write_set);
return ret;
}
#endif