/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2019 Alexander V. Chernikov
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR 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 AUTHOR 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.
 */

#include "rtsock_common.h"
#include "rtsock_config.h"

static void
jump_vnet(struct rtsock_test_config *c, const atf_tc_t *tc)
{
        char vnet_name[512];

        snprintf(vnet_name, sizeof(vnet_name), "vt-%s", atf_tc_get_ident(tc));
        RLOG("jumping to %s", vnet_name);

        vnet_switch_one(vnet_name, c->ifname);

        /* Update ifindex cache */
        c->ifindex = if_nametoindex(c->ifname);
}

static inline struct rtsock_test_config *
presetup_ipv6(const atf_tc_t *tc)
{
        struct rtsock_test_config *c;
        int ret;

        c = config_setup(tc, NULL);

        jump_vnet(c, tc);

        ret = iface_turn_up(c->ifname);
        ATF_REQUIRE_MSG(ret == 0, "Unable to turn up %s", c->ifname);
        ret = iface_enable_ipv6(c->ifname);
        ATF_REQUIRE_MSG(ret == 0, "Unable to enable IPv6 on %s", c->ifname);

        c->rtsock_fd = rtsock_setup_socket();

        return (c);
}

static inline struct rtsock_test_config *
presetup_ipv4(const atf_tc_t *tc)
{
        struct rtsock_test_config *c;
        int ret;

        c = config_setup(tc, NULL);

        jump_vnet(c, tc);

        /* assumes ifconfig doing IFF_UP */
        ret = iface_setup_addr(c->ifname, c->addr4_str, c->plen4);
        ATF_REQUIRE_MSG(ret == 0, "ifconfig failed");

        c->rtsock_fd = rtsock_setup_socket();

        return (c);
}

static void
prepare_route_message(struct rt_msghdr *rtm, int cmd, struct sockaddr *dst,
  struct sockaddr *gw)
{

        rtsock_prepare_route_message(rtm, cmd, dst, NULL, gw);

        rtm->rtm_flags |= (RTF_HOST | RTF_STATIC | RTF_LLDATA);
}

/* TESTS */
#define DECLARE_TEST_VARS                                       \
        char buffer[2048], msg[512];                            \
        ssize_t len;                                            \
        int ret;                                                \
        struct rtsock_test_config *c;                           \
        struct rt_msghdr *rtm = (struct rt_msghdr *)buffer;     \
        struct sockaddr *sa;                                    \
                                                                \

#define DECLARE_CLEANUP_VARS                                    \
        struct rtsock_test_config *c = config_setup(tc);        \
                                                                \

#define DESCRIBE_ROOT_TEST(_msg)        config_describe_root_test(tc, _msg)
#define CLEANUP_AFTER_TEST      config_generic_cleanup(tc)

#define RTM_DECLARE_ROOT_TEST(_name, _descr)                    \
ATF_TC_WITH_CLEANUP(_name);                                     \
ATF_TC_HEAD(_name, tc)                                          \
{                                                               \
        DESCRIBE_ROOT_TEST(_descr);                             \
}                                                               \
ATF_TC_CLEANUP(_name, tc)                                       \
{                                                               \
        CLEANUP_AFTER_TEST;                                     \
}

