enum2root.py

#!/usr/bin/python3
import nmap3
import subprocess
import sys
import xml.etree.ElementTree as ET
from pathlib import Path
import logging

logging.basicConfig(
    format='%(asctime)s %(levelname)-8s %(message)s',
    level=logging.INFO,
    datefmt='%H:%M:%S')

##prep##
def mkdirs():
    Path("nmap/basic").mkdir(parents=True, exist_ok=True)
    Path("nmap/full").mkdir(parents=True, exist_ok=True)

##masscan##
def masscanFull(IP):
    logging.info("Scanning all ports with masscan")
    scan = subprocess.run("masscan -p1-65535,U:1-65535 {} --rate=1000 -oX -".format(IP), stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell = True)
    out = scan.stdout.decode()
    return out #XML Output

def nmapFullDetailed(IP,ports):
    logging.info("Starting a detailed nmap scan with masscan results")
    nmap = nmap3.Nmap()
    scan = nmap.nmap_version_detection(str(IP), args="-sC -p {} -T5 --min-rate=1000 --max-retries=2".format(ports))
    return scan #JSON Object

def parseXML(XML):
    logging.info("parsing XML files")
    tree = ET.ElementTree(ET.fromstring(XML))
    root = tree.getroot()
    ports = [port.get('portid') for port in root.findall('.//port')]
    ports_string = ",".join(ports)
    return ports_string # A list with the ports ex: ['80', '22', '443']

def portEnum(IP):
    scan = masscanFull(IP)
    ports = parseXML(scan)
    final = nmapFullDetailed(IP,ports)
    return final #JSON Output with all the needed details

print(portEnum(sys.argv[1]))