Commit | Line | Data |
---|---|---|
1da177e4 LT |
1 | /* |
2 | * This is a module which is used for queueing IPv4 packets and | |
3 | * communicating with userspace via netlink. | |
4 | * | |
5 | * (C) 2000-2002 James Morris <jmorris@intercode.com.au> | |
9bb7bc94 | 6 | * (C) 2003-2005 Netfilter Core Team <coreteam@netfilter.org> |
1da177e4 LT |
7 | * |
8 | * This program is free software; you can redistribute it and/or modify | |
9 | * it under the terms of the GNU General Public License version 2 as | |
10 | * published by the Free Software Foundation. | |
1da177e4 LT |
11 | */ |
12 | #include <linux/module.h> | |
13 | #include <linux/skbuff.h> | |
14 | #include <linux/init.h> | |
15 | #include <linux/ip.h> | |
16 | #include <linux/notifier.h> | |
17 | #include <linux/netdevice.h> | |
18 | #include <linux/netfilter.h> | |
19 | #include <linux/netfilter_ipv4/ip_queue.h> | |
20 | #include <linux/netfilter_ipv4/ip_tables.h> | |
21 | #include <linux/netlink.h> | |
22 | #include <linux/spinlock.h> | |
23 | #include <linux/sysctl.h> | |
24 | #include <linux/proc_fs.h> | |
7351a22a | 25 | #include <linux/seq_file.h> |
1da177e4 | 26 | #include <linux/security.h> |
95ba434f | 27 | #include <linux/net.h> |
4a3e2f71 | 28 | #include <linux/mutex.h> |
5a0e3ad6 | 29 | #include <linux/slab.h> |
457c4cbc | 30 | #include <net/net_namespace.h> |
1da177e4 LT |
31 | #include <net/sock.h> |
32 | #include <net/route.h> | |
c01cd429 | 33 | #include <net/netfilter/nf_queue.h> |
3d7cc2ba | 34 | #include <net/ip.h> |
1da177e4 LT |
35 | |
36 | #define IPQ_QMAX_DEFAULT 1024 | |
37 | #define IPQ_PROC_FS_NAME "ip_queue" | |
38 | #define NET_IPQ_QMAX 2088 | |
39 | #define NET_IPQ_QMAX_NAME "ip_queue_maxlen" | |
40 | ||
02f014d8 | 41 | typedef int (*ipq_cmpfn)(struct nf_queue_entry *, unsigned long); |
1da177e4 | 42 | |
1192e403 | 43 | static unsigned char copy_mode __read_mostly = IPQ_COPY_NONE; |
94aec08e | 44 | static unsigned int queue_maxlen __read_mostly = IPQ_QMAX_DEFAULT; |
5756d346 | 45 | static DEFINE_SPINLOCK(queue_lock); |
1192e403 BH |
46 | static int peer_pid __read_mostly; |
47 | static unsigned int copy_range __read_mostly; | |
1da177e4 LT |
48 | static unsigned int queue_total; |
49 | static unsigned int queue_dropped = 0; | |
50 | static unsigned int queue_user_dropped = 0; | |
1192e403 | 51 | static struct sock *ipqnl __read_mostly; |
1da177e4 | 52 | static LIST_HEAD(queue_list); |
4a3e2f71 | 53 | static DEFINE_MUTEX(ipqnl_mutex); |
1da177e4 | 54 | |
1da177e4 | 55 | static inline void |
02f014d8 | 56 | __ipq_enqueue_entry(struct nf_queue_entry *entry) |
1da177e4 | 57 | { |
0ac41e81 | 58 | list_add_tail(&entry->list, &queue_list); |
1da177e4 LT |
59 | queue_total++; |
60 | } | |
61 | ||
1da177e4 LT |
62 | static inline int |
63 | __ipq_set_mode(unsigned char mode, unsigned int range) | |
64 | { | |
65 | int status = 0; | |
e905a9ed | 66 | |
1da177e4 LT |
67 | switch(mode) { |
68 | case IPQ_COPY_NONE: | |
69 | case IPQ_COPY_META: | |
70 | copy_mode = mode; | |
71 | copy_range = 0; | |
72 | break; | |
e905a9ed | 73 | |
1da177e4 | 74 | case IPQ_COPY_PACKET: |
5756d346 ED |
75 | if (range > 0xFFFF) |
76 | range = 0xFFFF; | |
1da177e4 | 77 | copy_range = range; |
5756d346 | 78 | copy_mode = mode; |
1da177e4 | 79 | break; |
e905a9ed | 80 | |
1da177e4 LT |
81 | default: |
82 | status = -EINVAL; | |
83 | ||
84 | } | |
85 | return status; | |
86 | } | |
87 | ||
95214092 PM |
88 | static void __ipq_flush(ipq_cmpfn cmpfn, unsigned long data); |
89 | ||
1da177e4 LT |
90 | static inline void |
91 | __ipq_reset(void) | |
92 | { | |
93 | peer_pid = 0; | |
94 | net_disable_timestamp(); | |
95 | __ipq_set_mode(IPQ_COPY_NONE, 0); | |
95214092 | 96 | __ipq_flush(NULL, 0); |
1da177e4 LT |
97 | } |
98 | ||
02f014d8 | 99 | static struct nf_queue_entry * |
95214092 | 100 | ipq_find_dequeue_entry(unsigned long id) |
1da177e4 | 101 | { |
02f014d8 | 102 | struct nf_queue_entry *entry = NULL, *i; |
e905a9ed | 103 | |
5756d346 | 104 | spin_lock_bh(&queue_lock); |
95214092 PM |
105 | |
106 | list_for_each_entry(i, &queue_list, list) { | |
107 | if ((unsigned long)i == id) { | |
108 | entry = i; | |
109 | break; | |
110 | } | |
111 | } | |
112 | ||
113 | if (entry) { | |
114 | list_del(&entry->list); | |
115 | queue_total--; | |
116 | } | |
117 | ||
5756d346 | 118 | spin_unlock_bh(&queue_lock); |
1da177e4 LT |
119 | return entry; |
120 | } | |
121 | ||
122 | static void | |
95214092 PM |
123 | __ipq_flush(ipq_cmpfn cmpfn, unsigned long data) |
124 | { | |
02f014d8 | 125 | struct nf_queue_entry *entry, *next; |
95214092 PM |
126 | |
127 | list_for_each_entry_safe(entry, next, &queue_list, list) { | |
128 | if (!cmpfn || cmpfn(entry, data)) { | |
129 | list_del(&entry->list); | |
130 | queue_total--; | |
4b3d15ef | 131 | nf_reinject(entry, NF_DROP); |
95214092 PM |
132 | } |
133 | } | |
134 | } | |
135 | ||
136 | static void | |
137 | ipq_flush(ipq_cmpfn cmpfn, unsigned long data) | |
1da177e4 | 138 | { |
5756d346 | 139 | spin_lock_bh(&queue_lock); |
95214092 | 140 | __ipq_flush(cmpfn, data); |
5756d346 | 141 | spin_unlock_bh(&queue_lock); |
1da177e4 LT |
142 | } |
143 | ||
144 | static struct sk_buff * | |
02f014d8 | 145 | ipq_build_packet_message(struct nf_queue_entry *entry, int *errp) |
1da177e4 | 146 | { |
27a884dc | 147 | sk_buff_data_t old_tail; |
1da177e4 LT |
148 | size_t size = 0; |
149 | size_t data_len = 0; | |
150 | struct sk_buff *skb; | |
151 | struct ipq_packet_msg *pmsg; | |
152 | struct nlmsghdr *nlh; | |
b7aa0bf7 | 153 | struct timeval tv; |
1da177e4 | 154 | |
5756d346 | 155 | switch (ACCESS_ONCE(copy_mode)) { |
1da177e4 LT |
156 | case IPQ_COPY_META: |
157 | case IPQ_COPY_NONE: | |
158 | size = NLMSG_SPACE(sizeof(*pmsg)); | |
1da177e4 | 159 | break; |
e905a9ed | 160 | |
1da177e4 | 161 | case IPQ_COPY_PACKET: |
9e56c214 | 162 | if (entry->skb->ip_summed == CHECKSUM_PARTIAL && |
5756d346 | 163 | (*errp = skb_checksum_help(entry->skb))) |
66a79a19 | 164 | return NULL; |
5756d346 ED |
165 | |
166 | data_len = ACCESS_ONCE(copy_range); | |
167 | if (data_len == 0 || data_len > entry->skb->len) | |
1da177e4 | 168 | data_len = entry->skb->len; |
e905a9ed | 169 | |
1da177e4 LT |
170 | size = NLMSG_SPACE(sizeof(*pmsg) + data_len); |
171 | break; | |
e905a9ed | 172 | |
1da177e4 LT |
173 | default: |
174 | *errp = -EINVAL; | |
1da177e4 LT |
175 | return NULL; |
176 | } | |
177 | ||
1da177e4 LT |
178 | skb = alloc_skb(size, GFP_ATOMIC); |
179 | if (!skb) | |
180 | goto nlmsg_failure; | |
e905a9ed | 181 | |
27a884dc | 182 | old_tail = skb->tail; |
1da177e4 LT |
183 | nlh = NLMSG_PUT(skb, 0, 0, IPQM_PACKET, size - sizeof(*nlh)); |
184 | pmsg = NLMSG_DATA(nlh); | |
185 | memset(pmsg, 0, sizeof(*pmsg)); | |
186 | ||
187 | pmsg->packet_id = (unsigned long )entry; | |
188 | pmsg->data_len = data_len; | |
b7aa0bf7 ED |
189 | tv = ktime_to_timeval(entry->skb->tstamp); |
190 | pmsg->timestamp_sec = tv.tv_sec; | |
191 | pmsg->timestamp_usec = tv.tv_usec; | |
82e91ffe | 192 | pmsg->mark = entry->skb->mark; |
02f014d8 | 193 | pmsg->hook = entry->hook; |
1da177e4 | 194 | pmsg->hw_protocol = entry->skb->protocol; |
e905a9ed | 195 | |
02f014d8 PM |
196 | if (entry->indev) |
197 | strcpy(pmsg->indev_name, entry->indev->name); | |
1da177e4 LT |
198 | else |
199 | pmsg->indev_name[0] = '\0'; | |
e905a9ed | 200 | |
02f014d8 PM |
201 | if (entry->outdev) |
202 | strcpy(pmsg->outdev_name, entry->outdev->name); | |
1da177e4 LT |
203 | else |
204 | pmsg->outdev_name[0] = '\0'; | |
e905a9ed | 205 | |
2c38de4c NC |
206 | if (entry->indev && entry->skb->dev && |
207 | entry->skb->mac_header != entry->skb->network_header) { | |
1da177e4 | 208 | pmsg->hw_type = entry->skb->dev->type; |
b95cce35 SH |
209 | pmsg->hw_addrlen = dev_parse_header(entry->skb, |
210 | pmsg->hw_addr); | |
1da177e4 | 211 | } |
e905a9ed | 212 | |
1da177e4 LT |
213 | if (data_len) |
214 | if (skb_copy_bits(entry->skb, 0, pmsg->payload, data_len)) | |
215 | BUG(); | |
e905a9ed | 216 | |
1da177e4 LT |
217 | nlh->nlmsg_len = skb->tail - old_tail; |
218 | return skb; | |
219 | ||
220 | nlmsg_failure: | |
91c66c68 | 221 | kfree_skb(skb); |
1da177e4 LT |
222 | *errp = -EINVAL; |
223 | printk(KERN_ERR "ip_queue: error creating packet message\n"); | |
224 | return NULL; | |
225 | } | |
226 | ||
227 | static int | |
02f014d8 | 228 | ipq_enqueue_packet(struct nf_queue_entry *entry, unsigned int queuenum) |
1da177e4 LT |
229 | { |
230 | int status = -EINVAL; | |
231 | struct sk_buff *nskb; | |
1da177e4 LT |
232 | |
233 | if (copy_mode == IPQ_COPY_NONE) | |
234 | return -EAGAIN; | |
235 | ||
1da177e4 LT |
236 | nskb = ipq_build_packet_message(entry, &status); |
237 | if (nskb == NULL) | |
02f014d8 | 238 | return status; |
e905a9ed | 239 | |
5756d346 | 240 | spin_lock_bh(&queue_lock); |
e905a9ed | 241 | |
1da177e4 | 242 | if (!peer_pid) |
e905a9ed | 243 | goto err_out_free_nskb; |
1da177e4 LT |
244 | |
245 | if (queue_total >= queue_maxlen) { | |
e905a9ed | 246 | queue_dropped++; |
1da177e4 LT |
247 | status = -ENOSPC; |
248 | if (net_ratelimit()) | |
e905a9ed | 249 | printk (KERN_WARNING "ip_queue: full at %d entries, " |
1da177e4 LT |
250 | "dropping packets(s). Dropped: %d\n", queue_total, |
251 | queue_dropped); | |
252 | goto err_out_free_nskb; | |
253 | } | |
254 | ||
e905a9ed | 255 | /* netlink_unicast will either free the nskb or attach it to a socket */ |
1da177e4 LT |
256 | status = netlink_unicast(ipqnl, nskb, peer_pid, MSG_DONTWAIT); |
257 | if (status < 0) { | |
e905a9ed | 258 | queue_user_dropped++; |
1da177e4 LT |
259 | goto err_out_unlock; |
260 | } | |
261 | ||
262 | __ipq_enqueue_entry(entry); | |
263 | ||
5756d346 | 264 | spin_unlock_bh(&queue_lock); |
1da177e4 LT |
265 | return status; |
266 | ||
267 | err_out_free_nskb: | |
e905a9ed YH |
268 | kfree_skb(nskb); |
269 | ||
1da177e4 | 270 | err_out_unlock: |
5756d346 | 271 | spin_unlock_bh(&queue_lock); |
1da177e4 LT |
272 | return status; |
273 | } | |
274 | ||
275 | static int | |
02f014d8 | 276 | ipq_mangle_ipv4(ipq_verdict_msg_t *v, struct nf_queue_entry *e) |
1da177e4 LT |
277 | { |
278 | int diff; | |
279 | struct iphdr *user_iph = (struct iphdr *)v->payload; | |
e2b58a67 | 280 | struct sk_buff *nskb; |
1da177e4 LT |
281 | |
282 | if (v->data_len < sizeof(*user_iph)) | |
283 | return 0; | |
284 | diff = v->data_len - e->skb->len; | |
d8a585d7 PM |
285 | if (diff < 0) { |
286 | if (pskb_trim(e->skb, v->data_len)) | |
287 | return -ENOMEM; | |
288 | } else if (diff > 0) { | |
1da177e4 LT |
289 | if (v->data_len > 0xFFFF) |
290 | return -EINVAL; | |
291 | if (diff > skb_tailroom(e->skb)) { | |
9a732ed6 AE |
292 | nskb = skb_copy_expand(e->skb, skb_headroom(e->skb), |
293 | diff, GFP_ATOMIC); | |
e2b58a67 | 294 | if (!nskb) { |
2ca7b0ac | 295 | printk(KERN_WARNING "ip_queue: error " |
e2b58a67 PM |
296 | "in mangle, dropping packet\n"); |
297 | return -ENOMEM; | |
1da177e4 | 298 | } |
e2b58a67 PM |
299 | kfree_skb(e->skb); |
300 | e->skb = nskb; | |
1da177e4 LT |
301 | } |
302 | skb_put(e->skb, diff); | |
303 | } | |
37d41879 | 304 | if (!skb_make_writable(e->skb, v->data_len)) |
1da177e4 | 305 | return -ENOMEM; |
27d7ff46 | 306 | skb_copy_to_linear_data(e->skb, v->payload, v->data_len); |
66a79a19 | 307 | e->skb->ip_summed = CHECKSUM_NONE; |
1da177e4 | 308 | |
1da177e4 LT |
309 | return 0; |
310 | } | |
311 | ||
1da177e4 LT |
312 | static int |
313 | ipq_set_verdict(struct ipq_verdict_msg *vmsg, unsigned int len) | |
314 | { | |
02f014d8 | 315 | struct nf_queue_entry *entry; |
1da177e4 | 316 | |
c6675233 | 317 | if (vmsg->value > NF_MAX_VERDICT || vmsg->value == NF_STOLEN) |
1da177e4 LT |
318 | return -EINVAL; |
319 | ||
95214092 | 320 | entry = ipq_find_dequeue_entry(vmsg->id); |
1da177e4 LT |
321 | if (entry == NULL) |
322 | return -ENOENT; | |
323 | else { | |
324 | int verdict = vmsg->value; | |
e905a9ed | 325 | |
1da177e4 LT |
326 | if (vmsg->data_len && vmsg->data_len == len) |
327 | if (ipq_mangle_ipv4(vmsg, entry) < 0) | |
328 | verdict = NF_DROP; | |
e905a9ed | 329 | |
4b3d15ef | 330 | nf_reinject(entry, verdict); |
1da177e4 LT |
331 | return 0; |
332 | } | |
333 | } | |
334 | ||
335 | static int | |
336 | ipq_set_mode(unsigned char mode, unsigned int range) | |
337 | { | |
338 | int status; | |
339 | ||
5756d346 | 340 | spin_lock_bh(&queue_lock); |
1da177e4 | 341 | status = __ipq_set_mode(mode, range); |
5756d346 | 342 | spin_unlock_bh(&queue_lock); |
1da177e4 LT |
343 | return status; |
344 | } | |
345 | ||
346 | static int | |
347 | ipq_receive_peer(struct ipq_peer_msg *pmsg, | |
e905a9ed | 348 | unsigned char type, unsigned int len) |
1da177e4 LT |
349 | { |
350 | int status = 0; | |
351 | ||
352 | if (len < sizeof(*pmsg)) | |
353 | return -EINVAL; | |
354 | ||
355 | switch (type) { | |
356 | case IPQM_MODE: | |
357 | status = ipq_set_mode(pmsg->msg.mode.value, | |
e905a9ed | 358 | pmsg->msg.mode.range); |
1da177e4 | 359 | break; |
e905a9ed | 360 | |
1da177e4 | 361 | case IPQM_VERDICT: |
c6675233 FW |
362 | status = ipq_set_verdict(&pmsg->msg.verdict, |
363 | len - sizeof(*pmsg)); | |
364 | break; | |
1da177e4 LT |
365 | default: |
366 | status = -EINVAL; | |
367 | } | |
368 | return status; | |
369 | } | |
370 | ||
371 | static int | |
02f014d8 | 372 | dev_cmp(struct nf_queue_entry *entry, unsigned long ifindex) |
1da177e4 | 373 | { |
02f014d8 PM |
374 | if (entry->indev) |
375 | if (entry->indev->ifindex == ifindex) | |
1da177e4 | 376 | return 1; |
02f014d8 PM |
377 | if (entry->outdev) |
378 | if (entry->outdev->ifindex == ifindex) | |
1da177e4 | 379 | return 1; |
ef47c6a7 PM |
380 | #ifdef CONFIG_BRIDGE_NETFILTER |
381 | if (entry->skb->nf_bridge) { | |
382 | if (entry->skb->nf_bridge->physindev && | |
383 | entry->skb->nf_bridge->physindev->ifindex == ifindex) | |
384 | return 1; | |
385 | if (entry->skb->nf_bridge->physoutdev && | |
386 | entry->skb->nf_bridge->physoutdev->ifindex == ifindex) | |
e905a9ed | 387 | return 1; |
ef47c6a7 PM |
388 | } |
389 | #endif | |
1da177e4 LT |
390 | return 0; |
391 | } | |
392 | ||
393 | static void | |
394 | ipq_dev_drop(int ifindex) | |
395 | { | |
95214092 | 396 | ipq_flush(dev_cmp, ifindex); |
1da177e4 LT |
397 | } |
398 | ||
399 | #define RCV_SKB_FAIL(err) do { netlink_ack(skb, nlh, (err)); return; } while (0) | |
400 | ||
401 | static inline void | |
cd40b7d3 | 402 | __ipq_rcv_skb(struct sk_buff *skb) |
1da177e4 | 403 | { |
d232b8dd DJ |
404 | int status, type, pid, flags; |
405 | unsigned int nlmsglen, skblen; | |
1da177e4 | 406 | struct nlmsghdr *nlh; |
b90e5794 | 407 | bool enable_timestamp = false; |
1da177e4 LT |
408 | |
409 | skblen = skb->len; | |
410 | if (skblen < sizeof(*nlh)) | |
411 | return; | |
412 | ||
b529ccf2 | 413 | nlh = nlmsg_hdr(skb); |
1da177e4 LT |
414 | nlmsglen = nlh->nlmsg_len; |
415 | if (nlmsglen < sizeof(*nlh) || skblen < nlmsglen) | |
416 | return; | |
417 | ||
418 | pid = nlh->nlmsg_pid; | |
419 | flags = nlh->nlmsg_flags; | |
e905a9ed | 420 | |
1da177e4 LT |
421 | if(pid <= 0 || !(flags & NLM_F_REQUEST) || flags & NLM_F_MULTI) |
422 | RCV_SKB_FAIL(-EINVAL); | |
e905a9ed | 423 | |
1da177e4 LT |
424 | if (flags & MSG_TRUNC) |
425 | RCV_SKB_FAIL(-ECOMM); | |
e905a9ed | 426 | |
1da177e4 LT |
427 | type = nlh->nlmsg_type; |
428 | if (type < NLMSG_NOOP || type >= IPQM_MAX) | |
429 | RCV_SKB_FAIL(-EINVAL); | |
e905a9ed | 430 | |
1da177e4 LT |
431 | if (type <= IPQM_BASE) |
432 | return; | |
e905a9ed | 433 | |
fd778461 | 434 | if (!capable(CAP_NET_ADMIN)) |
1da177e4 | 435 | RCV_SKB_FAIL(-EPERM); |
e905a9ed | 436 | |
5756d346 | 437 | spin_lock_bh(&queue_lock); |
e905a9ed | 438 | |
1da177e4 LT |
439 | if (peer_pid) { |
440 | if (peer_pid != pid) { | |
5756d346 | 441 | spin_unlock_bh(&queue_lock); |
1da177e4 LT |
442 | RCV_SKB_FAIL(-EBUSY); |
443 | } | |
444 | } else { | |
b90e5794 | 445 | enable_timestamp = true; |
1da177e4 LT |
446 | peer_pid = pid; |
447 | } | |
e905a9ed | 448 | |
5756d346 | 449 | spin_unlock_bh(&queue_lock); |
b90e5794 ED |
450 | if (enable_timestamp) |
451 | net_enable_timestamp(); | |
1da177e4 | 452 | status = ipq_receive_peer(NLMSG_DATA(nlh), type, |
e905a9ed | 453 | nlmsglen - NLMSG_LENGTH(0)); |
1da177e4 LT |
454 | if (status < 0) |
455 | RCV_SKB_FAIL(status); | |
e905a9ed | 456 | |
1da177e4 LT |
457 | if (flags & NLM_F_ACK) |
458 | netlink_ack(skb, nlh, 0); | |
1da177e4 LT |
459 | } |
460 | ||
461 | static void | |
cd40b7d3 | 462 | ipq_rcv_skb(struct sk_buff *skb) |
1da177e4 | 463 | { |
4a3e2f71 | 464 | mutex_lock(&ipqnl_mutex); |
cd40b7d3 | 465 | __ipq_rcv_skb(skb); |
4a3e2f71 | 466 | mutex_unlock(&ipqnl_mutex); |
1da177e4 LT |
467 | } |
468 | ||
469 | static int | |
470 | ipq_rcv_dev_event(struct notifier_block *this, | |
e905a9ed | 471 | unsigned long event, void *ptr) |
1da177e4 LT |
472 | { |
473 | struct net_device *dev = ptr; | |
474 | ||
721499e8 | 475 | if (!net_eq(dev_net(dev), &init_net)) |
e9dc8653 EB |
476 | return NOTIFY_DONE; |
477 | ||
1da177e4 LT |
478 | /* Drop any packets associated with the downed device */ |
479 | if (event == NETDEV_DOWN) | |
480 | ipq_dev_drop(dev->ifindex); | |
481 | return NOTIFY_DONE; | |
482 | } | |
483 | ||
484 | static struct notifier_block ipq_dev_notifier = { | |
485 | .notifier_call = ipq_rcv_dev_event, | |
486 | }; | |
487 | ||
488 | static int | |
489 | ipq_rcv_nl_event(struct notifier_block *this, | |
e905a9ed | 490 | unsigned long event, void *ptr) |
1da177e4 LT |
491 | { |
492 | struct netlink_notify *n = ptr; | |
493 | ||
dee5817e | 494 | if (event == NETLINK_URELEASE && n->protocol == NETLINK_FIREWALL) { |
5756d346 | 495 | spin_lock_bh(&queue_lock); |
09ad9bc7 | 496 | if ((net_eq(n->net, &init_net)) && (n->pid == peer_pid)) |
1da177e4 | 497 | __ipq_reset(); |
5756d346 | 498 | spin_unlock_bh(&queue_lock); |
1da177e4 LT |
499 | } |
500 | return NOTIFY_DONE; | |
501 | } | |
502 | ||
503 | static struct notifier_block ipq_nl_notifier = { | |
504 | .notifier_call = ipq_rcv_nl_event, | |
505 | }; | |
506 | ||
c392a740 | 507 | #ifdef CONFIG_SYSCTL |
1da177e4 LT |
508 | static struct ctl_table_header *ipq_sysctl_header; |
509 | ||
510 | static ctl_table ipq_table[] = { | |
511 | { | |
1da177e4 LT |
512 | .procname = NET_IPQ_QMAX_NAME, |
513 | .data = &queue_maxlen, | |
514 | .maxlen = sizeof(queue_maxlen), | |
515 | .mode = 0644, | |
516 | .proc_handler = proc_dointvec | |
517 | }, | |
f8572d8f | 518 | { } |
1da177e4 | 519 | }; |
c392a740 | 520 | #endif |
1da177e4 | 521 | |
c392a740 | 522 | #ifdef CONFIG_PROC_FS |
7351a22a | 523 | static int ip_queue_show(struct seq_file *m, void *v) |
1da177e4 | 524 | { |
5756d346 | 525 | spin_lock_bh(&queue_lock); |
e905a9ed | 526 | |
7351a22a | 527 | seq_printf(m, |
e905a9ed YH |
528 | "Peer PID : %d\n" |
529 | "Copy mode : %hu\n" | |
530 | "Copy range : %u\n" | |
531 | "Queue length : %u\n" | |
532 | "Queue max. length : %u\n" | |
1da177e4 LT |
533 | "Queue dropped : %u\n" |
534 | "Netlink dropped : %u\n", | |
e905a9ed YH |
535 | peer_pid, |
536 | copy_mode, | |
537 | copy_range, | |
538 | queue_total, | |
539 | queue_maxlen, | |
1da177e4 LT |
540 | queue_dropped, |
541 | queue_user_dropped); | |
542 | ||
5756d346 | 543 | spin_unlock_bh(&queue_lock); |
7351a22a AD |
544 | return 0; |
545 | } | |
e905a9ed | 546 | |
7351a22a AD |
547 | static int ip_queue_open(struct inode *inode, struct file *file) |
548 | { | |
549 | return single_open(file, ip_queue_show, NULL); | |
1da177e4 | 550 | } |
7351a22a AD |
551 | |
552 | static const struct file_operations ip_queue_proc_fops = { | |
553 | .open = ip_queue_open, | |
554 | .read = seq_read, | |
555 | .llseek = seq_lseek, | |
556 | .release = single_release, | |
557 | .owner = THIS_MODULE, | |
558 | }; | |
c392a740 | 559 | #endif |
1da177e4 | 560 | |
e3ac5298 | 561 | static const struct nf_queue_handler nfqh = { |
bbd86b9f HW |
562 | .name = "ip_queue", |
563 | .outfn = &ipq_enqueue_packet, | |
564 | }; | |
565 | ||
32292a7f | 566 | static int __init ip_queue_init(void) |
1da177e4 LT |
567 | { |
568 | int status = -ENOMEM; | |
c392a740 | 569 | struct proc_dir_entry *proc __maybe_unused; |
e905a9ed | 570 | |
1da177e4 | 571 | netlink_register_notifier(&ipq_nl_notifier); |
b4b51029 | 572 | ipqnl = netlink_kernel_create(&init_net, NETLINK_FIREWALL, 0, |
cd40b7d3 | 573 | ipq_rcv_skb, NULL, THIS_MODULE); |
1da177e4 LT |
574 | if (ipqnl == NULL) { |
575 | printk(KERN_ERR "ip_queue: failed to create netlink socket\n"); | |
576 | goto cleanup_netlink_notifier; | |
577 | } | |
578 | ||
c392a740 | 579 | #ifdef CONFIG_PROC_FS |
8eeee8b1 DL |
580 | proc = proc_create(IPQ_PROC_FS_NAME, 0, init_net.proc_net, |
581 | &ip_queue_proc_fops); | |
582 | if (!proc) { | |
1da177e4 LT |
583 | printk(KERN_ERR "ip_queue: failed to create proc entry\n"); |
584 | goto cleanup_ipqnl; | |
585 | } | |
c392a740 | 586 | #endif |
1da177e4 | 587 | register_netdevice_notifier(&ipq_dev_notifier); |
c392a740 | 588 | #ifdef CONFIG_SYSCTL |
5dd3df10 | 589 | ipq_sysctl_header = register_net_sysctl_table(&init_net, net_ipv4_ctl_path, ipq_table); |
c392a740 | 590 | #endif |
4b1e27e9 | 591 | status = nf_register_queue_handler(NFPROTO_IPV4, &nfqh); |
1da177e4 LT |
592 | if (status < 0) { |
593 | printk(KERN_ERR "ip_queue: failed to register queue handler\n"); | |
594 | goto cleanup_sysctl; | |
595 | } | |
596 | return status; | |
597 | ||
1da177e4 | 598 | cleanup_sysctl: |
c392a740 | 599 | #ifdef CONFIG_SYSCTL |
5dd3df10 | 600 | unregister_net_sysctl_table(ipq_sysctl_header); |
c392a740 | 601 | #endif |
1da177e4 | 602 | unregister_netdevice_notifier(&ipq_dev_notifier); |
457c4cbc | 603 | proc_net_remove(&init_net, IPQ_PROC_FS_NAME); |
c392a740 | 604 | cleanup_ipqnl: __maybe_unused |
b7c6ba6e | 605 | netlink_kernel_release(ipqnl); |
4a3e2f71 AV |
606 | mutex_lock(&ipqnl_mutex); |
607 | mutex_unlock(&ipqnl_mutex); | |
e905a9ed | 608 | |
1da177e4 LT |
609 | cleanup_netlink_notifier: |
610 | netlink_unregister_notifier(&ipq_nl_notifier); | |
611 | return status; | |
612 | } | |
613 | ||
65b4b4e8 | 614 | static void __exit ip_queue_fini(void) |
1da177e4 | 615 | { |
32292a7f | 616 | nf_unregister_queue_handlers(&nfqh); |
9481721b | 617 | |
95214092 | 618 | ipq_flush(NULL, 0); |
32292a7f | 619 | |
c392a740 | 620 | #ifdef CONFIG_SYSCTL |
5dd3df10 | 621 | unregister_net_sysctl_table(ipq_sysctl_header); |
c392a740 | 622 | #endif |
32292a7f | 623 | unregister_netdevice_notifier(&ipq_dev_notifier); |
457c4cbc | 624 | proc_net_remove(&init_net, IPQ_PROC_FS_NAME); |
32292a7f | 625 | |
b7c6ba6e | 626 | netlink_kernel_release(ipqnl); |
32292a7f PM |
627 | mutex_lock(&ipqnl_mutex); |
628 | mutex_unlock(&ipqnl_mutex); | |
629 | ||
630 | netlink_unregister_notifier(&ipq_nl_notifier); | |
1da177e4 LT |
631 | } |
632 | ||
633 | MODULE_DESCRIPTION("IPv4 packet queue handler"); | |
634 | MODULE_AUTHOR("James Morris <jmorris@intercode.com.au>"); | |
635 | MODULE_LICENSE("GPL"); | |
95ba434f | 636 | MODULE_ALIAS_NET_PF_PROTO(PF_NETLINK, NETLINK_FIREWALL); |
1da177e4 | 637 | |
65b4b4e8 AM |
638 | module_init(ip_queue_init); |
639 | module_exit(ip_queue_fini); |