/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * Copyright (C) 2013 Red Hat, Inc.
 */

#include "src/core/nm-default-daemon.h"

#include <arpa/inet.h>
#include <linux/if_addr.h>

#include "nm-ip6-config.h"

#include "libnm-platform/nm-platform.h"
#include "nm-test-utils-core.h"

static NMIP6Config *
build_test_config(void)
{
    NMIP6Config *config;

    /* Build up the config to subtract */
    config = nmtst_ip6_config_new(1);

    nm_ip6_config_add_address(config,
                              nmtst_platform_ip6_address("abcd:1234:4321::cdde", "1:2:3:4::5", 64));
    nm_ip6_config_add_route(
        config,
        nmtst_platform_ip6_route("abcd:1200::", 24, "abcd:1234:4321:cdde::2", NULL),
        NULL);
    nm_ip6_config_add_route(config,
                            nmtst_platform_ip6_route("2001::", 16, "2001:abba::2234", NULL),
                            NULL);

    nm_ip6_config_add_route(config,
                            nmtst_platform_ip6_route("::", 0, "3001:abba::3234", NULL),
                            NULL);

    nm_ip6_config_add_nameserver(config, nmtst_inet6_from_string("1:2:3:4::1"));
    nm_ip6_config_add_nameserver(config, nmtst_inet6_from_string("1:2:3:4::2"));
    nm_ip6_config_add_domain(config, "foobar.com");
    nm_ip6_config_add_domain(config, "baz.com");
    nm_ip6_config_add_search(config, "blahblah.com");
    nm_ip6_config_add_search(config, "beatbox.com");

    return config;
}

static void
test_subtract(void)
{
    NMIP6Config *               src, *dst;
    const NMPlatformIP6Address *test_addr;
    const NMPlatformIP6Route *  test_route;
    const char *                expected_addr           = "1122:3344:5566::7788";
    guint32                     expected_addr_plen      = 96;
    const char *                expected_route_dest     = "9991:8800::";
    guint32                     expected_route_plen     = 24;
    const char *                expected_route_next_hop = "1119:2228:3337:4446::5555";
    struct in6_addr             expected_ns1;
    struct in6_addr             expected_ns2;
    const char *                expected_domain = "wonderfalls.com";
    const char *                expected_search = "somewhere.com";
    struct in6_addr             tmp;

    src = build_test_config();

    /* add a couple more things to the test config */
    dst = build_test_config();
    nm_ip6_config_add_address(dst,
                              nmtst_platform_ip6_address(expected_addr, NULL, expected_addr_plen));
    nm_ip6_config_add_route(dst,
                            nmtst_platform_ip6_route(expected_route_dest,
                                                     expected_route_plen,
                                                     expected_route_next_hop,
                                                     NULL),
                            NULL);

    expected_ns1 = *nmtst_inet6_from_string("2222:3333:4444::5555");
    nm_ip6_config_add_nameserver(dst, &expected_ns1);
    expected_ns2 = *nmtst_inet6_from_string("2222:3333:4444::5556");
    nm_ip6_config_add_nameserver(dst, &expected_ns2);

    nm_ip6_config_add_domain(dst, expected_domain);
    nm_ip6_config_add_search(dst, expected_search);

    nm_ip6_config_subtract(dst, src, 0);

    /* ensure what's left is what we expect */
    g_assert_cmpuint(nm_ip6_config_get_num_addresses(dst), ==, 1);
    test_addr = _nmtst_ip6_config_get_address(dst, 0);
    g_assert(test_addr != NULL);
    tmp = *nmtst_inet6_from_string(expected_addr);
    g_assert(memcmp(&test_addr->address, &tmp, sizeof(tmp)) == 0);
    g_assert(memcmp(&test_addr->peer_address, &in6addr_any, sizeof(tmp)) == 0);
    g_assert_cmpuint(test_addr->plen, ==, expected_addr_plen);

    g_assert(nm_ip6_config_best_default_route_get(dst) == NULL);

    g_assert_cmpuint(nm_ip6_config_get_num_routes(dst), ==, 1);
    test_route = _nmtst_ip6_config_get_route(dst, 0);
    g_assert(test_route != NULL);

    tmp = *nmtst_inet6_from_string(expected_route_dest);
    g_assert(memcmp(&test_route->network, &tmp, sizeof(tmp)) == 0);
    g_assert_cmpuint(test_route->plen, ==, expected_route_plen);
    tmp = *nmtst_inet6_from_string(expected_route_next_hop);
    g_assert(memcmp(&test_route->gateway, &tmp, sizeof(tmp)) == 0);

    g_assert_cmpuint(nm_ip6_config_get_num_nameservers(dst), ==, 2);
    g_assert(memcmp(nm_ip6_config_get_nameserver(dst, 0), &expected_ns1, sizeof(expected_ns1))
             == 0);
    g_assert(memcmp(nm_ip6_config_get_nameserver(dst, 1), &expected_ns2, sizeof(expected_ns2))
             == 0);

    g_assert_cmpuint(nm_ip6_config_get_num_domains(dst), ==, 1);
    g_assert_cmpstr(nm_ip6_config_get_domain(dst, 0), ==, expected_domain);
    g_assert_cmpuint(nm_ip6_config_get_num_searches(dst), ==, 1);
    g_assert_cmpstr(nm_ip6_config_get_search(dst, 0), ==, expected_search);

    g_object_unref(src);
    g_object_unref(dst);
}

