X-Git-Url: https://git.lttng.org/?a=blobdiff_plain;f=urcu.h;h=c77b26f958e34b8521f1b8363d83b6492cb46b0f;hb=9d335088c7c7eb6219e32ec9426d336f3a211a77;hp=3c5b178d1a8254adb37331fea6b61698a58c459c;hpb=5e7e64b952ac49cf61c4c3bfc02e5938c3691d94;p=urcu.git

diff --git a/urcu.h b/urcu.h
index 3c5b178..c77b26f 100644
--- a/urcu.h
+++ b/urcu.h
@@ -66,6 +66,48 @@ static inline void atomic_inc(int *v)
 
 #define SIGURCU SIGUSR1
 
+#ifdef DEBUG_YIELD
+#include <sched.h>
+
+#define YIELD_READ 	(1 << 0)
+#define YIELD_WRITE	(1 << 1)
+
+extern unsigned int yield_active;
+extern unsigned int __thread rand_yield;
+
+static inline void debug_yield_read(void)
+{
+	if (yield_active & YIELD_READ)
+		if (rand_r(&rand_yield) & 0x1)
+			sched_yield();
+}
+
+static inline void debug_yield_write(void)
+{
+	if (yield_active & YIELD_WRITE)
+		if (rand_r(&rand_yield) & 0x1)
+			sched_yield();
+}
+
+static inline void debug_yield_init(void)
+{
+	rand_yield = time(NULL) ^ pthread_self();
+}
+#else
+static inline void debug_yield_read(void)
+{
+}
+
+static inline void debug_yield_write(void)
+{
+}
+
+static inline void debug_yield_init(void)
+{
+
+}
+#endif
+
 /* Global quiescent period parity */
 extern int urcu_qparity;
 
@@ -77,40 +119,37 @@ static inline int get_urcu_qparity(void)
 }
 
 /*
- * returns urcu_parity.
+ * urcu_parity should be declared on the caller's stack.
  */
-static inline int rcu_read_lock(void)
+static inline void rcu_read_lock(int *urcu_parity)
 {
-	int urcu_parity = get_urcu_qparity();
-	urcu_active_readers[urcu_parity]++;
+	debug_yield_read();
+	*urcu_parity = get_urcu_qparity();
+	debug_yield_read();
+	urcu_active_readers[*urcu_parity]++;
+	debug_yield_read();
 	/*
 	 * Increment active readers count before accessing the pointer.
 	 * See force_mb_all_threads().
 	 */
 	barrier();
-	return urcu_parity;
+	debug_yield_read();
 }
 
-static inline void rcu_read_unlock(int urcu_parity)
+static inline void rcu_read_unlock(int *urcu_parity)
 {
+	debug_yield_read();
 	barrier();
+	debug_yield_read();
 	/*
 	 * Finish using rcu before decrementing the pointer.
 	 * See force_mb_all_threads().
 	 */
-	urcu_active_readers[urcu_parity]--;
+	urcu_active_readers[*urcu_parity]--;
+	debug_yield_read();
 }
 
-extern void rcu_write_lock(void);
-extern void rcu_write_unlock(void);
-
-extern void *_urcu_publish_content(void **ptr, void *new);
-
-/*
- * gcc does not like automatic &struct ... * -> void **.
- * Remove the warning. (hopefully this is ok)
- */
-#define urcu_publish_content(ptr, new) _urcu_publish_content((void **)ptr, new)
+extern void *urcu_publish_content(void **ptr, void *new);
 
 /*
  * Reader thread registration.