import time
import Queue
import paramiko
-# import subprocess
+import subprocess
# import libvirt
import imp
import random
from jsonschema import validate as js_v, exceptions as js_e
from vim_schema import localinfo_schema, hostinfo_schema
+class RunCommandException(Exception):
+ pass
class host_thread(threading.Thread):
lvirt_module = None
def __init__(self, name, host, user, db, db_lock, test, image_path, host_id, version, develop_mode,
develop_bridge_iface, password=None, keyfile = None, logger_name=None, debug=None):
- '''Init a thread.
- Arguments:
- 'id' number of thead
- 'name' name of thread
- 'host','user': host ip or name to manage and user
- 'db', 'db_lock': database class and lock to use it in exclusion
- '''
+ """Init a thread to communicate with compute node or ovs_controller.
+ :param host_id: host identity
+ :param name: name of the thread
+ :param host: host ip or name to manage and user
+ :param user, password, keyfile: user and credentials to connect to host
+ :param db, db_lock': database class and lock to use it in exclusion
+ """
threading.Thread.__init__(self)
self.name = name
self.host = host
self.db_lock = db_lock
self.test = test
self.password = password
- self.keyfile = keyfile
+ self.keyfile = keyfile
self.localinfo_dirty = False
+ self.connectivity = True
if not test and not host_thread.lvirt_module:
try:
self.version = version
self.xml_level = 0
- #self.pending ={}
+ # self.pending ={}
self.server_status = {} #dictionary with pairs server_uuid:server_status
self.pending_terminate_server =[] #list with pairs (time,server_uuid) time to send a terminate for a server being destroyed
self.queueLock = threading.Lock()
self.taskQueue = Queue.Queue(2000)
self.ssh_conn = None
- self.connectivity = True
+ self.run_command_session = None
+ self.error = None
+ self.localhost = True if host == 'localhost' else False
self.lvirt_conn_uri = "qemu+ssh://{user}@{host}/system?no_tty=1&no_verify=1".format(
user=self.user, host=self.host)
if keyfile:
self.lvirt_conn_uri += "&keyfile=" + keyfile
+ self.remote_ip = None
+ self.local_ip = None
+
+ def run_command(self, command, keep_session=False, ignore_exit_status=False):
+ """Run a command passed as a str on a localhost or at remote machine.
+ :param command: text with the command to execute.
+ :param keep_session: if True it returns a <stdin> for sending input with '<stdin>.write("text\n")'.
+ A command with keep_session=True MUST be followed by a command with keep_session=False in order to
+ close the session and get the output
+ :param ignore_exit_status: Return stdout and not raise an exepction in case of error.
+ :return: the output of the command if 'keep_session=False' or the <stdin> object if 'keep_session=True'
+ :raises: RunCommandException if command fails
+ """
+ if self.run_command_session and keep_session:
+ raise RunCommandException("Internal error. A command with keep_session=True must be followed by another "
+ "command with keep_session=False to close session")
+ try:
+ if self.localhost:
+ if self.run_command_session:
+ p = self.run_command_session
+ self.run_command_session = None
+ (output, outerror) = p.communicate()
+ returncode = p.returncode
+ p.stdin.close()
+ elif keep_session:
+ p = subprocess.Popen(('bash', "-c", command), stdin=subprocess.PIPE, stdout=subprocess.PIPE,
+ stderr=subprocess.PIPE)
+ self.run_command_session = p
+ return p.stdin
+ else:
+ if not ignore_exit_status:
+ output = subprocess.check_output(('bash', "-c", command))
+ returncode = 0
+ else:
+ out = None
+ p = subprocess.Popen(('bash', "-c", command), stdout=subprocess.PIPE)
+ out, err = p.communicate()
+ return out
+ else:
+ if self.run_command_session:
+ (i, o, e) = self.run_command_session
+ self.run_command_session = None
+ i.channel.shutdown_write()
+ else:
+ if not self.ssh_conn:
+ self.ssh_connect()
+ (i, o, e) = self.ssh_conn.exec_command(command, timeout=10)
+ if keep_session:
+ self.run_command_session = (i, o, e)
+ return i
+ returncode = o.channel.recv_exit_status()
+ output = o.read()
+ outerror = e.read()
+ if returncode != 0 and not ignore_exit_status:
+ text = "run_command='{}' Error='{}'".format(command, outerror)
+ self.logger.error(text)
+ raise RunCommandException(text)
+
+ self.logger.debug("run_command='{}' result='{}'".format(command, output))
+ return output
+
+ except RunCommandException:
+ raise
+ except subprocess.CalledProcessError as e:
+ text = "run_command Exception '{}' '{}'".format(str(e), e.output)
+ except (paramiko.ssh_exception.SSHException, Exception) as e:
+ text = "run_command='{}' Exception='{}'".format(command, str(e))
+ self.ssh_conn = None
+ self.run_command_session = None
+ raise RunCommandException(text)
def ssh_connect(self):
try:
self.ssh_conn.set_missing_host_key_policy(paramiko.AutoAddPolicy())
self.ssh_conn.load_system_host_keys()
self.ssh_conn.connect(self.host, username=self.user, password=self.password, key_filename=self.keyfile,
- timeout=10) #, None)
- except paramiko.ssh_exception.SSHException as e:
- text = e.args[0]
- self.logger.error("ssh_connect ssh Exception: " + text)
+ timeout=10) # auth_timeout=10)
+ self.remote_ip = self.ssh_conn.get_transport().sock.getpeername()[0]
+ self.local_ip = self.ssh_conn.get_transport().sock.getsockname()[0]
+ except (paramiko.ssh_exception.SSHException, Exception) as e:
+ text = 'ssh connect Exception: {}'.format(e)
+ self.ssh_conn = None
+ self.error = text
+ raise
def check_connectivity(self):
if not self.test:
-
try:
- if not self.ssh_conn:
- self.ssh_connect()
-
command = 'sudo brctl show'
- (_, stdout, stderr) = self.ssh_conn.exec_command(command, timeout=10)
- content = stderr.read()
- if len(content) > 0:
- self.connectivity = False
- self.logger.error("ssh conection error")
- except paramiko.ssh_exception.SSHException as e:
- text = e.args[0]
+ self.run_command(command)
+ except RunCommandException as e:
self.connectivity = False
- self.logger.error("ssh_connect ssh Exception: " + text)
- raise paramiko.ssh_exception.SSHException("ssh error conection")
- except Exception as e:
- self.connectivity = False
- raise paramiko.ssh_exception.SSHException("ssh error conection")
+ self.logger.error("check_connectivity Exception: " + str(e))
def load_localinfo(self):
if not self.test:
try:
- # Connect SSH
- self.ssh_connect()
-
- command = 'mkdir -p ' + self.image_path
- # print self.name, ': command:', command
- (_, stdout, stderr) = self.ssh_conn.exec_command(command)
- content = stderr.read()
- if len(content) > 0:
- self.logger.error("command: '%s' stderr: '%s'", command, content)
-
- command = 'cat ' + self.image_path + '/.openvim.yaml'
- # print self.name, ': command:', command
- (_, stdout, stderr) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- if len(content) == 0:
- self.logger.error("command: '%s' stderr='%s'", command, stderr.read())
- raise paramiko.ssh_exception.SSHException("Error empty file, command: '{}'".format(command))
- self.localinfo = yaml.load(content)
+ self.run_command('sudo mkdir -p ' + self.image_path)
+ result = self.run_command('cat {}/.openvim.yaml'.format(self.image_path))
+ self.localinfo = yaml.load(result)
js_v(self.localinfo, localinfo_schema)
self.localinfo_dirty = False
if 'server_files' not in self.localinfo:
self.localinfo['server_files'] = {}
- self.logger.debug("localinfo load from host")
+ self.logger.debug("localinfo loaded from host")
return
-
- except paramiko.ssh_exception.SSHException as e:
- text = e.args[0]
- self.logger.error("load_localinfo ssh Exception: " + text)
+ except RunCommandException as e:
+ self.logger.error("load_localinfo Exception: " + str(e))
except host_thread.lvirt_module.libvirtError as e:
text = e.get_error_message()
self.logger.error("load_localinfo libvirt Exception: " + text)
text = str(e)
self.logger.error("load_localinfo Exception: " + text)
- #not loaded, insert a default data and force saving by activating dirty flag
+ # not loaded, insert a default data and force saving by activating dirty flag
self.localinfo = {'files':{}, 'server_files':{} }
- #self.localinfo_dirty=True
+ # self.localinfo_dirty=True
self.localinfo_dirty=False
def load_hostinfo(self):
if self.test:
- return;
+ return
try:
- #Connect SSH
- self.ssh_connect()
-
-
- command = 'cat ' + self.image_path + '/hostinfo.yaml'
- #print self.name, ': command:', command
- (_, stdout, stderr) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- if len(content) == 0:
- self.logger.error("command: '%s' stderr: '%s'", command, stderr.read())
- raise paramiko.ssh_exception.SSHException("Error empty file ")
- self.hostinfo = yaml.load(content)
+ result = self.run_command('cat {}/hostinfo.yaml'.format(self.image_path))
+ self.hostinfo = yaml.load(result)
js_v(self.hostinfo, hostinfo_schema)
- self.logger.debug("hostlinfo load from host " + str(self.hostinfo))
+ self.logger.debug("hostinfo load from host " + str(self.hostinfo))
return
-
- except paramiko.ssh_exception.SSHException as e:
- text = e.args[0]
- self.logger.error("load_hostinfo ssh Exception: " + text)
+ except RunCommandException as e:
+ self.logger.error("load_hostinfo ssh Exception: " + str(e))
except host_thread.lvirt_module.libvirtError as e:
text = e.get_error_message()
self.logger.error("load_hostinfo libvirt Exception: " + text)
except js_e.ValidationError as e:
text = ""
if len(e.path)>0: text=" at '" + ":".join(map(str, e.path))+"'"
- self.logger.error("load_hostinfo format Exception: %s %s", text, e.message)
+ self.logger.error("load_hostinfo format Exception: %s %s", text, str(e))
except Exception as e:
text = str(e)
self.logger.error("load_hostinfo Exception: " + text)
tries-=1
try:
- command = 'cat > ' + self.image_path + '/.openvim.yaml'
- self.logger.debug("command:" + command)
- (stdin, _, _) = self.ssh_conn.exec_command(command)
- yaml.safe_dump(self.localinfo, stdin, explicit_start=True, indent=4, default_flow_style=False, tags=False, encoding='utf-8', allow_unicode=True)
+ command = 'cat > {}/.openvim.yaml'.format(self.image_path)
+ in_stream = self.run_command(command, keep_session=True)
+ yaml.safe_dump(self.localinfo, in_stream, explicit_start=True, indent=4, default_flow_style=False,
+ tags=False, encoding='utf-8', allow_unicode=True)
+ result = self.run_command(command, keep_session=False) # to end session
self.localinfo_dirty = False
break #while tries
- except paramiko.ssh_exception.SSHException as e:
- text = e.args[0]
- self.logger.error("save_localinfo ssh Exception: " + text)
- if "SSH session not active" in text:
- self.ssh_connect()
+ except RunCommandException as e:
+ self.logger.error("save_localinfo ssh Exception: " + str(e))
except host_thread.lvirt_module.libvirtError as e:
text = e.get_error_message()
self.logger.error("save_localinfo libvirt Exception: " + text)
try:
self.logger.debug("deleting file '%s' of unused server '%s'", localfile['source file'], uuid)
self.delete_file(localfile['source file'])
- except paramiko.ssh_exception.SSHException as e:
+ except RunCommandException as e:
self.logger.error("Exception deleting file '%s': %s", localfile['source file'], str(e))
del self.localinfo['server_files'][uuid]
self.localinfo_dirty = True
if self.test or not self.connectivity:
return True
-
try:
- command = 'sudo ovs-vsctl --may-exist add-br br-int -- set Bridge br-int stp_enable=true'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- if len(content) == 0:
- return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
+ self.run_command('sudo ovs-vsctl --may-exist add-br br-int -- set Bridge br-int stp_enable=true')
+
+ return True
+ except RunCommandException as e:
self.logger.error("create_ovs_bridge ssh Exception: " + str(e))
if "SSH session not active" in str(e):
self.ssh_connect()
if self.test or not self.connectivity:
return True
try:
- port_name = 'ovim-' + str(vlan)
- command = 'sudo ovs-vsctl del-port br-int ' + port_name
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- if len(content) == 0:
- return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("delete_port_to_ovs_bridge ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+ port_name = 'ovim-{}'.format(str(vlan))
+ command = 'sudo ovs-vsctl del-port br-int {}'.format(port_name)
+ self.run_command(command)
+ return True
+ except RunCommandException as e:
+ self.logger.error("delete_port_to_ovs_bridge ssh Exception: {}".format(str(e)))
return False
def delete_dhcp_server(self, vlan, net_uuid, dhcp_path):
if not self.is_dhcp_port_free(vlan, net_uuid):
return True
try:
- dhcp_namespace = str(vlan) + '-dnsmasq'
+ dhcp_namespace = '{}-dnsmasq'.format(str(vlan))
dhcp_path = os.path.join(dhcp_path, dhcp_namespace)
pid_file = os.path.join(dhcp_path, 'dnsmasq.pid')
- command = 'sudo ip netns exec ' + dhcp_namespace + ' cat ' + pid_file
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- command = 'sudo ip netns exec ' + dhcp_namespace + ' kill -9 ' + content
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ command = 'sudo ip netns exec {} cat {}'.format(dhcp_namespace, pid_file)
+ content = self.run_command(command, ignore_exit_status=True)
+ dns_pid = content.replace('\n', '')
+ command = 'sudo ip netns exec {} kill -9 {}'.format(dhcp_namespace, dns_pid)
+ self.run_command(command, ignore_exit_status=True)
- # if len(content) == 0:
- # return True
- # else:
- # return False
- except paramiko.ssh_exception.SSHException as e:
+ except RunCommandException as e:
self.logger.error("delete_dhcp_server ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
return False
def is_dhcp_port_free(self, host_id, net_uuid):
if self.test:
return True
try:
- port_name = 'ovim-' + str(vlan)
- command = 'sudo ovs-vsctl add-port br-int ' + port_name + ' tag=' + str(vlan)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- if len(content) == 0:
- return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("add_port_to_ovs_bridge ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+ port_name = 'ovim-{}'.format(str(vlan))
+ command = 'sudo ovs-vsctl add-port br-int {} tag={}'.format(port_name, str(vlan))
+ self.run_command(command)
+ return True
+ except RunCommandException as e:
+ self.logger.error("add_port_to_ovs_bridge Exception: " + str(e))
return False
- def delete_dhcp_port(self, vlan, net_uuid):
+ def delete_dhcp_port(self, vlan, net_uuid, dhcp_path):
"""
Delete from an existing OVS bridge a linux bridge port attached and the linux bridge itself.
:param vlan: segmentation id
if not self.is_dhcp_port_free(vlan, net_uuid):
return True
- self.delete_dhcp_interfaces(vlan)
+ self.delete_dhcp_interfaces(vlan, dhcp_path)
return True
def delete_bridge_port_attached_to_ovs(self, vlan, net_uuid):
if self.test:
return True
try:
- port_name = 'ovim-' + str(vlan)
- command = 'sudo ip link set dev ovim-' + str(vlan) + ' down'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- # content = stdout.read()
- #
- # if len(content) != 0:
- # return False
- command = 'sudo ifconfig ' + port_name + ' down && sudo brctl delbr ' + port_name
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- if len(content) == 0:
- return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("delete_linux_bridge ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+ port_name = 'ovim-{}'.format(str(vlan))
+ command = 'sudo ip link set dev ovim-{} down'.format(str(vlan))
+ self.run_command(command)
+
+ command = 'sudo ip link delete {} && sudo brctl delbr {}'.format(port_name, port_name)
+ self.run_command(command)
+ return True
+ except RunCommandException as e:
+ self.logger.error("delete_linux_bridge Exception: {}".format(str(e)))
return False
def remove_link_bridge_to_ovs(self, vlan, link):
if self.test:
return True
try:
- br_tap_name = str(vlan) + '-vethBO'
- br_ovs_name = str(vlan) + '-vethOB'
+ br_tap_name = '{}-vethBO'.format(str(vlan))
+ br_ovs_name = '{}-vethOB'.format(str(vlan))
# Delete ovs veth pair
command = 'sudo ip link set dev {} down'.format(br_ovs_name)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ self.run_command(command, ignore_exit_status=True)
command = 'sudo ovs-vsctl del-port br-int {}'.format(br_ovs_name)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ self.run_command(command)
# Delete br veth pair
command = 'sudo ip link set dev {} down'.format(br_tap_name)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ self.run_command(command, ignore_exit_status=True)
# Delete br veth interface form bridge
command = 'sudo brctl delif {} {}'.format(link, br_tap_name)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ self.run_command(command)
# Delete br veth pair
command = 'sudo ip link set dev {} down'.format(link)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ self.run_command(command, ignore_exit_status=True)
- if len(content) == 0:
- return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("delete_linux_bridge ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+ return True
+ except RunCommandException as e:
+ self.logger.error("remove_link_bridge_to_ovs Exception: {}".format(str(e)))
return False
def create_ovs_bridge_port(self, vlan):
if self.test:
return True
try:
- port_name = 'ovim-' + str(vlan)
- command = 'sudo brctl show | grep ' + port_name
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # if exist nothing to create
- # if len(content) == 0:
- # return False
-
- command = 'sudo brctl addbr ' + port_name
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # if len(content) == 0:
- # return True
- # else:
- # return False
-
- command = 'sudo brctl stp ' + port_name + ' on'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # if len(content) == 0:
- # return True
- # else:
- # return False
- command = 'sudo ip link set dev ' + port_name + ' up'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- if len(content) == 0:
- return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("create_linux_bridge ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+ port_name = 'ovim-{}'.format(str(vlan))
+ command = 'sudo brctl show | grep {}'.format(port_name)
+ result = self.run_command(command, ignore_exit_status=True)
+ if not result:
+ command = 'sudo brctl addbr {}'.format(port_name)
+ self.run_command(command)
+
+ command = 'sudo brctl stp {} on'.format(port_name)
+ self.run_command(command)
+
+ command = 'sudo ip link set dev {} up'.format(port_name)
+ self.run_command(command)
+ return True
+ except RunCommandException as e:
+ self.logger.error("create_linux_bridge ssh Exception: {}".format(str(e)))
return False
def set_mac_dhcp_server(self, ip, mac, vlan, netmask, first_ip, dhcp_path):
if self.test:
return True
- dhcp_namespace = str(vlan) + '-dnsmasq'
+ dhcp_namespace = '{}-dnsmasq'.format(str(vlan))
dhcp_path = os.path.join(dhcp_path, dhcp_namespace)
dhcp_hostsdir = os.path.join(dhcp_path, dhcp_namespace)
+ ns_interface = '{}-vethDO'.format(str(vlan))
if not ip:
return False
try:
-
- ns_interface = str(vlan) + '-vethDO'
- command = 'sudo ip netns exec ' + dhcp_namespace + ' cat /sys/class/net/{}/address'.format(ns_interface)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- iface_listen_mac = stdout.read()
+ command = 'sudo ip netns exec {} cat /sys/class/net/{}/address'.format(dhcp_namespace, ns_interface)
+ iface_listen_mac = self.run_command(command, ignore_exit_status=True)
if iface_listen_mac > 0:
- command = 'sudo ip netns exec ' + dhcp_namespace + ' cat {} | grep {}'.format(dhcp_hostsdir, dhcp_hostsdir)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- if content > 0:
+ command = 'sudo ip netns exec {} cat {} | grep -i {}'.format(dhcp_namespace,
+ dhcp_hostsdir,
+ iface_listen_mac)
+ content = self.run_command(command, ignore_exit_status=True)
+ if content == '':
ip_data = iface_listen_mac.upper().replace('\n', '') + ',' + first_ip
dhcp_hostsdir = os.path.join(dhcp_path, dhcp_namespace)
- command = 'sudo ip netns exec ' + dhcp_namespace + ' sudo bash -ec "echo ' + ip_data + ' >> ' + dhcp_hostsdir + '"'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
-
- ip_data = mac.upper() + ',' + ip
+ command = 'sudo ip netns exec {} sudo bash -ec "echo {} >> {}"'.format(dhcp_namespace,
+ ip_data,
+ dhcp_hostsdir)
+ self.run_command(command)
- command = 'sudo ip netns exec ' + dhcp_namespace + ' touch ' + dhcp_hostsdir
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ ip_data = mac.upper() + ',' + ip
+ command = 'sudo ip netns exec {} sudo bash -ec "echo {} >> {}"'.format(dhcp_namespace,
+ ip_data,
+ dhcp_hostsdir)
+ self.run_command(command, ignore_exit_status=False)
- command = 'sudo ip netns exec ' + dhcp_namespace + ' sudo bash -ec "echo ' + ip_data + ' >> ' + dhcp_hostsdir + '"'
-
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- if len(content) == 0:
return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
+
+ return False
+ except RunCommandException as e:
self.logger.error("set_mac_dhcp_server ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
return False
def delete_mac_dhcp_server(self, ip, mac, vlan, dhcp_path):
ip_data = mac.upper() + ',' + ip
command = 'sudo ip netns exec ' + dhcp_namespace + ' sudo sed -i \'/' + ip_data + '/d\' ' + dhcp_hostsdir
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ self.run_command(command)
- if len(content) == 0:
- return True
- else:
- return False
-
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("set_mac_dhcp_server ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+ return True
+ except RunCommandException as e:
+ self.logger.error("delete_mac_dhcp_server Exception: " + str(e))
return False
def launch_dhcp_server(self, vlan, ip_range, netmask, dhcp_path, gateway, dns_list=None, routes=None):
leases_path = os.path.join(dhcp_path, "dnsmasq.leases")
pid_file = os.path.join(dhcp_path, 'dnsmasq.pid')
-
dhcp_range = ip_range[0] + ',' + ip_range[1] + ',' + netmask
- command = 'sudo ip netns exec ' + dhcp_namespace + ' mkdir -p ' + dhcp_path
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ command = 'sudo ip netns exec {} mkdir -p {}'.format(dhcp_namespace, dhcp_path)
+ self.run_command(command)
+ # check if dnsmasq process is running
+ dnsmasq_is_runing = False
pid_path = os.path.join(dhcp_path, 'dnsmasq.pid')
- command = 'sudo ip netns exec ' + dhcp_namespace + ' cat ' + pid_path
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ command = 'sudo ip netns exec ' + dhcp_namespace + ' ls ' + pid_path
+ content = self.run_command(command, ignore_exit_status=True)
# check if pid is runing
- pid_status_path = content
if content:
- command = "ps aux | awk '{print $2 }' | grep " + pid_status_path
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ pid_path = content.replace('\n', '')
+ command = "ps aux | awk '{print $2 }' | grep {}" + pid_path
+ dnsmasq_is_runing = self.run_command(command, ignore_exit_status=True)
gateway_option = ' --dhcp-option=3,' + gateway
for dns in dns_list:
dns_data += ',' + dns
- if not content:
+ if not dnsmasq_is_runing:
command = 'sudo ip netns exec ' + dhcp_namespace + ' /usr/sbin/dnsmasq --strict-order --except-interface=lo ' \
'--interface=' + ns_interface + \
' --bind-interfaces --dhcp-hostsdir=' + dhcp_path + \
dhcp_route_option + \
dns_data
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.readline()
-
- if len(content) == 0:
+ self.run_command(command)
return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
+ except RunCommandException as e:
self.logger.error("launch_dhcp_server ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
return False
- def delete_dhcp_interfaces(self, vlan):
+ def delete_dhcp_interfaces(self, vlan, dhcp_path):
"""
- Create a linux bridge with STP active
+ Delete a linux dnsmasq bridge and namespace
:param vlan: netowrk vlan id
+ :param dhcp_path:
:return:
"""
-
if self.test:
return True
try:
- br_veth_name = str(vlan) + '-vethDO'
- ovs_veth_name = str(vlan) + '-vethOD'
- dhcp_namespace = str(vlan) + '-dnsmasq'
+ br_veth_name ='{}-vethDO'.format(str(vlan))
+ ovs_veth_name = '{}-vethOD'.format(str(vlan))
+ dhcp_namespace = '{}-dnsmasq'.format(str(vlan))
- command = 'sudo ovs-vsctl del-port br-int ' + ovs_veth_name
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ dhcp_path = os.path.join(dhcp_path, dhcp_namespace)
+ command = 'sudo ovs-vsctl del-port br-int {}'.format(ovs_veth_name)
+ self.run_command(command, ignore_exit_status=True) # to end session
- command = 'sudo ip netns exec ' + dhcp_namespace + ' ip link set dev ' + br_veth_name + ' down'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ command = 'sudo ip link set dev {} down'.format(ovs_veth_name)
+ self.run_command(command, ignore_exit_status=True) # to end session
- command = 'sudo ip link set dev ' + dhcp_namespace + ' down'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ command = 'sudo ip link delete {} '.format(ovs_veth_name)
+ self.run_command(command, ignore_exit_status=True)
- command = 'sudo brctl delbr ' + dhcp_namespace
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ command = 'sudo ip netns exec {} ip link set dev {} down'.format(dhcp_namespace, br_veth_name)
+ self.run_command(command, ignore_exit_status=True)
- command = 'sudo ip netns del ' + dhcp_namespace
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ command = 'sudo rm -rf {}'.format(dhcp_path)
+ self.run_command(command)
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("delete_dhcp_interfaces ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+ command = 'sudo ip netns del {}'.format(dhcp_namespace)
+ self.run_command(command)
+
+ return True
+ except RunCommandException as e:
+ self.logger.error("delete_dhcp_interfaces ssh Exception: {}".format(str(e)))
return False
def create_dhcp_interfaces(self, vlan, ip_listen_address, netmask):
:param netmask: dhcp net CIDR
:return: True if success
"""
-
if self.test:
return True
try:
- ovs_veth_name = str(vlan) + '-vethOD'
- ns_veth = str(vlan) + '-vethDO'
- dhcp_namespace = str(vlan) + '-dnsmasq'
+ ovs_veth_name = '{}-vethOD'.format(str(vlan))
+ ns_veth = '{}-vethDO'.format(str(vlan))
+ dhcp_namespace = '{}-dnsmasq'.format(str(vlan))
- command = 'sudo ip netns add ' + dhcp_namespace
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- command = 'sudo ip link add ' + ns_veth + ' type veth peer name ' + ovs_veth_name
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- command = 'sudo ip link set ' + ns_veth + ' netns ' + dhcp_namespace
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- command = 'sudo ip netns exec ' + dhcp_namespace + ' ip link set dev ' + ns_veth + ' up'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- command = 'sudo ovs-vsctl add-port br-int ' + ovs_veth_name + ' tag=' + str(vlan)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- command = 'sudo ip link set dev ' + ovs_veth_name + ' up'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- command = 'sudo ip netns exec ' + dhcp_namespace + ' ip link set dev lo up'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- command = 'sudo ip netns exec ' + dhcp_namespace + ' ' + ' ifconfig ' + ns_veth \
- + ' ' + ip_listen_address + ' netmask ' + netmask
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- if len(content) == 0:
- return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("create_dhcp_interfaces ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+ command = 'sudo ip netns add {}'.format(dhcp_namespace)
+ self.run_command(command)
+
+ command = 'sudo ip link add {} type veth peer name {}'.format(ns_veth, ovs_veth_name)
+ self.run_command(command)
+
+ command = 'sudo ip link set {} netns {}'.format(ns_veth, dhcp_namespace)
+ self.run_command(command)
+
+ command = 'sudo ip netns exec {} ip link set dev {} up'.format(dhcp_namespace, ns_veth)
+ self.run_command(command)
+
+ command = 'sudo ovs-vsctl add-port br-int {} tag={}'.format(ovs_veth_name, str(vlan))
+ self.run_command(command, ignore_exit_status=True)
+
+ command = 'sudo ip link set dev {} up'.format(ovs_veth_name)
+ self.run_command(command)
+
+ command = 'sudo ip netns exec {} ip link set dev lo up'.format(dhcp_namespace)
+ self.run_command(command)
+
+ command = 'sudo ip netns exec {} ifconfig {} {} netmask {}'.format(dhcp_namespace,
+ ns_veth,
+ ip_listen_address,
+ netmask)
+ self.run_command(command)
+ return True
+ except RunCommandException as e:
+ self.logger.error("create_dhcp_interfaces ssh Exception: {}".format(str(e)))
return False
def delete_qrouter_connection(self, vlan, link):
:return:
"""
- ns_qouter = str(vlan) + '-qrouter'
- qrouter_ovs_veth = str(vlan) + '-vethOQ'
- qrouter_ns_veth = str(vlan) + '-vethQO'
-
- qrouter_br_veth = str(vlan) + '-vethBQ'
- qrouter_ns_router_veth = str(vlan) + '-vethQB'
-
- # delete ovs veth to ovs br-int
- command = 'sudo ovs-vsctl del-port br-int {}'.format(qrouter_ovs_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # down ns veth
- command = 'sudo ip netns exec {} ip link set dev {} down'.format(ns_qouter, qrouter_ns_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # down ovs veth interface
- command = 'sudo ip link set dev {} down'.format(qrouter_br_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # down br veth interface
- command = 'sudo ip link set dev {} down'.format(qrouter_ovs_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # down br veth interface
- command = 'sudo ip link set dev {} down'.format(qrouter_ns_router_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # down br veth interface
- command = 'sudo brctl delif {} {}'.format(link, qrouter_br_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
-
- # delete NS
- command = 'sudo ip netns del ' + ns_qouter
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ if self.test:
+ return True
+ try:
+ ns_qouter = '{}-qrouter'.format(str(vlan))
+ qrouter_ovs_veth = '{}-vethOQ'.format(str(vlan))
+ qrouter_ns_veth = '{}-vethQO'.format(str(vlan))
+ qrouter_br_veth = '{}-vethBQ'.format(str(vlan))
+ qrouter_ns_router_veth = '{}-vethQB'.format(str(vlan))
+
+ command = 'sudo ovs-vsctl del-port br-int {}'.format(qrouter_ovs_veth)
+ self.run_command(command, ignore_exit_status=True)
+
+ # down ns veth
+ command = 'sudo ip netns exec {} ip link set dev {} down'.format(ns_qouter, qrouter_ns_veth)
+ self.run_command(command, ignore_exit_status=True)
+
+ command = 'sudo ip netns exec {} ip link delete {} '.format(ns_qouter, qrouter_ns_veth)
+ self.run_command(command, ignore_exit_status=True)
+
+ command = 'sudo ip netns del ' + ns_qouter
+ self.run_command(command)
+
+ # down ovs veth interface
+ command = 'sudo ip link set dev {} down'.format(qrouter_br_veth)
+ self.run_command(command, ignore_exit_status=True)
+
+ # down br veth interface
+ command = 'sudo ip link set dev {} down'.format(qrouter_ovs_veth)
+ self.run_command(command, ignore_exit_status=True)
+
+ # delete veth interface
+ command = 'sudo ip link delete {} '.format(link, qrouter_ovs_veth)
+ self.run_command(command, ignore_exit_status=True)
+
+ # down br veth interface
+ command = 'sudo ip link set dev {} down'.format(qrouter_ns_router_veth)
+ self.run_command(command, ignore_exit_status=True)
+
+ # delete veth interface
+ command = 'sudo ip link delete {} '.format(link, qrouter_ns_router_veth)
+ self.run_command(command, ignore_exit_status=True)
+
+ # down br veth interface
+ command = 'sudo brctl delif {} {}'.format(link, qrouter_br_veth)
+ self.run_command(command)
+
+ # delete NS
+ return True
+ except RunCommandException as e:
+ self.logger.error("delete_qrouter_connection ssh Exception: {}".format(str(e)))
+ return False
def create_qrouter_ovs_connection(self, vlan, gateway, dhcp_cidr):
"""
:return:
"""
- ns_qouter = str(vlan) + '-qrouter'
- qrouter_ovs_veth = str(vlan) + '-vethOQ'
- qrouter_ns_veth = str(vlan) + '-vethQO'
-
- # Create NS
- command = 'sudo ip netns add ' + ns_qouter
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # Create pait veth
- command = 'sudo ip link add {} type veth peer name {}'.format(qrouter_ns_veth, qrouter_ovs_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # up ovs veth interface
- command = 'sudo ip link set dev {} up'.format(qrouter_ovs_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # add ovs veth to ovs br-int
- command = 'sudo ovs-vsctl add-port br-int {} tag={}'.format(qrouter_ovs_veth, vlan)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # add veth to ns
- command = 'sudo ip link set {} netns {}'.format(qrouter_ns_veth, ns_qouter)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # up ns loopback
- command = 'sudo ip netns exec {} ip link set dev lo up'.format(ns_qouter)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # up ns veth
- command = 'sudo ip netns exec {} ip link set dev {} up'.format(ns_qouter, qrouter_ns_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- from netaddr import IPNetwork
- ip_tools = IPNetwork(dhcp_cidr)
- cidr_len = ip_tools.prefixlen
-
- # set gw to ns veth
- command = 'sudo ip netns exec {} ip address add {}/{} dev {}'.format(ns_qouter, gateway, cidr_len, qrouter_ns_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ if self.test:
+ return True
- def add_ns_routes(self, vlan, routes):
+ try:
+ ns_qouter = '{}-qrouter'.format(str(vlan))
+ qrouter_ovs_veth ='{}-vethOQ'.format(str(vlan))
+ qrouter_ns_veth = '{}-vethQO'.format(str(vlan))
+
+ # Create NS
+ command = 'sudo ip netns add {}'.format(ns_qouter)
+ self.run_command(command)
+
+ # Create pait veth
+ command = 'sudo ip link add {} type veth peer name {}'.format(qrouter_ns_veth, qrouter_ovs_veth)
+ self.run_command(command, ignore_exit_status=True)
+
+ # up ovs veth interface
+ command = 'sudo ip link set dev {} up'.format(qrouter_ovs_veth)
+ self.run_command(command)
+
+ # add ovs veth to ovs br-int
+ command = 'sudo ovs-vsctl add-port br-int {} tag={}'.format(qrouter_ovs_veth, vlan)
+ self.run_command(command)
+
+ # add veth to ns
+ command = 'sudo ip link set {} netns {}'.format(qrouter_ns_veth, ns_qouter)
+ self.run_command(command)
+
+ # up ns loopback
+ command = 'sudo ip netns exec {} ip link set dev lo up'.format(ns_qouter)
+ self.run_command(command)
- for key, value in routes.iteritems():
- ns_qouter = str(vlan) + '-qrouter'
- qrouter_ns_router_veth = str(vlan) + '-vethQB'
# up ns veth
- if key == 'default':
- command = 'sudo ip netns exec {} ip route add {} via {} '.format(ns_qouter, key, value)
- else:
- command = 'sudo ip netns exec {} ip route add {} via {} dev {}'.format(ns_qouter, key, value,
- qrouter_ns_router_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ command = 'sudo ip netns exec {} ip link set dev {} up'.format(ns_qouter, qrouter_ns_veth)
+ self.run_command(command)
+
+ from netaddr import IPNetwork
+ ip_tools = IPNetwork(dhcp_cidr)
+ cidr_len = ip_tools.prefixlen
+
+ # set gw to ns veth
+ command = 'sudo ip netns exec {} ip address add {}/{} dev {}'.format(ns_qouter, gateway, cidr_len, qrouter_ns_veth)
+ self.run_command(command)
+
+ return True
+
+ except RunCommandException as e:
+ self.logger.error("Create_dhcp_interfaces ssh Exception: {}".format(str(e)))
+ return False
+
+ def add_ns_routes(self, vlan, routes):
+ """
+
+ :param vlan:
+ :param routes:
+ :return:
+ """
+
+ if self.test:
+ return True
+
+ try:
+ ns_qouter = '{}-qrouter'.format(str(vlan))
+ qrouter_ns_router_veth = '{}-vethQB'.format(str(vlan))
+
+ for key, value in routes.iteritems():
+ # up ns veth
+ if key == 'default':
+ command = 'sudo ip netns exec {} ip route add {} via {} '.format(ns_qouter, key, value)
+ else:
+ command = 'sudo ip netns exec {} ip route add {} via {} dev {}'.format(ns_qouter, key, value,
+ qrouter_ns_router_veth)
+
+ self.run_command(command)
+
+ return True
+
+ except RunCommandException as e:
+ self.logger.error("add_ns_routes, error adding routes to namesapce, {}".format(str(e)))
+ return False
def create_qrouter_br_connection(self, vlan, cidr, link):
"""
:return:
"""
- ns_qouter = str(vlan) + '-qrouter'
- qrouter_ns_router_veth = str(vlan) + '-vethQB'
- qrouter_br_veth = str(vlan) + '-vethBQ'
-
- # Create pait veth
- command = 'sudo ip link add {} type veth peer name {}'.format(qrouter_br_veth, qrouter_ns_router_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # up ovs veth interface
- command = 'sudo ip link set dev {} up'.format(qrouter_br_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # add veth to ns
- command = 'sudo ip link set {} netns {}'.format(qrouter_ns_router_veth, ns_qouter)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- # up ns veth
- command = 'sudo ip netns exec {} ip link set dev {} up'.format(ns_qouter, qrouter_ns_router_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- command = 'sudo ip netns exec {} ip address add {} dev {}'.format(ns_qouter, link['nat'], qrouter_ns_router_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- command = 'sudo brctl show | grep {}'.format(link['iface'])
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- if content > '':
- # up ns veth
- command = 'sudo brctl addif {} {}'.format(link['iface'], qrouter_br_veth)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ if self.test:
+ return True
- # up ns veth
- command = 'sudo ip netns exec {} iptables -t nat -A POSTROUTING -o {} -s {} -d {} -j MASQUERADE' \
- .format(ns_qouter, qrouter_ns_router_veth, link['nat'], cidr)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ try:
+ ns_qouter = '{}-qrouter'.format(str(vlan))
+ qrouter_ns_router_veth = '{}-vethQB'.format(str(vlan))
+ qrouter_br_veth = '{}-vethBQ'.format(str(vlan))
+ command = 'sudo brctl show | grep {}'.format(link['iface'])
+ content = self.run_command(command, ignore_exit_status=True)
- else:
- self.logger.error('Bridge {} given by user not exist'.format(qrouter_br_veth))
+ if content > '':
+ # Create pait veth
+ command = 'sudo ip link add {} type veth peer name {}'.format(qrouter_br_veth, qrouter_ns_router_veth)
+ self.run_command(command)
+
+ # up ovs veth interface
+ command = 'sudo ip link set dev {} up'.format(qrouter_br_veth)
+ self.run_command(command)
+
+ # add veth to ns
+ command = 'sudo ip link set {} netns {}'.format(qrouter_ns_router_veth, ns_qouter)
+ self.run_command(command)
+
+ # up ns veth
+ command = 'sudo ip netns exec {} ip link set dev {} up'.format(ns_qouter, qrouter_ns_router_veth)
+ self.run_command(command)
+
+ command = 'sudo ip netns exec {} ip address add {} dev {}'.format(ns_qouter,
+ link['nat'],
+ qrouter_ns_router_veth)
+ self.run_command(command)
+
+ # up ns veth
+ command = 'sudo brctl addif {} {}'.format(link['iface'], qrouter_br_veth)
+ self.run_command(command)
+
+ # up ns veth
+ command = 'sudo ip netns exec {} iptables -t nat -A POSTROUTING -o {} -s {} -d {} -j MASQUERADE' \
+ .format(ns_qouter, qrouter_ns_router_veth, link['nat'], cidr)
+ self.run_command(command)
+
+ return True
+ else:
+ self.logger.error('create_qrouter_br_connection, Bridge {} given by user not exist'.format(qrouter_br_veth))
+ return False
+ except RunCommandException as e:
+ self.logger.error("Error creating qrouter, {}".format(str(e)))
+ return False
def create_link_bridge_to_ovs(self, vlan, link):
"""
return True
try:
- br_tap_name = str(vlan) + '-vethBO'
- br_ovs_name = str(vlan) + '-vethOB'
+ br_tap_name = '{}-vethBO'.format(str(vlan))
+ br_ovs_name = '{}-vethOB'.format(str(vlan))
# is a bridge or a interface
command = 'sudo brctl show | grep {}'.format(link)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
+ content = self.run_command(command, ignore_exit_status=True)
if content > '':
command = 'sudo ip link add {} type veth peer name {}'.format(br_tap_name, br_ovs_name)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ self.run_command(command)
command = 'sudo ip link set dev {} up'.format(br_tap_name)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ self.run_command(command)
command = 'sudo ip link set dev {} up'.format(br_ovs_name)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ self.run_command(command)
command = 'sudo ovs-vsctl add-port br-int {} tag={}'.format(br_ovs_name, str(vlan))
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
+ self.run_command(command)
command = 'sudo brctl addif ' + link + ' {}'.format(br_tap_name)
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
-
- if len(content) == 0:
- return True
- else:
- return False
+ self.run_command(command)
+ return True
else:
self.logger.error('Link is not present, please check {}'.format(link))
return False
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("create_dhcp_interfaces ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+
+ except RunCommandException as e:
+ self.logger.error("create_link_bridge_to_ovs, Error creating link to ovs, {}".format(str(e)))
return False
def create_ovs_vxlan_tunnel(self, vxlan_interface, remote_ip):
"""
if self.test or not self.connectivity:
return True
+ if remote_ip == 'localhost':
+ if self.localhost:
+ return True # TODO: Cannot create a vxlan between localhost and localhost
+ remote_ip = self.local_ip
try:
- command = 'sudo ovs-vsctl add-port br-int ' + vxlan_interface + \
- ' -- set Interface ' + vxlan_interface + ' type=vxlan options:remote_ip=' + remote_ip + \
- ' -- set Port ' + vxlan_interface + ' other_config:stp-path-cost=10'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- # print content
- if len(content) == 0:
- return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("create_ovs_vxlan_tunnel ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+
+ command = 'sudo ovs-vsctl add-port br-int {} -- set Interface {} type=vxlan options:remote_ip={} ' \
+ '-- set Port {} other_config:stp-path-cost=10'.format(vxlan_interface,
+ vxlan_interface,
+ remote_ip,
+ vxlan_interface)
+ self.run_command(command)
+ return True
+ except RunCommandException as e:
+ self.logger.error("create_ovs_vxlan_tunnel, error creating vxlan tunnel, {}".format(str(e)))
return False
def delete_ovs_vxlan_tunnel(self, vxlan_interface):
if self.test or not self.connectivity:
return True
try:
- command = 'sudo ovs-vsctl del-port br-int ' + vxlan_interface
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- # print content
- if len(content) == 0:
- return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("delete_ovs_vxlan_tunnel ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+ command = 'sudo ovs-vsctl del-port br-int {}'.format(vxlan_interface)
+ self.run_command(command)
+ return True
+ except RunCommandException as e:
+ self.logger.error("delete_ovs_vxlan_tunnel, error deleting vxlan tunenl, {}".format(str(e)))
return False
def delete_ovs_bridge(self):
return True
try:
command = 'sudo ovs-vsctl del-br br-int'
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- if len(content) == 0:
- return True
- else:
- return False
- except paramiko.ssh_exception.SSHException as e:
- self.logger.error("delete_ovs_bridge ssh Exception: " + str(e))
- if "SSH session not active" in str(e):
- self.ssh_connect()
+ self.run_command(command)
+ return True
+ except RunCommandException as e:
+ self.logger.error("delete_ovs_bridge ssh Exception: {}".format(str(e)))
return False
def get_file_info(self, path):
command = 'ls -lL --time-style=+%Y-%m-%dT%H:%M:%S ' + path
- self.logger.debug("command: " + command)
- (_, stdout, _) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- if len(content) == 0:
- return None # file does not exist
- else:
+ try:
+ content = self.run_command(command)
return content.split(" ") # (permission, 1, owner, group, size, date, file)
+ except RunCommandException as e:
+ return None # file does not exist
def qemu_get_info(self, path):
command = 'qemu-img info ' + path
- self.logger.debug("command: " + command)
- (_, stdout, stderr) = self.ssh_conn.exec_command(command)
- content = stdout.read()
- if len(content) == 0:
- error = stderr.read()
- self.logger.error("get_qemu_info error " + error)
- raise paramiko.ssh_exception.SSHException("Error getting qemu_info: " + error)
- else:
- try:
- return yaml.load(content)
- except yaml.YAMLError as exc:
- text = ""
- if hasattr(exc, 'problem_mark'):
- mark = exc.problem_mark
- text = " at position: (%s:%s)" % (mark.line+1, mark.column+1)
- self.logger.error("get_qemu_info yaml format Exception " + text)
- raise paramiko.ssh_exception.SSHException("Error getting qemu_info yaml format" + text)
+ content = self.run_command(command)
+ try:
+ return yaml.load(content)
+ except yaml.YAMLError as exc:
+ text = ""
+ if hasattr(exc, 'problem_mark'):
+ mark = exc.problem_mark
+ text = " at position: (%s:%s)" % (mark.line+1, mark.column+1)
+ self.logger.error("get_qemu_info yaml format Exception " + text)
+ raise RunCommandException("Error getting qemu_info yaml format" + text)
def qemu_change_backing(self, inc_file, new_backing_file):
- command = 'qemu-img rebase -u -b ' + new_backing_file + ' ' + inc_file
- self.logger.debug("command: " + command)
- (_, _, stderr) = self.ssh_conn.exec_command(command)
- content = stderr.read()
- if len(content) == 0:
+ command = 'qemu-img rebase -u -b {} {}'.format(new_backing_file, inc_file)
+ try:
+ self.run_command(command)
return 0
- else:
- self.logger.error("qemu_change_backing error: " + content)
+ except RunCommandException as e:
+ self.logger.error("qemu_change_backing error: " + str(e))
return -1
def qemu_create_empty_disk(self, dev):
empty_disk_path = dev['source file']
- command = 'qemu-img create -f qcow2 ' + empty_disk_path + ' ' + str(dev['image_size']) + 'G'
- self.logger.debug("command: " + command)
- (_, _, stderr) = self.ssh_conn.exec_command(command)
- content = stderr.read()
- if len(content) == 0:
+ command = 'qemu-img create -f qcow2 {} {}G'.format(empty_disk_path, dev['image_size'])
+ try:
+ self.run_command(command)
return 0
- else:
- self.logger.error("qemu_create_empty_disk error: " + content)
+ except RunCommandException as e:
+ self.logger.error("qemu_create_empty_disk error: " + str(e))
return -1
def get_notused_filename(self, proposed_name, suffix=''):
def delete_file(self, file_name):
- command = 'rm -f '+file_name
- self.logger.debug("command: " + command)
- (_, _, stderr) = self.ssh_conn.exec_command(command)
- error_msg = stderr.read()
- if len(error_msg) > 0:
- raise paramiko.ssh_exception.SSHException("Error deleting file: " + error_msg)
+ command = 'rm -f ' + file_name
+ self.run_command(command)
def copy_file(self, source, destination, perserve_time=True):
if source[0:4]=="http":
command = 'cp --no-preserve=mode'
if perserve_time:
command += ' --preserve=timestamps'
- command += " '{}' '{}'".format(source, destination)
- self.logger.debug("command: " + command)
- (_, _, stderr) = self.ssh_conn.exec_command(command)
- error_msg = stderr.read()
- if len(error_msg) > 0:
- raise paramiko.ssh_exception.SSHException("Error copying image to local host: " + error_msg)
+ command += " '{}' '{}'".format(source, destination)
+ self.run_command(command)
def copy_remote_file(self, remote_file, use_incremental):
''' Copy a file from the repository to local folder and recursively
#create incremental image
if use_incremental_image:
local_file_inc = self.get_notused_filename(local_file, '.inc')
- command = 'qemu-img create -f qcow2 '+local_file_inc+ ' -o backing_file='+ local_file
- self.logger.debug("command: " + command)
- (_, _, stderr) = self.ssh_conn.exec_command(command)
- error_msg = stderr.read()
- if len(error_msg) > 0:
- raise paramiko.ssh_exception.SSHException("Error creating incremental file: " + error_msg)
+ command = 'qemu-img create -f qcow2 {} -o backing_file={}'.format(local_file_inc, local_file)
+ self.run_command(command)
local_file = local_file_inc
- qemu_info = {'file format':'qcow2'}
+ qemu_info = {'file format': 'qcow2'}
server_host_files[ dev['image_id'] ] = {'source file': local_file, 'file format': qemu_info['file format']}