static void
test_compare_with_source(void)
{
    NMIP6Config *        a, *b;
    NMPlatformIP6Address addr;
    NMPlatformIP6Route   route;

    a = nmtst_ip6_config_new(1);
    b = nmtst_ip6_config_new(2);

    /* Address */
    addr             = *nmtst_platform_ip6_address("1122:3344:5566::7788", NULL, 64);
    addr.addr_source = NM_IP_CONFIG_SOURCE_USER;
    nm_ip6_config_add_address(a, &addr);

    addr.addr_source = NM_IP_CONFIG_SOURCE_VPN;
    nm_ip6_config_add_address(b, &addr);

    /* Route */
    route           = *nmtst_platform_ip6_route("abcd:1200::", 24, "abcd:1234:4321:cdde::2", NULL);
    route.rt_source = NM_IP_CONFIG_SOURCE_USER;
    nm_ip6_config_add_route(a, &route, NULL);

    route.rt_source = NM_IP_CONFIG_SOURCE_VPN;
    nm_ip6_config_add_route(b, &route, NULL);

    /* Assert that the configs are basically the same, eg that the source is ignored */
    g_assert(nm_ip6_config_equal(a, b));

    g_object_unref(a);
    g_object_unref(b);
}

static void
test_add_address_with_source(void)
{
    NMIP6Config *               a;
    NMPlatformIP6Address        addr;
    const NMPlatformIP6Address *test_addr;

    a = nmtst_ip6_config_new(1);

    /* Test that a higher priority source is not overwritten */
    addr             = *nmtst_platform_ip6_address("1122:3344:5566::7788", NULL, 64);
    addr.addr_source = NM_IP_CONFIG_SOURCE_USER;
    nm_ip6_config_add_address(a, &addr);

    test_addr = _nmtst_ip6_config_get_address(a, 0);
    g_assert_cmpint(test_addr->addr_source, ==, NM_IP_CONFIG_SOURCE_USER);

    addr.addr_source = NM_IP_CONFIG_SOURCE_VPN;
    nm_ip6_config_add_address(a, &addr);

    test_addr = _nmtst_ip6_config_get_address(a, 0);
    g_assert_cmpint(test_addr->addr_source, ==, NM_IP_CONFIG_SOURCE_USER);

    /* Test that a lower priority address source is overwritten */
    _nmtst_ip6_config_del_address(a, 0);
    addr.addr_source = NM_IP_CONFIG_SOURCE_KERNEL;
    nm_ip6_config_add_address(a, &addr);

    test_addr = _nmtst_ip6_config_get_address(a, 0);
    g_assert_cmpint(test_addr->addr_source, ==, NM_IP_CONFIG_SOURCE_KERNEL);

    addr.addr_source = NM_IP_CONFIG_SOURCE_USER;
    nm_ip6_config_add_address(a, &addr);

    test_addr = _nmtst_ip6_config_get_address(a, 0);
    g_assert_cmpint(test_addr->addr_source, ==, NM_IP_CONFIG_SOURCE_USER);

    g_object_unref(a);
}

