1 files changed, 137 insertions, 0 deletions
diff --git a/zenmap/zenmapCore/UmitConfigParser.py b/zenmap/zenmapCore/UmitConfigParser.py
new file mode 100644
index 0000000..05b97a1
--- /dev/null
+++ b/zenmap/zenmapCore/UmitConfigParser.py
@@ -0,0 +1,137 @@
+#!/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/
+# *
+# ***************************************************************************/
+
+from configparser import ConfigParser, DEFAULTSECT, NoOptionError, \
+        NoSectionError
+from zenmapCore.UmitLogging import log
+
+
+class UmitConfigParser(ConfigParser):
+
+    def __init__(self, *args):
+        self.filenames = None
+        self.failed = False
+        ConfigParser.__init__(self, *args)
+
+    def set(self, section, option, value):
+        if not self.has_section(section):
+            self.add_section(section)
+
+        ConfigParser.set(self, section, option, str(value))
+        self.save_changes()
+
+    def read(self, filename):
+        log.debug(">>> Trying to parse: %s" % filename)
+
+        if ConfigParser.read(self, filename):
+            self.filenames = filename
+
+        return self.filenames
+
+    def save_changes(self):
+        if self.filenames:
+            log.debug("saving to %s" % self.filenames)
+            try:
+                with open(self.filenames, 'w') as fp:
+                    self.write(fp)
+            except Exception as e:
+                self.failed = e
+                log.error(">>> Can't save to %s: %s" % (self.filenames, e))
+                return
+            self.failed = False
+        else:
+            log.debug(">>> UmitConfigParser can't save changes: no filename")
+
+    def write(self, fp):
+        '''Write alphabetically sorted config files'''
+        if self._defaults:
+            fp.write("[%s]\n" % DEFAULTSECT)
+
+            items = sorted(self._defaults.items())
+
+            for (key, value) in items:
+                fp.write("%s = %s\n" % (key, str(value).replace('\n', '\n\t')))
+            fp.write("\n")
+
+        sects = sorted(self._sections.keys())
+
+        for section in sects:
+            fp.write("[%s]\n" % section)
+            for (key, value) in self._sections[section].items():
+                if key != "__name__":
+                    fp.write("%s = %s\n" %
+                             (key, str(value).replace('\n', '\n\t')))
+            fp.write("\n")
+
+
+def test_umit_conf_content(filename):
+    parser = ConfigParser()
+    parser.read(filename)
+
+    # Paths section
+    section = "paths"
+    assert get_or_false(parser, section, "nmap_command_path")
+
+
+def get_or_false(parser, section, option):
+    try:
+        result = parser.get(section, option)
+        return result
+    except NoOptionError:
+        return False
+    except NoSectionError:
+        return False