/*
 * Copyright (C) Copyright (C) 2021 ML!PA Consulting GmbH
 *
 * This file is subject to the terms and conditions of the GNU Lesser
 * General Public License v2.1. See the file LICENSE in the top level
 * directory for more details.
 */

/**
 * @{
 *
 * @file
 *
 * @author      Benjamin Valentin <benjamin.valentin@ml-pa.com>
 */
#include <errno.h>
#include <stdint.h>

#include "embUnit.h"

#include "net/dns_mock.h"
#include "net/gnrc/netif.h"
#include "net/sock/util.h"
#include "net/utils.h"

static gnrc_netif_t dummy_netif[2];

static void test_ipv6_addr_from_str__one_colon_start(void)
{
    ipv6_addr_t address;
    netif_t *netif;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv6(&address, &netif, ":ff::1"), -EINVAL);
}

static void test_ipv6_addr_from_str__three_colons(void)
{
    ipv6_addr_t address;
    netif_t *netif;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv6(&address, &netif, "ff02:::1"), -EINVAL);
}

static void test_ipv6_addr_from_str__illegal_chars(void)
{
    ipv6_addr_t address;
    netif_t *netif;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv6(&address, &netif, ":-D"), -EINVAL);
}

static void test_ipv6_addr_from_str__addr_NULL(void)
{
    ipv6_addr_t address;
    netif_t *netif;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv6(&address, &netif, NULL), -EINVAL);
}

static void test_ipv6_addr_from_str__address_NULL(void)
{
    TEST_ASSERT_NULL(ipv6_addr_from_str(NULL, "::"));
}

static void test_ipv6_addr_from_str__success(void)
{
    static const ipv6_addr_t a = { {
            0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
            0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f
        }
    };
    ipv6_addr_t address;
    netif_t *netif;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv6(&address, &netif, "1:203:405:607:809:a0b:c0d:e0f"), 0);
    TEST_ASSERT(ipv6_addr_equal(&a, &address));
    TEST_ASSERT_NULL(netif);
}

static void test_ipv6_addr_from_str__success2(void)
{
    static const ipv6_addr_t a = { {
            0xfe, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
            0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff
        }
    };
    ipv6_addr_t address;
    netif_t *netif;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv6(&address, &netif, "fe80::f8f9:fafb:fcfd:feff%1"), 0);
    TEST_ASSERT_EQUAL_INT((uintptr_t)netif, (uintptr_t)&dummy_netif[1]);
    TEST_ASSERT(ipv6_addr_equal(&a, &address));
}

static void test_ipv6_addr_from_str__invalid_interface(void)
{
    ipv6_addr_t address;
    netif_t *netif;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv6(&address, &netif, "fe80::f8f9:fafb:fcfd:feff%3"), -EINVAL);
}

static void test_ipv6_addr_from_str__success4(void)
{
    ipv6_addr_t address;
    netif_t *netif;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv6(&address, &netif, SOCK_DNS_MOCK_EXAMPLE_COM_HOSTNAME), 0);
    TEST_ASSERT(ipv6_addr_equal(&sock_dns_mock_example_com_addr_ipv6, &address));
}

static void test_ipv6_addr_from_str__success5(void)
{
    static const ipv6_addr_t a = IPV6_ADDR_LOOPBACK;
    ipv6_addr_t address;

    TEST_ASSERT_NOT_NULL(ipv6_addr_from_str(&address, "::1"));
    TEST_ASSERT(ipv6_addr_equal(&a, &address));
}

static void test_ipv4_addr_from_str__missing_parts(void)
{
    ipv4_addr_t address;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv4(&address, "1"), -EINVAL);
    TEST_ASSERT_EQUAL_INT(netutils_get_ipv4(&address, "1."), -EINVAL);
    TEST_ASSERT_EQUAL_INT(netutils_get_ipv4(&address, "1.2"), -EINVAL);
    TEST_ASSERT_EQUAL_INT(netutils_get_ipv4(&address, "1.2."), -EINVAL);
    TEST_ASSERT_EQUAL_INT(netutils_get_ipv4(&address, "1.2.3"), -EINVAL);
    TEST_ASSERT_EQUAL_INT(netutils_get_ipv4(&address, "1.2.3."), -EINVAL);
}

static void test_ipv4_addr_from_str__illegal_chars(void)
{
    ipv4_addr_t address;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv4(&address, ":-D"), -ENOTSUP);
}

static void test_ipv4_addr_from_str__addr_NULL(void)
{
    ipv4_addr_t address;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv4(&address, NULL), -EINVAL);
}

static void test_ipv4_addr_from_str__address_NULL(void)
{
    TEST_ASSERT_NULL(ipv4_addr_from_str(NULL, "1.2.3.4"));
}

static void test_ipv4_addr_from_str__success(void)
{
    static const ipv4_addr_t a = IPV4_ADDR_INIT(1, 2, 3, 4);
    ipv4_addr_t address;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv4(&address, "1.2.3.4"), 0);
    TEST_ASSERT(ipv4_addr_equal(&a, &address));
}

