skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fd

Browse source
This commit is contained in:
Aevann1 2022-02-17 03:30:59 +02:00
parent 71ede04c89
commit 172c4f725a
1 changed files with 13 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

13
files/helpers/sanitize.py
View file

@ -8,6 +8,8 @@ import re
from mistletoe import markdown from mistletoe import markdown
from json import loads, dump from json import loads, dump
from random import random, choice from random import random, choice
import signal
import time
db = db_session() db = db_session()
marseys = tuple(x[0] for x in db.query(Marsey.name).all()) marseys = tuple(x[0] for x in db.query(Marsey.name).all())
@ -102,8 +104,17 @@ allowed_protocols = ['http', 'https']
allowed_styles = ['color', 'background-color', 'font-weight', 'text-align'] allowed_styles = ['color', 'background-color', 'font-weight', 'text-align']
def handler(signum, frame):
print("Forever is over!")
raise Exception("end of time")
def sanitize(sanitized, noimages=False, alert=False, comment=False, edit=False): def sanitize(sanitized, noimages=False, alert=False, comment=False, edit=False):
signal.signal(signal.SIGALRM, handler)
signal.alarm(1)
if sanitized.count(':') > 100: abort(418) if sanitized.count(':') > 100: abort(418)
if sanitized.count('@') > 50: abort(418) if sanitized.count('@') > 50: abort(418)
@ -283,6 +294,8 @@ def sanitize(sanitized, noimages=False, alert=False, comment=False, edit=False):
marsey.count += 1 marsey.count += 1
g.db.add(marsey) g.db.add(marsey)
signal.alarm(0)
return sanitized return sanitized