tests/utils/utils.sh

   1 #!/src/bin/bash
   2 #
   3 # Copyright (C) - 2012 David Goulet <dgoulet@efficios.com>
   4 #
   5 # This library is free software; you can redistribute it and/or modify it under
   6 # the terms of the GNU Lesser General Public License as published by the Free
   7 # Software Foundation; version 2.1 of the License.
   8 #
   9 # This library is distributed in the hope that it will be useful, but WITHOUT
  10 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
  11 # FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
  12 # details.
  13 #
  14 # You should have received a copy of the GNU Lesser General Public License
  15 # along with this library; if not, write to the Free Software Foundation, Inc.,
  16 # 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA
  17
  18 SESSIOND_BIN="lttng-sessiond"
  19 CONSUMERD_BIN="lttng-consumerd"
  20 RELAYD_BIN="lttng-relayd"
  21 LTTNG_BIN="lttng"
  22 BABELTRACE_BIN="babeltrace"
  23 OUTPUT_DEST=/dev/null
  24 ERROR_OUTPUT_DEST=/dev/null
  25
  26 # Minimal kernel version supported for session daemon tests
  27 KERNEL_MAJOR_VERSION=2
  28 KERNEL_MINOR_VERSION=6
  29 KERNEL_PATCHLEVEL_VERSION=27
  30
  31 # We set the default UST register timeout to "wait forever", so that
  32 # basic tests don't have to worry about hitting timeouts on busy
  33 # systems. Specialized tests should test those corner-cases.
  34 export LTTNG_UST_REGISTER_TIMEOUT=-1
  35
  36 source $TESTDIR/utils/tap/tap.sh
  37
  38 function print_ok ()
  39 {
  40         # Check if we are a terminal
  41         if [ -t 1 ]; then
  42                 echo -e "\e[1;32mOK\e[0m"
  43         else
  44                 echo -e "OK"
  45         fi
  46 }
  47
  48 function print_fail ()
  49 {
  50         # Check if we are a terminal
  51         if [ -t 1 ]; then
  52                 echo -e "\e[1;31mFAIL\e[0m"
  53         else
  54                 echo -e "FAIL"
  55         fi
  56 }
  57
  58 function print_test_banner ()
  59 {
  60         local desc="$1"
  61         diag "$desc"
  62 }
  63
  64 function validate_kernel_version ()
  65 {
  66         local kern_version=($(uname -r | awk -F. '{ printf("%d.%d.%d\n",$1,$2,$3); }' | tr '.' '\n'))
  67         if [ ${kern_version[0]} -gt $KERNEL_MAJOR_VERSION ]; then
  68                 return 0
  69         fi
  70         if [ ${kern_version[1]} -gt $KERNEL_MINOR_VERSION ]; then
  71                 return 0
  72         fi
  73         if [ ${kern_version[2]} -ge $KERNEL_PATCHLEVEL_VERSION ]; then
  74                 return 0
  75         fi
  76         return 1
  77 }
  78
  79 # Generate a random string
  80 #  $1 = number of characters; defaults to 16
  81 #  $2 = include special characters; 1 = yes, 0 = no; defaults to yes
  82 function randstring()
  83 {
  84         [ "$2" == "0" ] && CHAR="[:alnum:]" || CHAR="[:graph:]"
  85         cat /dev/urandom | tr -cd "$CHAR" | head -c ${1:-16}
  86         echo
  87 }
  88
  89 function lttng_enable_kernel_event
  90 {
  91         local sess_name=$1
  92         local event_name=$2
  93         local channel_name=$3
  94
  95         if [ -z $event_name ]; then
  96                 # Enable all event if no event name specified
  97                 event_name="-a"
  98         fi
  99
 100         if [ -z $channel_name ]; then
 101                 # default channel if none specified
 102                 chan=""
 103         else
 104                 chan="-c $channel_name"
 105         fi
 106
 107         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -k 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 108         ok $? "Enable kernel event $event_name for session $sess_name"
 109 }
 110
 111 function lttng_enable_kernel_syscall()
 112 {
 113         local expected_to_fail=$1
 114         local sess_name=$2
 115         local syscall_name=$3
 116         local channel_name=$4
 117
 118         if [ -z $syscall_name ]; then
 119                 # Enable all event if no syscall name specified
 120                 syscall_name="-a"
 121         fi
 122
 123         if [ -z $channel_name ]; then
 124                 # default channel if none specified
 125                 chan=""
 126         else
 127                 chan="-c $channel_name"
 128         fi
 129
 130         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event --syscall "$syscall_name" $chan -s $sess_name -k 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 131         ret=$?
 132         if [[ $expected_to_fail -eq "1" ]]; then
 133                 test $ret -ne "0"
 134                 ok $? "Enable kernel syscall $syscall_name for session $sess_name on channel $channel_name fail as expected"
 135         else
 136                 ok $ret "Enable kernel syscall $syscall_name for session $sess_name on channel $channel_name"
 137         fi
 138 }
 139
 140 function lttng_enable_kernel_syscall_ok()
 141 {
 142         lttng_enable_kernel_syscall 0 ${*}
 143 }
 144
 145 function lttng_enable_kernel_syscall_fail()
 146 {
 147         lttng_enable_kernel_syscall 1 ${*}
 148 }
 149
 150 function lttng_disable_kernel_syscall()
 151 {
 152         local expected_to_fail=$1
 153         local sess_name=$2
 154         local syscall_name=$3
 155         local channel_name=$4
 156
 157         if [ -z $syscall_name ]; then
 158                 # Enable all event if no syscall name specified
 159                 syscall_name="-a"
 160         fi
 161
 162         if [ -z $channel_name ]; then
 163                 # default channel if none specified
 164                 chan=""
 165         else
 166                 chan="-c $channel_name"
 167         fi
 168
 169         $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event --syscall "$syscall_name" $chan -s $sess_name -k 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 170
 171         ret=$?
 172         if [[ $expected_to_fail -eq "1" ]]; then
 173                 test $ret -ne "0"
 174                 ok $? "Disable kernel syscall $syscall_name for session $sess_name on channel $channel_name fail as expected"
 175         else
 176                 ok $ret "Disable kernel syscall $syscall_name for session $sess_name on channel $channel_name"
 177         fi
 178 }
 179
 180 function lttng_disable_kernel_syscall_ok()
 181 {
 182         lttng_disable_kernel_syscall 0 ${*}
 183 }
 184
 185 function lttng_disable_kernel_syscall_fail()
 186 {
 187         lttng_disable_kernel_syscall 1 ${*}
 188 }
 189
 190 function lttng_enable_kernel_channel()
 191 {
 192         local expected_to_fail=$1
 193         local sess_name=$2
 194         local channel_name=$3
 195
 196         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -k $channel_name -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 197         ret=$?
 198         if [[ $expected_to_fail -eq "1" ]]; then
 199                 test "$ret" -ne "0"
 200                 ok $? "Expected failure on kernel channel creation $channel_name in $sess_name"
 201         else
 202                 ok $ret "Enable channel $channel_name for session $sess_name"
 203         fi
 204 }
 205
 206 function lttng_enable_kernel_channel_ok()
 207 {
 208         lttng_enable_kernel_channel 0 ${*}
 209 }
 210
 211 function lttng_enable_kernel_channel_fail()
 212 {
 213         lttng_enable_kernel_channel 1 ${*}
 214 }
 215
 216 function lttng_disable_kernel_channel()
 217 {
 218         local expected_to_fail=$1
 219         local sess_name=$2
 220         local channel_name=$3
 221
 222         $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-channel -k $channel_name -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 223         ret=$?
 224         if [[ $expected_to_fail -eq "1" ]]; then
 225                 test "$ret" -ne "0"
 226                 ok $? "Expected failure on kernel channel creation $channel_name in $sess_name"
 227         else
 228                 ok $ret "disable channel $channel_name for session $sess_name"
 229         fi
 230 }
 231
 232 function lttng_disable_kernel_channel_ok()
 233 {
 234         lttng_disable_kernel_channel 0 ${*}
 235 }
 236
 237 function lttng_disable_kernel_channel_fail()
 238 {
 239         lttng_disable_kernel_channel 1 ${*}
 240 }
 241
 242 function start_lttng_relayd
 243 {
 244         local opt=$1
 245
 246         DIR=$(readlink -f $TESTDIR)
 247
 248         if [ -z $(pidof lt-$RELAYD_BIN) ]; then
 249                 $DIR/../src/bin/lttng-relayd/$RELAYD_BIN -b $opt 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 250                 #$DIR/../src/bin/lttng-relayd/$RELAYD_BIN $opt -vvv >>/tmp/relayd.log 2>&1 &
 251                 if [ $? -eq 1 ]; then
 252                         fail "Start lttng-relayd (opt: $opt)"
 253                         return 1
 254                 else
 255                         pass "Start lttng-relayd (opt: $opt)"
 256                 fi
 257         else
 258                 pass "Start lttng-relayd (opt: $opt)"
 259         fi
 260 }
 261
 262 function stop_lttng_relayd_nocheck
 263 {
 264         PID_RELAYD=`pidof lt-$RELAYD_BIN`
 265
 266         diag "Killing lttng-relayd (pid: $PID_RELAYD)"
 267         kill $PID_RELAYD 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 268         retval=$?
 269
 270         if [ $retval -eq 1 ]; then
 271                 out=1
 272                 while [ -n "$out" ]; do
 273                         out=$(pidof lt-$RELAYD_BIN)
 274                         sleep 0.5
 275                 done
 276         fi
 277         return $retval
 278 }
 279
 280 function stop_lttng_relayd
 281 {
 282         stop_lttng_relayd_nocheck
 283
 284         if [ $? -eq 1 ]; then
 285                 fail "Killed lttng-relayd (pid: $PID_RELAYD)"
 286                 return 1
 287         else
 288                 pass "Killed lttng-relayd (pid: $PID_RELAYD)"
 289                 return 0
 290         fi
 291 }
 292
 293 #First argument: load path for automatic loading
 294 function start_lttng_sessiond()
 295 {
 296
 297         local load_path="$1"
 298         if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
 299                 # Env variable requested no session daemon
 300                 return
 301         fi
 302
 303         validate_kernel_version
 304         if [ $? -ne 0 ]; then
 305             fail "Start session daemon"
 306             BAIL_OUT "*** Kernel too old for session daemon tests ***"
 307         fi
 308
 309         DIR=$(readlink -f $TESTDIR)
 310         : ${LTTNG_SESSION_CONFIG_XSD_PATH=${DIR}/../src/common/config/}
 311         export LTTNG_SESSION_CONFIG_XSD_PATH
 312
 313         if [ -z $(pidof lt-$SESSIOND_BIN) ]; then
 314                 # Have a load path ?
 315                 if [ -n "$1" ]; then
 316                         $DIR/../src/bin/lttng-sessiond/$SESSIOND_BIN --load "$1" --background --consumerd32-path="$DIR/../src/bin/lttng-consumerd/lttng-consumerd" --consumerd64-path="$DIR/../src/bin/lttng-consumerd/lttng-consumerd"
 317                 else
 318                         $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"
 319                 fi
 320                 #$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
 321                 status=$?
 322                 ok $status "Start session daemon"
 323         fi
 324 }
 325
 326 function stop_lttng_sessiond ()
 327 {
 328         if [ -n $TEST_NO_SESSIOND ] && [ "$TEST_NO_SESSIOND" == "1" ]; then
 329                 # Env variable requested no session daemon
 330                 return
 331         fi
 332
 333         PID_SESSIOND=`pidof lt-$SESSIOND_BIN`
 334
 335         kill $PID_SESSIOND 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 336
 337         if [ $? -eq 1 ]; then
 338                 fail "Kill sessions daemon"
 339                 return 1
 340         else
 341                 out=1
 342                 while [ -n "$out" ]; do
 343                         out=$(pidof lt-$SESSIOND_BIN)
 344                         sleep 0.5
 345                 done
 346                 out=1
 347                 while [ -n "$out" ]; do
 348                         out=$(pidof $CONSUMERD_BIN)
 349                         sleep 0.5
 350                 done
 351                 pass "Kill session daemon"
 352         fi
 353 }
 354
 355 function list_lttng_with_opts ()
 356 {
 357         local opts=$1
 358         $TESTDIR/../src/bin/lttng/$LTTNG_BIN list $opts 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 359         ok $? "Lttng-tool list command with option $opts"
 360 }
 361
 362 function create_lttng_session_no_output ()
 363 {
 364         local sess_name=$1
 365
 366         $TESTDIR/../src/bin/lttng/$LTTNG_BIN create $sess_name --no-output 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 367         ok $? "Create session $sess_name in no-output mode"
 368 }
 369
 370 function create_lttng_session ()
 371 {
 372         local sess_name=$1
 373         local trace_path=$2
 374         local expected_to_fail=$3
 375
 376         $TESTDIR/../src/bin/lttng/$LTTNG_BIN create $sess_name -o $trace_path > $OUTPUT_DEST
 377         ret=$?
 378         if [[ $expected_to_fail ]]; then
 379                 test "$ret" -ne "0"
 380                 ok $? "Expected fail on session creation $sess_name in $trace_path"
 381         else
 382                 ok $ret "Create session $sess_name in $trace_path"
 383         fi
 384 }
 385
 386 function enable_ust_lttng_channel()
 387 {
 388         local sess_name=$1
 389         local channel_name=$2
 390         local expect_fail=$3
 391
 392         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -u $channel_name -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 393         ret=$?
 394         if [[ $expect_fail ]]; then
 395                 test "$ret" -ne "0"
 396                 ok $? "Expected fail on ust channel creation $channel_name in $sess_name"
 397         else
 398                 ok $ret "Enable channel $channel_name for session $sess_name"
 399         fi
 400 }
 401
 402 function disable_ust_lttng_channel()
 403 {
 404         local sess_name=$1
 405         local channel_name=$2
 406
 407         $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-channel -u $channel_name -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 408         ok $? "Disable channel $channel_name for session $sess_name"
 409 }
 410
 411 function enable_lttng_mmap_overwrite_kernel_channel()
 412 {
 413         local sess_name=$1
 414         local channel_name=$2
 415
 416         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -s $sess_name $channel_name -k --output mmap --overwrite 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 417         ok $? "Enable channel $channel_name for session $sess_name"
 418 }
 419
 420 function enable_lttng_mmap_overwrite_ust_channel()
 421 {
 422         local sess_name=$1
 423         local channel_name=$2
 424
 425         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-channel -s $sess_name $channel_name -u --output mmap --overwrite 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 426         ok $? "Enable channel $channel_name for session $sess_name"
 427 }
 428
 429 function enable_ust_lttng_event ()
 430 {
 431         local sess_name=$1
 432         local event_name="$2"
 433         local channel_name=$3
 434         local expected_to_fail=$4
 435
 436         if [ -z $channel_name ]; then
 437                 # default channel if none specified
 438                 chan=""
 439         else
 440                 chan="-c $channel_name"
 441         fi
 442
 443         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -u 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 444         ret=$?
 445         if [[ $expected_to_fail ]]; then
 446                 test $ret -ne "0"
 447                 ok $? "Enable ust event $event_name for session $session_name on channel $channel_name failed as expected"
 448         else
 449                 ok $ret "Enable event $event_name for session $sess_name"
 450         fi
 451 }
 452
 453 function enable_jul_lttng_event()
 454 {
 455         sess_name=$1
 456         event_name="$2"
 457         channel_name=$3
 458
 459         if [ -z $channel_name ]; then
 460                 # default channel if none specified
 461                 chan=""
 462         else
 463                 chan="-c $channel_name"
 464         fi
 465
 466         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -j 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 467         ok $? "Enable JUL event $event_name for session $sess_name"
 468 }
 469
 470 function enable_jul_lttng_event_loglevel()
 471 {
 472         local sess_name=$1
 473         local event_name="$2"
 474         local loglevel=$3
 475         local channel_name=$4
 476
 477         if [ -z $channel_name ]; then
 478                 # default channel if none specified
 479                 chan=""
 480         else
 481                 chan="-c $channel_name"
 482         fi
 483
 484         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event --loglevel $loglevel "$event_name" $chan -s $sess_name -j 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 485         ok $? "Enable JUL event $event_name for session $sess_name with loglevel $loglevel"
 486 }
 487
 488 function enable_log4j_lttng_event()
 489 {
 490         sess_name=$1
 491         event_name="$2"
 492         channel_name=$3
 493
 494         if [ -z $channel_name ]; then
 495                 # default channel if none specified
 496                 chan=""
 497         else
 498                 chan="-c $channel_name"
 499         fi
 500
 501         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" $chan -s $sess_name -l 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 502         ok $? "Enable LOG4J event $event_name for session $sess_name"
 503 }
 504
 505 function enable_log4j_lttng_event_loglevel()
 506 {
 507         local sess_name=$1
 508         local event_name="$2"
 509         local loglevel=$3
 510         local channel_name=$4
 511
 512         if [ -z $channel_name ]; then
 513                 # default channel if none specified
 514                 chan=""
 515         else
 516                 chan="-c $channel_name"
 517         fi
 518
 519         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event --loglevel $loglevel "$event_name" $chan -s $sess_name -l 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 520         ok $? "Enable LOG4J event $event_name for session $sess_name with loglevel $loglevel"
 521 }
 522
 523 function enable_ust_lttng_event_filter()
 524 {
 525         local sess_name="$1"
 526         local event_name="$2"
 527         local filter="$3"
 528
 529         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u --filter "$filter" 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 530         ok $? "Enable event $event_name with filtering for session $sess_name"
 531 }
 532
 533 function enable_ust_lttng_event_loglevel()
 534 {
 535         local sess_name="$1"
 536         local event_name="$2"
 537         local loglevel="$3"
 538
 539         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u --loglevel $loglevel 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 540         ok $? "Enable event $event_name with loglevel $loglevel"
 541 }
 542
 543 function enable_ust_lttng_event_loglevel_only()
 544 {
 545         local sess_name="$1"
 546         local event_name="$2"
 547         local loglevel="$3"
 548
 549         $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-event "$event_name" -s $sess_name -u --loglevel-only $loglevel 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 550         ok $? "Enable event $event_name with loglevel-only $loglevel"
 551 }
 552
 553 function disable_ust_lttng_event ()
 554 {
 555         local sess_name="$1"
 556         local event_name="$2"
 557         local channel_name="$3"
 558
 559         if [ -z $channel_name ]; then
 560                 # default channel if none specified
 561                 chan=""
 562         else
 563                 chan="-c $channel_name"
 564         fi
 565
 566         $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name $chan -u 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 567         ok $? "Disable event $event_name for session $sess_name"
 568 }
 569
 570 function disable_jul_lttng_event ()
 571 {
 572         local sess_name="$1"
 573         local event_name="$2"
 574
 575         $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name -j >/dev/null 2>&1
 576         ok $? "Disable JUL event $event_name for session $sess_name"
 577 }
 578
 579 function disable_log4j_lttng_event ()
 580 {
 581         local sess_name="$1"
 582         local event_name="$2"
 583
 584         $TESTDIR/../src/bin/lttng/$LTTNG_BIN disable-event "$event_name" -s $sess_name -l >/dev/null 2>&1
 585         ok $? "Disable LOG4J event $event_name for session $sess_name"
 586 }
 587
 588 function start_lttng_tracing ()
 589 {
 590         local sess_name=$1
 591         local expected_to_fail=$2
 592
 593         $TESTDIR/../src/bin/lttng/$LTTNG_BIN start $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 594         ret=$?
 595         if [[ $expected_to_fail ]]; then
 596                 test "$ret" -ne "0"
 597                 ok $? "Expected fail on start tracing for session: $sess_name"
 598         else
 599                 ok $ret "Start tracing for session $sess_name"
 600         fi
 601 }
 602
 603 function stop_lttng_tracing ()
 604 {
 605         local sess_name=$1
 606         local expected_to_fail=$2
 607
 608         $TESTDIR/../src/bin/lttng/$LTTNG_BIN stop $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 609         ret=$?
 610         if [[ $expected_to_fail ]]; then
 611                 test "$ret" -ne "0"
 612                 ok $? "Expected fail on stop tracing for session: $sess_name"
 613         else
 614                 ok $ret "Stop lttng tracing for session $sess_name"
 615         fi
 616 }
 617
 618 function destroy_lttng_session ()
 619 {
 620         local sess_name=$1
 621         local expected_to_fail=$2
 622
 623         $TESTDIR/../src/bin/lttng/$LTTNG_BIN destroy $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 624         ret=$?
 625         if [[ $expected_to_fail ]]; then
 626                 test "$ret" -ne "0"
 627                 ok $? "Expected fail on session deletion $sess_name"
 628         else
 629                 ok $ret "Destroy session $sess_name"
 630         fi
 631 }
 632
 633 function destroy_lttng_sessions ()
 634 {
 635         $TESTDIR/../src/bin/lttng/$LTTNG_BIN destroy --all 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 636         ok $? "Destroy all lttng sessions"
 637 }
 638
 639 function lttng_snapshot_add_output ()
 640 {
 641         local sess_name=$1
 642         local trace_path=$2
 643         local expected_to_fail=$3
 644
 645         $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot add-output -s $sess_name file://$trace_path 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 646         ret=$?
 647         if [[ $expected_to_fail ]]; then
 648                 test "$ret" -ne "0"
 649                 ok $? "Failed to add a  snapshot output file://$trace_path as expected"
 650         else
 651                 ok $ret "Added snapshot output file://$trace_path"
 652         fi
 653 }
 654
 655 function lttng_snapshot_del_output ()
 656 {
 657         local sess_name=$1
 658         local id=$2
 659         local expected_to_fail=$3
 660
 661         $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot del-output -s $sess_name $id 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 662         ret=$?
 663         if [[ $expected_to_fail ]]; then
 664                 test "$ret" -ne "0"
 665                 ok $? "Expect fail on deletion of snapshot output id $id"
 666         else
 667                 ok $ret "Deleted snapshot output id $id"
 668         fi
 669 }
 670
 671 function lttng_snapshot_record ()
 672 {
 673         local sess_name=$1
 674         local trace_path=$2
 675
 676         $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot record -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 677         ok $? "Snapshot recorded"
 678 }
 679
 680 function lttng_snapshot_list ()
 681 {
 682         local sess_name=$1
 683         $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot list-output -s $sess_name 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 684         ok $? "Snapshot list"
 685 }
 686
 687 function lttng_save()
 688 {
 689         local sess_name=$1
 690         local opts=$2
 691
 692         $TESTDIR/../src/bin/lttng/$LTTNG_BIN save $sess_name $opts 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 693         ok $? "Session successfully saved"
 694 }
 695
 696 function lttng_load()
 697 {
 698         local opts=$1
 699
 700         $TESTDIR/../src/bin/lttng/$LTTNG_BIN load $opts 1> $OUTPUT_DEST 2> $ERROR_OUTPUT_DEST
 701         ok $? "Load command successful"
 702 }
 703
 704 function trace_matches ()
 705 {
 706         local event_name=$1
 707         local nr_iter=$2
 708         local trace_path=$3
 709
 710         which $BABELTRACE_BIN >/dev/null
 711         skip $? -ne 0 "Babeltrace binary not found. Skipping trace matches"
 712
 713         local count=$($BABELTRACE_BIN $trace_path | grep $event_name | wc -l)
 714
 715         if [ "$count" -ne "$nr_iter" ]; then
 716                 fail "Trace match"
 717                 diag "$count events found in trace"
 718         else
 719                 pass "Trace match"
 720         fi
 721 }
 722
 723 function trace_match_only()
 724 {
 725         local event_name=$1
 726         local nr_iter=$2
 727         local trace_path=$3
 728
 729         which $BABELTRACE_BIN >/dev/null
 730         skip $? -ne 0 "Babeltrace binary not found. Skipping trace matches"
 731
 732         local count=$($BABELTRACE_BIN $trace_path | grep $event_name | wc -l)
 733         local total=$($BABELTRACE_BIN $trace_path | wc -l)
 734
 735     if [ "$nr_iter" -eq "$count" ] && [ "$total" -eq "$nr_iter" ]; then
 736         pass "Trace match with $total event $event_name"
 737     else
 738         fail "Trace match"
 739         diag "$total event(s) found, expecting $nr_iter of event $event_name and only found $count"
 740     fi
 741 }
 742
 743 function validate_trace
 744 {
 745         local event_name=$1
 746         local trace_path=$2
 747
 748         which $BABELTRACE_BIN >/dev/null
 749         if [ $? -ne 0 ]; then
 750             skip 0 "Babeltrace binary not found. Skipping trace validation"
 751         fi
 752
 753         OLDIFS=$IFS
 754         IFS=","
 755         for i in $event_name; do
 756                 traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | grep $i | wc -l)
 757                 if [ "$traced" -ne 0 ]; then
 758                         pass "Validate trace for event $i, $traced events"
 759                 else
 760                         fail "Validate trace for event $i"
 761                         diag "Found $traced occurences of $i"
 762                 fi
 763         done
 764         ret=$?
 765         IFS=$OLDIFS
 766         return $ret
 767 }
 768
 769 function validate_trace_exp()
 770 {
 771         local event_exp=$1
 772         local trace_path=$2
 773
 774         which $BABELTRACE_BIN >/dev/null
 775         skip $? -ne 0 "Babeltrace binary not found. Skipping trace validation"
 776
 777         traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | grep ${event_exp} | wc -l)
 778         if [ "$traced" -ne 0 ]; then
 779                 pass "Validate trace for expression '${event_exp}', $traced events"
 780         else
 781                 fail "Validate trace for expression '${event_exp}'"
 782                 diag "Found $traced occurences of '${event_exp}'"
 783         fi
 784         ret=$?
 785         return $ret
 786 }
 787
 788 function validate_trace_only_exp()
 789 {
 790         local event_exp=$1
 791         local trace_path=$2
 792
 793         which $BABELTRACE_BIN >/dev/null
 794         skip $? -ne 0 "Babeltrace binary not found. Skipping trace matches"
 795
 796         local count=$($BABELTRACE_BIN $trace_path | grep ${event_exp} | wc -l)
 797         local total=$($BABELTRACE_BIN $trace_path | wc -l)
 798
 799         if [ "$count" -ne 0 ] && [ "$total" -eq "$count" ]; then
 800                 pass "Trace match with $total for expression '${event_exp}"
 801         else
 802                 fail "Trace match"
 803                 diag "$total syscall event(s) found, only syscalls matching expression '${event_exp}' ($count occurrences) are expected"
 804         fi
 805         ret=$?
 806         return $ret
 807 }
 808
 809 function validate_trace_empty()
 810 {
 811         local trace_path=$1
 812
 813         which $BABELTRACE_BIN >/dev/null
 814         if [ $? -ne 0 ]; then
 815             skip 0 "Babeltrace binary not found. Skipping trace validation"
 816         fi
 817
 818         traced=$($BABELTRACE_BIN $trace_path 2>/dev/null | wc -l)
 819         if [ "$traced" -eq 0 ]; then
 820                 pass "Validate empty trace"
 821         else
 822                 fail "Validate empty trace"
 823                 diag "Found $traced events in trace"
 824         fi
 825         ret=$?
 826         return $ret
 827 }