1 files changed, 65 insertions, 0 deletions
diff --git a/utils/abtest/mark_aarch64fns.py b/utils/abtest/mark_aarch64fns.py
new file mode 100755
index 0000000000000..652014792849b
--- /dev/null
+++ b/utils/abtest/mark_aarch64fns.py
@@ -0,0 +1,65 @@
+#!/usr/bin/env python
+#
+# Mark functions in an arm assembly file. This is done by surrounding the
+# function with "# -- Begin Name" and "# -- End Name"
+# (This script is designed for aarch64 ios assembly syntax)
+import sys
+import re
+
+inp = open(sys.argv[1], "r").readlines()
+
+# First pass
+linenum = 0
+INVALID=-100
+last_align = INVALID
+last_code = INVALID
+last_globl = INVALID
+last_globl_name = None
+begin = INVALID
+in_text_section = False
+begins = dict()
+for line in inp:
+    linenum += 1
+    if re.search(r'.section\s+__TEXT,__text,regular,pure_instructions', line):
+        in_text_section = True
+        continue
+    elif ".section" in line:
+        in_text_section = False
+        continue
+
+    if not in_text_section:
+        continue
+
+    if ".align" in line:
+        last_align = linenum
+    gl = re.search(r'.globl\s+(\w+)', line)
+    if gl:
+        last_globl_name = gl.group(1)
+        last_globl = linenum
+    m = re.search(r'^(\w+):', line)
+    if m and begin == INVALID:
+        labelname = m.group(1)
+        if last_globl+2 == linenum and last_globl_name == labelname:
+            begin = last_globl
+            funcname = labelname
+    if line == "\n" and begin != INVALID:
+        end = linenum
+        triple = (funcname, begin, end)
+        begins[begin] = triple
+        begin = INVALID
+
+# Second pass: Mark
+out = open(sys.argv[1], "w")
+in_func = None
+linenum = 0
+for line in inp:
+    linenum += 1
+    if in_func is not None and linenum == end:
+        out.write("# -- End  %s\n" % in_func)
+        in_func = None
+
+    triple = begins.get(linenum)
+    if triple is not None:
+        in_func, begin, end = triple
+        out.write("# -- Begin  %s\n" % in_func)
+    out.write(line)