From a4dd0ad63c00f4dee3b86dfd3075d1d61b2b3180 Mon Sep 17 00:00:00 2001 From: sanine Date: Sat, 27 Aug 2022 23:52:56 -0500 Subject: add plibsys --- 3rdparty/plibsys/src/psocketaddress.c | 619 ++++++++++++++++++++++++++++++++++ 1 file changed, 619 insertions(+) create mode 100644 3rdparty/plibsys/src/psocketaddress.c (limited to '3rdparty/plibsys/src/psocketaddress.c') diff --git a/3rdparty/plibsys/src/psocketaddress.c b/3rdparty/plibsys/src/psocketaddress.c new file mode 100644 index 0000000..2338c83 --- /dev/null +++ b/3rdparty/plibsys/src/psocketaddress.c @@ -0,0 +1,619 @@ +/* + * The MIT License + * + * Copyright (C) 2010-2016 Alexander Saprykin + * + * Permission is hereby granted, free of charge, to any person obtaining + * a copy of this software and associated documentation files (the + * 'Software'), to deal in the Software without restriction, including + * without limitation the rights to use, copy, modify, merge, publish, + * distribute, sublicense, and/or sell copies of the Software, and to + * permit persons to whom the Software is furnished to do so, subject to + * the following conditions: + * + * The above copyright notice and this permission notice shall be + * included in all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED 'AS IS', WITHOUT WARRANTY OF ANY KIND, + * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF + * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. + * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY + * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, + * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE + * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. + */ + +#include "pmem.h" +#include "pstring.h" +#include "psocketaddress.h" +#include "plibsys-private.h" + +#include +#include + +#ifndef P_OS_WIN +# include +#endif + +#if defined (PLIBSYS_HAS_GETADDRINFO) && !defined (PLIBSYS_SOCKADDR_IN6_HAS_SCOPEID) +# undef PLIBSYS_HAS_GETADDRINFO +#endif + +#ifdef PLIBSYS_HAS_GETADDRINFO +# include +#endif + +/* According to Open Group specifications */ +#ifndef INET_ADDRSTRLEN +# ifdef P_OS_WIN + /* On Windows it includes port number */ +# define INET_ADDRSTRLEN 22 +# else +# define INET_ADDRSTRLEN 16 +# endif +#endif + +#ifdef AF_INET6 +# ifndef INET6_ADDRSTRLEN +# ifdef P_OS_WIN + /* On Windows it includes port number */ +# define INET6_ADDRSTRLEN 65 +# else +# define INET6_ADDRSTRLEN 46 +# endif +# endif +#endif + +#ifdef P_OS_VMS +# if PLIBSYS_SIZEOF_VOID_P == 8 +# define addrinfo __addrinfo64 +# endif +#endif + +#if defined (P_OS_BEOS) || defined (P_OS_OS2) +# ifdef AF_INET6 +# undef AF_INET6 +# endif +#endif + +struct PSocketAddress_ { + PSocketFamily family; + union addr_ { + struct in_addr sin_addr; +#ifdef AF_INET6 + struct in6_addr sin6_addr; +#endif + } addr; + puint16 port; + puint32 flowinfo; + puint32 scope_id; +}; + +P_LIB_API PSocketAddress * +p_socket_address_new_from_native (pconstpointer native, + psize len) +{ + PSocketAddress *ret; + puint16 family; + + if (P_UNLIKELY (native == NULL || len == 0)) + return NULL; + + if (P_UNLIKELY ((ret = p_malloc0 (sizeof (PSocketAddress))) == NULL)) + return NULL; + + family = ((struct sockaddr *) native)->sa_family; + + if (family == AF_INET) { + if (len < sizeof (struct sockaddr_in)) { + P_WARNING ("PSocketAddress::p_socket_address_new_from_native: invalid IPv4 native size"); + p_free (ret); + return NULL; + } + + memcpy (&ret->addr.sin_addr, &((struct sockaddr_in *) native)->sin_addr, sizeof (struct in_addr)); + ret->family = P_SOCKET_FAMILY_INET; + ret->port = p_ntohs (((struct sockaddr_in *) native)->sin_port); + return ret; + } +#ifdef AF_INET6 + else if (family == AF_INET6) { + if (len < sizeof (struct sockaddr_in6)) { + P_WARNING ("PSocketAddress::p_socket_address_new_from_native: invalid IPv6 native size"); + p_free (ret); + return NULL; + } + + memcpy (&ret->addr.sin6_addr, + &((struct sockaddr_in6 *) native)->sin6_addr, + sizeof (struct in6_addr)); + + ret->family = P_SOCKET_FAMILY_INET6; + ret->port = p_ntohs (((struct sockaddr_in *) native)->sin_port); +#ifdef PLIBSYS_SOCKADDR_IN6_HAS_FLOWINFO + ret->flowinfo = ((struct sockaddr_in6 *) native)->sin6_flowinfo; +#endif +#ifdef PLIBSYS_SOCKADDR_IN6_HAS_SCOPEID + ret->scope_id = ((struct sockaddr_in6 *) native)->sin6_scope_id; +#endif + return ret; + } +#endif + else { + p_free (ret); + return NULL; + } +} + +P_LIB_API PSocketAddress * +p_socket_address_new (const pchar *address, + puint16 port) +{ + PSocketAddress *ret; +#if defined (P_OS_WIN) || defined (PLIBSYS_HAS_GETADDRINFO) + struct addrinfo hints; + struct addrinfo *res; +#endif + +#ifdef P_OS_WIN + struct sockaddr_storage sa; + struct sockaddr_in *sin = (struct sockaddr_in *) &sa; +# ifdef AF_INET6 + struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *) &sa; +# endif /* AF_INET6 */ + pint len; +#endif /* P_OS_WIN */ + + if (P_UNLIKELY (address == NULL)) + return NULL; + +#if (defined (P_OS_WIN) || defined (PLIBSYS_HAS_GETADDRINFO)) && defined (AF_INET6) + if (strchr (address, ':') != NULL) { + memset (&hints, 0, sizeof (hints)); + + hints.ai_family = AF_UNSPEC; + hints.ai_socktype = SOCK_STREAM; + hints.ai_protocol = 0; +# ifndef P_OS_UNIXWARE + hints.ai_flags = AI_NUMERICHOST; +# endif + + if (P_UNLIKELY (getaddrinfo (address, NULL, &hints, &res) != 0)) + return NULL; + + if (P_LIKELY (res->ai_family == AF_INET6 && + res->ai_addrlen == sizeof (struct sockaddr_in6))) { + ((struct sockaddr_in6 *) res->ai_addr)->sin6_port = p_htons (port); + ret = p_socket_address_new_from_native (res->ai_addr, res->ai_addrlen); + } else + ret = NULL; + + freeaddrinfo (res); + + return ret; + } +#endif + + if (P_UNLIKELY ((ret = p_malloc0 (sizeof (PSocketAddress))) == NULL)) { + P_ERROR ("PSocketAddress::p_socket_address_new: failed to allocate memory"); + return NULL; + } + + ret->port = port; + +#ifdef P_OS_WIN + memset (&sa, 0, sizeof (sa)); + len = sizeof (sa); + sin->sin_family = AF_INET; + + if (WSAStringToAddressA ((LPSTR) address, AF_INET, NULL, (LPSOCKADDR) &sa, &len) == 0) { + memcpy (&ret->addr.sin_addr, &sin->sin_addr, sizeof (struct in_addr)); + ret->family = P_SOCKET_FAMILY_INET; + return ret; + } +# ifdef AF_INET6 + else { + sin6->sin6_family = AF_INET6; + + if (WSAStringToAddressA ((LPSTR) address, AF_INET6, NULL, (LPSOCKADDR) &sa, &len) == 0) { + memcpy (&ret->addr.sin6_addr, &sin6->sin6_addr, sizeof (struct in6_addr)); + ret->family = P_SOCKET_FAMILY_INET6; + return ret; + } + } +# endif /* AF_INET6 */ +#else /* P_OS_WIN */ + if (inet_pton (AF_INET, address, &ret->addr.sin_addr) > 0) { + ret->family = P_SOCKET_FAMILY_INET; + return ret; + } +# ifdef AF_INET6 + else if (inet_pton (AF_INET6, address, &ret->addr.sin6_addr) > 0) { + ret->family = P_SOCKET_FAMILY_INET6; + return ret; + } +# endif /* AF_INET6 */ +#endif /* P_OS_WIN */ + + p_free (ret); + return NULL; +} + +P_LIB_API PSocketAddress * +p_socket_address_new_any (PSocketFamily family, + puint16 port) +{ + PSocketAddress *ret; + puchar any_addr[] = {0, 0, 0, 0}; +#ifdef AF_INET6 + struct in6_addr any6_addr = IN6ADDR_ANY_INIT; +#endif + + if (P_UNLIKELY ((ret = p_malloc0 (sizeof (PSocketAddress))) == NULL)) { + P_ERROR ("PSocketAddress::p_socket_address_new_any: failed to allocate memory"); + return NULL; + } + + if (family == P_SOCKET_FAMILY_INET) + memcpy (&ret->addr.sin_addr, any_addr, sizeof (any_addr)); +#ifdef AF_INET6 + else if (family == P_SOCKET_FAMILY_INET6) + memcpy (&ret->addr.sin6_addr, &any6_addr.s6_addr, sizeof (any6_addr.s6_addr)); +#endif + else { + p_free (ret); + return NULL; + } + + ret->family = family; + ret->port = port; + + return ret; +} + +P_LIB_API PSocketAddress * +p_socket_address_new_loopback (PSocketFamily family, + puint16 port) +{ + PSocketAddress *ret; + puchar loop_addr[] = {127, 0, 0, 0}; +#ifdef AF_INET6 + struct in6_addr loop6_addr = IN6ADDR_LOOPBACK_INIT; +#endif + + if (P_UNLIKELY ((ret = p_malloc0 (sizeof (PSocketAddress))) == NULL)) { + P_ERROR ("PSocketAddress::p_socket_address_new_loopback: failed to allocate memory"); + return NULL; + } + + if (family == P_SOCKET_FAMILY_INET) + memcpy (&ret->addr.sin_addr, loop_addr, sizeof (loop_addr)); +#ifdef AF_INET6 + else if (family == P_SOCKET_FAMILY_INET6) + memcpy (&ret->addr.sin6_addr, &loop6_addr.s6_addr, sizeof (loop6_addr.s6_addr)); +#endif + else { + p_free (ret); + return NULL; + } + + ret->family = family; + ret->port = port; + + return ret; +} + +P_LIB_API pboolean +p_socket_address_to_native (const PSocketAddress *addr, + ppointer dest, + psize destlen) +{ + struct sockaddr_in *sin; +#ifdef AF_INET6 + struct sockaddr_in6 *sin6; +#endif + + if (P_UNLIKELY (addr == NULL || dest == NULL || destlen == 0)) + return FALSE; + + sin = (struct sockaddr_in *) dest; +#ifdef AF_INET6 + sin6 = (struct sockaddr_in6 *) dest; +#endif + + if (addr->family == P_SOCKET_FAMILY_INET) { + if (P_UNLIKELY (destlen < sizeof (struct sockaddr_in))) { + P_WARNING ("PSocketAddress::p_socket_address_to_native: invalid buffer size for IPv4"); + return FALSE; + } + + memcpy (&sin->sin_addr, &addr->addr.sin_addr, sizeof (struct in_addr)); + sin->sin_family = AF_INET; + sin->sin_port = p_htons (addr->port); + memset (sin->sin_zero, 0, sizeof (sin->sin_zero)); + return TRUE; + } +#ifdef AF_INET6 + else if (addr->family == P_SOCKET_FAMILY_INET6) { + if (P_UNLIKELY (destlen < sizeof (struct sockaddr_in6))) { + P_ERROR ("PSocketAddress::p_socket_address_to_native: invalid buffer size for IPv6"); + return FALSE; + } + + memcpy (&sin6->sin6_addr, &addr->addr.sin6_addr, sizeof (struct in6_addr)); + sin6->sin6_family = AF_INET6; + sin6->sin6_port = p_htons (addr->port); +#ifdef PLIBSYS_SOCKADDR_IN6_HAS_FLOWINFO + sin6->sin6_flowinfo = addr->flowinfo; +#endif +#ifdef PLIBSYS_SOCKADDR_IN6_HAS_SCOPEID + sin6->sin6_scope_id = addr->scope_id; +#endif + return TRUE; + } +#endif + else { + P_WARNING ("PSocketAddress::p_socket_address_to_native: unsupported socket address"); + return FALSE; + } +} + +P_LIB_API psize +p_socket_address_get_native_size (const PSocketAddress *addr) +{ + if (P_UNLIKELY (addr == NULL)) + return 0; + + if (addr->family == P_SOCKET_FAMILY_INET) + return sizeof (struct sockaddr_in); +#ifdef AF_INET6 + else if (addr->family == P_SOCKET_FAMILY_INET6) + return sizeof (struct sockaddr_in6); +#endif + else { + P_WARNING ("PSocketAddress::p_socket_address_get_native_size: unsupported socket family"); + return 0; + } +} + +P_LIB_API PSocketFamily +p_socket_address_get_family (const PSocketAddress *addr) +{ + if (P_UNLIKELY (addr == NULL)) + return P_SOCKET_FAMILY_UNKNOWN; + + return addr->family; +} + +P_LIB_API pchar * +p_socket_address_get_address (const PSocketAddress *addr) +{ +#ifdef AF_INET6 + pchar buffer[INET6_ADDRSTRLEN]; +#else + pchar buffer[INET_ADDRSTRLEN]; +#endif + +#ifdef P_OS_WIN + DWORD buflen = sizeof (buffer); + DWORD addrlen; + struct sockaddr_storage sa; + struct sockaddr_in *sin; +# ifdef AF_INET6 + struct sockaddr_in6 *sin6; +# endif /* AF_INET6 */ +#endif /* P_OS_WIN */ + + if (P_UNLIKELY (addr == NULL || addr->family == P_SOCKET_FAMILY_UNKNOWN)) + return NULL; +#ifdef P_OS_WIN + sin = (struct sockaddr_in *) &sa; +# ifdef AF_INET6 + sin6 = (struct sockaddr_in6 *) &sa; +# endif /* AF_INET6 */ +#endif /* P_OS_WIN */ + +#ifdef P_OS_WIN + memset (&sa, 0, sizeof (sa)); +#endif + +#ifdef P_OS_WIN + sa.ss_family = addr->family; + + if (addr->family == P_SOCKET_FAMILY_INET) { + addrlen = sizeof (struct sockaddr_in); + memcpy (&sin->sin_addr, &addr->addr.sin_addr, sizeof (struct in_addr)); + } +# ifdef AF_INET6 + else { + addrlen = sizeof (struct sockaddr_in6); + memcpy (&sin6->sin6_addr, &addr->addr.sin6_addr, sizeof (struct in6_addr)); + } +# endif /* AF_INET6 */ + + if (P_UNLIKELY (WSAAddressToStringA ((LPSOCKADDR) &sa, + addrlen, + NULL, + (LPSTR) buffer, + &buflen) != 0)) + return NULL; +#else /* !P_OS_WIN */ + if (addr->family == P_SOCKET_FAMILY_INET) + inet_ntop (AF_INET, &addr->addr.sin_addr, buffer, sizeof (buffer)); +# ifdef AF_INET6 + else + inet_ntop (AF_INET6, &addr->addr.sin6_addr, buffer, sizeof (buffer)); +# endif /* AF_INET6 */ +#endif /* P_OS_WIN */ + + return p_strdup (buffer); +} + +P_LIB_API puint16 +p_socket_address_get_port (const PSocketAddress *addr) +{ + if (P_UNLIKELY (addr == NULL)) + return 0; + + return addr->port; +} + +P_LIB_API puint32 +p_socket_address_get_flow_info (const PSocketAddress *addr) +{ + if (P_UNLIKELY (addr == NULL)) + return 0; + +#if !defined (AF_INET6) || !defined (PLIBSYS_SOCKADDR_IN6_HAS_FLOWINFO) + return 0; +#else + if (P_UNLIKELY (addr->family != P_SOCKET_FAMILY_INET6)) + return 0; + + return addr->flowinfo; +#endif +} + +P_LIB_API puint32 +p_socket_address_get_scope_id (const PSocketAddress *addr) +{ + if (P_UNLIKELY (addr == NULL)) + return 0; + +#if !defined (AF_INET6) || !defined (PLIBSYS_SOCKADDR_IN6_HAS_SCOPEID) + return 0; +#else + if (P_UNLIKELY (addr->family != P_SOCKET_FAMILY_INET6)) + return 0; + + return addr->scope_id; +#endif +} + +P_LIB_API void +p_socket_address_set_flow_info (PSocketAddress *addr, + puint32 flowinfo) +{ + if (P_UNLIKELY (addr == NULL)) + return; + +#if !defined (AF_INET6) || !defined (PLIBSYS_SOCKADDR_IN6_HAS_FLOWINFO) + P_UNUSED (flowinfo); + return; +#else + if (P_UNLIKELY (addr->family != P_SOCKET_FAMILY_INET6)) + return; + + addr->flowinfo = flowinfo; +#endif +} + +P_LIB_API void +p_socket_address_set_scope_id (PSocketAddress *addr, + puint32 scope_id) +{ + if (P_UNLIKELY (addr == NULL)) + return; + +#if !defined (AF_INET6) || !defined (PLIBSYS_SOCKADDR_IN6_HAS_SCOPEID) + P_UNUSED (scope_id); + return; +#else + if (P_UNLIKELY (addr->family != P_SOCKET_FAMILY_INET6)) + return; + + addr->scope_id = scope_id; +#endif +} + +P_LIB_API pboolean +p_socket_address_is_flow_info_supported (void) +{ +#ifdef AF_INET6 +# ifdef PLIBSYS_SOCKADDR_IN6_HAS_FLOWINFO + return TRUE; +# else + return FALSE; +# endif +#else + return FALSE; +#endif +} + +P_LIB_API pboolean +p_socket_address_is_scope_id_supported (void) +{ +#ifdef AF_INET6 +# ifdef PLIBSYS_SOCKADDR_IN6_HAS_SCOPEID + return TRUE; +# else + return FALSE; +# endif +#else + return FALSE; +#endif +} + +P_LIB_API pboolean +p_socket_address_is_ipv6_supported (void) +{ +#ifdef AF_INET6 + return TRUE; +#else + return FALSE; +#endif +} + +P_LIB_API pboolean +p_socket_address_is_any (const PSocketAddress *addr) +{ + puint32 addr4; + + if (P_UNLIKELY (addr == NULL || addr->family == P_SOCKET_FAMILY_UNKNOWN)) + return FALSE; + + if (addr->family == P_SOCKET_FAMILY_INET) { + addr4 = p_ntohl (* ((puint32 *) &addr->addr.sin_addr)); + + return (addr4 == INADDR_ANY); + } +#ifdef AF_INET6 + else + return IN6_IS_ADDR_UNSPECIFIED (&addr->addr.sin6_addr); +#else + else + return FALSE; +#endif +} + +P_LIB_API pboolean +p_socket_address_is_loopback (const PSocketAddress *addr) +{ + puint32 addr4; + + if (P_UNLIKELY (addr == NULL || addr->family == P_SOCKET_FAMILY_UNKNOWN)) + return FALSE; + + if (addr->family == P_SOCKET_FAMILY_INET) { + addr4 = p_ntohl (* ((puint32 *) &addr->addr.sin_addr)); + + /* 127.0.0.0/8 */ + return ((addr4 & 0xff000000) == 0x7f000000); + } +#ifdef AF_INET6 + else + return IN6_IS_ADDR_LOOPBACK (&addr->addr.sin6_addr); +#else + else + return FALSE; +#endif +} + +P_LIB_API void +p_socket_address_free (PSocketAddress *addr) +{ + if (P_UNLIKELY (addr == NULL)) + return; + + p_free (addr); +} -- cgit v1.2.1