Cdm: Validate License Message type in parse_license

2022-07-24 21:07:00 +01:00 · 2022-07-24 21:07:00 +01:00 · ac469383b8
parent b081d66ca2
commit ac469383b8
1 changed files with 6 additions and 0 deletions
--- a/pywidevine/cdm.py
+++ b/pywidevine/cdm.py
@ -211,6 +211,12 @@ class Cdm:
        if not isinstance(license_message, SignedMessage):
            raise ValueError(f"Expecting license_response to be a SignedMessage, got {license_message!r}")

+        if license_message.type != SignedMessage.MessageType.LICENSE:
+            raise ValueError(
+                f"Expecting a LICENSE message, not a "
+                f"'{SignedMessage.MessageType.Name(license_message.type)}' message."
+            )
+
        licence = License()
        licence.ParseFromString(license_message.msg)