Gen-ust-events: use options instead of arguments
[lttng-tools.git] / tests / regression / tools / streaming / test_high_throughput_limits
CommitLineData
17fe0490
CB
1#!/bin/bash
2#
3# Copyright (C) - 2012 Christian Babeux <christian.babeux@efficios.com>
4# David Goulet <dgoulet@efficios.com>
5#
6# This library is free software; you can redistribute it and/or modify it under
7# the terms of the GNU Lesser General Public License as published by the Free
8# Software Foundation; version 2.1 of the License.
9#
10# This library is distributed in the hope that it will be useful, but WITHOUT
11# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
12# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
13# details.
14#
15# You should have received a copy of the GNU Lesser General Public License
16# along with this library; if not, write to the Free Software Foundation, Inc.,
17# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
b2068598
DG
18
19TEST_DESC="Streaming - High throughput with bandwidth limits"
17fe0490
CB
20
21CURDIR=$(dirname $0)/
9ac429ef 22TESTDIR=$CURDIR/../../..
17fe0490
CB
23NR_APP_ITER=10
24NR_ITER=1000000
ea3a3cae
CB
25TESTAPP_PATH="$TESTDIR/utils/testapp"
26TESTAPP_NAME="gen-ust-events"
27TESTAPP_BIN="$TESTAPP_PATH/$TESTAPP_NAME/$TESTAPP_NAME"
17fe0490
CB
28SESSION_NAME="high-throughput"
29EVENT_NAME="tp:tptest"
30SESSIOND_CTRL_PORT=5342
31SESSIOND_DATA_PORT=5343
32DEFAULT_IF="lo"
33
34TRACE_PATH=$(mktemp -d)
35
5dca3876 36NUM_TESTS=104
d346ccad 37
9ac429ef 38source $TESTDIR/utils/utils.sh
17fe0490 39
ea3a3cae
CB
40if [ ! -x "$TESTAPP_BIN" ]; then
41 BAIL_OUT "No UST events binary detected."
17fe0490
CB
42fi
43
44function set_bw_limit
45{
46 limit=$1
ffaed7f7
JD
47 ctrlportlimit=$(($limit/10))
48 # failsafe to have at least 1kbit/s for control (in the case where $1 < 10)
49 [ $ctrlportlimit = 0 ] && ctrlportlimit=1
50 # if $1 < 10, we might bust the limit set here, but the
51 # parent qdisc (1:) will always limit us to the right max value
52 dataportlimit=$((9*${ctrlportlimit}))
53
d346ccad 54
17fe0490 55 tc qdisc add dev $DEFAULT_IF root handle 1: htb default 15 >/dev/null 2>&1
17fe0490 56
ffaed7f7
JD
57 # the total bandwidth is the limit set by the user
58 tc class add dev $DEFAULT_IF parent 1: classid 1:1 htb rate ${limit}kbit ceil ${limit}kbit >/dev/null 2>&1
59 # 1/10 of the bandwidth guaranteed and traffic prioritized for the control port
60 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
61 # 9/10 of the bandwidth guaranteed and can borrow up to the total bandwidth (if unused)
62 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
17fe0490 63
ffaed7f7
JD
64 # filter to assign control traffic to the 1:10 class
65 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
66 # filter to assign data traffic to the 1:11 class
67 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
d346ccad
CB
68
69 ok $? "Set bandwidth limits to ${limit}kbits, ${ctrlportlimit} for control and ${dataportlimit} for data"
17fe0490
CB
70}
71
72function reset_bw_limit
73{
17fe0490 74 tc qdisc del dev $DEFAULT_IF root >/dev/null 2>&1
d346ccad 75 ok $? "Reset bandwith limits"
17fe0490
CB
76}
77
78function create_lttng_session_with_uri
79{
80 sess_name=$1
81 uri=$2
82 # Create session with custom URI
83 $TESTDIR/../src/bin/lttng/$LTTNG_BIN create -U $uri $sess_name >/dev/null 2>&1
d346ccad 84 ok $? "Create session with uri $uri"
17fe0490
CB
85}
86
17fe0490
CB
87function run_apps
88{
89 for i in `seq 1 $NR_APP_ITER`; do
b2068598
DG
90 # With bandwidth limitation, unfortunately, application easily timeout
91 # due to very slow communication between the consumer and relayd making
92 # the status reply from the consumer quite slow thus delaying the
93 # registration done message.
a4c30524 94 LTTNG_UST_REGISTER_TIMEOUT=-1 $TESTAPP_BIN -i $NR_ITER & >/dev/null 2>&1
17fe0490
CB
95 done
96}
97
17fe0490
CB
98function test_high_throughput
99{
100 NETWORK_URI="net://localhost"
101 create_lttng_session_with_uri $SESSION_NAME $NETWORK_URI
c4926bb5 102 enable_ust_lttng_event_ok $SESSION_NAME $EVENT_NAME
e563bbdb 103 start_lttng_tracing_ok $SESSION_NAME
17fe0490 104 run_apps
5dca3876
MD
105 diag "Waiting for applications to end"
106 wait
107 pass "waiting done"
96340a01 108 stop_lttng_tracing_ok $SESSION_NAME
67b4c664 109 destroy_lttng_session_ok $SESSION_NAME
17fe0490
CB
110 validate_event_count
111}
112
113function validate_event_count
114{
115
116 TEMP_FILE=$(mktemp)
117 TEMP_FILE_2=$(mktemp)
118
119 traced=$(babeltrace $TRACE_PATH 2>/dev/null | wc -l)
120 babeltrace $TRACE_PATH >/dev/null 2>$TEMP_FILE_2
121
122 cat $TEMP_FILE_2 | cut -f4 -d " " >$TEMP_FILE
123
124 dropped=0
125 while read line;
126 do
127 let dropped=$dropped+$line
128 done < $TEMP_FILE
129
130 let total=$dropped+$traced
131 let wanted=$NR_APP_ITER*$NR_ITER
132
17fe0490 133 if [ $wanted -ne $total ]; then
d346ccad
CB
134 fail "Validate trace event count"
135 diag "Expected $wanted. Dropped $dropped. Recorded $traced. Total $total... "
17fe0490
CB
136 return 1
137 else
d346ccad
CB
138 pass "Validate trace event count"
139 diag "Expected $wanted. Dropped $dropped. Recorded $traced. Total $total... "
b2068598 140
b2068598
DG
141 rm -rf $TRACE_PATH
142 rm $TEMP_FILE $TEMP_FILE_2
143
17fe0490
CB
144 return 0
145 fi
146}
147
148function interrupt_cleanup()
149{
d346ccad 150 diag "*** Exiting ***"
17fe0490
CB
151 stop_lttng_relayd
152 stop_lttng_sessiond
153 reset_bw_limit
154 exit 1
155}
156
d346ccad 157plan_tests $NUM_TESTS
17fe0490 158
e3bef725
CB
159print_test_banner "$TEST_DESC"
160
d346ccad
CB
161if [ "$(id -u)" == "0" ]; then
162 isroot=1
163else
164 isroot=0
165fi
17fe0490 166
d346ccad
CB
167skip $isroot "Root access is needed to set bandwith limits. Skipping all tests." $NUM_TESTS ||
168{
17fe0490 169
d346ccad 170 # Catch sigint and try to cleanup limits
491c49eb 171 trap interrupt_cleanup SIGTERM
d346ccad
CB
172 trap interrupt_cleanup SIGINT
173
174 BW_LIMITS=(3200 1600 800 400 200 100 50 25)
175 for BW in ${BW_LIMITS[@]};
176 do
177 diag "Test high-throughput with bandwidth limit set to ${BW}kbits"
491c49eb 178
d346ccad
CB
179 set_bw_limit $BW
180
181 start_lttng_sessiond
182 start_lttng_relayd "-o $TRACE_PATH"
183 test_high_throughput
184 result=$?
185 stop_lttng_relayd
186 stop_lttng_sessiond
187 reset_bw_limit
188 done
189}
This page took 0.091159 seconds and 5 git commands to generate.