/*-
 * 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"
#include "sys/types.h"
#include <sys/time.h>
#include <sys/ioctl.h>

#include "net/bpf.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(vnet_name, c->ifnames, c->num_interfaces);

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

static inline struct rtsock_test_config *
presetup_ipv6_iface(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);
	ATF_REQUIRE_ERRNO(0, true);

	return (c);
}

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

	c = presetup_ipv6_iface(tc);

	ret = iface_setup_addr(c->ifname, c->addr6_str, c->plen6);

	c->rtsock_fd = rtsock_setup_socket();
	ATF_REQUIRE_ERRNO(0, true);

	return (c);
}

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

	c = config_setup(tc, NULL);
	ATF_REQUIRE(c != NULL);

	jump_vnet(c, tc);

	ret = iface_turn_up(c->ifname);
	ATF_REQUIRE_MSG(ret == 0, "Unable to turn up %s", c->ifname);
	ATF_REQUIRE_ERRNO(0, true);

	return (c);
}

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

	c = presetup_ipv4_iface(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();
	ATF_REQUIRE_ERRNO(0, true);

	return (c);
}


static void
prepare_v4_network(struct rtsock_test_config *c, struct sockaddr_in *dst,
  struct sockaddr_in *mask, struct sockaddr_in *gw)
{
	/* Create IPv4 subnetwork with smaller prefix */
	sa_fill_mask4(mask, c->plen4 + 1);
	*dst = c->net4;
	/* Calculate GW as last-net-address - 1 */
	*gw = c->net4;
	gw->sin_addr.s_addr = htonl((ntohl(c->net4.sin_addr.s_addr) | ~ntohl(c->mask4.sin_addr.s_addr)) - 1);
	sa_print((struct sockaddr *)dst, 0);
	sa_print((struct sockaddr *)mask, 0);
	sa_print((struct sockaddr *)gw, 0);
}

static void
prepare_v6_network(struct rtsock_test_config *c, struct sockaddr_in6 *dst,
  struct sockaddr_in6 *mask, struct sockaddr_in6 *gw)
{
	/* Create IPv6 subnetwork with smaller prefix */
	sa_fill_mask6(mask, c->plen6 + 1);
	*dst = c->net6;
	/* Calculate GW as last-net-address - 1 */
	*gw = c->net6;
#define _s6_addr32 __u6_addr.__u6_addr32
	gw->sin6_addr._s6_addr32[0] = htonl((ntohl(gw->sin6_addr._s6_addr32[0]) | ~ntohl(c->mask6.sin6_addr._s6_addr32[0])));
	gw->sin6_addr._s6_addr32[1] = htonl((ntohl(gw->sin6_addr._s6_addr32[1]) | ~ntohl(c->mask6.sin6_addr._s6_addr32[1])));
	gw->sin6_addr._s6_addr32[2] = htonl((ntohl(gw->sin6_addr._s6_addr32[2]) | ~ntohl(c->mask6.sin6_addr._s6_addr32[2])));
	gw->sin6_addr._s6_addr32[3] = htonl((ntohl(gw->sin6_addr._s6_addr32[3]) | ~ntohl(c->mask6.sin6_addr._s6_addr32[3])) - 1);
#undef _s6_addr32
	sa_print((struct sockaddr *)dst, 0);
	sa_print((struct sockaddr *)mask, 0);
	sa_print((struct sockaddr *)gw, 0);
}

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

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

	if (cmd == RTM_ADD || cmd == RTM_CHANGE)
		rtm->rtm_flags |= RTF_STATIC;
}

static void
verify_route_message(struct rt_msghdr *rtm, int cmd, struct sockaddr *dst,
  struct sockaddr *mask, struct sockaddr *gw)
{
	char msg[512];
	struct sockaddr *sa;
	int ret;

	RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_type == cmd,
	    "expected %s message, got %d (%s)", rtsock_print_cmdtype(cmd),
	    rtm->rtm_type, rtsock_print_cmdtype(rtm->rtm_type));
	RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_errno == 0,
	    "got got errno %d as message reply", rtm->rtm_errno);
	RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->_rtm_spare1 == 0,
	    "expected rtm_spare==0, got %d", rtm->_rtm_spare1);

	/* kernel MAY return more sockaddrs, including RTA_IFP / RTA_IFA, so verify the needed ones */
	if (dst != NULL) {
		sa = rtsock_find_rtm_sa(rtm, RTA_DST);
		RTSOCK_ATF_REQUIRE_MSG(rtm, sa != NULL, "DST is not set");
		ret = sa_equal_msg(sa, dst, msg, sizeof(msg));
		RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "DST sa diff: %s", msg);
	}

	if (mask != NULL) {
		sa = rtsock_find_rtm_sa(rtm, RTA_NETMASK);
		RTSOCK_ATF_REQUIRE_MSG(rtm, sa != NULL, "NETMASK is not set");
		ret = sa_equal_msg(sa, mask, msg, sizeof(msg));
		ret = 1;
		RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "NETMASK sa diff: %s", msg);
	}

	if (gw != NULL) {
		sa = rtsock_find_rtm_sa(rtm, RTA_GATEWAY);
		RTSOCK_ATF_REQUIRE_MSG(rtm, sa != NULL, "GATEWAY is not set");
		ret = sa_equal_msg(sa, gw, msg, sizeof(msg));
		RTSOCK_ATF_REQUIRE_MSG(rtm, ret != 0, "GATEWAY sa diff: %s", msg);
	}
}

static void
verify_route_message_extra(struct rt_msghdr *rtm, int ifindex, int rtm_flags)
{
	RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_index == ifindex,
	    "expected ifindex %d, got %d", ifindex, rtm->rtm_index);

	if (rtm->rtm_flags != rtm_flags) {
		char got_flags[64], expected_flags[64];
		rtsock_print_rtm_flags(got_flags, sizeof(got_flags),
		    rtm->rtm_flags);
		rtsock_print_rtm_flags(expected_flags, sizeof(expected_flags),
		    rtm_flags);

		RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_flags == rtm_flags,
		    "expected flags: 0x%X %s, got 0x%X %s",
		    rtm_flags, expected_flags,
		    rtm->rtm_flags, got_flags);
	}
}

