/* * Copyright (c) 2006-2021, RT-Thread Development Team * * SPDX-License-Identifier: Apache-2.0 * * Change Logs: * Date Author Notes * 2015-02-17 Bernard First version * 2018-05-17 ChenYong Add socket abstraction layer */ #include #include #include #include #include #include int accept(int s, struct sockaddr *addr, socklen_t *addrlen) { int new_socket = -1; int socket = dfs_net_getsocket(s); new_socket = sal_accept(socket, addr, addrlen); if (new_socket != -1) { /* this is a new socket, create it in file system fd */ int fd; struct dfs_file *d; /* allocate a fd */ fd = fd_new(); if (fd < 0) { rt_set_errno(-ENOMEM); sal_closesocket(new_socket); return -1; } d = fd_get(fd); if(d) { #ifdef RT_USING_DFS_V2 d->fops = dfs_net_get_fops(); #endif /* this is a socket fd */ d->vnode = (struct dfs_vnode *)rt_malloc(sizeof(struct dfs_vnode)); if (!d->vnode) { /* release fd */ fd_release(fd); rt_set_errno(-ENOMEM); return -1; } rt_memset(d->vnode, 0, sizeof(struct dfs_vnode)); dfs_vnode_init(d->vnode, FT_SOCKET, dfs_net_get_fops()); /* set socket to the data of dfs_file */ d->vnode->data = (void *)(size_t)new_socket; return fd; } rt_set_errno(-ENOMEM); sal_closesocket(new_socket); return -1; } return -1; } RTM_EXPORT(accept); int bind(int s, const struct sockaddr *name, socklen_t namelen) { int socket = dfs_net_getsocket(s); #ifdef SAL_USING_AF_UNIX struct sockaddr_in server_addr = {0}; if (name->sa_family == AF_UNIX) { server_addr.sin_family = AF_INET; server_addr.sin_port = htons(514); server_addr.sin_addr.s_addr = inet_addr("127.0.0.1"); return sal_bind(socket, (struct sockaddr *)&server_addr, namelen); } #endif /* SAL_USING_AF_UNIX */ return sal_bind(socket, name, namelen); } RTM_EXPORT(bind); int shutdown(int s, int how) { int error = 0; int socket = -1; struct dfs_file *d; socket = dfs_net_getsocket(s); if (socket < 0) { rt_set_errno(-ENOTSOCK); return -1; } d = fd_get(s); if (d == NULL) { rt_set_errno(-EBADF); return -1; } if (sal_shutdown(socket, how) == 0) { error = 0; } else { rt_set_errno(-ENOTSOCK); error = -1; } return error; } RTM_EXPORT(shutdown); int getpeername(int s, struct sockaddr *name, socklen_t *namelen) { int socket = dfs_net_getsocket(s); return sal_getpeername(socket, name, namelen); } RTM_EXPORT(getpeername); int getsockname(int s, struct sockaddr *name, socklen_t *namelen) { int socket = dfs_net_getsocket(s); return sal_getsockname(socket, name, namelen); } RTM_EXPORT(getsockname); int getsockopt(int s, int level, int optname, void *optval, socklen_t *optlen) { int socket = dfs_net_getsocket(s); return sal_getsockopt(socket, level, optname, optval, optlen); } RTM_EXPORT(getsockopt); int setsockopt(int s, int level, int optname, const void *optval, socklen_t optlen) { int socket = dfs_net_getsocket(s); return sal_setsockopt(socket, level, optname, optval, optlen); } RTM_EXPORT(setsockopt); int connect(int s, const struct sockaddr *name, socklen_t namelen) { int socket = dfs_net_getsocket(s); #ifdef SAL_USING_AF_UNIX struct sockaddr_in server_addr = {0}; if (name->sa_family == AF_UNIX) { server_addr.sin_family = AF_INET; server_addr.sin_port = htons(514); server_addr.sin_addr.s_addr = inet_addr("127.0.0.1"); return sal_connect(socket, (struct sockaddr *)&server_addr, namelen); } #endif /* SAL_USING_AF_UNIX */ return sal_connect(socket, name, namelen); } RTM_EXPORT(connect); int listen(int s, int backlog) { int socket = dfs_net_getsocket(s); return sal_listen(socket, backlog); } RTM_EXPORT(listen); int recv(int s, void *mem, size_t len, int flags) { int socket = dfs_net_getsocket(s); return sal_recvfrom(socket, mem, len, flags, NULL, NULL); } RTM_EXPORT(recv); int recvfrom(int s, void *mem, size_t len, int flags, struct sockaddr *from, socklen_t *fromlen) { int socket = dfs_net_getsocket(s); return sal_recvfrom(socket, mem, len, flags, from, fromlen); } RTM_EXPORT(recvfrom); int send(int s, const void *dataptr, size_t size, int flags) { int socket = dfs_net_getsocket(s); return sal_sendto(socket, dataptr, size, flags, NULL, 0); } RTM_EXPORT(send); int sendto(int s, const void *dataptr, size_t size, int flags, const struct sockaddr *to, socklen_t tolen) { int socket = dfs_net_getsocket(s); return sal_sendto(socket, dataptr, size, flags, to, tolen); } RTM_EXPORT(sendto); int socket(int domain, int type, int protocol) { /* create a BSD socket */ int fd; int socket; struct dfs_file *d; /* allocate a fd */ fd = fd_new(); if (fd < 0) { rt_set_errno(-ENOMEM); return -1; } d = fd_get(fd); #ifdef RT_USING_DFS_V2 d->fops = dfs_net_get_fops(); #endif d->vnode = (struct dfs_vnode *)rt_malloc(sizeof(struct dfs_vnode)); if (!d->vnode) { /* release fd */ fd_release(fd); rt_set_errno(-ENOMEM); return -1; } #ifdef SAL_USING_AF_UNIX if (domain == AF_UNIX) { domain = AF_INET; } #endif /* SAL_USING_AF_UNIX */ /* create socket and then put it to the dfs_file */ socket = sal_socket(domain, type, protocol); if (socket >= 0) { dfs_vnode_init(d->vnode, FT_SOCKET, dfs_net_get_fops()); /* set socket to the data of dfs_file */ d->vnode->data = (void *)(size_t)socket; } else { rt_free(d->vnode); /* release fd */ fd_release(fd); rt_set_errno(-ENOMEM); return -1; } return fd; } RTM_EXPORT(socket); int closesocket(int s) { int error = 0; int socket = -1; struct dfs_file *d; socket = dfs_net_getsocket(s); if (socket < 0) { rt_set_errno(-ENOTSOCK); return -1; } d = fd_get(s); if (d == RT_NULL) { rt_set_errno(-EBADF); return -1; } if (!d->vnode) { rt_set_errno(-EBADF); return -1; } if (sal_closesocket(socket) == 0) { error = 0; } else { rt_set_errno(-ENOTSOCK); error = -1; } rt_free(d->vnode); /* socket has been closed, delete it from file system fd */ fd_release(s); return error; } RTM_EXPORT(closesocket); int ioctlsocket(int s, long cmd, void *arg) { int socket = dfs_net_getsocket(s); return sal_ioctlsocket(socket, cmd, arg); } RTM_EXPORT(ioctlsocket);