高级套接口-(sendmsg和recvmsg)

转载自：http://www.tuicool.com/articles/Yre2Un

已经工作了接近一年的时间，工作之余也只能看看书，了解一下相关的技术细节，在网络设备公司不可避免的要和socket打交道，但通常都是调用公司封装好的接口，没有去考虑这些封装背后的工作，回过头来看真的觉得进步很小，我只能逼自己看看书，看看一些好的代码。 
sendmsg和recvmsg这两个接口是高级套接口，这两个接口支持一般数据的发送和接收，还支持多缓冲区的报文发送和接收(readv和sendv支持多缓冲区发送和接收)，还可以在报文中带辅助数据。这些功能是常用的send、recv等接口无法完成的。 
接口的声明如下： 

点击( 此处 )折叠或打开

1. #include  < sys / socket . h > 
   
2. 
   
3. ssize_t recvmsg ( int  sockfd ,  struct msghdr  * msg ,   int  flags ) ; 
   
4. ssize_t sendmsg ( int  sockfd ,  struct msghdr  * msg ,   int  flags ) ;

上述接口的参数分别是套接字描述符，消息的头部，已经对应的标识，这些标识主要用于对该套接口进行设置，如：MSG_DONTWAI(将本操作设置为非阻塞模式)，MSG_OOB(发送或接受带外数据)等。

在该接口的声明中包含了在其他发送接收函数中不用使用的接口体struct msghdr，该结构体是一个消息的头部结构体。 

点击( 此处 )折叠或打开

1. struct msghdr  { 
   
2.     void  * msg_name ;   / *  消息的协议地址  * / 
   
3.     socklen_t msg_namelen ;   / *  地址的长度  * / 
   
4.     struct iovec  * msg_iov ;   / *  多io缓冲区的地址  * / 
   
5.      int  msg_iovlen ;   / *  缓冲区的个数  * / 
   
6.     void  * msg_control ;   / *  辅助数据的地址  * / 
   
7.     socklen_t msg_controllen ;   / *  辅助数据的长度  * / 
   
8.      int  msg_flags ;   / *  接收消息的标识  * / 
   
9. } ;

其中的前两个成员主要用于保存当前使用的协议的地址，比如使用了tcp协议、udp协议、UNIX domain协议等，每种地址都存在一定的差异，比如unix domain的地址就是(AF_UNIX, file_path)。这样就使得该接口更加的通用，针对各种类型的协议都是有效的。 
接下来的两个成员是关于接受和发送数据的的。其中的strcut iovec是io向量，如下所示: 

点击( 此处 )折叠或打开

1. struct iovec  { 
   
2.     void  * io_base ;   / *  buffer空间的基地址  * / 
   
3.     size_t iov_len ;   / *  该buffer空间的长度  * / 
   
4. } ;

多缓冲区的发送和接收处理就是一个struct iovec的数组，每个成员的io_base都指向了不同的buffer的地址。io_len是指该buffer中的数据长度。而在struct msghdr中的msg_iovlen是指buffer缓冲区的个数，即iovec数组的长度。 
msg_control字段的也是指向一段内存，msg_controllen是指该内存的总大小长度，通常该内存被用来存储辅助数据，辅助数据可用于一些特殊的处理。msg_control通常指向一个控制消息头部，其结构体如下所示: 

点击( 此处 )折叠或打开

1. struct cmsghdr  { 
   
2.     socklen_t cmsg_len ;   / *  包含该头部的数据长度  * / 
   
3.      int  cmsg_level ;   / *  具体的协议标识  * / 
   
4.      int  cmsg_type ;   / *  协议中的类型  * / 
   
5. } ; 
   
6. 
   
7. 其中的cmsg_level主要包含IPPROTO_IP ( ipv ) ,  IPPROTO_IPV6 ( ipv6 ) , SOL_SOCKET ( unix domain ) .
8. 其中的cmsg_type是根据上述的类型有分别有不同的内容，比如SOL_SOCKET中主要包含:SCM_RIGHTS(发送接收描述字)， SCM_CREDS(发送接收用户凭证)

该cmsghdr是辅助数据的数据头部，类似一个tlv的封装形式。关于辅助数据的功能在最后采用进程间传递描述字的代码来说明。 
关于辅助数据，在recvmsg的msg_control中可能携带多个cmsghdr, 可以采用对应的宏协助处理: CMSG_FIRSTHDR(), CMSG_NXTHDR(), CMSG_DATA()， CMSG_LEN()， CMSG_SPACE(). 
最后的字段是msg_flags，该字段主要用于在接收消息的过程中用来获取消息的相关属性。 
接下来采用进程间传递描述字的例子来说明上述函数的使用。 
描述字的传递，就是将一个进程中的描述字传递到另一个进程中，使得该描述字依然有效，也就是使得在一个进程中的描述字传递到另一个描述字依然有效。 
在多进程的网络CS模式下，服务器fork产生的子进程在fork调用返回后，子进程共享父进程的所有打开的描述字。即使在子进程中调用exec函数，所有描述字通常还是保持打开的状态，也就是描述字是跨exec函数的。这也是为什么在exec只有的子进程仍然可以调用父进程共享的套接字的原因。 
但是这种实现并不能解决子进程的描述字传递给父进程的需求，对于无亲缘关系的进程之间传递描述字就更加不可能，在unix中的实现是：在两个进程之间创建一个unix domain socket套接口，然后调用sendmsg跨这个套接口发送一个特殊的消息，该消息由内核进行特殊的处理，从而把打开的描述字从发送进程传递到接收进程（采用recvmsg接收）。分析主要包含如下几个过程: 
(1) 创建一个字节流或者数据报的unix domain socket套接口， 
     1、在父子进程之间可采用socketpair实现流管道，类似于pipe的实现，两个进程中分别有一个socket套接口 
     2、在无亲缘关系的进程之间采用基本的domain socket过程（服务器Bind， listen, accpet返回的套接口， 客户端connect返回的套接口），将两个进程关联起来。 
