1 files changed, 277 insertions, 0 deletions
diff --git a/tests/sys/netgraph/util.c b/tests/sys/netgraph/util.c
new file mode 100644
index 000000000000..b25e63dfa76d
--- /dev/null
+++ b/tests/sys/netgraph/util.c
@@ -0,0 +1,277 @@
+/*-
+ * 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 <stdlib.h>
+#include <string.h>
+
+#include <sys/select.h>
+#include <sys/queue.h>
+
+#include "util.h"
+
+
+static int	cs = -1, ds = -1;
+static ng_error_t error_handling = FAIL;
+
+#define CHECK(r, x)	do {			\
+	if (!(x)) {				\
+		if (error_handling == PASS)	\
+		    return r;			\
+		atf_tc_fail_requirement(file, line, "%s (%s)", \
+		    #x " not met", strerror(errno));\
+	}					\
+} while(0)
+
+struct data_handler
+{
+	char const     *hook;
+	ng_data_handler_t handler;
+			SLIST_ENTRY(data_handler) next;
+};
+static SLIST_HEAD(, data_handler) data_head = SLIST_HEAD_INITIALIZER(data_head);
+static ng_msg_handler_t msg_handler = NULL;
+
+static void	handle_data(void *ctx);
+static void	handle_msg(void *ctx);
+
+void
+_ng_connect(char const *path1, char const *hook1,
+	    char const *path2, char const *hook2,
+	    char const *file, size_t line)
+{
+	struct ngm_connect c;
+
+	strncpy(c.ourhook, hook1, sizeof(c.ourhook));
+	strncpy(c.peerhook, hook2, sizeof(c.peerhook));
+	strncpy(c.path, path2, sizeof(c.path));
+
+	CHECK(, -1 != NgSendMsg(cs, path1,
+				NGM_GENERIC_COOKIE, NGM_CONNECT,
+				&c, sizeof(c)));
+}
+
+void
+_ng_mkpeer(char const *path1, char const *hook1,
+	   char const *type, char const *hook2,
+	   char const *file, size_t line)
+{
+	struct ngm_mkpeer p;
+
+	strncpy(p.ourhook, hook1, sizeof(p.ourhook));
+	strncpy(p.peerhook, hook2, sizeof(p.peerhook));
+	strncpy(p.type, type, sizeof(p.type));
+
+	CHECK(, -1 != NgSendMsg(cs, path1,
+				NGM_GENERIC_COOKIE, NGM_MKPEER,
+				&p, sizeof(p)));
+}
+
+void
+_ng_rmhook(char const *path, char const *hook,
+	   char const *file, size_t line)
+{
+	struct ngm_rmhook h;
+
+	strncpy(h.ourhook, hook, sizeof(h.ourhook));
+
+	CHECK(, -1 != NgSendMsg(cs, path,
+				NGM_GENERIC_COOKIE, NGM_RMHOOK,
+				&h, sizeof(h)));
+}
+
+void
+_ng_name(char const *path, char const *name,
+	 char const *file, size_t line)
+{
+	struct ngm_name	n;
+
+	strncpy(n.name, name, sizeof(n.name));
+
+	CHECK(, -1 != NgSendMsg(cs, path,
+				NGM_GENERIC_COOKIE, NGM_NAME,
+				&n, sizeof(n)));
+}
+
+void
+_ng_shutdown(char const *path,
+	     char const *file, size_t line)
+{
+	CHECK(, -1 != NgSendMsg(cs, path,
+				NGM_GENERIC_COOKIE, NGM_SHUTDOWN,
+				NULL, 0));
+}
+
+void
+ng_register_data(char const *hook, ng_data_handler_t proc)
+{
+	struct data_handler *p;
+
+	ATF_REQUIRE(NULL != (p = calloc(1, sizeof(struct data_handler))));
+	ATF_REQUIRE(NULL != (p->hook = strdup(hook)));
+	ATF_REQUIRE(NULL != (p->handler = proc));
+	SLIST_INSERT_HEAD(&data_head, p, next);
+}
+
+void
+_ng_send_data(char const *hook,
+	      void const *data, size_t len,
+	      char const *file, size_t line)
+{
+	CHECK(, -1 != NgSendData(ds, hook, data, len));
+}
+
+void
+ng_register_msg(ng_msg_handler_t proc)
+{
+	msg_handler = proc;
+}
+
+static void
+handle_msg(void *ctx)
+{
+	struct ng_mesg *m;
+	char		path[NG_PATHSIZ];
+
+	ATF_REQUIRE(-1 != NgAllocRecvMsg(cs, &m, path));
+
+	if (msg_handler != NULL)
+		(*msg_handler) (path, m, ctx);
+
+	free(m);
+}
+
+static void
+handle_data(void *ctx)
+{
+	char		hook[NG_HOOKSIZ];
+	struct data_handler *hnd;
+	u_char	       *data;
+	int		len;
+
+	ATF_REQUIRE(0 < (len = NgAllocRecvData(ds, &data, hook)));
+	SLIST_FOREACH(hnd, &data_head, next)
+	{
+		if (0 == strcmp(hnd->hook, hook))
+			break;
+	}
+
+	if (hnd != NULL)
+		(*(hnd->handler)) (data, len, ctx);
+
+	free(data);
+}
+
+int
+ng_handle_event(unsigned int ms, void *context)
+{
+	fd_set		fds;
+	int		maxfd = (ds < cs) ? cs : ds;
+	struct timeval	timeout = {0, ms * 1000lu};
+
+	FD_ZERO(&fds);
+	FD_SET(cs, &fds);
+	FD_SET(ds, &fds);
+retry:
+	switch (select(maxfd + 1, &fds, NULL, NULL, &timeout))
+	{
+	case -1:
+		ATF_REQUIRE_ERRNO(EINTR, 1);
+		goto retry;
+	case 0:			/* timeout */
+		return 0;
+	default:		/* something to do */
+		if (FD_ISSET(cs, &fds))
+			handle_msg(context);
+		if (FD_ISSET(ds, &fds))
+			handle_data(context);
+		return 1;
+	}
+}
+
+void
+ng_handle_events(unsigned int ms, void *context)
+{
+	while (ng_handle_event(ms, context))
+		;
+}
+
+int
+_ng_send_msg(char const *path, char const *msg,
+	     char const *file, size_t line)
+{
+	int		res;
+
+	CHECK(-1, -1 != (res = NgSendAsciiMsg(cs, path, "%s", msg)));
+	return (res);
+}
+
+ng_error_t
+ng_errors(ng_error_t n)
+{
+	ng_error_t	o = error_handling;
+
+	error_handling = n;
+	return (o);
+}
+
+void
+_ng_init(char const *file, size_t line)
+{
+	if (cs >= 0)		/* prevent reinit */
+		return;
+
+	CHECK(, 0 == NgMkSockNode(NULL, &cs, &ds));
+	NgSetDebug(3);
+}
+
+#define GD(x) void				\
+get_data##x(void *data, size_t len, void *ctx) {\
+	int	       *cnt = ctx;		\
+						\
+	(void)data;				\
+	(void)len;				\
+	cnt[x]++;				\
+}
+
+GD(0)
+GD(1)
+GD(2)
+GD(3)
+GD(4)
+GD(5)
+GD(6)
+GD(7)
+GD(8)
+GD(9)