nate/swebs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed frequent segfault with children

Browse Source
This commit is contained in:
Nate Choe
2022-04-03 20:42:13 -05:00
parent 12281a030d
commit 71482861ae
3 changed files with 35 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
src/util.c
View File

@@ -129,17 +129,39 @@ void sendFd(int fd, int dest) {
}
int recvFd(int source) {
union {
char buff[CMSG_SPACE(sizeof(int))];
struct cmsghdr align;
} cmsghdr;
struct msghdr msg;
struct cmsghdr *cmsg;
char cmsgbuf[CMSG_SPACE(sizeof(int))];
unsigned char *data;
struct iovec iov;
int data;
ssize_t nr;
int ret;
memset(&msg, 0, sizeof(msg));
msg.msg_control = cmsgbuf;
msg.msg_controllen = sizeof(cmsgbuf);
recvmsg(source, &msg, 0);
msg.msg_name = NULL;
msg.msg_namelen = 0;
iov.iov_base = &data;
iov.iov_len = sizeof(data);
msg.msg_iov = &iov;
msg.msg_iovlen = 1;
msg.msg_control = cmsghdr.buff;
msg.msg_controllen = sizeof(cmsghdr.buff);
nr = recvmsg(source, &msg, 0);
if (nr < 0)
return -1;
cmsg = CMSG_FIRSTHDR(&msg);
data = CMSG_DATA(cmsg);
memcpy(&ret, data, sizeof(ret));
if (cmsg == NULL || cmsg->cmsg_len != CMSG_LEN(sizeof(data)))
return -1;
if (cmsg->cmsg_level != SOL_SOCKET)
return -1;
if (cmsg->cmsg_type != SCM_RIGHTS)
return -1;
memcpy(&ret, CMSG_DATA(cmsg), sizeof(ret));
return ret;
}