X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=urcu.c;h=ee79d1efc7b7a98f87f30a3f5de4abe9e7c822b3;hb=b257a10b0b05064dcb5dc9d4d1c5f95c0364b9dd;hp=c55a5a2be9395ad691876e8e1b58d787729207d4;hpb=f69f195a06af55b7501ed2f59ed719970727ce5b;p=urcu.git

diff --git a/urcu.c b/urcu.c
index c55a5a2..ee79d1e 100644
--- a/urcu.c
+++ b/urcu.c
@@ -1,3 +1,13 @@
+/*
+ * urcu.c
+ *
+ * Userspace RCU library
+ *
+ * Copyright February 2009 - Mathieu Desnoyers <mathieu.desnoyers@polymtl.ca>
+ *
+ * Distributed under GPLv2
+ */
+
 #include <stdio.h>
 #include <pthread.h>
 #include <signal.h>
@@ -26,6 +36,27 @@ static struct reader_data *reader_data;
 static int num_readers, alloc_readers;
 static int sig_done;
 
+void rcu_write_lock(void)
+{
+	int ret;
+	ret = pthread_mutex_lock(&urcu_mutex);
+	if (ret) {
+		perror("Error in pthread mutex lock");
+		exit(-1);
+	}
+}
+
+void rcu_write_unlock(void)
+{
+	int ret;
+
+	ret = pthread_mutex_unlock(&urcu_mutex);
+	if (ret) {
+		perror("Error in pthread mutex unlock");
+		exit(-1);
+	}
+}
+
 /*
  * called with urcu_mutex held.
  */
@@ -84,18 +115,13 @@ void wait_for_quiescent_state(int parity)
 
 /*
  * Return old pointer, OK to free, no more reference exist.
+ * Called under rcu_write_lock.
  */
 void *urcu_publish_content(void **ptr, void *new)
 {
 	int ret, prev_parity;
 	void *oldptr;
 
-	ret = pthread_mutex_lock(&urcu_mutex);
-	if (ret) {
-		perror("Error in pthread mutex lock");
-		exit(-1);
-	}
-
 	/*
 	 * We can publish the new pointer before we change the current qparity.
 	 * Readers seeing the new pointer while being in the previous qparity
@@ -121,11 +147,6 @@ void *urcu_publish_content(void **ptr, void *new)
 	 * Deleting old data is ok !
 	 */
 	
-	ret = pthread_mutex_unlock(&urcu_mutex);
-	if (ret) {
-		perror("Error in pthread mutex lock");
-		exit(-1);
-	}
 	return oldptr;
 }
 
@@ -179,44 +200,17 @@ void urcu_remove_reader(pthread_t id)
 
 void urcu_register_thread(void)
 {
-	pthread_t self = pthread_self();
-	int ret;
-
-	ret = pthread_mutex_lock(&urcu_mutex);
-	if (ret) {
-		perror("Error in pthread mutex lock");
-		exit(-1);
-	}
-
-	urcu_add_reader(self);
-
-
-	ret = pthread_mutex_unlock(&urcu_mutex);
-	if (ret) {
-		perror("Error in pthread mutex unlock");
-		exit(-1);
-	}
+	rcu_write_lock();
+	urcu_add_reader(pthread_self());
+	rcu_write_unlock();
 }
 
 void urcu_unregister_thread(void)
 {
 	pthread_t self = pthread_self();
-	int ret;
-
-	ret = pthread_mutex_lock(&urcu_mutex);
-	if (ret) {
-		perror("Error in pthread mutex lock");
-		exit(-1);
-	}
-
-	urcu_remove_reader(self);
-
-	ret = pthread_mutex_unlock(&urcu_mutex);
-	if (ret) {
-		perror("Error in pthread mutex unlock");
-		exit(-1);
-	}
-
+	rcu_write_lock();
+	urcu_remove_reader(pthread_self());
+	rcu_write_unlock();
 }
 
 void sigurcu_handler(int signo, siginfo_t *siginfo, void *context)