1 ##############################################################################
2 # Copyright (c) 2015 Huawei Technologies Co.,Ltd and others.
4 # All rights reserved. This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
7 # http://www.apache.org/licenses/LICENSE-2.0
8 ##############################################################################
12 import vstf.common.decorator as deco
13 from vstf.common.utils import kill_by_name, my_popen
17 LOG = logging.getLogger(__name__)
23 self._send_processes = []
24 self._receive_processes = []
26 @deco.check("protocol", choices=['udp_bw'], defaults='udp_bw')
27 @deco.check("namespace", defaults=None)
30 @deco.check("size", defaults=64)
31 @deco.check("threads", defaults=1)
32 @deco.check("ratep", defaults=0)
33 def send_start(self, **kwargs):
34 cmd = self.format_send_start(**kwargs)
35 LOG.info("cmd:%s", cmd)
39 stdout=subprocess.PIPE,
40 stderr=subprocess.PIPE)
41 self._send_processes.append(process)
47 error_str = "start netmap send success"
49 error_str = "start netmap send failed, %s" % (str(kwargs))
51 self._send_processes.remove(process)
55 def send_stop(self, **kwargs):
59 for process in self._send_processes:
62 error_str = "stop netmap send success"
63 results.append((ret, error_str))
64 self._send_processes = []
67 def format_send_start(self, **kwargs):
68 cmd = "pkt-gen -i %(src_iface)s -f tx -l %(pkt_size)s -p %(threads)s -D %(dst_mac)s -R %(ratep)s"
70 'src_iface': kwargs['src'][0]['iface'],
71 'dst_mac': kwargs['dst'][0]['mac'],
72 'pkt_size': kwargs['size'],
73 'threads': kwargs['threads'],
74 'ratep': kwargs['ratep']
80 def format_receive_start(self, **kwargs):
81 cmd = "pkt-gen -i %(iface)s -f rx"
83 'iface': kwargs['dst'][0]['iface']
88 @deco.check("protocol", choices=['udp_bw'], defaults='udp_bw')
89 @deco.check("namespace", defaults=None)
91 def receive_start(self, **kwargs):
93 cmd = self.format_receive_start(**kwargs)
94 LOG.info("cmd:%s", cmd)
98 stdout=subprocess.PIPE,
99 stderr=subprocess.PIPE)
100 self._receive_processes.append(process)
106 error_str = "start netmap receive success"
108 error_str = "start netmap receive failed, %s" % (str(kwargs))
110 self._receive_processes.remove(process)
112 return ret, error_str
114 def receive_stop(self, **kwargs):
115 LOG.info("receive_stop")
117 for process in self._receive_processes:
120 self._receive_processes = []
121 error_str = "stop netmap receive success"
122 self._receive_processes = []
123 return ret, error_str
130 def force_clean(self):
131 LOG.info("%s %s start", self.__class__, self.force_clean.__name__)
132 kill_by_name('pkt-gen')
133 self._send_processes = []
134 self._receive_processes = []
141 "protocol": "udp_bw",
142 # "namespace": "receive",
147 ret = perf.receive_start(**receive)
148 LOG.info("*********receive_start***********")
151 # "namespace": "send",
152 "protocol": "udp_bw",
154 {"iface": "eth4", "mac": "90:e2:ba:20:1f:d8"}
157 {"mac": "90:e2:ba:20:1f:d9"}
163 print perf.send_start(**send)
164 print perf._send_processes
167 print perf.send_stop()
168 print perf.receive_stop()
171 if __name__ == "__main__":
172 from vstf.common.log import setup_logging
176 log_file="/var/log/vstf/vstf-netmap.log",