[dts] [PATCH V1] Add testsuite vm2vm virtio net perf

[lihong <lihongx.ma@intel.com>]

Signed-off-by: lihong <lihongx.ma@intel.com>
---
 tests/TestSuite_vm2vm_virtio_net_perf.py | 282 +++++++++++++++++++++++++++++++
 1 file changed, 282 insertions(+)
 create mode 100644 tests/TestSuite_vm2vm_virtio_net_perf.py

diff --git a/tests/TestSuite_vm2vm_virtio_net_perf.py b/tests/TestSuite_vm2vm_virtio_net_perf.py
new file mode 100644
index 0000000..dbcf5c4
--- /dev/null
+++ b/tests/TestSuite_vm2vm_virtio_net_perf.py
@@ -0,0 +1,282 @@
+# BSD LICENSE
+#
+# Copyright(c) <2019> Intel Corporation.
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions
+# are met:
+#
+#   * Redistributions of source code must retain the above copyright
+#     notice, this list of conditions and the following disclaimer.
+#   * Redistributions in binary form must reproduce the above copyright
+#     notice, this list of conditions and the following disclaimer in
+#     the documentation and/or other materials provided with the
+#     distribution.
+#   * Neither the name of Intel Corporation nor the names of its
+#     contributors may be used to endorse or promote products derived
+#     from this software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+"""
+DPDK Test suite.
+
+Test cases for vhost-user + virtio-net VM2VM tso/tso
+zero-copy/ufo/tso_ufo capability check.
+"""
+import re
+import time
+import utils
+from virt_common import VM
+from test_case import TestCase
+
+
+class TestVM2VMVirtioNetPerf(TestCase):
+    def set_up_all(self):
+        core_config = "1S/4C/1T"
+        cores_num = len([n for n in self.dut.cores if int(n['socket'])
+                            == 0])
+        self.verify(cores_num >= 4,
+                    "There has not enough cores to test this suite %s" %
+                    self.suite_name)
+        cores = self.dut.get_core_list(core_config)
+        self.coremask = utils.create_mask(cores)
+        self.memory_channel = self.dut.get_memory_channels()
+        self.vm_num = 2
+        self.virtio_ip1 = "1.1.1.2"
+        self.virtio_ip2 = "1.1.1.3"
+        self.vritio_mac1 = "52:54:00:00:00:01"
+        self.vritio_mac2 = "52:54:00:00:00:02"
+        self.vm_dut = None
+
+    def set_up(self):
+        """
+        run before each test case.
+        """
+        self.table_header = ['Mode', '[M|G]bits/sec']
+        self.result_table_create(self.table_header)
+        self.dut.send_expect("killall -s INT testpmd", "#")
+        self.dut.send_expect("killall -s INT qemu-system-x86_64", "#")
+        self.dut.send_expect("rm -rf ./vhost-net*", "#")
+        self.vhost = self.dut.new_session(suite="vhost")
+        self.vm_dut = []
+        self.vm = []
+
+    def start_vhost_testpmd(self, zerocopy=False):
+        """
+        launch the testpmd with different parameters
+        """
+        if zerocopy is True:
+            zerocopy_arg = ",dequeue-zero-copy=1"
+        else:
+            zerocopy_arg = ""
+        self.command_line = self.dut.target + "/app/testpmd -c %s -n %d " + \
+            "--socket-mem 2048,2048 --legacy-mem --no-pci --file-prefix=vhost " + \
+            "--vdev 'net_vhost0,iface=vhost-net0,queues=1%s' " + \
+            "--vdev 'net_vhost1,iface=vhost-net1,queues=1%s' " + \
+            "-- -i --nb-cores=1 --txd=1024 --rxd=1024 --txfreet=992 "
+
+        self.command_line = self.command_line % (
+                            self.coremask, self.memory_channel,
+                            zerocopy_arg, zerocopy_arg)
+        self.vhost.send_expect(self.command_line, "testpmd> ", 30)
+        self.vhost.send_expect("start", "testpmd> ", 30)
+
+    def start_vms(self, mode="tso"):
+        """
+        start two VM, each VM has one virtio device
+        """
+        setting_args = "mrg_rxbuf=on,csum=on,guest_csum=on,host_tso4=on,guest_tso4=on,guest_ecn=on"
+        if mode == "tso":
+            setting_args += ",gso=on"
+        elif mode == "ufo":
+            setting_args += ",guest_ufo=on,host_ufo=on"
+
+        for i in range(self.vm_num):
+            vm_dut = None
+            vm_info = VM(self.dut, 'vm%d' % i, 'vhost_sample')
+            vm_params = {}
+            vm_params['driver'] = 'vhost-user'
+            vm_params['opt_path'] = './vhost-net%d' % i
+            vm_params['opt_mac'] = "52:54:00:00:00:0%d" % (i+1)
+            vm_params['opt_settings'] = setting_args
+            vm_info.set_vm_device(**vm_params)
+            time.sleep(3)
+            try:
+                vm_dut = vm_info.start()
+                if vm_dut is None:
+                    raise Exception("Set up VM ENV failed")
+            except Exception as e:
+                print utils.RED("Failure for %s" % str(e))
+            vm_dut.restore_interfaces()
+
+            self.vm_dut.append(vm_dut)
+            self.vm.append(vm_info)
+
+    def config_vm_env(self):
+        """
+        set virtio device IP and run arp protocal
+        """
+        vm1_intf = self.vm_dut[0].ports_info[0]['intf']
+        vm2_intf = self.vm_dut[1].ports_info[0]['intf']
+        self.vm_dut[0].send_expect("ifconfig %s %s" % (vm1_intf, self.virtio_ip1), "#", 10)
+        self.vm_dut[1].send_expect("ifconfig %s %s" % (vm2_intf, self.virtio_ip2), "#", 10)
+        self.vm_dut[0].send_expect("arp -s %s %s" % (self.virtio_ip2, self.vritio_mac2), "#", 10)
+        self.vm_dut[1].send_expect("arp -s %s %s" % (self.virtio_ip1, self.vritio_mac1), "#", 10)
+
+    def start_iperf(self, mode):
+        """
+        run perf command between to vms
+        """
+        # clear the port xstats before iperf
+        self.vhost.send_expect("clear port xstats all", "testpmd> ", 10)
+
+        if mode == "tso":
+            iperf_server = "iperf -s -i 1"
+            iperf_client = "iperf -c 1.1.1.2 -i 1 -t 30"
+        elif mode == "ufo":
+            iperf_server = "iperf -s -u -i 1"
+            iperf_client = "iperf -c 1.1.1.2 -i 1 -t 30 -P 4 -u -b 1G -l 9000"
+        self.vm_dut[0].send_expect("%s > iperf_server.log &" % iperf_server, "", 10)
+        self.vm_dut[1].send_expect("%s > iperf_client.log &" % iperf_client, "", 60)
+        time.sleep(90)
+
+    def get_perf_result(self):
+        """
+        get the iperf test result
+        """
+        self.vm_dut[1].session.copy_file_from("%s/iperf_client.log" % self.dut.base_dir)
+        fp = open("./iperf_client.log")
+        fmsg = fp.read()
+        fp.close()
+        # remove the server report info from msg
+        index = fmsg.find("Server Report")
+        if index != -1:
+            fmsg = fmsg[:index]
+        iperfdata = re.compile('\S*\s*[M|G]bits/sec').findall(fmsg)
+        # the last data of iperf is the ave data from 0-30 sec
+        self.verify(len(iperfdata) != 0, "The iperf data between to vms is 0")
+        self.logger.info("The iperf data between vms is %s" % iperfdata[-1])
+
+        # put the result to table
+        results_row = ["vm2vm", iperfdata[-1]]
+        self.result_table_add(results_row)
+
+        # rm the iperf log file in vm
+        self.vm_dut[0].send_expect('rm iperf_server.log', '#', 10)
+        self.vm_dut[1].send_expect('rm iperf_client.log', '#', 10)
+
+    def verify_xstats_info_on_vhost(self):
+        """
+        check both 2VMs can receive and send big packets to each other
+        """
+        out_tx = self.vhost.send_expect("show port xstats 0", "testpmd> ", 20)
+        out_rx = self.vhost.send_expect("show port xstats 1", "testpmd> ", 20)
+
+        rx_info = re.search("rx_size_1523_to_max_packets:\s*(\d*)", out_rx)
+        tx_info = re.search("tx_size_1523_to_max_packets:\s*(\d*)", out_tx)
+
+        self.verify(int(rx_info.group(1)) > 0,
+                    "Port 1 not receive packet greater than 1522")
+        self.verify(int(tx_info.group(1)) > 0,
+                    "Port 0 not forward packet greater than 1522")
+
+    def send_and_verify(self, mode):
+        """
+        start to send packets and verify it
+        """
+        self.start_iperf(mode)
+        self.get_perf_result()
+        self.verify_xstats_info_on_vhost()
+        self.result_table_print()
+
+    def stop_all_apps(self):
+        for i in range(len(self.vm)):
+            self.vm[i].stop()
+        self.vhost.send_expect("quit", "#", 30)
+
+    def offload_capbility_check(self, vm_client):
+        """
+        check UFO and TSO offload status on for the Virtio-net driver in VM
+        """
+        vm_intf = vm_client.ports_info[0]['intf']
+        vm_client.send_expect('ethtool -k %s > offload.log' % vm_intf, '#', 10)
+        fmsg = vm_client.send_expect("cat ./offload.log", "#")
+        udp_info = re.search("udp-fragmentation-offload:\s*(\S*)", fmsg)
+        tcp_info = re.search("tx-tcp-segmentation:\s*(\S*)", fmsg)
+        tcp_enc_info = re.search("tx-tcp-ecn-segmentation:\s*(\S*)", fmsg)
+        tcp6_info = re.search("tx-tcp6-segmentation:\s*(\S*)", fmsg)
+
+        self.verify(udp_info is not None and udp_info.group(1) == "on",
+                    "the udp-fragmentation-offload in vm not right")
+        self.verify(tcp_info is not None and tcp_info.group(1) == "on",
+                    "tx-tcp-segmentation in vm not right")
+        self.verify(tcp_enc_info is not None and tcp_enc_info.group(1) == "on",
+                    "tx-tcp-ecn-segmentation in vm not right")
+        self.verify(tcp6_info is not None and tcp6_info.group(1) == "on",
+                    "tx-tcp6-segmentation in vm not right")
+
+    def test_vhost_vm2vm_tso_iperf(self):
+        """
+        vhost-user + virtio-net VM2VM with tcp traffic
+        """
+        self.start_vhost_testpmd(zerocopy=False)
+        self.start_vms(mode="tso")
+        self.config_vm_env()
+        self.send_and_verify(mode="tso")
+        self.stop_all_apps()
+
+    def test_vhost_vm2vm_tso_iperf_zerocopy(self):
+        """
+        vhost-user + virtio-net VM2VM zero-copy with tcp traffic
+        """
+        self.start_vhost_testpmd(zerocopy=True)
+        self.start_vms(mode="tso")
+        self.config_vm_env()
+        self.send_and_verify(mode="tso")
+        self.stop_all_apps()
+
+    def test_vhost_vm2vm_ufo_iperf(self):
+        """
+        vhost-user + virtio-net VM2VM with udp traffic
+        """
+        self.start_vhost_testpmd(zerocopy=False)
+        self.start_vms(mode="ufo")
+        self.config_vm_env()
+        self.send_and_verify(mode="ufo")
+        self.stop_all_apps()
+
+    def test_vhost_vm2vm_ufo_capbility(self):
+        """
+        check virtio-net device capability
+        """
+        self.start_vhost_testpmd(zerocopy=False)
+        self.start_vms(mode="ufo")
+        self.offload_capbility_check(self.vm_dut[0])
+        self.offload_capbility_check(self.vm_dut[1])
+        self.stop_all_apps()
+
+    def tear_down(self):
+        """
+        run after each test case.
+        """
+        self.dut.send_expect("killall -s INT testpmd", "#")
+        self.dut.send_expect("killall -s INT qemu-system-x86_64", "#")
+        time.sleep(2)
+
+    def tear_down_all(self):
+        """
+        Run after each test suite.
+        """
+        pass
-- 
2.7.4