static void
verify_link_gateway(struct rt_msghdr *rtm, int ifindex)
{
	struct sockaddr *sa;
	struct sockaddr_dl *sdl;

	sa = rtsock_find_rtm_sa(rtm, RTA_GATEWAY);
	RTSOCK_ATF_REQUIRE_MSG(rtm, sa != NULL, "GATEWAY is not set");
	RTSOCK_ATF_REQUIRE_MSG(rtm, sa->sa_family == AF_LINK, "GW sa family is %d", sa->sa_family);
	sdl = (struct sockaddr_dl *)sa;
	RTSOCK_ATF_REQUIRE_MSG(rtm, sdl->sdl_index == ifindex, "GW ifindex is %d", sdl->sdl_index);
}

/* TESTS */

#define	DECLARE_TEST_VARS					\
	char buffer[2048];					\
	struct rtsock_test_config *c;				\
	struct rt_msghdr *rtm = (struct rt_msghdr *)buffer;	\
	struct sockaddr *sa;					\
	int ret;						\
								\

#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;					\
}

ATF_TC_WITH_CLEANUP(rtm_get_v4_exact_success);
ATF_TC_HEAD(rtm_get_v4_exact_success, tc)
{
	DESCRIBE_ROOT_TEST("Tests RTM_GET with exact prefix lookup on an interface prefix");
}

ATF_TC_BODY(rtm_get_v4_exact_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv4(tc);

	prepare_route_message(rtm, RTM_GET, (struct sockaddr *)&c->net4,
	    (struct sockaddr *)&c->mask4, NULL);

	rtsock_send_rtm(c->rtsock_fd, rtm);

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

	/*
	 * RTM_GET: Report Metrics: len 240, pid: 45072, seq 42, errno 0, flags: <UP,DONE,PINNED>
	 * sockaddrs: 0x7 <DST,GATEWAY,NETMASK>
	 *  af=inet len=16 addr=192.0.2.0 hd={10, 02, 00{2}, C0, 00, 02, 00{9}}
	 *  af=link len=54 sdl_index=3 if_name=tap4242 hd={36, 12, 03, 00, 06, 00{49}}
	 *  af=inet len=16 addr=255.255.255.0 hd={10, 02, FF{5}, 00{9}}
	 */

	verify_route_message(rtm, RTM_GET, (struct sockaddr *)&c->net4,
	    (struct sockaddr *)&c->mask4, NULL);

	verify_route_message_extra(rtm, c->ifindex, RTF_UP | RTF_DONE | RTF_PINNED);

	/* Explicitly verify gateway for the interface route */
	verify_link_gateway(rtm, c->ifindex);
	sa = rtsock_find_rtm_sa(rtm, RTA_GATEWAY);
	RTSOCK_ATF_REQUIRE_MSG(rtm, sa != NULL, "GATEWAY is not set");
	RTSOCK_ATF_REQUIRE_MSG(rtm, sa->sa_family == AF_LINK, "GW sa family is %d", sa->sa_family);
	struct sockaddr_dl *sdl = (struct sockaddr_dl *)sa;
	RTSOCK_ATF_REQUIRE_MSG(rtm, sdl->sdl_index == c->ifindex, "GW ifindex is %d", sdl->sdl_index);
}

ATF_TC_CLEANUP(rtm_get_v4_exact_success, tc)
{
	CLEANUP_AFTER_TEST;
}

ATF_TC_WITH_CLEANUP(rtm_get_v4_lpm_success);
ATF_TC_HEAD(rtm_get_v4_lpm_success, tc)
{
	DESCRIBE_ROOT_TEST("Tests RTM_GET with address lookup on an existing prefix");
}

ATF_TC_BODY(rtm_get_v4_lpm_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv4(tc);

	prepare_route_message(rtm, RTM_GET, (struct sockaddr *)&c->net4, NULL, NULL);

	rtsock_send_rtm(c->rtsock_fd, rtm);

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

	/*
	 * RTM_GET: Report Metrics: len 312, pid: 67074, seq 1, errno 0, flags:<UP,DONE,PINNED>
	 * locks:  inits:
	 * sockaddrs: <DST,GATEWAY,NETMASK,IFP,IFA>
	 * 10.0.0.0 link#1 255.255.255.0 vtnet0:52.54.0.42.f.ef 10.0.0.157
	 */

	verify_route_message(rtm, RTM_GET, (struct sockaddr *)&c->net4,
	    (struct sockaddr *)&c->mask4, NULL);

	verify_route_message_extra(rtm, c->ifindex, RTF_UP | RTF_DONE | RTF_PINNED);
}

ATF_TC_CLEANUP(rtm_get_v4_lpm_success, tc)
{
	CLEANUP_AFTER_TEST;
}


ATF_TC_WITH_CLEANUP(rtm_get_v4_empty_dst_failure);
ATF_TC_HEAD(rtm_get_v4_empty_dst_failure, tc)
{

	DESCRIBE_ROOT_TEST("Tests RTM_GET with empty DST addr");
}

ATF_TC_BODY(rtm_get_v4_empty_dst_failure, tc)
{
	DECLARE_TEST_VARS;
	struct rtsock_config_options co;

	bzero(&co, sizeof(co));
	co.num_interfaces = 0;
	
	c = config_setup(tc,&co);
	c->rtsock_fd = rtsock_setup_socket();

	rtsock_prepare_route_message(rtm, RTM_GET, NULL,
	    (struct sockaddr *)&c->mask4, NULL);
	rtsock_update_rtm_len(rtm);

	ATF_CHECK_ERRNO(EINVAL, write(c->rtsock_fd, rtm, rtm->rtm_msglen) == -1);
}

ATF_TC_CLEANUP(rtm_get_v4_empty_dst_failure, tc)
{
	CLEANUP_AFTER_TEST;
}

ATF_TC_WITH_CLEANUP(rtm_get_v4_hostbits_success);
ATF_TC_HEAD(rtm_get_v4_hostbits_success, tc)
{
	DESCRIBE_ROOT_TEST("Tests RTM_GET with prefix with some hosts-bits set");
}

ATF_TC_BODY(rtm_get_v4_hostbits_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv4(tc);

	/* Q the same prefix */
	rtsock_prepare_route_message(rtm, RTM_GET, (struct sockaddr *)&c->addr4,
	    (struct sockaddr *)&c->mask4, NULL);
	rtsock_update_rtm_len(rtm);

	ATF_REQUIRE_ERRNO(0, true);
	ATF_CHECK_ERRNO(0, write(c->rtsock_fd, rtm, rtm->rtm_msglen) > 0);
}

ATF_TC_CLEANUP(rtm_get_v4_hostbits_success, tc)
{
	CLEANUP_AFTER_TEST;
}

