tests/regression/tools/snapshots/test_ust_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 UST tracing"
  18
  19 CURDIR=$(dirname $0)/
  20 TESTDIR=$CURDIR/../../..
  21 EVENT_NAME="tp:tptest"
  22 PID_RELAYD=0
  23 SESSION_NAME=""
  24 CHANNEL_NAME="chan1"
  25 BIN_NAME="gen-nevents"
  26 TESTAPP_PATH="$TESTDIR/utils/testapp"
  27 TESTAPP_NAME="gen-ust-events"
  28 TESTAPP_BIN="$TESTAPP_PATH/$TESTAPP_NAME/$TESTAPP_NAME"
  29 NR_ITER=2000000
  30 NR_USEC_WAIT=100
  31
  32 TRACE_PATH=$(mktemp -d)
  33
  34 NUM_TESTS=39
  35
  36 source $TESTDIR/utils/utils.sh
  37
  38 if [ ! -x "$TESTAPP_BIN" ]; then
  39         BAIL_OUT "No UST events binary detected."
  40 fi
  41
  42 function snapshot_add_output ()
  43 {
  44         local sess_name=$1
  45         local trace_path=$2
  46         local name=$3
  47         local extra_opt=""
  48
  49         if [ ! -z $name ]; then
  50                 extra_opt="-n $name"
  51         fi
  52
  53         $TESTDIR/../src/bin/lttng/$LTTNG_BIN snapshot add-output -s $sess_name $extra_opt $trace_path >/dev/null 2>&1
  54         ok $? "Added snapshot output $trace_path"
  55 }
  56
  57 # Test a snapshot using a default name for the output destination.
  58 function test_ust_default_name_with_del()
  59 {
  60         diag "Test UST snapshot streaming with default name with delete output"
  61         create_lttng_session_no_output $SESSION_NAME
  62         enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME
  63         enable_ust_lttng_event $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
  64         start_lttng_tracing $SESSION_NAME
  65         $TESTAPP_BIN $NR_ITER $NR_USEC_WAIT &
  66         ok $? "Start application to trace"
  67         snapshot_add_output $SESSION_NAME "net://localhost"
  68         lttng_snapshot_record $SESSION_NAME
  69
  70         # Validate test
  71         echo $TRACE_PATH/$HOSTNAME/snapshot-1
  72         validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1*
  73         if [ $? -ne 0 ]; then
  74                 return $?
  75         fi
  76
  77         lttng_snapshot_del_output $SESSION_NAME 1
  78         snapshot_add_output $SESSION_NAME "net://localhost"
  79         lttng_snapshot_record $SESSION_NAME
  80
  81         # Validate test with the next ID since a del output was done prior.
  82         validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-2*
  83         if [ $? -ne 0 ]; then
  84                 return $?
  85         fi
  86
  87         stop_lttng_tracing $SESSION_NAME
  88         destroy_lttng_session $SESSION_NAME
  89
  90         diag "Killing $TESTAPP_NAME"
  91         PID_APP=`pidof $TESTAPP_NAME`
  92         kill $PID_APP >/dev/null 2>&1
  93
  94         return 0
  95 }
  96
  97 # Test a snapshot using a default name for the output destination.
  98 function test_ust_default_name()
  99 {
 100         diag "Test UST snapshot streaming with default name"
 101         create_lttng_session_no_output $SESSION_NAME
 102         enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME
 103         enable_ust_lttng_event $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
 104         start_lttng_tracing $SESSION_NAME
 105         $TESTAPP_BIN $NR_ITER $NR_USEC_WAIT &
 106         ok $? "Start application to trace"
 107         snapshot_add_output $SESSION_NAME "net://localhost"
 108         lttng_snapshot_record $SESSION_NAME
 109         stop_lttng_tracing $SESSION_NAME
 110         destroy_lttng_session $SESSION_NAME
 111         # Validate test
 112         validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1*
 113         out=$?
 114
 115         diag "Killing $TESTAPP_NAME"
 116         PID_APP=`pidof $TESTAPP_NAME`
 117         kill $PID_APP >/dev/null 2>&1
 118
 119         return $out
 120 }
 121
 122 # Test a snapshot using a custom name for the output destination.
 123 function test_ust_custom_name()
 124 {
 125         local out
 126         local name="asnapshotname"
 127
 128         diag "Test UST snapshot streaming with custom name"
 129         create_lttng_session_no_output $SESSION_NAME
 130         enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME
 131         enable_ust_lttng_event $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
 132         start_lttng_tracing $SESSION_NAME
 133         $TESTAPP_BIN $NR_ITER $NR_USEC_WAIT &
 134         ok $? "Start application to trace"
 135         snapshot_add_output $SESSION_NAME "net://localhost" $name
 136         lttng_snapshot_record $SESSION_NAME
 137         stop_lttng_tracing $SESSION_NAME
 138         destroy_lttng_session $SESSION_NAME
 139
 140         if ls $TRACE_PATH/$HOSTNAME/$name* &> /dev/null; then
 141                 ok 0 "Custom name snapshot exists"
 142                 # Validate test
 143                 validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/$name-*
 144                 out=$?
 145         else
 146                 fail "No custom name snapshot found"
 147                 out=1
 148         fi
 149
 150         diag "Killing $TESTAPP_NAME"
 151         PID_APP=`pidof $TESTAPP_NAME`
 152         kill $PID_APP >/dev/null 2>&1
 153
 154         return $out
 155 }
 156
 157 plan_tests $NUM_TESTS
 158
 159 print_test_banner "$TEST_DESC"
 160
 161 if [ "$(id -u)" == "0" ]; then
 162         isroot=1
 163 else
 164         isroot=0
 165 fi
 166
 167 start_lttng_relayd "-o $TRACE_PATH"
 168 start_lttng_sessiond
 169
 170 tests=( test_ust_default_name_with_del test_ust_default_name test_ust_custom_name )
 171
 172 for fct_test in ${tests[@]};
 173 do
 174         SESSION_NAME=$(randstring 16 0)
 175         ${fct_test}
 176         if [ $? -eq 0 ]; then
 177                 # Only delete if successful
 178                 rm -rf $TRACE_PATH
 179         else
 180                 break
 181         fi
 182 done
 183
 184 stop_lttng_sessiond
 185 stop_lttng_relayd