Blame view
net/netfilter/nf_conntrack_helper.c
6.56 KB
7e5d03bb9
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
/* Helper handling for netfilter. */ /* (C) 1999-2001 Paul `Rusty' Russell * (C) 2002-2006 Netfilter Core Team <coreteam@netfilter.org> * (C) 2003,2004 USAGI/WIDE Project <http://www.linux-ipv6.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. */ #include <linux/types.h> #include <linux/netfilter.h> #include <linux/module.h> #include <linux/skbuff.h> #include <linux/vmalloc.h> #include <linux/stddef.h> #include <linux/slab.h> #include <linux/random.h> #include <linux/err.h> #include <linux/kernel.h> #include <linux/netdevice.h> |
2ba4cc319
|
23 |
#include <linux/rculist.h> |
efb9a8c28
|
24 |
#include <linux/rtnetlink.h> |
7e5d03bb9
|
25 |
|
7e5d03bb9
|
26 27 |
#include <net/netfilter/nf_conntrack.h> #include <net/netfilter/nf_conntrack_l3proto.h> |
605dcad6c
|
28 |
#include <net/netfilter/nf_conntrack_l4proto.h> |
7e5d03bb9
|
29 30 |
#include <net/netfilter/nf_conntrack_helper.h> #include <net/netfilter/nf_conntrack_core.h> |
ceceae1b1
|
31 |
#include <net/netfilter/nf_conntrack_extend.h> |
7e5d03bb9
|
32 |
|
58a3c9bb0
|
33 |
static DEFINE_MUTEX(nf_ct_helper_mutex); |
b8a7fe6c1
|
34 35 36 37 38 39 40 41 42 43 44 |
static struct hlist_head *nf_ct_helper_hash __read_mostly; static unsigned int nf_ct_helper_hsize __read_mostly; static unsigned int nf_ct_helper_count __read_mostly; static int nf_ct_helper_vmalloc; /* Stupid hash, but collision free for the default registrations of the * helpers currently in the kernel. */ static unsigned int helper_hash(const struct nf_conntrack_tuple *tuple) { return (((tuple->src.l3num << 8) | tuple->dst.protonum) ^ |
a34c45896
|
45 |
(__force __u16)tuple->src.u.all) % nf_ct_helper_hsize; |
b8a7fe6c1
|
46 |
} |
7e5d03bb9
|
47 |
|
226c0c0ef
|
48 |
static struct nf_conntrack_helper * |
7e5d03bb9
|
49 50 |
__nf_ct_helper_find(const struct nf_conntrack_tuple *tuple) { |
b8a7fe6c1
|
51 |
struct nf_conntrack_helper *helper; |
d4156e8cd
|
52 |
struct nf_conntrack_tuple_mask mask = { .src.u.all = htons(0xFFFF) }; |
b8a7fe6c1
|
53 54 |
struct hlist_node *n; unsigned int h; |
7e5d03bb9
|
55 |
|
b8a7fe6c1
|
56 57 58 59 |
if (!nf_ct_helper_count) return NULL; h = helper_hash(tuple); |
58a3c9bb0
|
60 |
hlist_for_each_entry_rcu(helper, n, &nf_ct_helper_hash[h], hnode) { |
b8a7fe6c1
|
61 62 |
if (nf_ct_tuple_src_mask_cmp(tuple, &helper->tuple, &mask)) return helper; |
7e5d03bb9
|
63 64 65 |
} return NULL; } |
7e5d03bb9
|
66 67 68 69 70 |
struct nf_conntrack_helper * __nf_conntrack_helper_find_byname(const char *name) { struct nf_conntrack_helper *h; |
b8a7fe6c1
|
71 72 |
struct hlist_node *n; unsigned int i; |
7e5d03bb9
|
73 |
|
b8a7fe6c1
|
74 |
for (i = 0; i < nf_ct_helper_hsize; i++) { |
58a3c9bb0
|
75 |
hlist_for_each_entry_rcu(h, n, &nf_ct_helper_hash[i], hnode) { |
b8a7fe6c1
|
76 77 78 |
if (!strcmp(h->name, name)) return h; } |
7e5d03bb9
|
79 |
} |
7e5d03bb9
|
80 81 |
return NULL; } |
13b183391
|
82 |
EXPORT_SYMBOL_GPL(__nf_conntrack_helper_find_byname); |
7e5d03bb9
|
83 |
|
b560580a1
|
84 85 86 87 88 89 90 91 92 93 94 95 |
struct nf_conn_help *nf_ct_helper_ext_add(struct nf_conn *ct, gfp_t gfp) { struct nf_conn_help *help; help = nf_ct_ext_add(ct, NF_CT_EXT_HELPER, gfp); if (help) INIT_HLIST_HEAD(&help->expectations); else pr_debug("failed to add helper extension area"); return help; } EXPORT_SYMBOL_GPL(nf_ct_helper_ext_add); |
226c0c0ef
|
96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 |
int __nf_ct_try_assign_helper(struct nf_conn *ct, gfp_t flags) { int ret = 0; struct nf_conntrack_helper *helper; struct nf_conn_help *help = nfct_help(ct); helper = __nf_ct_helper_find(&ct->tuplehash[IP_CT_DIR_REPLY].tuple); if (helper == NULL) { if (help) rcu_assign_pointer(help->helper, NULL); goto out; } if (help == NULL) { help = nf_ct_helper_ext_add(ct, flags); if (help == NULL) { ret = -ENOMEM; goto out; } } else { memset(&help->help, 0, sizeof(help->help)); } rcu_assign_pointer(help->helper, helper); out: return ret; } EXPORT_SYMBOL_GPL(__nf_ct_try_assign_helper); |
7e5d03bb9
|
124 125 126 127 128 129 130 131 |
static inline int unhelp(struct nf_conntrack_tuple_hash *i, const struct nf_conntrack_helper *me) { struct nf_conn *ct = nf_ct_tuplehash_to_ctrack(i); struct nf_conn_help *help = nfct_help(ct); if (help && help->helper == me) { nf_conntrack_event(IPCT_HELPER, ct); |
3c158f7f5
|
132 |
rcu_assign_pointer(help->helper, NULL); |
7e5d03bb9
|
133 134 135 |
} return 0; } |
9858a3ae1
|
136 137 138 139 140 141 142 143 144 145 146 147 148 |
void nf_ct_helper_destroy(struct nf_conn *ct) { struct nf_conn_help *help = nfct_help(ct); struct nf_conntrack_helper *helper; if (help) { rcu_read_lock(); helper = rcu_dereference(help->helper); if (helper && helper->destroy) helper->destroy(ct); rcu_read_unlock(); } } |
7e5d03bb9
|
149 150 |
int nf_conntrack_helper_register(struct nf_conntrack_helper *me) { |
b8a7fe6c1
|
151 |
unsigned int h = helper_hash(&me->tuple); |
6002f266b
|
152 153 |
BUG_ON(me->expect_policy == NULL); BUG_ON(me->expect_class_max >= NF_CT_MAX_EXPECT_CLASSES); |
af9d32ad6
|
154 |
BUG_ON(strlen(me->name) > NF_CT_HELPER_NAME_LEN - 1); |
7e5d03bb9
|
155 |
|
58a3c9bb0
|
156 157 |
mutex_lock(&nf_ct_helper_mutex); hlist_add_head_rcu(&me->hnode, &nf_ct_helper_hash[h]); |
b8a7fe6c1
|
158 |
nf_ct_helper_count++; |
58a3c9bb0
|
159 |
mutex_unlock(&nf_ct_helper_mutex); |
7e5d03bb9
|
160 161 162 |
return 0; } |
13b183391
|
163 |
EXPORT_SYMBOL_GPL(nf_conntrack_helper_register); |
7e5d03bb9
|
164 |
|
680479376
|
165 166 |
static void __nf_conntrack_helper_unregister(struct nf_conntrack_helper *me, struct net *net) |
7e5d03bb9
|
167 |
{ |
7e5d03bb9
|
168 |
struct nf_conntrack_tuple_hash *h; |
31f15875c
|
169 |
struct nf_conntrack_expect *exp; |
58c0fb0dd
|
170 |
const struct hlist_node *n, *next; |
ea781f197
|
171 |
const struct hlist_nulls_node *nn; |
31f15875c
|
172 |
unsigned int i; |
7e5d03bb9
|
173 |
|
7e5d03bb9
|
174 |
/* Get rid of expectations */ |
31f15875c
|
175 176 |
for (i = 0; i < nf_ct_expect_hsize; i++) { hlist_for_each_entry_safe(exp, n, next, |
680479376
|
177 |
&net->ct.expect_hash[i], hnode) { |
31f15875c
|
178 179 180 181 182 183 |
struct nf_conn_help *help = nfct_help(exp->master); if ((help->helper == me || exp->helper == me) && del_timer(&exp->timeout)) { nf_ct_unlink_expect(exp); nf_ct_expect_put(exp); } |
7e5d03bb9
|
184 185 186 187 |
} } /* Get rid of expecteds, set helpers to NULL. */ |
38fb0afcd
|
188 |
hlist_nulls_for_each_entry(h, nn, &net->ct.unconfirmed, hnnode) |
7e5d03bb9
|
189 190 |
unhelp(h, me); for (i = 0; i < nf_conntrack_htable_size; i++) { |
ea781f197
|
191 |
hlist_nulls_for_each_entry(h, nn, &net->ct.hash[i], hnnode) |
7e5d03bb9
|
192 193 |
unhelp(h, me); } |
680479376
|
194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 |
} void nf_conntrack_helper_unregister(struct nf_conntrack_helper *me) { struct net *net; mutex_lock(&nf_ct_helper_mutex); hlist_del_rcu(&me->hnode); nf_ct_helper_count--; mutex_unlock(&nf_ct_helper_mutex); /* Make sure every nothing is still using the helper unless its a * connection in the hash. */ synchronize_rcu(); |
efb9a8c28
|
209 |
rtnl_lock(); |
680479376
|
210 211 212 |
spin_lock_bh(&nf_conntrack_lock); for_each_net(net) __nf_conntrack_helper_unregister(me, net); |
f8ba1affa
|
213 |
spin_unlock_bh(&nf_conntrack_lock); |
efb9a8c28
|
214 |
rtnl_unlock(); |
7e5d03bb9
|
215 |
} |
13b183391
|
216 |
EXPORT_SYMBOL_GPL(nf_conntrack_helper_unregister); |
ceceae1b1
|
217 |
|
61eb3107c
|
218 |
static struct nf_ct_ext_type helper_extend __read_mostly = { |
ceceae1b1
|
219 220 221 222 |
.len = sizeof(struct nf_conn_help), .align = __alignof__(struct nf_conn_help), .id = NF_CT_EXT_HELPER, }; |
8d4bc5b6b
|
223 |
int nf_conntrack_helper_init(void) |
ceceae1b1
|
224 |
{ |
b8a7fe6c1
|
225 226 227 228 |
int err; nf_ct_helper_hsize = 1; /* gets rounded up to use one page */ nf_ct_helper_hash = nf_ct_alloc_hashtable(&nf_ct_helper_hsize, |
ea781f197
|
229 |
&nf_ct_helper_vmalloc, 0); |
b8a7fe6c1
|
230 231 232 233 234 235 236 237 238 239 240 241 242 |
if (!nf_ct_helper_hash) return -ENOMEM; err = nf_ct_extend_register(&helper_extend); if (err < 0) goto err1; return 0; err1: nf_ct_free_hashtable(nf_ct_helper_hash, nf_ct_helper_vmalloc, nf_ct_helper_hsize); return err; |
ceceae1b1
|
243 |
} |
8d4bc5b6b
|
244 |
void nf_conntrack_helper_fini(void) |
ceceae1b1
|
245 246 |
{ nf_ct_extend_unregister(&helper_extend); |
b8a7fe6c1
|
247 248 |
nf_ct_free_hashtable(nf_ct_helper_hash, nf_ct_helper_vmalloc, nf_ct_helper_hsize); |
ceceae1b1
|
249 |
} |