今天在看boost库的时候注意到异步I/O操作时,缓冲区有效性问题。
如何实现异步操作:以异步读操作为例async_read(buffer, handler);
void handler() {} void function(){ char buffer[1234];async_read(buffer, handler);}当运行function时会不会有安全隐患,因为这时候缓冲区buffer已经被释放了!需要了解异步操作做了哪些事情,async_read会将回调函数handler注册到能够检测到i/o事件的地方,并将buffer的地址也传递过去,用于存放接收到的数据。因此需要保证缓冲区buffer的有效性。一个例子:(从网上找来的,具体出处没找到)
- #include <iostream>
- #include <string>
- #include <boost/asio.hpp>
- #include <boost/bind.hpp>
- #include <boost/smart_ptr.hpp>
- using namespace boost::asio;
- using boost::system::error_code;
- using ip::tcp;
- struct CHelloWorld_Service{
- CHelloWorld_Service(io_service &iosev)
- :m_iosev(iosev),m_acceptor(iosev, tcp::endpoint(tcp::v4(), 1000))
- {
- }
- void start()
- {
- // 开始等待连接(非阻塞)
- boost::shared_ptr<tcp::socket> psocket(new tcp::socket(m_iosev));
- // 触发的事件只有error_code参数,所以用boost::bind把socket绑定进去
- m_acceptor.async_accept(*psocket,
- boost::bind(&CHelloWorld_Service::accept_handler, this , psocket, _1)
- );
- }
- // 有客户端连接时accept_handler触发
- void accept_handler(boost::shared_ptr<tcp::socket> psocket, error_code ec)
- {
- if (ec) return ;
- // 继续等待连接
- start();
- // 显示远程IP
- std::cout << psocket->remote_endpoint().address() << std::endl;
- // 发送信息(非阻塞)
- boost::shared_ptr<std::string> pstr(new std::string( "hello async world!" ));
- psocket->async_write_some(buffer(*pstr),
- boost::bind(&CHelloWorld_Service::write_handler, this , pstr, _1, _2)
- );
- }
- // 异步写操作完成后write_handler触发
- void write_handler(boost::shared_ptr<std::string> pstr,
- error_code ec, size_t bytes_transferred)
- {
- if (ec)
- std::cout<< "发送失败!" << std::endl;
- else
- std::cout<< *pstr << " 已发送" << std::endl;
- }
- private :
- io_service &m_iosev;
- ip::tcp::acceptor m_acceptor;
- };
- int main( int argc, char * argv[])
- {
- io_service iosev;
- CHelloWorld_Service sev(iosev);
- // 开始等待连接
- sev.start();
- iosev.run();
- return 0;
- }
在这个例子中,首先调用sev.start() 开 始接受客户端连接。由于async_accept 调 用后立即返回,start() 方 法 也就马上完成了。sev.start()在 瞬间返回后iosev.run() 开 始执行,iosev.run()方法是一个循环,负责分发异步回调事件,只 有所有异步操作全部完成才会返回。
这里有个问题,就是要保证start()方法中m_acceptor.async_accept 操 作所用的tcp::socket 对 象 在整个异步操作期间保持有效 (不 然系统底层异步操作了一半突然发现tcp::socket没了,不是拿人家开涮嘛-_-!!!),而且客户端连接进来后这个tcp::socket对象还 有用呢。这里的解决办法是使用一个带计数的智能指针boost::shared_ptr<tcp:: socket> ,并把这个指针作为参数绑定到回调函数上。
一旦有客户连接,我们在start()里给的回调函数accept_handler 就会被 调用,首先调用start()继续异步等待其 它客户端的连接,然后使用绑定进来的tcp::socket对象与当前客户端通信。
发送数据也使用了异步方式(async_write_some ), 同样要保证在整个异步发送期间缓冲区的有效性,所以也用boost::bind绑定了boost::shared_ptr<std:: string>。