Differential D30629 Diff 90408 tests/sys/netgraph/basic.c

Changeset View

Standalone View

tests/sys/netgraph/basic.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
				afedorovUnsubmitted Not Done Inline Actions Do you have any objection to using the two-clause version of the BSD license? afedorov: Do you have any objection to using the two-clause version of the BSD license?
				donnerAuthorUnsubmitted Done Inline Actions Any reasoning? $ fgrep -lR BSD-3-Clause /usr/include/ \| wc -l 445 $ fgrep -lR BSD-2-Clause /usr/include/ \| wc -l 621 donner: Any reasoning? ``` $ fgrep -lR BSD-3-Clause /usr/include/ \| wc -l 445 $ fgrep -lR BSD-2…
				* 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(send_recv);
				ATF_TC_HEAD(send_recv, conf)
				{
				atf_tc_set_md_var(conf, "require.user", "root");
				}
				ATF_TC_BODY(send_recv, dummy)
				{
				char msg[] = "test";
				int received;

				ng_init();
				ng_connect(".", "a", ".", "b");
				ng_register_data("b", get_data);
				ng_send_data("a", msg, sizeof(msg));

				received = 0;
				ng_handle_events(50, &received);
				ATF_CHECK(received == 1);
				}

				ATF_TC(node);
				ATF_TC_HEAD(node, conf)
				{
				atf_tc_set_md_var(conf, "require.user", "root");
				}
				ATF_TC_BODY(node, dummy)
				{
				char msg[] = "test";
				int received;

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


				ng_connect(".", "b", "test hub:", "b");
				ng_connect(".", "c", "test hub:", "c");
				ng_register_data("a", get_data);
				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);

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

				ng_shutdown("test hub:");
				}

				ATF_TC(message);
				ATF_TC_HEAD(message, conf)
				{
				atf_tc_set_md_var(conf, "require.user", "root");
				}
				ATF_TC_BODY(message, dummy)
				{
				ng_init();
				ng_mkpeer(".", "a", "hub", "a");
				ng_name("a", "test hub");

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

				ng_shutdown("test hub:");
				}

				ATF_TC(same_name);
				ATF_TC_HEAD(same_name, conf)
				{
				atf_tc_set_md_var(conf, "require.user", "root");
				}
				ATF_TC_BODY(same_name, dummy)
				{
				ng_init();
				ng_mkpeer(".", "a", "hub", "a");
				ng_name("a", "test");

				ng_errors(PASS);
				ng_connect(".", "a", ".", "b");
				ATF_CHECK_ERRNO(EEXIST, 1);
				ng_connect(".", "b", ".", "b");
				ATF_CHECK_ERRNO(EEXIST, 1);
				ng_name(".", "test");
				ATF_CHECK_ERRNO(EADDRINUSE, 1);

				ng_errors(FAIL);
				ng_shutdown("test:");
				}

				ATF_TC(queuelimit);
				ATF_TC_HEAD(queuelimit, conf)
				{
				atf_tc_set_md_var(conf, "require.user", "root");
				}
				ATF_TC_BODY(queuelimit, dummy)
				{
				int received, i;
				char msg[] = "test";
				const int MAX = 1000;

				ng_init();
				ng_connect(".", "a", ".", "b");
				ng_register_data("b", get_data);

				ng_errors(PASS);
				for (i = 0; i < MAX; i++) {
				ng_send_data("a", msg, sizeof(msg));
				if (errno != 0)
				break;
				/* no ng_handle_events -> messages stall */
				}
				ng_errors(FAIL);
				printf("queued %d\n", i);
				sleep(3);

				received = 0;
				ng_handle_events(50, &received);
				ATF_CHECK(received > 100);
				ATF_CHECK(received == i);
				atf_tc_expect_fail("Queue full (%d)", i);
				ATF_CHECK(received == MAX);
				atf_tc_expect_pass();
				}

				ATF_TP_ADD_TCS(basic)
				{
				ATF_TP_ADD_TC(basic, send_recv);
				ATF_TP_ADD_TC(basic, node);
				ATF_TP_ADD_TC(basic, message);
				ATF_TP_ADD_TC(basic, same_name);
				ATF_TP_ADD_TC(basic, queuelimit);

				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)++;
				}