abtest: aarch64: handle static functions

[thegameg]

D33303 is a better way to fix this.

utils/abtest/mark_aarch64fns.py

if re.search(r'.section\s+__TEXT,__text,regular,pure_instructions', line):
         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)
+    # We are now looking for something similar to this:
+    # .globl _func_name
+    # .p2align 2
+    # _func_name:                                           ; @func_name
+
+    # Look for .globl. This is optional.
+    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)
+
+    # Look for .align or .p2align. This is mandatory.
+    al = re.search(r'.(p2)?align\s+\d+$', line)
+    if al:
+        last_align = linenum
+
+    # Now look for the label. It should be followed by a comment of this form:
+    # _func_name:  ; @func_name
+    # The _ is darwin specific, so we're skipping it.
+    m = re.search(r'^(\w+):\s+; @(\w+)$', line)
     if m and begin == INVALID:
-        labelname = m.group(1)
-        if last_globl+2 == linenum and last_globl_name == labelname:
-            begin = last_globl
+        labelname = m.group(1)[1:] # Skip the _ in front of the label.
+        commentname = m.group(2)
+        if labelname == commentname:
+            if last_globl == linenum - 2:
+                begin = linenum - 2
+            elif last_align == linenum - 1:
+                begin = linenum - 1
+            else:
+                raise Exception('No .(p2)align or .globl directives.')
             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")