(2) 发送进程通过调用返回描述字的任一unix函数打开一个描述字(也就是返回fd类型的函数打开一个描述字)。 
(3) 发送进程创建一个msghdr结构，其中将待传递的描述字作为辅助数据发送，调用sendmsg跨越(1)中获得的套接口发送描述字。在发送完成以后，在发送进程即使关闭该描述字也不会影响接收进程的描述符，发送一个描述字导致该描述字的引用计数加1。 
(4) 接收进程调用recvmsg在(1)中获取的套接口上接收该描述字，该描述字在接收进程中的描述字号不同于在发送进程中的描述字号是正常的，也就是说如果在发送进程中描述字号是20，而在接收进程中对应的描述字号可能被使用，该进程会分配一个不一样的描述字号（如open对同一个文件进行多次打开，每一次的fd返回值都不一样是一个道理），此时就是说两个进程同时指向一个描述字。 
注意：在发送过程中由于没有报文，在接收的过程中会分不清是文件已经结束还是只是发送了辅助数据，因此通常在发送辅助数据的时候会传输至少一个字节的数据，该数据在接收过程中不做任何的处理。 
服务器端接收客户端发送过来的描述字: 

点击( 此处 )折叠或打开

1. 服务器端接收客户端发送过来的描述字： 
   
2. 
   
3. #include  "unp.h" 
   
4. 
   
5. int  main ( int  argc ,  char  * argv [ ] ) 
   
6. { 
   
7.      int  clifd ,  listenfd ; 
   
8.     struct sockaddr_un servaddr ,  cliaddr ; 
   
9.      int  ret ; 
   
10.     socklen_t clilen ; 
    
11.     struct msghdr msg ; 
    
12.     struct iovec iov [ 1 ] ; 
    
13.     char buf [ 100 ] ; 
    
14.     char  * testmsg  =   "test msg. " ; 
    
15.      
    
16.     union  { 
    
17.         struct cmsghdr cm ; 
    
18.         char control [ CMSG_SPACE ( sizeof ( int ) ) ] ; 
    
19.      }  control_un ; 
    
20.     struct cmsghdr  * pcmsg ; 
    
21.      int  recvfd ; 
    
22.      
    
23.     listenfd  =  socket ( AF_UNIX ,  SOCK_STREAM ,  0 ) ; 
    
24.      if   ( listenfd  <  0 )   { 
    
25.         printf ( "socket failed. " ) ; 
    
26.         return  - 1 ; 
    
27.      } 
    
28.      
    
29.     unlink ( UNIXSTR_PATH ) ; 
    
30.      
    
31.     bzero ( & servaddr ,  sizeof ( servaddr ) ) ; 
    
32.     servaddr . sun_family  =  AF_UNIX ; 
    
33.     strcpy ( servaddr . sun_path ,  UNIXSTR_PATH ) ; 
    
34.      
    
35.     ret  =  bind ( listenfd ,   ( SA  * ) & servaddr ,  sizeof ( servaddr ) ) ; 
    
36.      if   ( ret  <  0 )   { 
    
37.         printf ( "bind failed. errno = %d. " ,  errno ) ; 
    
38.         close  ( listenfd ) ; 
    
39.         return  - 1 ; 
    
40.      } 
    
41.      
    
42.     listen ( listenfd ,  5 ) ; 
    
43.      
    
44.      while   ( 1 )   { 
    
45.         clilen  =  sizeof ( cliaddr ) ; 
    
46.         clifd  =  accept ( listenfd ,   ( SA  * ) & cliaddr ,   & clilen ) ; 
    
47.          if   ( clifd  <  0 )   { 
    
48.             printf ( "accept failed. " ) ; 
    
49.             continue ; 
    
50.          } 
    
51.          
    
52.         msg . msg_name  =   NULL ; 
    
53.         msg . msg_namelen  =  0 ; 
    
54.         iov [ 0 ] . iov_base  =  buf ; 
    
55.         iov [ 0 ] . iov_len  =  100 ; 
    
56.         msg . msg_iov  =  iov ; 
    
57.         msg . msg_iovlen  =  1 ; 
    
58.         msg . msg_control  =  control_un . control ; 
    
59.         msg . msg_controllen  =  sizeof ( control_un . control ) ; 
    
60.          
    
61.         ret  =  recvmsg ( clifd ,   & msg ,  0 ) ; 
    
62.          if   ( ret  < =  0 )   { 
    
63.             return ret ; 
    
64.          } 
    
65.          
    
66.          if   ( ( pcmsg  =  CMSG_FIRSTHDR ( & msg ) )   ! =   NULL   & &   ( pcmsg - >cmsg_len  = =  CMSG_LEN ( sizeof ( int ) ) ) )   { 
    
67.              if   ( pcmsg - > cmsg_level  ! =  SOL_SOCKET )   { 
    
68.                 printf ( "cmsg_leval is not SOL_SOCKET " ) ; 
    
69.                 continue ; 
    
70.              } 
    
71.              
    
72.              if   ( pcmsg - > cmsg_type  ! =  SCM_RIGHTS )   { 
    
73.                 printf ( "cmsg_type is not SCM_RIGHTS" ) ; 
    
74.                 continue ; 
    
75.              } 
    
76.              
    
77.             recvfd  =   * ( ( int   * )  CMSG_DATA ( pcmsg ) ) ; 
    
78.             printf ( "recv fd = %d " ,  recvfd ) ; 
    
79.              
    
80.             write ( recvfd ,  testmsg ,  strlen ( testmsg )   +  1 ) ; 
    
81.          } 
    
82.      } 
    
83.      
    
84.     return 0 ; 
    
85. }

