blob: 4224c5abfbf003eecd140a37a631dfe4a1ed9ccb (
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
#!/bin/sh
##############################################################################
# Copyright (c) 2015 Ericsson AB and others.
#
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################
set -e
# Commandline arguments
DST_IP=$1 # destination IP address
NUM_PORTS=$2 # number of source ports
PKT_SIZE=$3 # packet size
DURATION=$4 # test duration (seconds)
# Configuration
UDP_SRC_MIN=1000 # UDP source port min
UDP_SRC_MAX=$(( UDP_SRC_MIN + NUM_PORTS - 1 )) # UDP source port max
UDP_DST_MIN=1000 # UDP destination port min
UDP_DST_MAX=$(( UDP_DST_MIN + NUM_PORTS )) # UDP destination port max
# helper function to send commands to pktgen
pgset()
{
local result
echo $1 > $PGDEV
result=$(cat $PGDEV | fgrep "Result: OK:")
if [ "$result" = "" ]; then
cat $PGDEV | fgrep "Result:" >/dev/stderr
exit 1
fi
}
# configure pktgen (see pktgen doc for details)
pgconfig()
{
#
# Thread commands
#
PGDEV=/proc/net/pktgen/kpktgend_0
# Remove all devices from this thread
pgset "rem_device_all"
# Add device to thread
pgset "add_device $DEV"
#
# Device commands
#
PGDEV=/proc/net/pktgen/$DEV
# 0 means continious sends untill explicitly stopped
pgset "count 0"
# use single SKB for all transmits
pgset "clone_skb 0"
# packet size, NIC adds 4 bytes CRC
pgset "pkt_size $PKT_SIZE"
# random address within the min-max range
pgset "flag IPDST_RND UDPSRC_RND UDPDST_RND"
# destination IP
pgset "dst_min $DST_IP"
pgset "dst_max $DST_IP"
# destination MAC address
pgset "dst_mac $MAC"
# source UDP port range
pgset "udp_src_min $UDP_SRC_MIN"
pgset "udp_src_max $UDP_SRC_MAX"
# destination UDP port range
pgset "udp_dst_min $UDP_DST_MIN"
pgset "udp_dst_max $UDP_DST_MAX"
}
# run pktgen
pgrun()
{
# Time to run, result can be vieved in /proc/net/pktgen/$DEV
PGDEV=/proc/net/pktgen/pgctrl
# Will hang, Ctrl-C or SIGINT to stop
pgset "start" start
}
# run pktgen for ${DURATION} seconds
run_test()
{
pgrun &
pid=$!
sleep $DURATION
kill -INT $pid
wait; sleep 1
}
# write the result to stdout in json format
output_json()
{
sent=$(awk '/^Result:/{print $5}' <$PGDEV)
pps=$(awk 'match($0,/'\([0-9]+\)pps'/, a) {print a[1]}' <$PGDEV)
errors=$(awk '/errors:/{print $5}' <$PGDEV)
flows=$(( NUM_PORTS * (NUM_PORTS + 1) ))
echo { '"packets_sent"':$sent , '"packets_per_second"':$pps, '"flows"':$flows, '"errors"':$errors }
}
# main entry
main()
{
modprobe pktgen
ping -c 3 $DST_IP >/dev/null
# destination MAC address
MAC=`arp -n | grep -w $DST_IP | awk '{print $3}'`
# outgoing interface
DEV=`arp -n | grep -w $DST_IP | awk '{print $5}'`
# setup the test
pgconfig
# run the test
run_test >/dev/null
PGDEV=/proc/net/pktgen/$DEV
# check result
result=$(cat $PGDEV | fgrep "Result: OK:")
if [ "$result" = "" ]; then
cat $PGDEV | fgrep Result: >/dev/stderr
exit 1
fi
# output result
output_json
}
main
|