linux 管道传递socket套接字

在前面我们介绍了UNIX域套接字编程，更重要的一点是UNIX域套接字可以在同一台主机上各进程之间传递文件描述符。

下面先来看两个函数：

#include <sys/types.h>
#include <sys/socket.h>

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;       /* optional address */
              socklen_t     msg_namelen;    /* size of address */
              struct iovec *msg_iov;        /* scatter/gather array */
              size_t        msg_iovlen;     /* # elements in msg_iov */
              void         *msg_control;    /* ancillary data, see below */
              size_t        msg_controllen; /* ancillary data buffer len */
              int           msg_flags;      /* flags on received message */
          };

如下图所示：

1、msg_name ：即对等方的地址指针，不关心时设为NULL即可；

2、msg_namelen：地址长度，不关心时设置为0即可；

3、msg_iov：是结构体iovec 的指针。  

struct iovec {
              void  *iov_base;    /* Starting address */
              size_t iov_len;     /* Number of bytes to transfer */
          };

成员iov_base 可以认为是传输正常数据时的buf，iov_len 是buf 的大小。

4、msg_iovlen：当有n个iovec 结构体时，此值为n；

5、msg_control：是一个指向cmsghdr 结构体的指针

 struct cmsghdr {
          socklen_t cmsg_len;    /* data byte count, including header */
          int       cmsg_level;  /* originating protocol */
          int       cmsg_type;   /* protocol-specific type */
          /* followed by unsigned char cmsg_data[]; */
      };

6、msg_controllen ：参见下图，即cmsghdr 结构体可能不止一个；

7、flags ： 一般设置为0即可；

为了对齐，可能存在一些填充字节，跟每个系统的实现有关，但我们不必关心，可以通过一些函数宏来获取相关的值，如下：

#include <sys/socket.h>

      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 来进一步认识这些函数宏的作用：

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 int sock_fd) {      int ret;      struct msghdr msg;      char recvchar;      struct iovec vec;      int recv_fd;      char cmsgbuf[CMSG_SPACE( sizeof (recv_fd))];      struct cmsghdr *p_cmsg;      int *p_fd;      vec.iov_base = &recvchar;      vec.iov_len = sizeof (recvchar);      msg.msg_name = NULL;      msg.msg_namelen = 0;      msg.msg_iov = &vec;      msg.msg_iovlen = 1;      msg.msg_control = cmsgbuf;      msg.msg_controllen = sizeof (cmsgbuf);      msg.msg_flags = 0;             p_fd = ( int *)CMSG_DATA(CMSG_FIRSTHDR(&msg));      *p_fd = -1;      ret = recvmsg(sock_fd, &msg, 0);      if (ret != 1)          ERR_EXIT( "recvmsg" );             p_cmsg = CMSG_FIRSTHDR(&msg);      if (p_cmsg == NULL)          ERR_EXIT( "no passed fd" );                    p_fd = ( int *)CMSG_DATA(p_cmsg);      recv_fd = *p_fd;      if (recv_fd == -1)          ERR_EXIT( "no passed fd" );             return recv_fd; }

/*************************************************************************
    > File Name: uxdomsock_sendfd.c
    > Author: Simba
    > Mail: dameng34@163.com
    > Created Time: Mon 04 Mar 2013 02:01:33 PM CST
 ************************************************************************/
     
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <fcntl.h>
     
#define ERR_EXIT(m) 
        do
        { 
                perror(m); 
                exit(EXIT_FAILURE); 
        } while(0)
     
int main(void)
{
    int sockfds[2];
    /* 只有unix域协议才能在进程间传递文件描述符，如果想要在没有亲缘关系的进程间
     * 传递，则不能用socketpair函数，要用socket()函数 */
    if (socketpair(PF_UNIX, SOCK_STREAM, 0, sockfds) < 0)
        ERR_EXIT("socketpair");
     
    pid_t pid;
    pid = fork();
    if (pid == -1)
        ERR_EXIT("fork");
    /* 如果是父进程打开的文件描述符，子进程可以共享
     * 这里演示的是子进程打开的文件描述符通过封装的函数传给父进程 */
    if (pid > 0)
    {
        close(sockfds[1]);
        int fd = recv_fd(sockfds[0]);
        char buf[1024] = {0};
        read(fd, buf, sizeof(buf));
        printf("buf=%sn", buf);
    }
    else if (pid == 0)
    {
        close(sockfds[0]);
        int fd;
        fd = open("test.txt", O_RDONLY);
        if (fd == -1);
        send_fd(sockfds[1], fd);
    }
    return 0;
}
我们知道，父进程在fork 之前打开的文件描述符，子进程是可以共享的，但是子进程打开的文件描述符，父进程是不能共享的，上述程序就是举例在子进程中打开了一个文件描述符，然后通过send_fd 函数将文件描述符传递给父进程，父进程可以通过recv_fd 函数接收到这个文件描述符。先建立一个文件test.txt 后输入几个字符，然后运行程序，输出如下：
simba@ubuntu:~/Documents/code/linux_programming/UNP/socket$ ./uxdomsock_sendfd 
buf=ilove
证明父进程确实可以打开test.txt 文件。
最后提醒一点，只有unix域协议才能在本机进程间传递文件描述符，如果想要在没有亲缘关系的进程间传递，则不能用socketpair函数，要用socket()函数 才行。