ust: improve message parsing

[ust.git] / libmarkers / marker.c

diff --git a/libmarkers/marker.c b/libmarkers/marker.c
index 198650b32d41846e0b8180c865ab5177e54119d7..2fa723dc49e1cdfdc22b8a566f556881026bb566 100644 (file)
--- a/libmarkers/marker.c
+++ b/libmarkers/marker.c
@@ -817,6 +817,7 @@ int marker_probe_register(const char *channel, const char *name,
        /* write rcu_pending before calling the RCU callback */
        smp_wmb();
        call_rcu_sched(&entry->rcu, free_old_closure);
+       /*synchronize_rcu(); free_old_closure();*/
        goto end;
 
 error_unregister_channel:
@@ -1470,6 +1471,23 @@ void lib_update_markers(void)
 //ust//        mutex_unlock(&module_mutex);
 }
 
+static void (*new_marker_cb)(struct marker *) = NULL;
+
+void marker_set_new_marker_cb(void (*cb)(struct marker *))
+{
+       new_marker_cb = cb;
+}
+
+static void new_markers(struct marker *start, struct marker *end)
+{
+       if(new_marker_cb) {
+               struct marker *m;
+               for(m=start; m < end; m++) {
+                       new_marker_cb(m);
+               }
+       }
+}
+
 int marker_register_lib(struct marker *markers_start, int markers_count)
 {
        struct lib *pl;
@@ -1479,16 +1497,29 @@ int marker_register_lib(struct marker *markers_start, int markers_count)
        pl->markers_start = markers_start;
        pl->markers_count = markers_count;
 
+       /* FIXME: maybe protect this with its own mutex? */
+       lock_markers();
        list_add(&pl->list, &libs);
+       unlock_markers();
+
+       new_markers(markers_start, markers_start + markers_count);
 
        /* FIXME: update just the loaded lib */
        lib_update_markers();
 
-       printf("just registered a markers section from %p and having %d markers\n", markers_start, markers_count);
+       DBG("just registered a markers section from %p and having %d markers", markers_start, markers_count);
        
        return 0;
 }
 
+int marker_unregister_lib(struct marker *markers_start, int markers_count)
+{
+       /*FIXME: implement; but before implementing, marker_register_lib must
+          have appropriate locking. */
+
+       return 0;
+}
+
 static int initialized = 0;
 
 void __attribute__((constructor)) init_markers(void)