RTM_DECLARE_ROOT_TEST(rtm_add_v6_ll_lle_success, "Tests addition of link-local IPv6 ND entry");
ATF_TC_BODY(rtm_add_v6_ll_lle_success, tc)
{
        DECLARE_TEST_VARS;

        c = presetup_ipv6(tc);

        char str_buf[128];
        struct sockaddr_in6 sin6;
        /* Interface here is optional. XXX: verify kernel side. */
        char *v6addr = "fe80::4242:4242";
        snprintf(str_buf, sizeof(str_buf), "%s%%%s", v6addr, c->ifname);
        sa_convert_str_to_sa(str_buf, (struct sockaddr *)&sin6);

        struct sockaddr_dl ether;
        snprintf(str_buf, sizeof(str_buf), "%s%%%s", c->remote_lladdr, c->ifname);
        sa_convert_str_to_sa(str_buf, (struct sockaddr *)&ether);

        prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&sin6, (struct sockaddr *)&ether);
        rtsock_send_rtm(c->rtsock_fd, rtm);

        /*
         * Got message of size 240 on 2019-12-17 15:06:51
         * RTM_ADD: Add Route: len 240, pid: 0, seq 0, errno 0, flags: <UP,HOST,DONE,LLINFO>
         * sockaddrs: 0x3 <DST,GATEWAY>
         *  af=inet6 len=28 addr=fe80::4242:4242 scope_id=3 if_name=tap4242
         *  af=link len=54 sdl_index=3 if_name=tap4242 addr=52:54:00:14:E3:10
         */

        rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);

        sa = rtsock_find_rtm_sa(rtm, RTA_DST);
        ret = sa_equal_msg(sa, (struct sockaddr *)&sin6, msg, sizeof(msg));
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "DST sa diff: %s", msg);

        sa = rtsock_find_rtm_sa(rtm, RTA_GATEWAY);
        int sa_flags = SA_F_IGNORE_IFNAME | SA_F_IGNORE_IFTYPE | SA_F_IGNORE_MEMCMP;
        ret = sa_equal_msg_flags(sa, (struct sockaddr *)&ether, msg, sizeof(msg), sa_flags);
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "GATEWAY sa diff: %s", msg);

#if 0
        /* Disable the check until https://reviews.freebsd.org/D22003 merge */
        /* Some additional checks to verify kernel has filled in interface data */
        struct sockaddr_dl *sdl = (struct sockaddr_dl *)sa;
        RTSOCK_ATF_REQUIRE_MSG(rtm, sdl->sdl_type > 0, "sdl_type not set");
#endif
}

RTM_DECLARE_ROOT_TEST(rtm_add_v6_gu_lle_success, "Tests addition of global IPv6 ND entry");
ATF_TC_BODY(rtm_add_v6_gu_lle_success, tc)
{
        DECLARE_TEST_VARS;

        c = presetup_ipv6(tc);

        char str_buf[128];

        struct sockaddr_in6 sin6;
        sin6 = c->net6;
#define _s6_addr32 __u6_addr.__u6_addr32
        sin6.sin6_addr._s6_addr32[3] = htonl(0x42424242);
#undef _s6_addr32

        struct sockaddr_dl ether;
        snprintf(str_buf, sizeof(str_buf), "%s%%%s", c->remote_lladdr, c->ifname);
        sa_convert_str_to_sa(str_buf, (struct sockaddr *)&ether);

        prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&sin6, (struct sockaddr *)&ether);

        rtsock_send_rtm(c->rtsock_fd, rtm);

        /*
         * Got message of size 240 on 2019-12-17 14:56:43
         * RTM_ADD: Add Route: len 240, pid: 0, seq 0, errno 0, flags: <UP,HOST,DONE,LLINFO>
         * sockaddrs: 0x3 <DST,GATEWAY>
         *  af=inet6 len=28 addr=2001:db8::4242:4242
         *  af=link len=54 sdl_index=3 if_name=tap4242 addr=52:54:00:14:E3:10
         */

        /* XXX: where is uRPF?! this should fail */

        rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);

        sa = rtsock_find_rtm_sa(rtm, RTA_DST);
        ret = sa_equal_msg(sa, (struct sockaddr *)&sin6, msg, sizeof(msg));
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "DST sa diff: %s", msg);

        sa = rtsock_find_rtm_sa(rtm, RTA_GATEWAY);
        int sa_flags = SA_F_IGNORE_IFNAME | SA_F_IGNORE_IFTYPE | SA_F_IGNORE_MEMCMP;
        ret = sa_equal_msg_flags(sa, (struct sockaddr *)&ether, msg, sizeof(msg), sa_flags);
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "GATEWAY sa diff: %s", msg);

#if 0
        /* Disable the check until https://reviews.freebsd.org/D22003 merge */
        /* Some additional checks to verify kernel has filled in interface data */
        struct sockaddr_dl *sdl = (struct sockaddr_dl *)sa;
        RTSOCK_ATF_REQUIRE_MSG(rtm, sdl->sdl_type > 0, "sdl_type not set");
#endif
}

