[lttng-tools.git] / tests / regression / tools / notification / test_notification_kernel_buffer_usage

#!/bin/bash
#
# Copyright (C) 2017 Jonathan Rajotte-Julien <jonathan.rajotte-julien@efficios.com>
#
# SPDX-License-Identifier: LGPL-2.1-only

CURDIR=$(dirname $0)/
TESTDIR=$CURDIR/../../../

TMPDIR=$(mktemp -d)

#This is needed since the testpoint create a pipe with the consumerd type suffixed
TESTPOINT_BASE_PATH=$(readlink -f "$TMPDIR/lttng.t_p_n")
TESTPOINT_PIPE_PATH=$(mktemp -u "${TESTPOINT_BASE_PATH}.XXXXXX")
TESTPOINT=$(readlink -f ${CURDIR}/.libs/libpause_consumer.so)


TESTAPP_PATH="$TESTDIR/utils/testapp"
TESTAPP_NAME="gen-ust-events"
TESTAPP_BIN="$TESTAPP_PATH/$TESTAPP_NAME/$TESTAPP_NAME"
TESTAPP_STATE_PATH=$(mktemp -u "$TMPDIR/application_state.XXXXXXXXXX")

SYSCALL_TESTAPP_NAME="gen-syscall-events"
SYSCALL_TESTAPP_BIN="$TESTAPP_PATH/$SYSCALL_TESTAPP_NAME/$SYSCALL_TESTAPP_NAME"

USERSPACE_PROBE_ELF_TESTAPP_NAME="userspace-probe-elf-binary"
USERSPACE_PROBE_ELF_TESTAPP_BIN="$TESTAPP_PATH/$USERSPACE_PROBE_ELF_TESTAPP_NAME/.libs/$USERSPACE_PROBE_ELF_TESTAPP_NAME"

SESSION_NAME="my_session"
CHANNEL_NAME="my_channel"

TRACE_PATH=$(mktemp -d)
PAGE_SIZE=$(getconf PAGE_SIZE)

DIR=$(readlink -f $TESTDIR)
NUM_TESTS=104

source $TESTDIR/utils/utils.sh

function kernel_event_generator_toggle_state
{
	kernel_event_generator_suspended=$((kernel_event_generator_suspended==0))
}

function generate_filter_events
{
	/bin/echo -n "10" > /proc/lttng-test-filter-event 2> /dev/null
}

function kernel_event_generator
{
	command_to_run=$1
	state_file=$2
	kernel_event_generator_suspended=0
	trap kernel_event_generator_toggle_state SIGUSR1

	while (true); do
		if [[ $kernel_event_generator_suspended -eq "1" ]]; then
			touch $state_file
			sleep 0.5
		else
			if [[ -f $state_file ]]; then
				rm $state_file 2> /dev/null
			fi
			$command_to_run
		fi
	done
}

function test_buffer_usage_notification
{
	local consumerd_pipe=$1
	local event_name="lttng_test_filter_event"

	create_lttng_session_notap $SESSION_NAME $TRACE_PATH

	lttng_enable_kernel_channel_notap $SESSION_NAME $CHANNEL_NAME \
		--subbuf-size=$PAGE_SIZE
	enable_kernel_lttng_event_notap $SESSION_NAME $event_name $CHANNEL_NAME

	kernel_event_generator generate_filter_events $TESTAPP_STATE_PATH &
	APP_PID=$!

	$CURDIR/notification 2 LTTNG_DOMAIN_KERNEL $APP_PID $TESTAPP_STATE_PATH \
		$SESSION_NAME $CHANNEL_NAME ${consumerd_pipe[@]}

	destroy_lttng_session_notap $SESSION_NAME

	kill -9 $APP_PID
	wait $APP_PID 2> /dev/null
}

if [ "$(id -u)" == "0" ]; then
	consumerd_pipe=()

	validate_lttng_modules_present

	# This is needed since the testpoint create a pipe with the consumer
	# type suffixed.
	for f in "$TESTPOINT_BASE_PATH"*; do
		consumerd_pipe+=("$f")
	done

	modprobe lttng-test

	LTTNG_SESSIOND_ENV_VARS="LTTNG_TESTPOINT_ENABLE=1 \
		CONSUMER_PAUSE_PIPE_PATH=${TESTPOINT_PIPE_PATH} \
		LD_PRELOAD=${TESTPOINT}"

	start_lttng_sessiond_notap

	test_buffer_usage_notification $consumerd_pipe

	stop_lttng_sessiond_notap
	rmmod lttng-test

	rm -rf ${consumerd_pipe[@]} 2> /dev/null
else
	# Kernel tests are skipped.
	plan_tests $NUM_TESTS
	skip 0 "Root access is needed. Skipping all kernel notification tests." $NUM_TESTS
fi

