/* SPDX-License-Identifier: GPL-2.0+ */

/*

 * Copyright (C) 2013 - 2014 Red Hat, Inc.

 */

#include "nm-default.h"

#include <arpa/inet.h>

#include "nm-ip4-config.h"

#include "platform/nm-platform.h"

#include "nm-test-utils-core.h"

static NMIP4Config *

build_test_config(void)

{

    NMIP4Config *        config;

    NMPlatformIP4Address addr;

    NMPlatformIP4Route   route;

    /* Build up the config to subtract */

    config = nmtst_ip4_config_new(1);

    nm_assert(NM_IP_CONFIG_CAST(config));

    addr = *nmtst_platform_ip4_address("192.168.1.10", "1.2.3.4", 24);

    nm_ip4_config_add_address(config, &addr);

    route = *nmtst_platform_ip4_route("10.0.0.0", 8, "192.168.1.1");

    nm_ip4_config_add_route(config, &route, NULL);

    route = *nmtst_platform_ip4_route("172.16.0.0", 16, "192.168.1.1");

    nm_ip4_config_add_route(config, &route, NULL);

    {

        const NMPlatformIP4Route r = {

            .rt_source     = NM_IP_CONFIG_SOURCE_DHCP,

            .gateway       = nmtst_inet4_from_string("192.168.1.1"),

            .table_coerced = 0,

            .metric        = 100,

        };

        nm_ip4_config_add_route(config, &r, NULL);

    }

    nm_ip4_config_add_nameserver(config, nmtst_inet4_from_string("4.2.2.1"));

    nm_ip4_config_add_nameserver(config, nmtst_inet4_from_string("4.2.2.2"));

    nm_ip4_config_add_domain(config, "foobar.com");

    nm_ip4_config_add_domain(config, "baz.com");

    nm_ip4_config_add_search(config, "blahblah.com");

    nm_ip4_config_add_search(config, "beatbox.com");

    nm_ip4_config_add_nis_server(config, nmtst_inet4_from_string("1.2.3.9"));

    nm_ip4_config_add_nis_server(config, nmtst_inet4_from_string("1.2.3.10"));

    nm_ip4_config_add_wins(config, nmtst_inet4_from_string("4.2.3.9"));

    nm_ip4_config_add_wins(config, nmtst_inet4_from_string("4.2.3.10"));

    return config;

}

static void

test_replace(void)

{

    gs_unref_object NMIP4Config *config1 = NULL;

    gs_unref_object NMIP4Config *config2 = NULL;

    NMPlatformIP4Address         addr;

    gboolean                     relevant_changes;

    config1 = nmtst_ip4_config_new(1);

    addr           = *nmtst_platform_ip4_address("172.16.0.1", NULL, 24);

    addr.timestamp = 10;

    addr.preferred = 3600;

    addr.lifetime  = 7200;

    nm_ip4_config_add_address(config1, &addr);

    addr           = *nmtst_platform_ip4_address("172.16.0.2", NULL, 24);

    addr.timestamp = 10;

    addr.preferred = 3600;

    addr.lifetime  = 7200;

    nm_ip4_config_add_address(config1, &addr);

    config2 = nmtst_ip4_config_new(1);

    addr           = *nmtst_platform_ip4_address("192.168.1.1", NULL, 24);

    addr.timestamp = 40;

    addr.preferred = 60;

    addr.lifetime  = 120;

    nm_ip4_config_add_address(config2, &addr);

    addr           = *nmtst_platform_ip4_address("172.16.0.2", NULL, 24);

    addr.timestamp = 40;

    addr.preferred = 60;

    addr.lifetime  = 120;

    nm_ip4_config_add_address(config2, &addr);

    g_assert(nm_ip4_config_replace(config2, config1, &relevant_changes));

    g_assert(relevant_changes);

    g_assert(nm_ip4_config_equal(config1, config2));

}

static void

test_subtract(void)

