1 files changed, 22 insertions, 0 deletions

diff --git a/tools/perf/util/db-export.c b/tools/perf/util/db-export.c
index 7123746edcf4..69fbb0a72d0c 100644
--- a/tools/perf/util/db-export.c
+++ b/tools/perf/util/db-export.c
@@ -463,6 +463,28 @@ int db_export__branch_types(struct db_export *dbe)
                 if (err)
                         break;
         }
+
+        /* Add trace begin / end variants */
+        for (i = 0; branch_types[i].name ; i++) {
+                const char *name = branch_types[i].name;
+                u32 type = branch_types[i].branch_type;
+                char buf[64];
+
+                if (type == PERF_IP_FLAG_BRANCH ||
+                    (type & (PERF_IP_FLAG_TRACE_BEGIN | PERF_IP_FLAG_TRACE_END)))
+                        continue;
+
+                snprintf(buf, sizeof(buf), "trace begin / %s", name);
+                err = db_export__branch_type(dbe, type | PERF_IP_FLAG_TRACE_BEGIN, buf);
+                if (err)
+                        break;
+
+                snprintf(buf, sizeof(buf), "%s / trace end", name);
+                err = db_export__branch_type(dbe, type | PERF_IP_FLAG_TRACE_END, buf);
+                if (err)
+                        break;
+        }
+
         return err;
 }