class NmapOutputViewer (gtk.VBox): HIGHLIGHT_PROPERTIES = ["details", "date", "hostname", "ip", "port_list", "open_port", "closed_port", "filtered_port"] def __init__(self, refresh=1, stop=1): self.nmap_highlight = NmapOutputHighlight() gtk.VBox.__init__(self) # Creating widgets self.__create_widgets() # Setting scrolled window self.__set_scrolled_window() # Setting text view self.__set_text_view() buffer = self.text_view.get_buffer() # The end mark is used to scroll to the bottom of the display. self.end_mark = buffer.create_mark(None, buffer.get_end_iter(), False) self.refreshing = True # Adding widgets to the VBox self.pack_start(self.scrolled, expand=True, fill=True) # The NmapCommand instance, if any, whose output is shown in this # display. self.command_execution = None # The position of the last read from the output stream. self.output_file_pointer = None def __create_widgets(self): # Creating widgets self.scrolled = gtk.ScrolledWindow() self.text_view = gtk.TextView() def __set_scrolled_window(self): # Seting scrolled window self.scrolled.set_border_width(5) self.scrolled.add(self.text_view) self.scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC) def __set_text_view(self): self.text_view.set_wrap_mode(gtk.WRAP_WORD) self.text_view.set_editable(False) self.tag_font = self.text_view.get_buffer().create_tag(None) self.tag_font.set_property("family", "Monospace") for property in self.HIGHLIGHT_PROPERTIES: settings = self.nmap_highlight.__getattribute__(property) tag = self.text_view.get_buffer().create_tag(property) if settings[0]: tag.set_property("weight", pango.WEIGHT_HEAVY) else: tag.set_property("weight", pango.WEIGHT_NORMAL) if settings[1]: tag.set_property("style", pango.STYLE_ITALIC) else: tag.set_property("style", pango.STYLE_NORMAL) if settings[2]: tag.set_property("underline", pango.UNDERLINE_SINGLE) else: tag.set_property("underline", pango.UNDERLINE_NONE) text_color = settings[3] highlight_color = settings[4] tag.set_property( "foreground", gtk.color_selection_palette_to_string( [gtk.gdk.Color(*text_color), ])) tag.set_property( "background", gtk.color_selection_palette_to_string( [gtk.gdk.Color(*highlight_color), ])) def go_to_host(self, host): """Go to host line on nmap output result""" buff = self.text_view.get_buffer() start_iter = buff.get_start_iter() found_tuple = start_iter.forward_search( "\nNmap scan report for %s\n" % host, gtk.TEXT_SEARCH_TEXT_ONLY ) if found_tuple is None: return found = found_tuple[0] if not found.forward_line(): return self.text_view.scroll_to_iter(found, 0, True, 0, 0) def show_output_properties(self, widget): nmap_out_prop = NmapOutputProperties(self.text_view) nmap_out_prop.run() for prop in nmap_out_prop.property_names: widget = nmap_out_prop.property_names[prop][8] wid_props = [] if widget.bold: wid_props.append(1) else: wid_props.append(0) if widget.italic: wid_props.append(1) else: wid_props.append(0) if widget.underline: wid_props.append(1) else: wid_props.append(0) wid_props.append("(%s, %s, %s)" % (widget.text_color.red, widget.text_color.green, widget.text_color.blue)) wid_props.append("(%s, %s, %s)" % (widget.highlight_color.red, widget.highlight_color.green, widget.highlight_color.blue)) self.nmap_highlight.__setattr__(widget.property_name, wid_props) nmap_out_prop.destroy() self.nmap_highlight.save_changes() self.apply_highlighting() def apply_highlighting(self, start_iter=None, end_iter=None): buf = self.text_view.get_buffer() if start_iter is None: start_iter = buf.get_start_iter() else: # Patterns are line-oriented; start on a line boundary. start_iter.backward_line() if end_iter is None: end_iter = buf.get_end_iter() buf.apply_tag(self.tag_font, start_iter, end_iter) if not self.nmap_highlight.enable: return text = buf.get_text(start_iter, end_iter) for property in self.HIGHLIGHT_PROPERTIES: settings = self.nmap_highlight.__getattribute__(property) for m in re.finditer(settings[5], text, re.M): m_start_iter = start_iter.copy() m_start_iter.forward_chars(m.start()) m_end_iter = start_iter.copy() m_end_iter.forward_chars(m.end()) buf.apply_tag_by_name(property, m_start_iter, m_end_iter) def show_nmap_output(self, output): """Show the string (or unicode) output in the output display.""" try: self.text_view.get_buffer().set_text(output) self.apply_highlighting() except MemoryError: self.show_large_output_message(self.command_execution) def set_command_execution(self, command): """Set the live running command whose output is shown by this display. The current output is extracted from the command object.""" self.command_execution = command if command is not None: self.text_view.get_buffer().set_text(u"") self.output_file_pointer = 0 else: self.output_file_pointer = None self.refresh_output() def show_large_output_message(self, command=None): buf = self.text_view.get_buffer() try: running = (command is not None and command.scan_state() is True) except Exception: running = False complete = False else: complete = not running if running: buf.set_text("Warning: You have insufficient resources for Zenmap " "to be able to display the complete output from Nmap here. \n" "Zenmap will continue to run the scan to completion. However," " some features of Zenmap might not work as expected.") elif complete: buf.set_text("Warning: You have insufficient resources for Zenmap " "to be able to display the complete output from Nmap here. \n" "The scan has completed. However, some features of Zenmap " "might not work as expected.") else: buf.set_text("Warning: You have insufficient resources for Zenmap " "to be able to display the complete output from Nmap here. \n" "The scan has been stopped. Some features of Zenmap might not " "work as expected.") def refresh_output(self, widget=None): """Update the output from the latest output of the command associated with this view, as set by set_command_execution. It has no effect if no command has been set.""" log.debug("Refresh nmap output") if self.command_execution is None: return # Seek to the end of the most recent read. self.command_execution.stdout_file.seek(self.output_file_pointer) try: new_output = self.command_execution.stdout_file.read() except MemoryError: self.show_large_output_message(self.command_execution) return self.output_file_pointer = self.command_execution.stdout_file.tell() v_adj = self.scrolled.get_vadjustment() if new_output and v_adj is not None: # Find out if the view is already scrolled to the bottom. at_end = (v_adj.value >= v_adj.upper - v_adj.page_size) buf = self.text_view.get_buffer() prev_end_mark = buf.create_mark( None, buf.get_end_iter(), left_gravity=True) try: buf.insert(buf.get_end_iter(), new_output) # Highlight the new text. self.apply_highlighting( buf.get_iter_at_mark(prev_end_mark), buf.get_end_iter()) except MemoryError: self.show_large_output_message(self.command_execution) return if at_end: # If we were already scrolled to the bottom, scroll back to the # bottom again. Also do it in an idle handler in case the added # text causes a scroll bar to appear and reflow the text, # making the text a bit taller. self.text_view.scroll_mark_onscreen(self.end_mark) gobject.idle_add( lambda: self.text_view.scroll_mark_onscreen( self.end_mark))
class NmapOutputViewer (gtk.VBox): HIGHLIGHT_PROPERTIES = ["details", "date", "hostname", "ip", "port_list", "open_port", "closed_port", "filtered_port"] def __init__(self, refresh=1, stop=1): self.nmap_highlight = NmapOutputHighlight() gtk.VBox.__init__(self) # Creating widgets self.__create_widgets() # Setting scrolled window self.__set_scrolled_window() # Setting text view self.__set_text_view() buffer = self.text_view.get_buffer() # The end mark is used to scroll to the bottom of the display. self.end_mark = buffer.create_mark(None, buffer.get_end_iter(), False) self.refreshing = True # Adding widgets to the VBox self.pack_start(self.scrolled, expand=True, fill=True) # The NmapCommand instance, if any, whose output is shown in this # display. self.command_execution = None # The position of the last read from the output stream. self.output_file_pointer = None def __create_widgets(self): # Creating widgets self.scrolled = gtk.ScrolledWindow() self.text_view = gtk.TextView() def __set_scrolled_window(self): # Seting scrolled window self.scrolled.set_border_width(5) self.scrolled.add(self.text_view) self.scrolled.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC) def __set_text_view(self): self.text_view.set_wrap_mode(gtk.WRAP_WORD) self.text_view.set_editable(False) self.tag_font = self.text_view.get_buffer().create_tag(None) self.tag_font.set_property("family", "Monospace") for property in self.HIGHLIGHT_PROPERTIES: settings = self.nmap_highlight.__getattribute__(property) tag = self.text_view.get_buffer().create_tag(property) if settings[0]: tag.set_property("weight", pango.WEIGHT_HEAVY) else: tag.set_property("weight", pango.WEIGHT_NORMAL) if settings[1]: tag.set_property("style", pango.STYLE_ITALIC) else: tag.set_property("style", pango.STYLE_NORMAL) if settings[2]: tag.set_property("underline", pango.UNDERLINE_SINGLE) else: tag.set_property("underline", pango.UNDERLINE_NONE) text_color = settings[3] highlight_color = settings[4] tag.set_property( "foreground", gtk.color_selection_palette_to_string( [gtk.gdk.Color(*text_color), ])) tag.set_property( "background", gtk.color_selection_palette_to_string( [gtk.gdk.Color(*highlight_color), ])) def go_to_host(self, host): """Go to host line on nmap output result""" buff = self.text_view.get_buffer() start_iter = buff.get_start_iter() found_tuple = start_iter.forward_search( "\nNmap scan report for %s\n" % host, gtk.TEXT_SEARCH_TEXT_ONLY ) if found_tuple is None: return found = found_tuple[0] if not found.forward_line(): return self.text_view.scroll_to_iter(found, 0, True, 0, 0) def show_output_properties(self, widget): nmap_out_prop = NmapOutputProperties(self.text_view) nmap_out_prop.run() for prop in nmap_out_prop.property_names: widget = nmap_out_prop.property_names[prop][8] wid_props = [] if widget.bold: wid_props.append(1) else: wid_props.append(0) if widget.italic: wid_props.append(1) else: wid_props.append(0) if widget.underline: wid_props.append(1) else: wid_props.append(0) wid_props.append("(%s, %s, %s)" % (widget.text_color.red, widget.text_color.green, widget.text_color.blue)) wid_props.append("(%s, %s, %s)" % (widget.highlight_color.red, widget.highlight_color.green, widget.highlight_color.blue)) self.nmap_highlight.__setattr__(widget.property_name, wid_props) nmap_out_prop.destroy() self.nmap_highlight.save_changes() self.apply_highlighting() def apply_highlighting(self, start_iter=None, end_iter=None): buf = self.text_view.get_buffer() if start_iter is None: start_iter = buf.get_start_iter() else: # Patterns are line-oriented; start on a line boundary. start_iter.backward_line() if end_iter is None: end_iter = buf.get_end_iter() buf.apply_tag(self.tag_font, start_iter, end_iter) if not self.nmap_highlight.enable: return text = buf.get_text(start_iter, end_iter) for property in self.HIGHLIGHT_PROPERTIES: settings = self.nmap_highlight.__getattribute__(property) for m in re.finditer(settings[5], text, re.M): m_start_iter = start_iter.copy() m_start_iter.forward_chars(m.start()) m_end_iter = start_iter.copy() m_end_iter.forward_chars(m.end()) buf.apply_tag_by_name(property, m_start_iter, m_end_iter) def show_nmap_output(self, output): """Show the string (or unicode) output in the output display.""" try: self.text_view.get_buffer().set_text(output) self.apply_highlighting() except MemoryError: self.show_large_output_message(self.command_execution) def set_command_execution(self, command): """Set the live running command whose output is shown by this display. The current output is extracted from the command object.""" self.command_execution = command if command is not None: self.text_view.get_buffer().set_text(u"") self.output_file_pointer = 0 else: self.output_file_pointer = None self.refresh_output() def show_large_output_message(self, command=None): buf = self.text_view.get_buffer() try: running = (command is not None and command.scan_state() is True) except: running = False complete = False else: complete = not running if running: buf.set_text("Warning: You have insufficient resources for Zenmap " "to be able to display the complete output from Nmap here. \n" "Zenmap will continue to run the scan to completion. However," " some features of Zenmap might not work as expected.") elif complete: buf.set_text("Warning: You have insufficient resources for Zenmap " "to be able to display the complete output from Nmap here. \n" "The scan has completed. However, some features of Zenmap " "might not work as expected.") else: buf.set_text("Warning: You have insufficient resources for Zenmap " "to be able to display the complete output from Nmap here. \n" "The scan has been stopped. Some features of Zenmap might not " "work as expected.") def refresh_output(self, widget=None): """Update the output from the latest output of the command associated with this view, as set by set_command_execution. It has no effect if no command has been set.""" log.debug("Refresh nmap output") if self.command_execution is None: return # Seek to the end of the most recent read. self.command_execution.stdout_file.seek(self.output_file_pointer) pos = self.command_execution.stdout_file.tell() try: new_output = self.command_execution.stdout_file.read() except MemoryError: self.show_large_output_message(self.command_execution) return self.output_file_pointer = self.command_execution.stdout_file.tell() # print "read %d -> %d %d" % ( # pos, self.output_file_pointer, len(new_output)) v_adj = self.scrolled.get_vadjustment() if new_output and v_adj is not None: # Find out if the view is already scrolled to the bottom. at_end = (v_adj.value >= v_adj.upper - v_adj.page_size) buf = self.text_view.get_buffer() prev_end_mark = buf.create_mark( None, buf.get_end_iter(), left_gravity=True) try: buf.insert(buf.get_end_iter(), new_output) # Highlight the new text. self.apply_highlighting( buf.get_iter_at_mark(prev_end_mark), buf.get_end_iter()) except MemoryError: self.show_large_output_message(self.command_execution) return if at_end: # If we were already scrolled to the bottom, scroll back to the # bottom again. Also do it in an idle handler in case the added # text causes a scroll bar to appear and reflow the text, # making the text a bit taller. self.text_view.scroll_mark_onscreen(self.end_mark) gobject.idle_add( lambda: self.text_view.scroll_mark_onscreen( self.end_mark))
class NmapOutputProperties(HIGDialog): def __init__(self, nmap_output_view): HIGDialog.__init__(self, _("Nmap Output Properties"), buttons=(gtk.STOCK_CLOSE, gtk.RESPONSE_CLOSE)) self.nmap_highlight = NmapOutputHighlight() self.__create_widgets() self.__pack_widgets() self.highlight_tab() self.vbox.show_all() def __create_widgets(self): self.properties_notebook = HIGNotebook() def __pack_widgets(self): self.vbox.pack_start(self.properties_notebook) 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 NmapOutputProperties(HIGDialog): def __init__(self, nmap_output_view): HIGDialog.__init__(self, _("Nmap Output Properties"), buttons=(gtk.STOCK_CLOSE, gtk.RESPONSE_CLOSE)) self.nmap_highlight = NmapOutputHighlight() self.__create_widgets() self.__pack_widgets() self.highlight_tab() self.vbox.show_all() def __create_widgets(self): self.properties_notebook = HIGNotebook() def __pack_widgets(self): self.vbox.pack_start(self.properties_notebook) 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")))