#!/src/bin/bash # # Copyright (C) - 2012 David Goulet # # This library is free software; you can redistribute it and/or modify it under # the terms of the GNU Lesser General Public License as published by the Free # Software Foundation; version 2.1 of the License. # # This library is distributed in the hope that it will be useful, but WITHOUT # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS # FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more # details. # # You should have received a copy of the GNU Lesser General Public License # along with this library; if not, write to the Free Software Foundation, Inc., # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA SESSIOND_BIN="lttng-sessiond" RELAYD_BIN="lttng-relayd" LTTNG_BIN="lttng" BABELTRACE_BIN="babeltrace" # Minimal kernel version supported for session daemon tests KERNEL_MAJOR_VERSION=2 KERNEL_MINOR_VERSION=6 KERNEL_PATCHLEVEL_VERSION=27 function print_ok () { # Check if we are a terminal if [ -t 1 ]; then echo -e "\e[1;32mOK\e[0m" else echo -e "OK" fi } function print_fail () { # Check if we are a terminal if [ -t 1 ]; then echo -e "\e[1;31mFAIL\e[0m" else echo -e "FAIL" fi } function print_test_banner () { desc="$1" count=$((${#desc}+2)) str=$(printf "%${count}s"); echo -e "\n" echo -e ${str// /-} echo -e " $desc " echo -e ${str// /-} } function validate_kernel_version () { kern_version=($(uname -r | awk -F. '{ printf("%d.%d.%d\n",$1,$2,$3); }' | tr '.' '\n')) if [ ${kern_version[0]} -gt $KERNEL_MAJOR_VERSION ]; then return 0 fi if [ ${kern_version[1]} -gt $KERNEL_MINOR_VERSION ]; then return 0 fi if [ ${kern_version[2]} -ge $KERNEL_PATCHLEVEL_VERSION ]; then return 0 fi return 1 } # Generate a random string # $1 = number of characters; defaults to 16 # $2 = include special characters; 1 = yes, 0 = no; defaults to yes function randstring() { [ "$2" == "0" ] && CHAR="[:alnum:]" || CHAR="[:graph:]" cat /dev/urandom | tr -cd "$CHAR" | head -c ${1:-16} echo } function spawn_sessiond () { echo "" echo -n "Starting session daemon... " validate_kernel_version if [ $? -ne 0 ]; then echo -e "\n*** Kernel too old for session daemon tests ***\n" return 2 fi DIR=$(readlink -f $TESTDIR) if [ -z $(pidof lt-$SESSIOND_BIN) ]; then $DIR/../src/bin/lttng-sessiond/$SESSIOND_BIN --daemonize --quiet --consumerd32-path="$DIR/../src/bin/lttng-consumerd/lttng-consumerd" --consumerd64-path="$DIR/../src/bin/lttng-consumerd/lttng-consumerd" #$DIR/../src/bin/lttng-sessiond/$SESSIOND_BIN --consumerd32-path="$DIR/../src/bin/lttng-consumerd/lttng-consumerd" --consumerd64-path="$DIR/../src/bin/lttng-consumerd/lttng-consumerd" --verbose-consumer >>/tmp/sessiond.log 2>&1 & if [ $? -eq 1 ]; then print_fail return 1 else print_ok fi fi return 0 } function lttng_enable_kernel_event { sess_name=$1 event_name=$2 if [ -z $event_name ]; then # Enable all event if no event name specified $event_name="-a" fi echo -n "Enabling kernel event $event_name for session $sess_name" $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -k >/dev/null 2>&1 if [ $? -eq 1 ]; then print_fail return 1 else print_ok fi } function start_lttng_relayd { local opt=$1 echo -e -n "Starting lttng-relayd (opt: $opt)... " DIR=$(readlink -f $TESTDIR) if [ -z $(pidof lt-$RELAYD_BIN) ]; then $DIR/../src/bin/lttng-relayd/$RELAYD_BIN $opt >/dev/null 2>&1 & #$DIR/../src/bin/lttng-relayd/$RELAYD_BIN $opt -vvv >>/tmp/relayd.log 2>&1 & if [ $? -eq 1 ]; then print_fail return 1 else print_ok fi else print_ok fi } function stop_lttng_relayd { PID_RELAYD=`pidof lt-$RELAYD_BIN` echo -e -n "Killing lttng-relayd (pid: $PID_RELAYD)... " kill $PID_RELAYD >/dev/null 2>&1 if [ $? -eq 1 ]; then print_fail return 1 else out=1 while [ -n "$out" ]; do out=$(pidof lt-$RELAYD_BIN) sleep 0.5 done print_ok return 0 fi } function start_lttng_sessiond() { if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then # Env variable requested no session daemon return fi spawn_sessiond out=$? if [ $out -eq 2 ]; then # Kernel version is not compatible. exit 0 elif [ $out -ne 0 ]; then echo "NOT bad $?" exit 1 fi # Simply wait for the session daemon bootstrap echo "Waiting for the session daemon to bootstrap (2 secs)" sleep 2 } function stop_lttng_sessiond () { if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then # Env variable requested no session daemon return fi PID_SESSIOND=`pidof lt-$SESSIOND_BIN` echo -e -n "Killing session daemon... " kill $PID_SESSIOND >/dev/null 2>&1 if [ $? -eq 1 ]; then print_fail return 1 else out=1 while [ -n "$out" ]; do out=$(pidof lt-$SESSIOND_BIN) sleep 0.5 done print_ok fi } function create_lttng_session () { sess_name=$1 trace_path=$2 echo -n "Creating lttng session $sess_name in $trace_path " $TESTDIR/../src/bin/lttng/$LTTNG_BIN create $sess_name -o $trace_path >/dev/null 2>&1 if [ $? -eq 1 ]; then print_fail return 1 else print_ok fi } function enable_lttng_channel() { sess_name=$1 channel_name=$2 echo -n "Enabling lttng channel $channel_name for session $sess_name" $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel $channel_name -s $sess_name >/dev/null 2>&1 if [ $? -eq 1 ]; then print_fail return 1 else print_ok fi } function disable_lttng_channel() { sess_name=$1 channel_name=$2 echo -n "Disabling lttng channel $channel_name for session $sess_name" $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-channel $channel_name -s $sess_name >/dev/null 2>&1 if [ $? -eq 1 ]; then print_fail return 1 else print_ok fi } function enable_ust_lttng_event () { sess_name=$1 event_name="$2" echo -n "Enabling lttng event $event_name for session $sess_name " $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u >/dev/null 2>&1 if [ $? -eq 1 ]; then print_fail return 1 else print_ok fi } function enable_ust_lttng_event_filter() { sess_name="$1" event_name="$2" filter="$3" echo -n "Enabling lttng event with filtering " $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u --filter "$filter" 2>&1 >/dev/null if [ $? -eq 0 ]; then print_ok return 0 else print_fail return 1 fi } function enable_ust_lttng_event_loglevel() { sess_name="$1" event_name="$2" loglevel="$3" echo -n "Enabling lttng event $event_name with loglevel $loglevel" $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u --loglevel $loglevel 2>&1 >/dev/null if [ $? -eq 0 ]; then print_ok return 0 else print_fail return 1 fi } function enable_ust_lttng_event_loglevel_only() { sess_name="$1" event_name="$2" loglevel="$3" echo -n "Enabling lttng event $event_name with loglevel-only $loglevel" $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u --loglevel-only $loglevel 2>&1 >/dev/null if [ $? -eq 0 ]; then print_ok return 0 else print_fail return 1 fi } function disable_ust_lttng_event () { sess_name="$1" event_name="$2" echo -n "Disabling lttng event $event_name for session $sess_name " $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name -u >/dev/null 2>&1 if [ $? -eq 1 ]; then print_fail return 1 else print_ok fi } function start_lttng_tracing () { sess_name=$1 echo -n "Start lttng tracing for session $sess_name " $TESTDIR/../src/bin/lttng/$LTTNG_BIN start $sess_name >/dev/null 2>&1 if [ $? -eq 1 ]; then print_fail return 1 else print_ok fi } function stop_lttng_tracing () { sess_name=$1 echo -n "Stop lttng tracing for session $sess_name " $TESTDIR/../src/bin/lttng/$LTTNG_BIN stop $sess_name >/dev/null 2>&1 if [ $? -eq 1 ]; then print_fail return 1 else print_ok fi } function destroy_lttng_session () { sess_name=$1 echo -n "Destroy lttng session $sess_name " $TESTDIR/../src/bin/lttng/$LTTNG_BIN destroy $sess_name >/dev/null 2>&1 if [ $? -eq 1 ]; then print_fail return 1 else print_ok fi } function trace_matches () { event_name=$1 nr_iter=$2 trace_path=$3 which $BABELTRACE_BIN >/dev/null if [ $? -eq 1 ]; then echo "Babeltrace binary not found. Skipping trace matches" return 0 fi echo -n "Looking for $nr_iter $event_name in $trace_path " count=$($BABELTRACE_BIN $trace_path | grep $event_name | wc -l) if [ "$count" -ne "$nr_iter" ]; then echo -n "$count found in trace " print_fail return 1 else echo -n "Trace is coherent " print_ok return 0 fi } function validate_trace { event_name=$1 trace_path=$2 which $BABELTRACE_BIN >/dev/null if [ $? -eq 1 ]; then echo "Babeltrace binary not found. Skipping trace matches" return 0 fi echo -n "Validating trace for event $event_name... " traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | grep $event_name | wc -l) if [ $traced -eq 0 ]; then print_fail return 1 else print_ok return 0 fi }