diff options
Diffstat (limited to 'nfvbench/chain_managers.py')
| -rw-r--r-- | nfvbench/chain_managers.py | 231 |
1 files changed, 231 insertions, 0 deletions
diff --git a/nfvbench/chain_managers.py b/nfvbench/chain_managers.py new file mode 100644 index 0000000..fe3a2d4 --- /dev/null +++ b/nfvbench/chain_managers.py @@ -0,0 +1,231 @@ +#!/usr/bin/env python +# Copyright 2016 Cisco Systems, Inc. All rights reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); you may +# not use this file except in compliance with the License. You may obtain +# a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT +# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the +# License for the specific language governing permissions and limitations +# under the License. +# + +from log import LOG +from network import Network +from packet_analyzer import PacketAnalyzer +from specs import ChainType +from stats_collector import IntervalCollector +import time + + +class StageManager(object): + + def __init__(self, config, cred, factory): + self.config = config + self.client = None + # conditions due to EXT chain special cases + if (config.vlan_tagging and not config.vlans) or not config.no_int_config: + VM_CLASS = factory.get_stage_class(config.service_chain) + self.client = VM_CLASS(config, cred) + self.client.setup() + + def get_vlans(self): + return self.client.get_vlans() if self.client else [] + + def get_host_ips(self): + return self.client.get_host_ips() + + def get_networks_uuids(self): + return self.client.get_networks_uuids() + + def disable_port_security(self): + self.client.disable_port_security() + + def get_vms(self): + return self.client.vms + + def get_nets(self): + return self.client.nets + + def get_ports(self): + return self.client.ports + + def get_compute_nodes(self): + return self.client.compute_nodes + + def set_vm_macs(self): + if self.client and self.config.service_chain != ChainType.EXT: + self.config.generator_config.set_vm_mac_list(self.client.get_end_port_macs()) + + def close(self): + if not self.config.no_cleanup and self.client: + self.client.dispose() + + +class StatsManager(object): + + def __init__(self, config, clients, specs, factory, vlans, notifier=None): + self.config = config + self.clients = clients + self.specs = specs + self.notifier = notifier + self.interval_collector = None + self.vlans = vlans + self.factory = factory + self._setup() + + def set_vlan_tag(self, device, vlan): + self.worker.set_vlan_tag(device, vlan) + + def _setup(self): + WORKER_CLASS = self.factory.get_chain_worker(self.specs.openstack.encaps, + self.config.service_chain) + self.worker = WORKER_CLASS(self.config, self.clients, self.specs) + self.worker.set_vlans(self.vlans) + self._config_interfaces() + + def _get_data(self): + return self.worker.get_data() + + def _get_network(self, traffic_port, index=None, reverse=False): + interfaces = [self.clients['traffic'].get_interface(traffic_port)] + interfaces.extend(self.worker.get_network_interfaces(index)) + return Network(interfaces, reverse) + + def _config_interfaces(self): + if self.config.service_chain != ChainType.EXT: + self.clients['vm'].disable_port_security() + + self.worker.config_interfaces() + + def _generate_traffic(self): + if self.config.no_traffic: + return + + self.interval_collector = IntervalCollector(time.time()) + self.interval_collector.attach_notifier(self.notifier) + LOG.info('Starting to generate traffic...') + stats = {} + for stats in self.clients['traffic'].run_traffic(): + self.interval_collector.add(stats) + + LOG.info('...traffic generating ended.') + return stats + + def get_stats(self): + return self.interval_collector.get() if self.interval_collector else [] + + def get_version(self): + return self.worker.get_version() + + def run(self): + """ + Run analysis in both direction and return the analysis + """ + self.worker.run() + + stats = self._generate_traffic() + result = { + 'raw_data': self._get_data(), + 'packet_analysis': {}, + 'stats': stats + } + + LOG.info('Requesting packet analysis on the forward direction...') + result['packet_analysis']['direction-forward'] = \ + self.get_analysis([self._get_network(0, 0), + self._get_network(0, 1, reverse=True)]) + LOG.info('Packet analysis on the forward direction completed') + + LOG.info('Requesting packet analysis on the reverse direction...') + result['packet_analysis']['direction-reverse'] = \ + self.get_analysis([self._get_network(1, 1), + self._get_network(1, 0, reverse=True)]) + + LOG.info('Packet analysis on the reverse direction completed') + return result + + def get_compute_nodes_bios(self): + return self.worker.get_compute_nodes_bios() + + @staticmethod + def get_analysis(nets): + LOG.info('Starting traffic analysis...') + + packet_analyzer = PacketAnalyzer() + # Traffic types are assumed to always alternate in every chain. Add a no stats interface in + # between if that is not the case. + tx = True + for network in nets: + for interface in network.get_interfaces(): + packet_analyzer.record(interface, 'tx' if tx else 'rx') + tx = not tx + + LOG.info('...traffic analysis completed') + return packet_analyzer.get_analysis() + + def close(self): + self.worker.close() + + +class PVPStatsManager(StatsManager): + + def __init__(self, config, clients, specs, factory, vlans, notifier=None): + StatsManager.__init__(self, config, clients, specs, factory, vlans, notifier) + + +class PVVPStatsManager(StatsManager): + + def __init__(self, config, clients, specs, factory, vlans, notifier=None): + StatsManager.__init__(self, config, clients, specs, factory, vlans, notifier) + + def run(self): + """ + Run analysis in both direction and return the analysis + """ + fwd_v2v_net, rev_v2v_net = self.worker.run() + + stats = self._generate_traffic() + result = { + 'raw_data': self._get_data(), + 'packet_analysis': {}, + 'stats': stats + } + + fwd_nets = [self._get_network(0, 0)] + if fwd_v2v_net: + fwd_nets.append(fwd_v2v_net) + fwd_nets.append(self._get_network(0, 1, reverse=True)) + + rev_nets = [self._get_network(1, 1)] + if rev_v2v_net: + rev_nets.append(rev_v2v_net) + rev_nets.append(self._get_network(1, 0, reverse=True)) + + LOG.info('Requesting packet analysis on the forward direction...') + result['packet_analysis']['direction-forward'] = self.get_analysis(fwd_nets) + LOG.info('Packet analysis on the forward direction completed') + + LOG.info('Requesting packet analysis on the reverse direction...') + result['packet_analysis']['direction-reverse'] = self.get_analysis(rev_nets) + + LOG.info('Packet analysis on the reverse direction completed') + return result + + +class EXTStatsManager(StatsManager): + def __init__(self, config, clients, specs, factory, vlans, notifier=None): + StatsManager.__init__(self, config, clients, specs, factory, vlans, notifier) + + def _setup(self): + WORKER_CLASS = self.factory.get_chain_worker(self.specs.openstack.encaps, + self.config.service_chain) + self.worker = WORKER_CLASS(self.config, self.clients, self.specs) + self.worker.set_vlans(self.vlans) + + if not self.config.no_int_config: + self._config_interfaces() |