{

    NMIP4Config *               src, *dst;

    NMPlatformIP4Address        addr;

    NMPlatformIP4Route          route;

    const NMPlatformIP4Address *test_addr;

    const NMPlatformIP4Route *  test_route;

    const char *                expected_addr           = "192.168.1.12";

    guint32                     expected_addr_plen      = 24;

    const char *                expected_route_dest     = "8.0.0.0";

    guint32                     expected_route_plen     = 8;

    const char *                expected_route_next_hop = "192.168.1.1";

    guint32                     expected_ns1            = nmtst_inet4_from_string("8.8.8.8");

    guint32                     expected_ns2            = nmtst_inet4_from_string("8.8.8.9");

    const char *                expected_domain         = "wonderfalls.com";

    const char *                expected_search         = "somewhere.com";

    guint32                     expected_nis            = nmtst_inet4_from_string("1.2.3.13");

    guint32                     expected_wins           = nmtst_inet4_from_string("2.3.4.5");

    guint32                     expected_mtu            = 1492;

    src = build_test_config();

    /* add a couple more things to the test config */

    dst  = build_test_config();

    addr = *nmtst_platform_ip4_address(expected_addr, NULL, expected_addr_plen);

    nm_ip4_config_add_address(dst, &addr);

    route = *nmtst_platform_ip4_route(expected_route_dest,

                                      expected_route_plen,

                                      expected_route_next_hop);

    nm_ip4_config_add_route(dst, &route, NULL);

    nm_ip4_config_add_nameserver(dst, expected_ns1);

    nm_ip4_config_add_nameserver(dst, expected_ns2);

    nm_ip4_config_add_domain(dst, expected_domain);

    nm_ip4_config_add_search(dst, expected_search);

    nm_ip4_config_add_nis_server(dst, expected_nis);

    nm_ip4_config_add_wins(dst, expected_wins);

    nm_ip4_config_set_mtu(dst, expected_mtu, NM_IP_CONFIG_SOURCE_UNKNOWN);

    nm_ip4_config_subtract(dst, src, 0);

    /* ensure what's left is what we expect */

    g_assert_cmpuint(nm_ip4_config_get_num_addresses(dst), ==, 1);

    test_addr = _nmtst_ip4_config_get_address(dst, 0);

    g_assert(test_addr != NULL);

    g_assert_cmpuint(test_addr->address, ==, nmtst_inet4_from_string(expected_addr));

    g_assert_cmpuint(test_addr->peer_address, ==, test_addr->address);

    g_assert_cmpuint(test_addr->plen, ==, expected_addr_plen);

    g_assert(!nm_ip4_config_best_default_route_get(dst));

    g_assert_cmpuint(nmtst_ip4_config_get_gateway(dst), ==, 0);

    g_assert_cmpuint(nm_ip4_config_get_num_routes(dst), ==, 1);

    test_route = _nmtst_ip4_config_get_route(dst, 0);

    g_assert(test_route != NULL);

    g_assert_cmpuint(test_route->network, ==, nmtst_inet4_from_string(expected_route_dest));

    g_assert_cmpuint(test_route->plen, ==, expected_route_plen);

    g_assert_cmpuint(test_route->gateway, ==, nmtst_inet4_from_string(expected_route_next_hop));

    g_assert_cmpuint(nm_ip4_config_get_num_nameservers(dst), ==, 2);

    g_assert_cmpuint(nm_ip4_config_get_nameserver(dst, 0), ==, expected_ns1);

    g_assert_cmpuint(nm_ip4_config_get_nameserver(dst, 1), ==, expected_ns2);

    g_assert_cmpuint(nm_ip4_config_get_num_domains(dst), ==, 1);

    g_assert_cmpstr(nm_ip4_config_get_domain(dst, 0), ==, expected_domain);

    g_assert_cmpuint(nm_ip4_config_get_num_searches(dst), ==, 1);

    g_assert_cmpstr(nm_ip4_config_get_search(dst, 0), ==, expected_search);

    g_assert_cmpuint(nm_ip4_config_get_num_nis_servers(dst), ==, 1);

    g_assert_cmpuint(nm_ip4_config_get_nis_server(dst, 0), ==, expected_nis);

    g_assert_cmpuint(nm_ip4_config_get_num_wins(dst), ==, 1);

    g_assert_cmpuint(nm_ip4_config_get_wins(dst, 0), ==, expected_wins);

    g_assert_cmpuint(nm_ip4_config_get_mtu(dst), ==, expected_mtu);

    g_object_unref(src);

    g_object_unref(dst);

}

