tests/regression/tools/snapshots/test_kernel_streaming

   1 #!/bin/bash
   2 #
   3 # Copyright (C) - 2013 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 TEST_DESC="Streaming - Snapshot Kernel tracing"
  18
  19 CURDIR=$(dirname $0)/
  20 TESTDIR=$CURDIR/../../..
  21 EVENT_NAME="sched_switch"
  22 PID_RELAYD=0
  23 SESSION_NAME=""
  24 CHANNEL_NAME="chan1"
  25
  26 TRACE_PATH=$(mktemp -d)
  27
  28 NUM_TESTS=36
  29
  30 source $TESTDIR/utils/utils.sh
  31
  32 # LTTng kernel modules check
  33 out=`ls /lib/modules/$(uname -r)/extra | grep lttng`
  34 if [ -z "$out" ]; then
  35         BAIL_OUT "LTTng modules not detected."
  36 fi
  37
  38 function snapshot_add_output ()
  39 {
  40         local sess_name=$1
  41         local trace_path=$2
  42         local name=$3
  43         local extra_opt=""
  44
  45         if [ ! -z $name ]; then
  46                 extra_opt="-n $name"
  47         fi
  48
  49         $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot add-output -s $sess_name $extra_opt $trace_path >/dev/null 2>&1
  50         ok $? "Added snapshot output $trace_path"
  51 }
  52
  53 # Test a snapshot using a default name for the output destination.
  54 function test_kernel_default_name_with_del()
  55 {
  56         diag "Test kernel snapshot streaming with default name with delete output"
  57         create_lttng_session_no_output $SESSION_NAME
  58         enable_lttng_mmap_overwrite_kernel_channel $SESSION_NAME $CHANNEL_NAME
  59         lttng_enable_kernel_event $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
  60         start_lttng_tracing_ok $SESSION_NAME
  61         snapshot_add_output $SESSION_NAME "net://localhost"
  62         lttng_snapshot_record $SESSION_NAME
  63
  64         # Validate test
  65         validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1*
  66         if [ $? -ne 0 ]; then
  67                 return $?
  68         fi
  69
  70         lttng_snapshot_del_output_ok $SESSION_NAME 1
  71         snapshot_add_output $SESSION_NAME "net://localhost"
  72         lttng_snapshot_record $SESSION_NAME
  73
  74         # Validate test with the next ID since a del output was done prior.
  75         validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-2*
  76         if [ $? -ne 0 ]; then
  77                 return $?
  78         fi
  79
  80         stop_lttng_tracing_ok $SESSION_NAME
  81         destroy_lttng_session_ok $SESSION_NAME
  82
  83         return 0
  84 }
  85
  86 # Test a snapshot using a default name for the output destination.
  87 function test_kernel_default_name()
  88 {
  89         diag "Test kernel snapshot streaming with default name"
  90         create_lttng_session_no_output $SESSION_NAME
  91         enable_lttng_mmap_overwrite_kernel_channel $SESSION_NAME $CHANNEL_NAME
  92         lttng_enable_kernel_event $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
  93         start_lttng_tracing_ok $SESSION_NAME
  94         snapshot_add_output $SESSION_NAME "net://localhost"
  95         lttng_snapshot_record $SESSION_NAME
  96         stop_lttng_tracing_ok $SESSION_NAME
  97         destroy_lttng_session_ok $SESSION_NAME
  98         # Validate test
  99         validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1*
 100         out=$?
 101
 102         return $out
 103 }
 104
 105 # Test a snapshot using a custom name for the output destination.
 106 function test_kernel_custom_name()
 107 {
 108         local out
 109         local name="asnapshotname"
 110
 111         diag "Test kernel snapshot streaming with custom name"
 112         create_lttng_session_no_output $SESSION_NAME
 113         enable_lttng_mmap_overwrite_kernel_channel $SESSION_NAME $CHANNEL_NAME
 114         lttng_enable_kernel_event $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
 115         start_lttng_tracing_ok $SESSION_NAME
 116         snapshot_add_output $SESSION_NAME "net://localhost" $name
 117         lttng_snapshot_record $SESSION_NAME
 118         stop_lttng_tracing_ok $SESSION_NAME
 119         destroy_lttng_session_ok $SESSION_NAME
 120
 121         if ls $TRACE_PATH/$HOSTNAME/$name* &> /dev/null; then
 122                 ok 0 "Custom name snapshot exists"
 123                 # Validate test
 124                 validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/$name-*
 125                 out=$?
 126         else
 127                 fail "No custom name snapshot found"
 128                 out=1
 129         fi
 130
 131         return $out
 132 }
 133
 134 plan_tests $NUM_TESTS
 135
 136 print_test_banner "$TEST_DESC"
 137
 138 if [ "$(id -u)" == "0" ]; then
 139         isroot=1
 140 else
 141         isroot=0
 142 fi
 143
 144 skip $isroot "Root access is needed. Skipping all kernel streaming tests." $NUM_TESTS ||
 145 {
 146         start_lttng_relayd "-o $TRACE_PATH"
 147         start_lttng_sessiond
 148
 149         tests=( test_kernel_default_name test_kernel_custom_name \
 150                         test_kernel_default_name_with_del )
 151
 152         for fct_test in ${tests[@]};
 153         do
 154                 SESSION_NAME=$(randstring 16 0)
 155                 ${fct_test}
 156                 if [ $? -eq 0 ]; then
 157                         # Only delete if successful
 158                         rm -rf $TRACE_PATH
 159                 else
 160                         break
 161                 fi
 162         done
 163
 164         stop_lttng_sessiond
 165         stop_lttng_relayd
 166 }