jeanlemotan
86 lines
2.2 KiB
C++
86 lines
2.2 KiB
C++
///////////////////////////////////////////////////////////////////////////////
|
|
// Copyright (c) Lewis Baker
|
|
// Licenced under MIT license. See LICENSE.txt for details.
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include "socket_helpers.hpp"
|
|
|
|
#include <cppcoro/net/ip_endpoint.hpp>
|
|
|
|
#if CPPCORO_OS_WINNT
|
|
#include <cstring>
|
|
#include <cassert>
|
|
|
|
#include <WinSock2.h>
|
|
#include <WS2tcpip.h>
|
|
#include <MSWSock.h>
|
|
#include <Windows.h>
|
|
|
|
|
|
cppcoro::net::ip_endpoint
|
|
cppcoro::net::detail::sockaddr_to_ip_endpoint(const sockaddr& address) noexcept
|
|
{
|
|
if (address.sa_family == AF_INET)
|
|
{
|
|
SOCKADDR_IN ipv4Address;
|
|
std::memcpy(&ipv4Address, &address, sizeof(ipv4Address));
|
|
|
|
std::uint8_t addressBytes[4];
|
|
std::memcpy(addressBytes, &ipv4Address.sin_addr, 4);
|
|
|
|
return ipv4_endpoint{
|
|
ipv4_address{ addressBytes },
|
|
ntohs(ipv4Address.sin_port)
|
|
};
|
|
}
|
|
else
|
|
{
|
|
assert(address.sa_family == AF_INET6);
|
|
|
|
SOCKADDR_IN6 ipv6Address;
|
|
std::memcpy(&ipv6Address, &address, sizeof(ipv6Address));
|
|
|
|
return ipv6_endpoint{
|
|
ipv6_address{ ipv6Address.sin6_addr.u.Byte },
|
|
ntohs(ipv6Address.sin6_port)
|
|
};
|
|
}
|
|
}
|
|
|
|
int cppcoro::net::detail::ip_endpoint_to_sockaddr(
|
|
const ip_endpoint& endPoint,
|
|
std::reference_wrapper<sockaddr_storage> address) noexcept
|
|
{
|
|
if (endPoint.is_ipv4())
|
|
{
|
|
const auto& ipv4EndPoint = endPoint.to_ipv4();
|
|
|
|
SOCKADDR_IN ipv4Address;
|
|
ipv4Address.sin_family = AF_INET;
|
|
std::memcpy(&ipv4Address.sin_addr, ipv4EndPoint.address().bytes(), 4);
|
|
ipv4Address.sin_port = htons(ipv4EndPoint.port());
|
|
std::memset(&ipv4Address.sin_zero, 0, sizeof(ipv4Address.sin_zero));
|
|
|
|
std::memcpy(&address.get(), &ipv4Address, sizeof(ipv4Address));
|
|
|
|
return sizeof(SOCKADDR_IN);
|
|
}
|
|
else
|
|
{
|
|
const auto& ipv6EndPoint = endPoint.to_ipv6();
|
|
|
|
SOCKADDR_IN6 ipv6Address;
|
|
ipv6Address.sin6_family = AF_INET6;
|
|
std::memcpy(&ipv6Address.sin6_addr, ipv6EndPoint.address().bytes(), 16);
|
|
ipv6Address.sin6_port = htons(ipv6EndPoint.port());
|
|
ipv6Address.sin6_flowinfo = 0;
|
|
ipv6Address.sin6_scope_struct = SCOPEID_UNSPECIFIED_INIT;
|
|
|
|
std::memcpy(&address.get(), &ipv6Address, sizeof(ipv6Address));
|
|
|
|
return sizeof(SOCKADDR_IN6);
|
|
}
|
|
}
|
|
|
|
#endif // CPPCORO_OS_WINNT
|