diff options
Diffstat (limited to 'zenmap/zenmapGUI/Icons.py')
-rw-r--r-- | zenmap/zenmapGUI/Icons.py | 206 |
1 files changed, 206 insertions, 0 deletions
diff --git a/zenmap/zenmapGUI/Icons.py b/zenmap/zenmapGUI/Icons.py new file mode 100644 index 0000000..9d8ea9c --- /dev/null +++ b/zenmap/zenmapGUI/Icons.py @@ -0,0 +1,206 @@ +#!/usr/bin/env python3 + +# ***********************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, GLib, GdkPixbuf + +import re +import os.path + +from zenmapCore.Paths import Path +from zenmapCore.UmitLogging import log + +icon_names = ( + # Operating Systems + 'default', + 'freebsd', + 'irix', + 'linux', + 'macosx', + 'openbsd', + 'redhat', + 'solaris', + 'ubuntu', + 'unknown', + 'win', + # Vulnerability Levels + 'vl_1', + 'vl_2', + 'vl_3', + 'vl_4', + 'vl_5') + +pixmap_path = Path.pixmaps_dir +if pixmap_path: + # This is a generator that returns file names for pixmaps in the order they + # should be tried. + def get_pixmap_file_names(icon_name, size): + yield '%s_%s.png' % (icon_name, size) + + iconfactory = Gtk.IconFactory() + for icon_name in icon_names: + for type, size in (('icon', '32'), ('logo', '75')): + key = '%s_%s' % (icon_name, type) + # Look for a usable image file. + for file_name in get_pixmap_file_names(icon_name, size): + file_path = os.path.join(pixmap_path, file_name) + try: + pixbuf = GdkPixbuf.Pixbuf.new_from_file(file_path) + break + except GLib.GError: + # Try again. + pass + else: + log.warn('Could not find the icon for %s at ' + 'any of (%s) in %s' % ( + icon_name, + ', '.join(get_pixmap_file_names(icon_name, size)), + pixmap_path)) + continue + iconset = Gtk.IconSet(pixbuf=pixbuf) + iconfactory.add(key, iconset) + log.debug('Register %s icon name for file %s' % (key, file_path)) + iconfactory.add_default() + + +def get_os_icon(host): + osmatch = host.get_best_osmatch() + if osmatch and osmatch['osclasses']: + osclass = osmatch['osclasses'][0] + else: + osclass = None + + if osclass and osmatch: + return get_os(osclass['osfamily'], osmatch['name'], 'icon') + else: + return get_os(None, None, 'icon') + + +def get_os_logo(host): + osmatch = host.get_best_osmatch() + if osmatch and osmatch['osclasses']: + osclass = osmatch['osclasses'][0] + else: + osclass = None + + if osclass and osmatch: + return get_os(osclass['osfamily'], osmatch['name'], 'logo') + else: + return get_os(None, None, 'logo') + + +def get_os(osfamily, osmatch, type): + if osfamily: + if osfamily == 'Linux': + if re.findall("ubuntu", osmatch.lower()): + # Ubuntu icon + return 'ubuntu_%s' % type + elif re.findall("red hat", osmatch.lower()): + # RedHat icon + return 'redhat_%s' % type + else: + # Generic Linux icon + return 'linux_%s' % type + elif osfamily == 'Windows': + # Windows icon + return 'win_%s' % type + elif osfamily == 'OpenBSD': + # OpenBSD icon + return 'openbsd_%s' % type + elif osfamily == 'FreeBSD': + # FreeBSD icon + return 'freebsd_%s' % type + elif osfamily == 'NetBSD': + # NetBSD icon + return 'default_%s' % type + elif osfamily == 'Solaris': + # Solaris icon + return 'solaris_%s' % type + elif osfamily == 'OpenSolaris': + # OpenSolaris icon + return 'solaris_%s' % type + elif osfamily == 'IRIX': + # Irix icon + return 'irix_%s' % type + elif osfamily == 'Mac OS X': + # Mac OS X icon + return 'macosx_%s' % type + elif osfamily == 'Mac OS': + # Mac OS icon + return 'macosx_%s' % type + else: + # Default OS icon + return 'default_%s' % type + else: + # Unknown OS icon + return 'unknown_%s' % type + + +def get_vulnerability_logo(open_ports): + open_ports = int(open_ports) + if open_ports < 3: + return 'vl_1_logo' + elif open_ports < 5: + return 'vl_2_logo' + elif open_ports < 7: + return 'vl_3_logo' + elif open_ports < 9: + return 'vl_4_logo' + else: + return 'vl_5_logo' |