From 508e01d81a96c8ede773e6d0dd332892e9600ea7 Mon Sep 17 00:00:00 2001 From: Jörgen Karlsson Date: Thu, 4 Jun 2015 09:11:46 +0200 Subject: add pktgen scenario and sample Supports measuring network throughput UDP. SLA can be verified for packet lost per million packets (ppm) Change-Id: Ie5972f189bbe58b39a2fae98630b2f117c176ae5 JIRA: YARDSTICK-5 Signed-off-by: Jorgen Karlsson --- yardstick/benchmark/scenarios/networking/pktgen.py | 135 ++++++++++++++++++ .../scenarios/networking/pktgen_benchmark.bash | 158 +++++++++++++++++++++ 2 files changed, 293 insertions(+) create mode 100644 yardstick/benchmark/scenarios/networking/pktgen.py create mode 100644 yardstick/benchmark/scenarios/networking/pktgen_benchmark.bash (limited to 'yardstick/benchmark/scenarios/networking') diff --git a/yardstick/benchmark/scenarios/networking/pktgen.py b/yardstick/benchmark/scenarios/networking/pktgen.py new file mode 100644 index 000000000..8038cad60 --- /dev/null +++ b/yardstick/benchmark/scenarios/networking/pktgen.py @@ -0,0 +1,135 @@ +############################################################################## +# 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 +############################################################################## +import pkg_resources +import logging +import json + +import yardstick.ssh as ssh +from yardstick.benchmark.scenarios import base + +LOG = logging.getLogger(__name__) +LOG.setLevel(logging.DEBUG) + + +class Pktgen(base.Scenario): + """Executes a pktgen benchmark between two hosts""" + __scenario_type__ = "Pktgen" + + TARGET_SCRIPT = 'pktgen_benchmark.bash' + + def __init__(self, context): + self.context = context + self.setup_done = False + + def setup(self): + '''scenario setup''' + self.target_script = pkg_resources.resource_filename( + 'yardstick.benchmark.scenarios.networking', + Pktgen.TARGET_SCRIPT) + user = self.context.get('user', 'ubuntu') + host = self.context.get('host', None) + target = self.context.get('target', None) + key_filename = self.context.get('key_filename', '~/.ssh/id_rsa') + + LOG.debug("user:%s, target:%s", user, target) + self.server = ssh.SSH(user, target, key_filename=key_filename) + self.server.wait(timeout=600) + + LOG.debug("user:%s, host:%s", user, host) + self.client = ssh.SSH(user, host, key_filename=key_filename) + self.client.wait(timeout=600) + + # copy script to host + self.client.run("cat > ~/pktgen.sh", + stdin=open(self.target_script, "rb")) + + self.setup_done = True + + def _iptables_setup(self): + """Setup iptables on server to monitor for received packets""" + cmd = "sudo iptables -F; " \ + "sudo iptables -A INPUT -p udp --dport 1000:%s -j DROP" \ + % (1000 + self.number_of_ports) + LOG.debug("Executing command: %s", cmd) + status, _, stderr = self.server.execute(cmd) + if status: + raise RuntimeError(stderr) + + def _iptables_get_result(self): + """Get packet statistics from server""" + cmd = "sudo iptables -L INPUT -vnx |" \ + "awk '/dpts:1000:%s/ {{printf \"%%s\", $1}}'" \ + % (1000 + self.number_of_ports) + LOG.debug("Executing command: %s", cmd) + status, stdout, stderr = self.server.execute(cmd) + if status: + raise RuntimeError(stderr) + return int(stdout) + + def run(self, args): + """execute the benchmark""" + + if not self.setup_done: + self.setup() + + ipaddr = args.get("ipaddr", '127.0.0.1') + + options = args['options'] + packetsize = options.get("packetsize", 60) + self.number_of_ports = options.get("number_of_ports", 10) + + self._iptables_setup() + + cmd = "sudo bash pktgen.sh %s %s %s" \ + % (ipaddr, self.number_of_ports, packetsize) + LOG.debug("Executing command: %s", cmd) + status, stdout, stderr = self.client.execute(cmd) + + if status: + raise RuntimeError(stderr) + + data = json.loads(stdout) + + data['packets_received'] = self._iptables_get_result() + + if "sla" in args: + sent = data['packets_sent'] + received = data['packets_received'] + ppm = 1000000 * (sent - received) / sent + sla_max_ppm = int(args["sla"]["max_ppm"]) + assert ppm <= sla_max_ppm, "ppm %d > sla_max_ppm %d" \ + % (ppm, sla_max_ppm) + + return data + + +def _test(): + '''internal test function''' + key_filename = pkg_resources.resource_filename('yardstick.resources', + 'files/yardstick_key') + ctx = {'host': '172.16.0.137', + 'target': '172.16.0.138', + 'user': 'ubuntu', + 'key_filename': key_filename + } + + logger = logging.getLogger('yardstick') + logger.setLevel(logging.DEBUG) + + p = Pktgen(ctx) + + options = {'packetsize': 120} + + args = {'options': options, + 'ipaddr': '192.168.111.31'} + result = p.run(args) + print result + +if __name__ == '__main__': + _test() diff --git a/yardstick/benchmark/scenarios/networking/pktgen_benchmark.bash b/yardstick/benchmark/scenarios/networking/pktgen_benchmark.bash new file mode 100644 index 000000000..f5df50f0f --- /dev/null +++ b/yardstick/benchmark/scenarios/networking/pktgen_benchmark.bash @@ -0,0 +1,158 @@ +#!/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 +shift +NUM_PORTS=$1 # number of source ports +shift +PKT_SIZE=$1 # packet size + +# 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 +DURATION=20 # test duration (seconds) + +# 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 + -- cgit 1.2.3-korg