1 /* SPDX-License-Identifier: GPL-2.0-only */
2 /* Copyright (C) 2013 Jozsef Kadlecsik <kadlec@netfilter.org> */
4 #ifndef __IP_SET_BITMAP_IP_GEN_H
5 #define __IP_SET_BITMAP_IP_GEN_H
7 #include <linux/rcupdate_wait.h>
9 #define mtype_do_test IPSET_TOKEN(MTYPE, _do_test)
10 #define mtype_gc_test IPSET_TOKEN(MTYPE, _gc_test)
11 #define mtype_is_filled IPSET_TOKEN(MTYPE, _is_filled)
12 #define mtype_do_add IPSET_TOKEN(MTYPE, _do_add)
13 #define mtype_ext_cleanup IPSET_TOKEN(MTYPE, _ext_cleanup)
14 #define mtype_do_del IPSET_TOKEN(MTYPE, _do_del)
15 #define mtype_do_list IPSET_TOKEN(MTYPE, _do_list)
16 #define mtype_do_head IPSET_TOKEN(MTYPE, _do_head)
17 #define mtype_adt_elem IPSET_TOKEN(MTYPE, _adt_elem)
18 #define mtype_add_timeout IPSET_TOKEN(MTYPE, _add_timeout)
19 #define mtype_gc_init IPSET_TOKEN(MTYPE, _gc_init)
20 #define mtype_kadt IPSET_TOKEN(MTYPE, _kadt)
21 #define mtype_uadt IPSET_TOKEN(MTYPE, _uadt)
22 #define mtype_destroy IPSET_TOKEN(MTYPE, _destroy)
23 #define mtype_memsize IPSET_TOKEN(MTYPE, _memsize)
24 #define mtype_flush IPSET_TOKEN(MTYPE, _flush)
25 #define mtype_head IPSET_TOKEN(MTYPE, _head)
26 #define mtype_same_set IPSET_TOKEN(MTYPE, _same_set)
27 #define mtype_elem IPSET_TOKEN(MTYPE, _elem)
28 #define mtype_test IPSET_TOKEN(MTYPE, _test)
29 #define mtype_add IPSET_TOKEN(MTYPE, _add)
30 #define mtype_del IPSET_TOKEN(MTYPE, _del)
31 #define mtype_list IPSET_TOKEN(MTYPE, _list)
32 #define mtype_gc IPSET_TOKEN(MTYPE, _gc)
35 #define get_ext(set, map, id) ((map)->extensions + ((set)->dsize * (id)))
38 mtype_gc_init(struct ip_set *set, void (*gc)(struct timer_list *t))
40 struct mtype *map = set->data;
42 timer_setup(&map->gc, gc, 0);
43 mod_timer(&map->gc, jiffies + IPSET_GC_PERIOD(set->timeout) * HZ);
47 mtype_ext_cleanup(struct ip_set *set)
49 struct mtype *map = set->data;
52 for (id = 0; id < map->elements; id++)
53 if (test_bit(id, map->members))
54 ip_set_ext_destroy(set, get_ext(set, map, id));
58 mtype_destroy(struct ip_set *set)
60 struct mtype *map = set->data;
62 if (SET_WITH_TIMEOUT(set))
63 del_timer_sync(&map->gc);
65 if (set->dsize && set->extensions & IPSET_EXT_DESTROY)
66 mtype_ext_cleanup(set);
67 ip_set_free(map->members);
74 mtype_flush(struct ip_set *set)
76 struct mtype *map = set->data;
78 if (set->extensions & IPSET_EXT_DESTROY)
79 mtype_ext_cleanup(set);
80 bitmap_zero(map->members, map->elements);
85 /* Calculate the actual memory size of the set data */
87 mtype_memsize(const struct mtype *map, size_t dsize)
89 return sizeof(*map) + map->memsize +
90 map->elements * dsize;
94 mtype_head(struct ip_set *set, struct sk_buff *skb)
96 const struct mtype *map = set->data;
97 struct nlattr *nested;
98 size_t memsize = mtype_memsize(map, set->dsize) + set->ext_size;
100 nested = nla_nest_start(skb, IPSET_ATTR_DATA);
102 goto nla_put_failure;
103 if (mtype_do_head(skb, map) ||
104 nla_put_net32(skb, IPSET_ATTR_REFERENCES, htonl(set->ref)) ||
105 nla_put_net32(skb, IPSET_ATTR_MEMSIZE, htonl(memsize)) ||
106 nla_put_net32(skb, IPSET_ATTR_ELEMENTS, htonl(set->elements)))
107 goto nla_put_failure;
108 if (unlikely(ip_set_put_flags(skb, set)))
109 goto nla_put_failure;
110 nla_nest_end(skb, nested);
118 mtype_test(struct ip_set *set, void *value, const struct ip_set_ext *ext,
119 struct ip_set_ext *mext, u32 flags)
121 struct mtype *map = set->data;
122 const struct mtype_adt_elem *e = value;
123 void *x = get_ext(set, map, e->id);
124 int ret = mtype_do_test(e, map, set->dsize);
128 return ip_set_match_extensions(set, ext, mext, flags, x);
132 mtype_add(struct ip_set *set, void *value, const struct ip_set_ext *ext,
133 struct ip_set_ext *mext, u32 flags)
135 struct mtype *map = set->data;
136 const struct mtype_adt_elem *e = value;
137 void *x = get_ext(set, map, e->id);
138 int ret = mtype_do_add(e, map, flags, set->dsize);
140 if (ret == IPSET_ADD_FAILED) {
141 if (SET_WITH_TIMEOUT(set) &&
142 ip_set_timeout_expired(ext_timeout(x, set))) {
145 } else if (!(flags & IPSET_FLAG_EXIST)) {
146 set_bit(e->id, map->members);
147 return -IPSET_ERR_EXIST;
149 /* Element is re-added, cleanup extensions */
150 ip_set_ext_destroy(set, x);
155 if (SET_WITH_TIMEOUT(set))
156 #ifdef IP_SET_BITMAP_STORED_TIMEOUT
157 mtype_add_timeout(ext_timeout(x, set), e, ext, set, map, ret);
159 ip_set_timeout_set(ext_timeout(x, set), ext->timeout);
162 if (SET_WITH_COUNTER(set))
163 ip_set_init_counter(ext_counter(x, set), ext);
164 if (SET_WITH_COMMENT(set))
165 ip_set_init_comment(set, ext_comment(x, set), ext);
166 if (SET_WITH_SKBINFO(set))
167 ip_set_init_skbinfo(ext_skbinfo(x, set), ext);
169 /* Activate element */
170 set_bit(e->id, map->members);
177 mtype_del(struct ip_set *set, void *value, const struct ip_set_ext *ext,
178 struct ip_set_ext *mext, u32 flags)
180 struct mtype *map = set->data;
181 const struct mtype_adt_elem *e = value;
182 void *x = get_ext(set, map, e->id);
184 if (mtype_do_del(e, map))
185 return -IPSET_ERR_EXIST;
187 ip_set_ext_destroy(set, x);
189 if (SET_WITH_TIMEOUT(set) &&
190 ip_set_timeout_expired(ext_timeout(x, set)))
191 return -IPSET_ERR_EXIST;
196 #ifndef IP_SET_BITMAP_STORED_TIMEOUT
198 mtype_is_filled(const struct mtype_elem *x)
205 mtype_list(const struct ip_set *set,
206 struct sk_buff *skb, struct netlink_callback *cb)
208 struct mtype *map = set->data;
209 struct nlattr *adt, *nested;
211 u32 id, first = cb->args[IPSET_CB_ARG0];
214 adt = nla_nest_start(skb, IPSET_ATTR_ADT);
217 /* Extensions may be replaced */
219 for (; cb->args[IPSET_CB_ARG0] < map->elements;
220 cb->args[IPSET_CB_ARG0]++) {
222 id = cb->args[IPSET_CB_ARG0];
223 x = get_ext(set, map, id);
224 if (!test_bit(id, map->members) ||
225 (SET_WITH_TIMEOUT(set) &&
226 #ifdef IP_SET_BITMAP_STORED_TIMEOUT
227 mtype_is_filled(x) &&
229 ip_set_timeout_expired(ext_timeout(x, set))))
231 nested = nla_nest_start(skb, IPSET_ATTR_DATA);
234 nla_nest_cancel(skb, adt);
239 goto nla_put_failure;
241 if (mtype_do_list(skb, map, id, set->dsize))
242 goto nla_put_failure;
243 if (ip_set_put_extensions(skb, set, x, mtype_is_filled(x)))
244 goto nla_put_failure;
245 nla_nest_end(skb, nested);
247 nla_nest_end(skb, adt);
249 /* Set listing finished */
250 cb->args[IPSET_CB_ARG0] = 0;
255 nla_nest_cancel(skb, nested);
256 if (unlikely(id == first)) {
257 cb->args[IPSET_CB_ARG0] = 0;
260 nla_nest_end(skb, adt);
267 mtype_gc(struct timer_list *t)
269 struct mtype *map = from_timer(map, t, gc);
270 struct ip_set *set = map->set;
274 /* We run parallel with other readers (test element)
275 * but adding/deleting new entries is locked out
277 spin_lock_bh(&set->lock);
278 for (id = 0; id < map->elements; id++)
279 if (mtype_gc_test(id, map, set->dsize)) {
280 x = get_ext(set, map, id);
281 if (ip_set_timeout_expired(ext_timeout(x, set))) {
282 clear_bit(id, map->members);
283 ip_set_ext_destroy(set, x);
287 spin_unlock_bh(&set->lock);
289 map->gc.expires = jiffies + IPSET_GC_PERIOD(set->timeout) * HZ;
293 static const struct ip_set_type_variant mtype = {
297 [IPSET_ADD] = mtype_add,
298 [IPSET_DEL] = mtype_del,
299 [IPSET_TEST] = mtype_test,
301 .destroy = mtype_destroy,
302 .flush = mtype_flush,
305 .same_set = mtype_same_set,
308 #endif /* __IP_SET_BITMAP_IP_GEN_H */