1 # Copyright (c) 2016-2017 Intel Corporation
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
7 # http://www.apache.org/licenses/LICENSE-2.0
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 """ Helper function to get Network Service testing configuration """
16 from __future__ import absolute_import
21 from oslo_config import cfg
22 from oslo_config.cfg import NoSuchOptError
23 from oslo_utils import encodeutils
25 NSB_ROOT = "/opt/nsb_bin"
29 cfg.StrOpt('bin_path',
31 help='bin_path for VNFs location.'),
32 cfg.StrOpt('trex_path',
33 default=os.path.join(NSB_ROOT, 'trex/scripts'),
34 help='trex automation lib path.'),
35 cfg.StrOpt('trex_client_lib',
36 default=os.path.join(NSB_ROOT, 'trex_client/stl'),
37 help='trex python library path.'),
39 CONF.register_opts(OPTS, group="nsb")
42 HEXADECIMAL = "[0-9a-zA-Z]"
45 class PciAddress(object):
47 PCI_PATTERN_STR = HEXADECIMAL.join([
49 "{4}):(", # domain (4 bytes)
50 "{2}):(", # bus (2 bytes)
51 "{2}).(", # function (2 bytes)
55 PCI_PATTERN = re.compile(PCI_PATTERN_STR)
58 def parse_address(cls, text, multi_line=False):
60 text = text.replace(os.linesep, '')
61 match = cls.PCI_PATTERN.search(text)
62 return cls(match.group(0))
64 def __init__(self, address):
65 super(PciAddress, self).__init__()
66 match = self.PCI_PATTERN.match(address)
68 raise ValueError('Invalid PCI address: {}'.format(address))
69 self.address = address
77 return self.match.group(1)
81 return self.match.group(2)
85 return self.match.group(3)
89 return self.match.group(4)
92 return [self.match.group(n) for n in range(1, 5)]
95 def get_nsb_option(option, default=None):
96 """return requested option for yardstick.conf"""
99 return CONF.nsb.__getitem__(option)
100 except NoSuchOptError:
101 logging.debug("Invalid key %s", option)
105 def provision_tool(connection, tool_path, tool_file=None):
107 verify if the tool path exits on the node,
108 if not push the local binary to remote node
113 tool_path = get_nsb_option('tool_path')
115 tool_path = os.path.join(tool_path, tool_file)
116 bin_path = get_nsb_option("bin_path")
117 exit_status = connection.execute("which %s > /dev/null 2>&1" % tool_path)[0]
119 return encodeutils.safe_decode(tool_path, incoming='utf-8').rstrip()
121 logging.warning("%s not found on %s, will try to copy from localhost",
122 tool_path, connection.host)
123 bin_path = get_nsb_option("bin_path")
124 connection.execute('mkdir -p "%s"' % bin_path)
125 connection.put(tool_path, tool_path)