program_analysis/milestone2_tmp.py at main · katja-kolos/program_analysis · GitHub

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
from os.path import join, exists
from os import sep

from inspect import getmembers, isclass
from importlib import import_module
from shutil import copyfile, move
from dynapyt.instrument.instrument import instrument_file
from dynapyt.utils.hooks import get_hooks_from_analysis
from dynapyt.analyses.BaseAnalysis import BaseAnalysis
import dynapyt.runtime as _rt
import sys
from src.dynamicslicing.slice_dataflow import SliceDataflow

print('Import successful')

abs_dir = rel_dir = r'tests/milestone2/test_6'
print(abs_dir)

def test_this_dir(abs_dir):
    import dynapyt.runtime as _rt
    rel_dir = abs_dir
    # gather hooks used by the analysis
    module_prefix = rel_dir.replace(sep, ".")
    module_name = "src.dynamicslicing.slice_dataflow" #adding src does not work -- relative import disallowed
    module = import_module(module_name)
    analysis_classes = getmembers(
        module, lambda c: isclass(c) and issubclass(c, BaseAnalysis) and c is not BaseAnalysis
    )

    # instrument
    program_file = join(abs_dir, "program.py")
    orig_program_file = join(abs_dir, "program.py.orig")
    # make sure to instrument the uninstrumented version
    with open(program_file, "r") as file:
        src = file.read()
        if "DYNAPYT: DO NOT INSTRUMENT" in src:
            if not exists(orig_program_file):
                pytest.fail(f"Could find only the instrumented program in {rel_dir}")
            copyfile(orig_program_file, program_file)

    selected_hooks = get_hooks_from_analysis([f"{module_name}.{ac[0]}:{program_file}" for ac in analysis_classes])

    instrument_file(program_file, selected_hooks)

    analysis_instances = [class_[1](orig_program_file) for class_ in analysis_classes]

    # analyze
    _rt.analyses = None
    _rt.set_analysis(analysis_instances)

    for analysis_instance in analysis_instances:
        if hasattr(analysis_instance, "begin_execution"):
            analysis_instance.begin_execution()
    import_module(f"{module_prefix}.program")
    _rt.end_execution()
    del sys.modules["dynapyt.runtime"]
    del _rt

    # slice_dfl_example = SliceDataflow(program_path)


abs_dirs = [
    r'tests/milestone2/test_0',
    r'tests/milestone2/test_1',
    r'tests/milestone2/test_2',
    r'tests/milestone2/test_3',
    r'tests/milestone2/test_4',
    r'tests/milestone2/test_5',
    r'tests/milestone2/test_6',
    r'tests/milestone2/test_7',
    r'tests/milestone2/test_8',
    r'tests/milestone2/test_9',
    r'tests/milestone2/test_10'
]
# for abs_dir in abs_dirs:
test_this_dir(abs_dir)