+++ /dev/null
-#!/bin/bash
-#
-# Copyright (C) - 2012 Christian Babeux <christian.babeux@efficios.com>
-# David Goulet <dgoulet@efficios.com>
-#
-# This library is free software; you can redistribute it and/or modify it under
-# the terms of the GNU Lesser General Public License as published by the Free
-# Software Foundation; version 2.1 of the License.
-#
-# This library is distributed in the hope that it will be useful, but WITHOUT
-# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
-# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
-# details.
-#
-# You should have received a copy of the GNU Lesser General Public License
-# along with this library; if not, write to the Free Software Foundation, Inc.,
-# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
-
-TEST_DESC="Streaming - High throughput with bandwidth limits"
-
-CURDIR=$(dirname $0)/
-TESTDIR=$CURDIR/../..
-NR_APP_ITER=10
-NR_ITER=1000000
-BIN_NAME="gen-ust-events"
-SESSION_NAME="high-throughput"
-EVENT_NAME="tp:tptest"
-SESSIOND_CTRL_PORT=5342
-SESSIOND_DATA_PORT=5343
-DEFAULT_IF="lo"
-
-TRACE_PATH=$(mktemp -d)
-
-source $TESTDIR/utils.sh
-
-print_test_banner "$TEST_DESC"
-
-if [ ! -x "$CURDIR/$BIN_NAME" ]; then
- echo -e "No UST nevents binary detected. Passing."
- exit 0
-fi
-
-if [ "$(id -u)" != "0" ]; then
- echo "This test must be running as root to set bandwidth limits. Aborting"
- # Exit status 0 so the tests can continue
- exit 0
-fi
-
-function set_bw_limit
-{
- limit=$1
- ctrlportlimit=$(($limit/10))
- # failsafe to have at least 1kbit/s for control (in the case where $1 < 10)
- [ $ctrlportlimit = 0 ] && ctrlportlimit=1
- # if $1 < 10, we might bust the limit set here, but the
- # parent qdisc (1:) will always limit us to the right max value
- dataportlimit=$((9*${ctrlportlimit}))
-
- echo -n "Setting bandwidth limits to ${limit}kbits, (${ctrlportlimit} for control and ${dataportlimit} for data)... "
- tc qdisc add dev $DEFAULT_IF root handle 1: htb default 15 >/dev/null 2>&1
-
- # the total bandwidth is the limit set by the user
- tc class add dev $DEFAULT_IF parent 1: classid 1:1 htb rate ${limit}kbit ceil ${limit}kbit >/dev/null 2>&1
- # 1/10 of the bandwidth guaranteed and traffic prioritized for the control port
- tc class add dev $DEFAULT_IF parent 1:1 classid 1:10 htb rate ${ctrlportlimit}kbit ceil ${limit}kbit prio 1 >/dev/null 2>&1
- # 9/10 of the bandwidth guaranteed and can borrow up to the total bandwidth (if unused)
- tc class add dev $DEFAULT_IF parent 1:1 classid 1:11 htb rate ${dataportlimit}kbit ceil ${limit}kbit prio 2 >/dev/null 2>&1
-
- # filter to assign control traffic to the 1:10 class
- tc filter add dev $DEFAULT_IF parent 1: protocol ip u32 match ip dport $SESSIOND_CTRL_PORT 0xffff flowid 1:10 >/dev/null 2>&1
- # filter to assign data traffic to the 1:11 class
- tc filter add dev $DEFAULT_IF parent 1: protocol ip u32 match ip dport $SESSIOND_DATA_PORT 0xffff flowid 1:11 >/dev/null 2>&1
- print_ok
-}
-
-function reset_bw_limit
-{
- echo -n "Resetting bandwidth limits... "
- tc qdisc del dev $DEFAULT_IF root >/dev/null 2>&1
- print_ok
-}
-
-function create_lttng_session_with_uri
-{
- sess_name=$1
- uri=$2
- # Create session with custom URI
- $TESTDIR/../src/bin/lttng/$LTTNG_BIN create -U $uri $sess_name >/dev/null 2>&1
-}
-
-function enable_lttng_consumer
-{
- uri=$1
- # Create session with custom URI
- $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-consumer -u $uri >/dev/null 2>&1
-}
-
-function run_apps
-{
- for i in `seq 1 $NR_APP_ITER`; do
- # With bandwidth limitation, unfortunately, application easily timeout
- # due to very slow communication between the consumer and relayd making
- # the status reply from the consumer quite slow thus delaying the
- # registration done message.
- LTTNG_UST_REGISTER_TIMEOUT=-1 ./$CURDIR/$BIN_NAME $NR_ITER & >/dev/null 2>&1
- done
-}
-
-function wait_apps
-{
- echo "Waiting for applications to end"
- while [ -n "$(pidof $BIN_NAME)" ]; do
- echo -n "."
- sleep 1
- done
- echo ""
-}
-
-function test_high_throughput
-{
- NETWORK_URI="net://localhost"
- create_lttng_session_with_uri $SESSION_NAME $NETWORK_URI
- enable_lttng_consumer $NETWORK_URI
- enable_ust_lttng_event $SESSION_NAME $EVENT_NAME
- start_lttng_tracing $SESSION_NAME
- run_apps
- wait_apps
-
- stop_lttng_tracing $SESSION_NAME
- destroy_lttng_session $SESSION_NAME
- validate_event_count
-}
-
-function validate_event_count
-{
-
- TEMP_FILE=$(mktemp)
- TEMP_FILE_2=$(mktemp)
-
- traced=$(babeltrace $TRACE_PATH 2>/dev/null | wc -l)
- babeltrace $TRACE_PATH >/dev/null 2>$TEMP_FILE_2
-
- cat $TEMP_FILE_2 | cut -f4 -d " " >$TEMP_FILE
-
- dropped=0
- while read line;
- do
- let dropped=$dropped+$line
- done < $TEMP_FILE
-
- let total=$dropped+$traced
- let wanted=$NR_APP_ITER*$NR_ITER
-
- if [ $wanted -ne $total ]; then
- echo -n "Expected $wanted. Dropped $dropped. Recorded $traced. Total $total... "
- print_fail
- return 1
- else
- echo -n "Expected $wanted. Dropped $dropped. Recorded $traced. Total $total... "
- print_ok
-
- # Cleanup only if everything is ok and passes.
- rm -rf $TRACE_PATH
- rm $TEMP_FILE $TEMP_FILE_2
-
- return 0
- fi
-}
-
-function interrupt_cleanup()
-{
- echo -en "\n*** Exiting ***\n"
- stop_lttng_relayd
- stop_lttng_sessiond
- reset_bw_limit
- exit 1
-}
-
-# Catch sigint and try to cleanup limits
-trap interrupt_cleanup SIGINT
-
-BW_LIMITS=(3200 1600 800 400 200 100 50 25)
-for BW in ${BW_LIMITS[@]};
-do
- echo ""
- echo -e "=== Testing high-throughput with bandwidth limit set to ${BW}kbits"
- set_bw_limit $BW
-
- start_lttng_sessiond
- start_lttng_relayd "-o $TRACE_PATH"
- test_high_throughput
- result=$?
- stop_lttng_relayd
- stop_lttng_sessiond
- reset_bw_limit
-
- if [ $result -ne 0 ]; then
- exit 1
- fi
-done