blob: 6926c0150c8872bfcb7cc312364f0e874b94a288 [file] [log] [blame]
#!/usr/bin/env python3
# Copyright (C) 2017 The Android Open Source Project
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
from __future__ import print_function
import argparse
import datetime
import os
import subprocess
import sys
"""Pulls all format files from an Android device.
Usage: ./tools/pull_ftrace_format_files.py [-s serial] [-p directory_prefix]
"""
ROOT_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
ADB_PATH = os.path.join(ROOT_DIR, 'buildtools/android_sdk/platform-tools/adb')
def adb(*cmd, **kwargs):
serial = kwargs.get('serial', None)
prefix = [ADB_PATH]
if serial:
prefix += ['-s', serial]
cmd = prefix + list(cmd)
output = subprocess.check_output(cmd, encoding='utf-8').replace('\r', '')
return output
def get_devices():
# adb devices output looks like:
# List of devices attached
# 557dccd8\tdevice
# With a trailing newline.
serials = [s.split('\t')[0] for s in adb('devices').split('\n')[1:] if s]
return serials
def ensure_output_directory_empty(path):
if os.path.isfile(path):
print('The output directory {} exists as a file.'.format(path))
sys.exit(1)
if os.path.isdir(path) and os.listdir(path):
print('The output directory {} exists but is not empty.'.format(path))
sys.exit(1)
if not os.path.isdir(path):
os.makedirs(path)
def ensure_dir(path):
try:
os.makedirs(path)
except OSError:
if not os.path.isdir(path):
raise
def ensure_single_device(serial):
serials = get_devices()
if serial is None and len(serials) == 1:
return serials[0]
if serial in serials:
return serial
if not serials:
print('No devices connected.')
elif serial is None:
print('More than one device connected, use -s.')
else:
print('No device with serial {} found.'.format(serial))
sys.exit(1)
def pull_format_files(serial, output_directory):
# Pulling each file individually is 100x slower so we pipe all together then
# split them on the host.
cmd = "find /sys/kernel/tracing/ " \
"-name available_events -o " \
"-name format -o " \
"-name header_event -o " \
"-name header_page | " \
"grep -v '/instances/' | " \
"while read f; do echo 'path:' $f; cat $f; done"
output = adb('shell', cmd, serial=serial)
sections = output.split('path: /sys/kernel/tracing/')
for section in sections:
if not section:
continue
path, rest = section.split('\n', 1)
path = os.path.join(output_directory, path)
ensure_dir(os.path.dirname(path))
with open(path, 'w') as f:
f.write(rest)
# Produces output of the form: prefix_android_seed_N2F62_3.10.49
def get_output_directory(prefix=None, serial=None):
build_id = adb('shell', 'getprop', 'ro.build.id', serial=serial).strip()
product = adb('shell', 'getprop', 'ro.build.product', serial=serial).strip()
kernel = adb('shell', 'uname', '-r', serial=serial).split('-')[0].strip()
parts = ['android', product, build_id, kernel]
if prefix:
parts = [prefix] + parts
dir_name = '_'.join(parts)
return os.path.join(ROOT_DIR, 'src/traced/probes/ftrace/test/data', dir_name)
def main():
parser = argparse.ArgumentParser(description='Pull format files.')
parser.add_argument(
'-p', dest='prefix', default=None, help='the output directory prefix')
parser.add_argument(
'-s',
dest='serial',
default=None,
help='use device with the given serial')
args = parser.parse_args()
prefix = args.prefix
serial = args.serial
serial = ensure_single_device(serial)
output_directory = get_output_directory(prefix, serial)
ensure_output_directory_empty(output_directory)
pull_format_files(serial, output_directory)
return 0
if __name__ == '__main__':
sys.exit(main())