1236 lines
34 KiB
C
1236 lines
34 KiB
C
/*
|
|
* Copyright (c) 2006-2022, RT-Thread Development Team
|
|
*
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*
|
|
* Change Logs:
|
|
* Date Author Notes
|
|
* 2018-05-23 ChenYong First version
|
|
* 2018-11-12 ChenYong Add TLS support
|
|
*/
|
|
|
|
#include <rtthread.h>
|
|
#include <rthw.h>
|
|
#include <sys/time.h>
|
|
#include <sys/ioctl.h>
|
|
|
|
#include <sal_socket.h>
|
|
#include <sal_netdb.h>
|
|
#ifdef SAL_USING_TLS
|
|
#include <sal_tls.h>
|
|
#endif
|
|
#include <sal_low_lvl.h>
|
|
#include <netdev.h>
|
|
|
|
#ifdef SAL_INTERNET_CHECK
|
|
#include <ipc/workqueue.h>
|
|
#endif
|
|
|
|
/* check system workqueue stack size */
|
|
#if defined(SAL_INTERNET_CHECK) && RT_SYSTEM_WORKQUEUE_STACKSIZE < 1536
|
|
#error "The system workqueue stack size must more than 1536 bytes"
|
|
#endif
|
|
|
|
#define DBG_TAG "sal.skt"
|
|
#define DBG_LVL DBG_INFO
|
|
#include <rtdbg.h>
|
|
|
|
#define SOCKET_TABLE_STEP_LEN 4
|
|
|
|
/* the socket table used to dynamic allocate sockets */
|
|
struct sal_socket_table
|
|
{
|
|
uint32_t max_socket;
|
|
struct sal_socket **sockets;
|
|
};
|
|
|
|
/* record the netdev and res table*/
|
|
struct sal_netdev_res_table
|
|
{
|
|
struct addrinfo *res;
|
|
struct netdev *netdev;
|
|
};
|
|
|
|
#ifdef SAL_USING_TLS
|
|
/* The global TLS protocol options */
|
|
static struct sal_proto_tls *proto_tls;
|
|
#endif
|
|
|
|
/* The global socket table */
|
|
static struct sal_socket_table socket_table;
|
|
static struct rt_mutex sal_core_lock;
|
|
static rt_bool_t init_ok = RT_FALSE;
|
|
static struct sal_netdev_res_table sal_dev_res_tbl[SAL_SOCKETS_NUM];
|
|
|
|
#define IS_SOCKET_PROTO_TLS(sock) (((sock)->protocol == PROTOCOL_TLS) || \
|
|
((sock)->protocol == PROTOCOL_DTLS))
|
|
#define SAL_SOCKOPS_PROTO_TLS_VALID(sock, name) (proto_tls && (proto_tls->ops->name) && IS_SOCKET_PROTO_TLS(sock))
|
|
|
|
#define SAL_SOCKOPT_PROTO_TLS_EXEC(sock, name, optval, optlen) \
|
|
do { \
|
|
if (SAL_SOCKOPS_PROTO_TLS_VALID(sock, name)){ \
|
|
return proto_tls->ops->name((sock)->user_data_tls, (optval), (optlen)); \
|
|
} \
|
|
}while(0) \
|
|
|
|
#define SAL_SOCKET_OBJ_GET(sock, socket) \
|
|
do { \
|
|
(sock) = sal_get_socket(socket); \
|
|
if ((sock) == RT_NULL) { \
|
|
return -1; \
|
|
} \
|
|
}while(0) \
|
|
|
|
#define SAL_NETDEV_IS_UP(netdev) \
|
|
do { \
|
|
if (!netdev_is_up(netdev)) { \
|
|
return -1; \
|
|
} \
|
|
}while(0) \
|
|
|
|
#define SAL_NETDEV_SOCKETOPS_VALID(netdev, pf, ops) \
|
|
do { \
|
|
(pf) = (struct sal_proto_family *) netdev->sal_user_data; \
|
|
if ((pf)->skt_ops->ops == RT_NULL){ \
|
|
return -1; \
|
|
} \
|
|
}while(0) \
|
|
|
|
#define SAL_NETDEV_NETDBOPS_VALID(netdev, pf, ops) \
|
|
((netdev) && netdev_is_up(netdev) && \
|
|
((pf) = (struct sal_proto_family *) (netdev)->sal_user_data) != RT_NULL && \
|
|
(pf)->netdb_ops->ops) \
|
|
|
|
#define SAL_NETDBOPS_VALID(netdev, pf, ops) \
|
|
((netdev) && \
|
|
((pf) = (struct sal_proto_family *) (netdev)->sal_user_data) != RT_NULL && \
|
|
(pf)->netdb_ops->ops) \
|
|
|
|
/**
|
|
* SAL (Socket Abstraction Layer) initialize.
|
|
*
|
|
* @return result 0: initialize success
|
|
* -1: initialize failed
|
|
*/
|
|
int sal_init(void)
|
|
{
|
|
int cn;
|
|
|
|
if (init_ok)
|
|
{
|
|
LOG_D("Socket Abstraction Layer is already initialized.");
|
|
return 0;
|
|
}
|
|
|
|
/* init sal socket table */
|
|
cn = SOCKET_TABLE_STEP_LEN < SAL_SOCKETS_NUM ? SOCKET_TABLE_STEP_LEN : SAL_SOCKETS_NUM;
|
|
socket_table.max_socket = cn;
|
|
socket_table.sockets = rt_calloc(1, cn * sizeof(struct sal_socket *));
|
|
if (socket_table.sockets == RT_NULL)
|
|
{
|
|
LOG_E("No memory for socket table.\n");
|
|
return -1;
|
|
}
|
|
|
|
/*init the dev_res table */
|
|
rt_memset(sal_dev_res_tbl, 0, sizeof(sal_dev_res_tbl));
|
|
|
|
/* create sal socket lock */
|
|
rt_mutex_init(&sal_core_lock, "sal_lock", RT_IPC_FLAG_PRIO);
|
|
|
|
LOG_I("Socket Abstraction Layer initialize success.");
|
|
init_ok = RT_TRUE;
|
|
|
|
return 0;
|
|
}
|
|
INIT_COMPONENT_EXPORT(sal_init);
|
|
|
|
#ifdef SAL_INTERNET_CHECK
|
|
/* check SAL network interface device internet status */
|
|
static void check_netdev_internet_up_work(struct rt_work *work, void *work_data)
|
|
{
|
|
#define SAL_INTERNET_VERSION 0x00
|
|
#define SAL_INTERNET_BUFF_LEN 12
|
|
#define SAL_INTERNET_TIMEOUT (2)
|
|
|
|
#define SAL_INTERNET_HOST "link.rt-thread.org"
|
|
#define SAL_INTERNET_PORT 8101
|
|
|
|
#define SAL_INTERNET_MONTH_LEN 4
|
|
#define SAL_INTERNET_DATE_LEN 16
|
|
|
|
unsigned int index;
|
|
int sockfd = -1, result = 0;
|
|
struct sockaddr_in server_addr;
|
|
struct hostent *host;
|
|
struct timeval timeout;
|
|
struct netdev *netdev = (struct netdev *)work_data;
|
|
socklen_t addr_len = sizeof(struct sockaddr_in);
|
|
char send_data[SAL_INTERNET_BUFF_LEN], recv_data = 0;
|
|
|
|
const char month[][SAL_INTERNET_MONTH_LEN] = {"Jan", "Feb", "Mar", "Apr", "May", "Jun", "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"};
|
|
char date[SAL_INTERNET_DATE_LEN];
|
|
unsigned int moth_num = 0;
|
|
|
|
struct sal_proto_family *pf = (struct sal_proto_family *) netdev->sal_user_data;
|
|
const struct sal_socket_ops *skt_ops;
|
|
|
|
if (work)
|
|
{
|
|
rt_free(work);
|
|
}
|
|
|
|
/* get network interface socket operations */
|
|
if (pf == RT_NULL || pf->skt_ops == RT_NULL)
|
|
{
|
|
result = -RT_ERROR;
|
|
goto __exit;
|
|
}
|
|
|
|
host = (struct hostent *) pf->netdb_ops->gethostbyname(SAL_INTERNET_HOST);
|
|
if (host == RT_NULL)
|
|
{
|
|
result = -RT_ERROR;
|
|
goto __exit;
|
|
}
|
|
|
|
skt_ops = pf->skt_ops;
|
|
if ((sockfd = skt_ops->socket(AF_INET, SOCK_DGRAM, 0)) < 0)
|
|
{
|
|
result = -RT_ERROR;
|
|
goto __exit;
|
|
}
|
|
|
|
server_addr.sin_family = AF_INET;
|
|
server_addr.sin_port = htons(SAL_INTERNET_PORT);
|
|
server_addr.sin_addr = *((struct in_addr *)host->h_addr);
|
|
rt_memset(&(server_addr.sin_zero), 0, sizeof(server_addr.sin_zero));
|
|
|
|
timeout.tv_sec = SAL_INTERNET_TIMEOUT;
|
|
timeout.tv_usec = 0;
|
|
|
|
/* set receive and send timeout */
|
|
skt_ops->setsockopt(sockfd, SOL_SOCKET, SO_RCVTIMEO, (void *) &timeout, sizeof(timeout));
|
|
skt_ops->setsockopt(sockfd, SOL_SOCKET, SO_SNDTIMEO, (void *) &timeout, sizeof(timeout));
|
|
|
|
/* get build moth value*/
|
|
rt_memset(date, 0x00, SAL_INTERNET_DATE_LEN);
|
|
rt_snprintf(date, SAL_INTERNET_DATE_LEN, "%s", __DATE__);
|
|
|
|
for (index = 0; index < sizeof(month) / SAL_INTERNET_MONTH_LEN; index++)
|
|
{
|
|
if (rt_memcmp(date, month[index], SAL_INTERNET_MONTH_LEN - 1) == 0)
|
|
{
|
|
moth_num = index + 1;
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* not find build month */
|
|
if (moth_num == 0 || moth_num > sizeof(month) / SAL_INTERNET_MONTH_LEN)
|
|
{
|
|
result = -RT_ERROR;
|
|
goto __exit;
|
|
}
|
|
|
|
rt_memset(send_data, 0x00, SAL_INTERNET_BUFF_LEN);
|
|
send_data[0] = SAL_INTERNET_VERSION;
|
|
for (index = 0; index < netdev->hwaddr_len; index++)
|
|
{
|
|
send_data[index + 1] = netdev->hwaddr[index] + moth_num;
|
|
}
|
|
send_data[9] = RT_VERSION_MAJOR;
|
|
send_data[10] = RT_VERSION_MINOR;
|
|
send_data[11] = RT_VERSION_PATCH;
|
|
|
|
skt_ops->sendto(sockfd, send_data, SAL_INTERNET_BUFF_LEN, 0,
|
|
(struct sockaddr *)&server_addr, sizeof(struct sockaddr));
|
|
|
|
result = skt_ops->recvfrom(sockfd, &recv_data, sizeof(recv_data), 0, (struct sockaddr *)&server_addr, &addr_len);
|
|
if (result < 0)
|
|
{
|
|
goto __exit;
|
|
}
|
|
|
|
if (recv_data == RT_FALSE)
|
|
{
|
|
result = -RT_ERROR;
|
|
goto __exit;
|
|
}
|
|
|
|
__exit:
|
|
if (result > 0)
|
|
{
|
|
LOG_D("Set network interface device(%s) internet status up.", netdev->name);
|
|
netdev_low_level_set_internet_status(netdev, RT_TRUE);
|
|
}
|
|
else
|
|
{
|
|
LOG_D("Set network interface device(%s) internet status down.", netdev->name);
|
|
netdev_low_level_set_internet_status(netdev, RT_FALSE);
|
|
}
|
|
|
|
if (sockfd >= 0)
|
|
{
|
|
skt_ops->closesocket(sockfd);
|
|
}
|
|
}
|
|
#endif /* SAL_INTERNET_CHECK */
|
|
|
|
/**
|
|
* This function will check SAL network interface device internet status.
|
|
*
|
|
* @param netdev the network interface device to check
|
|
*/
|
|
int sal_check_netdev_internet_up(struct netdev *netdev)
|
|
{
|
|
RT_ASSERT(netdev);
|
|
|
|
#ifdef SAL_INTERNET_CHECK
|
|
/* workqueue for network connect */
|
|
struct rt_work *net_work = RT_NULL;
|
|
|
|
|
|
net_work = (struct rt_work *)rt_calloc(1, sizeof(struct rt_work));
|
|
if (net_work == RT_NULL)
|
|
{
|
|
LOG_W("No memory for network interface device(%s) delay work.", netdev->name);
|
|
return -1;
|
|
}
|
|
|
|
rt_work_init(net_work, check_netdev_internet_up_work, (void *)netdev);
|
|
rt_work_submit(net_work, RT_TICK_PER_SECOND);
|
|
#endif /* SAL_INTERNET_CHECK */
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* This function will register TLS protocol to the global TLS protocol.
|
|
*
|
|
* @param pt TLS protocol object
|
|
*
|
|
* @return 0: TLS protocol object register success
|
|
*/
|
|
#ifdef SAL_USING_TLS
|
|
int sal_proto_tls_register(const struct sal_proto_tls *pt)
|
|
{
|
|
RT_ASSERT(pt);
|
|
proto_tls = (struct sal_proto_tls *) pt;
|
|
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
/**
|
|
* This function will get sal socket object by sal socket descriptor.
|
|
*
|
|
* @param socket sal socket index
|
|
*
|
|
* @return sal socket object of the current sal socket index
|
|
*/
|
|
struct sal_socket *sal_get_socket(int socket)
|
|
{
|
|
struct sal_socket_table *st = &socket_table;
|
|
|
|
socket = socket - SAL_SOCKET_OFFSET;
|
|
|
|
if (socket < 0 || socket >= (int) st->max_socket)
|
|
{
|
|
return RT_NULL;
|
|
}
|
|
|
|
/* check socket structure valid or not */
|
|
RT_ASSERT(st->sockets[socket]->magic == SAL_SOCKET_MAGIC);
|
|
|
|
return st->sockets[socket];
|
|
}
|
|
|
|
/**
|
|
* This function will lock sal socket.
|
|
*
|
|
* @note please don't invoke it on ISR.
|
|
*/
|
|
static void sal_lock(void)
|
|
{
|
|
rt_err_t result;
|
|
|
|
result = rt_mutex_take(&sal_core_lock, RT_WAITING_FOREVER);
|
|
if (result != RT_EOK)
|
|
{
|
|
RT_ASSERT(0);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* This function will lock sal socket.
|
|
*
|
|
* @note please don't invoke it on ISR.
|
|
*/
|
|
static void sal_unlock(void)
|
|
{
|
|
rt_mutex_release(&sal_core_lock);
|
|
}
|
|
|
|
/**
|
|
* This function will clean the netdev.
|
|
*
|
|
* @note please don't invoke it on ISR.
|
|
*/
|
|
int sal_netdev_cleanup(struct netdev *netdev)
|
|
{
|
|
uint32_t idx = 0;
|
|
int find_dev;
|
|
|
|
do
|
|
{
|
|
find_dev = 0;
|
|
sal_lock();
|
|
for (idx = 0; idx < socket_table.max_socket; idx++)
|
|
{
|
|
if (socket_table.sockets[idx] && socket_table.sockets[idx]->netdev == netdev)
|
|
{
|
|
find_dev = 1;
|
|
break;
|
|
}
|
|
}
|
|
sal_unlock();
|
|
if (find_dev)
|
|
{
|
|
rt_thread_mdelay(100);
|
|
}
|
|
}
|
|
while (find_dev);
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* This function will initialize sal socket object and set socket options
|
|
*
|
|
* @param family protocol family
|
|
* @param type socket type
|
|
* @param protocol transfer Protocol
|
|
* @param res sal socket object address
|
|
*
|
|
* @return 0 : socket initialize success
|
|
* -1 : input the wrong family
|
|
* -2 : input the wrong socket type
|
|
* -3 : get network interface failed
|
|
*/
|
|
static int socket_init(int family, int type, int protocol, struct sal_socket **res)
|
|
{
|
|
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
struct netdev *netdv_def = netdev_default;
|
|
struct netdev *netdev = RT_NULL;
|
|
rt_bool_t flag = RT_FALSE;
|
|
|
|
if (family < 0 || family > AF_MAX)
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
if (type < 0 || type > SOCK_MAX)
|
|
{
|
|
return -2;
|
|
}
|
|
|
|
sock = *res;
|
|
sock->domain = family;
|
|
sock->type = type;
|
|
sock->protocol = protocol;
|
|
|
|
if (netdv_def && netdev_is_up(netdv_def))
|
|
{
|
|
/* check default network interface device protocol family */
|
|
pf = (struct sal_proto_family *) netdv_def->sal_user_data;
|
|
if (pf != RT_NULL && pf->skt_ops && (pf->family == family || pf->sec_family == family))
|
|
{
|
|
sock->netdev = netdv_def;
|
|
flag = RT_TRUE;
|
|
}
|
|
}
|
|
|
|
if (flag == RT_FALSE)
|
|
{
|
|
/* get network interface device by protocol family */
|
|
netdev = netdev_get_by_family(family);
|
|
if (netdev == RT_NULL)
|
|
{
|
|
LOG_E("not find network interface device by protocol family(%d).", family);
|
|
return -3;
|
|
}
|
|
|
|
sock->netdev = netdev;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int socket_alloc(struct sal_socket_table *st, int f_socket)
|
|
{
|
|
int idx;
|
|
|
|
/* find an empty socket entry */
|
|
for (idx = f_socket; idx < (int) st->max_socket; idx++)
|
|
{
|
|
if (st->sockets[idx] == RT_NULL)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
/* allocate a larger sockte container */
|
|
if (idx == (int) st->max_socket && st->max_socket < SAL_SOCKETS_NUM)
|
|
{
|
|
int cnt, index;
|
|
struct sal_socket **sockets;
|
|
|
|
/* increase the number of socket with 4 step length */
|
|
cnt = st->max_socket + SOCKET_TABLE_STEP_LEN;
|
|
cnt = cnt > SAL_SOCKETS_NUM ? SAL_SOCKETS_NUM : cnt;
|
|
|
|
sockets = rt_realloc(st->sockets, cnt * sizeof(struct sal_socket *));
|
|
if (sockets == RT_NULL)
|
|
goto __result; /* return st->max_socket */
|
|
|
|
/* clean the new allocated fds */
|
|
for (index = st->max_socket; index < cnt; index++)
|
|
{
|
|
sockets[index] = RT_NULL;
|
|
}
|
|
|
|
st->sockets = sockets;
|
|
st->max_socket = cnt;
|
|
}
|
|
|
|
/* allocate 'struct sal_socket' */
|
|
if (idx < (int) st->max_socket && st->sockets[idx] == RT_NULL)
|
|
{
|
|
st->sockets[idx] = rt_calloc(1, sizeof(struct sal_socket));
|
|
if (st->sockets[idx] == RT_NULL)
|
|
{
|
|
idx = st->max_socket;
|
|
}
|
|
}
|
|
|
|
__result:
|
|
return idx;
|
|
}
|
|
|
|
static void socket_free(struct sal_socket_table *st, int idx)
|
|
{
|
|
struct sal_socket *sock;
|
|
|
|
sock = st->sockets[idx];
|
|
st->sockets[idx] = RT_NULL;
|
|
rt_free(sock);
|
|
}
|
|
|
|
static int socket_new(void)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_socket_table *st = &socket_table;
|
|
int idx;
|
|
|
|
sal_lock();
|
|
|
|
/* find an empty sal socket entry */
|
|
idx = socket_alloc(st, 0);
|
|
|
|
/* can't find an empty sal socket entry */
|
|
if (idx == (int) st->max_socket)
|
|
{
|
|
idx = -(1 + SAL_SOCKET_OFFSET);
|
|
goto __result;
|
|
}
|
|
|
|
sock = st->sockets[idx];
|
|
sock->socket = idx + SAL_SOCKET_OFFSET;
|
|
sock->magic = SAL_SOCKET_MAGIC;
|
|
sock->netdev = RT_NULL;
|
|
sock->user_data = RT_NULL;
|
|
#ifdef SAL_USING_TLS
|
|
sock->user_data_tls = RT_NULL;
|
|
#endif
|
|
|
|
__result:
|
|
sal_unlock();
|
|
return idx + SAL_SOCKET_OFFSET;
|
|
}
|
|
|
|
static void socket_delete(int socket)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_socket_table *st = &socket_table;
|
|
int idx;
|
|
|
|
idx = socket - SAL_SOCKET_OFFSET;
|
|
if (idx < 0 || idx >= (int) st->max_socket)
|
|
{
|
|
return;
|
|
}
|
|
sal_lock();
|
|
sock = sal_get_socket(socket);
|
|
RT_ASSERT(sock != RT_NULL);
|
|
sock->magic = 0;
|
|
sock->netdev = RT_NULL;
|
|
socket_free(st, idx);
|
|
sal_unlock();
|
|
}
|
|
|
|
int sal_accept(int socket, struct sockaddr *addr, socklen_t *addrlen)
|
|
{
|
|
int new_socket;
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* check the network interface is up status */
|
|
SAL_NETDEV_IS_UP(sock->netdev);
|
|
|
|
/* check the network interface socket operations */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, accept);
|
|
|
|
new_socket = pf->skt_ops->accept((int)(size_t)sock->user_data, addr, addrlen);
|
|
if (new_socket != -1)
|
|
{
|
|
int retval;
|
|
int new_sal_socket;
|
|
struct sal_socket *new_sock;
|
|
|
|
/* allocate a new socket structure and registered socket options */
|
|
new_sal_socket = socket_new();
|
|
new_sock = sal_get_socket(new_sal_socket);
|
|
if (new_sock == RT_NULL)
|
|
{
|
|
pf->skt_ops->closesocket(new_socket);
|
|
return -1;
|
|
}
|
|
|
|
retval = socket_init(sock->domain, sock->type, sock->protocol, &new_sock);
|
|
if (retval < 0)
|
|
{
|
|
pf->skt_ops->closesocket(new_socket);
|
|
rt_memset(new_sock, 0x00, sizeof(struct sal_socket));
|
|
/* socket init failed, delete socket */
|
|
socket_delete(new_sal_socket);
|
|
LOG_E("New socket registered failed, return error %d.", retval);
|
|
return -1;
|
|
}
|
|
|
|
/* new socket create by accept should have the same netdev with server*/
|
|
new_sock->netdev = sock->netdev;
|
|
/* socket structure user_data used to store the acquired new socket */
|
|
new_sock->user_data = (void *)(size_t)new_socket;
|
|
|
|
return new_sal_socket;
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
static void sal_sockaddr_to_ipaddr(const struct sockaddr *name, ip_addr_t *local_ipaddr)
|
|
{
|
|
const struct sockaddr_in *svr_addr = (const struct sockaddr_in *) name;
|
|
|
|
#if NETDEV_IPV4 && NETDEV_IPV6
|
|
local_ipaddr->u_addr.ip4.addr = svr_addr->sin_addr.s_addr;
|
|
local_ipaddr->type = IPADDR_TYPE_V4;
|
|
#elif NETDEV_IPV4
|
|
local_ipaddr->addr = svr_addr->sin_addr.s_addr;
|
|
#elif NETDEV_IPV6
|
|
#error "not only support IPV6"
|
|
#endif /* NETDEV_IPV4 && NETDEV_IPV6*/
|
|
}
|
|
|
|
int sal_bind(int socket, const struct sockaddr *name, socklen_t namelen)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
ip_addr_t input_ipaddr;
|
|
|
|
RT_ASSERT(name);
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* bind network interface by ip address */
|
|
sal_sockaddr_to_ipaddr(name, &input_ipaddr);
|
|
|
|
/* check input ipaddr is default netdev ipaddr */
|
|
if (!ip_addr_isany_val(input_ipaddr))
|
|
{
|
|
struct sal_proto_family *input_pf = RT_NULL, *local_pf = RT_NULL;
|
|
struct netdev *new_netdev = RT_NULL;
|
|
|
|
new_netdev = netdev_get_by_ipaddr(&input_ipaddr);
|
|
if (new_netdev == RT_NULL)
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
/* get input and local ip address proto_family */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, local_pf, bind);
|
|
SAL_NETDEV_SOCKETOPS_VALID(new_netdev, input_pf, bind);
|
|
|
|
/* check the network interface protocol family type */
|
|
if (input_pf->family != local_pf->family)
|
|
{
|
|
int new_socket = -1;
|
|
|
|
/* protocol family is different, close old socket and create new socket by input ip address */
|
|
local_pf->skt_ops->closesocket(socket);
|
|
|
|
new_socket = input_pf->skt_ops->socket(input_pf->family, sock->type, sock->protocol);
|
|
if (new_socket < 0)
|
|
{
|
|
return -1;
|
|
}
|
|
sock->netdev = new_netdev;
|
|
sock->user_data = (void *)(size_t)new_socket;
|
|
}
|
|
}
|
|
|
|
/* check and get protocol families by the network interface device */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, bind);
|
|
return pf->skt_ops->bind((int)(size_t)sock->user_data, name, namelen);
|
|
}
|
|
|
|
int sal_shutdown(int socket, int how)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
int error = 0;
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* shutdown operation not need to check network interface status */
|
|
/* check the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, shutdown);
|
|
|
|
if (pf->skt_ops->shutdown((int)(size_t)sock->user_data, how) == 0)
|
|
{
|
|
#ifdef SAL_USING_TLS
|
|
if (SAL_SOCKOPS_PROTO_TLS_VALID(sock, closesocket))
|
|
{
|
|
if (proto_tls->ops->closesocket(sock->user_data_tls) < 0)
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
#endif
|
|
error = 0;
|
|
}
|
|
else
|
|
{
|
|
error = -1;
|
|
}
|
|
|
|
|
|
return error;
|
|
}
|
|
|
|
int sal_getpeername(int socket, struct sockaddr *name, socklen_t *namelen)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* check the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, getpeername);
|
|
|
|
return pf->skt_ops->getpeername((int)(size_t)sock->user_data, name, namelen);
|
|
}
|
|
|
|
int sal_getsockname(int socket, struct sockaddr *name, socklen_t *namelen)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
|
|
/* get socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* check the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, getsockname);
|
|
|
|
return pf->skt_ops->getsockname((int)(size_t)sock->user_data, name, namelen);
|
|
}
|
|
|
|
int sal_getsockopt(int socket, int level, int optname, void *optval, socklen_t *optlen)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* check the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, getsockopt);
|
|
|
|
return pf->skt_ops->getsockopt((int)(size_t)sock->user_data, level, optname, optval, optlen);
|
|
}
|
|
|
|
int sal_setsockopt(int socket, int level, int optname, const void *optval, socklen_t optlen)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* check the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, setsockopt);
|
|
|
|
#ifdef SAL_USING_TLS
|
|
if (level == SOL_TLS)
|
|
{
|
|
switch (optname)
|
|
{
|
|
case TLS_CRET_LIST:
|
|
SAL_SOCKOPT_PROTO_TLS_EXEC(sock, set_cret_list, optval, optlen);
|
|
break;
|
|
|
|
case TLS_CIPHERSUITE_LIST:
|
|
SAL_SOCKOPT_PROTO_TLS_EXEC(sock, set_ciphersurite, optval, optlen);
|
|
break;
|
|
|
|
case TLS_PEER_VERIFY:
|
|
SAL_SOCKOPT_PROTO_TLS_EXEC(sock, set_peer_verify, optval, optlen);
|
|
break;
|
|
|
|
case TLS_DTLS_ROLE:
|
|
SAL_SOCKOPT_PROTO_TLS_EXEC(sock, set_dtls_role, optval, optlen);
|
|
break;
|
|
|
|
default:
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
return pf->skt_ops->setsockopt((int) sock->user_data, level, optname, optval, optlen);
|
|
}
|
|
#else
|
|
return pf->skt_ops->setsockopt((int)(size_t)sock->user_data, level, optname, optval, optlen);
|
|
#endif /* SAL_USING_TLS */
|
|
}
|
|
|
|
int sal_connect(int socket, const struct sockaddr *name, socklen_t namelen)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
int ret;
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* check the network interface is up status */
|
|
SAL_NETDEV_IS_UP(sock->netdev);
|
|
/* check the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, connect);
|
|
|
|
ret = pf->skt_ops->connect((int)(size_t)sock->user_data, name, namelen);
|
|
#ifdef SAL_USING_TLS
|
|
if (ret >= 0 && SAL_SOCKOPS_PROTO_TLS_VALID(sock, connect))
|
|
{
|
|
if (proto_tls->ops->connect(sock->user_data_tls) < 0)
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
#endif
|
|
|
|
return ret;
|
|
}
|
|
|
|
int sal_listen(int socket, int backlog)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* check the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, listen);
|
|
|
|
return pf->skt_ops->listen((int)(size_t)sock->user_data, backlog);
|
|
}
|
|
|
|
int sal_recvfrom(int socket, void *mem, size_t len, int flags,
|
|
struct sockaddr *from, socklen_t *fromlen)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* check the network interface is up status */
|
|
SAL_NETDEV_IS_UP(sock->netdev);
|
|
/* check the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, recvfrom);
|
|
|
|
#ifdef SAL_USING_TLS
|
|
if (SAL_SOCKOPS_PROTO_TLS_VALID(sock, recv))
|
|
{
|
|
int ret;
|
|
|
|
if ((ret = proto_tls->ops->recv(sock->user_data_tls, mem, len)) < 0)
|
|
{
|
|
return -1;
|
|
}
|
|
return ret;
|
|
}
|
|
else
|
|
{
|
|
return pf->skt_ops->recvfrom((int)(size_t)sock->user_data, mem, len, flags, from, fromlen);
|
|
}
|
|
#else
|
|
return pf->skt_ops->recvfrom((int)(size_t)sock->user_data, mem, len, flags, from, fromlen);
|
|
#endif
|
|
}
|
|
|
|
int sal_sendto(int socket, const void *dataptr, size_t size, int flags,
|
|
const struct sockaddr *to, socklen_t tolen)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* check the network interface is up status */
|
|
SAL_NETDEV_IS_UP(sock->netdev);
|
|
/* check the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, sendto);
|
|
|
|
#ifdef SAL_USING_TLS
|
|
if (SAL_SOCKOPS_PROTO_TLS_VALID(sock, send))
|
|
{
|
|
int ret;
|
|
|
|
if ((ret = proto_tls->ops->send(sock->user_data_tls, dataptr, size)) < 0)
|
|
{
|
|
return -1;
|
|
}
|
|
return ret;
|
|
}
|
|
else
|
|
{
|
|
return pf->skt_ops->sendto((int) sock->user_data, dataptr, size, flags, to, tolen);
|
|
}
|
|
#else
|
|
return pf->skt_ops->sendto((int)(size_t)sock->user_data, dataptr, size, flags, to, tolen);
|
|
#endif
|
|
}
|
|
|
|
int sal_socket(int domain, int type, int protocol)
|
|
{
|
|
int retval;
|
|
int socket, proto_socket;
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
|
|
/* allocate a new socket and registered socket options */
|
|
socket = socket_new();
|
|
if (socket < 0)
|
|
{
|
|
return -1;
|
|
}
|
|
|
|
/* get sal socket object by socket descriptor */
|
|
sock = sal_get_socket(socket);
|
|
if (sock == RT_NULL)
|
|
{
|
|
socket_delete(socket);
|
|
return -1;
|
|
}
|
|
|
|
/* Initialize sal socket object */
|
|
retval = socket_init(domain, type, protocol, &sock);
|
|
if (retval < 0)
|
|
{
|
|
LOG_E("SAL socket protocol family input failed, return error %d.", retval);
|
|
socket_delete(socket);
|
|
return -1;
|
|
}
|
|
|
|
/* valid the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, socket);
|
|
|
|
proto_socket = pf->skt_ops->socket(domain, type, protocol);
|
|
if (proto_socket >= 0)
|
|
{
|
|
#ifdef SAL_USING_TLS
|
|
if (SAL_SOCKOPS_PROTO_TLS_VALID(sock, socket))
|
|
{
|
|
sock->user_data_tls = proto_tls->ops->socket(socket);
|
|
if (sock->user_data_tls == RT_NULL)
|
|
{
|
|
socket_delete(socket);
|
|
return -1;
|
|
}
|
|
}
|
|
#endif
|
|
sock->user_data = (void *)(size_t)proto_socket;
|
|
return sock->socket;
|
|
}
|
|
socket_delete(socket);
|
|
return -1;
|
|
}
|
|
|
|
int sal_closesocket(int socket)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
int error = 0;
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* clsoesocket operation not need to vaild network interface status */
|
|
/* valid the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, closesocket);
|
|
|
|
if (pf->skt_ops->closesocket((int)(size_t)sock->user_data) == 0)
|
|
{
|
|
#ifdef SAL_USING_TLS
|
|
if (SAL_SOCKOPS_PROTO_TLS_VALID(sock, closesocket))
|
|
{
|
|
if (proto_tls->ops->closesocket(sock->user_data_tls) < 0)
|
|
{
|
|
return -1;
|
|
}
|
|
}
|
|
#endif
|
|
error = 0;
|
|
}
|
|
else
|
|
{
|
|
error = -1;
|
|
}
|
|
|
|
/* delete socket */
|
|
socket_delete(socket);
|
|
|
|
return error;
|
|
}
|
|
|
|
int sal_ioctlsocket(int socket, long cmd, void *arg)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
struct sockaddr_in *addr_in = RT_NULL;
|
|
struct sockaddr *addr = RT_NULL;
|
|
ip_addr_t input_ipaddr;
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* check the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, ioctlsocket);
|
|
|
|
struct sal_ifreq *ifr = (struct sal_ifreq *)arg;
|
|
|
|
if((sock->domain == AF_INET)&&(sock->netdev)&&(ifr != RT_NULL))
|
|
{
|
|
switch (cmd)
|
|
{
|
|
case SIOCGIFADDR:
|
|
addr_in = (struct sockaddr_in *)&(ifr->ifr_ifru.ifru_addr);
|
|
#if NETDEV_IPV4 && NETDEV_IPV6
|
|
addr_in->sin_addr.s_addr = sock->netdev->ip_addr.u_addr.ip4.addr;
|
|
#elif NETDEV_IPV4
|
|
addr_in->sin_addr.s_addr = sock->netdev->ip_addr.addr;
|
|
#elif NETDEV_IPV6
|
|
#error "not only support IPV6"
|
|
#endif /* NETDEV_IPV4 && NETDEV_IPV6*/
|
|
return 0;
|
|
|
|
case SIOCSIFADDR:
|
|
addr = (struct sockaddr *)&(ifr->ifr_ifru.ifru_addr);
|
|
sal_sockaddr_to_ipaddr(addr,&input_ipaddr);
|
|
netdev_set_ipaddr(sock->netdev,&input_ipaddr);
|
|
return 0;
|
|
|
|
case SIOCGIFNETMASK:
|
|
addr_in = (struct sockaddr_in *)&(ifr->ifr_ifru.ifru_netmask);
|
|
#if NETDEV_IPV4 && NETDEV_IPV6
|
|
addr_in->sin_addr.s_addr = sock->netdev->netmask.u_addr.ip4.addr;
|
|
#elif NETDEV_IPV4
|
|
addr_in->sin_addr.s_addr = sock->netdev->netmask.addr;
|
|
#elif NETDEV_IPV6
|
|
#error "not only support IPV6"
|
|
#endif /* NETDEV_IPV4 && NETDEV_IPV6*/
|
|
return 0;
|
|
|
|
case SIOCSIFNETMASK:
|
|
addr = (struct sockaddr *)&(ifr->ifr_ifru.ifru_netmask);
|
|
sal_sockaddr_to_ipaddr(addr,&input_ipaddr);
|
|
netdev_set_netmask(sock->netdev,&input_ipaddr);
|
|
return 0;
|
|
|
|
case SIOCGIFHWADDR:
|
|
addr = (struct sockaddr *)&(ifr->ifr_ifru.ifru_hwaddr);
|
|
rt_memcpy(addr->sa_data,sock->netdev->hwaddr,sock->netdev->hwaddr_len);
|
|
return 0;
|
|
|
|
case SIOCGIFMTU:
|
|
ifr->ifr_ifru.ifru_mtu = sock->netdev->mtu;
|
|
return 0;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
return pf->skt_ops->ioctlsocket((int)(size_t)sock->user_data, cmd, arg);
|
|
}
|
|
|
|
#ifdef SAL_USING_POSIX
|
|
int sal_poll(struct dfs_file *file, struct rt_pollreq *req)
|
|
{
|
|
struct sal_socket *sock;
|
|
struct sal_proto_family *pf;
|
|
int socket = (int)(size_t)file->vnode->data;
|
|
|
|
/* get the socket object by socket descriptor */
|
|
SAL_SOCKET_OBJ_GET(sock, socket);
|
|
|
|
/* check the network interface is up status */
|
|
SAL_NETDEV_IS_UP(sock->netdev);
|
|
/* check the network interface socket opreation */
|
|
SAL_NETDEV_SOCKETOPS_VALID(sock->netdev, pf, poll);
|
|
|
|
return pf->skt_ops->poll(file, req);
|
|
}
|
|
#endif
|
|
|
|
struct hostent *sal_gethostbyname(const char *name)
|
|
{
|
|
struct netdev *netdev = netdev_default;
|
|
struct sal_proto_family *pf;
|
|
|
|
if (SAL_NETDEV_NETDBOPS_VALID(netdev, pf, gethostbyname))
|
|
{
|
|
return pf->netdb_ops->gethostbyname(name);
|
|
}
|
|
else
|
|
{
|
|
/* get the first network interface device with up status */
|
|
netdev = netdev_get_first_by_flags(NETDEV_FLAG_UP);
|
|
if (SAL_NETDEV_NETDBOPS_VALID(netdev, pf, gethostbyname))
|
|
{
|
|
return pf->netdb_ops->gethostbyname(name);
|
|
}
|
|
}
|
|
|
|
return RT_NULL;
|
|
}
|
|
|
|
int sal_gethostbyname_r(const char *name, struct hostent *ret, char *buf,
|
|
size_t buflen, struct hostent **result, int *h_errnop)
|
|
{
|
|
struct netdev *netdev = netdev_default;
|
|
struct sal_proto_family *pf;
|
|
|
|
if (SAL_NETDEV_NETDBOPS_VALID(netdev, pf, gethostbyname_r))
|
|
{
|
|
return pf->netdb_ops->gethostbyname_r(name, ret, buf, buflen, result, h_errnop);
|
|
}
|
|
else
|
|
{
|
|
/* get the first network interface device with up status */
|
|
netdev = netdev_get_first_by_flags(NETDEV_FLAG_UP);
|
|
if (SAL_NETDEV_NETDBOPS_VALID(netdev, pf, gethostbyname_r))
|
|
{
|
|
return pf->netdb_ops->gethostbyname_r(name, ret, buf, buflen, result, h_errnop);
|
|
}
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
int sal_getaddrinfo(const char *nodename,
|
|
const char *servname,
|
|
const struct addrinfo *hints,
|
|
struct addrinfo **res)
|
|
{
|
|
struct netdev *netdev = netdev_default;
|
|
struct sal_proto_family *pf;
|
|
int ret = 0;
|
|
rt_uint32_t i = 0;
|
|
|
|
if (SAL_NETDEV_NETDBOPS_VALID(netdev, pf, getaddrinfo))
|
|
{
|
|
ret = pf->netdb_ops->getaddrinfo(nodename, servname, hints, res);
|
|
}
|
|
else
|
|
{
|
|
/* get the first network interface device with up status */
|
|
netdev = netdev_get_first_by_flags(NETDEV_FLAG_UP);
|
|
if (SAL_NETDEV_NETDBOPS_VALID(netdev, pf, getaddrinfo))
|
|
{
|
|
ret = pf->netdb_ops->getaddrinfo(nodename, servname, hints, res);
|
|
}
|
|
else
|
|
{
|
|
ret = -1;
|
|
}
|
|
}
|
|
|
|
if(ret == RT_EOK)
|
|
{
|
|
/*record the netdev and res*/
|
|
for(i = 0; i < SAL_SOCKETS_NUM; i++)
|
|
{
|
|
if(sal_dev_res_tbl[i].res == RT_NULL)
|
|
{
|
|
sal_dev_res_tbl[i].res = *res;
|
|
sal_dev_res_tbl[i].netdev = netdev;
|
|
break;
|
|
}
|
|
}
|
|
|
|
RT_ASSERT((i < SAL_SOCKETS_NUM));
|
|
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
void sal_freeaddrinfo(struct addrinfo *ai)
|
|
{
|
|
struct netdev *netdev = RT_NULL;
|
|
struct sal_proto_family *pf = RT_NULL;
|
|
rt_uint32_t i = 0;
|
|
|
|
/*when use the multi netdev, it must free the ai use the getaddrinfo netdev */
|
|
for(i = 0; i < SAL_SOCKETS_NUM; i++)
|
|
{
|
|
if(sal_dev_res_tbl[i].res == ai)
|
|
{
|
|
netdev = sal_dev_res_tbl[i].netdev;
|
|
sal_dev_res_tbl[i].res = RT_NULL;
|
|
sal_dev_res_tbl[i].netdev = RT_NULL;
|
|
break;
|
|
}
|
|
}
|
|
RT_ASSERT((i < SAL_SOCKETS_NUM));
|
|
|
|
if (SAL_NETDBOPS_VALID(netdev, pf, freeaddrinfo))
|
|
{
|
|
pf->netdb_ops->freeaddrinfo(ai);
|
|
}
|
|
}
|