[dts] [PATCH V1 4/6] framework/dut.py: add function create_eal_parameters

[Chen, Zhaoyan]

Acked-by: Chen, Zhaoyan <zhaoyan.chen at intel.com>



Regards,
Zhaoyan Chen


> -----Original Message-----
> From: Mei, JianweiX
> Sent: Saturday, September 28, 2019 1:35 AM
> To: dts at dpdk.org
> Cc: Chen, Zhaoyan <zhaoyan.chen at intel.com>; Ma, LihongX <lihongx.ma at intel.com>;
> Mei, JianweiX <jianweix.mei at intel.com>
> Subject: [dts][PATCH V1 4/6] framework/dut.py: add function
> create_eal_parameters
> 
> From: meijx <jianweix.mei at intel.com>
> 
> add new method to create EAL parameters and overwrite self.cores according to
> configured dut_cores.
> 
> Signed-off-by: meijx <jianweix.mei at intel.com>
> ---
>  framework/dut.py | 155
> +++++++++++++++++++++++++++++++++++++++++++++++++++++++
>  1 file changed, 155 insertions(+)
> 
> diff --git a/framework/dut.py b/framework/dut.py index a394b34..b30506f
> 100644
> --- a/framework/dut.py
> +++ b/framework/dut.py
> @@ -75,6 +75,160 @@ class Dut(Crb):
>          self.virt_pool = None
>          # hypervisor pid list, used for cleanup
>          self.virt_pids = []
> +        self.prefix_subfix = str(os.getpid()) + '_' + time.strftime("%Y%m%d%H%M%S",
> time.localtime())
> +        self.prefix_list = []
> +
> +    def filter_cores_from_crb_cfg(self):
> +        # get core list from crbs.cfg
> +        core_list = []
> +        all_core_list = [str(core['core']) for core in self.cores]
> +        core_list_str = self.crb['dut_cores']
> +        if core_list_str == '':
> +            core_list = all_core_list
> +        split_by_comma = core_list_str.split(',')
> +        range_cores = []
> +        for item in split_by_comma:
> +            if '-' in item:
> +                tmp = item.split('-')
> +                range_cores.extend([str(i) for i in range(int(tmp[0]), int(tmp[1]) + 1)])
> +            else:
> +                core_list.append(item)
> +        core_list.extend(range_cores)
> +
> +        abnormal_core_list = []
> +        for core in core_list:
> +            if core not in all_core_list:
> +                abnormal_core_list.append(core)
> +
> +        if abnormal_core_list:
> +            self.logger.info('those %s cores are out of range system, all core list of system
> are %s' % (abnormal_core_list, all_core_list))
> +            raise Exception('configured cores out of range system')
> +
> +        core_list = [core for core in self.cores if str(core['core']) in core_list]
> +        self.cores = core_list
> +        self.number_of_cores = len(self.cores)
> +
> +    def create_eal_parameters(self, fixed_prefix=False, socket=-1, **config):
> +        """
> +        generate eal parameters character string
> +        :param config:
> +        :return: eal_str eg:'-c 0xf -w 0000:88:00.0 -w 0000:88:00.1 --file-
> prefix=dpdk_1112_20190809143420'
> +        """
> +        default_cores = '1S/2C/1T'
> +        blank = ' '
> +        os_type = self.get_os_type()
> +        if config:
> +            # deal with cores
> +            if config.has_key('cores'):
> +                if config['cores'] == '' or config['cores'] == 'Default':
> +                    core_list = self.get_core_list(default_cores)
> +                elif type(config['cores']) == list:
> +                    core_list = config['cores']
> +                else:
> +                    core_list = self.get_core_list(config['cores'], socket=socket)
> +            else:
> +                core_list = self.get_core_list(default_cores)
> +
> +            # deal with ports
> +            w_pci_list = []
> +            if config.has_key('ports') and len(config['ports']) != 0:
> +                for port in config['ports']:
> +                    if type(port) == int:
> +                        if config.has_key('port_options') and port in
> config['port_options'].keys():
> +                            port_option = config['port_options'][port]
> +                            w_pci_list.append('-w %s,%s' % (self.ports_info[port]['pci'],
> port_option))
> +                        else:
> +                            w_pci_list.append('-w %s' % self.ports_info[port]['pci'])
> +                    else:
> +                        if config.has_key('port_options') and port in
> config['port_options'].keys():
> +                            port_option = config['port_options'][port]
> +                            w_pci_list.append('-w %s,%s' %
> (self.ports_info[config['ports'].index(port)]['pci'], port_option))
> +                        else:
> +                            w_pci_list = ['-w %s' % pci for pci in config['ports']]
> +            w_pci_str = ' '.join(w_pci_list)
> +
> +            # deal with black ports
> +            b_pci_list = []
> +            if config.has_key('b_ports') and len(config['b_ports']) != 0:
> +                for port in config['b_ports']:
> +                    if type(port) == int:
> +                        b_pci_list.append('-b %s' % self.ports_info[port]['pci'])
> +                    else:
> +                        b_pci_list = ['-b %s' % pci for pci in config['b_ports']]
> +            b_ports_str = ' '.join(b_pci_list)
> +
> +            # deal with no-pci
> +            if config.has_key('no_pci'):
> +                if config['no_pci'] == True:
> +                    no_pci = '--no-pci'
> +                else:
> +                    no_pci = ''
> +            else:
> +                no_pci = ''
> +
> +            # deal with file prefix
> +            if config.has_key('prefix') and config['prefix'] != '':
> +                if fixed_prefix == True:
> +                    file_prefix = config['prefix']
> +                else:
> +                    file_prefix = config['prefix'] + '_' + self.prefix_subfix
> +            else:
> +                file_prefix = 'dpdk' + '_' + self.prefix_subfix
> +            self.prefix_list.append(file_prefix)
> +
> +            # deal with vdev
> +            if config.has_key('vdevs') and len(config['vdevs']) != 0:
> +                vdev = '--vdev ' + ' --vdev '.join(config['vdevs'])
> +            else:
> +                vdev = ''
> +
> +            if os_type == 'freebsd':
> +                eal_str = '-l ' + ','.join(map(str, core_list)) \
> +                          + blank + '-n %d' % self.get_memory_channels() \
> +                          + blank + w_pci_str \
> +                          + blank + b_ports_str \
> +                          + blank + no_pci \
> +                          + blank + vdev
> +            else:
> +                eal_str = '-l ' + ','.join(map(str, core_list)) \
> +                          + blank + '-n %d' % self.get_memory_channels() \
> +                          + blank + w_pci_str \
> +                          + blank + b_ports_str \
> +                          + blank + '--file-prefix=' + file_prefix \
> +                          + blank + no_pci \
> +                          + blank + vdev
> +        else:
> +            # get pci from ports_info
> +            pci_list = []
> +            if len(self.ports_info) != 0:
> +                for port_info in self.ports_info:
> +                    pci_list.append('-w %s' % port_info['pci'])
> +            self.logger.info(pci_list)
> +            pci_str = ' '.join(pci_list)
> +            # default cores '1S/2C/1T'
> +            core_list = self.get_core_list(default_cores)
> +            file_prefix = 'dpdk' + '_' + self.prefix_subfix
> +            self.prefix_list.append(file_prefix)
> +            if os_type == 'freebsd':
> +                eal_str = '-l ' + ','.join(map(str, core_list)) \
> +                          + blank + '-n %d' % self.get_memory_channels() \
> +                          + blank + pci_str
> +            else:
> +                eal_str = '-l ' + ','.join(map(str, core_list)) \
> +                          + blank + '-n %d' % self.get_memory_channels() \
> +                          + blank + pci_str \
> +                          + blank + '--file-prefix=' + file_prefix
> +
> +        return eal_str
> +
> +    def get_eal_of_prefix(self, prefix=None):
> +
> +        if prefix:
> +            file_prefix = [prefix_name for prefix_name in self.prefix_list if prefix in
> prefix_name]
> +        else:
> +            file_prefix = 'dpdk' + '_' + self.prefix_subfix
> +
> +        return file_prefix
> 
>      def init_host_session(self, vm_name):
>          """
> @@ -199,6 +353,7 @@ class Dut(Crb):
>              self.send_expect('alias sed=gsed', '# ')
> 
>          self.init_core_list()
> +        self.filter_cores_from_crb_cfg()
>          self.pci_devices_information()
>          # make sure ipv6 enable before scan
>          self.enable_tester_ipv6()
> --
> 1.8.3.1