def test_timeout(self):
print("Test Dhclient: timeout")
router = self._create_router()
# Set VLAN that doesn't exist, so no response occur.
router._vlan_iface_id = 99
router._vlan_iface_name = "vlan99"
ipdb = IPDB()
try:
print("Get link-interface ...")
# Get the real link interface
link_iface = ipdb.interfaces["eth0"]
# Create a Vlan
print("Create VLAN ...")
with ipdb.create(kind="vlan", ifname=router.vlan_iface_name, link=link_iface,
vlan_id=router.vlan_iface_id).commit() as iface:
print("Update IP with Dhclient ...")
Dhclient.update_ip(router.vlan_iface_name, timeout=10)
iface.mtu = 1400
except TimeoutError:
print("[+] Test was successful")
except Exception as e:
print("Error: " + str(e))
raise e
finally:
print("Delete VLAN ...")
ipdb.interfaces[router.vlan_iface_name].remove().commit()
ipdb.release()
def test_normal_functionality(self):
print("Test Dhclient: update_ip")
router = self._create_router()
ipdb = IPDB()
try:
print("Get link-interface ...")
# Get the real link interface
link_iface = ipdb.interfaces["eth0"]
# Create a Vlan
print("Create VLAN ...")
with ipdb.create(kind="vlan", ifname=router.vlan_iface_name, link=link_iface,
vlan_id=router.vlan_iface_id).commit() as iface:
print("Update IP with Dhclient ...")
Dhclient.update_ip(router.vlan_iface_name)
iface.mtu = 1400
# Because the IPDB need some time to update the given IP
sleep(10)
process = Popen(["ping", "-c", "1", "-I", router.vlan_iface_name, router.ip], stdout=PIPE, stderr=PIPE)
stdout, sterr = process.communicate()
if not(sterr.decode('utf-8') == "" and "Unreachable" not in stdout.decode('utf-8')):
raise Exception(str(sterr.decode('utf-8')))
print("[+] Test was successful")
except Exception as e:
print("Error: " + str(e))
raise e
finally:
print("Delete VLAN ...")
ipdb.interfaces[router.vlan_iface_name].remove().commit()
ipdb.release()
def test_timeout(self):
print("Test Dhclient: timeout")
router = self._create_router()
# Set VLAN that doesn't exist, so no response occur.
router._vlan_iface_id = 99
router._vlan_iface_name = "vlan99"
ipdb = IPDB()
try:
print("Get link-interface ...")
# Get the real link interface
link_iface = ipdb.interfaces["eth0"]
# Create a Vlan
print("Create VLAN ...")
with ipdb.create(kind="vlan",
ifname=router.vlan_iface_name,
link=link_iface,
vlan_id=router.vlan_iface_id).commit() as iface:
print("Update IP with Dhclient ...")
Dhclient.update_ip(router.vlan_iface_name, timeout=10)
iface.mtu = 1400
except TimeoutError:
print("[+] Test was successful")
except Exception as e:
print("Error: " + str(e))
raise e
finally:
print("Delete VLAN ...")
ipdb.interfaces[router.vlan_iface_name].remove().commit()
ipdb.release()
def test_normal_functionality(self):
print("Test Dhclient: update_ip")
router = self._create_router()
ipdb = IPDB()
try:
print("Get link-interface ...")
# Get the real link interface
link_iface = ipdb.interfaces["eth0"]
# Create a Vlan
print("Create VLAN ...")
with ipdb.create(kind="vlan",
ifname=router.vlan_iface_name,
link=link_iface,
vlan_id=router.vlan_iface_id).commit() as iface:
print("Update IP with Dhclient ...")
Dhclient.update_ip(router.vlan_iface_name)
iface.mtu = 1400
# Because the IPDB need some time to update the given IP
sleep(10)
process = Popen(
["ping", "-c", "1", "-I", router.vlan_iface_name, router.ip],
stdout=PIPE,
stderr=PIPE)
stdout, sterr = process.communicate()
if not (sterr.decode('utf-8') == ""
and "Unreachable" not in stdout.decode('utf-8')):
raise Exception(str(sterr.decode('utf-8')))
print("[+] Test was successful")
except Exception as e:
print("Error: " + str(e))
raise e
finally:
print("Delete VLAN ...")
ipdb.interfaces[router.vlan_iface_name].remove().commit()
ipdb.release()
class Vlan:
def __init__(self,
link_iface_name: str,
vlan_iface_name: str,
vlan_iface_id: int,
vlan_iface_ip: str = None,
vlan_iface_ip_mask: int = None):
"""
Creats a virtual interface on a existing interface (like eth0).
It uses IPDB: IPDB is a transactional database, containing records, representing network stack objects.
Any change in the database is not reflected immidiately in OS, but waits until commit() is called.
:param link_iface_name: name of the existing interface (eth0, wlan0, ...)
:param vlan_iface_name: name of the vlan
:param vlan_iface_id: the id of the vlan
:param vlan_iface_ip: ip of the virtual interface
:param vlan_iface_ip_mask: network-mask of the virtual interface
"""
self.link_iface_name = link_iface_name
self.vlan_iface_name = vlan_iface_name
self.vlan_iface_id = vlan_iface_id
self.ipdb = IPDB()
# self.create_interface(link_iface_name, vlan_iface_name, vlan_iface_id, vlan_iface_ip, vlan_iface_ip_mask)
def create_interface(self,
vlan_iface_ip: str = None,
vlan_iface_ip_mask: int = None):
"""
Creats a virtual interface on a existing interface (like eth0)
:param vlan_iface_ip: ip of the virtual interface
:param vlan_iface_ip_mask: network-mask of the virtual interface
"""
Logger().debug("Create VLAN Interface ...", 2)
try:
link_iface = self.ipdb.interfaces[self.link_iface_name]
with self.ipdb.create(kind="vlan",
ifname=self.vlan_iface_name,
link=link_iface,
vlan_id=self.vlan_iface_id).commit() as i:
if vlan_iface_ip:
i.add_ip(vlan_iface_ip, vlan_iface_ip_mask)
i.mtu = 1400
if not vlan_iface_ip:
self._wait_for_ip_assignment()
vlan_iface_ip = self._get_ipv4_from_dictionary(
self.ipdb.interfaces[self.vlan_iface_name])
Logger().debug(
"[+] " + self.vlan_iface_name + " created with: Link=" +
self.link_iface_name + ", VLAN_ID=" + str(self.vlan_iface_id) +
", IP=" + vlan_iface_ip, 3)
except Exception as e:
Logger().debug(
"[-] " + self.vlan_iface_name + " couldn't be created", 3)
Logger().error(str(e), 3)
def delete_interface(self):
"""
Removes the virtual interface
"""
Logger().debug("Delete VLAN Interface ...", 2)
try:
self.ipdb.interfaces[self.vlan_iface_name].remove().commit()
self.ipdb.release()
Logger().debug(
"[+] Interface(" + self.vlan_iface_name +
") successfully deleted", 3)
except KeyError:
Logger().debug(
"[+] Interface(" + self.vlan_iface_name +
") is already deleted", 3)
return
except Exception as e:
Logger().debug(
"[-] Interface(" + self.vlan_iface_name +
") couldn't be deleted. Try 'ip link delete <vlan_name>'", 3)
Logger().error(str(e), 3)
def _wait_for_ip_assignment(self):
"""
Waits until the dhcp-client got an ip
"""
Logger().debug(
"Wait for ip assignment via dhcp for VLAN Interface(" +
self.vlan_iface_name + ") ...", 3)
if not self.get_ip():
os.system('dhclient ' + self.vlan_iface_name)
while self.get_ip() is None:
time.sleep(0.5)
def get_ip(self) -> str:
"""
Gets the ip of a specific interface
:return: the ip of an interface without network-mask
"""
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sockfd = sock.fileno()
ifreq = struct.pack('16sH14s', self.vlan_iface_name.encode('utf-8'),
socket.AF_INET, b'\x00' * 14)
try:
res = fcntl.ioctl(sockfd, 0x8915, ifreq)
except:
return None
ip = struct.unpack('16sH2x4s8x', res)[2]
return socket.inet_ntoa(ip)
def _get_ipv4_from_dictionary(self, iface) -> str:
"""
Gets the ip and network-mask from the ipdb
:param iface: the interface from ipdb
:return: ip with network-mask
"""
ipaddr_dictionary = iface.ipaddr
for i in range(len(ipaddr_dictionary)):
ip = ipaddr_dictionary[i]['address']
mask = ipaddr_dictionary[i]['prefixlen']
if re.match(
"((((\d|[1-9]\d|1\d{2}|2[0-4]\d|25[0-5])\.){3})(\d|[1-9]\d|1\d{2}|2[0-4]\d|25[0-5]))",
ip):
return ip + "/" + str(mask)
return None
class Vlan:
def __init__(self, link_iface_name: str, vlan_iface_name: str, vlan_iface_id: int, vlan_iface_ip: str=None,
vlan_iface_ip_mask: int=None):
"""
Creats a virtual interface on a existing interface (like eth0).
It uses IPDB: IPDB is a transactional database, containing records, representing network stack objects.
Any change in the database is not reflected immidiately in OS, but waits until commit() is called.
:param link_iface_name: name of the existing interface (eth0, wlan0, ...)
:param vlan_iface_name: name of the vlan
:param vlan_iface_id: the id of the vlan
:param vlan_iface_ip: ip of the virtual interface
:param vlan_iface_ip_mask: network-mask of the virtual interface
"""
self.link_iface_name = link_iface_name
self.vlan_iface_name = vlan_iface_name
self.vlan_iface_id = vlan_iface_id
self.ipdb = IPDB()
# self.create_interface(link_iface_name, vlan_iface_name, vlan_iface_id, vlan_iface_ip, vlan_iface_ip_mask)
def create_interface(self, vlan_iface_ip: str=None, vlan_iface_ip_mask: int=None):
"""
Creats a virtual interface on a existing interface (like eth0)
:param vlan_iface_ip: ip of the virtual interface
:param vlan_iface_ip_mask: network-mask of the virtual interface
"""
Logger().debug("Create VLAN Interface ...", 2)
try:
link_iface = self.ipdb.interfaces[self.link_iface_name]
with self.ipdb.create(kind="vlan", ifname=self.vlan_iface_name, link=link_iface,
vlan_id=self.vlan_iface_id).commit() as i:
if vlan_iface_ip:
i.add_ip(vlan_iface_ip, vlan_iface_ip_mask)
i.mtu = 1400
if not vlan_iface_ip:
self._wait_for_ip_assignment()
vlan_iface_ip = self._get_ipv4_from_dictionary(self.ipdb.interfaces[self.vlan_iface_name])
Logger().debug("[+] " + self.vlan_iface_name + " created with: Link=" + self.link_iface_name +
", VLAN_ID=" + str(self.vlan_iface_id) + ", IP=" + vlan_iface_ip, 3)
except Exception as e:
Logger().debug("[-] " + self.vlan_iface_name + " couldn't be created", 3)
Logger().error(str(e), 3)
def delete_interface(self):
"""
Removes the virtual interface
"""
Logger().debug("Delete VLAN Interface ...", 2)
try:
self.ipdb.interfaces[self.vlan_iface_name].remove().commit()
self.ipdb.release()
Logger().debug("[+] Interface(" + self.vlan_iface_name + ") successfully deleted", 3)
except KeyError:
Logger().debug("[+] Interface(" + self.vlan_iface_name + ") is already deleted", 3)
return
except Exception as e:
Logger().debug("[-] Interface(" + self.vlan_iface_name +
") couldn't be deleted. Try 'ip link delete <vlan_name>'", 3)
Logger().error(str(e), 3)
def _wait_for_ip_assignment(self):
"""
Waits until the dhcp-client got an ip
"""
Logger().debug("Wait for ip assignment via dhcp for VLAN Interface(" + self.vlan_iface_name + ") ...", 3)
if not self.get_ip():
os.system('dhclient ' + self.vlan_iface_name)
while self.get_ip() is None:
time.sleep(0.5)
def get_ip(self) -> str:
"""
Gets the ip of a specific interface
:return: the ip of an interface without network-mask
"""
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sockfd = sock.fileno()
ifreq = struct.pack('16sH14s', self.vlan_iface_name.encode('utf-8'), socket.AF_INET, b'\x00' * 14)
try:
res = fcntl.ioctl(sockfd, 0x8915, ifreq)
except:
return None
ip = struct.unpack('16sH2x4s8x', res)[2]
return socket.inet_ntoa(ip)
def _get_ipv4_from_dictionary(self, iface) -> str:
"""
Gets the ip and network-mask from the ipdb
:param iface: the interface from ipdb
:return: ip with network-mask
"""
ipaddr_dictionary = iface.ipaddr
for i in range(len(ipaddr_dictionary)):
ip = ipaddr_dictionary[i]['address']
mask = ipaddr_dictionary[i]['prefixlen']
if re.match("((((\d|[1-9]\d|1\d{2}|2[0-4]\d|25[0-5])\.){3})(\d|[1-9]\d|1\d{2}|2[0-4]\d|25[0-5]))", ip):
return ip + "/" + str(mask)
return None
