From 346b52df9cfc4178a75baeb92c0ec5c51fef82bb Mon Sep 17 00:00:00 2001
From: Enemy Submarine <enemy-submarine@github.com>
Date: Thu, 24 Jan 2019 02:26:30 +0300
Subject: [PATCH] Rewritten ip/port parser

---
 Parser.py | 65 +++++++++++++++++++++++++++++++++++++++++--------------
 1 file changed, 49 insertions(+), 16 deletions(-)

diff --git a/Parser.py b/Parser.py
index e106048..3da1ff2 100644
--- a/Parser.py
+++ b/Parser.py
@@ -1,21 +1,54 @@
-import netaddr
+import ipaddress
 
 
-def getCIDRFromRanges(str_ranges):
-    str_ranges = str_ranges.replace(' ', '')
-    ranges = []
-    ips = []
-    splitted_ranges = str_ranges.split(",")
-    for i in splitted_ranges:
-        ranges.append(i.split("-"))
-    for i in ranges:
-        if len(ranges[ranges.index(i)]) == 1:
-            ips.append(netaddr.iprange_to_cidrs(i[0], i[0]))
+def getPortsFromString(ports):
+    # Converts ports from form 20-40,100-900,40000-70000
+    # It will automatically prune off non-existent ports (<1 >65535)
+    if ports is None:
+        return [21, 22, 23, 25, 80, 443, 110, 111, 135, 139, 445, 8080, 8443, 53, 143, 989, 990, 3306, 1080, 5554, 6667, 2222, 4444, 666, 6666, 1337, 2020, 31337] # Change to default ports from constant
+    else:
+        if "-" not in ports:
+            tports = ports.split(",")
+            print(tports)
         else:
-            ips.append(netaddr.iprange_to_cidrs(i[0], i[1]))
-    return ips
+            ports = ports.split(",")
+            tports = []
+            for port in ports:
+                if "-" not in port:
+                    tports.append(int(port))
+                else:
+                    # I made this one line because I wanted to
+                    tports.extend(
+                        list(range(int(port.split("-")[0]), int(port.split("-")[1]) + 1)))
+    ports = [int(n) for n in tports if int(n) > 0 and int(n) < 65536]
+    return ports
 
 
-def getPortsFromString(str_ports):
-    str_ports = str_ports.replace(" ", "")
-    return [str_ports.split(",")]
+def getCIDRFromRanges(ips):
+    ip_objects = set()
+    inputs = [ip.strip() for ip in ips.split(',')]
+
+    for input_ in inputs:
+        try:
+            if '-' in input_:
+                input_ips = input_.split('-')
+                ranges = {ipaddr for ipaddr in ipaddress.summarize_address_range(
+                    ipaddress.IPv4Address(input_ips[0]),
+                    ipaddress.IPv4Address(input_ips[1]))
+                               }
+                ip_objects.update(ranges)
+            elif '/' in input_:
+                network = ipaddress.ip_network(input_)
+                ip_objects.add(network)
+            else:
+                ip = ipaddress.ip_address(input_)
+                ip_objects.add(ip)
+        except ValueError as e:
+            print(e)
+
+    for ip_obj in ip_objects:
+        if not isinstance(ip_obj, ipaddress.IPv4Address):
+            for ip in ip_obj.hosts():
+                yield ip
+        else:
+            yield ip_obj