|
|
@ -7,12 +7,14 @@ from util import sysid
|
|
|
|
from util import setup_child
|
|
|
|
from util import setup_child
|
|
|
|
from util import fprint
|
|
|
|
from util import fprint
|
|
|
|
from util import find_data_file
|
|
|
|
from util import find_data_file
|
|
|
|
|
|
|
|
import util
|
|
|
|
|
|
|
|
|
|
|
|
BG_IMG = 'icon.png'
|
|
|
|
BG_IMG = 'icon.png'
|
|
|
|
filename = sysid + "gendata.csv"
|
|
|
|
filename = sysid + "gendata.csv"
|
|
|
|
COLUMN_NAMES = np.flip(['Executable', 'Proto', 'Source IP', 'Destination IP', 'Status', 'PID'])
|
|
|
|
COLUMN_NAMES = np.flip(['Executable', 'Proto', 'Source IP', 'Destination IP', 'Status', 'PID'])
|
|
|
|
COLUMN_SIZES = np.flip([200, 50, 200, 200, 140, 100])
|
|
|
|
COLUMN_SIZES = np.flip([200, 50, 200, 200, 140, 100])
|
|
|
|
TEST_FILE = None
|
|
|
|
TEST_FILE = None
|
|
|
|
|
|
|
|
SEC_LEVELS = ["0: Backend analysis only.", "1: Kill offending processes.", "2: Block offending connections using firewall.", "3: Kill and block connections and processes."]
|
|
|
|
|
|
|
|
|
|
|
|
displaydata = None
|
|
|
|
displaydata = None
|
|
|
|
settings = None
|
|
|
|
settings = None
|
|
|
@ -38,8 +40,10 @@ class OtherFrame(wx.Frame):
|
|
|
|
Class used for creating frames other than the main one
|
|
|
|
Class used for creating frames other than the main one
|
|
|
|
"""
|
|
|
|
"""
|
|
|
|
def __init__(self):
|
|
|
|
def __init__(self):
|
|
|
|
wx.Frame.__init__(self, None, -1, 'Server Panel', size=(1000, 700))
|
|
|
|
wx.Frame.__init__(self, None, -1, 'Server Panel', size=(1000, 620))
|
|
|
|
panel = ServerPanel(self)
|
|
|
|
panel = ServerPanel(self)
|
|
|
|
|
|
|
|
self.SetIcon(wx.Icon(find_data_file("icon.ico"), wx.BITMAP_TYPE_ICO))
|
|
|
|
|
|
|
|
|
|
|
|
self.Show()
|
|
|
|
self.Show()
|
|
|
|
|
|
|
|
|
|
|
|
def on_edit(self, event):
|
|
|
|
def on_edit(self, event):
|
|
|
@ -56,6 +60,7 @@ class HelpFrame(wx.Frame):
|
|
|
|
def __init__(self):
|
|
|
|
def __init__(self):
|
|
|
|
wx.Frame.__init__(self, None, -1, 'Help', size=(600, 250))
|
|
|
|
wx.Frame.__init__(self, None, -1, 'Help', size=(600, 250))
|
|
|
|
panel = HelpPanel(self)
|
|
|
|
panel = HelpPanel(self)
|
|
|
|
|
|
|
|
self.SetIcon(wx.Icon(find_data_file("icon.ico"), wx.BITMAP_TYPE_ICO))
|
|
|
|
self.Show()
|
|
|
|
self.Show()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -143,13 +148,19 @@ class ServerPanel(wx.Panel):
|
|
|
|
self, size=(-1, 400),
|
|
|
|
self, size=(-1, 400),
|
|
|
|
style=wx.LC_REPORT | wx.BORDER_SUNKEN
|
|
|
|
style=wx.LC_REPORT | wx.BORDER_SUNKEN
|
|
|
|
)
|
|
|
|
)
|
|
|
|
tb = wx.ToolBar( self, -1)
|
|
|
|
#tb = wx.ToolBar( self, -1)
|
|
|
|
self.ToolBar = tb
|
|
|
|
#self.ToolBar = tb
|
|
|
|
tb.SetToolBitmapSize(wx.Size(30, 3))
|
|
|
|
#tb.SetToolBitmapSize(wx.Size(30, 3))
|
|
|
|
tb.AddTool(wx.ID_ANY, '',wx.Bitmap(find_data_file("WXPython_30x30.png")),)
|
|
|
|
#tb.AddTool(wx.ID_ANY, '',wx.Bitmap(find_data_file("WXPython_30x30.png")),)
|
|
|
|
tb.AddTool(wx.ID_ANY, '',wx.Bitmap(find_data_file("settings_30x30.png")))
|
|
|
|
#tb.AddTool(wx.ID_ANY, '',wx.Bitmap(find_data_file("settings_30x30.png")))
|
|
|
|
tb.Realize()
|
|
|
|
#tb.Realize()
|
|
|
|
self.main_sizer.Add(tb)
|
|
|
|
if settings["running"] == True:
|
|
|
|
|
|
|
|
txt = "Status: Running (" + str(settings["config"]["core"]["level"]) + ")"
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
|
|
|
txt = "Status: Not running"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.statustext = wx.StaticText(self, label=txt) # pos=(20,20))
|
|
|
|
|
|
|
|
#self.main_sizer.Add(tb)
|
|
|
|
#main_sizer.SetBackgroundColour((44, 51, 51))
|
|
|
|
#main_sizer.SetBackgroundColour((44, 51, 51))
|
|
|
|
# self.pnl1.SetBackgroundColour(wx.BLACK)
|
|
|
|
# self.pnl1.SetBackgroundColour(wx.BLACK)
|
|
|
|
self.handle_columns()
|
|
|
|
self.handle_columns()
|
|
|
@ -158,17 +169,18 @@ class ServerPanel(wx.Panel):
|
|
|
|
|
|
|
|
|
|
|
|
wx.GetApp().ProcessPendingEvents()
|
|
|
|
wx.GetApp().ProcessPendingEvents()
|
|
|
|
self.main_sizer.Add(self.list_ctrl, 0, wx.ALL | wx.EXPAND, 20)
|
|
|
|
self.main_sizer.Add(self.list_ctrl, 0, wx.ALL | wx.EXPAND, 20)
|
|
|
|
start_button = wx.Button(self, label='Start')
|
|
|
|
start_button = wx.Button(self, label='Start IPPigeon')
|
|
|
|
start_button.SetBackgroundColour((205, 215, 206))
|
|
|
|
start_button.SetBackgroundColour((205, 215, 206))
|
|
|
|
start_button.Bind(wx.EVT_BUTTON, self.on_start)
|
|
|
|
start_button.Bind(wx.EVT_BUTTON, self.on_start)
|
|
|
|
#start_button.Bind(wx.EVT_ENTER_WINDOW, self.on_start)
|
|
|
|
#start_button.Bind(wx.EVT_ENTER_WINDOW, self.on_start)
|
|
|
|
stop_button = wx.Button(self, label='Stop')
|
|
|
|
stop_button = wx.Button(self, label='Stop IPPigeon')
|
|
|
|
stop_button.SetBackgroundColour('#F08080')
|
|
|
|
stop_button.SetBackgroundColour('#F08080')
|
|
|
|
|
|
|
|
|
|
|
|
secondary_frame_button = wx.Button(self, label='Window')
|
|
|
|
secondary_frame_button = wx.Button(self, label='Settings')
|
|
|
|
secondary_frame_button.Bind(wx.EVT_BUTTON, self.on_window)
|
|
|
|
secondary_frame_button.Bind(wx.EVT_BUTTON, self.on_window)
|
|
|
|
# wx.BORDER(stop_button, wx.BORDER_NONE)
|
|
|
|
# wx.BORDER(stop_button, wx.BORDER_NONE)
|
|
|
|
stop_button.Bind(wx.EVT_BUTTON, self.on_stop)
|
|
|
|
stop_button.Bind(wx.EVT_BUTTON, self.on_stop)
|
|
|
|
|
|
|
|
self.main_sizer.Add(self.statustext, 0, wx.CENTER | wx.ALL | 100, 5)
|
|
|
|
self.main_sizer.Add(start_button, 0, wx.CENTER | wx.ALL | 100, 5)
|
|
|
|
self.main_sizer.Add(start_button, 0, wx.CENTER | wx.ALL | 100, 5)
|
|
|
|
self.main_sizer.Add(stop_button, 0, wx.CENTER | wx.ALL | 100, 5)
|
|
|
|
self.main_sizer.Add(stop_button, 0, wx.CENTER | wx.ALL | 100, 5)
|
|
|
|
self.main_sizer.Add(secondary_frame_button, 0, wx.CENTER | wx.ALL | 100, 5)
|
|
|
|
self.main_sizer.Add(secondary_frame_button, 0, wx.CENTER | wx.ALL | 100, 5)
|
|
|
@ -179,15 +191,27 @@ class ServerPanel(wx.Panel):
|
|
|
|
self.list_ctrl.InsertColumn(0, COLUMN_NAMES[col], width=COLUMN_SIZES[col])
|
|
|
|
self.list_ctrl.InsertColumn(0, COLUMN_NAMES[col], width=COLUMN_SIZES[col])
|
|
|
|
|
|
|
|
|
|
|
|
def updatedata(self):
|
|
|
|
def updatedata(self):
|
|
|
|
|
|
|
|
global settings
|
|
|
|
|
|
|
|
if settings["running"] == True:
|
|
|
|
|
|
|
|
txt = "Status: Running (" + str(settings["config"]["core"]["level"]) + ")"
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
|
|
|
txt = "Status: Not running"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if settings["newdata"] == True:
|
|
|
|
|
|
|
|
settings["newdata"] = False
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
|
|
|
wx.CallLater(1000, self.updatedata)
|
|
|
|
|
|
|
|
return
|
|
|
|
fprint("updatedata called")
|
|
|
|
fprint("updatedata called")
|
|
|
|
loaddata()
|
|
|
|
loaddata()
|
|
|
|
global settings
|
|
|
|
|
|
|
|
if self.list_ctrl.DeleteAllItems():
|
|
|
|
if self.list_ctrl.DeleteAllItems():
|
|
|
|
fprint("Items deleted")
|
|
|
|
fprint("Items deleted")
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
fprint("Unable to delete")
|
|
|
|
fprint("Unable to delete")
|
|
|
|
for i in range(len(TEST_FILE)):
|
|
|
|
for i in range(len(TEST_FILE)):
|
|
|
|
|
|
|
|
if str(TEST_FILE.iloc[i, 4]).find("TIME_WAIT") >= 0 or str(TEST_FILE.iloc[i, 4]).find("FIN_WAIT_2") >= 0:
|
|
|
|
|
|
|
|
continue
|
|
|
|
idx = 0
|
|
|
|
idx = 0
|
|
|
|
for ip in settings["badips"]:
|
|
|
|
for ip in settings["badips"]:
|
|
|
|
#fprint(pid)
|
|
|
|
#fprint(pid)
|
|
|
@ -208,18 +232,22 @@ class ServerPanel(wx.Panel):
|
|
|
|
|
|
|
|
|
|
|
|
#fprint(i, j, TEST_FILE.iloc[i, j])
|
|
|
|
#fprint(i, j, TEST_FILE.iloc[i, j])
|
|
|
|
#self.SetSizer(self.main_sizer)
|
|
|
|
#self.SetSizer(self.main_sizer)
|
|
|
|
wx.CallLater(5000, self.updatedata)
|
|
|
|
self.statustext.SetLabel(txt)
|
|
|
|
|
|
|
|
wx.CallLater(1000, self.updatedata)
|
|
|
|
|
|
|
|
|
|
|
|
def on_start(self, event):
|
|
|
|
def on_start(self, event):
|
|
|
|
fprint('in on_start')
|
|
|
|
fprint('in on_start')
|
|
|
|
settings["kill"] = True
|
|
|
|
settings["running"] = True
|
|
|
|
|
|
|
|
|
|
|
|
def on_stop(self, event):
|
|
|
|
def on_stop(self, event):
|
|
|
|
fprint('in on_stop')
|
|
|
|
fprint('in on_stop')
|
|
|
|
settings["kill"] = False
|
|
|
|
settings["running"] = False
|
|
|
|
|
|
|
|
util.clear_fwll()
|
|
|
|
|
|
|
|
|
|
|
|
def on_window(self, event):
|
|
|
|
def on_window(self, event):
|
|
|
|
OtherFrame()
|
|
|
|
fprint("open settings")
|
|
|
|
|
|
|
|
dg = GetData(parent = None)
|
|
|
|
|
|
|
|
dg.ShowModal()
|
|
|
|
|
|
|
|
|
|
|
|
def ShowImage(self, imageFile):
|
|
|
|
def ShowImage(self, imageFile):
|
|
|
|
if imageFile == "":
|
|
|
|
if imageFile == "":
|
|
|
@ -253,14 +281,84 @@ def openwindow(data, sets, kill):
|
|
|
|
loaddata()
|
|
|
|
loaddata()
|
|
|
|
app = wx.App(False)
|
|
|
|
app = wx.App(False)
|
|
|
|
frame = OtherFrame()
|
|
|
|
frame = OtherFrame()
|
|
|
|
|
|
|
|
frame.SetIcon(wx.Icon(find_data_file("icon.ico"), wx.BITMAP_TYPE_ICO))
|
|
|
|
app.MainLoop()
|
|
|
|
app.MainLoop()
|
|
|
|
else:
|
|
|
|
else:
|
|
|
|
fprint("Creating login panel")
|
|
|
|
fprint("Creating login panel")
|
|
|
|
|
|
|
|
|
|
|
|
app = wx.App(False)
|
|
|
|
app = wx.App(False)
|
|
|
|
frame = ServerFrame()
|
|
|
|
frame = ServerFrame()
|
|
|
|
|
|
|
|
frame.SetIcon(wx.Icon(find_data_file("icon.ico"), wx.BITMAP_TYPE_ICO))
|
|
|
|
app.MainLoop()
|
|
|
|
app.MainLoop()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class GetData(wx.Dialog):
|
|
|
|
|
|
|
|
def __init__(self, parent):
|
|
|
|
|
|
|
|
wx.Dialog.__init__(self, parent, wx.ID_ANY, "User Settings: ", size = (800,500))
|
|
|
|
|
|
|
|
self.p = wx.Panel(self, wx.ID_ANY)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.hostname = wx.StaticText(self.p, label="Host:", pos=(20,20))
|
|
|
|
|
|
|
|
self.hostnametext = wx.TextCtrl(self.p, value=settings["config"]["sftp"]["host"], pos=(180,20), size=(500,-1))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.user = wx.StaticText(self.p, label="User: ", pos=(20,60))
|
|
|
|
|
|
|
|
self.usertext = wx.TextCtrl(self.p, value=settings["config"]["sftp"]["user"], pos=(180,60), size=(500,-1))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.port = wx.StaticText(self.p, label="Port: ", pos=(20,100))
|
|
|
|
|
|
|
|
self.porttext = wx.TextCtrl(self.p, value=str(settings["config"]["sftp"]["port"]), pos=(180,100), size=(500,-1))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.keyfile = wx.StaticText(self.p, label="Keyfile: ", pos=(20,140))
|
|
|
|
|
|
|
|
self.keyfiletext = wx.TextCtrl(self.p, value=settings["config"]["sftp"]["keyfile"], pos=(180,140), size=(500,-1))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.filepathsend = wx.StaticText(self.p, label="Sending File Path: ", pos=(20,180))
|
|
|
|
|
|
|
|
self.filepathsendtext = wx.TextCtrl(self.p, value=settings["config"]["sftp"]["filepath"]["send"], pos=(180,180), size=(500,-1))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.filepathsendlogin = wx.StaticText(self.p, label="Sending Login Path: ", pos=(20,220))
|
|
|
|
|
|
|
|
self.filepathsendlogintext = wx.TextCtrl(self.p, value=settings["config"]["sftp"]["filepath"]["sendlogin"], pos=(180,220), size=(500,-1))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.filepathreceive = wx.StaticText(self.p, label="Receiving File Path: ", pos=(20,260))
|
|
|
|
|
|
|
|
self.filepathreceivetext = wx.TextCtrl(self.p, value=settings["config"]["sftp"]["filepath"]["receive"], pos=(180,260), size=(500,-1))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.filepathreceivelogin = wx.StaticText(self.p, label="Receiving Login Path: ", pos=(20,300))
|
|
|
|
|
|
|
|
self.filepathreceivelogintext = wx.TextCtrl(self.p, value=settings["config"]["sftp"]["filepath"]["receivelogin"], pos=(180,300), size=(500,-1))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.darkmode = wx.StaticText(self.p, label="Dark mode (On/Off): ", pos=(20,340))
|
|
|
|
|
|
|
|
self.darkmodetext = wx.TextCtrl(self.p, value=str(settings["config"]["ui"]["darkmode"]), pos=(180,340), size=(500,-1))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.interval = wx.StaticText(self.p, label="Interval (sec): ", pos=(20,380))
|
|
|
|
|
|
|
|
self.intervaltext = wx.TextCtrl(self.p, value=str(settings["config"]["core"]["interval"]), pos=(180,380), size=(500,-1))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.seclevel = wx.StaticText(self.p, label="Security Strength Level: ", pos=(20,420))
|
|
|
|
|
|
|
|
self.seclevelslider = wx.Slider(self.p, pos=(180,410), minValue=0, maxValue=3, style=wx.SL_HORIZONTAL | wx.SL_AUTOTICKS, value=settings["config"]["core"]["level"])
|
|
|
|
|
|
|
|
self.seclevel2 = wx.StaticText(self.p, label=SEC_LEVELS[settings["config"]["core"]["level"]], pos=(300,420))
|
|
|
|
|
|
|
|
self.seclevelslider.Bind(wx.EVT_SCROLL, self.level_upd)
|
|
|
|
|
|
|
|
self.Bind(wx.EVT_CLOSE, self.OnQuit)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def level_upd(self, event):
|
|
|
|
|
|
|
|
self.seclevel2.SetLabel(SEC_LEVELS[self.seclevelslider.GetValue()])
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def OnQuit(self, event):
|
|
|
|
|
|
|
|
# save changes
|
|
|
|
|
|
|
|
fprint("saving changes...")
|
|
|
|
|
|
|
|
config = settings["config"]
|
|
|
|
|
|
|
|
config["sftp"]["host"] = self.hostnametext.GetValue()
|
|
|
|
|
|
|
|
config["sftp"]["user"] = self.usertext.GetValue()
|
|
|
|
|
|
|
|
config["sftp"]["port"] = int(self.porttext.GetValue())
|
|
|
|
|
|
|
|
config["sftp"]["keyfile"] = self.keyfiletext.GetValue()
|
|
|
|
|
|
|
|
config["sftp"]["filepath"]["send"] = self.filepathsendtext.GetValue()
|
|
|
|
|
|
|
|
config["sftp"]["filepath"]["sendlogin"] = self.filepathsendlogintext.GetValue()
|
|
|
|
|
|
|
|
config["sftp"]["filepath"]["receive"] = self.filepathreceivetext.GetValue()
|
|
|
|
|
|
|
|
config["sftp"]["filepath"]["receivelogin"] = self.filepathreceivelogintext.GetValue()
|
|
|
|
|
|
|
|
config["ui"]["darkmode"] = bool(self.darkmodetext.GetValue())
|
|
|
|
|
|
|
|
config["core"]["interval"] = int(self.intervaltext.GetValue())
|
|
|
|
|
|
|
|
config["core"]["level"] = int(self.seclevelslider.GetValue())
|
|
|
|
|
|
|
|
settings["config"] = config
|
|
|
|
|
|
|
|
settings["applyconfig"] = True
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
self.result_name = None
|
|
|
|
|
|
|
|
self.Destroy()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def on_edit(self, event):
|
|
|
|
|
|
|
|
print('in on_edit')
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
if __name__ == '__main__':
|
|
|
|
openwindow(list(), dict(), int())
|
|
|
|
openwindow(list(), dict(), int())
|
|
|
|