PyNesca/modules/address_generation/Parser.py

import ipaddress
from core.prototypes.AbstractParser import AbstractParser


class Parser(AbstractParser):

    def parse_fields(self, ports:'port_field', ips:'address_field') -> {'ipv4_objects',
    'ports'}:
        result = dict()
        result['ports'] = self.parse_port_field(ports)
        result['ipv4_objects'] = self.get_all_addresses(ips)
        return result

    def parse_port_field(self, ports):
        """
        Parses ports from string, returns them as integers in the list.
        Handles non-existent ports and non-port values.
        """
        if ports:
            # Using set to avoid repetitions
            parsed = set()
            ports = ports.split(",")
            for port in ports:
                try:
                    # Input is in range form ("100-200"):
                    if '-' in port:
                        start, end = map(int, port.split('-'))
                        parsed.update(
                            [p for p in range(start, end + 1) if 65355 >= p > 0])
                    # Input is a single port ("80"):
                    else:
                        parsed.add(int(port))
                except ValueError as e:
                    # If we get any not integer just ignore it
                    pass
            return sorted(list(parsed))
        else:
            # Change to default ports from constant
            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]

    def parse_address_field(self, ips):
        """
        Parses ip input string, returns the generator over them.

        Supports next inputs:
            1) 1.2.3.4
            2) 192.168.0.0/24
            3) 1.2.3.4 - 5.6.7.8
        Any non-ip value will be ignored.
        """

        # A set to contain non repeating ip objects from ipaddress
        for ip_obj in self.get_all_addresses():
            # The object is just one ip, simply yield it:
            if isinstance(ip_obj, ipaddress.IPv4Address):
                yield ip_obj
            # The object is a network, yield every host in it:
            else:
                for host in ip_obj:
                    yield host

    def get_all_addresses(self, ips):
        ip_objects = set()
        inputs = [ip.strip() for ip in ips.split(',')]

        for input_ in inputs:
            try:
                # Input is in range form ("1.2.3.4 - 5.6.7.8"):
                if '-' in input_:
                    input_ips = input_.split('-')
                    ranges = set(
                    ipaddress.summarize_address_range(
                        *map(lambda x: ipaddress.IPv4Address(x.strip()), input_ips)
                        )
                    )
                    ip_objects.update(ranges)
                # Input is in CIDR form ("192.168.0.0/24"):
                elif '/' in input_:
                    network = ipaddress.ip_network(input_)
                    ip_objects.add(network)
                # Input is a single ip ("1.1.1.1"):
                else:
                    ip = ipaddress.ip_address(input_)
                    ip_objects.add(ip)
            except ValueError as e:
                print(e)
                # If we get any non-ip value just ignore it
                pass
            ip_objects = list(ip_objects)
        return ip_objects