X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=tests%2Futils.sh;h=8fcb0da38c2a8f4a3f9ca8af0484edf8b8aa4ee3;hp=24defc7f8d67c7172ef37c73cb4e60f673845623;hb=fec81a7e6e7d21bef1a16643404ac5016009feb3;hpb=fbb28ea917664ffc561d4897e3212d98da9e286f

diff --git a/tests/utils.sh b/tests/utils.sh
index 24defc7f8..8fcb0da38 100644
--- a/tests/utils.sh
+++ b/tests/utils.sh
@@ -16,6 +16,7 @@
 # 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
 
 SESSIOND_BIN="lttng-sessiond"
+RELAYD_BIN="lttng-relayd"
 LTTNG_BIN="lttng"
 BABELTRACE_BIN="babeltrace"
 
@@ -24,6 +25,38 @@ 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'))
@@ -39,6 +72,16 @@ function validate_kernel_version ()
 	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 ""
@@ -53,6 +96,7 @@ function spawn_sessiond ()
 
 	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"
 			return 1
@@ -64,6 +108,68 @@ function spawn_sessiond ()
 	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
+		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 &
+		#$DIR/../src/bin/lttng-relayd/$RELAYD_BIN $opt -vvv >>/tmp/relayd.log 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
@@ -235,3 +341,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
+}