Fix: Tests that assume CPU 0 is present
[lttng-tools.git] / tests / utils / utils.sh
... / ...
CommitLineData
1# Copyright (C) 2012 David Goulet <dgoulet@efficios.com>
2#
3# SPDX-License-Identifier: LGPL-2.1-only
4#
5
6SESSIOND_BIN="lttng-sessiond"
7SESSIOND_MATCH=".*lttng-sess.*"
8RUNAS_BIN="lttng-runas"
9RUNAS_MATCH=".*lttng-runas.*"
10CONSUMERD_BIN="lttng-consumerd"
11CONSUMERD_MATCH=".*lttng-consumerd.*"
12RELAYD_BIN="lttng-relayd"
13RELAYD_MATCH=".*lttng-relayd.*"
14LTTNG_BIN="lttng"
15BABELTRACE_BIN="babeltrace2"
16OUTPUT_DEST=/dev/null
17ERROR_OUTPUT_DEST=/dev/null
18MI_XSD_MAJOR_VERSION=4
19MI_XSD_MINOR_VERSION=1
20MI_XSD_PATH="$TESTDIR/../src/common/mi-lttng-${MI_XSD_MAJOR_VERSION}.${MI_XSD_MINOR_VERSION}.xsd"
21MI_VALIDATE="$TESTDIR/utils/xml-utils/validate_xml ${MI_XSD_PATH}"
22
23XML_PRETTY="$TESTDIR/utils/xml-utils/pretty_xml"
24XML_EXTRACT="$TESTDIR/utils/xml-utils/extract_xml"
25XML_NODE_CHECK="${XML_EXTRACT} -e"
26
27# To match 20201127-175802
28date_time_pattern="[0-9][0-9][0-9][0-9][0-9][0-9][0-9][0-9]-[0-9][0-9][0-9][0-9][0-9][0-9]"
29# The size of a long on this system
30system_long_bit_size=$(getconf LONG_BIT)
31
32# Minimal kernel version supported for session daemon tests
33KERNEL_MAJOR_VERSION=2
34KERNEL_MINOR_VERSION=6
35KERNEL_PATCHLEVEL_VERSION=27
36
37# We set the default UST register timeout and network and app socket timeout to
38# "wait forever", so that basic tests don't have to worry about hitting
39# timeouts on busy systems. Specialized tests should test those corner-cases.
40export LTTNG_UST_REGISTER_TIMEOUT=-1
41export LTTNG_NETWORK_SOCKET_TIMEOUT=-1
42export LTTNG_APP_SOCKET_TIMEOUT=-1
43
44# We set the default lttng-sessiond path to /bin/true to prevent the spawning
45# of a daemonized sessiond. This is necessary since 'lttng create' will spawn
46# its own sessiond if none is running. It also ensures that 'lttng create'
47# fails when no sessiond is running.
48export LTTNG_SESSIOND_PATH="/bin/true"
49
50source $TESTDIR/utils/tap/tap.sh
51
52if [ -z ${LTTNG_TEST_TEARDOWN_TIMEOUT+x} ]; then
53 LTTNG_TEST_TEARDOWN_TIMEOUT=60
54fi
55
56# Enable job monitor mode.
57# Here we are mostly interested in the following from the monitor mode:
58# All processes run in a separate process group.
59# This allows us to ensure that all subprocesses from all background tasks are
60# cleaned up correctly using signal to process group id.
61set -m
62
63kill_background_jobs ()
64{
65 local pids
66 pids=$(jobs -p)
67
68 if [ -z "$pids" ]; then
69 # Empty
70 return 0
71 fi
72
73 while read -r pid;
74 do
75 # Use negative number to send the signal to the process group.
76 # This ensure that any subprocesses receive the signal.
77 # /dev/null is used since there is an acceptable race between
78 # the moments the pids are listed and the moment we send a
79 # signal.
80 kill -SIGTERM -- "-$pid" 2>/dev/null
81 done <<< "$pids"
82}
83
84function cleanup ()
85{
86 # Try to kill daemons gracefully
87 stop_lttng_relayd_cleanup SIGTERM $LTTNG_TEST_TEARDOWN_TIMEOUT
88 stop_lttng_sessiond_cleanup SIGTERM $LTTNG_TEST_TEARDOWN_TIMEOUT
89
90 # If daemons are still present, forcibly kill them
91 stop_lttng_relayd_cleanup SIGKILL $LTTNG_TEST_TEARDOWN_TIMEOUT
92 stop_lttng_sessiond_cleanup SIGKILL $LTTNG_TEST_TEARDOWN_TIMEOUT
93 stop_lttng_consumerd_cleanup SIGKILL $LTTNG_TEST_TEARDOWN_TIMEOUT
94
95 kill_background_jobs
96}
97
98function full_cleanup ()
99{
100 cleanup
101 exit 1
102}
103
104function LTTNG_BAIL_OUT ()
105{
106 cleanup
107 BAIL_OUT "$@"
108}
109
110function null_pipes ()
111{
112 exec 0>/dev/null
113 exec 1>/dev/null
114 exec 2>/dev/null
115}
116
117trap full_cleanup SIGINT SIGTERM
118
119# perl prove closes its child pipes before giving it a chance to run its
120# signal trap handlers. Redirect pipes to /dev/null if SIGPIPE is caught
121# to allow those trap handlers to proceed.
122
123trap null_pipes SIGPIPE
124
125# Check pgrep from env, default to pgrep if none
126if [ -z "$PGREP" ]; then
127 PGREP=pgrep
128fi
129
130# Due to the renaming of threads we need to use the full command (pgrep -f) to
131# identify the pids for multiple lttng related processes. The problem with "pgrep
132# -f" is that it ends up also looking at the arguments. We use a two stage
133# lookup. The first one is using "pgrep -f" yielding potential candidate.
134# The second on perform grep on the basename of the first field of the
135# /proc/pid/cmdline of the previously identified pids. The first field
136# correspond to the actual command.
137function lttng_pgrep ()
138{
139 local pattern=$1
140 local possible_pids
141 local full_command_no_argument
142 local command_basename
143
144 possible_pids=$($PGREP -f "$pattern")
145 if [ -z "$possible_pids" ]; then
146 return 0
147 fi
148
149 while IFS= read -r pid ; do
150 # /proc/pid/cmdline is null separated.
151 if full_command_no_argument=$(tr '\0' '\n' < /proc/"$pid"/cmdline 2>/dev/null | head -n1); then
152 command_basename=$(basename "$full_command_no_argument")
153 if grep -q "$pattern" <<< "$command_basename"; then
154 echo "$pid"
155 fi
156 fi
157 done <<< "$possible_pids"
158 return 0
159}
160
161function print_ok ()
162{
163 # Check if we are a terminal
164 if [ -t 1 ]; then
165 echo -e "\e[1;32mOK\e[0m"
166 else
167 echo -e "OK"
168 fi
169}
170
171function print_fail ()
172{
173 # Check if we are a terminal
174 if [ -t 1 ]; then
175 echo -e "\e[1;31mFAIL\e[0m"
176 else
177 echo -e "FAIL"
178 fi
179}
180
181function print_test_banner ()
182{
183 local desc="$1"
184 diag "$desc"
185}
186
187function validate_kernel_version ()
188{
189 local kern_version=($(uname -r | awk -F. '{ printf("%d.%d.%d\n",$1,$2,$3); }' | tr '.' '\n'))
190 if [ ${kern_version[0]} -gt $KERNEL_MAJOR_VERSION ]; then
191 return 0
192 fi
193 if [ ${kern_version[1]} -gt $KERNEL_MINOR_VERSION ]; then
194 return 0
195 fi
196 if [ ${kern_version[2]} -ge $KERNEL_PATCHLEVEL_VERSION ]; then
197 return 0
198 fi
199 return 1
200}
201
202# Generate a random string
203# $1 = number of characters; defaults to 16
204# $2 = include special characters; 1 = yes, 0 = no; defaults to yes
205function randstring()
206{
207 local len="${1:-16}"
208
209 [ "$2" == "0" ] && CHAR="[:alnum:]" || CHAR="[:graph:]"
210 # /dev/urandom isn't guaranteed to generate valid multi-byte characters.
211 # Specifying the C locale eliminates the "Illegal byte sequence" error
212 # that 'tr' outputs in such cases.
213 LC_CTYPE=C tr -cd "$CHAR" < /dev/urandom 2>/dev/null | head -c "$len" 2>/dev/null
214 echo
215}
216
217# Helpers for get_possible_cpus.
218function get_possible_cpus_count_from_sysfs_possible_mask()
219{
220 local max_possible_cpu_id
221
222 # The Awk script extracts the highest CPU id from the possible CPU
223 # mask. Assuming a numerical order, a field separator '-' and a record
224 # separator ','. The last value parsed is the highest id.
225 if [ -f /sys/devices/system/cpu/possible ]; then
226 max_possible_cpu_id=$(awk -F '-' 'BEGIN { RS = ","} { last = $NF } END { printf("%d\n", last) }' \
227 /sys/devices/system/cpu/possible)
228 echo "$((max_possible_cpu_id+1))"
229 else
230 echo "0"
231 fi
232}
233
234# This is a fallback if the possible CPU mask is not available. This will not
235# take into account unplugged CPUs.
236function get_max_cpus_count_from_sysfs_cpu_directories()
237{
238 local max_possible_cpu_id=0
239 local current_cpu_id
240
241 for i in /sys/devices/system/cpu/cpu[0-9]*; do
242 current_cpu_id="${i#/sys/devices/system/cpu/cpu}"
243 if [ "$current_cpu_id" -gt "$max_possible_cpu_id" ]; then
244 max_possible_cpu_id="$current_cpu_id"
245 fi
246 done
247
248 echo "$((max_possible_cpu_id+1))"
249}
250
251# Return the number of possible CPUs.
252function get_possible_cpus_count()
253{
254 local possible_cpus_count
255 possible_cpus_count=$(get_possible_cpus_count_from_sysfs_possible_mask)
256
257 if [ "$possible_cpus_count" -eq "0" ]; then
258 local configured_cpus_count
259 configured_cpus_count=$(getconf _NPROCESSORS_CONF)
260 possible_cpus_count=$(get_max_cpus_count_from_sysfs_cpu_directories)
261 possible_cpus_count=$((configured_cpus_count > possible_cpus_count \
262 ? configured_cpus_count \
263 : possible_cpus_count))
264 fi
265
266 echo "$possible_cpus_count"
267}
268
269# Return the list of exposed CPU.
270#
271# NOTE! Use it like so:
272#
273# IFS=" " read -r -a VARIABLE <<< "$(get_exposed_cpus_list)"
274function get_exposed_cpus_list()
275{
276 local list=()
277
278 for i in /sys/devices/system/cpu/cpu[0-9]*; do
279 list+=("${i#/sys/devices/system/cpu/cpu}")
280 done
281
282 echo "${list[@]}"
283}
284
285# Return any available CPU found. Do not make assumption about the returned
286# value, e.g. that it could be 0.
287function get_any_available_cpu()
288{
289 for cpu in /sys/devices/system/cpu/cpu[0-9]*; do
290 echo "${cpu#/sys/devices/system/cpu/cpu}"
291 break;
292 done
293}
294
295# Return the number of _configured_ CPUs.
296function conf_proc_count()
297{
298 getconf _NPROCESSORS_CONF
299 if [ $? -ne 0 ]; then
300 diag "Failed to get the number of configured CPUs"
301 fi
302 echo
303}
304
305# Check if base lttng-modules are present.
306# Bail out on failure
307function validate_lttng_modules_present ()
308{
309 # Check for loadable modules.
310 modprobe -n lttng-tracer 2>/dev/null
311 if [ $? -eq 0 ]; then
312 return 0
313 fi
314
315 # Check for builtin modules.
316 ls /proc/lttng > /dev/null 2>&1
317 if [ $? -eq 0 ]; then
318 return 0
319 fi
320
321 LTTNG_BAIL_OUT "LTTng modules not detected."
322}
323
324# Run the lttng binary.
325#
326# The first two arguments are stdout and stderr redirect paths, respectively.
327# The rest of the arguments are forwarded to the lttng binary
328function _run_lttng_cmd
329{
330 local stdout_dest="$1"
331 local stderr_dest="$2"
332 shift 2
333
334 diag "$TESTDIR/../src/bin/lttng/$LTTNG_BIN $*"
335 $TESTDIR/../src/bin/lttng/$LTTNG_BIN "$@" 1> "$stdout_dest" 2> "$stderr_dest"
336}
337
338function enable_kernel_lttng_event
339{
340 local withtap="$1"
341 local expected_to_fail="$2"
342 local sess_name="$3"
343 local event_name="$4"
344 local channel_name="$5"
345
346 if [ -z "$event_name" ]; then
347 # Enable all event if no event name specified
348 event_name="-a"
349 fi
350
351 if [ -z "$channel_name" ]; then
352 # default channel if none specified
353 chan=""
354 else
355 chan="-c $channel_name"
356 fi
357
358 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
359 enable-event "$event_name" $chan -s $sess_name -k
360 ret=$?
361 if [[ $expected_to_fail -eq "1" ]]; then
362 test $ret -ne "0"
363 ret=$?
364 if [ $withtap -eq "1" ]; then
365 ok $ret "Enable kernel event $event_name for session $session_name on channel $channel_name failed as expected"
366 fi
367 else
368 if [ $withtap -eq "1" ]; then
369 ok $ret "Enable kernel event $event_name for session $sess_name"
370 fi
371 fi
372}
373
374function enable_kernel_lttng_event_ok ()
375{
376 enable_kernel_lttng_event 1 0 "$@"
377}
378
379function enable_kernel_lttng_event_fail ()
380{
381 enable_kernel_lttng_event 1 1 "$@"
382}
383
384function enable_kernel_lttng_event_notap ()
385{
386 enable_kernel_lttng_event 0 0 "$@"
387}
388
389# Old interface
390function lttng_enable_kernel_event
391{
392 enable_kernel_lttng_event_ok "$@"
393}
394
395function lttng_enable_kernel_syscall()
396{
397 local expected_to_fail=$1
398 local sess_name=$2
399 local syscall_name=$3
400 local channel_name=$4
401
402 if [ -z $syscall_name ]; then
403 # Enable all event if no syscall name specified
404 syscall_name="-a"
405 fi
406
407 if [ -z $channel_name ]; then
408 # default channel if none specified
409 chan=""
410 else
411 chan="-c $channel_name"
412 fi
413
414 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
415 enable-event --syscall "$syscall_name" $chan -s $sess_name -k
416 ret=$?
417 if [[ $expected_to_fail -eq "1" ]]; then
418 test $ret -ne "0"
419 ok $? "Enable kernel syscall $syscall_name for session $sess_name on channel $channel_name fail as expected"
420 else
421 ok $ret "Enable kernel syscall $syscall_name for session $sess_name on channel $channel_name"
422 fi
423}
424
425function lttng_enable_kernel_syscall_ok()
426{
427 lttng_enable_kernel_syscall 0 "$@"
428}
429
430function lttng_enable_kernel_syscall_fail()
431{
432 lttng_enable_kernel_syscall 1 "$@"
433}
434
435function lttng_disable_kernel_syscall()
436{
437 local expected_to_fail=$1
438 local sess_name=$2
439 local syscall_name=$3
440 local channel_name=$4
441
442 if [ -z $syscall_name ]; then
443 # Enable all event if no syscall name specified
444 syscall_name="-a"
445 fi
446
447 if [ -z $channel_name ]; then
448 # default channel if none specified
449 chan=""
450 else
451 chan="-c $channel_name"
452 fi
453
454 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
455 disable-event --syscall "$syscall_name" $chan -s $sess_name -k
456
457 ret=$?
458 if [[ $expected_to_fail -eq "1" ]]; then
459 test $ret -ne "0"
460 ok $? "Disable kernel syscall $syscall_name for session $sess_name on channel $channel_name failed as expected"
461 else
462 ok $ret "Disable kernel syscall $syscall_name for session $sess_name on channel $channel_name"
463 fi
464}
465
466function lttng_disable_kernel_syscall_ok()
467{
468 lttng_disable_kernel_syscall 0 "$@"
469}
470
471function lttng_disable_kernel_syscall_fail()
472{
473 lttng_disable_kernel_syscall 1 "$@"
474}
475
476function lttng_enable_kernel_function_event ()
477{
478 local expected_to_fail="$1"
479 local sess_name="$2"
480 local target="$3"
481 local event_name="$4"
482
483 "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" enable-event --kernel --function="$target" "$event_name" -s "$sess_name" > "$OUTPUT_DEST" 2> "$ERROR_OUTPUT_DEST"
484 ret=$?
485 if [[ $expected_to_fail -eq "1" ]]; then
486 test $ret -ne "0"
487 ok $? "Enable kernel function event for session $sess_name failed as expected"
488 else
489 ok $ret "Enable kernel function event for session $sess_name"
490 fi
491}
492
493function lttng_enable_kernel_function_event_ok ()
494{
495 lttng_enable_kernel_function_event 0 "$@"
496}
497
498function lttng_enable_kernel_userspace_probe_event ()
499{
500 local expected_to_fail="$1"
501 local sess_name="$2"
502 local target="$3"
503 local event_name="$4"
504
505 "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" enable-event --kernel --userspace-probe="$target" "$event_name" -s "$sess_name" > "$OUTPUT_DEST" 2> "$ERROR_OUTPUT_DEST"
506 ret=$?
507 if [[ $expected_to_fail -eq "1" ]]; then
508 test $ret -ne "0"
509 ok $? "Enable kernel userspace probe event for session $sess_name failed as expected"
510 else
511 ok $ret "Enable kernel userspace probe event for session $sess_name"
512 fi
513}
514
515function lttng_enable_kernel_userspace_probe_event_fail ()
516{
517 lttng_enable_kernel_userspace_probe_event 1 "$@"
518}
519
520function lttng_enable_kernel_userspace_probe_event_ok ()
521{
522 lttng_enable_kernel_userspace_probe_event 0 "$@"
523}
524
525function disable_kernel_lttng_userspace_probe_event_ok ()
526{
527 local sess_name="$1"
528 local event_name="$2"
529
530 "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" disable-event --kernel "$event_name" -s "$sess_name" > "$OUTPUT_DEST" 2> "$ERROR_OUTPUT_DEST"
531 ok $? "Disable kernel event $target for session $sess_name"
532}
533function lttng_enable_kernel_channel()
534{
535 local withtap=$1
536 local expected_to_fail=$2
537 local sess_name=$3
538 local channel_name=$4
539 local opts="${@:5}"
540
541 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
542 enable-channel -k $channel_name -s $sess_name $opts
543 ret=$?
544 if [[ $expected_to_fail -eq "1" ]]; then
545 test "$ret" -ne "0"
546 ret=$?
547 if [ $withtap -eq "1" ]; then
548 ok $ret "Enable channel $channel_name for session $sess_name failed as expected"
549 fi
550 else
551 if [ $withtap -eq "1" ]; then
552 ok $ret "Enable channel $channel_name for session $sess_name"
553 fi
554 fi
555}
556
557function lttng_enable_kernel_channel_ok()
558{
559 lttng_enable_kernel_channel 1 0 "$@"
560}
561
562function lttng_enable_kernel_channel_fail()
563{
564 lttng_enable_kernel_channel 1 1 "$@"
565}
566
567function lttng_enable_kernel_channel_notap()
568{
569 lttng_enable_kernel_channel 0 0 "$@"
570}
571
572function enable_kernel_lttng_channel_ok()
573{
574 lttng_enable_kernel_channel 1 0 "$@"
575}
576
577function lttng_disable_kernel_channel()
578{
579 local expected_to_fail=$1
580 local sess_name=$2
581 local channel_name=$3
582
583 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
584 disable-channel -k $channel_name -s $sess_name
585 ret=$?
586 if [[ $expected_to_fail -eq "1" ]]; then
587 test "$ret" -ne "0"
588 ok $? "Disable channel $channel_name for session $sess_name failed as expected"
589 else
590 ok $ret "Disable channel $channel_name for session $sess_name"
591 fi
592}
593
594function lttng_disable_kernel_channel_ok()
595{
596 lttng_disable_kernel_channel 0 "$@"
597}
598
599function lttng_disable_kernel_channel_fail()
600{
601 lttng_disable_kernel_channel 1 "$@"
602}
603
604function start_lttng_relayd_opt()
605{
606 local withtap=$1
607 local process_mode=$2
608 local opt=$3
609
610 DIR=$(readlink -f "$TESTDIR")
611
612 if [ -z $(lttng_pgrep "$RELAYD_MATCH") ]; then
613 # shellcheck disable=SC2086
614 $DIR/../src/bin/lttng-relayd/$RELAYD_BIN $process_mode $opt 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
615 #$DIR/../src/bin/lttng-relayd/$RELAYD_BIN $opt -vvv >>/tmp/relayd.log 2>&1 &
616 if [ $? -eq 1 ]; then
617 if [ $withtap -eq "1" ]; then
618 fail "Start lttng-relayd (process mode: $process_mode opt: $opt)"
619 fi
620 return 1
621 else
622 if [ $withtap -eq "1" ]; then
623 pass "Start lttng-relayd (process mode: $process_mode opt: $opt)"
624 fi
625 fi
626 else
627 pass "Start lttng-relayd (opt: $opt)"
628 fi
629}
630
631function start_lttng_relayd()
632{
633 start_lttng_relayd_opt 1 "-b" "$@"
634}
635
636function start_lttng_relayd_notap()
637{
638 start_lttng_relayd_opt 0 "-b" "$@"
639}
640
641function stop_lttng_relayd_opt()
642{
643 local withtap=$1
644 local is_cleanup=$2
645 local signal=$3
646 local timeout_s=$4
647 local dtimeleft_s=
648 local retval=0
649 local pids
650
651 if [ -z "$signal" ]; then
652 signal="SIGTERM"
653 fi
654
655
656 # Multiply time by 2 to simplify integer arithmetic
657 if [ -n "$timeout_s" ]; then
658 dtimeleft_s=$((timeout_s * 2))
659 fi
660
661
662 pids=$(lttng_pgrep "$RELAYD_MATCH")
663 if [ -z "$pids" ]; then
664 if [ "$is_cleanup" -eq 1 ]; then
665 :
666 elif [ "$withtap" -eq "1" ]; then
667 fail "No relay daemon to kill"
668 else
669 LTTNG_BAIL_OUT "No relay daemon to kill"
670 fi
671 return 0
672 fi
673
674 diag "Killing (signal $signal) lttng-relayd (pid: $pids)"
675
676 # shellcheck disable=SC2086
677 if ! kill -s $signal $pids 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST; then
678 retval=1
679 if [ "$withtap" -eq "1" ]; then
680 fail "Kill relay daemon"
681 fi
682 else
683 out=1
684 while [ -n "$out" ]; do
685 out=$(lttng_pgrep "$RELAYD_MATCH")
686 if [ -n "$dtimeleft_s" ]; then
687 if [ $dtimeleft_s -lt 0 ]; then
688 out=
689 retval=1
690 fi
691 dtimeleft_s=$((dtimeleft_s - 1))
692 fi
693 sleep 0.5
694 done
695 if [ "$withtap" -eq "1" ]; then
696 if [ "$retval" -eq "0" ]; then
697 pass "Wait after kill relay daemon"
698 else
699 fail "Wait after kill relay daemon"
700 fi
701 fi
702 fi
703 return $retval
704}
705
706function stop_lttng_relayd()
707{
708 stop_lttng_relayd_opt 1 0 "$@"
709}
710
711function stop_lttng_relayd_notap()
712{
713 stop_lttng_relayd_opt 0 0 "$@"
714}
715
716function stop_lttng_relayd_cleanup()
717{
718 stop_lttng_relayd_opt 0 1 "$@"
719}
720
721#First arg: show tap output
722#Second argument: load path for automatic loading
723function start_lttng_sessiond_opt()
724{
725 local withtap=$1
726 local load_path=$2
727
728 # The rest of the arguments will be passed directly to lttng-sessiond.
729 shift 2
730
731 local env_vars=""
732 local consumerd=""
733
734 local long_bit_value=
735 long_bit_value=$(getconf LONG_BIT)
736
737 if [ -n "$TEST_NO_SESSIOND" ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
738 # Env variable requested no session daemon
739 return
740 fi
741
742 DIR=$(readlink -f "$TESTDIR")
743
744 # Get long_bit value for 32/64 consumerd
745 case "$long_bit_value" in
746 32)
747 consumerd="--consumerd32-path=$DIR/../src/bin/lttng-consumerd/lttng-consumerd"
748 ;;
749 64)
750 consumerd="--consumerd64-path=$DIR/../src/bin/lttng-consumerd/lttng-consumerd"
751 ;;
752 *)
753 return
754 ;;
755 esac
756
757 # Check for env. variable. Allow the use of LD_PRELOAD etc.
758 if [[ "x${LTTNG_SESSIOND_ENV_VARS}" != "x" ]]; then
759 env_vars="${LTTNG_SESSIOND_ENV_VARS} "
760 fi
761 env_vars="${env_vars}$DIR/../src/bin/lttng-sessiond/$SESSIOND_BIN"
762
763 if ! validate_kernel_version; then
764 fail "Start session daemon"
765 LTTNG_BAIL_OUT "*** Kernel too old for session daemon tests ***"
766 fi
767
768 diag "export LTTNG_SESSION_CONFIG_XSD_PATH=${DIR}/../src/common/"
769 : "${LTTNG_SESSION_CONFIG_XSD_PATH="${DIR}/../src/common/"}"
770 export LTTNG_SESSION_CONFIG_XSD_PATH
771
772 if [ -z "$(lttng_pgrep "${SESSIOND_MATCH}")" ]; then
773 # Have a load path ?
774 if [ -n "$load_path" ]; then
775 diag "env $env_vars --load $load_path --background $consumerd $@"
776 # shellcheck disable=SC2086
777 env $env_vars --load "$load_path" --background "$consumerd" "$@"
778 else
779 diag "env $env_vars --background $consumerd $@"
780 # shellcheck disable=SC2086
781 env $env_vars --background "$consumerd" "$@"
782 fi
783 #$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
784 status=$?
785 if [ "$withtap" -eq "1" ]; then
786 ok $status "Start session daemon"
787 fi
788 fi
789}
790
791function start_lttng_sessiond()
792{
793 start_lttng_sessiond_opt 1 "$@"
794}
795
796function start_lttng_sessiond_notap()
797{
798 start_lttng_sessiond_opt 0 "$@"
799}
800
801function stop_lttng_sessiond_opt()
802{
803 local withtap=$1
804 local is_cleanup=$2
805 local signal=$3
806 local timeout_s=$4
807 local dtimeleft_s=
808 local retval=0
809 local runas_pids
810 local pids
811
812 if [ -z "$signal" ]; then
813 signal=SIGTERM
814 fi
815
816 # Multiply time by 2 to simplify integer arithmetic
817 if [ -n "$timeout_s" ]; then
818 dtimeleft_s=$((timeout_s * 2))
819 fi
820
821 if [ -n "$TEST_NO_SESSIOND" ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
822 # Env variable requested no session daemon
823 return 0
824 fi
825
826 runas_pids=$(lttng_pgrep "$RUNAS_MATCH")
827 pids=$(lttng_pgrep "$SESSIOND_MATCH")
828
829 if [ -n "$runas_pids" ]; then
830 pids="$pids $runas_pids"
831 fi
832
833 if [ -z "$pids" ]; then
834 if [ "$is_cleanup" -eq 1 ]; then
835 :
836 elif [ "$withtap" -eq "1" ]; then
837 fail "No session daemon to kill"
838 else
839 LTTNG_BAIL_OUT "No session daemon to kill"
840 fi
841 return 0
842 fi
843
844 diag "Killing (signal $signal) $SESSIOND_BIN and lt-$SESSIOND_BIN pids: $(echo "$pids" | tr '\n' ' ')"
845
846 # shellcheck disable=SC2086
847 if ! kill -s $signal $pids 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST; then
848 retval=1
849 if [ "$withtap" -eq "1" ]; then
850 fail "Kill sessions daemon"
851 fi
852 else
853 out=1
854 while [ -n "$out" ]; do
855 out=$(lttng_pgrep "${SESSIOND_MATCH}")
856 if [ -n "$dtimeleft_s" ]; then
857 if [ $dtimeleft_s -lt 0 ]; then
858 out=
859 retval=1
860 fi
861 dtimeleft_s=$((dtimeleft_s - 1))
862 fi
863 sleep 0.5
864 done
865 out=1
866 while [ -n "$out" ]; do
867 out=$(lttng_pgrep "$CONSUMERD_MATCH")
868 if [ -n "$dtimeleft_s" ]; then
869 if [ $dtimeleft_s -lt 0 ]; then
870 out=
871 retval=1
872 fi
873 dtimeleft_s=$((dtimeleft_s - 1))
874 fi
875 sleep 0.5
876 done
877
878 if [ "$withtap" -eq "1" ]; then
879 if [ "$retval" -eq "0" ]; then
880 pass "Wait after kill session daemon"
881 else
882 fail "Wait after kill session daemon"
883 fi
884 fi
885 fi
886 if [ "$signal" = "SIGKILL" ]; then
887 if [ "$(id -u)" -eq "0" ]; then
888 local modules=
889 modules="$(lsmod | grep ^lttng | awk '{print $1}')"
890
891 if [ -n "$modules" ]; then
892 diag "Unloading all LTTng modules"
893 modprobe --remove "$modules"
894 fi
895 fi
896 fi
897
898 return $retval
899}
900
901function stop_lttng_sessiond()
902{
903 stop_lttng_sessiond_opt 1 0 "$@"
904}
905
906function stop_lttng_sessiond_notap()
907{
908 stop_lttng_sessiond_opt 0 0 "$@"
909}
910
911function stop_lttng_sessiond_cleanup()
912{
913 stop_lttng_sessiond_opt 0 1 "$@"
914}
915
916function sigstop_lttng_sessiond_opt()
917{
918 local withtap=$1
919 local signal=SIGSTOP
920 local pids
921
922 if [ -n "$TEST_NO_SESSIOND" ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
923 # Env variable requested no session daemon
924 return
925 fi
926
927 pids="$(lttng_pgrep "${SESSIOND_MATCH}") $(lttng_pgrep "$RUNAS_MATCH")"
928
929 if [ "$withtap" -eq "1" ]; then
930 diag "Sending SIGSTOP to lt-$SESSIOND_BIN and $SESSIOND_BIN pids: $(echo "$pids" | tr '\n' ' ')"
931 fi
932
933 # shellcheck disable=SC2086
934 if ! kill -s $signal $pids 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST; then
935 if [ "$withtap" -eq "1" ]; then
936 fail "Sending SIGSTOP to session daemon"
937 fi
938 else
939 out=1
940 while [ $out -ne 0 ]; do
941 pids="$(lttng_pgrep "$SESSIOND_MATCH")"
942
943 # Wait until state becomes stopped for session
944 # daemon(s).
945 out=0
946 for sessiond_pid in $pids; do
947 state="$(ps -p "$sessiond_pid" -o state= )"
948 if [[ -n "$state" && "$state" != "T" ]]; then
949 out=1
950 fi
951 done
952 sleep 0.5
953 done
954 if [ "$withtap" -eq "1" ]; then
955 pass "Sending SIGSTOP to session daemon"
956 fi
957 fi
958}
959
960function sigstop_lttng_sessiond()
961{
962 sigstop_lttng_sessiond_opt 1 "$@"
963}
964
965function sigstop_lttng_sessiond_notap()
966{
967 sigstop_lttng_sessiond_opt 0 "$@"
968}
969
970function stop_lttng_consumerd_opt()
971{
972 local withtap=$1
973 local is_cleanup=$2
974 local signal=$3
975 local timeout_s=$4
976 local dtimeleft_s=
977 local retval=0
978 local pids
979
980 if [ -z "$signal" ]; then
981 signal=SIGTERM
982 fi
983
984 # Multiply time by 2 to simplify integer arithmetic
985 if [ -n "$timeout_s" ]; then
986 dtimeleft_s=$((timeout_s * 2))
987 fi
988
989 pids="$(lttng_pgrep "$CONSUMERD_MATCH")"
990
991 if [ -z "$pids" ]; then
992 if [ "$is_cleanup" -eq 1 ]; then
993 :
994 elif [ "$withtap" -eq "1" ]; then
995 fail "No consumerd daemon to kill"
996 else
997 LTTNG_BAIL_OUT "No consumerd daemon to kill"
998 fi
999 return 0
1000 fi
1001
1002 diag "Killing (signal $signal) $CONSUMERD_BIN pids: $(echo "$pids" | tr '\n' ' ')"
1003
1004 # shellcheck disable=SC2086
1005 if ! kill -s $signal $pids 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST; then
1006 retval=1
1007 if [ "$withtap" -eq "1" ]; then
1008 fail "Kill consumer daemon"
1009 fi
1010 else
1011 out=1
1012 while [ $out -ne 0 ]; do
1013 pids="$(lttng_pgrep "$CONSUMERD_MATCH")"
1014
1015 # If consumerds are still present check their status.
1016 # A zombie status qualifies the consumerd as *killed*
1017 out=0
1018 for consumer_pid in $pids; do
1019 state="$(ps -p "$consumer_pid" -o state= )"
1020 if [[ -n "$state" && "$state" != "Z" ]]; then
1021 out=1
1022 fi
1023 done
1024 if [ -n "$dtimeleft_s" ]; then
1025 if [ $dtimeleft_s -lt 0 ]; then
1026 out=0
1027 retval=1
1028 fi
1029 dtimeleft_s=$((dtimeleft_s - 1))
1030 fi
1031 sleep 0.5
1032 done
1033 if [ "$withtap" -eq "1" ]; then
1034 if [ "$retval" -eq "0" ]; then
1035 pass "Wait after kill consumer daemon"
1036 else
1037 fail "Wait after kill consumer daemon"
1038 fi
1039 fi
1040 fi
1041
1042 return $retval
1043}
1044
1045function stop_lttng_consumerd()
1046{
1047 stop_lttng_consumerd_opt 1 0 "$@"
1048}
1049
1050function stop_lttng_consumerd_notap()
1051{
1052 stop_lttng_consumerd_opt 0 0 "$@"
1053}
1054
1055function stop_lttng_consumerd_cleanup()
1056{
1057 stop_lttng_consumerd_opt 0 1 "$@"
1058}
1059
1060function sigstop_lttng_consumerd_opt()
1061{
1062 local withtap=$1
1063 local signal=SIGSTOP
1064 local pids
1065
1066 pids="$(lttng_pgrep "$CONSUMERD_MATCH")"
1067
1068 diag "Sending SIGSTOP to $CONSUMERD_BIN pids: $(echo "$pids" | tr '\n' ' ')"
1069
1070 # shellcheck disable=SC2086
1071 kill -s $signal $pids 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1072 retval=$?
1073
1074 if [ $retval -eq 1 ]; then
1075 if [ "$withtap" -eq "1" ]; then
1076 fail "Sending SIGSTOP to consumer daemon"
1077 fi
1078 return 1
1079 else
1080 out=1
1081 while [ $out -ne 0 ]; do
1082 pids="$(lttng_pgrep "$CONSUMERD_MATCH")"
1083
1084 # Wait until state becomes stopped for all
1085 # consumers.
1086 out=0
1087 for consumer_pid in $pids; do
1088 state="$(ps -p "$consumer_pid" -o state= )"
1089 if [[ -n "$state" && "$state" != "T" ]]; then
1090 out=1
1091 fi
1092 done
1093 sleep 0.5
1094 done
1095 if [ "$withtap" -eq "1" ]; then
1096 pass "Sending SIGSTOP to consumer daemon"
1097 fi
1098 fi
1099 return $retval
1100}
1101
1102function sigstop_lttng_consumerd()
1103{
1104 sigstop_lttng_consumerd_opt 1 "$@"
1105}
1106
1107function sigstop_lttng_consumerd_notap()
1108{
1109 sigstop_lttng_consumerd_opt 0 "$@"
1110}
1111
1112function list_lttng_with_opts ()
1113{
1114 local ret
1115 local withtap=$1
1116 shift
1117 local opts=$1
1118 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1119 list $opts
1120 ret=$?
1121 if [ $withtap -eq "1" ]; then
1122 ok $ret "Lttng-tool list command with option $opts"
1123 fi
1124}
1125
1126function list_lttng_ok ()
1127{
1128 list_lttng_with_opts 1 "$@"
1129}
1130
1131function list_lttng_notap ()
1132{
1133 list_lttng_with_opts 0 "$@"
1134}
1135
1136function create_lttng_session_no_output ()
1137{
1138 local sess_name=$1
1139 local opts="${@:2}"
1140
1141 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1142 create $sess_name --no-output $opts
1143 ok $? "Create session $sess_name in no-output mode"
1144}
1145
1146function create_lttng_session_uri () {
1147 local sess_name=$1
1148 local uri=$2
1149 local opts="${@:3}"
1150
1151 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1152 create $sess_name -U $uri $opts
1153 ok $? "Create session $sess_name with uri:$uri and opts: $opts"
1154}
1155
1156function create_lttng_session ()
1157{
1158 local withtap=$1
1159 local expected_to_fail=$2
1160 local sess_name=$3
1161 local trace_path=$4
1162 local opt=$5
1163
1164 if [ -z "$trace_path" ]; then
1165 # Use lttng-sessiond default output.
1166 trace_path=""
1167 else
1168 trace_path="-o $trace_path"
1169 fi
1170
1171 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1172 create "$sess_name" $trace_path $opt
1173 ret=$?
1174 if [ $expected_to_fail -eq "1" ]; then
1175 test "$ret" -ne "0"
1176 ret=$?
1177 if [ $withtap -eq "1" ]; then
1178 ok $ret "Create session $sess_name in $trace_path failed as expected"
1179 fi
1180 else
1181 if [ $withtap -eq "1" ]; then
1182 ok $ret "Create session $sess_name in $trace_path"
1183 fi
1184 fi
1185 return $ret
1186}
1187
1188function create_lttng_session_ok ()
1189{
1190 create_lttng_session 1 0 "$@"
1191}
1192
1193function create_lttng_session_fail ()
1194{
1195 create_lttng_session 1 1 "$@"
1196}
1197
1198function create_lttng_session_notap ()
1199{
1200 create_lttng_session 0 0 "$@"
1201}
1202
1203
1204function enable_ust_lttng_channel ()
1205{
1206 local withtap=$1
1207 local expected_to_fail=$2
1208 local sess_name=$3
1209 local channel_name=$4
1210 local opts="${@:5}"
1211
1212 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1213 enable-channel -u $channel_name -s $sess_name $opts
1214 ret=$?
1215 if [[ $expected_to_fail -eq "1" ]]; then
1216 test "$ret" -ne "0"
1217 ret=$?
1218 if [ $withtap -eq "1" ]; then
1219 ok $ret "Enable channel $channel_name for session $sess_name failed as expected"
1220 fi
1221 else
1222 if [ $withtap -eq "1" ]; then
1223 ok $ret "Enable channel $channel_name for session $sess_name"
1224 fi
1225 fi
1226 return $ret
1227}
1228
1229function enable_ust_lttng_channel_ok ()
1230{
1231 enable_ust_lttng_channel 1 0 "$@"
1232}
1233
1234function enable_ust_lttng_channel_fail ()
1235{
1236 enable_ust_lttng_channel 1 1 "$@"
1237}
1238
1239function enable_ust_lttng_channel_notap ()
1240{
1241 enable_ust_lttng_channel 0 0 "$@"
1242}
1243
1244function disable_ust_lttng_channel()
1245{
1246 local sess_name=$1
1247 local channel_name=$2
1248
1249 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1250 disable-channel -u $channel_name -s $sess_name
1251 ok $? "Disable channel $channel_name for session $sess_name"
1252}
1253
1254function enable_lttng_mmap_overwrite_kernel_channel()
1255{
1256 local sess_name=$1
1257 local channel_name=$2
1258
1259 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1260 enable-channel -s $sess_name $channel_name -k --output mmap --overwrite
1261 ok $? "Enable channel $channel_name for session $sess_name"
1262}
1263
1264function enable_lttng_mmap_discard_small_kernel_channel()
1265{
1266 local sess_name=$1
1267 local channel_name=$2
1268
1269 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1270 enable-channel -s $sess_name $channel_name -k --output mmap --discard --subbuf-size=$(getconf PAGE_SIZE) --num-subbuf=2
1271 ok $? "Enable small discard channel $channel_name for session $sess_name"
1272}
1273
1274function enable_lttng_mmap_overwrite_small_kernel_channel()
1275{
1276 local sess_name=$1
1277 local channel_name=$2
1278
1279 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1280 enable-channel -s $sess_name $channel_name -k --output mmap --overwrite --subbuf-size=$(getconf PAGE_SIZE) --num-subbuf=2
1281 ok $? "Enable small discard channel $channel_name for session $sess_name"
1282}
1283
1284function enable_lttng_mmap_overwrite_ust_channel()
1285{
1286 local sess_name=$1
1287 local channel_name=$2
1288
1289 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1290 enable-channel -s $sess_name $channel_name -u --output mmap --overwrite
1291 ok $? "Enable channel $channel_name for session $sess_name"
1292}
1293
1294function enable_ust_lttng_event ()
1295{
1296 local withtap=$1
1297 local expected_to_fail=$2
1298 local sess_name=$3
1299 local event_name="$4"
1300 local channel_name=$5
1301
1302 if [ -z $channel_name ]; then
1303 # default channel if none specified
1304 chan=""
1305 else
1306 chan="-c $channel_name"
1307 fi
1308
1309 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1310 enable-event "$event_name" $chan -s "$sess_name" -u
1311 ret=$?
1312 if [[ $expected_to_fail -eq "1" ]]; then
1313 test $ret -ne "0"
1314 ret=$?
1315 if [[ $withtap -eq "1" ]]; then
1316 ok $ret "Enable ust event $event_name for session $session_name failed as expected"
1317 fi
1318 else
1319 if [[ $withtap -eq "1" ]]; then
1320 ok $ret "Enable ust event $event_name for session $sess_name"
1321 fi
1322 fi
1323 return $ret
1324}
1325
1326function enable_ust_lttng_event_ok ()
1327{
1328 enable_ust_lttng_event 1 0 "$@"
1329}
1330
1331function enable_ust_lttng_event_fail ()
1332{
1333 enable_ust_lttng_event 1 1 "$@"
1334}
1335
1336function enable_ust_lttng_event_notap ()
1337{
1338 enable_ust_lttng_event 0 0 "$@"
1339}
1340
1341function enable_jul_lttng_event()
1342{
1343 sess_name=$1
1344 event_name="$2"
1345 channel_name=$3
1346
1347 if [ -z $channel_name ]; then
1348 # default channel if none specified
1349 chan=""
1350 else
1351 chan="-c $channel_name"
1352 fi
1353
1354 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1355 enable-event "$event_name" $chan -s $sess_name -j
1356 ok $? "Enable JUL event $event_name for session $sess_name"
1357}
1358
1359function enable_jul_lttng_event_loglevel()
1360{
1361 local sess_name=$1
1362 local event_name="$2"
1363 local loglevel=$3
1364 local channel_name=$4
1365
1366 if [ -z $channel_name ]; then
1367 # default channel if none specified
1368 chan=""
1369 else
1370 chan="-c $channel_name"
1371 fi
1372
1373 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1374 enable-event --loglevel $loglevel "$event_name" $chan -s $sess_name -j
1375 ok $? "Enable JUL event $event_name for session $sess_name with loglevel $loglevel"
1376}
1377
1378function enable_log4j_lttng_event()
1379{
1380 local sess_name=$1
1381 local event_name=$2
1382 local channel_name=$3
1383
1384 local chan_opt=()
1385
1386 # default channel if none specified
1387 if [ -n "$channel_name" ]; then
1388 chan_opt=("-c" "$channel_name")
1389 fi
1390
1391 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1392 enable-event "$event_name" "${chan_opt[@]}" -s "$sess_name" --log4j
1393 ok $? "Enable LOG4J event '$event_name' for session '$sess_name'"
1394}
1395
1396function enable_log4j_lttng_event_filter()
1397{
1398 local sess_name=$1
1399 local event_name=$2
1400 local filter=$3
1401
1402 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1403 enable-event "$event_name" -s "$sess_name" --log4j --filter "$filter"
1404 ok $? "Enable LOG4J event '$event_name' with filter '$filter' for session '$sess_name'"
1405}
1406
1407function enable_log4j_lttng_event_filter_loglevel_only()
1408{
1409 local sess_name=$1
1410 local event_name=$2
1411 local filter=$3
1412 local loglevel=$4
1413
1414 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1415 enable-event --loglevel-only "$loglevel" "$event_name" -s "$sess_name" -l --filter "$filter"
1416 ok $? "Enable LOG4J event '$event_name' with filter '$filter' and loglevel-only '$loglevel' for session '$sess_name'"
1417}
1418
1419function enable_log4j_lttng_event_loglevel()
1420{
1421 local sess_name=$1
1422 local event_name=$2
1423 local loglevel=$3
1424 local channel_name=$4
1425
1426
1427 # default channel if none specified
1428 if [ -n "$channel_name" ]; then
1429 chan_opt=("-c" "$channel_name")
1430 fi
1431
1432 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1433 enable-event --loglevel "$loglevel" "$event_name" "${chan_opt[@]}" -s "$sess_name" --log4j
1434 ok $? "Enable LOG4J event '$event_name' for session '$sess_name' with loglevel '$loglevel'"
1435}
1436
1437function enable_log4j_lttng_event_loglevel_only()
1438{
1439 local sess_name=$1
1440 local event_name=$2
1441 local loglevel=$3
1442 local channel_name=$4
1443
1444 local chan_opt=()
1445
1446 # default channel if none specified
1447 if [ -n "$channel_name" ]; then
1448 chan_opt=("-c" "$channel_name")
1449 fi
1450
1451 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1452 enable-event --loglevel-only "$loglevel" "$event_name" "${chan_opt[@]}" -s "$sess_name" --log4j
1453 ok $? "Enable LOG4J event '$event_name' for session '$sess_name' with loglevel-only '$loglevel'"
1454}
1455
1456function enable_python_lttng_event()
1457{
1458 sess_name=$1
1459 event_name="$2"
1460 channel_name=$3
1461
1462 if [ -z $channel_name ]; then
1463 # default channel if none specified
1464 chan=""
1465 else
1466 chan="-c $channel_name"
1467 fi
1468
1469 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1470 enable-event "$event_name" $chan -s $sess_name -p
1471 ok $? "Enable Python event $event_name for session $sess_name"
1472}
1473
1474function enable_python_lttng_event_loglevel()
1475{
1476 local sess_name=$1
1477 local event_name="$2"
1478 local loglevel=$3
1479 local channel_name=$4
1480
1481 if [ -z $channel_name ]; then
1482 # default channel if none specified
1483 chan=""
1484 else
1485 chan="-c $channel_name"
1486 fi
1487
1488 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1489 enable-event --loglevel $loglevel "$event_name" $chan -s $sess_name -p
1490 ok $? "Enable Python event $event_name for session $sess_name with loglevel $loglevel"
1491}
1492
1493function enable_ust_lttng_event_filter()
1494{
1495 local sess_name="$1"
1496 local event_name="$2"
1497 local filter="$3"
1498 local channel_name=$4
1499
1500 if [ -z $channel_name ]; then
1501 # default channel if none specified
1502 chan=""
1503 else
1504 chan="-c $channel_name"
1505 fi
1506
1507 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1508 enable-event $chan "$event_name" -s $sess_name -u --filter "$filter"
1509 ok $? "Enable event $event_name with filtering for session $sess_name"
1510}
1511
1512function enable_ust_lttng_event_loglevel()
1513{
1514 local sess_name="$1"
1515 local event_name="$2"
1516 local loglevel="$3"
1517
1518 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1519 enable-event "$event_name" -s $sess_name -u --loglevel $loglevel
1520 ok $? "Enable event $event_name with loglevel $loglevel"
1521}
1522
1523function enable_ust_lttng_event_loglevel_only()
1524{
1525 local sess_name="$1"
1526 local event_name="$2"
1527 local loglevel="$3"
1528
1529 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1530 enable-event "$event_name" -s $sess_name -u --loglevel-only $loglevel
1531 ok $? "Enable event $event_name with loglevel-only $loglevel"
1532}
1533
1534function disable_ust_lttng_event ()
1535{
1536 local sess_name="$1"
1537 local event_name="$2"
1538 local channel_name="$3"
1539
1540 if [ -z $channel_name ]; then
1541 # default channel if none specified
1542 chan=""
1543 else
1544 chan="-c $channel_name"
1545 fi
1546
1547 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1548 disable-event "$event_name" -s $sess_name $chan -u
1549 ok $? "Disable event $event_name for session $sess_name"
1550}
1551
1552function disable_jul_lttng_event ()
1553{
1554 local sess_name="$1"
1555 local event_name="$2"
1556
1557 $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name -j >/dev/null 2>&1
1558 ok $? "Disable JUL event $event_name for session $sess_name"
1559}
1560
1561function disable_log4j_lttng_event ()
1562{
1563 local sess_name="$1"
1564 local event_name="$2"
1565
1566 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1567 disable-event "$event_name" -s "$sess_name" --log4j
1568 ok $? "Disable LOG4J event '$event_name' for session '$sess_name'"
1569}
1570
1571function disable_python_lttng_event ()
1572{
1573 local sess_name="$1"
1574 local event_name="$2"
1575
1576 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1577 disable-event "$event_name" -s $sess_name -p
1578 ok $? "Disable Python event $event_name for session $sess_name"
1579}
1580
1581function start_lttng_tracing_opt ()
1582{
1583 local withtap=$1
1584 local expected_to_fail=$2
1585 local sess_name=$3
1586
1587 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1588 start $sess_name
1589 ret=$?
1590 if [[ $expected_to_fail -eq "1" ]]; then
1591 test "$ret" -ne "0"
1592 ret=$?
1593 if [ $withtap -eq "1" ]; then
1594 ok $? "Start tracing for session $sess_name failed as expected"
1595 fi
1596 else
1597 if [ $withtap -eq "1" ]; then
1598 ok $ret "Start tracing for session $sess_name"
1599 fi
1600 fi
1601}
1602
1603function start_lttng_tracing_ok ()
1604{
1605 start_lttng_tracing_opt 1 0 "$@"
1606}
1607
1608function start_lttng_tracing_fail ()
1609{
1610 start_lttng_tracing_opt 1 1 "$@"
1611}
1612
1613function start_lttng_tracing_notap ()
1614{
1615 start_lttng_tracing_opt 0 1 "$@"
1616}
1617
1618function stop_lttng_tracing_opt ()
1619{
1620 local withtap=$1
1621 local expected_to_fail=$2
1622 local sess_name=$3
1623
1624 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1625 stop $sess_name
1626 ret=$?
1627 if [[ $expected_to_fail -eq "1" ]]; then
1628 test "$ret" -ne "0"
1629 ret=$?
1630 if [ $withtap -eq "1" ]; then
1631 ok $? "Stop lttng tracing for session $sess_name failed as expected"
1632 fi
1633 else
1634 if [ $withtap -eq "1" ]; then
1635 ok $ret "Stop lttng tracing for session $sess_name"
1636 fi
1637 fi
1638}
1639
1640function stop_lttng_tracing_ok ()
1641{
1642 stop_lttng_tracing_opt 1 0 "$@"
1643}
1644
1645function stop_lttng_tracing_fail ()
1646{
1647 stop_lttng_tracing_opt 1 1 "$@"
1648}
1649
1650function stop_lttng_tracing_notap ()
1651{
1652 stop_lttng_tracing_opt 0 0 "$@"
1653}
1654
1655function destroy_lttng_session ()
1656{
1657 local withtap=$1
1658 local expected_to_fail=$2
1659 local sess_name=$3
1660
1661 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1662 destroy $sess_name
1663 ret=$?
1664 if [[ $expected_to_fail -eq "1" ]]; then
1665 test "$ret" -ne "0"
1666 ret=$?
1667 if [ $withtap -eq "1" ]; then
1668 ok $ret "Destroy session $sess_name failed as expected"
1669 fi
1670 else
1671 if [ $withtap -eq "1" ]; then
1672 ok $ret "Destroy session $sess_name"
1673 fi
1674 fi
1675}
1676
1677function destroy_lttng_session_ok ()
1678{
1679 destroy_lttng_session 1 0 "$@"
1680
1681}
1682
1683function destroy_lttng_session_fail ()
1684{
1685 destroy_lttng_session 1 1 "$@"
1686}
1687
1688function destroy_lttng_session_notap ()
1689{
1690 destroy_lttng_session 0 0 "$@"
1691}
1692
1693function destroy_lttng_sessions ()
1694{
1695 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1696 destroy --all
1697 ok $? "Destroy all lttng sessions"
1698}
1699
1700function lttng_snapshot_add_output ()
1701{
1702 local expected_to_fail=$1
1703 local sess_name=$2
1704 local trace_path=$3
1705 local opts=$4
1706
1707 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1708 snapshot add-output -s $sess_name $trace_path $opts
1709 ret=$?
1710 if [[ $expected_to_fail -eq 1 ]]; then
1711 test "$ret" -ne "0"
1712 ok $? "Added snapshot output $trace_path failed as expected"
1713 else
1714 ok $ret "Added snapshot output $trace_path"
1715 fi
1716}
1717
1718function lttng_snapshot_add_output_ok ()
1719{
1720 lttng_snapshot_add_output 0 "$@"
1721}
1722
1723function lttng_snapshot_add_output_fail ()
1724{
1725 lttng_snapshot_add_output 1 "$@"
1726}
1727
1728function lttng_snapshot_del_output ()
1729{
1730 local expected_to_fail=$1
1731 local sess_name=$2
1732 local id=$3
1733
1734 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1735 snapshot del-output -s $sess_name $id
1736 ret=$?
1737 if [[ $expected_to_fail -eq "1" ]]; then
1738 test "$ret" -ne "0"
1739 ok $? "Deleted snapshot output id $id failed as expected"
1740 else
1741 ok $ret "Deleted snapshot output id $id"
1742 fi
1743}
1744
1745function lttng_snapshot_del_output_ok ()
1746{
1747 lttng_snapshot_del_output 0 "$@"
1748}
1749
1750function lttng_snapshot_del_output_fail ()
1751{
1752 lttng_snapshot_del_output 1 "$@"
1753}
1754
1755function lttng_snapshot_record ()
1756{
1757 local sess_name=$1
1758 local trace_path=$2
1759
1760 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1761 snapshot record -s "$sess_name" "$trace_path"
1762 ok $? "Snapshot recorded"
1763}
1764
1765function lttng_snapshot_list ()
1766{
1767 local sess_name=$1
1768 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1769 snapshot list-output -s $sess_name
1770 ok $? "Snapshot list"
1771}
1772
1773function lttng_save()
1774{
1775 local sess_name=$1
1776 local opts=$2
1777
1778 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1779 save $sess_name $opts
1780 ok $? "Session saved"
1781}
1782
1783function lttng_load()
1784{
1785 local expected_to_fail=$1
1786 local opts=$2
1787
1788 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1789 load $opts
1790 ret=$?
1791 if [[ $expected_to_fail -eq "1" ]]; then
1792 test $ret -ne "0"
1793 ok $? "Load command failed as expected with opts: $opts"
1794 else
1795 ok $ret "Load command with opts: $opts"
1796 fi
1797}
1798
1799function lttng_load_ok()
1800{
1801 lttng_load 0 "$@"
1802}
1803
1804function lttng_load_fail()
1805{
1806 lttng_load 1 "$@"
1807}
1808
1809function lttng_track()
1810{
1811 local expected_to_fail="$1"
1812 shift 1
1813 local opts="$@"
1814 $TESTDIR/../src/bin/lttng/$LTTNG_BIN track $opts >$OUTPUT_DEST
1815 ret=$?
1816 if [[ $expected_to_fail -eq "1" ]]; then
1817 test $ret -ne "0"
1818 ok $? "Track command failed as expected with opts: $opts"
1819 else
1820 ok $ret "Track command with opts: $opts"
1821 fi
1822}
1823
1824function lttng_track_ok()
1825{
1826 lttng_track 0 "$@"
1827}
1828
1829function lttng_track_fail()
1830{
1831 lttng_track 1 "$@"
1832}
1833
1834function lttng_untrack()
1835{
1836 local expected_to_fail="$1"
1837 shift 1
1838 local opts="$@"
1839 $TESTDIR/../src/bin/lttng/$LTTNG_BIN untrack $opts >$OUTPUT_DEST
1840 ret=$?
1841 if [[ $expected_to_fail -eq "1" ]]; then
1842 test $ret -ne "0"
1843 ok $? "Untrack command failed as expected with opts: $opts"
1844 else
1845 ok $ret "Untrack command with opts: $opts"
1846 fi
1847}
1848
1849function lttng_untrack_ok()
1850{
1851 lttng_untrack 0 "$@"
1852}
1853
1854function lttng_untrack_fail()
1855{
1856 lttng_untrack 1 "$@"
1857}
1858
1859function lttng_track_pid_ok()
1860{
1861 PID=$1
1862 "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" track --kernel --pid=$PID 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1863 ok $? "Lttng track pid on the kernel domain"
1864}
1865
1866function lttng_untrack_kernel_all_ok()
1867{
1868 "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" untrack --kernel --pid --all 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
1869 ok $? "Lttng untrack all pid on the kernel domain"
1870}
1871
1872function lttng_track_ust_ok()
1873{
1874 lttng_track_ok -u "$@"
1875}
1876
1877function lttng_track_ust_fail()
1878{
1879 lttng_track_fail -u "$@"
1880}
1881
1882function lttng_track_kernel_ok()
1883{
1884 lttng_track_ok -k "$@"
1885}
1886
1887function lttng_track_kernel_fail()
1888{
1889 lttng_track_fail -k "$@"
1890}
1891
1892function lttng_untrack_ust_ok()
1893{
1894 lttng_untrack_ok -u "$@"
1895}
1896
1897function lttng_untrack_ust_fail()
1898{
1899 lttng_untrack_fail -u "$@"
1900}
1901
1902function lttng_untrack_kernel_ok()
1903{
1904 lttng_untrack_ok -k "$@"
1905}
1906
1907function lttng_untrack_kernel_fail()
1908{
1909 lttng_untrack_fail -k "$@"
1910}
1911
1912function lttng_add_context_list()
1913{
1914 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1915 add-context --list
1916 ret=$?
1917 ok $ret "Context listing"
1918}
1919
1920function add_context_lttng()
1921{
1922 local expected_to_fail="$1"
1923 local domain="$2"
1924 local session_name="$3"
1925 local channel_name="$4"
1926 local type="$5"
1927
1928 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
1929 add-context -s $session_name -c $channel_name -t $type $domain
1930 ret=$?
1931 if [[ $expected_to_fail -eq "1" ]]; then
1932 test $ret -ne "0"
1933 ok $? "Add context command failed as expected for type: $type"
1934 else
1935 ok $ret "Add context command for type: $type"
1936 fi
1937}
1938
1939function add_context_ust_ok()
1940{
1941 add_context_lttng 0 -u "$@"
1942}
1943
1944function add_context_ust_fail()
1945{
1946 add_context_lttng 1 -u "$@"
1947}
1948
1949function add_context_kernel_ok()
1950{
1951 add_context_lttng 0 -k "$@"
1952}
1953
1954function add_context_kernel_fail()
1955{
1956 add_context_lttng 1 -k "$@"
1957}
1958
1959function wait_live_trace_ready ()
1960{
1961 local url=$1
1962 local zero_client_match=0
1963
1964 diag "Waiting for live trace at url: $url"
1965 while [ $zero_client_match -eq 0 ]; do
1966 zero_client_match=$($BABELTRACE_BIN -i lttng-live $url | grep "0 client(s) connected" | wc -l)
1967 sleep 0.5
1968 done
1969 pass "Waiting for live trace at url: $url"
1970}
1971
1972function wait_live_viewer_connect ()
1973{
1974 local url=$1
1975 local one_client_match=0
1976
1977 diag "Waiting for live viewers on url: $url"
1978 while [ $one_client_match -eq 0 ]; do
1979 one_client_match=$($BABELTRACE_BIN -i lttng-live $url | grep "1 client(s) connected" | wc -l)
1980 sleep 0.5
1981 done
1982 pass "Waiting for live viewers on url: $url"
1983}
1984
1985function bail_out_if_no_babeltrace()
1986{
1987 which "$BABELTRACE_BIN" >/dev/null
1988 if [ $? -ne 0 ]; then
1989 LTTNG_BAIL_OUT "\"$BABELTRACE_BIN\" binary not found. Skipping tests"
1990 fi
1991}
1992
1993# Check that the trace metadata contains '$expected' event ids matching '$event_name'.
1994function validate_metadata_event()
1995{
1996 local event_name=$1
1997 local expected=$2
1998 local trace_path=$3
1999
2000 local metadata_file
2001 local metadata_path
2002 local count
2003
2004 metadata_file=$(find "$trace_path" -name "metadata")
2005 metadata_path=$(dirname "$metadata_file")
2006
2007 bail_out_if_no_babeltrace
2008
2009 count=$($BABELTRACE_BIN --output-format=ctf-metadata "$metadata_path" | grep -c "$event_name")
2010
2011 test "$count" -eq "$expected"
2012 ok $? "Found $count / $expected metadata event id matching '$event_name'"
2013}
2014
2015# Check that the trace contains '$expected' events matching '$event_name', other
2016# events not matching '$event_name' can be present.
2017function trace_matches()
2018{
2019 local event_name=$1
2020 local expected=$2
2021 local trace_path=$3
2022
2023 local count
2024 local total
2025
2026 bail_out_if_no_babeltrace
2027
2028 count=$($BABELTRACE_BIN "$trace_path" | grep -c "$event_name")
2029 total=$($BABELTRACE_BIN "$trace_path" | wc -l)
2030
2031 test "$count" -eq "$expected"
2032
2033 ok $? "Found $count / $expected events matching '$event_name' out of $total events"
2034}
2035
2036# Check that the trace contains '$expected' events matching '$event_name' and no
2037# other events.
2038function trace_match_only()
2039{
2040 local event_name=$1
2041 local expected=$2
2042 local trace_path=$3
2043
2044 local count
2045 local total
2046
2047 bail_out_if_no_babeltrace
2048
2049 count=$($BABELTRACE_BIN "$trace_path" | grep -c "$event_name")
2050 total=$($BABELTRACE_BIN "$trace_path" | wc -l)
2051
2052 test "$expected" -eq "$count" && test "$total" -eq "$expected"
2053
2054 ok $? "Found $count / $expected events matching '$event_name' amongst $total events"
2055}
2056
2057# Check that the trace contains at least 1 event matching each name in the
2058# comma separated list '$event_names'.
2059function validate_trace()
2060{
2061 local event_names=$1
2062 local trace_path=$2
2063
2064 local count
2065
2066 bail_out_if_no_babeltrace
2067
2068 OLDIFS=$IFS
2069 IFS=","
2070 for event_name in $event_names; do
2071 # trace_path is unquoted since callers make use of globbing
2072 count=$($BABELTRACE_BIN $trace_path | grep -c "$event_name")
2073 test "$count" -gt 0
2074 ok $? "Found $count events matching '$event_name'"
2075 done
2076 IFS=$OLDIFS
2077}
2078
2079# Check that the trace contains at least 1 event matching each name in the
2080# comma separated list '$event_names' and a total of '$expected' events.
2081function validate_trace_count()
2082{
2083 local event_names=$1
2084 local trace_path=$2
2085 local expected=$3
2086
2087 local count
2088 local total=0
2089
2090 bail_out_if_no_babeltrace
2091
2092 OLDIFS=$IFS
2093 IFS=","
2094 for event_name in $event_names; do
2095 count=$($BABELTRACE_BIN "$trace_path" | grep -c "$event_name")
2096 test "$count" -gt 0
2097 ok $? "Found '$count' events matching '$event_name'"
2098 total=$(( total + count ))
2099 done
2100 IFS=$OLDIFS
2101 test $total -eq "$expected"
2102 ok $? "Found $total events, expected $expected events"
2103}
2104
2105# Check that the trace contains at least '$expected_min' event matching each
2106# name in the comma separated list '$event_names' and a total at least
2107# '$expected_min' and less than '$expected_max' events.
2108function validate_trace_count_range_incl_min_excl_max()
2109{
2110 local event_names=$1
2111 local trace_path=$2
2112 local expected_min=$3
2113 local expected_max=$4
2114
2115 local count
2116 local total=0
2117
2118 bail_out_if_no_babeltrace
2119
2120 OLDIFS=$IFS
2121 IFS=","
2122 for event_name in $event_names; do
2123 count=$($BABELTRACE_BIN "$trace_path" | grep -c "$event_name")
2124 test "$count" -ge "$expected_min"
2125 ok $? "Found $count events matching '$event_name', expected at least $expected_min"
2126 total=$(( total + count ))
2127 done
2128 IFS=$OLDIFS
2129 test $total -ge "$expected_min" && test $total -lt "$expected_max"
2130 ok $? "Found a total of $total events, expected at least $expected_min and less than $expected_max"
2131}
2132
2133function trace_first_line()
2134{
2135 local trace_path=$1
2136
2137 $BABELTRACE_BIN "$trace_path" | head -n 1
2138}
2139
2140# Check that the trace contains at least 1 event matching the grep extended
2141# regexp '$event_exp'.
2142function validate_trace_exp()
2143{
2144 local event_exp=$1
2145 local trace_path=$2
2146
2147 local count
2148
2149 bail_out_if_no_babeltrace
2150
2151 # event_exp is unquoted since it contains multiple grep arguments
2152 count=$($BABELTRACE_BIN "$trace_path" | grep -c --extended-regexp $event_exp)
2153 test "$count" -gt 0
2154 ok $? "Found $count events matching expression '$event_exp'"
2155}
2156
2157# Check that the trace contains at least 1 event matching the grep extended
2158# regexp '$event_exp' and zero event not matching it.
2159function validate_trace_only_exp()
2160{
2161 local event_exp=$1
2162 local trace_path=$2
2163
2164 local count
2165 local total
2166
2167 bail_out_if_no_babeltrace
2168
2169 # event_exp is unquoted since it contains multiple grep arguments
2170 count=$($BABELTRACE_BIN "$trace_path" | grep -c --extended-regexp $event_exp)
2171 total=$($BABELTRACE_BIN "$trace_path" | wc -l)
2172
2173 test "$count" -gt 0 && test "$total" -eq "$count"
2174 ok $? "Found $count events matching expression '$event_exp' amongst $total events"
2175}
2176
2177# Check that the trace is valid and contains 0 event.
2178function validate_trace_empty()
2179{
2180 local trace_path=$1
2181
2182 local ret
2183 local count
2184
2185 bail_out_if_no_babeltrace
2186
2187 events=$($BABELTRACE_BIN "$trace_path")
2188 ret=$?
2189 if [ $ret -ne 0 ]; then
2190 fail "Failed to parse trace"
2191 return $ret
2192 fi
2193
2194 count=$(echo -n "$events" | wc -l)
2195 test "$count" -eq 0
2196 ok $? "Validate trace is empty, found $count events"
2197}
2198
2199function validate_directory_empty ()
2200{
2201 local trace_path="$1"
2202
2203 local files
2204 local ret
2205 local nb_files
2206
2207 # Do not double quote `$trace_path` below as we want wildcards to be
2208 # expanded.
2209 files="$(ls -A $trace_path)"
2210 ret=$?
2211 if [ $ret -ne 0 ]; then
2212 fail "Failed to list content of directory \"$trace_path\""
2213 return $ret
2214 fi
2215
2216 nb_files="$(echo -n "$files" | wc -l)"
2217 test "$nb_files" -eq 0
2218 ok $? "Directory \"$trace_path\" is empty"
2219}
2220
2221function validate_trace_session_ust_empty()
2222{
2223 validate_directory_empty "$1"/ust
2224}
2225
2226function validate_trace_session_kernel_empty()
2227{
2228 validate_trace_empty "$1"/kernel
2229}
2230
2231function regenerate_metadata ()
2232{
2233 local expected_to_fail=$1
2234 local sess_name=$2
2235
2236 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
2237 regenerate metadata -s $sess_name
2238 ret=$?
2239 if [[ $expected_to_fail -eq "1" ]]; then
2240 test "$ret" -ne "0"
2241 ok $? "Expected fail on regenerate metadata $sess_name"
2242 else
2243 ok $ret "Metadata regenerate $sess_name"
2244 fi
2245}
2246
2247function regenerate_metadata_ok ()
2248{
2249 regenerate_metadata 0 "$@"
2250}
2251
2252function regenerate_metadata_fail ()
2253{
2254 regenerate_metadata 1 "$@"
2255}
2256
2257function regenerate_statedump ()
2258{
2259 local expected_to_fail=$1
2260 local sess_name=$2
2261
2262 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
2263 regenerate statedump -s $sess_name
2264 ret=$?
2265 if [[ $expected_to_fail -eq "1" ]]; then
2266 test "$ret" -ne "0"
2267 ok $? "Expected fail on regenerate statedump $sess_name"
2268 else
2269 ok $ret "Statedump regenerate $sess_name"
2270 fi
2271}
2272
2273function regenerate_statedump_ok ()
2274{
2275 regenerate_statedump 0 "$@"
2276}
2277
2278function regenerate_statedump_fail ()
2279{
2280 regenerate_statedump 1 "$@"
2281}
2282
2283function rotate_session ()
2284{
2285 local expected_to_fail=$1
2286 local sess_name=$2
2287
2288 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
2289 rotate $sess_name
2290 ret=$?
2291 if [[ $expected_to_fail -eq "1" ]]; then
2292 test "$ret" -ne "0"
2293 ok $? "Expected fail on rotate session $sess_name"
2294 else
2295 ok $ret "Rotate session $sess_name"
2296 fi
2297}
2298
2299function rotate_session_ok ()
2300{
2301 rotate_session 0 "$@"
2302}
2303
2304function rotate_session_fail ()
2305{
2306 rotate_session 1 "$@"
2307}
2308
2309function destructive_tests_enabled ()
2310{
2311 if [ "$LTTNG_ENABLE_DESTRUCTIVE_TESTS" = "will-break-my-system" ]; then
2312 return 0
2313 else
2314 return 1
2315 fi
2316}
2317
2318function lttng_enable_rotation_timer ()
2319{
2320 local expected_to_fail=$1
2321 local sess_name=$2
2322 local period=$3
2323
2324 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
2325 enable-rotation -s $sess_name --timer $period
2326 ret=$?
2327 if [[ $expected_to_fail -eq "1" ]]; then
2328 test "$ret" -ne "0"
2329 ok $? "Expected fail when setting periodic rotation ($period) of session $sess_name"
2330 else
2331 ok $ret "Set periodic rotation ($period) of session $sess_name"
2332 fi
2333}
2334
2335function lttng_enable_rotation_timer_ok ()
2336{
2337 lttng_enable_rotation_timer 0 $@
2338}
2339
2340function lttng_enable_rotation_timer_fail ()
2341{
2342 lttng_enable_rotation_timer 1 $@
2343}
2344
2345function lttng_enable_rotation_size ()
2346{
2347 local expected_to_fail=$1
2348 local sess_name=$2
2349 local size=$3
2350
2351 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
2352 enable-rotation -s $sess_name --size $size
2353 ret=$?
2354 if [[ $expected_to_fail -eq "1" ]]; then
2355 test "$ret" -ne "0"
2356 ok $? "Expected to fail to set a periodic rotation of session $sess_name" "every " $size " bytes"
2357 else
2358 ok $ret "Set a scheduled rotation of session $sess_name" "every " $size " bytes"
2359 fi
2360}
2361
2362function lttng_enable_rotation_size_ok ()
2363{
2364 lttng_enable_rotation_size 0 $@
2365}
2366
2367function lttng_enable_rotation_size_fail ()
2368{
2369 lttng_enable_rotation_size 1 $@
2370}
2371
2372function lttng_clear_session ()
2373{
2374 local expected_to_fail=$1
2375 local sess_name=$2
2376
2377 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
2378 clear $sess_name
2379 ret=$?
2380 if [[ $expected_to_fail -eq "1" ]]; then
2381 test "$ret" -ne "0"
2382 ok $? "Expected fail on clear session $sess_name"
2383 else
2384 ok $ret "Clear session $sess_name"
2385 fi
2386}
2387
2388function lttng_clear_session_ok ()
2389{
2390 lttng_clear_session 0 $@
2391}
2392
2393function lttng_clear_session_fail ()
2394{
2395 lttng_clear_session 1 $@
2396}
2397
2398function lttng_clear_all ()
2399{
2400 _run_lttng_cmd "$OUTPUT_DEST" "$ERROR_OUTPUT_DEST" \
2401 clear --all
2402 ok $? "Clear all lttng sessions"
2403}
2404
2405function lttng_add_trigger()
2406{
2407 local expected_to_fail="$1"
2408 local trigger_name="$2"
2409 shift 2
2410 local args=("$@")
2411
2412 diag "$TESTDIR/../src/bin/lttng/$LTTNG_BIN add-trigger --name $trigger_name ${args[*]}"
2413 $TESTDIR/../src/bin/lttng/$LTTNG_BIN add-trigger --name "$trigger_name" "${args[@]}" 1> /dev/null 2> /dev/null
2414 ret=$?
2415 if [[ $expected_to_fail -eq "1" ]]; then
2416 test "$ret" -ne "0"
2417 ok $? "Add trigger $trigger_name failed as expected"
2418 else
2419 ok $ret "Add trigger $trigger_name"
2420 fi
2421}
2422
2423function lttng_remove_trigger()
2424{
2425 local expected_to_fail="$1"
2426 local trigger_name="$2"
2427 shift 2
2428
2429 diag "$TESTDIR/../src/bin/lttng/$LTTNG_BIN remove-trigger $trigger_name $*"
2430 "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" remove-trigger "$trigger_name" "$@" 1> /dev/null 2> /dev/null
2431 ret=$?
2432 if [[ $expected_to_fail -eq "1" ]]; then
2433 test "$ret" -ne "0"
2434 ok $? "Remove trigger $trigger_name failed as expected"
2435 else
2436 ok $ret "Remove trigger $trigger_name"
2437 fi
2438}
2439
2440function lttng_add_trigger_ok()
2441{
2442 lttng_add_trigger 0 "$@"
2443}
2444
2445function lttng_add_trigger_fail()
2446{
2447 lttng_add_trigger 1 "$@"
2448}
2449
2450function lttng_remove_trigger_ok()
2451{
2452 lttng_remove_trigger 0 "$@"
2453}
2454
2455function list_triggers_matches_ok ()
2456{
2457 local tmp_stdout=$(mktemp -t "tmp.${FUNCNAME[0]}_stdout.XXXXXX")
2458 local tmp_stderr=$(mktemp -t "tmp.${FUNCNAME[0]}_stderr.XXXXXX")
2459
2460 local test_name="$1"
2461 local expected_stdout_file="$2"
2462
2463 diag "$TESTDIR/../src/bin/lttng/$LTTNG_BIN list-triggers"
2464
2465 "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" list-triggers > "${tmp_stdout}" 2> "${tmp_stderr}"
2466 ok $? "${test_name}: exit code is 0"
2467
2468 diff -u "${expected_stdout_file}" "${tmp_stdout}"
2469 ok $? "${test_name}: expected stdout"
2470
2471 diff -u /dev/null "${tmp_stderr}"
2472 ok $? "${test_name}: expected stderr"
2473
2474 rm -f "${tmp_stdout}"
2475 rm -f "${tmp_stderr}"
2476}
2477
2478function list_triggers_matches_mi_ok ()
2479{
2480 local tmp_stdout
2481 local tmp_stdout_raw
2482 local tmp_stderr
2483
2484 local test_name="$1"
2485 local expected_stdout_file="$2"
2486
2487 tmp_stdout_raw=$(mktemp -t "tmp.${FUNCNAME[0]}_stdout.XXXXXX")
2488 tmp_stdout=$(mktemp -t "tmp.${FUNCNAME[0]}_stdout.XXXXXX")
2489 tmp_stderr=$(mktemp -t "tmp.${FUNCNAME[0]}_stderr.XXXXXX")
2490
2491 diag "$TESTDIR/../src/bin/lttng/$LTTNG_BIN --mi xml list-triggers"
2492
2493 "$TESTDIR/../src/bin/lttng/$LTTNG_BIN" --mi=xml list-triggers > "${tmp_stdout_raw}" 2> "${tmp_stderr}"
2494 ok $? "${test_name}: exit code is 0"
2495
2496 # Pretty-fy xml before further test.
2497 $XML_PRETTY < "${tmp_stdout_raw}" > "${tmp_stdout}"
2498
2499 $MI_VALIDATE "${tmp_stdout}"
2500 ok $? "list-trigger mi is valid"
2501
2502 diff -u "${expected_stdout_file}" "${tmp_stdout}"
2503 ok $? "${test_name}: expected stdout"
2504
2505 diff -u /dev/null "${tmp_stderr}"
2506 ok $? "${test_name}: expected stderr"
2507
2508 rm -f "${tmp_stdout}"
2509 rm -f "${tmp_stdout_raw}"
2510 rm -f "${tmp_stderr}"
2511}
2512
2513function validate_path_pattern ()
2514{
2515 local message=$1
2516 local pattern=$2
2517 # Base path is only used in error case and is used to list the content
2518 # of the base path.
2519 local base_path=$3
2520
2521
2522 [ -f $pattern ]
2523 ret=$?
2524 ok $ret "$message"
2525
2526 if [ "$ret" -ne "0" ]; then
2527 diag "Path pattern expected: $pattern"
2528 # List the tracepath for more info. We use find as a recursive
2529 # directory lister.
2530 diag "The base path content:"
2531 find "$base_path" -print
2532 fi
2533}
2534
2535function validate_trace_path_ust_uid ()
2536{
2537 local trace_path=$1
2538 local session_name=$2
2539 local uid=$UID
2540 local pattern="$trace_path/$session_name-$date_time_pattern/ust/uid/$uid/${system_long_bit_size}-bit/metadata"
2541
2542 validate_path_pattern "UST per-uid trace path is valid" "$pattern" "$trace_path"
2543}
2544
2545function validate_trace_path_ust_uid_network ()
2546{
2547 local trace_path=$1
2548 local session_name=$2
2549 local base_path=$3
2550 local uid=$UID
2551 local hostname=$HOSTNAME
2552 local pattern
2553 local ret
2554
2555 # If the session was given a network base path (e.g
2556 # 127.0.0.1/my/custom/path on creation, there is no session name
2557 # component to the path on the relayd side. Caller can simply not pass a
2558 # session name for this scenario.
2559 if [ -n "$session_name" ]; then
2560 session_name="$session_name-$date_time_pattern"
2561 if [ -n "$base_path" ]; then
2562 fail "Session name and base path are mutually exclusive"
2563 return
2564 fi
2565 fi
2566
2567 pattern="$trace_path/$hostname/$base_path/$session_name/ust/uid/$uid/${system_long_bit_size}-bit/metadata"
2568
2569 validate_path_pattern "UST per-uid network trace path is valid" "$pattern" "$trace_path"
2570}
2571
2572function validate_trace_path_ust_uid_snapshot_network ()
2573{
2574 local trace_path=$1
2575 local session_name=$2
2576 local snapshot_name=$3
2577 local snapshot_number=$4
2578 local base_path=$5
2579 local hostname=$HOSTNAME
2580 local uid=$UID
2581 local pattern
2582 local ret
2583
2584 # If the session/output was given a network base path (e.g
2585 # 127.0.0.1/my/custom/path on creation, there is no session name
2586 # component to the path on the relayd side. Caller can simply not pass a
2587 # session name for this scenario.
2588 if [ -n "$session_name" ]; then
2589 session_name="$session_name-$date_time_pattern"
2590 if [ -n "$base_path" ]; then
2591 fail "Session name and base path are mutually exclusive"
2592 return
2593 fi
2594 fi
2595
2596 pattern="$trace_path/$hostname/$base_path/$session_name/$snapshot_name-$date_time_pattern-$snapshot_number/ust/uid/$uid/${system_long_bit_size}-bit/metadata"
2597
2598 validate_path_pattern "UST per-uid network snapshot trace path is valid" "$pattern" "$trace_path"
2599}
2600
2601function validate_trace_path_ust_uid_snapshot ()
2602{
2603 local trace_path=$1
2604 local session_name=$2
2605 local snapshot_name=$3
2606 local snapshot_number=$4
2607 local base_path=$5
2608 local uid=$UID
2609 local pattern
2610 local ret
2611
2612 # If the session/output was given a network base path (e.g
2613 # 127.0.0.1/my/custom/path) on creation, there is no session name
2614 # component to the path on the relayd side. Caller can simply not pass a
2615 # session name for this scenario.
2616 if [ -n "$session_name" ]; then
2617 session_name="$session_name-$date_time_pattern"
2618 if [ -n "$base_path" ]; then
2619 fail "Session name and base path are mutually exclusive"
2620 return
2621 fi
2622 fi
2623
2624 pattern="$trace_path/$base_path/$session_name/$snapshot_name-$date_time_pattern-$snapshot_number/ust/uid/$uid/${system_long_bit_size}-bit/metadata"
2625
2626 validate_path_pattern "UST per-uid snapshot trace path is valid" "$pattern" "$trace_path"
2627}
2628
2629function validate_trace_path_ust_pid ()
2630{
2631 local trace_path=$1
2632 local session_name=$2
2633 local app_string=$3
2634 local pid=$4
2635 local pattern
2636 local ret
2637
2638 # If the session was given a trace path on creation, there is no session
2639 # name component to the path. Caller can simply not pass a session name
2640 # for this scenario.
2641 if [ -n "$session_name" ]; then
2642 session_name="$session_name-$date_time_pattern"
2643 fi
2644
2645 pattern="$trace_path/$session_name/ust/pid/$pid/$app_string-*-$date_time_pattern/metadata"
2646
2647 validate_path_pattern "UST per-pid trace path is valid" "$pattern" "$trace_path"
2648}
2649
2650function validate_trace_path_kernel ()
2651{
2652 local trace_path=$1
2653 local session_name=$2
2654 local pattern
2655
2656 # If the session was given a trace path on creation, there is no session
2657 # name component to the path. Caller can simply not pass a session name
2658 # for this scenario.
2659 if [ -n "$session_name" ]; then
2660 session_name="$session_name-$date_time_pattern"
2661 fi
2662
2663 pattern="$trace_path/$session_name/kernel/metadata"
2664
2665 validate_path_pattern "Kernel trace path is valid" "$pattern" "$trace_path"
2666}
2667
2668function validate_trace_path_kernel_network ()
2669{
2670 local trace_path=$1
2671 local session_name=$2
2672 local hostname=$HOSTNAME
2673 local pattern="$trace_path/$hostname/$session_name-$date_time_pattern/kernel/metadata"
2674
2675 validate_path_pattern "Kernel network trace path is valid" "$pattern" "$trace_path"
2676}
2677
2678function validate_trace_path_kernel_snapshot ()
2679{
2680 local trace_path=$1
2681 local session_name=$2
2682 local snapshot_name=$3
2683 local snapshot_number=$4
2684 local base_path=$5
2685 local pattern
2686 local ret
2687
2688 # If the session/output was given a network base path (e.g
2689 # 127.0.0.1/my/custom/path on creation, there is no session name
2690 # component to the path on the relayd side. Caller can simply not pass a
2691 # session name for this scenario.
2692 if [ -n "$session_name" ]; then
2693 session_name="$session_name-$date_time_pattern"
2694 if [ -n "$base_path" ]; then
2695 fail "Session name and base path are mutually exclusive"
2696 return
2697 fi
2698 fi
2699
2700 pattern="$trace_path/$base_path/$session_name/$snapshot_name-$date_time_pattern-$snapshot_number/kernel/metadata"
2701
2702 validate_path_pattern "Kernel snapshot trace path is valid" "$pattern" "$trace_path"
2703}
2704
2705function validate_trace_path_kernel_snapshot_network ()
2706{
2707 local trace_path=$1
2708 local session_name=$2
2709 local snapshot_name=$3
2710 local snapshot_number=$4
2711 local base_path=$5
2712 local hostname=$HOSTNAME
2713 local pattern
2714 local ret
2715
2716 # If the session/output was given a network base path (e.g
2717 # 127.0.0.1/my/custom/path on creation, there is no session name
2718 # component to the path on the relayd side. Caller can simply not pass a
2719 # session name for this scenario.
2720 if [ -n "$session_name" ]; then
2721 session_name="$session_name-$date_time_pattern"
2722 if [ -n "$base_path" ]; then
2723 fail "Session name and base path are mutually exclusive"
2724 return
2725 fi
2726 fi
2727
2728 pattern="$trace_path/$hostname/$base_path/$session_name/$snapshot_name-$date_time_pattern-$snapshot_number/kernel/metadata"
2729
2730 validate_path_pattern "Kernel network snapshot trace path is valid" "$pattern" "$trace_path"
2731}
This page took 0.031342 seconds and 4 git commands to generate.