Tool.py
1.82 KB
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
# DExTer : Debugging Experience Tester
# ~~~~~~ ~ ~~ ~ ~~
#
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
"""View tool."""
import os
import pickle
from dex.heuristic import Heuristic
from dex.heuristic.Heuristic import add_heuristic_tool_arguments
from dex.tools import ToolBase
from dex.utils.Exceptions import Error, HeuristicException
from dex.utils.ReturnCode import ReturnCode
class Tool(ToolBase):
"""Given a dextIR file, display the information in a human-readable form.
"""
@property
def name(self):
return 'DExTer view'
def add_tool_arguments(self, parser, defaults):
add_heuristic_tool_arguments(parser)
parser.add_argument(
'input_path',
metavar='dextIR-file',
type=str,
default=None,
help='dexter dextIR file to view')
parser.description = Tool.__doc__
def handle_options(self, defaults):
options = self.context.options
options.input_path = os.path.abspath(options.input_path)
if not os.path.isfile(options.input_path):
raise Error('<d>could not find dextIR file</> <r>"{}"</>'.format(
options.input_path))
def go(self) -> ReturnCode:
options = self.context.options
with open(options.input_path, 'rb') as fp:
steps = pickle.load(fp)
try:
heuristic = Heuristic(self.context, steps)
except HeuristicException as e:
raise Error('could not apply heuristic: {}'.format(e))
self.context.o.auto('{}\n\n{}\n\n{}\n\n'.format(
heuristic.summary_string, steps, heuristic.verbose_output))
return ReturnCode.OK