下面先来看两个函数:
#include #include
ssize_t sendmsg(int sockfd, const struct msghdr *msg, int
flags); ssize_t recvmsg(int sockfd, struct msghdr *msg, int
flags);
它们与sendto 和 recvfrom
函数相似,只不过可以传输更复杂的数据结构,不仅可以传输一般数据,还可以传输额外的数据,即文件描述符。下面来看结构体msghdr
:
struct msghdr { void *msg_name;
socklen_t msg_namelen; struct iovec *msg_iov;
size_t msg_iovlen;
void *msg_control; size_t msg_controllen; int msg_flags; };
如下图所示:
1、msg_name :即对等方的地址指针,不关心时设为NULL即可;
2、msg_namelen:地址长度,不关心时设置为0即可;
3、msg_iov:是结构体iovec 的指针。
struct iovec { void
*iov_base; size_t iov_len;
};
成员iov_base 可以认为是传输正常数据时的buf,iov_len 是buf 的大小。
4、msg_iovlen:当有n个iovec 结构体时,此值为n;
5、msg_control:是一个指向cmsghdr 结构体的指针
struct cmsghdr { socklen_t
cmsg_len; int
cmsg_level; int
cmsg_type;
};
6、msg_controllen :参见下图,即cmsghdr 结构体可能不止一个;
7、flags : 一般设置为0即可;
为了对齐,可能存在一些填充字节,跟每个系统的实现有关,但我们不必关心,可以通过一些函数宏来获取相关的值,如下:
#include struct cmsghdr *CMSG_FIRSTHDR(struct msghdr
*msgh); struct cmsghdr *CMSG_NXTHDR(struct msghdr *msgh,
struct cmsghdr *cmsg); size_t CMSG_ALIGN(size_t
length); size_t CMSG_SPACE(size_t
length); size_t CMSG_LEN(size_t
length); unsigned char *CMSG_DATA(struct cmsghdr
*cmsg);
下面通过封装两个函数,send_fd 和 recv_fd 来进一步认识这些函数宏的作用:
C++
Code
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
void send_fd( int sock_fd, int send_fd) { int ret; struct msghdr msg; struct cmsghdr *p_cmsg; struct iovec vec; char cmsgbuf[CMSG_SPACE( sizeof (send_fd))]; int *p_fds; char sendchar = 0 ; msg.msg_control = cmsgbuf; msg.msg_controllen = sizeof (cmsgbuf); p_cmsg = CMSG_FIRSTHDR(&msg); p_cmsg->cmsg_level = SOL_SOCKET; p_cmsg->cmsg_type = SCM_RIGHTS; p_cmsg->cmsg_len = CMSG_LEN( sizeof (send_fd)); p_fds = ( int *)CMSG_DATA(p_cmsg); *p_fds = send_fd; // 通过传递辅助数据的方式传递文件描述符
msg.msg_name = NULL ; msg.msg_namelen = 0 ; msg.msg_iov = &vec; msg.msg_iovlen = 1 ; //主要目的不是传递数据,故只传1个字符 msg.msg_flags = 0 ; vec.iov_base = &sendchar; vec.iov_len = sizeof (sendchar); ret = sendmsg(sock_fd, &msg, 0 ); if (ret != 1 ) ERR_EXIT( "sendmsg" ); } int recv_fd( const