1 /* (C) 1999-2001 Michal Ludvig <michal@logix.cz> 2 * 3 * This program is free software; you can redistribute it and/or modify 4 * it under the terms of the GNU General Public License version 2 as 5 * published by the Free Software Foundation. 6 */ 7 8 #include <linux/module.h> 9 #include <linux/skbuff.h> 10 #include <linux/if_ether.h> 11 #include <linux/if_packet.h> 12 13 #include <linux/netfilter/xt_pkttype.h> 14 #include <linux/netfilter/x_tables.h> 15 16 MODULE_LICENSE("GPL"); 17 MODULE_AUTHOR("Michal Ludvig <michal@logix.cz>"); 18 MODULE_DESCRIPTION("IP tables match to match on linklayer packet type"); 19 MODULE_ALIAS("ipt_pkttype"); 20 MODULE_ALIAS("ip6t_pkttype"); 21 22 static int match(const struct sk_buff *skb, 23 const struct net_device *in, 24 const struct net_device *out, 25 const struct xt_match *match, 26 const void *matchinfo, 27 int offset, 28 unsigned int protoff, 29 int *hotdrop) 30 { 31 const struct xt_pkttype_info *info = matchinfo; 32 33 return (skb->pkt_type == info->pkttype) ^ info->invert; 34 } 35 36 static struct xt_match pkttype_match = { 37 .name = "pkttype", 38 .match = match, 39 .matchsize = sizeof(struct xt_pkttype_info), 40 .me = THIS_MODULE, 41 }; 42 43 static struct xt_match pkttype6_match = { 44 .name = "pkttype", 45 .match = match, 46 .matchsize = sizeof(struct xt_pkttype_info), 47 .me = THIS_MODULE, 48 }; 49 50 static int __init init(void) 51 { 52 int ret; 53 ret = xt_register_match(AF_INET, &pkttype_match); 54 if (ret) 55 return ret; 56 57 ret = xt_register_match(AF_INET6, &pkttype6_match); 58 if (ret) 59 xt_unregister_match(AF_INET, &pkttype_match); 60 61 return ret; 62 } 63 64 static void __exit fini(void) 65 { 66 xt_unregister_match(AF_INET, &pkttype_match); 67 xt_unregister_match(AF_INET6, &pkttype6_match); 68 } 69 70 module_init(init); 71 module_exit(fini); 72