def __init__(self):
HIGVBox.__init__(self)
self._create_widgets()
self._set_port_list()
self._set_host_list()
self._pack_widgets()
def _create_widgets(self):
self.rn_hbox = gtk.HBox()
self.rn_hbox.set_spacing(4)
self.rn_vbox = gtk.VBox()
# RadialNet's widgets
self.radialnet = RadialNet(LAYOUT_WEIGHTED)
self.control = ControlWidget(self.radialnet)
self.fisheye = ControlFisheye(self.radialnet)
self.rn_toolbar = Toolbar(self.radialnet,
self,
self.control,
self.fisheye)
self.display_panel = HIGVBox()
self.radialnet.set_no_show_all(True)
self.slow_vbox = HIGVBox()
self.slow_label = gtk.Label()
self.slow_vbox.pack_start(self.slow_label, False, False)
show_button = gtk.Button(_("Show the topology anyway"))
show_button.connect("clicked", self.show_anyway)
self.slow_vbox.pack_start(show_button, False, False)
self.slow_vbox.show_all()
self.slow_vbox.set_no_show_all(True)
self.slow_vbox.hide()
self.radialnet.show()
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 __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 __init__(self, inventory):
HIGVBox.__init__(self)
self.set_border_width(6)
self.set_spacing(4)
self.network_inventory = inventory
self._create_widgets()
self._pack_widgets()
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()
class ScanResultNotebook(HIGNotebook):
"""This is the right side of a ScanResult, the notebook with the tabs such
as "Nmap Output"."""
def __init__(self, inventory, scans_store):
HIGNotebook.__init__(self)
self.set_border_width(5)
self.__create_widgets(inventory, scans_store)
self.scans_list.scans_list.connect(
"row-activated", self._scan_row_activated)
self.append_page(self.nmap_output_page, gtk.Label(_('Nmap Output')))
self.append_page(self.open_ports_page, gtk.Label(_('Ports / Hosts')))
self.append_page(self.topology_page, gtk.Label(_('Topology')))
self.append_page(self.host_details_page, gtk.Label(_('Host Details')))
self.append_page(self.scans_list_page, gtk.Label(_('Scans')))
def host_mode(self):
self.open_ports.host.host_mode()
def port_mode(self):
self.open_ports.host.port_mode()
def __create_widgets(self, inventory, scans_store):
self.open_ports_page = HIGVBox()
self.nmap_output_page = HIGVBox()
self.topology_page = HIGVBox()
self.host_details_page = HIGScrolledWindow()
self.host_details_vbox = HIGVBox()
self.scans_list_page = HIGVBox()
self.open_ports = ScanOpenPortsPage()
self.nmap_output = ScanNmapOutputPage(scans_store)
self.topology = TopologyPage(inventory)
self.scans_list = ScanScanListPage(scans_store)
self.no_selected = gtk.Label(_('No host selected.'))
self.host_details = self.no_selected
self.open_ports_page.add(self.open_ports)
self.nmap_output_page.add(self.nmap_output)
self.topology_page.add(self.topology)
self.scans_list_page.add(self.scans_list)
self.host_details_page.add_with_viewport(self.host_details_vbox)
self.host_details_vbox._pack_expand_fill(self.host_details)
def _scan_row_activated(self, treeview, path, view_column):
"""Switch back to the Nmap Output view when a scan is activated
(double-clicked) on the scans list."""
self.nmap_output.set_active_iter(treeview.get_model().get_iter(path))
self.set_current_page(0)
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 __init__(self, scans_store):
HIGVBox.__init__(self)
self.set_spacing(4)
scans_store.connect("row-changed", self._row_changed)
self.scans_list = gtk.TreeView(scans_store)
self.scans_list.get_selection().connect(
"changed", self._selection_changed)
status_col = gtk.TreeViewColumn(_("Status"))
cell = gtk.CellRendererText()
status_col.pack_start(cell)
status_col.set_cell_data_func(cell, status_data_func)
self.scans_list.append_column(status_col)
command_col = gtk.TreeViewColumn(_("Command"))
cell = gtk.CellRendererText()
command_col.pack_start(cell)
command_col.set_cell_data_func(cell, command_data_func)
self.scans_list.append_column(command_col)
scrolled_window = HIGScrolledWindow()
scrolled_window.set_border_width(0)
scrolled_window.add(self.scans_list)
self.pack_start(scrolled_window, True, True)
hbox = HIGHBox()
buttonbox = gtk.HButtonBox()
buttonbox.set_layout(gtk.BUTTONBOX_START)
buttonbox.set_spacing(4)
self.append_button = HIGButton(_("Append Scan"), gtk.STOCK_ADD)
buttonbox.pack_start(self.append_button, False)
self.remove_button = HIGButton(_("Remove Scan"), gtk.STOCK_REMOVE)
buttonbox.pack_start(self.remove_button, False)
self.cancel_button = HIGButton(_("Cancel Scan"), gtk.STOCK_CANCEL)
buttonbox.pack_start(self.cancel_button, False)
hbox.pack_start(buttonbox, padding=4)
self.pack_start(hbox, False, padding=4)
self._update()
def _create_widgets(self):
self.vbox = HIGVBox()
self.button_box = gtk.HButtonBox()
self.text = gtk.Label()
self.btn_ok = gtk.Button(stock=gtk.STOCK_OK)
def _create_widgets(self):
# Mode buttons
self.host_mode_button = gtk.ToggleButton(_("Hosts"))
self.service_mode_button = gtk.ToggleButton(_("Services"))
self.buttons_box = gtk.HBox()
# Main window vbox
self.main_vbox = HIGVBox()
# Host list
self.host_list = gtk.ListStore(object, str, str)
self.host_list.set_sort_func(1000, cmp_treemodel_addr)
self.host_list.set_sort_column_id(1000, gtk.SORT_ASCENDING)
self.host_view = gtk.TreeView(self.host_list)
self.pic_column = gtk.TreeViewColumn(_('OS'))
self.host_column = gtk.TreeViewColumn(_('Host'))
self.os_cell = gtk.CellRendererPixbuf()
self.host_cell = gtk.CellRendererText()
# Service list
self.service_list = gtk.ListStore(str)
self.service_list.set_sort_column_id(0, gtk.SORT_ASCENDING)
self.service_view = gtk.TreeView(self.service_list)
self.service_column = gtk.TreeViewColumn(_('Service'))
self.service_cell = gtk.CellRendererText()
self.scrolled = gtk.ScrolledWindow()
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 __init__(self, scans):
gtk.Window.__init__(self)
self.set_title(_("Compare Results"))
self.ndiff_process = None
# We allow the user to start a new diff before the old one has
# finished. We have to keep references to old processes until they
# finish to avoid problems when tearing down the Python interpreter at
# program exit.
self.old_processes = []
self.timer_id = None
self.main_vbox = HIGVBox()
self.diff_view = DiffView()
self.diff_view.set_size_request(-1, 100)
self.hbox_buttons = HIGHBox()
self.progress = gtk.ProgressBar()
self.btn_close = HIGButton(stock=gtk.STOCK_CLOSE)
self.hbox_selection = HIGHBox()
self.scan_chooser_a = ScanChooser(scans, _(u"A Scan"))
self.scan_chooser_b = ScanChooser(scans, _(u"B Scan"))
self._pack_widgets()
self._connect_widgets()
self.set_default_size(-1, 500)
# Initial Size Request
self.initial_size = self.get_size()
def highlight_tab(self):
# Creating highlight tab main box
self.highlight_main_vbox = HIGVBox()
# Creating highlight tab main table
self.highlight_main_table = HIGTable()
self.highlight_main_table.set_border_width(6)
#############
# Properties:
self.property_names = {
"details": [_("details"), "MAC Address:"],
"port_list": [_("port listing title"), "PORT STATE SERVICE"],
"open_port": [_("open port"), "22/tcp open ssh"],
"closed_port": [_("closed port"), "70/tcp closed gopher"],
"filtered_port": [_("filtered port"), "80/tcp filtered http"],
"date": [_("date"), "2006-05-26 11:14 BRT"],
"hostname": [_("hostname"), "scanme.nmap.org"],
"ip": [_("ip"), "127.0.0.1"],
}
for p in self.property_names:
settings = self.nmap_highlight.__getattribute__(p)
self.property_names[p].append(settings[0])
self.property_names[p].append(settings[1])
self.property_names[p].append(settings[2])
self.property_names[p].append(gtk.gdk.Color(*settings[3]))
self.property_names[p].append(gtk.gdk.Color(*settings[4]))
self.property_names[p].append(settings[5])
# Creating properties and related widgets and attaching it to main
# table
y1 = 0
y2 = 1
for p in self.property_names:
hp = HighlightProperty(p, self.property_names[p])
self.highlight_main_table.attach(hp.property_name_label, 0, 1, y1, y2)
self.highlight_main_table.attach(hp.example_label, 1, 2, y1, y2)
self.highlight_main_table.attach(hp.bold_tg_button, 2, 3, y1, y2)
self.highlight_main_table.attach(hp.italic_tg_button, 3, 4, y1, y2)
self.highlight_main_table.attach(hp.underline_tg_button, 4, 5, y1, y2)
self.highlight_main_table.attach(hp.text_color_button, 5, 6, y1, y2)
self.highlight_main_table.attach(hp.highlight_color_button, 6, 7, y1, y2)
# Setting example styles and colors
hp.update_example()
self.property_names[p].append(hp)
y1 += 1
y2 += 1
# Packing main table into main vbox
self.highlight_main_vbox.pack_start(self.highlight_main_table)
# Adding color tab
self.properties_notebook.append_page(self.highlight_main_vbox, gtk.Label(_("Highlight definitions")))
def _create_widgets(self):
self.main_vbox = HIGVBox()
self.diff_view = DiffView()
self.diff_view.set_size_request(-1, 100)
self.hbox_buttons = HIGHBox()
self.progress = gtk.ProgressBar()
self.btn_close = HIGButton(stock=gtk.STOCK_CLOSE)
self.hbox_selection = HIGHBox()
self.scan_chooser_a = ScanChooser(self.scans, _(u"A Scan"))
self.scan_chooser_b = ScanChooser(self.scans, _(u"B Scan"))
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 __init__(self, scan_interface):
HIGVBox.__init__(self)
self._scan_interface = scan_interface
self._create_widgets()
self._connect_widgets()
self._pack_widgets()
self._set_scrolled()
self._set_host_list()
self._set_service_list()
self._pack_expand_fill(self.main_vbox)
self.mode = None
# Default mode is host mode
self.host_mode(self.host_mode_button)
self.host_view.show_all()
self.service_view.show_all()
def _create_widgets(self):
self.vbox = HIGVBox()
self.bottom_hbox = gtk.HBox()
self.bottom_label = gtk.Label()
self.btn_box = gtk.HButtonBox()
self.btn_open = HIGButton(stock=gtk.STOCK_OPEN)
self.btn_append = HIGButton(_("Append"), gtk.STOCK_ADD)
self.btn_close = HIGButton(stock=gtk.STOCK_CLOSE)
self.search_gui = SearchGUI(self)
def __init__(self, host):
HIGVBox.__init__(self)
self.__create_widgets()
self.set_os_image(get_os_logo(host))
self.set_vulnerability_image(get_vulnerability_logo(host.get_open_ports()))
self.set_host_status(
{
"state": host.get_state(),
"open": str(host.get_open_ports()),
"filtered": str(host.get_filtered_ports()),
"closed": str(host.get_closed_ports()),
"scanned": str(host.get_scanned_ports()),
"uptime": host.get_uptime()["seconds"],
"lastboot": host.get_uptime()["lastboot"],
}
)
addresses = {}
if host.ip is not None:
addresses["ipv4"] = host.ip["addr"]
if host.ipv6 is not None:
addresses["ipv6"] = host.ipv6["addr"]
if host.mac is not None:
addresses["mac"] = host.mac["addr"]
self.set_addresses(addresses)
self.set_hostnames(host.get_hostnames())
os = host.get_best_osmatch()
if os:
os["portsused"] = host.get_ports_used()
self.set_os(os)
self.set_tcpseq(host.get_tcpsequence())
self.set_ipseq(host.get_ipidsequence())
self.set_tcptsseq(host.get_tcptssequence())
self.set_comment(host.comment)
def __init__(self, host):
HIGVBox.__init__(self)
self.__create_widgets()
self.set_os_image(get_os_logo(host))
self.set_vulnerability_image(
get_vulnerability_logo(host.get_open_ports()))
self.set_host_status({'state': host.get_state(),
'open': str(host.get_open_ports()),
'filtered': str(host.get_filtered_ports()),
'closed': str(host.get_closed_ports()),
'scanned': str(host.get_scanned_ports()),
'uptime': host.get_uptime()['seconds'],
'lastboot': host.get_uptime()['lastboot']})
addresses = {}
if host.ip is not None:
addresses['ipv4'] = host.ip['addr']
if host.ipv6 is not None:
addresses['ipv6'] = host.ipv6['addr']
if host.mac is not None:
addresses['mac'] = host.mac['addr']
self.set_addresses(addresses)
self.set_hostnames(host.get_hostnames())
os = host.get_best_osmatch()
if os:
os['portsused'] = host.get_ports_used()
self.set_os(os)
self.set_tcpseq(host.get_tcpsequence())
self.set_ipseq(host.get_ipidsequence())
self.set_tcptsseq(host.get_tcptssequence())
self.set_comment(host.comment)
class BugReport(gtk.Window, object):
def __init__(self):
gtk.Window.__init__(self)
self.set_title(_('How to Report a Bug'))
self.set_position(gtk.WIN_POS_CENTER_ALWAYS)
self._create_widgets()
self._pack_widgets()
self._connect_widgets()
def _create_widgets(self):
self.vbox = HIGVBox()
self.button_box = gtk.HButtonBox()
self.text = gtk.Label()
self.btn_ok = gtk.Button(stock=gtk.STOCK_OK)
def _pack_widgets(self):
self.vbox.set_border_width(6)
self.text.set_line_wrap(True)
self.text.set_markup(_("""\
<big><b>How to report a bug</b></big>
Like their author, %(nmap)s and %(app)s aren't perfect. But you can help \
make it better by sending bug reports or even writing patches. If \
%(nmap)s doesn't behave the way you expect, first upgrade to the latest \
version available from <b>%(nmap_web)s</b>. If the problem persists, do \
some research to determine whether it has already been discovered and \
addressed. Try Googling the error message or browsing the nmap-dev \
archives at http://seclists.org/. Read the full manual page as well. If \
nothing comes of this, mail a bug report to \
<b><[email protected]></b>. Please include everything you have \
learned about the problem, as well as what version of Nmap you are \
running and what operating system version it is running on. Problem \
reports and %(nmap)s usage questions sent to [email protected] are \
far more likely to be answered than those sent to Fyodor directly.
Code patches to fix bugs are even better than bug reports. Basic \
instructions for creating patch files with your changes are available at \
http://nmap.org/data/HACKING. Patches may be sent to nmap-dev \
(recommended) or to Fyodor directly.
""") % {"app": escape(APP_DISPLAY_NAME), "nmap": escape(NMAP_DISPLAY_NAME), "nmap_web": escape(NMAP_WEB_SITE)})
self.vbox.add(self.text)
self.button_box.set_layout(gtk.BUTTONBOX_END)
self.button_box.pack_start(self.btn_ok)
self.vbox._pack_noexpand_nofill(self.button_box)
self.add(self.vbox)
def _connect_widgets(self):
self.btn_ok.connect("clicked", self.close)
self.connect("delete-event", self.close)
def close(self, widget=None, event=None):
self.destroy()
def __create_widgets(self, inventory, scans_store):
self.open_ports_page = HIGVBox()
self.nmap_output_page = HIGVBox()
self.topology_page = HIGVBox()
self.host_details_page = HIGScrolledWindow()
self.host_details_vbox = HIGVBox()
self.scans_list_page = HIGVBox()
self.open_ports = ScanOpenPortsPage()
self.nmap_output = ScanNmapOutputPage(scans_store)
self.topology = TopologyPage(inventory)
self.scans_list = ScanScanListPage(scans_store)
self.no_selected = gtk.Label(_('No host selected.'))
self.host_details = self.no_selected
self.open_ports_page.add(self.open_ports)
self.nmap_output_page.add(self.nmap_output)
self.topology_page.add(self.topology)
self.scans_list_page.add(self.scans_list)
self.host_details_page.add_with_viewport(self.host_details_vbox)
self.host_details_vbox._pack_expand_fill(self.host_details)
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()
class DiffWindow(gtk.Window):
def __init__(self, scans):
gtk.Window.__init__(self)
self.set_title(_("Compare Results"))
self.ndiff_process = None
# We allow the user to start a new diff before the old one has
# finished. We have to keep references to old processes until they
# finish to avoid problems when tearing down the Python interpreter at
# program exit.
self.old_processes = []
self.timer_id = None
self.main_vbox = HIGVBox()
self.diff_view = DiffView()
self.diff_view.set_size_request(-1, 100)
self.hbox_buttons = HIGHBox()
self.progress = gtk.ProgressBar()
self.btn_close = HIGButton(stock=gtk.STOCK_CLOSE)
self.hbox_selection = HIGHBox()
self.scan_chooser_a = ScanChooser(scans, _(u"A Scan"))
self.scan_chooser_b = ScanChooser(scans, _(u"B Scan"))
self._pack_widgets()
self._connect_widgets()
self.set_default_size(-1, 500)
# Initial Size Request
self.initial_size = self.get_size()
def _pack_widgets(self):
self.main_vbox.set_border_width(6)
self.hbox_selection.pack_start(self.scan_chooser_a, True, True)
self.hbox_selection.pack_start(self.scan_chooser_b, True, True)
self.main_vbox.pack_start(self.hbox_selection, False)
scroll = gtk.ScrolledWindow()
scroll.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
scroll.add(self.diff_view)
self.main_vbox.pack_start(scroll, True, True)
self.progress.hide()
self.progress.set_no_show_all(True)
self.hbox_buttons.pack_start(self.progress, False)
self.hbox_buttons.pack_end(self.btn_close, False)
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_close.connect("clicked", self.close)
self.scan_chooser_a.connect('changed', self.refresh_diff)
self.scan_chooser_b.connect('changed', self.refresh_diff)
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."""
log.debug("Refresh diff.")
if (self.ndiff_process is not None and
self.ndiff_process.poll() is None):
# Put this in the list of old processes we keep track of.
self.old_processes.append(self.ndiff_process)
self.ndiff_process = None
scan_a = self.scan_chooser_a.parsed_scan
scan_b = self.scan_chooser_b.parsed_scan
if scan_a is None or scan_b is None:
self.diff_view.clear()
else:
try:
self.ndiff_process = zenmapCore.Diff.ndiff(scan_a, scan_b)
except OSError, e:
alert = HIGAlertDialog(
message_format=_("Error running ndiff"),
secondary_text=_(
"There was an error running the ndiff program.\n\n"
) + str(e).decode(sys.getdefaultencoding(), "replace"))
alert.run()
alert.destroy()
else:
class ScanHostsView(HIGVBox, object):
HOST_MODE, SERVICE_MODE = range(2)
def __init__(self, scan_interface):
HIGVBox.__init__(self)
self._scan_interface = scan_interface
self._create_widgets()
self._connect_widgets()
self._pack_widgets()
self._set_scrolled()
self._set_host_list()
self._set_service_list()
self._pack_expand_fill(self.main_vbox)
self.mode = None
# Default mode is host mode
self.host_mode(self.host_mode_button)
self.host_view.show_all()
self.service_view.show_all()
def _create_widgets(self):
# Mode buttons
self.host_mode_button = gtk.ToggleButton(_("Hosts"))
self.service_mode_button = gtk.ToggleButton(_("Services"))
self.buttons_box = gtk.HBox()
# Main window vbox
self.main_vbox = HIGVBox()
# Host list
self.host_list = gtk.ListStore(object, str, str)
self.host_list.set_sort_func(1000, cmp_treemodel_addr)
self.host_list.set_sort_column_id(1000, gtk.SORT_ASCENDING)
self.host_view = gtk.TreeView(self.host_list)
self.pic_column = gtk.TreeViewColumn(_('OS'))
self.host_column = gtk.TreeViewColumn(_('Host'))
self.os_cell = gtk.CellRendererPixbuf()
self.host_cell = gtk.CellRendererText()
# Service list
self.service_list = gtk.ListStore(str)
self.service_list.set_sort_column_id(0, gtk.SORT_ASCENDING)
self.service_view = gtk.TreeView(self.service_list)
self.service_column = gtk.TreeViewColumn(_('Service'))
self.service_cell = gtk.CellRendererText()
self.scrolled = gtk.ScrolledWindow()
def _pack_widgets(self):
self.main_vbox.set_spacing(0)
self.main_vbox.set_border_width(0)
self.main_vbox._pack_noexpand_nofill(self.buttons_box)
self.main_vbox._pack_expand_fill(self.scrolled)
self.host_mode_button.set_active(True)
self.buttons_box.set_border_width(5)
self.buttons_box.pack_start(self.host_mode_button)
self.buttons_box.pack_start(self.service_mode_button)
def _connect_widgets(self):
self.host_mode_button.connect("toggled", self.host_mode)
self.service_mode_button.connect("toggled", self.service_mode)
def host_mode(self, widget):
self._remove_scrolled_child()
if widget.get_active():
self.mode = self.HOST_MODE
self.service_mode_button.set_active(False)
self.scrolled.add(self.host_view)
else:
self.service_mode_button.set_active(True)
def service_mode(self, widget):
self._remove_scrolled_child()
if widget.get_active():
self.mode = self.SERVICE_MODE
self.host_mode_button.set_active(False)
self.scrolled.add(self.service_view)
else:
self.host_mode_button.set_active(True)
def _remove_scrolled_child(self):
try:
child = self.scrolled.get_child()
self.scrolled.remove(child)
except:
pass
def _set_scrolled(self):
self.scrolled.set_border_width(5)
self.scrolled.set_size_request(150, -1)
self.scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
def _set_service_list(self):
self.service_view.set_enable_search(True)
self.service_view.set_search_column(0)
selection = self.service_view.get_selection()
selection.set_mode(gtk.SELECTION_MULTIPLE)
self.service_view.append_column(self.service_column)
self.service_column.set_resizable(True)
self.service_column.set_sort_column_id(0)
self.service_column.set_reorderable(True)
self.service_column.pack_start(self.service_cell, True)
self.service_column.set_attributes(self.service_cell, text=0)
def _set_host_list(self):
self.host_view.set_enable_search(True)
self.host_view.set_search_column(1)
selection = self.host_view.get_selection()
selection.set_mode(gtk.SELECTION_MULTIPLE)
self.host_view.append_column(self.pic_column)
self.host_view.append_column(self.host_column)
self.host_column.set_resizable(True)
self.pic_column.set_resizable(True)
self.host_column.set_sort_column_id(1000)
self.pic_column.set_sort_column_id(1)
self.host_column.set_reorderable(True)
self.pic_column.set_reorderable(True)
self.pic_column.pack_start(self.os_cell, True)
self.host_column.pack_start(self.host_cell, True)
self.pic_column.set_min_width(35)
self.pic_column.set_attributes(self.os_cell, stock_id = 1)
self.host_column.set_attributes(self.host_cell, text = 2)
def mass_update(self, hosts):
"""Update the internal ListStores to reflect the hosts and services
passed in. Hosts that have not changed are left alone."""
hosts = set(hosts)
services = set()
for h in hosts:
services.update([s["service_name"] for s in h.services])
# Disable sorting while elements are added. See the PyGTK FAQ 13.43,
# "Are there tips for improving performance when adding many rows to a
# Treeview?"
sort_column_id = self.host_list.get_sort_column_id()
self.host_list.set_default_sort_func(lambda *args: -1)
self.host_list.set_sort_column_id(-1, gtk.SORT_ASCENDING)
self.host_view.freeze_child_notify()
self.host_view.set_model(None)
it = self.host_list.get_iter_first()
# Remove any of our ListStore hosts that aren't in the list passed in.
while it:
host = self.host_list.get_value(it, 0)
if host in hosts:
hosts.remove(host)
self.host_list.set(it, 1, get_os_icon(host))
it = self.host_list.iter_next(it)
else:
if not self.host_list.remove(it):
it = None
# Add any remaining hosts into our ListStore.
for host in hosts:
self.add_host(host)
# Reenable sorting.
if sort_column_id != (None, None):
self.host_list.set_sort_column_id(*sort_column_id)
self.host_view.set_model(self.host_list)
self.host_view.thaw_child_notify()
it = self.service_list.get_iter_first()
# Remove any of our ListStore services that aren't in the list passed
# in.
while it:
service_name = self.service_list.get_value(it, 0)
if service_name in services:
services.remove(service_name)
it = self.service_list.iter_next(it)
else:
if not self.service_list.remove(it):
it = None
# Add any remaining services into our ListStore.
for service_name in services:
self.add_service(service_name)
def add_host(self, host):
self.host_list.append([host, get_os_icon(host), host.get_hostname()])
def add_service(self, service):
self.service_list.append([service])
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 BugReport(gtk.Window, object):
def __init__(self):
gtk.Window.__init__(self)
self.set_title(_('How to Report a Bug'))
self.set_position(gtk.WIN_POS_CENTER_ALWAYS)
self._create_widgets()
self._pack_widgets()
self._connect_widgets()
def _create_widgets(self):
self.vbox = HIGVBox()
self.button_box = gtk.HButtonBox()
self.text = gtk.Label()
self.btn_ok = gtk.Button(stock=gtk.STOCK_OK)
def _pack_widgets(self):
self.vbox.set_border_width(6)
self.text.set_line_wrap(True)
self.text.set_markup(
_("""\
<big><b>How to report a bug</b></big>
Like their author, %(nmap)s and %(app)s aren't perfect. But you can help \
make it better by sending bug reports or even writing patches. If \
%(nmap)s doesn't behave the way you expect, first upgrade to the latest \
version available from <b>%(nmap_web)s</b>. If the problem persists, do \
some research to determine whether it has already been discovered and \
addressed. Try Googling the error message or browsing the nmap-dev \
archives at http://seclists.org/. Read the full manual page as well. If \
nothing comes of this, mail a bug report to \
<b><[email protected]></b>. Please include everything you have \
learned about the problem, as well as what version of Nmap you are \
running and what operating system version it is running on. Problem \
reports and %(nmap)s usage questions sent to [email protected] are \
far more likely to be answered than those sent to Fyodor directly.
Code patches to fix bugs are even better than bug reports. Basic \
instructions for creating patch files with your changes are available at \
http://nmap.org/data/HACKING. Patches may be sent to nmap-dev \
(recommended) or to Fyodor directly.
""") % {
"app": escape(APP_DISPLAY_NAME),
"nmap": escape(NMAP_DISPLAY_NAME),
"nmap_web": escape(NMAP_WEB_SITE)
})
self.vbox.add(self.text)
self.button_box.set_layout(gtk.BUTTONBOX_END)
self.button_box.pack_start(self.btn_ok)
self.vbox._pack_noexpand_nofill(self.button_box)
self.add(self.vbox)
def _connect_widgets(self):
self.btn_ok.connect("clicked", self.close)
self.connect("delete-event", self.close)
def close(self, widget=None, event=None):
self.destroy()
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 TopologyPage(HIGVBox):
def __init__(self, inventory):
HIGVBox.__init__(self)
self.set_border_width(6)
self.set_spacing(4)
self.network_inventory = inventory
self._create_widgets()
self._pack_widgets()
def _create_widgets(self):
self.rn_hbox = gtk.HBox()
self.rn_hbox.set_spacing(4)
self.rn_vbox = gtk.VBox()
# RadialNet's widgets
self.radialnet = RadialNet(LAYOUT_WEIGHTED)
self.control = ControlWidget(self.radialnet)
self.fisheye = ControlFisheye(self.radialnet)
self.rn_toolbar = Toolbar(self.radialnet,
self,
self.control,
self.fisheye)
self.display_panel = HIGVBox()
self.radialnet.set_no_show_all(True)
self.slow_vbox = HIGVBox()
self.slow_label = gtk.Label()
self.slow_vbox.pack_start(self.slow_label, False, False)
show_button = gtk.Button(_("Show the topology anyway"))
show_button.connect("clicked", self.show_anyway)
self.slow_vbox.pack_start(show_button, False, False)
self.slow_vbox.show_all()
self.slow_vbox.set_no_show_all(True)
self.slow_vbox.hide()
self.radialnet.show()
def _pack_widgets(self):
self.rn_hbox.pack_start(self.display_panel, True, True)
self.rn_hbox.pack_start(self.control, False)
self.rn_vbox.pack_start(self.rn_hbox, True, True)
self.rn_vbox.pack_start(self.fisheye, False)
self.pack_start(self.rn_toolbar, False, False)
self.pack_start(self.rn_vbox, True, True)
self.display_panel.pack_start(self.slow_vbox, True, False)
self.display_panel.pack_start(self.radialnet, True, True)
def add_scan(self, scan):
"""Parses a given XML file and adds the parsed result to the network inventory."""
self.network_inventory.add_scan(scan)
self.update_radialnet()
def update_radialnet(self):
"""Creates a graph from network inventory's host list and displays it."""
hosts_up = self.network_inventory.get_hosts_up()
self.slow_label.set_text(_("""\
Topology is disabled because too many hosts can cause it
to run slowly. The limit is %d hosts and there are %d.\
""" % (SLOW_LIMIT, len(hosts_up))))
if len(hosts_up) <= SLOW_LIMIT:
self.radialnet.show()
self.slow_vbox.hide()
self.update_radialnet_unchecked()
else:
self.radialnet.hide()
self.slow_vbox.show()
def update_radialnet_unchecked(self):
hosts_up = self.network_inventory.get_hosts_up()
graph = make_graph_from_hosts(hosts_up)
self.radialnet.set_empty()
self.radialnet.set_graph(graph)
self.radialnet.show()
def show_anyway(self, widget):
self.radialnet.show()
self.slow_vbox.hide()
self.update_radialnet_unchecked()
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 ScanHostsView(HIGVBox, object):
def __init__(self, hosts={}, services={}):
HIGVBox.__init__(self)
self._create_widgets()
self._connect_widgets()
self._pack_widgets()
self._set_scrolled()
self._set_host_list(hosts)
self._set_service_list(services)
self._pack_expand_fill(self.main_vbox)
# Default mode is host mode
self.host_mode(self.host_mode_button)
self.host_view.show_all()
self.service_view.show_all()
def _create_widgets(self):
# Mode buttons
self.host_mode_button = gtk.ToggleButton(_("Hosts"))
self.service_mode_button = gtk.ToggleButton(_("Services"))
self.buttons_box = gtk.HBox()
# Main window vbox
self.main_vbox = HIGVBox()
# Host list
self.host_list = gtk.ListStore(str, str)
self.host_view = gtk.TreeView(self.host_list)
self.pic_column = gtk.TreeViewColumn(_('OS'))
self.host_column = gtk.TreeViewColumn(_('Host'))
self.os_cell = gtk.CellRendererPixbuf()
self.host_cell = gtk.CellRendererText()
# Service list
self.service_list = gtk.ListStore(str)
self.service_view = gtk.TreeView(self.service_list)
self.service_column = gtk.TreeViewColumn(_('Service'))
self.service_cell = gtk.CellRendererText()
self.scrolled = gtk.ScrolledWindow()
def _pack_widgets(self):
self.main_vbox.set_spacing(0)
self.main_vbox.set_border_width(0)
self.main_vbox._pack_noexpand_nofill(self.buttons_box)
self.main_vbox._pack_expand_fill(self.scrolled)
self.host_mode_button.set_active(True)
self.buttons_box.set_border_width(5)
self.buttons_box.pack_start(self.host_mode_button)
self.buttons_box.pack_start(self.service_mode_button)
def _connect_widgets(self):
self.host_mode_button.connect("toggled", self.host_mode)
self.service_mode_button.connect("toggled", self.service_mode)
def host_mode(self, widget):
self._remove_scrolled_child()
if widget.get_active():
self.service_mode_button.set_active(False)
self.scrolled.add(self.host_view)
else:
self.service_mode_button.set_active(True)
def service_mode(self, widget):
self._remove_scrolled_child()
if widget.get_active():
self.host_mode_button.set_active(False)
self.scrolled.add(self.service_view)
else:
self.host_mode_button.set_active(True)
def _remove_scrolled_child(self):
try:
child = self.scrolled.get_child()
self.scrolled.remove(child)
except:
pass
def _set_scrolled(self):
self.scrolled.set_border_width(5)
self.scrolled.set_size_request(150, -1)
self.scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
def _set_service_list(self, services):
self.service_view.set_enable_search(True)
self.service_view.set_search_column(0)
selection = self.service_view.get_selection()
selection.set_mode(gtk.SELECTION_MULTIPLE)
self.service_view.append_column(self.service_column)
self.service_column.set_resizable(True)
self.service_column.set_sort_column_id(0)
self.service_column.set_reorderable(True)
self.service_column.pack_start(self.service_cell, True)
self.service_column.set_attributes(self.service_cell, text=0)
self.set_services(services)
def _set_host_list(self, hosts):
self.host_view.set_enable_search(True)
self.host_view.set_search_column(1)
selection = self.host_view.get_selection()
selection.set_mode(gtk.SELECTION_MULTIPLE)
self.host_view.append_column(self.pic_column)
self.host_view.append_column(self.host_column)
self.host_column.set_resizable(True)
self.pic_column.set_resizable(True)
self.host_column.set_sort_column_id(1)
self.pic_column.set_sort_column_id(1)
self.host_column.set_reorderable(True)
self.pic_column.set_reorderable(True)
self.pic_column.pack_start(self.os_cell, True)
self.host_column.pack_start(self.host_cell, True)
self.pic_column.set_attributes(self.os_cell, stock_id=0)
self.pic_column.set_min_width(35)
self.host_column.set_attributes(self.host_cell, text=1)
self.set_hosts(hosts)
def clear_host_list(self):
for i in range(len(self.host_list)):
iter = self.host_list.get_iter_root()
del (self.host_list[iter])
def clear_service_list(self):
for i in range(len(self.service_list)):
iter = self.service_list.get_iter_root()
del (self.service_list[iter])
def set_hosts(self, hosts):
self.hosts = hosts
self.clear_host_list()
for host in hosts:
self.host_list.append([hosts[host]['stock'], host])
def set_services(self, services):
self.services = services
self.clear_service_list()
for service in services:
self.service_list.append([service])
def add_host(self, host):
for h in host:
self.host_list.append([host[h]['stock'], h])
def add_service(self, service):
if type(service) == type([]):
for s in service:
self.service_list.append([s])
elif type(service) in StringTypes:
self.service_list.append([service])
def get_action(self, host):
try:
return self.hosts[host]['action']
except:
return False
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 ScanInterface(HIGVBox):
"""ScanInterface contains the scan toolbar and the scan results. Each
ScanInterface represents a single NetworkInventory as well as a set of
running scans."""
# The time delay between when you stop typing a filter string and filtering
# actually begins, in milliseconds.
FILTER_DELAY = 1000
def __init__(self):
HIGVBox.__init__(self)
# The borders are consuming too much space on Maemo. Setting it to
# 0 pixels while on Maemo
if is_maemo():
self.set_border_width(0)
self.set_spacing(0)
# True if nothing has happened here page yet, i.e., it's okay to load a
# scan from a file here.
self.empty = True
# The most recent name the inventory on this page has been saved under.
self.saved_filename = None
# The network inventory shown by this page. It may consist of multiple
# scans.
self.inventory = FilteredNetworkInventory()
# The list of currently running scans (NmapCommand objects).
self.jobs = []
# The list of running and finished scans shown on the Nmap Output page.
self.scans_store = ScansListStore()
self.top_box = HIGVBox()
self.__create_toolbar()
self.__create_command_toolbar()
self.select_default_profile()
self.scan_result = ScanResult(self.inventory, self.scans_store,
scan_interface=self)
self.host_view_selection = self.scan_result.get_host_selection()
self.service_view_selection = self.scan_result.get_service_selection()
self.host_view_selection.connect(
'changed', self.host_selection_changed)
self.service_view_selection.connect(
'changed', self.service_selection_changed)
host_page = self.scan_result.scan_result_notebook.open_ports.host
host_page.host_view.get_selection().connect(
'changed', self.service_host_selection_changed)
self.host_view_selection.connect(
'changed', self.host_selection_changed)
self.scan_result.scan_result_notebook.nmap_output.connect(
"changed", self._displayed_scan_change_cb)
self.scan_result.scan_result_notebook.scans_list.remove_button.connect(
"clicked", self._remove_scan_cb)
# The hosts dict maps hostnames (as returned by HostInfo.get_hostname)
# to HostInfo objects.
self.hosts = {}
# The services dict maps service names ("http") to lists of dicts of
# the form
# {'host': <HostInfo object>, 'hostname': u'example.com',
# 'port_state': u'open', 'portid': u'22', 'protocol': u'tcp',
# 'service_conf': u'10', 'service_extrainfo': u'protocol 2.0',
# 'service_method': u'probed', 'service_name': u'ssh',
# 'service_product': u'OpenSSH', 'service_version': u'4.3'}
# In other words each dict has the same keys as an entry in
# HostInfo.ports, with the addition of "host" and "hostname" keys.
self.services = {}
self.top_box.set_border_width(6)
self.top_box.set_spacing(5)
self.top_box._pack_noexpand_nofill(self.toolbar)
self.top_box._pack_noexpand_nofill(self.command_toolbar)
self._pack_noexpand_nofill(self.top_box)
self._pack_expand_fill(self.scan_result)
self.scan_result.scan_result_notebook.scans_list.cancel_button.connect(
"clicked", self._cancel_scans_list_cb)
self.update_cancel_button()
# Create the filter GUI
self.filter_bar = FilterBar()
self.pack_start(self.filter_bar, False, True, 0)
self.filter_bar.set_no_show_all(True)
self.filter_timeout_id = None
self.filter_bar.connect("changed", self.filter_changed)
self.scan_result.filter_toggle_button.connect("toggled",
self.filter_toggle_toggled)
self.scan_result.filter_toggle_button.show()
def toggle_filter_bar(self):
self.scan_result.filter_toggle_button.clicked()
def filter_toggle_toggled(self, widget):
if self.scan_result.filter_toggle_button.get_active():
# Show the filter bar
self.filter_bar.show()
self.filter_bar.grab_focus()
self.filter_hosts(self.filter_bar.get_filter_string())
else:
# Hide the filter bar
self.filter_bar.hide()
self.filter_hosts("")
self.update_ui()
def filter_changed(self, filter_bar):
# Restart the timer to start the filter.
if self.filter_timeout_id:
gobject.source_remove(self.filter_timeout_id)
self.filter_timeout_id = gobject.timeout_add(
self.FILTER_DELAY, self.filter_hosts,
filter_bar.get_filter_string())
def filter_hosts(self, filter_string):
start = time.clock()
self.inventory.apply_filter(filter_string)
filter_time = time.clock() - start
# Update the gui
start = time.clock()
self.update_ui()
gui_time = time.clock() - start
if filter_time + gui_time > 0.0:
log.debug("apply_filter %g ms update_ui %g ms (%.0f%% filter)" %
(filter_time * 1000.0, gui_time * 1000.0,
100.0 * filter_time / (filter_time + gui_time)))
self.filter_timeout_id = None
return False
def is_changed(self):
"""Return true if this window has unsaved changes."""
for scan in self.inventory.get_scans():
if scan.unsaved:
return True
return False
changed = property(is_changed)
def num_scans_running(self):
return len(self.jobs)
def select_default_profile(self):
"""Select a "default" profile. Currently this is defined to be the
first profile."""
if len(self.toolbar.profile_entry.get_model()) > 0:
self.toolbar.profile_entry.set_active(0)
def go_to_host(self, hostname):
"""Scroll the text output to the appearance of the named host."""
self.scan_result.scan_result_notebook.nmap_output.nmap_output.go_to_host(hostname) # noqa
def __create_toolbar(self):
self.toolbar = ScanToolbar()
self.target_entry_changed_handler = self.toolbar.target_entry.connect(
'changed', self._target_entry_changed)
self.profile_entry_changed_handler = \
self.toolbar.profile_entry.connect(
'changed', self._profile_entry_changed)
self.toolbar.scan_button.connect('clicked', self.start_scan_cb)
self.toolbar.cancel_button.connect('clicked', self._cancel_scan_cb)
def __create_command_toolbar(self):
self.command_toolbar = ScanCommandToolbar()
self.command_toolbar.command_entry.connect(
'activate', lambda x: self.toolbar.scan_button.clicked())
self.command_entry_changed_handler = \
self.command_toolbar.command_entry.connect(
'changed', self._command_entry_changed)
def _command_entry_changed(self, editable):
ops = NmapOptions()
ops.parse_string(self.command_toolbar.get_command())
# Set the target and profile without propagating the "changed" signal
# back to the command entry.
self.set_target_quiet(join_quoted(ops.target_specs))
self.set_profile_name_quiet("")
def _target_entry_changed(self, editable):
target_string = self.toolbar.get_selected_target()
targets = split_quoted(target_string)
ops = NmapOptions()
ops.parse_string(self.command_toolbar.get_command())
ops.target_specs = targets
self.set_command_quiet(ops.render_string())
def _profile_entry_changed(self, widget):
"""Update the command based on the contents of the target and profile
entries. If the command corresponding to the current profile is not
blank, use it. Otherwise use the current contents of the command
entry."""
profile_name = self.toolbar.get_selected_profile()
target_string = self.toolbar.get_selected_target()
cmd_profile = CommandProfile()
command_string = cmd_profile.get_command(profile_name)
del(cmd_profile)
if command_string == "":
command_string = self.command_toolbar.get_command()
ops = NmapOptions()
ops.parse_string(command_string)
# Use the targets from the command entry, if there are any, otherwise
# use any targets from the profile.
targets = split_quoted(target_string)
if len(targets) > 0:
ops.target_specs = targets
else:
self.toolbar.set_selected_target(join_quoted(ops.target_specs))
self.set_command_quiet(ops.render_string())
def set_command_quiet(self, command_string):
"""Set the command used by this scan interface, ignoring any further
"changed" signals."""
self.command_toolbar.command_entry.handler_block(
self.command_entry_changed_handler)
self.command_toolbar.set_command(command_string)
self.command_toolbar.command_entry.handler_unblock(
self.command_entry_changed_handler)
def set_target_quiet(self, target_string):
"""Set the target string used by this scan interface, ignoring any
further "changed" signals."""
self.toolbar.target_entry.handler_block(
self.target_entry_changed_handler)
self.toolbar.set_selected_target(target_string)
self.toolbar.target_entry.handler_unblock(
self.target_entry_changed_handler)
def set_profile_name_quiet(self, profile_name):
"""Set the profile name used by this scan interface, ignoring any
further "changed" signals."""
self.toolbar.profile_entry.handler_block(
self.profile_entry_changed_handler)
self.toolbar.set_selected_profile(profile_name)
self.toolbar.profile_entry.handler_unblock(
self.profile_entry_changed_handler)
def start_scan_cb(self, widget=None):
target = self.toolbar.selected_target
command = self.command_toolbar.command
profile = self.toolbar.selected_profile
log.debug(">>> Start Scan:")
log.debug(">>> Target: '%s'" % target)
log.debug(">>> Profile: '%s'" % profile)
log.debug(">>> Command: '%s'" % command)
if target != '':
try:
self.toolbar.add_new_target(target)
except IOError, e:
# We failed to save target_list.txt; treat it as read-only.
# Probably it's owned by root and this is a normal user.
log.debug(">>> Error saving %s: %s" % (
Path.target_list, str(e)))
if command == '':
warn_dialog = HIGAlertDialog(
message_format=_("Empty Nmap Command"),
secondary_text=_("There is no command to execute. "
"Maybe the selected/typed profile doesn't exist. "
"Please check the profile name or type the nmap "
"command you would like to execute."),
type=gtk.MESSAGE_ERROR)
warn_dialog.run()
warn_dialog.destroy()
return
self.execute_command(command, target, profile)
class DiffWindow(gtk.Window):
def __init__(self, scans):
gtk.Window.__init__(self)
self.set_title(_("Compare Results"))
self.ndiff_process = None
# We allow the user to start a new diff before the old one has
# finished. We have to keep references to old processes until they
# finish to avoid problems when tearing down the Python interpreter at
# program exit.
self.old_processes = []
self.timer_id = None
self.main_vbox = HIGVBox()
self.diff_view = DiffView()
self.diff_view.set_size_request(-1, 100)
self.hbox_buttons = HIGHBox()
self.progress = gtk.ProgressBar()
self.btn_close = HIGButton(stock=gtk.STOCK_CLOSE)
self.hbox_selection = HIGHBox()
self.scan_chooser_a = ScanChooser(scans, _(u"A Scan"))
self.scan_chooser_b = ScanChooser(scans, _(u"B Scan"))
self._pack_widgets()
self._connect_widgets()
self.set_default_size(-1, 500)
# Initial Size Request
self.initial_size = self.get_size()
def _pack_widgets(self):
self.main_vbox.set_border_width(6)
self.hbox_selection.pack_start(self.scan_chooser_a, True, True)
self.hbox_selection.pack_start(self.scan_chooser_b, True, True)
self.main_vbox.pack_start(self.hbox_selection, False)
scroll = gtk.ScrolledWindow()
scroll.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
scroll.add(self.diff_view)
self.main_vbox.pack_start(scroll, True, True)
self.progress.hide()
self.progress.set_no_show_all(True)
self.hbox_buttons.pack_start(self.progress, False)
self.hbox_buttons.pack_end(self.btn_close, False)
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_close.connect("clicked", self.close)
self.scan_chooser_a.connect('changed', self.refresh_diff)
self.scan_chooser_b.connect('changed', self.refresh_diff)
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."""
log.debug("Refresh diff.")
if (self.ndiff_process is not None
and self.ndiff_process.poll() is None):
# Put this in the list of old processes we keep track of.
self.old_processes.append(self.ndiff_process)
self.ndiff_process = None
scan_a = self.scan_chooser_a.parsed_scan
scan_b = self.scan_chooser_b.parsed_scan
if scan_a is None or scan_b is None:
self.diff_view.clear()
else:
try:
self.ndiff_process = zenmapCore.Diff.ndiff(scan_a, scan_b)
except OSError as e:
alert = HIGAlertDialog(
message_format=_("Error running ndiff"),
secondary_text=_(
"There was an error running the ndiff program.\n\n") +
str(e).decode(sys.getdefaultencoding(), "replace"))
alert.run()
alert.destroy()
else:
self.progress.show()
if self.timer_id is None:
self.timer_id = gobject.timeout_add(
NDIFF_CHECK_TIMEOUT, self.check_ndiff_process)
def check_ndiff_process(self):
"""Check if the ndiff subprocess is done and show the diff if it is.
Also remove any finished processes from the old process list."""
# Check if any old background processes have finished.
for p in self.old_processes[:]:
if p.poll() is not None:
p.close()
self.old_processes.remove(p)
if self.ndiff_process is not None:
# We're running the most recent scan. Check if it's done.
status = self.ndiff_process.poll()
if status is None:
# Keep calling this function on a timer until the process
# finishes.
self.progress.pulse()
return True
if status == 0 or status == 1:
# Successful completion.
try:
diff = self.ndiff_process.get_scan_diff()
except zenmapCore.Diff.NdiffParseException as e:
alert = HIGAlertDialog(
message_format=_("Error parsing ndiff output"),
secondary_text=str(e))
alert.run()
alert.destroy()
else:
self.diff_view.show_diff(diff)
else:
# Unsuccessful completion.
error_text = _(
"The ndiff process terminated with status code %d."
) % status
stderr = self.ndiff_process.stderr.read()
if len(stderr) > 0:
error_text += "\n\n" + stderr
alert = HIGAlertDialog(message_format=_("Error running ndiff"),
secondary_text=error_text)
alert.run()
alert.destroy()
self.progress.hide()
self.ndiff_process.close()
self.ndiff_process = None
if len(self.old_processes) > 0:
# Keep calling this callback.
return True
else:
# All done.
self.timer_id = None
return False
def close(self, widget=None, extra=None):
self.destroy()
class TopologyPage(HIGVBox):
def __init__(self, inventory):
HIGVBox.__init__(self)
self.set_border_width(6)
self.set_spacing(4)
self.network_inventory = inventory
self._create_widgets()
self._pack_widgets()
def _create_widgets(self):
self.rn_hbox = gtk.HBox()
self.rn_hbox.set_spacing(4)
self.rn_vbox = gtk.VBox()
# RadialNet's widgets
self.radialnet = RadialNet(LAYOUT_WEIGHTED)
self.control = ControlWidget(self.radialnet)
self.fisheye = ControlFisheye(self.radialnet)
self.rn_toolbar = Toolbar(self.radialnet, self, self.control,
self.fisheye)
self.display_panel = HIGVBox()
self.radialnet.set_no_show_all(True)
self.slow_vbox = HIGVBox()
self.slow_label = gtk.Label()
self.slow_vbox.pack_start(self.slow_label, False, False)
show_button = gtk.Button(_("Show the topology anyway"))
show_button.connect("clicked", self.show_anyway)
self.slow_vbox.pack_start(show_button, False, False)
self.slow_vbox.show_all()
self.slow_vbox.set_no_show_all(True)
self.slow_vbox.hide()
self.radialnet.show()
def _pack_widgets(self):
self.rn_hbox.pack_start(self.display_panel, True, True)
self.rn_hbox.pack_start(self.control, False)
self.rn_vbox.pack_start(self.rn_hbox, True, True)
self.rn_vbox.pack_start(self.fisheye, False)
self.pack_start(self.rn_toolbar, False, False)
self.pack_start(self.rn_vbox, True, True)
self.display_panel.pack_start(self.slow_vbox, True, False)
self.display_panel.pack_start(self.radialnet, True, True)
def add_scan(self, scan):
"""Parses a given XML file and adds the parsed result to the network
inventory."""
self.network_inventory.add_scan(scan)
self.update_radialnet()
def update_radialnet(self):
"""Creates a graph from network inventory's host list and displays
it."""
hosts_up = self.network_inventory.get_hosts_up()
self.slow_label.set_text(
_("""\
Topology is disabled because too many hosts can cause it
to run slowly. The limit is %d hosts and there are %d.\
""" % (SLOW_LIMIT, len(hosts_up))))
if len(hosts_up) <= SLOW_LIMIT:
self.radialnet.show()
self.slow_vbox.hide()
self.update_radialnet_unchecked()
else:
self.radialnet.hide()
self.slow_vbox.show()
def update_radialnet_unchecked(self):
hosts_up = self.network_inventory.get_hosts_up()
graph = make_graph_from_hosts(hosts_up)
self.radialnet.set_empty()
self.radialnet.set_graph(graph)
self.radialnet.show()
def show_anyway(self, widget):
self.radialnet.show()
self.slow_vbox.hide()
self.update_radialnet_unchecked()
class ScanInterface(HIGVBox):
"""ScanInterface contains the scan toolbar and the scan results. Each
ScanInterface represents a single NetworkInventory as well as a set of
running scans."""
# The time delay between when you stop typing a filter string and filtering
# actually begins, in milliseconds.
FILTER_DELAY = 1000
def __init__(self):
HIGVBox.__init__(self)
# The borders are consuming too much space on Maemo. Setting it to
# 0 pixels while on Maemo
if is_maemo():
self.set_border_width(0)
self.set_spacing(0)
# True if nothing has happened here page yet, i.e., it's okay to load a
# scan from a file here.
self.empty = True
# The most recent name the inventory on this page has been saved under.
self.saved_filename = None
# The network inventory shown by this page. It may consist of multiple
# scans.
self.inventory = FilteredNetworkInventory()
# The list of currently running scans (NmapCommand objects).
self.jobs = []
# The list of running and finished scans shown on the Nmap Output page.
self.scans_store = ScansListStore()
self.top_box = HIGVBox()
self.__create_toolbar()
self.__create_command_toolbar()
self.select_default_profile()
self.scan_result = ScanResult(self.inventory,
self.scans_store,
scan_interface=self)
self.host_view_selection = self.scan_result.get_host_selection()
self.service_view_selection = self.scan_result.get_service_selection()
self.host_view_selection.connect('changed',
self.host_selection_changed)
self.service_view_selection.connect('changed',
self.service_selection_changed)
host_page = self.scan_result.scan_result_notebook.open_ports.host
host_page.host_view.get_selection().connect(
'changed', self.service_host_selection_changed)
self.host_view_selection.connect('changed',
self.host_selection_changed)
self.scan_result.scan_result_notebook.nmap_output.connect(
"changed", self._displayed_scan_change_cb)
self.scan_result.scan_result_notebook.scans_list.remove_button.connect(
"clicked", self._remove_scan_cb)
# The hosts dict maps hostnames (as returned by HostInfo.get_hostname)
# to HostInfo objects.
self.hosts = {}
# The services dict maps service names ("http") to lists of dicts of
# the form
# {'host': <HostInfo object>, 'hostname': u'example.com',
# 'port_state': u'open', 'portid': u'22', 'protocol': u'tcp',
# 'service_conf': u'10', 'service_extrainfo': u'protocol 2.0',
# 'service_method': u'probed', 'service_name': u'ssh',
# 'service_product': u'OpenSSH', 'service_version': u'4.3'}
# In other words each dict has the same keys as an entry in
# HostInfo.ports, with the addition of "host" and "hostname" keys.
self.services = {}
self.top_box.set_border_width(6)
self.top_box.set_spacing(5)
self.top_box._pack_noexpand_nofill(self.toolbar)
self.top_box._pack_noexpand_nofill(self.command_toolbar)
self._pack_noexpand_nofill(self.top_box)
self._pack_expand_fill(self.scan_result)
self.scan_result.scan_result_notebook.scans_list.cancel_button.connect(
"clicked", self._cancel_scans_list_cb)
self.update_cancel_button()
# Create the filter GUI
self.filter_bar = FilterBar()
self.pack_start(self.filter_bar, False, True, 0)
self.filter_bar.set_no_show_all(True)
self.filter_timeout_id = None
self.filter_bar.connect("changed", self.filter_changed)
self.scan_result.filter_toggle_button.connect(
"toggled", self.filter_toggle_toggled)
self.scan_result.filter_toggle_button.show()
def toggle_filter_bar(self):
self.scan_result.filter_toggle_button.clicked()
def filter_toggle_toggled(self, widget):
if self.scan_result.filter_toggle_button.get_active():
# Show the filter bar
self.filter_bar.show()
self.filter_bar.grab_focus()
self.filter_hosts(self.filter_bar.get_filter_string())
else:
# Hide the filter bar
self.filter_bar.hide()
self.filter_hosts("")
self.update_ui()
def filter_changed(self, filter_bar):
# Restart the timer to start the filter.
if self.filter_timeout_id:
gobject.source_remove(self.filter_timeout_id)
self.filter_timeout_id = gobject.timeout_add(
self.FILTER_DELAY, self.filter_hosts,
filter_bar.get_filter_string())
def filter_hosts(self, filter_string):
start = time.clock()
self.inventory.apply_filter(filter_string)
filter_time = time.clock() - start
# Update the gui
start = time.clock()
self.update_ui()
gui_time = time.clock() - start
if filter_time + gui_time > 0.0:
log.debug(
"apply_filter %g ms update_ui %g ms (%.0f%% filter)" %
(filter_time * 1000.0, gui_time * 1000.0, 100.0 * filter_time /
(filter_time + gui_time)))
self.filter_timeout_id = None
return False
def is_changed(self):
"""Return true if this window has unsaved changes."""
for scan in self.inventory.get_scans():
if scan.unsaved:
return True
return False
changed = property(is_changed)
def num_scans_running(self):
return len(self.jobs)
def select_default_profile(self):
"""Select a "default" profile. Currently this is defined to be the
first profile."""
if len(self.toolbar.profile_entry.get_model()) > 0:
self.toolbar.profile_entry.set_active(0)
def go_to_host(self, hostname):
"""Scroll the text output to the appearance of the named host."""
self.scan_result.scan_result_notebook.nmap_output.nmap_output.go_to_host(
hostname)
def __create_toolbar(self):
self.toolbar = ScanToolbar()
self.target_entry_changed_handler = self.toolbar.target_entry.connect(
'changed', self._target_entry_changed)
self.profile_entry_changed_handler = \
self.toolbar.profile_entry.connect(
'changed', self._profile_entry_changed)
self.toolbar.scan_button.connect('clicked', self.start_scan_cb)
self.toolbar.cancel_button.connect('clicked', self._cancel_scan_cb)
def __create_command_toolbar(self):
self.command_toolbar = ScanCommandToolbar()
self.command_toolbar.command_entry.connect(
'activate', lambda x: self.toolbar.scan_button.clicked())
self.command_entry_changed_handler = \
self.command_toolbar.command_entry.connect(
'changed', self._command_entry_changed)
def _command_entry_changed(self, editable):
ops = NmapOptions()
ops.parse_string(self.command_toolbar.get_command())
# Set the target and profile without propagating the "changed" signal
# back to the command entry.
self.set_target_quiet(join_quoted(ops.target_specs))
self.set_profile_name_quiet("")
def _target_entry_changed(self, editable):
target_string = self.toolbar.get_selected_target()
targets = split_quoted(target_string)
ops = NmapOptions()
ops.parse_string(self.command_toolbar.get_command())
ops.target_specs = targets
self.set_command_quiet(ops.render_string())
def _profile_entry_changed(self, widget):
"""Update the command based on the contents of the target and profile
entries. If the command corresponding to the current profile is not
blank, use it. Otherwise use the current contents of the command
entry."""
profile_name = self.toolbar.get_selected_profile()
target_string = self.toolbar.get_selected_target()
cmd_profile = CommandProfile()
command_string = cmd_profile.get_command(profile_name)
del (cmd_profile)
if command_string == "":
command_string = self.command_toolbar.get_command()
ops = NmapOptions()
ops.parse_string(command_string)
# Use the targets from the command entry, if there are any, otherwise
# use any targets from the profile.
targets = split_quoted(target_string)
if len(targets) > 0:
ops.target_specs = targets
else:
self.toolbar.set_selected_target(join_quoted(ops.target_specs))
self.set_command_quiet(ops.render_string())
def set_command_quiet(self, command_string):
"""Set the command used by this scan interface, ignoring any further
"changed" signals."""
self.command_toolbar.command_entry.handler_block(
self.command_entry_changed_handler)
self.command_toolbar.set_command(command_string)
self.command_toolbar.command_entry.handler_unblock(
self.command_entry_changed_handler)
def set_target_quiet(self, target_string):
"""Set the target string used by this scan interface, ignoring any
further "changed" signals."""
self.toolbar.target_entry.handler_block(
self.target_entry_changed_handler)
self.toolbar.set_selected_target(target_string)
self.toolbar.target_entry.handler_unblock(
self.target_entry_changed_handler)
def set_profile_name_quiet(self, profile_name):
"""Set the profile name used by this scan interface, ignoring any
further "changed" signals."""
self.toolbar.profile_entry.handler_block(
self.profile_entry_changed_handler)
self.toolbar.set_selected_profile(profile_name)
self.toolbar.profile_entry.handler_unblock(
self.profile_entry_changed_handler)
def start_scan_cb(self, widget=None):
target = self.toolbar.selected_target
command = self.command_toolbar.command
profile = self.toolbar.selected_profile
log.debug(">>> Start Scan:")
log.debug(">>> Target: '%s'" % target)
log.debug(">>> Profile: '%s'" % profile)
log.debug(">>> Command: '%s'" % command)
if target != '':
try:
self.toolbar.add_new_target(target)
except IOError, e:
# We failed to save target_list.txt; treat it as read-only.
# Probably it's owned by root and this is a normal user.
log.debug(">>> Error saving %s: %s" %
(Path.target_list, str(e)))
if command == '':
warn_dialog = HIGAlertDialog(
message_format=_("Empty Nmap Command"),
secondary_text=_(
"There is no command to execute. "
"Maybe the selected/typed profile doesn't exist. "
"Please check the profile name or type the nmap "
"command you would like to execute."),
type=gtk.MESSAGE_ERROR)
warn_dialog.run()
warn_dialog.destroy()
return
self.execute_command(command, target, profile)
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()
def __init__(self):
HIGVBox.__init__(self)
# The borders are consuming too much space on Maemo. Setting it to
# 0 pixels while on Maemo
if is_maemo():
self.set_border_width(0)
self.set_spacing(0)
# True if nothing has happened here page yet, i.e., it's okay to load a
# scan from a file here.
self.empty = True
# The most recent name the inventory on this page has been saved under.
self.saved_filename = None
# The network inventory shown by this page. It may consist of multiple
# scans.
self.inventory = FilteredNetworkInventory()
# The list of currently running scans (NmapCommand objects).
self.jobs = []
# The list of running and finished scans shown on the Nmap Output page.
self.scans_store = ScansListStore()
self.top_box = HIGVBox()
self.__create_toolbar()
self.__create_command_toolbar()
self.select_default_profile()
self.scan_result = ScanResult(self.inventory,
self.scans_store,
scan_interface=self)
self.host_view_selection = self.scan_result.get_host_selection()
self.service_view_selection = self.scan_result.get_service_selection()
self.host_view_selection.connect('changed',
self.host_selection_changed)
self.service_view_selection.connect('changed',
self.service_selection_changed)
host_page = self.scan_result.scan_result_notebook.open_ports.host
host_page.host_view.get_selection().connect(
'changed', self.service_host_selection_changed)
self.host_view_selection.connect('changed',
self.host_selection_changed)
self.scan_result.scan_result_notebook.nmap_output.connect(
"changed", self._displayed_scan_change_cb)
self.scan_result.scan_result_notebook.scans_list.remove_button.connect(
"clicked", self._remove_scan_cb)
# The hosts dict maps hostnames (as returned by HostInfo.get_hostname)
# to HostInfo objects.
self.hosts = {}
# The services dict maps service names ("http") to lists of dicts of
# the form
# {'host': <HostInfo object>, 'hostname': u'example.com',
# 'port_state': u'open', 'portid': u'22', 'protocol': u'tcp',
# 'service_conf': u'10', 'service_extrainfo': u'protocol 2.0',
# 'service_method': u'probed', 'service_name': u'ssh',
# 'service_product': u'OpenSSH', 'service_version': u'4.3'}
# In other words each dict has the same keys as an entry in
# HostInfo.ports, with the addition of "host" and "hostname" keys.
self.services = {}
self.top_box.set_border_width(6)
self.top_box.set_spacing(5)
self.top_box._pack_noexpand_nofill(self.toolbar)
self.top_box._pack_noexpand_nofill(self.command_toolbar)
self._pack_noexpand_nofill(self.top_box)
self._pack_expand_fill(self.scan_result)
self.scan_result.scan_result_notebook.scans_list.cancel_button.connect(
"clicked", self._cancel_scans_list_cb)
self.update_cancel_button()
# Create the filter GUI
self.filter_bar = FilterBar()
self.pack_start(self.filter_bar, False, True, 0)
self.filter_bar.set_no_show_all(True)
self.filter_timeout_id = None
self.filter_bar.connect("changed", self.filter_changed)
self.scan_result.filter_toggle_button.connect(
"toggled", self.filter_toggle_toggled)
self.scan_result.filter_toggle_button.show()
class DiffWindow(gtk.Window):
def __init__(self, scans):
gtk.Window.__init__(self)
self.set_title(_("Compare Results"))
self.ndiff_process = None
# We allow the user to start a new diff before the old one has
# finished. We have to keep references to old processes until they
# finish to avoid problems when tearing down the Python interpreter at
# program exit.
self.old_processes = []
self.timer_id = None
self.main_vbox = HIGVBox()
self.diff_view = DiffView()
self.diff_view.set_size_request(-1, 100)
self.hbox_buttons = HIGHBox()
self.progress = gtk.ProgressBar()
self.btn_close = HIGButton(stock=gtk.STOCK_CLOSE)
self.hbox_selection = HIGHBox()
self.scan_chooser_a = ScanChooser(scans, _(u"A Scan"))
self.scan_chooser_b = ScanChooser(scans, _(u"B Scan"))
self._pack_widgets()
self._connect_widgets()
self.set_default_size(-1, 500)
# Initial Size Request
self.initial_size = self.get_size()
def _pack_widgets(self):
self.main_vbox.set_border_width(6)
self.hbox_selection.pack_start(self.scan_chooser_a, True, True)
self.hbox_selection.pack_start(self.scan_chooser_b, True, True)
self.main_vbox.pack_start(self.hbox_selection, False)
scroll = gtk.ScrolledWindow()
scroll.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
scroll.add(self.diff_view)
self.main_vbox.pack_start(scroll, True, True)
self.progress.hide()
self.progress.set_no_show_all(True)
self.hbox_buttons.pack_start(self.progress, False)
self.hbox_buttons.pack_end(self.btn_close, False)
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_close.connect("clicked", self.close)
self.scan_chooser_a.connect('changed', self.refresh_diff)
self.scan_chooser_b.connect('changed', self.refresh_diff)
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."""
log.debug("Refresh diff.")
if (self.ndiff_process is not None and
self.ndiff_process.poll() is None):
# Put this in the list of old processes we keep track of.
self.old_processes.append(self.ndiff_process)
self.ndiff_process = None
scan_a = self.scan_chooser_a.parsed_scan
scan_b = self.scan_chooser_b.parsed_scan
if scan_a is None or scan_b is None:
self.diff_view.clear()
else:
try:
self.ndiff_process = zenmapCore.Diff.ndiff(scan_a, scan_b)
except OSError, e:
alert = HIGAlertDialog(
message_format=_("Error running ndiff"),
secondary_text=_(
"There was an error running the ndiff program.\n\n"
) + str(e).decode(sys.getdefaultencoding(), "replace"))
alert.run()
alert.destroy()
else:
class ScanHostsView(HIGVBox, object):
HOST_MODE, SERVICE_MODE = range(2)
def __init__(self, scan_interface):
HIGVBox.__init__(self)
self._scan_interface = scan_interface
self._create_widgets()
self._connect_widgets()
self._pack_widgets()
self._set_scrolled()
self._set_host_list()
self._set_service_list()
self._pack_expand_fill(self.main_vbox)
self.mode = None
# Default mode is host mode
self.host_mode(self.host_mode_button)
self.host_view.show_all()
self.service_view.show_all()
def _create_widgets(self):
# Mode buttons
self.host_mode_button = gtk.ToggleButton(_("Hosts"))
self.service_mode_button = gtk.ToggleButton(_("Services"))
self.buttons_box = gtk.HBox()
# Main window vbox
self.main_vbox = HIGVBox()
# Host list
self.host_list = gtk.ListStore(object, str, str)
self.host_list.set_sort_func(1000, cmp_treemodel_addr)
self.host_list.set_sort_column_id(1000, gtk.SORT_ASCENDING)
self.host_view = gtk.TreeView(self.host_list)
self.pic_column = gtk.TreeViewColumn(_('OS'))
self.host_column = gtk.TreeViewColumn(_('Host'))
self.os_cell = gtk.CellRendererPixbuf()
self.host_cell = gtk.CellRendererText()
# Service list
self.service_list = gtk.ListStore(str)
self.service_list.set_sort_column_id(0, gtk.SORT_ASCENDING)
self.service_view = gtk.TreeView(self.service_list)
self.service_column = gtk.TreeViewColumn(_('Service'))
self.service_cell = gtk.CellRendererText()
self.scrolled = gtk.ScrolledWindow()
def _pack_widgets(self):
self.main_vbox.set_spacing(0)
self.main_vbox.set_border_width(0)
self.main_vbox._pack_noexpand_nofill(self.buttons_box)
self.main_vbox._pack_expand_fill(self.scrolled)
self.host_mode_button.set_active(True)
self.buttons_box.set_border_width(5)
self.buttons_box.pack_start(self.host_mode_button)
self.buttons_box.pack_start(self.service_mode_button)
def _connect_widgets(self):
self.host_mode_button.connect("toggled", self.host_mode)
self.service_mode_button.connect("toggled", self.service_mode)
def host_mode(self, widget):
self._remove_scrolled_child()
if widget.get_active():
self.mode = self.HOST_MODE
self.service_mode_button.set_active(False)
self.scrolled.add(self.host_view)
else:
self.service_mode_button.set_active(True)
def service_mode(self, widget):
self._remove_scrolled_child()
if widget.get_active():
self.mode = self.SERVICE_MODE
self.host_mode_button.set_active(False)
self.scrolled.add(self.service_view)
else:
self.host_mode_button.set_active(True)
def _remove_scrolled_child(self):
try:
child = self.scrolled.get_child()
self.scrolled.remove(child)
except:
pass
def _set_scrolled(self):
self.scrolled.set_border_width(5)
self.scrolled.set_size_request(150, -1)
self.scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC)
def _set_service_list(self):
self.service_view.set_enable_search(True)
self.service_view.set_search_column(0)
selection = self.service_view.get_selection()
selection.set_mode(gtk.SELECTION_MULTIPLE)
self.service_view.append_column(self.service_column)
self.service_column.set_resizable(True)
self.service_column.set_sort_column_id(0)
self.service_column.set_reorderable(True)
self.service_column.pack_start(self.service_cell, True)
self.service_column.set_attributes(self.service_cell, text=0)
def _set_host_list(self):
self.host_view.set_enable_search(True)
self.host_view.set_search_column(1)
selection = self.host_view.get_selection()
selection.set_mode(gtk.SELECTION_MULTIPLE)
self.host_view.append_column(self.pic_column)
self.host_view.append_column(self.host_column)
self.host_column.set_resizable(True)
self.pic_column.set_resizable(True)
self.host_column.set_sort_column_id(1000)
self.pic_column.set_sort_column_id(1)
self.host_column.set_reorderable(True)
self.pic_column.set_reorderable(True)
self.pic_column.pack_start(self.os_cell, True)
self.host_column.pack_start(self.host_cell, True)
self.pic_column.set_min_width(35)
self.pic_column.set_attributes(self.os_cell, stock_id = 1)
self.host_column.set_attributes(self.host_cell, text = 2)
def mass_update(self, hosts):
"""Update the internal ListStores to reflect the hosts and services
passed in. Hosts that have not changed are left alone."""
hosts = set(hosts)
services = set()
for h in hosts:
services.update([s["service_name"] for s in h.services])
# Disable sorting while elements are added. See the PyGTK FAQ 13.43,
# "Are there tips for improving performance when adding many rows to a
# Treeview?"
sort_column_id = self.host_list.get_sort_column_id()
self.host_list.set_default_sort_func(lambda *args: -1)
self.host_list.set_sort_column_id(-1, gtk.SORT_ASCENDING)
self.host_view.freeze_child_notify()
self.host_view.set_model(None)
it = self.host_list.get_iter_first()
# Remove any of our ListStore hosts that aren't in the list passed in.
while it:
host = self.host_list.get_value(it, 0)
if host in hosts:
hosts.remove(host)
self.host_list.set(it, 1, get_os_icon(host))
it = self.host_list.iter_next(it)
else:
if not self.host_list.remove(it):
it = None
# Add any remaining hosts into our ListStore.
for host in hosts:
self.add_host(host)
# Reenable sorting.
if sort_column_id != (None, None):
self.host_list.set_sort_column_id(*sort_column_id)
self.host_view.set_model(self.host_list)
self.host_view.thaw_child_notify()
it = self.service_list.get_iter_first()
# Remove any of our ListStore services that aren't in the list passed
# in.
while it:
service_name = self.service_list.get_value(it, 0)
if service_name in services:
services.remove(service_name)
it = self.service_list.iter_next(it)
else:
if not self.service_list.remove(it):
it = None
# Add any remaining services into our ListStore.
for service_name in services:
self.add_service(service_name)
def add_host(self, host):
self.host_list.append([host, get_os_icon(host), host.get_hostname()])
def add_service(self, service):
self.service_list.append([service])
def __init__(self):
HIGVBox.__init__(self)
# The borders are consuming too much space on Maemo. Setting it to
# 0 pixels while on Maemo
if is_maemo():
self.set_border_width(0)
self.set_spacing(0)
# True if nothing has happened here page yet, i.e., it's okay to load a
# scan from a file here.
self.empty = True
# The most recent name the inventory on this page has been saved under.
self.saved_filename = None
# The network inventory shown by this page. It may consist of multiple
# scans.
self.inventory = FilteredNetworkInventory()
# The list of currently running scans (NmapCommand objects).
self.jobs = []
# The list of running and finished scans shown on the Nmap Output page.
self.scans_store = ScansListStore()
self.top_box = HIGVBox()
self.__create_toolbar()
self.__create_command_toolbar()
self.select_default_profile()
self.scan_result = ScanResult(self.inventory, self.scans_store,
scan_interface=self)
self.host_view_selection = self.scan_result.get_host_selection()
self.service_view_selection = self.scan_result.get_service_selection()
self.host_view_selection.connect(
'changed', self.host_selection_changed)
self.service_view_selection.connect(
'changed', self.service_selection_changed)
host_page = self.scan_result.scan_result_notebook.open_ports.host
host_page.host_view.get_selection().connect(
'changed', self.service_host_selection_changed)
self.host_view_selection.connect(
'changed', self.host_selection_changed)
self.scan_result.scan_result_notebook.nmap_output.connect(
"changed", self._displayed_scan_change_cb)
self.scan_result.scan_result_notebook.scans_list.remove_button.connect(
"clicked", self._remove_scan_cb)
# The hosts dict maps hostnames (as returned by HostInfo.get_hostname)
# to HostInfo objects.
self.hosts = {}
# The services dict maps service names ("http") to lists of dicts of
# the form
# {'host': <HostInfo object>, 'hostname': u'example.com',
# 'port_state': u'open', 'portid': u'22', 'protocol': u'tcp',
# 'service_conf': u'10', 'service_extrainfo': u'protocol 2.0',
# 'service_method': u'probed', 'service_name': u'ssh',
# 'service_product': u'OpenSSH', 'service_version': u'4.3'}
# In other words each dict has the same keys as an entry in
# HostInfo.ports, with the addition of "host" and "hostname" keys.
self.services = {}
self.top_box.set_border_width(6)
self.top_box.set_spacing(5)
self.top_box._pack_noexpand_nofill(self.toolbar)
self.top_box._pack_noexpand_nofill(self.command_toolbar)
self._pack_noexpand_nofill(self.top_box)
self._pack_expand_fill(self.scan_result)
self.scan_result.scan_result_notebook.scans_list.cancel_button.connect(
"clicked", self._cancel_scans_list_cb)
self.update_cancel_button()
# Create the filter GUI
self.filter_bar = FilterBar()
self.pack_start(self.filter_bar, False, True, 0)
self.filter_bar.set_no_show_all(True)
self.filter_timeout_id = None
self.filter_bar.connect("changed", self.filter_changed)
self.scan_result.filter_toggle_button.connect("toggled",
self.filter_toggle_toggled)
self.scan_result.filter_toggle_button.show()
def __init__(self, scan):
HIGVBox.__init__(self)
na = _('Not available')
# Command info
self.command_label = HIGEntryLabel(_('Command:'))
self.info_command_label = HIGEntryLabel(na)
self.nmap_version_label = HIGEntryLabel(_('Nmap Version:'))
self.info_nmap_version_label = HIGEntryLabel(na)
self.verbose_label = HIGEntryLabel(_('Verbosity level:'))
self.info_verbose_label = HIGEntryLabel(na)
self.debug_label = HIGEntryLabel(_('Debug level:'))
self.info_debug_label = HIGEntryLabel(na)
self.command_expander = gtk.Expander("<b>" + _("Command Info") +
"</b>")
self.command_expander.set_use_markup(True)
self.command_table = HIGTable()
self.command_table.set_border_width(5)
self.command_table.set_row_spacings(6)
self.command_table.set_col_spacings(6)
self.command_hbox = HIGHBox()
self.command_hbox._pack_noexpand_nofill(hig_box_space_holder())
self.command_hbox._pack_noexpand_nofill(self.command_table)
self.command_table.attach(self.command_label, 0, 1, 0, 1)
self.command_table.attach(self.info_command_label, 1, 2, 0, 1)
self.command_table.attach(self.nmap_version_label, 0, 1, 1, 2)
self.command_table.attach(self.info_nmap_version_label, 1, 2, 1, 2)
self.command_table.attach(self.verbose_label, 0, 1, 2, 3)
self.command_table.attach(self.info_verbose_label, 1, 2, 2, 3)
self.command_table.attach(self.debug_label, 0, 1, 3, 4)
self.command_table.attach(self.info_debug_label, 1, 2, 3, 4)
self.command_expander.add(self.command_hbox)
self._pack_noexpand_nofill(self.command_expander)
self.command_expander.set_expanded(True)
# General info:
self.start_label = HIGEntryLabel(_('Started on:'))
self.info_start_label = HIGEntryLabel(na)
self.finished_label = HIGEntryLabel(_('Finished on:'))
self.info_finished_label = HIGEntryLabel(na)
self.host_up_label = HIGEntryLabel(_('Hosts up:'))
self.info_hosts_up_label = HIGEntryLabel(na)
self.host_down_label = HIGEntryLabel(_('Hosts down:'))
self.info_hosts_down_label = HIGEntryLabel(na)
self.host_scanned_label = HIGEntryLabel(_('Hosts scanned:'))
self.info_hosts_scanned_label = HIGEntryLabel(na)
self.open_label = HIGEntryLabel(_('Open ports:'))
self.info_open_label = HIGEntryLabel(na)
self.filtered_label = HIGEntryLabel(_('Filtered ports:'))
self.info_filtered_label = HIGEntryLabel(na)
self.closed_label = HIGEntryLabel(_('Closed ports:'))
self.info_closed_label = HIGEntryLabel(na)
self.general_expander = gtk.Expander("<b>" + _("General Info") +
"</b>")
self.general_expander.set_use_markup(True)
self.general_table = HIGTable()
self.general_table.set_border_width(5)
self.general_table.set_row_spacings(6)
self.general_table.set_col_spacings(6)
self.general_hbox = HIGHBox()
self.general_hbox._pack_noexpand_nofill(hig_box_space_holder())
self.general_hbox._pack_noexpand_nofill(self.general_table)
self.general_table.attach(self.start_label, 0, 1, 0, 1)
self.general_table.attach(self.info_start_label, 1, 2, 0, 1)
self.general_table.attach(self.finished_label, 0, 1, 1, 2)
self.general_table.attach(self.info_finished_label, 1, 2, 1, 2)
self.general_table.attach(self.host_up_label, 0, 1, 2, 3)
self.general_table.attach(self.info_hosts_up_label, 1, 2, 2, 3)
self.general_table.attach(self.host_down_label, 0, 1, 3, 4)
self.general_table.attach(self.info_hosts_down_label, 1, 2, 3, 4)
self.general_table.attach(self.host_scanned_label, 0, 1, 4, 5)
self.general_table.attach(self.info_hosts_scanned_label, 1, 2, 4, 5)
self.general_table.attach(self.open_label, 0, 1, 5, 6)
self.general_table.attach(self.info_open_label, 1, 2, 5, 6)
self.general_table.attach(self.filtered_label, 0, 1, 6, 7)
self.general_table.attach(self.info_filtered_label, 1, 2, 6, 7)
self.general_table.attach(self.closed_label, 0, 1, 7, 8)
self.general_table.attach(self.info_closed_label, 1, 2, 7, 8)
self.general_expander.add(self.general_hbox)
self._pack_noexpand_nofill(self.general_expander)
self.general_expander.set_expanded(True)
self._set_from_scan(scan)
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)
def highlight_tab(self):
# Creating highlight tab main box
self.highlight_main_vbox = HIGVBox()
# Creating highlight tab main table
self.highlight_main_table = HIGTable()
self.highlight_main_table.set_border_width(6)
#############
# Properties:
self.property_names = {
"details": [_("details"), "MAC Address:"],
"port_list": [_("port listing title"), "PORT STATE SERVICE"],
"open_port": [_("open port"), "22/tcp open ssh"],
"closed_port": [_("closed port"), "70/tcp closed gopher"],
"filtered_port": [_("filtered port"), "80/tcp filtered http"],
"date": [_("date"), "2006-05-26 11:14 BRT"],
"hostname": [_("hostname"), "scanme.nmap.org"],
"ip": [_("ip"), "127.0.0.1"]
}
for p in self.property_names:
settings = self.nmap_highlight.__getattribute__(p)
self.property_names[p].append(settings[0])
self.property_names[p].append(settings[1])
self.property_names[p].append(settings[2])
self.property_names[p].append(gtk.gdk.Color(*settings[3]))
self.property_names[p].append(gtk.gdk.Color(*settings[4]))
self.property_names[p].append(settings[5])
# Creating properties and related widgets and attaching it to main table
y1 = 0
y2 = 1
for p in self.property_names:
hp = HighlightProperty(p, self.property_names[p])
self.highlight_main_table.attach(hp.property_name_label, 0, 1, y1,
y2)
self.highlight_main_table.attach(hp.example_label, 1, 2, y1, y2)
self.highlight_main_table.attach(hp.bold_tg_button, 2, 3, y1, y2)
self.highlight_main_table.attach(hp.italic_tg_button, 3, 4, y1, y2)
self.highlight_main_table.attach(hp.underline_tg_button, 4, 5, y1,
y2)
self.highlight_main_table.attach(hp.text_color_button, 5, 6, y1,
y2)
self.highlight_main_table.attach(hp.highlight_color_button, 6, 7,
y1, y2)
# Setting example styles and colors
hp.update_example()
self.property_names[p].append(hp)
y1 += 1
y2 += 1
# Packing main table into main vbox
self.highlight_main_vbox.pack_start(self.highlight_main_table)
# Adding color tab
self.properties_notebook.append_page(
self.highlight_main_vbox, gtk.Label(_("Highlight definitions")))
class SearchWindow(BaseSearchWindow, object):
def __init__(self, load_method, append_method):
BaseSearchWindow.__init__(self)
self.set_default_size(600, 400)
self.load_method = load_method
self.append_method = append_method
self._create_widgets()
self._pack_widgets()
self._connect_widgets()
def _create_widgets(self):
self.vbox = HIGVBox()
self.bottom_hbox = gtk.HBox()
self.bottom_label = gtk.Label()
self.btn_box = gtk.HButtonBox()
self.btn_open = HIGButton(stock=gtk.STOCK_OPEN)
self.btn_append = HIGButton(_("Append"), gtk.STOCK_ADD)
self.btn_close = HIGButton(stock=gtk.STOCK_CLOSE)
self.search_gui = SearchGUI(self)
def _pack_widgets(self):
BaseSearchWindow._pack_widgets(self)
self.btn_box.set_layout(gtk.BUTTONBOX_END)
self.btn_box.set_spacing(4)
self.btn_box.pack_start(self.btn_close)
self.btn_box.pack_start(self.btn_append)
self.btn_box.pack_start(self.btn_open)
self.bottom_label.set_alignment(0.0, 0.5)
self.bottom_label.set_use_markup(True)
self.bottom_hbox.set_spacing(4)
self.bottom_hbox.pack_start(self.bottom_label, True)
self.bottom_hbox.pack_start(self.btn_box, False)
self.vbox.set_spacing(4)
self.vbox.pack_start(self.search_gui, True, True)
self.vbox.pack_start(self.bottom_hbox, False)
self.add(self.vbox)
def _connect_widgets(self):
# Double click on result, opens it
self.search_gui.result_view.connect("row-activated",
self.open_selected)
self.btn_open.connect("clicked", self.open_selected)
self.btn_append.connect("clicked", self.append_selected)
self.btn_close.connect("clicked", self.close)
self.connect("delete-event", self.close)
def close(self, widget=None, event=None):
self.search_gui.close()
self.destroy()
def set_label_text(self, text):
self.bottom_label.set_label(text)
def open_selected(self,
widget=None,
path=None,
view_column=None,
extra=None):
# Open selected results
self.load_method(self.results)
# Close Search Window
self.close()
def append_selected(self,
widget=None,
path=None,
view_column=None,
extra=None):
# Append selected results
self.append_method(self.results)
# Close Search Window
self.close()
def get_results(self):
# Return list with parsed objects from result list store
return self.search_gui.selected_results
results = property(get_results)
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)