include/linux/netfilter/ipset/ip_set_comment.h

   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*
  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
  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() */
  47 static inline int
  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(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
  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