class NetworkCtrl:
"""
The NetworkCtrl manages:
1. Creates a Vlan and a Namespace
2. Encapsulates the Vlan inside the Namespace
3. Provides a SSH-Connection via paramiko
4. Provides a WebServer
5. Provides a Web_configuration_Assistent
"""
def __init__(self, remote_system: RemoteSystem, link_iface_name="eth0"):
"""
Creats a VLAN and a Namespace for the specific RemoteSystem(Router,PowerStrip) and 'eth0' as the link-interface.
The VLAN will be encapsulate in the Namespace.
Also the a SSHClient will be created.
:param remote_system: Could e a Router or a powerstrip object
"""
Logger().info("Create Network Controller ...", 1)
self.remote_system = remote_system
# TODO: ausgelagert in NVAssisten. soll beides aber in Zukunft gelöscht/ausgelagert werden
"""
self.vlan = Vlan(link_iface_name, router.vlan_iface_name, router.vlan_iface_id,
vlan_iface_ip=None, vlan_iface_ip_mask=None)
self.vlan.create_interface()
self.namespace = Namespace(self.router.namespace_name, self.vlan.ipdb)
self.namespace.encapsulate_interface(self.vlan.vlan_iface_name)
"""
self.nv_assisten = NVAssistent()
self.nv_assisten.create_namespace_vlan(
str(self.remote_system.namespace_name),
link_iface_name,
str(self.remote_system.vlan_iface_name),
int(self.remote_system.vlan_iface_id),
)
self.ssh = paramiko.SSHClient()
def connect_with_remote_system(self):
"""
Connects to the remote_system via SSH(Paramiko).
Ignores a missing signatur.
"""
Logger().info("Connect with RemoteSystem ...", 1)
try:
self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
self.ssh.connect(
str(self.remote_system.ip),
port=22,
username=str(self.remote_system.usr_name),
password=str(self.remote_system.usr_password),
)
Logger().debug("[+] Successfully connected", 2)
except Exception as e:
Logger().error("[-] Couldn't connect", 2)
Logger().error("" + str(e), 1)
def send_command(self, command) -> str:
"""
Sends the given command via SSH to the RemoteSystem.
:param command: like "ping 8.8.8.8"
:return: The output of the command given by the RemoteSystem
"""
try:
stdin, stdout, stderr = self.ssh.exec_command(command)
output = stdout.readlines()
Logger().debug("[+] Sent the command (" + command + ") to the RemoteSystem", 2)
return str(output)
except Exception as e:
Logger().error("[-] Couldn't send the command (" + command + ")", 2)
raise e
def send_data(self, local_file: str, remote_file: str):
"""
Sends Data via sftp to the RemoteSystem
:param local_file: Path to the local file
:param remote_file: Path on the Router, where the file should be saved
"""
try:
# TODO: If sftp is installed on the Router
"""
sftp = self.ssh.open_sftp()
sftp.put(local_file, remote_file)
sftp.close()
"""
command = (
"sshpass -p"
+ str(self.remote_system.usr_password)
+ " scp "
+ local_file
+ " "
+ str(self.remote_system.usr_name)
+ "@"
+ str(self.remote_system.ip)
+ ":"
+ remote_file
)
os.system(command)
# TODO: Paramiko_scp have to installed
"""
scp = SCPClient(self.ssh.get_transport())
scp.put(local_file, remote_file)
"""
Logger().debug(
"[+] Sent data '"
+ local_file
+ "' to RemoteSystem '"
+ str(self.remote_system.usr_name)
+ "@"
+ str(self.remote_system.ip)
+ ":"
+ remote_file
+ "'",
2,
)
except Exception as e:
Logger().error(
"[-] Couldn't send '"
+ local_file
+ "' to RemoteSystem '"
+ str(self.remote_system.usr_name)
+ "@"
+ str(self.remote_system.ip)
+ ":"
+ remote_file
+ "'",
2,
)
Logger().error(str(e), 2)
def remote_system_wget(self, file: str, remote_path: str):
"""
The RemoteSystem downloads the file from the PI and stores it at remote_file
:param file: like /root/TestFramework/firmware/.../<firmware>.bin
:param remote_path: like /tmp/
"""
try:
webserver = WebServer()
webserver.start()
# Proves first if file already exists
self.send_command(
"test -f /"
+ remote_path
+ "/"
+ file.split("/")[-1]
+ " || wget http://"
+ self.nv_assisten.namespace.get_ip_of_encapsulate_interface()
+ ":"
+ str(WebServer.PORT_WEBSERVER)
+ file.replace(WebServer.BASE_DIR, "")
+ " -P "
+ remote_path
)
webserver.join()
except Exception as e:
Logger().error(str(e), 2)
def wca_setup_wizard(self, config):
"""
Starts the WebConfigurationAssist and
sets the values provided by the wizard-mode (in the WebConfiguration)
:param config: {node_name, mesh_vpn, limit_bandwidth, show_location, latitude, longitude, altitude, contact,...}
"""
try:
# remote_sytem has to be a router object
wca = WebConfigurationAssist(config, self.remote_system)
wca.setup_wizard()
wca.exit()
except Exception as e:
Logger().error(str(e), 2)
self.exit()
raise e
def wca_setup_expert(self, config):
"""
Starts the WebConfigurationAssist and
sets the values provided by the expert-mode(in the WebConfiguration)
:param config: {node_name, mesh_vpn, limit_bandwidth, show_location, latitude, longitude, altitude, contact,...}
"""
try:
# remote_sytem has to be a router object
wca = WebConfigurationAssist(config, self.remote_system)
wca.setup_expert_private_wlan()
wca.setup_expert_remote_access()
wca.setup_expert_network()
wca.setup_expert_mesh_vpn()
wca.setup_expert_wlan()
wca.setup_expert_autoupdate()
wca.exit()
except Exception as e:
Logger().error(str(e), 2)
self.exit()
raise e
def test_connection(self) -> bool:
"""
Sends a 'Ping' to the RemoteSystem
:return:
"""
output = os.system("ping -c 1 " + str(self.remote_system.ip))
if not output:
return True
else:
return False
def exit(self):
"""
Delete the VLAN resp. the Namespace with the VLAN
"""
Logger().info("Close Network Controller ...", 1)
self.nv_assisten.delete_namespace()
class NetworkCtrl:
"""
The NetworkCtrl manages:
1. Creates a Vlan and a Namespace
2. Encapsulates the Vlan inside the Namespace
3. Provides a SSH-Connection via paramiko
4. Provides a WebServer
"""
def __init__(self, remote_system: RemoteSystem, link_iface_name='eth0'):
"""
Creats a VLAN and a Namespace for the specific RemoteSystem(Router,PowerStrip) and 'eth0' as the link-interface.
The VLAN will be encapsulate in the Namespace.
Also the a SSHClient will be created.
:param remote_system: Could e a Router or a powerstrip object
"""
Logger().info("Create Network Controller ...", 1)
self.remote_system = remote_system
# TODO: ausgelagert in NVAssisten. soll beides aber in Zukunft gelöscht/ausgelagert werden
'''
self.vlan = Vlan(link_iface_name, router.vlan_iface_name, router.vlan_iface_id,
vlan_iface_ip=None, vlan_iface_ip_mask=None)
self.vlan.create_interface()
self.namespace = Namespace(self.router.namespace_name, self.vlan.ipdb)
self.namespace.encapsulate_interface(self.vlan.vlan_iface_name)
'''
self.nv_assisten = NVAssistent()
self.nv_assisten.create_namespace_vlan(str(self.remote_system.namespace_name), link_iface_name,
str(self.remote_system.vlan_iface_name),
int(self.remote_system.vlan_iface_id))
self.ssh = paramiko.SSHClient()
def connect_with_remote_system(self):
"""
Connects to the remote_system via SSH(Paramiko).
Ignores a missing signatur.
"""
Logger().info("Connect with RemoteSystem ...", 1)
try:
self.ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
self.ssh.connect(str(self.remote_system.ip), port=22,
username=str(self.remote_system.usr_name),
password=str(self.remote_system.usr_password))
Logger().debug("[+] Successfully connected", 2)
except Exception as e:
Logger().error("[-] Couldn't connect", 2)
Logger().error("" + str(e), 1)
def send_command(self, command) -> str:
"""
Sends the given command via SSH to the RemoteSystem.
:param command: like "ping 8.8.8.8"
:return: The output of the command given by the RemoteSystem
"""
try:
stdin, stdout, stderr = self.ssh.exec_command(command)
output = stdout.readlines()
Logger().debug("[+] Sent the command (" + command + ") to the RemoteSystem", 2)
return str(output)
except Exception as e:
Logger().error("[-] Couldn't send the command (" + command + ")", 2)
raise e
def send_data(self, local_file: str, remote_file: str):
"""
Sends Data via sftp to the RemoteSystem
:param local_file: Path to the local file
:param remote_file: Path on the Router, where the file should be saved
"""
try:
# TODO: If sftp is installed on the Router
'''
sftp = self.ssh.open_sftp()
sftp.put(local_file, remote_file)
sftp.close()
'''
command = 'sshpass -p' + str(self.remote_system.usr_password) + ' scp ' + local_file + ' ' + \
str(self.remote_system.usr_name) + '@' + str(self.remote_system.ip) + ':' + remote_file
os.system(command)
# TODO: Paramiko_scp have to installed
'''
scp = SCPClient(self.ssh.get_transport())
scp.put(local_file, remote_file)
'''
Logger().debug("[+] Sent data '" + local_file + "' to RemoteSystem '" +
str(self.remote_system.usr_name) + "@" + str(self.remote_system.ip) +
":" + remote_file + "'", 2)
except Exception as e:
Logger().error("[-] Couldn't send '" + local_file + "' to RemoteSystem '" +
str(self.remote_system.usr_name) + "@" + str(self.remote_system.ip) +
":" + remote_file + "'", 2)
Logger().error(str(e), 2)
def remote_system_wget(self, file: str, remote_path: str):
"""
The RemoteSystem downloads the file from the PI and stores it at remote_file
:param file: like /root/TestFramework/firmware/.../<firmware>.bin
:param remote_path: like /tmp/
"""
try:
webserver = WebServer()
webserver.start()
# Proves first if file already exists
self.send_command('test -f /' + remote_path + '/' + file.split('/')[-1] +
' || wget http://' + self.nv_assisten.namespace.get_ip_of_encapsulate_interface() + ':' +
str(WebServer.PORT_WEBSERVER) +
file.replace(WebServer.BASE_DIR, '') + ' -P ' + remote_path)
webserver.join()
except Exception as e:
Logger().error(str(e), 2)
def test_connection(self) -> bool:
"""
Sends a 'Ping' to the RemoteSystem
:return:
"""
output = os.system("ping -c 1 " + str(self.remote_system.ip))
if not output:
return True
else:
return False
def exit(self):
"""
Delete the VLAN resp. the Namespace with the VLAN
"""
Logger().info("Close Network Controller ...", 1)
self.nv_assisten.delete_namespace()
