Activate streaming tests in make check
[lttng-tools.git] / tests / utils.sh
index 352a19837ccd9b5ba07eff98338aba5c571a78dd..1ca298562aeb8afca9b2f9bd29494383fabc4d45 100644 (file)
@@ -1,4 +1,4 @@
-#!/bin/bash
+#!/src/bin/bash
 #
 # Copyright (C) - 2012 David Goulet <dgoulet@efficios.com>
 #
 # 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 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"
+               if [ $? -eq 1 ]; then
+                       echo -e "\e[1;31mFAILED\e[0m"
+                       return 1
+               else
+                       echo -e "\e[1;32mOK\e[0m"
+               fi
+       fi
+
+       return 0
+}
+
+function lttng_enable_kernel_event
 {
-       if [ -z $(pidof $SESSIOND_BIN) ]; then
-               echo -n "Starting session daemon... "
-               $TESTDIR/../lttng-sessiond/$SESSIOND_BIN --daemonize --quiet
+       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
+               echo -e '\e[1;31mFAILED\e[0m'
+               return 1
+       else
+               echo -e "\e[1;32mOK\e[0m"
+       fi
+}
+
+function lttng_start_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 &
                if [ $? -eq 1 ]; then
                        echo -e "\e[1;31mFAILED\e[0m"
                        return 1
                else
                        echo -e "\e[1;32mOK\e[0m"
                fi
+       else
+               echo -e "\e[1;32mOK\e[0m"
+       fi
+}
+
+function lttng_stop_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
+               echo -e "\e[1;31mFAILED\e[0m"
+               return 1
+       else
+               out=1
+               while [ -n "$out" ]; do
+                       out=$(pidof lt-$RELAYD_BIN)
+                       sleep 0.5
+               done
+               echo -e "\e[1;32mOK\e[0m"
+               return 0
+       fi
+}
+
+function start_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_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... "
@@ -43,6 +173,11 @@ function stop_sessiond ()
                echo -e "\e[1;31mFAILED\e[0m"
                return 1
        else
+               out=1
+               while [ -n "$out" ]; do
+                       out=$(pidof lt-$SESSIOND_BIN)
+                       sleep 0.5
+               done
                echo -e "\e[1;32mOK\e[0m"
        fi
 }
@@ -52,14 +187,43 @@ function create_lttng_session ()
        sess_name=$1
        trace_path=$2
 
-       echo -n "Creating lttng session $SESSION_NAME in $TRACE_PATH "
-       $TESTDIR/../lttng/$LTTNG_BIN create $sess_name -o $trace_path >/dev/null 2>&1
+       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"
+               return 1
+       else
+               echo -e "\e[1;32mOK\e[0m"
+       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
+               echo -e "\e[1;31mFAILED\e[0m"
+               return 1
+       else
+               echo -e "\e[1;32mOK\e[0m"
+       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
                echo -e "\e[1;31mFAILED\e[0m"
                return 1
        else
                echo -e "\e[1;32mOK\e[0m"
-               #echo $out | grep "written in" | cut -d' ' -f6
        fi
 }
 
@@ -69,7 +233,7 @@ function enable_ust_lttng_event ()
        event_name=$2
 
        echo -n "Enabling lttng event $event_name for session $sess_name "
-       $TESTDIR/../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'
                return 1
@@ -83,7 +247,7 @@ function start_tracing ()
        sess_name=$1
 
        echo -n "Start lttng tracing for session $sess_name "
-       $TESTDIR/../lttng/$LTTNG_BIN start $sess_name >/dev/null 2>&1
+       $TESTDIR/../src/bin/lttng/$LTTNG_BIN start $sess_name >/dev/null 2>&1
        if [ $? -eq 1 ]; then
                echo -e '\e[1;31mFAILED\e[0m'
                return 1
@@ -97,7 +261,7 @@ function stop_tracing ()
        sess_name=$1
 
        echo -n "Stop lttng tracing for session $sess_name "
-       $TESTDIR/../lttng/$LTTNG_BIN stop $sess_name >/dev/null 2>&1
+       $TESTDIR/../src/bin/lttng/$LTTNG_BIN stop $sess_name >/dev/null 2>&1
        if [ $? -eq 1 ]; then
                echo -e '\e[1;31mFAILED\e[0m'
                return 1
@@ -111,7 +275,7 @@ function destroy_lttng_session ()
        sess_name=$1
 
        echo -n "Destroy lttng session $sess_name "
-       $TESTDIR/../lttng/$LTTNG_BIN destroy $sess_name >/dev/null 2>&1 
+       $TESTDIR/../src/bin/lttng/$LTTNG_BIN destroy $sess_name >/dev/null 2>&1
        if [ $? -eq 1 ]; then
                echo -e '\e[1;31mFAILED\e[0m'
                return 1
@@ -126,6 +290,12 @@ function trace_matches ()
        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)
@@ -137,3 +307,25 @@ function trace_matches ()
                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
+               echo -e "\e[1;31mFAILED\e[0m"
+               return 1
+       else
+               echo -e "\e[1;32mOK\e[0m"
+               return 0
+       fi
+}
This page took 0.026982 seconds and 4 git commands to generate.