From 4563e600fa80a564c38c8400653bfcfa18e7e2b4 Mon Sep 17 00:00:00 2001
From: Jake Miller <laconicwolf@gmail.com>
Date: Thu, 29 Mar 2018 17:57:01 -0500
Subject: [PATCH] Update nmap_xml_parser.py

---
 nmap_xml_parser.py | 17 ++++++++++++++---
 1 file changed, 14 insertions(+), 3 deletions(-)

diff --git a/nmap_xml_parser.py b/nmap_xml_parser.py
index 49650b4..e99832e 100644
--- a/nmap_xml_parser.py
+++ b/nmap_xml_parser.py
@@ -54,6 +54,11 @@ def get_host_data(root):
         except IndexError:
             host_name = ''
 
+        if args.ip_addresses:
+            addr_info.extend((ip_address, host_name))
+            host_data.append(addr_info)
+            continue
+            
         try:
             os_element = host.findall('os')
             os_name = os_element[0].findall('osmatch')[0].attrib['name']
@@ -147,9 +152,13 @@ def list_ip_addresses(data):
     ip_list = []
     for item in data:
         ip_list.append(item[0])
+
     sorted_set = sorted(set(ip_list))
+    addr_list = []
     for ip in sorted_set:
-        print(ip)
+        addr_list.append(ip)
+
+    return addr_list
 
 
 def print_web_ports(data):
@@ -242,7 +251,9 @@ def main():
         if args.csv:
             parse_to_csv(data)
         if args.ip_addresses:
-            list_ip_addresses(data)
+            addrs = list_ip_addresses(data)
+            for addr in addrs:
+                print(addr)
         if args.print_all:
             print_data(data)
         if args.filter_by_port:
@@ -282,4 +293,4 @@ if __name__ == '__main__':
         exit()
         
     csv_name = args.csv
-    main()
\ No newline at end of file
+    main()