Mirrors_Framework/rt-thread
1
0
Fork 0
mirror of https://github.com/RT-Thread/rt-thread.git synced 2026-05-31 20:05:59 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[net][sal_socket] improve closesocke/shutdown error handling

Browse Source 
Signed-off-by: chenyong <1521761801@qq.com>
This commit is contained in:
chenyong
2019-03-05 19:25:05 +08:00
parent aa7166df01
commit 0d9b7d60db
1 changed files with 44 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
components/net/sal_socket/socket/net_sockets.c
+43 -19
View File
@@ -78,29 +78,39 @@ RTM_EXPORT(bind);
int shutdown(int s, int how) int shutdown(int s, int how)
{ {
int socket; int error = 0;
int socket = -1;
struct dfs_fd *d; struct dfs_fd *d;
socket = dfs_net_getsocket(s);
if (socket < 0)
{
rt_set_errno(-ENOTSOCK);
return -1;
}
d = fd_get(s); d = fd_get(s);
if (d == NULL) if (d == NULL)
{ {
rt_set_errno(-EBADF); rt_set_errno(-EBADF);
return -1; return -1;
} }
socket = dfs_net_getsocket(s);
if (sal_shutdown(socket, how) == 0) if (sal_shutdown(socket, how) == 0)
{ {
/* socket has been closed, delete it from file system fd */ error = 0;
fd_put(d); }
fd_put(d); else
{
return 0; rt_set_errno(-ENOTSOCK);
error = -1;
} }
return -1; /* socket has been closed, delete it from file system fd */
fd_put(d);
fd_put(d);
return error;
} }
RTM_EXPORT(shutdown); RTM_EXPORT(shutdown);
@@ -228,7 +238,7 @@ int socket(int domain, int type, int protocol)
rt_set_errno(-ENOMEM); rt_set_errno(-ENOMEM);
return -1; return -1;
} }
/* release the ref-count of fd */ /* release the ref-count of fd */
@@ -240,25 +250,39 @@ RTM_EXPORT(socket);
int closesocket(int s) int closesocket(int s)
{ {
int socket = dfs_net_getsocket(s); int error = 0;
int socket = -1;
struct dfs_fd *d; struct dfs_fd *d;
d = fd_get(s); socket = dfs_net_getsocket(s);
if(!d) if (socket < 0)
{ {
rt_set_errno(-ENOTSOCK);
return -1;
}
d = fd_get(s);
if (d == RT_NULL)
{
rt_set_errno(-EBADF);
return -1; return -1;
} }
if (sal_closesocket(socket) == 0) if (sal_closesocket(socket) == 0)
{ {
/* socket has been closed, delete it from file system fd */ error = 0;
fd_put(d); }
fd_put(d); else
{
return 0; rt_set_errno(-ENOTSOCK);
error = -1;
} }
return -1; /* socket has been closed, delete it from file system fd */
fd_put(d);
fd_put(d);
return error;
} }
RTM_EXPORT(closesocket); RTM_EXPORT(closesocket);