/*------------------------------------------------------------------------------
 *
 * Copyright (c) 2011-2025, EURid vzw. All rights reserved.
 * The YADIFA TM software product is provided under the BSD 3-clause license:
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 *        * Redistributions of source code must retain the above copyright
 *          notice, this list of conditions and the following disclaimer.
 *        * Redistributions in binary form must reproduce the above copyright
 *          notice, this list of conditions and the following disclaimer in the
 *          documentation and/or other materials provided with the distribution.
 *        * Neither the name of EURid nor the names of its contributors may be
 *          used to endorse or promote products derived from this software
 *          without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 *----------------------------------------------------------------------------*/

/**-----------------------------------------------------------------------------
 * @defgroup network Network functions
 * @ingroup dnscore
 * @brief
 *
 * @{
 *----------------------------------------------------------------------------*/

#ifndef NETWORK_H
#define NETWORK_H

#include <sys/types.h> /* Required for BSD */
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>

#include <dnscore/sys_types.h>

#define SOCKADD4_TAG 0x344444414b434f53
#define SOCKADD6_TAG 0x364444414b434f53

/*
 * In order to avoid casting, this is the type that should be used to store sockaddr
 */

union socketaddress_u
{
    /// @note 20200629 edf -- In order to avoid a potential issue with the sockaddr structure in FreeBSD and possibly
    /// other OSes, I've removed the ss_* fields.
    //                        Every access must be done through "sa".

    struct sockaddr         sa;
    struct sockaddr_in      sa4;
    struct sockaddr_in6     sa6;
    struct sockaddr_storage ss;
    uint16_t                sa_family;
};

typedef union socketaddress_u socketaddress_t;

union socketaddressp_u
{
    /// @note 20200629 edf -- In order to avoid a potential issue with the sockaddr structure in FreeBSD and possibly
    /// other OSes, I've removed the ss_* fields.
    //                        Every access must be done through "sa".

    struct sockaddr         *sa;
    struct sockaddr_in      *sa4;
    struct sockaddr_in6     *sa6;
    struct sockaddr_storage *ss;
    socketaddress_t         *sat;
    uint16_t                *sa_family;
};

typedef union socketaddressp_u socketaddressp_t;

union socketaddresscp_u
{
    /// @note 20200629 edf -- In order to avoid a potential issue with the sockaddr structure in FreeBSD and possibly
    /// other OSes, I've removed the ss_* fields.
    //                        Every access must be done through "sa".

    const struct sockaddr         *sa;
    const struct sockaddr_in      *sa4;
    const struct sockaddr_in6     *sa6;
    const struct sockaddr_storage *ss;
    const socketaddress_t         *sat;
    const uint16_t                *sa_family;
};

typedef union socketaddresscp_u socketaddresscp_t;

static inline socklen_t         socketaddress_len(const socketaddress_t *sa)
{
    switch(sa->sa.sa_family)
    {
        case AF_INET:
            return sizeof(struct sockaddr_in);
        case AF_INET6:
            return sizeof(struct sockaddr_in6);
        default:
            return sizeof(struct sockaddr_storage);
    }
}

static inline socklen_t sockaddr_len(const struct sockaddr *sa)
{
    switch(sa->sa_family)
    {
        case AF_INET:
            return sizeof(struct sockaddr_in);
        case AF_INET6:
            return sizeof(struct sockaddr_in6);
        default:
            return sizeof(struct sockaddr_storage);
    }
}

// minimal storage for IPv4 & IPv6

union socketaddress_46
{
    struct sockaddr     sa;
    struct sockaddr_in  sa4;
    struct sockaddr_in6 sa6;
};

static inline bool sockaddr_equals(const struct sockaddr *a, const struct sockaddr *b)
{
    if(a->sa_family == b->sa_family)
    {
        switch(a->sa_family)
        {
            case AF_INET:
            {
                const struct sockaddr_in *sa4 = (const void *)a;
                const struct sockaddr_in *sb4 = (const void *)b;

                return memcmp(&sa4->sin_addr.s_addr, &sb4->sin_addr.s_addr, 4) == 0;
            }
            case AF_INET6:
            {

                const struct sockaddr_in6 *sa6 = (const void *)a;
                const struct sockaddr_in6 *sb6 = (const void *)b;

                return memcmp(&sa6->sin6_addr, &sb6->sin6_addr, 16) == 0;
            }
        }
    }

    return false;
}

int  sockaddr_compare_addr_port(const struct sockaddr *a, const struct sockaddr *b);

int  socketaddress_compare_ip(const void *a, const void *b);

void socketaddress_copy(socketaddress_t *dst, const socketaddress_t *src);

int  sockaddr_storage_compare_ip(const void *key_a, const void *key_b);

void sockaddr_storage_copy(struct sockaddr_storage *dest, const struct sockaddr_storage *src);

/**
 * Wrapper around accept that handles retries.
 */

int accept_ex(int sockfd, struct sockaddr *address, socklen_t *address_lenp);

/**
 * Wrapper around accept using poll that handles timeouts
 */

int                    accept_ex2(int sockfd, struct sockaddr *address, socklen_t *address_lenp, int timeout_ms);

static inline uint16_t sockaddr_port_ne(socketaddress_t *sa)
{
    switch(sa->sa.sa_family)
    {
        case AF_INET:
        {
            return sa->sa4.sin_port;
        }
        case AF_INET6:
        {
            return sa->sa6.sin6_port;
        }
        default:
        {
            return 0;
        }
    }
}

static inline uint16_t sockaddr_port(socketaddress_t *sa) { return ntohs(sockaddr_port_ne(sa)); }

static inline int32_t  sockaddr_inet_port(const struct sockaddr *sa)
{
    switch(sa->sa_family)
    {
        case AF_INET:
        {
            struct sockaddr_in *ipv4 = (void *)sa;
            return (uint32_t)ipv4->sin_port;
        }
        case AF_INET6:
        {
            struct sockaddr_in6 *ipv6 = (void *)sa;
            return (uint32_t)ipv6->sin6_port;
        }
        default:
        {
            return ERROR;
        }
    }
}

static inline void sockaddr_set_inet_port(struct sockaddr *sa, uint16_t ne_port)
{
    switch(sa->sa_family)
    {
        case AF_INET:
        {
            struct sockaddr_in *ipv4 = (void *)sa;
            ipv4->sin_port = ne_port;
            break;
        }
        case AF_INET6:
        {
            struct sockaddr_in6 *ipv6 = (void *)sa;
            ipv6->sin6_port = ne_port;
            break;
        }
        default:
        {
            break;
        }
    }
}

/**
 * Tool function to tell if a (listening) address is ANY
 */

bool      addr_info_is_any(struct addrinfo *addr);

ya_result socketaddress_init_parse_with_port(socketaddress_t *sa, const char *text, int port);

int       socket_server(struct sockaddr *sa, socklen_t sa_len, int family, int listen_queue_size);

/**
 * Enumerates all network interfaces.
 */

typedef ya_result(network_interfaces_forall_callback)(const char *itf_name, const socketaddress_t *ss, void *data);

ya_result network_interfaces_forall(network_interfaces_forall_callback *cb, void *data);

#endif /* HOST_ADDRESS_H */

/** @} */
