Fix: clear the fixme in high_throughput_limits
[lttng-tools.git] / tests / tools / streaming / 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
18TEST_DESC="Streaming - High throughput with bandwith limits"
19
20CURDIR=$(dirname $0)/
21TESTDIR=$CURDIR/../..
22NR_APP_ITER=10
23NR_ITER=1000000
24BIN_NAME="gen-ust-events"
25SESSION_NAME="high-throughput"
26EVENT_NAME="tp:tptest"
27SESSIOND_CTRL_PORT=5342
28SESSIOND_DATA_PORT=5343
29DEFAULT_IF="lo"
30
31TRACE_PATH=$(mktemp -d)
32
33source $TESTDIR/utils.sh
34
35print_test_banner "$TEST_DESC"
36
37if [ ! -x "$CURDIR/$BIN_NAME" ]; then
38 echo -e "No UST nevents binary detected. Passing."
39 exit 0
40fi
41
42if [ "$(id -u)" != "0" ]; then
43 echo "This test must be running as root to set bandwith limits. Aborting"
44 # Exit status 0 so the tests can continue
45 exit 0
46fi
47
48function set_bw_limit
49{
50 limit=$1
51 echo -n "Setting bandwith limits to ${limit}kbits... "
52 tc qdisc add dev $DEFAULT_IF root handle 1: htb default 15 >/dev/null 2>&1
53 tc class add dev $DEFAULT_IF parent 1: classid 1:1 htb rate ${limit}kbit ceil ${limit}kbit >/dev/null 2>&1
54
55 # FIXME: Timeout when setting limits on ctrl port.
bfdaf0fd 56 tc filter add dev $DEFAULT_IF parent 1: protocol ip u32 match ip dport $SESSIOND_CTRL_PORT 0xffff flowid 1:1 >/dev/null 2>&1
17fe0490
CB
57
58 tc filter add dev $DEFAULT_IF parent 1: protocol ip u32 match ip dport $SESSIOND_DATA_PORT 0xffff flowid 1:1 >/dev/null 2>&1
59 print_ok
60}
61
62function reset_bw_limit
63{
64 echo -n "Resetting bandwith limits... "
65 tc qdisc del dev $DEFAULT_IF root >/dev/null 2>&1
66 print_ok
67}
68
69function create_lttng_session_with_uri
70{
71 sess_name=$1
72 uri=$2
73 # Create session with custom URI
74 $TESTDIR/../src/bin/lttng/$LTTNG_BIN create -U $uri $sess_name >/dev/null 2>&1
75}
76
77function enable_lttng_consumer
78{
79 uri=$1
80 # Create session with custom URI
81 $TESTDIR/../src/bin/lttng/$LTTNG_BIN enable-consumer -u $uri >/dev/null 2>&1
82}
83
84function run_apps
85{
86 for i in `seq 1 $NR_APP_ITER`; do
bfdaf0fd 87 LTTNG_UST_REGISTER_TIMEOUT=-1 ./$CURDIR/$BIN_NAME $NR_ITER & >/dev/null 2>&1
17fe0490
CB
88 done
89}
90
91function wait_apps
92{
93 echo "Waiting for applications to end"
94 while [ -n "$(pidof $BIN_NAME)" ]; do
95 echo -n "."
96 sleep 1
97 done
98 echo ""
99}
100
101function test_high_throughput
102{
103 NETWORK_URI="net://localhost"
104 create_lttng_session_with_uri $SESSION_NAME $NETWORK_URI
105 enable_lttng_consumer $NETWORK_URI
106 enable_ust_lttng_event $SESSION_NAME $EVENT_NAME
107 start_lttng_tracing $SESSION_NAME
108 run_apps
109 wait_apps
110
111 # FIXME: Should not be necessary...
112 #echo "Sleeping..."
113 #sleep 180
114
115 stop_lttng_tracing $SESSION_NAME
116 destroy_lttng_session $SESSION_NAME
117 validate_event_count
118}
119
120function validate_event_count
121{
122
123 TEMP_FILE=$(mktemp)
124 TEMP_FILE_2=$(mktemp)
125
126 traced=$(babeltrace $TRACE_PATH 2>/dev/null | wc -l)
127 babeltrace $TRACE_PATH >/dev/null 2>$TEMP_FILE_2
128
129 cat $TEMP_FILE_2 | cut -f4 -d " " >$TEMP_FILE
130
131 dropped=0
132 while read line;
133 do
134 let dropped=$dropped+$line
135 done < $TEMP_FILE
136
137 let total=$dropped+$traced
138 let wanted=$NR_APP_ITER*$NR_ITER
139
140 rm -rf $TRACE_PATH
141 rm $TEMP_FILE $TEMP_FILE_2
142
143 if [ $wanted -ne $total ]; then
144 echo -n "Expected $wanted. Dropped $dropped. Recorded $traced. Total $total... "
145 print_fail
146 return 1
147 else
148 echo -n "Expected $wanted. Dropped $dropped. Recorded $traced. Total $total... "
149 print_ok
150 return 0
151 fi
152}
153
154function interrupt_cleanup()
155{
156 echo -en "\n*** Exiting ***\n"
157 stop_lttng_relayd
158 stop_lttng_sessiond
159 reset_bw_limit
160 exit 1
161}
162
163# Catch sigint and try to cleanup limits
164trap interrupt_cleanup SIGINT
165
166BW_LIMITS=(3200 1600 800 400 200 100 50 25)
167for BW in ${BW_LIMITS[@]};
168do
169 echo ""
170 echo -e "=== Testing high-throughput with bandwith limit set to ${BW}kbits"
171 set_bw_limit $BW
172
173 start_lttng_sessiond
174 start_lttng_relayd "-o $TRACE_PATH"
175 test_high_throughput
176 result=$?
177 stop_lttng_relayd
178 stop_lttng_sessiond
179 reset_bw_limit
180
181 if [ $result -ne 0 ]; then
182 exit 1
183 fi
184done
This page took 0.041541 seconds and 4 git commands to generate.