Question

我在CentOS 6.5上使用内核版本kernel-2.6.32-431.el6。

我试图找到recvmmsg（）的实现，希望我没有看错了源〜/ rpmbuild / BUILD / kernel-2.6.32-431.el6 / linux-2.6.32-431。 el6.x86_64 / NET / socket.c中。如果我是，请将我指向正确的来源。

int __sys_recvmmsg(int fd, struct mmsghdr __user *mmsg, unsigned int vlen,
                   unsigned int flags, struct timespec *timeout)
{
        int fput_needed, err, datagrams;
        struct socket *sock;
        struct mmsghdr __user *entry;
        struct compat_mmsghdr __user *compat_entry;
        struct msghdr msg_sys;
        struct timespec end_time;

        if (timeout &&
            poll_select_set_timeout(&end_time, timeout->tv_sec,
                                    timeout->tv_nsec))
                return -EINVAL;

        datagrams = 0;

        sock = sockfd_lookup_light(fd, &err, &fput_needed);
        if (!sock)
                return err;

        err = sock_error(sock->sk);
        if (err)
                goto out_put;

        entry = mmsg;
        compat_entry = (struct compat_mmsghdr __user *)mmsg;

        while (datagrams < vlen) {
                /*
                 * No need to ask LSM for more than the first datagram.
                 */
                if (MSG_CMSG_COMPAT & flags) {
                        err = __sys_recvmsg(sock, (struct msghdr __user *)compat_entry,
                                            &msg_sys, flags, datagrams);
                        if (err < 0)
                                break;
                        err = __put_user(err, &compat_entry->msg_len);
                        ++compat_entry;
                } else {
                        err = __sys_recvmsg(sock, (struct msghdr __user *)entry,
                                            &msg_sys, flags, datagrams);
                        if (err < 0)
                                break;
                        err = put_user(err, &entry->msg_len);
                        ++entry;
                }

                if (err)
                        break;
                ++datagrams;

                if (timeout) {
                        ktime_get_ts(timeout);
                        *timeout = timespec_sub(end_time, *timeout);
                        if (timeout->tv_sec < 0) {
                                timeout->tv_sec = timeout->tv_nsec = 0;
                                break;
                        }

                        /* Timeout, return less than vlen datagrams */
                        if (timeout->tv_nsec == 0 && timeout->tv_sec == 0)
                                break;
                }

                /* Out of band data, return right away */
                if (msg_sys.msg_flags & MSG_OOB)
                        break;
        }

out_put:
        fput_light(sock->file, fput_needed);

        if (err == 0)
                return datagrams;

        if (datagrams != 0) {
                /*
                 * We may return less entries than requested (vlen) if the
                 * sock is non block and there aren't enough datagrams...
                 */
                if (err != -EAGAIN) {
                        /*
                         * ... or  if recvmsg returns an error after we
                         * received some datagrams, where we record the
                         * error to return on the next call or if the
                         * app asks about it using getsockopt(SO_ERROR).
                         */
                        sock->sk->sk_err = -err;
                }

                return datagrams;
        }

        return err;
}

让我们说套接字上有2个数据包，我试试：

timespec t = {0, 0};
recvmmsg(fd, mmsg, vlen, 0, &t);

然后，由于这些行：

                    if (timeout->tv_nsec == 0 && timeout->tv_sec == 0)
                            break;

recvmmsg（）将只返回1个数据包而不是2.我是对的吗？

似乎缺少recvmmsg（）的主要目的之一 - 尝试在有数据包排队时减少系统调用数。

也许我错过了什么？

欢迎提出任何建议。

谢谢！

============================

更新

我尝试编写一小段代码，等待数据包排队，然后使用timeout = 0（非NULL）调用recvmmsg（）。它仍然只能获得1个数据包。因此，我认为我正在查看正确的源代码。

Answer 1

没有。零超时意味着无穷大。

Answer 2

是的，你是对的。我认为如果你想读取队列中的所有数据包，最好使用NULL timeout + O_NONBLOCK

Answer 3

忽略recvmmsg超时参数，并在必要时使用SOL_SOCKET和SO_RCVTIMEO。正常工作。

Answer 4

如果您想在不阻塞的情况下接收接收缓冲区中存在的最多 vlen 消息，这将解决您的问题：

recvmmsg(fd, mmsg, vlen, MSG_DONTWAIT, NULL);

它将循环直到内部 recvmsg 调用返回错误代码（如 EAGAIN/EWOULDBLOCK）或直到在缓冲区中填充了 vlen 的消息计数而不会阻塞。

具有超时= 0的recvmmsg（）等效于非阻塞recvmsg（）？

4 个答案: