src/tools/deep_memory_profiler/subcommands/expand.py

   1 # Copyright 2013 The Chromium Authors. All rights reserved.
   2 # Use of this source code is governed by a BSD-style license that can be
   3 # found in the LICENSE file.
   4
   5 import logging
   6 import sys
   7
   8 from lib.policy import PolicySet
   9 from lib.subcommand import SubCommand
  10
  11
  12 LOGGER = logging.getLogger('dmprof')
  13
  14
  15 class ExpandCommand(SubCommand):
  16   def __init__(self):
  17     super(ExpandCommand, self).__init__(
  18         'Usage: %prog expand <dump> <policy> <component> <depth>')
  19     self._parser.add_option('--alternative-dirs', dest='alternative_dirs',
  20                             metavar='/path/on/target@/path/on/host[:...]',
  21                             help='Read files in /path/on/host/ instead of '
  22                                  'files in /path/on/target/.')
  23
  24   def do(self, sys_argv):
  25     options, args = self._parse_args(sys_argv, 4)
  26     dump_path = args[1]
  27     target_policy = args[2]
  28     component_name = args[3]
  29     depth = args[4]
  30     alternative_dirs_dict = {}
  31
  32     policy_set = PolicySet.load(SubCommand._parse_policy_list(target_policy))
  33     if not policy_set[target_policy].find_rule(component_name):
  34       sys.stderr.write("ERROR: Component %s not found in policy %s\n"
  35           % (component_name, target_policy))
  36       return 1
  37
  38     if options.alternative_dirs:
  39       for alternative_dir_pair in options.alternative_dirs.split(':'):
  40         target_path, host_path = alternative_dir_pair.split('@', 1)
  41         alternative_dirs_dict[target_path] = host_path
  42     (bucket_set, dump) = SubCommand.load_basic_files(
  43         dump_path, False, alternative_dirs=alternative_dirs_dict)
  44
  45     ExpandCommand._output(dump, policy_set[target_policy], bucket_set,
  46                           component_name, int(depth), sys.stdout)
  47     return 0
  48
  49   @staticmethod
  50   def _output(dump, policy, bucket_set, component_name, depth, out):
  51     """Prints all stacktraces in a given component of given depth.
  52
  53     Args:
  54         dump: A Dump object.
  55         policy: A Policy object.
  56         bucket_set: A BucketSet object.
  57         component_name: A name of component for filtering.
  58         depth: An integer representing depth to be printed.
  59         out: An IO object to output.
  60     """
  61     sizes = {}
  62
  63     ExpandCommand._accumulate(
  64         dump, policy, bucket_set, component_name, depth, sizes)
  65
  66     sorted_sizes_list = sorted(
  67         sizes.iteritems(), key=(lambda x: x[1]), reverse=True)
  68     total = 0
  69     # TODO(dmikurube): Better formatting.
  70     for size_pair in sorted_sizes_list:
  71       out.write('%10d %s\n' % (size_pair[1], size_pair[0]))
  72       total += size_pair[1]
  73     LOGGER.info('total: %d\n' % total)
  74
  75   @staticmethod
  76   def _add_size(precedence, bucket, depth, committed, sizes):
  77     stacktrace_sequence = precedence
  78     for function, sourcefile in zip(
  79         bucket.symbolized_stackfunction[
  80             0 : min(len(bucket.symbolized_stackfunction), 1 + depth)],
  81         bucket.symbolized_stacksourcefile[
  82             0 : min(len(bucket.symbolized_stacksourcefile), 1 + depth)]):
  83       stacktrace_sequence += '%s(@%s) ' % (function, sourcefile)
  84     if not stacktrace_sequence in sizes:
  85       sizes[stacktrace_sequence] = 0
  86     sizes[stacktrace_sequence] += committed
  87
  88   @staticmethod
  89   def _accumulate(dump, policy, bucket_set, component_name, depth, sizes):
  90     rule = policy.find_rule(component_name)
  91     if not rule:
  92       pass
  93     elif rule.allocator_type == 'malloc':
  94       for bucket_id, _, committed, allocs, frees in dump.iter_stacktrace:
  95         bucket = bucket_set.get(bucket_id)
  96         if not bucket or bucket.allocator_type == 'malloc':
  97           component_match = policy.find_malloc(bucket)
  98         elif bucket.allocator_type == 'mmap':
  99           continue
 100         else:
 101           assert False
 102         if component_match == component_name:
 103           precedence = ''
 104           precedence += '(alloc=%d) ' % allocs
 105           precedence += '(free=%d) ' % frees
 106           if bucket.typeinfo:
 107             precedence += '(type=%s) ' % bucket.symbolized_typeinfo
 108             precedence += '(type.name=%s) ' % bucket.typeinfo_name
 109           ExpandCommand._add_size(precedence, bucket, depth, committed, sizes)
 110     elif rule.allocator_type == 'mmap':
 111       for _, region in dump.iter_map:
 112         if region[0] != 'hooked':
 113           continue
 114         component_match, bucket = policy.find_mmap(region, bucket_set)
 115         if component_match == component_name:
 116           ExpandCommand._add_size('', bucket, depth,
 117                                   region[1]['committed'], sizes)
 118     elif rule.allocator_type == 'unhooked':
 119       for addr, region in dump.iter_map:
 120         if region[0] != 'unhooked':
 121           continue
 122         component_match = policy.find_unhooked(region)
 123         if component_match == component_name:
 124           precedence = ''
 125           precedence += '%s-' % hex(addr[0])[2:]
 126           precedence += '%s' % hex(addr[1])[2:]
 127           precedence += ' %s' % region[1]['vma']['readable']
 128           precedence += '%s' % region[1]['vma']['writable']
 129           precedence += '%s' % region[1]['vma']['executable']
 130           precedence += '%s' % region[1]['vma']['private']
 131           precedence += ' %s' % region[1]['vma']['offset']
 132           precedence += ' %s:' % region[1]['vma']['major']
 133           precedence += '%s' % region[1]['vma']['minor']
 134           precedence += ' %s' % region[1]['vma']['inode']
 135           precedence += '   %s' % region[1]['vma']['name']
 136           if not precedence in sizes:
 137             sizes[precedence] = 0
 138           sizes[precedence] += region[1]['committed']