mirror of
https://github.com/thecyberlearn/quantum-ai-v3.git
synced 2026-08-18 09:52:58 +00:00
238 lines
11 KiB
Python
238 lines
11 KiB
Python
"""
|
|
Security middleware for enhanced security headers and CSP.
|
|
"""
|
|
|
|
from django.conf import settings
|
|
from django.utils import timezone
|
|
import logging
|
|
|
|
logger = logging.getLogger('core.security')
|
|
|
|
|
|
class SecurityHeadersMiddleware:
|
|
"""
|
|
Middleware to add comprehensive security headers to all responses.
|
|
Implements Content Security Policy, security headers, and security monitoring.
|
|
"""
|
|
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
response = self.get_response(request)
|
|
|
|
# Content Security Policy
|
|
if not settings.DEBUG:
|
|
# Check if this is a direct access agent page that needs external frames
|
|
is_direct_access_page = (
|
|
'/agents/' in request.path and
|
|
request.path.count('/') >= 3 and
|
|
not request.path.endswith('/api/execute/')
|
|
)
|
|
|
|
if is_direct_access_page:
|
|
# Relaxed CSP for direct access agent pages (external forms)
|
|
csp_policy = (
|
|
"default-src 'self'; "
|
|
"script-src 'self' 'unsafe-inline' https://js.stripe.com https://checkout.stripe.com https://form.jotform.com https://www.jotform.com https://agent.jotform.com https://cdn.jotfor.ms; "
|
|
"style-src 'self' 'unsafe-inline' https://fonts.googleapis.com https://form.jotform.com https://www.jotform.com https://agent.jotform.com https://cdn.jotfor.ms; "
|
|
"font-src 'self' https://fonts.gstatic.com https://form.jotform.com https://www.jotform.com https://agent.jotform.com https://cdn.jotfor.ms; "
|
|
"img-src 'self' data: https: blob:; "
|
|
"connect-src 'self' https: wss: ws:; "
|
|
"frame-src 'self' https: http:; "
|
|
"child-src 'self' https: http:; "
|
|
"object-src 'none'; "
|
|
"base-uri 'self'; "
|
|
"form-action 'self' https: http:; "
|
|
"frame-ancestors 'none';"
|
|
)
|
|
else:
|
|
# Production CSP - Strict security for other pages
|
|
csp_policy = (
|
|
"default-src 'self'; "
|
|
"script-src 'self' 'unsafe-inline' https://js.stripe.com https://checkout.stripe.com; "
|
|
"style-src 'self' 'unsafe-inline' https://fonts.googleapis.com; "
|
|
"font-src 'self' https://fonts.gstatic.com; "
|
|
"img-src 'self' data: https: blob:; "
|
|
"connect-src 'self' https://api.stripe.com https://checkout.stripe.com; "
|
|
"frame-src 'self' https://js.stripe.com https://hooks.stripe.com; "
|
|
"object-src 'none'; "
|
|
"base-uri 'self'; "
|
|
"form-action 'self'; "
|
|
"frame-ancestors 'none'; "
|
|
"upgrade-insecure-requests;"
|
|
)
|
|
else:
|
|
# Development CSP - More permissive for development tools
|
|
csp_policy = (
|
|
"default-src 'self' 'unsafe-inline' 'unsafe-eval'; "
|
|
"script-src 'self' 'unsafe-inline' 'unsafe-eval' https://js.stripe.com; "
|
|
"style-src 'self' 'unsafe-inline' https://fonts.googleapis.com; "
|
|
"font-src 'self' https://fonts.gstatic.com; "
|
|
"img-src 'self' data: https: blob:; "
|
|
"connect-src 'self' ws: wss: https:; "
|
|
"frame-src 'self' https: http:;"
|
|
)
|
|
|
|
response['Content-Security-Policy'] = csp_policy
|
|
|
|
# Additional Security Headers
|
|
response['X-Content-Type-Options'] = 'nosniff'
|
|
response['X-XSS-Protection'] = '1; mode=block'
|
|
response['Referrer-Policy'] = 'strict-origin-when-cross-origin'
|
|
response['Permissions-Policy'] = (
|
|
'geolocation=(), microphone=(), camera=(), '
|
|
'payment=(self "https://js.stripe.com"), '
|
|
'usb=(), magnetometer=(), gyroscope=(), accelerometer=()'
|
|
)
|
|
|
|
# X-Frame-Options handling
|
|
if request.path.endswith('/display/') and '/agents/' in request.path:
|
|
# Allow direct access agent display pages to be framed (they contain external iframes)
|
|
response['X-Frame-Options'] = 'SAMEORIGIN'
|
|
else:
|
|
# Deny framing for all other pages
|
|
response['X-Frame-Options'] = 'DENY'
|
|
|
|
# Security for critical pages
|
|
if request.path.startswith('/admin/') or request.path.startswith('/wallet/'):
|
|
response['X-Frame-Options'] = 'DENY'
|
|
response['Cache-Control'] = 'no-store, no-cache, must-revalidate, max-age=0'
|
|
response['Pragma'] = 'no-cache'
|
|
response['Expires'] = '0'
|
|
|
|
# Log security events for monitoring
|
|
if hasattr(request, 'user') and request.user.is_authenticated:
|
|
# Log administrative actions
|
|
if request.path.startswith('/admin/') and request.method == 'POST':
|
|
logger.info(f"Admin action by user {request.user.id} from IP {request.META.get('REMOTE_ADDR')}")
|
|
|
|
# Log sensitive financial operations
|
|
if request.path.startswith('/wallet/') and request.method == 'POST':
|
|
logger.info(f"Wallet operation by user {request.user.id} from IP {request.META.get('REMOTE_ADDR')}")
|
|
|
|
# Log agent executions
|
|
if request.path.startswith('/agents/api/execute') and request.method == 'POST':
|
|
logger.info(f"Agent execution by user {request.user.id} from IP {request.META.get('REMOTE_ADDR')}")
|
|
|
|
# Log authentication failures
|
|
if hasattr(request, 'user') and not request.user.is_authenticated:
|
|
if request.path.startswith('/auth/') and request.method == 'POST':
|
|
logger.warning(f"Failed authentication attempt from IP {request.META.get('REMOTE_ADDR')}")
|
|
|
|
return response
|
|
|
|
|
|
class SecurityMonitoringMiddleware:
|
|
"""
|
|
Middleware for security event monitoring and threat detection.
|
|
"""
|
|
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
self.suspicious_patterns = [
|
|
'.env', 'wp-admin', 'phpmyadmin', '../', '<script', 'SELECT * FROM',
|
|
'UNION SELECT', 'DROP TABLE', 'INSERT INTO', 'DELETE FROM',
|
|
'etc/passwd', 'windows/system32', '../../../../', '../../../',
|
|
'cmd.exe', '/bin/bash', 'eval(', 'exec(', 'system(',
|
|
'base64_decode', 'shell_exec', 'file_get_contents',
|
|
'fopen(', 'include(', 'require(', 'curl_exec',
|
|
'<?php', '<%', '<jsp:', 'javascript:', 'vbscript:',
|
|
'onload=', 'onerror=', 'onclick=', 'onfocus=',
|
|
'document.cookie', 'document.location', 'window.location'
|
|
]
|
|
|
|
# Track suspicious IPs for rate limiting
|
|
self.suspicious_ips = set()
|
|
self.failed_attempts = {}
|
|
|
|
def __call__(self, request):
|
|
# Check for suspicious patterns in URL and parameters
|
|
self._check_suspicious_activity(request)
|
|
|
|
response = self.get_response(request)
|
|
|
|
# Log failed authentication attempts and track suspicious IPs
|
|
if response.status_code == 401 or response.status_code == 403:
|
|
self._log_security_event(request, 'auth_failure', f"Status: {response.status_code}")
|
|
self._track_failed_attempt(request)
|
|
|
|
# Log rate limit violations
|
|
if hasattr(response, 'status_code') and response.status_code == 429:
|
|
self._log_security_event(request, 'rate_limit_exceeded', f"Path: {request.path}")
|
|
|
|
# Log suspicious response patterns
|
|
if response.status_code == 500:
|
|
self._log_security_event(request, 'server_error', f"Path: {request.path}")
|
|
|
|
return response
|
|
|
|
def _check_suspicious_activity(self, request):
|
|
"""Check for suspicious patterns in requests"""
|
|
full_path = request.get_full_path()
|
|
|
|
# Check URL for suspicious patterns
|
|
for pattern in self.suspicious_patterns:
|
|
if pattern.lower() in full_path.lower():
|
|
self._log_security_event(
|
|
request,
|
|
'suspicious_request',
|
|
f"Pattern: {pattern}, Path: {full_path}"
|
|
)
|
|
break
|
|
|
|
# Check for potential SQLi in parameters
|
|
if request.GET:
|
|
for key, value in request.GET.items():
|
|
for pattern in ['SELECT', 'UNION', 'DROP', 'INSERT', 'DELETE']:
|
|
if pattern in str(value).upper():
|
|
self._log_security_event(
|
|
request,
|
|
'potential_sqli',
|
|
f"Parameter: {key}, Value: {value[:100]}"
|
|
)
|
|
break
|
|
|
|
def _track_failed_attempt(self, request):
|
|
"""Track failed authentication attempts by IP"""
|
|
ip = request.META.get('REMOTE_ADDR', 'unknown')
|
|
|
|
if ip not in self.failed_attempts:
|
|
self.failed_attempts[ip] = {'count': 0, 'last_attempt': None}
|
|
|
|
self.failed_attempts[ip]['count'] += 1
|
|
self.failed_attempts[ip]['last_attempt'] = timezone.now()
|
|
|
|
# Mark IP as suspicious after 5 failed attempts
|
|
if self.failed_attempts[ip]['count'] >= 5:
|
|
self.suspicious_ips.add(ip)
|
|
self._log_security_event(
|
|
request,
|
|
'suspicious_ip_detected',
|
|
f"IP {ip} marked suspicious after {self.failed_attempts[ip]['count']} failed attempts"
|
|
)
|
|
|
|
def _log_security_event(self, request, event_type, details):
|
|
"""Log security events for monitoring"""
|
|
ip = request.META.get('REMOTE_ADDR', 'unknown')
|
|
user_id = request.user.id if hasattr(request, 'user') and request.user.is_authenticated else 'anonymous'
|
|
user_agent = request.META.get('HTTP_USER_AGENT', '')[:100]
|
|
|
|
# Enhanced logging with more context
|
|
logger.warning(
|
|
f"Security Event: {event_type} - "
|
|
f"IP: {ip} - "
|
|
f"User: {user_id} - "
|
|
f"Path: {request.path} - "
|
|
f"Method: {request.method} - "
|
|
f"UA: {user_agent} - "
|
|
f"Referer: {request.META.get('HTTP_REFERER', 'none')[:100]} - "
|
|
f"Details: {details}"
|
|
)
|
|
|
|
# Additional context for critical events
|
|
if event_type in ['suspicious_request', 'potential_sqli', 'suspicious_ip_detected']:
|
|
logger.critical(
|
|
f"CRITICAL SECURITY ALERT: {event_type} - "
|
|
f"IP: {ip} - User: {user_id} - {details}"
|
|
) |