Commit | Line | Data |
---|---|---|
7af4cc3f HW |
1 | /* |
2 | * This is a module which is used for queueing packets and communicating with | |
3 | * userspace via nfetlink. | |
4 | * | |
5 | * (C) 2005 by Harald Welte <laforge@netfilter.org> | |
4ad9d4fa | 6 | * (C) 2007 by Patrick McHardy <kaber@trash.net> |
7af4cc3f HW |
7 | * |
8 | * Based on the old ipv4-only ip_queue.c: | |
9 | * (C) 2000-2002 James Morris <jmorris@intercode.com.au> | |
10 | * (C) 2003-2005 Netfilter Core Team <coreteam@netfilter.org> | |
11 | * | |
12 | * This program is free software; you can redistribute it and/or modify | |
13 | * it under the terms of the GNU General Public License version 2 as | |
14 | * published by the Free Software Foundation. | |
15 | * | |
16 | */ | |
17 | #include <linux/module.h> | |
18 | #include <linux/skbuff.h> | |
19 | #include <linux/init.h> | |
20 | #include <linux/spinlock.h> | |
21 | #include <linux/notifier.h> | |
22 | #include <linux/netdevice.h> | |
23 | #include <linux/netfilter.h> | |
838ab636 | 24 | #include <linux/proc_fs.h> |
7af4cc3f HW |
25 | #include <linux/netfilter_ipv4.h> |
26 | #include <linux/netfilter_ipv6.h> | |
27 | #include <linux/netfilter/nfnetlink.h> | |
28 | #include <linux/netfilter/nfnetlink_queue.h> | |
29 | #include <linux/list.h> | |
30 | #include <net/sock.h> | |
c01cd429 | 31 | #include <net/netfilter/nf_queue.h> |
7af4cc3f HW |
32 | |
33 | #include <asm/atomic.h> | |
34 | ||
fbcd923c HW |
35 | #ifdef CONFIG_BRIDGE_NETFILTER |
36 | #include "../bridge/br_private.h" | |
37 | #endif | |
38 | ||
7af4cc3f HW |
39 | #define NFQNL_QMAX_DEFAULT 1024 |
40 | ||
7af4cc3f HW |
41 | struct nfqnl_instance { |
42 | struct hlist_node hlist; /* global list of queues */ | |
9872bec7 | 43 | struct rcu_head rcu; |
7af4cc3f HW |
44 | |
45 | int peer_pid; | |
46 | unsigned int queue_maxlen; | |
47 | unsigned int copy_range; | |
48 | unsigned int queue_total; | |
49 | unsigned int queue_dropped; | |
50 | unsigned int queue_user_dropped; | |
51 | ||
e48b9b2f | 52 | unsigned int id_sequence; /* 'sequence' of pkt ids */ |
7af4cc3f HW |
53 | |
54 | u_int16_t queue_num; /* number of this queue */ | |
55 | u_int8_t copy_mode; | |
56 | ||
57 | spinlock_t lock; | |
58 | ||
59 | struct list_head queue_list; /* packets in queue */ | |
60 | }; | |
61 | ||
02f014d8 | 62 | typedef int (*nfqnl_cmpfn)(struct nf_queue_entry *, unsigned long); |
7af4cc3f | 63 | |
9872bec7 | 64 | static DEFINE_SPINLOCK(instances_lock); |
7af4cc3f | 65 | |
7af4cc3f | 66 | #define INSTANCE_BUCKETS 16 |
9d6023ab | 67 | static struct hlist_head instance_table[INSTANCE_BUCKETS] __read_mostly; |
7af4cc3f HW |
68 | |
69 | static inline u_int8_t instance_hashfn(u_int16_t queue_num) | |
70 | { | |
71 | return ((queue_num >> 8) | queue_num) % INSTANCE_BUCKETS; | |
72 | } | |
73 | ||
74 | static struct nfqnl_instance * | |
9872bec7 | 75 | instance_lookup(u_int16_t queue_num) |
7af4cc3f HW |
76 | { |
77 | struct hlist_head *head; | |
78 | struct hlist_node *pos; | |
79 | struct nfqnl_instance *inst; | |
80 | ||
81 | head = &instance_table[instance_hashfn(queue_num)]; | |
9872bec7 | 82 | hlist_for_each_entry_rcu(inst, pos, head, hlist) { |
7af4cc3f HW |
83 | if (inst->queue_num == queue_num) |
84 | return inst; | |
85 | } | |
86 | return NULL; | |
87 | } | |
88 | ||
7af4cc3f HW |
89 | static struct nfqnl_instance * |
90 | instance_create(u_int16_t queue_num, int pid) | |
91 | { | |
baab2ce7 | 92 | struct nfqnl_instance *inst; |
9872bec7 | 93 | unsigned int h; |
baab2ce7 | 94 | int err; |
7af4cc3f | 95 | |
9872bec7 | 96 | spin_lock(&instances_lock); |
baab2ce7 PM |
97 | if (instance_lookup(queue_num)) { |
98 | err = -EEXIST; | |
7af4cc3f | 99 | goto out_unlock; |
baab2ce7 | 100 | } |
7af4cc3f | 101 | |
10dfdc69 | 102 | inst = kzalloc(sizeof(*inst), GFP_ATOMIC); |
baab2ce7 PM |
103 | if (!inst) { |
104 | err = -ENOMEM; | |
7af4cc3f | 105 | goto out_unlock; |
baab2ce7 | 106 | } |
7af4cc3f | 107 | |
7af4cc3f HW |
108 | inst->queue_num = queue_num; |
109 | inst->peer_pid = pid; | |
110 | inst->queue_maxlen = NFQNL_QMAX_DEFAULT; | |
111 | inst->copy_range = 0xfffff; | |
112 | inst->copy_mode = NFQNL_COPY_NONE; | |
181a46a5 | 113 | spin_lock_init(&inst->lock); |
7af4cc3f | 114 | INIT_LIST_HEAD(&inst->queue_list); |
9872bec7 | 115 | INIT_RCU_HEAD(&inst->rcu); |
7af4cc3f | 116 | |
baab2ce7 PM |
117 | if (!try_module_get(THIS_MODULE)) { |
118 | err = -EAGAIN; | |
7af4cc3f | 119 | goto out_free; |
baab2ce7 | 120 | } |
7af4cc3f | 121 | |
9872bec7 PM |
122 | h = instance_hashfn(queue_num); |
123 | hlist_add_head_rcu(&inst->hlist, &instance_table[h]); | |
7af4cc3f | 124 | |
9872bec7 | 125 | spin_unlock(&instances_lock); |
7af4cc3f | 126 | |
7af4cc3f HW |
127 | return inst; |
128 | ||
129 | out_free: | |
130 | kfree(inst); | |
131 | out_unlock: | |
9872bec7 | 132 | spin_unlock(&instances_lock); |
baab2ce7 | 133 | return ERR_PTR(err); |
7af4cc3f HW |
134 | } |
135 | ||
b43d8d85 PM |
136 | static void nfqnl_flush(struct nfqnl_instance *queue, nfqnl_cmpfn cmpfn, |
137 | unsigned long data); | |
7af4cc3f HW |
138 | |
139 | static void | |
9872bec7 | 140 | instance_destroy_rcu(struct rcu_head *head) |
7af4cc3f | 141 | { |
9872bec7 PM |
142 | struct nfqnl_instance *inst = container_of(head, struct nfqnl_instance, |
143 | rcu); | |
7af4cc3f | 144 | |
b43d8d85 | 145 | nfqnl_flush(inst, NULL, 0); |
9872bec7 | 146 | kfree(inst); |
7af4cc3f HW |
147 | module_put(THIS_MODULE); |
148 | } | |
149 | ||
9872bec7 | 150 | static void |
7af4cc3f HW |
151 | __instance_destroy(struct nfqnl_instance *inst) |
152 | { | |
9872bec7 PM |
153 | hlist_del_rcu(&inst->hlist); |
154 | call_rcu(&inst->rcu, instance_destroy_rcu); | |
7af4cc3f HW |
155 | } |
156 | ||
9872bec7 | 157 | static void |
7af4cc3f HW |
158 | instance_destroy(struct nfqnl_instance *inst) |
159 | { | |
9872bec7 PM |
160 | spin_lock(&instances_lock); |
161 | __instance_destroy(inst); | |
162 | spin_unlock(&instances_lock); | |
7af4cc3f HW |
163 | } |
164 | ||
7af4cc3f | 165 | static inline void |
02f014d8 | 166 | __enqueue_entry(struct nfqnl_instance *queue, struct nf_queue_entry *entry) |
7af4cc3f | 167 | { |
0ac41e81 | 168 | list_add_tail(&entry->list, &queue->queue_list); |
7af4cc3f HW |
169 | queue->queue_total++; |
170 | } | |
171 | ||
02f014d8 | 172 | static struct nf_queue_entry * |
b43d8d85 | 173 | find_dequeue_entry(struct nfqnl_instance *queue, unsigned int id) |
7af4cc3f | 174 | { |
02f014d8 | 175 | struct nf_queue_entry *entry = NULL, *i; |
601e68e1 | 176 | |
7af4cc3f | 177 | spin_lock_bh(&queue->lock); |
b43d8d85 PM |
178 | |
179 | list_for_each_entry(i, &queue->queue_list, list) { | |
180 | if (i->id == id) { | |
181 | entry = i; | |
182 | break; | |
183 | } | |
184 | } | |
185 | ||
186 | if (entry) { | |
187 | list_del(&entry->list); | |
188 | queue->queue_total--; | |
189 | } | |
190 | ||
7af4cc3f HW |
191 | spin_unlock_bh(&queue->lock); |
192 | ||
193 | return entry; | |
194 | } | |
195 | ||
196 | static void | |
b43d8d85 | 197 | nfqnl_flush(struct nfqnl_instance *queue, nfqnl_cmpfn cmpfn, unsigned long data) |
7af4cc3f | 198 | { |
02f014d8 | 199 | struct nf_queue_entry *entry, *next; |
b43d8d85 | 200 | |
7af4cc3f | 201 | spin_lock_bh(&queue->lock); |
b43d8d85 PM |
202 | list_for_each_entry_safe(entry, next, &queue->queue_list, list) { |
203 | if (!cmpfn || cmpfn(entry, data)) { | |
204 | list_del(&entry->list); | |
205 | queue->queue_total--; | |
4b3d15ef | 206 | nf_reinject(entry, NF_DROP); |
b43d8d85 PM |
207 | } |
208 | } | |
7af4cc3f HW |
209 | spin_unlock_bh(&queue->lock); |
210 | } | |
211 | ||
212 | static struct sk_buff * | |
213 | nfqnl_build_packet_message(struct nfqnl_instance *queue, | |
0ef0f465 | 214 | struct nf_queue_entry *entry) |
7af4cc3f | 215 | { |
27a884dc | 216 | sk_buff_data_t old_tail; |
7af4cc3f HW |
217 | size_t size; |
218 | size_t data_len = 0; | |
219 | struct sk_buff *skb; | |
220 | struct nfqnl_msg_packet_hdr pmsg; | |
221 | struct nlmsghdr *nlh; | |
222 | struct nfgenmsg *nfmsg; | |
3e4ead4f JJ |
223 | struct sk_buff *entskb = entry->skb; |
224 | struct net_device *indev; | |
225 | struct net_device *outdev; | |
7af4cc3f | 226 | |
cabaa9bf | 227 | size = NLMSG_SPACE(sizeof(struct nfgenmsg)) |
df6fb868 PM |
228 | + nla_total_size(sizeof(struct nfqnl_msg_packet_hdr)) |
229 | + nla_total_size(sizeof(u_int32_t)) /* ifindex */ | |
230 | + nla_total_size(sizeof(u_int32_t)) /* ifindex */ | |
fbcd923c | 231 | #ifdef CONFIG_BRIDGE_NETFILTER |
df6fb868 PM |
232 | + nla_total_size(sizeof(u_int32_t)) /* ifindex */ |
233 | + nla_total_size(sizeof(u_int32_t)) /* ifindex */ | |
fbcd923c | 234 | #endif |
df6fb868 PM |
235 | + nla_total_size(sizeof(u_int32_t)) /* mark */ |
236 | + nla_total_size(sizeof(struct nfqnl_msg_packet_hw)) | |
237 | + nla_total_size(sizeof(struct nfqnl_msg_packet_timestamp)); | |
7af4cc3f | 238 | |
02f014d8 | 239 | outdev = entry->outdev; |
3e4ead4f | 240 | |
7af4cc3f | 241 | spin_lock_bh(&queue->lock); |
601e68e1 | 242 | |
861934c7 | 243 | switch ((enum nfqnl_config_mode)queue->copy_mode) { |
7af4cc3f HW |
244 | case NFQNL_COPY_META: |
245 | case NFQNL_COPY_NONE: | |
7af4cc3f | 246 | break; |
601e68e1 | 247 | |
7af4cc3f | 248 | case NFQNL_COPY_PACKET: |
84fa7933 PM |
249 | if ((entskb->ip_summed == CHECKSUM_PARTIAL || |
250 | entskb->ip_summed == CHECKSUM_COMPLETE) && | |
0ef0f465 | 251 | skb_checksum_help(entskb)) { |
e7dfb09a PM |
252 | spin_unlock_bh(&queue->lock); |
253 | return NULL; | |
254 | } | |
601e68e1 | 255 | if (queue->copy_range == 0 |
3e4ead4f JJ |
256 | || queue->copy_range > entskb->len) |
257 | data_len = entskb->len; | |
7af4cc3f HW |
258 | else |
259 | data_len = queue->copy_range; | |
601e68e1 | 260 | |
df6fb868 | 261 | size += nla_total_size(data_len); |
7af4cc3f | 262 | break; |
7af4cc3f HW |
263 | } |
264 | ||
e48b9b2f PM |
265 | entry->id = queue->id_sequence++; |
266 | ||
7af4cc3f HW |
267 | spin_unlock_bh(&queue->lock); |
268 | ||
269 | skb = alloc_skb(size, GFP_ATOMIC); | |
270 | if (!skb) | |
271 | goto nlmsg_failure; | |
601e68e1 | 272 | |
27a884dc | 273 | old_tail = skb->tail; |
601e68e1 | 274 | nlh = NLMSG_PUT(skb, 0, 0, |
7af4cc3f HW |
275 | NFNL_SUBSYS_QUEUE << 8 | NFQNL_MSG_PACKET, |
276 | sizeof(struct nfgenmsg)); | |
277 | nfmsg = NLMSG_DATA(nlh); | |
02f014d8 | 278 | nfmsg->nfgen_family = entry->pf; |
7af4cc3f HW |
279 | nfmsg->version = NFNETLINK_V0; |
280 | nfmsg->res_id = htons(queue->queue_num); | |
281 | ||
282 | pmsg.packet_id = htonl(entry->id); | |
febf0a43 | 283 | pmsg.hw_protocol = entskb->protocol; |
02f014d8 | 284 | pmsg.hook = entry->hook; |
7af4cc3f | 285 | |
df6fb868 | 286 | NLA_PUT(skb, NFQA_PACKET_HDR, sizeof(pmsg), &pmsg); |
7af4cc3f | 287 | |
02f014d8 | 288 | indev = entry->indev; |
3e4ead4f | 289 | if (indev) { |
fbcd923c | 290 | #ifndef CONFIG_BRIDGE_NETFILTER |
ea3a66ff | 291 | NLA_PUT_BE32(skb, NFQA_IFINDEX_INDEV, htonl(indev->ifindex)); |
fbcd923c | 292 | #else |
02f014d8 | 293 | if (entry->pf == PF_BRIDGE) { |
fbcd923c | 294 | /* Case 1: indev is physical input device, we need to |
601e68e1 | 295 | * look for bridge group (when called from |
fbcd923c | 296 | * netfilter_bridge) */ |
ea3a66ff PM |
297 | NLA_PUT_BE32(skb, NFQA_IFINDEX_PHYSINDEV, |
298 | htonl(indev->ifindex)); | |
fbcd923c | 299 | /* this is the bridge group "brX" */ |
ea3a66ff PM |
300 | NLA_PUT_BE32(skb, NFQA_IFINDEX_INDEV, |
301 | htonl(indev->br_port->br->dev->ifindex)); | |
fbcd923c HW |
302 | } else { |
303 | /* Case 2: indev is bridge group, we need to look for | |
304 | * physical device (when called from ipv4) */ | |
ea3a66ff PM |
305 | NLA_PUT_BE32(skb, NFQA_IFINDEX_INDEV, |
306 | htonl(indev->ifindex)); | |
307 | if (entskb->nf_bridge && entskb->nf_bridge->physindev) | |
308 | NLA_PUT_BE32(skb, NFQA_IFINDEX_PHYSINDEV, | |
309 | htonl(entskb->nf_bridge->physindev->ifindex)); | |
fbcd923c HW |
310 | } |
311 | #endif | |
7af4cc3f HW |
312 | } |
313 | ||
3e4ead4f | 314 | if (outdev) { |
fbcd923c | 315 | #ifndef CONFIG_BRIDGE_NETFILTER |
ea3a66ff | 316 | NLA_PUT_BE32(skb, NFQA_IFINDEX_OUTDEV, htonl(outdev->ifindex)); |
fbcd923c | 317 | #else |
02f014d8 | 318 | if (entry->pf == PF_BRIDGE) { |
fbcd923c | 319 | /* Case 1: outdev is physical output device, we need to |
601e68e1 | 320 | * look for bridge group (when called from |
fbcd923c | 321 | * netfilter_bridge) */ |
ea3a66ff PM |
322 | NLA_PUT_BE32(skb, NFQA_IFINDEX_PHYSOUTDEV, |
323 | htonl(outdev->ifindex)); | |
fbcd923c | 324 | /* this is the bridge group "brX" */ |
ea3a66ff PM |
325 | NLA_PUT_BE32(skb, NFQA_IFINDEX_OUTDEV, |
326 | htonl(outdev->br_port->br->dev->ifindex)); | |
fbcd923c HW |
327 | } else { |
328 | /* Case 2: outdev is bridge group, we need to look for | |
329 | * physical output device (when called from ipv4) */ | |
ea3a66ff PM |
330 | NLA_PUT_BE32(skb, NFQA_IFINDEX_OUTDEV, |
331 | htonl(outdev->ifindex)); | |
332 | if (entskb->nf_bridge && entskb->nf_bridge->physoutdev) | |
333 | NLA_PUT_BE32(skb, NFQA_IFINDEX_PHYSOUTDEV, | |
334 | htonl(entskb->nf_bridge->physoutdev->ifindex)); | |
fbcd923c HW |
335 | } |
336 | #endif | |
7af4cc3f HW |
337 | } |
338 | ||
ea3a66ff PM |
339 | if (entskb->mark) |
340 | NLA_PUT_BE32(skb, NFQA_MARK, htonl(entskb->mark)); | |
7af4cc3f | 341 | |
b95cce35 | 342 | if (indev && entskb->dev) { |
7af4cc3f | 343 | struct nfqnl_msg_packet_hw phw; |
b95cce35 SH |
344 | int len = dev_parse_header(entskb, phw.hw_addr); |
345 | if (len) { | |
346 | phw.hw_addrlen = htons(len); | |
df6fb868 | 347 | NLA_PUT(skb, NFQA_HWADDR, sizeof(phw), &phw); |
b95cce35 | 348 | } |
7af4cc3f HW |
349 | } |
350 | ||
b7aa0bf7 | 351 | if (entskb->tstamp.tv64) { |
7af4cc3f | 352 | struct nfqnl_msg_packet_timestamp ts; |
b7aa0bf7 ED |
353 | struct timeval tv = ktime_to_timeval(entskb->tstamp); |
354 | ts.sec = cpu_to_be64(tv.tv_sec); | |
355 | ts.usec = cpu_to_be64(tv.tv_usec); | |
7af4cc3f | 356 | |
df6fb868 | 357 | NLA_PUT(skb, NFQA_TIMESTAMP, sizeof(ts), &ts); |
7af4cc3f HW |
358 | } |
359 | ||
360 | if (data_len) { | |
df6fb868 | 361 | struct nlattr *nla; |
ca7c48ca | 362 | int sz = nla_attr_size(data_len); |
7af4cc3f | 363 | |
df6fb868 | 364 | if (skb_tailroom(skb) < nla_total_size(data_len)) { |
7af4cc3f HW |
365 | printk(KERN_WARNING "nf_queue: no tailroom!\n"); |
366 | goto nlmsg_failure; | |
367 | } | |
368 | ||
df6fb868 PM |
369 | nla = (struct nlattr *)skb_put(skb, nla_total_size(data_len)); |
370 | nla->nla_type = NFQA_PAYLOAD; | |
ca7c48ca | 371 | nla->nla_len = sz; |
7af4cc3f | 372 | |
df6fb868 | 373 | if (skb_copy_bits(entskb, 0, nla_data(nla), data_len)) |
7af4cc3f HW |
374 | BUG(); |
375 | } | |
601e68e1 | 376 | |
7af4cc3f HW |
377 | nlh->nlmsg_len = skb->tail - old_tail; |
378 | return skb; | |
379 | ||
380 | nlmsg_failure: | |
df6fb868 | 381 | nla_put_failure: |
7af4cc3f HW |
382 | if (skb) |
383 | kfree_skb(skb); | |
7af4cc3f HW |
384 | if (net_ratelimit()) |
385 | printk(KERN_ERR "nf_queue: error creating packet message\n"); | |
386 | return NULL; | |
387 | } | |
388 | ||
389 | static int | |
02f014d8 | 390 | nfqnl_enqueue_packet(struct nf_queue_entry *entry, unsigned int queuenum) |
7af4cc3f | 391 | { |
7af4cc3f HW |
392 | struct sk_buff *nskb; |
393 | struct nfqnl_instance *queue; | |
0ef0f465 | 394 | int err; |
7af4cc3f | 395 | |
9872bec7 PM |
396 | /* rcu_read_lock()ed by nf_hook_slow() */ |
397 | queue = instance_lookup(queuenum); | |
2bd01197 | 398 | if (!queue) |
0ef0f465 | 399 | goto err_out; |
7af4cc3f | 400 | |
2bd01197 | 401 | if (queue->copy_mode == NFQNL_COPY_NONE) |
0ef0f465 | 402 | goto err_out; |
7af4cc3f | 403 | |
0ef0f465 | 404 | nskb = nfqnl_build_packet_message(queue, entry); |
7af4cc3f | 405 | if (nskb == NULL) |
0ef0f465 | 406 | goto err_out; |
601e68e1 | 407 | |
7af4cc3f | 408 | spin_lock_bh(&queue->lock); |
601e68e1 | 409 | |
7af4cc3f | 410 | if (!queue->peer_pid) |
601e68e1 | 411 | goto err_out_free_nskb; |
7af4cc3f HW |
412 | |
413 | if (queue->queue_total >= queue->queue_maxlen) { | |
601e68e1 | 414 | queue->queue_dropped++; |
7af4cc3f | 415 | if (net_ratelimit()) |
601e68e1 YH |
416 | printk(KERN_WARNING "nf_queue: full at %d entries, " |
417 | "dropping packets(s). Dropped: %d\n", | |
7af4cc3f HW |
418 | queue->queue_total, queue->queue_dropped); |
419 | goto err_out_free_nskb; | |
420 | } | |
421 | ||
422 | /* nfnetlink_unicast will either free the nskb or add it to a socket */ | |
0ef0f465 PM |
423 | err = nfnetlink_unicast(nskb, queue->peer_pid, MSG_DONTWAIT); |
424 | if (err < 0) { | |
601e68e1 | 425 | queue->queue_user_dropped++; |
7af4cc3f HW |
426 | goto err_out_unlock; |
427 | } | |
428 | ||
429 | __enqueue_entry(queue, entry); | |
430 | ||
431 | spin_unlock_bh(&queue->lock); | |
0ef0f465 | 432 | return 0; |
7af4cc3f HW |
433 | |
434 | err_out_free_nskb: | |
601e68e1 | 435 | kfree_skb(nskb); |
7af4cc3f HW |
436 | err_out_unlock: |
437 | spin_unlock_bh(&queue->lock); | |
0ef0f465 PM |
438 | err_out: |
439 | return -1; | |
7af4cc3f HW |
440 | } |
441 | ||
442 | static int | |
02f014d8 | 443 | nfqnl_mangle(void *data, int data_len, struct nf_queue_entry *e) |
7af4cc3f | 444 | { |
e2b58a67 | 445 | struct sk_buff *nskb; |
7af4cc3f HW |
446 | int diff; |
447 | ||
448 | diff = data_len - e->skb->len; | |
d8a585d7 PM |
449 | if (diff < 0) { |
450 | if (pskb_trim(e->skb, data_len)) | |
451 | return -ENOMEM; | |
452 | } else if (diff > 0) { | |
7af4cc3f HW |
453 | if (data_len > 0xFFFF) |
454 | return -EINVAL; | |
455 | if (diff > skb_tailroom(e->skb)) { | |
9a732ed6 AE |
456 | nskb = skb_copy_expand(e->skb, skb_headroom(e->skb), |
457 | diff, GFP_ATOMIC); | |
e2b58a67 | 458 | if (!nskb) { |
1158ba27 | 459 | printk(KERN_WARNING "nf_queue: OOM " |
7af4cc3f | 460 | "in mangle, dropping packet\n"); |
e2b58a67 | 461 | return -ENOMEM; |
7af4cc3f | 462 | } |
e2b58a67 PM |
463 | kfree_skb(e->skb); |
464 | e->skb = nskb; | |
7af4cc3f HW |
465 | } |
466 | skb_put(e->skb, diff); | |
467 | } | |
37d41879 | 468 | if (!skb_make_writable(e->skb, data_len)) |
7af4cc3f | 469 | return -ENOMEM; |
27d7ff46 | 470 | skb_copy_to_linear_data(e->skb, data, data_len); |
e7dfb09a | 471 | e->skb->ip_summed = CHECKSUM_NONE; |
7af4cc3f HW |
472 | return 0; |
473 | } | |
474 | ||
7af4cc3f HW |
475 | static int |
476 | nfqnl_set_mode(struct nfqnl_instance *queue, | |
477 | unsigned char mode, unsigned int range) | |
478 | { | |
c5de0dfd | 479 | int status = 0; |
7af4cc3f HW |
480 | |
481 | spin_lock_bh(&queue->lock); | |
c5de0dfd PM |
482 | switch (mode) { |
483 | case NFQNL_COPY_NONE: | |
484 | case NFQNL_COPY_META: | |
485 | queue->copy_mode = mode; | |
486 | queue->copy_range = 0; | |
487 | break; | |
488 | ||
489 | case NFQNL_COPY_PACKET: | |
490 | queue->copy_mode = mode; | |
491 | /* we're using struct nlattr which has 16bit nla_len */ | |
492 | if (range > 0xffff) | |
493 | queue->copy_range = 0xffff; | |
494 | else | |
495 | queue->copy_range = range; | |
496 | break; | |
497 | ||
498 | default: | |
499 | status = -EINVAL; | |
500 | ||
501 | } | |
7af4cc3f HW |
502 | spin_unlock_bh(&queue->lock); |
503 | ||
504 | return status; | |
505 | } | |
506 | ||
507 | static int | |
02f014d8 | 508 | dev_cmp(struct nf_queue_entry *entry, unsigned long ifindex) |
7af4cc3f | 509 | { |
02f014d8 PM |
510 | if (entry->indev) |
511 | if (entry->indev->ifindex == ifindex) | |
7af4cc3f | 512 | return 1; |
02f014d8 PM |
513 | if (entry->outdev) |
514 | if (entry->outdev->ifindex == ifindex) | |
7af4cc3f | 515 | return 1; |
ef47c6a7 PM |
516 | #ifdef CONFIG_BRIDGE_NETFILTER |
517 | if (entry->skb->nf_bridge) { | |
518 | if (entry->skb->nf_bridge->physindev && | |
519 | entry->skb->nf_bridge->physindev->ifindex == ifindex) | |
520 | return 1; | |
521 | if (entry->skb->nf_bridge->physoutdev && | |
522 | entry->skb->nf_bridge->physoutdev->ifindex == ifindex) | |
523 | return 1; | |
524 | } | |
525 | #endif | |
7af4cc3f HW |
526 | return 0; |
527 | } | |
528 | ||
529 | /* drop all packets with either indev or outdev == ifindex from all queue | |
530 | * instances */ | |
531 | static void | |
532 | nfqnl_dev_drop(int ifindex) | |
533 | { | |
534 | int i; | |
601e68e1 | 535 | |
9872bec7 | 536 | rcu_read_lock(); |
7af4cc3f | 537 | |
9872bec7 | 538 | for (i = 0; i < INSTANCE_BUCKETS; i++) { |
7af4cc3f HW |
539 | struct hlist_node *tmp; |
540 | struct nfqnl_instance *inst; | |
541 | struct hlist_head *head = &instance_table[i]; | |
542 | ||
9872bec7 | 543 | hlist_for_each_entry_rcu(inst, tmp, head, hlist) |
b43d8d85 | 544 | nfqnl_flush(inst, dev_cmp, ifindex); |
7af4cc3f HW |
545 | } |
546 | ||
9872bec7 | 547 | rcu_read_unlock(); |
7af4cc3f HW |
548 | } |
549 | ||
550 | #define RCV_SKB_FAIL(err) do { netlink_ack(skb, nlh, (err)); return; } while (0) | |
551 | ||
552 | static int | |
553 | nfqnl_rcv_dev_event(struct notifier_block *this, | |
554 | unsigned long event, void *ptr) | |
555 | { | |
556 | struct net_device *dev = ptr; | |
557 | ||
721499e8 | 558 | if (!net_eq(dev_net(dev), &init_net)) |
e9dc8653 EB |
559 | return NOTIFY_DONE; |
560 | ||
7af4cc3f HW |
561 | /* Drop any packets associated with the downed device */ |
562 | if (event == NETDEV_DOWN) | |
563 | nfqnl_dev_drop(dev->ifindex); | |
564 | return NOTIFY_DONE; | |
565 | } | |
566 | ||
567 | static struct notifier_block nfqnl_dev_notifier = { | |
568 | .notifier_call = nfqnl_rcv_dev_event, | |
569 | }; | |
570 | ||
571 | static int | |
572 | nfqnl_rcv_nl_event(struct notifier_block *this, | |
573 | unsigned long event, void *ptr) | |
574 | { | |
575 | struct netlink_notify *n = ptr; | |
576 | ||
577 | if (event == NETLINK_URELEASE && | |
578 | n->protocol == NETLINK_NETFILTER && n->pid) { | |
579 | int i; | |
580 | ||
581 | /* destroy all instances for this pid */ | |
9872bec7 PM |
582 | spin_lock(&instances_lock); |
583 | for (i = 0; i < INSTANCE_BUCKETS; i++) { | |
7af4cc3f HW |
584 | struct hlist_node *tmp, *t2; |
585 | struct nfqnl_instance *inst; | |
586 | struct hlist_head *head = &instance_table[i]; | |
587 | ||
588 | hlist_for_each_entry_safe(inst, tmp, t2, head, hlist) { | |
b4b51029 EB |
589 | if ((n->net == &init_net) && |
590 | (n->pid == inst->peer_pid)) | |
7af4cc3f HW |
591 | __instance_destroy(inst); |
592 | } | |
593 | } | |
9872bec7 | 594 | spin_unlock(&instances_lock); |
7af4cc3f HW |
595 | } |
596 | return NOTIFY_DONE; | |
597 | } | |
598 | ||
599 | static struct notifier_block nfqnl_rtnl_notifier = { | |
600 | .notifier_call = nfqnl_rcv_nl_event, | |
601 | }; | |
602 | ||
5bf75853 PM |
603 | static const struct nla_policy nfqa_verdict_policy[NFQA_MAX+1] = { |
604 | [NFQA_VERDICT_HDR] = { .len = sizeof(struct nfqnl_msg_verdict_hdr) }, | |
605 | [NFQA_MARK] = { .type = NLA_U32 }, | |
606 | [NFQA_PAYLOAD] = { .type = NLA_UNSPEC }, | |
838ab636 HW |
607 | }; |
608 | ||
7af4cc3f HW |
609 | static int |
610 | nfqnl_recv_verdict(struct sock *ctnl, struct sk_buff *skb, | |
df6fb868 | 611 | struct nlmsghdr *nlh, struct nlattr *nfqa[]) |
7af4cc3f HW |
612 | { |
613 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
614 | u_int16_t queue_num = ntohs(nfmsg->res_id); | |
615 | ||
616 | struct nfqnl_msg_verdict_hdr *vhdr; | |
617 | struct nfqnl_instance *queue; | |
618 | unsigned int verdict; | |
02f014d8 | 619 | struct nf_queue_entry *entry; |
838ab636 | 620 | int err; |
7af4cc3f | 621 | |
9872bec7 PM |
622 | rcu_read_lock(); |
623 | queue = instance_lookup(queue_num); | |
624 | if (!queue) { | |
625 | err = -ENODEV; | |
626 | goto err_out_unlock; | |
627 | } | |
7af4cc3f | 628 | |
838ab636 HW |
629 | if (queue->peer_pid != NETLINK_CB(skb).pid) { |
630 | err = -EPERM; | |
9872bec7 | 631 | goto err_out_unlock; |
838ab636 | 632 | } |
7af4cc3f | 633 | |
df6fb868 | 634 | if (!nfqa[NFQA_VERDICT_HDR]) { |
838ab636 | 635 | err = -EINVAL; |
9872bec7 | 636 | goto err_out_unlock; |
838ab636 | 637 | } |
7af4cc3f | 638 | |
df6fb868 | 639 | vhdr = nla_data(nfqa[NFQA_VERDICT_HDR]); |
7af4cc3f HW |
640 | verdict = ntohl(vhdr->verdict); |
641 | ||
838ab636 HW |
642 | if ((verdict & NF_VERDICT_MASK) > NF_MAX_VERDICT) { |
643 | err = -EINVAL; | |
9872bec7 | 644 | goto err_out_unlock; |
838ab636 | 645 | } |
7af4cc3f | 646 | |
b43d8d85 | 647 | entry = find_dequeue_entry(queue, ntohl(vhdr->id)); |
838ab636 HW |
648 | if (entry == NULL) { |
649 | err = -ENOENT; | |
9872bec7 | 650 | goto err_out_unlock; |
838ab636 | 651 | } |
9872bec7 | 652 | rcu_read_unlock(); |
7af4cc3f | 653 | |
df6fb868 PM |
654 | if (nfqa[NFQA_PAYLOAD]) { |
655 | if (nfqnl_mangle(nla_data(nfqa[NFQA_PAYLOAD]), | |
656 | nla_len(nfqa[NFQA_PAYLOAD]), entry) < 0) | |
7af4cc3f HW |
657 | verdict = NF_DROP; |
658 | } | |
659 | ||
df6fb868 | 660 | if (nfqa[NFQA_MARK]) |
ea3a66ff | 661 | entry->skb->mark = ntohl(nla_get_be32(nfqa[NFQA_MARK])); |
601e68e1 | 662 | |
4b3d15ef | 663 | nf_reinject(entry, verdict); |
7af4cc3f | 664 | return 0; |
838ab636 | 665 | |
9872bec7 PM |
666 | err_out_unlock: |
667 | rcu_read_unlock(); | |
838ab636 | 668 | return err; |
7af4cc3f HW |
669 | } |
670 | ||
671 | static int | |
672 | nfqnl_recv_unsupp(struct sock *ctnl, struct sk_buff *skb, | |
df6fb868 | 673 | struct nlmsghdr *nlh, struct nlattr *nfqa[]) |
7af4cc3f HW |
674 | { |
675 | return -ENOTSUPP; | |
676 | } | |
677 | ||
5bf75853 PM |
678 | static const struct nla_policy nfqa_cfg_policy[NFQA_CFG_MAX+1] = { |
679 | [NFQA_CFG_CMD] = { .len = sizeof(struct nfqnl_msg_config_cmd) }, | |
680 | [NFQA_CFG_PARAMS] = { .len = sizeof(struct nfqnl_msg_config_params) }, | |
838ab636 HW |
681 | }; |
682 | ||
e3ac5298 | 683 | static const struct nf_queue_handler nfqh = { |
bbd86b9f HW |
684 | .name = "nf_queue", |
685 | .outfn = &nfqnl_enqueue_packet, | |
686 | }; | |
687 | ||
7af4cc3f HW |
688 | static int |
689 | nfqnl_recv_config(struct sock *ctnl, struct sk_buff *skb, | |
df6fb868 | 690 | struct nlmsghdr *nlh, struct nlattr *nfqa[]) |
7af4cc3f HW |
691 | { |
692 | struct nfgenmsg *nfmsg = NLMSG_DATA(nlh); | |
693 | u_int16_t queue_num = ntohs(nfmsg->res_id); | |
694 | struct nfqnl_instance *queue; | |
9872bec7 | 695 | struct nfqnl_msg_config_cmd *cmd = NULL; |
838ab636 | 696 | int ret = 0; |
7af4cc3f | 697 | |
9872bec7 PM |
698 | if (nfqa[NFQA_CFG_CMD]) { |
699 | cmd = nla_data(nfqa[NFQA_CFG_CMD]); | |
700 | ||
701 | /* Commands without queue context - might sleep */ | |
702 | switch (cmd->command) { | |
703 | case NFQNL_CFG_CMD_PF_BIND: | |
914afea8 PM |
704 | return nf_register_queue_handler(ntohs(cmd->pf), |
705 | &nfqh); | |
9872bec7 | 706 | case NFQNL_CFG_CMD_PF_UNBIND: |
914afea8 PM |
707 | return nf_unregister_queue_handler(ntohs(cmd->pf), |
708 | &nfqh); | |
9872bec7 | 709 | } |
9872bec7 PM |
710 | } |
711 | ||
712 | rcu_read_lock(); | |
713 | queue = instance_lookup(queue_num); | |
a3c8e7fd PM |
714 | if (queue && queue->peer_pid != NETLINK_CB(skb).pid) { |
715 | ret = -EPERM; | |
9872bec7 | 716 | goto err_out_unlock; |
a3c8e7fd PM |
717 | } |
718 | ||
9872bec7 | 719 | if (cmd != NULL) { |
7af4cc3f HW |
720 | switch (cmd->command) { |
721 | case NFQNL_CFG_CMD_BIND: | |
9872bec7 PM |
722 | if (queue) { |
723 | ret = -EBUSY; | |
724 | goto err_out_unlock; | |
725 | } | |
7af4cc3f | 726 | queue = instance_create(queue_num, NETLINK_CB(skb).pid); |
baab2ce7 PM |
727 | if (IS_ERR(queue)) { |
728 | ret = PTR_ERR(queue); | |
9872bec7 PM |
729 | goto err_out_unlock; |
730 | } | |
7af4cc3f HW |
731 | break; |
732 | case NFQNL_CFG_CMD_UNBIND: | |
9872bec7 PM |
733 | if (!queue) { |
734 | ret = -ENODEV; | |
735 | goto err_out_unlock; | |
736 | } | |
7af4cc3f HW |
737 | instance_destroy(queue); |
738 | break; | |
739 | case NFQNL_CFG_CMD_PF_BIND: | |
7af4cc3f | 740 | case NFQNL_CFG_CMD_PF_UNBIND: |
7af4cc3f HW |
741 | break; |
742 | default: | |
cd21f0ac | 743 | ret = -ENOTSUPP; |
838ab636 | 744 | break; |
7af4cc3f | 745 | } |
7af4cc3f HW |
746 | } |
747 | ||
df6fb868 | 748 | if (nfqa[NFQA_CFG_PARAMS]) { |
7af4cc3f | 749 | struct nfqnl_msg_config_params *params; |
7af4cc3f | 750 | |
406dbfc9 | 751 | if (!queue) { |
a3c8e7fd | 752 | ret = -ENODEV; |
9872bec7 | 753 | goto err_out_unlock; |
406dbfc9 | 754 | } |
df6fb868 | 755 | params = nla_data(nfqa[NFQA_CFG_PARAMS]); |
7af4cc3f HW |
756 | nfqnl_set_mode(queue, params->copy_mode, |
757 | ntohl(params->copy_range)); | |
758 | } | |
759 | ||
df6fb868 | 760 | if (nfqa[NFQA_CFG_QUEUE_MAXLEN]) { |
829e17a1 | 761 | __be32 *queue_maxlen; |
a3c8e7fd PM |
762 | |
763 | if (!queue) { | |
764 | ret = -ENODEV; | |
9872bec7 | 765 | goto err_out_unlock; |
a3c8e7fd | 766 | } |
df6fb868 | 767 | queue_maxlen = nla_data(nfqa[NFQA_CFG_QUEUE_MAXLEN]); |
829e17a1 EL |
768 | spin_lock_bh(&queue->lock); |
769 | queue->queue_maxlen = ntohl(*queue_maxlen); | |
770 | spin_unlock_bh(&queue->lock); | |
771 | } | |
772 | ||
9872bec7 PM |
773 | err_out_unlock: |
774 | rcu_read_unlock(); | |
838ab636 | 775 | return ret; |
7af4cc3f HW |
776 | } |
777 | ||
7c8d4cb4 | 778 | static const struct nfnl_callback nfqnl_cb[NFQNL_MSG_MAX] = { |
7af4cc3f | 779 | [NFQNL_MSG_PACKET] = { .call = nfqnl_recv_unsupp, |
37d2e7a2 | 780 | .attr_count = NFQA_MAX, }, |
7af4cc3f | 781 | [NFQNL_MSG_VERDICT] = { .call = nfqnl_recv_verdict, |
5bf75853 PM |
782 | .attr_count = NFQA_MAX, |
783 | .policy = nfqa_verdict_policy }, | |
7af4cc3f | 784 | [NFQNL_MSG_CONFIG] = { .call = nfqnl_recv_config, |
5bf75853 PM |
785 | .attr_count = NFQA_CFG_MAX, |
786 | .policy = nfqa_cfg_policy }, | |
7af4cc3f HW |
787 | }; |
788 | ||
7c8d4cb4 | 789 | static const struct nfnetlink_subsystem nfqnl_subsys = { |
7af4cc3f HW |
790 | .name = "nf_queue", |
791 | .subsys_id = NFNL_SUBSYS_QUEUE, | |
792 | .cb_count = NFQNL_MSG_MAX, | |
7af4cc3f HW |
793 | .cb = nfqnl_cb, |
794 | }; | |
795 | ||
838ab636 HW |
796 | #ifdef CONFIG_PROC_FS |
797 | struct iter_state { | |
798 | unsigned int bucket; | |
799 | }; | |
800 | ||
801 | static struct hlist_node *get_first(struct seq_file *seq) | |
802 | { | |
803 | struct iter_state *st = seq->private; | |
804 | ||
805 | if (!st) | |
806 | return NULL; | |
807 | ||
808 | for (st->bucket = 0; st->bucket < INSTANCE_BUCKETS; st->bucket++) { | |
809 | if (!hlist_empty(&instance_table[st->bucket])) | |
810 | return instance_table[st->bucket].first; | |
811 | } | |
812 | return NULL; | |
813 | } | |
814 | ||
815 | static struct hlist_node *get_next(struct seq_file *seq, struct hlist_node *h) | |
816 | { | |
817 | struct iter_state *st = seq->private; | |
818 | ||
819 | h = h->next; | |
820 | while (!h) { | |
821 | if (++st->bucket >= INSTANCE_BUCKETS) | |
822 | return NULL; | |
823 | ||
824 | h = instance_table[st->bucket].first; | |
825 | } | |
826 | return h; | |
827 | } | |
828 | ||
829 | static struct hlist_node *get_idx(struct seq_file *seq, loff_t pos) | |
830 | { | |
831 | struct hlist_node *head; | |
832 | head = get_first(seq); | |
833 | ||
834 | if (head) | |
835 | while (pos && (head = get_next(seq, head))) | |
836 | pos--; | |
837 | return pos ? NULL : head; | |
838 | } | |
839 | ||
840 | static void *seq_start(struct seq_file *seq, loff_t *pos) | |
ca7c48ca | 841 | __acquires(instances_lock) |
838ab636 | 842 | { |
9872bec7 | 843 | spin_lock(&instances_lock); |
838ab636 HW |
844 | return get_idx(seq, *pos); |
845 | } | |
846 | ||
847 | static void *seq_next(struct seq_file *s, void *v, loff_t *pos) | |
848 | { | |
849 | (*pos)++; | |
850 | return get_next(s, v); | |
851 | } | |
852 | ||
853 | static void seq_stop(struct seq_file *s, void *v) | |
ca7c48ca | 854 | __releases(instances_lock) |
838ab636 | 855 | { |
9872bec7 | 856 | spin_unlock(&instances_lock); |
838ab636 HW |
857 | } |
858 | ||
859 | static int seq_show(struct seq_file *s, void *v) | |
860 | { | |
861 | const struct nfqnl_instance *inst = v; | |
862 | ||
863 | return seq_printf(s, "%5d %6d %5d %1d %5d %5d %5d %8d %2d\n", | |
864 | inst->queue_num, | |
865 | inst->peer_pid, inst->queue_total, | |
866 | inst->copy_mode, inst->copy_range, | |
867 | inst->queue_dropped, inst->queue_user_dropped, | |
9872bec7 | 868 | inst->id_sequence, 1); |
838ab636 HW |
869 | } |
870 | ||
56b3d975 | 871 | static const struct seq_operations nfqnl_seq_ops = { |
838ab636 HW |
872 | .start = seq_start, |
873 | .next = seq_next, | |
874 | .stop = seq_stop, | |
875 | .show = seq_show, | |
876 | }; | |
877 | ||
878 | static int nfqnl_open(struct inode *inode, struct file *file) | |
879 | { | |
e2da5913 PE |
880 | return seq_open_private(file, &nfqnl_seq_ops, |
881 | sizeof(struct iter_state)); | |
838ab636 HW |
882 | } |
883 | ||
da7071d7 | 884 | static const struct file_operations nfqnl_file_ops = { |
838ab636 HW |
885 | .owner = THIS_MODULE, |
886 | .open = nfqnl_open, | |
887 | .read = seq_read, | |
888 | .llseek = seq_lseek, | |
889 | .release = seq_release_private, | |
890 | }; | |
891 | ||
892 | #endif /* PROC_FS */ | |
893 | ||
32292a7f | 894 | static int __init nfnetlink_queue_init(void) |
7af4cc3f | 895 | { |
838ab636 | 896 | int i, status = -ENOMEM; |
601e68e1 | 897 | |
838ab636 HW |
898 | for (i = 0; i < INSTANCE_BUCKETS; i++) |
899 | INIT_HLIST_HEAD(&instance_table[i]); | |
900 | ||
7af4cc3f HW |
901 | netlink_register_notifier(&nfqnl_rtnl_notifier); |
902 | status = nfnetlink_subsys_register(&nfqnl_subsys); | |
903 | if (status < 0) { | |
904 | printk(KERN_ERR "nf_queue: failed to create netlink socket\n"); | |
905 | goto cleanup_netlink_notifier; | |
906 | } | |
907 | ||
838ab636 | 908 | #ifdef CONFIG_PROC_FS |
8eeee8b1 DL |
909 | if (!proc_create("nfnetlink_queue", 0440, |
910 | proc_net_netfilter, &nfqnl_file_ops)) | |
838ab636 | 911 | goto cleanup_subsys; |
838ab636 HW |
912 | #endif |
913 | ||
7af4cc3f HW |
914 | register_netdevice_notifier(&nfqnl_dev_notifier); |
915 | return status; | |
916 | ||
838ab636 HW |
917 | #ifdef CONFIG_PROC_FS |
918 | cleanup_subsys: | |
7af4cc3f | 919 | nfnetlink_subsys_unregister(&nfqnl_subsys); |
32292a7f | 920 | #endif |
7af4cc3f HW |
921 | cleanup_netlink_notifier: |
922 | netlink_unregister_notifier(&nfqnl_rtnl_notifier); | |
923 | return status; | |
924 | } | |
925 | ||
65b4b4e8 | 926 | static void __exit nfnetlink_queue_fini(void) |
7af4cc3f | 927 | { |
32292a7f PM |
928 | nf_unregister_queue_handlers(&nfqh); |
929 | unregister_netdevice_notifier(&nfqnl_dev_notifier); | |
930 | #ifdef CONFIG_PROC_FS | |
931 | remove_proc_entry("nfnetlink_queue", proc_net_netfilter); | |
932 | #endif | |
933 | nfnetlink_subsys_unregister(&nfqnl_subsys); | |
934 | netlink_unregister_notifier(&nfqnl_rtnl_notifier); | |
7af4cc3f HW |
935 | } |
936 | ||
937 | MODULE_DESCRIPTION("netfilter packet queue handler"); | |
938 | MODULE_AUTHOR("Harald Welte <laforge@netfilter.org>"); | |
939 | MODULE_LICENSE("GPL"); | |
940 | MODULE_ALIAS_NFNL_SUBSYS(NFNL_SUBSYS_QUEUE); | |
941 | ||
65b4b4e8 AM |
942 | module_init(nfnetlink_queue_init); |
943 | module_exit(nfnetlink_queue_fini); |