| 1 | #!/bin/bash |
| 2 | # |
| 3 | # Copyright (C) 2017 Jonathan Rajotte <jonathan.rajotte-julien@efficios.com> |
| 4 | # |
| 5 | # SPDX-License-Identifier: LGPL-2.1-only |
| 6 | |
| 7 | CURDIR=$(dirname "$0")/ |
| 8 | TESTDIR=$CURDIR/../../../ |
| 9 | |
| 10 | TEST_TMPDIR=$(mktemp --tmpdir -d tmp.test_notif_ust_buffer_usage_dir.XXXXXX) |
| 11 | |
| 12 | set -x |
| 13 | #This is needed since the testpoint create a pipe with the consumerd type suffixed |
| 14 | TESTPOINT_BASE_PATH=$(readlink -f "$TEST_TMPDIR/lttng.t_p_n") |
| 15 | TESTPOINT_PIPE_PATH=$(mktemp -u --tmpdir="$TEST_TMPDIR" "lttng.t_p_n.XXXXXX") |
| 16 | TESTPOINT=$(readlink -f "${CURDIR}/.libs/libpause_consumer.so") |
| 17 | |
| 18 | TESTAPP_PATH="$TESTDIR/utils/testapp" |
| 19 | |
| 20 | GEN_UST_EVENTS_TESTAPP_NAME="gen-ust-events" |
| 21 | GEN_UST_EVENTS_TESTAPP_BIN="$TESTAPP_PATH/$GEN_UST_EVENTS_TESTAPP_NAME/$GEN_UST_EVENTS_TESTAPP_NAME" |
| 22 | |
| 23 | TESTAPP_STATE_PATH=$(mktemp -u --tmpdir="$TEST_TMPDIR" "application_state.XXXXXXXXXX") |
| 24 | |
| 25 | SESSION_NAME="my_session" |
| 26 | CHANNEL_NAME="my_channel" |
| 27 | |
| 28 | TRACE_PATH=$(mktemp -d --tmpdir="$TEST_TMPDIR" "trace_path.XXXXXX") |
| 29 | PAGE_SIZE=$(getconf PAGE_SIZE) |
| 30 | set +x |
| 31 | |
| 32 | # shellcheck source=../../../utils/utils.sh |
| 33 | source "$TESTDIR/utils/utils.sh" |
| 34 | # shellcheck source=./util_event_generator.sh |
| 35 | source "$CURDIR/util_event_generator.sh" |
| 36 | |
| 37 | function test_buffer_usage_notification |
| 38 | { |
| 39 | consumerd_pipe=() |
| 40 | event_name="tp:tptest" |
| 41 | |
| 42 | create_lttng_session_notap $SESSION_NAME "$TRACE_PATH" |
| 43 | |
| 44 | enable_ust_lttng_channel_notap $SESSION_NAME $CHANNEL_NAME --subbuf-size="$PAGE_SIZE" |
| 45 | enable_ust_lttng_event_notap $SESSION_NAME $event_name $CHANNEL_NAME |
| 46 | |
| 47 | # This is needed since the testpoint create a pipe with the consumer type suffixed |
| 48 | for f in "$TESTPOINT_BASE_PATH"*; do |
| 49 | consumerd_pipe+=("$f") |
| 50 | done |
| 51 | |
| 52 | ust_event_generator "$GEN_UST_EVENTS_TESTAPP_BIN" "$TESTAPP_STATE_PATH" & |
| 53 | APP_PID=$! |
| 54 | |
| 55 | "$CURDIR/notification" 2 LTTNG_DOMAIN_UST $APP_PID "$TESTAPP_STATE_PATH" \ |
| 56 | $SESSION_NAME $CHANNEL_NAME "${consumerd_pipe[@]}" |
| 57 | |
| 58 | destroy_lttng_session_notap $SESSION_NAME |
| 59 | |
| 60 | # On ungraceful kill the app is cleaned up via the full_cleanup call |
| 61 | # Suppress kill message |
| 62 | kill -SIGUSR2 $APP_PID |
| 63 | wait $APP_PID 2> /dev/null |
| 64 | |
| 65 | # Just in case cleanup |
| 66 | rm -rf "$TRACE_PATH" |
| 67 | } |
| 68 | |
| 69 | LTTNG_SESSIOND_ENV_VARS="LTTNG_TESTPOINT_ENABLE=1 CONSUMER_PAUSE_PIPE_PATH=${TESTPOINT_PIPE_PATH} LD_PRELOAD=${TESTPOINT}" |
| 70 | start_lttng_sessiond_notap |
| 71 | |
| 72 | test_buffer_usage_notification |
| 73 | |
| 74 | stop_lttng_sessiond_notap |
| 75 | |
| 76 | rm -rf "$TEST_TMPDIR" |