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

fix RCE, thank you @Camas

master
Aevann1 2022-06-17 20:36:34 +02:00
parent 4bae98dab0
commit 30ed6daa26
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
files/helpers/media.py
View File

@ -10,7 +10,7 @@ from .const import *
def process_audio(file): def process_audio(file):
name = f'/audio/{time.time()}'.replace('.','') + '.' + file.filename.split('.')[-1].lower() name = f'/audio/{time.time()}'.replace('.','') + '.mp3'
file.save(name) file.save(name)
if os.stat(name).st_size > 8 * 1024 * 1024: if os.stat(name).st_size > 8 * 1024 * 1024:
@ -24,8 +24,7 @@ def process_audio(file):
def process_video(file): def process_video(file):
old = f'/videos/{time.time()}'.replace('.','') old = f'/videos/{time.time()}'.replace('.','')
extension = file.filename.split('.')[-1].lower() new = old + '.mp4'
new = old + '.' + extension
if extension == 'webm': if extension == 'webm':
file.save(new) file.save(new)
@ -71,4 +70,4 @@ def process_image(patron, filename=None, resize=0):
i = ImageOps.exif_transpose(i) i = ImageOps.exif_transpose(i)
i.save(filename, format="WEBP", method=6) i.save(filename, format="WEBP", method=6)
return filename return filename