unbalanced_allocs.py
2.33 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
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
#!/usr/bin/env python
#===- lib/fuzzer/scripts/unbalanced_allocs.py ------------------------------===#
#
# 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
#
#===------------------------------------------------------------------------===#
#
# Post-process -trace_malloc=2 output and printout only allocations and frees
# unbalanced inside of fuzzer runs.
# Usage:
# my_fuzzer -trace_malloc=2 -runs=10 2>&1 | unbalanced_allocs.py -skip=5
#
#===------------------------------------------------------------------------===#
import argparse
import sys
_skip = 0
def PrintStack(line, stack):
global _skip
if _skip > 0:
return
print('Unbalanced ' + line.rstrip());
for l in stack:
print(l.rstrip())
def ProcessStack(line, f):
stack = []
while line and line.startswith(' #'):
stack += [line]
line = f.readline()
return line, stack
def ProcessFree(line, f, allocs):
if not line.startswith('FREE['):
return f.readline()
addr = int(line.split()[1], 16)
next_line, stack = ProcessStack(f.readline(), f)
if addr in allocs:
del allocs[addr]
else:
PrintStack(line, stack)
return next_line
def ProcessMalloc(line, f, allocs):
if not line.startswith('MALLOC['):
return ProcessFree(line, f, allocs)
addr = int(line.split()[1], 16)
assert not addr in allocs
next_line, stack = ProcessStack(f.readline(), f)
allocs[addr] = (line, stack)
return next_line
def ProcessRun(line, f):
if not line.startswith('MallocFreeTracer: START'):
return ProcessMalloc(line, f, {})
allocs = {}
print(line.rstrip())
line = f.readline()
while line:
if line.startswith('MallocFreeTracer: STOP'):
global _skip
_skip = _skip - 1
for _, (l, s) in allocs.items():
PrintStack(l, s)
print(line.rstrip())
return f.readline()
line = ProcessMalloc(line, f, allocs)
return line
def ProcessFile(f):
line = f.readline()
while line:
line = ProcessRun(line, f);
def main(argv):
parser = argparse.ArgumentParser()
parser.add_argument('--skip', default=0, help='number of runs to ignore')
args = parser.parse_args()
global _skip
_skip = int(args.skip) + 1
ProcessFile(sys.stdin)
if __name__ == '__main__':
main(sys.argv)