Devine
/
pywidevine
mirror of https://github.com/devine-dl/pywidevine.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

RemoteCdm: Improve Server Version testing 

Some systems like Caddy or Nginx will prefix their own word to the Server header, e.g., `Caddy, pywidevine server v1.2.3` so I had to change a fair bit of the code to have wider compatibility across some unknowns that may occur with the Serve header.
This commit is contained in:
rlaphoenix 2022-08-04 08:33:33 +01:00
parent a1494a3742
commit f21a21712b
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pywidevine/remotecdm.py
View File

@ -2,6 +2,7 @@ from __future__ import annotations
import base64 import base64
import binascii import binascii
import re
from typing import Union, Optional from typing import Union, Optional
import requests import requests
@ -82,9 +83,12 @@ class RemoteCdm(Cdm):
if r.status_code != 200: if r.status_code != 200:
raise ValueError(f"Could not test Remote API version [{r.status_code}]") raise ValueError(f"Could not test Remote API version [{r.status_code}]")
server = r.headers.get("Server") server = r.headers.get("Server")
if not server or not server.startswith("https://github.com/rlaphoenix/pywidevine serve"): if not server or "pywidevine serve" not in server.lower():
raise ValueError(f"This Remote CDM API does not seem to be a pywidevine serve API ({server}).") raise ValueError(f"This Remote CDM API does not seem to be a pywidevine serve API ({server}).")
server_version = server.split("v")[-1] server_version = re.search(r"pywidevine serve v([\d.]+)", server, re.IGNORECASE)
if not server_version:
raise ValueError(f"The pywidevine server API is not stating the version correctly, cannot continue.")
server_version = server_version.group(1)
if server_version < "1.3.0": if server_version < "1.3.0":
raise ValueError(f"This pywidevine serve API version ({server_version}) is not supported.") raise ValueError(f"This pywidevine serve API version ({server_version}) is not supported.")