def __init__(self, scans_store):
HIGVBox.__init__(self)
# This is a cache of details windows we have open.
self._details_windows = {}
self.set_spacing(0)
hbox = HIGHBox()
self.scans_list = gtk.ComboBox(scans_store)
cell = gtk.CellRendererText()
self.scans_list.pack_start(cell, True)
self.scans_list.set_cell_data_func(cell, scan_entry_data_func)
hbox._pack_expand_fill(self.scans_list)
self.scans_list.connect("changed", self._selection_changed)
scans_store.connect("row-changed", self._row_changed)
scans_store.connect("row-deleted", self._row_deleted)
self.throbber = Throbber()
hbox._pack_noexpand_nofill(self.throbber)
self.details_button = gtk.Button(_("Details"))
self.details_button.connect("clicked", self._show_details)
hbox._pack_noexpand_nofill(self.details_button)
self._pack_noexpand_nofill(hbox)
self.nmap_output = NmapOutputViewer()
self._pack_expand_fill(self.nmap_output)
self._update()
def __init__(self, scans_store):
HIGVBox.__init__(self)
# This is a cache of details windows we have open.
self._details_windows = {}
self.set_spacing(0)
hbox = HIGHBox()
self.scans_list = gtk.ComboBox(scans_store)
cell = gtk.CellRendererText()
self.scans_list.pack_start(cell, True)
self.scans_list.set_cell_data_func(cell, scan_entry_data_func)
hbox._pack_expand_fill(self.scans_list)
self.scans_list.connect("changed", self._selection_changed)
scans_store.connect("row-changed", self._row_changed)
scans_store.connect("row-deleted", self._row_deleted)
self.throbber = Throbber()
hbox._pack_noexpand_nofill(self.throbber)
self.details_button = gtk.Button(_("Details"))
self.details_button.connect("clicked", self._show_details)
hbox._pack_noexpand_nofill(self.details_button)
self._pack_noexpand_nofill(hbox)
self.nmap_output = NmapOutputViewer()
self._pack_expand_fill(self.nmap_output)
self._update()
def set_comment(self, comment=""):
self.comment_expander.set_use_markup(True)
if comment:
self.comment_expander.set_expanded(True)
hbox = HIGHBox()
self.comment_scrolled = gtk.ScrolledWindow()
self.comment_scrolled.set_border_width(5)
self.comment_scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
self.comment_txt_vw = gtk.TextView()
self.comment_txt_vw.set_wrap_mode(gtk.WRAP_WORD)
self.comment_txt_vw.get_buffer().set_text(comment)
self.comment_scrolled.add(self.comment_txt_vw)
hbox._pack_expand_fill(self.comment_scrolled)
self.comment_expander.add(hbox)
self._pack_noexpand_nofill(self.comment_expander)
def set_comment(self, comment=''):
self.comment_expander.set_use_markup(True)
if comment:
self.comment_expander.set_expanded(True)
hbox = HIGHBox()
self.comment_scrolled = gtk.ScrolledWindow()
self.comment_scrolled.set_border_width(5)
self.comment_scrolled.set_policy(gtk.POLICY_AUTOMATIC,
gtk.POLICY_AUTOMATIC)
self.comment_txt_vw = gtk.TextView()
self.comment_txt_vw.set_wrap_mode(gtk.WRAP_WORD)
self.comment_txt_vw.get_buffer().set_text(comment)
self.comment_scrolled.add(self.comment_txt_vw)
hbox._pack_expand_fill(self.comment_scrolled)
self.comment_expander.add(hbox)
self._pack_noexpand_nofill(self.comment_expander)
class CrashReport(HIGWindow):
def __init__(self, summary, description):
HIGWindow.__init__(self)
gtk.Window.__init__(self)
self.set_title(_('Crash Report'))
self.set_position(gtk.WIN_POS_CENTER_ALWAYS)
self.response_id = False
self._create_widgets()
self._pack_widgets()
self._connect_widgets()
self.summary = summary
self.description = "\n----\n" + description
def _create_widgets(self):
self.vbox = HIGVBox()
self.button_box = gtk.HButtonBox()
self.email_label = gtk.Label(_("""\
An email address is optional. Sometimes we use it to get more information. If \
you provide an email address your report will be marked private so only \
project developers can read it.\
"""))
self.email_entry = gtk.Entry()
self.summary_entry = gtk.Entry()
self.description_label = gtk.Label(_("\
What were you doing when the crash happened?\
"))
self.description_scrolled = gtk.ScrolledWindow()
self.description_text = gtk.TextView()
self.bug_text = gtk.Label(_("""\
An unhandled exception has crashed Zenmap. This dialog allows you to tell us \
what you did to cause the crash and help us to fix it. Submitting the report \
will open a description of the new bug at the bug tracker. Feel free to edit \
the report to remove any identifying information such as your home directory.\
"""))
self.btn_ok = gtk.Button(stock=gtk.STOCK_OK)
self.btn_cancel = gtk.Button(stock=gtk.STOCK_CANCEL)
self.hbox = HIGHBox()
self.table = HIGTable()
def _pack_widgets(self):
self.description_label.set_line_wrap(True)
self.description_label.set_alignment(0.0, 0.5)
self.description_scrolled.add(self.description_text)
self.description_scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
self.description_scrolled.set_size_request(400, 150)
self.description_text.set_wrap_mode(gtk.WRAP_WORD)
self.email_label.set_line_wrap(True)
self.email_label.set_alignment(0.0, 0.5)
self.bug_text.set_line_wrap(True)
self.hbox.set_border_width(12)
self.vbox.set_border_width(6)
self.table.attach_label(gtk.Label(_("Summary")), 0, 1, 0, 1)
self.table.attach_entry(self.summary_entry, 1, 2, 0, 1)
self.table.attach_entry(self.description_label, 0, 2, 1, 2)
self.table.attach_entry(self.description_scrolled, 0, 2, 2, 3)
self.table.attach(self.email_label, 0, 2, 3, 4)
self.table.attach_label(gtk.Label(_("Email")), 0, 1, 4, 5)
self.table.attach_entry(self.email_entry, 1, 2, 4, 5)
self.hbox._pack_expand_fill(self.bug_text)
self.button_box.set_layout(gtk.BUTTONBOX_END)
self.button_box.pack_start(self.btn_ok)
self.button_box.pack_start(self.btn_cancel)
self.vbox._pack_noexpand_nofill(self.hbox)
self.vbox._pack_expand_fill(self.table)
self.vbox._pack_noexpand_nofill(self.button_box)
self.add(self.vbox)
def _connect_widgets(self):
self.btn_ok.connect("clicked", self.send_report)
self.btn_cancel.connect("clicked", self.close)
self.connect("delete-event", self.close)
def send_report(self, widget):
bug_register = BugRegister()
description_text = self.description
private = False
if self.email != "":
description_text += "\n\n" + self.email
private = True
report_url = bug_register.get_report_url(self.summary,
description_text, private = private)
# First we try reporting the bug with a web browser because that gives
# better feedback on what happened. If that doesn't work try with
# urllib2.
try:
webbrowser.open(report_url)
except webbrowser.Error:
try:
urllib2.urlopen(report_url)
ok_dialog = HIGAlertDialog(type=gtk.MESSAGE_INFO,
message_format=_("Bug reported"),
secondary_text=_("The bug was successfully reported."))
ok_dialog.run()
ok_dialog.destroy()
except urllib2.URLError, e:
cancel_dialog = HIGAlertDialog(type=gtk.MESSAGE_ERROR,
message_format=_("Bug not reported"),
secondary_text=_("This error occurred while trying to report the bug:\n\n" + str(e)))
cancel_dialog.run()
cancel_dialog.destroy()
self.close()
class UmitCredits(HIGWindow):
def __init__(self):
HIGWindow.__init__(self)
self.set_title(_("%s credits") % UMIT_DISPLAY_NAME)
self.set_size_request(-1, 250)
self.set_position(gtk.WIN_POS_CENTER)
self.__create_widgets()
self.__packing()
self.set_text()
def __create_widgets(self):
self.vbox = HIGVBox()
self.hbox = HIGHBox()
self.notebook = HIGNotebook()
self.btn_close = HIGButton(stock=gtk.STOCK_CLOSE)
self.written_by_scroll = HIGScrolledWindow()
self.written_by_text = HIGTextView()
self.design_scroll = HIGScrolledWindow()
self.design_text = HIGTextView()
self.soc2007_scroll = HIGScrolledWindow()
self.soc2007_text = HIGTextView()
self.contributors_scroll = HIGScrolledWindow()
self.contributors_text = HIGTextView()
self.translation_scroll = HIGScrolledWindow()
self.translation_text = HIGTextView()
self.nokia_scroll = HIGScrolledWindow()
self.nokia_text = HIGTextView()
def __packing(self):
self.add(self.vbox)
self.vbox.set_spacing(12)
self.vbox._pack_expand_fill(self.notebook)
self.vbox._pack_noexpand_nofill(self.hbox)
self.hbox._pack_expand_fill(hig_box_space_holder())
self.hbox._pack_noexpand_nofill(self.btn_close)
self.notebook.append_page(self.written_by_scroll,
gtk.Label(_("Written by")))
self.notebook.append_page(self.design_scroll, gtk.Label(_("Design")))
self.notebook.append_page(self.soc2007_scroll,
gtk.Label(_("SoC 2007")))
self.notebook.append_page(self.contributors_scroll,
gtk.Label(_("Contributors")))
self.notebook.append_page(self.translation_scroll,
gtk.Label(_("Translation")))
self.notebook.append_page(self.nokia_scroll, gtk.Label(_("Maemo")))
self.written_by_scroll.add(self.written_by_text)
self.written_by_text.set_wrap_mode(gtk.WRAP_NONE)
self.design_scroll.add(self.design_text)
self.design_text.set_wrap_mode(gtk.WRAP_NONE)
self.soc2007_scroll.add(self.soc2007_text)
self.soc2007_text.set_wrap_mode(gtk.WRAP_NONE)
self.contributors_scroll.add(self.contributors_text)
self.contributors_text.set_wrap_mode(gtk.WRAP_NONE)
self.translation_scroll.add(self.translation_text)
self.translation_text.set_wrap_mode(gtk.WRAP_NONE)
self.nokia_scroll.add(self.nokia_text)
self.nokia_text.set_wrap_mode(gtk.WRAP_NONE)
self.btn_close.connect('clicked', lambda x, y=None: self.destroy())
def set_text(self):
b = self.written_by_text.get_buffer()
b.set_text("""Adriano Monteiro Marques <*****@*****.**>""")
b = self.design_text.get_buffer()
b.set_text("""Operating System and Vulnerability Icons:
Takeshi Alexandre Gondo <*****@*****.**>
Logo, Application Icons and Splash screen:
Virgílio Carlo de Menezes Vasconcelos <*****@*****.**>
The Umit Project Web Site Design:
Joao Paulo Pacheco <*****@*****.**>""")
b = self.soc2007_text.get_buffer()
b.set_text("""Independent Features:
Adriano Monteiro Marques <*****@*****.**>
Frederico Silva Ribeiro <*****@*****.**>
Network Inventory:
Guilherme Henrique Polo Gonçalves <*****@*****.**>
Umit Radial Mapper:
João Paulo de Souza Medeiros <*****@*****.**>
Profile/Wizard interface editor:
Luis Antonio Bastião Silva <*****@*****.**>
NSE Facilitator:
Maxim I. Gavrilov <*****@*****.**>
Umit Web:
Rodolfo da Silva Carvalho <*****@*****.**>""")
b = self.contributors_text.get_buffer()
b.set_text("""Sponsored by (SoC 2005, 2006 and 2007):
Google <*****@*****.**>
Mentor of Umit for Google SoC 2005 and 2006:
Fyodor <*****@*****.**>
Mentor of Umit for Google SoC 2007 Projects:
Adriano Monteiro Marques <*****@*****.**>
Initial development:
Adriano Monteiro Marques <*****@*****.**>
Cleber Rodrigues Rosa Junior <*****@*****.**>
Nmap students from Google SoC 2007 that helped Umit:
Eddie Bell <*****@*****.**>
David Fifield <*****@*****.**>
Kris Katterjohn <*****@*****.**>
The Umit Project WebSite:
AbraoBarbosa dos Santos Neto <*****@*****.**>
Adriano Monteiro Marques <*****@*****.**>
Heitor de Lima Matos <*****@*****.**>
Joao Paulo Pacheco <*****@*****.**>
João Paulo de Souza Medeiros <*****@*****.**>
Luis Antonio Bastião Silva <*****@*****.**>
Rodolfo da Silva Carvalho <*****@*****.**>
Beta testers for 0.9.5RC1:
Drew Miller <*****@*****.**>
Igor Feghali <*****@*****.**>
Joao Paulo Pacheco <*****@*****.**>
Luis Antonio Bastião Silva <*****@*****.**>
<*****@*****.**>
<*****@*****.**>
<*****@*****.**>
Initial attempt on Maemo port:
Adriano Monteiro Marques <*****@*****.**>
Osvaldo Santana Neto <*****@*****.**>""")
b = self.translation_text.get_buffer()
b.set_text("""Brazilian Portuguese:
Adriano Monteiro Marques <*****@*****.**>""")
b = self.nokia_text.get_buffer()
b.set_text("""Adriano Monteiro Marques <*****@*****.**>""")
class ProfileEditor(HIGWindow):
def __init__(self, command=None, profile_name=None, deletable=True, overwrite=False):
HIGWindow.__init__(self)
self.connect("delete_event", self.exit)
self.set_title(_('Profile Editor'))
self.set_position(gtk.WIN_POS_CENTER)
self.deletable = deletable
self.profile_name = profile_name
self.overwrite = overwrite
# Used to block recursive updating of the command entry when the command
# entry causes the OptionBuilder widgets to change.
self.inhibit_command_update = False
self.__create_widgets()
self.__pack_widgets()
self.profile = CommandProfile()
self.ops = NmapOptions()
if profile_name:
log.debug("Showing profile %s" % profile_name)
prof = self.profile.get_profile(profile_name)
# Interface settings
self.profile_name_entry.set_text(profile_name)
self.profile_description_text.get_buffer().set_text(prof['description'])
command_string = prof['command']
self.ops.parse_string(command_string)
if command:
self.ops.parse_string(command)
self.option_builder = OptionBuilder(Path.profile_editor, self.ops, self.update_command, self.help_field.get_buffer())
log.debug("Option groups: %s" % str(self.option_builder.groups))
log.debug("Option section names: %s" % str(self.option_builder.section_names))
#log.debug("Option tabs: %s" % str(self.option_builder.tabs))
for tab in self.option_builder.groups:
self.__create_tab(_(tab), _(self.option_builder.section_names[tab]), self.option_builder.tabs[tab])
self.update_command()
def command_entry_changed_cb(self, widget):
command_string = self.command_entry.get_text().decode("UTF-8")
self.ops.parse_string(command_string)
self.inhibit_command_update = True
self.option_builder.update()
self.inhibit_command_update = False
def update_command(self):
"""Regenerate and display the command."""
if not self.inhibit_command_update:
# Block recursive updating of the OptionBuilder widgets when they
# cause a change in the command entry.
self.command_entry.handler_block(self.command_entry_changed_cb_id)
self.command_entry.set_text(self.ops.render_string())
self.command_entry.handler_unblock(self.command_entry_changed_cb_id)
def update_help_name(self, widget, extra):
self.help_field.get_buffer().set_text("Profile name\n\nThis is how the"
+" profile will be identified in the drop-down combo box in the"
+" scan tab.")
def update_help_desc(self, widget, extra):
self.help_field.get_buffer().set_text("Description\n\nThe description is a"
+ " full description of what the scan does, which may be long.")
def __create_widgets(self):
###
# Vertical box to keep 3 boxes
self.main_whole_box = HIGVBox()
self.upper_box = HIGHBox()
self.middle_box = HIGHBox()
self.lower_box = HIGHBox()
#self.main_vbox = HIGVBox()
self.command_entry = gtk.Entry()
self.command_entry_changed_cb_id = \
self.command_entry.connect("changed", self.command_entry_changed_cb)
self.scan_button = HIGButton(_("Scan"))
self.scan_button.connect("clicked", self.run_scan)
self.notebook = gtk.Notebook()
# Profile info page
self.profile_info_vbox = HIGVBox()
self.profile_info_label = HIGSectionLabel(_('Profile Information'))
self.profile_name_label = HIGEntryLabel(_('Profile name'))
self.profile_name_entry = gtk.Entry()
self.profile_name_entry.connect('enter-notify-event', self.update_help_name)
self.profile_description_label = HIGEntryLabel(_('Description'))
self.profile_description_scroll = HIGScrolledWindow()
self.profile_description_scroll.set_border_width(0)
self.profile_description_text = HIGTextView()
self.profile_description_text.connect('motion-notify-event', self.update_help_desc)
# Buttons
self.buttons_hbox = HIGHBox()
self.cancel_button = HIGButton(stock=gtk.STOCK_CANCEL)
self.cancel_button.connect('clicked', self.exit)
self.delete_button = HIGButton(stock=gtk.STOCK_DELETE)
self.delete_button.connect('clicked', self.delete_profile)
self.save_button = HIGButton(_("Save Changes"), stock=gtk.STOCK_SAVE)
self.save_button.connect('clicked', self.save_profile)
###
self.help_vbox = HIGVBox()
self.help_label = HIGSectionLabel(_('Help'))
self.help_scroll = HIGScrolledWindow()
self.help_scroll.set_border_width(0)
self.help_field = HIGTextView()
self.help_field.set_cursor_visible(False)
self.help_field.set_left_margin(5)
self.help_field.set_editable(False)
self.help_vbox.set_size_request(200,-1)
###
def __pack_widgets(self):
###
self.add(self.main_whole_box)
# Packing command entry to upper box
self.upper_box._pack_expand_fill(self.command_entry)
self.upper_box._pack_noexpand_nofill(self.scan_button)
# Packing notebook (left) and help box (right) to middle box
self.middle_box._pack_expand_fill(self.notebook)
self.middle_box._pack_expand_fill(self.help_vbox)
# Packing buttons to lower box
self.lower_box.pack_end(self.buttons_hbox)
# Packing the three vertical boxes to the main box
self.main_whole_box._pack_noexpand_nofill(self.upper_box)
self.main_whole_box._pack_expand_fill(self.middle_box)
self.main_whole_box._pack_noexpand_nofill(self.lower_box)
###
# Packing profile information tab on notebook
self.notebook.append_page(self.profile_info_vbox, gtk.Label(_('Profile')))
self.profile_info_vbox.set_border_width(5)
table = HIGTable()
self.profile_info_vbox._pack_noexpand_nofill(self.profile_info_label)
self.profile_info_vbox._pack_expand_fill(HIGSpacer(table))
self.profile_description_scroll.add(self.profile_description_text)
vbox_desc = HIGVBox()
vbox_desc._pack_noexpand_nofill(self.profile_description_label)
vbox_desc._pack_expand_fill(hig_box_space_holder())
vbox_ann = HIGVBox()
vbox_ann._pack_expand_fill(hig_box_space_holder())
table.attach(self.profile_name_label,0,1,0,1,xoptions=0,yoptions=0)
table.attach(self.profile_name_entry,1,2,0,1,yoptions=0)
table.attach(vbox_desc,0,1,1,2,xoptions=0)
table.attach(self.profile_description_scroll,1,2,1,2)
# Packing buttons on button_hbox
self.buttons_hbox._pack_expand_fill(hig_box_space_holder())
if self.deletable:
self.buttons_hbox._pack_noexpand_nofill(self.delete_button)
self.buttons_hbox._pack_noexpand_nofill(self.cancel_button)
self.buttons_hbox._pack_noexpand_nofill(self.save_button)
self.buttons_hbox.set_border_width(5)
self.buttons_hbox.set_spacing(6)
###
self.help_vbox._pack_noexpand_nofill(self.help_label)
self.help_vbox._pack_expand_fill(self.help_scroll)
self.help_scroll.add(self.help_field)
self.help_vbox.set_border_width(1)
self.help_vbox.set_spacing(1)
###
def __create_tab(self, tab_name, section_name, tab):
log.debug(">>> Tab name: %s" % tab_name)
log.debug(">>>Creating profile editor section: %s" % section_name)
vbox = HIGVBox()
if tab.notscripttab: # if notscripttab is set
table = HIGTable()
table.set_row_spacings(2)
section = HIGSectionLabel(section_name)
vbox._pack_noexpand_nofill(section)
vbox._pack_noexpand_nofill(HIGSpacer(table))
vbox.set_border_width(5)
tab.fill_table(table, True)
else:
hbox = tab.get_hmain_box()
vbox.pack_start(hbox,True,True,0)
self.notebook.append_page(vbox, gtk.Label(tab_name))
def save_profile(self, widget):
if self.overwrite:
self.profile.remove_profile(self.profile_name)
profile_name = self.profile_name_entry.get_text()
if profile_name == '':
alert = HIGAlertDialog(message_format=_('Unnamed profile'),\
secondary_text=_('You must provide a name \
for this profile.'))
alert.run()
alert.destroy()
self.profile_name_entry.grab_focus()
return None
command = self.ops.render_string()
buf = self.profile_description_text.get_buffer()
description = buf.get_text(buf.get_start_iter(),\
buf.get_end_iter())
try:
self.profile.add_profile(profile_name,\
command=command,\
description=description)
except ValueError:
alert = HIGAlertDialog(message_format=_('Disallowed profile name'),\
secondary_text=_('Sorry, the name "%s" \
is not allowed due to technical limitations. (The underlying ConfigParser \
used to store profiles does not allow it.) Choose a different \
name.' % profile_name))
alert.run()
alert.destroy()
return
self.scan_interface.toolbar.profile_entry.update()
self.destroy()
def clean_profile_info(self):
self.profile_name_entry.set_text('')
self.profile_description_text.get_buffer().set_text('')
def set_scan_interface(self, interface):
self.scan_interface = interface
def exit(self, *args):
self.destroy()
def delete_profile(self, widget=None, extra=None):
if self.deletable:
dialog = HIGDialog(buttons=(gtk.STOCK_OK, gtk.RESPONSE_OK,
gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))
alert = HIGEntryLabel('<b>'+_("Deleting Profile")+'</b>')
text = HIGEntryLabel(_('Your profile is going to be deleted! Click\
Ok to continue, or Cancel to go back to Profile Editor.'))
hbox = HIGHBox()
hbox.set_border_width(5)
hbox.set_spacing(12)
vbox = HIGVBox()
vbox.set_border_width(5)
vbox.set_spacing(12)
image = gtk.Image()
image.set_from_stock(gtk.STOCK_DIALOG_WARNING, gtk.ICON_SIZE_DIALOG)
vbox.pack_start(alert)
vbox.pack_start(text)
hbox.pack_start(image)
hbox.pack_start(vbox)
dialog.vbox.pack_start(hbox)
dialog.vbox.show_all()
response = dialog.run()
dialog.destroy()
if response == gtk.RESPONSE_CANCEL:
return True
self.profile.remove_profile(self.profile_name)
self.update_profile_entry()
self.destroy()
def run_scan(self, widget=None):
command_string = self.command_entry.get_text().decode("UTF-8")
self.scan_interface.command_toolbar.command = command_string
self.scan_interface.start_scan_cb()
self.exit()
def update_profile_entry(self, widget=None, extra=None):
self.scan_interface.toolbar.profile_entry.update()
list = self.scan_interface.toolbar.profile_entry.get_model()
length = len(list)
if length >0 :
self.scan_interface.toolbar.profile_entry.set_active(0)
class UmitCredits(HIGWindow):
def __init__(self):
HIGWindow.__init__(self)
self.set_title(_("%s credits") % UMIT_DISPLAY_NAME)
self.set_size_request(-1, 250)
self.set_position(gtk.WIN_POS_CENTER)
self.__create_widgets()
self.__packing()
self.set_text()
def __create_widgets(self):
self.vbox = HIGVBox()
self.hbox = HIGHBox()
self.notebook = HIGNotebook()
self.btn_close = HIGButton(stock=gtk.STOCK_CLOSE)
self.written_by_scroll = HIGScrolledWindow()
self.written_by_text = HIGTextView()
self.design_scroll = HIGScrolledWindow()
self.design_text = HIGTextView()
self.soc2007_scroll = HIGScrolledWindow()
self.soc2007_text = HIGTextView()
self.contributors_scroll = HIGScrolledWindow()
self.contributors_text = HIGTextView()
self.translation_scroll = HIGScrolledWindow()
self.translation_text = HIGTextView()
self.nokia_scroll = HIGScrolledWindow()
self.nokia_text = HIGTextView()
def __packing(self):
self.add(self.vbox)
self.vbox.set_spacing(12)
self.vbox._pack_expand_fill(self.notebook)
self.vbox._pack_noexpand_nofill(self.hbox)
self.hbox._pack_expand_fill(hig_box_space_holder())
self.hbox._pack_noexpand_nofill(self.btn_close)
self.notebook.append_page(
self.written_by_scroll, gtk.Label(_("Written by")))
self.notebook.append_page(
self.design_scroll, gtk.Label(_("Design")))
self.notebook.append_page(
self.soc2007_scroll, gtk.Label(_("SoC 2007")))
self.notebook.append_page(
self.contributors_scroll, gtk.Label(_("Contributors")))
self.notebook.append_page(
self.translation_scroll, gtk.Label(_("Translation")))
self.notebook.append_page(
self.nokia_scroll, gtk.Label(_("Maemo")))
self.written_by_scroll.add(self.written_by_text)
self.written_by_text.set_wrap_mode(gtk.WRAP_NONE)
self.design_scroll.add(self.design_text)
self.design_text.set_wrap_mode(gtk.WRAP_NONE)
self.soc2007_scroll.add(self.soc2007_text)
self.soc2007_text.set_wrap_mode(gtk.WRAP_NONE)
self.contributors_scroll.add(self.contributors_text)
self.contributors_text.set_wrap_mode(gtk.WRAP_NONE)
self.translation_scroll.add(self.translation_text)
self.translation_text.set_wrap_mode(gtk.WRAP_NONE)
self.nokia_scroll.add(self.nokia_text)
self.nokia_text.set_wrap_mode(gtk.WRAP_NONE)
self.btn_close.connect('clicked', lambda x, y=None: self.destroy())
def set_text(self):
b = self.written_by_text.get_buffer()
b.set_text("""Adriano Monteiro Marques <*****@*****.**>""")
b = self.design_text.get_buffer()
b.set_text("""Operating System and Vulnerability Icons:
Takeshi Alexandre Gondo <*****@*****.**>
Logo, Application Icons and Splash screen:
Virgílio Carlo de Menezes Vasconcelos <*****@*****.**>
The Umit Project Web Site Design:
Joao Paulo Pacheco <*****@*****.**>""")
b = self.soc2007_text.get_buffer()
b.set_text("""Independent Features:
Adriano Monteiro Marques <*****@*****.**>
Frederico Silva Ribeiro <*****@*****.**>
Network Inventory:
Guilherme Henrique Polo Gonçalves <*****@*****.**>
Umit Radial Mapper:
João Paulo de Souza Medeiros <*****@*****.**>
Profile/Wizard interface editor:
Luis Antonio Bastião Silva <*****@*****.**>
NSE Facilitator:
Maxim I. Gavrilov <*****@*****.**>
Umit Web:
Rodolfo da Silva Carvalho <*****@*****.**>""")
b = self.contributors_text.get_buffer()
b.set_text("""Sponsored by (SoC 2005, 2006 and 2007):
Google <*****@*****.**>
Mentor of Umit for Google SoC 2005 and 2006:
Fyodor <*****@*****.**>
Mentor of Umit for Google SoC 2007 Projects:
Adriano Monteiro Marques <*****@*****.**>
Initial development:
Adriano Monteiro Marques <*****@*****.**>
Cleber Rodrigues Rosa Junior <*****@*****.**>
Nmap students from Google SoC 2007 that helped Umit:
Eddie Bell <*****@*****.**>
David Fifield <*****@*****.**>
Kris Katterjohn <*****@*****.**>
The Umit Project WebSite:
AbraoBarbosa dos Santos Neto <*****@*****.**>
Adriano Monteiro Marques <*****@*****.**>
Heitor de Lima Matos <*****@*****.**>
Joao Paulo Pacheco <*****@*****.**>
João Paulo de Souza Medeiros <*****@*****.**>
Luis Antonio Bastião Silva <*****@*****.**>
Rodolfo da Silva Carvalho <*****@*****.**>
Beta testers for 0.9.5RC1:
Drew Miller <*****@*****.**>
Igor Feghali <*****@*****.**>
Joao Paulo Pacheco <*****@*****.**>
Luis Antonio Bastião Silva <*****@*****.**>
<*****@*****.**>
<*****@*****.**>
<*****@*****.**>
Initial attempt on Maemo port:
Adriano Monteiro Marques <*****@*****.**>
Osvaldo Santana Neto <*****@*****.**>""")
b = self.translation_text.get_buffer()
b.set_text("""Brazilian Portuguese:
Adriano Monteiro Marques <*****@*****.**>""")
b = self.nokia_text.get_buffer()
b.set_text("""Adriano Monteiro Marques <*****@*****.**>""")
class DiffText(HIGVBox, object):
def __init__ (self, colors, check_color):
HIGVBox.__init__(self)
self.set_border_width(5)
self.set_spacing(6)
self.colors = colors
self.check_color = check_color
self._create_widgets()
self._pack_hbox()
self._set_text_view()
self._set_scrolled()
self._pack_noexpand_nofill(self.lbl_diff)
self._pack_expand_fill(self.hbox)
def _create_widgets (self):
self.hbox = HIGHBox ()
self.lbl_diff = HIGSectionLabel ("<b>%s</b>" % _("Diff Result"))
self.scrolled = gtk.ScrolledWindow()
self.txt_diff_result = gtk.TextView()
self.txg_tag = gtk.TextTag ("diff_style")
self.txg_added = gtk.TextTag ('added style')
self.txg_removed = gtk.TextTag ('removed style')
def _pack_hbox (self):
self.hbox.set_border_width(5)
self.hbox._pack_noexpand_nofill (hig_box_space_holder())
self.hbox._pack_expand_fill(self.scrolled)
def _set_scrolled (self):
self.scrolled.set_size_request(-1, 250)
# Packing text view into scrolled window
self.scrolled.add_with_viewport(self.txt_diff_result)
# Setting scrolled window
self.scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
def set_text(self, text):
"""Set the raw diff text of this diff view."""
self.txt_diff_result.get_buffer().set_text(text)
def clear(self):
"""Make this diff view empty."""
self.set_text("")
def activate_color(self, activate):
self.check_color = activate
self._refresh()
def _set_text_view(self):
self.txg_table = self.txt_diff_result.get_buffer().get_tag_table()
self.txg_table.add(self.txg_tag)
self.txg_table.add(self.txg_added)
self.txg_table.add(self.txg_removed)
self.txg_tag.set_property("family", "Monospace")
self.txg_added.set_property("background-gdk", self.colors.added)
self.txg_removed.set_property("background-gdk", self.colors.not_present)
self.txt_diff_result.set_wrap_mode(gtk.WRAP_WORD)
self.txt_diff_result.set_editable(False)
self.txt_diff_result.get_buffer().connect("changed", self._refresh)
def _refresh(self, *args):
self.txg_added.set_property("background-gdk", self.colors.added)
self.txg_removed.set_property("background-gdk", self.colors.not_present)
buff = self.txt_diff_result.get_buffer()
buff.apply_tag(self.txg_tag, buff.get_start_iter(), buff.get_end_iter())
if self.check_color:
positions = self._take_changes(buff)
for i in positions['added']:
buff.apply_tag(self.txg_added, i[0],i[1])
for i in positions['removed']:
buff.apply_tag(self.txg_removed, i[0], i[1])
else:
buff.remove_tag(self.txg_added, buff.get_start_iter (), buff.get_end_iter())
buff.remove_tag(self.txg_removed, buff.get_start_iter (), buff.get_end_iter())
def _take_changes (self, buffer):
positions = {'added':[], 'removed':[]}
in_line = 0
type = ''
iter = buffer.get_start_iter ()
last_char = '\n'
while iter.forward_char():
char = iter.get_char ()
offset = iter.get_offset()
if not in_line:
pos = []
if char == '+' and last_char == '\n':
pos.append (buffer.get_iter_at_offset(offset))
type = 'added'
in_line = 1
elif char == '-' and last_char == '\n':
pos.append (buffer.get_iter_at_offset(offset))
type = 'removed'
in_line = 1
else:
if char == '\n':
pos.append (buffer.get_iter_at_offset(offset))
positions [type].append (pos)
in_line = 0
type = ''
last_char = char
return positions
class ScanChooser(HIGVBox):
"""This class allows the selection of scan results from the list of open
tabs or from a file. It emits the "changed" signal when the scan selection
has changed."""
__gsignals__ = {
"changed": (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ())
}
def __init__(self, scans, title):
self.__gobject_init__()
self.title = title
self.scan_dict = {}
# Setting HIGVBox
self.set_border_width(5)
self.set_spacing(6)
self._create_widgets()
self._pack_hbox()
self._attaching_widgets()
self._set_scrolled()
self._set_text_view()
self._set_open_button()
for scan in scans:
self.add_scan(scan.scan_name or scan.get_nmap_command(), scan)
self.combo_scan.connect('changed', self.show_scan)
self.combo_scan.connect('changed', lambda x: self.emit('changed'))
self._pack_noexpand_nofill(self.lbl_scan)
self._pack_expand_fill(self.hbox)
def _create_widgets(self):
self.lbl_scan = HIGSectionLabel(self.title)
self.hbox = HIGHBox()
self.table = HIGTable()
self.list_scan = gtk.ListStore(str)
self.combo_scan = gtk.ComboBoxEntry(self.list_scan, 0)
self.btn_open_scan = gtk.Button(stock=gtk.STOCK_OPEN)
self.exp_scan = gtk.Expander(_("Scan Output"))
self.scrolled = gtk.ScrolledWindow()
self.txt_scan_result = gtk.TextView()
self.txg_tag = gtk.TextTag("scan_style")
def get_buffer(self):
return self.txt_scan_result.get_buffer()
def show_scan(self, widget):
nmap_output = self.get_nmap_output()
if nmap_output:
self.txt_scan_result.get_buffer().set_text(nmap_output)
def normalize_output(self, output):
return "\n".join(output.split("\\n"))
def _pack_hbox(self):
self.hbox._pack_noexpand_nofill(hig_box_space_holder())
self.hbox._pack_expand_fill(self.table)
def _attaching_widgets(self):
self.table.attach(self.combo_scan, 0, 1, 0, 1, yoptions=0)
self.table.attach(self.btn_open_scan,
1,
2,
0,
1,
yoptions=0,
xoptions=0)
self.table.attach(self.exp_scan, 0, 2, 1, 2)
def _set_scrolled(self):
self.scrolled.set_border_width(5)
self.scrolled.set_size_request(-1, 130)
# Packing scrolled window into expander
self.exp_scan.add(self.scrolled)
# Packing text view into scrolled window
self.scrolled.add_with_viewport(self.txt_scan_result)
# Setting scrolled window
self.scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
def _set_text_view(self):
self.txg_table = self.txt_scan_result.get_buffer().get_tag_table()
self.txg_table.add(self.txg_tag)
self.txg_tag.set_property("family", "Monospace")
self.txt_scan_result.set_wrap_mode(gtk.WRAP_WORD)
self.txt_scan_result.set_editable(False)
self.txt_scan_result.get_buffer().connect("changed",
self._text_changed_cb)
def _set_open_button(self):
self.btn_open_scan.connect('clicked', self.open_file)
def open_file(self, widget):
file_chooser = ResultsFileSingleChooserDialog(_("Select Scan Result"))
response = file_chooser.run()
file_chosen = file_chooser.get_filename()
file_chooser.destroy()
if response == gtk.RESPONSE_OK:
try:
parser = NmapParser()
parser.parse_file(file_chosen)
except xml.sax.SAXParseException as e:
alert = HIGAlertDialog(
message_format='<b>%s</b>' % _('Error parsing file'),
secondary_text=_(
"The file is not an Nmap XML output file. "
"The parsing error that occurred was\n%s") % str(e))
alert.run()
alert.destroy()
return False
except Exception as e:
alert = HIGAlertDialog(message_format='<b>%s</b>' %
_('Cannot open selected file'),
secondary_text=_("""\
This error occurred while trying to open the file:
%s""") % str(e))
alert.run()
alert.destroy()
return False
scan_name = os.path.split(file_chosen)[-1]
self.add_scan(scan_name, parser)
self.combo_scan.set_active(len(self.list_scan) - 1)
def add_scan(self, scan_name, parser):
scan_id = 1
new_scan_name = scan_name
while new_scan_name in self.scan_dict.keys():
new_scan_name = "%s (%s)" % (scan_name, scan_id)
scan_id += 1
self.list_scan.append([new_scan_name])
self.scan_dict[new_scan_name] = parser
def _text_changed_cb(self, widget):
buff = self.txt_scan_result.get_buffer()
buff.apply_tag(self.txg_tag, buff.get_start_iter(),
buff.get_end_iter())
def get_parsed_scan(self):
"""Return the currently selected scan's parsed output as an NmapParser
object, or None if no valid scan is selected."""
selected_scan = self.combo_scan.child.get_text()
return self.scan_dict.get(selected_scan)
def get_nmap_output(self):
"""Return the currently selected scan's output as a string, or None if
no valid scan is selected."""
if self.parsed_scan is not None:
return self.parsed_scan.get_nmap_output()
else:
return None
nmap_output = property(get_nmap_output)
parsed_scan = property(get_parsed_scan)
class CrashReport(HIGDialog):
def __init__(self, type, value, tb):
HIGDialog.__init__(self)
gtk.Window.__init__(self)
self.set_title(_('Crash Report'))
self.set_position(gtk.WIN_POS_CENTER_ALWAYS)
self._create_widgets()
self._pack_widgets()
self._connect_widgets()
trace = "".join(traceback.format_exception(type, value, tb))
text = "Version: " + VERSION + "\n" + trace
self.description_text.get_buffer().set_text(text)
def _create_widgets(self):
self.button_box = gtk.HButtonBox()
self.button_box_ok = gtk.HButtonBox()
self.description_scrolled = gtk.ScrolledWindow()
self.description_text = gtk.TextView()
self.description_text.set_editable(False)
self.bug_text = gtk.Label()
self.bug_text.set_markup(_("""\
An unexpected error has crashed %(app_name)s. Please copy the stack trace below and \
send it to the <a href="mailto:[email protected]">[email protected]</a> \
mailing list. (<a href="http://seclists.org/nmap-dev/">More about the list.</a>) \
The developers will see your report and try to fix the problem.""") % \
{ "app_name": escape(APP_DISPLAY_NAME) })
self.email_frame = gtk.Frame()
self.email_label = gtk.Label()
self.email_label.set_markup(_("""\
<b>Copy and email to <a href="mailto:[email protected]">[email protected]</a>:</b>\
"""))
self.btn_copy = gtk.Button(stock=gtk.STOCK_COPY)
self.btn_ok = gtk.Button(stock=gtk.STOCK_OK)
self.hbox = HIGHBox()
def _pack_widgets(self):
self.description_scrolled.add(self.description_text)
self.description_scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
self.description_scrolled.set_size_request(400, 150)
self.description_text.set_wrap_mode(gtk.WRAP_WORD)
self.bug_text.set_line_wrap(True)
self.email_label.set_line_wrap(True)
self.email_frame.set_label_widget(self.email_label)
self.email_frame.set_shadow_type(gtk.SHADOW_NONE)
self.hbox.set_border_width(6)
self.vbox.set_border_width(6)
self.hbox._pack_expand_fill(self.bug_text)
self.button_box.set_layout(gtk.BUTTONBOX_START)
self.button_box_ok.set_layout(gtk.BUTTONBOX_END)
self.button_box.pack_start(self.btn_copy)
self.button_box_ok.pack_start(self.btn_ok)
self.vbox.pack_start(self.hbox)
self.vbox.pack_start(self.email_frame)
self.vbox.pack_start(self.description_scrolled)
self.vbox.pack_start(self.button_box)
self.action_area.pack_start(self.button_box_ok)
def _connect_widgets(self):
self.btn_ok.connect("clicked", self.close)
self.btn_copy.connect("clicked", self.copy)
self.connect("delete-event", self.close)
def get_description(self):
buff = self.description_text.get_buffer()
return buff.get_text(buff.get_start_iter(), buff.get_end_iter())
def copy(self, widget=None, event=None):
clipboard = gtk.clipboard_get()
clipboard.set_text(self.get_description())
clipboard.store()
def close(self, widget=None, event=None):
self.destroy()
gtk.main_quit()
sys.exit(0)
class CrashReport(HIGWindow):
def __init__(self, summary, description):
HIGWindow.__init__(self)
gtk.Window.__init__(self)
self.set_title(_('Crash Report'))
self.set_position(gtk.WIN_POS_CENTER_ALWAYS)
self.response_id = False
self._create_widgets()
self._pack_widgets()
self._connect_widgets()
self.summary = summary
self.description = description
def _create_widgets(self):
self.vbox = HIGVBox()
self.button_box = gtk.HButtonBox()
self.email_label = HIGHintSectionLabel(_("Email"),
_("\
Optional. We occasionally contact the reporters of crashes in order to get \
more information."))
self.email_entry = gtk.Entry()
self.description_label = gtk.Label(_("\
At the top of this form please write what you were doing when the crash \
happened, as well as any other information you think is relevant."))
self.description_scrolled = gtk.ScrolledWindow()
self.description_text = gtk.TextView()
self.bug_icon = gtk.Image()
self.bug_text = gtk.Label(_("""\
This Bug Report dialog allows you to easily tell us about a problem that you \
may have found in %s. Doing so, you help us to help you, by fixing and \
improving %s faster than usual. Submitting the report will open a description \
of the new bug at the bug tracker.\
""" % (APP_DISPLAY_NAME, APP_DISPLAY_NAME)))
self.btn_ok = gtk.Button(stock=gtk.STOCK_OK)
self.btn_cancel = gtk.Button(stock=gtk.STOCK_CANCEL)
self.hbox = HIGHBox()
self.table = HIGTable()
def _pack_widgets(self):
self.description_label.set_line_wrap(True)
self.description_label.set_alignment(0.0, 0.5)
self.description_scrolled.add(self.description_text)
self.description_scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
self.description_scrolled.set_size_request(400, 150)
self.description_text.set_wrap_mode(gtk.WRAP_WORD)
self.bug_icon.set_from_stock(gtk.STOCK_DIALOG_INFO, gtk.ICON_SIZE_DIALOG)
self.bug_icon.set_padding(10, 0)
self.bug_text.set_line_wrap(True)
self.hbox.set_border_width(12)
self.vbox.set_border_width(6)
self.table.attach_label(self.email_label, 0, 1, 0, 1)
self.table.attach_entry(self.email_entry, 1, 2, 0, 1)
self.table.attach_entry(self.description_label, 0, 2, 1, 2)
self.table.attach_entry(self.description_scrolled, 0, 2, 2, 3)
self.hbox._pack_noexpand_nofill(self.bug_icon)
self.hbox._pack_expand_fill(self.bug_text)
self.button_box.set_layout(gtk.BUTTONBOX_END)
self.button_box.pack_start(self.btn_ok)
self.button_box.pack_start(self.btn_cancel)
self.vbox._pack_noexpand_nofill(self.hbox)
self.vbox._pack_expand_fill(self.table)
self.vbox._pack_noexpand_nofill(self.button_box)
self.add(self.vbox)
def _connect_widgets(self):
self.btn_ok.connect("clicked", self.send_report)
self.btn_cancel.connect("clicked", self.close)
self.connect("delete-event", self.close)
def send_report(self, widget):
bug_register = BugRegister()
report_url = bug_register.get_report_url(self.summary, "%s\n\nEmail: %s" % (self.description, self.email))
# First we try reporting the bug with a web browser because that gives
# better feedback on what happened. If that doesn't work try with
# urllib2.
try:
webbrowser.open(report_url)
except webbrowser.Error:
try:
urllib2.urlopen(report_url)
ok_dialog = HIGAlertDialog(type=gtk.MESSAGE_INFO,
message_format=_("Bug reported"),
secondary_text=_("The bug was successfully reported."))
ok_dialog.run()
ok_dialog.destroy()
except urllib2.URLError, e:
cancel_dialog = HIGAlertDialog(type=gtk.MESSAGE_ERROR,
message_format=_("Bug not reported"),
secondary_text=_("This error occurred while trying to report the bug:\n\n" + str(e)))
cancel_dialog.run()
cancel_dialog.destroy()
self.close()
class ProfileEditor(HIGWindow):
def __init__(self, profile_name=None, deletable=True, overwrite=False):
HIGWindow.__init__(self)
self.connect("delete_event", self.exit)
self.set_title(_('Profile Editor'))
self.set_position(gtk.WIN_POS_CENTER)
self.deletable = deletable
self.profile_name = profile_name
self.overwrite = overwrite
self.__create_widgets()
self.__pack_widgets()
self.profile = CommandProfile()
options_used = {}
if profile_name:
log.debug("Showing profile %s" % profile_name)
prof = self.profile.get_profile(profile_name)
options_used = prof['options']
# Interface settings
self.profile_name_entry.set_text(profile_name)
self.profile_description_text.get_buffer().set_text(
prof['description'])
self.constructor = CommandConstructor(options_used)
###
self.profilehelp = ProfileHelp(options_used)
###
self.options = OptionBuilder(Path.profile_editor, self.constructor,
self.update_command, self.profilehelp)
log.debug("Option groups: %s" % str(self.options.groups))
log.debug("Option section names: %s" % str(self.options.section_names))
#log.debug("Option tabs: %s" % str(self.options.tabs))
for tab in self.options.groups:
self.__create_tab(tab, self.options.section_names[tab],
self.options.tabs[tab])
self.update_command()
def update_command(self):
"""Regenerate command with target '<target>' and set the value for the command entry"""
self.command_entry.set_text(self.constructor.get_command('<target>'))
### whenever the command would be updated, thats when we update the help
self.update_help()
def update_help(self, text=None):
helpText = self.help_field.get_buffer()
if text:
helpText.set_text(text)
else:
tempText = ""
if self.profilehelp.get_currentstate() == "Default":
helpText.set_text(" ")
else:
tempText += self.profilehelp.get_label()
tempText += "\n\n"
tempText += self.profilehelp.get_shortdesc()
if self.profilehelp.get_example():
tempText += "\n\nExample input:\n"
tempText += self.profilehelp.get_example()
helpText.set_text(tempText)
def update_help_name(self, widget, extra):
self.update_help(
text="Profile name\n\nThis is how the" +
" profile will be identified in the drop-down combo box in the" +
" scan tab.")
def update_help_desc(self, widget, extra):
self.update_help(
text="Description\n\nThe description is a" +
" full description of what the scan does, which may be long.")
def help(self, widget):
d = HIGAlertDialog(
parent=self,
message_format=_("Help not implemented"),
secondary_text=_("Profile editor help is not implemented yet."))
d.run()
d.destroy()
def __create_widgets(self):
###
# Vertical box to keep 3 boxes
self.main_whole_box = HIGVBox()
self.upper_box = HIGVBox()
self.middle_box = HIGHBox()
self.lower_box = HIGHBox()
#self.main_vbox = HIGVBox()
self.command_expander = HIGExpander('<b>' + _('Command') + '</b>')
self.command_expander.set_expanded(True)
self.command_entry = gtk.Entry()
self.command_entry.set_editable(False)
self.notebook = gtk.Notebook()
# Profile info page
self.profile_info_vbox = HIGVBox()
self.profile_info_label = HIGSectionLabel(_('Profile Information'))
self.profile_name_label = HIGEntryLabel(_('Profile name'))
self.profile_name_entry = gtk.Entry()
self.profile_name_entry.connect('enter-notify-event',
self.update_help_name)
self.profile_description_label = HIGEntryLabel(_('Description'))
self.profile_description_scroll = HIGScrolledWindow()
self.profile_description_scroll.set_border_width(0)
self.profile_description_text = HIGTextView()
self.profile_description_text.connect('motion-notify-event',
self.update_help_desc)
# Buttons
self.buttons_hbox = HIGHBox()
self.help_button = HIGButton(stock=gtk.STOCK_HELP)
self.help_button.connect('clicked', self.help)
self.cancel_button = HIGButton(stock=gtk.STOCK_CANCEL)
self.cancel_button.connect('clicked', self.exit)
self.delete_button = HIGButton(stock=gtk.STOCK_DELETE)
self.delete_button.connect('clicked', self.delete_profile)
self.ok_button = HIGButton(stock=gtk.STOCK_OK)
self.ok_button.connect('clicked', self.save_profile)
###
self.help_vbox = HIGVBox()
self.help_label = HIGSectionLabel(_('Help'))
self.help_scroll = HIGScrolledWindow()
self.help_scroll.set_border_width(0)
self.help_field = HIGTextView()
self.help_field.set_cursor_visible(False)
self.help_field.set_left_margin(5)
self.help_field.set_editable(False)
self.help_vbox.set_size_request(200, -1)
###
def __pack_widgets(self):
###
self.add(self.main_whole_box)
# Packing command expander to upper box
self.upper_box._pack_noexpand_nofill(self.command_expander)
# Packing notebook (left) and help box (right) to middle box
self.middle_box._pack_expand_fill(self.notebook)
self.middle_box._pack_expand_fill(self.help_vbox)
# Packing buttons to lower box
self.lower_box.pack_end(self.buttons_hbox)
# Packing the three vertical boxes to the main box
self.main_whole_box._pack_noexpand_nofill(self.upper_box)
self.main_whole_box._pack_noexpand_nofill(self.middle_box)
self.main_whole_box._pack_noexpand_nofill(self.lower_box)
###
# Packing command_entry on command_expander
self.command_expander.hbox.pack_start(self.command_entry)
# Packing profile information tab on notebook
self.notebook.append_page(self.profile_info_vbox,
gtk.Label(_('Profile')))
self.profile_info_vbox.set_border_width(5)
table = HIGTable()
self.profile_info_vbox._pack_noexpand_nofill(self.profile_info_label)
self.profile_info_vbox._pack_expand_fill(HIGSpacer(table))
self.profile_description_scroll.add(self.profile_description_text)
vbox_desc = HIGVBox()
vbox_desc._pack_noexpand_nofill(self.profile_description_label)
vbox_desc._pack_expand_fill(hig_box_space_holder())
vbox_ann = HIGVBox()
vbox_ann._pack_expand_fill(hig_box_space_holder())
table.attach(self.profile_name_label,
0,
1,
0,
1,
xoptions=0,
yoptions=0)
table.attach(self.profile_name_entry, 1, 2, 0, 1, yoptions=0)
table.attach(vbox_desc, 0, 1, 1, 2, xoptions=0)
table.attach(self.profile_description_scroll, 1, 2, 1, 2)
# Packing buttons on button_hbox
#self.buttons_hbox.pack_start(self.help_button)
self.buttons_hbox._pack_expand_fill(hig_box_space_holder())
if self.deletable:
self.buttons_hbox._pack_noexpand_nofill(self.delete_button)
self.buttons_hbox._pack_noexpand_nofill(self.cancel_button)
self.buttons_hbox._pack_noexpand_nofill(self.ok_button)
self.buttons_hbox.set_border_width(5)
self.buttons_hbox.set_spacing(6)
###
self.help_vbox._pack_noexpand_nofill(self.help_label)
self.help_vbox._pack_expand_fill(self.help_scroll)
self.help_scroll.add(self.help_field)
self.help_vbox.set_border_width(1)
self.help_vbox.set_spacing(1)
###
def __create_tab(self, tab_name, section_name, tab):
log.debug(">>> Tab name: %s" % tab_name)
log.debug(">>>Creating profile editor section: %s" % section_name)
vbox = HIGVBox()
table = HIGTable()
table.set_row_spacings(2)
section = HIGSectionLabel(section_name)
vbox._pack_noexpand_nofill(section)
vbox._pack_noexpand_nofill(HIGSpacer(table))
vbox.set_border_width(5)
tab.fill_table(table, True)
self.notebook.append_page(vbox, gtk.Label(tab_name))
def save_profile(self, widget):
if self.overwrite:
self.profile.remove_profile(self.profile_name)
profile_name = self.profile_name_entry.get_text()
if profile_name == '':
alert = HIGAlertDialog(message_format=_('Unnamed profile'),\
secondary_text=_('You must provide a name \
for this profile.' ))
alert.run()
alert.destroy()
self.profile_name_entry.grab_focus()
return None
command = self.constructor.get_command('%s')
buf = self.profile_description_text.get_buffer()
description = buf.get_text(buf.get_start_iter(),\
buf.get_end_iter())
self.profile.add_profile(profile_name,\
command=command,\
description=description,\
options=self.constructor.get_options())
self.scan_interface.toolbar.profile_entry.update()
self.scan_interface.refresh_command(None)
self.destroy()
def clean_profile_info(self):
self.profile_name_entry.set_text('')
self.profile_description_text.get_buffer().set_text('')
def set_scan_interface(self, interface):
self.scan_interface = interface
def exit(self, *args):
self.destroy()
def delete_profile(self, widget=None, extra=None):
if self.deletable:
dialog = HIGDialog(buttons=(gtk.STOCK_OK, gtk.RESPONSE_OK,
gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))
alert = HIGEntryLabel('<b>' + _("Deleting Profile") + '</b>')
text = HIGEntryLabel(
_('Your profile is going to be deleted! Click\
Ok to continue, or Cancel to go back to Profile Editor.'))
hbox = HIGHBox()
hbox.set_border_width(5)
hbox.set_spacing(12)
vbox = HIGVBox()
vbox.set_border_width(5)
vbox.set_spacing(12)
image = gtk.Image()
image.set_from_stock(gtk.STOCK_DIALOG_WARNING,
gtk.ICON_SIZE_DIALOG)
vbox.pack_start(alert)
vbox.pack_start(text)
hbox.pack_start(image)
hbox.pack_start(vbox)
dialog.vbox.pack_start(hbox)
dialog.vbox.show_all()
response = dialog.run()
dialog.destroy()
if response == gtk.RESPONSE_CANCEL:
return True
self.profile.remove_profile(self.profile_name)
self.update_profile_entry()
self.destroy()
def update_profile_entry(self, widget=None, extra=None):
self.scan_interface.toolbar.profile_entry.update()
list = self.scan_interface.toolbar.profile_entry.get_model()
length = len(list)
if length > 0:
self.scan_interface.toolbar.profile_entry.set_active(0)
class DiffTree(HIGVBox, object):
def __init__(self, colors):
HIGVBox.__init__(self)
self.colors = colors
self.set_border_width(5)
self.set_spacing(6)
self._create_widgets()
self._set_diff_view()
self._pack_widgets()
def _create_widgets(self):
self.diff_box = HIGHBox()
self.diff_title = HIGSectionLabel("Comparison")
self.diff_scrolled = gtk.ScrolledWindow()
self.diff_tree = gtk.TreeStore(str, str, str, str, str, str)
self.diff_view = gtk.TreeView(self.diff_tree)
self.diff_column1 = gtk.TreeViewColumn("")
self.diff_column2 = gtk.TreeViewColumn(_("Section"))
self.diff_column3 = gtk.TreeViewColumn(_("Property"))
self.diff_column4 = gtk.TreeViewColumn(_("Original value"))
self.diff_column5 = gtk.TreeViewColumn(_("Current value"))
self.diff_cell = gtk.CellRendererText()
def activate_color(self, activate):
if activate:
self.diff_column1.set_attributes(self.diff_cell, text=0, background=5)
self.diff_column2.set_attributes(self.diff_cell, text=1, background=5)
self.diff_column3.set_attributes(self.diff_cell, text=2, background=5)
self.diff_column4.set_attributes(self.diff_cell, text=3, background=5)
self.diff_column5.set_attributes(self.diff_cell, text=4, background=5)
else:
self.diff_column1.clear_attributes(self.diff_cell)
self.diff_column2.clear_attributes(self.diff_cell)
self.diff_column3.clear_attributes(self.diff_cell)
self.diff_column4.clear_attributes(self.diff_cell)
self.diff_column5.clear_attributes(self.diff_cell)
self.diff_column1.set_attributes(self.diff_cell, text=0)
self.diff_column2.set_attributes(self.diff_cell, text=1)
self.diff_column3.set_attributes(self.diff_cell, text=2)
self.diff_column4.set_attributes(self.diff_cell, text=3)
self.diff_column5.set_attributes(self.diff_cell, text=4)
self.diff_cell.set_property("background", "white")
def _set_diff_view(self):
font_desc = pango.FontDescription("Normal 9")
self.diff_cell.set_property("font-desc", font_desc)
self.diff_scrolled.set_size_request(-1, 260)
self.diff_view.set_enable_search(True)
self.diff_view.set_search_column(2)
selection = self.diff_view.get_selection()
selection.set_mode(gtk.SELECTION_MULTIPLE)
self.diff_view.append_column(self.diff_column1)
self.diff_view.append_column(self.diff_column2)
self.diff_view.append_column(self.diff_column3)
self.diff_view.append_column(self.diff_column4)
self.diff_view.append_column(self.diff_column5)
# Diff Column 1
self.diff_column1.set_reorderable(True)
self.diff_column1.set_resizable(True)
self.diff_column1.connect("clicked", self.search_column, 0)
self.diff_column1.set_sort_column_id(0)
self.diff_column1.pack_start(self.diff_cell, True)
self.diff_column1.set_attributes(self.diff_cell, text=0)
# Diff Column 2
self.diff_column2.set_reorderable(True)
self.diff_column2.set_resizable(True)
self.diff_column2.connect("clicked", self.search_column, 1)
self.diff_column2.set_sort_column_id(1)
self.diff_column2.pack_start(self.diff_cell, True)
self.diff_column2.set_attributes(self.diff_cell, text=1)
# Diff Column 3
self.diff_column3.set_reorderable(True)
self.diff_column3.set_resizable(True)
self.diff_column3.connect("clicked", self.search_column, 2)
self.diff_column3.set_sort_column_id(2)
self.diff_column3.pack_start(self.diff_cell, True)
self.diff_column3.set_attributes(self.diff_cell, text=2)
# Diff Column 4
self.diff_column4.set_reorderable(True)
self.diff_column4.set_resizable(True)
self.diff_column4.connect("clicked", self.search_column, 3)
self.diff_column4.set_sort_column_id(3)
self.diff_column4.pack_start(self.diff_cell, True)
self.diff_column4.set_attributes(self.diff_cell, text=3)
# Diff Column 5
self.diff_column5.set_reorderable(True)
self.diff_column5.set_resizable(True)
self.diff_column5.connect("clicked", self.search_column, 4)
self.diff_column5.set_sort_column_id(4)
self.diff_column5.pack_start(self.diff_cell, True)
self.diff_column5.set_attributes(self.diff_cell, text=4)
self.diff_scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
def clear_diff_tree(self):
for i in range(len(self.diff_tree)):
iter = self.diff_tree.get_iter_root()
del(self.diff_tree[iter])
def change_status(self, iter, status):
self.diff_tree[iter][0] = status
self.diff_tree[iter][5] = self.colors.get_hex_color(status)
def set_parent_status(self, parent, status):
if status != "" and status.upper() != "U":
while parent:
self.change_status(parent, "M")
parent = self.diff_tree.iter_parent(parent)
def make_diff(self, parsed1, parsed2):
self.clear_diff_tree()
section = _("%s Info" % APP_DISPLAY_NAME)
parent = self.append_parent(None, section, "")
self.diff_it(parent, "", _("Profile"), parsed1.profile, parsed2.profile)
self.diff_it(parent, "", _("Profile Name"), parsed1.profile_name,
parsed2.profile_name)
self.diff_it(parent, "", _("Profile Options"),
",".join(parsed1.profile_options.keys()),
",".join(parsed2.profile_options.keys()))
self.diff_it(parent, "", _("Target"), parsed1.target, parsed2.target)
section = _("%s Info" % NMAP_DISPLAY_NAME)
parent = self.append_parent(None, section, "")
self.diff_it(parent, "", _("Debugging"), parsed1.debugging_level,
parsed2.debugging_level)
self.diff_it(parent, "", _("Verbosity"), parsed1.verbose_level,
parsed2.verbose_level)
self.diff_it(parent, "", _("Command"), parsed1.nmap_command,
parsed2.nmap_command)
self.diff_it(parent, "", _("Scanner version"), parsed1.scanner_version,
parsed2.scanner_version)
section = _("Scan Info")
parent = self.append_parent(None, section, "")
self.diff_it(parent, "", _("Open Ports"), parsed1.open_ports,
parsed2.open_ports)
self.diff_it(parent, "", _("Filtered Ports"), parsed1.filtered_ports,
parsed2.filtered_ports)
self.diff_it(parent, "", _("Closed Ports"), parsed1.closed_ports,
parsed2.closed_ports)
self.diff_it(parent, "", _("Hosts Up"), parsed1.hosts_up,
parsed2.hosts_up)
self.diff_it(parent, "", _("Hosts Down"), parsed1.hosts_down,
parsed2.hosts_down)
self.diff_it(parent, "", _("Hosts Scanned"), parsed1.hosts_scanned,
parsed2.hosts_scanned)
self.diff_it(parent, "", _("Finish date"), parsed1.formated_finish_date,
parsed2.formated_finish_date)
hosts1 = parsed1.hosts[:]
hosts2 = parsed2.hosts[:]
while hosts1:
host = hosts1.pop()
second_host = HostInfo()
host_state = "N"
for host2 in hosts2:
if (host.mac and host.mac == host2.mac) or \
(host.ip and host.ip == host2.ip) or \
(host.ipv6 and host.ipv6 == host2.ipv6):
second_host = host2
host_state = ""
del(hosts2[hosts2.index(host2)]) # Remove it from the hosts2
break
self.add_host_diff(host_state, host, second_host)
for host in hosts2:
self.add_host_diff("A", host, host)
def add_host_diff(self, host_state, host, host2=None):
section = _("Host")
if host.ip:
section = _("Host %s") % (host.ip["addr"])
elif host.ipv6:
section = _("Host %s") % (host.ipv6["addr"])
elif host.mac:
section = _("Host %s") % (host.mac["addr"])
parent = self.append_parent(None, section, host_state)
self.diff_it(parent, "", _("Comment"), host.comment,
host2.comment)
self.diff_it(parent, "", _("LastBoot"), host.uptime.get("lastboot", ""),
host2.uptime.get("lastboot", ""))
self.diff_it(parent, "", _("OS Match"), host.get_best_osmatch().get("name", ""),
host2.get_best_osmatch().get("name", ""))
host_ports = host.ports[:]
host2_ports = host2.ports[:]
for port in xrange(len(host_ports)):
# Making sure that extraports1 will get a sanity value to be processed
try:
extraports1 = host_ports[port].get("extraports", [])
except:
extraports1 = {}
else:
if len(extraports1) == 0:
extraports1 = {}
else:
extraports1 = extraports1[0]
# Making sure that extraports2 will get a sanity value to be processed
try:
extraports2 = host2_ports[port].get("extraports", [])
except:
extraports2 = {}
else:
if len(extraports2) == 0:
extraports2 = {}
else:
extraports2 = extraports2[0]
if extraports1 and extraports2:
self.add_extraports_diff(parent, "", extraports1, extraports2)
elif extraports1 and not extraports2:
self.add_extraports_diff(parent, "N", extraports1, extraports2)
elif not extraports1 and extraports2:
self.add_extraports_diff(parent, "A", extraports1, extraports2)
section = _("Ports")
parent = self.append_parent(parent, section, "")
# Making sure that ports1 will get a sanity value to be processed
try:
ports1 = host_ports[port].get("port", [])
except:
ports1 = {}
else:
if len(ports1) == 0:
ports1 = {}
elif len(ports1) == 1:
ports1 = ports1[0]
# Making sure that ports2 will get a sanity value to be processed
try:
ports2 = host2_ports[port].get("port", [])
except:
ports2 = {}
else:
if len(ports2) == 0:
ports2 = [{}]
elif len(ports2) == 1:
ports2 = ports2[0]
if type(ports2)!= type([]):
ports2 = [ports2]
if type(ports1) != type([]):
ports1 = [ports1]
for p1 in ports1:
if not p1:
continue
p2 = [port2 for port2 in ports2 \
if port2.get("portid", "a") == p1.get("portid", "b")]
if p2: # Removing found port
ports2.remove(p2[0])
if p1 and p2:
self.add_port_diff(parent, "", p1, p2[0])
elif p1 and not p2:
self.add_port_diff(parent, "N", p1, {})
for p2 in ports2: # If there is something left...
self.add_port_diff(parent, "A", {}, p2)
def add_port_diff(self, port_parent, state, port1, port2):
if (port1 or port2) and (type(port1) == type({})) and (type(port2) == type({})):
section = port1.get("portid", False)
if not section: # If port1 is empty, then, try port2
section = port2.get("portid", "")
parent = self.append_parent(port_parent, section, state)
self.diff_it(parent, "",
_("State"), port1.get("port_state", ""),
port2.get("port_state", ""))
self.diff_it(parent, "",
_("Service Name"), port1.get("service_name", ""),
port2.get("service_name", ""))
self.diff_it(parent, "",
_("Product"), port1.get("service_product", ""),
port2.get("service_product", ""))
self.diff_it(parent, "",
_("Service Version"), port1.get("service_version", ""),
port2.get("service_version", ""))
self.diff_it(parent, "",
_("Protocol"), port1.get("protocol", ""),
port2.get("protocol", ""))
self.diff_it(parent, "",
_("Extra Info"), port1.get("service_extrainfo", ""),
port2.get("service_extrainfo", ""))
self.diff_it(parent, "",
_("Service Conf"), port1.get("service_conf", ""),
port2.get("service_conf", ""))
# Last parent status modification
if state.upper() == "A":
self.change_status(parent, "A")
def add_extraports_diff(self, host_parent, state, extraports1, extraports2):
if extraports1 or extraports2:
section = _("Extraports")
parent = self.append_parent(host_parent, section, state)
self.set_parent_status(parent, state)
self.diff_it(parent, "", _("Count"), extraports1.get("count"),
extraports2.get("count"))
self.diff_it(parent, "", _("State"), extraports1.get("state"),
extraports2.get("state"))
def diff_it(self, parent, section, prop_name, prop1, prop2):
if prop1 or prop2:
state = diff_state(prop1, prop2)
self.set_parent_status(parent, state)
self.diff_tree.append(parent, [state,
section,
prop_name,
prop1,
prop2,
self.colors.get_hex_color(state)])
return state
def append_parent(self, parent, section, state):
self.set_parent_status(parent, state)
return self.diff_tree.append(parent, [state, section, "", "", "",
self.colors.get_hex_color(state)])
def search_column(self, widget, column_id):
self.diff_view.set_search_column(column_id)
def _pack_widgets(self):
self._pack_noexpand_nofill(self.diff_title)
self._pack_expand_fill(self.diff_box)
self.diff_box._pack_noexpand_nofill(hig_box_space_holder())
self.diff_box._pack_expand_fill(self.diff_scrolled)
self.diff_scrolled.add(self.diff_view)
class ScanChooser(HIGVBox):
"""This class allows the selection of scan results from the list of open
tabs or from a file. It emits the "changed" signal when the scan selection
has changed."""
__gsignals__ = {
"changed": (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ())
}
def __init__(self, scans, title):
self.__gobject_init__()
self.title = title
self.scan_dict = {}
# Setting HIGVBox
self.set_border_width(5)
self.set_spacing(6)
self._create_widgets()
self._pack_hbox()
self._attaching_widgets()
self._set_scrolled()
self._set_text_view()
self._set_open_button()
for scan in scans:
self.add_scan(scan.scan_name or scan.get_nmap_command(), scan)
self.combo_scan.connect('changed', self.show_scan)
self.combo_scan.connect('changed', lambda x: self.emit('changed'))
self._pack_noexpand_nofill(self.lbl_scan)
self._pack_expand_fill(self.hbox)
def _create_widgets(self):
self.lbl_scan = HIGSectionLabel(self.title)
self.hbox = HIGHBox()
self.table = HIGTable()
self.list_scan = gtk.ListStore(str)
self.combo_scan = gtk.ComboBoxEntry(self.list_scan, 0)
self.btn_open_scan = gtk.Button(stock=gtk.STOCK_OPEN)
self.exp_scan = gtk.Expander(_("Scan Output"))
self.scrolled = gtk.ScrolledWindow()
self.txt_scan_result = gtk.TextView()
self.txg_tag = gtk.TextTag("scan_style")
def get_buffer(self):
return self.txt_scan_result.get_buffer()
def show_scan(self, widget):
nmap_output = self.get_nmap_output()
if nmap_output:
self.txt_scan_result.get_buffer().set_text(nmap_output)
def normalize_output(self, output):
return "\n".join(output.split("\\n"))
def _pack_hbox(self):
self.hbox._pack_noexpand_nofill(hig_box_space_holder())
self.hbox._pack_expand_fill(self.table)
def _attaching_widgets(self):
self.table.attach(self.combo_scan, 0, 1, 0, 1, yoptions=0)
self.table.attach(
self.btn_open_scan, 1, 2, 0, 1, yoptions=0, xoptions=0)
self.table.attach(self.exp_scan, 0, 2, 1, 2)
def _set_scrolled(self):
self.scrolled.set_border_width(5)
self.scrolled.set_size_request(-1, 130)
# Packing scrolled window into expander
self.exp_scan.add(self.scrolled)
# Packing text view into scrolled window
self.scrolled.add_with_viewport(self.txt_scan_result)
# Setting scrolled window
self.scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
def _set_text_view(self):
self.txg_table = self.txt_scan_result.get_buffer().get_tag_table()
self.txg_table.add(self.txg_tag)
self.txg_tag.set_property("family", "Monospace")
self.txt_scan_result.set_wrap_mode(gtk.WRAP_WORD)
self.txt_scan_result.set_editable(False)
self.txt_scan_result.get_buffer().connect(
"changed", self._text_changed_cb)
def _set_open_button(self):
self.btn_open_scan.connect('clicked', self.open_file)
def open_file(self, widget):
file_chooser = ResultsFileSingleChooserDialog(_("Select Scan Result"))
response = file_chooser.run()
file_chosen = file_chooser.get_filename()
file_chooser.destroy()
if response == gtk.RESPONSE_OK:
try:
parser = NmapParser()
parser.parse_file(file_chosen)
except xml.sax.SAXParseException, e:
alert = HIGAlertDialog(
message_format='<b>%s</b>' % _('Error parsing file'),
secondary_text=_(
"The file is not an Nmap XML output file. "
"The parsing error that occurred was\n%s") % str(e))
alert.run()
alert.destroy()
return False
except Exception, e:
alert = HIGAlertDialog(
message_format='<b>%s</b>' % _(
'Cannot open selected file'),
secondary_text=_("""\
This error occurred while trying to open the file:
%s""") % str(e))
alert.run()
alert.destroy()
return False
scan_name = os.path.split(file_chosen)[-1]
self.add_scan(scan_name, parser)
self.combo_scan.set_active(len(self.list_scan) - 1)
class ProfileEditor(HIGWindow):
def __init__(self, command=None, profile_name=None,
deletable=True, overwrite=False):
HIGWindow.__init__(self)
self.connect("delete_event", self.exit)
self.set_title(_('Profile Editor'))
self.set_position(gtk.WIN_POS_CENTER)
self.deletable = deletable
self.profile_name = profile_name
self.overwrite = overwrite
# Used to block recursive updating of the command entry when the
# command entry causes the OptionBuilder widgets to change.
self.inhibit_command_update = False
self.__create_widgets()
self.__pack_widgets()
self.profile = CommandProfile()
self.ops = NmapOptions()
if profile_name:
log.debug("Showing profile %s" % profile_name)
prof = self.profile.get_profile(profile_name)
# Interface settings
self.profile_name_entry.set_text(profile_name)
self.profile_description_text.get_buffer().set_text(
prof['description'])
command_string = prof['command']
self.ops.parse_string(command_string)
if command:
self.ops.parse_string(command)
self.option_builder = OptionBuilder(
Path.profile_editor, self.ops,
self.update_command, self.help_field.get_buffer())
log.debug("Option groups: %s" % str(self.option_builder.groups))
log.debug("Option section names: %s" % str(
self.option_builder.section_names))
#log.debug("Option tabs: %s" % str(self.option_builder.tabs))
for tab in self.option_builder.groups:
self.__create_tab(
_(tab),
_(self.option_builder.section_names[tab]),
self.option_builder.tabs[tab])
self.update_command()
def command_entry_changed_cb(self, widget):
command_string = self.command_entry.get_text().decode("UTF-8")
self.ops.parse_string(command_string)
self.inhibit_command_update = True
self.option_builder.update()
self.inhibit_command_update = False
def update_command(self):
"""Regenerate and display the command."""
if not self.inhibit_command_update:
# Block recursive updating of the OptionBuilder widgets when they
# cause a change in the command entry.
self.command_entry.handler_block(self.command_entry_changed_cb_id)
self.command_entry.set_text(self.ops.render_string())
self.command_entry.handler_unblock(
self.command_entry_changed_cb_id)
def update_help_name(self, widget, extra):
self.help_field.get_buffer().set_text(
"Profile name\n\nThis is how the profile will be identified "
"in the drop-down combo box in the scan tab.")
def update_help_desc(self, widget, extra):
self.help_field.get_buffer().set_text(
"Description\n\nThe description is a full description of what "
"the scan does, which may be long.")
def __create_widgets(self):
###
# Vertical box to keep 3 boxes
self.main_whole_box = HIGVBox()
self.upper_box = HIGHBox()
self.middle_box = HIGHBox()
self.lower_box = HIGHBox()
#self.main_vbox = HIGVBox()
self.command_entry = gtk.Entry()
self.command_entry_changed_cb_id = self.command_entry.connect(
"changed", self.command_entry_changed_cb)
self.scan_button = HIGButton(_("Scan"))
self.scan_button.connect("clicked", self.run_scan)
self.notebook = gtk.Notebook()
# Profile info page
self.profile_info_vbox = HIGVBox()
self.profile_info_label = HIGSectionLabel(_('Profile Information'))
self.profile_name_label = HIGEntryLabel(_('Profile name'))
self.profile_name_entry = gtk.Entry()
self.profile_name_entry.connect(
'enter-notify-event', self.update_help_name)
self.profile_description_label = HIGEntryLabel(_('Description'))
self.profile_description_scroll = HIGScrolledWindow()
self.profile_description_scroll.set_border_width(0)
self.profile_description_text = HIGTextView()
self.profile_description_text.connect(
'motion-notify-event', self.update_help_desc)
# Buttons
self.buttons_hbox = HIGHBox()
self.cancel_button = HIGButton(stock=gtk.STOCK_CANCEL)
self.cancel_button.connect('clicked', self.exit)
self.delete_button = HIGButton(stock=gtk.STOCK_DELETE)
self.delete_button.connect('clicked', self.delete_profile)
self.save_button = HIGButton(_("Save Changes"), stock=gtk.STOCK_SAVE)
self.save_button.connect('clicked', self.save_profile)
###
self.help_vbox = HIGVBox()
self.help_label = HIGSectionLabel(_('Help'))
self.help_scroll = HIGScrolledWindow()
self.help_scroll.set_border_width(0)
self.help_field = HIGTextView()
self.help_field.set_cursor_visible(False)
self.help_field.set_left_margin(5)
self.help_field.set_editable(False)
self.help_vbox.set_size_request(200, -1)
###
def __pack_widgets(self):
###
self.add(self.main_whole_box)
# Packing command entry to upper box
self.upper_box._pack_expand_fill(self.command_entry)
self.upper_box._pack_noexpand_nofill(self.scan_button)
# Packing notebook (left) and help box (right) to middle box
self.middle_box._pack_expand_fill(self.notebook)
self.middle_box._pack_expand_fill(self.help_vbox)
# Packing buttons to lower box
self.lower_box.pack_end(self.buttons_hbox)
# Packing the three vertical boxes to the main box
self.main_whole_box._pack_noexpand_nofill(self.upper_box)
self.main_whole_box._pack_expand_fill(self.middle_box)
self.main_whole_box._pack_noexpand_nofill(self.lower_box)
###
# Packing profile information tab on notebook
self.notebook.append_page(
self.profile_info_vbox, gtk.Label(_('Profile')))
self.profile_info_vbox.set_border_width(5)
table = HIGTable()
self.profile_info_vbox._pack_noexpand_nofill(self.profile_info_label)
self.profile_info_vbox._pack_expand_fill(HIGSpacer(table))
self.profile_description_scroll.add(self.profile_description_text)
vbox_desc = HIGVBox()
vbox_desc._pack_noexpand_nofill(self.profile_description_label)
vbox_desc._pack_expand_fill(hig_box_space_holder())
vbox_ann = HIGVBox()
vbox_ann._pack_expand_fill(hig_box_space_holder())
table.attach(
self.profile_name_label, 0, 1, 0, 1, xoptions=0, yoptions=0)
table.attach(self.profile_name_entry, 1, 2, 0, 1, yoptions=0)
table.attach(vbox_desc, 0, 1, 1, 2, xoptions=0)
table.attach(self.profile_description_scroll, 1, 2, 1, 2)
# Packing buttons on button_hbox
self.buttons_hbox._pack_expand_fill(hig_box_space_holder())
if self.deletable:
self.buttons_hbox._pack_noexpand_nofill(self.delete_button)
self.buttons_hbox._pack_noexpand_nofill(self.cancel_button)
self.buttons_hbox._pack_noexpand_nofill(self.save_button)
self.buttons_hbox.set_border_width(5)
self.buttons_hbox.set_spacing(6)
###
self.help_vbox._pack_noexpand_nofill(self.help_label)
self.help_vbox._pack_expand_fill(self.help_scroll)
self.help_scroll.add(self.help_field)
self.help_vbox.set_border_width(1)
self.help_vbox.set_spacing(1)
###
def __create_tab(self, tab_name, section_name, tab):
log.debug(">>> Tab name: %s" % tab_name)
log.debug(">>>Creating profile editor section: %s" % section_name)
vbox = HIGVBox()
if tab.notscripttab: # if notscripttab is set
table = HIGTable()
table.set_row_spacings(2)
section = HIGSectionLabel(section_name)
vbox._pack_noexpand_nofill(section)
vbox._pack_noexpand_nofill(HIGSpacer(table))
vbox.set_border_width(5)
tab.fill_table(table, True)
else:
hbox = tab.get_hmain_box()
vbox.pack_start(hbox, True, True, 0)
self.notebook.append_page(vbox, gtk.Label(tab_name))
def save_profile(self, widget):
if self.overwrite:
self.profile.remove_profile(self.profile_name)
profile_name = self.profile_name_entry.get_text()
if profile_name == '':
alert = HIGAlertDialog(
message_format=_('Unnamed profile'),
secondary_text=_(
'You must provide a name for this profile.'))
alert.run()
alert.destroy()
self.profile_name_entry.grab_focus()
return None
command = self.ops.render_string()
buf = self.profile_description_text.get_buffer()
description = buf.get_text(
buf.get_start_iter(), buf.get_end_iter())
try:
self.profile.add_profile(
profile_name,
command=command,
description=description)
except ValueError:
alert = HIGAlertDialog(
message_format=_('Disallowed profile name'),
secondary_text=_('Sorry, the name "%s" is not allowed due '
'to technical limitations. (The underlying '
'ConfigParser used to store profiles does not allow '
'it.) Choose a different name.' % profile_name))
alert.run()
alert.destroy()
return
self.scan_interface.toolbar.profile_entry.update()
self.destroy()
def clean_profile_info(self):
self.profile_name_entry.set_text('')
self.profile_description_text.get_buffer().set_text('')
def set_scan_interface(self, interface):
self.scan_interface = interface
def exit(self, *args):
self.destroy()
def delete_profile(self, widget=None, extra=None):
if self.deletable:
dialog = HIGDialog(buttons=(gtk.STOCK_OK, gtk.RESPONSE_OK,
gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL))
alert = HIGEntryLabel('<b>' + _("Deleting Profile") + '</b>')
text = HIGEntryLabel(_(
'Your profile is going to be deleted! ClickOk to continue, '
'or Cancel to go back to Profile Editor.'))
hbox = HIGHBox()
hbox.set_border_width(5)
hbox.set_spacing(12)
vbox = HIGVBox()
vbox.set_border_width(5)
vbox.set_spacing(12)
image = gtk.Image()
image.set_from_stock(
gtk.STOCK_DIALOG_WARNING, gtk.ICON_SIZE_DIALOG)
vbox.pack_start(alert)
vbox.pack_start(text)
hbox.pack_start(image)
hbox.pack_start(vbox)
dialog.vbox.pack_start(hbox)
dialog.vbox.show_all()
response = dialog.run()
dialog.destroy()
if response == gtk.RESPONSE_CANCEL:
return True
self.profile.remove_profile(self.profile_name)
self.update_profile_entry()
self.destroy()
def run_scan(self, widget=None):
command_string = self.command_entry.get_text().decode("UTF-8")
self.scan_interface.command_toolbar.command = command_string
self.scan_interface.start_scan_cb()
self.exit()
def update_profile_entry(self, widget=None, extra=None):
self.scan_interface.toolbar.profile_entry.update()
list = self.scan_interface.toolbar.profile_entry.get_model()
length = len(list)
if length > 0:
self.scan_interface.toolbar.profile_entry.set_active(0)
class DiffWindow(gtk.Window):
def __init__(self, scans):
"""scans in the format: {"scan_title":parsed_scan}
"""
gtk.Window.__init__(self)
self.set_title(_("Compare Results"))
self.scans = scans
self.umit_conf = UmitConf()
self.colors = Colors()
# Diff views
self.text_view = DiffText(self.colors, self.umit_conf.colored_diff)
self.compare_view = DiffTree(self.colors)
self.temp_html_file = None
self._create_widgets()
self._pack_widgets()
self._connect_widgets()
# Settings
if self.umit_conf.diff_mode == "text":
self.text_mode.set_active(True)
else:
self.compare_mode.set_active(True)
self.check_color.set_active(self.umit_conf.colored_diff)
# Initial Size Request
self.initial_size = self.size_request()
def _show_help(self, action):
webbrowser.open("file://%s" % os.path.join(Path.docs_dir, "help.html"), new=2)
def _create_widgets(self):
self.main_vbox = HIGVBox()
self.hbox_mode = HIGHBox()
self.hbox_settings = HIGHBox()
self.hbox_buttons = HIGHBox()
self.hbox_result = HIGHBox()
self.btn_open_browser = HIGButton(_("Open in Browser"), stock=gtk.STOCK_EXECUTE)
self.btn_help = HIGButton(stock=gtk.STOCK_HELP)
self.btn_close = HIGButton(stock=gtk.STOCK_CLOSE)
self.check_color = gtk.CheckButton(_("Enable colored diffies"))
self.btn_legend = HIGButton(_("Color Descriptions"), stock=gtk.STOCK_SELECT_COLOR)
self.text_mode = gtk.ToggleButton(_("Text Mode"))
self.compare_mode = gtk.ToggleButton(_("Compare Mode"))
self.vpaned = gtk.VPaned()
self.hpaned = gtk.HPaned()
self.scan_chooser1 = ScanChooser(self.scans, "1")
self.scan_chooser2 = ScanChooser(self.scans, "2")
self.scan_buffer1 = self.scan_chooser1.get_buffer()
self.scan_buffer2 = self.scan_chooser2.get_buffer()
def _pack_widgets(self):
self.main_vbox.set_border_width(6)
self.vpaned.pack1(self.hpaned, True, False)
self.vpaned.pack2(self.hbox_result)
self.hpaned.pack1(self.scan_chooser1, True, False)
self.hpaned.pack2(self.scan_chooser2, True, False)
self.hbox_buttons._pack_expand_fill(self.btn_help)
self.hbox_buttons._pack_expand_fill(self.btn_legend)
self.hbox_buttons._pack_expand_fill(self.btn_open_browser)
self.hbox_buttons._pack_expand_fill(self.btn_close)
self.hbox_buttons.set_homogeneous(True)
self.hbox_mode.set_homogeneous(True)
self.hbox_mode.pack_start(self.text_mode)
self.hbox_mode.pack_start(self.compare_mode)
self.hbox_settings._pack_noexpand_nofill(self.hbox_mode)
self.hbox_settings._pack_expand_fill(self.check_color)
self.main_vbox._pack_expand_fill(self.vpaned)
self.main_vbox._pack_noexpand_nofill(self.hbox_settings)
self.main_vbox._pack_noexpand_nofill(self.hbox_buttons)
self.add(self.main_vbox)
def _connect_widgets(self):
self.connect("delete-event", self.close)
self.btn_legend.connect("clicked", self.show_legend_window)
self.btn_help.connect("clicked", self._show_help)
self.btn_close.connect("clicked", self.close)
self.btn_open_browser.connect("clicked", self.open_browser)
self.check_color.connect("toggled", self._set_color)
self.text_mode.connect("clicked", self._change_to_text)
self.compare_mode.connect("clicked", self._change_to_compare)
self.scan_chooser1.exp_scan.connect('activate', self.resize_vpane)
self.scan_chooser2.exp_scan.connect('activate', self.resize_vpane)
self.scan_chooser1.connect('changed', self.refresh_diff)
self.scan_chooser2.connect('changed', self.refresh_diff)
def open_browser(self, widget):
text1 = self.scan_chooser1.nmap_output
text2 = self.scan_chooser2.nmap_output
if text1 is None or text2 is None:
alert = HIGAlertDialog(
message_format='<b>'+_('Select Scan')+'</b>',
secondary_text=_("You must select two different scans to \
generate diff."))
alert.run()
alert.destroy()
return False
if text1 == '' and text2 == '':
alert = HIGAlertDialog(
message_format='<b>'+_('No Text Output')+'</b>',
secondary_text=_("Neither of the scans you selected has \
any text output. (Scans loaded from plain Nmap XML output files do not contain \
text output.) The HTML diff shows only differences between text output, so \
there is nothing to show."))
alert.run()
alert.destroy()
return False
# True tells splitlines to keep line endings.
text1 = text1.splitlines(True)
text2 = text2.splitlines(True)
if self.temp_html_file is not None:
self.temp_html_file.close()
# A NamedTemporaryFile is deleted when it is closed.
self.temp_html_file = tempfile.NamedTemporaryFile(suffix = ".html", prefix = "zenmap-diff-")
if use_html:
diff = DiffHtml(text1, text2)
diff = diff.generate()
self.temp_html_file.write(''.join(diff))
else:
diff = Diff(text1, text2)
diff = diff.generate ()
diff.insert(0, '''<pre>(This diff is been shown in pure text \
because you dont have Python 2.4 or higher.)\n''')
diff.append('</pre>')
self.temp_html_file.writelines(diff)
self.temp_html_file.flush()
webbrowser.open("file://" + self.temp_html_file.name, autoraise=1)
def show_legend_window(self, widget):
legend_window = DiffLegendWindow(self.colors)
legend_window.run()
legend_window.destroy()
self.refresh_diff(None)
def refresh_diff (self, widget):
"""This method is called whenever the diff output might have changed,
such as when a different scan was selected in one of the choosers."""
if self.compare_mode.get_active():
# Graphical comparison mode.
parsed1 = self.scan_chooser1.parsed_scan
parsed2 = self.scan_chooser2.parsed_scan
if parsed1 is not None and parsed2 is not None:
self.compare_view.make_diff(parsed1, parsed2)
else:
self.compare_view.clear_diff_tree()
self.compare_view.activate_color(self.check_color.get_active())
else:
# Text comparison mode.
text1 = self.scan_chooser1.nmap_output
text2 = self.scan_chooser2.nmap_output
if text1 is not None and text2 is not None:
diff = Diff(text1.split('\n'), text2.split('\n'))
self.text_view.set_text('\n'.join(diff.generate_without_banner()))
else:
self.text_view.clear()
self.text_view.activate_color(self.check_color.get_active())
def resize_vpane(self, widget):
exp1 = not widget.get_expanded()
if widget == self.scan_chooser1.exp_scan:
exp2 = self.scan_chooser2.exp_scan.get_expanded()
else:
exp2 = self.scan_chooser1.exp_scan.get_expanded()
if not exp1 and not exp2:
self.vpaned.compute_position(-1, 0, 500)
self.size_allocate(gtk.gdk.Rectangle(width=self.initial_size[0],
height=self.initial_size[1]))
self.queue_resize()
def _change_to_text(self, widget):
if not widget.get_active():
return
self.umit_conf.diff_mode = "text"
children = self.hbox_result.get_children()
if children:
self.hbox_result.remove(children[0])
self.compare_view.hide()
self.hbox_result._pack_expand_fill(self.text_view)
self.text_view.show_all()
self.compare_mode.set_active(False)
self.refresh_diff(None)
def _change_to_compare(self, widget):
if not widget.get_active():
return
self.umit_conf.diff_mode = "compare"
children = self.hbox_result.get_children()
if children:
self.hbox_result.remove(children[0])
self.text_view.hide()
self.hbox_result._pack_expand_fill(self.compare_view)
self.compare_view.show_all()
self.text_mode.set_active(False)
self.refresh_diff(None)
def _set_color(self, widget):
activate = widget.get_active()
self.umit_conf.colored_diff = activate
self.compare_view.activate_color(activate)
self.text_view.activate_color(activate)
def close(self, widget=None, extra=None):
if self.temp_html_file is not None:
self.temp_html_file.close()
self.destroy()
