First

lib/socket_helpers.cpp

@@ -0,0 +1,85 @@
+///////////////////////////////////////////////////////////////////////////////
+// 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