ATF_TC_WITH_CLEANUP(rtm_add_v4_gw_direct_success);
ATF_TC_HEAD(rtm_add_v4_gw_direct_success, tc)
{
	DESCRIBE_ROOT_TEST("Tests IPv4 route addition with directly-reachable GW specified by IP");
}

ATF_TC_BODY(rtm_add_v4_gw_direct_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv4(tc);

	/* Create IPv4 subnetwork with smaller prefix */
	struct sockaddr_in mask4;
	struct sockaddr_in net4;
	struct sockaddr_in gw4;
	prepare_v4_network(c, &net4, &mask4, &gw4);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);

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

	/*
	 * RTM_ADD: Add Route: len 200, pid: 46068, seq 42, errno 0, flags:<GATEWAY,DONE,STATIC>
	 * locks:  inits:
	 * sockaddrs: <DST,GATEWAY,NETMASK>
	 *  192.0.2.0 192.0.2.254 255.255.255.128
	 */

	verify_route_message(rtm, RTM_ADD, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);
	verify_route_message_extra(rtm, c->ifindex,
	    RTF_UP | RTF_DONE | RTF_GATEWAY | RTF_STATIC);
}

ATF_TC_CLEANUP(rtm_add_v4_gw_direct_success, tc)
{
	CLEANUP_AFTER_TEST;
}

RTM_DECLARE_ROOT_TEST(rtm_add_v4_no_rtf_host_success,
    "Tests success with netmask sa and RTF_HOST inconsistency");

ATF_TC_BODY(rtm_add_v4_no_rtf_host_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv4(tc);

	/* Create IPv4 subnetwork with smaller prefix */
	struct sockaddr_in mask4;
	struct sockaddr_in net4;
	struct sockaddr_in gw4;
	prepare_v4_network(c, &net4, &mask4, &gw4);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net4,
	    NULL, (struct sockaddr *)&gw4);
	rtsock_update_rtm_len(rtm);

	/* RTF_HOST is NOT specified, while netmask is empty */
	ATF_REQUIRE_ERRNO(0, true);
	ATF_CHECK_ERRNO(0, write(c->rtsock_fd, rtm, rtm->rtm_msglen) > 0);
}

ATF_TC_WITH_CLEANUP(rtm_del_v4_prefix_nogw_success);
ATF_TC_HEAD(rtm_del_v4_prefix_nogw_success, tc)
{
	DESCRIBE_ROOT_TEST("Tests IPv4 route removal without specifying gateway");
}

ATF_TC_BODY(rtm_del_v4_prefix_nogw_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv4(tc);

	/* Create IPv4 subnetwork with smaller prefix */
	struct sockaddr_in mask4;
	struct sockaddr_in net4;
	struct sockaddr_in gw4;
	prepare_v4_network(c, &net4, &mask4, &gw4);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);

	rtsock_send_rtm(c->rtsock_fd, rtm);

	/* Route has been added successfully, try to delete it */
	prepare_route_message(rtm, RTM_DELETE, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, NULL);

	rtsock_send_rtm(c->rtsock_fd, rtm);

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

	/*
	 * RTM_DELETE: Delete Route: len 200, pid: 46417, seq 43, errno 0, flags: <GATEWAY,DONE,STATIC>
	 * sockaddrs: 0x7 <DST,GATEWAY,NETMASK>
	 *  af=inet len=16 addr=192.0.2.0 hd={10, 02, 00{2}, C0, 00, 02, 00{9}}
	 *  af=inet len=16 addr=192.0.2.254 hd={10, 02, 00{2}, C0, 00, 02, FE, 00{8}}
	 *  af=inet len=16 addr=255.255.255.128 hd={10, 02, FF{5}, 80, 00{8}}
	 */
	verify_route_message(rtm, RTM_DELETE, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);

	verify_route_message_extra(rtm, c->ifindex, RTF_DONE | RTF_GATEWAY | RTF_STATIC);
}

ATF_TC_CLEANUP(rtm_del_v4_prefix_nogw_success, tc)
{
	CLEANUP_AFTER_TEST;
}

RTM_DECLARE_ROOT_TEST(rtm_change_v4_gw_success,
    "Tests IPv4 gateway change");

ATF_TC_BODY(rtm_change_v4_gw_success, tc)
{
	DECLARE_TEST_VARS;
	struct rtsock_config_options co;

	bzero(&co, sizeof(co));
	co.num_interfaces = 2;

	c = config_setup(tc, &co);
	jump_vnet(c, tc);

	ret = iface_turn_up(c->ifnames[0]);
	ATF_REQUIRE_MSG(ret == 0, "Unable to turn up %s", c->ifnames[0]);
	ret = iface_turn_up(c->ifnames[1]);
	ATF_REQUIRE_MSG(ret == 0, "Unable to turn up %s", c->ifnames[1]);

	ret = iface_setup_addr(c->ifnames[0], c->addr4_str, c->plen4);
	ATF_REQUIRE_MSG(ret == 0, "ifconfig failed");

	/* Use 198.51.100.0/24 "TEST-NET-2" for the second interface */
	ret = iface_setup_addr(c->ifnames[1], "198.51.100.1", 24);
	ATF_REQUIRE_MSG(ret == 0, "ifconfig failed");

	c->rtsock_fd = rtsock_setup_socket();

	/* Create IPv4 subnetwork with smaller prefix */
	struct sockaddr_in mask4;
	struct sockaddr_in net4;
	struct sockaddr_in gw4;
	prepare_v4_network(c, &net4, &mask4, &gw4);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);

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

	verify_route_message(rtm, RTM_ADD, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);

	/* Change gateway to the one on desiding on the other interface */
	inet_pton(AF_INET, "198.51.100.2", &gw4.sin_addr.s_addr);
	prepare_route_message(rtm, RTM_CHANGE, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);
	rtsock_send_rtm(c->rtsock_fd, rtm);

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

	verify_route_message(rtm, RTM_CHANGE, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);

	verify_route_message_extra(rtm, if_nametoindex(c->ifnames[1]),
	    RTF_UP | RTF_DONE | RTF_GATEWAY | RTF_STATIC);

	/* Verify the change has actually taken place */
	prepare_route_message(rtm, RTM_GET, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, NULL);

	rtsock_send_rtm(c->rtsock_fd, rtm);

	/*
	 * RTM_GET: len 200, pid: 3894, seq 44, errno 0, flags: <UP,GATEWAY,DONE,STATIC>
	 *  sockaddrs: 0x7 <DST,GATEWAY,NETMASK>
 	 *  af=inet len=16 addr=192.0.2.0 hd={x10, x02, x00{2}, xC0, x00, x02, x00{9}}
	 *  af=inet len=16 addr=198.51.100.2 hd={x10, x02, x00{2}, xC6, x33, x64, x02, x00{8}}
	 *  af=inet len=16 addr=255.255.255.128 hd={x10, x02, xFF, xFF, xFF, xFF, xFF, x80, x00{8}}
	 */

	rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);
	verify_route_message_extra(rtm, if_nametoindex(c->ifnames[1]),
	    RTF_UP | RTF_DONE | RTF_GATEWAY | RTF_STATIC);

}

