2 * xt_hashlimit - Netfilter module to limit the number of packets per time
3 * separately for each hashbucket (sourceip/sourceport/dstip/dstport)
5 * (C) 2003-2004 by Harald Welte <laforge@netfilter.org>
6 * Copyright © CC Computer Consultants GmbH, 2007 - 2008
8 * Development of this code was funded by Astaro AG, http://www.astaro.com/
10 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
11 #include <linux/module.h>
12 #include <linux/spinlock.h>
13 #include <linux/random.h>
14 #include <linux/jhash.h>
15 #include <linux/slab.h>
16 #include <linux/vmalloc.h>
17 #include <linux/proc_fs.h>
18 #include <linux/seq_file.h>
19 #include <linux/list.h>
20 #include <linux/skbuff.h>
24 #if IS_ENABLED(CONFIG_IP6_NF_IPTABLES)
25 #include <linux/ipv6.h>
29 #include <net/net_namespace.h>
30 #include <net/netns/generic.h>
32 #include <linux/netfilter/x_tables.h>
33 #include <linux/netfilter_ipv4/ip_tables.h>
34 #include <linux/netfilter_ipv6/ip6_tables.h>
35 #include <linux/netfilter/xt_hashlimit.h>
36 #include <linux/mutex.h>
38 MODULE_LICENSE("GPL");
39 MODULE_AUTHOR("Harald Welte <laforge@netfilter.org>");
40 MODULE_AUTHOR("Jan Engelhardt <jengelh@medozas.de>");
41 MODULE_DESCRIPTION("Xtables: per hash-bucket rate-limit match");
42 MODULE_ALIAS("ipt_hashlimit");
43 MODULE_ALIAS("ip6t_hashlimit");
45 struct hashlimit_net
{
46 struct hlist_head htables
;
47 struct proc_dir_entry
*ipt_hashlimit
;
48 struct proc_dir_entry
*ip6t_hashlimit
;
51 static int hashlimit_net_id
;
52 static inline struct hashlimit_net
*hashlimit_pernet(struct net
*net
)
54 return net_generic(net
, hashlimit_net_id
);
57 /* need to declare this at the top */
58 static const struct file_operations dl_file_ops
;
67 #if IS_ENABLED(CONFIG_IP6_NF_IPTABLES)
79 /* static / read-only parts in the beginning */
80 struct hlist_node node
;
81 struct dsthash_dst dst
;
83 /* modified structure members in the end */
85 unsigned long expires
; /* precalculated expiry time */
87 unsigned long prev
; /* last modification */
89 u_int32_t credit_cap
, cost
;
94 struct xt_hashlimit_htable
{
95 struct hlist_node node
; /* global list of all htables */
100 struct hashlimit_cfg1 cfg
; /* config */
102 /* used internally */
103 spinlock_t lock
; /* lock for list_head */
104 u_int32_t rnd
; /* random seed for hash */
105 unsigned int count
; /* number entries in table */
106 struct timer_list timer
; /* timer for gc */
109 struct proc_dir_entry
*pde
;
112 struct hlist_head hash
[0]; /* hashtable itself */
115 static DEFINE_MUTEX(hashlimit_mutex
); /* protects htables list */
116 static struct kmem_cache
*hashlimit_cachep __read_mostly
;
118 static inline bool dst_cmp(const struct dsthash_ent
*ent
,
119 const struct dsthash_dst
*b
)
121 return !memcmp(&ent
->dst
, b
, sizeof(ent
->dst
));
125 hash_dst(const struct xt_hashlimit_htable
*ht
, const struct dsthash_dst
*dst
)
127 u_int32_t hash
= jhash2((const u32
*)dst
,
128 sizeof(*dst
)/sizeof(u32
),
131 * Instead of returning hash % ht->cfg.size (implying a divide)
132 * we return the high 32 bits of the (hash * ht->cfg.size) that will
133 * give results between [0 and cfg.size-1] and same hash distribution,
134 * but using a multiply, less expensive than a divide
136 return ((u64
)hash
* ht
->cfg
.size
) >> 32;
139 static struct dsthash_ent
*
140 dsthash_find(const struct xt_hashlimit_htable
*ht
,
141 const struct dsthash_dst
*dst
)
143 struct dsthash_ent
*ent
;
144 struct hlist_node
*pos
;
145 u_int32_t hash
= hash_dst(ht
, dst
);
147 if (!hlist_empty(&ht
->hash
[hash
])) {
148 hlist_for_each_entry_rcu(ent
, pos
, &ht
->hash
[hash
], node
)
149 if (dst_cmp(ent
, dst
)) {
150 spin_lock(&ent
->lock
);
157 /* allocate dsthash_ent, initialize dst, put in htable and lock it */
158 static struct dsthash_ent
*
159 dsthash_alloc_init(struct xt_hashlimit_htable
*ht
,
160 const struct dsthash_dst
*dst
, bool *race
)
162 struct dsthash_ent
*ent
;
164 spin_lock(&ht
->lock
);
166 /* Two or more packets may race to create the same entry in the
167 * hashtable, double check if this packet lost race.
169 ent
= dsthash_find(ht
, dst
);
171 spin_unlock(&ht
->lock
);
176 /* initialize hash with random val at the time we allocate
177 * the first hashtable entry */
178 if (unlikely(!ht
->rnd_initialized
)) {
179 get_random_bytes(&ht
->rnd
, sizeof(ht
->rnd
));
180 ht
->rnd_initialized
= true;
183 if (ht
->cfg
.max
&& ht
->count
>= ht
->cfg
.max
) {
184 /* FIXME: do something. question is what.. */
185 net_err_ratelimited("max count of %u reached\n", ht
->cfg
.max
);
188 ent
= kmem_cache_alloc(hashlimit_cachep
, GFP_ATOMIC
);
190 memcpy(&ent
->dst
, dst
, sizeof(ent
->dst
));
191 spin_lock_init(&ent
->lock
);
193 spin_lock(&ent
->lock
);
194 hlist_add_head_rcu(&ent
->node
, &ht
->hash
[hash_dst(ht
, dst
)]);
197 spin_unlock(&ht
->lock
);
201 static void dsthash_free_rcu(struct rcu_head
*head
)
203 struct dsthash_ent
*ent
= container_of(head
, struct dsthash_ent
, rcu
);
205 kmem_cache_free(hashlimit_cachep
, ent
);
209 dsthash_free(struct xt_hashlimit_htable
*ht
, struct dsthash_ent
*ent
)
211 hlist_del_rcu(&ent
->node
);
212 call_rcu_bh(&ent
->rcu
, dsthash_free_rcu
);
215 static void htable_gc(unsigned long htlong
);
217 static int htable_create(struct net
*net
, struct xt_hashlimit_mtinfo1
*minfo
,
220 struct hashlimit_net
*hashlimit_net
= hashlimit_pernet(net
);
221 struct xt_hashlimit_htable
*hinfo
;
225 if (minfo
->cfg
.size
) {
226 size
= minfo
->cfg
.size
;
228 size
= (totalram_pages
<< PAGE_SHIFT
) / 16384 /
229 sizeof(struct list_head
);
230 if (totalram_pages
> 1024 * 1024 * 1024 / PAGE_SIZE
)
235 /* FIXME: don't use vmalloc() here or anywhere else -HW */
236 hinfo
= vmalloc(sizeof(struct xt_hashlimit_htable
) +
237 sizeof(struct list_head
) * size
);
240 minfo
->hinfo
= hinfo
;
242 /* copy match config into hashtable config */
243 memcpy(&hinfo
->cfg
, &minfo
->cfg
, sizeof(hinfo
->cfg
));
244 hinfo
->cfg
.size
= size
;
245 if (hinfo
->cfg
.max
== 0)
246 hinfo
->cfg
.max
= 8 * hinfo
->cfg
.size
;
247 else if (hinfo
->cfg
.max
< hinfo
->cfg
.size
)
248 hinfo
->cfg
.max
= hinfo
->cfg
.size
;
250 for (i
= 0; i
< hinfo
->cfg
.size
; i
++)
251 INIT_HLIST_HEAD(&hinfo
->hash
[i
]);
255 hinfo
->family
= family
;
256 hinfo
->rnd_initialized
= false;
257 spin_lock_init(&hinfo
->lock
);
259 hinfo
->pde
= proc_create_data(minfo
->name
, 0,
260 (family
== NFPROTO_IPV4
) ?
261 hashlimit_net
->ipt_hashlimit
: hashlimit_net
->ip6t_hashlimit
,
262 &dl_file_ops
, hinfo
);
263 if (hinfo
->pde
== NULL
) {
269 setup_timer(&hinfo
->timer
, htable_gc
, (unsigned long)hinfo
);
270 hinfo
->timer
.expires
= jiffies
+ msecs_to_jiffies(hinfo
->cfg
.gc_interval
);
271 add_timer(&hinfo
->timer
);
273 hlist_add_head(&hinfo
->node
, &hashlimit_net
->htables
);
278 static bool select_all(const struct xt_hashlimit_htable
*ht
,
279 const struct dsthash_ent
*he
)
284 static bool select_gc(const struct xt_hashlimit_htable
*ht
,
285 const struct dsthash_ent
*he
)
287 return time_after_eq(jiffies
, he
->expires
);
290 static void htable_selective_cleanup(struct xt_hashlimit_htable
*ht
,
291 bool (*select
)(const struct xt_hashlimit_htable
*ht
,
292 const struct dsthash_ent
*he
))
296 /* lock hash table and iterate over it */
297 spin_lock_bh(&ht
->lock
);
298 for (i
= 0; i
< ht
->cfg
.size
; i
++) {
299 struct dsthash_ent
*dh
;
300 struct hlist_node
*pos
, *n
;
301 hlist_for_each_entry_safe(dh
, pos
, n
, &ht
->hash
[i
], node
) {
302 if ((*select
)(ht
, dh
))
303 dsthash_free(ht
, dh
);
306 spin_unlock_bh(&ht
->lock
);
309 /* hash table garbage collector, run by timer */
310 static void htable_gc(unsigned long htlong
)
312 struct xt_hashlimit_htable
*ht
= (struct xt_hashlimit_htable
*)htlong
;
314 htable_selective_cleanup(ht
, select_gc
);
316 /* re-add the timer accordingly */
317 ht
->timer
.expires
= jiffies
+ msecs_to_jiffies(ht
->cfg
.gc_interval
);
318 add_timer(&ht
->timer
);
321 static void htable_destroy(struct xt_hashlimit_htable
*hinfo
)
323 struct hashlimit_net
*hashlimit_net
= hashlimit_pernet(hinfo
->net
);
324 struct proc_dir_entry
*parent
;
326 del_timer_sync(&hinfo
->timer
);
328 if (hinfo
->family
== NFPROTO_IPV4
)
329 parent
= hashlimit_net
->ipt_hashlimit
;
331 parent
= hashlimit_net
->ip6t_hashlimit
;
332 remove_proc_entry(hinfo
->pde
->name
, parent
);
333 htable_selective_cleanup(hinfo
, select_all
);
337 static struct xt_hashlimit_htable
*htable_find_get(struct net
*net
,
341 struct hashlimit_net
*hashlimit_net
= hashlimit_pernet(net
);
342 struct xt_hashlimit_htable
*hinfo
;
343 struct hlist_node
*pos
;
345 hlist_for_each_entry(hinfo
, pos
, &hashlimit_net
->htables
, node
) {
346 if (!strcmp(name
, hinfo
->pde
->name
) &&
347 hinfo
->family
== family
) {
355 static void htable_put(struct xt_hashlimit_htable
*hinfo
)
357 mutex_lock(&hashlimit_mutex
);
358 if (--hinfo
->use
== 0) {
359 hlist_del(&hinfo
->node
);
360 htable_destroy(hinfo
);
362 mutex_unlock(&hashlimit_mutex
);
365 /* The algorithm used is the Simple Token Bucket Filter (TBF)
366 * see net/sched/sch_tbf.c in the linux source tree
369 /* Rusty: This is my (non-mathematically-inclined) understanding of
370 this algorithm. The `average rate' in jiffies becomes your initial
371 amount of credit `credit' and the most credit you can ever have
372 `credit_cap'. The `peak rate' becomes the cost of passing the
375 `prev' tracks the last packet hit: you gain one credit per jiffy.
376 If you get credit balance more than this, the extra credit is
377 discarded. Every time the match passes, you lose `cost' credits;
378 if you don't have that many, the test fails.
380 See Alexey's formal explanation in net/sched/sch_tbf.c.
382 To get the maximum range, we multiply by this factor (ie. you get N
383 credits per jiffy). We want to allow a rate as low as 1 per day
384 (slowest userspace tool allows), which means
385 CREDITS_PER_JIFFY*HZ*60*60*24 < 2^32 ie.
387 #define MAX_CPJ (0xFFFFFFFF / (HZ*60*60*24))
389 /* Repeated shift and or gives us all 1s, final shift and add 1 gives
390 * us the power of 2 below the theoretical max, so GCC simply does a
392 #define _POW2_BELOW2(x) ((x)|((x)>>1))
393 #define _POW2_BELOW4(x) (_POW2_BELOW2(x)|_POW2_BELOW2((x)>>2))
394 #define _POW2_BELOW8(x) (_POW2_BELOW4(x)|_POW2_BELOW4((x)>>4))
395 #define _POW2_BELOW16(x) (_POW2_BELOW8(x)|_POW2_BELOW8((x)>>8))
396 #define _POW2_BELOW32(x) (_POW2_BELOW16(x)|_POW2_BELOW16((x)>>16))
397 #define POW2_BELOW32(x) ((_POW2_BELOW32(x)>>1) + 1)
399 #define CREDITS_PER_JIFFY POW2_BELOW32(MAX_CPJ)
401 /* in byte mode, the lowest possible rate is one packet/second.
402 * credit_cap is used as a counter that tells us how many times we can
403 * refill the "credits available" counter when it becomes empty.
405 #define MAX_CPJ_BYTES (0xFFFFFFFF / HZ)
406 #define CREDITS_PER_JIFFY_BYTES POW2_BELOW32(MAX_CPJ_BYTES)
408 static u32
xt_hashlimit_len_to_chunks(u32 len
)
410 return (len
>> XT_HASHLIMIT_BYTE_SHIFT
) + 1;
413 /* Precision saver. */
414 static u32
user2credits(u32 user
)
416 /* If multiplying would overflow... */
417 if (user
> 0xFFFFFFFF / (HZ
*CREDITS_PER_JIFFY
))
419 return (user
/ XT_HASHLIMIT_SCALE
) * HZ
* CREDITS_PER_JIFFY
;
421 return (user
* HZ
* CREDITS_PER_JIFFY
) / XT_HASHLIMIT_SCALE
;
424 static u32
user2credits_byte(u32 user
)
427 us
*= HZ
* CREDITS_PER_JIFFY_BYTES
;
428 return (u32
) (us
>> 32);
431 static void rateinfo_recalc(struct dsthash_ent
*dh
, unsigned long now
, u32 mode
)
433 unsigned long delta
= now
- dh
->rateinfo
.prev
;
439 dh
->rateinfo
.prev
= now
;
441 if (mode
& XT_HASHLIMIT_BYTES
) {
442 u32 tmp
= dh
->rateinfo
.credit
;
443 dh
->rateinfo
.credit
+= CREDITS_PER_JIFFY_BYTES
* delta
;
444 cap
= CREDITS_PER_JIFFY_BYTES
* HZ
;
445 if (tmp
>= dh
->rateinfo
.credit
) {/* overflow */
446 dh
->rateinfo
.credit
= cap
;
450 dh
->rateinfo
.credit
+= delta
* CREDITS_PER_JIFFY
;
451 cap
= dh
->rateinfo
.credit_cap
;
453 if (dh
->rateinfo
.credit
> cap
)
454 dh
->rateinfo
.credit
= cap
;
457 static void rateinfo_init(struct dsthash_ent
*dh
,
458 struct xt_hashlimit_htable
*hinfo
)
460 dh
->rateinfo
.prev
= jiffies
;
461 if (hinfo
->cfg
.mode
& XT_HASHLIMIT_BYTES
) {
462 dh
->rateinfo
.credit
= CREDITS_PER_JIFFY_BYTES
* HZ
;
463 dh
->rateinfo
.cost
= user2credits_byte(hinfo
->cfg
.avg
);
464 dh
->rateinfo
.credit_cap
= hinfo
->cfg
.burst
;
466 dh
->rateinfo
.credit
= user2credits(hinfo
->cfg
.avg
*
468 dh
->rateinfo
.cost
= user2credits(hinfo
->cfg
.avg
);
469 dh
->rateinfo
.credit_cap
= dh
->rateinfo
.credit
;
473 static inline __be32
maskl(__be32 a
, unsigned int l
)
475 return l
? htonl(ntohl(a
) & ~0 << (32 - l
)) : 0;
478 #if IS_ENABLED(CONFIG_IP6_NF_IPTABLES)
479 static void hashlimit_ipv6_mask(__be32
*i
, unsigned int p
)
483 i
[0] = maskl(i
[0], p
);
484 i
[1] = i
[2] = i
[3] = 0;
487 i
[1] = maskl(i
[1], p
- 32);
491 i
[2] = maskl(i
[2], p
- 64);
495 i
[3] = maskl(i
[3], p
- 96);
504 hashlimit_init_dst(const struct xt_hashlimit_htable
*hinfo
,
505 struct dsthash_dst
*dst
,
506 const struct sk_buff
*skb
, unsigned int protoff
)
508 __be16 _ports
[2], *ports
;
512 memset(dst
, 0, sizeof(*dst
));
514 switch (hinfo
->family
) {
516 if (hinfo
->cfg
.mode
& XT_HASHLIMIT_HASH_DIP
)
517 dst
->ip
.dst
= maskl(ip_hdr(skb
)->daddr
,
519 if (hinfo
->cfg
.mode
& XT_HASHLIMIT_HASH_SIP
)
520 dst
->ip
.src
= maskl(ip_hdr(skb
)->saddr
,
523 if (!(hinfo
->cfg
.mode
&
524 (XT_HASHLIMIT_HASH_DPT
| XT_HASHLIMIT_HASH_SPT
)))
526 nexthdr
= ip_hdr(skb
)->protocol
;
528 #if IS_ENABLED(CONFIG_IP6_NF_IPTABLES)
533 if (hinfo
->cfg
.mode
& XT_HASHLIMIT_HASH_DIP
) {
534 memcpy(&dst
->ip6
.dst
, &ipv6_hdr(skb
)->daddr
,
535 sizeof(dst
->ip6
.dst
));
536 hashlimit_ipv6_mask(dst
->ip6
.dst
, hinfo
->cfg
.dstmask
);
538 if (hinfo
->cfg
.mode
& XT_HASHLIMIT_HASH_SIP
) {
539 memcpy(&dst
->ip6
.src
, &ipv6_hdr(skb
)->saddr
,
540 sizeof(dst
->ip6
.src
));
541 hashlimit_ipv6_mask(dst
->ip6
.src
, hinfo
->cfg
.srcmask
);
544 if (!(hinfo
->cfg
.mode
&
545 (XT_HASHLIMIT_HASH_DPT
| XT_HASHLIMIT_HASH_SPT
)))
547 nexthdr
= ipv6_hdr(skb
)->nexthdr
;
548 protoff
= ipv6_skip_exthdr(skb
, sizeof(struct ipv6hdr
), &nexthdr
, &frag_off
);
549 if ((int)protoff
< 0)
559 poff
= proto_ports_offset(nexthdr
);
561 ports
= skb_header_pointer(skb
, protoff
+ poff
, sizeof(_ports
),
564 _ports
[0] = _ports
[1] = 0;
569 if (hinfo
->cfg
.mode
& XT_HASHLIMIT_HASH_SPT
)
570 dst
->src_port
= ports
[0];
571 if (hinfo
->cfg
.mode
& XT_HASHLIMIT_HASH_DPT
)
572 dst
->dst_port
= ports
[1];
576 static u32
hashlimit_byte_cost(unsigned int len
, struct dsthash_ent
*dh
)
578 u64 tmp
= xt_hashlimit_len_to_chunks(len
);
579 tmp
= tmp
* dh
->rateinfo
.cost
;
581 if (unlikely(tmp
> CREDITS_PER_JIFFY_BYTES
* HZ
))
582 tmp
= CREDITS_PER_JIFFY_BYTES
* HZ
;
584 if (dh
->rateinfo
.credit
< tmp
&& dh
->rateinfo
.credit_cap
) {
585 dh
->rateinfo
.credit_cap
--;
586 dh
->rateinfo
.credit
= CREDITS_PER_JIFFY_BYTES
* HZ
;
592 hashlimit_mt(const struct sk_buff
*skb
, struct xt_action_param
*par
)
594 const struct xt_hashlimit_mtinfo1
*info
= par
->matchinfo
;
595 struct xt_hashlimit_htable
*hinfo
= info
->hinfo
;
596 unsigned long now
= jiffies
;
597 struct dsthash_ent
*dh
;
598 struct dsthash_dst dst
;
602 if (hashlimit_init_dst(hinfo
, &dst
, skb
, par
->thoff
) < 0)
606 dh
= dsthash_find(hinfo
, &dst
);
608 dh
= dsthash_alloc_init(hinfo
, &dst
, &race
);
610 rcu_read_unlock_bh();
613 /* Already got an entry, update expiration timeout */
614 dh
->expires
= now
+ msecs_to_jiffies(hinfo
->cfg
.expire
);
615 rateinfo_recalc(dh
, now
, hinfo
->cfg
.mode
);
617 dh
->expires
= jiffies
+ msecs_to_jiffies(hinfo
->cfg
.expire
);
618 rateinfo_init(dh
, hinfo
);
621 /* update expiration timeout */
622 dh
->expires
= now
+ msecs_to_jiffies(hinfo
->cfg
.expire
);
623 rateinfo_recalc(dh
, now
, hinfo
->cfg
.mode
);
626 if (info
->cfg
.mode
& XT_HASHLIMIT_BYTES
)
627 cost
= hashlimit_byte_cost(skb
->len
, dh
);
629 cost
= dh
->rateinfo
.cost
;
631 if (dh
->rateinfo
.credit
>= cost
) {
632 /* below the limit */
633 dh
->rateinfo
.credit
-= cost
;
634 spin_unlock(&dh
->lock
);
635 rcu_read_unlock_bh();
636 return !(info
->cfg
.mode
& XT_HASHLIMIT_INVERT
);
639 spin_unlock(&dh
->lock
);
640 rcu_read_unlock_bh();
641 /* default match is underlimit - so over the limit, we need to invert */
642 return info
->cfg
.mode
& XT_HASHLIMIT_INVERT
;
649 static int hashlimit_mt_check(const struct xt_mtchk_param
*par
)
651 struct net
*net
= par
->net
;
652 struct xt_hashlimit_mtinfo1
*info
= par
->matchinfo
;
655 if (info
->cfg
.gc_interval
== 0 || info
->cfg
.expire
== 0)
657 if (info
->name
[sizeof(info
->name
)-1] != '\0')
659 if (par
->family
== NFPROTO_IPV4
) {
660 if (info
->cfg
.srcmask
> 32 || info
->cfg
.dstmask
> 32)
663 if (info
->cfg
.srcmask
> 128 || info
->cfg
.dstmask
> 128)
667 if (info
->cfg
.mode
& ~XT_HASHLIMIT_ALL
) {
668 pr_info("Unknown mode mask %X, kernel too old?\n",
673 /* Check for overflow. */
674 if (info
->cfg
.mode
& XT_HASHLIMIT_BYTES
) {
675 if (user2credits_byte(info
->cfg
.avg
) == 0) {
676 pr_info("overflow, rate too high: %u\n", info
->cfg
.avg
);
679 } else if (info
->cfg
.burst
== 0 ||
680 user2credits(info
->cfg
.avg
* info
->cfg
.burst
) <
681 user2credits(info
->cfg
.avg
)) {
682 pr_info("overflow, try lower: %u/%u\n",
683 info
->cfg
.avg
, info
->cfg
.burst
);
687 mutex_lock(&hashlimit_mutex
);
688 info
->hinfo
= htable_find_get(net
, info
->name
, par
->family
);
689 if (info
->hinfo
== NULL
) {
690 ret
= htable_create(net
, info
, par
->family
);
692 mutex_unlock(&hashlimit_mutex
);
696 mutex_unlock(&hashlimit_mutex
);
700 static void hashlimit_mt_destroy(const struct xt_mtdtor_param
*par
)
702 const struct xt_hashlimit_mtinfo1
*info
= par
->matchinfo
;
704 htable_put(info
->hinfo
);
707 static struct xt_match hashlimit_mt_reg
[] __read_mostly
= {
711 .family
= NFPROTO_IPV4
,
712 .match
= hashlimit_mt
,
713 .matchsize
= sizeof(struct xt_hashlimit_mtinfo1
),
714 .checkentry
= hashlimit_mt_check
,
715 .destroy
= hashlimit_mt_destroy
,
718 #if IS_ENABLED(CONFIG_IP6_NF_IPTABLES)
722 .family
= NFPROTO_IPV6
,
723 .match
= hashlimit_mt
,
724 .matchsize
= sizeof(struct xt_hashlimit_mtinfo1
),
725 .checkentry
= hashlimit_mt_check
,
726 .destroy
= hashlimit_mt_destroy
,
733 static void *dl_seq_start(struct seq_file
*s
, loff_t
*pos
)
734 __acquires(htable
->lock
)
736 struct xt_hashlimit_htable
*htable
= s
->private;
737 unsigned int *bucket
;
739 spin_lock_bh(&htable
->lock
);
740 if (*pos
>= htable
->cfg
.size
)
743 bucket
= kmalloc(sizeof(unsigned int), GFP_ATOMIC
);
745 return ERR_PTR(-ENOMEM
);
751 static void *dl_seq_next(struct seq_file
*s
, void *v
, loff_t
*pos
)
753 struct xt_hashlimit_htable
*htable
= s
->private;
754 unsigned int *bucket
= (unsigned int *)v
;
757 if (*pos
>= htable
->cfg
.size
) {
764 static void dl_seq_stop(struct seq_file
*s
, void *v
)
765 __releases(htable
->lock
)
767 struct xt_hashlimit_htable
*htable
= s
->private;
768 unsigned int *bucket
= (unsigned int *)v
;
772 spin_unlock_bh(&htable
->lock
);
775 static int dl_seq_real_show(struct dsthash_ent
*ent
, u_int8_t family
,
779 const struct xt_hashlimit_htable
*ht
= s
->private;
781 spin_lock(&ent
->lock
);
782 /* recalculate to show accurate numbers */
783 rateinfo_recalc(ent
, jiffies
, ht
->cfg
.mode
);
787 res
= seq_printf(s
, "%ld %pI4:%u->%pI4:%u %u %u %u\n",
788 (long)(ent
->expires
- jiffies
)/HZ
,
790 ntohs(ent
->dst
.src_port
),
792 ntohs(ent
->dst
.dst_port
),
793 ent
->rateinfo
.credit
, ent
->rateinfo
.credit_cap
,
796 #if IS_ENABLED(CONFIG_IP6_NF_IPTABLES)
798 res
= seq_printf(s
, "%ld %pI6:%u->%pI6:%u %u %u %u\n",
799 (long)(ent
->expires
- jiffies
)/HZ
,
801 ntohs(ent
->dst
.src_port
),
803 ntohs(ent
->dst
.dst_port
),
804 ent
->rateinfo
.credit
, ent
->rateinfo
.credit_cap
,
812 spin_unlock(&ent
->lock
);
816 static int dl_seq_show(struct seq_file
*s
, void *v
)
818 struct xt_hashlimit_htable
*htable
= s
->private;
819 unsigned int *bucket
= (unsigned int *)v
;
820 struct dsthash_ent
*ent
;
821 struct hlist_node
*pos
;
823 if (!hlist_empty(&htable
->hash
[*bucket
])) {
824 hlist_for_each_entry(ent
, pos
, &htable
->hash
[*bucket
], node
)
825 if (dl_seq_real_show(ent
, htable
->family
, s
))
831 static const struct seq_operations dl_seq_ops
= {
832 .start
= dl_seq_start
,
838 static int dl_proc_open(struct inode
*inode
, struct file
*file
)
840 int ret
= seq_open(file
, &dl_seq_ops
);
843 struct seq_file
*sf
= file
->private_data
;
844 sf
->private = PDE(inode
)->data
;
849 static const struct file_operations dl_file_ops
= {
850 .owner
= THIS_MODULE
,
851 .open
= dl_proc_open
,
854 .release
= seq_release
857 static int __net_init
hashlimit_proc_net_init(struct net
*net
)
859 struct hashlimit_net
*hashlimit_net
= hashlimit_pernet(net
);
861 hashlimit_net
->ipt_hashlimit
= proc_mkdir("ipt_hashlimit", net
->proc_net
);
862 if (!hashlimit_net
->ipt_hashlimit
)
864 #if IS_ENABLED(CONFIG_IP6_NF_IPTABLES)
865 hashlimit_net
->ip6t_hashlimit
= proc_mkdir("ip6t_hashlimit", net
->proc_net
);
866 if (!hashlimit_net
->ip6t_hashlimit
) {
867 proc_net_remove(net
, "ipt_hashlimit");
874 static void __net_exit
hashlimit_proc_net_exit(struct net
*net
)
876 proc_net_remove(net
, "ipt_hashlimit");
877 #if IS_ENABLED(CONFIG_IP6_NF_IPTABLES)
878 proc_net_remove(net
, "ip6t_hashlimit");
882 static int __net_init
hashlimit_net_init(struct net
*net
)
884 struct hashlimit_net
*hashlimit_net
= hashlimit_pernet(net
);
886 INIT_HLIST_HEAD(&hashlimit_net
->htables
);
887 return hashlimit_proc_net_init(net
);
890 static void __net_exit
hashlimit_net_exit(struct net
*net
)
892 struct hashlimit_net
*hashlimit_net
= hashlimit_pernet(net
);
894 BUG_ON(!hlist_empty(&hashlimit_net
->htables
));
895 hashlimit_proc_net_exit(net
);
898 static struct pernet_operations hashlimit_net_ops
= {
899 .init
= hashlimit_net_init
,
900 .exit
= hashlimit_net_exit
,
901 .id
= &hashlimit_net_id
,
902 .size
= sizeof(struct hashlimit_net
),
905 static int __init
hashlimit_mt_init(void)
909 err
= register_pernet_subsys(&hashlimit_net_ops
);
912 err
= xt_register_matches(hashlimit_mt_reg
,
913 ARRAY_SIZE(hashlimit_mt_reg
));
918 hashlimit_cachep
= kmem_cache_create("xt_hashlimit",
919 sizeof(struct dsthash_ent
), 0, 0,
921 if (!hashlimit_cachep
) {
922 pr_warning("unable to create slab cache\n");
928 xt_unregister_matches(hashlimit_mt_reg
, ARRAY_SIZE(hashlimit_mt_reg
));
930 unregister_pernet_subsys(&hashlimit_net_ops
);
935 static void __exit
hashlimit_mt_exit(void)
937 xt_unregister_matches(hashlimit_mt_reg
, ARRAY_SIZE(hashlimit_mt_reg
));
938 unregister_pernet_subsys(&hashlimit_net_ops
);
941 kmem_cache_destroy(hashlimit_cachep
);
944 module_init(hashlimit_mt_init
);
945 module_exit(hashlimit_mt_exit
);