def __init__(self, application=None, *args, **kwargs):
self.application = application
wx.Frame.__init__(self, None, *args, **kwargs)
self.Bind(wx.EVT_CLOSE, lambda x: self.Destroy())
# self.Show()
import sys
def val(key):
if key in sys.argv:
return sys.argv[sys.argv.index("-t") + 1]
else:
return None
def is_yaml(filename):
ext = ".yaml"
return filename[-len(ext) :] == ext
template_file = val("-t")
content_file = val("-c")
kb_file = val("-k")
scan_file = val("-s")
report_file = val("-r")
if template_file and report_file:
report = Report()
report.template_load_xml(template_file)
if content_file:
if is_yaml(content_file):
report.content_load_yaml(content_file)
else:
report.content_load_json(content_file)
if kb_file:
if is_yaml(kb_file):
report.kb_load_yaml(kb_file)
else:
report.kb_load_json(kb_file)
if scan_file:
report.scan = Scan(scan_file)
report.xml_apply_meta(
vulnparam_highlighting=self.menu_view_v.IsChecked(), truncation=self.menu_view_i.IsChecked()
)
report.save_report_xml(report_file)
else:
print "Usage: "
print
print " " + self.application.title + ".exe"
print " start GUI application"
print
print " " + self.application.title + ".exe -t template-file [-c content-file] [-k kb-file] [-s scan-file] -r report-file"
print " generate report"
print
print " " + self.application.title + ".exe [any other arguments]"
print " display usage and exit"
self.Close()
def __init__(self, application=None, *args, **kwargs):
self.application = application
wx.Frame.__init__(self, None, *args, **kwargs)
self.Bind(wx.EVT_CLOSE, lambda x: self.Destroy())
#self.Show()
import sys
def val(key):
if key in sys.argv:
return sys.argv[sys.argv.index('-t') + 1]
else:
return None
def is_yaml(filename):
ext = '.yaml'
return filename[-len(ext):] == ext
template_file = val('-t')
content_file = val('-c')
kb_file = val('-k')
scan_file = val('-s')
report_file = val('-r')
if template_file and report_file:
report = Report()
report.template_load_xml(template_file)
if content_file:
if is_yaml(content_file):
report.content_load_yaml(content_file)
else:
report.content_load_json(content_file)
if kb_file:
if is_yaml(kb_file):
report.kb_load_yaml(kb_file)
else:
report.kb_load_json(kb_file)
if scan_file:
report.scan = Scan(scan_file)
report.xml_apply_meta(
vulnparam_highlighting=self.menu_view_v.IsChecked(),
truncation=self.menu_view_i.IsChecked())
report.save_report_xml(report_file)
else:
print 'Usage: '
print
print ' ' + self.application.title + '.exe'
print ' start GUI application'
print
print ' ' + self.application.title + '.exe -t template-file [-c content-file] [-k kb-file] [-s scan-file] -r report-file'
print ' generate report'
print
print ' ' + self.application.title + '.exe [any other arguments]'
print ' display usage and exit'
self.Close()
class __MainWindow(wx.Frame):
# Variables set during __init__
# report
# application
# scan
# template
# icon
# statusbar
# save_into_directory
# menu_file_open_c
# menu_file_open_k
# menu_file_generate_c
# menu_file_generate_k
# menu_file_generate_r
# menu_file_save_t
# menu_file_save_c
# menu_file_save_s
# menu_file_save_k
# menu_file_save_r
# menu_view_c
# menu_view_y
# menu_view_j
# menu_view_s
# menu_view_v
# menu_view_i
# menu_view_r
# menu_view_t
# menu_tools_template_structure_preview
# menu_tools_merge_scan_into_content
# menu_tools_generate_few_passwords
# ctrl_st_t
# ctrl_tc_t
# ctrl_st_c
# ctrl_tc_c
# ctrl_tc_c_b
# ctrl_st_s
# ctrl_tc_s
# ctrl_tc_s_b
# ctrl_st_k
# ctrl_tc_k
# ctrl_st_r
# ctrl_tc_r
# color_tc_bg_e
# color_tc_bg_d
# _statusbar_h
# children
def __init__(
self, application=None, parent=None, *args, **kwargs
): # style=wx.DEFAULT_FRAME_STYLE^wx.RESIZE_BORDER
self.children = []
# self.report = None
self.report = Report()
self.application = application
self.scan = None
self.save_into_directory = ""
wx.Frame.__init__(
self, parent, title=self.application.title + " " + self.application.version, *args, **kwargs
) # style=style
self.Bind(wx.EVT_CLOSE, lambda x: self.Destroy())
myStream = cStringIO.StringIO(base64.b64decode(icon))
myImage = wx.ImageFromStream(myStream)
myBitmap = wx.BitmapFromImage(myImage)
self.icon = wx.EmptyIcon()
self.icon.CopyFromBitmap(myBitmap)
self.SetIcon(self.icon)
# Menu arrangement
menu = wx.MenuBar()
class Index(object):
def __init__(self, current):
self.__current = current - 1
@property
def current(self):
return self.__current
@current.setter
def current(self, x):
self.__current = x
def next(self):
self.__current += 1
return self.__current
index = Index(100)
menu_file = wx.Menu()
menu_file.Append(index.next(), "Open Report &Template...")
self.Bind(wx.EVT_MENU, self.Open_Template, id=index.current)
self.menu_file_open_c = menu_file.Append(index.next(), "Open &Content...")
self.menu_file_open_c.Enable(False)
self.Bind(wx.EVT_MENU, self.Open_Content, id=index.current)
menu_file.Append(index.next(), "Open &Scan...")
self.Bind(wx.EVT_MENU, self.Open_Scan, id=index.current)
self.menu_file_open_k = menu_file.Append(index.next(), "Open &Knowledge Base...")
self.menu_file_open_k.Enable(False)
self.Bind(wx.EVT_MENU, self.Open_Knowledge_Base, id=index.current)
# menu_file.AppendSeparator()
# self.menu_file_generate_c = menu_file.Append(index.next(), '&Generate Content')
# self.menu_file_generate_c.Enable(False)
# self.Bind(wx.EVT_MENU, self.Generate_Content, id=index.current)
# self.menu_file_generate_k = menu_file.Append(index.next(), 'G&enerate Knowledge Base')
# self.menu_file_generate_k.Enable(False)
# self.Bind(wx.EVT_MENU, self.Generate_Knowledge_Base, id=index.current)
# self.menu_file_generate_r = menu_file.Append(index.next(), 'Ge&nerate Report')
# self.menu_file_generate_r.Enable(False)
# self.Bind(wx.EVT_MENU, self.Generate_Report, id=index.current)
menu_file.AppendSeparator()
self.menu_file_save_t = menu_file.Append(index.next(), "&Save Template As...")
self.menu_file_save_t.Enable(False)
self.Bind(wx.EVT_MENU, self.Save_Template_As, id=index.current)
self.menu_file_save_c = menu_file.Append(index.next(), "Sav&e Content As...")
self.menu_file_save_c.Enable(False)
self.Bind(wx.EVT_MENU, self.Save_Content_As, id=index.current)
# self.menu_file_save_k = menu_file.Append(index.next(), 'S&ave Knowledge Base As...')
# self.menu_file_save_k.Enable(False)
# self.Bind(wx.EVT_MENU, self.Save_Knowledge_Base_As, id=index.current)
self.menu_file_save_s = menu_file.Append(index.next(), "Sa&ve Scan As...")
self.menu_file_save_s.Enable(False)
self.Bind(wx.EVT_MENU, self.Save_Scan_As, id=index.current)
self.menu_file_save_r = menu_file.Append(index.next(), "Save &Report As...")
self.menu_file_save_r.Enable(False)
self.Bind(wx.EVT_MENU, self.Save_Report_As, id=index.current)
menu_file.AppendSeparator()
menu_file.Append(wx.ID_EXIT, "E&xit\tCtrl+Q", "Exit application")
self.Bind(wx.EVT_MENU, self.Exit, id=wx.ID_EXIT)
menu.Append(menu_file, "&File")
menu_view = wx.Menu()
self.menu_view_c = menu_view.Append(index.next(), "C&lean template", kind=wx.ITEM_CHECK)
self.Bind(wx.EVT_MENU, self.Clean_template, id=index.current)
self.menu_view_c.Check(True)
menu_view.AppendSeparator()
self.menu_view_y = menu_view.Append(index.next(), "&yaml", kind=wx.ITEM_RADIO)
self.Bind(wx.EVT_MENU, self.Use_yaml, id=index.current)
self.menu_view_j = menu_view.Append(index.next(), "&json", kind=wx.ITEM_RADIO)
self.Bind(wx.EVT_MENU, self.Use_json, id=index.current)
self.menu_view_y.Check(True)
menu.Append(menu_view, "&View")
# menu_view.AppendSeparator()
# self.menu_view_s = menu_view.Append(index.next(), '&Status Preview', kind=wx.ITEM_CHECK)
# self.Bind(wx.EVT_MENU, self.Status_Preview, id=index.current)
# self.menu_view_s.Check(False)
menu_view.AppendSeparator()
self.menu_view_v = menu_view.Append(index.next(), "&VulnParam highlighting", kind=wx.ITEM_CHECK)
self.Bind(wx.EVT_MENU, self.VulnParam_highlighting, id=index.current)
self.menu_view_v.Check(True)
self.menu_view_i = menu_view.Append(
index.next(),
"V&iewState truncation",
"Warning! Application performance will noticeably decrease!",
kind=wx.ITEM_CHECK,
)
self.Bind(wx.EVT_MENU, self.Viewstate_truncation, id=index.current)
self.menu_view_i.Check(True)
self.menu_view_r = menu_view.Append(
index.next(),
"Include &requests and responses",
"Warning! Have a small scan or be very patient!",
kind=wx.ITEM_CHECK,
)
self.menu_view_r.Check(False)
menu_view.AppendSeparator()
self.menu_view_t = menu_view.Append(index.next(), "Always on &top", kind=wx.ITEM_CHECK)
self.Bind(wx.EVT_MENU, self.Always_on_top, id=index.current)
self.menu_view_t.Check(True)
menu_tools = wx.Menu()
self.menu_tools_template_structure_preview = menu_tools.Append(index.next(), "Te&mplate structure preview")
self.menu_tools_template_structure_preview.Enable(False)
self.Bind(wx.EVT_MENU, self.Template_Structure_Preview, id=index.current)
self.menu_tools_merge_scan_into_content = menu_tools.Append(index.next(), "Mer&ge Scan into Content")
self.menu_tools_merge_scan_into_content.Enable(False)
self.Bind(wx.EVT_MENU, self.Merge_Scan_Into_Content, id=index.current)
self.menu_tools_generate_few_passwords = menu_tools.Append(index.next(), "Generate &few passwords")
self.Bind(wx.EVT_MENU, self.Generate_few_passwords, id=index.current)
menu.Append(menu_tools, "&Tools")
menu_help = wx.Menu()
menu_help.Append(index.next(), "&Usage")
self.Bind(wx.EVT_MENU, self.Usage, id=index.current)
menu_help.Append(index.next(), "&Changelog")
self.Bind(wx.EVT_MENU, self.Changelog, id=index.current)
menu_help.AppendSeparator()
menu_help.Append(wx.ID_ABOUT, "&About")
self.Bind(wx.EVT_MENU, self.About, id=wx.ID_ABOUT)
menu.Append(menu_help, "&Help")
self.SetMenuBar(menu)
# Frame layout arrangement
class FileDropTarget(wx.FileDropTarget):
def __init__(self, target, handler):
wx.FileDropTarget.__init__(self)
self.target = target
self.handler = handler
def OnDropFiles(self, x, y, filenames):
self.handler(filenames)
panel = wx.Panel(self)
vbox = wx.BoxSizer(wx.VERTICAL)
fgs = wx.FlexGridSizer(5, 2, 9, 25)
# Template
self.ctrl_st_t = wx.StaticText(panel, label="Template:")
self.ctrl_st_t.Enable(False)
self.ctrl_tc_t = wx.TextCtrl(panel, style=wx.TE_MULTILINE | wx.TE_READONLY, size=(200, 3 * 17))
def ctrl_tc_t_OnFocus(e):
self.ctrl_tc_t.ShowNativeCaret(False)
# for unknown reason this refuse to work in wxpython 3.0
e.Skip()
def ctrl_tc_t_OnDoubleclick(e):
if self.ctrl_st_t.IsEnabled():
self.application.TextWindow(self, title="Template Preview", content=self.ctrl_tc_t.GetValue())
e.Skip()
self.ctrl_tc_t.Bind(wx.EVT_SET_FOCUS, ctrl_tc_t_OnFocus)
self.ctrl_tc_t.Bind(wx.EVT_LEFT_DCLICK, ctrl_tc_t_OnDoubleclick)
def ctrl_tc_t_OnMouseOver(e):
self.status("You might use drag & drop", hint=True)
e.Skip()
# def ctrl_tc_t_OnMouseLeave(e):
# self.status('')
# e.Skip()
self.ctrl_tc_t.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_t_OnMouseOver)
# self.ctrl_tc_t.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_t_OnMouseLeave)
def ctrl_tc_t_OnDropFiles(filenames):
if len(filenames) != 1:
wx.MessageBox("Single file is expected!", "Error", wx.OK | wx.ICON_ERROR)
return
self._open_template(filenames[0])
ctrl_tc_t_dt = FileDropTarget(self.ctrl_tc_t, ctrl_tc_t_OnDropFiles)
self.ctrl_tc_t.SetDropTarget(ctrl_tc_t_dt)
fgs.AddMany([(self.ctrl_st_t, 1, wx.EXPAND), (self.ctrl_tc_t, 1, wx.EXPAND)])
# Content + Edit
self.ctrl_st_c = wx.StaticText(panel, label="Content:")
self.ctrl_st_c.Enable(False)
self.ctrl_tc_c = wx.TextCtrl(panel, style=wx.TE_MULTILINE | wx.TE_READONLY, size=(200, 3 * 17))
self.color_tc_bg_e = self.ctrl_tc_c.GetBackgroundColour()
self.ctrl_tc_c.Enable(False)
self.color_tc_bg_d = wx.SystemSettings.GetColour(wx.SYS_COLOUR_BTNFACE)
self.ctrl_tc_c.SetBackgroundColour(self.color_tc_bg_d)
def ctrl_tc_c_OnFocus(e):
self.ctrl_tc_c.ShowNativeCaret(False)
e.Skip()
def ctrl_tc_c_OnDoubleclick(e):
if self.ctrl_st_c.IsEnabled():
self.application.TextWindow(self, title="Content Preview", content=self.ctrl_tc_c.GetValue())
e.Skip()
self.ctrl_tc_c.Bind(wx.EVT_SET_FOCUS, ctrl_tc_c_OnFocus)
self.ctrl_tc_c.Bind(wx.EVT_LEFT_DCLICK, ctrl_tc_c_OnDoubleclick)
def ctrl_tc_c_b_onClick(e):
self.application.YamledWindowWrapper(self, title="Content", content=self.report._content)
pass
def ctrl_tc_c_onResize(e):
size = self.ctrl_tc_c.GetSize()
self.ctrl_tc_c_b.SetPosition((size[0] - 36 - 1, -1))
self.ctrl_tc_c_b = wx.Button(self.ctrl_tc_c, index.next(), "E", size=(16, 16))
self.ctrl_tc_c_b.Bind(wx.EVT_BUTTON, ctrl_tc_c_b_onClick)
self.ctrl_tc_c.Bind(wx.EVT_SIZE, ctrl_tc_c_onResize)
self.ctrl_tc_c_b.Hide()
def ctrl_tc_c_b_OnMouseOver(e):
self.status("Send Content to Yaml Editor", hint=True)
e.Skip()
# def ctrl_tc_c_b_OnMouseLeave(e):
# self.status('')
# e.Skip()
self.ctrl_tc_c_b.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_c_b_OnMouseOver)
# self.ctrl_tc_c_b.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_c_b_OnMouseLeave)
def ctrl_tc_c_OnMouseOver(e):
if self.ctrl_st_c.IsEnabled(): # Yamled
self.ctrl_tc_c_b.Show()
self.status("You might use drag & drop", hint=True)
e.Skip()
# def ctrl_tc_c_OnMouseLeave(e):
# self.status('')
# e.Skip()
self.ctrl_tc_c.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_c_OnMouseOver)
# self.ctrl_tc_c.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_c_OnMouseLeave)
def ctrl_tc_c_OnDropFiles(filenames):
if len(filenames) != 1:
wx.MessageBox("Single file is expected!", "Error", wx.OK | wx.ICON_ERROR)
return
self._open_content(filenames[0])
if self.ctrl_st_c.IsEnabled(): # Yamled
self.ctrl_tc_c_b.Show()
ctrl_tc_c_dt = FileDropTarget(self.ctrl_tc_c, ctrl_tc_c_OnDropFiles)
self.ctrl_tc_c.SetDropTarget(ctrl_tc_c_dt)
fgs.AddMany([(self.ctrl_st_c, 1, wx.EXPAND), (self.ctrl_tc_c, 1, wx.EXPAND)])
# Scan + Edit button
self.ctrl_st_s = wx.StaticText(panel, label="Scan:")
self.ctrl_st_s.Enable(False)
self.ctrl_tc_s = wx.TextCtrl(panel, style=wx.TE_MULTILINE | wx.TE_READONLY, size=(200, 3 * 17))
def ctrl_tc_s_OnFocus(e):
self.ctrl_tc_s.ShowNativeCaret(False)
e.Skip()
def ctrl_tc_s_OnDoubleclick(e):
if self.ctrl_st_s.IsEnabled():
self.application.TextWindow(self, title="Scan Preview", content=self.ctrl_tc_s.GetValue())
e.Skip()
self.ctrl_tc_s.Bind(wx.EVT_SET_FOCUS, ctrl_tc_s_OnFocus)
self.ctrl_tc_s.Bind(wx.EVT_LEFT_DCLICK, ctrl_tc_s_OnDoubleclick)
def ctrl_tc_s_b_onClick(e):
self.application.YamledWindowWrapper(self, title="Scan", content=self.scan._scan)
pass
def ctrl_tc_s_onResize(e):
size = self.ctrl_tc_s.GetSize()
self.ctrl_tc_s_b.SetPosition((size[0] - 36 - 1, -1))
self.ctrl_tc_s_b = wx.Button(self.ctrl_tc_s, index.next(), "E", size=(16, 16))
self.ctrl_tc_s_b.Bind(wx.EVT_BUTTON, ctrl_tc_s_b_onClick)
self.ctrl_tc_s.Bind(wx.EVT_SIZE, ctrl_tc_s_onResize)
self.ctrl_tc_s_b.Hide()
def ctrl_tc_s_b_OnMouseOver(e):
self.status("Send Scan to Yaml Editor", hint=True)
e.Skip()
# def ctrl_tc_s_b_OnMouseLeave(e):
# self.status('')
# e.Skip()
self.ctrl_tc_s_b.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_s_b_OnMouseOver)
# self.ctrl_tc_s_b.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_s_b_OnMouseLeave)
def ctrl_tc_s_OnMouseOver(e):
if self.ctrl_st_s.IsEnabled(): # Yamled
self.ctrl_tc_s_b.Show()
self.status("You might use drag & drop", hint=True)
e.Skip()
# def ctrl_tc_s_OnMouseLeave(e):
# #self.ctrl_tc_s_b.Hide()
# self.status('')
# e.Skip()
self.ctrl_tc_s.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_s_OnMouseOver)
# self.ctrl_tc_s.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_s_OnMouseLeave)
def ctrl_tc_s_OnDropFiles(filenames):
if len(filenames) != 1:
wx.MessageBox("Single file is expected!", "Error", wx.OK | wx.ICON_ERROR)
return
self._open_scan(filenames[0])
if self.ctrl_st_s.IsEnabled(): # Yamled
self.ctrl_tc_s_b.Show()
ctrl_tc_s_dt = FileDropTarget(self.ctrl_tc_s, ctrl_tc_s_OnDropFiles)
self.ctrl_tc_s.SetDropTarget(ctrl_tc_s_dt)
fgs.AddMany([(self.ctrl_st_s, 1, wx.EXPAND), (self.ctrl_tc_s, 1, wx.EXPAND)])
# Knowledge Base
self.ctrl_st_k = wx.StaticText(panel, label="Knowledge Base:")
self.ctrl_st_k.Enable(False)
self.ctrl_tc_k = wx.TextCtrl(panel, style=wx.TE_MULTILINE | wx.TE_READONLY, size=(200, 3 * 17))
self.ctrl_tc_k.Enable(False)
self.ctrl_tc_k.SetBackgroundColour(self.color_tc_bg_d)
def ctrl_tc_k_OnFocus(e):
self.ctrl_tc_k.ShowNativeCaret(False)
e.Skip()
def ctrl_tc_k_OnDoubleclick(e):
if self.ctrl_st_k.IsEnabled():
self.application.TextWindow(self, title="KB Preview", content=self.ctrl_tc_k.GetValue())
e.Skip()
self.ctrl_tc_k.Bind(wx.EVT_SET_FOCUS, ctrl_tc_k_OnFocus)
self.ctrl_tc_k.Bind(wx.EVT_LEFT_DCLICK, ctrl_tc_k_OnDoubleclick)
def ctrl_tc_k_OnMouseOver(e):
self.status("You might use drag & drop", hint=True)
e.Skip()
def ctrl_tc_k_OnMouseLeave(e):
self.status("")
e.Skip()
self.ctrl_tc_k.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_k_OnMouseOver)
self.ctrl_tc_k.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_k_OnMouseLeave)
def ctrl_tc_k_OnDropFiles(filenames):
if len(filenames) != 1:
wx.MessageBox("Single file is expected!", "Error", wx.OK | wx.ICON_ERROR)
return
self._open_kb(filenames[0])
ctrl_tc_k_dt = FileDropTarget(self.ctrl_tc_k, ctrl_tc_k_OnDropFiles)
self.ctrl_tc_k.SetDropTarget(ctrl_tc_k_dt)
fgs.AddMany([(self.ctrl_st_k, 1, wx.EXPAND), (self.ctrl_tc_k, 1, wx.EXPAND)])
def panel_OnMouseOver(e):
self.status("")
self.ctrl_tc_c_b.Hide()
self.ctrl_tc_s_b.Hide()
e.Skip()
panel.Bind(wx.EVT_ENTER_WINDOW, panel_OnMouseOver)
# Report
# self.ctrl_st_r = wx.StaticText(panel, label='Report:')
# self.ctrl_st_r.Enable (False)
# self.ctrl_tc_r = wx.TextCtrl(panel, style=wx.TE_MULTILINE|wx.TE_READONLY, size=(200, 3*17,))
# self.ctrl_tc_r.Enable(False)
# self.ctrl_tc_r.SetBackgroundColour (self.color_tc_bg_d)
# def ctrl_tc_r_OnFocus (e):
# self.ctrl_tc_r.ShowNativeCaret (False)
# e.Skip()
# self.ctrl_tc_r.Bind (wx.EVT_SET_FOCUS, ctrl_tc_r_OnFocus)
# fgs.AddMany ([(self.ctrl_st_r, 1, wx.EXPAND), (self.ctrl_tc_r, 1, wx.EXPAND)])
fgs.AddGrowableRow(0, 1)
fgs.AddGrowableRow(1, 1)
fgs.AddGrowableRow(2, 1)
fgs.AddGrowableRow(3, 1)
# fgs.AddGrowableRow (4, 1)
fgs.AddGrowableCol(1, 1)
vbox.Add(fgs, proportion=1, flag=wx.ALL | wx.EXPAND, border=10)
# data = wx.TextCtrl(panel)
# hbox1 = wx.BoxSizer (wx.HORIZONTAL)
# hbox1.Add(data, proportion=1, flag=wx.LEFT|wx.RIGHT|wx.BOTTOM|wx.EXPAND, border=10)
# vbox.Add (hbox1, 0, wx.ALL|wx.EXPAND, 0)
panel.SetSizer(vbox)
vbox.Fit(self)
# self.SetMinSize(self.GetSize())
self.statusbar = self.CreateStatusBar()
self._statusbar_h = self.statusbar.GetSize()[1]
# self.statusbar.Hide()
self.status("Started")
self.SetMinSize((self.GetSize()[0], self.GetSize()[1] + self._statusbar_h))
# panel = wx.Panel (self)
# vbox = wx.BoxSizer (wx.VERTICAL)
# hbox1 = wx.BoxSizer (wx.HORIZONTAL)
##st1 = wx.StaticText (panel, wx.ID_ANY, label='Not yet ready')
# st1 = wx.StaticText (panel, wx.ID_ANY, label='Template:', size=(100, -1,))
# hbox1.Add (st1, 0, wx.ALL, 5)
# tc1 = wx.TextCtrl (panel, wx.ID_ANY, size=(300, -1,))
# hbox1.Add (tc1, 1, wx.ALL|wx.EXPAND, 0)
# vbox.Add (hbox1, 0, wx.ALL|wx.EXPAND, 0)
# hbox2 = wx.BoxSizer (wx.HORIZONTAL)
# st2 = wx.StaticText (panel, wx.ID_ANY, label='Scan:', size=(100, -1,))
# hbox2.Add (st2, 0, wx.ALL, 5)
# tc2 = wx.TextCtrl (panel, wx.ID_ANY, size=(300, -1,))
# hbox2.Add (tc2, 1, wx.ALL|wx.EXPAND, 0)
# vbox.Add (hbox2, 0, wx.ALL|wx.EXPAND, 0)
##vbox.Add (hbox1, 0, wx.CENTER, 5)
# panel.SetSizer (vbox)
# vbox.Fit (self)
# self.Center()
self.alignVMiddleRight()
self.Show()
self.Always_on_top(None)
# print 'loaded'
def alignVMiddleRight(self):
dw, dh = wx.DisplaySize()
w, h = self.GetSize()
x = dw - w - 40 * 2
y = dh / 2 - h / 2
self.SetPosition((x, y))
def Exit(self, e):
self.Close()
def About(self, e):
dialog = wx.AboutDialogInfo()
# dialog.SetIcon (wx.Icon('icon.ico', wx.BITMAP_TYPE_PNG))
dialog.SetIcon(self.icon)
dialog.SetName(self.application.title + ": " + self.application.long_title)
dialog.SetVersion(self.application.version)
dialog.SetCopyright(self.application.c)
dialog.SetDescription("\n".join(map(lambda x: x[4:], self.application.about.split("\n")[1:][:-1])))
dialog.SetWebSite(self.application.url)
dialog.SetLicence(self.application.license)
wx.AboutBox(dialog)
def Template_Structure_Preview(self, e):
self.application.TextWindow(
self, title="Template Structure Preview", content=self.report.template_dump_struct()
)
def HtmlUsageView(self, content):
page = ""
for i in content.split("\n\n"):
if i[:2] == "# ":
page += "<h1>" + cgi.escape(i[2:]) + "</h1>"
elif i[:3] == "## ":
page += "<h2>" + cgi.escape(i[3:]) + "</h2>"
else:
page += "<pre>" + cgi.escape(i) + "</pre>"
return (
"<html><head><style>"
"body {margin:0;padding:0}"
"h1 {font-size:20px}"
"h2 {font-size:16px}"
"pre {font-size:13px; font-family:sans-serif}"
"</style></head><body>" + page + "<p/></body></html>"
)
def Usage(self, e):
# self.application.TextWindow(self, title=self.Usage.__name__, content='\n'.join(
# map(lambda x: x[4:], self.application.usage.split('\n')[1:][:-1])))
self.application.HtmlWindow(
self,
title=self.Usage.__name__,
content=self.HtmlUsageView(
"\n".join(map(lambda x: x[4:], self.application.usage.split("\n")[1:][:-1]))
),
)
def Changelog(self, e):
self.application.TextWindow(
self,
title=self.Changelog.__name__,
content="\n".join(map(lambda x: x[4:], self.application.changelog.split("\n")[1:][:-1])),
)
def Destroy(self):
map(lambda x: x.Close(), filter(lambda x: isinstance(x, wx.Frame), self.GetChildren()))
wx.WakeUpIdle()
# print 'destroying MainWindow'
super(wx.Frame, self).Destroy()
def Open_Template(self, e):
openFileDialog = wx.FileDialog(
self,
"Open Template",
"",
"",
"XML files (*.xml)|*.xml|All files (*.*)|*.*",
wx.FD_OPEN | wx.FD_FILE_MUST_EXIST,
)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
self._open_template(openFileDialog.GetPath())
def _open_template(self, filename):
self.status("Loading template...")
self.ctrl_st_t.Enable(False)
self.ctrl_tc_t.SetValue("")
self.ctrl_st_c.Enable(False)
self.ctrl_tc_c.SetValue("")
self.menu_file_open_k.Enable(False)
self.menu_file_save_t.Enable(False)
self.menu_file_save_r.Enable(False)
self.menu_tools_template_structure_preview.Enable(False)
# if self.report:
# del self.report
# self.report = Report()
# print self.report._skel
self.report.template_load_xml(filename, clean=self.menu_view_c.IsChecked())
if self.menu_view_y.IsChecked():
self.ctrl_tc_t.SetValue(self.report.template_dump_yaml())
else:
self.ctrl_tc_t.SetValue(self.report.template_dump_json())
self.ctrl_st_t.Enable(True)
self.ctrl_tc_c.Enable(True)
self.ctrl_tc_c.SetBackgroundColour(self.color_tc_bg_e)
self.ctrl_tc_k.Enable(True)
self.ctrl_tc_k.SetBackgroundColour(self.color_tc_bg_e)
self.menu_file_open_k.Enable(True)
self.menu_file_open_c.Enable(True)
self.menu_file_save_t.Enable(True)
self.menu_tools_template_structure_preview.Enable(True)
if self.scan:
self.menu_file_save_r.Enable(True)
if self.ctrl_st_s.IsEnabled():
self.menu_tools_merge_scan_into_content.Enable(True)
self.status("Template loaded")
def Open_Content(self, e):
openFileDialog = wx.FileDialog(
self,
"Open Content",
"",
"",
"Content files (*.yaml; *.json)|*.yaml;*.json|All files (*.*)|*.*",
wx.FD_OPEN | wx.FD_FILE_MUST_EXIST,
)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
self._open_content(openFileDialog.GetPath())
def __show_content(self):
if self.menu_view_y.IsChecked():
self.ctrl_tc_c.SetValue(self.report.content_dump_yaml())
else:
self.ctrl_tc_c.SetValue(self.report.content_dump_json())
def _open_content(self, filename):
self.status("Loading content...")
self.ctrl_st_c.Enable(False)
json_ext = ".json"
if filename[-len(json_ext) :] == json_ext:
self.report.content_load_json(filename)
else:
self.report.content_load_yaml(filename)
self.save_into_directory = os.path.dirname(filename)
self.__show_content()
self.ctrl_st_c.Enable(True)
self.menu_file_save_r.Enable(True)
self.menu_file_save_c.Enable(True)
# if self.scan:
# self.menu_tools_merge_scan_into_content.Enable(True)
self.status("Content loaded")
def Open_Scan(self, e):
openFileDialog = wx.FileDialog(
self,
"Open Scan",
"",
"",
"Scan files (*.xml; *.yaml; *.json)|*.xml;*.yaml;*.json|All files (*.*)|*.*",
wx.FD_OPEN | wx.FD_FILE_MUST_EXIST,
)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
self._open_scan(openFileDialog.GetPath())
def _open_scan(self, filename):
self.status("Loading scan...")
self.menu_file_save_s.Enable(False)
if self.scan is not None:
del self.scan
self.scan = Scan(filename, requests_and_responses=self.menu_view_r.IsChecked())
if self.menu_view_y.IsChecked():
self.ctrl_tc_s.SetValue(self.scan.dump_yaml(truncate=self.menu_view_i.IsChecked()))
else:
self.ctrl_tc_s.SetValue(self.scan.dump_json(truncate=self.menu_view_i.IsChecked()))
self.ctrl_st_s.Enable(True)
self.menu_file_save_s.Enable(True)
self.menu_file_save_r.Enable(True)
# if self.ctrl_st_c.IsEnabled():
if self.ctrl_st_t.IsEnabled():
self.menu_tools_merge_scan_into_content.Enable(True)
self.status("Scan loaded")
def Merge_Scan_Into_Content(self, e):
self.status("Merging Scan into Content...")
self.report.merge_scan(self.scan.modify(truncate=self.menu_view_i.IsChecked()))
self.ctrl_st_c.Enable(True)
self.menu_file_save_c.Enable(True)
self.menu_file_save_s.Enable(False)
del self.scan
self.scan = None
self.ctrl_tc_s.SetValue("")
self.ctrl_st_s.Enable(False)
self.menu_file_save_s.Enable(False)
self.__show_content()
self.menu_tools_merge_scan_into_content.Enable(False)
self.report.content_refresh()
self.status("Merged")
def Generate_few_passwords(self, e):
self.application.TextWindow(
self, title="Random Password Generator", content="\n".join(pwgen.Few(15)), size=(235, 270)
)
# def Open_Knowledge_Base (self, e):
# pass
# def Generate_Content (self, e):
# pass
# def Generate_Knowledge_Base (self, e):
# pass
# def Generate_Report (self, e):
# pass
def Save_Template_As(self, e):
openFileDialog = wx.FileDialog(
self,
"Save Template As",
self.save_into_directory,
"",
"Content files (*.yaml; *.json)|*.yaml;*.json|All files (*.*)|*.*",
wx.FD_SAVE | wx.wx.FD_OVERWRITE_PROMPT,
)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
json_ext = ".json"
filename = openFileDialog.GetPath()
self.status("Saving Template content...")
h = open(filename, "w")
if filename[-len(json_ext) :] == json_ext:
h.write(self.report.template_dump_json().encode("utf-8"))
else:
h.write(self.report.template_dump_yaml().encode("utf-8"))
h.close()
self.status("Template content saved")
def Save_Content_As(self, e):
openFileDialog = wx.FileDialog(
self,
"Save Content As",
self.save_into_directory,
"",
"Content files (*.yaml; *.json)|*.yaml;*.json|All files (*.*)|*.*",
wx.FD_SAVE | wx.wx.FD_OVERWRITE_PROMPT,
)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
json_ext = ".json"
filename = openFileDialog.GetPath()
self.status("Saving Content...")
h = open(filename, "w")
if filename[-len(json_ext) :] == json_ext:
h.write(self.report.content_dump_json().encode("utf-8"))
else:
h.write(self.report.content_dump_yaml().encode("utf-8"))
h.close()
self.status("Content saved")
def Save_Scan_As(self, e):
openFileDialog = wx.FileDialog(
self,
"Save Scan As",
self.save_into_directory,
"",
"Content files (*.yaml; *.json)|*.yaml;*.json|All files (*.*)|*.*",
wx.FD_SAVE | wx.wx.FD_OVERWRITE_PROMPT,
)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
json_ext = ".json"
filename = openFileDialog.GetPath()
h = open(filename, "w")
if filename[-len(json_ext) :] == json_ext:
h.write(self.scan.dump_json(truncate=self.menu_view_i.IsChecked()).encode("utf-8"))
else:
h.write(self.scan.dump_yaml(truncate=self.menu_view_i.IsChecked()).encode("utf-8"))
h.close()
self.status("Scan saved")
def _refresh(self):
self.status("Reloading previews...")
if self.menu_view_y.IsChecked():
self._Use_yaml()
else:
self._Use_json()
self.status("Ready")
def _clean_template(self, force=False):
if force == True or self.report.template_cleanup_required == True:
if self.ctrl_st_t.IsEnabled():
self.report.template_reload(clean=self.menu_view_c.IsChecked())
if self.ctrl_st_c.IsEnabled():
self.report.content_reload()
if self.ctrl_st_k.IsEnabled():
self.report.kb_reload()
self._refresh()
# print 'cleanup performed.'
# else:
# print 'cleanup omitted.'
def Clean_template(self, e):
self._clean_template(force=True)
# def Save_Knowledge_Base_As (self, e):
# pass
def Save_Report_As(self, e):
openFileDialog = wx.FileDialog(
self,
"Save Report As",
self.save_into_directory,
"",
"XML files (*.xml)|*.xml|All files (*.*)|*.*",
wx.FD_SAVE | wx.wx.FD_OVERWRITE_PROMPT,
)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
filename = openFileDialog.GetPath()
if filename == self.report._template_filename:
wx.MessageBox(
"For safety reasons, template overwriting with generated report is not allowed!",
"Error",
wx.OK | wx.ICON_ERROR,
)
return
self.status("Generating and saving the report...")
self.report.scan = self.scan
self._clean_template()
# self.report.xml_apply_meta()
self.report.xml_apply_meta(
vulnparam_highlighting=self.menu_view_v.IsChecked(), truncation=self.menu_view_i.IsChecked()
)
self.report.save_report_xml(filename)
# self._clean_template()
self.status("Report saved")
def _Use_yaml(self):
if self.ctrl_st_t.IsEnabled():
self.ctrl_tc_t.SetValue(self.report.template_dump_yaml())
if self.ctrl_st_c.IsEnabled():
self.ctrl_tc_c.SetValue(self.report.content_dump_yaml())
if self.ctrl_st_s.IsEnabled():
self.ctrl_tc_s.SetValue(self.scan.dump_yaml(truncate=self.menu_view_i.IsChecked()))
if self.ctrl_st_k.IsEnabled():
self.ctrl_tc_k.SetValue(self.report.kb_dump_yaml())
def Use_yaml(self, e):
self._Use_yaml()
def _Use_json(self):
if self.ctrl_st_t.IsEnabled():
self.ctrl_tc_t.SetValue(self.report.template_dump_json())
if self.ctrl_st_c.IsEnabled():
self.ctrl_tc_c.SetValue(self.report.content_dump_json())
if self.ctrl_st_s.IsEnabled():
self.ctrl_tc_s.SetValue(self.scan.dump_json(truncate=self.menu_view_i.IsChecked()))
if self.ctrl_st_k.IsEnabled():
self.ctrl_tc_k.SetValue(self.report.kb_dump_json())
def Use_json(self, e):
self._Use_json()
# def Status_Preview(self, e):
# if self.menu_view_s.IsChecked():
# self.statusbar.Show()
# self.SetSize((-1,self.GetSize()[1]+self._statusbar_h,))
# else:
# self.statusbar.Hide()
# self.SetSize((-1,self.GetSize()[1]-self._statusbar_h,))
def status(self, text, hint=False):
# print text
self.SetStatusText(["", "Hint: "][hint] + text)
def VulnParam_highlighting(self, e):
pass
def Viewstate_truncation(self, e):
self._refresh()
def Always_on_top(self, e):
if self.menu_view_t.IsChecked():
self.SetWindowStyle(self.GetWindowStyle() | wx.STAY_ON_TOP)
for i in self.children:
i.SetWindowStyle(i.GetWindowStyle() | wx.STAY_ON_TOP)
else:
self.SetWindowStyle(self.GetWindowStyle() & ~wx.STAY_ON_TOP)
for i in self.children:
i.SetWindowStyle(i.GetWindowStyle() & ~wx.STAY_ON_TOP)
def Open_Knowledge_Base(self, e):
openFileDialog = wx.FileDialog(
self,
"Open Knowledge Base",
"",
"",
"Knowledge Base files (*.yaml; *.json; *.csv)|*.yaml;*.json;*.csv|All files (*.*)|*.*",
wx.FD_OPEN | wx.FD_FILE_MUST_EXIST,
)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
self._open_kb(openFileDialog.GetPath())
def _open_kb(self, filename):
self.ctrl_st_k.Enable(False)
csv_ext = ".csv"
json_ext = ".json"
if filename[-len(csv_ext) :].lower() == csv_ext:
self.report.kb_load_csv(filename)
elif filename[-len(json_ext) :] == json_ext:
self.report.kb_load_json(filename)
else:
self.report.kb_load_yaml(filename)
if self.menu_view_y.IsChecked():
self.ctrl_tc_k.SetValue(self.report.kb_dump_yaml())
else:
self.ctrl_tc_k.SetValue(self.report.kb_dump_json())
self.ctrl_st_k.Enable(True)
# self.menu_file_save_k.Enable (True)
self.menu_file_save_r.Enable(True)
class __MainWindow(wx.Frame):
# Variables set during __init__
#report
#application
#scan
#template
#icon
#statusbar
#menu_file_open_c
#menu_file_open_k
#menu_file_generate_c
#menu_file_generate_k
#menu_file_generate_r
#menu_file_save_t
#menu_file_save_c
#menu_file_save_s
#menu_file_save_k
#menu_file_save_r
#menu_view_c
#menu_view_y
#menu_view_j
#menu_view_s
#menu_view_v
#menu_view_i
#menu_view_r
#menu_view_t
#menu_tools_template_structure_preview
#menu_tools_merge_scan_into_content
#menu_tools_generate_few_passwords
#ctrl_st_t
#ctrl_tc_t
#ctrl_st_c
#ctrl_tc_c
#ctrl_tc_c_b
#ctrl_st_s
#ctrl_tc_s
#ctrl_tc_s_b
#ctrl_st_k
#ctrl_tc_k
#ctrl_st_r
#ctrl_tc_r
#color_tc_bg_e
#color_tc_bg_d
#_statusbar_h
#children
def __init__(self,
application=None,
parent=None,
*args,
**kwargs): #style=wx.DEFAULT_FRAME_STYLE^wx.RESIZE_BORDER
self.children = []
#self.report = None
self.report = Report()
self.application = application
self.scan = None
wx.Frame.__init__(self,
parent,
title=self.application.title + ' ' +
self.application.version,
*args,
**kwargs) #style=style
self.Bind(wx.EVT_CLOSE, lambda x: self.Destroy())
myStream = cStringIO.StringIO(base64.b64decode(icon))
myImage = wx.ImageFromStream(myStream)
myBitmap = wx.BitmapFromImage(myImage)
self.icon = wx.EmptyIcon()
self.icon.CopyFromBitmap(myBitmap)
self.SetIcon(self.icon)
# Menu arrangement
menu = wx.MenuBar()
class Index(object):
def __init__(self, current):
self.__current = current - 1
@property
def current(self):
return self.__current
@current.setter
def current(self, x):
self.__current = x
def next(self):
self.__current += 1
return self.__current
index = Index(100)
menu_file = wx.Menu()
menu_file.Append(index.next(), 'Open Report &Template...')
self.Bind(wx.EVT_MENU, self.Open_Template, id=index.current)
self.menu_file_open_c = menu_file.Append(index.next(),
'Open &Content...')
self.menu_file_open_c.Enable(False)
self.Bind(wx.EVT_MENU, self.Open_Content, id=index.current)
menu_file.Append(index.next(), 'Open &Scan...')
self.Bind(wx.EVT_MENU, self.Open_Scan, id=index.current)
self.menu_file_open_k = menu_file.Append(
index.next(), 'Open &Knowledge Base...')
self.menu_file_open_k.Enable(False)
self.Bind(wx.EVT_MENU, self.Open_Knowledge_Base, id=index.current)
#menu_file.AppendSeparator()
#self.menu_file_generate_c = menu_file.Append(index.next(), '&Generate Content')
#self.menu_file_generate_c.Enable(False)
#self.Bind(wx.EVT_MENU, self.Generate_Content, id=index.current)
#self.menu_file_generate_k = menu_file.Append(index.next(), 'G&enerate Knowledge Base')
#self.menu_file_generate_k.Enable(False)
#self.Bind(wx.EVT_MENU, self.Generate_Knowledge_Base, id=index.current)
#self.menu_file_generate_r = menu_file.Append(index.next(), 'Ge&nerate Report')
#self.menu_file_generate_r.Enable(False)
#self.Bind(wx.EVT_MENU, self.Generate_Report, id=index.current)
menu_file.AppendSeparator()
self.menu_file_save_t = menu_file.Append(index.next(),
'&Save Template As...')
self.menu_file_save_t.Enable(False)
self.Bind(wx.EVT_MENU, self.Save_Template_As, id=index.current)
self.menu_file_save_c = menu_file.Append(index.next(),
'Sav&e Content As...')
self.menu_file_save_c.Enable(False)
self.Bind(wx.EVT_MENU, self.Save_Content_As, id=index.current)
#self.menu_file_save_k = menu_file.Append(index.next(), 'S&ave Knowledge Base As...')
#self.menu_file_save_k.Enable(False)
#self.Bind(wx.EVT_MENU, self.Save_Knowledge_Base_As, id=index.current)
self.menu_file_save_s = menu_file.Append(index.next(),
'Sa&ve Scan As...')
self.menu_file_save_s.Enable(False)
self.Bind(wx.EVT_MENU, self.Save_Scan_As, id=index.current)
self.menu_file_save_r = menu_file.Append(index.next(),
'Save &Report As...')
self.menu_file_save_r.Enable(False)
self.Bind(wx.EVT_MENU, self.Save_Report_As, id=index.current)
menu_file.AppendSeparator()
menu_file.Append(wx.ID_EXIT, 'E&xit\tCtrl+Q', 'Exit application')
self.Bind(wx.EVT_MENU, self.Exit, id=wx.ID_EXIT)
menu.Append(menu_file, '&File')
menu_view = wx.Menu()
self.menu_view_c = menu_view.Append(index.next(),
'C&lean template',
kind=wx.ITEM_CHECK)
self.Bind(wx.EVT_MENU, self.Clean_template, id=index.current)
self.menu_view_c.Check(True)
menu_view.AppendSeparator()
self.menu_view_y = menu_view.Append(index.next(),
'&yaml',
kind=wx.ITEM_RADIO)
self.Bind(wx.EVT_MENU, self.Use_yaml, id=index.current)
self.menu_view_j = menu_view.Append(index.next(),
'&json',
kind=wx.ITEM_RADIO)
self.Bind(wx.EVT_MENU, self.Use_json, id=index.current)
self.menu_view_y.Check(True)
menu.Append(menu_view, '&View')
#menu_view.AppendSeparator()
#self.menu_view_s = menu_view.Append(index.next(), '&Status Preview', kind=wx.ITEM_CHECK)
#self.Bind(wx.EVT_MENU, self.Status_Preview, id=index.current)
#self.menu_view_s.Check(False)
menu_view.AppendSeparator()
self.menu_view_v = menu_view.Append(index.next(),
'&VulnParam highlighting',
kind=wx.ITEM_CHECK)
self.Bind(wx.EVT_MENU,
self.VulnParam_highlighting,
id=index.current)
self.menu_view_v.Check(True)
self.menu_view_i = menu_view.Append(
index.next(),
'V&iewState truncation',
'Warning! Application performance will noticeably decrease!',
kind=wx.ITEM_CHECK)
self.Bind(wx.EVT_MENU, self.Viewstate_truncation, id=index.current)
self.menu_view_i.Check(True)
self.menu_view_r = menu_view.Append(
index.next(),
'Include &requests and responses',
'Warning! Have a small scan or be very patient!',
kind=wx.ITEM_CHECK)
self.menu_view_r.Check(False)
menu_view.AppendSeparator()
self.menu_view_t = menu_view.Append(index.next(),
'Always on &top',
kind=wx.ITEM_CHECK)
self.Bind(wx.EVT_MENU, self.Always_on_top, id=index.current)
self.menu_view_t.Check(True)
menu_tools = wx.Menu()
self.menu_tools_template_structure_preview = menu_tools.Append(
index.next(), 'Te&mplate structure preview')
self.menu_tools_template_structure_preview.Enable(False)
self.Bind(wx.EVT_MENU,
self.Template_Structure_Preview,
id=index.current)
self.menu_tools_merge_scan_into_content = menu_tools.Append(
index.next(), 'Mer&ge Scan into Content')
self.menu_tools_merge_scan_into_content.Enable(False)
self.Bind(wx.EVT_MENU,
self.Merge_Scan_Into_Content,
id=index.current)
self.menu_tools_generate_few_passwords = menu_tools.Append(
index.next(), 'Generate &few passwords')
self.Bind(wx.EVT_MENU,
self.Generate_few_passwords,
id=index.current)
menu.Append(menu_tools, '&Tools')
menu_help = wx.Menu()
menu_help.Append(index.next(), '&Usage')
self.Bind(wx.EVT_MENU, self.Usage, id=index.current)
menu_help.Append(index.next(), '&Changelog')
self.Bind(wx.EVT_MENU, self.Changelog, id=index.current)
menu_help.AppendSeparator()
menu_help.Append(wx.ID_ABOUT, '&About')
self.Bind(wx.EVT_MENU, self.About, id=wx.ID_ABOUT)
menu.Append(menu_help, '&Help')
self.SetMenuBar(menu)
# Frame layout arrangement
class FileDropTarget(wx.FileDropTarget):
def __init__(self, target, handler):
wx.FileDropTarget.__init__(self)
self.target = target
self.handler = handler
def OnDropFiles(self, x, y, filenames):
self.handler(filenames)
panel = wx.Panel(self)
vbox = wx.BoxSizer(wx.VERTICAL)
fgs = wx.FlexGridSizer(5, 2, 9, 25)
# Template
self.ctrl_st_t = wx.StaticText(panel, label='Template:')
self.ctrl_st_t.Enable(False)
self.ctrl_tc_t = wx.TextCtrl(panel,
style=wx.TE_MULTILINE
| wx.TE_READONLY,
size=(
200,
3 * 17,
))
def ctrl_tc_t_OnFocus(e):
self.ctrl_tc_t.ShowNativeCaret(False)
# for unknown reason this refuse to work in wxpython 3.0
e.Skip()
def ctrl_tc_t_OnDoubleclick(e):
if self.ctrl_st_t.IsEnabled():
self.application.TextWindow(
self,
title='Template Preview',
content=self.ctrl_tc_t.GetValue())
e.Skip()
self.ctrl_tc_t.Bind(wx.EVT_SET_FOCUS, ctrl_tc_t_OnFocus)
self.ctrl_tc_t.Bind(wx.EVT_LEFT_DCLICK, ctrl_tc_t_OnDoubleclick)
def ctrl_tc_t_OnMouseOver(e):
self.status('You might use drag & drop', hint=True)
e.Skip()
#def ctrl_tc_t_OnMouseLeave(e):
# self.status('')
# e.Skip()
self.ctrl_tc_t.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_t_OnMouseOver)
#self.ctrl_tc_t.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_t_OnMouseLeave)
def ctrl_tc_t_OnDropFiles(filenames):
if len(filenames) != 1:
wx.MessageBox('Single file is expected!', 'Error',
wx.OK | wx.ICON_ERROR)
return
self._open_template(filenames[0])
ctrl_tc_t_dt = FileDropTarget(self.ctrl_tc_t,
ctrl_tc_t_OnDropFiles)
self.ctrl_tc_t.SetDropTarget(ctrl_tc_t_dt)
fgs.AddMany([(self.ctrl_st_t, 1, wx.EXPAND),
(self.ctrl_tc_t, 1, wx.EXPAND)])
# Content + Edit
self.ctrl_st_c = wx.StaticText(panel, label='Content:')
self.ctrl_st_c.Enable(False)
self.ctrl_tc_c = wx.TextCtrl(panel,
style=wx.TE_MULTILINE
| wx.TE_READONLY,
size=(
200,
3 * 17,
))
self.color_tc_bg_e = self.ctrl_tc_c.GetBackgroundColour()
self.ctrl_tc_c.Enable(False)
self.color_tc_bg_d = wx.SystemSettings.GetColour(
wx.SYS_COLOUR_BTNFACE)
self.ctrl_tc_c.SetBackgroundColour(self.color_tc_bg_d)
def ctrl_tc_c_OnFocus(e):
self.ctrl_tc_c.ShowNativeCaret(False)
e.Skip()
def ctrl_tc_c_OnDoubleclick(e):
if self.ctrl_st_c.IsEnabled():
self.application.TextWindow(
self,
title='Content Preview',
content=self.ctrl_tc_c.GetValue())
e.Skip()
self.ctrl_tc_c.Bind(wx.EVT_SET_FOCUS, ctrl_tc_c_OnFocus)
self.ctrl_tc_c.Bind(wx.EVT_LEFT_DCLICK, ctrl_tc_c_OnDoubleclick)
def ctrl_tc_c_b_onClick(e):
self.application.YamledWindowWrapper(
self, title='Content', content=self.report._content)
pass
def ctrl_tc_c_onResize(e):
size = self.ctrl_tc_c.GetSize()
self.ctrl_tc_c_b.SetPosition((size[0] - 36 - 1, -1))
self.ctrl_tc_c_b = wx.Button(self.ctrl_tc_c,
index.next(),
'E',
size=(16, 16))
self.ctrl_tc_c_b.Bind(wx.EVT_BUTTON, ctrl_tc_c_b_onClick)
self.ctrl_tc_c.Bind(wx.EVT_SIZE, ctrl_tc_c_onResize)
self.ctrl_tc_c_b.Hide()
def ctrl_tc_c_b_OnMouseOver(e):
self.status('Send Content to Yaml Editor', hint=True)
e.Skip()
#def ctrl_tc_c_b_OnMouseLeave(e):
# self.status('')
# e.Skip()
self.ctrl_tc_c_b.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_c_b_OnMouseOver)
#self.ctrl_tc_c_b.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_c_b_OnMouseLeave)
def ctrl_tc_c_OnMouseOver(e):
if self.ctrl_st_c.IsEnabled(): # Yamled
self.ctrl_tc_c_b.Show()
self.status('You might use drag & drop', hint=True)
e.Skip()
#def ctrl_tc_c_OnMouseLeave(e):
# self.status('')
# e.Skip()
self.ctrl_tc_c.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_c_OnMouseOver)
#self.ctrl_tc_c.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_c_OnMouseLeave)
def ctrl_tc_c_OnDropFiles(filenames):
if len(filenames) != 1:
wx.MessageBox('Single file is expected!', 'Error',
wx.OK | wx.ICON_ERROR)
return
self._open_content(filenames[0])
if self.ctrl_st_c.IsEnabled(): # Yamled
self.ctrl_tc_c_b.Show()
ctrl_tc_c_dt = FileDropTarget(self.ctrl_tc_c,
ctrl_tc_c_OnDropFiles)
self.ctrl_tc_c.SetDropTarget(ctrl_tc_c_dt)
fgs.AddMany([(self.ctrl_st_c, 1, wx.EXPAND),
(self.ctrl_tc_c, 1, wx.EXPAND)])
# Scan + Edit button
self.ctrl_st_s = wx.StaticText(panel, label='Scan:')
self.ctrl_st_s.Enable(False)
self.ctrl_tc_s = wx.TextCtrl(panel,
style=wx.TE_MULTILINE
| wx.TE_READONLY,
size=(
200,
3 * 17,
))
def ctrl_tc_s_OnFocus(e):
self.ctrl_tc_s.ShowNativeCaret(False)
e.Skip()
def ctrl_tc_s_OnDoubleclick(e):
if self.ctrl_st_s.IsEnabled():
self.application.TextWindow(
self,
title='Scan Preview',
content=self.ctrl_tc_s.GetValue())
e.Skip()
self.ctrl_tc_s.Bind(wx.EVT_SET_FOCUS, ctrl_tc_s_OnFocus)
self.ctrl_tc_s.Bind(wx.EVT_LEFT_DCLICK, ctrl_tc_s_OnDoubleclick)
def ctrl_tc_s_b_onClick(e):
self.application.YamledWindowWrapper(self,
title='Scan',
content=self.scan._scan)
pass
def ctrl_tc_s_onResize(e):
size = self.ctrl_tc_s.GetSize()
self.ctrl_tc_s_b.SetPosition((size[0] - 36 - 1, -1))
self.ctrl_tc_s_b = wx.Button(self.ctrl_tc_s,
index.next(),
'E',
size=(16, 16))
self.ctrl_tc_s_b.Bind(wx.EVT_BUTTON, ctrl_tc_s_b_onClick)
self.ctrl_tc_s.Bind(wx.EVT_SIZE, ctrl_tc_s_onResize)
self.ctrl_tc_s_b.Hide()
def ctrl_tc_s_b_OnMouseOver(e):
self.status('Send Scan to Yaml Editor', hint=True)
e.Skip()
#def ctrl_tc_s_b_OnMouseLeave(e):
# self.status('')
# e.Skip()
self.ctrl_tc_s_b.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_s_b_OnMouseOver)
#self.ctrl_tc_s_b.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_s_b_OnMouseLeave)
def ctrl_tc_s_OnMouseOver(e):
if self.ctrl_st_s.IsEnabled(): # Yamled
self.ctrl_tc_s_b.Show()
self.status('You might use drag & drop', hint=True)
e.Skip()
#def ctrl_tc_s_OnMouseLeave(e):
# #self.ctrl_tc_s_b.Hide()
# self.status('')
# e.Skip()
self.ctrl_tc_s.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_s_OnMouseOver)
#self.ctrl_tc_s.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_s_OnMouseLeave)
def ctrl_tc_s_OnDropFiles(filenames):
if len(filenames) != 1:
wx.MessageBox('Single file is expected!', 'Error',
wx.OK | wx.ICON_ERROR)
return
self._open_scan(filenames[0])
if self.ctrl_st_s.IsEnabled(): # Yamled
self.ctrl_tc_s_b.Show()
ctrl_tc_s_dt = FileDropTarget(self.ctrl_tc_s,
ctrl_tc_s_OnDropFiles)
self.ctrl_tc_s.SetDropTarget(ctrl_tc_s_dt)
fgs.AddMany([(self.ctrl_st_s, 1, wx.EXPAND),
(self.ctrl_tc_s, 1, wx.EXPAND)])
# Knowledge Base
self.ctrl_st_k = wx.StaticText(panel, label='Knowledge Base:')
self.ctrl_st_k.Enable(False)
self.ctrl_tc_k = wx.TextCtrl(panel,
style=wx.TE_MULTILINE
| wx.TE_READONLY,
size=(
200,
3 * 17,
))
self.ctrl_tc_k.Enable(False)
self.ctrl_tc_k.SetBackgroundColour(self.color_tc_bg_d)
def ctrl_tc_k_OnFocus(e):
self.ctrl_tc_k.ShowNativeCaret(False)
e.Skip()
def ctrl_tc_k_OnDoubleclick(e):
if self.ctrl_st_k.IsEnabled():
self.application.TextWindow(
self,
title='KB Preview',
content=self.ctrl_tc_k.GetValue())
e.Skip()
self.ctrl_tc_k.Bind(wx.EVT_SET_FOCUS, ctrl_tc_k_OnFocus)
self.ctrl_tc_k.Bind(wx.EVT_LEFT_DCLICK, ctrl_tc_k_OnDoubleclick)
def ctrl_tc_k_OnMouseOver(e):
self.status('You might use drag & drop', hint=True)
e.Skip()
def ctrl_tc_k_OnMouseLeave(e):
self.status('')
e.Skip()
self.ctrl_tc_k.Bind(wx.EVT_ENTER_WINDOW, ctrl_tc_k_OnMouseOver)
self.ctrl_tc_k.Bind(wx.EVT_LEAVE_WINDOW, ctrl_tc_k_OnMouseLeave)
def ctrl_tc_k_OnDropFiles(filenames):
if len(filenames) != 1:
wx.MessageBox('Single file is expected!', 'Error',
wx.OK | wx.ICON_ERROR)
return
self._open_kb(filenames[0])
ctrl_tc_k_dt = FileDropTarget(self.ctrl_tc_k,
ctrl_tc_k_OnDropFiles)
self.ctrl_tc_k.SetDropTarget(ctrl_tc_k_dt)
fgs.AddMany([(self.ctrl_st_k, 1, wx.EXPAND),
(self.ctrl_tc_k, 1, wx.EXPAND)])
def panel_OnMouseOver(e):
self.status('')
self.ctrl_tc_c_b.Hide()
self.ctrl_tc_s_b.Hide()
e.Skip()
panel.Bind(wx.EVT_ENTER_WINDOW, panel_OnMouseOver)
# Report
#self.ctrl_st_r = wx.StaticText(panel, label='Report:')
#self.ctrl_st_r.Enable (False)
#self.ctrl_tc_r = wx.TextCtrl(panel, style=wx.TE_MULTILINE|wx.TE_READONLY, size=(200, 3*17,))
#self.ctrl_tc_r.Enable(False)
#self.ctrl_tc_r.SetBackgroundColour (self.color_tc_bg_d)
#def ctrl_tc_r_OnFocus (e):
# self.ctrl_tc_r.ShowNativeCaret (False)
# e.Skip()
#self.ctrl_tc_r.Bind (wx.EVT_SET_FOCUS, ctrl_tc_r_OnFocus)
#fgs.AddMany ([(self.ctrl_st_r, 1, wx.EXPAND), (self.ctrl_tc_r, 1, wx.EXPAND)])
fgs.AddGrowableRow(0, 1)
fgs.AddGrowableRow(1, 1)
fgs.AddGrowableRow(2, 1)
fgs.AddGrowableRow(3, 1)
#fgs.AddGrowableRow (4, 1)
fgs.AddGrowableCol(1, 1)
vbox.Add(fgs, proportion=1, flag=wx.ALL | wx.EXPAND, border=10)
#data = wx.TextCtrl(panel)
#hbox1 = wx.BoxSizer (wx.HORIZONTAL)
#hbox1.Add(data, proportion=1, flag=wx.LEFT|wx.RIGHT|wx.BOTTOM|wx.EXPAND, border=10)
#vbox.Add (hbox1, 0, wx.ALL|wx.EXPAND, 0)
panel.SetSizer(vbox)
vbox.Fit(self)
#self.SetMinSize(self.GetSize())
self.statusbar = self.CreateStatusBar()
self._statusbar_h = self.statusbar.GetSize()[1]
#self.statusbar.Hide()
self.status('Started')
self.SetMinSize((
self.GetSize()[0],
self.GetSize()[1] + self._statusbar_h,
))
#panel = wx.Panel (self)
#vbox = wx.BoxSizer (wx.VERTICAL)
#hbox1 = wx.BoxSizer (wx.HORIZONTAL)
##st1 = wx.StaticText (panel, wx.ID_ANY, label='Not yet ready')
#st1 = wx.StaticText (panel, wx.ID_ANY, label='Template:', size=(100, -1,))
#hbox1.Add (st1, 0, wx.ALL, 5)
#tc1 = wx.TextCtrl (panel, wx.ID_ANY, size=(300, -1,))
#hbox1.Add (tc1, 1, wx.ALL|wx.EXPAND, 0)
#vbox.Add (hbox1, 0, wx.ALL|wx.EXPAND, 0)
#hbox2 = wx.BoxSizer (wx.HORIZONTAL)
#st2 = wx.StaticText (panel, wx.ID_ANY, label='Scan:', size=(100, -1,))
#hbox2.Add (st2, 0, wx.ALL, 5)
#tc2 = wx.TextCtrl (panel, wx.ID_ANY, size=(300, -1,))
#hbox2.Add (tc2, 1, wx.ALL|wx.EXPAND, 0)
#vbox.Add (hbox2, 0, wx.ALL|wx.EXPAND, 0)
##vbox.Add (hbox1, 0, wx.CENTER, 5)
#panel.SetSizer (vbox)
#vbox.Fit (self)
#self.Center()
self.alignVMiddleRight()
self.Show()
self.Always_on_top(None)
#print 'loaded'
def alignVMiddleRight(self):
dw, dh = wx.DisplaySize()
w, h = self.GetSize()
x = dw - w - 40 * 2
y = dh / 2 - h / 2
self.SetPosition((x, y))
def Exit(self, e):
self.Close()
def About(self, e):
dialog = wx.AboutDialogInfo()
#dialog.SetIcon (wx.Icon('icon.ico', wx.BITMAP_TYPE_PNG))
dialog.SetIcon(self.icon)
dialog.SetName(self.application.title + ': ' +
self.application.long_title)
dialog.SetVersion(self.application.version)
dialog.SetCopyright(self.application.c)
dialog.SetDescription('\n'.join(
map(lambda x: x[4:],
self.application.about.split('\n')[1:][:-1])))
dialog.SetWebSite(self.application.url)
dialog.SetLicence(self.application.license)
wx.AboutBox(dialog)
def Template_Structure_Preview(self, e):
self.application.TextWindow(
self,
title='Template Structure Preview',
content=self.report.template_dump_struct())
def HtmlUsageView(self, content):
page = ''
for i in content.split('\n\n'):
if i[:2] == '# ':
page += '<h1>' + cgi.escape(i[2:]) + '</h1>'
elif i[:3] == '## ':
page += '<h2>' + cgi.escape(i[3:]) + '</h2>'
else:
page += '<pre>' + cgi.escape(i) + '</pre>'
return ('<html><head><style>'
'body {margin:0;padding:0}'
'h1 {font-size:20px}'
'h2 {font-size:16px}'
'pre {font-size:13px; font-family:sans-serif}'
'</style></head><body>' + page + '<p/></body></html>')
def Usage(self, e):
#self.application.TextWindow(self, title=self.Usage.__name__, content='\n'.join(
# map(lambda x: x[4:], self.application.usage.split('\n')[1:][:-1])))
self.application.HtmlWindow(
self,
title=self.Usage.__name__,
content=self.HtmlUsageView('\n'.join(
map(lambda x: x[4:],
self.application.usage.split('\n')[1:][:-1]))))
def Changelog(self, e):
self.application.TextWindow(
self,
title=self.Changelog.__name__,
content='\n'.join(
map(lambda x: x[4:],
self.application.changelog.split('\n')[1:][:-1])))
def Destroy(self):
map(lambda x: x.Close(),
filter(lambda x: isinstance(x, wx.Frame), self.GetChildren()))
wx.WakeUpIdle()
#print 'destroying MainWindow'
super(wx.Frame, self).Destroy()
def Open_Template(self, e):
openFileDialog = wx.FileDialog(
self, 'Open Template', '', '',
'XML files (*.xml)|*.xml|All files (*.*)|*.*',
wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
self._open_template(openFileDialog.GetPath())
def _open_template(self, filename):
self.status('Loading template...')
self.ctrl_st_t.Enable(False)
self.ctrl_tc_t.SetValue('')
self.ctrl_st_c.Enable(False)
self.ctrl_tc_c.SetValue('')
self.menu_file_open_k.Enable(False)
self.menu_file_save_t.Enable(False)
self.menu_file_save_r.Enable(False)
self.menu_tools_template_structure_preview.Enable(False)
#if self.report:
# del self.report
#self.report = Report()
#print self.report._skel
self.report.template_load_xml(filename,
clean=self.menu_view_c.IsChecked())
if self.menu_view_y.IsChecked():
self.ctrl_tc_t.SetValue(self.report.template_dump_yaml())
else:
self.ctrl_tc_t.SetValue(self.report.template_dump_json())
self.ctrl_st_t.Enable(True)
self.ctrl_tc_c.Enable(True)
self.ctrl_tc_c.SetBackgroundColour(self.color_tc_bg_e)
self.ctrl_tc_k.Enable(True)
self.ctrl_tc_k.SetBackgroundColour(self.color_tc_bg_e)
self.menu_file_open_k.Enable(True)
self.menu_file_open_c.Enable(True)
self.menu_file_save_t.Enable(True)
self.menu_tools_template_structure_preview.Enable(True)
if self.scan:
self.menu_file_save_r.Enable(True)
if self.ctrl_st_s.IsEnabled():
self.menu_tools_merge_scan_into_content.Enable(True)
self.status('Template loaded')
def Open_Content(self, e):
openFileDialog = wx.FileDialog(
self, 'Open Content', '', '',
'Content files (*.yaml; *.json)|*.yaml;*.json|All files (*.*)|*.*',
wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
self._open_content(openFileDialog.GetPath())
def __show_content(self):
if self.menu_view_y.IsChecked():
self.ctrl_tc_c.SetValue(self.report.content_dump_yaml())
else:
self.ctrl_tc_c.SetValue(self.report.content_dump_json())
def _open_content(self, filename):
self.status('Loading content...')
self.ctrl_st_c.Enable(False)
json_ext = '.json'
if filename[-len(json_ext):] == json_ext:
self.report.content_load_json(filename)
else:
self.report.content_load_yaml(filename)
self.__show_content()
self.ctrl_st_c.Enable(True)
self.menu_file_save_r.Enable(True)
self.menu_file_save_c.Enable(True)
#if self.scan:
# self.menu_tools_merge_scan_into_content.Enable(True)
self.status('Content loaded')
def Open_Scan(self, e):
openFileDialog = wx.FileDialog(
self, 'Open Scan', '', '',
'Scan files (*.xml; *.yaml; *.json)|*.xml;*.yaml;*.json|All files (*.*)|*.*',
wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
self._open_scan(openFileDialog.GetPath())
def _open_scan(self, filename):
self.status('Loading scan...')
self.menu_file_save_s.Enable(False)
if self.scan is not None:
del self.scan
self.scan = Scan(
filename, requests_and_responses=self.menu_view_r.IsChecked())
if self.menu_view_y.IsChecked():
self.ctrl_tc_s.SetValue(
self.scan.dump_yaml(truncate=self.menu_view_i.IsChecked()))
else:
self.ctrl_tc_s.SetValue(
self.scan.dump_json(truncate=self.menu_view_i.IsChecked()))
self.ctrl_st_s.Enable(True)
self.menu_file_save_s.Enable(True)
self.menu_file_save_r.Enable(True)
#if self.ctrl_st_c.IsEnabled():
if self.ctrl_st_t.IsEnabled():
self.menu_tools_merge_scan_into_content.Enable(True)
self.status('Scan loaded')
def Merge_Scan_Into_Content(self, e):
self.status('Merging Scan into Content...')
self.report.merge_scan(
self.scan.modify(truncate=self.menu_view_i.IsChecked()))
self.ctrl_st_c.Enable(True)
self.menu_file_save_c.Enable(True)
self.menu_file_save_s.Enable(False)
del self.scan
self.scan = None
self.ctrl_tc_s.SetValue('')
self.ctrl_st_s.Enable(False)
self.menu_file_save_s.Enable(False)
self.__show_content()
self.menu_tools_merge_scan_into_content.Enable(False)
self.report.content_refresh()
self.status('Merged')
def Generate_few_passwords(self, e):
self.application.TextWindow(self,
title='Random Password Generator',
content='\n'.join(pwgen.Few(15)),
size=(
235,
270,
))
#def Open_Knowledge_Base (self, e):
# pass
#def Generate_Content (self, e):
# pass
#def Generate_Knowledge_Base (self, e):
# pass
#def Generate_Report (self, e):
# pass
def Save_Template_As(self, e):
openFileDialog = wx.FileDialog(
self, 'Save Template As', '', '',
'Content files (*.yaml; *.json)|*.yaml;*.json|All files (*.*)|*.*',
wx.FD_SAVE | wx.wx.FD_OVERWRITE_PROMPT)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
json_ext = '.json'
filename = openFileDialog.GetPath()
self.status('Saving Template content...')
h = open(filename, 'w')
if filename[-len(json_ext):] == json_ext:
h.write(self.report.template_dump_json().encode('utf-8'))
else:
h.write(self.report.template_dump_yaml().encode('utf-8'))
h.close()
self.status('Template content saved')
def Save_Content_As(self, e):
openFileDialog = wx.FileDialog(
self, 'Save Content As', '', '',
'Content files (*.yaml; *.json)|*.yaml;*.json|All files (*.*)|*.*',
wx.FD_SAVE | wx.wx.FD_OVERWRITE_PROMPT)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
json_ext = '.json'
filename = openFileDialog.GetPath()
self.status('Saving Content...')
h = open(filename, 'w')
if filename[-len(json_ext):] == json_ext:
h.write(self.report.content_dump_json().encode('utf-8'))
else:
h.write(self.report.content_dump_yaml().encode('utf-8'))
h.close()
self.status('Content saved')
def Save_Scan_As(self, e):
openFileDialog = wx.FileDialog(
self, 'Save Scan As', '', '',
'Content files (*.yaml; *.json)|*.yaml;*.json|All files (*.*)|*.*',
wx.FD_SAVE | wx.wx.FD_OVERWRITE_PROMPT)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
json_ext = '.json'
filename = openFileDialog.GetPath()
h = open(filename, 'w')
if filename[-len(json_ext):] == json_ext:
h.write(
self.scan.dump_json(
truncate=self.menu_view_i.IsChecked()).encode('utf-8'))
else:
h.write(
self.scan.dump_yaml(
truncate=self.menu_view_i.IsChecked()).encode('utf-8'))
h.close()
self.status('Scan saved')
def _refresh(self):
self.status('Reloading previews...')
if self.menu_view_y.IsChecked():
self._Use_yaml()
else:
self._Use_json()
self.status('Ready')
def _clean_template(self, force=False):
if force == True or self.report.template_cleanup_required == True:
if self.ctrl_st_t.IsEnabled():
self.report.template_reload(
clean=self.menu_view_c.IsChecked())
if self.ctrl_st_c.IsEnabled():
self.report.content_reload()
if self.ctrl_st_k.IsEnabled():
self.report.kb_reload()
self._refresh()
# print 'cleanup performed.'
#else:
# print 'cleanup omitted.'
def Clean_template(self, e):
self._clean_template(force=True)
#def Save_Knowledge_Base_As (self, e):
# pass
def Save_Report_As(self, e):
openFileDialog = wx.FileDialog(
self, 'Save Report As', '', '',
'XML files (*.xml)|*.xml|All files (*.*)|*.*',
wx.FD_SAVE | wx.wx.FD_OVERWRITE_PROMPT)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
filename = openFileDialog.GetPath()
if filename == self.report._template_filename:
wx.MessageBox(
'For safety reasons, template overwriting with generated report is not allowed!',
'Error', wx.OK | wx.ICON_ERROR)
return
self.status('Generating and saving the report...')
self.report.scan = self.scan
self._clean_template()
#self.report.xml_apply_meta()
self.report.xml_apply_meta(
vulnparam_highlighting=self.menu_view_v.IsChecked(),
truncation=self.menu_view_i.IsChecked())
self.report.save_report_xml(filename)
#self._clean_template()
self.status('Report saved')
def _Use_yaml(self):
if self.ctrl_st_t.IsEnabled():
self.ctrl_tc_t.SetValue(self.report.template_dump_yaml())
if self.ctrl_st_c.IsEnabled():
self.ctrl_tc_c.SetValue(self.report.content_dump_yaml())
if self.ctrl_st_s.IsEnabled():
self.ctrl_tc_s.SetValue(
self.scan.dump_yaml(truncate=self.menu_view_i.IsChecked()))
if self.ctrl_st_k.IsEnabled():
self.ctrl_tc_k.SetValue(self.report.kb_dump_yaml())
def Use_yaml(self, e):
self._Use_yaml()
def _Use_json(self):
if self.ctrl_st_t.IsEnabled():
self.ctrl_tc_t.SetValue(self.report.template_dump_json())
if self.ctrl_st_c.IsEnabled():
self.ctrl_tc_c.SetValue(self.report.content_dump_json())
if self.ctrl_st_s.IsEnabled():
self.ctrl_tc_s.SetValue(
self.scan.dump_json(truncate=self.menu_view_i.IsChecked()))
if self.ctrl_st_k.IsEnabled():
self.ctrl_tc_k.SetValue(self.report.kb_dump_json())
def Use_json(self, e):
self._Use_json()
#def Status_Preview(self, e):
# if self.menu_view_s.IsChecked():
# self.statusbar.Show()
# self.SetSize((-1,self.GetSize()[1]+self._statusbar_h,))
# else:
# self.statusbar.Hide()
# self.SetSize((-1,self.GetSize()[1]-self._statusbar_h,))
def status(self, text, hint=False):
#print text
self.SetStatusText(['', 'Hint: '][hint] + text)
def VulnParam_highlighting(self, e):
pass
def Viewstate_truncation(self, e):
self._refresh()
def Always_on_top(self, e):
if self.menu_view_t.IsChecked():
self.SetWindowStyle(self.GetWindowStyle() | wx.STAY_ON_TOP)
for i in self.children:
i.SetWindowStyle(i.GetWindowStyle() | wx.STAY_ON_TOP)
else:
self.SetWindowStyle(self.GetWindowStyle() & ~wx.STAY_ON_TOP)
for i in self.children:
i.SetWindowStyle(i.GetWindowStyle() & ~wx.STAY_ON_TOP)
def Open_Knowledge_Base(self, e):
openFileDialog = wx.FileDialog(
self, 'Open Knowledge Base', '', '',
'Knowledge Base files (*.yaml; *.json; *.csv)|*.yaml;*.json;*.csv|All files (*.*)|*.*',
wx.FD_OPEN | wx.FD_FILE_MUST_EXIST)
if openFileDialog.ShowModal() == wx.ID_CANCEL:
return
self._open_kb(openFileDialog.GetPath())
def _open_kb(self, filename):
self.ctrl_st_k.Enable(False)
csv_ext = '.csv'
json_ext = '.json'
if filename[-len(csv_ext):].lower() == csv_ext:
self.report.kb_load_csv(filename)
elif filename[-len(json_ext):] == json_ext:
self.report.kb_load_json(filename)
else:
self.report.kb_load_yaml(filename)
if self.menu_view_y.IsChecked():
self.ctrl_tc_k.SetValue(self.report.kb_dump_yaml())
else:
self.ctrl_tc_k.SetValue(self.report.kb_dump_json())
self.ctrl_st_k.Enable(True)
#self.menu_file_save_k.Enable (True)
self.menu_file_save_r.Enable(True)
def __init__(self):
#def flagged (key):
# return key in sys.argv
def value(key):
if key in sys.argv:
index = sys.argv.index(key)
if len(sys.argv) > index + 1:
return sys.argv[index + 1]
return None
def values(key):
if key in sys.argv:
return map(
lambda x: sys.argv[x + 1],
filter(
lambda x: x + 1 < len(sys.argv),
filter(lambda x: sys.argv[x] == key,
range(len(sys.argv)))))
return [] #None caused error if empty.
def is_csv(filename):
ext = '.csv'
return filename[-len(ext):] == ext
def is_yaml(filename):
ext = '.yaml'
return filename[-len(ext):] == ext
template_file = value('-t')
content_file = value('-c')
kb_file = value('-k')
scan_files = values('-s')
nmap_files = values('-n')
report_file = value('-r')
output_file = value('-o')
summary_file = value('-u')
if template_file and report_file:
report = Report()
report.template_load_xml(template_file)
if content_file:
if is_yaml(content_file):
report.content_load_yaml(content_file)
else:
report.content_load_json(content_file)
for scan_file in scan_files:
#report.scan = Scan(scan_file)
scan = Scan(scan_file)
report.merge_scan(scan.modify())
report.content_refresh()
del scan
for nmap_file in nmap_files:
report.nmap_load_xml(nmap_file)
if kb_file:
if is_csv(kb_file):
report.kb_load_csv(kb_file)
elif is_yaml(kb_file):
report.kb_load_yaml(kb_file)
else:
report.kb_load_json(kb_file)
if output_file:
json_ext = '.json'
with open(output_file, 'w') as h:
if output_file[-len(json_ext):] == json_ext:
h.write(report.content_dump_json().encode('utf-8'))
else:
h.write(report.content_dump_yaml().encode('utf-8'))
print 'Output content file saved.'
report.xml_apply_meta()
report.save_report_xml(report_file)
print 'Report saved.'
elif scan_files and output_file:
if len(scan_files) > 1:
print 'Only single scan-file can be provided!'
else:
scan = Scan(scan_files[0])
json_ext = '.json'
with open(output_file, 'w') as h:
if output_file[-len(json_ext):] == json_ext:
h.write(scan.dump_json().encode('utf-8'))
else:
h.write(scan.dump_yaml().encode('utf-8'))
print 'Output scan file saved.'
elif scan_files and summary_file:
if len(scan_files) > 1:
print 'Only single scan-file can be provided!'
else:
scan = Scan(scan_files[0])
with open(summary_file, 'w') as h:
for i in scan.findings():
h.write('\t'.join([
i['Severity'],
str(len(i['Occurrences'])), i['Name']
]) + '\n')
else:
print 'Usage: '
print
print ' ' + self.title + '.exe -t template-file [-c content-file] [-k kb-file] [[-s scan-file] ...] [[-n nmap-file] ...] -r output-report-file [-o output-content-file]'
print ' generate report (and optionally - content as yaml / json)'
print
print ' ' + self.title + '.exe -s scan-file -o output-scan-file'
print ' convert scan xml to yaml or json format (derived from output file extension)'
print
print ' ' + self.title + '.exe -s scan-file -u output-summary-file'
print ' save findings summary to file (severities and names)'
print
print ' ' + self.title + '.exe [--help]'
print ' display usage and exit'
