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 static void 19 require_netlink(void) 20 { 21 if (modfind("netlink") == -1) 22 atf_tc_skip("netlink module not loaded"); 23 } 24 25 ATF_TC(snl_verify_genl_parsers); 26 ATF_TC_HEAD(snl_verify_genl_parsers, tc) 27 { 28 atf_tc_set_md_var(tc, "descr", "Tests snl(3) generic parsers are correct"); 29 } 30 31 ATF_TC_BODY(snl_verify_genl_parsers, tc) 32 { 33 SNL_VERIFY_PARSERS(snl_all_genl_parsers); 34 35 } 36 37 ATF_TC(test_snl_get_genl_family_success); 38 ATF_TC_HEAD(test_snl_get_genl_family_success, tc) 39 { 40 atf_tc_set_md_var(tc, "descr", "Tests successfull resolution of the 'nlctrl' family"); 41 } 42 43 ATF_TC_BODY(test_snl_get_genl_family_success, tc) 44 { 45 struct snl_state ss; 46 47 require_netlink(); 48 49 if (!snl_init(&ss, NETLINK_GENERIC)) 50 atf_tc_fail("snl_init() failed"); 51 52 ATF_CHECK_EQ(snl_get_genl_family(&ss, "nlctrl"), GENL_ID_CTRL); 53 } 54 55 ATF_TC(test_snl_get_genl_family_failure); 56 ATF_TC_HEAD(test_snl_get_genl_family_failure, tc) 57 { 58 atf_tc_set_md_var(tc, "descr", "Tests unsuccessfull resolution of 'no-such-family' family"); 59 } 60 61 ATF_TC_BODY(test_snl_get_genl_family_failure, tc) 62 { 63 struct snl_state ss; 64 65 require_netlink(); 66 67 if (!snl_init(&ss, NETLINK_GENERIC)) 68 atf_tc_fail("snl_init() failed"); 69 70 ATF_CHECK_EQ(snl_get_genl_family(&ss, "no-such-family"), 0); 71 } 72 73 ATF_TC(test_snl_get_genl_family_groups); 74 ATF_TC_HEAD(test_snl_get_genl_family_groups, tc) 75 { 76 atf_tc_set_md_var(tc, "descr", "Tests getting 'nlctrl' groups"); 77 } 78 79 ATF_TC_BODY(test_snl_get_genl_family_groups, tc) 80 { 81 struct snl_state ss; 82 struct snl_writer nw; 83 struct nlmsghdr *hdr; 84 85 require_netlink(); 86 87 if (!snl_init(&ss, NETLINK_GENERIC)) 88 atf_tc_fail("snl_init() failed"); 89 90 snl_init_writer(&ss, &nw); 91 hdr = snl_create_genl_msg_request(&nw, GENL_ID_CTRL, CTRL_CMD_GETFAMILY); 92 snl_add_msg_attr_string(&nw, CTRL_ATTR_FAMILY_NAME, "nlctrl"); 93 hdr = snl_finalize_msg(&nw); 94 snl_send_message(&ss, hdr); 95 96 hdr = snl_read_reply(&ss, hdr->nlmsg_seq); 97 ATF_CHECK(hdr != NULL); 98 ATF_CHECK(hdr->nlmsg_type != NLMSG_ERROR); 99 100 struct _getfamily_attrs attrs = {}; 101 102 ATF_CHECK(snl_parse_nlmsg(&ss, hdr, &_genl_ctrl_getfam_parser, &attrs)); 103 ATF_CHECK_EQ(attrs.mcast_groups.num_groups, 1); 104 105 struct _snl_genl_ctrl_mcast_group *group = attrs.mcast_groups.groups[0]; 106 107 ATF_CHECK(group->mcast_grp_id > 0); 108 ATF_CHECK(!strcmp(group->mcast_grp_name, "notify")); 109 } 110 111 ATF_TP_ADD_TCS(tp) 112 { 113 ATF_TP_ADD_TC(tp, snl_verify_genl_parsers); 114 ATF_TP_ADD_TC(tp, test_snl_get_genl_family_success); 115 ATF_TP_ADD_TC(tp, test_snl_get_genl_family_failure); 116 ATF_TP_ADD_TC(tp, test_snl_get_genl_family_groups); 117 118 return (atf_no_error()); 119 } 120 121