blob: 06594b8760f7b05b7b236ec88e7ebcb7014645ad [file] [log] [blame]
#!/bin/bash
# Copyright (C) 2018 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.
set -e
SCRIPT_DIR="$(realpath "$(dirname "${BASH_SOURCE[0]}")")"
if [ "$TMPDIR" == "" ]; then
TMPDIR=/tmp
fi
function get_gn_value() {
local out=$1
local key=$2
"$SCRIPT_DIR/gn" args "$out" --list=$key --short | awk '{print $3}' \
| tr -d '"'
}
function is_monolithic() {
local out=$1
value=$(get_gn_value "$out" "monolithic_binaries")
test "$value" == "true"
}
function is_android() {
local out=$1
value=$(get_gn_value "$out" "target_os")
test "$value" == "android"
}
function is_ssh_target() {
[[ -n "$SSH_TARGET" ]]
}
function is_mac() {
# shellcheck disable=2251
! test -d /proc
return $?
}
function tmux_ensure_bash() {
if [[ $SHELL == *"fish" ]]; then
tmux send-keys "bash" Enter
fi
}
function reset_tracing() {
if is_android "$OUT"; then
adb shell 'test -d /sys/kernel/tracing && echo 0 > /sys/kernel/tracing/tracing_on'
elif ! is_mac; then
# shellcheck disable=SC2016
local script='
if [ ! -w /sys/kernel/tracing ]; then
echo "tracefs not accessible, try sudo chown -R $USER /sys/kernel/tracing"
sudo chown -R "$USER" /sys/kernel/tracing
fi
echo 0 > /sys/kernel/tracing/tracing_on
'
if is_ssh_target; then
# shellcheck disable=SC2029
ssh -t "$SSH_TARGET" "sh -c '$script'"
else
sh -c "$script"
fi
fi
}
function adb_supports_push_sync() {
adb --help 2>&1 | grep 'push.*\[--sync\]' >/dev/null 2>&1
}
function push() {
if is_android "$OUT"; then
local maybe_sync=''
if adb_supports_push_sync; then
maybe_sync='--sync'
fi
echo adb push $maybe_sync "$1" "$DIR"
adb push $maybe_sync "$1" "$DIR"
elif is_ssh_target; then
echo scp "$1" "$SSH_TARGET:$DIR"
scp "$1" "$SSH_TARGET:$DIR"
else
echo cp "$1" "$DIR"
cp "$1" "$DIR"
fi
}
function pull() {
if is_android "$OUT"; then
echo adb pull "$DIR/$1" "$2"
adb pull "$DIR/$1" "$2"
elif is_ssh_target; then
echo scp "$SSH_TARGET:$DIR/$1" "$2"
scp "$SSH_TARGET:$DIR/$1" "$2"
else
echo mv "$DIR/$1" "$2"
mv "$DIR/$1" "$2"
fi
}
BACKGROUND=0
SKIP_CONVERTERS=0
TMUX_LAYOUT="even-vertical"
CPU_MASK=""
while getopts "bl:nt:c:C:z:s:" o; do
case "$o" in
b) BACKGROUND=1 ;;
l) TMUX_LAYOUT=${OPTARG} ;;
n) SKIP_CONVERTERS=1 ;;
t) SSH_TARGET=${OPTARG} ;;
c) CONFIG=${OPTARG} ;;
C) OUT=${OPTARG} ;;
z) CPU_MASK=${OPTARG} ;;
s) SCRIPT=${OPTARG} ;;
*)
echo >&2 "Invalid option $o"
exit
;;
esac
done
# Allow out to be passed as argument
shift $((OPTIND - 1))
OUT="${OUT:-$1}"
# Provide useful usage information
if [ -z "$OUT" ]; then
echo "Usage: $0 [OPTION]... [OUTPUT]"
echo ""
echo "Options:"
echo " -b run in the background"
echo " -l LAYOUT tmux pane layout"
echo " -n skip post-trace convertors"
echo " -t TARGET SSH device target"
echo " -c CONFIG trace configuration file"
echo " -C OUTPUT output directory"
echo " -z MASK constrain binaries to given cpu mask (taskset syntax)"
echo " -s SCRIPT a script to put into a tmux pane"
echo ""
echo "Environment variables:"
echo " SSH_TARGET SSH device target"
echo " CONFIG trace configuration file"
echo " OUT output directory"
exit 1
fi
# Warn about invalid output directories
if [ ! -f "$OUT/args.gn" ]; then
echo >&2 "OUT=$OUT doesn't look like an output directory."
echo >&2 "Please specify a directory by doing:"
echo >&2 " export OUT=out/xxx $0"
exit 1
fi
# Check SSH target is valid
if is_ssh_target && ! ssh -q "$SSH_TARGET" exit; then
echo >&2 "SSH_TARGET=$SSH_TARGET doesn't look like a valid SSH target."
echo >&2 "Please specify a SSH cross-compilation target by doing:"
echo >&2 " export SSH_TARGET=<user>@<host> $0"
exit 1
fi
if ! builtin type -P tmux &>/dev/null; then
echo >&2 "tmux not found"
exit 1
fi
# You can set the config to one of the files under test/configs e.g.
# CONFIG=ftrace.cfg or to :test. Defaults to :test.
CONFIG="${CONFIG:-:test}"
if is_android "$OUT"; then
DIR=/data/local/tmp
elif is_ssh_target; then
DIR=$(ssh "$SSH_TARGET" mktemp -d $TMPDIR/perfetto.XXXXXX)
elif is_mac; then
DIR=$(mktemp -d $TMPDIR/perfetto.XXXXXX)
else
DIR=$(mktemp -p $TMPDIR -d perfetto.XXXXXX)
fi
if is_android "$OUT"; then
TRACECONV="gcc_like_host/traceconv"
else
TRACECONV="traceconv"
fi
# (re)build the binaries
BUILD_TARGETS=(traced traced_probes perfetto test/configs)
if [[ SKIP_CONVERTERS -eq 0 ]]; then
BUILD_TARGETS+=($TRACECONV)
fi
"$SCRIPT_DIR/ninja" -C "$OUT" ${BUILD_TARGETS[*]}
push "$OUT/traced"
push "$OUT/traced_probes"
push "$OUT/perfetto"
reset_tracing
if is_android "$OUT"; then
PREFIX="PERFETTO_CONSUMER_SOCK_NAME=@perfetto_test_consumer PERFETTO_PRODUCER_SOCK_NAME=@perfetto_test_producer"
else
PREFIX=""
fi
if ! is_monolithic "$OUT"; then
PREFIX="$PREFIX LD_LIBRARY_PATH=$DIR"
push "$OUT/libperfetto.so"
fi
CONFIG_DEVICE_PATH="$CONFIG"
CMD_OPTS=""
# Shorthand for using serialized test configs.
if [[ "$CONFIG" == *.protobuf ]]; then
CONFIG_DEVICE_PATH="$CONFIG"
CONFIG_PATH=$OUT/$CONFIG
if [[ ! -f $CONFIG_PATH ]]; then
echo >&2 "Config \"$CONFIG_PATH\" not known."
exit 1
fi
push "$CONFIG_PATH"
elif [[ "$CONFIG" != ":test" ]]; then
CONFIG_DEVICE_PATH="$(basename "$CONFIG")"
CONFIG_PATH=$CONFIG
# If path isn't valid, assume it's a name of a test config.
if [[ ! -f $CONFIG_PATH ]]; then
CONFIG_PATH=test/configs/$CONFIG
if [[ ! -f $CONFIG_PATH ]]; then
echo >&2 "Config \"$CONFIG\" not known."
exit 1
fi
fi
CMD_OPTS="--txt $CMD_OPTS"
push "$CONFIG_PATH"
fi
if [[ -f "$SCRIPT" ]]; then
push "$SCRIPT"
fi
POSTFIX=""
if [[ -n "$CPU_MASK" ]]; then
PREFIX="$PREFIX taskset $CPU_MASK"
fi
if [[ BACKGROUND -eq 1 ]]; then
PREFIX="$PREFIX nohup"
POSTFIX=" &> /dev/null &"
fi
if tmux has-session -t demo; then
tmux kill-session -t demo
fi
tmux -2 new-session -d -s demo
if [ ! -z "$ANDROID_ADB_SERVER_PORT" ]; then
tmux set-environment -t demo ANDROID_ADB_SERVER_PORT $ANDROID_ADB_SERVER_PORT
fi
if tmux -V | awk '{split($2, ver, "."); if (ver[1] < 2) exit 1 ; else if (ver[1] == 2 && ver[2] < 1) exit 1 }'; then
tmux set-option -g mouse on
else
tmux set-option -g mode-mouse on
tmux set-option -g mouse-resize-pane on
tmux set-option -g mouse-select-pane on
tmux set-option -g mouse-select-window on
fi
tmux split-window -v
tmux split-window -v
if [[ -n "$SCRIPT" ]]; then
tmux split-window -v
fi
tmux select-layout "${TMUX_LAYOUT}"
tmux select-pane -t 0
tmux send-keys "clear" C-m
if is_android "$OUT"; then
tmux send-keys "adb shell" C-m
fi
tmux select-pane -t 1
tmux send-keys "clear" C-m
if is_android "$OUT"; then
tmux send-keys "adb shell" C-m
fi
tmux select-pane -t 2
tmux send-keys "clear" C-m
if is_android "$OUT"; then
tmux send-keys "adb shell" C-m
fi
if [[ -n "$SCRIPT" ]]; then
tmux select-pane -t 3
tmux send-keys "clear" C-m
if is_android "$OUT"; then
tmux send-keys "adb shell" C-m
fi
fi
sleep 2
tmux select-pane -t 1
if is_ssh_target; then
tmux send-keys "ssh $SSH_TARGET" Enter
fi
tmux_ensure_bash
tmux send-keys "PS1='[traced]$ '" Enter
tmux send-keys "cd $DIR" Enter
tmux send-keys "clear" C-m
tmux send-keys "$PREFIX ./traced $POSTFIX" Enter
tmux select-pane -t 0
if is_ssh_target; then
tmux send-keys "ssh $SSH_TARGET" Enter
fi
tmux_ensure_bash
tmux send-keys "PS1='[traced_probes]$ '" Enter
tmux send-keys "cd $DIR" Enter
tmux send-keys "clear" C-m
tmux send-keys "$PREFIX ./traced_probes $POSTFIX" Enter
tmux select-pane -t 2
if is_ssh_target; then
tmux send-keys "ssh $SSH_TARGET" Enter
fi
tmux_ensure_bash
tmux send-keys "PS1='[consumer]$ '" Enter
tmux send-keys "cd $DIR" Enter
tmux send-keys "clear" C-m
tmux send-keys "$PREFIX ./perfetto $CMD_OPTS -c $CONFIG_DEVICE_PATH -o trace $POSTFIX"
if [[ -n "$SCRIPT" ]]; then
tmux select-pane -t 3
if is_ssh_target; then
tmux send-keys "ssh $SSH_TARGET" Enter
fi
tmux_ensure_bash
tmux send-keys "PS1='[script]$ '" Enter
tmux send-keys "cd $DIR" Enter
tmux send-keys "clear" C-m
if [[ -f "$SCRIPT" ]]; then
tmux send-keys "./$(basename "$SCRIPT")"
else
tmux send-keys "$SCRIPT"
fi
fi
# Select consumer pane.
tmux select-pane -t 2
tmux -2 attach-session -t demo
if [[ BACKGROUND -eq 1 ]]; then
exit 0
fi
reset_tracing
TRACE=$TMPDIR/trace
echo -e "\n\x1b[32mPulling trace into $TRACE.perfetto-trace\x1b[0m"
pull trace "$TRACE.perfetto-trace"
if [[ SKIP_CONVERTERS -eq 0 ]]; then
echo -e "\n\x1b[32mPulling trace into $TRACE.pbtext\x1b[0m"
"$OUT/$TRACECONV" text <"$TRACE.perfetto-trace" >"$TRACE.pbtext"
fi