File: unbalanced_allocs.py

package info (click to toggle)
llvm-toolchain-19 1%3A19.1.7-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, trixie
  • size: 1,998,520 kB
  • sloc: cpp: 6,951,680; ansic: 1,486,157; asm: 913,598; python: 232,024; f90: 80,126; objc: 75,281; lisp: 37,276; pascal: 16,990; sh: 10,009; ml: 5,058; perl: 4,724; awk: 3,523; makefile: 3,167; javascript: 2,504; xml: 892; fortran: 664; cs: 573
file content (100 lines) | stat: -rwxr-xr-x 2,548 bytes parent folder | download | duplicates (12)
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
#!/usr/bin/env python3
# ===- 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)