update

security-management-ui.py

@@ -4,15 +4,7 @@ import json
 import os
 import sniff
 import multiprocessing
-
-def get_interfaces():
-    interfaces = []
-    output = os.popen('route print').read()
-    lines = output.split('\n')
-    for line in lines:
-        if ("......" in line):
-            interfaces.append(line.split("......")[1])
-    return interfaces
+from utilities.list_interfaces import get_interfaces
 
 
 app = Flask(__name__, template_folder='template')

sniff.py

 from scapy.all import *
-import socket
 import datetime
 import os
-import ssl
 import binascii
 import yaml
 import sqlite3
-import xml.etree.ElementTree as ET
 from datetime import datetime
 import protocol_actions
-import asset_actions
 import threading
 import requests
 lock = threading.Lock()

utilities/list_interfaces.py

 from scapy.all import *
-import yaml
-import sqlite3
 
-yamlfile=open("config.yaml")
-data = yaml.load(yamlfile, Loader=yaml.FullLoader)
-mydb=sqlite3.connect("assets.db",check_same_thread=False)
-query=mydb.cursor()
-
-def show_list():
-	for i in get_if_list():
-		if(not query.execute("SELECT interface_name FROM interfaces WHERE interface_name='"+i+"'").fetchone()):
-			updateiface='INSERT INTO interfaces\
-					(interface_name,plant)\
-					VALUES\
-					("'+i+'","'+data["configuration"]["edge_device_location"]+'")'
-			query.execute(updateiface)
-			mydb.commit()
-	query.close()
-	print(get_if_list())
+def get_interfaces():
+	interfaces = []
+	if platform == "linux" or platform == "linux2":
+		for i in get_if_list():
+			interfaces.append(i)
+	elif platform == "darwin":
+		for i in get_if_list():
+			interfaces.append(i)
+	elif platform == "win32":
+		output = os.popen('route print').read()
+		lines = output.split('\n')
+		for line in lines:
+			if ("......" in line):
+				interfaces.append(line.split("......")[1])
+	return interfaces
\ No newline at end of file
\ No newline at end of file