1 #ifndef _IP_SET_COMMENT_H
2 #define _IP_SET_COMMENT_H
3
4 /* Copyright (C) 2013 Oliver Smith <oliver@8.c.9.b.0.7.4.0.1.0.0.2.ip6.arpa>
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License version 2 as
8 * published by the Free Software Foundation.
9 */
10
11 #ifdef __KERNEL__
12
13 static inline char*
ip_set_comment_uget(struct nlattr * tb)14 ip_set_comment_uget(struct nlattr *tb)
15 {
16 return nla_data(tb);
17 }
18
19 /* Called from uadd only, protected by the set spinlock.
20 * The kadt functions don't use the comment extensions in any way.
21 */
22 static inline void
ip_set_init_comment(struct ip_set * set,struct ip_set_comment * comment,const struct ip_set_ext * ext)23 ip_set_init_comment(struct ip_set *set, struct ip_set_comment *comment,
24 const struct ip_set_ext *ext)
25 {
26 struct ip_set_comment_rcu *c = rcu_dereference_protected(comment->c, 1);
27 size_t len = ext->comment ? strlen(ext->comment) : 0;
28
29 if (unlikely(c)) {
30 set->ext_size -= sizeof(*c) + strlen(c->str) + 1;
31 kfree_rcu(c, rcu);
32 rcu_assign_pointer(comment->c, NULL);
33 }
34 if (!len)
35 return;
36 if (unlikely(len > IPSET_MAX_COMMENT_SIZE))
37 len = IPSET_MAX_COMMENT_SIZE;
38 c = kmalloc(sizeof(*c) + len + 1, GFP_ATOMIC);
39 if (unlikely(!c))
40 return;
41 strlcpy(c->str, ext->comment, len + 1);
42 set->ext_size += sizeof(*c) + strlen(c->str) + 1;
43 rcu_assign_pointer(comment->c, c);
44 }
45
46 /* Used only when dumping a set, protected by rcu_read_lock_bh() */
47 static inline int
ip_set_put_comment(struct sk_buff * skb,const struct ip_set_comment * comment)48 ip_set_put_comment(struct sk_buff *skb, const struct ip_set_comment *comment)
49 {
50 struct ip_set_comment_rcu *c = rcu_dereference_bh(comment->c);
51
52 if (!c)
53 return 0;
54 return nla_put_string(skb, IPSET_ATTR_COMMENT, c->str);
55 }
56
57 /* Called from uadd/udel, flush or the garbage collectors protected
58 * by the set spinlock.
59 * Called when the set is destroyed and when there can't be any user
60 * of the set data anymore.
61 */
62 static inline void
ip_set_comment_free(struct ip_set * set,struct ip_set_comment * comment)63 ip_set_comment_free(struct ip_set *set, struct ip_set_comment *comment)
64 {
65 struct ip_set_comment_rcu *c;
66
67 c = rcu_dereference_protected(comment->c, 1);
68 if (unlikely(!c))
69 return;
70 set->ext_size -= sizeof(*c) + strlen(c->str) + 1;
71 kfree_rcu(c, rcu);
72 rcu_assign_pointer(comment->c, NULL);
73 }
74
75 #endif
76 #endif
77
