Merge pull request #2 from DefensiveOrigins/rfc1918

Rfc1918

Author: Relkci

HostCount.py

@@ -3,6 +3,20 @@
 import os
 from collections import defaultdict
 
+# RFC1918 networks
+RFC1918_NETWORKS = [
+    ipaddress.ip_network("10.0.0.0/8"),
+    ipaddress.ip_network("172.16.0.0/12"),
+    ipaddress.ip_network("192.168.0.0/16"),
+]
+
+def is_rfc1918(ip_str):
+    try:
+        ip = ipaddress.ip_address(ip_str)
+        return any(ip in net for net in RFC1918_NETWORKS)
+    except ValueError:
+        return False
+
 def extract_unique_hosts_from_file(file_path):
     unique_hosts = set()
 
@@ -37,6 +51,8 @@ def main():
     parser.add_argument("-d", "--directory", help="Directory of files to process.")
     parser.add_argument("-D", "--duplicates", action="store_true",
                         help="When processing a directory, list hosts that appear in more than one file and which files they appear in.")
+    parser.add_argument("-r", "--rfc1918", action="store_true",
+                        help="Optionally show how many hosts are in RFC1918 space vs not (per-file and totals).")
     args = parser.parse_args()
 
     if args.directory:
@@ -52,28 +68,49 @@ def main():
                 grand_total_hosts.update(file_hosts)
         print(f"\nTotal unique hosts across all files: {len(grand_total_hosts)}")
 
-        if args.duplicates:
-            # Build host -> set(files) mapping
-            host_to_files = defaultdict(set)
-            for fname, hosts in file_hosts_map.items():
-                for h in hosts:
-                    host_to_files[h].add(fname)
+        # Build host -> set(files) mapping once for both -D output and the duplicate note
+        host_to_files = defaultdict(set)
+        for fname, hosts in file_hosts_map.items():
+            for h in hosts:
+                host_to_files[h].add(fname)
+        duplicates_map = {host: sorted(list(files)) for host, files in host_to_files.items() if len(files) > 1}
+        duplicate_count = len(duplicates_map)
 
-            # Find hosts present in more than one file
-            duplicates = {host: sorted(list(files)) for host, files in host_to_files.items() if len(files) > 1}
+        # If -D not used, but duplicates exist, print a concise note that duplicates were found and are only counted once
+        if not args.duplicates and duplicate_count > 0:
+            print(f"\nNote: {duplicate_count} host(s) were found in more than one file but are only counted once in the total unique host count. Use -D to list them.")
 
-            if not duplicates:
+        if args.rfc1918:
+            print("\nRFC1918 breakdown per file:")
+            for fname in sorted(file_hosts_map.keys()):
+                hosts = file_hosts_map[fname]
+                rfc_count = sum(1 for h in hosts if is_rfc1918(h))
+                non_rfc = len(hosts) - rfc_count
+                print(f"{fname}: {rfc_count} RFC1918, {non_rfc} non-RFC1918")
+
+            total_rfc = sum(1 for h in grand_total_hosts if is_rfc1918(h))
+            total_non_rfc = len(grand_total_hosts) - total_rfc
+            print(f"\nTotals across all files: {total_rfc} RFC1918, {total_non_rfc} non-RFC1918")
+
+        if args.duplicates:
+            # Use the mapping we already built to list duplicates
+            if not duplicates_map:
                 print("\nNo duplicate hosts found between files.")
             else:
-                print(f"\nDuplicate hosts found between files: {len(duplicates)} hosts\n")
-                for host in sorted(duplicates.keys(), key=lambda x: tuple(int(p) for p in x.split('.'))):
-                    files_list = ", ".join(duplicates[host])
+                print(f"\nDuplicate hosts found between files: {len(duplicates_map)} hosts\n")
+                for host in sorted(duplicates_map.keys(), key=lambda x: tuple(int(p) for p in x.split('.'))):
+                    files_list = ", ".join(duplicates_map[host])
                     print(f"{host}: {files_list}")
 
     elif args.file:
         unique_hosts = extract_unique_hosts_from_file(args.file)
         print(f"\nTotal unique hosts in {args.file} (excluding network and broadcast): {len(unique_hosts)}")
 
+        if args.rfc1918:
+            rfc_count = sum(1 for h in unique_hosts if is_rfc1918(h))
+            non_rfc = len(unique_hosts) - rfc_count
+            print(f"{args.file}: {rfc_count} RFC1918, {non_rfc} non-RFC1918")
+
     else:
         print("ERROR: Either a file or a directory must be specified.")
         parser.print_help()