ip过滤器提升async_receive_from

时间:2011-12-13 11:30:46

标签: c++ udp boost-asio

我正在使用boost::asio来捕获udp端口上的数据包。我只是新手来提升。 如何使async_receive_from复制数据仅缓冲指定源ip的数据包?

1 个答案:

答案 0 :(得分:2)

根据 capture 数据包的含义,这样的一些代码可以正常工作。这是从Boost Asio Async UDP示例修改的。如果将端口设置为0,则socket_连接到指定port的本地接口。我相信它会侦听所有端口。

一旦使用async_receive_from收到数据包,它也会从解码的数据报返回sender_endpoint_(即问题数据包来自哪里。)在handle_receive_from函数中添加一个条件语句检查所需的sender_endpoint_和“将数据复制到缓冲区”。

class server
{ 
public: 
    server(boost::asio::io_service& io_service, short port)
    : io_service_(io_service),
    socket_(io_service, udp::endpoint(udp::v4(), port))
    {
        boost::asio::socket_base::receive_buffer_size option(1<<12); <br>
        socket_.set_option(option);
        start_receive(); 
    }

    void handle_receive_from(const boost::system::error_code& error,
        size_t bytes_recvd)
    {
        if (!error && bytes_recvd > 0)
        {
           if(sender_endpoint_ == <desired_IP_here>) 
               messages_.push(data_);
        } 
        start_receive();
    }

private: 
    boost::asio::io_service& io_service_; 
    udp::socket socket_;
    udp::endpoint sender_endpoint_;
    enum { max_length = 256}; 
    boost::array < boost::uint32_t, max_length > data_; 
    std::queue<boost::array<boost::uint32_t, max_length> messages_; 

    void start_receive()
    {
        socket_.async_receive_from(
            boost::asio::buffer(data_, (sizeof(boost::uint32_t)*max_length)),
            sender_endpoint_,
            boost::bind(&server::handle_receive_from, this,
            boost::asio::placeholders::error,
            boost::asio::placeholders::bytes_transferred));

    }
};

几乎忘了 - 主要功能!

int main(void) 
{
    boost::asio::io_service io_service;
    int port_rx = 0;


    using namespace std;
    server rx(io_service, port_rx);
    io_service.run();

    return 0;
}

希望这有帮助!