X-Git-Url: http://git.liburcu.org/?a=blobdiff_plain;f=libust%2Fmarker.c;h=b543b166bcd52bd26928da736f69481e9b73c03d;hb=aa08b4413291fabcbd1b1144377d37034ad361de;hp=8690d2b1151e0a5861c2c81327226cb55e46757a;hpb=fbca6b624335eef18c8d86194aeb101a720168f4;p=ust.git

diff --git a/libust/marker.c b/libust/marker.c
index 8690d2b..b543b16 100644
--- a/libust/marker.c
+++ b/libust/marker.c
@@ -43,6 +43,8 @@
 
 extern struct marker __start___markers[] __attribute__((visibility("hidden")));
 extern struct marker __stop___markers[] __attribute__((visibility("hidden")));
+extern struct marker_addr __start___marker_addr[] __attribute__((visibility("hidden")));
+extern struct marker_addr __stop___marker_addr[] __attribute__((visibility("hidden")));
 
 /* Set to 1 to enable marker debug output */
 static const int marker_debug;
@@ -87,7 +89,7 @@ struct marker_entry {
 	char *format;
 	char *name;
 			/* Probe wrapper */
-	void (*call)(const struct marker *mdata, void *call_private, ...);
+	void (*call)(const struct marker *mdata, void *call_private, struct registers *regs, ...);
 	struct marker_probe_closure single;
 	struct marker_probe_closure *multi;
 	int refcount;	/* Number of times armed. 0 if disarmed. */
@@ -123,7 +125,7 @@ static void marker_update_processes(void)
  * operations that modifies the execution flow of preemptible code.
  */
 notrace void __mark_empty_function(const struct marker *mdata,
-	void *probe_private, void *call_private, const char *fmt, va_list *args)
+	void *probe_private, struct registers *regs, void *call_private, const char *fmt, va_list *args)
 {
 }
 //ust// EXPORT_SYMBOL_GPL(__mark_empty_function);
@@ -139,7 +141,7 @@ notrace void __mark_empty_function(const struct marker *mdata,
  * rcu_dereference() for the pointer read.
  */
 notrace void marker_probe_cb(const struct marker *mdata,
-		void *call_private, ...)
+		void *call_private, struct registers *regs, ...)
 {
 	va_list args;
 	char ptype;
@@ -160,8 +162,8 @@ notrace void marker_probe_cb(const struct marker *mdata,
 		/* Must read the ptr before private data. They are not data
 		 * dependant, so we put an explicit smp_rmb() here. */
 		smp_rmb();
-		va_start(args, call_private);
-		func(mdata, mdata->single.probe_private, call_private,
+		va_start(args, regs);
+		func(mdata, mdata->single.probe_private, regs, call_private,
 			mdata->format, &args);
 		va_end(args);
 	} else {
@@ -181,9 +183,9 @@ notrace void marker_probe_cb(const struct marker *mdata,
 		 */
 		smp_read_barrier_depends();
 		for (i = 0; multi[i].func; i++) {
-			va_start(args, call_private);
+			va_start(args, regs);
 			multi[i].func(mdata, multi[i].probe_private,
-				call_private, mdata->format, &args);
+				regs, call_private, mdata->format, &args);
 			va_end(args);
 		}
 	}
@@ -200,7 +202,7 @@ notrace void marker_probe_cb(const struct marker *mdata,
  * Should be connected to markers "MARK_NOARGS".
  */
 static notrace void marker_probe_cb_noarg(const struct marker *mdata,
-		void *call_private, ...)
+		void *call_private, struct registers *regs, ...)
 {
 	va_list args;	/* not initialized */
 	char ptype;
@@ -216,7 +218,7 @@ static notrace void marker_probe_cb_noarg(const struct marker *mdata,
 		/* Must read the ptr before private data. They are not data
 		 * dependant, so we put an explicit smp_rmb() here. */
 		smp_rmb();
-		func(mdata, mdata->single.probe_private, call_private,
+		func(mdata, mdata->single.probe_private, regs, call_private,
 			mdata->format, &args);
 	} else {
 		struct marker_probe_closure *multi;
@@ -235,7 +237,7 @@ static notrace void marker_probe_cb_noarg(const struct marker *mdata,
 		 */
 		smp_read_barrier_depends();
 		for (i = 0; multi[i].func; i++)
-			multi[i].func(mdata, multi[i].probe_private,
+			multi[i].func(mdata, multi[i].probe_private, regs,
 				call_private, mdata->format, &args);
 	}
 //ust//	rcu_read_unlock_sched_notrace();
@@ -823,6 +825,7 @@ int marker_probe_register(const char *channel, const char *name,
 	}
 	mutex_unlock(&markers_mutex);
 
+	/* Activate marker if necessary */
 	marker_update_probes();
 
 	mutex_lock(&markers_mutex);
@@ -1495,15 +1498,22 @@ static void new_markers(struct marker *start, struct marker *end)
 	}
 }
 
-int marker_register_lib(struct marker *markers_start, int markers_count)
+int marker_register_lib(struct marker *markers_start, struct marker_addr *marker_addr_start, int markers_count)
 {
 	struct lib *pl;
+	struct marker_addr *addr;
 
 	pl = (struct lib *) malloc(sizeof(struct lib));
 
 	pl->markers_start = markers_start;
+	pl->markers_addr_start = marker_addr_start;
 	pl->markers_count = markers_count;
 
+	lock_markers();
+	for(addr = marker_addr_start; addr < marker_addr_start + markers_count; addr++)
+		addr->marker->location = addr->addr;
+	unlock_markers();
+
 	/* FIXME: maybe protect this with its own mutex? */
 	lock_markers();
 	list_add(&pl->list, &libs);
@@ -1532,8 +1542,8 @@ static int initialized = 0;
 void __attribute__((constructor)) init_markers(void)
 {
 	if(!initialized) {
-		marker_register_lib(__start___markers, (((long)__stop___markers)-((long)__start___markers))/sizeof(struct marker));
-		DBG("markers_start: %p, markers_stop: %p\n", __start___markers, __stop___markers);
+		marker_register_lib(__start___markers, __start___marker_addr, (((long)__stop___markers)-((long)__start___markers))/sizeof(struct marker));
+		//DBG("markers_start: %p, markers_stop: %p\n", __start___markers, __stop___markers);
 		initialized = 1;
 	}
 }