static void
test_add_route_with_source(void)
{
    gs_unref_object NMIP6Config *a = NULL;
    NMPlatformIP6Route           route;
    const NMPlatformIP6Route *   test_route;

    a = nmtst_ip6_config_new(1);

    /* Test that a higher priority source is not overwritten */
    route           = *nmtst_platform_ip6_route("abcd:1200::", 24, "abcd:1234:4321:cdde::2", NULL);
    route.rt_source = NM_IP_CONFIG_SOURCE_USER;
    nm_ip6_config_add_route(a, &route, NULL);

    g_assert_cmpint(nm_ip6_config_get_num_routes(a), ==, 1);
    test_route = _nmtst_ip6_config_get_route(a, 0);
    g_assert_cmpint(test_route->rt_source, ==, NM_IP_CONFIG_SOURCE_USER);

    route.rt_source = NM_IP_CONFIG_SOURCE_VPN;
    nm_ip6_config_add_route(a, &route, NULL);

    g_assert_cmpint(nm_ip6_config_get_num_routes(a), ==, 1);
    test_route = _nmtst_ip6_config_get_route(a, 0);
    g_assert_cmpint(test_route->rt_source, ==, NM_IP_CONFIG_SOURCE_USER);

    _nmtst_ip6_config_del_route(a, 0);
    g_assert_cmpint(nm_ip6_config_get_num_routes(a), ==, 0);

    /* Test that a lower priority address source is overwritten */
    route.rt_source = NM_IP_CONFIG_SOURCE_KERNEL;
    nm_ip6_config_add_route(a, &route, NULL);

    g_assert_cmpint(nm_ip6_config_get_num_routes(a), ==, 1);
    test_route = _nmtst_ip6_config_get_route(a, 0);
    g_assert_cmpint(test_route->rt_source, ==, NM_IP_CONFIG_SOURCE_KERNEL);

    route.rt_source = NM_IP_CONFIG_SOURCE_USER;
    nm_ip6_config_add_route(a, &route, NULL);

    g_assert_cmpint(nm_ip6_config_get_num_routes(a), ==, 1);
    test_route = _nmtst_ip6_config_get_route(a, 0);
    g_assert_cmpint(test_route->rt_source, ==, NM_IP_CONFIG_SOURCE_USER);
}

