1#!/usr/bin/env python3 2# Copyright © Microsoft Corporation 3# 4# Permission is hereby granted, free of charge, to any person obtaining a 5# copy of this software and associated documentation files (the "Software"), 6# to deal in the Software without restriction, including without limitation 7# the rights to use, copy, modify, merge, publish, distribute, sublicense, 8# and/or sell copies of the Software, and to permit persons to whom the 9# Software is furnished to do so, subject to the following conditions: 10# 11# The above copyright notice and this permission notice (including the next 12# paragraph) shall be included in all copies or substantial portions of the 13# Software. 14# 15# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 18# THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING 20# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS 21# IN THE SOFTWARE. 22 23import argparse 24 25# Take a log file produced by GALLIUM_REFCNT_LOG, filter it to the objects that 26# weren't destroyed by the end of the log, and write the results out sorted. 27# Strips stacks by default to prevent OOM. Could probably be rewritten to walk 28# the file twice to preserve stacks without OOM, but this was the easy way. 29def main(): 30 parser = argparse.ArgumentParser() 31 parser.add_argument('--input', 32 action='store', 33 required=True, 34 help='path to file containing refcount log') 35 parser.add_argument('--output', 36 action='store', 37 required=True, 38 help='path to trimmed log') 39 parser.add_argument('--filter', 40 help='object type filter') 41 parser.add_argument('--keep-stacks', 42 help='keep stacks, otherwise only headers') 43 args = parser.parse_args() 44 45 objects = {} 46 47 with open(args.input) as in_file: 48 stack = [] 49 cur_object = '' 50 51 for line in in_file: 52 if line[0] == '<': 53 54 parts = line.split(' ') 55 prev_object = cur_object 56 cur_object = parts[1] 57 if parts[3].strip() == 'Destroy': 58 if cur_object in objects: 59 del objects[cur_object] 60 else: 61 if parts[3].strip() == 'Create': 62 if (not args.filter) or (args.filter in parts[0]): 63 objects[cur_object] = [] 64 if prev_object in objects: 65 objects[prev_object] += stack 66 67 stack = [line] 68 elif args.keep_stacks: 69 stack += line 70 71 with open(args.output, 'wt') as out_file: 72 for stack in objects.values(): 73 for stack_line in stack: 74 out_file.write(stack_line) 75 76 77if __name__ == '__main__': 78 main() 79