DDoS攻击防护定价

# 假设我们有一个简单的Web服务器
from flask import Flask, request

app = Flask(__name__)

@app.route('/')
def home():
    return "Welcome to our website!"

# 简单的DDoS防护机制示例
MAX_REQUESTS_PER_SECOND = 100
request_counts = {}

@app.before_request
def limit_requests():
    client_ip = request.remote_addr
    current_time = int(time.time())
    
    if client_ip not in request_counts:
        request_counts[client_ip] = []
    
    # 移除超过1秒的请求记录
    request_counts[client_ip] = [t for t in request_counts[client_ip] if current_time - t < 1]
    
    if len(request_counts[client_ip]) >= MAX_REQUESTS_PER_SECOND:
        return "Too many requests!", 429
    
    request_counts[client_ip].append(current_time)
    return None

if __name__ == '__main__':
    app.run()