mirror of
https://github.com/Ekultek/Zeus-Scanner.git
synced 2026-03-11 08:55:51 +00:00
135 lines
4.3 KiB
Python
135 lines
4.3 KiB
Python
import os
|
|
import nmap
|
|
import json
|
|
import time
|
|
import socket
|
|
|
|
from lib.settings import (
|
|
logger,
|
|
set_color,
|
|
create_dir,
|
|
find_application
|
|
)
|
|
|
|
|
|
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",
|
|
ports=None):
|
|
self.ip = ip
|
|
self.verbose = verbose
|
|
self.pretty = pretty
|
|
self.dir = dirname
|
|
self.file = filename
|
|
self.ports = ports
|
|
|
|
def _get_all_info(self):
|
|
"""
|
|
get all the information from the scan
|
|
"""
|
|
scanned_data = self.NM.scan(self.ip, ports=self.ports)
|
|
if self.pretty:
|
|
scanned_data = json.dumps(scanned_data, indent=4, sort_keys=True)
|
|
return scanned_data
|
|
|
|
def send_to_file(self):
|
|
"""
|
|
send all the information to a JSON file for further use
|
|
"""
|
|
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(self._get_all_info())
|
|
return full_nmap_path
|
|
|
|
def show_open_ports(self, sep="-" * 30):
|
|
"""
|
|
outputs the current scan information
|
|
"""
|
|
logger.info(set_color("data found for IP '{}'...".format(self.ip)))
|
|
for host in self.NM.all_hosts():
|
|
if host:
|
|
print(
|
|
"{}\nScanned: {} ({})\nHost state: {}".format(
|
|
sep, self.ip, self.NM[self.ip].hostname(),
|
|
self.NM[self.ip].state()
|
|
)
|
|
)
|
|
else:
|
|
logger.warning(set_color(
|
|
"nothing found skipping...", level=30
|
|
))
|
|
for proto in self.NM[host].all_protocols():
|
|
print(
|
|
"Protocol: {}".format(proto)
|
|
)
|
|
oports = self.NM[host][proto].keys()
|
|
oports.sort()
|
|
for port in oports:
|
|
print(
|
|
"Port: {}\tStatus: {}".format(
|
|
port, self.NM[host][proto][port]["state"]
|
|
)
|
|
)
|
|
print(sep)
|
|
|
|
|
|
def find_nmap(item_name="nmap", given_search_path=None, verbose=False):
|
|
"""
|
|
find nmap on the users system if they do not specify a path for it or it is not in their PATH
|
|
"""
|
|
return find_application(item_name, given_search_path=given_search_path, verbose=verbose)
|
|
|
|
|
|
def perform_port_scan(url, ports=None, scanner=NmapHook, verbose=False, full_path=None, **kwargs):
|
|
"""
|
|
main function that will initalize the port scanning
|
|
"""
|
|
url = url.strip()
|
|
logger.info(set_color(
|
|
"attempting to find IP address for hostname '{}'...".format(url)
|
|
))
|
|
found_ip_address = socket.gethostbyname(url)
|
|
logger.info(set_color(
|
|
"found IP address for given URL -> '{}'...".format(found_ip_address)
|
|
))
|
|
if verbose:
|
|
logger.debug(set_color(
|
|
"checking for nmap on your system...", level=10
|
|
))
|
|
nmap_exists = find_nmap(verbose=verbose)
|
|
if nmap_exists:
|
|
if verbose:
|
|
logger.debug(set_color(
|
|
"nmap has been found under '{}'...".format(nmap_exists), level=10
|
|
))
|
|
logger.info(set_color(
|
|
"starting port scan on IP address '{}'...".format(found_ip_address)
|
|
))
|
|
try:
|
|
data = scanner(found_ip_address, ports=ports)
|
|
logger.warning(set_color(
|
|
"sleeping for 15 seconds to given nmap time to complete...", level=30
|
|
))
|
|
time.sleep(15)
|
|
data.show_open_ports()
|
|
file_path = data.send_to_file()
|
|
logger.info(set_color(
|
|
"port scan completed, saved to JSON file under '{}'...".format(file_path)
|
|
))
|
|
except Exception as e:
|
|
logger.exception(set_color(
|
|
"ran into exception '{}', cannot continue quitting...".format(e), level=50
|
|
))
|
|
pass
|
|
else:
|
|
logger.fatal(set_color(
|
|
"nmap was not found on your system, please install it...", level=50
|
|
))
|