• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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
5import logging
6import sys
7
8from lib.bucket import BUCKET_ID, COMMITTED, ALLOC_COUNT, FREE_COUNT
9from lib.policy import PolicySet
10from lib.subcommand import SubCommand
11
12
13LOGGER = logging.getLogger('dmprof')
14
15
16class PProfCommand(SubCommand):
17  def __init__(self):
18    super(PProfCommand, self).__init__(
19        'Usage: %prog pprof [-c COMPONENT] <dump> <policy>')
20    self._parser.add_option('-c', '--component', type='string',
21                            dest='component',
22                            help='restrict to COMPONENT', metavar='COMPONENT')
23
24  def do(self, sys_argv):
25    options, args = self._parse_args(sys_argv, 2)
26
27    dump_path = args[1]
28    target_policy = args[2]
29    component = options.component
30
31    (bucket_set, dump) = SubCommand.load_basic_files(dump_path, False)
32    policy_set = PolicySet.load(SubCommand._parse_policy_list(target_policy))
33
34    with open(SubCommand._find_prefix(dump_path) + '.maps', 'r') as maps_f:
35      maps_lines = maps_f.readlines()
36    PProfCommand._output(
37        dump, policy_set[target_policy], bucket_set, maps_lines, component,
38        sys.stdout)
39
40    return 0
41
42  @staticmethod
43  def _output(dump, policy, bucket_set, maps_lines, component_name, out):
44    """Converts the heap profile dump so it can be processed by pprof.
45
46    Args:
47        dump: A Dump object.
48        policy: A Policy object.
49        bucket_set: A BucketSet object.
50        maps_lines: A list of strings containing /proc/.../maps.
51        component_name: A name of component for filtering.
52        out: An IO object to output.
53    """
54    out.write('heap profile: ')
55    com_committed, com_allocs = PProfCommand._accumulate(
56        dump, policy, bucket_set, component_name)
57
58    out.write('%6d: %8s [%6d: %8s] @ heapprofile\n' % (
59        com_allocs, com_committed, com_allocs, com_committed))
60
61    PProfCommand._output_stacktrace_lines(
62        dump, policy, bucket_set, component_name, out)
63
64    out.write('MAPPED_LIBRARIES:\n')
65    for line in maps_lines:
66      out.write(line)
67
68  @staticmethod
69  def _accumulate(dump, policy, bucket_set, component_name):
70    """Accumulates size of committed chunks and the number of allocated chunks.
71
72    Args:
73        dump: A Dump object.
74        policy: A Policy object.
75        bucket_set: A BucketSet object.
76        component_name: A name of component for filtering.
77
78    Returns:
79        Two integers which are the accumulated size of committed regions and the
80        number of allocated chunks, respectively.
81    """
82    com_committed = 0
83    com_allocs = 0
84
85    for _, region in dump.iter_map:
86      if region[0] != 'hooked':
87        continue
88      component_match, bucket = policy.find_mmap(region, bucket_set)
89
90      if (component_name and component_name != component_match) or (
91          region[1]['committed'] == 0):
92        continue
93
94      com_committed += region[1]['committed']
95      com_allocs += 1
96
97    for line in dump.iter_stacktrace:
98      words = line.split()
99      bucket = bucket_set.get(int(words[BUCKET_ID]))
100      if not bucket or bucket.allocator_type == 'malloc':
101        component_match = policy.find_malloc(bucket)
102      elif bucket.allocator_type == 'mmap':
103        continue
104      else:
105        assert False
106      if (not bucket or
107          (component_name and component_name != component_match)):
108        continue
109
110      com_committed += int(words[COMMITTED])
111      com_allocs += int(words[ALLOC_COUNT]) - int(words[FREE_COUNT])
112
113    return com_committed, com_allocs
114
115  @staticmethod
116  def _output_stacktrace_lines(dump, policy, bucket_set, component_name, out):
117    """Prints information of stacktrace lines for pprof.
118
119    Args:
120        dump: A Dump object.
121        policy: A Policy object.
122        bucket_set: A BucketSet object.
123        component_name: A name of component for filtering.
124        out: An IO object to output.
125    """
126    for _, region in dump.iter_map:
127      if region[0] != 'hooked':
128        continue
129      component_match, bucket = policy.find_mmap(region, bucket_set)
130
131      if (component_name and component_name != component_match) or (
132          region[1]['committed'] == 0):
133        continue
134
135      out.write('     1: %8s [     1: %8s] @' % (
136          region[1]['committed'], region[1]['committed']))
137      for address in bucket.stacktrace:
138        out.write(' 0x%016x' % address)
139      out.write('\n')
140
141    for line in dump.iter_stacktrace:
142      words = line.split()
143      bucket = bucket_set.get(int(words[BUCKET_ID]))
144      if not bucket or bucket.allocator_type == 'malloc':
145        component_match = policy.find_malloc(bucket)
146      elif bucket.allocator_type == 'mmap':
147        continue
148      else:
149        assert False
150      if (not bucket or
151          (component_name and component_name != component_match)):
152        continue
153
154      out.write('%6d: %8s [%6d: %8s] @' % (
155          int(words[ALLOC_COUNT]) - int(words[FREE_COUNT]),
156          words[COMMITTED],
157          int(words[ALLOC_COUNT]) - int(words[FREE_COUNT]),
158          words[COMMITTED]))
159      for address in bucket.stacktrace:
160        out.write(' 0x%016x' % address)
161      out.write('\n')
162