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 APPS_PID=
  32
  33 TRACE_PATH=$(mktemp -d)
  34
  35 NUM_TESTS=49
  36
  37 source $TESTDIR/utils/utils.sh
  38
  39 if [ ! -x "$TESTAPP_BIN" ]; then
  40         BAIL_OUT "No UST events binary detected."
  41 fi
  42
  43 # Start trace application and return once one event has been hit.
  44 function start_test_app()
  45 {
  46         local tmp_file=$(mktemp -u)
  47
  48         # Start application with a temporary file.
  49         $TESTAPP_BIN $NR_ITER $NR_USEC_WAIT $tmp_file &
  50         ret=$?
  51         APPS_PID="${APPS_PID} ${!}"
  52         ok $ret "Start application to trace"
  53
  54         # Wait for the application file to appear indicating that at least one
  55         # tracepoint has been fired.
  56         while [ ! -f "$tmp_file" ]; do
  57                 sleep 0.5
  58         done
  59         diag "Removing test app temporary file $tmp_file"
  60         rm -f $tmp_file
  61 }
  62
  63 function stop_test_apps()
  64 {
  65         diag "Stopping $TESTAPP_NAME"
  66         for p in ${APPS_PID}; do
  67                 kill ${p}
  68                 wait ${p} 2>/dev/null
  69         done
  70         APPS_PID=
  71 }
  72
  73 # Test a snapshot using a default name for the output destination.
  74 function test_ust_default_name_with_del()
  75 {
  76         diag "Test UST snapshot streaming with default name with delete output"
  77         create_lttng_session_no_output $SESSION_NAME
  78         enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME
  79         enable_ust_lttng_event_ok $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
  80         start_lttng_tracing_ok $SESSION_NAME
  81
  82         start_test_app
  83
  84         lttng_snapshot_add_output_ok $SESSION_NAME "net://localhost"
  85         lttng_snapshot_record $SESSION_NAME
  86
  87         # Validate test
  88         echo $TRACE_PATH/$HOSTNAME/snapshot-1
  89         validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1*
  90         if [ $? -ne 0 ]; then
  91                 stop_test_apps
  92                 return $?
  93         fi
  94
  95         lttng_snapshot_del_output_ok $SESSION_NAME 1
  96         lttng_snapshot_add_output_ok $SESSION_NAME "net://localhost"
  97         lttng_snapshot_record $SESSION_NAME
  98
  99         # Validate test with the next ID since a del output was done prior.
 100         validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-2*
 101         if [ $? -ne 0 ]; then
 102                 stop_test_apps
 103                 return $?
 104         fi
 105
 106         stop_lttng_tracing_ok $SESSION_NAME
 107         destroy_lttng_session_ok $SESSION_NAME
 108
 109         stop_test_apps
 110
 111         return 0
 112 }
 113
 114 # Test a snapshot using a default name for the output destination.
 115 function test_ust_default_name()
 116 {
 117         diag "Test UST snapshot streaming with default name"
 118         create_lttng_session_no_output $SESSION_NAME
 119         enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME
 120         enable_ust_lttng_event_ok $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
 121         start_lttng_tracing_ok $SESSION_NAME
 122
 123         start_test_app
 124
 125         lttng_snapshot_add_output_ok $SESSION_NAME "net://localhost"
 126         lttng_snapshot_record $SESSION_NAME
 127         stop_lttng_tracing_ok $SESSION_NAME
 128         destroy_lttng_session_ok $SESSION_NAME
 129         # Validate test
 130         validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1*
 131         out=$?
 132
 133         stop_test_apps
 134
 135         return $out
 136 }
 137
 138 function test_ust_default_name_custom_uri()
 139 {
 140         diag "Test UST snapshot streaming with default name with custom URL"
 141         create_lttng_session_no_output $SESSION_NAME
 142         enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME
 143         enable_ust_lttng_event_ok $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
 144         start_lttng_tracing_ok $SESSION_NAME
 145
 146         start_test_app
 147
 148         lttng_snapshot_add_output_ok $SESSION_NAME "-C tcp://localhost:5342 -D tcp://localhost:5343"
 149         lttng_snapshot_record $SESSION_NAME
 150         stop_lttng_tracing_ok $SESSION_NAME
 151         destroy_lttng_session_ok $SESSION_NAME
 152         # Validate test
 153         validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/snapshot-1*
 154         out=$?
 155
 156         stop_test_apps
 157
 158         return $out
 159 }
 160
 161 # Test a snapshot using a custom name for the output destination.
 162 function test_ust_custom_name()
 163 {
 164         local out
 165         local name="asnapshotname"
 166
 167         diag "Test UST snapshot streaming with custom name"
 168         create_lttng_session_no_output $SESSION_NAME
 169         enable_lttng_mmap_overwrite_ust_channel $SESSION_NAME $CHANNEL_NAME
 170         enable_ust_lttng_event_ok $SESSION_NAME $EVENT_NAME $CHANNEL_NAME
 171         start_lttng_tracing_ok $SESSION_NAME
 172
 173         start_test_app
 174
 175         lttng_snapshot_add_output_ok $SESSION_NAME "net://localhost" "-n $name"
 176         lttng_snapshot_record $SESSION_NAME
 177         stop_lttng_tracing_ok $SESSION_NAME
 178         destroy_lttng_session_ok $SESSION_NAME
 179
 180         if ls $TRACE_PATH/$HOSTNAME/$name* &> /dev/null; then
 181                 ok 0 "Custom name snapshot exists"
 182                 # Validate test
 183                 validate_trace $EVENT_NAME $TRACE_PATH/$HOSTNAME/$name-*
 184                 out=$?
 185         else
 186                 fail "No custom name snapshot found"
 187                 out=1
 188         fi
 189
 190         stop_test_apps
 191
 192         return $out
 193 }
 194
 195 plan_tests $NUM_TESTS
 196
 197 print_test_banner "$TEST_DESC"
 198
 199 if [ "$(id -u)" == "0" ]; then
 200         isroot=1
 201 else
 202         isroot=0
 203 fi
 204
 205 start_lttng_relayd "-o $TRACE_PATH"
 206 start_lttng_sessiond
 207
 208 tests=( test_ust_default_name_with_del test_ust_default_name test_ust_custom_name test_ust_default_name_custom_uri )
 209
 210 for fct_test in ${tests[@]};
 211 do
 212         SESSION_NAME=$(randstring 16 0)
 213         ${fct_test}
 214         if [ $? -eq 0 ]; then
 215                 # Only delete if successful
 216                 rm -rf $TRACE_PATH
 217         else
 218                 break
 219         fi
 220 done
 221
 222 stop_lttng_sessiond
 223 stop_lttng_relayd