发送文件描述符函数

发送文件描述符函数

 

/*
* 从fd中接收消息，并将文件描述符放在指针recvfd中
*/
ssize_t recv_fd(int fd, void*data, size_t bytes, int*recvfd)
{
   struct msghdr msghdr_recv; /*建立一个消息，之后填充此消息的成员数据，并发送给fd*/
   struct iovec iov[1]; /*为向量，消息的数据在此项中保存*/
   size_t n;
   int newfd;
   /*建立一个联合结构，便于进行消息的处理*/
   union{
       struct cmsghdr cm;
       char control[CMSG_SPACE(sizeof(int))];
   }control_un;
   struct cmsghdr*pcmsghdr; /*消息头部*/
   msghdr_recv.msg_control = control_un.control; /*控制消息*/
   msghdr_recv.msg_controllen = sizeof(control_un.control); /*控制消息的长度*/

   msghdr_recv.msg_name = NULL; /*消息的名称为空*/
   msghdr_recv.msg_namelen = 0; /*消息的长度为空*/

   iov[0].iov_base = data; /*向量的数据为传入的数据*/
   iov[0].iov_len = bytes; /*向量的长度为传入数据的长度*/
   msghdr_recv.msg_iov = iov; /*消息向量指针*/
   msghdr_recv.msg_iovlen = 1; /*消息向量的个数为1个*/
   if((n = recvmsg(fd, &msghdr_recv, 0))<=0) /*接收消息*/
       return n;

   if((pcmsghdr = CMSG_FIRSTHDR(&msghdr_recv))!= NULL && /*获得消息的头部*/
       pcmsghdr->cmsg_len == CMSG_LEN(sizeof(int))){ /*获得消息的长度为int*/
       if(pcmsghdr->cmsg_level != SOL_SOCKET) /*消息的level应该为SOL_SOCKET*/
           printf("control level != SOL_SOCKET\n");

       if(pcmsghdr->cmsg_type != SCM_RIGHTS) /*消息的类型判断*/
           printf("control type != SCM_RIGHTS\n");

           *recvfd =*((int*)CMSG_DATA(pcmsghdr)); /*获得打开文件的描述符*/
   }else
       *recvfd = -1;

   return n; /*返回接收消息的长度*/
}