diff options
Diffstat (limited to '')
-rw-r--r-- | zenmap/radialnet/gui/HostsViewer.py | 231 |
1 files changed, 231 insertions, 0 deletions
diff --git a/zenmap/radialnet/gui/HostsViewer.py b/zenmap/radialnet/gui/HostsViewer.py new file mode 100644 index 0000000..feb0ae8 --- /dev/null +++ b/zenmap/radialnet/gui/HostsViewer.py @@ -0,0 +1,231 @@ +# vim: set fileencoding=utf-8 : + +# ***********************IMPORTANT NMAP LICENSE TERMS************************ +# * +# * The Nmap Security Scanner is (C) 1996-2023 Nmap Software LLC ("The Nmap +# * Project"). Nmap is also a registered trademark of the Nmap Project. +# * +# * This program is distributed under the terms of the Nmap Public Source +# * License (NPSL). The exact license text applying to a particular Nmap +# * release or source code control revision is contained in the LICENSE +# * file distributed with that version of Nmap or source code control +# * revision. More Nmap copyright/legal information is available from +# * https://nmap.org/book/man-legal.html, and further information on the +# * NPSL license itself can be found at https://nmap.org/npsl/ . This +# * header summarizes some key points from the Nmap license, but is no +# * substitute for the actual license text. +# * +# * Nmap is generally free for end users to download and use themselves, +# * including commercial use. It is available from https://nmap.org. +# * +# * The Nmap license generally prohibits companies from using and +# * redistributing Nmap in commercial products, but we sell a special Nmap +# * OEM Edition with a more permissive license and special features for +# * this purpose. See https://nmap.org/oem/ +# * +# * If you have received a written Nmap license agreement or contract +# * stating terms other than these (such as an Nmap OEM license), you may +# * choose to use and redistribute Nmap under those terms instead. +# * +# * The official Nmap Windows builds include the Npcap software +# * (https://npcap.com) for packet capture and transmission. It is under +# * separate license terms which forbid redistribution without special +# * permission. So the official Nmap Windows builds may not be redistributed +# * without special permission (such as an Nmap OEM license). +# * +# * Source is provided to this software because we believe users have a +# * right to know exactly what a program is going to do before they run it. +# * This also allows you to audit the software for security holes. +# * +# * Source code also allows you to port Nmap to new platforms, fix bugs, and add +# * new features. You are highly encouraged to submit your changes as a Github PR +# * or by email to the dev@nmap.org mailing list for possible incorporation into +# * the main distribution. Unless you specify otherwise, it is understood that +# * you are offering us very broad rights to use your submissions as described in +# * the Nmap Public Source License Contributor Agreement. This is important +# * because we fund the project by selling licenses with various terms, and also +# * because the inability to relicense code has caused devastating problems for +# * other Free Software projects (such as KDE and NASM). +# * +# * The free version of Nmap is distributed in the hope that it will be +# * useful, but WITHOUT ANY WARRANTY; without even the implied warranty of +# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. Warranties, +# * indemnification and commercial support are all available through the +# * Npcap OEM program--see https://nmap.org/oem/ +# * +# ***************************************************************************/ + +import gi + +gi.require_version("Gtk", "3.0") +from gi.repository import Gtk + +import re + +from radialnet.bestwidgets.windows import BWMainWindow + +from radialnet.gui.NodeNotebook import NodeNotebook +from radialnet.util.misc import ipv4_compare + + +HOSTS_COLORS = ['#d5ffd5', '#ffffd5', '#ffd5d5'] + +HOSTS_HEADER = ['ID', '#', 'Hosts'] + +DIMENSION = (700, 400) + +IP_RE = re.compile(r'^[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}$') + + +class HostsViewer(BWMainWindow): + """ + """ + def __init__(self, nodes): + """ + """ + BWMainWindow.__init__(self) + self.set_title(_('Hosts Viewer')) + self.set_default_size(DIMENSION[0], DIMENSION[1]) + + self.__nodes = nodes + self.__default_view = Gtk.Label.new(_("No node selected")) + self.__view = self.__default_view + + self.__create_widgets() + + def __create_widgets(self): + """ + """ + self.__panel = Gtk.Paned.new(Gtk.Orientation.HORIZONTAL) + self.__panel.set_border_width(6) + + self.__list = HostsList(self, self.__nodes) + + self.__panel.add1(self.__list) + self.__panel.add2(self.__view) + self.__panel.set_position(int(DIMENSION[0] / 5)) + + self.add(self.__panel) + + def change_notebook(self, node): + """ + """ + if self.__view is not None: + self.__view.destroy() + + if node is not None: + self.__view = NodeNotebook(node) + else: + self.__view = self.__default_view + self.__view.show_all() + + self.__panel.add2(self.__view) + + +class HostsList(Gtk.ScrolledWindow): + """ + """ + def __init__(self, parent, nodes): + """ + """ + super(HostsList, self).__init__() + self.set_policy(Gtk.PolicyType.AUTOMATIC, Gtk.PolicyType.AUTOMATIC) + self.set_shadow_type(Gtk.ShadowType.NONE) + + self.__parent = parent + self.__nodes = nodes + + self.__create_widgets() + + def __create_widgets(self): + """ + """ + self.__cell = Gtk.CellRendererText() + + self.__hosts_store = Gtk.ListStore.new([int, int, str, str, bool]) + + self.__hosts_treeview = Gtk.TreeView.new_with_model(self.__hosts_store) + self.__hosts_treeview.connect('cursor-changed', self.__cursor_callback) + + for i in range(len(self.__nodes)): + + node = self.__nodes[i] + + ports = node.get_info('number_of_open_ports') + color = HOSTS_COLORS[node.get_info('vulnerability_score')] + + host = node.get_info('hostname') or node.get_info('ip') or "" + + self.__hosts_store.append([i, + ports, + host, + color, + True]) + + self.__hosts_column = list() + + for i in range(0, len(HOSTS_HEADER)): + + column = Gtk.TreeViewColumn(title=HOSTS_HEADER[i], + cell_renderer=self.__cell, + text=i) + + self.__hosts_column.append(column) + + self.__hosts_column[i].set_reorderable(True) + self.__hosts_column[i].set_resizable(True) + self.__hosts_column[i].set_attributes(self.__cell, + text=i, + background=3, + editable=4) + + self.__hosts_treeview.append_column(self.__hosts_column[2]) + + self.__hosts_store.set_sort_func(2, self.__host_sort) + + self.__hosts_column[2].set_sort_column_id(2) + + self.add_with_viewport(self.__hosts_treeview) + + if len(self.__hosts_treeview.get_model()) > 0: + self.__hosts_treeview.set_cursor((0,)) + self.__cursor_callback(self.__hosts_treeview) + + def __cursor_callback(self, widget): + """ + """ + path = widget.get_cursor()[0] + if path is None: + return + + iter = self.__hosts_store.get_iter(path) + + node = self.__nodes[self.__hosts_store.get_value(iter, 0)] + + self.__parent.change_notebook(node) + + def __host_sort(self, treemodel, iter1, iter2, *_): + """ + """ + value1 = treemodel.get_value(iter1, 2) + value2 = treemodel.get_value(iter2, 2) + + value1_is_ip = IP_RE.match(value1) + value2_is_ip = IP_RE.match(value2) + + if value1_is_ip and value2_is_ip: + return ipv4_compare(value1, value2) + + if value1_is_ip: + return -1 + + if value2_is_ip: + return 1 + + if value1 < value2: + return -1 + + if value1 > value2: + return 1 + + return 0 |