perf scripting python: Add sample flags

Add sample flags to python scripting.

Signed-off-by: Adrian Hunter <adrian.hunter@intel.com>
Cc: Andi Kleen <ak@linux.intel.com>
Cc: Jiri Olsa <jolsa@redhat.com>
Link: https://lore.kernel.org/r/20210525095112.1399-6-adrian.hunter@intel.com
Signed-off-by: Arnaldo Carvalho de Melo <acme@redhat.com>

diff --git a/tools/perf/util/scripting-engines/trace-event-python.c b/tools/perf/util/scripting-engines/trace-event-python.c
index a434f4b..5d01e4f 100644 (file)
--- a/tools/perf/util/scripting-engines/trace-event-python.c
+++ b/tools/perf/util/scripting-engines/trace-event-python.c
@@ -742,6 +742,29 @@ static void set_sym_in_dict(PyObject *dict, struct addr_location *al,
        }
 }
 
+static void set_sample_flags(PyObject *dict, u32 flags)
+{
+       const char *ch = PERF_IP_FLAG_CHARS;
+       char *p, str[33];
+
+       for (p = str; *ch; ch++, flags >>= 1) {
+               if (flags & 1)
+                       *p++ = *ch;
+       }
+       *p = 0;
+       pydict_set_item_string_decref(dict, "flags", _PyUnicode_FromString(str));
+}
+
+static void python_process_sample_flags(struct perf_sample *sample, PyObject *dict_sample)
+{
+       char flags_disp[SAMPLE_FLAGS_BUF_SIZE];
+
+       set_sample_flags(dict_sample, sample->flags);
+       perf_sample__sprintf_flags(sample->flags, flags_disp, sizeof(flags_disp));
+       pydict_set_item_string_decref(dict_sample, "flags_disp",
+               _PyUnicode_FromString(flags_disp));
+}
+
 static PyObject *get_perf_sample_dict(struct perf_sample *sample,
                                         struct evsel *evsel,
                                         struct addr_location *al,
@@ -805,6 +828,9 @@ static PyObject *get_perf_sample_dict(struct perf_sample *sample,
                set_sym_in_dict(dict_sample, addr_al, "addr_dso", "addr_symbol", "addr_symoff");
        }
 
+       if (sample->flags)
+               python_process_sample_flags(sample, dict_sample);
+
        set_regs_in_dict(dict, sample, evsel);
 
        return dict;