vendor/lldb/lldb-trunk-r256945

1 files changed, 210 insertions, 0 deletions

diff --git a/packages/Python/lldbsuite/test/lldbinline.py b/packages/Python/lldbsuite/test/lldbinline.py
new file mode 100644
index 000000000000..691a3fba6dc2
--- /dev/null
+++ b/packages/Python/lldbsuite/test/lldbinline.py
@@ -0,0 +1,210 @@
+from __future__ import print_function
+from __future__ import absolute_import
+
+# System modules
+import os
+import sys
+
+# Third-party modules
+
+# LLDB modules
+import lldb
+from .lldbtest import *
+from . import lldbutil
+
+def source_type(filename):
+    _, extension = os.path.splitext(filename)
+    return {
+        '.c' : 'C_SOURCES',
+        '.cpp' : 'CXX_SOURCES',
+        '.cxx' : 'CXX_SOURCES',
+        '.cc' : 'CXX_SOURCES',
+        '.m' : 'OBJC_SOURCES',
+        '.mm' : 'OBJCXX_SOURCES'
+    }.get(extension, None)
+
+class CommandParser:
+    def __init__(self):
+        self.breakpoints = []
+
+    def parse_one_command(self, line):
+        parts = line.split('//%')
+
+        command = None
+        new_breakpoint = True
+
+        if len(parts) == 2:
+            command = parts[1].strip() # take off whitespace
+            new_breakpoint = parts[0].strip() != ""
+
+        return (command, new_breakpoint)
+
+    def parse_source_files(self, source_files):
+        for source_file in source_files:
+            file_handle = open(source_file)
+            lines = file_handle.readlines()
+            line_number = 0
+            current_breakpoint = None # non-NULL means we're looking through whitespace to find additional commands
+            for line in lines:
+                line_number = line_number + 1 # 1-based, so we do this first
+                (command, new_breakpoint) = self.parse_one_command(line)
+
+                if new_breakpoint:
+                    current_breakpoint = None
+
+                if command != None:
+                    if current_breakpoint == None:
+                        current_breakpoint = {}
+                        current_breakpoint['file_name'] = source_file
+                        current_breakpoint['line_number'] = line_number
+                        current_breakpoint['command'] = command
+                        self.breakpoints.append(current_breakpoint)
+                    else:
+                        current_breakpoint['command'] = current_breakpoint['command'] + "\n" + command
+
+    def set_breakpoints(self, target):
+        for breakpoint in self.breakpoints:
+            breakpoint['breakpoint'] = target.BreakpointCreateByLocation(breakpoint['file_name'], breakpoint['line_number'])
+
+    def handle_breakpoint(self, test, breakpoint_id):
+        for breakpoint in self.breakpoints:
+            if breakpoint['breakpoint'].GetID() == breakpoint_id:
+                test.execute_user_command(breakpoint['command'])
+                return
+
+class InlineTest(TestBase):
+    # Internal implementation
+
+    def getRerunArgs(self):
+        # The -N option says to NOT run a if it matches the option argument, so
+        # if we are using dSYM we say to NOT run dwarf (-N dwarf) and vice versa.
+        if self.using_dsym is None:
+            # The test was skipped altogether.
+            return ""
+        elif self.using_dsym:
+            return "-N dwarf %s" % (self.mydir)
+        else:
+            return "-N dsym %s" % (self.mydir)
+        
+    def BuildMakefile(self):
+        if os.path.exists("Makefile"):
+            return
+
+        categories = {}
+
+        for f in os.listdir(os.getcwd()):
+            t = source_type(f)
+            if t:
+                if t in list(categories.keys()):
+                    categories[t].append(f)
+                else:
+                    categories[t] = [f]
+
+        makefile = open("Makefile", 'w+')
+
+        level = os.sep.join([".."] * len(self.mydir.split(os.sep))) + os.sep + "make"
+
+        makefile.write("LEVEL = " + level + "\n")
+
+        for t in list(categories.keys()):
+            line = t + " := " + " ".join(categories[t])
+            makefile.write(line + "\n")
+
+        if ('OBJCXX_SOURCES' in list(categories.keys())) or ('OBJC_SOURCES' in list(categories.keys())):
+            makefile.write("LDFLAGS = $(CFLAGS) -lobjc -framework Foundation\n")
+
+        if ('CXX_SOURCES' in list(categories.keys())):
+            makefile.write("CXXFLAGS += -std=c++11\n")
+
+        makefile.write("include $(LEVEL)/Makefile.rules\n")
+        makefile.flush()
+        makefile.close()
+
+
+    @skipUnlessDarwin
+    def __test_with_dsym(self):
+        self.using_dsym = True
+        self.BuildMakefile()
+        self.buildDsym()
+        self.do_test()
+
+    def __test_with_dwarf(self):
+        self.using_dsym = False
+        self.BuildMakefile()
+        self.buildDwarf()
+        self.do_test()
+
+    def __test_with_dwo(self):
+        self.using_dsym = False
+        self.BuildMakefile()
+        self.buildDwo()
+        self.do_test()
+
+    def execute_user_command(self, __command):
+        exec(__command, globals(), locals())
+
+    def do_test(self):
+        exe_name = "a.out"
+        exe = os.path.join(os.getcwd(), exe_name)
+        source_files = [ f for f in os.listdir(os.getcwd()) if source_type(f) ]
+        target = self.dbg.CreateTarget(exe)
+
+        parser = CommandParser()
+        parser.parse_source_files(source_files)
+        parser.set_breakpoints(target)
+
+        process = target.LaunchSimple(None, None, os.getcwd())
+
+        while lldbutil.get_stopped_thread(process, lldb.eStopReasonBreakpoint):
+            thread = lldbutil.get_stopped_thread(process, lldb.eStopReasonBreakpoint)
+            breakpoint_id = thread.GetStopReasonDataAtIndex (0)
+            parser.handle_breakpoint(self, breakpoint_id)
+            process.Continue()
+
+
+    # Utilities for testcases
+
+    def check_expression (self, expression, expected_result, use_summary = True):
+        value = self.frame().EvaluateExpression (expression)
+        self.assertTrue(value.IsValid(), expression+"returned a valid value")
+        if self.TraceOn():
+            print(value.GetSummary())
+            print(value.GetValue())
+        if use_summary:
+            answer = value.GetSummary()
+        else:
+            answer = value.GetValue()
+        report_str = "%s expected: %s got: %s"%(expression, expected_result, answer)
+        self.assertTrue(answer == expected_result, report_str)
+
+def ApplyDecoratorsToFunction(func, decorators):
+    tmp = func
+    if type(decorators) == list:
+        for decorator in decorators:
+            tmp = decorator(tmp)
+    elif hasattr(decorators, '__call__'):
+        tmp = decorators(tmp)
+    return tmp
+    
+
+def MakeInlineTest(__file, __globals, decorators=None):
+    # Derive the test name from the current file name
+    file_basename = os.path.basename(__file)
+    InlineTest.mydir = TestBase.compute_mydir(__file)
+
+    test_name, _ = os.path.splitext(file_basename)
+    # Build the test case 
+    test = type(test_name, (InlineTest,), {'using_dsym': None})
+    test.name = test_name
+
+    test.test_with_dsym = ApplyDecoratorsToFunction(test._InlineTest__test_with_dsym, decorators)
+    test.test_with_dwarf = ApplyDecoratorsToFunction(test._InlineTest__test_with_dwarf, decorators)
+    test.test_with_dwo = ApplyDecoratorsToFunction(test._InlineTest__test_with_dwo, decorators)
+
+    # Add the test case to the globals, and hide InlineTest
+    __globals.update({test_name : test})
+
+    # Store the name of the originating file.o
+    test.test_filename = __file
+    return test
+