RTM_DECLARE_ROOT_TEST(rtm_change_v4_mtu_success,
    "Tests IPv4 path mtu change");

ATF_TC_BODY(rtm_change_v4_mtu_success, tc)
{
	DECLARE_TEST_VARS;

	unsigned long test_mtu = 1442;

	c = presetup_ipv4(tc);

	/* Create IPv4 subnetwork with smaller prefix */
	struct sockaddr_in mask4;
	struct sockaddr_in net4;
	struct sockaddr_in gw4;
	prepare_v4_network(c, &net4, &mask4, &gw4);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);

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

	/* Change MTU */
	prepare_route_message(rtm, RTM_CHANGE, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, NULL);
	rtm->rtm_inits |= RTV_MTU;
	rtm->rtm_rmx.rmx_mtu = test_mtu;

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

	verify_route_message(rtm, RTM_CHANGE, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, NULL);

	RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_rmx.rmx_mtu == test_mtu,
	    "expected mtu: %lu, got %lu", test_mtu, rtm->rtm_rmx.rmx_mtu);

	/* Verify the change has actually taken place */
	prepare_route_message(rtm, RTM_GET, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, NULL);

	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_rmx.rmx_mtu == test_mtu,
	    "expected mtu: %lu, got %lu", test_mtu, rtm->rtm_rmx.rmx_mtu);
}

RTM_DECLARE_ROOT_TEST(rtm_change_v4_flags_success,
    "Tests IPv4 path flags change");

ATF_TC_BODY(rtm_change_v4_flags_success, tc)
{
	DECLARE_TEST_VARS;

	uint32_t test_flags = RTF_PROTO1 | RTF_PROTO2 | RTF_PROTO3 | RTF_STATIC;
	uint32_t desired_flags;

	c = presetup_ipv4(tc);

	/* Create IPv4 subnetwork with smaller prefix */
	struct sockaddr_in mask4;
	struct sockaddr_in net4;
	struct sockaddr_in gw4;
	prepare_v4_network(c, &net4, &mask4, &gw4);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);

	/* Set test flags during route addition */
	desired_flags = RTF_UP | RTF_DONE | RTF_GATEWAY | test_flags;
	rtm->rtm_flags |= test_flags;
	rtsock_send_rtm(c->rtsock_fd, rtm);
	rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);

	/* Change flags */
	prepare_route_message(rtm, RTM_CHANGE, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, NULL);
	rtm->rtm_flags &= ~test_flags;
	desired_flags &= ~test_flags;

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

	/* Verify updated flags */
	verify_route_message_extra(rtm, c->ifindex, desired_flags | RTF_DONE);

	/* Verify the change has actually taken place */
	prepare_route_message(rtm, RTM_GET, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, NULL);

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

	verify_route_message_extra(rtm, c->ifindex, desired_flags | RTF_DONE);
}


ATF_TC_WITH_CLEANUP(rtm_add_v6_gu_gw_gu_direct_success);
ATF_TC_HEAD(rtm_add_v6_gu_gw_gu_direct_success, tc)
{
	DESCRIBE_ROOT_TEST("Tests IPv6 global unicast prefix addition with directly-reachable GU GW");
}

ATF_TC_BODY(rtm_add_v6_gu_gw_gu_direct_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv6(tc);

	/* Create IPv6 subnetwork with smaller prefix */
	struct sockaddr_in6 mask6;
	struct sockaddr_in6 net6;
	struct sockaddr_in6 gw6;
	prepare_v6_network(c, &net6, &mask6, &gw6);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);

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

	/*
	 * RTM_ADD: Add Route: len 200, pid: 46068, seq 42, errno 0, flags:<GATEWAY,DONE,STATIC>
	 * locks:  inits:
	 * sockaddrs: <DST,GATEWAY,NETMASK>
	 *  192.0.2.0 192.0.2.254 255.255.255.128
	 */

	verify_route_message(rtm, RTM_ADD, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);

	verify_route_message_extra(rtm, c->ifindex,
	    RTF_UP | RTF_DONE | RTF_GATEWAY | RTF_STATIC);
}

ATF_TC_CLEANUP(rtm_add_v6_gu_gw_gu_direct_success, tc)
{
	CLEANUP_AFTER_TEST;
}

ATF_TC_WITH_CLEANUP(rtm_del_v6_gu_prefix_nogw_success);
ATF_TC_HEAD(rtm_del_v6_gu_prefix_nogw_success, tc)
{

	DESCRIBE_ROOT_TEST("Tests IPv6 global unicast prefix removal without specifying gateway");
}

ATF_TC_BODY(rtm_del_v6_gu_prefix_nogw_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv6(tc);

	/* Create IPv6 subnetwork with smaller prefix */
	struct sockaddr_in6 mask6;
	struct sockaddr_in6 net6;
	struct sockaddr_in6 gw6;
	prepare_v6_network(c, &net6, &mask6, &gw6);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);

	rtsock_send_rtm(c->rtsock_fd, rtm);

	/* Route has been added successfully, try to delete it */
	prepare_route_message(rtm, RTM_DELETE, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, NULL);

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

	/*
	 * RTM_DELETE: Delete Route: len 200, pid: 46417, seq 43, errno 0, flags: <GATEWAY,DONE,STATIC>
	 * sockaddrs: 0x7 <DST,GATEWAY,NETMASK>
	 *  af=inet len=16 addr=192.0.2.0 hd={10, 02, 00{2}, C0, 00, 02, 00{9}}
	 *  af=inet len=16 addr=192.0.2.254 hd={10, 02, 00{2}, C0, 00, 02, FE, 00{8}}
	 *  af=inet len=16 addr=255.255.255.128 hd={10, 02, FF{5}, 80, 00{8}}
	 */

	verify_route_message(rtm, RTM_DELETE, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);
	verify_route_message_extra(rtm, c->ifindex, RTF_DONE | RTF_GATEWAY | RTF_STATIC);
}