客户端发送描述字: 

点击( 此处 )折叠或打开

1. #include  "unp.h" 
   
2. #define OPEN_FILE  "test" 
   
3. int  main ( int  argc ,  char  * argv [ ] ) 
   
4. { 
   
5.      int  clifd ; 
   
6.     struct sockaddr_un servaddr ; 
   
7.      int  ret ; 
   
8.     struct msghdr msg ; 
   
9.     struct iovec iov [ 1 ] ; 
   
10.     char buf [ 100 ] ; 
    
11.     union  { 
    
12.         struct cmsghdr cm ; 
    
13.         char control [ CMSG_SPACE ( sizeof ( int ) ) ] ; 
    
14.      }  control_un ; 
    
15.     struct cmsghdr  * pcmsg ; 
    
16.      int  fd ; 
    
17.      
    
18.     clifd  =  socket ( AF_UNIX ,  SOCK_STREAM ,  0 ) ; 
    
19.      if   ( clifd  <  0 )   { 
    
20.         printf ( "socket failed. " ) ; 
    
21.         return  - 1 ; 
    
22.      } 
    
23.      
    
24.     fd  =  open ( OPEN_FILE ,  O_CREAT |  O_RDWR ,  0777 ) ; 
    
25.      if   ( fd  <  0 )   { 
    
26.         printf ( "open test failed. " ) ; 
    
27.         return  - 1 ; 
    
28.      } 
    
29.      
    
30.     bzero ( & servaddr ,  sizeof ( servaddr ) ) ; 
    
31.     servaddr . sun_family  =  AF_UNIX ; 
    
32.     strcpy ( servaddr . sun_path ,  UNIXSTR_PATH ) ; 
    
33.      
    
34.     ret  =  connect ( clifd , ( SA  * ) & servaddr ,  sizeof ( servaddr ) ) ; 
    
35.      if   ( ret  <  0 )   { 
    
36.         printf ( "connect failed. " ) ; 
    
37.         return 0 ; 
    
38.      } 
    
39.      
    
40.     msg . msg_name  =   NULL ; 
    
41.     msg . msg_namelen  =  0 ; 
    
42.     iov [ 0 ] . iov_base  =  buf ; 
    
43.     iov [ 0 ] . iov_len  =  100 ; 
    
44.     msg . msg_iov  =  iov ; 
    
45.     msg . msg_iovlen  =  1 ; 
    
46.     msg . msg_control  =  control_un . control ; 
    
47.     msg . msg_controllen  =  sizeof ( control_un . control ) ; 
    
48.      
    
49.     pcmsg  =  CMSG_FIRSTHDR ( & msg ) ; 
    
50.     pcmsg - > cmsg_len  =  CMSG_LEN ( sizeof ( int ) ) ; 
    
51.     pcmsg - > cmsg_level  =  SOL_SOCKET ; 
    
52.     pcmsg - > cmsg_type  =  SCM_RIGHTS ; 
    
53.      * ( ( int   * ) CMSG_DATA ( pcmsg ) )   = =  fd ; 
    
54.      
    
55.     ret  =  sendmsg ( clifd ,   & msg ,  0 ) ; 
    
56.     printf ( "ret = %d. " ,  ret ) ; 
    
57.     return 0 ; 
    
58. }

上面的程序主要实现了两个进程（无亲缘关系）之间传递描述符的实现，主要使用了sendmsg和recvmsg的强大功能，利用了辅助数据来传递描述字（内核经过特殊处理的消息）。 
关于这两个函数的另一个应用是获取认证相关的处理，也是通过辅助数据的方式将用户的认证数据发送给对应的客户端。需要将cmsg->cmsg_type = SCM_CREDS。其余的处理都差不多。