aboutsummaryrefslogtreecommitdiff
path: root/src/lib9/sendfd.c
blob: b3a2448ff895e19894227093fd5685d641f1a47a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
#include <u.h>
#define NOPLAN9DEFINES
#include <libc.h>
#include <sys/socket.h>
#include <sys/uio.h>
#include <unistd.h>
#include <errno.h>

typedef struct Sendfd Sendfd;
struct Sendfd {
	struct cmsghdr cmsg;
	int fd;
};

int
sendfd(int s, int fd)
{
	char buf[1];
	struct iovec iov;
	struct msghdr msg;
	int n;
	Sendfd sfd;

	buf[0] = 0;
	iov.iov_base = buf;
	iov.iov_len = 1;

	memset(&msg, 0, sizeof msg);
	msg.msg_iov = &iov;
	msg.msg_iovlen = 1;

	sfd.cmsg.cmsg_len = sizeof sfd;
	sfd.cmsg.cmsg_level = SOL_SOCKET;
	sfd.cmsg.cmsg_type = SCM_RIGHTS;
	sfd.fd = fd;

	msg.msg_control = &sfd;
	msg.msg_controllen = sizeof sfd;

	if((n=sendmsg(s, &msg, 0)) != iov.iov_len)
		return -1;
	return 0;
}

int
recvfd(int s)
{
	int n;
	char buf[1];
	struct iovec iov;
	struct msghdr msg;
	Sendfd sfd;

	iov.iov_base = buf;
	iov.iov_len = 1;

	memset(&msg, 0, sizeof msg);
	msg.msg_name = 0;
	msg.msg_namelen = 0;
	msg.msg_iov = &iov;
	msg.msg_iovlen = 1;

	memset(&sfd, 0, sizeof sfd);
	sfd.fd = -1;
	sfd.cmsg.cmsg_len = sizeof sfd;
	sfd.cmsg.cmsg_level = SOL_SOCKET;
	sfd.cmsg.cmsg_type = SCM_RIGHTS;

	msg.msg_control = &sfd;
	msg.msg_controllen = sizeof sfd;

	if((n=recvmsg(s, &msg, 0)) < 0)
		return -1;
	if(n==0 && sfd.fd==-1){
		werrstr("eof in recvfd");
		return -1;
	}
	return sfd.fd;
}