mirror of
https://github.com/boostorg/asio.git
synced 2026-01-23 05:22:08 +00:00
449 lines
15 KiB
C++
449 lines
15 KiB
C++
//
|
|
// udp.cpp
|
|
// ~~~~~~~
|
|
//
|
|
// Copyright (c) 2003-2012 Christopher M. Kohlhoff (chris at kohlhoff dot com)
|
|
//
|
|
// Distributed under the Boost Software License, Version 1.0. (See accompanying
|
|
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
|
|
//
|
|
|
|
// Disable autolinking for unit tests.
|
|
#if !defined(BOOST_ALL_NO_LIB)
|
|
#define BOOST_ALL_NO_LIB 1
|
|
#endif // !defined(BOOST_ALL_NO_LIB)
|
|
|
|
// Test that header file is self-contained.
|
|
#include <boost/asio/ip/udp.hpp>
|
|
|
|
#include <boost/bind.hpp>
|
|
#include <cstring>
|
|
#include <boost/asio/io_service.hpp>
|
|
#include <boost/asio/placeholders.hpp>
|
|
#include "../unit_test.hpp"
|
|
#include "../archetypes/gettable_socket_option.hpp"
|
|
#include "../archetypes/io_control_command.hpp"
|
|
#include "../archetypes/settable_socket_option.hpp"
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
// ip_udp_socket_compile test
|
|
// ~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
// The following test checks that all public member functions on the class
|
|
// ip::udp::socket compile and link correctly. Runtime failures are ignored.
|
|
|
|
namespace ip_udp_socket_compile {
|
|
|
|
void connect_handler(const boost::system::error_code&)
|
|
{
|
|
}
|
|
|
|
void send_handler(const boost::system::error_code&, std::size_t)
|
|
{
|
|
}
|
|
|
|
void receive_handler(const boost::system::error_code&, std::size_t)
|
|
{
|
|
}
|
|
|
|
void test()
|
|
{
|
|
using namespace boost::asio;
|
|
namespace ip = boost::asio::ip;
|
|
|
|
try
|
|
{
|
|
io_service ios;
|
|
char mutable_char_buffer[128] = "";
|
|
const char const_char_buffer[128] = "";
|
|
socket_base::message_flags in_flags = 0;
|
|
archetypes::settable_socket_option<void> settable_socket_option1;
|
|
archetypes::settable_socket_option<int> settable_socket_option2;
|
|
archetypes::settable_socket_option<double> settable_socket_option3;
|
|
archetypes::gettable_socket_option<void> gettable_socket_option1;
|
|
archetypes::gettable_socket_option<int> gettable_socket_option2;
|
|
archetypes::gettable_socket_option<double> gettable_socket_option3;
|
|
archetypes::io_control_command io_control_command;
|
|
boost::system::error_code ec;
|
|
|
|
// basic_datagram_socket constructors.
|
|
|
|
ip::udp::socket socket1(ios);
|
|
ip::udp::socket socket2(ios, ip::udp::v4());
|
|
ip::udp::socket socket3(ios, ip::udp::v6());
|
|
ip::udp::socket socket4(ios, ip::udp::endpoint(ip::udp::v4(), 0));
|
|
ip::udp::socket socket5(ios, ip::udp::endpoint(ip::udp::v6(), 0));
|
|
int native_socket1 = ::socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
|
ip::udp::socket socket6(ios, ip::udp::v4(), native_socket1);
|
|
|
|
#if defined(BOOST_ASIO_HAS_MOVE)
|
|
ip::udp::socket socket7(std::move(socket6));
|
|
#endif // defined(BOOST_ASIO_HAS_MOVE)
|
|
|
|
// basic_datagram_socket operators.
|
|
|
|
#if defined(BOOST_ASIO_HAS_MOVE)
|
|
socket1 = ip::udp::socket(ios);
|
|
socket1 = std::move(socket2);
|
|
#endif // defined(BOOST_ASIO_HAS_MOVE)
|
|
|
|
// basic_io_object functions.
|
|
|
|
io_service& ios_ref = socket1.get_io_service();
|
|
(void)ios_ref;
|
|
|
|
// basic_socket functions.
|
|
|
|
ip::udp::socket::lowest_layer_type& lowest_layer = socket1.lowest_layer();
|
|
(void)lowest_layer;
|
|
|
|
const ip::udp::socket& socket8 = socket1;
|
|
const ip::udp::socket::lowest_layer_type& lowest_layer2
|
|
= socket8.lowest_layer();
|
|
(void)lowest_layer2;
|
|
|
|
socket1.open(ip::udp::v4());
|
|
socket1.open(ip::udp::v6());
|
|
socket1.open(ip::udp::v4(), ec);
|
|
socket1.open(ip::udp::v6(), ec);
|
|
|
|
int native_socket2 = ::socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
|
socket1.assign(ip::udp::v4(), native_socket2);
|
|
int native_socket3 = ::socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
|
|
socket1.assign(ip::udp::v4(), native_socket3, ec);
|
|
|
|
bool is_open = socket1.is_open();
|
|
(void)is_open;
|
|
|
|
socket1.close();
|
|
socket1.close(ec);
|
|
|
|
ip::udp::socket::native_type native_socket4 = socket1.native();
|
|
(void)native_socket4;
|
|
|
|
ip::udp::socket::native_handle_type native_socket5
|
|
= socket1.native_handle();
|
|
(void)native_socket5;
|
|
|
|
socket1.cancel();
|
|
socket1.cancel(ec);
|
|
|
|
bool at_mark1 = socket1.at_mark();
|
|
(void)at_mark1;
|
|
bool at_mark2 = socket1.at_mark(ec);
|
|
(void)at_mark2;
|
|
|
|
std::size_t available1 = socket1.available();
|
|
(void)available1;
|
|
std::size_t available2 = socket1.available(ec);
|
|
(void)available2;
|
|
|
|
socket1.bind(ip::udp::endpoint(ip::udp::v4(), 0));
|
|
socket1.bind(ip::udp::endpoint(ip::udp::v6(), 0));
|
|
socket1.bind(ip::udp::endpoint(ip::udp::v4(), 0), ec);
|
|
socket1.bind(ip::udp::endpoint(ip::udp::v6(), 0), ec);
|
|
|
|
socket1.connect(ip::udp::endpoint(ip::udp::v4(), 0));
|
|
socket1.connect(ip::udp::endpoint(ip::udp::v6(), 0));
|
|
socket1.connect(ip::udp::endpoint(ip::udp::v4(), 0), ec);
|
|
socket1.connect(ip::udp::endpoint(ip::udp::v6(), 0), ec);
|
|
|
|
socket1.async_connect(ip::udp::endpoint(ip::udp::v4(), 0),
|
|
&connect_handler);
|
|
socket1.async_connect(ip::udp::endpoint(ip::udp::v6(), 0),
|
|
&connect_handler);
|
|
|
|
socket1.set_option(settable_socket_option1);
|
|
socket1.set_option(settable_socket_option1, ec);
|
|
socket1.set_option(settable_socket_option2);
|
|
socket1.set_option(settable_socket_option2, ec);
|
|
socket1.set_option(settable_socket_option3);
|
|
socket1.set_option(settable_socket_option3, ec);
|
|
|
|
socket1.get_option(gettable_socket_option1);
|
|
socket1.get_option(gettable_socket_option1, ec);
|
|
socket1.get_option(gettable_socket_option2);
|
|
socket1.get_option(gettable_socket_option2, ec);
|
|
socket1.get_option(gettable_socket_option3);
|
|
socket1.get_option(gettable_socket_option3, ec);
|
|
|
|
socket1.io_control(io_control_command);
|
|
socket1.io_control(io_control_command, ec);
|
|
|
|
bool non_blocking1 = socket1.non_blocking();
|
|
(void)non_blocking1;
|
|
socket1.non_blocking(true);
|
|
socket1.non_blocking(false, ec);
|
|
|
|
bool non_blocking2 = socket1.native_non_blocking();
|
|
(void)non_blocking2;
|
|
socket1.native_non_blocking(true);
|
|
socket1.native_non_blocking(false, ec);
|
|
|
|
ip::udp::endpoint endpoint1 = socket1.local_endpoint();
|
|
ip::udp::endpoint endpoint2 = socket1.local_endpoint(ec);
|
|
|
|
ip::udp::endpoint endpoint3 = socket1.remote_endpoint();
|
|
ip::udp::endpoint endpoint4 = socket1.remote_endpoint(ec);
|
|
|
|
socket1.shutdown(socket_base::shutdown_both);
|
|
socket1.shutdown(socket_base::shutdown_both, ec);
|
|
|
|
// basic_datagram_socket functions.
|
|
|
|
socket1.send(buffer(mutable_char_buffer));
|
|
socket1.send(buffer(const_char_buffer));
|
|
socket1.send(null_buffers());
|
|
socket1.send(buffer(mutable_char_buffer), in_flags);
|
|
socket1.send(buffer(const_char_buffer), in_flags);
|
|
socket1.send(null_buffers(), in_flags);
|
|
socket1.send(buffer(mutable_char_buffer), in_flags, ec);
|
|
socket1.send(buffer(const_char_buffer), in_flags, ec);
|
|
socket1.send(null_buffers(), in_flags, ec);
|
|
|
|
socket1.async_send(buffer(mutable_char_buffer), &send_handler);
|
|
socket1.async_send(buffer(const_char_buffer), &send_handler);
|
|
socket1.async_send(null_buffers(), &send_handler);
|
|
socket1.async_send(buffer(mutable_char_buffer), in_flags, &send_handler);
|
|
socket1.async_send(buffer(const_char_buffer), in_flags, &send_handler);
|
|
socket1.async_send(null_buffers(), in_flags, &send_handler);
|
|
|
|
socket1.send_to(buffer(mutable_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v4(), 0));
|
|
socket1.send_to(buffer(mutable_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v6(), 0));
|
|
socket1.send_to(buffer(const_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v4(), 0));
|
|
socket1.send_to(buffer(const_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v6(), 0));
|
|
socket1.send_to(null_buffers(),
|
|
ip::udp::endpoint(ip::udp::v4(), 0));
|
|
socket1.send_to(null_buffers(),
|
|
ip::udp::endpoint(ip::udp::v6(), 0));
|
|
socket1.send_to(buffer(mutable_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), in_flags);
|
|
socket1.send_to(buffer(mutable_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), in_flags);
|
|
socket1.send_to(buffer(const_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), in_flags);
|
|
socket1.send_to(buffer(const_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), in_flags);
|
|
socket1.send_to(null_buffers(),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), in_flags);
|
|
socket1.send_to(null_buffers(),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), in_flags);
|
|
socket1.send_to(buffer(mutable_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), in_flags, ec);
|
|
socket1.send_to(buffer(mutable_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), in_flags, ec);
|
|
socket1.send_to(buffer(const_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), in_flags, ec);
|
|
socket1.send_to(buffer(const_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), in_flags, ec);
|
|
socket1.send_to(null_buffers(),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), in_flags, ec);
|
|
socket1.send_to(null_buffers(),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), in_flags, ec);
|
|
|
|
socket1.async_send_to(buffer(mutable_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), &send_handler);
|
|
socket1.async_send_to(buffer(mutable_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), &send_handler);
|
|
socket1.async_send_to(buffer(const_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), &send_handler);
|
|
socket1.async_send_to(buffer(const_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), &send_handler);
|
|
socket1.async_send_to(null_buffers(),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), &send_handler);
|
|
socket1.async_send_to(null_buffers(),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), &send_handler);
|
|
socket1.async_send_to(buffer(mutable_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), in_flags, &send_handler);
|
|
socket1.async_send_to(buffer(mutable_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), in_flags, &send_handler);
|
|
socket1.async_send_to(buffer(const_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), in_flags, &send_handler);
|
|
socket1.async_send_to(buffer(const_char_buffer),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), in_flags, &send_handler);
|
|
socket1.async_send_to(null_buffers(),
|
|
ip::udp::endpoint(ip::udp::v4(), 0), in_flags, &send_handler);
|
|
socket1.async_send_to(null_buffers(),
|
|
ip::udp::endpoint(ip::udp::v6(), 0), in_flags, &send_handler);
|
|
|
|
socket1.receive(buffer(mutable_char_buffer));
|
|
socket1.receive(null_buffers());
|
|
socket1.receive(buffer(mutable_char_buffer), in_flags);
|
|
socket1.receive(null_buffers(), in_flags);
|
|
socket1.receive(buffer(mutable_char_buffer), in_flags, ec);
|
|
socket1.receive(null_buffers(), in_flags, ec);
|
|
|
|
socket1.async_receive(buffer(mutable_char_buffer), &receive_handler);
|
|
socket1.async_receive(null_buffers(), &receive_handler);
|
|
socket1.async_receive(buffer(mutable_char_buffer), in_flags,
|
|
&receive_handler);
|
|
socket1.async_receive(null_buffers(), in_flags, &receive_handler);
|
|
|
|
ip::udp::endpoint endpoint;
|
|
socket1.receive_from(buffer(mutable_char_buffer), endpoint);
|
|
socket1.receive_from(null_buffers(), endpoint);
|
|
socket1.receive_from(buffer(mutable_char_buffer), endpoint, in_flags);
|
|
socket1.receive_from(null_buffers(), endpoint, in_flags);
|
|
socket1.receive_from(buffer(mutable_char_buffer), endpoint, in_flags, ec);
|
|
socket1.receive_from(null_buffers(), endpoint, in_flags, ec);
|
|
|
|
socket1.async_receive_from(buffer(mutable_char_buffer),
|
|
endpoint, &receive_handler);
|
|
socket1.async_receive_from(null_buffers(),
|
|
endpoint, &receive_handler);
|
|
socket1.async_receive_from(buffer(mutable_char_buffer),
|
|
endpoint, in_flags, &receive_handler);
|
|
socket1.async_receive_from(null_buffers(),
|
|
endpoint, in_flags, &receive_handler);
|
|
}
|
|
catch (std::exception&)
|
|
{
|
|
}
|
|
}
|
|
|
|
} // namespace ip_udp_socket_compile
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
// ip_udp_socket_runtime test
|
|
// ~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
// The following test checks the runtime operation of the ip::udp::socket class.
|
|
|
|
namespace ip_udp_socket_runtime {
|
|
|
|
void handle_send(size_t expected_bytes_sent,
|
|
const boost::system::error_code& err, size_t bytes_sent)
|
|
{
|
|
BOOST_CHECK(!err);
|
|
BOOST_CHECK(expected_bytes_sent == bytes_sent);
|
|
}
|
|
|
|
void handle_recv(size_t expected_bytes_recvd,
|
|
const boost::system::error_code& err, size_t bytes_recvd)
|
|
{
|
|
BOOST_CHECK(!err);
|
|
BOOST_CHECK(expected_bytes_recvd == bytes_recvd);
|
|
}
|
|
|
|
void test()
|
|
{
|
|
using namespace std; // For memcmp and memset.
|
|
using namespace boost::asio;
|
|
namespace ip = boost::asio::ip;
|
|
|
|
io_service ios;
|
|
|
|
ip::udp::socket s1(ios, ip::udp::endpoint(ip::udp::v4(), 0));
|
|
ip::udp::endpoint target_endpoint = s1.local_endpoint();
|
|
target_endpoint.address(ip::address_v4::loopback());
|
|
|
|
ip::udp::socket s2(ios);
|
|
s2.open(ip::udp::v4());
|
|
s2.bind(ip::udp::endpoint(ip::udp::v4(), 0));
|
|
char send_msg[] = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
|
|
s2.send_to(buffer(send_msg, sizeof(send_msg)), target_endpoint);
|
|
|
|
char recv_msg[sizeof(send_msg)];
|
|
ip::udp::endpoint sender_endpoint;
|
|
size_t bytes_recvd = s1.receive_from(buffer(recv_msg, sizeof(recv_msg)),
|
|
sender_endpoint);
|
|
|
|
BOOST_CHECK(bytes_recvd == sizeof(send_msg));
|
|
BOOST_CHECK(memcmp(send_msg, recv_msg, sizeof(send_msg)) == 0);
|
|
|
|
memset(recv_msg, 0, sizeof(recv_msg));
|
|
|
|
target_endpoint = sender_endpoint;
|
|
s1.async_send_to(buffer(send_msg, sizeof(send_msg)), target_endpoint,
|
|
boost::bind(handle_send, sizeof(send_msg),
|
|
boost::asio::placeholders::error,
|
|
boost::asio::placeholders::bytes_transferred));
|
|
s2.async_receive_from(buffer(recv_msg, sizeof(recv_msg)), sender_endpoint,
|
|
boost::bind(handle_recv, sizeof(recv_msg),
|
|
boost::asio::placeholders::error,
|
|
boost::asio::placeholders::bytes_transferred));
|
|
|
|
ios.run();
|
|
|
|
BOOST_CHECK(memcmp(send_msg, recv_msg, sizeof(send_msg)) == 0);
|
|
}
|
|
|
|
} // namespace ip_udp_socket_runtime
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
// ip_udp_resolver_compile test
|
|
// ~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
// The following test checks that all public member functions on the class
|
|
// ip::udp::resolver compile and link correctly. Runtime failures are ignored.
|
|
|
|
namespace ip_udp_resolver_compile {
|
|
|
|
void resolve_handler(const boost::system::error_code&,
|
|
boost::asio::ip::udp::resolver::iterator)
|
|
{
|
|
}
|
|
|
|
void test()
|
|
{
|
|
using namespace boost::asio;
|
|
namespace ip = boost::asio::ip;
|
|
|
|
try
|
|
{
|
|
io_service ios;
|
|
boost::system::error_code ec;
|
|
ip::udp::resolver::query q(ip::udp::v4(), "localhost", "0");
|
|
ip::udp::endpoint e(ip::address_v4::loopback(), 0);
|
|
|
|
// basic_resolver constructors.
|
|
|
|
ip::udp::resolver resolver(ios);
|
|
|
|
// basic_io_object functions.
|
|
|
|
io_service& ios_ref = resolver.get_io_service();
|
|
(void)ios_ref;
|
|
|
|
// basic_resolver functions.
|
|
|
|
resolver.cancel();
|
|
|
|
ip::udp::resolver::iterator iter1 = resolver.resolve(q);
|
|
(void)iter1;
|
|
|
|
ip::udp::resolver::iterator iter2 = resolver.resolve(q, ec);
|
|
(void)iter2;
|
|
|
|
ip::udp::resolver::iterator iter3 = resolver.resolve(e);
|
|
(void)iter3;
|
|
|
|
ip::udp::resolver::iterator iter4 = resolver.resolve(e, ec);
|
|
(void)iter4;
|
|
|
|
resolver.async_resolve(q, &resolve_handler);
|
|
|
|
resolver.async_resolve(e, &resolve_handler);
|
|
}
|
|
catch (std::exception&)
|
|
{
|
|
}
|
|
}
|
|
|
|
} // namespace ip_udp_resolver_compile
|
|
|
|
//------------------------------------------------------------------------------
|
|
|
|
test_suite* init_unit_test_suite(int, char*[])
|
|
{
|
|
test_suite* test = BOOST_TEST_SUITE("ip/udp");
|
|
test->add(BOOST_TEST_CASE(&ip_udp_socket_compile::test));
|
|
test->add(BOOST_TEST_CASE(&ip_udp_socket_runtime::test));
|
|
test->add(BOOST_TEST_CASE(&ip_udp_resolver_compile::test));
|
|
return test;
|
|
}
|