static void

test_compare_with_source(void)

{

    NMIP4Config *        a, *b;

    NMPlatformIP4Address addr;

    NMPlatformIP4Route   route;

    a = nmtst_ip4_config_new(1);

    b = nmtst_ip4_config_new(2);

    /* Address */

    addr             = *nmtst_platform_ip4_address("1.2.3.4", NULL, 24);

    addr.addr_source = NM_IP_CONFIG_SOURCE_USER;

    nm_ip4_config_add_address(a, &addr);

    addr.addr_source = NM_IP_CONFIG_SOURCE_VPN;

    nm_ip4_config_add_address(b, &addr);

    /* Route */

    route           = *nmtst_platform_ip4_route("10.0.0.0", 8, "192.168.1.1");

    route.rt_source = NM_IP_CONFIG_SOURCE_USER;

    nm_ip4_config_add_route(a, &route, NULL);

    route.rt_source = NM_IP_CONFIG_SOURCE_VPN;

    nm_ip4_config_add_route(b, &route, NULL);

    /* Assert that the configs are basically the same, eg that the source is ignored */

    g_assert(nm_ip4_config_equal(a, b));

    g_object_unref(a);

    g_object_unref(b);

}

static void

test_add_address_with_source(void)

{

    NMIP4Config *               a;

    NMPlatformIP4Address        addr;

    const NMPlatformIP4Address *test_addr;

    a = nmtst_ip4_config_new(1);

    /* Test that a higher priority source is not overwritten */

    addr             = *nmtst_platform_ip4_address("1.2.3.4", NULL, 24);

    addr.addr_source = NM_IP_CONFIG_SOURCE_USER;

    nm_ip4_config_add_address(a, &addr);

    test_addr = _nmtst_ip4_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_ip4_config_add_address(a, &addr);

    test_addr = _nmtst_ip4_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_ip4_config_del_address(a, 0);

    addr.addr_source = NM_IP_CONFIG_SOURCE_KERNEL;

    nm_ip4_config_add_address(a, &addr);

    test_addr = _nmtst_ip4_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_ip4_config_add_address(a, &addr);

    test_addr = _nmtst_ip4_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 NMIP4Config *a = NULL;

    NMPlatformIP4Route           route;

    const NMPlatformIP4Route *   test_route;

    a = nmtst_ip4_config_new(1);

    /* Test that a higher priority source is not overwritten */

    route           = *nmtst_platform_ip4_route("1.2.3.0", 24, "1.2.3.1");

    route.rt_source = NM_IP_CONFIG_SOURCE_USER;

    nm_ip4_config_add_route(a, &route, NULL);

    g_assert_cmpint(nm_ip4_config_get_num_routes(a), ==, 1);

    test_route = _nmtst_ip4_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_ip4_config_add_route(a, &route, NULL);

    g_assert_cmpint(nm_ip4_config_get_num_routes(a), ==, 1);

    test_route = _nmtst_ip4_config_get_route(a, 0);

    g_assert_cmpint(test_route->rt_source, ==, NM_IP_CONFIG_SOURCE_USER);

    _nmtst_ip4_config_del_route(a, 0);

    g_assert_cmpint(nm_ip4_config_get_num_routes(a), ==, 0);

    /* Test that a lower priority address source is overwritten */

    route.rt_source = NM_IP_CONFIG_SOURCE_RTPROT_KERNEL;

    nm_ip4_config_add_route(a, &route, NULL);

    g_assert_cmpint(nm_ip4_config_get_num_routes(a), ==, 1);

    test_route = _nmtst_ip4_config_get_route(a, 0);

    g_assert_cmpint(test_route->rt_source, ==, NM_IP_CONFIG_SOURCE_RTPROT_KERNEL);

    route.rt_source = NM_IP_CONFIG_SOURCE_KERNEL;

    nm_ip4_config_add_route(a, &route, NULL);

    g_assert_cmpint(nm_ip4_config_get_num_routes(a), ==, 1);

    test_route = _nmtst_ip4_config_get_route(a, 0);

    g_assert_cmpint(test_route->rt_source, ==, NM_IP_CONFIG_SOURCE_KERNEL);

}

