-
Notifications
You must be signed in to change notification settings - Fork 0
/
trace.py
157 lines (124 loc) · 5.94 KB
/
trace.py
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
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
import json
import subprocess
import sys
class ListChangeIdentifier:
def __init__(self, objects_index, list_index):
self.objects_index = objects_index
self.list_indices = list_index
print(f"New ListChangeIdentifier({objects_index}, {list_index})")
class JavaObject:
def __init__(self, instance, name, value):
self.instance = instance
self.name = name
self.value = value
print(f"New JavaObject('{instance}', {name}, {value})")
class State:
empty_state = {
'line': 0,
'stdout': '',
'heap': {},
'stack_to_render': []
}
def __init__(self, state_dict, previous_state=empty_state):
self.objects = []
self.changes = []
self.line_num = state_dict['line']
self.stdout = state_dict['stdout']
self.heap = state_dict['heap']
## For now, we only parse arrays, linkedlists, and primitives ##
encoded_locals = state_dict['stack_to_render'][0]['encoded_locals']
for varname in state_dict['stack_to_render'][0]['ordered_varnames']:
if varname == '__return__':
continue
if type(encoded_locals[varname]) is list:
# Handle heap reference
ref = encoded_locals[varname][1]
heap_item = self.heap[str(ref)]
if heap_item[0] == 'LIST': # handle list
list_var_old = None
for obj in previous_state.objects:
if obj.name == varname:
list_var_old = obj.value
break
list_var = []
for list_item in heap_item[1:]:
if type(list_item) is list: # Assume item 0 is "ELIDE"
for i in range(list_item[1]):
list_var.append(list_var[-1])
else:
list_var.append(list_item)
self.objects.append(JavaObject('LIST', varname, list_var))
for i in range(len(list_var)):
try:
if list_var[i] != list_var_old[i]: # Changed item
print(f"{varname}[{i}] changed from {list_var_old[i]} to {list_var[i]}")
self.changes.append(ListChangeIdentifier(len(self.objects) - 1, i))
except: # New item
self.changes.append(ListChangeIdentifier(len(self.objects) - 1, i))
elif heap_item[0] == 'INSTANCE':
if heap_item[1] == 'LinkedList': # handle LinkedList
list_var_old = None
for obj in previous_state.objects:
if obj.name == varname:
list_var_old = obj.value
break
list_var = []
if heap_item[2][1] != None: # if head != null
node = self.heap[str(heap_item[2][1][1])]
list_var.append(node[2][1])
while(node[3][1] != None): # Iterate through LinkedList
node = self.heap[str(node[3][1][1])]
list_var.append(node[2][1])
self.objects.append(JavaObject('LIST', varname, list_var))
for i in range(len(list_var)):
try:
if list_var[i] != list_var_old[i]: # Changed item
print(f"{varname}[{i}] changed from {list_var_old[i]} to {list_var[i]}")
self.changes.append(ListChangeIdentifier(len(self.objects) - 1, i))
except: # New item
self.changes.append(ListChangeIdentifier(len(self.objects) - 1, i))
else: # Handle primitive data type
self.objects.append(JavaObject('primitive', varname, encoded_locals[varname]))
var_old = None
for obj in previous_state.objects:
if obj.name == varname:
var_old = obj.value
break
if var_old != encoded_locals[varname]:
pass
# TODO: PrimitiveChangeIdentifier
class Tracer:
def __init__(self, usercode):
self.states = []
## Run code through Traceprinter ##
trace_input_dict = {
"usercode": usercode,
"options": {},
"args": [], # massed to main()
"stdin": ""
}
trace_input_json = json.dumps(trace_input_dict)
# print(trace_input_json) # DEBUG
traceprinter_command = "../java/bin/java -cp .:javax.json-1.0.jar:../java/lib/tools.jar traceprinter.InMemory"
trace_proc = subprocess.Popen(traceprinter_command.split(),
stdout=subprocess.PIPE,
stdin=subprocess.PIPE,
cwd="./traceprinter_backend/cp")
trace_json = trace_proc.communicate(input=trace_input_json.encode())[0]
print(trace_json.decode('utf-8')) # DEBUG
self.states = Tracer.parse(trace_json)
def parse(trace_json):
states = []
trace_dict = json.loads(trace_json)
trace = trace_dict['trace']
for state in trace:
if state['stack_to_render'][0]['func_name'].split(':')[0] != 'main':
# Ignore states outside of main
continue
print(f"\nNew State at states[{len(states)}]")
if len(states) > 0:
states.append(State(state, states[-1]))
else:
states.append(State(state))
return states
# tracer = Tracer(open('Test.java').read()) # DEBUG