// SPDX-License-Identifier: GPL-2.0
#include <stdio.h>
#include <string.h>

#include <ynl.h>

#include <arpa/inet.h>
#include <net/if.h>

#include <kselftest_harness.h>

#include "rt-route-user.h"

static void rt_route_print(struct __test_metadata *_metadata,
                           struct rt_route_getroute_rsp *r)
{
        char ifname[IF_NAMESIZE];
        char route_str[64];
        const char *route;
        const char *name;

        /* Ignore local */
        if (r->_hdr.rtm_table == RT_TABLE_LOCAL)
                return;

        if (r->_present.oif) {
                name = if_indextoname(r->oif, ifname);
                EXPECT_NE(NULL, name);
                if (name)
                        ksft_print_msg("oif: %-16s ", name);
        }

        if (r->_len.dst) {
                route = inet_ntop(r->_hdr.rtm_family, r->dst,
                                  route_str, sizeof(route_str));
                printf("dst: %s/%d", route, r->_hdr.rtm_dst_len);
        }

        if (r->_len.gateway) {
                route = inet_ntop(r->_hdr.rtm_family, r->gateway,
                                  route_str, sizeof(route_str));
                printf("gateway: %s ", route);
        }

        printf("\n");
}

FIXTURE(rt_route)
{
        struct ynl_sock *ys;
};

FIXTURE_SETUP(rt_route)
{
        struct ynl_error yerr;

        self->ys = ynl_sock_create(&ynl_rt_route_family, &yerr);
        ASSERT_NE(NULL, self->ys)
                TH_LOG("failed to create rt-route socket: %s", yerr.msg);
}

FIXTURE_TEARDOWN(rt_route)
{
        ynl_sock_destroy(self->ys);
}

TEST_F(rt_route, dump)
{
        struct rt_route_getroute_req_dump *req;
        struct rt_route_getroute_list *rsp;
        struct in6_addr v6_expected;
        struct in_addr v4_expected;
        bool found_v4 = false;
        bool found_v6 = false;

        /* The bash wrapper configures 192.168.1.1/24 and 2001:db8::1/64,
         * make sure we can find the connected routes in the dump.
         */
        inet_pton(AF_INET, "192.168.1.0", &v4_expected);
        inet_pton(AF_INET6, "2001:db8::", &v6_expected);

        req = rt_route_getroute_req_dump_alloc();
        ASSERT_NE(NULL, req);

        rsp = rt_route_getroute_dump(self->ys, req);
        rt_route_getroute_req_dump_free(req);
        ASSERT_NE(NULL, rsp) {
                TH_LOG("dump failed: %s", self->ys->err.msg);
        }

        ASSERT_FALSE(ynl_dump_empty(rsp)) {
                rt_route_getroute_list_free(rsp);
                TH_LOG("no routes reported");
        }

        ynl_dump_foreach(rsp, route) {
                rt_route_print(_metadata, route);

                if (route->_hdr.rtm_table == RT_TABLE_LOCAL)
                        continue;

                if (route->_len.dst == 4 && route->_hdr.rtm_dst_len == 24)
                        found_v4 |= !memcmp(route->dst, &v4_expected, 4);
                if (route->_len.dst == 16 && route->_hdr.rtm_dst_len == 64)
                        found_v6 |= !memcmp(route->dst, &v6_expected, 16);
        }
        rt_route_getroute_list_free(rsp);

        EXPECT_TRUE(found_v4);
        EXPECT_TRUE(found_v6);
}

TEST_HARNESS_MAIN