tools/pkt_fwd/testpmd.py

   1 # Copyright 2016-2017 Intel Corporation.
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #   http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 """VSPERF TestPMD implementation
  16 """
  17
  18 import logging
  19 import pexpect
  20 from conf import settings
  21 from src.dpdk import dpdk
  22 from src.dpdk import TestPMDProcess
  23 from tools.pkt_fwd.pkt_fwd import IPktFwd
  24
  25 _LOGGER = logging.getLogger(__name__)
  26 _VSWITCHD_CONST_ARGS = ['--', '-i']
  27 _TESTPMD_PVP_CONST_ARGS = ['--vdev', 'net_vhost0,iface=/tmp/dpdkvhostuser0',
  28                            '--vdev', 'net_vhost1,iface=/tmp/dpdkvhostuser1',]
  29
  30 class TestPMD(IPktFwd):
  31     """TestPMD implementation (only phy2phy deployment is supported)
  32
  33     This is wrapper for functionality implemented in ${DPDK}/app/test-pmd.
  34
  35     The method docstrings document only considerations specific to this
  36     implementation. For generic information of the nature of the methods,
  37     see the interface definition.
  38     """
  39
  40     _logger = logging.getLogger()
  41
  42     def __init__(self, guest=False):
  43         vswitchd_args = settings.getValue('VSWITCHD_DPDK_ARGS')
  44
  45         # override socket-mem settings
  46         for tmp_arg in vswitchd_args:
  47             if tmp_arg.startswith('--socket-mem'):
  48                 vswitchd_args.remove(tmp_arg)
  49         vswitchd_args += ['--socket-mem ' +
  50                           ','.join(settings.getValue('DPDK_SOCKET_MEM'))]
  51
  52         if guest:
  53             vswitchd_args += _TESTPMD_PVP_CONST_ARGS
  54         vswitchd_args += _VSWITCHD_CONST_ARGS
  55         vswitchd_args += settings.getValue('TESTPMD_ARGS')
  56
  57         self._nports = len(settings.getValue('NICS'))
  58         self._fwdmode = settings.getValue('TESTPMD_FWD_MODE')
  59         self._csum_layer = settings.getValue('TESTPMD_CSUM_LAYER')
  60         self._csum_calc = settings.getValue('TESTPMD_CSUM_CALC')
  61         self._csum_tunnel = settings.getValue('TESTPMD_CSUM_PARSE_TUNNEL')
  62
  63         self._testpmd = TestPMDProcess(testpmd_args=vswitchd_args)
  64
  65     def start(self):
  66         """See IPktFwd for general description
  67
  68         Activates testpmd.
  69         """
  70         self._logger.info("Starting TestPMD...")
  71         dpdk.init()
  72         self._testpmd.start()
  73         self._logger.info("TestPMD...Started.")
  74
  75         self._testpmd.send('set fwd {}'.format(self._fwdmode), 1)
  76
  77         for port in range(self._nports):
  78             self._testpmd.send('csum set {} {} {}'.format(
  79                 self._csum_layer, self._csum_calc, port), 1)
  80             self._testpmd.send('csum parse_tunnel {} {}'.format(
  81                 self._csum_tunnel, port), 1)
  82
  83         self._testpmd.send('start', 1)
  84
  85     def start_for_guest(self):
  86         """See IPktFwd for general description
  87
  88         Activates testpmd for guest config
  89         """
  90         self._logger.info("Starting TestPMD for one guest...")
  91         dpdk.init()
  92         self._testpmd.start()
  93         self._logger.info("TestPMD...Started.")
  94         self._testpmd.send('set portlist 0,2,1,3')
  95
  96         self._testpmd.send('start', 1)
  97
  98     def stop(self):
  99         """See IPktFwd for general description
 100
 101         Kills testpmd.
 102         """
 103         try:
 104             self._testpmd.send('stop')
 105             self._testpmd.wait('Done.', 5)
 106             self._testpmd.send('quit', 2)
 107             self._testpmd.kill()
 108         except pexpect.EOF:
 109             pass
 110         dpdk.cleanup()