82 lines
2.7 KiB
C++
82 lines
2.7 KiB
C++
|
#include <fc/udp_socket.hpp>
|
||
|
|
#include <fc/ip.hpp>
|
||
|
|
#include <fc/fwd_impl.hpp>
|
||
|
|
#include <fc/asio.hpp>
|
||
|
|
|
||
|
|
|
||
|
|
namespace fc {
|
||
|
|
|
||
|
|
class udp_socket::impl {
|
||
|
|
public:
|
||
|
|
impl():_sock( fc::asio::default_io_service() ){}
|
||
|
|
|
||
|
|
boost::asio::ip::udp::socket _sock;
|
||
|
|
};
|
||
|
|
|
||
|
|
boost::asio::ip::udp::endpoint to_asio_ep( const fc::ip::endpoint& e ) {
|
||
|
|
return boost::asio::ip::udp::endpoint(boost::asio::ip::address_v4(e.get_address()), e.port() );
|
||
|
|
}
|
||
|
|
fc::ip::endpoint to_fc_ep( const boost::asio::ip::udp::endpoint& e ) {
|
||
|
|
return fc::ip::endpoint( e.address().to_v4().to_ulong(), e.port() );
|
||
|
|
}
|
||
|
|
|
||
|
|
udp_socket::udp_socket() {
|
||
|
|
}
|
||
|
|
udp_socket::~udp_socket() {
|
||
|
|
}
|
||
|
|
|
||
|
|
size_t udp_socket::send_to( const char* b, size_t l, const ip::endpoint& to ) {
|
||
|
|
try {
|
||
|
|
return my->_sock.send_to( boost::asio::buffer(b, l), to_asio_ep(to) );
|
||
|
|
} catch( const boost::system::system_error& e ) {
|
||
|
|
if( e.code() == boost::asio::error::would_block ) {
|
||
|
|
promise<size_t>::ptr p(new promise<size_t>("udp_socket::send_to"));
|
||
|
|
my->_sock.async_send_to( boost::asio::buffer(b,l), to_asio_ep(to),
|
||
|
|
[=]( const boost::system::error_code& ec, size_t bt ) {
|
||
|
|
if( !ec ) p->set_value(bt);
|
||
|
|
else p->set_exception( fc::copy_exception( boost::system::system_error(ec) ) );
|
||
|
|
});
|
||
|
|
return p->wait();
|
||
|
|
}
|
||
|
|
throw;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
void udp_socket::open() {
|
||
|
|
my->_sock.open( boost::asio::ip::udp::v4() );
|
||
|
|
my->_sock.non_blocking(true);
|
||
|
|
}
|
||
|
|
void udp_socket::set_receive_buffer_size( size_t s ) {
|
||
|
|
my->_sock.set_option(boost::asio::socket_base::receive_buffer_size(s) );
|
||
|
|
}
|
||
|
|
void udp_socket::bind( const fc::ip::endpoint& e ) {
|
||
|
|
my->_sock.bind( to_asio_ep(e) );
|
||
|
|
}
|
||
|
|
size_t udp_socket::receive_from( char* b, size_t l, fc::ip::endpoint& _from ) {
|
||
|
|
try {
|
||
|
|
boost::asio::ip::udp::endpoint from;
|
||
|
|
size_t r = my->_sock.receive_from( boost::asio::buffer(b, l), from );
|
||
|
|
_from = to_fc_ep(from);
|
||
|
|
return r;
|
||
|
|
} catch( const boost::system::system_error& e ) {
|
||
|
|
if( e.code() == boost::asio::error::would_block ) {
|
||
|
|
boost::asio::ip::udp::endpoint from;
|
||
|
|
promise<size_t>::ptr p(new promise<size_t>("udp_socket::send_to"));
|
||
|
|
my->_sock.async_receive_from( boost::asio::buffer(b,l), from,
|
||
|
|
[=]( const boost::system::error_code& ec, size_t bt ) {
|
||
|
|
if( !ec ) p->set_value(bt);
|
||
|
|
else p->set_exception( fc::copy_exception( boost::system::system_error(ec) ) );
|
||
|
|
});
|
||
|
|
auto r = p->wait();
|
||
|
|
_from = to_fc_ep(from);
|
||
|
|
return r;
|
||
|
|
}
|
||
|
|
throw;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
void udp_socket::close() {
|
||
|
|
my->_sock->cancel();
|
||
|
|
my->_sock.close();
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|