X-Git-Url: http://git.lttng.org./?a=blobdiff_plain;f=rcuja%2Frcuja-internal.h;h=96e9dabf6225d58ebb91e445c574c8a21ecc3551;hb=7d67da99e337f495c1b4c48636d93fa522bd9dfd;hp=28908e050b06e88d4dc9003487ee4f912596b819;hpb=b4540e8adb5a4d726e9dbcf2388b6e9b628ce990;p=userspace-rcu.git

diff --git a/rcuja/rcuja-internal.h b/rcuja/rcuja-internal.h
index 28908e0..96e9dab 100644
--- a/rcuja/rcuja-internal.h
+++ b/rcuja/rcuja-internal.h
@@ -23,18 +23,50 @@
  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  */
 
+#define _GNU_SOURCE
 #include <pthread.h>
+#include <stdio.h>
+#include <inttypes.h>
+#include <unistd.h>
 #include <urcu/rculfhash.h>
 
+/*
+ * Number of least significant pointer bits reserved to represent the
+ * child type.
+ */
+#define JA_TYPE_BITS	3
+#define JA_TYPE_MAX_NR	(1UL << JA_TYPE_BITS)
+#define JA_TYPE_MASK	(JA_TYPE_MAX_NR - 1)
+#define JA_PTR_MASK	(~JA_TYPE_MASK)
+
+#define JA_ENTRY_PER_NODE	256UL
+#define JA_LOG2_BITS_PER_BYTE	3U
+#define JA_BITS_PER_BYTE	(1U << JA_LOG2_BITS_PER_BYTE)
+
+#define JA_MAX_DEPTH	9	/* Maximum depth, including leafs */
+
+/*
+ * Entry for NULL node is at index 8 of the table. It is never encoded
+ * in flags.
+ */
+#define NODE_INDEX_NULL		8
+
+/*
+ * Number of removals needed on a fallback node before we try to shrink
+ * it.
+ */
+#define JA_FALLBACK_REMOVAL_COUNT	8
+
 /* Never declared. Opaque type used to store flagged node pointers. */
 struct cds_ja_inode_flag;
+struct cds_ja_inode;
 
 /*
  * Shadow node contains mutex and call_rcu head associated with a node.
  */
 struct cds_ja_shadow_node {
 	struct cds_lfht_node ht_node;	/* hash table node */
-	struct cds_ja_inode *node;	/* reverse mapping and hash table key */
+	struct cds_ja_inode_flag *node_flag;	/* reverse mapping and hash table key */
 	/*
 	 * mutual exclusion on all nodes belonging to the same tree
 	 * position (e.g. both nodes before and after recompaction
@@ -43,6 +75,9 @@ struct cds_ja_shadow_node {
 	pthread_mutex_t *lock;
 	unsigned int nr_child;		/* number of children in node */
 	struct rcu_head head;		/* for deferred node and shadow node reclaim */
+	int fallback_removal_count;	/* removals left keeping fallback */
+	int level;			/* level in the tree */
+	struct cds_ja *ja;		/* toplevel judy array */
 };
 
 struct cds_ja {
@@ -55,19 +90,53 @@ struct cds_ja {
 	 * cache footprint, especially for very small nodes.
 	 */
 	struct cds_lfht *ht;
+	unsigned long nr_fallback;	/* Number of fallback nodes used */
 };
 
+static inline
+struct cds_ja_inode_flag *ja_node_flag(struct cds_ja_inode *node,
+		unsigned long type)
+{
+	assert(type < (1UL << JA_TYPE_BITS));
+	return (struct cds_ja_inode_flag *) (((unsigned long) node) | type);
+}
+
+static inline
+struct cds_ja_inode *ja_node_ptr(struct cds_ja_inode_flag *node)
+{
+	return (struct cds_ja_inode *) (((unsigned long) node) & JA_PTR_MASK);
+}
+
+static inline
+unsigned long ja_node_type(struct cds_ja_inode_flag *node)
+{
+	unsigned long type;
+
+	if (ja_node_ptr(node) == NULL) {
+		return NODE_INDEX_NULL;
+	}
+	type = (unsigned int) ((unsigned long) node & JA_TYPE_MASK);
+	assert(type < (1UL << JA_TYPE_BITS));
+	return type;
+}
+
+__attribute__((visibility("protected")))
+void rcuja_free_all_children(struct cds_ja_shadow_node *shadow_node,
+		struct cds_ja_inode_flag *node_flag,
+		void (*free_node_cb)(struct rcu_head *head));
+
 __attribute__((visibility("protected")))
 struct cds_ja_shadow_node *rcuja_shadow_lookup_lock(struct cds_lfht *ht,
-		struct cds_ja_inode *node);
+		struct cds_ja_inode_flag *node_flag);
 
 __attribute__((visibility("protected")))
 void rcuja_shadow_unlock(struct cds_ja_shadow_node *shadow_node);
 
 __attribute__((visibility("protected")))
-int rcuja_shadow_set(struct cds_lfht *ht,
-		struct cds_ja_inode *new_node,
-		struct cds_ja_shadow_node *inherit_from);
+struct cds_ja_shadow_node *rcuja_shadow_set(struct cds_lfht *ht,
+		struct cds_ja_inode_flag *new_node_flag,
+		struct cds_ja_shadow_node *inherit_from,
+		struct cds_ja *ja);
 
 /* rcuja_shadow_clear flags */
 enum {
@@ -77,12 +146,14 @@ enum {
 
 __attribute__((visibility("protected")))
 int rcuja_shadow_clear(struct cds_lfht *ht,
-		struct cds_ja_inode *node,
+		struct cds_ja_inode_flag *node_flag,
+		struct cds_ja_shadow_node *shadow_node,
 		unsigned int flags);
 
 __attribute__((visibility("protected")))
 void rcuja_shadow_prune(struct cds_lfht *ht,
-		unsigned int flags);
+		unsigned int flags,
+		void (*free_node_cb)(struct rcu_head *head));
 
 __attribute__((visibility("protected")))
 struct cds_lfht *rcuja_create_ht(const struct rcu_flavor_struct *flavor);
@@ -90,4 +161,41 @@ struct cds_lfht *rcuja_create_ht(const struct rcu_flavor_struct *flavor);
 __attribute__((visibility("protected")))
 int rcuja_delete_ht(struct cds_lfht *ht);
 
+//#define DEBUG
+
+#ifdef __linux__
+#include <syscall.h>
+#endif
+
+#if defined(_syscall0)
+_syscall0(pid_t, gettid)
+#elif defined(__NR_gettid)
+static inline pid_t gettid(void)
+{
+	return syscall(__NR_gettid);
+}
+#else
+#warning "use pid as tid"
+static inline pid_t gettid(void)
+{
+	return getpid();
+}
+#endif
+
+#ifdef DEBUG
+#define dbg_printf(fmt, args...)				\
+	fprintf(stderr, "[debug rcuja %lu %s()@%s:%u] " fmt,	\
+		(unsigned long) gettid(), __func__,		\
+		__FILE__, __LINE__, ## args)
+#else
+#define dbg_printf(fmt, args...)				\
+do {								\
+	/* do nothing but check printf format */		\
+	if (0)							\
+		fprintf(stderr, "[debug rcuja %lu %s()@%s:%u] " fmt, \
+			(unsigned long) gettid(), __func__,	\
+			__FILE__, __LINE__, ## args);		\
+} while (0)
+#endif
+
 #endif /* _URCU_RCUJA_INTERNAL_H */