Tests: use modprobe to test for the presence of lttng-modules
[lttng-tools.git] / tests / utils / utils.sh
CommitLineData
d3e8f6bb
DG
1# Copyright (C) - 2012 David Goulet <dgoulet@efficios.com>
2#
3# This library is free software; you can redistribute it and/or modify it under
4# the terms of the GNU Lesser General Public License as published by the Free
5# Software Foundation; version 2.1 of the License.
6#
7# This library is distributed in the hope that it will be useful, but WITHOUT
8# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
9# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
10# details.
11#
12# You should have received a copy of the GNU Lesser General Public License
13# along with this library; if not, write to the Free Software Foundation, Inc.,
14# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
15
16SESSIOND_BIN="lttng-sessiond"
725f658e 17SESSIOND_MATCH=".*lttng-sess.*"
1c362dc7 18SESSIOND_PIDS=""
19356f3a 19RUNAS_BIN="lttng-runas"
725f658e 20RUNAS_MATCH=".*lttng-runas.*"
f7613992 21CONSUMERD_BIN="lttng-consumerd"
725f658e 22CONSUMERD_MATCH=".*lttng-consumerd.*"
f4e40ab6 23RELAYD_BIN="lttng-relayd"
725f658e 24RELAYD_MATCH=".*lttng-relayd.*"
1c362dc7 25RELAYD_PIDS=""
d3e8f6bb
DG
26LTTNG_BIN="lttng"
27BABELTRACE_BIN="babeltrace"
2cf48300
JR
28OUTPUT_DEST=/dev/null
29ERROR_OUTPUT_DEST=/dev/null
d3e8f6bb 30
fd4dfcec
DG
31# Minimal kernel version supported for session daemon tests
32KERNEL_MAJOR_VERSION=2
33KERNEL_MINOR_VERSION=6
34KERNEL_PATCHLEVEL_VERSION=27
35
651b8bb3
JG
36# We set the default UST register timeout and network and app socket timeout to
37# "wait forever", so that basic tests don't have to worry about hitting
38# timeouts on busy systems. Specialized tests should test those corner-cases.
629b9335 39export LTTNG_UST_REGISTER_TIMEOUT=-1
651b8bb3
JG
40export LTTNG_NETWORK_SOCKET_TIMEOUT=-1
41export LTTNG_APP_SOCKET_TIMEOUT=-1
629b9335 42
fd7fe1a8
JR
43# We set the default lttng-sessiond path to /bin/true to prevent the spawning
44# of a daemonized sessiond. This is necessary since 'lttng create' will spawn
45# its own sessiond if none is running. It also ensures that 'lttng create'
46# fails when no sessiond is running.
47export LTTNG_SESSIOND_PATH="/bin/true"
48
29655db7
CB
49source $TESTDIR/utils/tap/tap.sh
50
1c362dc7
JR
51function full_cleanup ()
52{
53 if [ -n "${SESSIOND_PIDS}" ] || [ -n "${RELAYD_PIDS}" ]; then
54 kill -9 ${SESSIOND_PIDS} ${RELAYD_PIDS} > /dev/null 2>&1
55 fi
56
57 # Disable trap for SIGTERM since the following kill to the
58 # pidgroup will be SIGTERM. Otherwise it loops.
59 # The '-' before the pid number ($$) indicates 'kill' to signal the
60 # whole process group.
61 trap - SIGTERM && kill -- -$$
62}
63
64
65trap full_cleanup SIGINT SIGTERM
66
fec81a7e
CB
67function print_ok ()
68{
69 # Check if we are a terminal
70 if [ -t 1 ]; then
71 echo -e "\e[1;32mOK\e[0m"
72 else
73 echo -e "OK"
74 fi
75}
76
77function print_fail ()
78{
79 # Check if we are a terminal
80 if [ -t 1 ]; then
81 echo -e "\e[1;31mFAIL\e[0m"
82 else
83 echo -e "FAIL"
84 fi
85}
86
87function print_test_banner ()
88{
7d0ad314 89 local desc="$1"
29655db7 90 diag "$desc"
fec81a7e
CB
91}
92
fd4dfcec
DG
93function validate_kernel_version ()
94{
7d0ad314 95 local kern_version=($(uname -r | awk -F. '{ printf("%d.%d.%d\n",$1,$2,$3); }' | tr '.' '\n'))
fd4dfcec
DG
96 if [ ${kern_version[0]} -gt $KERNEL_MAJOR_VERSION ]; then
97 return 0
98 fi
99 if [ ${kern_version[1]} -gt $KERNEL_MINOR_VERSION ]; then
100 return 0
101 fi
102 if [ ${kern_version[2]} -ge $KERNEL_PATCHLEVEL_VERSION ]; then
103 return 0
104 fi
105 return 1
106}
107
9ac429ef 108# Generate a random string
f4e40ab6
DG
109# $1 = number of characters; defaults to 16
110# $2 = include special characters; 1 = yes, 0 = no; defaults to yes
9ac429ef 111function randstring()
f4e40ab6
DG
112{
113 [ "$2" == "0" ] && CHAR="[:alnum:]" || CHAR="[:graph:]"
b89c3a68 114 cat /dev/urandom 2>/dev/null | tr -cd "$CHAR" 2>/dev/null | head -c ${1:-16} 2>/dev/null
f4e40ab6
DG
115 echo
116}
117
9e136324
JG
118# Return the number of _configured_ CPUs.
119function conf_proc_count()
120{
121 getconf _NPROCESSORS_CONF
122 if [ $? -ne 0 ]; then
123 diag "Failed to get the number of configured CPUs"
124 fi
125 echo
126}
127
9c8a3964
JR
128# Check if base lttng-modules are present.
129# Bail out on failure
130function validate_lttng_modules_present ()
131{
132 modprobe -n lttng-tracer 2>/dev/null
133 if [ $? -ne 0 ]; then
134 BAIL_OUT "LTTng modules not detected."
135 fi
136}
137
4a180d9f 138function enable_kernel_lttng_event
f4e40ab6 139{
854382b8
JR
140 local withtap="$1"
141 local expected_to_fail="$2"
142 local sess_name="$3"
143 local event_name="$4"
144 local channel_name="$5"
f4e40ab6 145
4a180d9f 146 if [ -z "$event_name" ]; then
f4e40ab6 147 # Enable all event if no event name specified
29655db7 148 event_name="-a"
f4e40ab6
DG
149 fi
150
4a180d9f 151 if [ -z "$channel_name" ]; then
07b86b52
JD
152 # default channel if none specified
153 chan=""
154 else
155 chan="-c $channel_name"
156 fi
157
2cf48300 158 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -k 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
4a180d9f
MD
159 ret=$?
160 if [[ $expected_to_fail -eq "1" ]]; then
161 test $ret -ne "0"
854382b8
JR
162 ret=$?
163 if [ $withtap -eq "1" ]; then
164 ok $ret "Enable kernel event $event_name for session $session_name on channel $channel_name failed as expected"
165 fi
4a180d9f 166 else
854382b8
JR
167 if [ $withtap -eq "1" ]; then
168 ok $ret "Enable kernel event $event_name for session $sess_name"
169 fi
4a180d9f
MD
170 fi
171}
172
173function enable_kernel_lttng_event_ok ()
174{
854382b8 175 enable_kernel_lttng_event 1 0 "$@"
4a180d9f
MD
176}
177
178function enable_kernel_lttng_event_fail ()
179{
854382b8
JR
180 enable_kernel_lttng_event 1 1 "$@"
181}
182
183function enable_kernel_lttng_event_notap ()
184{
185 enable_kernel_lttng_event 0 0 "$@"
4a180d9f
MD
186}
187
188# Old interface
189function lttng_enable_kernel_event
190{
191 enable_kernel_lttng_event_ok "$@"
f4e40ab6
DG
192}
193
8cfcd41c
MD
194function lttng_enable_kernel_syscall()
195{
196 local expected_to_fail=$1
197 local sess_name=$2
198 local syscall_name=$3
199 local channel_name=$4
200
201 if [ -z $syscall_name ]; then
202 # Enable all event if no syscall name specified
203 syscall_name="-a"
204 fi
205
206 if [ -z $channel_name ]; then
207 # default channel if none specified
208 chan=""
209 else
210 chan="-c $channel_name"
211 fi
212
2cf48300 213 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event --syscall "$syscall_name" $chan -s $sess_name -k 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
8cfcd41c
MD
214 ret=$?
215 if [[ $expected_to_fail -eq "1" ]]; then
216 test $ret -ne "0"
217 ok $? "Enable kernel syscall $syscall_name for session $sess_name on channel $channel_name fail as expected"
218 else
219 ok $ret "Enable kernel syscall $syscall_name for session $sess_name on channel $channel_name"
220 fi
221}
222
223function lttng_enable_kernel_syscall_ok()
224{
421b83dc 225 lttng_enable_kernel_syscall 0 "$@"
8cfcd41c
MD
226}
227
228function lttng_enable_kernel_syscall_fail()
229{
421b83dc 230 lttng_enable_kernel_syscall 1 "$@"
8cfcd41c
MD
231}
232
233function lttng_disable_kernel_syscall()
234{
235 local expected_to_fail=$1
236 local sess_name=$2
237 local syscall_name=$3
238 local channel_name=$4
239
240 if [ -z $syscall_name ]; then
241 # Enable all event if no syscall name specified
242 syscall_name="-a"
243 fi
244
245 if [ -z $channel_name ]; then
246 # default channel if none specified
247 chan=""
248 else
249 chan="-c $channel_name"
250 fi
251
2cf48300 252 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event --syscall "$syscall_name" $chan -s $sess_name -k 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
8cfcd41c
MD
253
254 ret=$?
255 if [[ $expected_to_fail -eq "1" ]]; then
256 test $ret -ne "0"
34ab15c5 257 ok $? "Disable kernel syscall $syscall_name for session $sess_name on channel $channel_name failed as expected"
8cfcd41c
MD
258 else
259 ok $ret "Disable kernel syscall $syscall_name for session $sess_name on channel $channel_name"
260 fi
261}
262
263function lttng_disable_kernel_syscall_ok()
264{
421b83dc 265 lttng_disable_kernel_syscall 0 "$@"
8cfcd41c
MD
266}
267
268function lttng_disable_kernel_syscall_fail()
269{
421b83dc 270 lttng_disable_kernel_syscall 1 "$@"
8cfcd41c
MD
271}
272
d96f6315
MD
273function lttng_enable_kernel_channel()
274{
854382b8
JR
275 local withtap=$1
276 local expected_to_fail=$2
277 local sess_name=$3
278 local channel_name=$4
279 local opt=$5
d96f6315 280
854382b8 281 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -k $channel_name -s $sess_name $opt 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
d96f6315
MD
282 ret=$?
283 if [[ $expected_to_fail -eq "1" ]]; then
284 test "$ret" -ne "0"
854382b8
JR
285 ret=$?
286 if [ $withtap -eq "1" ]; then
287 ok $ret "Enable channel $channel_name for session $sess_name failed as expected"
288 fi
d96f6315 289 else
854382b8
JR
290 if [ $withtap -eq "1" ]; then
291 ok $ret "Enable channel $channel_name for session $sess_name"
292 fi
d96f6315
MD
293 fi
294}
295
296function lttng_enable_kernel_channel_ok()
297{
854382b8 298 lttng_enable_kernel_channel 1 0 "$@"
d96f6315
MD
299}
300
301function lttng_enable_kernel_channel_fail()
302{
854382b8
JR
303 lttng_enable_kernel_channel 1 1 "$@"
304}
305
306function lttng_enable_kernel_channel_notap()
307{
308 lttng_enable_kernel_channel 0 0 "$@"
309}
310
311function enable_kernel_lttng_channel_ok()
312{
313 lttng_enable_kernel_channel 1 0 "$@"
d96f6315
MD
314}
315
316function lttng_disable_kernel_channel()
317{
318 local expected_to_fail=$1
319 local sess_name=$2
320 local channel_name=$3
321
2cf48300 322 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-channel -k $channel_name -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
d96f6315
MD
323 ret=$?
324 if [[ $expected_to_fail -eq "1" ]]; then
325 test "$ret" -ne "0"
34ab15c5 326 ok $? "Disable channel $channel_name for session $sess_name failed as expected"
d96f6315 327 else
34ab15c5 328 ok $ret "Disable channel $channel_name for session $sess_name"
d96f6315
MD
329 fi
330}
331
332function lttng_disable_kernel_channel_ok()
333{
421b83dc 334 lttng_disable_kernel_channel 0 "$@"
d96f6315
MD
335}
336
337function lttng_disable_kernel_channel_fail()
338{
421b83dc 339 lttng_disable_kernel_channel 1 "$@"
d96f6315
MD
340}
341
05aa48da 342function start_lttng_relayd_opt()
f4e40ab6 343{
05aa48da
MD
344 local withtap=$1
345 local opt=$2
173af62f 346
f4e40ab6
DG
347 DIR=$(readlink -f $TESTDIR)
348
725f658e 349 if [ -z $(pgrep $RELAYD_MATCH) ]; then
2cf48300 350 $DIR/../src/bin/lttng-relayd/$RELAYD_BIN -b $opt 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
25a818af 351 #$DIR/../src/bin/lttng-relayd/$RELAYD_BIN $opt -vvv >>/tmp/relayd.log 2>&1 &
f4e40ab6 352 if [ $? -eq 1 ]; then
05aa48da
MD
353 if [ $withtap -eq "1" ]; then
354 fail "Start lttng-relayd (opt: $opt)"
355 fi
f4e40ab6
DG
356 return 1
357 else
05aa48da
MD
358 if [ $withtap -eq "1" ]; then
359 pass "Start lttng-relayd (opt: $opt)"
360 fi
f4e40ab6
DG
361 fi
362 else
29655db7 363 pass "Start lttng-relayd (opt: $opt)"
f4e40ab6 364 fi
1c362dc7 365
725f658e 366 RELAYD_PIDS=$(pgrep $RELAYD_MATCH)
f4e40ab6
DG
367}
368
05aa48da 369function start_lttng_relayd()
f4e40ab6 370{
05aa48da
MD
371 start_lttng_relayd_opt 1 "$@"
372}
373
374function start_lttng_relayd_notap()
375{
376 start_lttng_relayd_opt 0 "$@"
377}
378
379function stop_lttng_relayd_opt()
380{
381 local withtap=$1
382
05aa48da 383 if [ $withtap -eq "1" ]; then
1c362dc7 384 diag "Killing lttng-relayd (pid: $RELAYD_PIDS)"
05aa48da 385 fi
1c362dc7 386 kill $RELAYD_PIDS 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1fb23888 387 retval=$?
29655db7 388
05aa48da
MD
389 if [ $? -eq 1 ]; then
390 if [ $withtap -eq "1" ]; then
391 fail "Kill relay daemon"
392 fi
393 return 1
394 else
f4e40ab6
DG
395 out=1
396 while [ -n "$out" ]; do
725f658e 397 out=$(pgrep $RELAYD_MATCH)
f4e40ab6
DG
398 sleep 0.5
399 done
05aa48da
MD
400 if [ $withtap -eq "1" ]; then
401 pass "Kill relay daemon"
402 fi
1fb23888 403 fi
1c362dc7 404 RELAYD_PIDS=""
1fb23888
MD
405 return $retval
406}
407
05aa48da 408function stop_lttng_relayd()
1fb23888 409{
05aa48da
MD
410 stop_lttng_relayd_opt 1 "$@"
411}
1fb23888 412
05aa48da
MD
413function stop_lttng_relayd_notap()
414{
415 stop_lttng_relayd_opt 0 "$@"
f4e40ab6
DG
416}
417
05aa48da
MD
418#First arg: show tap output
419#Second argument: load path for automatic loading
420function start_lttng_sessiond_opt()
355f483d 421{
05aa48da
MD
422 local withtap=$1
423 local load_path=$2
8d51ddbc 424
4b01971f 425 local env_vars=""
b916da6b
JR
426 local consumerd=""
427 local long_bit_value=$(getconf LONG_BIT)
428
355f483d
DG
429 if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
430 # Env variable requested no session daemon
431 return
432 fi
433
b916da6b
JR
434 DIR=$(readlink -f $TESTDIR)
435
436 # Get long_bit value for 32/64 consumerd
437 case "$long_bit_value" in
438 32)
439 consumerd="--consumerd32-path=$DIR/../src/bin/lttng-consumerd/lttng-consumerd"
440 ;;
441 64)
442 consumerd="--consumerd64-path=$DIR/../src/bin/lttng-consumerd/lttng-consumerd"
443 ;;
444 *)
445 return
446 ;;
447 esac
448
4b01971f
JR
449 # Check for env. variable. Allow the use of LD_PRELOAD etc.
450 if [[ "x${LTTNG_SESSIOND_ENV_VARS}" != "x" ]]; then
451 env_vars=${LTTNG_SESSIOND_ENV_VARS}
452 fi
453
29655db7
CB
454 validate_kernel_version
455 if [ $? -ne 0 ]; then
456 fail "Start session daemon"
457 BAIL_OUT "*** Kernel too old for session daemon tests ***"
355f483d
DG
458 fi
459
d3d97763
JG
460 : ${LTTNG_SESSION_CONFIG_XSD_PATH=${DIR}/../src/common/config/}
461 export LTTNG_SESSION_CONFIG_XSD_PATH
29655db7 462
725f658e 463 if [ -z $(pgrep ${SESSIOND_MATCH}) ]; then
8d51ddbc 464 # Have a load path ?
05aa48da 465 if [ -n "$load_path" ]; then
4b01971f 466 env $env_vars $DIR/../src/bin/lttng-sessiond/$SESSIOND_BIN --load "$load_path" --background $consumerd
8d51ddbc 467 else
4b01971f 468 env $env_vars $DIR/../src/bin/lttng-sessiond/$SESSIOND_BIN --background $consumerd
8d51ddbc 469 fi
0fc2834c 470 #$DIR/../src/bin/lttng-sessiond/$SESSIOND_BIN --background --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
29655db7 471 status=$?
05aa48da
MD
472 if [ $withtap -eq "1" ]; then
473 ok $status "Start session daemon"
474 fi
29655db7 475 fi
725f658e 476 SESSIOND_PIDS=$(pgrep $SESSIOND_MATCH)
355f483d
DG
477}
478
05aa48da
MD
479function start_lttng_sessiond()
480{
481 start_lttng_sessiond_opt 1 "$@"
482}
483
484function start_lttng_sessiond_notap()
d3e8f6bb 485{
05aa48da
MD
486 start_lttng_sessiond_opt 0 "$@"
487}
488
489function stop_lttng_sessiond_opt()
490{
491 local withtap=$1
4c80129b
JR
492 local signal=$2
493 local kill_opt=""
05aa48da 494
355f483d
DG
495 if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
496 # Env variable requested no session daemon
497 return
498 fi
499
725f658e 500 local pids="${SESSIOND_PIDS} $(pgrep $RUNAS_MATCH)"
d3e8f6bb 501
4c80129b
JR
502 if [ -n "$2" ]; then
503 kill_opt="$kill_opt -s $signal"
504 fi
8490897a 505 if [ $withtap -eq "1" ]; then
725f658e 506 diag "Killing $SESSIOND_BIN and lt-$SESSIOND_BIN pids: $(echo $pids | tr '\n' ' ')"
8490897a 507 fi
1c362dc7 508 kill $kill_opt $pids 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
29655db7 509
d3e8f6bb 510 if [ $? -eq 1 ]; then
05aa48da
MD
511 if [ $withtap -eq "1" ]; then
512 fail "Kill sessions daemon"
513 fi
d3e8f6bb 514 else
5fa32580
DG
515 out=1
516 while [ -n "$out" ]; do
725f658e 517 out=$(pgrep ${SESSIOND_MATCH})
5fa32580
DG
518 sleep 0.5
519 done
f7613992
MD
520 out=1
521 while [ -n "$out" ]; do
725f658e 522 out=$(pgrep $CONSUMERD_MATCH)
f7613992
MD
523 sleep 0.5
524 done
1c362dc7
JR
525
526 SESSIOND_PIDS=""
05aa48da
MD
527 if [ $withtap -eq "1" ]; then
528 pass "Kill session daemon"
529 fi
d3e8f6bb
DG
530 fi
531}
532
05aa48da
MD
533function stop_lttng_sessiond()
534{
535 stop_lttng_sessiond_opt 1 "$@"
536}
537
538function stop_lttng_sessiond_notap()
539{
540 stop_lttng_sessiond_opt 0 "$@"
541}
542
8490897a
MD
543function sigstop_lttng_sessiond_opt()
544{
545 local withtap=$1
546 local signal=SIGSTOP
547 local kill_opt=""
548
549 if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
550 # Env variable requested no session daemon
551 return
552 fi
553
725f658e 554 PID_SESSIOND="$(pgrep ${SESSIOND_MATCH}) $(pgrep $RUNAS_MATCH)"
8490897a
MD
555
556 kill_opt="$kill_opt -s $signal"
557
558 if [ $withtap -eq "1" ]; then
725f658e 559 diag "Sending SIGSTOP to lt-$SESSIOND_BIN and $SESSIOND_BIN pids: $(echo $PID_SESSIOND | tr '\n' ' ')"
8490897a
MD
560 fi
561 kill $kill_opt $PID_SESSIOND 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
562
563 if [ $? -eq 1 ]; then
564 if [ $withtap -eq "1" ]; then
565 fail "Sending SIGSTOP to session daemon"
566 fi
567 else
568 out=1
569 while [ $out -ne 0 ]; do
725f658e 570 pid=$(pgrep $SESSIOND_MATCH)
8490897a
MD
571
572 # Wait until state becomes stopped for session
573 # daemon(s).
574 out=0
575 for sessiond_pid in $pid; do
576 state=$(ps -p $sessiond_pid -o state= )
577 if [[ -n "$state" && "$state" != "T" ]]; then
578 out=1
579 fi
580 done
581 sleep 0.5
582 done
583 if [ $withtap -eq "1" ]; then
584 pass "Sending SIGSTOP to session daemon"
585 fi
586 fi
587}
588
589function sigstop_lttng_sessiond()
590{
591 sigstop_lttng_sessiond_opt 1 "$@"
592}
593
594function sigstop_lttng_sessiond_notap()
595{
596 sigstop_lttng_sessiond_opt 0 "$@"
597}
598
4c80129b
JR
599function stop_lttng_consumerd_opt()
600{
601 local withtap=$1
602 local signal=$2
603 local kill_opt=""
604
725f658e 605 PID_CONSUMERD=$(pgrep $CONSUMERD_MATCH)
4c80129b
JR
606
607 if [ -n "$2" ]; then
608 kill_opt="$kill_opt -s $signal"
609 fi
610
611 if [ $withtap -eq "1" ]; then
8490897a 612 diag "Killing $CONSUMERD_BIN pids: $(echo $PID_CONSUMERD | tr '\n' ' ')"
4c80129b
JR
613 fi
614 kill $kill_opt $PID_CONSUMERD 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
615 retval=$?
616 set +x
617
618 if [ $? -eq 1 ]; then
619 if [ $withtap -eq "1" ]; then
620 fail "Kill consumer daemon"
621 fi
622 return 1
623 else
624 out=1
625 while [ $out -ne 0 ]; do
725f658e 626 pid=$(pgrep $CONSUMERD_MATCH)
4c80129b
JR
627
628 # If consumerds are still present check their status.
629 # A zombie status qualifies the consumerd as *killed*
630 out=0
631 for consumer_pid in $pid; do
632 state=$(ps -p $consumer_pid -o state= )
633 if [[ -n "$state" && "$state" != "Z" ]]; then
634 out=1
635 fi
636 done
637 sleep 0.5
638 done
639 if [ $withtap -eq "1" ]; then
640 pass "Kill consumer daemon"
641 fi
642 fi
643 return $retval
644}
645
646function stop_lttng_consumerd()
647{
648 stop_lttng_consumerd_opt 1 "$@"
649}
650
651function stop_lttng_consumerd_notap()
652{
653 stop_lttng_consumerd_opt 0 "$@"
654}
655
8490897a
MD
656function sigstop_lttng_consumerd_opt()
657{
658 local withtap=$1
659 local signal=SIGSTOP
660 local kill_opt=""
661
725f658e 662 PID_CONSUMERD=$(pgrep $CONSUMERD_MATCH)
8490897a
MD
663
664 kill_opt="$kill_opt -s $signal"
665
666 if [ $withtap -eq "1" ]; then
667 diag "Sending SIGSTOP to $CONSUMERD_BIN pids: $(echo $PID_CONSUMERD | tr '\n' ' ')"
668 fi
669 kill $kill_opt $PID_CONSUMERD 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
670 retval=$?
671 set +x
672
673 if [ $? -eq 1 ]; then
674 if [ $withtap -eq "1" ]; then
675 fail "Sending SIGSTOP to consumer daemon"
676 fi
677 return 1
678 else
679 out=1
680 while [ $out -ne 0 ]; do
725f658e 681 pid=$(pgrep $CONSUMERD_MATCH)
8490897a
MD
682
683 # Wait until state becomes stopped for all
684 # consumers.
685 out=0
686 for consumer_pid in $pid; do
687 state=$(ps -p $consumer_pid -o state= )
688 if [[ -n "$state" && "$state" != "T" ]]; then
689 out=1
690 fi
691 done
692 sleep 0.5
693 done
694 if [ $withtap -eq "1" ]; then
695 pass "Sending SIGSTOP to consumer daemon"
696 fi
697 fi
698 return $retval
699}
700
701function sigstop_lttng_consumerd()
702{
703 sigstop_lttng_consumerd_opt 1 "$@"
704}
705
706function sigstop_lttng_consumerd_notap()
707{
708 sigstop_lttng_consumerd_opt 0 "$@"
709}
710
873c2aae
JRJ
711function list_lttng_with_opts ()
712{
7d0ad314 713 local opts=$1
2cf48300 714 $TESTDIR/../src/bin/lttng/$LTTNG_BIN list $opts 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
873c2aae
JRJ
715 ok $? "Lttng-tool list command with option $opts"
716}
717
07b86b52
JD
718function create_lttng_session_no_output ()
719{
7d0ad314 720 local sess_name=$1
07b86b52 721
2cf48300 722 $TESTDIR/../src/bin/lttng/$LTTNG_BIN create $sess_name --no-output 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
07b86b52
JD
723 ok $? "Create session $sess_name in no-output mode"
724}
725
d3e8f6bb
DG
726function create_lttng_session ()
727{
434f8068
JR
728 local withtap=$1
729 local expected_to_fail=$2
730 local sess_name=$3
731 local trace_path=$4
732 local opt=$5
d3e8f6bb 733
4c80129b 734 $TESTDIR/../src/bin/lttng/$LTTNG_BIN create $sess_name -o $trace_path $opt > $OUTPUT_DEST
873c2aae 735 ret=$?
434f8068 736 if [ $expected_to_fail -eq "1" ]; then
01513c3e 737 test "$ret" -ne "0"
434f8068
JR
738 ret=$?
739 if [ $withtap -eq "1" ]; then
740 ok $ret "Create session $sess_name in $trace_path failed as expected"
741 fi
873c2aae 742 else
434f8068
JR
743 if [ $withtap -eq "1" ]; then
744 ok $ret "Create session $sess_name in $trace_path"
745 fi
873c2aae 746 fi
434f8068 747 return $ret
d4018451
DG
748}
749
bf6ae429
JR
750function create_lttng_session_ok ()
751{
434f8068 752 create_lttng_session 1 0 "$@"
bf6ae429
JR
753}
754
755function create_lttng_session_fail ()
756{
434f8068
JR
757 create_lttng_session 1 1 "$@"
758}
759
760function create_lttng_session_notap ()
761{
762 create_lttng_session 0 0 "$@"
bf6ae429
JR
763}
764
765
827caf52 766function enable_ust_lttng_channel ()
d4018451 767{
434f8068
JR
768 local withtap=$1
769 local expected_to_fail=$2
770 local sess_name=$3
771 local channel_name=$4
772 local opt=$5
d4018451 773
4c80129b 774 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -u $channel_name -s $sess_name $opt 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
312dabc3 775 ret=$?
34ab15c5 776 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 777 test "$ret" -ne "0"
434f8068
JR
778 ret=$?
779 if [ $withtap -eq "1" ]; then
780 ok $ret "Enable channel $channel_name for session $sess_name failed as expected"
781 fi
312dabc3 782 else
434f8068
JR
783 if [ $withtap -eq "1" ]; then
784 ok $ret "Enable channel $channel_name for session $sess_name"
785 fi
312dabc3 786 fi
434f8068 787 return $ret
d4018451
DG
788}
789
827caf52
JR
790function enable_ust_lttng_channel_ok ()
791{
434f8068 792 enable_ust_lttng_channel 1 0 "$@"
827caf52
JR
793}
794
795function enable_ust_lttng_channel_fail ()
796{
434f8068
JR
797 enable_ust_lttng_channel 1 1 "$@"
798}
799
800function enable_ust_lttng_channel_notap ()
801{
802 enable_ust_lttng_channel 0 0 "$@"
827caf52
JR
803}
804
29655db7 805function disable_ust_lttng_channel()
d4018451 806{
7d0ad314
JRJ
807 local sess_name=$1
808 local channel_name=$2
d4018451 809
2cf48300 810 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-channel -u $channel_name -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
29655db7 811 ok $? "Disable channel $channel_name for session $sess_name"
d3e8f6bb
DG
812}
813
07b86b52
JD
814function enable_lttng_mmap_overwrite_kernel_channel()
815{
7d0ad314
JRJ
816 local sess_name=$1
817 local channel_name=$2
07b86b52 818
2cf48300 819 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -s $sess_name $channel_name -k --output mmap --overwrite 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
07b86b52
JD
820 ok $? "Enable channel $channel_name for session $sess_name"
821}
822
086e6add
MD
823function enable_lttng_mmap_discard_small_kernel_channel()
824{
825 local sess_name=$1
826 local channel_name=$2
827
828 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -s $sess_name $channel_name -k --output mmap --discard --subbuf-size=$(getconf PAGE_SIZE) --num-subbuf=2 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
829 ok $? "Enable small discard channel $channel_name for session $sess_name"
830}
831
832function enable_lttng_mmap_overwrite_small_kernel_channel()
833{
834 local sess_name=$1
835 local channel_name=$2
836
837 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -s $sess_name $channel_name -k --output mmap --overwrite --subbuf-size=$(getconf PAGE_SIZE) --num-subbuf=2 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
838 ok $? "Enable small discard channel $channel_name for session $sess_name"
839}
840
ebaaaf5e
JD
841function enable_lttng_mmap_overwrite_ust_channel()
842{
7d0ad314
JRJ
843 local sess_name=$1
844 local channel_name=$2
ebaaaf5e 845
2cf48300 846 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -s $sess_name $channel_name -u --output mmap --overwrite 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
ebaaaf5e
JD
847 ok $? "Enable channel $channel_name for session $sess_name"
848}
849
d3e8f6bb
DG
850function enable_ust_lttng_event ()
851{
434f8068
JR
852 local withtap=$1
853 local expected_to_fail=$2
854 local sess_name=$3
855 local event_name="$4"
856 local channel_name=$5
ebaaaf5e
JD
857
858 if [ -z $channel_name ]; then
859 # default channel if none specified
860 chan=""
861 else
862 chan="-c $channel_name"
863 fi
d3e8f6bb 864
2cf48300 865 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -u 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 866 ret=$?
c4926bb5 867 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 868 test $ret -ne "0"
434f8068
JR
869 ret=$?
870 if [[ $withtap -eq "1" ]]; then
871 ok $ret "Enable ust event $event_name for session $session_name failed as expected"
872 fi
01513c3e 873 else
434f8068
JR
874 if [[ $withtap -eq "1" ]]; then
875 ok $ret "Enable ust event $event_name for session $sess_name"
876 fi
01513c3e 877 fi
434f8068 878 return $ret
26b53d3b
DG
879}
880
c4926bb5
JR
881function enable_ust_lttng_event_ok ()
882{
434f8068 883 enable_ust_lttng_event 1 0 "$@"
c4926bb5
JR
884}
885
886function enable_ust_lttng_event_fail ()
887{
434f8068
JR
888 enable_ust_lttng_event 1 1 "$@"
889}
890
891function enable_ust_lttng_event_notap ()
892{
893 enable_ust_lttng_event 0 0 "$@"
c4926bb5
JR
894}
895
37175ce4
DG
896function enable_jul_lttng_event()
897{
898 sess_name=$1
899 event_name="$2"
900 channel_name=$3
901
902 if [ -z $channel_name ]; then
903 # default channel if none specified
904 chan=""
905 else
906 chan="-c $channel_name"
907 fi
908
2cf48300 909 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -j 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
37175ce4
DG
910 ok $? "Enable JUL event $event_name for session $sess_name"
911}
912
b2064f54
DG
913function enable_jul_lttng_event_loglevel()
914{
7d0ad314
JRJ
915 local sess_name=$1
916 local event_name="$2"
917 local loglevel=$3
918 local channel_name=$4
b2064f54
DG
919
920 if [ -z $channel_name ]; then
921 # default channel if none specified
922 chan=""
923 else
924 chan="-c $channel_name"
925 fi
926
2cf48300 927 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event --loglevel $loglevel "$event_name" $chan -s $sess_name -j 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
b2064f54
DG
928 ok $? "Enable JUL event $event_name for session $sess_name with loglevel $loglevel"
929}
930
504d4ace
DG
931function enable_log4j_lttng_event()
932{
933 sess_name=$1
934 event_name="$2"
935 channel_name=$3
936
937 if [ -z $channel_name ]; then
938 # default channel if none specified
939 chan=""
940 else
941 chan="-c $channel_name"
942 fi
943
2cf48300 944 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -l 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
504d4ace
DG
945 ok $? "Enable LOG4J event $event_name for session $sess_name"
946}
947
948function enable_log4j_lttng_event_loglevel()
949{
950 local sess_name=$1
951 local event_name="$2"
952 local loglevel=$3
953 local channel_name=$4
954
955 if [ -z $channel_name ]; then
956 # default channel if none specified
957 chan=""
958 else
959 chan="-c $channel_name"
960 fi
961
2cf48300 962 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event --loglevel $loglevel "$event_name" $chan -s $sess_name -l 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
504d4ace
DG
963 ok $? "Enable LOG4J event $event_name for session $sess_name with loglevel $loglevel"
964}
965
0e115563
DG
966function enable_python_lttng_event()
967{
968 sess_name=$1
969 event_name="$2"
970 channel_name=$3
971
972 if [ -z $channel_name ]; then
973 # default channel if none specified
974 chan=""
975 else
976 chan="-c $channel_name"
977 fi
978
2cf48300 979 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -p 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
0e115563
DG
980 ok $? "Enable Python event $event_name for session $sess_name"
981}
982
983function enable_python_lttng_event_loglevel()
984{
985 local sess_name=$1
986 local event_name="$2"
987 local loglevel=$3
988 local channel_name=$4
989
990 if [ -z $channel_name ]; then
991 # default channel if none specified
992 chan=""
993 else
994 chan="-c $channel_name"
995 fi
996
2cf48300 997 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event --loglevel $loglevel "$event_name" $chan -s $sess_name -p 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
0e115563
DG
998 ok $? "Enable Python event $event_name for session $sess_name with loglevel $loglevel"
999}
1000
26b53d3b
DG
1001function enable_ust_lttng_event_filter()
1002{
7d0ad314
JRJ
1003 local sess_name="$1"
1004 local event_name="$2"
1005 local filter="$3"
26b53d3b 1006
2cf48300 1007 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u --filter "$filter" 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
29655db7 1008 ok $? "Enable event $event_name with filtering for session $sess_name"
26b53d3b
DG
1009}
1010
1011function enable_ust_lttng_event_loglevel()
1012{
7d0ad314
JRJ
1013 local sess_name="$1"
1014 local event_name="$2"
1015 local loglevel="$3"
26b53d3b 1016
2cf48300 1017 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u --loglevel $loglevel 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
29655db7 1018 ok $? "Enable event $event_name with loglevel $loglevel"
26b53d3b
DG
1019}
1020
1021function enable_ust_lttng_event_loglevel_only()
1022{
7d0ad314
JRJ
1023 local sess_name="$1"
1024 local event_name="$2"
1025 local loglevel="$3"
26b53d3b 1026
2cf48300 1027 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u --loglevel-only $loglevel 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
29655db7 1028 ok $? "Enable event $event_name with loglevel-only $loglevel"
26b53d3b
DG
1029}
1030
1031function disable_ust_lttng_event ()
1032{
7d0ad314
JRJ
1033 local sess_name="$1"
1034 local event_name="$2"
01513c3e 1035 local channel_name="$3"
26b53d3b 1036
01513c3e
JRJ
1037 if [ -z $channel_name ]; then
1038 # default channel if none specified
1039 chan=""
1040 else
1041 chan="-c $channel_name"
1042 fi
1043
2cf48300 1044 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name $chan -u 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
29655db7 1045 ok $? "Disable event $event_name for session $sess_name"
d3e8f6bb
DG
1046}
1047
1d842d5a
DG
1048function disable_jul_lttng_event ()
1049{
1050 local sess_name="$1"
1051 local event_name="$2"
1052
1053 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name -j >/dev/null 2>&1
1054 ok $? "Disable JUL event $event_name for session $sess_name"
1055}
1056
504d4ace
DG
1057function disable_log4j_lttng_event ()
1058{
1059 local sess_name="$1"
1060 local event_name="$2"
1061
1062 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name -l >/dev/null 2>&1
1063 ok $? "Disable LOG4J event $event_name for session $sess_name"
1064}
1065
0e115563
DG
1066function disable_python_lttng_event ()
1067{
1068 local sess_name="$1"
1069 local event_name="$2"
1070
2cf48300 1071 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name -p 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
0e115563
DG
1072 ok $? "Disable Python event $event_name for session $sess_name"
1073}
1074
fb3268e3 1075function start_lttng_tracing ()
d3e8f6bb 1076{
e563bbdb
JR
1077 local expected_to_fail=$1
1078 local sess_name=$2
d3e8f6bb 1079
2cf48300 1080 $TESTDIR/../src/bin/lttng/$LTTNG_BIN start $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 1081 ret=$?
e563bbdb 1082 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 1083 test "$ret" -ne "0"
34ab15c5 1084 ok $? "Start tracing for session $sess_name failed as expected"
01513c3e
JRJ
1085 else
1086 ok $ret "Start tracing for session $sess_name"
1087 fi
d3e8f6bb
DG
1088}
1089
e563bbdb
JR
1090function start_lttng_tracing_ok ()
1091{
1092 start_lttng_tracing 0 "$@"
1093}
1094
1095function start_lttng_tracing_fail ()
1096{
1097 start_lttng_tracing 1 "$@"
1098}
1099
fb3268e3 1100function stop_lttng_tracing ()
d3e8f6bb 1101{
96340a01
JR
1102 local expected_to_fail=$1
1103 local sess_name=$2
d3e8f6bb 1104
2cf48300 1105 $TESTDIR/../src/bin/lttng/$LTTNG_BIN stop $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 1106 ret=$?
96340a01 1107 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 1108 test "$ret" -ne "0"
34ab15c5 1109 ok $? "Stop lttng tracing for session $sess_name failed as expected"
01513c3e
JRJ
1110 else
1111 ok $ret "Stop lttng tracing for session $sess_name"
1112 fi
d3e8f6bb
DG
1113}
1114
96340a01
JR
1115function stop_lttng_tracing_ok ()
1116{
1117 stop_lttng_tracing 0 "$@"
1118}
1119
1120function stop_lttng_tracing_fail ()
1121{
1122 stop_lttng_tracing 1 "$@"
1123}
1124
d3e8f6bb
DG
1125function destroy_lttng_session ()
1126{
854382b8
JR
1127 local withtap=$1
1128 local expected_to_fail=$2
1129 local sess_name=$3
d3e8f6bb 1130
2cf48300 1131 $TESTDIR/../src/bin/lttng/$LTTNG_BIN destroy $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 1132 ret=$?
96340a01 1133 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 1134 test "$ret" -ne "0"
854382b8
JR
1135 ret=$?
1136 if [ $withtap -eq "1" ]; then
1137 ok $ret "Destroy session $sess_name failed as expected"
1138 fi
01513c3e 1139 else
854382b8
JR
1140 if [ $withtap -eq "1" ]; then
1141 ok $ret "Destroy session $sess_name"
1142 fi
01513c3e 1143 fi
d3e8f6bb
DG
1144}
1145
67b4c664
JR
1146function destroy_lttng_session_ok ()
1147{
854382b8 1148 destroy_lttng_session 1 0 "$@"
67b4c664
JR
1149
1150}
1151
1152function destroy_lttng_session_fail ()
1153{
854382b8 1154 destroy_lttng_session 1 1 "$@"
67b4c664
JR
1155}
1156
854382b8
JR
1157function destroy_lttng_session_notap ()
1158{
1159 destroy_lttng_session 0 0 "$@"
1160}
67b4c664 1161
873c2aae
JRJ
1162function destroy_lttng_sessions ()
1163{
2cf48300 1164 $TESTDIR/../src/bin/lttng/$LTTNG_BIN destroy --all 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
873c2aae
JRJ
1165 ok $? "Destroy all lttng sessions"
1166}
1167
07b86b52
JD
1168function lttng_snapshot_add_output ()
1169{
b5633831
JR
1170 local expected_to_fail=$1
1171 local sess_name=$2
1172 local trace_path=$3
07b86b52 1173
2cf48300 1174 $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot add-output -s $sess_name file://$trace_path 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 1175 ret=$?
b5633831 1176 if [[ $expected_to_fail -eq 1 ]]; then
01513c3e 1177 test "$ret" -ne "0"
34ab15c5 1178 ok $? "Added snapshot output file://$trace_path failed as expected"
01513c3e
JRJ
1179 else
1180 ok $ret "Added snapshot output file://$trace_path"
1181 fi
07b86b52
JD
1182}
1183
b5633831
JR
1184function lttng_snapshot_add_output_ok ()
1185{
1186 lttng_snapshot_add_output 0 "$@"
1187}
1188
1189function lttng_snapshot_add_output_fail ()
1190{
1191 lttng_snapshot_add_output 1 "$@"
1192}
1193
26402e0c
DG
1194function lttng_snapshot_del_output ()
1195{
31580dc7
JR
1196 local expected_to_fail=$1
1197 local sess_name=$2
1198 local id=$3
26402e0c 1199
2cf48300 1200 $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot del-output -s $sess_name $id 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 1201 ret=$?
31580dc7 1202 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 1203 test "$ret" -ne "0"
34ab15c5 1204 ok $? "Deleted snapshot output id $id failed as expected"
01513c3e
JRJ
1205 else
1206 ok $ret "Deleted snapshot output id $id"
1207 fi
26402e0c
DG
1208}
1209
31580dc7
JR
1210function lttng_snapshot_del_output_ok ()
1211{
1212 lttng_snapshot_del_output 0 "$@"
1213}
1214
1215function lttng_snapshot_del_output_fail ()
1216{
1217 lttng_snapshot_del_output 1 "$@"
1218}
1219
07b86b52
JD
1220function lttng_snapshot_record ()
1221{
7d0ad314
JRJ
1222 local sess_name=$1
1223 local trace_path=$2
07b86b52 1224
2cf48300 1225 $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot record -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
07b86b52
JD
1226 ok $? "Snapshot recorded"
1227}
1228
01513c3e
JRJ
1229function lttng_snapshot_list ()
1230{
1231 local sess_name=$1
2cf48300 1232 $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot list-output -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e
JRJ
1233 ok $? "Snapshot list"
1234}
1235
e02b109b
DG
1236function lttng_save()
1237{
1238 local sess_name=$1
1239 local opts=$2
1240
2cf48300 1241 $TESTDIR/../src/bin/lttng/$LTTNG_BIN save $sess_name $opts 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
34ab15c5 1242 ok $? "Session saved"
e02b109b
DG
1243}
1244
1245function lttng_load()
1246{
192ac418
JR
1247 local expected_to_fail=$1
1248 local opts=$2
e02b109b 1249
2cf48300 1250 $TESTDIR/../src/bin/lttng/$LTTNG_BIN load $opts 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
192ac418
JR
1251 ret=$?
1252 if [[ $expected_to_fail -eq "1" ]]; then
1253 test $ret -ne "0"
1254 ok $? "Load command failed as expected with opts: $opts"
1255 else
1256 ok $ret "Load command with opts: $opts"
1257 fi
1258}
1259
1260function lttng_load_ok()
1261{
1262 lttng_load 0 "$@"
1263}
1264
1265function lttng_load_fail()
1266{
1267 lttng_load 1 "$@"
e02b109b
DG
1268}
1269
e83a8bdb
JR
1270function lttng_track()
1271{
1272 local expected_to_fail=$1
1273 local opts=$2
1274 $TESTDIR/../src/bin/lttng/$LTTNG_BIN track $opts >$OUTPUT_DEST
1275 ret=$?
1276 if [[ $expected_to_fail -eq "1" ]]; then
1277 test $ret -ne "0"
1278 ok $? "Track command failed as expected with opts: $opts"
1279 else
34ab15c5 1280 ok $ret "Track command with opts: $opts"
e83a8bdb
JR
1281 fi
1282}
1283
1284function lttng_track_ok()
1285{
1286 lttng_track 0 "$@"
1287}
1288
1289function lttng_track_fail()
1290{
1291 lttng_track 1 "$@"
1292}
1293
1294function lttng_untrack()
1295{
1296 local expected_to_fail=$1
1297 local opts=$2
1298 $TESTDIR/../src/bin/lttng/$LTTNG_BIN untrack $opts >$OUTPUT_DEST
1299 ret=$?
1300 if [[ $expected_to_fail -eq "1" ]]; then
1301 test $ret -ne "0"
1302 ok $? "Untrack command failed as expected with opts: $opts"
1303 else
34ab15c5 1304 ok $ret "Untrack command with opts: $opts"
e83a8bdb
JR
1305 fi
1306}
1307
1308function lttng_untrack_ok()
1309{
1310 lttng_untrack 0 "$@"
1311}
1312
1313function lttng_untrack_fail()
1314{
1315 lttng_untrack 1 "$@"
1316}
1317
c8e51d15
FD
1318function lttng_track_pid_ok()
1319{
1320 PID=$1
1321 "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" track --kernel --pid=$PID 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1322 ok $? "Lttng track pid on the kernel domain"
1323}
1324
1325function lttng_untrack_kernel_all_ok()
1326{
1327 "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" untrack --kernel --pid --all 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1328 ok $? "Lttng untrack all pid on the kernel domain"
1329}
1330
8dcef147
JR
1331function add_context_lttng()
1332{
1333 local expected_to_fail="$1"
1334 local domain="$2"
1335 local session_name="$3"
1336 local channel_name="$4"
1337 local type="$5"
1338
1339 $TESTDIR/../src/bin/lttng/$LTTNG_BIN add-context -s $session_name -c $channel_name -t $type $domain 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1340 ret=$?
1341 if [[ $expected_to_fail -eq "1" ]]; then
1342 test $ret -ne "0"
1343 ok $? "Add context command failed as expected for type: $type"
1344 else
1345 ok $ret "Add context command for type: $type"
1346 fi
1347}
1348
1349function add_context_ust_ok()
1350{
1351 add_context_lttng 0 -u "$@"
1352}
1353
1354function add_context_ust_fail()
1355{
1356 add_context_lttng 1 -u "$@"
1357}
1358
1359function add_context_kernel_ok()
1360{
1361 add_context_lttng 0 -k "$@"
1362}
1363
1364function add_context_kernel_fail()
1365{
1366 add_context_lttng 1 -k "$@"
1367}
1368
d3e8f6bb
DG
1369function trace_matches ()
1370{
7d0ad314
JRJ
1371 local event_name=$1
1372 local nr_iter=$2
1373 local trace_path=$3
d3e8f6bb 1374
317eef93 1375 which $BABELTRACE_BIN >/dev/null
29655db7 1376 skip $? -ne 0 "Babeltrace binary not found. Skipping trace matches"
d3e8f6bb 1377
7d0ad314 1378 local count=$($BABELTRACE_BIN $trace_path | grep $event_name | wc -l)
29655db7 1379
d3e8f6bb 1380 if [ "$count" -ne "$nr_iter" ]; then
29655db7 1381 fail "Trace match"
af91abc9 1382 diag "$count matching events found in trace"
d3e8f6bb 1383 else
29655db7 1384 pass "Trace match"
d3e8f6bb
DG
1385 fi
1386}
f4e40ab6 1387
d53addeb
DG
1388function trace_match_only()
1389{
1390 local event_name=$1
1391 local nr_iter=$2
1392 local trace_path=$3
1393
1394 which $BABELTRACE_BIN >/dev/null
1395 skip $? -ne 0 "Babeltrace binary not found. Skipping trace matches"
1396
1397 local count=$($BABELTRACE_BIN $trace_path | grep $event_name | wc -l)
1398 local total=$($BABELTRACE_BIN $trace_path | wc -l)
1399
1400 if [ "$nr_iter" -eq "$count" ] && [ "$total" -eq "$nr_iter" ]; then
1401 pass "Trace match with $total event $event_name"
1402 else
1403 fail "Trace match"
1404 diag "$total event(s) found, expecting $nr_iter of event $event_name and only found $count"
1405 fi
1406}
1407
f4e40ab6
DG
1408function validate_trace
1409{
7d0ad314
JRJ
1410 local event_name=$1
1411 local trace_path=$2
f4e40ab6
DG
1412
1413 which $BABELTRACE_BIN >/dev/null
29655db7
CB
1414 if [ $? -ne 0 ]; then
1415 skip 0 "Babeltrace binary not found. Skipping trace validation"
f4e40ab6
DG
1416 fi
1417
07b86b52
JD
1418 OLDIFS=$IFS
1419 IFS=","
1420 for i in $event_name; do
1421 traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | grep $i | wc -l)
1422 if [ "$traced" -ne 0 ]; then
ebaaaf5e 1423 pass "Validate trace for event $i, $traced events"
07b86b52
JD
1424 else
1425 fail "Validate trace for event $i"
1426 diag "Found $traced occurences of $i"
1427 fi
1428 done
1429 ret=$?
1430 IFS=$OLDIFS
1431 return $ret
f4e40ab6 1432}
8cfcd41c 1433
54cd6107
JD
1434function validate_trace_count
1435{
1436 local event_name=$1
1437 local trace_path=$2
1438 local expected_count=$3
1439
1440 which $BABELTRACE_BIN >/dev/null
1441 if [ $? -ne 0 ]; then
1442 skip 0 "Babeltrace binary not found. Skipping trace validation"
1443 fi
1444
1445 cnt=0
1446 OLDIFS=$IFS
1447 IFS=","
1448 for i in $event_name; do
1449 traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | grep $i | wc -l)
1450 if [ "$traced" -ne 0 ]; then
1451 pass "Validate trace for event $i, $traced events"
1452 else
1453 fail "Validate trace for event $i"
1454 diag "Found $traced occurences of $i"
1455 fi
1456 cnt=$(($cnt + $traced))
1457 done
1458 IFS=$OLDIFS
1459 test $cnt -eq $expected_count
1460 ok $? "Read a total of $cnt events, expected $expected_count"
1461}
1462
086e6add
MD
1463function trace_first_line
1464{
1465 local trace_path=$1
1466
1467 which $BABELTRACE_BIN >/dev/null
1468 if [ $? -ne 0 ]; then
1469 skip 0 "Babeltrace binary not found. Skipping trace validation"
1470 fi
1471
1472 $BABELTRACE_BIN $trace_path 2>/dev/null | head -n 1
1473}
1474
8cfcd41c
MD
1475function validate_trace_exp()
1476{
1477 local event_exp=$1
1478 local trace_path=$2
1479
1480 which $BABELTRACE_BIN >/dev/null
1481 skip $? -ne 0 "Babeltrace binary not found. Skipping trace validation"
1482
1483 traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | grep ${event_exp} | wc -l)
1484 if [ "$traced" -ne 0 ]; then
1485 pass "Validate trace for expression '${event_exp}', $traced events"
1486 else
1487 fail "Validate trace for expression '${event_exp}'"
1488 diag "Found $traced occurences of '${event_exp}'"
1489 fi
1490 ret=$?
1491 return $ret
1492}
1493
1494function validate_trace_only_exp()
1495{
1496 local event_exp=$1
1497 local trace_path=$2
1498
1499 which $BABELTRACE_BIN >/dev/null
1500 skip $? -ne 0 "Babeltrace binary not found. Skipping trace matches"
1501
1502 local count=$($BABELTRACE_BIN $trace_path | grep ${event_exp} | wc -l)
1503 local total=$($BABELTRACE_BIN $trace_path | wc -l)
1504
1505 if [ "$count" -ne 0 ] && [ "$total" -eq "$count" ]; then
1506 pass "Trace match with $total for expression '${event_exp}"
1507 else
1508 fail "Trace match"
1509 diag "$total syscall event(s) found, only syscalls matching expression '${event_exp}' ($count occurrences) are expected"
1510 fi
1511 ret=$?
1512 return $ret
1513}
1514
1515function validate_trace_empty()
1516{
1517 local trace_path=$1
1518
1519 which $BABELTRACE_BIN >/dev/null
1520 if [ $? -ne 0 ]; then
1521 skip 0 "Babeltrace binary not found. Skipping trace validation"
1522 fi
1523
2462eee7
JD
1524 events=$($BABELTRACE_BIN $trace_path 2>/dev/null)
1525 ret=$?
1526 if [ $ret -ne 0 ]; then
1527 fail "Failed to parse trace"
1528 return $ret
1529 fi
1530
1531 traced=$(echo -n "$events" | wc -l)
8cfcd41c
MD
1532 if [ "$traced" -eq 0 ]; then
1533 pass "Validate empty trace"
1534 else
1535 fail "Validate empty trace"
1536 diag "Found $traced events in trace"
1537 fi
1538 ret=$?
1539 return $ret
1540}
801236b0 1541
eded6438 1542function regenerate_metadata ()
801236b0
JD
1543{
1544 local expected_to_fail=$1
1545 local sess_name=$2
1546
eded6438 1547 $TESTDIR/../src/bin/lttng/$LTTNG_BIN regenerate metadata -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
801236b0
JD
1548 ret=$?
1549 if [[ $expected_to_fail -eq "1" ]]; then
1550 test "$ret" -ne "0"
eded6438 1551 ok $? "Expected fail on regenerate metadata $sess_name"
801236b0
JD
1552 else
1553 ok $ret "Metadata regenerate $sess_name"
1554 fi
1555}
1556
eded6438 1557function regenerate_metadata_ok ()
801236b0 1558{
eded6438 1559 regenerate_metadata 0 "$@"
801236b0
JD
1560}
1561
eded6438 1562function regenerate_metadata_fail ()
801236b0 1563{
eded6438 1564 regenerate_metadata 1 "$@"
801236b0 1565}
512eb148 1566
54cd6107
JD
1567function regenerate_statedump ()
1568{
1569 local expected_to_fail=$1
1570 local sess_name=$2
1571
1572 $TESTDIR/../src/bin/lttng/$LTTNG_BIN regenerate statedump -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1573 ret=$?
1574 if [[ $expected_to_fail -eq "1" ]]; then
1575 test "$ret" -ne "0"
1576 ok $? "Expected fail on regenerate statedump $sess_name"
1577 else
1578 ok $ret "Metadata regenerate $sess_name"
1579 fi
1580}
1581
1582function regenerate_statedump_ok ()
1583{
1584 regenerate_statedump 0 "$@"
1585}
1586
1587function regenerate_statedump_fail ()
1588{
1589 regenerate_statedump 1 "$@"
1590}
1591
e7716c6a
JD
1592function rotate_session ()
1593{
1594 local expected_to_fail=$1
1595 local sess_name=$2
1596
1597 $TESTDIR/../src/bin/lttng/$LTTNG_BIN rotate $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1598 ret=$?
1599 if [[ $expected_to_fail -eq "1" ]]; then
1600 test "$ret" -ne "0"
1601 ok $? "Expected fail on rotate session $sess_name"
1602 else
1603 ok $ret "Rotate session $sess_name"
1604 fi
1605}
1606
1607function rotate_session_ok ()
1608{
1609 rotate_session 0 "$@"
1610}
1611
1612function rotate_session_fail ()
1613{
1614 rotate_session 1 "$@"
1615}
1616
512eb148
JD
1617function destructive_tests_enabled ()
1618{
1619 if [ ${LTTNG_ENABLE_DESTRUCTIVE_TESTS} = "will-break-my-system" ]; then
1620 return 0
1621 else
1622 return 1
1623 fi
1624}
e7716c6a
JD
1625
1626function lttng_enable_rotation_timer ()
1627{
1628 local expected_to_fail=$1
1629 local sess_name=$2
1630 local period=$3
1631
1632 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-rotation -s $sess_name --timer $period 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1633 ret=$?
1634 if [[ $expected_to_fail -eq "1" ]]; then
1635 test "$ret" -ne "0"
1636 ok $? "Expected fail on rotate session $sess_name"
1637 else
1638 ok $ret "Rotate session $sess_name"
1639 fi
1640}
1641
1642function lttng_enable_rotation_timer_ok ()
1643{
1644 lttng_enable_rotation_timer 0 $@
1645}
1646
1647function lttng_enable_rotation_timer_fail ()
1648{
1649 lttng_enable_rotation_timer 1 $@
1650}
1651
1652function lttng_enable_rotation_size ()
1653{
1654 local expected_to_fail=$1
1655 local sess_name=$2
1656 local size=$3
1657
1658 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-rotation -s $sess_name --size $size 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1659 ret=$?
1660 if [[ $expected_to_fail -eq "1" ]]; then
1661 test "$ret" -ne "0"
1662 ok $? "Expected fail on rotate session $sess_name"
1663 else
1664 ok $ret "Rotate session $sess_name"
1665 fi
1666}
1667
1668function lttng_enable_rotation_size_ok ()
1669{
1670 lttng_enable_rotation_size 0 $@
1671}
1672
1673function lttng_enable_rotation_size_fail ()
1674{
1675 lttng_enable_rotation_size 1 $@
1676}
This page took 0.130362 seconds and 5 git commands to generate.