ChatGPT防火墙实战指南：从零搭建到生产环境部署

1次阅读

共计 1777 个字符，预计需要花费 5 分钟才能阅读完成。

随着 ChatGPT API 的广泛应用，开发者面临恶意请求、滥用和 API 密钥泄露等安全风险。本文详细介绍如何从零搭建一个高效的 ChatGPT 防火墙，涵盖请求限流、内容过滤和异常检测等核心功能。通过 Nginx+Lua 的实战方案，开发者可以快速部署可扩展的防护层，有效降低 API 滥用风险，同时保持低延迟和高可用性。

ChatGPT API 在提供强大功能的同时，也面临诸多安全挑战：

API 密钥泄露：密钥被恶意获取后可能导致高额费用或数据泄露。
请求滥用：自动化脚本发送大量请求，影响服务稳定性。
恶意内容生成：利用 API 生成不当或违规内容。
DDoS 攻击：大规模请求导致服务不可用。

常见方案对比：

Nginx+Lua：
优点：高性能、低延迟、灵活定制。
缺点：需要一定的 Lua 编程经验。
云原生 API 网关：
优点：开箱即用，集成方便。
缺点：成本较高，定制性有限。
自研中间件：
优点：完全可控，高度定制。
缺点：开发维护成本高。

综合考虑性能和灵活性，我们选择 Nginx+Lua 方案。

使用 OpenResty 的 lua-resty-limit-traffic 模块实现令牌桶算法，限制每个 IP 的请求速率。

lua_shared_dict limit_req_store 100m;

init_by_lua_block {
    local limit_req = require "resty.limit.req"
    local lim = limit_req.new("limit_req_store", 10, 5) -- 10 请求 / 秒，桶容量 5
}

access_by_lua_block {
    local limit_req = require "resty.limit.req"
    local lim = limit_req.new("limit_req_store", 10, 5)

    local key = ngx.var.remote_addr
    local delay, err = lim:incoming(key, true)

    if not delay then
        if err == "rejected" then
            return ngx.exit(429)
        end
        ngx.log(ngx.ERR, "failed to limit req:", err)
        return ngx.exit(500)
    end
}

基于正则表达式检测恶意 Prompt，防止生成不当内容。

local function filter_prompt(prompt)
    local banned_patterns = {"暴力", "色情", "诈骗", "仇恨言论"}

    for _, pattern in ipairs(banned_patterns) do
        if string.find(prompt, pattern) then
            return false
        end
    end
    return true
end

location /chat {
    content_by_lua_block {local prompt = ngx.req.get_post_args().prompt
        if not filter_prompt(prompt) then
            ngx.status = 400
            ngx.say("请求包含违规内容")
            return
        end
        -- 转发到 ChatGPT API
    }
}

集成公开的 IP 黑名单，防范已知恶意 IP。

local ip_blacklist = {"1.2.3.4", "5.6.7.8"}

access_by_lua_block {for _, ip in ipairs(ip_blacklist) do
        if ngx.var.remote_addr == ip then
            ngx.exit(403)
        end
    end
}