AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-06-04 11:14:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(audio_service): validate message_id format using UUID (#12087)

Browse Source 
Signed-off-by: -LAN- <laipz8200@outlook.com>
This commit is contained in:
-LAN- 2024-12-25 18:36:52 +08:00 committed by GitHub
parent 754791efd3
commit a1c78ad9d4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/services/audio_service.py
View File

@ -1,5 +1,6 @@
import io import io
import logging import logging
import uuid
from typing import Optional from typing import Optional
from werkzeug.datastructures import FileStorage from werkzeug.datastructures import FileStorage
@ -122,6 +123,10 @@ class AudioService:
raise e raise e
if message_id: if message_id:
try:
uuid.UUID(message_id)
except ValueError:
return None
message = db.session.query(Message).filter(Message.id == message_id).first() message = db.session.query(Message).filter(Message.id == message_id).first()
if message is None: if message is None:
return None return None