File: collect_process_dump.py

package info (click to toggle)
chromium 139.0.7258.127-1
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 6,122,068 kB
  • sloc: cpp: 35,100,771; ansic: 7,163,530; javascript: 4,103,002; python: 1,436,920; asm: 946,517; xml: 746,709; pascal: 187,653; perl: 88,691; sh: 88,436; objc: 79,953; sql: 51,488; cs: 44,583; fortran: 24,137; makefile: 22,147; tcl: 15,277; php: 13,980; yacc: 8,984; ruby: 7,485; awk: 3,720; lisp: 3,096; lex: 1,327; ada: 727; jsp: 228; sed: 36
file content (68 lines) | stat: -rwxr-xr-x 2,185 bytes parent folder | download | duplicates (11)
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
#!/usr/bin/env python3
#
# Copyright 2019 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""Triggers and collects data from a dump created by dump_process.cc.

The resulting files are ready to be analyzed by analyze_dumps.py.
"""

import argparse
import logging
import os
import sys

_SRC_PATH = os.path.join(os.path.dirname(__file__), os.pardir, os.pardir)
sys.path.append(os.path.join(_SRC_PATH, 'third_party', 'catapult', 'devil'))
from devil.android import device_utils


def _CreateArgumentParser():
  parser = argparse.ArgumentParser()
  parser.add_argument('--package', type=str, required=True,
                      help='Chrome package')
  parser.add_argument('--output-directory', type=str, required=True,
                      help='Dumps destination directory')
  return parser


def main():
  logging.basicConfig(level=logging.INFO)
  parser = _CreateArgumentParser()
  args = parser.parse_args()

  logging.info('Configuring the device')
  devices = device_utils.DeviceUtils.HealthyDevices()
  assert len(devices) == 1, 'Expected exactly one connected device'
  device = devices[0]
  device.EnableRoot()
  device.RunShellCommand(('setenforce', '0'))

  logging.info('Finding the first renderer PID')
  renderer_name = '%s:sandboxed_process' % args.package
  renderer_pids = device.GetPids(renderer_name)
  pid = int(renderer_pids.items()[0][1][0])
  logging.info('PID = %d', pid)

  logging.info('Setting up directories')
  dumps_path = '/data/local/tmp/dumps'
  device.RemovePath(dumps_path, force=True, recursive=True)
  device.RunShellCommand(('mkdir', dumps_path))

  logging.info('Dumping the renderer\'s memory')
  command = (
      'mkdir -p %(path)s && cd %(path)s && /data/local/tmp/dump_process %(pid)d'
      % {'path': dumps_path, 'pid': pid})
  device.RunShellCommand(command, shell=True)

  logging.info('Pulling results')
  filenames = device.ListDirectory(dumps_path)
  for filename in filenames:
    device_path = os.path.join(dumps_path, filename)
    device.PullFile(device_path, os.path.join(args.output_directory, filename))


if __name__ == '__main__':
  sys.exit(main())