GuardianUnivalle-Benito-Yucra 0.1.45__tar.gz → 0.1.46__tar.gz

{guardianunivalle_benito_yucra-0.1.45 → guardianunivalle_benito_yucra-0.1.46}/GuardianUnivalle_Benito_Yucra/detectores/detector_dos.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 import time
 import logging
 import json
+from collections import deque
 from typing import Dict, List
 from django.conf import settings
 from django.utils.deprecation import MiddlewareMixin
@@ -9,7 +10,7 @@ from ..mitigacion.limitador_peticion import limitar_peticion
 from ..auditoria.registro_auditoria import registrar_evento
 
 # =====================================================
-# ===            CONFIGURACIÓN DEL LOGGER            ===
+# === CONFIGURACIÓN DEL LOGGER ===
 # =====================================================
 logger = logging.getLogger("dosdefense")
 logger.setLevel(logging.INFO)
@@ -19,108 +20,148 @@ if not logger.handlers:
     logger.addHandler(handler)
 
 # =====================================================
-# ===       PARÁMETROS DE CONFIGURACIÓN BASE         ===
+# === PARÁMETROS DE CONFIGURACIÓN BASE ===
 # =====================================================
 LIMITE_PETICIONES = getattr(settings, "DOS_LIMITE_PETICIONES", 100)  # por minuto
 VENTANA_SEGUNDOS = getattr(settings, "DOS_VENTANA_SEGUNDOS", 60)
 PESO_DOS = getattr(settings, "DOS_PESO", 0.6)
+LIMITE_ENDPOINTS_DISTINTOS = getattr(settings, "DOS_LIMITE_ENDPOINTS", 50)
+TRUSTED_IPS = getattr(settings, "DOS_TRUSTED_IPS", [])
 
 # =====================================================
-# ===  REGISTRO TEMPORAL DE SOLICITUDES POR IP      ===
+# === REGISTRO TEMPORAL EN MEMORIA ===
 # =====================================================
-# En producción se recomienda Redis o Memcached
-REGISTRO_SOLICITUDES: Dict[str, List[float]] = {}
-
+# Estructura: { ip: deque([timestamps]), ... }
+# deque es eficiente para ventanas deslizantes
+REGISTRO_SOLICITUDES: Dict[str, deque] = {}
+REGISTRO_ENDPOINTS: Dict[str, set] = {}
 
 # =====================================================
-# ===  FUNCIONES AUXILIARES                         ===
+# === FUNCIONES AUXILIARES ===
 # =====================================================
 def get_client_ip(request) -> str:
     """Obtiene la IP real del cliente (considera proxies)."""
     x_forwarded_for = request.META.get("HTTP_X_FORWARDED_FOR")
     if x_forwarded_for:
         return x_forwarded_for.split(",")[0].strip()
-    return request.META.get("REMOTE_ADDR", "")
+    return request.META.get("REMOTE_ADDR", "") or "0.0.0.0"
+
+
+def limpiar_registro_global():
+    """Elimina IPs sin actividad reciente para evitar uso excesivo de memoria."""
+    ahora = time.time()
+    expiracion = VENTANA_SEGUNDOS * 2  # doble ventana
+    inactivas = [
+        ip for ip, tiempos in REGISTRO_SOLICITUDES.items()
+        if tiempos and ahora - tiempos[-1] > expiracion
+    ]
+    for ip in inactivas:
+        REGISTRO_SOLICITUDES.pop(ip, None)
+        REGISTRO_ENDPOINTS.pop(ip, None)
 
 
 def limpiar_registro(ip: str):
     """Limpia peticiones antiguas fuera de la ventana de tiempo."""
     ahora = time.time()
-    REGISTRO_SOLICITUDES[ip] = [
-        t for t in REGISTRO_SOLICITUDES.get(ip, []) if ahora - t < VENTANA_SEGUNDOS
-    ]
+    if ip not in REGISTRO_SOLICITUDES:
+        REGISTRO_SOLICITUDES[ip] = deque()
+    tiempos = REGISTRO_SOLICITUDES[ip]
+    while tiempos and ahora - tiempos[0] > VENTANA_SEGUNDOS:
+        tiempos.popleft()
 
 
-def calcular_nivel_amenaza_dos(
-    tasa_peticion: int, limite: int = LIMITE_PETICIONES
-) -> float:
+def calcular_nivel_amenaza_dos(tasa_peticion: int, limite: int = LIMITE_PETICIONES) -> float:
     """
-    Calcula la puntuación de amenaza DoS basada en el peso configurado.
+    Calcula la puntuación de amenaza DoS.
     Fórmula: S_dos = w_dos * (tasa_peticion / limite)
     """
-    proporcion = tasa_peticion / limite
-    s_dos = PESO_DOS * proporcion
+    proporcion = tasa_peticion / max(limite, 1)
+    s_dos = PESO_DOS * min(proporcion, 2.0)
     return round(min(s_dos, 1.0), 3)
 
 
 def detectar_dos(ip: str, tasa_peticion: int, limite: int = LIMITE_PETICIONES) -> bool:
     """Evalúa si la tasa de peticiones excede el umbral permitido."""
     if tasa_peticion > limite:
-        # Registrar evento en auditoría
         registrar_evento(
             tipo="DoS",
             descripcion=f"Alta tasa de peticiones desde {ip}: {tasa_peticion} req/min (límite {limite})",
             severidad="ALTA",
         )
