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 """ PING acts as traffic generation and vnf definitions based on IETS Spec """
16 from __future__ import absolute_import
20 from multiprocessing import Queue
21 from ipaddress import IPv4Interface
23 from yardstick.network_services.vnf_generic.vnf.sample_vnf import SampleVNFTrafficGen
24 from yardstick.network_services.vnf_generic.vnf.sample_vnf import DpdkVnfSetupEnvHelper
25 from yardstick.network_services.vnf_generic.vnf.sample_vnf import ClientResourceHelper
27 LOG = logging.getLogger(__name__)
30 class PingParser(object):
31 """ Class providing file-like API for talking with SSH connection """
33 def __init__(self, q_out):
37 def write(self, chunk):
38 """ 64 bytes from 10.102.22.93: icmp_seq=1 ttl=64 time=0.296 ms """
39 match = re.search(r"icmp_seq=(\d+).*time=([0-9.]+)", chunk)
40 LOG.debug("Parser called on %s", chunk)
42 # IMPORTANT: in order for the data to be properly taken
43 # in by InfluxDB, it needs to be converted to numeric types
45 "packets_received": float(match.group(1)),
46 "rtt": float(match.group(2)),
50 """ close the ssh connection """
54 """ clear queue till Empty """
55 while self.queue.qsize() > 0:
59 class PingSetupEnvHelper(DpdkVnfSetupEnvHelper):
61 def setup_vnf_environment(self):
62 for intf in self.vnfd_helper.interfaces:
63 vi = intf['virtual-interface']
64 vi['local_iface_name'] = self.get_local_iface_name_by_vpci(vi['vpci'])
67 class PingResourceHelper(ClientResourceHelper):
69 def __init__(self, setup_helper):
70 super(PingResourceHelper, self).__init__(setup_helper)
72 self._parser = PingParser(self._queue)
74 def run_traffic(self, traffic_profile, *args):
75 # drop the connection in order to force a new one
76 self.ssh_helper.drop_connection()
78 self.client_started.value = 1
80 "sudo ip addr flush {local_if_name}",
81 "sudo ip addr add {local_ip}/24 dev {local_if_name}",
82 "sudo ip link set {local_if_name} up",
85 self.cmd_kwargs['packet_size'] = traffic_profile.params['traffic_profile']['frame_size']
88 self.ssh_helper.execute(cmd.format(**self.cmd_kwargs))
90 ping_cmd = "nohup ping -s {packet_size} {target_ip}&"
91 self.ssh_helper.run(ping_cmd.format(**self.cmd_kwargs),
93 keep_stdin_open=True, pty=True)
96 class PingTrafficGen(SampleVNFTrafficGen):
98 This traffic generator can ping a single IP with pingsize
99 and target given in traffic profile
106 def __init__(self, name, vnfd, setup_env_helper_type=None, resource_helper_type=None):
107 if setup_env_helper_type is None:
108 setup_env_helper_type = PingSetupEnvHelper
109 if resource_helper_type is None:
110 resource_helper_type = PingResourceHelper
112 super(PingTrafficGen, self).__init__(name, vnfd, setup_env_helper_type,
113 resource_helper_type)
116 def _check_status(self):
117 return self._tg_process.is_alive()
119 def instantiate(self, scenario_cfg, context_cfg):
122 "packets_received": 0,
125 self.setup_helper.setup_vnf_environment()
126 intf = self.vnfd_helper.interfaces[0]["virtual-interface"]
127 self.resource_helper.cmd_kwargs = {
128 'target_ip': IPv4Interface(intf["dst_ip"]).ip.exploded,
129 'local_ip': IPv4Interface(intf["local_ip"]).ip.exploded,
130 'local_if_name': intf["local_iface_name"].split('/')[0],