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

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

View File

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