# Just in case cleanup
rm -rf $TRACE_PATH
rm -rf $TMPDIR
Commit	Line	Data
	1	#!/bin/bash
	2	#
	3	# Copyright (C) 2017 Jonathan Rajotte-Julien <jonathan.rajotte-julien@efficios.com>
	4	#
	5	# SPDX-License-Identifier: LGPL-2.1-only
	6
	7	CURDIR=$(dirname $0)/
	8	TESTDIR=$CURDIR/../../../
	9
	10	TMPDIR=$(mktemp -d)
	11
	12	#This is needed since the testpoint create a pipe with the consumerd type suffixed
	13	TESTPOINT_BASE_PATH=$(readlink -f "$TMPDIR/lttng.t_p_n")
	14	TESTPOINT_PIPE_PATH=$(mktemp -u "${TESTPOINT_BASE_PATH}.XXXXXX")
	15	TESTPOINT=$(readlink -f ${CURDIR}/.libs/libpause_consumer.so)
	16
	17
	18	TESTAPP_PATH="$TESTDIR/utils/testapp"
	19	TESTAPP_NAME="gen-ust-events"
	20	TESTAPP_BIN="$TESTAPP_PATH/$TESTAPP_NAME/$TESTAPP_NAME"
	21	TESTAPP_STATE_PATH=$(mktemp -u "$TMPDIR/application_state.XXXXXXXXXX")
	22
	23	SYSCALL_TESTAPP_NAME="gen-syscall-events"
	24	SYSCALL_TESTAPP_BIN="$TESTAPP_PATH/$SYSCALL_TESTAPP_NAME/$SYSCALL_TESTAPP_NAME"
	25
	26	USERSPACE_PROBE_ELF_TESTAPP_NAME="userspace-probe-elf-binary"
	27	USERSPACE_PROBE_ELF_TESTAPP_BIN="$TESTAPP_PATH/$USERSPACE_PROBE_ELF_TESTAPP_NAME/.libs/$USERSPACE_PROBE_ELF_TESTAPP_NAME"
	28
	29	SESSION_NAME="my_session"
	30	CHANNEL_NAME="my_channel"
	31
	32	TRACE_PATH=$(mktemp -d)
	33	PAGE_SIZE=$(getconf PAGE_SIZE)
	34
	35	DIR=$(readlink -f $TESTDIR)
	36	NUM_TESTS=104
	37
	38	source $TESTDIR/utils/utils.sh
	39
	40	function kernel_event_generator_toggle_state
	41	{
	42	kernel_event_generator_suspended=$((kernel_event_generator_suspended==0))
	43	}
	44
	45	function generate_filter_events
	46	{
	47	/bin/echo -n "10" > /proc/lttng-test-filter-event 2> /dev/null
	48	}
	49
	50	function kernel_event_generator
	51	{
	52	command_to_run=$1
	53	state_file=$2
	54	kernel_event_generator_suspended=0
	55	trap kernel_event_generator_toggle_state SIGUSR1
	56
	57	while (true); do
	58	if [[ $kernel_event_generator_suspended -eq "1" ]]; then
	59	touch $state_file
	60	sleep 0.5
	61	else
	62	if [[ -f $state_file ]]; then
	63	rm $state_file 2> /dev/null
	64	fi
	65	$command_to_run
	66	fi
	67	done
	68	}
	69
	70	function test_buffer_usage_notification
	71	{
	72	local consumerd_pipe=$1
	73	local event_name="lttng_test_filter_event"
	74
	75	create_lttng_session_notap $SESSION_NAME $TRACE_PATH
	76
	77	lttng_enable_kernel_channel_notap $SESSION_NAME $CHANNEL_NAME \
	78	--subbuf-size=$PAGE_SIZE
	79	enable_kernel_lttng_event_notap $SESSION_NAME $event_name $CHANNEL_NAME
	80
	81	kernel_event_generator generate_filter_events $TESTAPP_STATE_PATH &
	82	APP_PID=$!
	83
	84	$CURDIR/notification 2 LTTNG_DOMAIN_KERNEL $APP_PID $TESTAPP_STATE_PATH \
	85	$SESSION_NAME $CHANNEL_NAME ${consumerd_pipe[@]}
	86
	87	destroy_lttng_session_notap $SESSION_NAME
	88
	89	kill -9 $APP_PID
	90	wait $APP_PID 2> /dev/null
	91	}
	92
	93	if [ "$(id -u)" == "0" ]; then
	94	consumerd_pipe=()
	95
	96	validate_lttng_modules_present
	97
	98	# This is needed since the testpoint create a pipe with the consumer
	99	# type suffixed.
	100	for f in "$TESTPOINT_BASE_PATH"*; do
	101	consumerd_pipe+=("$f")
	102	done
	103
	104	modprobe lttng-test
	105
	106	LTTNG_SESSIOND_ENV_VARS="LTTNG_TESTPOINT_ENABLE=1 \
	107	CONSUMER_PAUSE_PIPE_PATH=${TESTPOINT_PIPE_PATH} \
	108	LD_PRELOAD=${TESTPOINT}"
	109
	110	start_lttng_sessiond_notap
	111
	112	test_buffer_usage_notification $consumerd_pipe
	113
	114	stop_lttng_sessiond_notap
	115	rmmod lttng-test
	116
	117	rm -rf ${consumerd_pipe[@]} 2> /dev/null
	118	else
	119	# Kernel tests are skipped.
	120	plan_tests $NUM_TESTS
	121	skip 0 "Root access is needed. Skipping all kernel notification tests." $NUM_TESTS
	122	fi
	123
	124	# Just in case cleanup
	125	rm -rf $TRACE_PATH
	126	rm -rf $TMPDIR