1 #include <stdio.h>
2 #include <string.h>
3 #include <stdlib.h>
4
5 #include <sys/param.h>
6 #include <sys/module.h>
7
8 #include <netlink/netlink.h>
9 #include "netlink/netlink_snl.h"
10 #include "netlink/netlink_snl_generic.h"
11
12 #include <atf-c.h>
13
14 static const struct snl_hdr_parser *snl_all_genl_parsers[] = {
15 &_genl_ctrl_getfam_parser, &_genl_ctrl_mc_parser,
16 };
17
18 ATF_TC(snl_verify_genl_parsers);
ATF_TC_HEAD(snl_verify_genl_parsers,tc)19 ATF_TC_HEAD(snl_verify_genl_parsers, tc)
20 {
21 atf_tc_set_md_var(tc, "descr", "Tests snl(3) generic parsers are correct");
22 }
23
ATF_TC_BODY(snl_verify_genl_parsers,tc)24 ATF_TC_BODY(snl_verify_genl_parsers, tc)
25 {
26 SNL_VERIFY_PARSERS(snl_all_genl_parsers);
27
28 }
29
30 ATF_TC(test_snl_get_genl_family_success);
ATF_TC_HEAD(test_snl_get_genl_family_success,tc)31 ATF_TC_HEAD(test_snl_get_genl_family_success, tc)
32 {
33 atf_tc_set_md_var(tc, "descr", "Tests successfull resolution of the 'nlctrl' family");
34 atf_tc_set_md_var(tc, "require.kmods", "netlink");
35 }
36
ATF_TC_BODY(test_snl_get_genl_family_success,tc)37 ATF_TC_BODY(test_snl_get_genl_family_success, tc)
38 {
39 struct snl_state ss;
40
41 if (!snl_init(&ss, NETLINK_GENERIC))
42 atf_tc_fail("snl_init() failed");
43
44 ATF_CHECK_EQ(snl_get_genl_family(&ss, "nlctrl"), GENL_ID_CTRL);
45 }
46
47 ATF_TC(test_snl_get_genl_family_failure);
ATF_TC_HEAD(test_snl_get_genl_family_failure,tc)48 ATF_TC_HEAD(test_snl_get_genl_family_failure, tc)
49 {
50 atf_tc_set_md_var(tc, "descr", "Tests unsuccessfull resolution of 'no-such-family' family");
51 atf_tc_set_md_var(tc, "require.kmods", "netlink");
52 }
53
ATF_TC_BODY(test_snl_get_genl_family_failure,tc)54 ATF_TC_BODY(test_snl_get_genl_family_failure, tc)
55 {
56 struct snl_state ss;
57
58 if (!snl_init(&ss, NETLINK_GENERIC))
59 atf_tc_fail("snl_init() failed");
60
61 ATF_CHECK_EQ(snl_get_genl_family(&ss, "no-such-family"), 0);
62 }
63
64 ATF_TC(test_snl_get_genl_family_groups);
ATF_TC_HEAD(test_snl_get_genl_family_groups,tc)65 ATF_TC_HEAD(test_snl_get_genl_family_groups, tc)
66 {
67 atf_tc_set_md_var(tc, "descr", "Tests getting 'nlctrl' groups");
68 atf_tc_set_md_var(tc, "require.kmods", "netlink");
69 }
70
ATF_TC_BODY(test_snl_get_genl_family_groups,tc)71 ATF_TC_BODY(test_snl_get_genl_family_groups, tc)
72 {
73 struct snl_state ss;
74 struct snl_writer nw;
75 struct nlmsghdr *hdr;
76
77 if (!snl_init(&ss, NETLINK_GENERIC))
78 atf_tc_fail("snl_init() failed");
79
80 snl_init_writer(&ss, &nw);
81 hdr = snl_create_genl_msg_request(&nw, GENL_ID_CTRL, CTRL_CMD_GETFAMILY);
82 snl_add_msg_attr_string(&nw, CTRL_ATTR_FAMILY_NAME, "nlctrl");
83 hdr = snl_finalize_msg(&nw);
84 snl_send_message(&ss, hdr);
85
86 hdr = snl_read_reply(&ss, hdr->nlmsg_seq);
87 ATF_CHECK(hdr != NULL);
88 ATF_CHECK(hdr->nlmsg_type != NLMSG_ERROR);
89
90 struct _getfamily_attrs attrs = {};
91
92 ATF_CHECK(snl_parse_nlmsg(&ss, hdr, &_genl_ctrl_getfam_parser, &attrs));
93 ATF_CHECK_EQ(attrs.mcast_groups.num_groups, 1);
94
95 struct _snl_genl_ctrl_mcast_group *group = attrs.mcast_groups.groups[0];
96
97 ATF_CHECK(group->mcast_grp_id > 0);
98 ATF_CHECK(!strcmp(group->mcast_grp_name, "notify"));
99 }
100
ATF_TP_ADD_TCS(tp)101 ATF_TP_ADD_TCS(tp)
102 {
103 ATF_TP_ADD_TC(tp, snl_verify_genl_parsers);
104 ATF_TP_ADD_TC(tp, test_snl_get_genl_family_success);
105 ATF_TP_ADD_TC(tp, test_snl_get_genl_family_failure);
106 ATF_TP_ADD_TC(tp, test_snl_get_genl_family_groups);
107
108 return (atf_no_error());
109 }
110
111