the ip address generator is ready, just need to tie it together

Author: Ekultek

lib/creation/ip_generator.py

@@ -0,0 +1,66 @@
+import socket
+import itertools
+
+from multiprocessing import Pool
+
+
+def generate_ip_range(selected_range):
+    """
+    generate an IP address range from each provided node.
+    for example `10.0.1-10.1-10` will return a generator
+    object that has IP `10.0.1.1 - 10.0.10.10` in it
+    """
+    octets = selected_range.split(".")
+    chunks = [map(int, octet.split("-")) for octet in octets]
+    ranges = [range(c[0], c[1] + 1) if len(c) == 2 else c for c in chunks]
+    for address in itertools.product(*ranges):
+        yield ".".join(map(str, address))
+
+
+def check_ip_alive(ip):
+    """
+    efficiently check if an IP address is alive or not
+    by using the socket.gethostbyaddr function
+    """
+    def is_valid_ip(ip):
+        try:
+            socket.inet_aton(ip)
+            return True
+        except:
+            return False
+
+    try:
+        if not is_valid_ip(ip):
+            return False
+        else:
+            return socket.gethostbyaddr(ip)
+    except socket.herror:
+        return False
+
+
+def check_ip_wrapper(generated_ips, limit=350):
+    """
+    multiprocess the check_ip_alive function in order
+    to proces a large amount of IP addresses quickly
+    """
+    alive_ips = []
+    ips_to_use = []
+    i = 0
+    proc_pool = Pool(processes=35)
+
+    for ip in generated_ips:
+        ips_to_use.append(ip)
+        i += 1
+        if i == limit:
+            break
+    for ip in ips_to_use:
+        try:
+            result = proc_pool.apply_async(check_ip_alive, args=(ip,)).get()
+            if not result:
+                pass
+            else:
+                alive_ips.append(ip)
+        except Exception:
+            pass
+    proc_pool.close()
+    return alive_ips