tests/regression/tools/notification/test_notification_ust_buffer_usage

   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 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 TESTAPP_PATH="$TESTDIR/utils/testapp"
  18
  19 GEN_UST_EVENTS_TESTAPP_NAME="gen-ust-events"
  20 GEN_UST_EVENTS_TESTAPP_BIN="$TESTAPP_PATH/$GEN_UST_EVENTS_TESTAPP_NAME/$GEN_UST_EVENTS_TESTAPP_NAME"
  21
  22 TESTAPP_STATE_PATH=$(mktemp -u "$TMPDIR/application_state.XXXXXXXXXX")
  23
  24 SESSION_NAME="my_session"
  25 CHANNEL_NAME="my_channel"
  26
  27 TRACE_PATH=$(mktemp -d)
  28 PAGE_SIZE=$(getconf PAGE_SIZE)
  29
  30 # shellcheck source=../../../utils/utils.sh
  31 source "$TESTDIR/utils/utils.sh"
  32
  33 function ust_event_generator_toggle_state
  34 {
  35         ust_event_generator_suspended=$((ust_event_generator_suspended==0))
  36
  37 }
  38 function ust_event_generator
  39 {
  40         test_app=$1
  41         state_file=$2
  42         ust_event_generator_suspended=0
  43         trap ust_event_generator_toggle_state SIGUSR1
  44
  45         while (true); do
  46                 if [[ $ust_event_generator_suspended -eq "1" ]]; then
  47                         touch $state_file
  48                         sleep 0.5
  49                 else
  50                         if [[ -f $state_file ]]; then
  51                                 rm -rf $state_file 2> /dev/null
  52                         fi
  53                         taskset  -c 0 $test_app -i $NR_ITER -w $NR_USEC_WAIT > /dev/null 2>&1
  54                 fi
  55         done
  56 }
  57
  58 function test_buffer_usage_notification
  59 {
  60         consumerd_pipe=()
  61         event_name="tp:tptest"
  62
  63         create_lttng_session_notap $SESSION_NAME "$TRACE_PATH"
  64
  65         enable_ust_lttng_channel_notap $SESSION_NAME $CHANNEL_NAME --subbuf-size="$PAGE_SIZE"
  66         enable_ust_lttng_event_notap $SESSION_NAME $event_name $CHANNEL_NAME
  67
  68         # This is needed since the testpoint create a pipe with the consumer type suffixed
  69         for f in "$TESTPOINT_BASE_PATH"*; do
  70                 consumerd_pipe+=("$f")
  71         done
  72
  73         ust_event_generator $GEN_UST_EVENTS_TESTAPP_BIN $TESTAPP_STATE_PATH &
  74         APP_PID=$!
  75
  76         "$CURDIR/notification" 2 LTTNG_DOMAIN_UST $SESSION_NAME $CHANNEL_NAME $APP_PID "$TESTAPP_STATE_PATH" "${consumerd_pipe[@]}"
  77
  78         destroy_lttng_session_notap $SESSION_NAME
  79
  80         # On ungraceful kill the app is cleaned up via the full_cleanup call
  81         # Suppress kill message
  82         kill -9 $APP_PID
  83         wait $APP_PID 2> /dev/null
  84
  85         # Just in case cleanup
  86         rm -rf "$TRACE_PATH"
  87 }
  88
  89 LTTNG_SESSIOND_ENV_VARS="LTTNG_TESTPOINT_ENABLE=1 CONSUMER_PAUSE_PIPE_PATH=${TESTPOINT_PIPE_PATH} LD_PRELOAD=${TESTPOINT}"
  90 start_lttng_sessiond_notap
  91
  92 test_buffer_usage_notification
  93
  94 stop_lttng_sessiond_notap
  95
  96 rm -rf "$TMPDIR"