1 files changed, 110 insertions, 0 deletions
diff --git a/tests/sys/netlink/test_snl_generic.c b/tests/sys/netlink/test_snl_generic.c
new file mode 100644
index 000000000000..8613bf04a45c
--- /dev/null
+++ b/tests/sys/netlink/test_snl_generic.c
@@ -0,0 +1,110 @@
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+
+#include <sys/param.h>
+#include <sys/module.h>
+
+#include <netlink/netlink.h>
+#include "netlink/netlink_snl.h"
+#include "netlink/netlink_snl_generic.h"
+
+#include <atf-c.h>
+
+static const struct snl_hdr_parser *snl_all_genl_parsers[] = {
+	&_genl_ctrl_getfam_parser, &_genl_ctrl_mc_parser,
+};
+
+ATF_TC(snl_verify_genl_parsers);
+ATF_TC_HEAD(snl_verify_genl_parsers, tc)
+{
+	atf_tc_set_md_var(tc, "descr", "Tests snl(3) generic parsers are correct");
+}
+
+ATF_TC_BODY(snl_verify_genl_parsers, tc)
+{
+	SNL_VERIFY_PARSERS(snl_all_genl_parsers);
+
+}
+
+ATF_TC(test_snl_get_genl_family_success);
+ATF_TC_HEAD(test_snl_get_genl_family_success, tc)
+{
+	atf_tc_set_md_var(tc, "descr", "Tests successfull resolution of the 'nlctrl' family");
+	atf_tc_set_md_var(tc, "require.kmods", "netlink");
+}
+
+ATF_TC_BODY(test_snl_get_genl_family_success, tc)
+{
+	struct snl_state ss;
+
+	if (!snl_init(&ss, NETLINK_GENERIC))
+		atf_tc_fail("snl_init() failed");
+
+	ATF_CHECK_EQ(snl_get_genl_family(&ss, "nlctrl"), GENL_ID_CTRL);
+}
+
+ATF_TC(test_snl_get_genl_family_failure);
+ATF_TC_HEAD(test_snl_get_genl_family_failure, tc)
+{
+	atf_tc_set_md_var(tc, "descr", "Tests unsuccessfull resolution of 'no-such-family' family");
+	atf_tc_set_md_var(tc, "require.kmods", "netlink");
+}
+
+ATF_TC_BODY(test_snl_get_genl_family_failure, tc)
+{
+	struct snl_state ss;
+
+	if (!snl_init(&ss, NETLINK_GENERIC))
+		atf_tc_fail("snl_init() failed");
+
+	ATF_CHECK_EQ(snl_get_genl_family(&ss, "no-such-family"), 0);
+}
+
+ATF_TC(test_snl_get_genl_family_groups);
+ATF_TC_HEAD(test_snl_get_genl_family_groups, tc)
+{
+	atf_tc_set_md_var(tc, "descr", "Tests getting 'nlctrl' groups");
+	atf_tc_set_md_var(tc, "require.kmods", "netlink");
+}
+
+ATF_TC_BODY(test_snl_get_genl_family_groups, tc)
+{
+	struct snl_state ss;
+	struct snl_writer nw;
+	struct nlmsghdr *hdr;
+
+	if (!snl_init(&ss, NETLINK_GENERIC))
+		atf_tc_fail("snl_init() failed");
+
+	snl_init_writer(&ss, &nw);
+	hdr = snl_create_genl_msg_request(&nw, GENL_ID_CTRL, CTRL_CMD_GETFAMILY);
+	snl_add_msg_attr_string(&nw, CTRL_ATTR_FAMILY_NAME, "nlctrl");
+	hdr = snl_finalize_msg(&nw);
+	snl_send_message(&ss, hdr);
+
+	hdr = snl_read_reply(&ss, hdr->nlmsg_seq);
+	ATF_CHECK(hdr != NULL);
+	ATF_CHECK(hdr->nlmsg_type != NLMSG_ERROR);
+
+	struct _getfamily_attrs attrs = {};
+
+	ATF_CHECK(snl_parse_nlmsg(&ss, hdr, &_genl_ctrl_getfam_parser, &attrs));
+	ATF_CHECK_EQ(attrs.mcast_groups.num_groups, 1);
+
+	struct _snl_genl_ctrl_mcast_group *group = attrs.mcast_groups.groups[0];
+
+	ATF_CHECK(group->mcast_grp_id > 0);
+	ATF_CHECK(!strcmp(group->mcast_grp_name, "notify"));
+}
+
+ATF_TP_ADD_TCS(tp)
+{
+	ATF_TP_ADD_TC(tp, snl_verify_genl_parsers);
+	ATF_TP_ADD_TC(tp, test_snl_get_genl_family_success);
+	ATF_TP_ADD_TC(tp, test_snl_get_genl_family_failure);
+	ATF_TP_ADD_TC(tp, test_snl_get_genl_family_groups);
+
+	return (atf_no_error());
+}
+