static void
test_nm_ip6_config_addresses_sort_check(NMIP6Config *             config,
                                        NMSettingIP6ConfigPrivacy use_tempaddr,
                                        int                       repeat)
{
    int          addr_count = nm_ip6_config_get_num_addresses(config);
    int          i, irepeat;
    NMIP6Config *copy, *copy2;
    int *        idx = g_new(int, addr_count);

    nm_ip6_config_set_privacy(config, use_tempaddr);
    copy = nm_ip6_config_clone(config);
    g_assert(copy);
    copy2 = nm_ip6_config_clone(config);
    g_assert(copy2);

    /* initialize the array of indices, and keep shuffling them for every @repeat iteration. */
    for (i = 0; i < addr_count; i++)
        idx[i] = i;

    for (irepeat = 0; irepeat < repeat; irepeat++) {
        /* randomly shuffle the addresses. */
        nm_ip6_config_reset_addresses(copy);
        for (i = 0; i < addr_count; i++) {
            int j = g_rand_int_range(nmtst_get_rand(), i, addr_count);

            NM_SWAP(&idx[i], &idx[j]);
            nm_ip6_config_add_address(copy, _nmtst_ip6_config_get_address(config, idx[i]));
        }

        /* reorder them again */
        _nmtst_ip6_config_addresses_sort(copy);

        /* check equality using nm_ip6_config_equal() */
        if (!nm_ip6_config_equal(copy, config)) {
            g_message("%s", "SORTING yields unexpected output:");
            for (i = 0; i < addr_count; i++) {
                g_message(
                    "   >> [%d] = %s",
                    i,
                    nm_platform_ip6_address_to_string(_nmtst_ip6_config_get_address(config, i),
                                                      NULL,
                                                      0));
                g_message("   << [%d] = %s",
                          i,
                          nm_platform_ip6_address_to_string(_nmtst_ip6_config_get_address(copy, i),
                                                            NULL,
                                                            0));
            }
            g_assert_not_reached();
        }

        /* also check equality using nm_ip6_config_replace() */
        g_assert(nm_ip6_config_replace(copy2, copy, NULL) == FALSE);
    }

    g_free(idx);
    g_object_unref(copy);
    g_object_unref(copy2);
}

