Commit | Line | Data |
---|---|---|
069fc1c5 FD |
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" |