self-hosted/Mailu
1
0
Fork 0
mirror of https://github.com/Mailu/Mailu.git synced 2024-12-14 10:53:30 +02:00
Code Issues Releases Activity

Merge pull request #513 from mprihoda/feature/better-ratelimit-error

Browse Source 
Return correct status codes from auth rate limiter failure.
This commit is contained in:
kaiyou 2018-09-16 19:58:34 +02:00 committed by GitHub
commit 313b79538e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
core/admin/mailu/internal/__init__.py
View File

@ -1,3 +1,5 @@
from flask_limiter import RateLimitExceeded
from mailu import limiter from mailu import limiter
import socket import socket
@ -6,6 +8,14 @@ import flask
internal = flask.Blueprint('internal', __name__) internal = flask.Blueprint('internal', __name__)
@internal.app_errorhandler(RateLimitExceeded)
def rate_limit_handler(e):
response = flask.Response()
response.headers['Auth-Status'] = 'Authentication rate limit from one source exceeded'
response.headers['Auth-Error-Code'] = '451 4.3.2'
if int(flask.request.headers['Auth-Login-Attempt']) < 10:
response.headers['Auth-Wait'] = '3'
return response
@limiter.request_filter @limiter.request_filter
def whitelist_webmail(): def whitelist_webmail():