116 lines
4.7 KiB
Python
Executable File
116 lines
4.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
#
|
|
# Copyright (C) 2014 The Android Open Source Project
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
import argparse
|
|
import os
|
|
|
|
from common.archs import archs_list
|
|
from common.logger import Logger
|
|
from file_format.c1visualizer.parser import parse_c1_visualizer_stream
|
|
from file_format.checker.parser import parse_checker_stream
|
|
from match.file import match_files
|
|
|
|
|
|
def parse_arguments():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("tested_file",
|
|
help="text file the checks should be verified against")
|
|
parser.add_argument("source_path", nargs="?",
|
|
help="path to file/folder with checking annotations")
|
|
parser.add_argument("--check-prefix", dest="check_prefix", default="CHECK", metavar="PREFIX",
|
|
help="prefix of checks in the test files (default: CHECK)")
|
|
parser.add_argument("--list-passes", dest="list_passes", action="store_true",
|
|
help="print a list of all passes found in the tested file")
|
|
parser.add_argument("--dump-pass", dest="dump_pass", metavar="PASS",
|
|
help="print a compiler pass dump")
|
|
parser.add_argument("--arch", dest="arch", choices=archs_list,
|
|
help="Run tests for the specified target architecture.")
|
|
parser.add_argument("--debuggable", action="store_true",
|
|
help="Run tests for debuggable code.")
|
|
parser.add_argument("--print-cfg", action="store_true", default="True", dest="print_cfg",
|
|
help="Print the whole cfg file in case of test failure (default)")
|
|
parser.add_argument("--no-print-cfg", action="store_false", default="True", dest="print_cfg",
|
|
help="Don't print the whole cfg file in case of test failure")
|
|
parser.add_argument("-q", "--quiet", action="store_true",
|
|
help="print only errors")
|
|
return parser.parse_args()
|
|
|
|
|
|
def list_passes(output_filename):
|
|
c1_file = parse_c1_visualizer_stream(output_filename, open(output_filename, "r"))
|
|
for compiler_pass in c1_file.passes:
|
|
Logger.log(compiler_pass.name)
|
|
|
|
|
|
def dump_pass(output_filename, pass_name):
|
|
c1_file = parse_c1_visualizer_stream(output_filename, open(output_filename, "r"))
|
|
compiler_pass = c1_file.find_pass(pass_name)
|
|
if compiler_pass:
|
|
max_line_no = compiler_pass.start_line_no + len(compiler_pass.body)
|
|
len_line_no = len(str(max_line_no)) + 2
|
|
cur_line_no = compiler_pass.start_line_no
|
|
for line in compiler_pass.body:
|
|
Logger.log((str(cur_line_no) + ":").ljust(len_line_no) + line)
|
|
cur_line_no += 1
|
|
else:
|
|
Logger.fail('Pass "{}" not found in the output'.format(pass_name))
|
|
|
|
|
|
def find_checker_files(path):
|
|
""" Returns a list of files to scan for check annotations in the given path.
|
|
Path to a file is returned as a single-element list, directories are
|
|
recursively traversed and all '.java', '.j', and '.smali' files returned.
|
|
"""
|
|
if not path:
|
|
Logger.fail("No source path provided")
|
|
elif os.path.isfile(path):
|
|
return [path]
|
|
elif os.path.isdir(path):
|
|
found_files = []
|
|
for root, dirs, files in os.walk(path):
|
|
for file in files:
|
|
extension = os.path.splitext(file)[1]
|
|
if extension in [".java", ".smali", ".j"]:
|
|
found_files.append(os.path.join(root, file))
|
|
return found_files
|
|
else:
|
|
Logger.fail('Source path "{}" not found'.format(path))
|
|
|
|
|
|
def run_tests(check_prefix, check_path, output_filename, target_arch, debuggable_mode, print_cfg):
|
|
c1_file = parse_c1_visualizer_stream(output_filename, open(output_filename, "r"))
|
|
for check_filename in find_checker_files(check_path):
|
|
checker_file = parse_checker_stream(os.path.basename(check_filename),
|
|
check_prefix,
|
|
open(check_filename, "r"),
|
|
target_arch)
|
|
match_files(checker_file, c1_file, target_arch, debuggable_mode, print_cfg)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
args = parse_arguments()
|
|
|
|
if args.quiet:
|
|
Logger.Verbosity = Logger.Level.ERROR
|
|
|
|
if args.list_passes:
|
|
list_passes(args.tested_file)
|
|
elif args.dump_pass:
|
|
dump_pass(args.tested_file, args.dump_pass)
|
|
else:
|
|
run_tests(args.check_prefix, args.source_path, args.tested_file, args.arch, args.debuggable,
|
|
args.print_cfg)
|