Tests: Add filtering tests for uncovered cases
[lttng-tools.git] / tests / utils.sh
1 #!/src/bin/bash
2 #
3 # Copyright (C) - 2012 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
18 SESSIOND_BIN="lttng-sessiond"
19 RELAYD_BIN="lttng-relayd"
20 LTTNG_BIN="lttng"
21 BABELTRACE_BIN="babeltrace"
22
23 # Minimal kernel version supported for session daemon tests
24 KERNEL_MAJOR_VERSION=2
25 KERNEL_MINOR_VERSION=6
26 KERNEL_PATCHLEVEL_VERSION=27
27
28 function print_ok ()
29 {
30 # Check if we are a terminal
31 if [ -t 1 ]; then
32 echo -e "\e[1;32mOK\e[0m"
33 else
34 echo -e "OK"
35 fi
36 }
37
38 function print_fail ()
39 {
40 # Check if we are a terminal
41 if [ -t 1 ]; then
42 echo -e "\e[1;31mFAIL\e[0m"
43 else
44 echo -e "FAIL"
45 fi
46 }
47
48 function print_test_banner ()
49 {
50 desc="$1"
51
52 count=$((${#desc}+2))
53 str=$(printf "%${count}s");
54 echo -e "\n"
55 echo -e ${str// /-}
56 echo -e " $desc "
57 echo -e ${str// /-}
58 }
59
60 function validate_kernel_version ()
61 {
62 kern_version=($(uname -r | awk -F. '{ printf("%d.%d.%d\n",$1,$2,$3); }' | tr '.' '\n'))
63 if [ ${kern_version[0]} -gt $KERNEL_MAJOR_VERSION ]; then
64 return 0
65 fi
66 if [ ${kern_version[1]} -gt $KERNEL_MINOR_VERSION ]; then
67 return 0
68 fi
69 if [ ${kern_version[2]} -ge $KERNEL_PATCHLEVEL_VERSION ]; then
70 return 0
71 fi
72 return 1
73 }
74
75 # Generate a random string
76 # $1 = number of characters; defaults to 16
77 # $2 = include special characters; 1 = yes, 0 = no; defaults to yes
78 function randstring()
79 {
80 [ "$2" == "0" ] && CHAR="[:alnum:]" || CHAR="[:graph:]"
81 cat /dev/urandom | tr -cd "$CHAR" | head -c ${1:-16}
82 echo
83 }
84
85 function spawn_sessiond ()
86 {
87 echo ""
88 echo -n "Starting session daemon... "
89 validate_kernel_version
90 if [ $? -ne 0 ]; then
91 echo -e "\n*** Kernel too old for session daemon tests ***\n"
92 return 2
93 fi
94
95 DIR=$(readlink -f $TESTDIR)
96
97 if [ -z $(pidof lt-$SESSIOND_BIN) ]; then
98 $DIR/../src/bin/lttng-sessiond/$SESSIOND_BIN --daemonize --quiet --consumerd32-path="$DIR/../src/bin/lttng-consumerd/lttng-consumerd" --consumerd64-path="$DIR/../src/bin/lttng-consumerd/lttng-consumerd"
99 #$DIR/../src/bin/lttng-sessiond/$SESSIOND_BIN --consumerd32-path="$DIR/../src/bin/lttng-consumerd/lttng-consumerd" --consumerd64-path="$DIR/../src/bin/lttng-consumerd/lttng-consumerd" --verbose-consumer >>/tmp/sessiond.log 2>&1 &
100 if [ $? -eq 1 ]; then
101 print_fail
102 return 1
103 else
104 print_ok
105 fi
106 fi
107
108 return 0
109 }
110
111 function lttng_enable_kernel_event
112 {
113 sess_name=$1
114 event_name=$2
115
116 if [ -z $event_name ]; then
117 # Enable all event if no event name specified
118 $event_name="-a"
119 fi
120
121 echo -n "Enabling kernel event $event_name for session $sess_name"
122 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event $event_name -s $sess_name -k >/dev/null 2>&1
123 if [ $? -eq 1 ]; then
124 print_fail
125 return 1
126 else
127 print_ok
128 fi
129 }
130
131 function start_lttng_relayd
132 {
133 local opt="$1"
134
135 echo -e -n "Starting lttng-relayd (opt: $opt)... "
136
137 DIR=$(readlink -f $TESTDIR)
138
139 if [ -z $(pidof lt-$RELAYD_BIN) ]; then
140 $DIR/../src/bin/lttng-relayd/$RELAYD_BIN $opt >/dev/null 2>&1 &
141 #$DIR/../src/bin/lttng-relayd/$RELAYD_BIN $opt -vvv >>/tmp/relayd.log 2>&1 &
142 if [ $? -eq 1 ]; then
143 print_fail
144 return 1
145 else
146 print_ok
147 fi
148 else
149 print_ok
150 fi
151 }
152
153 function stop_lttng_relayd
154 {
155 PID_RELAYD=`pidof lt-$RELAYD_BIN`
156
157 echo -e -n "Killing lttng-relayd (pid: $PID_RELAYD)... "
158 kill $PID_RELAYD >/dev/null 2>&1
159 if [ $? -eq 1 ]; then
160 print_fail
161 return 1
162 else
163 out=1
164 while [ -n "$out" ]; do
165 out=$(pidof lt-$RELAYD_BIN)
166 sleep 0.5
167 done
168 print_ok
169 return 0
170 fi
171 }
172
173 function start_lttng_sessiond()
174 {
175 if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
176 # Env variable requested no session daemon
177 return
178 fi
179
180 spawn_sessiond
181 out=$?
182 if [ $out -eq 2 ]; then
183 # Kernel version is not compatible.
184 exit 0
185 elif [ $out -ne 0 ]; then
186 echo "NOT bad $?"
187 exit 1
188 fi
189
190 # Simply wait for the session daemon bootstrap
191 echo "Waiting for the session daemon to bootstrap (2 secs)"
192 sleep 2
193 }
194
195 function stop_lttng_sessiond ()
196 {
197 if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
198 # Env variable requested no session daemon
199 return
200 fi
201
202 PID_SESSIOND=`pidof lt-$SESSIOND_BIN`
203
204 echo -e -n "Killing session daemon... "
205 kill $PID_SESSIOND >/dev/null 2>&1
206 if [ $? -eq 1 ]; then
207 print_fail
208 return 1
209 else
210 out=1
211 while [ -n "$out" ]; do
212 out=$(pidof lt-$SESSIOND_BIN)
213 sleep 0.5
214 done
215 print_ok
216 fi
217 }
218
219 function create_lttng_session ()
220 {
221 sess_name=$1
222 trace_path=$2
223
224 echo -n "Creating lttng session $sess_name in $trace_path "
225 $TESTDIR/../src/bin/lttng/$LTTNG_BIN create $sess_name -o $trace_path >/dev/null 2>&1
226 if [ $? -eq 1 ]; then
227 print_fail
228 return 1
229 else
230 print_ok
231 fi
232 }
233
234 function enable_lttng_channel()
235 {
236 sess_name=$1
237 channel_name=$2
238
239 echo -n "Enabling lttng channel $channel_name for session $sess_name"
240 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel $channel_name -s $sess_name >/dev/null 2>&1
241 if [ $? -eq 1 ]; then
242 print_fail
243 return 1
244 else
245 print_ok
246 fi
247 }
248
249 function disable_lttng_channel()
250 {
251 sess_name=$1
252 channel_name=$2
253
254 echo -n "Disabling lttng channel $channel_name for session $sess_name"
255 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-channel $channel_name -s $sess_name >/dev/null 2>&1
256 if [ $? -eq 1 ]; then
257 print_fail
258 return 1
259 else
260 print_ok
261 fi
262 }
263
264 function enable_ust_lttng_event ()
265 {
266 sess_name=$1
267 event_name=$2
268
269 echo -n "Enabling lttng event $event_name for session $sess_name "
270 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event $event_name -s $sess_name -u >/dev/null 2>&1
271 if [ $? -eq 1 ]; then
272 print_fail
273 return 1
274 else
275 print_ok
276 fi
277 }
278
279 function start_lttng_tracing ()
280 {
281 sess_name=$1
282
283 echo -n "Start lttng tracing for session $sess_name "
284 $TESTDIR/../src/bin/lttng/$LTTNG_BIN start $sess_name >/dev/null 2>&1
285 if [ $? -eq 1 ]; then
286 print_fail
287 return 1
288 else
289 print_ok
290 fi
291 }
292
293 function stop_lttng_tracing ()
294 {
295 sess_name=$1
296
297 echo -n "Stop lttng tracing for session $sess_name "
298 $TESTDIR/../src/bin/lttng/$LTTNG_BIN stop $sess_name >/dev/null 2>&1
299 if [ $? -eq 1 ]; then
300 print_fail
301 return 1
302 else
303 print_ok
304 fi
305 }
306
307 function destroy_lttng_session ()
308 {
309 sess_name=$1
310
311 echo -n "Destroy lttng session $sess_name "
312 $TESTDIR/../src/bin/lttng/$LTTNG_BIN destroy $sess_name >/dev/null 2>&1
313 if [ $? -eq 1 ]; then
314 print_fail
315 return 1
316 else
317 print_ok
318 fi
319 }
320
321 function trace_matches ()
322 {
323 event_name=$1
324 nr_iter=$2
325 trace_path=$3
326
327 which $BABELTRACE_BIN >/dev/null
328 if [ $? -eq 1 ]; then
329 echo "Babeltrace binary not found. Skipping trace matches"
330 return 0
331 fi
332
333 echo -n "Looking for $nr_iter $event_name in $trace_path "
334
335 count=$($BABELTRACE_BIN $trace_path | grep $event_name | wc -l)
336 if [ "$count" -ne "$nr_iter" ]; then
337 echo -n "$count found in trace "
338 print_fail
339 return 1
340 else
341 echo -n "Trace is coherent "
342 print_ok
343 return 0
344 fi
345 }
346
347 function validate_trace
348 {
349 event_name=$1
350 trace_path=$2
351
352 which $BABELTRACE_BIN >/dev/null
353 if [ $? -eq 1 ]; then
354 echo "Babeltrace binary not found. Skipping trace matches"
355 return 0
356 fi
357
358 echo -n "Validating trace for event $event_name... "
359 traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | grep $event_name | wc -l)
360 if [ $traced -eq 0 ]; then
361 print_fail
362 return 1
363 else
364 print_ok
365 return 0
366 fi
367 }
This page took 0.061504 seconds and 4 git commands to generate.