Estimates a path in host from a corresponding path in Android device.

tools/deep_memory_profiler/dmprof.py

 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """The deep heap profiler script for Chrome."""
 
 import copy
 import datetime
 import json
 import logging
@@ skipping 23 matching lines @@
 
 BUCKET_ID = 5
 VIRTUAL = 0
 COMMITTED = 1
 ALLOC_COUNT = 2
 FREE_COUNT = 3
 NULL_REGEX = re.compile('')
 
 LOGGER = logging.getLogger('dmprof')
 POLICIES_JSON_PATH = os.path.join(BASE_PATH, 'policies.json')
+CHROME_SRC_PATH = os.path.join(BASE_PATH, os.pardir, os.pardir)
 
 
 # Heap Profile Dump versions
 
 # DUMP_DEEP_[1-4] are obsolete.
 # DUMP_DEEP_2+ distinct mmap regions and malloc chunks.
 # DUMP_DEEP_3+ don't include allocation functions in their stack dumps.
 # DUMP_DEEP_4+ support comments with '#' and global stats "nonprofiled-*".
 # DUMP_DEEP_[1-2] should be processed by POLICY_DEEP_1.
 # DUMP_DEEP_[3-4] should be processed by POLICY_DEEP_2 or POLICY_DEEP_3.
@@ skipping 989 matching lines @@
 
   def __getitem__(self, index):
     return self._dump_list[index]
 
 
 class Command(object):
   """Subclasses are a subcommand for this executable.
 
   See COMMANDS in main().
   """
+  _DEVICE_LIB_PATTERN = re.compile(r'(/data/app-lib/.*-[0-9])/.*')
+
   def __init__(self, usage):
     self._parser = optparse.OptionParser(usage)
 
   @staticmethod
   def load_basic_files(
       dump_path, multiple, no_dump=False, fake_directories=None):
     prefix = Command._find_prefix(dump_path)
-    symbol_data_sources = SymbolDataSources(prefix, fake_directories or {})
+
+    # Auto-estimating the binary with symbol information located on the host

[bulach, 2013/05/14 14:44:44]

nit, I think this would be clearer:
# Translate the symbol information from the device to the
# corresponding files in the host.

[Dai Mikurube (NOT FULLTIME), 2013/05/14 17:21:47]

To make it clear that it's an estimation,

1) Renamed all fake_directories to alternative_dirs

2) Rewrote this comment to:

# Translate the symbol information from the Android device to the
# corresponding files in the host.  The location of the files in the host
# is estimated from the binary path on the device.  Use --alternative-dirs
# to specify the location manually.

3) Changed the behavior and added a comment on _estimate_alternative_dirs (fka
_suppose_fake_directories)

+    # from the binary path on the Android device.
+    if not fake_directories:
+      fake_directories = {}
+      lib_on_device = Command._suppose_fake_directories(prefix)

[bulach, 2013/05/14 14:44:44]

nit: perhaps
device_lib = Command._get_device_lib(prefix)

[Dai Mikurube (NOT FULLTIME), 2013/05/14 17:21:47]

Finally, it's _estimate_alternative_dirs.

+      if lib_on_device:
+        fake_directories[lib_on_device] = os.path.join(
+            CHROME_SRC_PATH, 'out', 'Debug', 'lib')
+    if fake_directories:
+      for device, host in fake_directories.iteritems():
+        LOGGER.info('Assuming %s on device as %s on host' % (device, host))
+    symbol_data_sources = SymbolDataSources(prefix, fake_directories)
     symbol_data_sources.prepare()
     bucket_set = BucketSet()
     bucket_set.load(prefix)
     if not no_dump:
       if multiple:
         dump_list = DumpList.load(Command._find_all_dumps(dump_path))
       else:
         dump = Dump.load(dump_path)
     symbol_mapping_cache = SymbolMappingCache()
     with open(prefix + '.cache.function', 'a+') as cache_f:
@@ skipping 14 matching lines @@
     elif multiple:
       return (bucket_set, dump_list)
     else:
       return (bucket_set, dump)
 
   @staticmethod
   def _find_prefix(path):
     return re.sub('\.[0-9][0-9][0-9][0-9]\.heap', '', path)
 
   @staticmethod
+  def _suppose_fake_directories(prefix):

[bulach, 2013/05/14 14:44:44]

nit: perhaps a comment

"""Return the path for the application library.

On android, returns the name if the map contains only one application library
(that is, an .so inside "/data/app-lib").
Returns None otherwise.
"""

[Dai Mikurube (NOT FULLTIME), 2013/05/14 17:21:47]

As written above.

+    device_lib_path_candidates = set()
+
+    with open(prefix + '.maps') as maps_f:
+      maps = proc_maps.ProcMaps.load(maps_f)
+      for entry in maps:
+        matched = Command._DEVICE_LIB_PATTERN.match(entry.as_dict()['name'])
+        if matched:
+          device_lib_path_candidates.add(matched.group(1))
+
+    if len(device_lib_path_candidates) == 1:
+      return device_lib_path_candidates.pop()
+    else:
+      return None
+
+  @staticmethod
   def _find_all_dumps(dump_path):
     prefix = Command._find_prefix(dump_path)
     dump_path_list = [dump_path]
 
     n = int(dump_path[len(dump_path) - 9 : len(dump_path) - 5])
     n += 1
     while True:
       p = '%s.%04d.heap' % (prefix, n)
       if os.path.exists(p):
         dump_path_list.append(p)
@@ skipping 665 matching lines @@
     errorcode = COMMANDS[action]().do(sys.argv)
   except ParsingException, e:
     errorcode = 1
     sys.stderr.write('Exit by parsing error: %s\n' % e)
 
   return errorcode
 
 
 if __name__ == '__main__':
   sys.exit(main())