def teardown_iperf_server(self, pid):
iperf = iperf_driver.IPerfDriver()
try:
iperf.stop_server(pid)
return agent_utils.make_response(code=0)
except Exception as e:
message = e.message
return agent_utils.make_response(code=1, message=message)
def teardown_iperf_server(self, pid):
iperf = iperf_driver.IPerfDriver()
try:
iperf.stop_server(pid)
return agent_utils.make_response(code=0)
except Exception as e:
message = e.message
return agent_utils.make_response(code=1, message=message)
def start_iperf_client(self, host, protocol="TCP", timeout=5, parallel=None, bandwidth=None):
iperf = iperf_driver.IPerfDriver()
try:
data = iperf.start_client(host, protocol="TCP", timeout=5, parallel=None, bandwidth=None)
data["server_ip"] = host
return agent_utils.make_response(code=0, data=data)
except Exception as e:
message = e.message
return agent_utils.make_response(code=1, message=message)
def setup_iperf_server(self, protocol='TCP', port=5001, window=None):
"""iperf -s
"""
iperf = iperf_driver.IPerfDriver()
try:
data = iperf.start_server(protocol='TCP', port=5001, window=None)
return agent_utils.make_response(code=0, data=data)
except:
message = 'Start iperf server failed!'
return agent_utils.make_response(code=1, message=message)
def setup_iperf_server(self, protocol='TCP', port=5001, window=None):
"""iperf -s
"""
iperf = iperf_driver.IPerfDriver()
try:
data = iperf.start_server(protocol='TCP', port=5001, window=None)
return agent_utils.make_response(code=0, data=data)
except:
message = 'Start iperf server failed!'
return agent_utils.make_response(code=1, message=message)
def teardown_link(self, interface):
"""ip link
"""
cmd = ['ip', 'link', 'delete', interface]
stdcode, stdout = agent_utils.execute(cmd, root=True)
if stdcode == 0:
return agent_utils.make_response(code=stdcode)
# execute failed.
message = stdout.pop(0)
return agent_utils.make_response(code=stdcode, message=message)
def teardown_link(self, interface):
"""ip link
"""
cmd = ['ip', 'link', 'delete', interface]
stdcode, stdout = agent_utils.execute(cmd, root=True)
if stdcode == 0:
return agent_utils.make_response(code=stdcode)
# execute failed.
message = stdout.pop(0)
return agent_utils.make_response(code=stdcode, message=message)
def add_vlan_to_interface(self, interface, vlan_id):
"""Add vlan interface.
ip link add link eth0 name eth0.10 type vlan id 10
"""
subif = "%s.%s" % (interface, vlan_id)
vlan_id = "%s" % vlan_id
cmd = ["ip", "link", "add", "link", interface, "name", subif, "type", "vlan", "id", vlan_id]
stdcode, stdout = agent_utils.execute(cmd, root=True)
if stdcode == 0:
return agent_utils.make_response(code=stdcode)
# execute failed.
message = stdout.pop(0)
return agent_utils.make_response(code=stdcode, message=message)
def add_vlan_to_interface(self, interface, vlan_id):
"""Add vlan interface.
ip link add link eth0 name eth0.10 type vlan id 10
"""
subif = '%s.%s' % (interface, vlan_id)
vlan_id = '%s' % vlan_id
cmd = ['ip', 'link', 'add', 'link', interface, 'name',
subif, 'type', 'vlan', 'id', vlan_id]
stdcode, stdout = agent_utils.execute(cmd, root=True)
if stdcode == 0:
return agent_utils.make_response(code=stdcode)
# execute failed.
message = stdout.pop(0)
return agent_utils.make_response(code=stdcode, message=message)
def test_make_response(self):
para = dict()
para["code"] = 0
para["message"] = "Test make response."
para["data"] = dict()
result = utils.make_response(para["code"], para["message"], para["data"])
self.assertEqual(para, result)
def test_make_response(self):
para = dict()
para['code'] = 0
para['message'] = 'Test make response.'
para['data'] = dict()
result = utils.make_response(para['code'], para['message'],
para['data'])
self.assertEqual(para, result)
def test_make_response(self):
para = dict()
para['code'] = 0
para['message'] = 'Test make response.'
para['data'] = dict()
result = utils.make_response(para['code'], para['message'],
para['data'])
self.assertEqual(para, result)
def get_interface(self, interface='eth0'):
"""Interface info.
ifconfig interface
"""
LOG.info("RPC: get_interface interfae: %s" % interface)
code, message, data = agent_utils.get_interface(interface)
return agent_utils.make_response(code, message, data)
def get_interface(self, interface='eth0'):
"""Interface info.
ifconfig interface
"""
LOG.info("RPC: get_interface interfae: %s" % interface)
code, message, data = agent_utils.get_interface(interface)
return agent_utils.make_response(code, message, data)
def add_vlan_to_interface(self, interface, vlan_id):
"""Add vlan interface.
ip link add link eth0 name eth0.10 type vlan id 10
"""
subif = '%s.%s' % (interface, vlan_id)
vlan_id = '%s' % vlan_id
cmd = [
'ip', 'link', 'add', 'link', interface, 'name', subif, 'type',
'vlan', 'id', vlan_id
]
stdcode, stdout = agent_utils.execute(cmd, root=True)
if stdcode == 0:
return agent_utils.make_response(code=stdcode)
# execute failed.
message = stdout.pop(0)
return agent_utils.make_response(code=stdcode, message=message)
def start_iperf_client(self,
host,
protocol='TCP',
timeout=5,
parallel=None,
bandwidth=None):
iperf = iperf_driver.IPerfDriver()
try:
data = iperf.start_client(host,
protocol='TCP',
timeout=5,
parallel=None,
bandwidth=None)
data['server_ip'] = host
return agent_utils.make_response(code=0, data=data)
except Exception as e:
message = e.message
return agent_utils.make_response(code=1, message=message)
def check_ports_on_br(self, bridge="br-ex", ports=["eth3"]):
"""Check ports exist on bridge.
ovs-vsctl list-ports bridge
"""
LOG.info("RPC: check_ports_on_br bridge: %s, ports: %s" % (bridge, ports))
cmd = ["ovs-vsctl", "list-ports", bridge]
stdcode, stdout = agent_utils.execute(cmd, root=True)
data = dict()
if stdcode == 0:
for port in ports:
if port in stdout:
data[port] = True
stdout.remove(port)
else:
data[port] = False
return agent_utils.make_response(code=stdcode, data=data)
# execute failed.
message = stdout.pop(0)
return agent_utils.make_response(code=stdcode, message=message)
def setup_link(self, interface, cidr):
"""Setup a link.
ip addr add dev interface
ip link set dev interface up
"""
# clear old ipaddr in interface
cmd = ["ip", "addr", "flush", "dev", interface]
agent_utils.execute(cmd, root=True)
ip = IPNetwork(cidr)
cmd = ["ip", "addr", "add", cidr, "broadcast", str(ip.broadcast), "dev", interface]
stdcode, stdout = agent_utils.execute(cmd, root=True)
if stdcode == 0:
cmd = ["ip", "link", "set", "dev", interface, "up"]
stdcode, stdout = agent_utils.execute(cmd, root=True)
if stdcode == 0:
return agent_utils.make_response(code=stdcode)
# execute failed.
message = stdout.pop(0)
return agent_utils.make_response(code=stdcode, message=message)
def check_ports_on_br(self, bridge='br-ex', ports=['eth3']):
"""Check ports exist on bridge.
ovs-vsctl list-ports bridge
"""
LOG.info("RPC: check_ports_on_br bridge: %s, ports: %s" %
(bridge, ports))
cmd = ['ovs-vsctl', 'list-ports', bridge]
stdcode, stdout = agent_utils.execute(cmd, root=True)
data = dict()
if stdcode == 0:
for port in ports:
if port in stdout:
data[port] = True
stdout.remove(port)
else:
data[port] = False
return agent_utils.make_response(code=stdcode, data=data)
# execute failed.
message = stdout.pop(0)
return agent_utils.make_response(code=stdcode, message=message)
def setup_link(self, interface, cidr):
"""Setup a link.
ip addr add dev interface
ip link set dev interface up
"""
# clear old ipaddr in interface
cmd = ['ip', 'addr', 'flush', 'dev', interface]
agent_utils.execute(cmd, root=True)
ip = IPNetwork(cidr)
cmd = ['ip', 'addr', 'add', cidr, 'broadcast',
str(ip.broadcast), 'dev', interface]
stdcode, stdout = agent_utils.execute(cmd, root=True)
if stdcode == 0:
cmd = ['ip', 'link', 'set', 'dev', interface, 'up']
stdcode, stdout = agent_utils.execute(cmd, root=True)
if stdcode == 0:
return agent_utils.make_response(code=stdcode)
# execute failed.
message = stdout.pop(0)
return agent_utils.make_response(code=stdcode, message=message)
def ping(self, ips, boardcast=False, count=2, timeout=2, interface=None):
"""Ping host or broadcast.
ping host -c 2 -W 2
"""
cmd = ['ping', '-c', str(count), '-W', str(timeout)]
True if not interface else cmd.extend(['-I', interface])
True if not boardcast else cmd.append('-b')
# Batch create subprocess
data = dict()
try:
for ip in ips:
stdcode, stdout = agent_utils.execute(cmd + [ip])
if stdcode:
data[ip] = 100
else:
pattern = r',\s([0-9]+)%\spacket\sloss'
data[ip] = re.search(pattern, stdout[-2]).groups()[0]
return agent_utils.make_response(code=0, data=data)
except Exception as e:
message = e.message
return agent_utils.make_response(code=1, message=message)
def ping(self, ips, boardcast=False, count=2, timeout=2, interface=None):
"""Ping host or broadcast.
ping host -c 2 -W 2
"""
cmd = ["ping", "-c", str(count), "-W", str(timeout)]
True if not interface else cmd.extend(["-I", interface])
True if not boardcast else cmd.append("-b")
# Batch create subprocess
data = dict()
try:
for ip in ips:
stdcode, stdout = agent_utils.execute(cmd + [ip])
if stdcode:
data[ip] = 100
else:
pattern = r",\s([0-9]+)%\spacket\sloss"
data[ip] = re.search(pattern, stdout[-2]).groups()[0]
return agent_utils.make_response(code=0, data=data)
except Exception as e:
message = e.message
return agent_utils.make_response(code=1, message=message)
def setup_link(self, interface, cidr):
"""Setup a link.
ip addr add dev interface
ip link set dev interface up
"""
# clear old ipaddr in interface
cmd = ['ip', 'addr', 'flush', 'dev', interface]
agent_utils.execute(cmd, root=True)
ip = IPNetwork(cidr)
cmd = [
'ip', 'addr', 'add', cidr, 'broadcast',
str(ip.broadcast), 'dev', interface
]
stdcode, stdout = agent_utils.execute(cmd, root=True)
if stdcode == 0:
cmd = ['ip', 'link', 'set', 'dev', interface, 'up']
stdcode, stdout = agent_utils.execute(cmd, root=True)
if stdcode == 0:
return agent_utils.make_response(code=stdcode)
# execute failed.
message = stdout.pop(0)
return agent_utils.make_response(code=stdcode, message=message)
