main.py

import yaml
from core.recon import run_nmap
from core.fetcher import fetch_exploit
from core.detector import detect_type
from core.generator import generate_http
from core.runner import run_exploit
from core.report import save_report


def load_config():
    with open("config.yaml") as f:
        return yaml.safe_load(f)


def main():

    config = load_config()

    target = input("Target IP: ")

    exploits = []

    if config["recon"]:
        exploits = run_nmap(target)

    for exploit in exploits:

        file = fetch_exploit(exploit)

        if not file:
            continue

        etype = detect_type(file)

        if etype == "http":

            exploit_file = generate_http(file)

            if config["run"]:
                run_exploit(exploit_file)

    if config["report"]:
        save_report(target, exploits)


if __name__ == "__main__":
    main()