mirrors/Astral
1
0
Fork 0
mirror of https://github.com/Mathewnd/Astral.git synced 2026-10-29 13:48:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

sockets: MSG_WAITALL support

Browse Source
This commit is contained in:
Matheus
2024-08-29 22:48:35 -03:00
parent d3f3ccbb11
commit 066b89c40a
5 changed files with 30 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
kernel-src/sys/syscalls/recvmsg.c
+10 -2
View File
@@ -6,7 +6,7 @@
#include <kernel/file.h>
#include <kernel/alloc.h>
#define KNOWN_FLAGS (MSG_PEEK)
#define KNOWN_FLAGS (MSG_PEEK | MSG_WAITALL)
syscallret_t syscall_recvmsg(context_t *, int fd, msghdr_t *umsghdr, int flags) {
__assert((flags & ~KNOWN_FLAGS) == 0);
@@ -41,7 +41,15 @@ syscallret_t syscall_recvmsg(context_t *, int fd, msghdr_t *umsghdr, int flags)
sockaddr_t sockaddr;
socket_t *socket = SOCKFS_SOCKET_FROM_NODE(file->vnode);
size_t recvcount;
ret.errno = socket->ops->recv(socket, &sockaddr, buffer, buffersize, fileflagstovnodeflags(file->flags) | ((flags & MSG_PEEK) ? SOCKET_RECV_FLAGS_PEEK : 0), &recvcount);
uintmax_t recvflags = 0;
if (flags & MSG_PEEK)
recvflags |= SOCKET_RECV_FLAGS_PEEK;
if (flags & MSG_WAITALL)
recvflags |= SOCKET_RECV_FLAGS_WAITALL;
ret.errno = socket->ops->recv(socket, &sockaddr, buffer, buffersize, fileflagstovnodeflags(file->flags) | recvflags, &recvcount);
ret.ret = ret.errno ? -1 : recvcount;
if (ret.errno)
goto cleanup;