-        # Mitigación: pasar un usuario genérico
         limitar_peticion(usuario_id="anonimo")
         return True
+    elif tasa_peticion > limite * 0.75:
+        # Umbral de advertencia
+        registrar_evento(
+            tipo="DoS",
+            descripcion=f"Posible saturación desde {ip}: {tasa_peticion} req/min",
+            severidad="MEDIA",
+        )
     return False
 
 
+def analizar_headers(user_agent: str, referer: str) -> List[str]:
+    """Detecta patrones de agentes o cabeceras sospechosas."""
+    sospechas = []
+    if not user_agent or len(user_agent) < 10:
+        sospechas.append("User-Agent vacío o anómalo")
+    if "curl" in user_agent.lower() or "python" in user_agent.lower():
+        sospechas.append("User-Agent indica script o bot")
+    if referer and any(palabra in referer.lower() for palabra in ["attack", "scan", "bot"]):
+        sospechas.append("Referer sospechoso")
+    return sospechas
+
+
 # =====================================================
-# ===      MIDDLEWARE DE DETECCIÓN DE DoS           ===
+# === MIDDLEWARE PRINCIPAL DE DEFENSA DoS ===
 # =====================================================
 class DOSDefenseMiddleware(MiddlewareMixin):
     """
-    Middleware para detección y registro de ataques DoS.
-    - Captura IP, agente y cabeceras sospechosas.
-    - Evalúa la frecuencia de peticiones por IP.
-    - Marca request.dos_attack_info con información del intento.
+    Middleware de detección y registro de ataques DoS.
     """
 
     def process_request(self, request):
+        limpiar_registro_global()
+
         client_ip = get_client_ip(request)
+        if client_ip in TRUSTED_IPS:
+            return None
+
         user_agent = request.META.get("HTTP_USER_AGENT", "Desconocido")
         referer = request.META.get("HTTP_REFERER", "")
         path = request.path
 
-        # Limpieza de registro anterior
+        # Registrar endpoint accedido
+        REGISTRO_ENDPOINTS.setdefault(client_ip, set()).add(path)
+
+        # Mantener ventana deslizante
         limpiar_registro(client_ip)
+        REGISTRO_SOLICITUDES[client_ip].append(time.time())
 
-        # Registrar nueva petición
-        REGISTRO_SOLICITUDES.setdefault(client_ip, []).append(time.time())
         tasa = len(REGISTRO_SOLICITUDES[client_ip])
         nivel = calcular_nivel_amenaza_dos(tasa)
         es_dos = detectar_dos(client_ip, tasa)
 
-        if es_dos:
-            descripcion = [
-                f"Tasa de {tasa} req/min excede límite {LIMITE_PETICIONES}",
-                f"User-Agent: {user_agent}",
-                f"Referer: {referer or 'N/A'}",
-                f"Ruta: {path}",
-            ]
+        descripcion = []
+        sospechas_headers = analizar_headers(user_agent, referer)
+        if sospechas_headers:
+            descripcion.extend(sospechas_headers)
+
+        # Verificar exceso de endpoints distintos (indicativo de escaneo)
+        if len(REGISTRO_ENDPOINTS[client_ip]) > LIMITE_ENDPOINTS_DISTINTOS:
+            descripcion.append("Número anormal de endpoints distintos accedidos")
+
+        if es_dos or descripcion:
+            descripcion.insert(0, f"Tasa actual: {tasa} req/min (nivel {nivel:.2f})")
+            descripcion.append(f"Ruta: {path}")
 
-            # Log profesional
             logger.warning(
-                "DoS detectado desde IP %s: %s ; nivel: %.2f",
+                "DoS detectado o sospechoso desde IP %s: %s ; nivel: %.2f",
                 client_ip,
                 descripcion,
                 nivel,
             )
 
-            # Enviar a sistema de auditoría
             request.dos_attack_info = {
                 "ip": client_ip,
                 "tipos": ["DoS"],

{guardianunivalle_benito_yucra-0.1.45 → guardianunivalle_benito_yucra-0.1.46}/GuardianUnivalle_Benito_Yucra.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GuardianUnivalle-Benito-Yucra
-Version: 0.1.45
+Version: 0.1.46
 Summary: Middleware y detectores de seguridad (SQLi, XSS, CSRF, DoS, Keylogger) para Django/Flask
 Author-email: Andres Benito Calle Yucra <benitoandrescalle035@gmail.com>
 License: MIT

{guardianunivalle_benito_yucra-0.1.45 → guardianunivalle_benito_yucra-0.1.46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: GuardianUnivalle-Benito-Yucra
-Version: 0.1.45
+Version: 0.1.46
 Summary: Middleware y detectores de seguridad (SQLi, XSS, CSRF, DoS, Keylogger) para Django/Flask
 Author-email: Andres Benito Calle Yucra <benitoandrescalle035@gmail.com>
 License: MIT

{guardianunivalle_benito_yucra-0.1.45 → guardianunivalle_benito_yucra-0.1.46}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "GuardianUnivalle-Benito-Yucra"  # usar mayúsculas consistente
-version = "0.1.45"
+version = "0.1.46"
 description = "Middleware y detectores de seguridad (SQLi, XSS, CSRF, DoS, Keylogger) para Django/Flask"
 authors = [
     { name = "Andres Benito Calle Yucra", email = "benitoandrescalle035@gmail.com" }