1 /* Kernel module to match connection tracking information. */ 2 3 /* (C) 1999-2001 Paul `Rusty' Russell 4 * (C) 2002-2005 Netfilter Core Team <coreteam@netfilter.org> 5 * 6 * This program is free software; you can redistribute it and/or modify 7 * it under the terms of the GNU General Public License version 2 as 8 * published by the Free Software Foundation. 9 */ 10 11 #include <linux/module.h> 12 #include <linux/skbuff.h> 13 #include <net/netfilter/nf_conntrack.h> 14 #include <linux/netfilter/x_tables.h> 15 #include <linux/netfilter/xt_state.h> 16 17 MODULE_LICENSE("GPL"); 18 MODULE_AUTHOR("Rusty Russell <rusty@rustcorp.com.au>"); 19 MODULE_DESCRIPTION("ip[6]_tables connection tracking state match module"); 20 MODULE_ALIAS("ipt_state"); 21 MODULE_ALIAS("ip6t_state"); 22 23 static bool 24 state_mt(const struct sk_buff *skb, const struct xt_match_param *par) 25 { 26 const struct xt_state_info *sinfo = par->matchinfo; 27 enum ip_conntrack_info ctinfo; 28 unsigned int statebit; 29 30 if (nf_ct_is_untracked(skb)) 31 statebit = XT_STATE_UNTRACKED; 32 else if (!nf_ct_get(skb, &ctinfo)) 33 statebit = XT_STATE_INVALID; 34 else 35 statebit = XT_STATE_BIT(ctinfo); 36 37 return (sinfo->statemask & statebit); 38 } 39 40 static bool 41 state_mt_check(const char *tablename, const void *inf, 42 const struct xt_match *match, void *matchinfo, 43 unsigned int hook_mask) 44 { 45 if (nf_ct_l3proto_try_module_get(match->family) < 0) { 46 printk(KERN_WARNING "can't load conntrack support for " 47 "proto=%u\n", match->family); 48 return false; 49 } 50 return true; 51 } 52 53 static void state_mt_destroy(const struct xt_match *match, void *matchinfo) 54 { 55 nf_ct_l3proto_module_put(match->family); 56 } 57 58 static struct xt_match state_mt_reg[] __read_mostly = { 59 { 60 .name = "state", 61 .family = NFPROTO_IPV4, 62 .checkentry = state_mt_check, 63 .match = state_mt, 64 .destroy = state_mt_destroy, 65 .matchsize = sizeof(struct xt_state_info), 66 .me = THIS_MODULE, 67 }, 68 { 69 .name = "state", 70 .family = NFPROTO_IPV6, 71 .checkentry = state_mt_check, 72 .match = state_mt, 73 .destroy = state_mt_destroy, 74 .matchsize = sizeof(struct xt_state_info), 75 .me = THIS_MODULE, 76 }, 77 }; 78 79 static int __init state_mt_init(void) 80 { 81 return xt_register_matches(state_mt_reg, ARRAY_SIZE(state_mt_reg)); 82 } 83 84 static void __exit state_mt_exit(void) 85 { 86 xt_unregister_matches(state_mt_reg, ARRAY_SIZE(state_mt_reg)); 87 } 88 89 module_init(state_mt_init); 90 module_exit(state_mt_exit); 91