tpd94/CDRM-Extension
3
2
Fork 2
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Update inject.js

Browse Source
This commit is contained in:
TPD94 2025-06-29 12:07:18 -04:00
parent 388afaa171
commit 3fea6e54ba
1 changed files with 170 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

190
inject.js
View File

@ -15,6 +15,7 @@ let remoteListenerMounted = false;
let injectionSuccess = false; let injectionSuccess = false;
let foundChallengeInBody = false; let foundChallengeInBody = false;
let licenseResponseCounter = 0; let licenseResponseCounter = 0;
let keysRetrieved = false;
// Post message to content.js to get DRM override // Post message to content.js to get DRM override
window.postMessage({ type: "__GET_DRM_OVERRIDE__" }, "*"); window.postMessage({ type: "__GET_DRM_OVERRIDE__" }, "*");
@ -321,20 +322,20 @@ function jsonContainsValue(obj, prefix = "CAES") {
return false; return false;
} }
function jsonReplaceValue(obj, target, newValue) { function jsonReplaceValue(obj, newValue) {
if (typeof obj === "string") { if (typeof obj === "string") {
return obj === target ? newValue : obj; return obj.startsWith("CAES") || obj.startsWith("PD94") ? newValue : obj;
} }
if (Array.isArray(obj)) { if (Array.isArray(obj)) {
return obj.map(item => jsonReplaceValue(item, target, newValue)); return obj.map(item => jsonReplaceValue(item, newValue));
} }
if (typeof obj === "object" && obj !== null) { if (typeof obj === "object" && obj !== null) {
const newObj = {}; const newObj = {};
for (const key in obj) { for (const key in obj) {
if (Object.hasOwn(obj, key)) { if (Object.hasOwn(obj, key)) {
newObj[key] = jsonReplaceValue(obj[key], target, newValue); newObj[key] = jsonReplaceValue(obj[key], newValue);
} }
} }
return newObj; return newObj;
@ -430,7 +431,7 @@ MediaKeySession.prototype.generateRequest = function(initDataType, initData) {
event.stopImmediatePropagation(); event.stopImmediatePropagation();
const uint8Array = new Uint8Array(event.message); const uint8Array = new Uint8Array(event.message);
const base64challenge = arrayBufferToBase64(uint8Array); const base64challenge = arrayBufferToBase64(uint8Array);
if (base64challenge === "CAQ=" && interceptType !== "DISABLED") { if (base64challenge === "CAQ=" && interceptType !== "DISABLED" && !serviceCertFound) {
const { const {
device_type, system_id, security_level, host, secret, device_name device_type, system_id, security_level, host, secret, device_name
} = widevineDeviceInfo; } = widevineDeviceInfo;
@ -501,19 +502,21 @@ const originalUpdate = MediaKeySession.prototype.update;
MediaKeySession.prototype.update = function(response) { MediaKeySession.prototype.update = function(response) {
const uint8 = response instanceof Uint8Array ? response : new Uint8Array(response); const uint8 = response instanceof Uint8Array ? response : new Uint8Array(response);
const base64Response = window.btoa(String.fromCharCode(...uint8)); const base64Response = window.btoa(String.fromCharCode(...uint8));
if (base64Response.startsWith("CAUS") && foundWidevinePssh && remoteCDM) { if (base64Response.startsWith("CAUS") && foundWidevinePssh && remoteCDM && !serviceCertFound) {
remoteCDM.setServiceCertificate(base64Response); remoteCDM.setServiceCertificate(base64Response);
if (interceptType === "EME") { if (interceptType === "EME" && !remoteCDM.challenge) {
remoteCDM.getChallenge(foundWidevinePssh); remoteCDM.getChallenge(foundWidevinePssh);
} }
window.postMessage({ type: "__DRM_TYPE__", data: "Widevine" }, "*"); window.postMessage({ type: "__DRM_TYPE__", data: "Widevine" }, "*");
window.postMessage({ type: "__PSSH_DATA__", data: foundWidevinePssh }, "*"); window.postMessage({ type: "__PSSH_DATA__", data: foundWidevinePssh }, "*");
serviceCertFound = true;
} }
if (!base64Response.startsWith("CAUS") && (foundWidevinePssh || foundPlayreadyPssh)) { if (!base64Response.startsWith("CAUS") && (foundWidevinePssh || foundPlayreadyPssh) && !keysRetrieved) {
if (licenseResponseCounter === 1 || foundChallengeInBody) { if (licenseResponseCounter === 1 || foundChallengeInBody) {
remoteCDM.parseLicense(base64Response); remoteCDM.parseLicense(base64Response);
remoteCDM.getKeys(); remoteCDM.getKeys();
remoteCDM.closeSession(); remoteCDM.closeSession();
keysRetrieved = true;
window.postMessage({ type: "__KEYS_DATA__", data: remoteCDM.keys }, "*"); window.postMessage({ type: "__KEYS_DATA__", data: remoteCDM.keys }, "*");
} }
licenseResponseCounter++; licenseResponseCounter++;
@ -553,15 +556,120 @@ MediaKeySession.prototype.update = function(response) {
const method = (config.method || 'GET').toUpperCase(); const method = (config.method || 'GET').toUpperCase();
if (method === 'POST') { if (method === 'POST') {
if (config.body) { let body = config.body;
if (config.body instanceof ArrayBuffer || body instanceof Uint8Array) { if (body) {
const buffer = body instanceof Uint8Array ? body : new Uint8Array(body); if (body instanceof ArrayBuffer || body instanceof Uint8Array) {
const base64Body = window.btoa(String.fromCharCode(...buffer)); const buffer = body instanceof Uint8Array ? body : new Uint8Array(body);
if (base64Body.startsWith("CAES") && base64Body !== remoteCDM.challenge && interceptType === "EME") { const base64Body = window.btoa(String.fromCharCode(...buffer));
foundChallengeInBody = true; if ((base64Body.startsWith("CAES") || base64Body.startsWith("PD94")) && (!remoteCDM || remoteCDM.challenge === null || base64Body !== remoteCDM.challenge) && interceptType === "EME") {
window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*"); foundChallengeInBody = true;
return; window.postMessage({ type: "__LICENSE_URL__", data: resource }, "*");
} // Block the request
return;
}
if ((base64Body.startsWith("CAES") || base64Body.startsWith("PD94")) && interceptType == "LICENSE" &&!foundChallengeInBody) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: resource }, "*");
if (!remoteCDM) {
if (base64Body.startsWith("CAES")) {
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("PD94")) {
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("CAES") || base64EncodedBody.startsWith("PD94")) && (!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("CAES") || base64EncodedBody.startsWith("PD94")) && interceptType == "LICENSE" && !foundChallengeInBody) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: resource }, "*");
if (!remoteCDM) {
if (base64EncodedBody.startsWith("CAES")) {
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("PD94")) {
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);
if ((jsonContainsValue(jsonBody, "CAES") || jsonContainsValue(jsonBody, "PD94")) && (!remoteCDM || remoteCDM.challenge === null) && interceptType === "EME") {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: resource }, "*");
// Block the request
return;
}
if ((jsonContainsValue(jsonBody, "CAES") || jsonContainsValue(jsonBody, "PD94")) && interceptType === "LICENSE" && !foundChallengeInBody) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: resource }, "*");
if (!remoteCDM) {
if (jsonContainsValue(jsonBody, "CAES")) {
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 (jsonContainsValue(jsonBody, "PD94")) {
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);
}
} }
} }
} }
@ -594,7 +702,7 @@ MediaKeySession.prototype.update = function(response) {
// Block the request // Block the request
return; return;
} }
if ((base64Body.startsWith("CAES") || base64Body.startsWith("PD94")) && interceptType == "LICENSE") { if ((base64Body.startsWith("CAES") || base64Body.startsWith("PD94")) && interceptType == "LICENSE" &&!foundChallengeInBody) {
foundChallengeInBody = true; foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*"); window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*");
if (!remoteCDM) { if (!remoteCDM) {
@ -615,7 +723,7 @@ MediaKeySession.prototype.update = function(response) {
remoteCDM.getChallenge(foundPlayreadyPssh); remoteCDM.getChallenge(foundPlayreadyPssh);
} }
} }
if (remoteCDM) { if (remoteCDM && remoteCDM.challenge === null) {
remoteCDM.getChallenge(foundWidevinePssh); remoteCDM.getChallenge(foundWidevinePssh);
} }
const injectedBody = base64ToUint8Array(remoteCDM.challenge); const injectedBody = base64ToUint8Array(remoteCDM.challenge);
@ -631,7 +739,7 @@ MediaKeySession.prototype.update = function(response) {
// Block the request // Block the request
return; return;
} }
if ((base64EncodedBody.startsWith("CAES") || base64EncodedBody.startsWith("PD94")) && interceptType == "LICENSE") { if ((base64EncodedBody.startsWith("CAES") || base64EncodedBody.startsWith("PD94")) && interceptType == "LICENSE" && !foundChallengeInBody) {
foundChallengeInBody = true; foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*"); window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*");
if (!remoteCDM) { if (!remoteCDM) {
@ -652,10 +760,52 @@ MediaKeySession.prototype.update = function(response) {
remoteCDM.getChallenge(foundPlayreadyPssh); remoteCDM.getChallenge(foundPlayreadyPssh);
} }
} }
if (remoteCDM && remoteCDM.challenge === null) {
remoteCDM.getChallenge(foundWidevinePssh);
}
const injectedBody = atob(remoteCDM.challenge); const injectedBody = atob(remoteCDM.challenge);
return originalSend.call(this, injectedBody); return originalSend.call(this, injectedBody);
} }
} }
if (typeof body === 'string' && isJson(body)) {
const jsonBody = JSON.parse(body);
if ((jsonContainsValue(jsonBody, "CAES") || jsonContainsValue(jsonBody, "PD94")) && (!remoteCDM || remoteCDM.challenge === null) && interceptType === "EME") {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*");
// Block the request
return;
}
if ((jsonContainsValue(jsonBody, "CAES") || jsonContainsValue(jsonBody, "PD94")) && interceptType === "LICENSE" && !foundChallengeInBody) {
foundChallengeInBody = true;
window.postMessage({ type: "__LICENSE_URL__", data: this._url }, "*");
if (!remoteCDM) {
if (jsonContainsValue(jsonBody, "CAES")) {
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 (jsonContainsValue(jsonBody, "PD94")) {
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));
}
}
} }
} }
return originalSend.apply(this, arguments); return originalSend.apply(this, arguments);