Decryption-Tools/pywidevine
1
0
Fork 0
mirror of https://github.com/devine-dl/pywidevine.git synced 2024-10-30 05:29:21 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

serve: Add ability to get all types of keys in /keys

Browse Source
This commit is contained in:
rlaphoenix 2022-08-01 23:07:59 +01:00
parent 64ae5709d3
commit 290da707ea
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
pywidevine/serve.py
View File

@ -150,16 +150,19 @@ async def keys(request: web.Request) -> web.Response:
# get key type
key_type = request.match_info["key_type"]
try:
if key_type.isdigit():
key_type = License.KeyContainer.KeyType.Name(int(key_type))
else:
License.KeyContainer.KeyType.Value(key_type) # only test
except ValueError as e:
return web.json_response({
"status": 400,
"message": f"The Key Type value is invalid, {e}"
}, status=400)
if key_type == "ALL":
key_type = None
else:
try:
if key_type.isdigit():
key_type = License.KeyContainer.KeyType.Name(int(key_type))
else:
License.KeyContainer.KeyType.Value(key_type) # only test
except ValueError as e:
return web.json_response({
"status": 400,
"message": f"The Key Type value is invalid, {e}"
}, status=400)
# get cdm
if session_id not in request.app["sessions"]:
@ -183,7 +186,7 @@ async def keys(request: web.Request) -> web.Response:
"permissions": key.permissions,
}
for key in cdm._sessions[session_id].keys
if key.type == key_type
if not key_type or key.type == key_type
]
return web.json_response({