freebsd-src/tests/sys/netgraph/hub.c

/*-
 * SPDX-License-Identifier: BSD-3-Clause
 *
 * Copyright 2021 Lutz Donnerhacke
 *
 * 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.
 * 3. Neither the name of the copyright holder nor the names of its
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS 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 COPYRIGHT HOLDER 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 <atf-c.h>
#include <errno.h>
#include <stdio.h>

#include "util.h"

ATF_TC(basic);
ATF_TC_HEAD(basic, conf)
{
	atf_tc_set_md_var(conf, "require.user", "root");
}

ATF_TC_BODY(basic, dummy)
{
	char		msg[] = "test";
	ng_counter_t	r;

	ng_errors(PASS);
	ng_shutdown("hub:");
	ng_errors(FAIL);

	ng_init();
	ng_mkpeer(".", "a", "hub", "a");
	ng_name("a", "hub");
	ng_connect(".", "b", "hub:", "b");
	ng_connect(".", "c", "hub:", "c");

	/* do not bounce back */
	ng_register_data("a", get_data0);
	ng_counter_clear(r);
	ng_send_data("a", msg, sizeof(msg));
	ng_handle_events(50, r);
	ATF_CHECK(r[0] == 0);

	/* send to others */
	ng_register_data("b", get_data0);
	ng_register_data("c", get_data0);
	ng_counter_clear(r);
	ng_send_data("a", msg, sizeof(msg));
	ng_handle_events(50, r);
	ATF_CHECK(r[0] == 2);

	ng_counter_clear(r);
	ng_send_data("b", msg, sizeof(msg));
	ng_handle_events(50, r);
	ATF_CHECK(r[0] == 2);

	ng_counter_clear(r);
	ng_send_data("c", msg, sizeof(msg));
	ng_handle_events(50, r);
	ATF_CHECK(r[0] == 2);

	/* remove a link */
	ng_rmhook(".", "b");
	ng_counter_clear(r);
	ng_send_data("a", msg, sizeof(msg));
	ng_handle_events(50, r);
	ATF_CHECK(r[0] == 1);

	ng_shutdown("hub:");
}

ATF_TC(persistence);
ATF_TC_HEAD(persistence, conf)
{
	atf_tc_set_md_var(conf, "require.user", "root");
}

ATF_TC_BODY(persistence, dummy)
{
	ng_errors(PASS);
	ng_shutdown("hub:");
	ng_errors(FAIL);

	ng_init();
	ng_mkpeer(".", "a", "hub", "a");
	ng_name("a", "hub");

	ng_send_msg("hub:", "setpersistent");
	ng_rmhook(".", "a");

	ng_shutdown("hub:");
}

ATF_TC(loop);
ATF_TC_HEAD(loop, conf)
{
	atf_tc_set_md_var(conf, "require.user", "root");
}

ATF_TC_BODY(loop, dummy)
{
	ng_counter_t	r;
	int		i;
	char		msg[] = "LOOP Alert!";

	ng_errors(PASS);
	ng_shutdown("hub1:");
	ng_shutdown("hub2:");
	ng_errors(FAIL);

	ng_init();
	ng_mkpeer(".", "a", "hub", "a");
	ng_name("a", "hub1");
	ng_mkpeer(".", "b", "hub", "b");
	ng_name("b", "hub2");

	ng_register_data("a", get_data0);
	ng_register_data("b", get_data0);

	/*-
	 * Open loop
	 *
	 *    /-- hub1
	 * . <    |
	 *    \-- hub2
	 */
	ng_connect("hub1:", "xc1", "hub2:", "xc1");

	ng_counter_clear(r);
	ng_send_data("a", msg, sizeof(msg));
	ng_handle_events(50, r);
	ATF_CHECK(r[0] == 1);

	/*-
	 * Closed loop, DANGEROUS!
	 *
	 *    /-- hub1 -\
	 * . <     |    |
	 *    \-- hub2 -/
	 */
	ng_connect("hub1:", "xc2", "hub2:", "xc2");

	ng_counter_clear(r);
	ng_send_data("a", msg, sizeof(msg));
	for (i = 0; i < 10; i++)	/* don't run forever */
		if (!ng_handle_event(50, r))
			break;
	ATF_CHECK(r[0] > 7);

	ng_shutdown("hub1:");
	ng_shutdown("hub2:");
}

ATF_TC(many_hooks);
ATF_TC_HEAD(many_hooks, conf)
{
	atf_tc_set_md_var(conf, "require.user", "root");
}

ATF_TC_BODY(many_hooks, dummy)
{
	ng_counter_t	r;
	int		i;
	char		msg[] = "test";
	const int	HOOKS = 1000;

	ng_errors(PASS);
	ng_shutdown("hub:");
	ng_errors(FAIL);

	ng_init();
	ng_mkpeer(".", "a", "hub", "a");
	ng_name("a", "hub");

	ng_register_data("a", get_data0);
	ng_counter_clear(r);
	for (i = 0; i < HOOKS; i++)
	{
		char		hook[20];

		snprintf(hook, sizeof(hook), "hook%d", i);
		ng_connect(".", hook, "hub:", hook);
		ng_errors(PASS);
		ng_send_data(hook, msg, sizeof(msg));
		ng_errors(FAIL);
		if (errno != 0)
			break;
		ng_handle_events(50, r);
	}
	ATF_CHECK(r[0] > 100);
	atf_tc_expect_fail("Implementation limitation (%d)", i);
	ATF_CHECK(r[0] == HOOKS);
	atf_tc_expect_pass();

	ng_shutdown("hub:");
}


ATF_TP_ADD_TCS(hub)
{
	ATF_TP_ADD_TC(hub, basic);
	ATF_TP_ADD_TC(hub, loop);
	ATF_TP_ADD_TC(hub, persistence);
	ATF_TP_ADD_TC(hub, many_hooks);

	return atf_no_error();
}