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