forked from OSchip/llvm-project
97 lines
4.0 KiB
Python
97 lines
4.0 KiB
Python
"""Test the integrity of the lldb public api directory containing SB*.h headers.
|
|
|
|
There should be nothing unwanted there and a simpe main.cpp which includes SB*.h
|
|
should compile and link with the LLDB framework."""
|
|
|
|
import os, re, StringIO
|
|
import unittest2
|
|
from lldbtest import *
|
|
import lldbutil
|
|
|
|
class SBDirCheckerCase(TestBase):
|
|
|
|
mydir = os.path.join("api", "check_public_api_headers")
|
|
|
|
def setUp(self):
|
|
TestBase.setUp(self)
|
|
self.lib_dir = os.environ["LLDB_LIB_DIR"]
|
|
self.template = 'main.cpp.template'
|
|
self.source = 'main.cpp'
|
|
|
|
def test_sb_api_directory(self):
|
|
"""Test the SB API directory and make sure there's no unwanted stuff."""
|
|
|
|
if self.getArchitecture() == "i386":
|
|
self.skipTest("LLDB is 64-bit and cannot be linked to 32-bit test program.")
|
|
|
|
# Call the program generator to produce main.cpp.
|
|
self.generate_main_cpp()
|
|
|
|
if sys.platform.startswith("darwin"):
|
|
d = {'FRAMEWORK_INCLUDES' : "-F%s" % self.lib_dir}
|
|
if sys.platform.startswith("linux") or os.environ.get('LLDB_BUILD_TYPE') == 'Makefile':
|
|
d = {'FRAMEWORK_INCLUDES' : "-I%s" % os.path.join(os.environ["LLDB_SRC"], "include")}
|
|
self.buildDefault(dictionary=d)
|
|
self.exe_name = 'a.out'
|
|
self.sanity_check_executable(self.exe_name)
|
|
|
|
def generate_main_cpp(self):
|
|
"""Generate main.cpp from main.cpp.template."""
|
|
temp = os.path.join(os.getcwd(), self.template)
|
|
with open(temp, 'r') as f:
|
|
content = f.read()
|
|
|
|
public_api_dir = os.path.join(os.environ["LLDB_SRC"], "include", "lldb", "API")
|
|
|
|
# Look under the include/lldb/API directory and add #include statements
|
|
# for all the SB API headers.
|
|
public_headers = os.listdir(public_api_dir)
|
|
# For different platforms, the include statement can vary.
|
|
if sys.platform.startswith("darwin"):
|
|
include_stmt = "'#include <%s>' % os.path.join('LLDB', header)"
|
|
if sys.platform.startswith("linux") or os.environ.get('LLDB_BUILD_TYPE') == 'Makefile':
|
|
include_stmt = "'#include <%s>' % os.path.join(public_api_dir, header)"
|
|
list = [eval(include_stmt) for header in public_headers if (header.startswith("SB") and
|
|
header.endswith(".h"))]
|
|
includes = '\n'.join(list)
|
|
new_content = content.replace('%include_SB_APIs%', includes)
|
|
src = os.path.join(os.getcwd(), self.source)
|
|
with open(src, 'w') as f:
|
|
f.write(new_content)
|
|
|
|
# The main.cpp has been generated, add a teardown hook to remove it.
|
|
self.addTearDownHook(lambda: os.remove(src))
|
|
|
|
def sanity_check_executable(self, exe_name):
|
|
"""Sanity check executable compiled from the auto-generated program."""
|
|
exe = os.path.join(os.getcwd(), exe_name)
|
|
self.runCmd("file %s" % exe, CURRENT_EXECUTABLE_SET)
|
|
|
|
self.line_to_break = line_number(self.source, '// Set breakpoint here.')
|
|
|
|
existing_library_path = os.environ[self.dylibPath] if self.dylibPath in os.environ else None
|
|
|
|
env_val = self.lib_dir if not existing_library_path else "%s:%s" % (existing_library_path, self.lib_dir)
|
|
env_cmd = "settings set target.env-vars %s=%s" %(self.dylibPath, env_val)
|
|
if self.TraceOn():
|
|
print "Set environment to: ", env_cmd
|
|
self.runCmd(env_cmd)
|
|
self.addTearDownHook(lambda: self.runCmd("settings remove target.env-vars %s" % self.dylibPath))
|
|
|
|
lldbutil.run_break_set_by_file_and_line (self, self.source, self.line_to_break, num_expected_locations = -1)
|
|
|
|
self.runCmd("run", RUN_SUCCEEDED)
|
|
|
|
# The stop reason of the thread should be breakpoint.
|
|
self.expect("thread list", STOPPED_DUE_TO_BREAKPOINT,
|
|
substrs = ['stopped',
|
|
'stop reason = breakpoint'])
|
|
|
|
self.runCmd('frame variable')
|
|
|
|
if __name__ == '__main__':
|
|
import atexit
|
|
lldb.SBDebugger.Initialize()
|
|
atexit.register(lambda: lldb.SBDebugger.Terminate())
|
|
unittest2.main()
|