ATF_TC_CLEANUP(rtm_del_v6_gu_prefix_nogw_success, tc)
{
	CLEANUP_AFTER_TEST;
}

RTM_DECLARE_ROOT_TEST(rtm_change_v6_gw_success,
    "Tests IPv6 gateway change");

ATF_TC_BODY(rtm_change_v6_gw_success, tc)
{
	DECLARE_TEST_VARS;
	struct rtsock_config_options co;

	bzero(&co, sizeof(co));
	co.num_interfaces = 2;

	c = config_setup(tc, &co);
	jump_vnet(c, tc);

	ret = iface_turn_up(c->ifnames[0]);
	ATF_REQUIRE_MSG(ret == 0, "Unable to turn up %s", c->ifnames[0]);
	ret = iface_turn_up(c->ifnames[1]);
	ATF_REQUIRE_MSG(ret == 0, "Unable to turn up %s", c->ifnames[1]);

	ret = iface_enable_ipv6(c->ifnames[0]);
	ATF_REQUIRE_MSG(ret == 0, "Unable to enable IPv6 on %s", c->ifnames[0]);
	ret = iface_enable_ipv6(c->ifnames[1]);
	ATF_REQUIRE_MSG(ret == 0, "Unable to enable IPv6 on %s", c->ifnames[1]);

	ret = iface_setup_addr(c->ifnames[0], c->addr6_str, c->plen6);
	ATF_REQUIRE_MSG(ret == 0, "ifconfig failed");

	ret = iface_setup_addr(c->ifnames[1], "2001:DB8:4242::1", 64);
	ATF_REQUIRE_MSG(ret == 0, "ifconfig failed");

	c->rtsock_fd = rtsock_setup_socket();

	/* Create IPv6 subnetwork with smaller prefix */
	struct sockaddr_in6 mask6;
	struct sockaddr_in6 net6;
	struct sockaddr_in6 gw6;
	prepare_v6_network(c, &net6, &mask6, &gw6);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);

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

	verify_route_message(rtm, RTM_ADD, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);

	/* Change gateway to the one on residing on the other interface */
	inet_pton(AF_INET6, "2001:DB8:4242::4242", &gw6.sin6_addr);
	prepare_route_message(rtm, RTM_CHANGE, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);
	rtsock_send_rtm(c->rtsock_fd, rtm);

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

	verify_route_message(rtm, RTM_CHANGE, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);

	verify_route_message_extra(rtm, if_nametoindex(c->ifnames[1]),
	    RTF_UP | RTF_DONE | RTF_GATEWAY | RTF_STATIC);

	/* Verify the change has actually taken place */
	prepare_route_message(rtm, RTM_GET, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, NULL);

	rtsock_send_rtm(c->rtsock_fd, rtm);

	/*
	 * RTM_GET: len 248, pid: 2268, seq 44, errno 0, flags: <UP,GATEWAY,DONE,STATIC>
	 *  sockaddrs: 0x7 <DST,GATEWAY,NETMASK>
	 *  af=inet6 len=28 addr=2001:db8:: hd={x1C, x1C, x00{6}, x20, x01, x0D, xB8, x00{16}}
	 *  af=inet6 len=28 addr=2001:db8:4242::4242 hd={x1C, x1C, x00{6}, x20, x01, x0D, xB8, x42, x42, x00{8}, x42, x42, x00{4}}
	 *  af=inet6 len=28 addr=ffff:ffff:8000:: hd={x1C, x1C, xFF, xFF, xFF, xFF, xFF, xFF, xFF, xFF, xFF, xFF, x80, x00{15}}
	 */

	rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);
	verify_route_message_extra(rtm, if_nametoindex(c->ifnames[1]),
	    RTF_UP | RTF_DONE | RTF_GATEWAY | RTF_STATIC);
}

RTM_DECLARE_ROOT_TEST(rtm_change_v6_mtu_success,
    "Tests IPv6 path mtu change");

ATF_TC_BODY(rtm_change_v6_mtu_success, tc)
{
	DECLARE_TEST_VARS;

	unsigned long test_mtu = 1442;

	c = presetup_ipv6(tc);

	/* Create IPv6 subnetwork with smaller prefix */
	struct sockaddr_in6 mask6;
	struct sockaddr_in6 net6;
	struct sockaddr_in6 gw6;
	prepare_v6_network(c, &net6, &mask6, &gw6);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);

	/* Send route add */
	rtsock_send_rtm(c->rtsock_fd, rtm);
	rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);

	/* Change MTU */
	prepare_route_message(rtm, RTM_CHANGE, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, NULL);
	rtm->rtm_inits |= RTV_MTU;
	rtm->rtm_rmx.rmx_mtu = test_mtu;

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

	verify_route_message(rtm, RTM_CHANGE, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, NULL);

	RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_rmx.rmx_mtu == test_mtu,
	    "expected mtu: %lu, got %lu", test_mtu, rtm->rtm_rmx.rmx_mtu);

	/* Verify the change has actually taken place */
	prepare_route_message(rtm, RTM_GET, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, NULL);

	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_rmx.rmx_mtu == test_mtu,
	    "expected mtu: %lu, got %lu", test_mtu, rtm->rtm_rmx.rmx_mtu);
}

RTM_DECLARE_ROOT_TEST(rtm_change_v6_flags_success,
    "Tests IPv6 path flags change");

ATF_TC_BODY(rtm_change_v6_flags_success, tc)
{
	DECLARE_TEST_VARS;

	uint32_t test_flags = RTF_PROTO1 | RTF_PROTO2 | RTF_PROTO3 | RTF_STATIC;
	uint32_t desired_flags;

	c = presetup_ipv6(tc);

	/* Create IPv6 subnetwork with smaller prefix */
	struct sockaddr_in6 mask6;
	struct sockaddr_in6 net6;
	struct sockaddr_in6 gw6;
	prepare_v6_network(c, &net6, &mask6, &gw6);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);

	/* Set test flags during route addition */
	desired_flags = RTF_UP | RTF_DONE | RTF_GATEWAY | test_flags;
	rtm->rtm_flags |= test_flags;
	rtsock_send_rtm(c->rtsock_fd, rtm);
	rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);

	/* Change flags */
	prepare_route_message(rtm, RTM_CHANGE, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, NULL);
	rtm->rtm_flags &= ~test_flags;
	desired_flags &= ~test_flags;

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

	/* Verify updated flags */
	verify_route_message_extra(rtm, c->ifindex, desired_flags | RTF_DONE);

	/* Verify the change has actually taken place */
	prepare_route_message(rtm, RTM_GET, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, NULL);

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

	verify_route_message_extra(rtm, c->ifindex, desired_flags | RTF_DONE);
}