RTM_DECLARE_ROOT_TEST(rtm_add_v4_gu_lle_success, "Tests addition of IPv4 ARP entry");
ATF_TC_BODY(rtm_add_v4_gu_lle_success, tc)
{
        DECLARE_TEST_VARS;

        c = presetup_ipv4(tc);

        char str_buf[128];

        struct sockaddr_in sin;
        sin = c->addr4;
        /* Use the next IPv4 address after self */
        sin.sin_addr.s_addr = htonl(ntohl(sin.sin_addr.s_addr) + 1);

        struct sockaddr_dl ether;
        snprintf(str_buf, sizeof(str_buf), "%s%%%s", c->remote_lladdr, c->ifname);
        sa_convert_str_to_sa(str_buf, (struct sockaddr *)&ether);

        prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&sin, (struct sockaddr *)&ether);

        len = rtsock_send_rtm(c->rtsock_fd, rtm);

        /*
         * RTM_ADD: Add Route: len 224, pid: 43131, seq 42, errno 0, flags: <HOST,DONE,LLINFO,STATIC>
         * sockaddrs: 0x3 <DST,GATEWAY>
         *  af=inet len=16 addr=192.0.2.2
         *  af=link len=54 sdl_index=3 if_name=tap4242 addr=52:54:00:14:E3:10
         */

        rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);

        sa = rtsock_find_rtm_sa(rtm, RTA_DST);
        ret = sa_equal_msg(sa, (struct sockaddr *)&sin, msg, sizeof(msg));
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "DST sa diff: %s", msg);

        sa = rtsock_find_rtm_sa(rtm, RTA_GATEWAY);
        int sa_flags = SA_F_IGNORE_IFNAME | SA_F_IGNORE_IFTYPE | SA_F_IGNORE_MEMCMP;
        ret = sa_equal_msg_flags(sa, (struct sockaddr *)&ether, msg, sizeof(msg), sa_flags);
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "GATEWAY sa diff: %s", msg);

        /*
         * TODO: Currently kernel code does not set sdl_type, contrary to IPv6.
         */
}

RTM_DECLARE_ROOT_TEST(rtm_del_v6_ll_lle_success, "Tests removal of link-local IPv6 ND entry");
ATF_TC_BODY(rtm_del_v6_ll_lle_success, tc)
{
        DECLARE_TEST_VARS;

        c = presetup_ipv6(tc);

        char str_buf[128];

        struct sockaddr_in6 sin6;
        /* Interface here is optional. XXX: verify kernel side. */
        char *v6addr = "fe80::4242:4242";
        snprintf(str_buf, sizeof(str_buf), "%s%%%s", v6addr, c->ifname);
        sa_convert_str_to_sa(str_buf, (struct sockaddr *)&sin6);

        struct sockaddr_dl ether;
        snprintf(str_buf, sizeof(str_buf), "%s%%%s", c->remote_lladdr, c->ifname);
        sa_convert_str_to_sa(str_buf, (struct sockaddr *)&ether);

        prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&sin6, (struct sockaddr *)&ether);

        rtsock_send_rtm(c->rtsock_fd, rtm);

        /* Successfully added an entry, let's try to remove it. */
        prepare_route_message(rtm, RTM_DELETE, (struct sockaddr *)&sin6, (struct sockaddr *)&ether);

        rtsock_send_rtm(c->rtsock_fd, rtm);

        rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);

        RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_type == RTM_DELETE, "rtm_type is not delete");

        sa = rtsock_find_rtm_sa(rtm, RTA_DST);
        ret = sa_equal_msg(sa, (struct sockaddr *)&sin6, msg, sizeof(msg));
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "DST sa diff: %s", msg);

        sa = rtsock_find_rtm_sa(rtm, RTA_GATEWAY);
        int sa_flags = SA_F_IGNORE_IFNAME | SA_F_IGNORE_IFTYPE | SA_F_IGNORE_MEMCMP;
        ret = sa_equal_msg_flags(sa, (struct sockaddr *)&ether, msg, sizeof(msg), sa_flags);
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "GATEWAY sa diff: %s", msg);

        /*
         * TODO: Currently kernel code does not set sdl_type on delete.
         */
}

