boost asio SO_REUSEPORT

Question

我正在使用boost库开发一个多进程套接字服务器。

每个进程运行一个io_service。

我希望这个进程都在同一个端口上接受。

我知道SO_REUSEPORT (在Linux内核3.9之后)会有帮助。

就像这个python脚本

import socket                                                                                                                                                       

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)                                                                                                               
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1)                                                                                                             

s.bind(('0.0.0.0', 9091))
s.listen(1)

while True:
    conn, addr = s.accept()
    print "new connection"
    while True:
        data = conn.recv(100)
        print "got data", data
        if not data or data == 'exit':
            break
    conn.close()

但是我不知道如何在boost asio io_service中使用这个选项？

Answer 1

由我自己回答。

#include <iostream>
#include <string>
#include <array>
#include <boost/asio.hpp>
#include <arpa/inet.h>


using boost::asio::ip::tcp;

int main()
{
    boost::asio::io_service io;
    tcp::acceptor acceptor(io);
    acceptor.open(tcp::v4());

    int one = 1;
    setsockopt(acceptor.native_handle(), SOL_SOCKET, SO_REUSEADDR | SO_REUSEPORT, &one, sizeof(one));

    acceptor.bind(tcp::endpoint(tcp::v4(), 9091));
    acceptor.listen();

    std::cout << "start" << std::endl;
    for(;;)
    {
        tcp::socket socket(io);

        acceptor.accept(socket);
        std::cout << "new connections" << std::endl;
        for(;;)
        {                                                                                                                                                               
            std::array<char, 4> buf;
            boost::system::error_code error;
            boost::asio::read(socket, boost::asio::buffer(buf), error);
            if(error)
            {
                std::cout << "read error: " << error << std::endl;
                break;
            }
            std::cout << "read: " << std::string(buf.data()) << std::endl;
        }
    }
}

Answer 2

对于2019年阅读这篇文章的人来说: Asio现在包含了boost/asio/detail/impl/socket_ops.ipp中的一个解决方案

#if defined(__MACH__) && defined(__APPLE__) \
  || defined(__NetBSD__) || defined(__FreeBSD__) || defined(__OpenBSD__)
    // To implement portable behaviour for SO_REUSEADDR with UDP sockets we
    // need to also set SO_REUSEPORT on BSD-based platforms.
    if ((state & datagram_oriented)
        && level == SOL_SOCKET && optname == SO_REUSEADDR)
    {
      call_setsockopt(&msghdr::msg_namelen, s,
          SOL_SOCKET, SO_REUSEPORT, optval, optlen);
    }
#endif

因此，如果需要，socket_->set_option(udp::socket::reuse_address(true));将自动设置SO_REUSEPORT选项。

Answer 3

按照boost/asio/socket_base.hpp定义reuse_address的方式，我这样做了：

typedef boost::asio::detail::socket_option::boolean<SOL_SOCKET, SO_REUSEPORT> reuse_port;
socket_.set_option(reuse_port(true));