1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
#!/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)
|
