X-Git-Url: http://git.efficios.com/?p=lttng-tools.git;a=blobdiff_plain;f=tests%2Fregression%2Ftools%2Fnotification%2Futil_event_generator.sh;fp=tests%2Fregression%2Ftools%2Fnotification%2Futil_event_generator.sh;h=cafd0643ecd627360506ff2114b2fce33e57a33b;hp=0000000000000000000000000000000000000000;hb=2463b7879c00298daa79744cdaae82ac061a4ed8;hpb=3a4595c2469472dee1656cde5f8882c2123efd3c

diff --git a/tests/regression/tools/notification/util_event_generator.sh b/tests/regression/tools/notification/util_event_generator.sh
new file mode 100644
index 000000000..cafd0643e
--- /dev/null
+++ b/tests/regression/tools/notification/util_event_generator.sh
@@ -0,0 +1,195 @@
+#!/bin/bash
+#
+# Copyright (C) 2020 Jonathan Rajotte-Julien <jonathan.rajotte-julien@efficios.com>
+#
+# SPDX-License-Identifier: LGPL-2.1-only
+
+GENERATOR_CURDIR=$(dirname "$0")/
+GENERATOR_TESTDIR=$GENERATOR_CURDIR/../../../
+TESTAPP_PATH=${TESTAPP_PATH:-"$GENERATOR_TESTDIR/utils/testapp"}
+
+SYSCALL_TESTAPP_NAME=${SYSCALL_TESTAPP_NAME:-"gen-syscall-events"}
+SYSCALL_TESTAPP_BIN=${SYSCALL_TESTAPP_BIN:-"$TESTAPP_PATH/$SYSCALL_TESTAPP_NAME/$SYSCALL_TESTAPP_NAME"}
+
+USERSPACE_PROBE_ELF_TESTAPP_NAME=${USERSPACE_PROBE_ELF_TESTAPP_NAME:-"userspace-probe-elf-binary"}
+USERSPACE_PROBE_ELF_TESTAPP_BIN=${USERSPACE_PROBE_ELF_TESTAPP_BIN:-"$TESTAPP_PATH/$USERSPACE_PROBE_ELF_TESTAPP_NAME/.libs/$USERSPACE_PROBE_ELF_TESTAPP_NAME"}
+
+function generate_filter_events
+{
+	local nr=$1
+	/bin/echo -n "$nr" > /proc/lttng-test-filter-event 2> /dev/null
+}
+
+function generate_syscalls
+{
+	local nr=$1
+	shift
+
+	for i in $(seq 1 "$nr"); do
+		# Pass /dev/null so to generate the syscall right away.
+		$SYSCALL_TESTAPP_BIN /dev/null "$@"
+	done
+}
+
+function userspace_probe_testapp
+{
+	local nr=$1
+	shift 
+
+	for i in $(seq 1 "$nr"); do
+		$USERSPACE_PROBE_ELF_TESTAPP_BIN "$@"
+	done
+}
+
+function ust_event_generator_toggle_state
+{
+	ust_event_generator_suspended=$((ust_event_generator_suspended==0))
+}
+
+function generator_quit
+{
+	generator_quit=0
+}
+
+# Note: Only one generator can be used at a time per domain type
+function ust_event_generator_run_once_per_transition
+{
+	# Used by the signal trap
+	ust_event_generator_suspended=0
+	# Used by the signal trap for SIGUSR2 to end the generator
+	generator_quit=1
+
+	local test_app=$1
+	local state_file=$2
+	local nr_iter=$3
+	local nr_usec_wait=$4
+	local run=false
+
+	# Pass any of the remaining arguments to the generator.
+	shift 4
+
+	trap ust_event_generator_toggle_state SIGUSR1
+	trap generator_quit SIGUSR2
+
+	while [ $generator_quit -ne 0  ]; do
+		if [[ $ust_event_generator_suspended -eq "1" ]]; then
+			touch "$state_file"
+			# Reset the "run" state
+			run=true
+			sleep 0.5
+		elif [ "$run" = true ]; then
+			taskset  -c 0 "$test_app" -i "$nr_iter" -w "$nr_usec_wait" "$@"> /dev/null 2>&1
+			run=false;
+			if [[ -f $state_file ]]; then
+				rm -rf "$state_file" 2> /dev/null
+			fi
+		else
+			# Wait for a "suspend" to reset the run state
+			sleep 0.1
+		fi
+	done
+}
+
+# Note: Only one generator can be used at a time per domain type
+function ust_event_generator
+{
+	# Used by the signal trap
+	ust_event_generator_suspended=0
+	# Used by the signal trap for SIGUSR2 to end the generator
+	generator_quit=1
+
+	local test_app=$1
+	local state_file=$2
+	local nr_iter=1000
+	local nr_usec_wait=5
+
+	# Pass any of the remaining arguments to the generator.
+	shift 2
+
+	trap ust_event_generator_toggle_state SIGUSR1
+	trap generator_quit SIGUSR2
+
+	while [ $generator_quit -ne 0 ]; do
+		if [[ $ust_event_generator_suspended -eq "1" ]]; then
+			touch "$state_file"
+			# Reset the "run" state
+			sleep 0.5
+		else
+			taskset  -c 0 "$test_app" -i $nr_iter -w $nr_usec_wait "$@" > /dev/null 2>&1
+			if [[ -f $state_file ]]; then
+				rm -rf "$state_file" 2> /dev/null
+			fi
+		fi
+	done
+}
+
+function kernel_event_generator_toggle_state
+{
+	kernel_event_generator_suspended=$((kernel_event_generator_suspended==0))
+
+}
+
+function kernel_event_generator_run_once_per_transition
+{
+	# Used by the signal trap
+	kernel_event_generator_suspended=0
+	# Used by the signal trap for SIGUSR2 to end the generator
+	generator_quit=1
+
+	local generator=$1
+	local state_file=$2
+	local nr_iter=$3
+
+	# Pass any of the remaining arguments to the generator.
+	shift 3
+
+	local run=false
+	trap kernel_event_generator_toggle_state SIGUSR1
+	trap generator_quit SIGUSR2
+
+	while [ $generator_quit -ne 0 ]; do
+		if [[ $kernel_event_generator_suspended -eq "1" ]]; then
+			touch "$state_file"
+			run=true
+			sleep 0.5
+		elif [ "$run" = true ]; then
+			$generator "$nr_iter" "$@"
+			run=false
+			if [[ -f $state_file ]]; then
+				rm "$state_file" 2> /dev/null
+			fi
+		else
+			# Wait for a "suspend" to reset the run state
+			sleep 0.1
+		fi
+	done
+}
+
+function kernel_event_generator
+{
+	# Used by the signal trap
+	kernel_event_generator_suspended=0
+	# Used by the signal trap for SIGUSR2 to end the generator
+	generator_quit=1
+
+	local generator=$1
+	local state_file=$2
+
+	# Pass any of the remaining arguments to the generator.
+	shift 2
+
+	trap kernel_event_generator_toggle_state SIGUSR1
+	trap generator_quit SIGUSR2
+
+	while [ $generator_quit -ne 0 ]; do
+		if [[ $kernel_event_generator_suspended -eq "1" ]]; then
+			touch "$state_file"
+			sleep 0.5
+		else
+			$generator "10" "$@"
+			if [[ -f $state_file ]]; then
+				rm "$state_file" 2> /dev/null
+			fi
+		fi
+	done
+}