aiptx 2.0.2__py3-none-any.whl

aipt_v2/tools/scanners/config.py

@@ -0,0 +1,156 @@
+"""
+AIPT Scanner Configuration - Centralized Scanner Settings
+
+This file contains all configuration for external security scanners.
+Modify these settings to connect to different scanner instances.
+"""
+
+import os
+from dataclasses import dataclass
+from typing import Optional
+
+
+# ==================== Server Configuration ====================
+
+# Remote Scanner Server
+SCANNER_SERVER_IP = os.getenv("AIPT_SCANNER_IP", "13.127.28.41")
+
+# Port Configuration
+ACUNETIX_PORT = int(os.getenv("AIPT_ACUNETIX_PORT", "3443"))
+BURP_PORT = int(os.getenv("AIPT_BURP_PORT", "1337"))
+
+# Full URLs
+ACUNETIX_URL = os.getenv("AIPT_ACUNETIX_URL", f"https://{SCANNER_SERVER_IP}:{ACUNETIX_PORT}")
+BURP_URL = os.getenv("AIPT_BURP_URL", f"http://{SCANNER_SERVER_IP}:{BURP_PORT}/v0.1")
+
+
+# ==================== Acunetix Configuration ====================
+
+@dataclass
+class AcunetixSettings:
+    """Acunetix scanner settings."""
+    base_url: str = ACUNETIX_URL
+    api_key: str = os.getenv(
+        "AIPT_ACUNETIX_API_KEY",
+        "1986ad8c0a5b3df4d7028d5f3c06e936c83ef0a486ef74537812989cff1a41a7c"
+    )
+    verify_ssl: bool = False
+    timeout: int = 30
+
+    # Default scan settings
+    default_profile: str = "full"  # full, high_risk, xss, sqli, crawl, malware
+    default_criticality: int = 10  # 0-30 (10=normal, 30=critical)
+
+    # Rate limiting
+    max_concurrent_scans: int = 5
+    poll_interval: int = 30  # seconds
+
+
+# ==================== Burp Suite Configuration ====================
+
+@dataclass
+class BurpSettings:
+    """Burp Suite scanner settings."""
+    base_url: str = BURP_URL
+    api_key: str = os.getenv("AIPT_BURP_API_KEY", "t7thBWbImyiP8SA9hojkiFhq9QbHqlcm")
+    verify_ssl: bool = False
+    timeout: int = 30
+
+    # Default scan settings
+    default_config: Optional[str] = None  # Scan configuration ID
+
+    # Rate limiting
+    max_concurrent_scans: int = 3
+    poll_interval: int = 30  # seconds
+
+
+# ==================== Global Settings ====================
+
+@dataclass
+class ScannerSettings:
+    """Global scanner settings."""
+    # Timeouts
+    scan_timeout: int = 3600  # 1 hour max per scan
+    connection_timeout: int = 30
+
+    # Output
+    save_reports: bool = True
+    report_dir: str = "reports/scanners"
+    report_format: str = "html"  # html, pdf, xml
+
+    # Finding thresholds
+    min_severity: str = "info"  # info, low, medium, high, critical
+    deduplicate_findings: bool = True
+
+    # Retry settings
+    max_retries: int = 3
+    retry_delay: int = 5  # seconds
+
+
+# ==================== Default Instances ====================
+
+ACUNETIX = AcunetixSettings()
+BURP = BurpSettings()
+SCANNER = ScannerSettings()
+
+
+# ==================== Configuration Helpers ====================
+
+def get_acunetix_config() -> dict:
+    """Get Acunetix configuration as dict for tool initialization."""
+    return {
+        "base_url": ACUNETIX.base_url,
+        "api_key": ACUNETIX.api_key,
+        "verify_ssl": ACUNETIX.verify_ssl,
+        "timeout": ACUNETIX.timeout
+    }
+
+
+def get_burp_config() -> dict:
+    """Get Burp Suite configuration as dict for tool initialization."""
+    return {
+        "base_url": BURP.base_url,
+        "api_key": BURP.api_key,
+        "verify_ssl": BURP.verify_ssl,
+        "timeout": BURP.timeout
+    }
+
+
+def print_config():
+    """Print current scanner configuration."""
+    print("=" * 60)
+    print("AIPT Scanner Configuration")
+    print("=" * 60)
+    print(f"\nServer IP: {SCANNER_SERVER_IP}")
+    print(f"\nAcunetix:")
+    print(f"  URL:     {ACUNETIX.base_url}")
+    print(f"  API Key: {ACUNETIX.api_key[:20]}..." if ACUNETIX.api_key else "  API Key: Not set")
+    print(f"  SSL:     {ACUNETIX.verify_ssl}")
+    print(f"\nBurp Suite:")
+    print(f"  URL:     {BURP.base_url}")
+    print(f"  API Key: {BURP.api_key[:20]}..." if BURP.api_key else "  API Key: Not set")
+    print(f"  SSL:     {BURP.verify_ssl}")
+    print("=" * 60)
+
+
+# ==================== Environment Variable Reference ====================
+"""
+Environment variables for configuration override:
+
+    AIPT_SCANNER_IP         - Scanner server IP (default: 13.127.28.41)
+    AIPT_ACUNETIX_PORT      - Acunetix port (default: 3443)
+    AIPT_BURP_PORT          - Burp port (default: 1337)
+    AIPT_ACUNETIX_URL       - Full Acunetix URL (overrides IP+port)
+    AIPT_BURP_URL           - Full Burp URL (overrides IP+port)
+    AIPT_ACUNETIX_API_KEY   - Acunetix API key
+    AIPT_BURP_API_KEY       - Burp Suite API key
+
+Example:
+    export AIPT_SCANNER_IP="192.168.1.100"
+    export AIPT_ACUNETIX_API_KEY="your-api-key-here"
+    export AIPT_BURP_API_KEY="your-burp-api-key"
+"""
+
+
+if __name__ == "__main__":
+    print_config()