From e0899c73270ce79d95e22c83e78cce9f82447397 Mon Sep 17 00:00:00 2001
From: Jake Miller <laconicwolf@gmail.com>
Date: Mon, 23 Jul 2018 20:33:37 -0500
Subject: [PATCH] Update nmap_xml_parser.py

---
 nmap_xml_parser.py | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/nmap_xml_parser.py b/nmap_xml_parser.py
index 2f6c917..3740794 100644
--- a/nmap_xml_parser.py
+++ b/nmap_xml_parser.py
@@ -237,12 +237,14 @@ def main():
             print("\n[-] The file {} cannot be found or you do not have permission to open the file.".format(filename))
             continue
 
-        # Read the file and check for entities
-        with open(filename) as fh:
-            contents = fh.read()
-            if '<!entity' in contents.lower():
-                print("[-] Error! This program does not permit XML entities. Exiting!")
-                continue
+        if not args.skip_entity_check:
+            # Read the file and check for entities
+            with open(filename) as fh:
+                contents = fh.read()
+                if '<!entity' in contents.lower():
+                    print("[-] Error! This program does not permit XML entities. Ignoring {}".format(filename))
+                    print("[*] Use -s (--skip_entity_check) to ignore this check for XML entities.")
+                    continue
         data = parse_xml(filename)
         if not data:
             print("[*] Zero hosts identitified as 'Up'. Exiting.")
@@ -272,6 +274,9 @@ if __name__ == '__main__':
     parser.add_argument("-d", "--debug",
                         help="Display error information",
                         action="store_true")
+    parser.add_argument("-s", "--skip_entity_check",
+                        help="Skip the check for XML entities",
+                        action="store_true")
     parser.add_argument("-p", "--print_all",
                         help="Display scan information to the screen", 
                         action="store_true")