diff --git a/tests/sys/netgraph/basic.c b/tests/sys/netgraph/basic.c --- a/tests/sys/netgraph/basic.c +++ b/tests/sys/netgraph/basic.c @@ -1,4 +1,4 @@ -/* +/*- * SPDX-License-Identifier: BSD-3-Clause * * Copyright 2021 Lutz Donnerhacke @@ -37,17 +37,18 @@ #include "util.h" -static void get_data(void *data, size_t len, void *ctx); +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; + char msg[] = "test"; + int received; ng_init(); ng_connect(".", "a", ".", "b"); @@ -64,10 +65,11 @@ { atf_tc_set_md_var(conf, "require.user", "root"); } + ATF_TC_BODY(node, dummy) { - char msg[] = "test"; - int received; + char msg[] = "test"; + int received; ng_init(); ng_mkpeer(".", "a", "hub", "a"); @@ -99,6 +101,7 @@ { atf_tc_set_md_var(conf, "require.user", "root"); } + ATF_TC_BODY(message, dummy) { ng_init(); @@ -116,6 +119,7 @@ { atf_tc_set_md_var(conf, "require.user", "root"); } + ATF_TC_BODY(same_name, dummy) { ng_init(); @@ -139,18 +143,20 @@ { atf_tc_set_md_var(conf, "require.user", "root"); } + ATF_TC_BODY(queuelimit, dummy) { - int received, i; - char msg[] = "test"; - const int MAX = 1000; + 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++) { + for (i = 0; i < MAX; i++) + { ng_send_data("a", msg, sizeof(msg)); if (errno != 0) break; @@ -183,7 +189,7 @@ static void get_data(void *data, size_t len, void *ctx) { - int *cnt = ctx; + int *cnt = ctx; (void)data; printf("Got %zu bytes of data.\n", len); diff --git a/tests/sys/netgraph/util.h b/tests/sys/netgraph/util.h --- a/tests/sys/netgraph/util.h +++ b/tests/sys/netgraph/util.h @@ -1,4 +1,4 @@ -/* +/*- * SPDX-License-Identifier: BSD-3-Clause * * Copyright 2021 Lutz Donnerhacke @@ -34,54 +34,64 @@ #include -void _ng_connect (char const *path1, char const *hook1, - char const *path2, char const *hook2, - char const *file, size_t line); +void +_ng_connect(char const *path1, char const *hook1, + char const *path2, char const *hook2, + char const *file, size_t line); #define ng_connect(p1,h1,p2,h2) \ _ng_connect(p1,h1,p2,h2,__FILE__,__LINE__) -void _ng_mkpeer (char const *path1, char const *hook1, - char const *type, char const *hook2, - char const *file, size_t line); +void +_ng_mkpeer(char const *path1, char const *hook1, + char const *type, char const *hook2, + char const *file, size_t line); #define ng_mkpeer(p1,h1,t,h2) \ _ng_mkpeer(p1,h1,t,h2,__FILE__,__LINE__) -void _ng_shutdown(char const *path, - char const *file, size_t line); +void +_ng_shutdown(char const *path, + char const *file, size_t line); #define ng_shutdown(p) \ _ng_shutdown(p,__FILE__,__LINE__) -void _ng_rmhook (char const *path, char const *hook, - char const *file, size_t line); +void +_ng_rmhook(char const *path, char const *hook, + char const *file, size_t line); #define ng_rmhook(p,h) \ _ng_rmhook(p,h,__FILE__,__LINE__) -void _ng_name (char const *path, char const *name, - char const *file, size_t line); +void +_ng_name(char const *path, char const *name, + char const *file, size_t line); #define ng_name(p,n) \ _ng_name(p,n,__FILE__,__LINE__) typedef void (*ng_data_handler_t)(void *, size_t, void *ctx); -void ng_register_data(char const *hook, ng_data_handler_t proc); -void _ng_send_data(char const *hook, void const *, size_t, - char const *file, size_t line); +void ng_register_data(char const *hook, ng_data_handler_t proc); +void +_ng_send_data(char const *hook, void const *, size_t, + char const *file, size_t line); #define ng_send_data(h,d,l) \ _ng_send_data(h,d,l,__FILE__,__LINE__) typedef void (*ng_msg_handler_t)(char const *, struct ng_mesg *, void *); -void ng_register_msg(ng_msg_handler_t proc); -int _ng_send_msg(char const *path, char const *msg, - char const *file, size_t line); +void ng_register_msg(ng_msg_handler_t proc); +int +_ng_send_msg(char const *path, char const *msg, + char const *file, size_t line); #define ng_send_msg(p,m) \ _ng_send_msg(p,m,__FILE__,__LINE__) -int ng_handle_event (unsigned int ms, void *ctx); -void ng_handle_events(unsigned int ms, void *ctx); +int ng_handle_event(unsigned int ms, void *ctx); +void ng_handle_events(unsigned int ms, void *ctx); -typedef enum { FAIL, PASS } ng_error_t; -ng_error_t ng_errors(ng_error_t); +typedef enum +{ + FAIL, PASS +} ng_error_t; +ng_error_t ng_errors(ng_error_t); -void _ng_init(char const *file, size_t line); +void _ng_init(char const *file, size_t line); #define ng_init() \ _ng_init(__FILE__,__LINE__) diff --git a/tests/sys/netgraph/util.c b/tests/sys/netgraph/util.c --- a/tests/sys/netgraph/util.c +++ b/tests/sys/netgraph/util.c @@ -1,4 +1,4 @@ -/* +/*- * SPDX-License-Identifier: BSD-3-Clause * * Copyright 2021 Lutz Donnerhacke @@ -42,7 +42,7 @@ #include "util.h" -static int cs = -1, ds = -1; +static int cs = -1, ds = -1; static ng_error_t error_handling = FAIL; #define CHECK(r, x) do { \ @@ -54,16 +54,17 @@ } \ } while(0) -struct data_handler { - char const *hook; +struct data_handler +{ + char const *hook; ng_data_handler_t handler; - SLIST_ENTRY(data_handler) next; + 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); +static void handle_data(void *ctx); +static void handle_msg(void *ctx); void _ng_connect(char const *path1, char const *hook1, @@ -72,29 +73,29 @@ { struct ngm_connect c; - strncpy(c.ourhook, hook1, sizeof(c.ourhook)); + strncpy(c.ourhook, hook1, sizeof(c.ourhook)); strncpy(c.peerhook, hook2, sizeof(c.peerhook)); - strncpy(c.path, path2, sizeof(c.path)); + strncpy(c.path, path2, sizeof(c.path)); CHECK(, -1 != NgSendMsg(cs, path1, - NGM_GENERIC_COOKIE, NGM_CONNECT, - &c, sizeof(c))); + 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 *type, char const *hook2, char const *file, size_t line) { struct ngm_mkpeer p; - strncpy(p.ourhook, hook1, sizeof(p.ourhook)); + strncpy(p.ourhook, hook1, sizeof(p.ourhook)); strncpy(p.peerhook, hook2, sizeof(p.peerhook)); - strncpy(p.type, type, sizeof(p.type)); + strncpy(p.type, type, sizeof(p.type)); CHECK(, -1 != NgSendMsg(cs, path1, - NGM_GENERIC_COOKIE, NGM_MKPEER, - &p, sizeof(p))); + NGM_GENERIC_COOKIE, NGM_MKPEER, + &p, sizeof(p))); } void @@ -106,21 +107,21 @@ strncpy(h.ourhook, hook, sizeof(h.ourhook)); CHECK(, -1 != NgSendMsg(cs, path, - NGM_GENERIC_COOKIE, NGM_RMHOOK, - &h, sizeof(h))); + 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; + struct ngm_name n; strncpy(n.name, name, sizeof(n.name)); CHECK(, -1 != NgSendMsg(cs, path, - NGM_GENERIC_COOKIE, NGM_NAME, - &n, sizeof(n))); + NGM_GENERIC_COOKIE, NGM_NAME, + &n, sizeof(n))); } void @@ -128,8 +129,8 @@ char const *file, size_t line) { CHECK(, -1 != NgSendMsg(cs, path, - NGM_GENERIC_COOKIE, NGM_SHUTDOWN, - NULL, 0)); + NGM_GENERIC_COOKIE, NGM_SHUTDOWN, + NULL, 0)); } void @@ -152,37 +153,42 @@ } void -ng_register_msg(ng_msg_handler_t proc) { +ng_register_msg(ng_msg_handler_t proc) +{ msg_handler = proc; } static void -handle_msg(void *ctx) { +handle_msg(void *ctx) +{ struct ng_mesg *m; - char path[NG_PATHSIZ]; + char path[NG_PATHSIZ]; ATF_REQUIRE(-1 != NgAllocRecvMsg(cs, &m, path)); - if(msg_handler != NULL) - (*msg_handler)(path, m, ctx); + if (msg_handler != NULL) + (*msg_handler) (path, m, ctx); free(m); } static void -handle_data(void *ctx) { - char hook[NG_HOOKSIZ]; +handle_data(void *ctx) +{ + char hook[NG_HOOKSIZ]; struct data_handler *hnd; - u_char *data; - int len; + 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); + (*(hnd->handler)) (data, len, ctx); free(data); } @@ -190,25 +196,26 @@ 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_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)) { + switch (select(maxfd + 1, &fds, NULL, NULL, &timeout)) + { case -1: ATF_REQUIRE_ERRNO(EINTR, 1); goto retry; - case 0: /* timeout */ + case 0: /* timeout */ return 0; - default: /* something to do */ + default: /* something to do */ if (FD_ISSET(cs, &fds)) - handle_msg(context); + handle_msg(context); if (FD_ISSET(ds, &fds)) - handle_data(context); + handle_data(context); return 1; } } @@ -216,7 +223,7 @@ void ng_handle_events(unsigned int ms, void *context) { - while(ng_handle_event(ms, context)) + while (ng_handle_event(ms, context)) ; } @@ -224,7 +231,7 @@ _ng_send_msg(char const *path, char const *msg, char const *file, size_t line) { - int res; + int res; CHECK(-1, -1 != (res = NgSendAsciiMsg(cs, path, "%s", msg))); return (res); @@ -233,15 +240,16 @@ ng_error_t ng_errors(ng_error_t n) { - ng_error_t o = error_handling; + 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 */ +_ng_init(char const *file, size_t line) +{ + if (cs >= 0) /* prevent reinit */ return; CHECK(, 0 == NgMkSockNode(NULL, &cs, &ds));