1 files changed, 56 insertions, 0 deletions

diff --git a/tools/perf/util/session.c b/tools/perf/util/session.c
index 6b305fbcc986..f5baff1495e6 100644
--- a/tools/perf/util/session.c
+++ b/tools/perf/util/session.c
@@ -14,6 +14,7 @@
 #include "sort.h"
 #include "util.h"
 #include "cpumap.h"
+#include "event-parse.h"
 
 static int perf_session__open(struct perf_session *self, bool force)
 {
@@ -1610,3 +1611,58 @@ void perf_session__fprintf_info(struct perf_session *session, FILE *fp,
         perf_header__fprintf_info(session, fp, full);
         fprintf(fp, "# ========\n#\n");
 }
+
+
+int __perf_session__set_tracepoints_handlers(struct perf_session *session,
+                                             const struct perf_evsel_str_handler *assocs,
+                                             size_t nr_assocs)
+{
+        struct perf_evlist *evlist = session->evlist;
+        struct event_format *format;
+        struct perf_evsel *evsel;
+        char *tracepoint, *name;
+        size_t i;
+        int err;
+
+        for (i = 0; i < nr_assocs; i++) {
+                err = -ENOMEM;
+                tracepoint = strdup(assocs[i].name);
+                if (tracepoint == NULL)
+                        goto out;
+
+                err = -ENOENT;
+                name = strchr(tracepoint, ':');
+                if (name == NULL)
+                        goto out_free;
+
+                *name++ = '\0';
+                format = pevent_find_event_by_name(session->pevent,
+                                                   tracepoint, name);
+                if (format == NULL) {
+                        /*
+                         * Adding a handler for an event not in the session,
+                         * just ignore it.
+                         */
+                        goto next;
+                }
+
+                evsel = perf_evlist__find_tracepoint_by_id(evlist, format->id);
+                if (evsel == NULL)
+                        goto next;
+
+                err = -EEXIST;
+                if (evsel->handler.func != NULL)
+                        goto out_free;
+                evsel->handler.func = assocs[i].handler;
+next:
+                free(tracepoint);
+        }
+
+        err = 0;
+out:
+        return err;
+
+out_free:
+        free(tracepoint);
+        goto out;
+}