mirror of
https://github.com/retspen/webvirtcloud
synced 2024-12-25 15:45:23 +00:00
162 lines
5.5 KiB
Python
162 lines
5.5 KiB
Python
from xml.etree import ElementTree
|
|
|
|
from libvirt import VIR_INTERFACE_XML_INACTIVE
|
|
from vrtManager import util
|
|
from vrtManager.connection import wvmConnect
|
|
|
|
|
|
class wvmInterfaces(wvmConnect):
|
|
|
|
def define_iface(self, xml, flag=0):
|
|
self.wvm.interfaceDefineXML(xml, flag)
|
|
|
|
def create_iface(
|
|
self, name, itype, mode, netdev, ipv4_type, ipv4_addr, ipv4_gw, ipv6_type, ipv6_addr, ipv6_gw, stp, delay
|
|
):
|
|
xml = f"""<interface type='{itype}' name='{name}'>
|
|
<start mode='{mode}'/>"""
|
|
if ipv4_type == "dhcp":
|
|
xml += """<protocol family='ipv4'>
|
|
<dhcp/>
|
|
</protocol>"""
|
|
if ipv4_type == "static":
|
|
address, prefix = ipv4_addr.split("/")
|
|
xml += f"""<protocol family='ipv4'>
|
|
<ip address='{address}' prefix='{prefix}'/>
|
|
<route gateway='{ipv4_gw}'/>
|
|
</protocol>"""
|
|
if ipv6_type == "dhcp":
|
|
xml += """<protocol family='ipv6'>
|
|
<dhcp/>
|
|
</protocol>"""
|
|
if ipv6_type == "static":
|
|
address, prefix = ipv6_addr.split("/")
|
|
xml += f"""<protocol family='ipv6'>
|
|
<ip address='{address}' prefix='{prefix}'/>
|
|
<route gateway='{ipv6_gw}'/>
|
|
</protocol>"""
|
|
if itype == "bridge":
|
|
xml += f"""<bridge stp='{stp}' delay='{delay}'>
|
|
<interface name='{netdev}' type='ethernet'/>
|
|
</bridge>"""
|
|
xml += """</interface>"""
|
|
self.define_iface(xml)
|
|
iface = self.get_iface(name)
|
|
iface.create()
|
|
|
|
|
|
class wvmInterface(wvmConnect):
|
|
def __init__(self, host, login, passwd, conn, iface):
|
|
wvmConnect.__init__(self, host, login, passwd, conn)
|
|
self.iface = self.get_iface(iface)
|
|
|
|
def _XMLDesc(self, flags=0):
|
|
return self.iface.XMLDesc(flags)
|
|
|
|
def get_start_mode(self):
|
|
try:
|
|
xml = self._XMLDesc(VIR_INTERFACE_XML_INACTIVE)
|
|
return util.get_xml_path(xml, "/interface/start/@mode")
|
|
except:
|
|
return None
|
|
|
|
def is_active(self):
|
|
return self.iface.isActive()
|
|
|
|
def get_mac(self):
|
|
mac = self.iface.MACString()
|
|
if mac:
|
|
return mac
|
|
else:
|
|
return None
|
|
|
|
def get_type(self):
|
|
xml = self._XMLDesc()
|
|
return util.get_xml_path(xml, "/interface/@type")
|
|
|
|
def get_ipv4_type(self):
|
|
try:
|
|
xml = self._XMLDesc(VIR_INTERFACE_XML_INACTIVE)
|
|
ipaddr = util.get_xml_path(xml, "/interface/protocol[@family='ipv4']/ip/@address")
|
|
if ipaddr:
|
|
return "static"
|
|
else:
|
|
return "dhcp"
|
|
except:
|
|
return None
|
|
|
|
def get_ipv4(self):
|
|
xml = self._XMLDesc()
|
|
int_ipv4_ip = util.get_xml_path(xml, "/interface/protocol[@family='ipv4']/ip/@address")
|
|
int_ipv4_mask = util.get_xml_path(xml, "/interface/protocol[@family='ipv4']/ip/@prefix")
|
|
if not int_ipv4_ip or not int_ipv4_mask:
|
|
return None
|
|
else:
|
|
return int_ipv4_ip + "/" + int_ipv4_mask
|
|
|
|
def get_ipv6_type(self):
|
|
try:
|
|
xml = self._XMLDesc(VIR_INTERFACE_XML_INACTIVE)
|
|
ipaddr = util.get_xml_path(xml, "/interface/protocol[@family='ipv6']/ip/@address")
|
|
if ipaddr:
|
|
return "static"
|
|
else:
|
|
return "dhcp"
|
|
except:
|
|
return None
|
|
|
|
def get_ipv6(self):
|
|
xml = self._XMLDesc()
|
|
int_ipv6_ip = util.get_xml_path(xml, "/interface/protocol[@family='ipv6']/ip/@address")
|
|
int_ipv6_mask = util.get_xml_path(xml, "/interface/protocol[@family='ipv6']/ip/@prefix")
|
|
if not int_ipv6_ip or not int_ipv6_mask:
|
|
return None
|
|
else:
|
|
return int_ipv6_ip + "/" + int_ipv6_mask
|
|
|
|
def get_bridge(self):
|
|
bridge = None
|
|
if self.get_type() == "bridge":
|
|
bridge = util.get_xml_path(self._XMLDesc(), "/interface/bridge/interface/@name")
|
|
for iface in self.get_bridge_slave_ifaces():
|
|
if iface.get("state") == "up" and iface.get("speed") != "unknown":
|
|
bridge = iface.get("name")
|
|
return bridge
|
|
return bridge
|
|
else:
|
|
return None
|
|
|
|
def get_bridge_slave_ifaces(self):
|
|
ifaces = list()
|
|
if self.get_type() == "bridge":
|
|
tree = ElementTree.fromstring(self._XMLDesc())
|
|
for iface in tree.findall("./bridge/"):
|
|
address = state = speed = None
|
|
name = iface.get("name")
|
|
if_type = iface.get("type")
|
|
link = iface.find("link")
|
|
if link is not None:
|
|
state = link.get("state")
|
|
speed = link.get("speed")
|
|
mac = iface.find("mac")
|
|
if mac is not None:
|
|
address = mac.get("address")
|
|
ifaces.append({"name": name, "type": if_type, "state": state, "speed": speed, "mac": address})
|
|
return ifaces
|
|
else:
|
|
return None
|
|
|
|
def get_details(self):
|
|
mac = self.get_mac()
|
|
itype = self.get_type()
|
|
state = self.is_active()
|
|
return {"name": self.iface, "type": itype, "state": state, "mac": mac}
|
|
|
|
def stop_iface(self):
|
|
self.iface.destroy()
|
|
|
|
def start_iface(self):
|
|
self.iface.create()
|
|
|
|
def delete_iface(self):
|
|
self.iface.undefine()
|