Socket::Socket (int timeout)
: _deadline (_io_service)
, _socket (_io_service)
- , _acceptor (0)
, _timeout (timeout)
{
_deadline.expires_at (boost::posix_time::pos_infin);
check ();
}
-Socket::~Socket ()
-{
- delete _acceptor;
-}
-
void
Socket::check ()
{
if (_deadline.expires_at() <= boost::asio::deadline_timer::traits_type::now ()) {
- if (_acceptor) {
- _acceptor->cancel ();
- } else {
- _socket.close ();
- }
+ _socket.close ();
_deadline.expires_at (boost::posix_time::pos_infin);
}
}
}
-void
-Socket::accept (int port)
-{
- _acceptor = new boost::asio::ip::tcp::acceptor (_io_service, boost::asio::ip::tcp::endpoint (boost::asio::ip::tcp::v4(), port));
-
- _deadline.expires_from_now (boost::posix_time::seconds (_timeout));
- boost::system::error_code ec = boost::asio::error::would_block;
- _acceptor->async_accept (_socket, boost::lambda::var(ec) = boost::lambda::_1);
- do {
- _io_service.run_one ();
- } while (ec == boost::asio::error::would_block);
-
- delete _acceptor;
- _acceptor = 0;
-
- if (ec) {
- throw NetworkError (String::compose (_("error during async_accept (%1)"), ec.value ()));
- }
-}
-
/** Blocking write.
* @param data Buffer to write.
* @param size Number of bytes to write.
{
public:
Socket (int timeout = 30);
- ~Socket ();
/** @return Our underlying socket */
boost::asio::ip::tcp::socket& socket () {
}
void connect (boost::asio::ip::tcp::endpoint);
- void accept (int);
void write (uint32_t n);
void write (uint8_t const * data, int size);
boost::asio::io_service _io_service;
boost::asio::deadline_timer _deadline;
boost::asio::ip::tcp::socket _socket;
- boost::asio::ip::tcp::acceptor* _acceptor;
int _timeout;
};