写一个用于CTF比赛的WAF通用防御脚本

在CTF比赛中，Web应用程序防火墙(WAF)脚本主要用于检测并阻止恶意攻击，保护Web服务免受常见的Web安全威胁，如SQL注入、跨站脚本(XSS)、CSRF等。下面是一个简单的通用防御脚本概念示例，使用Python编写，但它实际应用中会更复杂，需要结合OWASP Top 10规范：

import re
from urllib.parse import urlparse

# 定义黑名单关键词和模式
whitelist = ['example.com', 'login']
sql_injection_patterns = [r'\b(?:union|select|insert)\b', r'\b\w+@\w+\.\w+\b'] # 示例SQL关键字和标识符
xss_cleaning_regex = r'<[^>]*script[^>]*>|<\/script>|<[^>]*onload[^>]*>|<[^>]*onmouseout[^>]*>|javascript:'

def is_waf_bypass(url, request_data):
    # 检查URL是否在白名单内
    if not any(urlparse(url).netloc in w for w in whitelist):
        return True

    # SQL注入检测
    for pattern in sql_injection_patterns:
        if pattern in request_data or pattern in url:
            return False

    # XSS过滤
    cleaned_data = re.sub(xss_cleaning_regex, '', request_data)
    if 'javascript:' in cleaned_data or 'vbscript:' in cleaned_data:
        return False

    # 其他基本验证，比如检查POST数据长度、禁止特殊字符等...
    if len(request_data) > 1000 or '..' in request_data:
        return False

    return True

# 使用示例
if is_waf_bypass('/admin panel', 'user=admin&password=123'):
    block_request()
else:
    pass # 允许请求通过