static void
test_nm_ip6_config_addresses_sort(void)
{
    NMIP6Config *config = build_test_config();

#define ADDR_ADD(...) \
    nm_ip6_config_add_address(config, nmtst_platform_ip6_address_full(__VA_ARGS__))

    nm_ip6_config_reset_addresses(config);
    ADDR_ADD("2607:f0d0:1002:51::4", NULL, 64, 0, NM_IP_CONFIG_SOURCE_USER, 0, 0, 0, 0);
    ADDR_ADD("2607:f0d0:1002:51::5", NULL, 64, 0, NM_IP_CONFIG_SOURCE_USER, 0, 0, 0, 0);
    ADDR_ADD("2607:f0d0:1002:51::6",
             NULL,
             64,
             0,
             NM_IP_CONFIG_SOURCE_NDISC,
             0,
             0,
             0,
             IFA_F_MANAGETEMPADDR);
    ADDR_ADD("2607:f0d0:1002:51::3",
             NULL,
             64,
             0,
             NM_IP_CONFIG_SOURCE_USER,
             0,
             0,
             0,
             IFA_F_TEMPORARY);
    ADDR_ADD("2607:f0d0:1002:51::8",
             NULL,
             64,
             0,
             NM_IP_CONFIG_SOURCE_USER,
             0,
             0,
             0,
             IFA_F_TEMPORARY);
    ADDR_ADD("2607:f0d0:1002:51::0",
             NULL,
             64,
             0,
             NM_IP_CONFIG_SOURCE_KERNEL,
             0,
             0,
             0,
             IFA_F_TEMPORARY);
    ADDR_ADD("fec0::1", NULL, 128, 0, NM_IP_CONFIG_SOURCE_KERNEL, 0, 0, 0, 0);
    ADDR_ADD("fe80::208:74ff:feda:625c", NULL, 128, 0, NM_IP_CONFIG_SOURCE_KERNEL, 0, 0, 0, 0);
    ADDR_ADD("fe80::208:74ff:feda:625d", NULL, 128, 0, NM_IP_CONFIG_SOURCE_KERNEL, 0, 0, 0, 0);
    ADDR_ADD("::1", NULL, 128, 0, NM_IP_CONFIG_SOURCE_USER, 0, 0, 0, 0);
    ADDR_ADD("2607:f0d0:1002:51::2",
             NULL,
             64,
             0,
             NM_IP_CONFIG_SOURCE_USER,
             0,
             0,
             0,
             IFA_F_TENTATIVE);
    test_nm_ip6_config_addresses_sort_check(config, NM_SETTING_IP6_CONFIG_PRIVACY_UNKNOWN, 8);
    test_nm_ip6_config_addresses_sort_check(config, NM_SETTING_IP6_CONFIG_PRIVACY_DISABLED, 8);
    test_nm_ip6_config_addresses_sort_check(config,
                                            NM_SETTING_IP6_CONFIG_PRIVACY_PREFER_PUBLIC_ADDR,
                                            8);

    nm_ip6_config_reset_addresses(config);
    ADDR_ADD("2607:f0d0:1002:51::3",
             NULL,
             64,
             0,
             NM_IP_CONFIG_SOURCE_USER,
             0,
             0,
             0,
             IFA_F_TEMPORARY);
    ADDR_ADD("2607:f0d0:1002:51::4", NULL, 64, 0, NM_IP_CONFIG_SOURCE_USER, 0, 0, 0, 0);
    ADDR_ADD("2607:f0d0:1002:51::5", NULL, 64, 0, NM_IP_CONFIG_SOURCE_USER, 0, 0, 0, 0);
    ADDR_ADD("2607:f0d0:1002:51::8",
             NULL,
             64,
             0,
             NM_IP_CONFIG_SOURCE_USER,
             0,
             0,
             0,
             IFA_F_TEMPORARY);
    ADDR_ADD("2607:f0d0:1002:51::0",
             NULL,
             64,
             0,
             NM_IP_CONFIG_SOURCE_KERNEL,
             0,
             0,
             0,
             IFA_F_TEMPORARY);
    ADDR_ADD("2607:f0d0:1002:51::6",
             NULL,
             64,
             0,
             NM_IP_CONFIG_SOURCE_NDISC,
             0,
             0,
             0,
             IFA_F_MANAGETEMPADDR);
    ADDR_ADD("fec0::1", NULL, 128, 0, NM_IP_CONFIG_SOURCE_KERNEL, 0, 0, 0, 0);
    ADDR_ADD("fe80::208:74ff:feda:625c", NULL, 128, 0, NM_IP_CONFIG_SOURCE_KERNEL, 0, 0, 0, 0);
    ADDR_ADD("fe80::208:74ff:feda:625d", NULL, 128, 0, NM_IP_CONFIG_SOURCE_KERNEL, 0, 0, 0, 0);
    ADDR_ADD("::1", NULL, 128, 0, NM_IP_CONFIG_SOURCE_USER, 0, 0, 0, 0);
    ADDR_ADD("2607:f0d0:1002:51::2",
             NULL,
             64,
             0,
             NM_IP_CONFIG_SOURCE_USER,
             0,
             0,
             0,
             IFA_F_TENTATIVE);
    test_nm_ip6_config_addresses_sort_check(config,
                                            NM_SETTING_IP6_CONFIG_PRIVACY_PREFER_TEMP_ADDR,
                                            8);

#undef ADDR_ADD
    g_object_unref(config);
}

static void
test_strip_search_trailing_dot(void)
{
    NMIP6Config *config;

    config = nmtst_ip6_config_new(1);

    nm_ip6_config_add_search(config, ".");
    nm_ip6_config_add_search(config, "foo");
    nm_ip6_config_add_search(config, "bar.");
    nm_ip6_config_add_search(config, "baz.com");
    nm_ip6_config_add_search(config, "baz.com.");
    nm_ip6_config_add_search(config, "foobar..");
    nm_ip6_config_add_search(config, ".foobar");
    nm_ip6_config_add_search(config, "~.");

    g_assert_cmpuint(nm_ip6_config_get_num_searches(config), ==, 4);
    g_assert_cmpstr(nm_ip6_config_get_search(config, 0), ==, "foo");
    g_assert_cmpstr(nm_ip6_config_get_search(config, 1), ==, "bar");
    g_assert_cmpstr(nm_ip6_config_get_search(config, 2), ==, "baz.com");
    g_assert_cmpstr(nm_ip6_config_get_search(config, 3), ==, "~");

    g_object_unref(config);
}

