carpathianflorist
/
MarseyWorld
forked from MarseyWorld/MarseyWorld
1
0
Fork 0
Code Pull Requests Activity

disable signups when ddos detected

master
Aevann 2023-08-10 00:09:15 +03:00
parent 73bb8f6153
commit 6bf8f67d0f
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
files/routes/errors.py
View File

@ -48,7 +48,8 @@ def error_401(e):
qs = urlencode(dict(request.values))
argval = quote(f"{path}?{qs}", safe='').replace('/logged_out','')
if not argval: argval = '/'
if session.get("history") or not get_setting("signups"): return redirect(f"/login?redirect={argval}")
if session.get("history") or not get_setting("signups") or get_setting("ddos_detected"):
return redirect(f"/login?redirect={argval}")
else: return redirect(f"/signup?redirect={argval}")
@app.errorhandler(500)

4
files/routes/login.py
View File

@ -146,7 +146,7 @@ def logout(v):
@limiter.limit(DEFAULT_RATELIMIT, deduct_when=lambda response: response.status_code < 400)
@auth_desired
def sign_up_get(v):
if not get_setting('signups'):
if not get_setting('signups') or get_setting("ddos_detected"):
abort(403, "New account registration is currently closed. Please come back later!")
if v: return redirect(SITE_FULL)
@ -195,7 +195,7 @@ def sign_up_get(v):
@limiter.limit("10/day", deduct_when=lambda response: response.status_code < 400)
@auth_desired
def sign_up_post(v):
if not get_setting('signups'):
if not get_setting('signups') or get_setting("ddos_detected"):
abort(403, "New account registration is currently closed. Please come back later!")
if v: abort(403)