通过UNIX域套接字发送文件描述符，并使用select()函数

Question

我正在使用UNIX域套接字将文件描述符传输到另一个进程。这个方法是有效的，但当我首次尝试使用select()检测套接字是否可写时，sendmsg()调用会失败，并出现Bad File Descriptor错误。

如果我不将文件描述符信息添加到msghdr结构中，那么sendmsg()函数与select()一起正常工作，因此冲突似乎出现在select()和传输文件描述符之间。

我在select()、recvmsg()或其他任何文档中都没有找到有关此问题的信息。由于需要将其变成一个向多个进程分发文件描述符的服务器，我仍然想能够使用select()。

有没有办法使这个方法能够正常工作，或者是否有其他替代方案？

平台为Ubuntu 10.4。

这是初始化结构的代码：

struct cmsghdr_fd : public cmsghdr
{
  int fd;
};

int sendfd(int sock, int fd)
{
  struct msghdr hdr;
  struct iovec data;
  struct cmsghdr_fd msgdata;

  char dummy = '*';
  data.iov_base = &dummy;
  data.iov_len = sizeof(dummy);

  hdr.msg_name = NULL;
  hdr.msg_namelen = 0;
  hdr.msg_iov = &data;
  hdr.msg_iovlen = 1;
  hdr.msg_flags = 0;

  hdr.msg_control = &msgdata;
  hdr.msg_controllen = sizeof(msgdata);

  struct cmsghdr* cmsg = CMSG_FIRSTHDR(&hdr);
  cmsg->cmsg_len   = hdr.msg_controllen;
  cmsg->cmsg_level = SOL_SOCKET;
  cmsg->cmsg_type  = SCM_RIGHTS;

  *(int*)CMSG_DATA(cmsg) = fd;

  int n = sendmsg(sock, &hdr, 0);

  if(n == -1)
    printf("sendmsg() failed: %s (socket fd = %d)\n", strerror(errno), sock);

  return n;
}

只要我不先调用select()检查套接字是否准备好写入，这段代码就能正常工作。

- svdree

---

你能否包含填充 msghdr 和 cmsghdr 的代码？ - Nikolai Fetissov

我已经将填充结构体的代码添加到问题中。 - svdree

你应该使用CMSG_LEN宏来设置cmsg_len，但我不认为这是实际的问题所在... - Hasturkun

这段代码和我看过的其他典型的sendfd()之间有一些差异 - 这些中的任何一个会有影响吗：http://code.swtch.com/plan9port/src/0e6ae8ed3276/src/lib9/sendfd.c或http://www.koders.com/c/fidAA01FB46DDB5D83E39B335DA2639B226F39CF21E.aspx - nos

我们可能还需要看 select() 的使用方法。 - Nikolai Fetissov

Answer 1

我尝试了这个页面上提供的sendfd代码，这个代码由友好的nos提供。即使只有略微不同，但它在与select()结合使用时仍然有效。现在代码看起来像这样:

    int sendfd(int sock, int fd)
    {
      struct msghdr hdr;
      struct iovec data;

      char cmsgbuf[CMSG_SPACE(sizeof(int))];

      char dummy = '*';
      data.iov_base = &dummy;
      data.iov_len = sizeof(dummy);

      memset(&hdr, 0, sizeof(hdr));
      hdr.msg_name = NULL;
      hdr.msg_namelen = 0;
      hdr.msg_iov = &data;
      hdr.msg_iovlen = 1;
      hdr.msg_flags = 0;

      hdr.msg_control = cmsgbuf;
      hdr.msg_controllen = CMSG_LEN(sizeof(int));

      struct cmsghdr* cmsg = CMSG_FIRSTHDR(&hdr);
      cmsg->cmsg_len   = CMSG_LEN(sizeof(int));
      cmsg->cmsg_level = SOL_SOCKET;
      cmsg->cmsg_type  = SCM_RIGHTS;

      *(int*)CMSG_DATA(cmsg) = fd;

      int n = sendmsg(sock, &hdr, 0);

      if(n == -1)
        printf("sendmsg() failed: %s (socket fd = %d)\n", strerror(errno), sock);

      return n;
        }

- svdree

---

不再使用 select。我在原始帖子的评论中给出了理由。使用 poll 系统调用（可移植到BSD）或 epoll 系统调用。 - Omnifarious