/*****************************************************************************/

static void
test_replace(gconstpointer user_data)
{
    nm_auto_unref_dedup_multi_index NMDedupMultiIndex *multi_idx = nm_dedup_multi_index_new();
    const int                                          TEST_IDX  = GPOINTER_TO_INT(user_data);
    const int                                          IFINDEX   = 1;
    gs_unref_object NMIP6Config *src_conf                        = NULL;
    gs_unref_object NMIP6Config *dst_conf                        = NULL;
    NMPlatformIP6Address *       addr;
    NMPlatformIP6Address         addrs[5] = {};
    guint                        addrs_n  = 0;
    guint                        i;

    dst_conf = nm_ip6_config_new(multi_idx, IFINDEX);
    src_conf = nm_ip6_config_new(multi_idx, IFINDEX);

    switch (TEST_IDX) {
    case 1:
        addr              = &addrs[addrs_n++];
        addr->ifindex     = IFINDEX;
        addr->address     = *nmtst_inet6_from_string("fe80::78ec:7a6d:602d:20f2");
        addr->plen        = 64;
        addr->n_ifa_flags = IFA_F_PERMANENT;
        addr->addr_source = NM_IP_CONFIG_SOURCE_KERNEL;
        break;
    case 2:
        addr              = &addrs[addrs_n++];
        addr->ifindex     = IFINDEX;
        addr->address     = *nmtst_inet6_from_string("fe80::78ec:7a6d:602d:20f2");
        addr->plen        = 64;
        addr->n_ifa_flags = IFA_F_PERMANENT;
        addr->addr_source = NM_IP_CONFIG_SOURCE_KERNEL;

        addr              = &addrs[addrs_n++];
        addr->ifindex     = IFINDEX;
        addr->address     = *nmtst_inet6_from_string("1::1");
        addr->plen        = 64;
        addr->addr_source = NM_IP_CONFIG_SOURCE_USER;

        nm_ip6_config_add_address(dst_conf, addr);
        break;
    default:
        g_assert_not_reached();
    }

    g_assert(addrs_n < G_N_ELEMENTS(addrs));

    for (i = 0; i < addrs_n; i++)
        nm_ip6_config_add_address(src_conf, &addrs[i]);

    nm_ip6_config_replace(dst_conf, src_conf, NULL);

    for (i = 0; i < addrs_n; i++) {
        const NMPlatformIP6Address *a = _nmtst_ip6_config_get_address(dst_conf, i);
        const NMPlatformIP6Address *b = _nmtst_ip6_config_get_address(src_conf, i);

        g_assert(nm_platform_ip6_address_cmp(&addrs[i], a) == 0);
        g_assert(nm_platform_ip6_address_cmp(&addrs[i], b) == 0);
    }
    g_assert(addrs_n == nm_ip6_config_get_num_addresses(dst_conf));
    g_assert(addrs_n == nm_ip6_config_get_num_addresses(src_conf));
}

/*****************************************************************************/

NMTST_DEFINE();

int
main(int argc, char **argv)
{
    nmtst_init_with_logging(&argc, &argv, NULL, "ALL");

    g_test_add_func("/ip6-config/subtract", test_subtract);
    g_test_add_func("/ip6-config/compare-with-source", test_compare_with_source);
    g_test_add_func("/ip6-config/add-address-with-source", test_add_address_with_source);
    g_test_add_func("/ip6-config/add-route-with-source", test_add_route_with_source);
    g_test_add_func("/ip6-config/test_nm_ip6_config_addresses_sort",
                    test_nm_ip6_config_addresses_sort);
    g_test_add_func("/ip6-config/strip-search-trailing-dot", test_strip_search_trailing_dot);
    g_test_add_data_func("/ip6-config/replace/1", GINT_TO_POINTER(1), test_replace);
    g_test_add_data_func("/ip6-config/replace/2", GINT_TO_POINTER(2), test_replace);

    return g_test_run();
}
