TriHard
/
rDrama
forked from rDrama/rDrama
1
0
Fork 0
Code Pull Requests Activity

fd

master
Aevann1 2022-02-17 03:30:59 +02:00
parent 431a3db3e8
commit cd22b37f61
1 changed files with 13 additions and 0 deletions
Show Stats 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