1 # Copyright (c) 2019 Cisco and/or its affiliates.
2 # Licensed under the Apache License, Version 2.0 (the "License");
3 # you may not use this file except in compliance with the License.
4 # You may obtain a copy of the License at:
6 # http://www.apache.org/licenses/LICENSE-2.0
8 # Unless required by applicable law or agreed to in writing, software
9 # distributed under the License is distributed on an "AS IS" BASIS,
10 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11 # See the License for the specific language governing permissions and
12 # limitations under the License.
14 """VPP Configuration File Generator library."""
19 from resources.libraries.python.ssh import SSH
20 from resources.libraries.python.Constants import Constants
21 from resources.libraries.python.DUTSetup import DUTSetup
22 from resources.libraries.python.topology import NodeType
23 from resources.libraries.python.topology import Topology
25 __all__ = ['VppConfigGenerator']
28 def pci_dev_check(pci_dev):
29 """Check if provided PCI address is in correct format.
31 :param pci_dev: PCI address (expected format: xxxx:xx:xx.x).
33 :returns: True if PCI address is in correct format.
35 :raises ValueError: If PCI address is in incorrect format.
37 pattern = re.compile("^[0-9A-Fa-f]{4}:[0-9A-Fa-f]{2}:"
38 "[0-9A-Fa-f]{2}\\.[0-9A-Fa-f]$")
39 if not pattern.match(pci_dev):
40 raise ValueError('PCI address {addr} is not in valid format '
41 'xxxx:xx:xx.x'.format(addr=pci_dev))
45 class VppConfigGenerator(object):
46 """VPP Configuration File Generator."""
49 """Initialize library."""
50 # VPP Node to apply configuration on
56 # Serialized VPP Configuration
59 self._vpp_service_name = 'vpp'
60 # VPP Logfile location
61 self._vpp_logfile = '/tmp/vpe.log'
62 # VPP Startup config location
63 self._vpp_startup_conf = '/etc/vpp/startup.conf'
64 # VPP Startup config backup location
65 self._vpp_startup_conf_backup = None
67 def set_node(self, node):
70 :param node: Node to store configuration on.
72 :raises RuntimeError: If Node type is not DUT.
74 if node['type'] != NodeType.DUT:
75 raise RuntimeError('Startup config can only be applied to DUT'
78 self._hostname = Topology.get_node_hostname(node)
80 def set_vpp_logfile(self, logfile):
81 """Set VPP logfile location.
83 :param logfile: VPP logfile location.
86 self._vpp_logfile = logfile
88 def set_vpp_startup_conf_backup(self, backup='/etc/vpp/startup.backup'):
89 """Set VPP startup configuration backup.
91 :param backup: VPP logfile location.
94 self._vpp_startup_conf_backup = backup
96 def get_config_str(self):
97 """Get dumped startup configuration in VPP config format.
99 :returns: Startup configuration in VPP config format.
102 self.dump_config(self._nodeconfig)
103 return self._vpp_config
105 def add_config_item(self, config, value, path):
106 """Add startup configuration item.
108 :param config: Startup configuration of node.
109 :param value: Value to insert.
110 :param path: Path where to insert item.
116 config[path[0]] = value
118 if path[0] not in config:
120 elif isinstance(config[path[0]], str):
121 config[path[0]] = {} if config[path[0]] == '' \
122 else {config[path[0]]: ''}
123 self.add_config_item(config[path[0]], value, path[1:])
125 def dump_config(self, obj, level=-1):
126 """Dump the startup configuration in VPP config format.
128 :param obj: Python Object to print.
129 :param level: Nested level for indentation.
136 self._vpp_config += '{}{{\n'.format((level) * indent)
137 if isinstance(obj, dict):
138 for key, val in obj.items():
139 if hasattr(val, '__iter__'):
140 self._vpp_config += '{}{}\n'.format((level + 1) * indent,
142 self.dump_config(val, level + 1)
144 self._vpp_config += '{}{} {}\n'.format((level + 1) * indent,
148 self._vpp_config += '{}{}\n'.format((level + 1) * indent, val)
150 self._vpp_config += '{}}}\n'.format(level * indent)
152 def add_unix_log(self, value=None):
153 """Add UNIX log configuration.
155 :param value: Log file.
158 path = ['unix', 'log']
160 value = self._vpp_logfile
161 self.add_config_item(self._nodeconfig, value, path)
163 def add_unix_cli_listen(self, value='/run/vpp/cli.sock'):
164 """Add UNIX cli-listen configuration.
166 :param value: CLI listen address and port or path to CLI socket.
169 path = ['unix', 'cli-listen']
170 self.add_config_item(self._nodeconfig, value, path)
172 def add_unix_gid(self, value='vpp'):
173 """Add UNIX gid configuration.
178 path = ['unix', 'gid']
179 self.add_config_item(self._nodeconfig, value, path)
181 def add_unix_nodaemon(self):
182 """Add UNIX nodaemon configuration."""
183 path = ['unix', 'nodaemon']
184 self.add_config_item(self._nodeconfig, '', path)
186 def add_unix_coredump(self):
187 """Add UNIX full-coredump configuration."""
188 path = ['unix', 'full-coredump']
189 self.add_config_item(self._nodeconfig, '', path)
191 def add_unix_exec(self, value):
192 """Add UNIX exec configuration."""
193 path = ['unix', 'exec']
194 self.add_config_item(self._nodeconfig, value, path)
196 def add_api_segment_gid(self, value='vpp'):
197 """Add API-SEGMENT gid configuration.
202 path = ['api-segment', 'gid']
203 self.add_config_item(self._nodeconfig, value, path)
205 def add_api_segment_global_size(self, value):
206 """Add API-SEGMENT global-size configuration.
208 :param value: Global size.
211 path = ['api-segment', 'global-size']
212 self.add_config_item(self._nodeconfig, value, path)
214 def add_api_segment_api_size(self, value):
215 """Add API-SEGMENT api-size configuration.
217 :param value: API size.
220 path = ['api-segment', 'api-size']
221 self.add_config_item(self._nodeconfig, value, path)
223 def add_dpdk_dev(self, *devices):
224 """Add DPDK PCI device configuration.
226 :param devices: PCI device(s) (format xxxx:xx:xx.x)
229 for device in devices:
230 if pci_dev_check(device):
231 path = ['dpdk', 'dev {0}'.format(device)]
232 self.add_config_item(self._nodeconfig, '', path)
234 def add_dpdk_dev_parameter(self, device, parameter, value):
235 """Add parameter for DPDK device.
237 :param device: PCI device (format xxxx:xx:xx.x).
238 :param parameter: Parameter name.
239 :param value: Parameter value.
244 if pci_dev_check(device):
245 path = ['dpdk', 'dev {0}'.format(device), parameter]
246 self.add_config_item(self._nodeconfig, value, path)
248 def add_dpdk_cryptodev(self, count):
249 """Add DPDK Crypto PCI device configuration.
251 :param count: Number of HW crypto devices to add.
254 cryptodev = Topology.get_cryptodev(self._node)
255 for i in range(count):
256 cryptodev_config = 'dev {0}'.format(
257 re.sub(r'\d.\d$', '1.'+str(i), cryptodev))
258 path = ['dpdk', cryptodev_config]
259 self.add_config_item(self._nodeconfig, '', path)
260 self.add_dpdk_uio_driver('igb_uio')
262 def add_dpdk_sw_cryptodev(self, sw_pmd_type, socket_id, count):
263 """Add DPDK SW Crypto device configuration.
265 :param sw_pmd_type: Type of SW crypto device PMD to add.
266 :param socket_id: Socket ID.
267 :param count: Number of SW crypto devices to add.
268 :type sw_pmd_type: str
272 for _ in range(count):
273 cryptodev_config = 'vdev cryptodev_{0}_pmd,socket_id={1}'.\
274 format(sw_pmd_type, str(socket_id))
275 path = ['dpdk', cryptodev_config]
276 self.add_config_item(self._nodeconfig, '', path)
278 def add_dpdk_eth_bond_dev(self, ethbond_id, mode, xmit_policy, *slaves):
279 """Add DPDK Eth_bond device configuration.
281 :param ethbond_id: Eth_bond device ID.
282 :param mode: Link bonding mode.
283 :param xmit_policy: Transmission policy.
284 :param slaves: PCI device(s) to be bonded (format xxxx:xx:xx.x).
285 :type ethbond_id: str or int
286 :type mode: str or int
287 :type xmit_policy: str
290 slaves_config = ',slave=' + \
291 ',slave='.join(slave if pci_dev_check(slave) else ''
293 ethbond_config = 'vdev eth_bond{id},mode={mode}{slaves},' \
294 'xmit_policy={xmit_pol}'.format(id=ethbond_id,
296 slaves=slaves_config,
297 xmit_pol=xmit_policy)
298 path = ['dpdk', ethbond_config]
299 self.add_config_item(self._nodeconfig, '', path)
301 def add_dpdk_dev_default_rxq(self, value):
302 """Add DPDK dev default rxq configuration.
304 :param value: Default number of rxqs.
307 path = ['dpdk', 'dev default', 'num-rx-queues']
308 self.add_config_item(self._nodeconfig, value, path)
310 def add_dpdk_dev_default_txq(self, value):
311 """Add DPDK dev default txq configuration.
313 :param value: Default number of txqs.
316 path = ['dpdk', 'dev default', 'num-tx-queues']
317 self.add_config_item(self._nodeconfig, value, path)
319 def add_dpdk_dev_default_rxd(self, value):
320 """Add DPDK dev default rxd configuration.
322 :param value: Default number of rxds.
325 path = ['dpdk', 'dev default', 'num-rx-desc']
326 self.add_config_item(self._nodeconfig, value, path)
328 def add_dpdk_dev_default_txd(self, value):
329 """Add DPDK dev default txd configuration.
331 :param value: Default number of txds.
334 path = ['dpdk', 'dev default', 'num-tx-desc']
335 self.add_config_item(self._nodeconfig, value, path)
337 def add_dpdk_log_level(self, value):
338 """Add DPDK log-level configuration.
340 :param value: Log level.
343 path = ['dpdk', 'log-level']
344 self.add_config_item(self._nodeconfig, value, path)
346 def add_dpdk_num_mbufs(self, value):
347 """Add DPDK number of I/O buffers.
349 :param value: Number of I/O buffers.
352 path = ['dpdk', 'num-mbufs']
353 self.add_config_item(self._nodeconfig, value, path)
355 def add_dpdk_no_pci(self):
356 """Add DPDK no-pci."""
357 path = ['dpdk', 'no-pci']
358 self.add_config_item(self._nodeconfig, '', path)
360 def add_dpdk_uio_driver(self, value=None):
361 """Add DPDK uio-driver configuration.
363 :param value: DPDK uio-driver configuration. By default, driver will be
364 loaded automatically from Topology file, still leaving
365 option to manually override by parameter.
369 value = Topology.get_uio_driver(self._node)
370 path = ['dpdk', 'uio-driver']
371 self.add_config_item(self._nodeconfig, value, path)
373 def add_cpu_main_core(self, value):
374 """Add CPU main core configuration.
376 :param value: Main core option.
379 path = ['cpu', 'main-core']
380 self.add_config_item(self._nodeconfig, value, path)
382 def add_cpu_corelist_workers(self, value):
383 """Add CPU corelist-workers configuration.
385 :param value: Corelist-workers option.
388 path = ['cpu', 'corelist-workers']
389 self.add_config_item(self._nodeconfig, value, path)
391 def add_heapsize(self, value):
392 """Add Heapsize configuration.
394 :param value: Amount of heapsize.
398 self.add_config_item(self._nodeconfig, value, path)
400 def add_api_trace(self):
401 """Add API trace configuration."""
402 path = ['api-trace', 'on']
403 self.add_config_item(self._nodeconfig, '', path)
405 def add_ip6_hash_buckets(self, value):
406 """Add IP6 hash buckets configuration.
408 :param value: Number of IP6 hash buckets.
411 path = ['ip6', 'hash-buckets']
412 self.add_config_item(self._nodeconfig, value, path)
414 def add_ip6_heap_size(self, value):
415 """Add IP6 heap-size configuration.
417 :param value: IP6 Heapsize amount.
420 path = ['ip6', 'heap-size']
421 self.add_config_item(self._nodeconfig, value, path)
423 def add_ip_heap_size(self, value):
424 """Add IP heap-size configuration.
426 :param value: IP Heapsize amount.
429 path = ['ip', 'heap-size']
430 self.add_config_item(self._nodeconfig, value, path)
432 def add_statseg_size(self, value):
433 """Add stats segment heap size configuration.
435 :param value: Stats heapsize amount.
438 path = ['statseg', 'size']
439 self.add_config_item(self._nodeconfig, value, path)
441 def add_plugin(self, state, *plugins):
442 """Add plugin section for specific plugin(s).
444 :param state: State of plugin [enable|disable].
445 :param plugins: Plugin(s) to disable.
449 for plugin in plugins:
450 path = ['plugins', 'plugin {0}'.format(plugin), state]
451 self.add_config_item(self._nodeconfig, ' ', path)
453 def add_dpdk_no_multi_seg(self):
454 """Add DPDK no-multi-seg configuration."""
455 path = ['dpdk', 'no-multi-seg']
456 self.add_config_item(self._nodeconfig, '', path)
458 def add_dpdk_no_tx_checksum_offload(self):
459 """Add DPDK no-tx-checksum-offload configuration."""
460 path = ['dpdk', 'no-tx-checksum-offload']
461 self.add_config_item(self._nodeconfig, '', path)
463 def add_nat(self, value='deterministic'):
464 """Add NAT configuration.
466 :param value: NAT mode.
470 self.add_config_item(self._nodeconfig, value, path)
472 def add_tcp_preallocated_connections(self, value):
473 """Add TCP pre-allocated connections.
475 :param value: The number of pre-allocated connections.
478 path = ['tcp', 'preallocated-connections']
479 self.add_config_item(self._nodeconfig, value, path)
481 def add_tcp_preallocated_half_open_connections(self, value):
482 """Add TCP pre-allocated half open connections.
484 :param value: The number of pre-allocated half open connections.
487 path = ['tcp', 'preallocated-half-open-connections']
488 self.add_config_item(self._nodeconfig, value, path)
490 def add_session_event_queue_length(self, value):
491 """Add session event queue length.
493 :param value: Session event queue length.
496 path = ['session', 'event-queue-length']
497 self.add_config_item(self._nodeconfig, value, path)
499 def add_session_preallocated_sessions(self, value):
500 """Add the number of pre-allocated sessions.
502 :param value: Number of pre-allocated sessions.
505 path = ['session', 'preallocated-sessions']
506 self.add_config_item(self._nodeconfig, value, path)
508 def add_session_v4_session_table_buckets(self, value):
509 """Add number of v4 session table buckets to the config.
511 :param value: Number of v4 session table buckets.
514 path = ['session', 'v4-session-table-buckets']
515 self.add_config_item(self._nodeconfig, value, path)
517 def add_session_v4_session_table_memory(self, value):
518 """Add the size of v4 session table memory.
520 :param value: Size of v4 session table memory.
523 path = ['session', 'v4-session-table-memory']
524 self.add_config_item(self._nodeconfig, value, path)
526 def add_session_v4_halfopen_table_buckets(self, value):
527 """Add the number of v4 halfopen table buckets.
529 :param value: Number of v4 halfopen table buckets.
532 path = ['session', 'v4-halfopen-table-buckets']
533 self.add_config_item(self._nodeconfig, value, path)
535 def add_session_v4_halfopen_table_memory(self, value):
536 """Add the size of v4 halfopen table memory.
538 :param value: Size of v4 halfopen table memory.
541 path = ['session', 'v4-halfopen-table-memory']
542 self.add_config_item(self._nodeconfig, value, path)
544 def add_session_local_endpoints_table_buckets(self, value):
545 """Add the number of local endpoints table buckets.
547 :param value: Number of local endpoints table buckets.
550 path = ['session', 'local-endpoints-table-buckets']
551 self.add_config_item(self._nodeconfig, value, path)
553 def add_session_local_endpoints_table_memory(self, value):
554 """Add the size of local endpoints table memory.
556 :param value: Size of local endpoints table memory.
559 path = ['session', 'local-endpoints-table-memory']
560 self.add_config_item(self._nodeconfig, value, path)
562 def apply_config(self, filename=None, retries=60, restart_vpp=True):
563 """Generate and apply VPP configuration for node.
565 Use data from calls to this class to form a startup.conf file and
566 replace /etc/vpp/startup.conf with it on node.
568 :param filename: Startup configuration file name.
569 :param retries: Number of times (default 60) to re-try waiting.
570 :param restart_vpp: Whether to restart VPP.
573 :type restart_vpp: bool.
574 :raises RuntimeError: If writing config file failed or restart of VPP
575 failed or backup of VPP startup.conf failed.
577 self.dump_config(self._nodeconfig)
580 ssh.connect(self._node)
583 filename = self._vpp_startup_conf
585 if self._vpp_startup_conf_backup is not None:
587 ssh.exec_command('sudo cp {src} {dest}'.
588 format(src=self._vpp_startup_conf,
589 dest=self._vpp_startup_conf_backup))
591 raise RuntimeError('Backup of config file failed on node '
592 '{name}'.format(name=self._hostname))
595 ssh.exec_command('echo "{config}" | sudo tee {filename}'.
596 format(config=self._vpp_config,
600 raise RuntimeError('Writing config file failed to node {name}'.
601 format(name=self._hostname))
604 DUTSetup.start_service(self._node, Constants.VPP_UNIT)
606 # Sleep <waittime> seconds, up to <retry> times,
607 # and verify if VPP is running.
608 for _ in range(retries):
611 ssh.exec_command_sudo('vppctl show pci')
612 if ret == 0 and 'Connection refused' not in stdout:
615 raise RuntimeError('VPP failed to restart on node {name}'.
616 format(name=self._hostname))
618 def restore_config(self):
619 """Restore VPP startup.conf from backup.
621 :raises RuntimeError: When restoration of startup.conf file failed.
625 ssh.connect(self._node)
627 ret, _, _ = ssh.exec_command('sudo cp {src} {dest}'.
628 format(src=self._vpp_startup_conf_backup,
629 dest=self._vpp_startup_conf))
631 raise RuntimeError('Restoration of config file failed on node '
632 '{name}'.format(name=self._hostname))