Fix: cleanup high_throughput_limits test
[lttng-tools.git] / tests / tools / streaming / high_throughput_limits
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
18
19 TEST_DESC="Streaming - High throughput with bandwidth limits"
20
21 CURDIR=$(dirname $0)/
22 TESTDIR=$CURDIR/../..
23 NR_APP_ITER=10
24 NR_ITER=1000000
25 BIN_NAME="gen-ust-events"
26 SESSION_NAME="high-throughput"
27 EVENT_NAME="tp:tptest"
28 SESSIOND_CTRL_PORT=5342
29 SESSIOND_DATA_PORT=5343
30 DEFAULT_IF="lo"
31
32 TRACE_PATH=$(mktemp -d)
33
34 source $TESTDIR/utils.sh
35
36 print_test_banner "$TEST_DESC"
37
38 if [ ! -x "$CURDIR/$BIN_NAME" ]; then
39 echo -e "No UST nevents binary detected. Passing."
40 exit 0
41 fi
42
43 if [ "$(id -u)" != "0" ]; then
44 echo "This test must be running as root to set bandwidth limits. Aborting"
45 # Exit status 0 so the tests can continue
46 exit 0
47 fi
48
49 function set_bw_limit
50 {
51 limit=$1
52 echo -n "Setting bandwidth limits to ${limit}kbits... "
53 tc qdisc add dev $DEFAULT_IF root handle 1: htb default 15 >/dev/null 2>&1
54 tc class add dev $DEFAULT_IF parent 1: classid 1:1 htb rate ${limit}kbit ceil ${limit}kbit >/dev/null 2>&1
55
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
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
62 function reset_bw_limit
63 {
64 echo -n "Resetting bandwidth limits... "
65 tc qdisc del dev $DEFAULT_IF root >/dev/null 2>&1
66 print_ok
67 }
68
69 function 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
77 function 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
84 function run_apps
85 {
86 for i in `seq 1 $NR_APP_ITER`; do
87 # With bandwidth limitation, unfortunately, application easily timeout
88 # due to very slow communication between the consumer and relayd making
89 # the status reply from the consumer quite slow thus delaying the
90 # registration done message.
91 LTTNG_UST_REGISTER_TIMEOUT=-1 ./$CURDIR/$BIN_NAME $NR_ITER & >/dev/null 2>&1
92 done
93 }
94
95 function wait_apps
96 {
97 echo "Waiting for applications to end"
98 while [ -n "$(pidof $BIN_NAME)" ]; do
99 echo -n "."
100 sleep 1
101 done
102 echo ""
103 }
104
105 function test_high_throughput
106 {
107 NETWORK_URI="net://localhost"
108 create_lttng_session_with_uri $SESSION_NAME $NETWORK_URI
109 enable_lttng_consumer $NETWORK_URI
110 enable_ust_lttng_event $SESSION_NAME $EVENT_NAME
111 start_lttng_tracing $SESSION_NAME
112 run_apps
113 wait_apps
114
115 stop_lttng_tracing $SESSION_NAME
116 destroy_lttng_session $SESSION_NAME
117 validate_event_count
118 }
119
120 function 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 if [ $wanted -ne $total ]; then
141 echo -n "Expected $wanted. Dropped $dropped. Recorded $traced. Total $total... "
142 print_fail
143 return 1
144 else
145 echo -n "Expected $wanted. Dropped $dropped. Recorded $traced. Total $total... "
146 print_ok
147
148 # Cleanup only if everything is ok and passes.
149 rm -rf $TRACE_PATH
150 rm $TEMP_FILE $TEMP_FILE_2
151
152 return 0
153 fi
154 }
155
156 function interrupt_cleanup()
157 {
158 echo -en "\n*** Exiting ***\n"
159 stop_lttng_relayd
160 stop_lttng_sessiond
161 reset_bw_limit
162 exit 1
163 }
164
165 # Catch sigint and try to cleanup limits
166 trap interrupt_cleanup SIGINT
167
168 BW_LIMITS=(3200 1600 800 400 200 100 50 25)
169 for BW in ${BW_LIMITS[@]};
170 do
171 echo ""
172 echo -e "=== Testing high-throughput with bandwidth limit set to ${BW}kbits"
173 set_bw_limit $BW
174
175 start_lttng_sessiond
176 start_lttng_relayd "-o $TRACE_PATH"
177 test_high_throughput
178 result=$?
179 stop_lttng_relayd
180 stop_lttng_sessiond
181 reset_bw_limit
182
183 if [ $result -ne 0 ]; then
184 exit 1
185 fi
186 done
This page took 0.03434 seconds and 4 git commands to generate.