sys/sys/ref.h

   1 /*
   2  * Copyright (c) 2010, Venkatesh Srinivas <me@endeavour.zapto.org>
   3  *
   4  * Permission to use, copy, modify, and distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #ifndef _SYS_REF_H_
  18 #define _SYS_REF_H_
  19
  20 struct kref {
  21         int refcount;
  22 };
  23
  24 void kref_init(struct kref *ref, int count);
  25 void kref_inc(struct kref *ref);
  26 int kref_dec(struct kref *ref, void (*deconstruct)(void*, void*),
  27              void *priv1, void *priv2);
  28
  29 #define KREF_DEC(refp, deconstruct)                                     \
  30         ({                                                              \
  31                 int _val = atomic_fetchadd_int(&(refp)->refcount, -1);  \
  32                 if (_val == 1)                                          \
  33                         deconstruct;                                    \
  34                 (_val != 1);                                            \
  35         })
  36
  37 #endif
  38