trappy/systrace: Provide custom data string parsing

Here we provide custom data string parsing for Android userspace events.
Borrowed the regex and parsing logic from Patrick's patch.

Co-developed-by: Patrick Bellasi <patrick.bellasi@arm.com>
Signed-off-by: Joel Fernandes <joelaf@google.com>
Reviewed-by: KP Singh <kpsingh@google.com>

1 files changed, 17 insertions, 0 deletions

diff --git a/trappy/systrace.py b/trappy/systrace.py
index 6e917a6..e18abf8 100644
--- a/trappy/systrace.py
+++ b/trappy/systrace.py
@@ -14,6 +14,10 @@
 #
 
 from trappy.ftrace import GenericFTrace
+import re
+
+SYSTRACE_EVENT = re.compile(
+    r'^(?P<event>[A-Z])(\|(?P<pid>\d+)\|(?P<func>.*)(\|(?P<data>\d+))?)?')
 
 class drop_before_trace(object):
     """Object that, when called, returns True if the line is not part of
@@ -75,3 +79,16 @@ class SysTrace(GenericFTrace):
 
         """
         return lambda x: not x.endswith("</script>\n")
+
+    def generate_data_dict(self, data_str):
+        """ Custom parsing for systrace's userspace events """
+        data_dict = None
+
+        match = SYSTRACE_EVENT.match(data_str)
+        if match:
+            data_dict = { 'event': match.group('event'),
+                          'pid'  : match.group('pid'),
+                          'func' : match.group('func'),
+                          'data' : match.group('data') }
+
+        return data_dict