voldemort/CDRM-Extension
1
0
Fork 0
forked from tpd94/CDRM-Extension
Code Pull Requests Packages Releases Activity

Compare commits

base: voldemort:9e071365e372dda6424041f37eff597be50819cb
Branches Tags
voldemort:main
voldemort:rewrite
tpd94:main
tpd94:rewrite
...
compare: voldemort:867294f7f60135ae89d896bb01d809243dc3908b
Branches Tags
voldemort:main
voldemort:rewrite
tpd94:main
tpd94:rewrite

2 Commits
9e071365e3 ... 867294f7f6

Author SHA1 Message Date
voldemort
867294f7f6 refactor: extract DRM challenge detection and handling for fetch and XHR requests 2025-07-21 20:25:47 +07:00
voldemort
f8712d7726 refactor: consolidate remote cdm initialization into a helper function 2025-07-21 18:28:59 +07:00
1 changed files with 158 additions and 413 deletions
Show Stats Expand all files Collapse all files

519
src/inject.js
View File

@ -537,12 +537,11 @@ function postDRMTypeAndPssh(type, pssh) {
window.postMessage({ type: "__PSSH_DATA__", data: pssh }, "*"); window.postMessage({ type: "__PSSH_DATA__", data: pssh }, "*");
} }
function ensureRemoteCDM(type, deviceInfo, pssh) { function createAndOpenRemoteCDM(type, deviceInfo, pssh) {
if (!remoteCDM) { let cdm;
if (type === "Widevine") { if (type === "Widevine") {
const { device_type, system_id, security_level, host, secret, device_name } = const { device_type, system_id, security_level, host, secret, device_name } = deviceInfo;
deviceInfo; cdm = new remoteWidevineCDM(
remoteCDM = new remoteWidevineCDM(
device_type, device_type,
system_id, system_id,
security_level, security_level,
@ -550,14 +549,20 @@ function ensureRemoteCDM(type, deviceInfo, pssh) {
secret, secret,
device_name device_name
); );
remoteCDM.openSession(); cdm.openSession();
remoteCDM.getChallenge(pssh); cdm.getChallenge(pssh);
} else if (type === "PlayReady") { } else if (type === "PlayReady") {
const { security_level, host, secret, device_name } = deviceInfo; const { security_level, host, secret, device_name } = deviceInfo;
remoteCDM = new remotePlayReadyCDM(security_level, host, secret, device_name); cdm = new remotePlayReadyCDM(security_level, host, secret, device_name);
remoteCDM.openSession(); cdm.openSession();
remoteCDM.getChallenge(pssh); cdm.getChallenge(pssh);
} }
return cdm;
}
function ensureRemoteCDM(type, deviceInfo, pssh) {
if (!remoteCDM) {
remoteCDM = createAndOpenRemoteCDM(type, deviceInfo, pssh);
} }
} }
@ -589,17 +594,11 @@ MediaKeySession.prototype.generateRequest = function (initDataType, initData) {
interceptType !== "DISABLED" && interceptType !== "DISABLED" &&
!serviceCertFound !serviceCertFound
) { ) {
const { device_type, system_id, security_level, host, secret, device_name } = remoteCDM = createAndOpenRemoteCDM(
widevineDeviceInfo; "Widevine",
remoteCDM = new remoteWidevineCDM( widevineDeviceInfo,
device_type, foundWidevinePssh
system_id,
security_level,
host,
secret,
device_name
); );
remoteCDM.openSession();
} }
if ( if (
!injectionSuccess && !injectionSuccess &&
@ -715,6 +714,81 @@ MediaKeySession.prototype.update = function (response) {
return updatePromise; return updatePromise;
}; };
// Helpers
function detectDRMChallenge(body) {
// Handles ArrayBuffer, Uint8Array, string, and JSON string
// Returns: { type: "Widevine"|"PlayReady"|null, base64: string|null, bodyType: "buffer"|"string"|"json"|null }
if (body instanceof ArrayBuffer || body instanceof Uint8Array) {
const buffer = body instanceof Uint8Array ? body : new Uint8Array(body);
const base64Body = window.btoa(String.fromCharCode(...buffer));
if (base64Body.startsWith(DRM_SIGNATURES.WIDEVINE)) {
return { type: "Widevine", base64: base64Body, bodyType: "buffer" };
}
if (base64Body.startsWith(DRM_SIGNATURES.PLAYREADY)) {
return { type: "PlayReady", base64: base64Body, bodyType: "buffer" };
}
} else if (typeof body === "string" && !isJson(body)) {
const base64EncodedBody = btoa(body);
if (base64EncodedBody.startsWith(DRM_SIGNATURES.WIDEVINE)) {
return { type: "Widevine", base64: base64EncodedBody, bodyType: "string" };
}
if (base64EncodedBody.startsWith(DRM_SIGNATURES.PLAYREADY)) {
return { type: "PlayReady", base64: base64EncodedBody, bodyType: "string" };
}
} else if (typeof body === "string" && isJson(body)) {
const jsonBody = JSON.parse(body);
if (jsonContainsValue(jsonBody, DRM_SIGNATURES.WIDEVINE)) {
return { type: "Widevine", base64: null, bodyType: "json" };
}
if (jsonContainsValue(jsonBody, DRM_SIGNATURES.PLAYREADY)) {
return { type: "PlayReady", base64: null, bodyType: "json" };
}
}
return { type: null, base64: null, bodyType: null };
}
function handleLicenseMode({
drmInfo,
body,
setBody, // function to set the new body (for fetch: (b) => config.body = b, for XHR: (b) => originalSend.call(this, b))
urlOrResource,
getWidevinePssh,
getPlayreadyPssh,
widevineDeviceInfo,
playreadyDeviceInfo,
}) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: urlOrResource }, "*");
// Create remoteCDM if needed
if (!remoteCDM) {
if (drmInfo.type === "Widevine") {
remoteCDM = createAndOpenRemoteCDM("Widevine", widevineDeviceInfo, getWidevinePssh());
}
if (drmInfo.type === "PlayReady") {
remoteCDM = createAndOpenRemoteCDM(
"PlayReady",
playreadyDeviceInfo,
getPlayreadyPssh()
);
}
}
if (remoteCDM && remoteCDM.challenge === null) {
remoteCDM.getChallenge(getWidevinePssh());
}
// Inject the new challenge into the request body
if (drmInfo.bodyType === "json") {
const jsonBody = JSON.parse(body);
const injectedBody = jsonReplaceValue(jsonBody, remoteCDM.challenge);
setBody(JSON.stringify(injectedBody));
} else if (drmInfo.bodyType === "buffer") {
setBody(base64ToUint8Array(remoteCDM.challenge));
} else {
setBody(atob(remoteCDM.challenge));
}
}
// fetch POST interceptor // fetch POST interceptor
(function () { (function () {
const originalFetch = window.fetch; const originalFetch = window.fetch;
@ -725,144 +799,14 @@ MediaKeySession.prototype.update = function (response) {
if (method === "POST") { if (method === "POST") {
let body = config.body; let body = config.body;
if (body) { if (body) {
if (body instanceof ArrayBuffer || body instanceof Uint8Array) { const drmInfo = detectDRMChallenge(body);
const buffer = body instanceof Uint8Array ? body : new Uint8Array(body);
const base64Body = window.btoa(String.fromCharCode(...buffer));
if (
(base64Body.startsWith(DRM_SIGNATURES.WIDEVINE) ||
base64Body.startsWith(DRM_SIGNATURES.PLAYREADY)) &&
(!remoteCDM ||
remoteCDM.challenge === null ||
base64Body !== remoteCDM.challenge) &&
interceptType === "EME"
) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: resource }, "*");
// Block the request
return;
}
if (
(base64Body.startsWith(DRM_SIGNATURES.WIDEVINE) ||
base64Body.startsWith(DRM_SIGNATURES.PLAYREADY)) &&
interceptType == "LICENSE" &&
!foundChallengeInBody
) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: resource }, "*");
if (!remoteCDM) {
if (base64Body.startsWith(DRM_SIGNATURES.WIDEVINE)) {
const {
device_type,
system_id,
security_level,
host,
secret,
device_name,
} = widevineDeviceInfo;
remoteCDM = new remoteWidevineCDM(
device_type,
system_id,
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundWidevinePssh);
}
if (base64Body.startsWith(DRM_SIGNATURES.PLAYREADY)) {
const { security_level, host, secret, device_name } =
playreadyDeviceInfo;
remoteCDM = new remotePlayReadyCDM(
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundPlayreadyPssh);
}
}
if (remoteCDM && remoteCDM.challenge === null) {
remoteCDM.getChallenge(foundWidevinePssh);
}
const injectedBody = base64ToUint8Array(remoteCDM.challenge);
config.body = injectedBody;
return originalFetch(resource, config);
}
}
if (typeof body === "string" && !isJson(body)) {
const base64EncodedBody = btoa(body);
if (
(base64EncodedBody.startsWith(DRM_SIGNATURES.WIDEVINE) ||
base64EncodedBody.startsWith(DRM_SIGNATURES.PLAYREADY)) &&
(!remoteCDM ||
remoteCDM.challenge === null ||
base64EncodedBody !== remoteCDM.challenge) &&
interceptType === "EME"
) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: resource }, "*");
// Block the request
return;
}
if (
(base64EncodedBody.startsWith(DRM_SIGNATURES.WIDEVINE) ||
base64EncodedBody.startsWith(DRM_SIGNATURES.PLAYREADY)) &&
interceptType == "LICENSE" &&
!foundChallengeInBody
) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: resource }, "*");
if (!remoteCDM) {
if (base64EncodedBody.startsWith(DRM_SIGNATURES.WIDEVINE)) {
const {
device_type,
system_id,
security_level,
host,
secret,
device_name,
} = widevineDeviceInfo;
remoteCDM = new remoteWidevineCDM(
device_type,
system_id,
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundWidevinePssh);
}
if (base64EncodedBody.startsWith(DRM_SIGNATURES.PLAYREADY)) {
const { security_level, host, secret, device_name } =
playreadyDeviceInfo;
remoteCDM = new remotePlayReadyCDM(
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundPlayreadyPssh);
}
}
if (remoteCDM && remoteCDM.challenge === null) {
remoteCDM.getChallenge(foundWidevinePssh);
}
const injectedBody = atob(remoteCDM.challenge);
config.body = injectedBody;
return originalFetch(resource, config);
}
}
if (typeof body === "string" && isJson(body)) {
const jsonBody = JSON.parse(body);
// EME mode: block the request if a DRM challenge is detected
if ( if (
(jsonContainsValue(jsonBody, DRM_SIGNATURES.WIDEVINE) || drmInfo.type &&
jsonContainsValue(jsonBody, DRM_SIGNATURES.PLAYREADY)) && (!remoteCDM ||
(!remoteCDM || remoteCDM.challenge === null) && remoteCDM.challenge === null ||
drmInfo.base64 !== remoteCDM.challenge) &&
interceptType === "EME" interceptType === "EME"
) { ) {
foundChallengeInBody = true; foundChallengeInBody = true;
@ -871,54 +815,21 @@ MediaKeySession.prototype.update = function (response) {
return; return;
} }
if ( // LICENSE mode: replace the challenge in the request
(jsonContainsValue(jsonBody, DRM_SIGNATURES.WIDEVINE) || if (drmInfo.type && interceptType === "LICENSE" && !foundChallengeInBody) {
jsonContainsValue(jsonBody, DRM_SIGNATURES.PLAYREADY)) && handleLicenseMode({
interceptType === "LICENSE" && drmInfo,
!foundChallengeInBody body,
) { setBody: (b) => {
foundChallengeInBody = true; config.body = b;
window.postMessage({ type: "__LICENSE_URL__", data: resource }, "*"); },
if (!remoteCDM) { urlOrResource: resource,
if (jsonContainsValue(jsonBody, DRM_SIGNATURES.WIDEVINE)) { getWidevinePssh: () => foundWidevinePssh,
const { getPlayreadyPssh: () => foundPlayreadyPssh,
device_type, widevineDeviceInfo,
system_id, playreadyDeviceInfo,
security_level, });
host, return originalFetch(resource, config);
secret,
device_name,
} = widevineDeviceInfo;
remoteCDM = new remoteWidevineCDM(
device_type,
system_id,
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundWidevinePssh);
}
if (jsonContainsValue(jsonBody, DRM_SIGNATURES.PLAYREADY)) {
const { security_level, host, secret, device_name } =
playreadyDeviceInfo;
remoteCDM = new remotePlayReadyCDM(
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundPlayreadyPssh);
}
}
if (remoteCDM && remoteCDM.challenge === null) {
remoteCDM.getChallenge(foundWidevinePssh);
}
const injectedBody = jsonReplaceValue(jsonBody, remoteCDM.challenge);
config.body = JSON.stringify(injectedBody);
}
} }
} }
} }
@ -941,144 +852,14 @@ MediaKeySession.prototype.update = function (response) {
XMLHttpRequest.prototype.send = function (body) { XMLHttpRequest.prototype.send = function (body) {
if (this._method && this._method.toUpperCase() === "POST") { if (this._method && this._method.toUpperCase() === "POST") {
if (body) { if (body) {
if (body instanceof ArrayBuffer || body instanceof Uint8Array) { const drmInfo = detectDRMChallenge(body);
const buffer = body instanceof Uint8Array ? body : new Uint8Array(body);
const base64Body = window.btoa(String.fromCharCode(...buffer)); // EME mode: block the request if a DRM challenge is detected
if ( if (
(base64Body.startsWith(DRM_SIGNATURES.WIDEVINE) || drmInfo.type &&
base64Body.startsWith(DRM_SIGNATURES.PLAYREADY)) &&
(!remoteCDM || (!remoteCDM ||
remoteCDM.challenge === null || remoteCDM.challenge === null ||
base64Body !== remoteCDM.challenge) && drmInfo.base64 !== remoteCDM.challenge) &&
interceptType === "EME"
) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*");
// Block the request
return;
}
if (
(base64Body.startsWith(DRM_SIGNATURES.WIDEVINE) ||
base64Body.startsWith(DRM_SIGNATURES.PLAYREADY)) &&
interceptType == "LICENSE" &&
!foundChallengeInBody
) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*");
if (!remoteCDM) {
if (base64Body.startsWith(DRM_SIGNATURES.WIDEVINE)) {
const {
device_type,
system_id,
security_level,
host,
secret,
device_name,
} = widevineDeviceInfo;
remoteCDM = new remoteWidevineCDM(
device_type,
system_id,
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundWidevinePssh);
}
if (base64Body.startsWith(DRM_SIGNATURES.PLAYREADY)) {
const { security_level, host, secret, device_name } =
playreadyDeviceInfo;
remoteCDM = new remotePlayReadyCDM(
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundPlayreadyPssh);
}
}
if (remoteCDM && remoteCDM.challenge === null) {
remoteCDM.getChallenge(foundWidevinePssh);
}
const injectedBody = base64ToUint8Array(remoteCDM.challenge);
return originalSend.call(this, injectedBody);
}
}
if (typeof body === "string" && !isJson(body)) {
const base64EncodedBody = btoa(body);
if (
(base64EncodedBody.startsWith(DRM_SIGNATURES.WIDEVINE) ||
base64EncodedBody.startsWith(DRM_SIGNATURES.PLAYREADY)) &&
(!remoteCDM ||
remoteCDM.challenge === null ||
base64EncodedBody !== remoteCDM.challenge) &&
interceptType === "EME"
) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*");
// Block the request
return;
}
if (
(base64EncodedBody.startsWith(DRM_SIGNATURES.WIDEVINE) ||
base64EncodedBody.startsWith(DRM_SIGNATURES.PLAYREADY)) &&
interceptType == "LICENSE" &&
!foundChallengeInBody
) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*");
if (!remoteCDM) {
if (base64EncodedBody.startsWith(DRM_SIGNATURES.WIDEVINE)) {
const {
device_type,
system_id,
security_level,
host,
secret,
device_name,
} = widevineDeviceInfo;
remoteCDM = new remoteWidevineCDM(
device_type,
system_id,
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundWidevinePssh);
}
if (base64EncodedBody.startsWith(DRM_SIGNATURES.PLAYREADY)) {
const { security_level, host, secret, device_name } =
playreadyDeviceInfo;
remoteCDM = new remotePlayReadyCDM(
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundPlayreadyPssh);
}
}
if (remoteCDM && remoteCDM.challenge === null) {
remoteCDM.getChallenge(foundWidevinePssh);
}
const injectedBody = atob(remoteCDM.challenge);
return originalSend.call(this, injectedBody);
}
}
if (typeof body === "string" && isJson(body)) {
const jsonBody = JSON.parse(body);
if (
(jsonContainsValue(jsonBody, DRM_SIGNATURES.WIDEVINE) ||
jsonContainsValue(jsonBody, DRM_SIGNATURES.PLAYREADY)) &&
(!remoteCDM || remoteCDM.challenge === null) &&
interceptType === "EME" interceptType === "EME"
) { ) {
foundChallengeInBody = true; foundChallengeInBody = true;
@ -1087,54 +868,18 @@ MediaKeySession.prototype.update = function (response) {
return; return;
} }
if ( // LICENSE mode: replace the challenge in the request
(jsonContainsValue(jsonBody, DRM_SIGNATURES.WIDEVINE) || if (drmInfo.type && interceptType === "LICENSE" && !foundChallengeInBody) {
jsonContainsValue(jsonBody, DRM_SIGNATURES.PLAYREADY)) && return handleLicenseMode({
interceptType === "LICENSE" && drmInfo,
!foundChallengeInBody body,
) { setBody: (b) => originalSend.call(this, b),
foundChallengeInBody = true; urlOrResource: this._url,
window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*"); getWidevinePssh: () => foundWidevinePssh,
if (!remoteCDM) { getPlayreadyPssh: () => foundPlayreadyPssh,
if (jsonContainsValue(jsonBody, DRM_SIGNATURES.WIDEVINE)) { widevineDeviceInfo,
const { playreadyDeviceInfo,
device_type, });
system_id,
security_level,
host,
secret,
device_name,
} = widevineDeviceInfo;
remoteCDM = new remoteWidevineCDM(
device_type,
system_id,
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundWidevinePssh);
}
if (jsonContainsValue(jsonBody, DRM_SIGNATURES.PLAYREADY)) {
const { security_level, host, secret, device_name } =
playreadyDeviceInfo;
remoteCDM = new remotePlayReadyCDM(
security_level,
host,
secret,
device_name
);
remoteCDM.openSession();
remoteCDM.getChallenge(foundPlayreadyPssh);
}
}
if (remoteCDM && remoteCDM.challenge === null) {
remoteCDM.getChallenge(foundWidevinePssh);
}
const injectedBody = jsonReplaceValue(jsonBody, remoteCDM.challenge);
return originalSend.call(this, JSON.stringify(injectedBody));
}
} }
} }
} }