# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
SESSIOND_BIN="lttng-sessiond"
+RELAYD_BIN="lttng-relayd"
LTTNG_BIN="lttng"
BABELTRACE_BIN="babeltrace"
-function start_sessiond ()
+# 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()
{
- if [ -z $(pidof $SESSIOND_BIN) ]; then
- echo -n "Starting session daemon... "
- $TESTDIR/../src/bin/lttng-sessiond/$SESSIOND_BIN --daemonize --quiet
+ [ "$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
- echo -e "\e[1;31mFAILED\e[0m"
+ print_fail
return 1
else
- echo -e "\e[1;32mOK\e[0m"
+ print_ok
fi
fi
+
+ return 0
}
-function stop_sessiond ()
+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
- echo -e "\e[1;31mFAILED\e[0m"
+ print_fail
return 1
else
- echo -e "\e[1;32mOK\e[0m"
+ out=1
+ while [ -n "$out" ]; do
+ out=$(pidof lt-$SESSIOND_BIN)
+ sleep 0.5
+ done
+ print_ok
fi
}
sess_name=$1
trace_path=$2
- echo -n "Creating lttng session $SESSION_NAME in $TRACE_PATH "
+ 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
- echo -e "\e[1;31mFAILED\e[0m"
+ print_fail
return 1
else
- echo -e "\e[1;32mOK\e[0m"
- #echo $out | grep "written in" | cut -d' ' -f6
+ 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
+ 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
+ $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u >/dev/null 2>&1
if [ $? -eq 1 ]; then
- echo -e '\e[1;31mFAILED\e[0m'
+ print_fail
return 1
else
- echo -e "\e[1;32mOK\e[0m"
+ 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 start_tracing ()
+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
- echo -e '\e[1;31mFAILED\e[0m'
+ print_fail
return 1
else
- echo -e "\e[1;32mOK\e[0m"
+ print_ok
fi
}
-function stop_tracing ()
+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
- echo -e '\e[1;31mFAILED\e[0m'
+ print_fail
return 1
else
- echo -e "\e[1;32mOK\e[0m"
+ print_ok
fi
}
echo -n "Destroy lttng session $sess_name "
$TESTDIR/../src/bin/lttng/$LTTNG_BIN destroy $sess_name >/dev/null 2>&1
if [ $? -eq 1 ]; then
- echo -e '\e[1;31mFAILED\e[0m'
+ print_fail
return 1
else
- echo -e "\e[1;32mOK\e[0m"
+ print_ok
fi
}
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 -e "$count found in trace \e[1;31mFAILED\e[0m"
+ 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
- echo -e "Trace is coherent \e[1;32mOK\e[0m"
+ print_ok
return 0
fi
}