blob: a141885da39c20df0fcc5e66da6e2e50bc3c8007 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
#!/bin/bash
# This function has to execute the given arguments on the target machine.
# For local execution, this could look like:
# sudo bash -c "$@"
# Or for remote execution:
# ssh -i ~/.ssh/id_rsa root@remote "$@"
# Make sure that the command is executed by root.
function execute_command {
bash -c "$@"
}
function measure_packets {
TCPREPLAY_SPEED=$1
TCPREPLAY_COUNT=$2
BRO_PID=$(execute_command "bro -i \"${BRO_INTERFACE}\" -C -b Log::default_writer=Log::WRITER_NONE \"${BRO_SCRIPT}\" > ${BRO_DIR}/bro-out.txt 2> ${BRO_DIR}/bro-err.txt & echo \$!")
tcpreplay -i "${BRO_INTERFACE}" -M "${TCPREPLAY_SPEED}" -L "${TCPREPLAY_COUNT}" "${TCPREPLAY_DUMP}" > /dev/null 2> /dev/null
PCPU="100.0"
while [[ $(echo "${PCPU}>50" | bc) -eq 1 ]]
do
sleep 1
PCPU=$(execute_command "ps -q ${BRO_PID} -o pcpu --no-headers")
done
execute_command "kill -SIGINT \"${BRO_PID}\""
execute_command "while kill -0 ${BRO_PID} 2>/dev/null ; do sleep 0.1 ; done"
execute_command "tail -1 ${BRO_DIR}/bro-err.txt" | sed 's/.* \([0-9]\+\) packets received.*/\1/'
}
if [[ $# -ne 3 ]]
then
echo "Executes Bro and tcpreplay and measures the number of packages"
echo "received and handled by Bro."
echo
echo "Usage:"
echo " $0 SCRIPT IFACE DUMP"
echo "Arguments:"
echo " SCRIPT the Bro script to execute"
echo " IFACE the interface for Bro to listen on"
echo " DUMP the network dump to replay"
exit 1
fi
BRO_SCRIPT=$1
BRO_INTERFACE=$2
TCPREPLAY_DUMP=$3
SPEEDS=(100 50 25)
COUNTS=(1000000 2000000 4000000)
if [[ ! -r "${TCPREPLAY_DUMP}" ]]
then
echo "The network dump '${TCPREPLAY_DUMP}' does not exist. Aborting."
exit 1
fi
TCPREPLAY_DUMP=$(realpath "${TCPREPLAY_DUMP}")
BRO_DIR=$(execute_command "mktemp --directory --tmpdir bro.XXX")
echo -ne "sent\t"
for SPEED in ${SPEEDS[@]}
do
echo -ne "${SPEED}\t"
done
echo
for COUNT in ${COUNTS[@]}
do
echo -ne "${COUNT}\t"
for SPEED in ${SPEEDS[@]}
do
COUNT_RECEIVED=$(measure_packets ${SPEED} ${COUNT})
echo -ne "${COUNT_RECEIVED}\t"
done
echo
done
execute_command "rm -rf \"${BRO_DIR}\""
|
