Differential D30633 Diff 90434 tests/sys/netgraph/hub.c

Changeset View

Standalone View

tests/sys/netgraph/hub.c

This file was added.

				/*
				* 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"

				static void get_data(void data, size_t len, void ctx);

				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";
				int received;

				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_data);
				received = 0;
				ng_send_data("a", msg, sizeof(msg));
				ng_handle_events(50, &received);
				ATF_CHECK(received == 0);

				/* send to others */
				ng_register_data("b", get_data);
				ng_register_data("c", get_data);
				received = 0;
				ng_send_data("a", msg, sizeof(msg));
				ng_handle_events(50, &received);
				ATF_CHECK(received == 2);

				received = 0;
				ng_send_data("b", msg, sizeof(msg));
				ng_handle_events(50, &received);
				ATF_CHECK(received == 2);

				received = 0;
				ng_send_data("c", msg, sizeof(msg));
				ng_handle_events(50, &received);
				ATF_CHECK(received == 2);

				/* remove a link */
				ng_rmhook(".", "b");
				received = 0;
				ng_send_data("a", msg, sizeof(msg));
				ng_handle_events(50, &received);
				ATF_CHECK(received == 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)
				{
				int received, 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_data);
				ng_register_data("b", get_data);

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

				received = 0;
				ng_send_data("a", msg, sizeof(msg));
				ng_handle_events(50, &received);
				ATF_CHECK(received == 1);

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

				received = 0;
				ng_send_data("a", msg, sizeof(msg));
				for(i = 0; i < 10; i++) /* don't run forever */
				if (!ng_handle_event(50, &received))
				break;
				ATF_CHECK(received > 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)
				{
				int received, 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_data);
				received = 0;
				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, &received);
				}
				ATF_CHECK(received > 100);
				atf_tc_expect_fail("Implementation limitation");
				ATF_CHECK(received == 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();
				}

				static void
				get_data(void data, size_t len, void ctx)
				{
				int *cnt = ctx;

				(void)data;
				printf("Got %zu bytes of data.\n", len);
				(*cnt)++;
				}