ATF_TC_WITH_CLEANUP(rtm_add_v4_temporal1_success);
ATF_TC_HEAD(rtm_add_v4_temporal1_success, tc)
{
	DESCRIBE_ROOT_TEST("Tests IPv4 route expiration with expire time set");
}

ATF_TC_BODY(rtm_add_v4_temporal1_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv4(tc);

	/* Create IPv4 subnetwork with smaller prefix */
	struct sockaddr_in mask4;
	struct sockaddr_in net4;
	struct sockaddr_in gw4;
	prepare_v4_network(c, &net4, &mask4, &gw4);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);

	/* Set expire time to now */
	struct timeval tv;
	gettimeofday(&tv, NULL);
	rtm->rtm_rmx.rmx_expire = tv.tv_sec - 1;
	rtm->rtm_inits |= RTV_EXPIRE;

	rtsock_send_rtm(c->rtsock_fd, rtm);
	rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);
	ATF_REQUIRE_MSG(rtm != NULL, "unable to get rtsock reply for RTM_ADD");
	RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_inits & RTV_EXPIRE, "RTV_EXPIRE not set");

	/* The next should be route deletion */
	rtm = rtsock_read_rtm(c->rtsock_fd, buffer, sizeof(buffer));

	verify_route_message(rtm, RTM_DELETE, (struct sockaddr *)&net4,
	    (struct sockaddr *)&mask4, (struct sockaddr *)&gw4);

	verify_route_message_extra(rtm, c->ifindex,
	    RTF_DONE | RTF_GATEWAY | RTF_STATIC);
}

ATF_TC_CLEANUP(rtm_add_v4_temporal1_success, tc)
{
	CLEANUP_AFTER_TEST;
}

ATF_TC_WITH_CLEANUP(rtm_add_v6_temporal1_success);
ATF_TC_HEAD(rtm_add_v6_temporal1_success, tc)
{
	DESCRIBE_ROOT_TEST("Tests IPv6 global unicast prefix addition with directly-reachable GU GW");
}

ATF_TC_BODY(rtm_add_v6_temporal1_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv6(tc);

	/* Create IPv6 subnetwork with smaller prefix */
	struct sockaddr_in6 mask6;
	struct sockaddr_in6 net6;
	struct sockaddr_in6 gw6;
	prepare_v6_network(c, &net6, &mask6, &gw6);

	prepare_route_message(rtm, RTM_ADD, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);

	/* Set expire time to now */
	struct timeval tv;
	gettimeofday(&tv, NULL);
	rtm->rtm_rmx.rmx_expire = tv.tv_sec - 1;
	rtm->rtm_inits |= RTV_EXPIRE;

	rtsock_send_rtm(c->rtsock_fd, rtm);
	rtm = rtsock_read_rtm_reply(c->rtsock_fd, buffer, sizeof(buffer), rtm->rtm_seq);
	ATF_REQUIRE_MSG(rtm != NULL, "unable to get rtsock reply for RTM_ADD");
	RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_inits & RTV_EXPIRE, "RTV_EXPIRE not set");

	/* The next should be route deletion */
	rtm = rtsock_read_rtm(c->rtsock_fd, buffer, sizeof(buffer));

	verify_route_message(rtm, RTM_DELETE, (struct sockaddr *)&net6,
	    (struct sockaddr *)&mask6, (struct sockaddr *)&gw6);

	verify_route_message_extra(rtm, c->ifindex,
	    RTF_DONE | RTF_GATEWAY | RTF_STATIC);
}

ATF_TC_CLEANUP(rtm_add_v6_temporal1_success, tc)
{
	CLEANUP_AFTER_TEST;
}

/* Interface address messages tests */

RTM_DECLARE_ROOT_TEST(rtm_add_v6_gu_ifa_hostroute_success,
    "Tests validness for /128 host route announce after ifaddr assignment");

ATF_TC_BODY(rtm_add_v6_gu_ifa_hostroute_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv6_iface(tc);

	c->rtsock_fd = rtsock_setup_socket();

	ret = iface_setup_addr(c->ifname, c->addr6_str, c->plen6);

	/*
	 * There will be multiple.
	 * RTM_ADD without llinfo.
	 */

	while (true) {
		rtm = rtsock_read_rtm(c->rtsock_fd, buffer, sizeof(buffer));
		if ((rtm->rtm_type == RTM_ADD) && ((rtm->rtm_flags & RTF_LLINFO) == 0))
			break;
	}
	/* This should be a message for the host route */

	verify_route_message(rtm, RTM_ADD, (struct sockaddr *)&c->addr6, NULL, NULL);
	rtsock_validate_pid_kernel(rtm);
	/* No netmask should be set */
	RTSOCK_ATF_REQUIRE_MSG(rtm, rtsock_find_rtm_sa(rtm, RTA_NETMASK) == NULL, "netmask is set");

	/* gateway should be link sdl with ifindex of an address interface */
	verify_link_gateway(rtm, c->ifindex);

	int expected_rt_flags = RTF_UP | RTF_HOST | RTF_DONE | RTF_STATIC | RTF_PINNED;
	verify_route_message_extra(rtm, if_nametoindex("lo0"), expected_rt_flags);
}

RTM_DECLARE_ROOT_TEST(rtm_add_v6_gu_ifa_prefixroute_success,
    "Tests validness for the prefix route announce after ifaddr assignment");

ATF_TC_BODY(rtm_add_v6_gu_ifa_prefixroute_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv6_iface(tc);

	c->rtsock_fd = rtsock_setup_socket();

	ret = iface_setup_addr(c->ifname, c->addr6_str, c->plen6);

	/*
	 * Multiple RTM_ADD messages will be generated:
	 * 1) lladdr mapping (RTF_LLDATA)
	 * 2) host route (one w/o netmask)
	 * 3) prefix route
	 */

	while (true) {
		rtm = rtsock_read_rtm(c->rtsock_fd, buffer, sizeof(buffer));
		/* Find RTM_ADD with netmask - this should skip both host route and LLADDR */
		if ((rtm->rtm_type == RTM_ADD) && (rtsock_find_rtm_sa(rtm, RTA_NETMASK)))
			break;
	}

	/* This should be a message for the prefix route */
	verify_route_message(rtm, RTM_ADD, (struct sockaddr *)&c->net6,
	    (struct sockaddr *)&c->mask6, NULL);

	/* gateway should be link sdl with ifindex of an address interface */
	verify_link_gateway(rtm, c->ifindex);

	int expected_rt_flags = RTF_UP | RTF_DONE | RTF_PINNED;
	verify_route_message_extra(rtm, c->ifindex, expected_rt_flags);
}

