进程间传递描述符二

发送、接收描述符

发送描述符

经过了前面的准备工作&＃xff0c;是时候发送描述符了&＃xff0c;先来看看函数原型&＃xff1a;

int write_fd(int fd, void *ptr, int nbytes, int sendfd);

参数说明如下&＃xff1a;

&＃64;fd &＃xff1a;发送 TCP 套接字接口&＃xff1b;这个可以是使用socketpair返回的发送套接字接口

&＃64;ptr &＃xff1a;发送数据的缓冲区指针&＃xff1b;

&＃64;nbytes &＃xff1a;发送的字节数&＃xff1b;

&＃64;sendfd &＃xff1a;向接收进程发送的描述符&＃xff1b;

函数返回值为写入的字节数&＃xff0c; <0 说明发送失败&＃xff1b;

废话少说&＃xff0c;代码先上&＃xff0c;发送描述符的代码相对简单一些&＃xff0c;说明见代码内注释。

先说明一下&＃xff0c;旧的 Unix 系统使用的是 msg_accrights 域来传递描述符&＃xff0c;因此我们需要使用宏 HAVE_MSGHDR_MSG_CONTROL 以期能同时支持这两种版本。

int write_fd(int fd, void *ptr, int nbytes, int sendfd)<br>{<br>struct msghdr msg;<br>struct iovec iov[1];<br>// 有些系统使用的是旧的msg_accrights域来传递描述符&＃xff0c;Linux下是新的msg_control字段<br>#ifdef HAVE_MSGHDR_MSG_CONTROL<br>union{ // 前面说过&＃xff0c;保证cmsghdr和msg_control的对齐<br>struct cmsghdr cm;<br>char control[CMSG_SPACE(sizeof(int))];<br>}control_un;<br>struct cmsghdr *cmptr; <br>// 设置辅助缓冲区和长度<br>msg.msg_control &＃61; control_un.control; <br>msg.msg_controllen &＃61; sizeof(control_un.control);<br>// 只需要一组附属数据就够了&＃xff0c;直接通过CMSG_FIRSTHDR取得<br>cmptr &＃61; CMSG_FIRSTHDR(&msg);<br>// 设置必要的字段&＃xff0c;数据和长度<br>cmptr->cmsg_len &＃61; CMSG_LEN(sizeof(int)); // fd类型是int&＃xff0c;设置长度<br>cmptr->cmsg_level &＃61; SOL_SOCKET; <br>cmptr->cmsg_type &＃61; SCM_RIGHTS; // 指明发送的是描述符<br>*((int*)CMSG_DATA(cmptr)) &＃61; sendfd; // 把fd写入辅助数据中<br>#else<br>msg.msg_accrights &＃61; (caddr_t)&sendfd; // 这个旧的更方便啊<br>msg.msg_accrightslen &＃61; sizeof(int);<br>#endif<br>// UDP才需要&＃xff0c;无视<br>msg.msg_name &＃61; NULL;<br>msg.msg_namelen &＃61; 0;<br>// 别忘了设置数据缓冲区&＃xff0c;实际上1个字节就够了<br>iov[0].iov_base &＃61; ptr;<br>iov[0].iov_len &＃61; nbytes;<br>msg.msg_iov &＃61; iov;<br>msg.msg_iovlen &＃61; 1;<br>return sendmsg(fd, &msg, 0);<br>} 

  接收描述符

发送方准备好之后&＃xff0c;接收方准备接收&＃xff0c;函数原型为&＃xff1a;

int read_fd(int fd, void *ptr, int nbytes, int *recvfd);

参数说明如下&＃xff1a;

&＃64;fd &＃xff1a;接收 TCP 套接字接口&＃xff1b; 这个可以是使用 socketpair返回的接收套接字接口

&＃64;ptr &＃xff1a;接收数据的缓冲区指针&＃xff1b;

&＃64;nbytes &＃xff1a;接收缓冲区大小&＃xff1b;

&＃64;recvfd &＃xff1a;用来接收发送进程发送来的描述符&＃xff1b;

函数返回值为读取的字节数&＃xff0c; <0 说明读取失败&＃xff1b;

接收函数代码如下&＃xff0c;相比发送要复杂一些。

int read_fd(int fd, void *ptr, int nbytes, int *recvfd)<br>{<br>struct msghdr msg;<br>struct iovec iov[1];<br>int n;<br>int newfd;<br>#ifdef HAVE_MSGHDR_MSG_CONTROL<br>union{ // 对齐<br>struct cmsghdr cm;<br>char control[CMSG_SPACE(sizeof(int))];<br>}control_un;<br>struct cmsghdr *cmptr;<br>// 设置辅助数据缓冲区和长度<br>msg.msg_control &＃61; control_un.control;<br>msg.msg_controllen &＃61; sizeof(control_un.control);<br>#else<br>msg.msg_accrights &＃61; (caddr_t) &newfd; // 这个简单<br>msg.msg_accrightslen &＃61; sizeof(int);<br>#endif <br>// TCP无视<br>msg.msg_name &＃61; NULL;<br>msg.msg_namelen &＃61; 0;<br>// 设置数据缓冲区<br>iov[0].iov_base &＃61; ptr;<br>iov[0].iov_len &＃61; nbytes;<br>msg.msg_iov &＃61; iov;<br>msg.msg_iovlen &＃61; 1;<br>// 设置结束&＃xff0c;准备接收<br>if((n &＃61; recvmsg(fd, &msg, 0)) <&＃61; 0)<br>{<br>return n;<br>}<br>#ifdef HAVE_MSGHDR_MSG_CONTROL<br>// 检查是否收到了辅助数据&＃xff0c;以及长度&＃xff0c;回忆上一节的CMSG宏<br>cmptr &＃61; CMSG_FIRSTHDR(&msg);<br>if((cmptr !&＃61; NULL) && (cmptr->cmsg_len &＃61;&＃61; CMSG_LEN(sizeof(int))))<br>{<br>// 还是必要的检查<br>if(cmptr->cmsg_level !&＃61; SOL_SOCKET)<br>{<br>printf("control level !&＃61; SOL_SOCKET/n");<br>exit(-1);<br>}<br>if(cmptr->cmsg_type !&＃61; SCM_RIGHTS)<br>{<br>printf("control type !&＃61; SCM_RIGHTS/n");<br>exit(-1);<br>}<br>// 好了&＃xff0c;描述符在这<br>*recvfd &＃61; *((int*)CMSG_DATA(cmptr));<br>}<br>else<br>{<br>if(cmptr &＃61;&＃61; NULL) printf("null cmptr, fd not passed./n");<br>else printf("message len[%d] if incorrect./n", cmptr->cmsg_len);<br>*recvfd &＃61; -1; // descriptor was not passed<br>}<br>#else<br>if(msg.msg_accrightslen &＃61;&＃61; sizeof(int)) *recvfd &＃61; newfd; <br>else *recvfd &＃61; -1;<br>#endif<br>return n;<br>}<br>

发送和接收函数就这么多&＃xff0c;就像上面看到的&＃xff0c;进程间传递套接字还是有点麻烦的。Linux的就介绍完了&＃xff0c;后面在简单说说Windows是如何传递的&＃xff0c;话说MSDN真的方便哪。