2021-01-05

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

       下面先来看两个函数:


    
    
  1. #include <sys/types.h>
  2. #include <sys/socket.h>
  3. ssize_t sendmsg(int sockfd, const struct msghdr *msg, int flags);
  4. ssize_t recvmsg(int sockfd, struct msghdr *msg, int flags);
      它们与sendto 和 recvfrom 函数相似,只不过可以传输更复杂的数据结构,不仅可以传输一般数据,还可以传输额外的数据,即文件描述符。下面来看结构体msghdr :


    
    
  1. struct msghdr {
  2. void *msg_name; /* optional address */
  3. socklen_t msg_namelen; /* size of address */
  4. struct iovec *msg_iov; /* scatter/gather array */
  5. size_t msg_iovlen; /* # elements in msg_iov */
  6. void *msg_control; /* ancillary data, see below */
  7. size_t msg_controllen; /* ancillary data buffer len */
  8. int msg_flags; /* flags on received message */
  9. };
       如下图所示:


                                            

1、msg_name :即对等方的地址指针,不关心时设为NULL即可;

2、msg_namelen:地址长度,不关心时设置为0即可;

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


    
    
  1. struct iovec {
  2. void *iov_base; /* Starting address */
  3. size_t iov_len; /* Number of bytes to transfer */
  4. };

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

4、msg_iovlen:当有n个iovec 结构体时,此值为n;

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


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

6、msg_controllen :参见下图,即cmsghdr 结构体可能不止一个;

7、flags : 一般设置为0即可;

                          

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


    
    
  1. #include <sys/socket.h>
  2. struct cmsghdr *CMSG_FIRSTHDR(struct msghdr *msgh);
  3. struct cmsghdr *CMSG_NXTHDR(struct msghdr *msgh, struct cmsghdr *cmsg);
  4. size_t CMSG_ALIGN(size_t length);
  5. size_t CMSG_SPACE(size_t length);
  6. size_t CMSG_LEN(size_t length);
  7. unsigned char *CMSG_DATA(struct cmsghdr *cmsg);
下面通过封装两个函数,send_fd 和 recv_fd 来进一步认识这些函数宏的作用:

    
    
  1. void send_fd(int sock_fd, int send_fd)
  2. {
  3. int ret;
  4. struct msghdr msg;
  5. struct cmsghdr *p_cmsg;
  6. struct iovec vec;
  7. char cmsgbuf[CMSG_SPACE( sizeof(send_fd))];
  8. int *p_fds;
  9. char sendchar = 0;
  10. msg.msg_control = cmsgbuf;
  11. msg.msg_controllen = sizeof(cmsgbuf);
  12. p_cmsg = CMSG_FIRSTHDR(&msg);
  13. p_cmsg->cmsg_level = SOL_SOCKET;
  14. p_cmsg->cmsg_type = SCM_RIGHTS;
  15. p_cmsg->cmsg_len = CMSG_LEN( sizeof(send_fd));
  16. p_fds = ( int *)CMSG_DATA(p_cmsg);
  17. *p_fds = send_fd; // 通过传递辅助数据的方式传递文件描述符
  18. msg.msg_name = NULL;
  19. msg.msg_namelen = 0;
  20. msg.msg_iov = &vec;
  21. msg.msg_iovlen = 1; //主要目的不是传递数据,故只传1个字符
  22. msg.msg_flags = 0;
  23. vec.iov_base = &sendchar;
  24. vec.iov_len = sizeof(sendchar);
  25. ret = sendmsg(sock_fd, &msg, 0);
  26. if (ret != 1)
  27. ERR_EXIT( "sendmsg");
  28. }
  29. int recv_fd(const int sock_fd)
  30. {
  31. int ret;
  32. struct msghdr msg;
  33. char recvchar;
  34. struct iovec vec;
  35. int recv_fd;
  36. char cmsgbuf[CMSG_SPACE( sizeof(recv_fd))];
  37. struct cmsghdr *p_cmsg;
  38. int *p_fd;
  39. vec.iov_base = &recvchar;
  40. vec.iov_len = sizeof(recvchar);
  41. msg.msg_name = NULL;
  42. msg.msg_namelen = 0;
  43. msg.msg_iov = &vec;
  44. msg.msg_iovlen = 1;
  45. msg.msg_control = cmsgbuf;
  46. msg.msg_controllen = sizeof(cmsgbuf);
  47. msg.msg_flags = 0;
  48. p_fd = ( int *)CMSG_DATA(CMSG_FIRSTHDR(&msg));
  49. *p_fd = -1;
  50. ret = recvmsg(sock_fd, &msg, 0);
  51. if (ret != 1)
  52. ERR_EXIT( "recvmsg");
  53. p_cmsg = CMSG_FIRSTHDR(&msg);
  54. if (p_cmsg == NULL)
  55. ERR_EXIT( "no passed fd");
  56. p_fd = ( int *)CMSG_DATA(p_cmsg);
  57. recv_fd = *p_fd;
  58. if (recv_fd == -1)
  59. ERR_EXIT( "no passed fd");
  60. return recv_fd;
  61. }
