socket: handle control data, change recv and send arguments

This commit is contained in:
Matheus
2024-09-01 13:20:12 -03:00
parent fe1102224e
commit 997fbc4e2e
7 changed files with 170 additions and 37 deletions
+1
View File
@@ -88,6 +88,7 @@ typedef struct {
#define SOCK_DGRAM 2
#define SOCK_RAW 3
#define MSG_CTRUNC 0x8
#define MSG_PEEK 2
#define MSG_WAITALL 0x100
#define MSG_NOSIGNAL 0x4000
+44 -2
View File
@@ -29,13 +29,33 @@ typedef struct {
};
} sockaddr_t;
typedef struct {
size_t length;
int level;
int type;
uint8_t data[];
} sockctrl_t;
typedef struct {
sockaddr_t *addr;
void *buffer;
size_t count;
uintmax_t flags;
size_t donecount;
sockctrl_t *ctrl;
size_t ctrllen;
size_t ctrldone;
} sockdesc_t;
#define SOCKET_RECV_FLAGS_PEEK 0x8000000000000000l
#define SOCKET_RECV_FLAGS_WAITALL 0x4000000000000000l
#define SOCKET_RECV_FLAGS_CTRLTRUNCATED 0x2000000000000000l
#define SOCKET_SEND_FLAGS_NOSIGNAL 0x8000000000000000l
typedef struct socketops_t {
int (*bind)(socket_t *socket, sockaddr_t *addr, cred_t *cred);
int (*send)(socket_t *socket, sockaddr_t *addr, void *buffer, size_t count, uintmax_t flags, size_t *sendcount);
int (*recv)(socket_t *socket, sockaddr_t *addr, void *buffer, size_t count, uintmax_t flags, size_t *recvcount);
int (*send)(socket_t *socket, sockdesc_t *desc);
int (*recv)(socket_t *socket, sockdesc_t *desc);
int (*poll)(socket_t *socket, polldata_t *data, int events);
int (*connect)(socket_t *socket, sockaddr_t *addr, uintmax_t data, cred_t *cred);
int (*listen)(socket_t *socket, int backlog);
@@ -155,6 +175,28 @@ static inline void sock_freemsghdr(msghdr_t *hdr) {
free(hdr->msgctrl);
}
#define SOCK_CTRL_NEXT(c) ((sockctrl_t *)((uintptr_t)(c) + (c)->length))
static inline size_t sock_countctrl(sockctrl_t *ctrl, size_t len) {
size_t count = 0;
uintmax_t offset = 0;
while (offset < len) {
if (ctrl->length < sizeof(sockctrl_t))
break;
offset += ctrl->length;
if (offset > len || offset < ctrl->length)
break;
++count;
ctrl = SOCK_CTRL_NEXT(ctrl);
}
return count;
}
void localsock_leavebinding(vnode_t *vnode);
int localsock_pair(socket_t **ret1, socket_t **ret2);
socket_t *localsock_createsocket();