RTM_DECLARE_ROOT_TEST(rtm_del_v6_gu_lle_success, "Tests removal of global IPv6 ND entry");
ATF_TC_BODY(rtm_del_v6_gu_lle_success, tc)
{
        DECLARE_TEST_VARS;

        c = presetup_ipv6(tc);

        char str_buf[128];

        struct sockaddr_in6 sin6;
        sin6 = c->net6;
#define _s6_addr32 __u6_addr.__u6_addr32
        sin6.sin6_addr._s6_addr32[3] = htonl(0x42424242);
#undef _s6_addr32

        struct sockaddr_dl ether;
        snprintf(str_buf, sizeof(str_buf), "%s%%%s", c->remote_lladdr, c->ifname);
        sa_convert_str_to_sa(str_buf, (struct sockaddr *)&ether);

        prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&sin6, (struct sockaddr *)&ether);

        len = rtsock_send_rtm(c->rtsock_fd, rtm);

        /* Successfully added an entry, let's try to remove it. */
        prepare_route_message(rtm, RTM_DELETE, (struct sockaddr *)&sin6, (struct sockaddr *)&ether);

        rtsock_send_rtm(c->rtsock_fd, rtm);

        rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);

        RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_type == RTM_DELETE, "rtm_type is not delete");

        sa = rtsock_find_rtm_sa(rtm, RTA_DST);
        ret = sa_equal_msg(sa, (struct sockaddr *)&sin6, msg, sizeof(msg));
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "DST sa diff: %s", msg);

        sa = rtsock_find_rtm_sa(rtm, RTA_GATEWAY);
        int sa_flags = SA_F_IGNORE_IFNAME | SA_F_IGNORE_IFTYPE | SA_F_IGNORE_MEMCMP;
        ret = sa_equal_msg_flags(sa, (struct sockaddr *)&ether, msg, sizeof(msg), sa_flags);
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "GATEWAY sa diff: %s", msg);

        /*
         * TODO: Currently kernel code does not set sdl_type on delete.
         */
}

RTM_DECLARE_ROOT_TEST(rtm_del_v4_gu_lle_success, "Tests removal of IPv4 ARP entry");
ATF_TC_BODY(rtm_del_v4_gu_lle_success, tc)
{
        DECLARE_TEST_VARS;

        c = presetup_ipv4(tc);

        char str_buf[128];

        struct sockaddr_in sin;
        sin = c->addr4;
        /* Use the next IPv4 address after self */
        sin.sin_addr.s_addr = htonl(ntohl(sin.sin_addr.s_addr) + 1);

        struct sockaddr_dl ether;
        snprintf(str_buf, sizeof(str_buf), "%s%%%s", c->remote_lladdr, c->ifname);
        sa_convert_str_to_sa(str_buf, (struct sockaddr *)&ether);

        prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&sin, (struct sockaddr *)&ether);

        rtsock_send_rtm(c->rtsock_fd, rtm);

        /* We successfully added an entry, let's try to remove it. */
        prepare_route_message(rtm, RTM_DELETE, (struct sockaddr *)&sin, (struct sockaddr *)&ether);

        rtsock_send_rtm(c->rtsock_fd, rtm);

        rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);

        RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_type == RTM_DELETE, "rtm_type is not delete");

        sa = rtsock_find_rtm_sa(rtm, RTA_DST);
        ret = sa_equal_msg(sa, (struct sockaddr *)&sin, msg, sizeof(msg));
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "DST sa diff: %s", msg);

        sa = rtsock_find_rtm_sa(rtm, RTA_GATEWAY);
        int sa_flags = SA_F_IGNORE_IFNAME | SA_F_IGNORE_IFTYPE | SA_F_IGNORE_MEMCMP;
        ret = sa_equal_msg_flags(sa, (struct sockaddr *)&ether, msg, sizeof(msg), sa_flags);
        RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "GATEWAY sa diff: %s", msg);

        /*
         * TODO: Currently kernel code does not set sdl_type, contrary to IPv6.
         */
}

ATF_TP_ADD_TCS(tp)
{
        ATF_TP_ADD_TC(tp, rtm_add_v6_ll_lle_success);
        ATF_TP_ADD_TC(tp, rtm_add_v6_gu_lle_success);
        ATF_TP_ADD_TC(tp, rtm_add_v4_gu_lle_success);
        ATF_TP_ADD_TC(tp, rtm_del_v6_ll_lle_success);
        ATF_TP_ADD_TC(tp, rtm_del_v6_gu_lle_success);
        ATF_TP_ADD_TC(tp, rtm_del_v4_gu_lle_success);

        return (atf_no_error());
}