Преглед на файлове

Reject requests that were not addressed to the proper vtnId

Signed-off-by: Stan Janssen <stan.janssen@elaad.nl>
Stan Janssen преди 3 години
родител
ревизия
54b22f9b3a
променени са 1 файла, в които са добавени 6 реда и са изтрити 0 реда
  1. 6 0
      openleadr/service/vtn_service.py

+ 6 - 0
openleadr/service/vtn_service.py

@@ -59,6 +59,12 @@ class VTNService:
             # Parse the message to a type and payload dict
             message_type, message_payload = parse_message(content)
 
+            if 'vtn_id' in message_payload \
+                    and message_payload['vtn_id'] is not None \
+                    and message_payload['vtn_id'] != self.vtn_id:
+                raise errors.InvalidIdError(f"The supplied vtnID is invalid. It should be '{self.vtn_id}', "
+                                            f"you supplied {message_payload['vtn_id']}.")
+
             # Authenticate the message
             if request.secure and 'ven_id' in message_payload:
                 await authenticate_message(request, message_tree, message_payload,