static void

test_merge_subtract_mtu(void)

{

    NMIP4Config *cfg1, *cfg2, *cfg3;

    guint32      expected_mtu2 = 1492;

    guint32      expected_mtu3 = 666;

    cfg1 = build_test_config();

    cfg2 = build_test_config();

    cfg3 = build_test_config();

    /* add MSS, MTU to configs to test them */

    nm_ip4_config_set_mtu(cfg2, expected_mtu2, NM_IP_CONFIG_SOURCE_UNKNOWN);

    nm_ip4_config_set_mtu(cfg3, expected_mtu3, NM_IP_CONFIG_SOURCE_UNKNOWN);

    nm_ip4_config_merge(cfg1, cfg2, NM_IP_CONFIG_MERGE_DEFAULT, 0);

    /* ensure MSS and MTU are in cfg1 */

    g_assert_cmpuint(nm_ip4_config_get_mtu(cfg1), ==, expected_mtu2);

    nm_ip4_config_merge(cfg1, cfg3, NM_IP_CONFIG_MERGE_DEFAULT, 0);

    /* ensure again the MSS and MTU in cfg1 got overridden */

    g_assert_cmpuint(nm_ip4_config_get_mtu(cfg1), ==, expected_mtu3);

    nm_ip4_config_subtract(cfg1, cfg3, 0);

    /* ensure MSS and MTU are zero in cfg1 */

    g_assert_cmpuint(nm_ip4_config_get_mtu(cfg1), ==, 0);

    g_object_unref(cfg1);

    g_object_unref(cfg2);

    g_object_unref(cfg3);

}

static void

test_strip_search_trailing_dot(void)

{

    NMIP4Config *config;

    config = nmtst_ip4_config_new(1);

    nm_ip4_config_add_search(config, ".");

    nm_ip4_config_add_search(config, "foo");

    nm_ip4_config_add_search(config, "bar.");

    nm_ip4_config_add_search(config, "baz.com");

    nm_ip4_config_add_search(config, "baz.com.");

    nm_ip4_config_add_search(config, "foobar..");

    nm_ip4_config_add_search(config, ".foobar");

    nm_ip4_config_add_search(config, "~.");

    g_assert_cmpuint(nm_ip4_config_get_num_searches(config), ==, 4);

    g_assert_cmpstr(nm_ip4_config_get_search(config, 0), ==, "foo");

    g_assert_cmpstr(nm_ip4_config_get_search(config, 1), ==, "bar");

    g_assert_cmpstr(nm_ip4_config_get_search(config, 2), ==, "baz.com");

    g_assert_cmpstr(nm_ip4_config_get_search(config, 3), ==, "~");

    g_object_unref(config);

}

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

NMTST_DEFINE();

int

main(int argc, char **argv)

{

    nmtst_init_with_logging(&argc, &argv, NULL, "DEFAULT");

    g_test_add_func("/ip4-config/replace", test_replace);

    g_test_add_func("/ip4-config/subtract", test_subtract);

    g_test_add_func("/ip4-config/compare-with-source", test_compare_with_source);

    g_test_add_func("/ip4-config/add-address-with-source", test_add_address_with_source);

    g_test_add_func("/ip4-config/add-route-with-source", test_add_route_with_source);

    g_test_add_func("/ip4-config/merge-subtract-mtu", test_merge_subtract_mtu);

    g_test_add_func("/ip4-config/strip-search-trailing-dot", test_strip_search_trailing_dot);

    return g_test_run();

}