Blame view
net/netfilter/xt_dscp.c
2.82 KB
9ba162761 [NETFILTER]: x_ta... |
1 2 |
/* IP tables module for matching the value of the IPv4/IPv6 DSCP field * |
9ba162761 [NETFILTER]: x_ta... |
3 4 5 6 7 8 |
* (C) 2002 by Harald Welte <laforge@netfilter.org> * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License version 2 as * published by the Free Software Foundation. */ |
8bee4bad0 netfilter: xt ext... |
9 |
#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt |
9ba162761 [NETFILTER]: x_ta... |
10 11 12 13 14 |
#include <linux/module.h> #include <linux/skbuff.h> #include <linux/ip.h> #include <linux/ipv6.h> #include <net/dsfield.h> |
9ba162761 [NETFILTER]: x_ta... |
15 |
#include <linux/netfilter/x_tables.h> |
c3b33e6a2 [NETFILTER]: Merg... |
16 |
#include <linux/netfilter/xt_dscp.h> |
9ba162761 [NETFILTER]: x_ta... |
17 18 |
MODULE_AUTHOR("Harald Welte <laforge@netfilter.org>"); |
2ae15b64e [NETFILTER]: Upda... |
19 |
MODULE_DESCRIPTION("Xtables: DSCP/TOS field match"); |
9ba162761 [NETFILTER]: x_ta... |
20 21 22 |
MODULE_LICENSE("GPL"); MODULE_ALIAS("ipt_dscp"); MODULE_ALIAS("ip6t_dscp"); |
c3b33e6a2 [NETFILTER]: Merg... |
23 |
MODULE_ALIAS("ipt_tos"); |
f1095ab51 [NETFILTER]: IPv6... |
24 |
MODULE_ALIAS("ip6t_tos"); |
9ba162761 [NETFILTER]: x_ta... |
25 |
|
d3c5ee6d5 [NETFILTER]: x_ta... |
26 |
static bool |
62fc80510 netfilter: xtable... |
27 |
dscp_mt(const struct sk_buff *skb, struct xt_action_param *par) |
9ba162761 [NETFILTER]: x_ta... |
28 |
{ |
f7108a20d netfilter: xtable... |
29 |
const struct xt_dscp_info *info = par->matchinfo; |
1d93a9cba [NETFILTER]: x_ta... |
30 31 32 33 |
u_int8_t dscp = ipv4_get_dsfield(ip_hdr(skb)) >> XT_DSCP_SHIFT; return (dscp == info->dscp) ^ !!info->invert; } |
d3c5ee6d5 [NETFILTER]: x_ta... |
34 |
static bool |
62fc80510 netfilter: xtable... |
35 |
dscp_mt6(const struct sk_buff *skb, struct xt_action_param *par) |
1d93a9cba [NETFILTER]: x_ta... |
36 |
{ |
f7108a20d netfilter: xtable... |
37 |
const struct xt_dscp_info *info = par->matchinfo; |
0660e03f6 [SK_BUFF]: Introd... |
38 |
u_int8_t dscp = ipv6_get_dsfield(ipv6_hdr(skb)) >> XT_DSCP_SHIFT; |
9ba162761 [NETFILTER]: x_ta... |
39 40 41 |
return (dscp == info->dscp) ^ !!info->invert; } |
b0f38452f netfilter: xtable... |
42 |
static int dscp_mt_check(const struct xt_mtchk_param *par) |
9ba162761 [NETFILTER]: x_ta... |
43 |
{ |
9b4fce7a3 netfilter: xtable... |
44 |
const struct xt_dscp_info *info = par->matchinfo; |
9ba162761 [NETFILTER]: x_ta... |
45 |
|
9b4fce7a3 netfilter: xtable... |
46 |
if (info->dscp > XT_DSCP_MAX) { |
8bee4bad0 netfilter: xt ext... |
47 48 |
pr_info("dscp %x out of range ", info->dscp); |
4a5a5c73b netfilter: xtable... |
49 |
return -EDOM; |
9ba162761 [NETFILTER]: x_ta... |
50 |
} |
bd414ee60 netfilter: xtable... |
51 |
return 0; |
9ba162761 [NETFILTER]: x_ta... |
52 |
} |
62fc80510 netfilter: xtable... |
53 |
static bool tos_mt(const struct sk_buff *skb, struct xt_action_param *par) |
f1095ab51 [NETFILTER]: IPv6... |
54 |
{ |
f7108a20d netfilter: xtable... |
55 |
const struct xt_tos_match_info *info = par->matchinfo; |
f1095ab51 [NETFILTER]: IPv6... |
56 |
|
aa5fa3185 netfilter: xtable... |
57 |
if (par->family == NFPROTO_IPV4) |
f1095ab51 [NETFILTER]: IPv6... |
58 59 60 61 62 63 |
return ((ip_hdr(skb)->tos & info->tos_mask) == info->tos_value) ^ !!info->invert; else return ((ipv6_get_dsfield(ipv6_hdr(skb)) & info->tos_mask) == info->tos_value) ^ !!info->invert; } |
d3c5ee6d5 [NETFILTER]: x_ta... |
64 |
static struct xt_match dscp_mt_reg[] __read_mostly = { |
4470bbc74 [NETFILTER]: x_ta... |
65 66 |
{ .name = "dscp", |
ee999d8b9 netfilter: x_tabl... |
67 |
.family = NFPROTO_IPV4, |
d3c5ee6d5 [NETFILTER]: x_ta... |
68 69 |
.checkentry = dscp_mt_check, .match = dscp_mt, |
4470bbc74 [NETFILTER]: x_ta... |
70 71 72 73 74 |
.matchsize = sizeof(struct xt_dscp_info), .me = THIS_MODULE, }, { .name = "dscp", |
ee999d8b9 netfilter: x_tabl... |
75 |
.family = NFPROTO_IPV6, |
d3c5ee6d5 [NETFILTER]: x_ta... |
76 77 |
.checkentry = dscp_mt_check, .match = dscp_mt6, |
4470bbc74 [NETFILTER]: x_ta... |
78 79 80 |
.matchsize = sizeof(struct xt_dscp_info), .me = THIS_MODULE, }, |
c3b33e6a2 [NETFILTER]: Merg... |
81 82 |
{ .name = "tos", |
f1095ab51 [NETFILTER]: IPv6... |
83 |
.revision = 1, |
ee999d8b9 netfilter: x_tabl... |
84 |
.family = NFPROTO_IPV4, |
f1095ab51 [NETFILTER]: IPv6... |
85 86 87 88 89 90 91 |
.match = tos_mt, .matchsize = sizeof(struct xt_tos_match_info), .me = THIS_MODULE, }, { .name = "tos", .revision = 1, |
ee999d8b9 netfilter: x_tabl... |
92 |
.family = NFPROTO_IPV6, |
f1095ab51 [NETFILTER]: IPv6... |
93 94 95 96 |
.match = tos_mt, .matchsize = sizeof(struct xt_tos_match_info), .me = THIS_MODULE, }, |
9ba162761 [NETFILTER]: x_ta... |
97 |
}; |
d3c5ee6d5 [NETFILTER]: x_ta... |
98 |
static int __init dscp_mt_init(void) |
9ba162761 [NETFILTER]: x_ta... |
99 |
{ |
d3c5ee6d5 [NETFILTER]: x_ta... |
100 |
return xt_register_matches(dscp_mt_reg, ARRAY_SIZE(dscp_mt_reg)); |
9ba162761 [NETFILTER]: x_ta... |
101 |
} |
d3c5ee6d5 [NETFILTER]: x_ta... |
102 |
static void __exit dscp_mt_exit(void) |
9ba162761 [NETFILTER]: x_ta... |
103 |
{ |
d3c5ee6d5 [NETFILTER]: x_ta... |
104 |
xt_unregister_matches(dscp_mt_reg, ARRAY_SIZE(dscp_mt_reg)); |
9ba162761 [NETFILTER]: x_ta... |
105 |
} |
d3c5ee6d5 [NETFILTER]: x_ta... |
106 107 |
module_init(dscp_mt_init); module_exit(dscp_mt_exit); |