RTM_DECLARE_ROOT_TEST(rtm_add_v6_gu_ifa_ordered_success,
    "Tests ordering of the messages for IPv6 global unicast ifaddr assignment");

ATF_TC_BODY(rtm_add_v6_gu_ifa_ordered_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv6_iface(tc);

	c->rtsock_fd = rtsock_setup_socket();

	ret = iface_setup_addr(c->ifname, c->addr6_str, c->plen6);

	int count = 0, tries = 0;

	enum msgtype {
		MSG_IFADDR,
		MSG_HOSTROUTE,
		MSG_PREFIXROUTE,
		MSG_MAX,
	};

	int msg_array[MSG_MAX];

	bzero(msg_array, sizeof(msg_array));

	while (count < 3 && tries < 20) {
		rtm = rtsock_read_rtm(c->rtsock_fd, buffer, sizeof(buffer));
		tries++;
		/* Classify */
		if (rtm->rtm_type == RTM_NEWADDR) {
			RLOG("MSG_IFADDR: %d", count);
			msg_array[MSG_IFADDR] = count++;
			continue;
		}

		/* Find RTM_ADD with netmask - this should skip both host route and LLADDR */
		if ((rtm->rtm_type == RTM_ADD) && (rtsock_find_rtm_sa(rtm, RTA_NETMASK))) {
			RLOG("MSG_PREFIXROUTE: %d", count);
			msg_array[MSG_PREFIXROUTE] = count++;
			continue;
		}

		if ((rtm->rtm_type == RTM_ADD) && ((rtm->rtm_flags & RTF_LLDATA) == 0)) {
			RLOG("MSG_HOSTROUTE: %d", count);
			msg_array[MSG_HOSTROUTE] = count++;
			continue;
		}

		RLOG("skipping msg type %s, try: %d", rtsock_print_cmdtype(rtm->rtm_type),
		    tries);
	}

	/* TODO: verify multicast */
	ATF_REQUIRE_MSG(count == 3, "Received only %d/3 messages", count);
	ATF_REQUIRE_MSG(msg_array[MSG_IFADDR] == 0, "ifaddr message is not the first");
}

RTM_DECLARE_ROOT_TEST(rtm_del_v6_gu_ifa_hostroute_success,
    "Tests validness for /128 host route removal after ifaddr removal");

ATF_TC_BODY(rtm_del_v6_gu_ifa_hostroute_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv6_iface(tc);

	ret = iface_setup_addr(c->ifname, c->addr6_str, c->plen6);

	c->rtsock_fd = rtsock_setup_socket();

	ret = iface_delete_addr(c->ifname, c->addr6_str);

	while (true) {
		rtm = rtsock_read_rtm(c->rtsock_fd, buffer, sizeof(buffer));
		if ((rtm->rtm_type == RTM_DELETE) &&
		    ((rtm->rtm_flags & RTF_LLINFO) == 0) &&
		    rtsock_find_rtm_sa(rtm, RTA_NETMASK) == NULL)
			break;
	}
	/* This should be a message for the host route */

	verify_route_message(rtm, RTM_DELETE, (struct sockaddr *)&c->addr6, NULL, NULL);
	rtsock_validate_pid_kernel(rtm);
	/* No netmask should be set */
	RTSOCK_ATF_REQUIRE_MSG(rtm, rtsock_find_rtm_sa(rtm, RTA_NETMASK) == NULL, "netmask is set");

	/* gateway should be link sdl with ifindex of an address interface */
	verify_link_gateway(rtm, c->ifindex);

	/* XXX: consider passing ifindex in rtm_index as done in RTM_ADD. */
	int expected_rt_flags = RTF_HOST | RTF_DONE | RTF_STATIC | RTF_PINNED;
	RTSOCK_ATF_REQUIRE_MSG(rtm, rtm->rtm_flags == expected_rt_flags,
	    "expected rtm flags: 0x%X, got 0x%X", expected_rt_flags, rtm->rtm_flags);
}

RTM_DECLARE_ROOT_TEST(rtm_del_v6_gu_ifa_prefixroute_success,
    "Tests validness for the prefix route removal after ifaddr assignment");

ATF_TC_BODY(rtm_del_v6_gu_ifa_prefixroute_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv6_iface(tc);

	ret = iface_setup_addr(c->ifname, c->addr6_str, c->plen6);

	c->rtsock_fd = rtsock_setup_socket();

	ret = iface_delete_addr(c->ifname, c->addr6_str);

	while (true) {
		rtm = rtsock_read_rtm(c->rtsock_fd, buffer, sizeof(buffer));
		/* Find RTM_DELETE with netmask - this should skip both host route and LLADDR */
		if ((rtm->rtm_type == RTM_DELETE) && (rtsock_find_rtm_sa(rtm, RTA_NETMASK)))
			break;
	}

	/* This should be a message for the prefix route */
	verify_route_message(rtm, RTM_DELETE, (struct sockaddr *)&c->net6,
	    (struct sockaddr *)&c->mask6, NULL);

	/* gateway should be link sdl with ifindex of an address interface */
	verify_link_gateway(rtm, c->ifindex);

	int expected_rt_flags = RTF_DONE | RTF_PINNED;
	verify_route_message_extra(rtm, c->ifindex, expected_rt_flags);
}

RTM_DECLARE_ROOT_TEST(rtm_add_v4_gu_ifa_prefixroute_success,
    "Tests validness for the prefix route announce after ifaddr assignment");

