Tests: add consumer testpoint to pause data consumption
[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
4a180d9f 128function enable_kernel_lttng_event
f4e40ab6 129{
4a180d9f
MD
130 local expected_to_fail="$1"
131 local sess_name="$2"
132 local event_name="$3"
133 local channel_name="$4"
f4e40ab6 134
4a180d9f 135 if [ -z "$event_name" ]; then
f4e40ab6 136 # Enable all event if no event name specified
29655db7 137 event_name="-a"
f4e40ab6
DG
138 fi
139
4a180d9f 140 if [ -z "$channel_name" ]; then
07b86b52
JD
141 # default channel if none specified
142 chan=""
143 else
144 chan="-c $channel_name"
145 fi
146
2cf48300 147 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -k 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
4a180d9f
MD
148 ret=$?
149 if [[ $expected_to_fail -eq "1" ]]; then
150 test $ret -ne "0"
151 ok $? "Enable kernel event $event_name for session $session_name on channel $channel_name failed as expected"
152 else
153 ok $ret "Enable kernel event $event_name for session $sess_name"
154 fi
155}
156
157function enable_kernel_lttng_event_ok ()
158{
159 enable_kernel_lttng_event 0 "$@"
160}
161
162function enable_kernel_lttng_event_fail ()
163{
164 enable_kernel_lttng_event 1 "$@"
165}
166
167# Old interface
168function lttng_enable_kernel_event
169{
170 enable_kernel_lttng_event_ok "$@"
f4e40ab6
DG
171}
172
8cfcd41c
MD
173function lttng_enable_kernel_syscall()
174{
175 local expected_to_fail=$1
176 local sess_name=$2
177 local syscall_name=$3
178 local channel_name=$4
179
180 if [ -z $syscall_name ]; then
181 # Enable all event if no syscall name specified
182 syscall_name="-a"
183 fi
184
185 if [ -z $channel_name ]; then
186 # default channel if none specified
187 chan=""
188 else
189 chan="-c $channel_name"
190 fi
191
2cf48300 192 $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
193 ret=$?
194 if [[ $expected_to_fail -eq "1" ]]; then
195 test $ret -ne "0"
196 ok $? "Enable kernel syscall $syscall_name for session $sess_name on channel $channel_name fail as expected"
197 else
198 ok $ret "Enable kernel syscall $syscall_name for session $sess_name on channel $channel_name"
199 fi
200}
201
202function lttng_enable_kernel_syscall_ok()
203{
421b83dc 204 lttng_enable_kernel_syscall 0 "$@"
8cfcd41c
MD
205}
206
207function lttng_enable_kernel_syscall_fail()
208{
421b83dc 209 lttng_enable_kernel_syscall 1 "$@"
8cfcd41c
MD
210}
211
212function lttng_disable_kernel_syscall()
213{
214 local expected_to_fail=$1
215 local sess_name=$2
216 local syscall_name=$3
217 local channel_name=$4
218
219 if [ -z $syscall_name ]; then
220 # Enable all event if no syscall name specified
221 syscall_name="-a"
222 fi
223
224 if [ -z $channel_name ]; then
225 # default channel if none specified
226 chan=""
227 else
228 chan="-c $channel_name"
229 fi
230
2cf48300 231 $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
232
233 ret=$?
234 if [[ $expected_to_fail -eq "1" ]]; then
235 test $ret -ne "0"
34ab15c5 236 ok $? "Disable kernel syscall $syscall_name for session $sess_name on channel $channel_name failed as expected"
8cfcd41c
MD
237 else
238 ok $ret "Disable kernel syscall $syscall_name for session $sess_name on channel $channel_name"
239 fi
240}
241
242function lttng_disable_kernel_syscall_ok()
243{
421b83dc 244 lttng_disable_kernel_syscall 0 "$@"
8cfcd41c
MD
245}
246
247function lttng_disable_kernel_syscall_fail()
248{
421b83dc 249 lttng_disable_kernel_syscall 1 "$@"
8cfcd41c
MD
250}
251
d96f6315
MD
252function lttng_enable_kernel_channel()
253{
254 local expected_to_fail=$1
255 local sess_name=$2
256 local channel_name=$3
257
2cf48300 258 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -k $channel_name -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
d96f6315
MD
259 ret=$?
260 if [[ $expected_to_fail -eq "1" ]]; then
261 test "$ret" -ne "0"
34ab15c5 262 ok $? "Enable channel $channel_name for session $sess_name failed as expected"
d96f6315
MD
263 else
264 ok $ret "Enable channel $channel_name for session $sess_name"
265 fi
266}
267
268function lttng_enable_kernel_channel_ok()
269{
421b83dc 270 lttng_enable_kernel_channel 0 "$@"
d96f6315
MD
271}
272
273function lttng_enable_kernel_channel_fail()
274{
421b83dc 275 lttng_enable_kernel_channel 1 "$@"
d96f6315
MD
276}
277
278function lttng_disable_kernel_channel()
279{
280 local expected_to_fail=$1
281 local sess_name=$2
282 local channel_name=$3
283
2cf48300 284 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-channel -k $channel_name -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
d96f6315
MD
285 ret=$?
286 if [[ $expected_to_fail -eq "1" ]]; then
287 test "$ret" -ne "0"
34ab15c5 288 ok $? "Disable channel $channel_name for session $sess_name failed as expected"
d96f6315 289 else
34ab15c5 290 ok $ret "Disable channel $channel_name for session $sess_name"
d96f6315
MD
291 fi
292}
293
294function lttng_disable_kernel_channel_ok()
295{
421b83dc 296 lttng_disable_kernel_channel 0 "$@"
d96f6315
MD
297}
298
299function lttng_disable_kernel_channel_fail()
300{
421b83dc 301 lttng_disable_kernel_channel 1 "$@"
d96f6315
MD
302}
303
05aa48da 304function start_lttng_relayd_opt()
f4e40ab6 305{
05aa48da
MD
306 local withtap=$1
307 local opt=$2
173af62f 308
f4e40ab6
DG
309 DIR=$(readlink -f $TESTDIR)
310
725f658e 311 if [ -z $(pgrep $RELAYD_MATCH) ]; then
2cf48300 312 $DIR/../src/bin/lttng-relayd/$RELAYD_BIN -b $opt 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
25a818af 313 #$DIR/../src/bin/lttng-relayd/$RELAYD_BIN $opt -vvv >>/tmp/relayd.log 2>&1 &
f4e40ab6 314 if [ $? -eq 1 ]; then
05aa48da
MD
315 if [ $withtap -eq "1" ]; then
316 fail "Start lttng-relayd (opt: $opt)"
317 fi
f4e40ab6
DG
318 return 1
319 else
05aa48da
MD
320 if [ $withtap -eq "1" ]; then
321 pass "Start lttng-relayd (opt: $opt)"
322 fi
f4e40ab6
DG
323 fi
324 else
29655db7 325 pass "Start lttng-relayd (opt: $opt)"
f4e40ab6 326 fi
1c362dc7 327
725f658e 328 RELAYD_PIDS=$(pgrep $RELAYD_MATCH)
f4e40ab6
DG
329}
330
05aa48da 331function start_lttng_relayd()
f4e40ab6 332{
05aa48da
MD
333 start_lttng_relayd_opt 1 "$@"
334}
335
336function start_lttng_relayd_notap()
337{
338 start_lttng_relayd_opt 0 "$@"
339}
340
341function stop_lttng_relayd_opt()
342{
343 local withtap=$1
344
05aa48da 345 if [ $withtap -eq "1" ]; then
1c362dc7 346 diag "Killing lttng-relayd (pid: $RELAYD_PIDS)"
05aa48da 347 fi
1c362dc7 348 kill $RELAYD_PIDS 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1fb23888 349 retval=$?
29655db7 350
05aa48da
MD
351 if [ $? -eq 1 ]; then
352 if [ $withtap -eq "1" ]; then
353 fail "Kill relay daemon"
354 fi
355 return 1
356 else
f4e40ab6
DG
357 out=1
358 while [ -n "$out" ]; do
725f658e 359 out=$(pgrep $RELAYD_MATCH)
f4e40ab6
DG
360 sleep 0.5
361 done
05aa48da
MD
362 if [ $withtap -eq "1" ]; then
363 pass "Kill relay daemon"
364 fi
1fb23888 365 fi
1c362dc7 366 RELAYD_PIDS=""
1fb23888
MD
367 return $retval
368}
369
05aa48da 370function stop_lttng_relayd()
1fb23888 371{
05aa48da
MD
372 stop_lttng_relayd_opt 1 "$@"
373}
1fb23888 374
05aa48da
MD
375function stop_lttng_relayd_notap()
376{
377 stop_lttng_relayd_opt 0 "$@"
f4e40ab6
DG
378}
379
05aa48da
MD
380#First arg: show tap output
381#Second argument: load path for automatic loading
382function start_lttng_sessiond_opt()
355f483d 383{
05aa48da
MD
384 local withtap=$1
385 local load_path=$2
8d51ddbc 386
4b01971f 387 local env_vars=""
b916da6b
JR
388 local consumerd=""
389 local long_bit_value=$(getconf LONG_BIT)
390
355f483d
DG
391 if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
392 # Env variable requested no session daemon
393 return
394 fi
395
b916da6b
JR
396 DIR=$(readlink -f $TESTDIR)
397
398 # Get long_bit value for 32/64 consumerd
399 case "$long_bit_value" in
400 32)
401 consumerd="--consumerd32-path=$DIR/../src/bin/lttng-consumerd/lttng-consumerd"
402 ;;
403 64)
404 consumerd="--consumerd64-path=$DIR/../src/bin/lttng-consumerd/lttng-consumerd"
405 ;;
406 *)
407 return
408 ;;
409 esac
410
4b01971f
JR
411 # Check for env. variable. Allow the use of LD_PRELOAD etc.
412 if [[ "x${LTTNG_SESSIOND_ENV_VARS}" != "x" ]]; then
413 env_vars=${LTTNG_SESSIOND_ENV_VARS}
414 fi
415
29655db7
CB
416 validate_kernel_version
417 if [ $? -ne 0 ]; then
418 fail "Start session daemon"
419 BAIL_OUT "*** Kernel too old for session daemon tests ***"
355f483d
DG
420 fi
421
d3d97763
JG
422 : ${LTTNG_SESSION_CONFIG_XSD_PATH=${DIR}/../src/common/config/}
423 export LTTNG_SESSION_CONFIG_XSD_PATH
29655db7 424
725f658e 425 if [ -z $(pgrep ${SESSIOND_MATCH}) ]; then
8d51ddbc 426 # Have a load path ?
05aa48da 427 if [ -n "$load_path" ]; then
4b01971f 428 env $env_vars $DIR/../src/bin/lttng-sessiond/$SESSIOND_BIN --load "$load_path" --background $consumerd
8d51ddbc 429 else
4b01971f 430 env $env_vars $DIR/../src/bin/lttng-sessiond/$SESSIOND_BIN --background $consumerd
8d51ddbc 431 fi
0fc2834c 432 #$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 433 status=$?
05aa48da
MD
434 if [ $withtap -eq "1" ]; then
435 ok $status "Start session daemon"
436 fi
29655db7 437 fi
725f658e 438 SESSIOND_PIDS=$(pgrep $SESSIOND_MATCH)
355f483d
DG
439}
440
05aa48da
MD
441function start_lttng_sessiond()
442{
443 start_lttng_sessiond_opt 1 "$@"
444}
445
446function start_lttng_sessiond_notap()
d3e8f6bb 447{
05aa48da
MD
448 start_lttng_sessiond_opt 0 "$@"
449}
450
451function stop_lttng_sessiond_opt()
452{
453 local withtap=$1
4c80129b
JR
454 local signal=$2
455 local kill_opt=""
05aa48da 456
355f483d
DG
457 if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
458 # Env variable requested no session daemon
459 return
460 fi
461
725f658e 462 local pids="${SESSIOND_PIDS} $(pgrep $RUNAS_MATCH)"
d3e8f6bb 463
4c80129b
JR
464 if [ -n "$2" ]; then
465 kill_opt="$kill_opt -s $signal"
466 fi
8490897a 467 if [ $withtap -eq "1" ]; then
725f658e 468 diag "Killing $SESSIOND_BIN and lt-$SESSIOND_BIN pids: $(echo $pids | tr '\n' ' ')"
8490897a 469 fi
1c362dc7 470 kill $kill_opt $pids 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
29655db7 471
d3e8f6bb 472 if [ $? -eq 1 ]; then
05aa48da
MD
473 if [ $withtap -eq "1" ]; then
474 fail "Kill sessions daemon"
475 fi
d3e8f6bb 476 else
5fa32580
DG
477 out=1
478 while [ -n "$out" ]; do
725f658e 479 out=$(pgrep ${SESSIOND_MATCH})
5fa32580
DG
480 sleep 0.5
481 done
f7613992
MD
482 out=1
483 while [ -n "$out" ]; do
725f658e 484 out=$(pgrep $CONSUMERD_MATCH)
f7613992
MD
485 sleep 0.5
486 done
1c362dc7
JR
487
488 SESSIOND_PIDS=""
05aa48da
MD
489 if [ $withtap -eq "1" ]; then
490 pass "Kill session daemon"
491 fi
d3e8f6bb
DG
492 fi
493}
494
05aa48da
MD
495function stop_lttng_sessiond()
496{
497 stop_lttng_sessiond_opt 1 "$@"
498}
499
500function stop_lttng_sessiond_notap()
501{
502 stop_lttng_sessiond_opt 0 "$@"
503}
504
8490897a
MD
505function sigstop_lttng_sessiond_opt()
506{
507 local withtap=$1
508 local signal=SIGSTOP
509 local kill_opt=""
510
511 if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
512 # Env variable requested no session daemon
513 return
514 fi
515
725f658e 516 PID_SESSIOND="$(pgrep ${SESSIOND_MATCH}) $(pgrep $RUNAS_MATCH)"
8490897a
MD
517
518 kill_opt="$kill_opt -s $signal"
519
520 if [ $withtap -eq "1" ]; then
725f658e 521 diag "Sending SIGSTOP to lt-$SESSIOND_BIN and $SESSIOND_BIN pids: $(echo $PID_SESSIOND | tr '\n' ' ')"
8490897a
MD
522 fi
523 kill $kill_opt $PID_SESSIOND 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
524
525 if [ $? -eq 1 ]; then
526 if [ $withtap -eq "1" ]; then
527 fail "Sending SIGSTOP to session daemon"
528 fi
529 else
530 out=1
531 while [ $out -ne 0 ]; do
725f658e 532 pid=$(pgrep $SESSIOND_MATCH)
8490897a
MD
533
534 # Wait until state becomes stopped for session
535 # daemon(s).
536 out=0
537 for sessiond_pid in $pid; do
538 state=$(ps -p $sessiond_pid -o state= )
539 if [[ -n "$state" && "$state" != "T" ]]; then
540 out=1
541 fi
542 done
543 sleep 0.5
544 done
545 if [ $withtap -eq "1" ]; then
546 pass "Sending SIGSTOP to session daemon"
547 fi
548 fi
549}
550
551function sigstop_lttng_sessiond()
552{
553 sigstop_lttng_sessiond_opt 1 "$@"
554}
555
556function sigstop_lttng_sessiond_notap()
557{
558 sigstop_lttng_sessiond_opt 0 "$@"
559}
560
4c80129b
JR
561function stop_lttng_consumerd_opt()
562{
563 local withtap=$1
564 local signal=$2
565 local kill_opt=""
566
725f658e 567 PID_CONSUMERD=$(pgrep $CONSUMERD_MATCH)
4c80129b
JR
568
569 if [ -n "$2" ]; then
570 kill_opt="$kill_opt -s $signal"
571 fi
572
573 if [ $withtap -eq "1" ]; then
8490897a 574 diag "Killing $CONSUMERD_BIN pids: $(echo $PID_CONSUMERD | tr '\n' ' ')"
4c80129b
JR
575 fi
576 kill $kill_opt $PID_CONSUMERD 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
577 retval=$?
578 set +x
579
580 if [ $? -eq 1 ]; then
581 if [ $withtap -eq "1" ]; then
582 fail "Kill consumer daemon"
583 fi
584 return 1
585 else
586 out=1
587 while [ $out -ne 0 ]; do
725f658e 588 pid=$(pgrep $CONSUMERD_MATCH)
4c80129b
JR
589
590 # If consumerds are still present check their status.
591 # A zombie status qualifies the consumerd as *killed*
592 out=0
593 for consumer_pid in $pid; do
594 state=$(ps -p $consumer_pid -o state= )
595 if [[ -n "$state" && "$state" != "Z" ]]; then
596 out=1
597 fi
598 done
599 sleep 0.5
600 done
601 if [ $withtap -eq "1" ]; then
602 pass "Kill consumer daemon"
603 fi
604 fi
605 return $retval
606}
607
608function stop_lttng_consumerd()
609{
610 stop_lttng_consumerd_opt 1 "$@"
611}
612
613function stop_lttng_consumerd_notap()
614{
615 stop_lttng_consumerd_opt 0 "$@"
616}
617
8490897a
MD
618function sigstop_lttng_consumerd_opt()
619{
620 local withtap=$1
621 local signal=SIGSTOP
622 local kill_opt=""
623
725f658e 624 PID_CONSUMERD=$(pgrep $CONSUMERD_MATCH)
8490897a
MD
625
626 kill_opt="$kill_opt -s $signal"
627
628 if [ $withtap -eq "1" ]; then
629 diag "Sending SIGSTOP to $CONSUMERD_BIN pids: $(echo $PID_CONSUMERD | tr '\n' ' ')"
630 fi
631 kill $kill_opt $PID_CONSUMERD 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
632 retval=$?
633 set +x
634
635 if [ $? -eq 1 ]; then
636 if [ $withtap -eq "1" ]; then
637 fail "Sending SIGSTOP to consumer daemon"
638 fi
639 return 1
640 else
641 out=1
642 while [ $out -ne 0 ]; do
725f658e 643 pid=$(pgrep $CONSUMERD_MATCH)
8490897a
MD
644
645 # Wait until state becomes stopped for all
646 # consumers.
647 out=0
648 for consumer_pid in $pid; do
649 state=$(ps -p $consumer_pid -o state= )
650 if [[ -n "$state" && "$state" != "T" ]]; then
651 out=1
652 fi
653 done
654 sleep 0.5
655 done
656 if [ $withtap -eq "1" ]; then
657 pass "Sending SIGSTOP to consumer daemon"
658 fi
659 fi
660 return $retval
661}
662
663function sigstop_lttng_consumerd()
664{
665 sigstop_lttng_consumerd_opt 1 "$@"
666}
667
668function sigstop_lttng_consumerd_notap()
669{
670 sigstop_lttng_consumerd_opt 0 "$@"
671}
672
873c2aae
JRJ
673function list_lttng_with_opts ()
674{
7d0ad314 675 local opts=$1
2cf48300 676 $TESTDIR/../src/bin/lttng/$LTTNG_BIN list $opts 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
873c2aae
JRJ
677 ok $? "Lttng-tool list command with option $opts"
678}
679
07b86b52
JD
680function create_lttng_session_no_output ()
681{
7d0ad314 682 local sess_name=$1
07b86b52 683
2cf48300 684 $TESTDIR/../src/bin/lttng/$LTTNG_BIN create $sess_name --no-output 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
07b86b52
JD
685 ok $? "Create session $sess_name in no-output mode"
686}
687
d3e8f6bb
DG
688function create_lttng_session ()
689{
bf6ae429
JR
690 local expected_to_fail=$1
691 local sess_name=$2
692 local trace_path=$3
4c80129b 693 local opt=$4
d3e8f6bb 694
4c80129b 695 $TESTDIR/../src/bin/lttng/$LTTNG_BIN create $sess_name -o $trace_path $opt > $OUTPUT_DEST
873c2aae 696 ret=$?
bf6ae429 697 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 698 test "$ret" -ne "0"
34ab15c5 699 ok $? "Create session $sess_name in $trace_path failed as expected"
873c2aae
JRJ
700 else
701 ok $ret "Create session $sess_name in $trace_path"
702 fi
d4018451
DG
703}
704
bf6ae429
JR
705function create_lttng_session_ok ()
706{
707 create_lttng_session 0 "$@"
708}
709
710function create_lttng_session_fail ()
711{
712 create_lttng_session 1 "$@"
713}
714
715
827caf52 716function enable_ust_lttng_channel ()
d4018451 717{
34ab15c5 718 local expected_to_fail=$1
827caf52
JR
719 local sess_name=$2
720 local channel_name=$3
4c80129b 721 local opt=$4
d4018451 722
4c80129b 723 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -u $channel_name -s $sess_name $opt 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
312dabc3 724 ret=$?
34ab15c5 725 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 726 test "$ret" -ne "0"
34ab15c5 727 ok $? "Enable channel $channel_name for session $sess_name failed as expected"
312dabc3
JRJ
728 else
729 ok $ret "Enable channel $channel_name for session $sess_name"
730 fi
d4018451
DG
731}
732
827caf52
JR
733function enable_ust_lttng_channel_ok ()
734{
735 enable_ust_lttng_channel 0 "$@"
736}
737
738function enable_ust_lttng_channel_fail ()
739{
740 enable_ust_lttng_channel 1 "$@"
741}
742
29655db7 743function disable_ust_lttng_channel()
d4018451 744{
7d0ad314
JRJ
745 local sess_name=$1
746 local channel_name=$2
d4018451 747
2cf48300 748 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-channel -u $channel_name -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
29655db7 749 ok $? "Disable channel $channel_name for session $sess_name"
d3e8f6bb
DG
750}
751
07b86b52
JD
752function enable_lttng_mmap_overwrite_kernel_channel()
753{
7d0ad314
JRJ
754 local sess_name=$1
755 local channel_name=$2
07b86b52 756
2cf48300 757 $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
758 ok $? "Enable channel $channel_name for session $sess_name"
759}
760
086e6add
MD
761function enable_lttng_mmap_discard_small_kernel_channel()
762{
763 local sess_name=$1
764 local channel_name=$2
765
766 $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
767 ok $? "Enable small discard channel $channel_name for session $sess_name"
768}
769
770function enable_lttng_mmap_overwrite_small_kernel_channel()
771{
772 local sess_name=$1
773 local channel_name=$2
774
775 $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
776 ok $? "Enable small discard channel $channel_name for session $sess_name"
777}
778
ebaaaf5e
JD
779function enable_lttng_mmap_overwrite_ust_channel()
780{
7d0ad314
JRJ
781 local sess_name=$1
782 local channel_name=$2
ebaaaf5e 783
2cf48300 784 $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
785 ok $? "Enable channel $channel_name for session $sess_name"
786}
787
d3e8f6bb
DG
788function enable_ust_lttng_event ()
789{
c4926bb5
JR
790 local expected_to_fail=$1
791 local sess_name=$2
792 local event_name="$3"
793 local channel_name=$4
ebaaaf5e
JD
794
795 if [ -z $channel_name ]; then
796 # default channel if none specified
797 chan=""
798 else
799 chan="-c $channel_name"
800 fi
d3e8f6bb 801
2cf48300 802 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -u 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 803 ret=$?
c4926bb5 804 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 805 test $ret -ne "0"
34ab15c5 806 ok $? "Enable ust event $event_name for session $session_name failed as expected"
01513c3e 807 else
34ab15c5 808 ok $ret "Enable ust event $event_name for session $sess_name"
01513c3e 809 fi
26b53d3b
DG
810}
811
c4926bb5
JR
812function enable_ust_lttng_event_ok ()
813{
814 enable_ust_lttng_event 0 "$@"
815}
816
817function enable_ust_lttng_event_fail ()
818{
819 enable_ust_lttng_event 1 "$@"
820}
821
37175ce4
DG
822function enable_jul_lttng_event()
823{
824 sess_name=$1
825 event_name="$2"
826 channel_name=$3
827
828 if [ -z $channel_name ]; then
829 # default channel if none specified
830 chan=""
831 else
832 chan="-c $channel_name"
833 fi
834
2cf48300 835 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -j 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
37175ce4
DG
836 ok $? "Enable JUL event $event_name for session $sess_name"
837}
838
b2064f54
DG
839function enable_jul_lttng_event_loglevel()
840{
7d0ad314
JRJ
841 local sess_name=$1
842 local event_name="$2"
843 local loglevel=$3
844 local channel_name=$4
b2064f54
DG
845
846 if [ -z $channel_name ]; then
847 # default channel if none specified
848 chan=""
849 else
850 chan="-c $channel_name"
851 fi
852
2cf48300 853 $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
854 ok $? "Enable JUL event $event_name for session $sess_name with loglevel $loglevel"
855}
856
504d4ace
DG
857function enable_log4j_lttng_event()
858{
859 sess_name=$1
860 event_name="$2"
861 channel_name=$3
862
863 if [ -z $channel_name ]; then
864 # default channel if none specified
865 chan=""
866 else
867 chan="-c $channel_name"
868 fi
869
2cf48300 870 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -l 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
504d4ace
DG
871 ok $? "Enable LOG4J event $event_name for session $sess_name"
872}
873
874function enable_log4j_lttng_event_loglevel()
875{
876 local sess_name=$1
877 local event_name="$2"
878 local loglevel=$3
879 local channel_name=$4
880
881 if [ -z $channel_name ]; then
882 # default channel if none specified
883 chan=""
884 else
885 chan="-c $channel_name"
886 fi
887
2cf48300 888 $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
889 ok $? "Enable LOG4J event $event_name for session $sess_name with loglevel $loglevel"
890}
891
0e115563
DG
892function enable_python_lttng_event()
893{
894 sess_name=$1
895 event_name="$2"
896 channel_name=$3
897
898 if [ -z $channel_name ]; then
899 # default channel if none specified
900 chan=""
901 else
902 chan="-c $channel_name"
903 fi
904
2cf48300 905 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -p 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
0e115563
DG
906 ok $? "Enable Python event $event_name for session $sess_name"
907}
908
909function enable_python_lttng_event_loglevel()
910{
911 local sess_name=$1
912 local event_name="$2"
913 local loglevel=$3
914 local channel_name=$4
915
916 if [ -z $channel_name ]; then
917 # default channel if none specified
918 chan=""
919 else
920 chan="-c $channel_name"
921 fi
922
2cf48300 923 $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
924 ok $? "Enable Python event $event_name for session $sess_name with loglevel $loglevel"
925}
926
26b53d3b
DG
927function enable_ust_lttng_event_filter()
928{
7d0ad314
JRJ
929 local sess_name="$1"
930 local event_name="$2"
931 local filter="$3"
26b53d3b 932
2cf48300 933 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u --filter "$filter" 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
29655db7 934 ok $? "Enable event $event_name with filtering for session $sess_name"
26b53d3b
DG
935}
936
937function enable_ust_lttng_event_loglevel()
938{
7d0ad314
JRJ
939 local sess_name="$1"
940 local event_name="$2"
941 local loglevel="$3"
26b53d3b 942
2cf48300 943 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u --loglevel $loglevel 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
29655db7 944 ok $? "Enable event $event_name with loglevel $loglevel"
26b53d3b
DG
945}
946
947function enable_ust_lttng_event_loglevel_only()
948{
7d0ad314
JRJ
949 local sess_name="$1"
950 local event_name="$2"
951 local loglevel="$3"
26b53d3b 952
2cf48300 953 $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 954 ok $? "Enable event $event_name with loglevel-only $loglevel"
26b53d3b
DG
955}
956
957function disable_ust_lttng_event ()
958{
7d0ad314
JRJ
959 local sess_name="$1"
960 local event_name="$2"
01513c3e 961 local channel_name="$3"
26b53d3b 962
01513c3e
JRJ
963 if [ -z $channel_name ]; then
964 # default channel if none specified
965 chan=""
966 else
967 chan="-c $channel_name"
968 fi
969
2cf48300 970 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name $chan -u 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
29655db7 971 ok $? "Disable event $event_name for session $sess_name"
d3e8f6bb
DG
972}
973
1d842d5a
DG
974function disable_jul_lttng_event ()
975{
976 local sess_name="$1"
977 local event_name="$2"
978
979 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name -j >/dev/null 2>&1
980 ok $? "Disable JUL event $event_name for session $sess_name"
981}
982
504d4ace
DG
983function disable_log4j_lttng_event ()
984{
985 local sess_name="$1"
986 local event_name="$2"
987
988 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name -l >/dev/null 2>&1
989 ok $? "Disable LOG4J event $event_name for session $sess_name"
990}
991
0e115563
DG
992function disable_python_lttng_event ()
993{
994 local sess_name="$1"
995 local event_name="$2"
996
2cf48300 997 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name -p 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
0e115563
DG
998 ok $? "Disable Python event $event_name for session $sess_name"
999}
1000
fb3268e3 1001function start_lttng_tracing ()
d3e8f6bb 1002{
e563bbdb
JR
1003 local expected_to_fail=$1
1004 local sess_name=$2
d3e8f6bb 1005
2cf48300 1006 $TESTDIR/../src/bin/lttng/$LTTNG_BIN start $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 1007 ret=$?
e563bbdb 1008 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 1009 test "$ret" -ne "0"
34ab15c5 1010 ok $? "Start tracing for session $sess_name failed as expected"
01513c3e
JRJ
1011 else
1012 ok $ret "Start tracing for session $sess_name"
1013 fi
d3e8f6bb
DG
1014}
1015
e563bbdb
JR
1016function start_lttng_tracing_ok ()
1017{
1018 start_lttng_tracing 0 "$@"
1019}
1020
1021function start_lttng_tracing_fail ()
1022{
1023 start_lttng_tracing 1 "$@"
1024}
1025
fb3268e3 1026function stop_lttng_tracing ()
d3e8f6bb 1027{
96340a01
JR
1028 local expected_to_fail=$1
1029 local sess_name=$2
d3e8f6bb 1030
2cf48300 1031 $TESTDIR/../src/bin/lttng/$LTTNG_BIN stop $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 1032 ret=$?
96340a01 1033 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 1034 test "$ret" -ne "0"
34ab15c5 1035 ok $? "Stop lttng tracing for session $sess_name failed as expected"
01513c3e
JRJ
1036 else
1037 ok $ret "Stop lttng tracing for session $sess_name"
1038 fi
d3e8f6bb
DG
1039}
1040
96340a01
JR
1041function stop_lttng_tracing_ok ()
1042{
1043 stop_lttng_tracing 0 "$@"
1044}
1045
1046function stop_lttng_tracing_fail ()
1047{
1048 stop_lttng_tracing 1 "$@"
1049}
1050
d3e8f6bb
DG
1051function destroy_lttng_session ()
1052{
67b4c664
JR
1053 local expected_to_fail=$1
1054 local sess_name=$2
d3e8f6bb 1055
2cf48300 1056 $TESTDIR/../src/bin/lttng/$LTTNG_BIN destroy $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 1057 ret=$?
96340a01 1058 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 1059 test "$ret" -ne "0"
34ab15c5 1060 ok $? "Destroy session $sess_name failed as expected"
01513c3e
JRJ
1061 else
1062 ok $ret "Destroy session $sess_name"
1063 fi
d3e8f6bb
DG
1064}
1065
67b4c664
JR
1066function destroy_lttng_session_ok ()
1067{
1068 destroy_lttng_session 0 "$@"
1069
1070}
1071
1072function destroy_lttng_session_fail ()
1073{
1074 destroy_lttng_session 1 "$@"
1075}
1076
1077
873c2aae
JRJ
1078function destroy_lttng_sessions ()
1079{
2cf48300 1080 $TESTDIR/../src/bin/lttng/$LTTNG_BIN destroy --all 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
873c2aae
JRJ
1081 ok $? "Destroy all lttng sessions"
1082}
1083
07b86b52
JD
1084function lttng_snapshot_add_output ()
1085{
b5633831
JR
1086 local expected_to_fail=$1
1087 local sess_name=$2
1088 local trace_path=$3
07b86b52 1089
2cf48300 1090 $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot add-output -s $sess_name file://$trace_path 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 1091 ret=$?
b5633831 1092 if [[ $expected_to_fail -eq 1 ]]; then
01513c3e 1093 test "$ret" -ne "0"
34ab15c5 1094 ok $? "Added snapshot output file://$trace_path failed as expected"
01513c3e
JRJ
1095 else
1096 ok $ret "Added snapshot output file://$trace_path"
1097 fi
07b86b52
JD
1098}
1099
b5633831
JR
1100function lttng_snapshot_add_output_ok ()
1101{
1102 lttng_snapshot_add_output 0 "$@"
1103}
1104
1105function lttng_snapshot_add_output_fail ()
1106{
1107 lttng_snapshot_add_output 1 "$@"
1108}
1109
26402e0c
DG
1110function lttng_snapshot_del_output ()
1111{
31580dc7
JR
1112 local expected_to_fail=$1
1113 local sess_name=$2
1114 local id=$3
26402e0c 1115
2cf48300 1116 $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot del-output -s $sess_name $id 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e 1117 ret=$?
31580dc7 1118 if [[ $expected_to_fail -eq "1" ]]; then
01513c3e 1119 test "$ret" -ne "0"
34ab15c5 1120 ok $? "Deleted snapshot output id $id failed as expected"
01513c3e
JRJ
1121 else
1122 ok $ret "Deleted snapshot output id $id"
1123 fi
26402e0c
DG
1124}
1125
31580dc7
JR
1126function lttng_snapshot_del_output_ok ()
1127{
1128 lttng_snapshot_del_output 0 "$@"
1129}
1130
1131function lttng_snapshot_del_output_fail ()
1132{
1133 lttng_snapshot_del_output 1 "$@"
1134}
1135
07b86b52
JD
1136function lttng_snapshot_record ()
1137{
7d0ad314
JRJ
1138 local sess_name=$1
1139 local trace_path=$2
07b86b52 1140
2cf48300 1141 $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot record -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
07b86b52
JD
1142 ok $? "Snapshot recorded"
1143}
1144
01513c3e
JRJ
1145function lttng_snapshot_list ()
1146{
1147 local sess_name=$1
2cf48300 1148 $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot list-output -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
01513c3e
JRJ
1149 ok $? "Snapshot list"
1150}
1151
e02b109b
DG
1152function lttng_save()
1153{
1154 local sess_name=$1
1155 local opts=$2
1156
2cf48300 1157 $TESTDIR/../src/bin/lttng/$LTTNG_BIN save $sess_name $opts 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
34ab15c5 1158 ok $? "Session saved"
e02b109b
DG
1159}
1160
1161function lttng_load()
1162{
192ac418
JR
1163 local expected_to_fail=$1
1164 local opts=$2
e02b109b 1165
2cf48300 1166 $TESTDIR/../src/bin/lttng/$LTTNG_BIN load $opts 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
192ac418
JR
1167 ret=$?
1168 if [[ $expected_to_fail -eq "1" ]]; then
1169 test $ret -ne "0"
1170 ok $? "Load command failed as expected with opts: $opts"
1171 else
1172 ok $ret "Load command with opts: $opts"
1173 fi
1174}
1175
1176function lttng_load_ok()
1177{
1178 lttng_load 0 "$@"
1179}
1180
1181function lttng_load_fail()
1182{
1183 lttng_load 1 "$@"
e02b109b
DG
1184}
1185
e83a8bdb
JR
1186function lttng_track()
1187{
1188 local expected_to_fail=$1
1189 local opts=$2
1190 $TESTDIR/../src/bin/lttng/$LTTNG_BIN track $opts >$OUTPUT_DEST
1191 ret=$?
1192 if [[ $expected_to_fail -eq "1" ]]; then
1193 test $ret -ne "0"
1194 ok $? "Track command failed as expected with opts: $opts"
1195 else
34ab15c5 1196 ok $ret "Track command with opts: $opts"
e83a8bdb
JR
1197 fi
1198}
1199
1200function lttng_track_ok()
1201{
1202 lttng_track 0 "$@"
1203}
1204
1205function lttng_track_fail()
1206{
1207 lttng_track 1 "$@"
1208}
1209
1210function lttng_untrack()
1211{
1212 local expected_to_fail=$1
1213 local opts=$2
1214 $TESTDIR/../src/bin/lttng/$LTTNG_BIN untrack $opts >$OUTPUT_DEST
1215 ret=$?
1216 if [[ $expected_to_fail -eq "1" ]]; then
1217 test $ret -ne "0"
1218 ok $? "Untrack command failed as expected with opts: $opts"
1219 else
34ab15c5 1220 ok $ret "Untrack command with opts: $opts"
e83a8bdb
JR
1221 fi
1222}
1223
1224function lttng_untrack_ok()
1225{
1226 lttng_untrack 0 "$@"
1227}
1228
1229function lttng_untrack_fail()
1230{
1231 lttng_untrack 1 "$@"
1232}
1233
8dcef147
JR
1234function add_context_lttng()
1235{
1236 local expected_to_fail="$1"
1237 local domain="$2"
1238 local session_name="$3"
1239 local channel_name="$4"
1240 local type="$5"
1241
1242 $TESTDIR/../src/bin/lttng/$LTTNG_BIN add-context -s $session_name -c $channel_name -t $type $domain 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1243 ret=$?
1244 if [[ $expected_to_fail -eq "1" ]]; then
1245 test $ret -ne "0"
1246 ok $? "Add context command failed as expected for type: $type"
1247 else
1248 ok $ret "Add context command for type: $type"
1249 fi
1250}
1251
1252function add_context_ust_ok()
1253{
1254 add_context_lttng 0 -u "$@"
1255}
1256
1257function add_context_ust_fail()
1258{
1259 add_context_lttng 1 -u "$@"
1260}
1261
1262function add_context_kernel_ok()
1263{
1264 add_context_lttng 0 -k "$@"
1265}
1266
1267function add_context_kernel_fail()
1268{
1269 add_context_lttng 1 -k "$@"
1270}
1271
d3e8f6bb
DG
1272function trace_matches ()
1273{
7d0ad314
JRJ
1274 local event_name=$1
1275 local nr_iter=$2
1276 local trace_path=$3
d3e8f6bb 1277
317eef93 1278 which $BABELTRACE_BIN >/dev/null
29655db7 1279 skip $? -ne 0 "Babeltrace binary not found. Skipping trace matches"
d3e8f6bb 1280
7d0ad314 1281 local count=$($BABELTRACE_BIN $trace_path | grep $event_name | wc -l)
29655db7 1282
d3e8f6bb 1283 if [ "$count" -ne "$nr_iter" ]; then
29655db7 1284 fail "Trace match"
af91abc9 1285 diag "$count matching events found in trace"
d3e8f6bb 1286 else
29655db7 1287 pass "Trace match"
d3e8f6bb
DG
1288 fi
1289}
f4e40ab6 1290
d53addeb
DG
1291function trace_match_only()
1292{
1293 local event_name=$1
1294 local nr_iter=$2
1295 local trace_path=$3
1296
1297 which $BABELTRACE_BIN >/dev/null
1298 skip $? -ne 0 "Babeltrace binary not found. Skipping trace matches"
1299
1300 local count=$($BABELTRACE_BIN $trace_path | grep $event_name | wc -l)
1301 local total=$($BABELTRACE_BIN $trace_path | wc -l)
1302
1303 if [ "$nr_iter" -eq "$count" ] && [ "$total" -eq "$nr_iter" ]; then
1304 pass "Trace match with $total event $event_name"
1305 else
1306 fail "Trace match"
1307 diag "$total event(s) found, expecting $nr_iter of event $event_name and only found $count"
1308 fi
1309}
1310
f4e40ab6
DG
1311function validate_trace
1312{
7d0ad314
JRJ
1313 local event_name=$1
1314 local trace_path=$2
f4e40ab6
DG
1315
1316 which $BABELTRACE_BIN >/dev/null
29655db7
CB
1317 if [ $? -ne 0 ]; then
1318 skip 0 "Babeltrace binary not found. Skipping trace validation"
f4e40ab6
DG
1319 fi
1320
07b86b52
JD
1321 OLDIFS=$IFS
1322 IFS=","
1323 for i in $event_name; do
1324 traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | grep $i | wc -l)
1325 if [ "$traced" -ne 0 ]; then
ebaaaf5e 1326 pass "Validate trace for event $i, $traced events"
07b86b52
JD
1327 else
1328 fail "Validate trace for event $i"
1329 diag "Found $traced occurences of $i"
1330 fi
1331 done
1332 ret=$?
1333 IFS=$OLDIFS
1334 return $ret
f4e40ab6 1335}
8cfcd41c 1336
54cd6107
JD
1337function validate_trace_count
1338{
1339 local event_name=$1
1340 local trace_path=$2
1341 local expected_count=$3
1342
1343 which $BABELTRACE_BIN >/dev/null
1344 if [ $? -ne 0 ]; then
1345 skip 0 "Babeltrace binary not found. Skipping trace validation"
1346 fi
1347
1348 cnt=0
1349 OLDIFS=$IFS
1350 IFS=","
1351 for i in $event_name; do
1352 traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | grep $i | wc -l)
1353 if [ "$traced" -ne 0 ]; then
1354 pass "Validate trace for event $i, $traced events"
1355 else
1356 fail "Validate trace for event $i"
1357 diag "Found $traced occurences of $i"
1358 fi
1359 cnt=$(($cnt + $traced))
1360 done
1361 IFS=$OLDIFS
1362 test $cnt -eq $expected_count
1363 ok $? "Read a total of $cnt events, expected $expected_count"
1364}
1365
086e6add
MD
1366function trace_first_line
1367{
1368 local trace_path=$1
1369
1370 which $BABELTRACE_BIN >/dev/null
1371 if [ $? -ne 0 ]; then
1372 skip 0 "Babeltrace binary not found. Skipping trace validation"
1373 fi
1374
1375 $BABELTRACE_BIN $trace_path 2>/dev/null | head -n 1
1376}
1377
8cfcd41c
MD
1378function validate_trace_exp()
1379{
1380 local event_exp=$1
1381 local trace_path=$2
1382
1383 which $BABELTRACE_BIN >/dev/null
1384 skip $? -ne 0 "Babeltrace binary not found. Skipping trace validation"
1385
1386 traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | grep ${event_exp} | wc -l)
1387 if [ "$traced" -ne 0 ]; then
1388 pass "Validate trace for expression '${event_exp}', $traced events"
1389 else
1390 fail "Validate trace for expression '${event_exp}'"
1391 diag "Found $traced occurences of '${event_exp}'"
1392 fi
1393 ret=$?
1394 return $ret
1395}
1396
1397function validate_trace_only_exp()
1398{
1399 local event_exp=$1
1400 local trace_path=$2
1401
1402 which $BABELTRACE_BIN >/dev/null
1403 skip $? -ne 0 "Babeltrace binary not found. Skipping trace matches"
1404
1405 local count=$($BABELTRACE_BIN $trace_path | grep ${event_exp} | wc -l)
1406 local total=$($BABELTRACE_BIN $trace_path | wc -l)
1407
1408 if [ "$count" -ne 0 ] && [ "$total" -eq "$count" ]; then
1409 pass "Trace match with $total for expression '${event_exp}"
1410 else
1411 fail "Trace match"
1412 diag "$total syscall event(s) found, only syscalls matching expression '${event_exp}' ($count occurrences) are expected"
1413 fi
1414 ret=$?
1415 return $ret
1416}
1417
1418function validate_trace_empty()
1419{
1420 local trace_path=$1
1421
1422 which $BABELTRACE_BIN >/dev/null
1423 if [ $? -ne 0 ]; then
1424 skip 0 "Babeltrace binary not found. Skipping trace validation"
1425 fi
1426
1427 traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | wc -l)
1428 if [ "$traced" -eq 0 ]; then
1429 pass "Validate empty trace"
1430 else
1431 fail "Validate empty trace"
1432 diag "Found $traced events in trace"
1433 fi
1434 ret=$?
1435 return $ret
1436}
801236b0 1437
eded6438 1438function regenerate_metadata ()
801236b0
JD
1439{
1440 local expected_to_fail=$1
1441 local sess_name=$2
1442
eded6438 1443 $TESTDIR/../src/bin/lttng/$LTTNG_BIN regenerate metadata -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
801236b0
JD
1444 ret=$?
1445 if [[ $expected_to_fail -eq "1" ]]; then
1446 test "$ret" -ne "0"
eded6438 1447 ok $? "Expected fail on regenerate metadata $sess_name"
801236b0
JD
1448 else
1449 ok $ret "Metadata regenerate $sess_name"
1450 fi
1451}
1452
eded6438 1453function regenerate_metadata_ok ()
801236b0 1454{
eded6438 1455 regenerate_metadata 0 "$@"
801236b0
JD
1456}
1457
eded6438 1458function regenerate_metadata_fail ()
801236b0 1459{
eded6438 1460 regenerate_metadata 1 "$@"
801236b0 1461}
512eb148 1462
54cd6107
JD
1463function regenerate_statedump ()
1464{
1465 local expected_to_fail=$1
1466 local sess_name=$2
1467
1468 $TESTDIR/../src/bin/lttng/$LTTNG_BIN regenerate statedump -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1469 ret=$?
1470 if [[ $expected_to_fail -eq "1" ]]; then
1471 test "$ret" -ne "0"
1472 ok $? "Expected fail on regenerate statedump $sess_name"
1473 else
1474 ok $ret "Metadata regenerate $sess_name"
1475 fi
1476}
1477
1478function regenerate_statedump_ok ()
1479{
1480 regenerate_statedump 0 "$@"
1481}
1482
1483function regenerate_statedump_fail ()
1484{
1485 regenerate_statedump 1 "$@"
1486}
1487
512eb148
JD
1488function destructive_tests_enabled ()
1489{
1490 if [ ${LTTNG_ENABLE_DESTRUCTIVE_TESTS} = "will-break-my-system" ]; then
1491 return 0
1492 else
1493 return 1
1494 fi
1495}
This page took 0.119126 seconds and 5 git commands to generate.