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