Add zfs_refcount_transfer_ownership_many()

[mirror_zfs.git] / include / sys / refcount.h

diff --git a/include/sys/refcount.h b/include/sys/refcount.h
index 1752c64e3e8be2dbc636dd7dcaa188526178780d..e982faeba0f29031b2f5ad035da40374bd25f783 100644 (file)
--- a/include/sys/refcount.h
+++ b/include/sys/refcount.h
@@ -20,6 +20,7 @@
  */
 /*
  * Copyright (c) 2005, 2010, Oracle and/or its affiliates. All rights reserved.
+ * Copyright (c) 2012, 2015 by Delphix. All rights reserved.
  */
 
 #ifndef        _SYS_REFCOUNT_H
@@ -38,7 +39,7 @@ extern "C" {
  * particular object, use FTAG (which is a string) for the holder_tag.
  * Otherwise, use the object that holds the reference.
  */
-#define        FTAG ((char *)__func__)
+#define        FTAG ((char *)(uintptr_t)__func__)
 
 #ifdef ZFS_DEBUG
 typedef struct reference {
@@ -50,53 +51,70 @@ typedef struct reference {
 
 typedef struct refcount {
        kmutex_t rc_mtx;
+       boolean_t rc_tracked;
        list_t rc_list;
        list_t rc_removed;
-       int64_t rc_count;
-       int64_t rc_removed_count;
-} refcount_t;
-
-/* Note: refcount_t must be initialized with refcount_create() */
-
-void refcount_create(refcount_t *rc);
-void refcount_destroy(refcount_t *rc);
-void refcount_destroy_many(refcount_t *rc, uint64_t number);
-int refcount_is_zero(refcount_t *rc);
-int64_t refcount_count(refcount_t *rc);
-int64_t refcount_add(refcount_t *rc, void *holder_tag);
-int64_t refcount_remove(refcount_t *rc, void *holder_tag);
-int64_t refcount_add_many(refcount_t *rc, uint64_t number, void *holder_tag);
-int64_t refcount_remove_many(refcount_t *rc, uint64_t number, void *holder_tag);
-void refcount_transfer(refcount_t *dst, refcount_t *src);
-
-void refcount_init(void);
-void refcount_fini(void);
+       uint64_t rc_count;
+       uint64_t rc_removed_count;
+} zfs_refcount_t;
+
+/*
+ * Note: zfs_refcount_t must be initialized with
+ * refcount_create[_untracked]()
+ */
+
+void zfs_refcount_create(zfs_refcount_t *);
+void zfs_refcount_create_untracked(zfs_refcount_t *);
+void zfs_refcount_create_tracked(zfs_refcount_t *);
+void zfs_refcount_destroy(zfs_refcount_t *);
+void zfs_refcount_destroy_many(zfs_refcount_t *, uint64_t);
+int zfs_refcount_is_zero(zfs_refcount_t *);
+int64_t zfs_refcount_count(zfs_refcount_t *);
+int64_t zfs_refcount_add(zfs_refcount_t *, void *);
+int64_t zfs_refcount_remove(zfs_refcount_t *, void *);
+int64_t zfs_refcount_add_many(zfs_refcount_t *, uint64_t, void *);
+int64_t zfs_refcount_remove_many(zfs_refcount_t *, uint64_t, void *);
+void zfs_refcount_transfer(zfs_refcount_t *, zfs_refcount_t *);
+void zfs_refcount_transfer_ownership(zfs_refcount_t *, void *, void *);
+void zfs_refcount_transfer_ownership_many(zfs_refcount_t *, uint64_t,
+    void *, void *);
+boolean_t zfs_refcount_held(zfs_refcount_t *, void *);
+boolean_t zfs_refcount_not_held(zfs_refcount_t *, void *);
+
+void zfs_refcount_init(void);
+void zfs_refcount_fini(void);
 
 #else  /* ZFS_DEBUG */
 
 typedef struct refcount {
        uint64_t rc_count;
-} refcount_t;
-
-#define        refcount_create(rc) ((rc)->rc_count = 0)
-#define        refcount_destroy(rc) ((rc)->rc_count = 0)
-#define        refcount_destroy_many(rc, number) ((rc)->rc_count = 0)
-#define        refcount_is_zero(rc) ((rc)->rc_count == 0)
-#define        refcount_count(rc) ((rc)->rc_count)
-#define        refcount_add(rc, holder) atomic_add_64_nv(&(rc)->rc_count, 1)
-#define        refcount_remove(rc, holder) atomic_add_64_nv(&(rc)->rc_count, -1)
-#define        refcount_add_many(rc, number, holder) \
+} zfs_refcount_t;
+
+#define        zfs_refcount_create(rc) ((rc)->rc_count = 0)
+#define        zfs_refcount_create_untracked(rc) ((rc)->rc_count = 0)
+#define        zfs_refcount_create_tracked(rc) ((rc)->rc_count = 0)
+#define        zfs_refcount_destroy(rc) ((rc)->rc_count = 0)
+#define        zfs_refcount_destroy_many(rc, number) ((rc)->rc_count = 0)
+#define        zfs_refcount_is_zero(rc) ((rc)->rc_count == 0)
+#define        zfs_refcount_count(rc) ((rc)->rc_count)
+#define        zfs_refcount_add(rc, holder) atomic_inc_64_nv(&(rc)->rc_count)
+#define        zfs_refcount_remove(rc, holder) atomic_dec_64_nv(&(rc)->rc_count)
+#define        zfs_refcount_add_many(rc, number, holder) \
        atomic_add_64_nv(&(rc)->rc_count, number)
-#define        refcount_remove_many(rc, number, holder) \
+#define        zfs_refcount_remove_many(rc, number, holder) \
        atomic_add_64_nv(&(rc)->rc_count, -number)
-#define        refcount_transfer(dst, src) { \
+#define        zfs_refcount_transfer(dst, src) { \
        uint64_t __tmp = (src)->rc_count; \
        atomic_add_64(&(src)->rc_count, -__tmp); \
        atomic_add_64(&(dst)->rc_count, __tmp); \
 }
+#define        zfs_refcount_transfer_ownership(rc, ch, nh)             ((void)0)
+#define        zfs_refcount_transfer_ownership_many(rc, nr, ch, nh)    ((void)0)
+#define        zfs_refcount_held(rc, holder)                   ((rc)->rc_count > 0)
+#define        zfs_refcount_not_held(rc, holder)               (B_TRUE)
 
-#define        refcount_init()
-#define        refcount_fini()
+#define        zfs_refcount_init()
+#define        zfs_refcount_fini()
 
 #endif /* ZFS_DEBUG */