Prechádzať zdrojové kódy

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

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- 3 mesiacov pred
rodič
commit
a1c78ad9d4
1 zmenil súbory, kde vykonal 5 pridanie a 0 odobranie
  1. 5 0
      api/services/audio_service.py

+ 5 - 0
api/services/audio_service.py

@@ -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