ATF_TC_BODY(rtm_add_v4_gu_ifa_prefixroute_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv4_iface(tc);

	c->rtsock_fd = rtsock_setup_socket();

	ret = iface_setup_addr(c->ifname, c->addr6_str, c->plen6);

	/*
	 * Multiple RTM_ADD messages will be generated:
	 * 1) lladdr mapping (RTF_LLDATA)
	 * 3) prefix route
	 */

	while (true) {
		rtm = rtsock_read_rtm(c->rtsock_fd, buffer, sizeof(buffer));
		/* Find RTM_ADD with netmask - this should skip both host route and LLADDR */
		if ((rtm->rtm_type == RTM_ADD) && (rtsock_find_rtm_sa(rtm, RTA_NETMASK)))
			break;
	}

	/* This should be a message for the prefix route */
	verify_route_message(rtm, RTM_ADD, (struct sockaddr *)&c->net4,
	    (struct sockaddr *)&c->mask4, NULL);

	/* gateway should be link sdl with ifindex of an address interface */
	verify_link_gateway(rtm, c->ifindex);

	int expected_rt_flags = RTF_UP | RTF_DONE | RTF_PINNED;
	verify_route_message_extra(rtm, c->ifindex, expected_rt_flags);
}

RTM_DECLARE_ROOT_TEST(rtm_add_v4_gu_ifa_ordered_success,
    "Tests ordering of the messages for IPv4 unicast ifaddr assignment");

ATF_TC_BODY(rtm_add_v4_gu_ifa_ordered_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv4_iface(tc);

	c->rtsock_fd = rtsock_setup_socket();

	ret = iface_setup_addr(c->ifname, c->addr4_str, c->plen4);

	int count = 0, tries = 0;

	enum msgtype {
		MSG_IFADDR,
		MSG_PREFIXROUTE,
		MSG_MAX,
	};

	int msg_array[MSG_MAX];

	bzero(msg_array, sizeof(msg_array));

	while (count < 2 && tries < 20) {
		rtm = rtsock_read_rtm(c->rtsock_fd, buffer, sizeof(buffer));
		tries++;
		/* Classify */
		if (rtm->rtm_type == RTM_NEWADDR) {
			RLOG("MSG_IFADDR: %d", count);
			msg_array[MSG_IFADDR] = count++;
			continue;
		}

		/* Find RTM_ADD with netmask - this should skip both host route and LLADDR */
		if ((rtm->rtm_type == RTM_ADD) && (rtsock_find_rtm_sa(rtm, RTA_NETMASK))) {
			RLOG("MSG_PREFIXROUTE: %d", count);
			msg_array[MSG_PREFIXROUTE] = count++;
			continue;
		}

		RLOG("skipping msg type %s, try: %d", rtsock_print_cmdtype(rtm->rtm_type),
		    tries);
	}

	/* TODO: verify multicast */
	ATF_REQUIRE_MSG(count == 2, "Received only %d/2 messages", count);
	ATF_REQUIRE_MSG(msg_array[MSG_IFADDR] == 0, "ifaddr message is not the first");
}

RTM_DECLARE_ROOT_TEST(rtm_del_v4_gu_ifa_prefixroute_success,
    "Tests validness for the prefix route removal after ifaddr assignment");

ATF_TC_BODY(rtm_del_v4_gu_ifa_prefixroute_success, tc)
{
	DECLARE_TEST_VARS;

	c = presetup_ipv4_iface(tc);


	ret = iface_setup_addr(c->ifname, c->addr4_str, c->plen4);

	c->rtsock_fd = rtsock_setup_socket();

	ret = iface_delete_addr(c->ifname, c->addr4_str);

	while (true) {
		rtm = rtsock_read_rtm(c->rtsock_fd, buffer, sizeof(buffer));
		/* Find RTM_ADD with netmask - this should skip both host route and LLADDR */
		if ((rtm->rtm_type == RTM_DELETE) && (rtsock_find_rtm_sa(rtm, RTA_NETMASK)))
			break;
	}

	/* This should be a message for the prefix route */
	verify_route_message(rtm, RTM_DELETE, (struct sockaddr *)&c->net4,
	    (struct sockaddr *)&c->mask4, NULL);

	/* gateway should be link sdl with ifindex of an address interface */
	verify_link_gateway(rtm, c->ifindex);

	int expected_rt_flags = RTF_DONE | RTF_PINNED;
	verify_route_message_extra(rtm, c->ifindex, expected_rt_flags);
}


ATF_TP_ADD_TCS(tp)
{
	ATF_TP_ADD_TC(tp, rtm_get_v4_exact_success);
	ATF_TP_ADD_TC(tp, rtm_get_v4_lpm_success);
	ATF_TP_ADD_TC(tp, rtm_get_v4_hostbits_success);
	ATF_TP_ADD_TC(tp, rtm_get_v4_empty_dst_failure);
	ATF_TP_ADD_TC(tp, rtm_add_v4_no_rtf_host_success);
	ATF_TP_ADD_TC(tp, rtm_add_v4_gw_direct_success);
	ATF_TP_ADD_TC(tp, rtm_del_v4_prefix_nogw_success);
	ATF_TP_ADD_TC(tp, rtm_add_v6_gu_gw_gu_direct_success);
	ATF_TP_ADD_TC(tp, rtm_del_v6_gu_prefix_nogw_success);
	ATF_TP_ADD_TC(tp, rtm_change_v4_gw_success);
	ATF_TP_ADD_TC(tp, rtm_change_v4_mtu_success);
	ATF_TP_ADD_TC(tp, rtm_change_v4_flags_success);
	ATF_TP_ADD_TC(tp, rtm_change_v6_gw_success);
	ATF_TP_ADD_TC(tp, rtm_change_v6_mtu_success);
	ATF_TP_ADD_TC(tp, rtm_change_v6_flags_success);
	/* ifaddr tests */
	ATF_TP_ADD_TC(tp, rtm_add_v6_gu_ifa_hostroute_success);
	ATF_TP_ADD_TC(tp, rtm_add_v6_gu_ifa_prefixroute_success);
	ATF_TP_ADD_TC(tp, rtm_add_v6_gu_ifa_ordered_success);
	ATF_TP_ADD_TC(tp, rtm_del_v6_gu_ifa_hostroute_success);
	ATF_TP_ADD_TC(tp, rtm_del_v6_gu_ifa_prefixroute_success);
	ATF_TP_ADD_TC(tp, rtm_add_v4_gu_ifa_ordered_success);
	ATF_TP_ADD_TC(tp, rtm_del_v4_gu_ifa_prefixroute_success);
	/* temporal routes */
	ATF_TP_ADD_TC(tp, rtm_add_v4_temporal1_success);
	ATF_TP_ADD_TC(tp, rtm_add_v6_temporal1_success);

	return (atf_no_error());
}