Selaa lähdekoodia

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

Signed-off-by: -LAN- <laipz8200@outlook.com>
-LAN- 4 kuukautta sitten
vanhempi
commit
a1c78ad9d4
1 muutettua tiedostoa jossa 5 lisäystä ja 0 poistoa
  1. 5 0
      api/services/audio_service.py

+ 5 - 0
api/services/audio_service.py

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