static void test_ipv4_addr_from_str__success2(void)
{
    ipv4_addr_t address;

    TEST_ASSERT_EQUAL_INT(netutils_get_ipv4(&address, SOCK_DNS_MOCK_EXAMPLE_COM_HOSTNAME), 0);
    TEST_ASSERT(ipv4_addr_equal(&sock_dns_mock_example_com_addr_ipv4, &address));
}

Test *tests_netutils_ipv4_tests(void)
{
    EMB_UNIT_TESTFIXTURES(fixtures) {
        /* IPv4 tests */
        new_TestFixture(test_ipv4_addr_from_str__missing_parts),
        new_TestFixture(test_ipv4_addr_from_str__illegal_chars),
        new_TestFixture(test_ipv4_addr_from_str__addr_NULL),
        new_TestFixture(test_ipv4_addr_from_str__address_NULL),
        new_TestFixture(test_ipv4_addr_from_str__success),
        new_TestFixture(test_ipv4_addr_from_str__success2),
    };

    EMB_UNIT_TESTCALLER(ipv4_addr_tests, NULL, NULL, fixtures);

    return (Test *)&ipv4_addr_tests;
}

static void test_sock_tl_name2ep__ip_if_port(void)
{
    static const ipv6_addr_t a = { {
            0xfe, 0x80, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
            0xf8, 0xf9, 0xfa, 0xfb, 0xfc, 0xfd, 0xfe, 0xff
        }
    };
    struct _sock_tl_ep ep;

    int res = sock_tl_name2ep(&ep, "[fe80::f8f9:fafb:fcfd:feff%1]:1234");
    TEST_ASSERT_EQUAL_INT(res, 0);
    TEST_ASSERT_EQUAL_INT(AF_INET6, ep.family);
    TEST_ASSERT_EQUAL_INT(1234, ep.port);
    TEST_ASSERT_EQUAL_INT(1, ep.netif);
    TEST_ASSERT(ipv6_addr_equal(&a, (ipv6_addr_t *)&ep.addr.ipv6));
}

static void test_sock_tl_name2ep__name_port(void)
{
    static const ipv6_addr_t a = { {
            0x26, 0x06, 0x28, 0x00, 0x02, 0x20, 0x00, 0x01,
            0x02, 0x48, 0x18, 0x93, 0x25, 0xc8, 0x19, 0x46
        }
    };
    struct _sock_tl_ep ep;

    int res = sock_tl_name2ep(&ep, "example.com:1234");
    TEST_ASSERT_EQUAL_INT(0, res);
    TEST_ASSERT_EQUAL_INT(AF_INET6, ep.family);
    TEST_ASSERT_EQUAL_INT(1234, ep.port);
    TEST_ASSERT(ipv6_addr_equal(&a, (ipv6_addr_t *)&ep.addr.ipv6));
}

static void test_sock_tl_name2ep__name_only(void)
{
    static const ipv6_addr_t a = { {
            0x26, 0x06, 0x28, 0x00, 0x02, 0x20, 0x00, 0x01,
            0x02, 0x48, 0x18, 0x93, 0x25, 0xc8, 0x19, 0x46
        }
    };
    struct _sock_tl_ep ep;

    int res = sock_tl_name2ep(&ep, "example.com");
    TEST_ASSERT_EQUAL_INT(0, res);
    TEST_ASSERT_EQUAL_INT(AF_INET6, ep.family);
    TEST_ASSERT_EQUAL_INT(0, ep.port);
    TEST_ASSERT(ipv6_addr_equal(&a, (ipv6_addr_t *)&ep.addr.ipv6));
}

Test *tests_netutils_ipv6_tests(void)
{
    for (unsigned i = 0; i < ARRAY_SIZE(dummy_netif); ++i) {
        netif_register(&dummy_netif[i].netif);
        dummy_netif[i].pid = i;
    }

    EMB_UNIT_TESTFIXTURES(fixtures) {
        new_TestFixture(test_ipv6_addr_from_str__one_colon_start),
        new_TestFixture(test_ipv6_addr_from_str__three_colons),
        new_TestFixture(test_ipv6_addr_from_str__illegal_chars),
        new_TestFixture(test_ipv6_addr_from_str__addr_NULL),
        new_TestFixture(test_ipv6_addr_from_str__address_NULL),
        new_TestFixture(test_ipv6_addr_from_str__success),
        new_TestFixture(test_ipv6_addr_from_str__success2),
        new_TestFixture(test_ipv6_addr_from_str__invalid_interface),
        new_TestFixture(test_ipv6_addr_from_str__success4),
        new_TestFixture(test_ipv6_addr_from_str__success5),
        new_TestFixture(test_sock_tl_name2ep__ip_if_port),
        new_TestFixture(test_sock_tl_name2ep__name_port),
        new_TestFixture(test_sock_tl_name2ep__name_only),
    };

    EMB_UNIT_TESTCALLER(ipv6_addr_tests, NULL, NULL, fixtures);

    return (Test *)&ipv6_addr_tests;
}

int main(void)
{
    TESTS_START();
    TESTS_RUN(tests_netutils_ipv4_tests());
    TESTS_RUN(tests_netutils_ipv6_tests());
    TESTS_END();

    return 0;
}
/** @} */