537 lines
13 KiB
C++
537 lines
13 KiB
C++
|
//
|
||
|
// socket_option.hpp
|
||
|
// ~~~~~~~~~~~~~~~~~
|
||
|
//
|
||
|
// Copyright (c) 2003-2007 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)
|
||
|
//
|
||
|
|
||
|
#ifndef ASIO_IP_DETAIL_SOCKET_OPTION_HPP
|
||
|
#define ASIO_IP_DETAIL_SOCKET_OPTION_HPP
|
||
|
|
||
|
#if defined(_MSC_VER) && (_MSC_VER >= 1200)
|
||
|
# pragma once
|
||
|
#endif // defined(_MSC_VER) && (_MSC_VER >= 1200)
|
||
|
|
||
|
#include "asio/detail/push_options.hpp"
|
||
|
|
||
|
#include "asio/detail/push_options.hpp"
|
||
|
#include <cstddef>
|
||
|
#include <cstring>
|
||
|
#include <boost/config.hpp>
|
||
|
#include "asio/detail/pop_options.hpp"
|
||
|
|
||
|
#include "asio/ip/address.hpp"
|
||
|
#include "asio/detail/socket_ops.hpp"
|
||
|
#include "asio/detail/socket_types.hpp"
|
||
|
|
||
|
namespace asio {
|
||
|
namespace ip {
|
||
|
namespace detail {
|
||
|
namespace socket_option {
|
||
|
|
||
|
// Helper template for implementing boolean-based options.
|
||
|
template <int IPv4_Level, int IPv4_Name, int IPv6_Level, int IPv6_Name>
|
||
|
class boolean
|
||
|
{
|
||
|
public:
|
||
|
#if defined(__sun)
|
||
|
typedef unsigned char value_type;
|
||
|
#else
|
||
|
typedef int value_type;
|
||
|
#endif
|
||
|
|
||
|
// Default constructor.
|
||
|
boolean()
|
||
|
: value_(0)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
// Construct with a specific option value.
|
||
|
explicit boolean(bool v)
|
||
|
: value_(v ? 1 : 0)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
// Set the value of the boolean.
|
||
|
boolean& operator=(bool v)
|
||
|
{
|
||
|
value_ = v ? 1 : 0;
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
// Get the current value of the boolean.
|
||
|
bool value() const
|
||
|
{
|
||
|
return !!value_;
|
||
|
}
|
||
|
|
||
|
// Convert to bool.
|
||
|
operator bool() const
|
||
|
{
|
||
|
return !!value_;
|
||
|
}
|
||
|
|
||
|
// Test for false.
|
||
|
bool operator!() const
|
||
|
{
|
||
|
return !value_;
|
||
|
}
|
||
|
|
||
|
// Get the level of the socket option.
|
||
|
template <typename Protocol>
|
||
|
int level(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return IPv6_Level;
|
||
|
return IPv4_Level;
|
||
|
}
|
||
|
|
||
|
// Get the name of the socket option.
|
||
|
template <typename Protocol>
|
||
|
int name(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return IPv6_Name;
|
||
|
return IPv4_Name;
|
||
|
}
|
||
|
|
||
|
// Get the address of the boolean data.
|
||
|
template <typename Protocol>
|
||
|
value_type* data(const Protocol&)
|
||
|
{
|
||
|
return &value_;
|
||
|
}
|
||
|
|
||
|
// Get the address of the boolean data.
|
||
|
template <typename Protocol>
|
||
|
const value_type* data(const Protocol&) const
|
||
|
{
|
||
|
return &value_;
|
||
|
}
|
||
|
|
||
|
// Get the size of the boolean data.
|
||
|
template <typename Protocol>
|
||
|
std::size_t size(const Protocol&) const
|
||
|
{
|
||
|
return sizeof(value_);
|
||
|
}
|
||
|
|
||
|
// Set the size of the boolean data.
|
||
|
template <typename Protocol>
|
||
|
void resize(const Protocol&, std::size_t s)
|
||
|
{
|
||
|
if (s != sizeof(value_))
|
||
|
throw std::length_error("boolean socket option resize");
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
value_type value_;
|
||
|
};
|
||
|
|
||
|
// Helper template for implementing unicast hops options.
|
||
|
template <int IPv4_Level, int IPv4_Name, int IPv6_Level, int IPv6_Name>
|
||
|
class unicast_hops
|
||
|
{
|
||
|
public:
|
||
|
// Default constructor.
|
||
|
unicast_hops()
|
||
|
: value_(0)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
// Construct with a specific option value.
|
||
|
explicit unicast_hops(int v)
|
||
|
: value_(v)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
// Set the value of the option.
|
||
|
unicast_hops& operator=(int v)
|
||
|
{
|
||
|
value_ = v;
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
// Get the current value of the option.
|
||
|
int value() const
|
||
|
{
|
||
|
return value_;
|
||
|
}
|
||
|
|
||
|
// Get the level of the socket option.
|
||
|
template <typename Protocol>
|
||
|
int level(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return IPv6_Level;
|
||
|
return IPv4_Level;
|
||
|
}
|
||
|
|
||
|
// Get the name of the socket option.
|
||
|
template <typename Protocol>
|
||
|
int name(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return IPv6_Name;
|
||
|
return IPv4_Name;
|
||
|
}
|
||
|
|
||
|
// Get the address of the data.
|
||
|
template <typename Protocol>
|
||
|
int* data(const Protocol&)
|
||
|
{
|
||
|
return &value_;
|
||
|
}
|
||
|
|
||
|
// Get the address of the data.
|
||
|
template <typename Protocol>
|
||
|
const int* data(const Protocol&) const
|
||
|
{
|
||
|
return &value_;
|
||
|
}
|
||
|
|
||
|
// Get the size of the data.
|
||
|
template <typename Protocol>
|
||
|
std::size_t size(const Protocol&) const
|
||
|
{
|
||
|
return sizeof(value_);
|
||
|
}
|
||
|
|
||
|
// Set the size of the data.
|
||
|
template <typename Protocol>
|
||
|
void resize(const Protocol&, std::size_t s)
|
||
|
{
|
||
|
if (s != sizeof(value_))
|
||
|
throw std::length_error("unicast hops socket option resize");
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
int value_;
|
||
|
};
|
||
|
|
||
|
// Helper template for implementing multicast hops options.
|
||
|
template <int IPv4_Level, int IPv4_Name, int IPv6_Level, int IPv6_Name>
|
||
|
class multicast_hops
|
||
|
{
|
||
|
public:
|
||
|
// Default constructor.
|
||
|
multicast_hops()
|
||
|
: ipv4_value_(0),
|
||
|
ipv6_value_(0)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
// Construct with a specific option value.
|
||
|
explicit multicast_hops(int v)
|
||
|
{
|
||
|
if (v < 0 || v > 255)
|
||
|
throw std::out_of_range("multicast hops value out of range");
|
||
|
ipv4_value_ = static_cast<unsigned char>(v);
|
||
|
ipv6_value_ = v;
|
||
|
}
|
||
|
|
||
|
// Set the value of the option.
|
||
|
multicast_hops& operator=(int v)
|
||
|
{
|
||
|
if (v < 0 || v > 255)
|
||
|
throw std::out_of_range("multicast hops value out of range");
|
||
|
ipv4_value_ = static_cast<unsigned char>(v);
|
||
|
ipv6_value_ = v;
|
||
|
return *this;
|
||
|
}
|
||
|
|
||
|
// Get the current value of the option.
|
||
|
int value() const
|
||
|
{
|
||
|
return ipv6_value_;
|
||
|
}
|
||
|
|
||
|
// Get the level of the socket option.
|
||
|
template <typename Protocol>
|
||
|
int level(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return IPv6_Level;
|
||
|
return IPv4_Level;
|
||
|
}
|
||
|
|
||
|
// Get the name of the socket option.
|
||
|
template <typename Protocol>
|
||
|
int name(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return IPv6_Name;
|
||
|
return IPv4_Name;
|
||
|
}
|
||
|
|
||
|
// Get the address of the data.
|
||
|
template <typename Protocol>
|
||
|
void* data(const Protocol& protocol)
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return &ipv6_value_;
|
||
|
return &ipv4_value_;
|
||
|
}
|
||
|
|
||
|
// Get the address of the data.
|
||
|
template <typename Protocol>
|
||
|
const void* data(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return &ipv6_value_;
|
||
|
return &ipv4_value_;
|
||
|
}
|
||
|
|
||
|
// Get the size of the data.
|
||
|
template <typename Protocol>
|
||
|
std::size_t size(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return sizeof(ipv6_value_);
|
||
|
return sizeof(ipv4_value_);
|
||
|
}
|
||
|
|
||
|
// Set the size of the data.
|
||
|
template <typename Protocol>
|
||
|
void resize(const Protocol& protocol, std::size_t s)
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
{
|
||
|
if (s != sizeof(ipv6_value_))
|
||
|
throw std::length_error("multicast hops socket option resize");
|
||
|
if (ipv6_value_ < 0)
|
||
|
ipv4_value_ = 0;
|
||
|
else if (ipv6_value_ > 255)
|
||
|
ipv4_value_ = 255;
|
||
|
else
|
||
|
ipv4_value_ = static_cast<unsigned char>(ipv6_value_);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
if (s != sizeof(ipv4_value_))
|
||
|
throw std::length_error("multicast hops socket option resize");
|
||
|
ipv6_value_ = ipv4_value_;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
unsigned char ipv4_value_;
|
||
|
int ipv6_value_;
|
||
|
};
|
||
|
|
||
|
// Helper template for implementing ip_mreq-based options.
|
||
|
template <int IPv4_Level, int IPv4_Name, int IPv6_Level, int IPv6_Name>
|
||
|
class multicast_request
|
||
|
{
|
||
|
public:
|
||
|
// Default constructor.
|
||
|
multicast_request()
|
||
|
{
|
||
|
ipv4_value_.imr_multiaddr.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
asio::ip::address_v4::any().to_ulong());
|
||
|
ipv4_value_.imr_interface.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
asio::ip::address_v4::any().to_ulong());
|
||
|
|
||
|
asio::detail::in6_addr_type tmp_addr = IN6ADDR_ANY_INIT;
|
||
|
ipv6_value_.ipv6mr_multiaddr = tmp_addr;
|
||
|
ipv6_value_.ipv6mr_interface = 0;
|
||
|
}
|
||
|
|
||
|
// Construct with multicast address only.
|
||
|
explicit multicast_request(const asio::ip::address& multicast_address)
|
||
|
{
|
||
|
if (multicast_address.is_v6())
|
||
|
{
|
||
|
ipv4_value_.imr_multiaddr.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
asio::ip::address_v4::any().to_ulong());
|
||
|
ipv4_value_.imr_interface.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
asio::ip::address_v4::any().to_ulong());
|
||
|
|
||
|
using namespace std; // For memcpy.
|
||
|
asio::ip::address_v6 ipv6_address = multicast_address.to_v6();
|
||
|
asio::ip::address_v6::bytes_type bytes = ipv6_address.to_bytes();
|
||
|
memcpy(ipv6_value_.ipv6mr_multiaddr.s6_addr, bytes.elems, 16);
|
||
|
ipv6_value_.ipv6mr_interface = 0;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
ipv4_value_.imr_multiaddr.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
multicast_address.to_v4().to_ulong());
|
||
|
ipv4_value_.imr_interface.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
asio::ip::address_v4::any().to_ulong());
|
||
|
|
||
|
asio::detail::in6_addr_type tmp_addr = IN6ADDR_ANY_INIT;
|
||
|
ipv6_value_.ipv6mr_multiaddr = tmp_addr;
|
||
|
ipv6_value_.ipv6mr_interface = 0;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// Construct with multicast address and IPv4 address specifying an interface.
|
||
|
explicit multicast_request(
|
||
|
const asio::ip::address_v4& multicast_address,
|
||
|
const asio::ip::address_v4& network_interface
|
||
|
= asio::ip::address_v4::any())
|
||
|
{
|
||
|
ipv4_value_.imr_multiaddr.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
multicast_address.to_ulong());
|
||
|
ipv4_value_.imr_interface.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
network_interface.to_ulong());
|
||
|
|
||
|
asio::detail::in6_addr_type tmp_addr = IN6ADDR_ANY_INIT;
|
||
|
ipv6_value_.ipv6mr_multiaddr = tmp_addr;
|
||
|
ipv6_value_.ipv6mr_interface = 0;
|
||
|
}
|
||
|
|
||
|
// Construct with multicast address and IPv6 network interface index.
|
||
|
explicit multicast_request(
|
||
|
const asio::ip::address_v6& multicast_address,
|
||
|
unsigned long network_interface = 0)
|
||
|
{
|
||
|
ipv4_value_.imr_multiaddr.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
asio::ip::address_v4::any().to_ulong());
|
||
|
ipv4_value_.imr_interface.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
asio::ip::address_v4::any().to_ulong());
|
||
|
|
||
|
using namespace std; // For memcpy.
|
||
|
asio::ip::address_v6::bytes_type bytes =
|
||
|
multicast_address.to_bytes();
|
||
|
memcpy(ipv6_value_.ipv6mr_multiaddr.s6_addr, bytes.elems, 16);
|
||
|
ipv6_value_.ipv6mr_interface = network_interface;
|
||
|
}
|
||
|
|
||
|
// Get the level of the socket option.
|
||
|
template <typename Protocol>
|
||
|
int level(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return IPv6_Level;
|
||
|
return IPv4_Level;
|
||
|
}
|
||
|
|
||
|
// Get the name of the socket option.
|
||
|
template <typename Protocol>
|
||
|
int name(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return IPv6_Name;
|
||
|
return IPv4_Name;
|
||
|
}
|
||
|
|
||
|
// Get the address of the option data.
|
||
|
template <typename Protocol>
|
||
|
const void* data(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return &ipv6_value_;
|
||
|
return &ipv4_value_;
|
||
|
}
|
||
|
|
||
|
// Get the size of the option data.
|
||
|
template <typename Protocol>
|
||
|
std::size_t size(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return sizeof(ipv6_value_);
|
||
|
return sizeof(ipv4_value_);
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
asio::detail::in4_mreq_type ipv4_value_;
|
||
|
asio::detail::in6_mreq_type ipv6_value_;
|
||
|
};
|
||
|
|
||
|
// Helper template for implementing options that specify a network interface.
|
||
|
template <int IPv4_Level, int IPv4_Name, int IPv6_Level, int IPv6_Name>
|
||
|
class network_interface
|
||
|
{
|
||
|
public:
|
||
|
// Default constructor.
|
||
|
network_interface()
|
||
|
{
|
||
|
ipv4_value_.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
asio::ip::address_v4::any().to_ulong());
|
||
|
ipv6_value_ = 0;
|
||
|
}
|
||
|
|
||
|
// Construct with IPv4 interface.
|
||
|
explicit network_interface(const asio::ip::address_v4& ipv4_interface)
|
||
|
{
|
||
|
ipv4_value_.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
ipv4_interface.to_ulong());
|
||
|
ipv6_value_ = 0;
|
||
|
}
|
||
|
|
||
|
// Construct with IPv6 interface.
|
||
|
explicit network_interface(unsigned long ipv6_interface)
|
||
|
{
|
||
|
ipv4_value_.s_addr =
|
||
|
asio::detail::socket_ops::host_to_network_long(
|
||
|
asio::ip::address_v4::any().to_ulong());
|
||
|
ipv6_value_ = ipv6_interface;
|
||
|
}
|
||
|
|
||
|
// Get the level of the socket option.
|
||
|
template <typename Protocol>
|
||
|
int level(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return IPv6_Level;
|
||
|
return IPv4_Level;
|
||
|
}
|
||
|
|
||
|
// Get the name of the socket option.
|
||
|
template <typename Protocol>
|
||
|
int name(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return IPv6_Name;
|
||
|
return IPv4_Name;
|
||
|
}
|
||
|
|
||
|
// Get the address of the option data.
|
||
|
template <typename Protocol>
|
||
|
const void* data(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return &ipv6_value_;
|
||
|
return &ipv4_value_;
|
||
|
}
|
||
|
|
||
|
// Get the size of the option data.
|
||
|
template <typename Protocol>
|
||
|
std::size_t size(const Protocol& protocol) const
|
||
|
{
|
||
|
if (protocol.family() == PF_INET6)
|
||
|
return sizeof(ipv6_value_);
|
||
|
return sizeof(ipv4_value_);
|
||
|
}
|
||
|
|
||
|
private:
|
||
|
asio::detail::in4_addr_type ipv4_value_;
|
||
|
unsigned long ipv6_value_;
|
||
|
};
|
||
|
|
||
|
} // namespace socket_option
|
||
|
} // namespace detail
|
||
|
} // namespace ip
|
||
|
} // namespace asio
|
||
|
|
||
|
#include "asio/detail/pop_options.hpp"
|
||
|
|
||
|
#endif // ASIO_IP_DETAIL_SOCKET_OPTION_HPP
|