python - 是否可以限制 Flask 请求触发的 CPU/内存使用？

Question

我用 flask 开发后端我目前正在阅读有关诸如 Billion laughs attack 之类的攻击。尽管我知道其他防御方法，但我提出这个问题的动机是深度防御(在不同的组件中采取几种不同的防御 - 如果一个被绕过，希望其他方法有效)。

一个用户不应该为其他用户造成 DOS。如果我通过 gunicorn 运行 Flask，我认为它会为每个请求生成一个线程。是否可以为这些线程设置硬 CPU 使用率/内存使用率限制？

我发现了什么

根据我的发现，我猜这可能是不可能的，或者我看错了层次:

• gunicorn security : 似乎只是关于限制请求的 header /有效负载
• Flask security : 好像只是关于XSS、CSRF、CSP和各种header
• Flask-Limiter : flask 的速率限制

我认为主进程需要负责管理它的线程，但也许操作系统也可以/必须这样做？

Answer 1

我想我找到了an answer :

post_worker_init(worker)和 resource.setrlimit(resource.RLIMIT_AS, 10 * 2**20) .也可能是 resource.RLIMIT_STACK 或 resource.RLIMIT_HEAP。