Commit | Line | Data |
---|---|---|
ebaaaf5e JD |
1 | #!/bin/bash |
2 | # | |
3 | # Copyright (C) - 2013 David Goulet <dgoulet@efficios.com> | |
4 | # | |
5 | # This library is free software; you can redistribute it and/or modify it under | |
6 | # the terms of the GNU Lesser General Public License as published by the Free | |
7 | # Software Foundation; version 2.1 of the License. | |
8 | # | |
9 | # This library is distributed in the hope that it will be useful, but WITHOUT | |
10 | # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS | |
11 | # FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more | |
12 | # details. | |
13 | # | |
14 | # You should have received a copy of the GNU Lesser General Public License | |
15 | # along with this library; if not, write to the Free Software Foundation, Inc., | |
16 | # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
17 | TEST_DESC="Streaming - Snapshot UST tracing" | |
18 | ||
19 | CURDIR=$(dirname $0)/ | |
20 | TESTDIR=$CURDIR/../../.. | |
21 | EVENT_NAME="tp:tptest" | |
22 | PID_RELAYD=0 | |
23 | SESSION_NAME="" | |
24 | CHANNEL_NAME="chan1" | |
25 | BIN_NAME="gen-nevents" | |
26 | TESTAPP_PATH="$TESTDIR/utils/testapp" | |
27 | TESTAPP_NAME="gen-ust-events" | |
28 | TESTAPP_BIN="$TESTAPP_PATH/$TESTAPP_NAME/$TESTAPP_NAME" | |
29 | NR_ITER=2000000 | |
30 | NR_USEC_WAIT=100 | |
d8b43e5c | 31 | APPS_PID= |
ebaaaf5e JD |
32 | |
33 | TRACE_PATH=$(mktemp -d) | |
34 | ||
847b88a9 | 35 | NUM_TESTS=49 |
ebaaaf5e JD |
36 | |
37 | source $TESTDIR/utils/utils.sh | |
38 | ||
39 | if [ ! -x "$TESTAPP_BIN" ]; then | |
40 | BAIL_OUT "No UST events binary detected." | |
41 | fi | |
42 | ||
43 | function snapshot_add_output () | |
44 | { | |
45 | local sess_name=$1 | |
46 | local trace_path=$2 | |
47 | local name=$3 | |
48 | local extra_opt="" | |
49 | ||
50 | if [ ! -z $name ]; then | |
51 | extra_opt="-n $name" | |
52 | fi | |
53 | ||
54 | $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot add-output -s $sess_name $extra_opt $trace_path >/dev/null 2>&1 | |
55 | ok $? "Added snapshot output $trace_path" | |
56 | } | |
57 | ||
0062bdcc | 58 | # Start trace application and return once one event has been hit. |
d8b43e5c | 59 | function start_test_app() |
0062bdcc DG |
60 | { |
61 | local tmp_file="/tmp/lttng_test_ust.42.file" | |
62 | ||
63 | # Start application with a temporary file. | |
64 | $TESTAPP_BIN $NR_ITER $NR_USEC_WAIT $tmp_file & | |
d8b43e5c MD |
65 | ret=$? |
66 | APPS_PID="${APPS_PID} ${!}" | |
67 | ok $ret "Start application to trace" | |
0062bdcc DG |
68 | |
69 | # Wait for the application file to appear indicating that at least one | |
70 | # tracepoint has been fired. | |
71 | while [ ! -f "$tmp_file" ]; do | |
72 | sleep 0.5 | |
73 | done | |
74 | diag "Removing test app temporary file $tmp_file" | |
75 | rm -f $tmp_file | |
76 | } | |
77 | ||
d8b43e5c | 78 | function stop_test_apps() |
19cb8605 | 79 | { |
d8b43e5c MD |
80 | diag "Stopping $TESTAPP_NAME" |
81 | for p in ${APPS_PID}; do | |
82 | kill ${p} | |
83 | wait ${p} 2>&1 | |
84 | done | |
19cb8605 MD |
85 | } |
86 | ||
ebaaaf5e JD |
87 | # Test a snapshot using a default name for the output destination. |
88 | function test_ust_default_name_with_del() | |
89 | { | |
90 | diag "Test UST snapshot streaming with default name with delete output" | |
91 | create_lttng_session_no_output $SESSION_NAME | |
92 | enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME | |
93 | enable_ust_lttng_event $SESSION_NAME $EVENT_NAME $CHANNEL_NAME | |
94 | start_lttng_tracing $SESSION_NAME | |
0062bdcc | 95 | |
d8b43e5c | 96 | start_test_app |
0062bdcc | 97 | |
ebaaaf5e JD |
98 | snapshot_add_output $SESSION_NAME "net://localhost" |
99 | lttng_snapshot_record $SESSION_NAME | |
100 | ||
101 | # Validate test | |
102 | echo $TRACE_PATH/$HOSTNAME/snapshot-1 | |
103 | validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1* | |
104 | if [ $? -ne 0 ]; then | |
d8b43e5c | 105 | stop_test_apps |
ebaaaf5e JD |
106 | return $? |
107 | fi | |
108 | ||
109 | lttng_snapshot_del_output $SESSION_NAME 1 | |
110 | snapshot_add_output $SESSION_NAME "net://localhost" | |
111 | lttng_snapshot_record $SESSION_NAME | |
112 | ||
113 | # Validate test with the next ID since a del output was done prior. | |
114 | validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-2* | |
115 | if [ $? -ne 0 ]; then | |
d8b43e5c | 116 | stop_test_apps |
ebaaaf5e JD |
117 | return $? |
118 | fi | |
119 | ||
120 | stop_lttng_tracing $SESSION_NAME | |
121 | destroy_lttng_session $SESSION_NAME | |
122 | ||
d8b43e5c | 123 | stop_test_apps |
ebaaaf5e JD |
124 | |
125 | return 0 | |
126 | } | |
127 | ||
128 | # Test a snapshot using a default name for the output destination. | |
129 | function test_ust_default_name() | |
130 | { | |
131 | diag "Test UST snapshot streaming with default name" | |
132 | create_lttng_session_no_output $SESSION_NAME | |
133 | enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME | |
134 | enable_ust_lttng_event $SESSION_NAME $EVENT_NAME $CHANNEL_NAME | |
135 | start_lttng_tracing $SESSION_NAME | |
0062bdcc | 136 | |
d8b43e5c | 137 | start_test_app |
0062bdcc | 138 | |
ebaaaf5e JD |
139 | snapshot_add_output $SESSION_NAME "net://localhost" |
140 | lttng_snapshot_record $SESSION_NAME | |
141 | stop_lttng_tracing $SESSION_NAME | |
142 | destroy_lttng_session $SESSION_NAME | |
143 | # Validate test | |
144 | validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1* | |
145 | out=$? | |
146 | ||
d8b43e5c | 147 | stop_test_apps |
ebaaaf5e JD |
148 | |
149 | return $out | |
150 | } | |
151 | ||
847b88a9 CB |
152 | function test_ust_default_name_custom_uri() |
153 | { | |
154 | diag "Test UST snapshot streaming with default name with custom URL" | |
155 | create_lttng_session_no_output $SESSION_NAME | |
156 | enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME | |
157 | enable_ust_lttng_event $SESSION_NAME $EVENT_NAME $CHANNEL_NAME | |
158 | start_lttng_tracing $SESSION_NAME | |
0062bdcc | 159 | |
d8b43e5c | 160 | start_test_app |
0062bdcc | 161 | |
847b88a9 CB |
162 | snapshot_add_output $SESSION_NAME "-C tcp://localhost:5342 -D tcp://localhost:5343" |
163 | lttng_snapshot_record $SESSION_NAME | |
164 | stop_lttng_tracing $SESSION_NAME | |
165 | destroy_lttng_session $SESSION_NAME | |
166 | # Validate test | |
167 | validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1* | |
168 | out=$? | |
169 | ||
d8b43e5c | 170 | stop_test_apps |
847b88a9 CB |
171 | |
172 | return $out | |
173 | } | |
174 | ||
ebaaaf5e JD |
175 | # Test a snapshot using a custom name for the output destination. |
176 | function test_ust_custom_name() | |
177 | { | |
178 | local out | |
179 | local name="asnapshotname" | |
180 | ||
181 | diag "Test UST snapshot streaming with custom name" | |
182 | create_lttng_session_no_output $SESSION_NAME | |
183 | enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME | |
184 | enable_ust_lttng_event $SESSION_NAME $EVENT_NAME $CHANNEL_NAME | |
185 | start_lttng_tracing $SESSION_NAME | |
0062bdcc | 186 | |
d8b43e5c | 187 | start_test_app |
0062bdcc | 188 | |
ebaaaf5e JD |
189 | snapshot_add_output $SESSION_NAME "net://localhost" $name |
190 | lttng_snapshot_record $SESSION_NAME | |
191 | stop_lttng_tracing $SESSION_NAME | |
192 | destroy_lttng_session $SESSION_NAME | |
193 | ||
194 | if ls $TRACE_PATH/$HOSTNAME/$name* &> /dev/null; then | |
195 | ok 0 "Custom name snapshot exists" | |
196 | # Validate test | |
197 | validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/$name-* | |
198 | out=$? | |
199 | else | |
200 | fail "No custom name snapshot found" | |
201 | out=1 | |
202 | fi | |
203 | ||
d8b43e5c | 204 | stop_test_apps |
ebaaaf5e JD |
205 | |
206 | return $out | |
207 | } | |
208 | ||
209 | plan_tests $NUM_TESTS | |
210 | ||
211 | print_test_banner "$TEST_DESC" | |
212 | ||
213 | if [ "$(id -u)" == "0" ]; then | |
214 | isroot=1 | |
215 | else | |
216 | isroot=0 | |
217 | fi | |
218 | ||
219 | start_lttng_relayd "-o $TRACE_PATH" | |
220 | start_lttng_sessiond | |
221 | ||
847b88a9 | 222 | tests=( test_ust_default_name_with_del test_ust_default_name test_ust_custom_name test_ust_default_name_custom_uri ) |
ebaaaf5e JD |
223 | |
224 | for fct_test in ${tests[@]}; | |
225 | do | |
226 | SESSION_NAME=$(randstring 16 0) | |
227 | ${fct_test} | |
228 | if [ $? -eq 0 ]; then | |
229 | # Only delete if successful | |
230 | rm -rf $TRACE_PATH | |
231 | else | |
232 | break | |
233 | fi | |
234 | done | |
235 | ||
236 | stop_lttng_sessiond | |
237 | stop_lttng_relayd |