来解释一下send_fd 函数:


    
    
  1. msg.msg_name = NULL;
  2. msg.msg_namelen = 0;
  3. msg.msg_iov = &vec;
  4. msg.msg_iovlen = 1; //主要目的不是传递数据,故只传1个字符
  5. msg.msg_flags = 0;
  6. vec.iov_base = &sendchar;
  7. vec.iov_len = sizeof(sendchar);

       这几行中需要注意的是我们现在的目的不是传输正常数据,而是为了传递文件描述符,所以只定义一个1字节的char,其余参照前面对参数的解释可以理解。

       现在我们只有一个cmsghdr 结构体,把需要传递的文件描述符send_fd 长度,也就是需要传输的额外数据大小,当作参数传给CMSG_SPACE 宏,可以得到整个结构体的大小,包括一些填充字节,如上图所示,也即 

char cmsgbuf[CMSG_SPACE(sizeof(send_fd))]; 
    
    
也就可以进一步得出以下两行:


    
    
  1. msg.msg_control = cmsgbuf;
  2. msg.msg_controllen = sizeof(cmsgbuf);
接着,需要填充cmsghdr 结构体,传入msghdr 指针,CMSG_FIRSTHDR宏可以得到首个cmsghdr 结构体的指针,即

p_cmsg = CMSG_FIRSTHDR(&msg);
    
    
然后使用指针来填充各字段,如下:


    
    
  1. p_cmsg->cmsg_level = SOL_SOCKET;
  2. p_cmsg->cmsg_type = SCM_RIGHTS;
  3. p_cmsg->cmsg_len = CMSG_LEN( sizeof(send_fd));

传入send_fd 的大小,CMSG_LEN宏可以得到cmsg_len 字段的大小。

最后,传入结构体指针 p_cmsg ,宏CMSG_DATA 可以得到准备存放send_fd 的位置指针,将send_fd 放进去,如下:


    
    
  1. p_fds = ( int*)CMSG_DATA(p_cmsg);
  2. *p_fds = send_fd; // 通过传递辅助数据的方式传递文件描述符

recv_fd 函数就类似了,不再赘述。


可以写个小程序测试一下:


    
    
  1. #include <unistd.h>
  2. #include <sys/types.h>
  3. #include <sys/socket.h>
  4. #include <sys/un.h>
  5. #include <stdlib.h>
  6. #include <stdio.h>
  7. #include <errno.h>
  8. #include <string.h>
  9. #include <fcntl.h>
  10. #define ERR_EXIT(m) \
  11. do \
  12. { \
  13. perror(m); \
  14. exit(EXIT_FAILURE); \
  15. } while(0)
  16. int main(void)
  17. {
  18. int sockfds[ 2];
  19. /* 只有unix域协议才能在进程间传递文件描述符,如果想要在没有亲缘关系的进程间
  20. * 传递,则不能用socketpair函数,要用socket()函数 */
  21. if (socketpair(PF_UNIX, SOCK_STREAM, 0, sockfds) < 0)
  22. ERR_EXIT( "socketpair");
  23. pid_t pid;
  24. pid = fork();
  25. if (pid == -1)
  26. ERR_EXIT( "fork");
  27. /* 如果是父进程打开的文件描述符,子进程可以共享
  28. * 这里演示的是子进程打开的文件描述符通过封装的函数传给父进程 */
  29. if (pid > 0)
  30. {
  31. close(sockfds[ 1]);
  32. int fd = recv_fd(sockfds[ 0]);
  33. char buf[ 1024] = { 0};
  34. read(fd, buf, sizeof(buf));
  35. printf( "buf=%s\n", buf);
  36. }
  37. else if (pid == 0)
  38. {
  39. close(sockfds[ 0]);
  40. int fd;
  41. fd = open( "test.txt", O_RDONLY);
  42. if (fd == -1);
  43. send_fd(sockfds[ 1], fd);
  44. }
  45. return 0;
  46. }
我们知道,父进程在fork 之前打开的文件描述符,子进程是可以共享的,但是子进程打开的文件描述符,父进程是不能共享的,上述程序就是举例在子进程中打开了一个文件描述符,然后通过send_fd 函数将文件描述符传递给父进程,父进程可以通过recv_fd 函数接收到这个文件描述符。先建立一个文件test.txt 后输入几个字符,然后运行程序,输出如下:


    
    
  1. huangcheng@ubuntu:~$ cat test.txt
  2. ctthuangcheng
  3. huangcheng@ubuntu:~$ ./a.out
  4. buf=ctthuangcheng
  5. huangcheng@ubuntu:~$

证明父进程确实可以打开test.txt 文件。
最后提醒一点,只有unix域协议才能在本机进程间传递文件描述符,如果想要在没有亲缘关系的进程间传递,则不能用socketpair函数,要用socket()函数 才行。

转载于:https://www.cnblogs.com/hehehaha/p/6332479.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值