使用boost:asio与select?阻止TCP输入或文件更新

时间:2012-09-04 23:32:17

标签: c++ boost boost-asio

我本来打算在我的程序中有一个线程,它会等待两个文件描述符,一个用于套接字,另一个用于描述文件系统的FD(特别是等待查看是否有新文件添加到目录中) )。由于我希望很少看到添加的新文件或新的TCP消息,我希望有一个线程等待输入和处理发生时检测到的任何输入,而不是单独的线程。

然后我(终于!)得到了'老板'的许可来使用提升。所以现在我想用boost:asio替换基本套接字。只是我遇到了一个小问题。似乎asio对它自己的select版本进行了修改,而不是提供一个我可以直接使用select的FD。这让我不确定如何阻止这两个条件,新文件和TCP输入,同时当一个只使用select而另一个似乎不支持select的使用。是否有一个简单的工作,我错过了?

1 个答案:

答案 0 :(得分:7)

ASIO最好异步使用(这就是它的意思):你可以为TCP读取和文件描述符活动设置处理程序,并为你调用处理程序。

这是一个让你入门的演示示例(为具有inotify支持的Linux编写):

#include <iostream>
#include <boost/asio.hpp>
#include <boost/bind.hpp>
#include <sys/inotify.h>

namespace asio = boost::asio;
void start_notify_handler();
void start_accept_handler();

// this stuff goes into your class, only global for the simplistic demo
asio::streambuf buf(1024);
asio::io_service io_svc;
asio::posix::stream_descriptor stream_desc(io_svc);
asio::ip::tcp::socket sock(io_svc);
asio::ip::tcp::endpoint end(asio::ip::tcp::v4(), 1234);
asio::ip::tcp::acceptor acceptor(io_svc, end);

// this gets called on file system activity
void notify_handler(const boost::system::error_code&,
                    std::size_t transferred)
{
    size_t processed = 0;
    while(transferred - processed >= sizeof(inotify_event))
    {
        const char* cdata = processed
                            + asio::buffer_cast<const char*>(buf.data());
        const inotify_event* ievent =
                                 reinterpret_cast<const inotify_event*>(cdata);
        processed += sizeof(inotify_event) + ievent->len;
        if(ievent->len > 0 && ievent->mask & IN_OPEN)
            std::cout << "Someone opened " << ievent->name << '\n';
    }
    start_notify_handler();
}

// this gets called when nsomeone connects to you on TCP port 1234
void accept_handler(const boost::system::error_code&)
{
    std::cout << "Someone connected from " 
              << sock.remote_endpoint().address() << '\n';
    sock.close(); // dropping connection: this is just a demo
    start_accept_handler();
}

void start_notify_handler()
{
    stream_desc.async_read_some( buf.prepare(buf.max_size()),
        boost::bind(&notify_handler, asio::placeholders::error,
                    asio::placeholders::bytes_transferred));
}

void start_accept_handler()
{
    acceptor.async_accept(sock,
        boost::bind(&accept_handler, asio::placeholders::error));
}

int main()
{
    int raw_fd = inotify_init(); // error handling ignored
    stream_desc.assign(raw_fd);
    inotify_add_watch(raw_fd, ".", IN_OPEN);
    start_notify_handler();
    start_accept_handler();
    io_svc.run();
}