Zeus-Scanner/lib/attacks/nmap_scan/__init__.py

import json
import os
import socket
import shlex
import subprocess

import nmap

import lib.core.settings
from var.auto_issue.github import request_issue_creation


class NmapHook(object):

    """
    Nmap API hook, uses python, must have nmap on your system
    """

    NM = nmap.PortScanner()

    def __init__(self, ip, verbose=False, pretty=True,
                 dirname="{}/log/scanner-log".format(os.getcwd()), filename="nmap_scan-results-{}.json",
                 opts=None):
        self.ip = ip
        self.verbose = verbose
        self.pretty = pretty
        self.dir = dirname
        self.file = filename
        if opts is None:
            self.opts = ""
        else:
            self.opts = " ".join(opts)

    def _get_all_info(self):
        """
        get all the information from the scan
        """
        scanned_data = self.NM.scan(self.ip, arguments=self.opts)
        if self.pretty:
            scanned_data = json.dumps(scanned_data, indent=4, sort_keys=True)
        return scanned_data

    def send_to_file(self, data):
        """
        send all the information to a JSON file for further use
        """
        lib.core.settings.create_dir(self.dir)
        full_nmap_path = "{}/{}".format(self.dir, self.file.format(self.ip))
        with open(full_nmap_path, "a+") as log:
            log.write(data)
        return full_nmap_path

    def show_open_ports(self, json_data, sep="-" * 30):
        """
        outputs the current scan information
        """
        # have to create a spacer or the output comes out funky..
        spacer_data = {4: " " * 8, 6: " " * 6, 8: " " * 4}
        lib.core.settings.logger.info(lib.core.settings.set_color("finding data for IP '{}'...".format(self.ip)))
        json_data = json.loads(json_data)["scan"]
        print(
            "{}\nScanned: {} ({})\tStatus: {}\nProtocol: {}\n".format(
                sep, self.ip,
                json_data[self.ip]["hostnames"][0]["name"],
                json_data[self.ip]["status"]["state"],
                "TCP"
            )
        )
        oports = json_data[self.ip]["tcp"].keys()
        oports.sort()
        for port in oports:
            port_status = json_data[self.ip]["tcp"][port]["state"]
            # output the found port information..
            print(
                "Port: {}\tStatus: {}{}Type: {}".format(
                    port, json_data[self.ip]["tcp"][port]["state"],
                    spacer_data[len(port_status)],
                    json_data[self.ip]["tcp"][port]["name"]
                )
            )
        print("{}".format(sep))


def find_nmap(item_name="nmap"):
    """
    find nmap on the users system if they do not specify a path for it or it is not in their PATH
    """
    return lib.core.settings.find_application(item_name)


def perform_port_scan(url, scanner=NmapHook, verbose=False, opts=None, **kwargs):
    """
    main function that will initalize the port scanning
    """
    url = url.strip()
    lib.core.settings.logger.info(lib.core.settings.set_color(
        "attempting to find IP address for hostname '{}'...".format(url)
    ))
    found_ip_address = socket.gethostbyname(url)
    lib.core.settings.logger.info(lib.core.settings.set_color(
        "found IP address for given URL -> '{}'...".format(found_ip_address)
    ))
    if verbose:
        lib.core.settings.logger.debug(lib.core.settings.set_color(
            "checking for nmap on your system...", level=10
        ))
    nmap_exists = "".join(find_nmap())
    if nmap_exists:
        if verbose:
            lib.core.settings.logger.debug(lib.core.settings.set_color(
                "nmap has been found under '{}'...".format(nmap_exists), level=10
            ))
        lib.core.settings.logger.info(lib.core.settings.set_color(
            "starting port scan on IP address '{}'...".format(found_ip_address)
        ))
        try:
            data = scanner(found_ip_address, opts=opts)
            json_data = data._get_all_info()
            data.show_open_ports(json_data)
            file_path = data.send_to_file(json_data)
            lib.core.settings.logger.info(lib.core.settings.set_color(
                "port scan completed, all data saved to JSON file under '{}'...".format(file_path)
            ))
        except KeyError:
            lib.core.settings.logger.fatal(lib.core.settings.set_color(
                "no port information found for '{}({})'...".format(
                    url, found_ip_address
                ), level=50
            ))
        except Exception as e:
            lib.core.settings.logger.exception(lib.core.settings.set_color(
                "ran into exception '{}', cannot continue quitting...".format(e), level=50
            ))
            request_issue_creation()
            pass
    else:
        lib.core.settings.logger.fatal(lib.core.settings.set_color(
            "nmap was not found on your system...", level=50
        ))
        question = lib.core.settings.prompt(
            "would you like to automatically install it", opts="yN"
        )
        if question.lower().startswith("y"):
            install_nmap_command = shlex.split("sudo sh {}".format(lib.core.settings.NMAP_INSTALLER_TOOL))
            subprocess.call(install_nmap_command)
            lib.core.settings.logger.info(lib.core.settings.set_color(
                "nmap has been successfully installed, re-running..."
            ))
            perform_port_scan(url, verbose=verbose, opts=opts)
        else:
            lib.core.settings.logger.fatal(lib.core.settings.set_color(
                "nmap is not installed, please install it in order to continue...", level=50
            ))