Files
Vineless/content_script.js
2025-07-02 21:04:47 +09:00

538 lines
20 KiB
JavaScript

function uint8ArrayToBase64(uint8array) {
return btoa(String.fromCharCode.apply(null, uint8array));
}
function uint8ArrayToString(uint8array) {
return String.fromCharCode.apply(null, uint8array)
}
function base64toUint8Array(base64_string){
return Uint8Array.from(atob(base64_string), c => c.charCodeAt(0))
}
function compareUint8Arrays(arr1, arr2) {
if (arr1.length !== arr2.length)
return false;
return Array.from(arr1).every((value, index) => value === arr2[index]);
}
function base64ToBase64Url(b64) {
return b64.replace(/\+/g, '-').replace(/\//g, '_').replace(/=+$/, '');
}
function hexToBase64(hexstring) {
return btoa(hexstring.match(/\w{2}/g).map(function(a) {
return String.fromCharCode(parseInt(a, 16));
}).join(""));
}
function hexToUint8Array(hex) {
if (typeof hex !== 'string' || hex.length % 2 !== 0)
throw new Error("Invalid hex string");
const bytes = new Uint8Array(hex.length / 2);
for (let i = 0; i < hex.length; i += 2) {
bytes[i / 2] = parseInt(hex.substr(i, 2), 16);
}
return bytes;
}
function generateClearKeyLicense(keys) {
return JSON.stringify({
keys: keys.map(({ k, kid }) => ({
kty: "oct",
alg: "A128KW",
k: base64ToBase64Url(hexToBase64(k)),
kid: base64ToBase64Url(hexToBase64(kid))
})),
type: "temporary"
});
}
function makeCkInitData(keys) {
const systemId = new Uint8Array([
0x10, 0x77, 0xef, 0xec,
0xc0, 0xb2,
0x4d, 0x02,
0xac, 0xe3,
0x3c, 0x1e, 0x52, 0xe2, 0xfb, 0x4b
]);
const kidCount = keys.length;
const kidDataLength = kidCount * 16;
const dataSize = 0;
const size = 4 + 4 + 4 + 16 + 4 + kidDataLength + 4 + dataSize;
const buffer = new ArrayBuffer(size);
const view = new DataView(buffer);
let offset = 0;
view.setUint32(offset, size); offset += 4;
view.setUint32(offset, 0x70737368); offset += 4; // 'pssh'
view.setUint8(offset++, 0x01); // version 1
view.setUint8(offset++, 0x00); // flags (3 bytes)
view.setUint8(offset++, 0x00);
view.setUint8(offset++, 0x00);
new Uint8Array(buffer, offset, 16).set(systemId); offset += 16;
view.setUint32(offset, kidCount); offset += 4;
for (const key of keys) {
const kidBytes = hexToUint8Array(key.kid);
if (kidBytes.length !== 16) throw new Error("Invalid KID length");
new Uint8Array(buffer, offset, 16).set(kidBytes);
offset += 16;
}
view.setUint32(offset, dataSize); offset += 4;
return new Uint8Array(buffer);
}
function emitAndWaitForResponse(type, data) {
return new Promise((resolve) => {
const requestId = Math.random().toString(16).substring(2, 9);
const responseHandler = (event) => {
const { detail } = event;
if (detail.substring(0, 7) === requestId) {
document.removeEventListener('responseReceived', responseHandler);
resolve(detail.substring(7));
}
};
document.addEventListener('responseReceived', responseHandler);
const requestEvent = new CustomEvent('response', {
detail: {
type: type,
body: data,
requestId: requestId,
}
});
document.dispatchEvent(requestEvent);
});
}
const fnproxy = (object, func) => new Proxy(object, { apply: func });
const proxy = (object, key, func) => Object.hasOwnProperty.call(object, key) && Object.defineProperty(object, key, {
value: fnproxy(object[key], func)
});
function getEventListeners(type) {
if (this == null) return [];
const store = this[Symbol.for(getEventListeners)];
if (store == null || store[type] == null) return [];
return store[type];
}
class Evaluator {
static isDASH(text) {
return text.includes('<mpd') && text.includes('</mpd>');
}
static isHLS(text) {
return text.includes('#extm3u');
}
static isHLSMaster(text) {
return text.includes('#ext-x-stream-inf');
}
static isMSS(text) {
return text.includes('<smoothstreamingmedia') && text.includes('</smoothstreamingmedia>');
}
static getManifestType(text) {
const lower = text.toLowerCase();
if (this.isDASH(lower)) {
return "DASH";
} else if (this.isHLS(lower)) {
if (this.isHLSMaster(lower)) {
return "HLS_MASTER";
} else {
return "HLS_PLAYLIST";
}
} else if (this.isMSS(lower)) {
return "MSS";
}
}
}
function sanitizeConfigForClearKey(config) {
const sanitizeCapabilities = (caps, fallbackType) => {
if (!Array.isArray(caps) || caps.length === 0) {
return [{
contentType: fallbackType,
robustness: ""
}];
}
return caps.map(c => ({
contentType: c.contentType,
robustness: "" // ClearKey only works with empty robustness
}));
};
return {
initDataTypes: config.initDataTypes || ["cenc"],
audioCapabilities: sanitizeCapabilities(config.audioCapabilities, "audio/mp4; codecs=\"mp4a.40.2\""),
videoCapabilities: sanitizeCapabilities(config.videoCapabilities, "video/mp4; codecs=\"avc1.42E01E\""),
distinctiveIdentifier: "not-allowed",
persistentState: "optional",
sessionTypes: ["temporary"]
};
}
function hookKeySystem(interface) {
const origKeySystemDescriptor = Object.getOwnPropertyDescriptor(interface.prototype, 'keySystem');
const origKeySystemGetter = origKeySystemDescriptor?.get;
if (typeof origKeySystemGetter !== 'undefined') {
Object.defineProperty(interface.prototype, 'keySystem', {
get() {
if (this._emeShim) {
console.log("[Vineless] Shimmed keySystem");
return this._emeShim.origKeySystem;
}
return origKeySystemGetter.call(this);
}
});
}
}
(async () => {
const requestMediaKeySystemAccessUnaltered = navigator.requestMediaKeySystemAccess;
if (typeof Navigator !== 'undefined') {
proxy(Navigator.prototype, 'requestMediaKeySystemAccess', async (_target, _this, _args) => {
console.log("[Vineless] requestMediaKeySystemAccess", _args);
const enabledData = JSON.parse(await emitAndWaitForResponse("GET_ENABLED"));
const origKeySystem = _args[0];
if (enabledData &&
(enabledData.wv && origKeySystem === "com.widevine.alpha") ||
(enabledData.pr && origKeySystem.startsWith("com.microsoft.playready"))
) {
_args[0] = "org.w3.clearkey";
_args[1] = [sanitizeConfigForClearKey(_args[1])];
}
const systemAccess = await _target.apply(_this, _args);
systemAccess._emeShim = {
origKeySystem
};
console.log("[Vineless] requestMediaKeySystemAccess SUCCESS", systemAccess);
return systemAccess;
});
}
if (typeof MediaCapabilities !== 'undefined') {
proxy(MediaCapabilities.prototype, 'decodingInfo', async (_target, _this, _args) => {
const enabledData = JSON.parse(await emitAndWaitForResponse("GET_ENABLED"));
const [config] = _args;
const origKeySystem = config?.keySystemConfiguration?.keySystem;
if (enabledData &&
(enabledData.wv && origKeySystem === "com.widevine.alpha") ||
(enabledData.pr && origKeySystem.startsWith("com.microsoft.playready"))
) {
console.log("[Vineless] Intercepted decodingInfo for", origKeySystem);
const ckConfig = structuredClone(config);
ckConfig.keySystemConfiguration = sanitizeConfigForClearKey(ckConfig.keySystemConfiguration);
ckConfig.keySystemConfiguration.keySystem = "org.w3.clearkey";
try {
const ckResult = await _target.call(_this, ckConfig);
// Generate a real MediaKeySystemAccess to attach
const access = await requestMediaKeySystemAccessUnaltered.call(navigator, "org.w3.clearkey", [ckConfig.keySystemConfiguration]);
// Shim .keySystem and .getConfiguration()
access._emeShim = { origKeySystem };
// Optionally also patch `getConfiguration()` to reflect original input
const originalGetConfig = access.getConfiguration.bind(access);
access.getConfiguration = () => ({
...originalGetConfig(),
videoCapabilities: config.keySystemConfiguration.videoCapabilities,
audioCapabilities: config.keySystemConfiguration.audioCapabilities,
sessionTypes: config.keySystemConfiguration.sessionTypes,
initDataTypes: config.keySystemConfiguration.initDataTypes
});
return {
...ckResult,
supported: true,
smooth: true,
powerEfficient: true,
keySystemAccess: access
};
} catch (e) {
console.warn("[Vineless] decodingInfo fallback failed:", e);
return {
supported: true,
smooth: true,
powerEfficient: false,
keySystemAccess: null
};
}
}
return _target.apply(_this, _args);
});
}
if (typeof HTMLMediaElement !== 'undefined') {
proxy(HTMLMediaElement.prototype, 'setMediaKeys', async (_target, _this, _args) => {
console.log("[Vineless] setMediaKeys", _args);
const enabledData = JSON.parse(await emitAndWaitForResponse("GET_ENABLED"));
const keys = _args[0];
const keySystem = keys?._emeShim?.origKeySystem;
if (!enabledData || !keys || !keySystem) {
return await _target.apply(_this, _args);
}
if (keySystem === "com.widevine.alpha") {
if (!enabledData.wv) {
return await _target.apply(_this, _args);
}
} else if (keySystem.startsWith("com.microsoft.playready")) {
if (!enabledData.pr) {
return await _target.apply(_this, _args);
}
} else if (keySystem !== "org.w3.clearkey") {
console.error("[Vineless] Unsupported keySystem!");
return await _target.apply(_this, _args);
}
// Replace with our own ClearKey MediaKeys
if (keys._ckConfig) {
const ckAccess = await requestMediaKeySystemAccessUnaltered.call(navigator, 'org.w3.clearkey', [keys._ckConfig]);
keys._ckKeys = await ckAccess.createMediaKeys();
console.log("[Vineless] Replaced mediaKeys with ClearKey one");
return _target.call(_this, keys._ckKeys);
}
return _target.apply(_this, _args);
});
}
if (typeof MediaKeySystemAccess !== 'undefined') {
proxy(MediaKeySystemAccess.prototype, 'createMediaKeys', async (_target, _this, _args) => {
console.log("[Vineless] createMediaKeys");
const realKeys = _target.apply(_this, _args);
realKeys.then(res => {
const config = _this.getConfiguration();
res._ckConfig = sanitizeConfigForClearKey(config);
res._emeShim = _this._emeShim;
});
return realKeys;
});
hookKeySystem(MediaKeySystemAccess);
}
if (typeof MediaKeys !== 'undefined') {
proxy(MediaKeys.prototype, 'createSession', (_target, _this, _args) => {
console.log("[Vineless] createSession");
const session = _target.apply(_this, _args);
session._mediaKeys = _this;
return session;
});
hookKeySystem(MediaKeys);
}
if (typeof MediaKeySession !== 'undefined') {
proxy(MediaKeySession.prototype, 'generateRequest', async (_target, _this, _args) => {
const enabledData = JSON.parse(await emitAndWaitForResponse("GET_ENABLED"));
const keySystem = _this._mediaKeys?._emeShim?.origKeySystem;
if (!enabledData || _this._ck || !keySystem) {
return await _target.apply(_this, _args);
}
if (keySystem === "com.widevine.alpha") {
if (!enabledData.wv) {
return await _target.apply(_this, _args);
}
} else if (keySystem.startsWith("com.microsoft.playready")) {
if (!enabledData.pr) {
return await _target.apply(_this, _args);
}
} else if (keySystem !== "org.w3.clearkey") {
console.error("[Vineless] Unsupported keySystem!");
return await _target.apply(_this, _args);
}
console.log("[Vineless] generateRequest", _args);
try {
_this.sessionId = "vl-" + Math.random().toString(36);
const base64Pssh = uint8ArrayToBase64(new Uint8Array(_args[1]));
const data = keySystem.startsWith("com.microsoft.playready") ? `pr:${_this.sessionId}:${base64Pssh}` : base64Pssh;
const challenge = await emitAndWaitForResponse("REQUEST", data);
const challengeBytes = base64toUint8Array(challenge);
const evt = new MediaKeyMessageEvent("message", {
message: challengeBytes.buffer,
messageType: "license-request"
});
_this.dispatchEvent(evt);
} catch (e) {
console.error("[Vineless] generateRequest FAILED,", e);
throw e;
}
return;
});
proxy(MediaKeySession.prototype, 'update', async (_target, _this, _args) => {
const enabledData = JSON.parse(await emitAndWaitForResponse("GET_ENABLED"));
const keySystem = _this._mediaKeys?._emeShim?.origKeySystem;
if (!enabledData || _this._ck || !keySystem) {
return await _target.apply(_this, _args);
}
if (keySystem === "com.widevine.alpha") {
if (!enabledData.wv) {
return await _target.apply(_this, _args);
}
} else if (keySystem.startsWith("com.microsoft.playready")) {
if (!enabledData.pr) {
return await _target.apply(_this, _args);
}
} else if (keySystem !== "org.w3.clearkey") {
console.error("[Vineless] Unsupported keySystem!");
return await _target.apply(_this, _args);
}
const [response] = _args;
console.log("[Vineless] update");
const base64Response = uint8ArrayToBase64(new Uint8Array(response));
const data = keySystem.startsWith("com.microsoft.playready") ? `pr:${_this.sessionId}:${base64Response}` : base64Response;
const bgResponse = await emitAndWaitForResponse("RESPONSE", data);
try {
const parsed = JSON.parse(bgResponse);
console.log(parsed, _this);
if (parsed && _this._mediaKeys._ckKeys) {
const ckLicense = generateClearKeyLicense(parsed.keys);
const ckSession = _this._mediaKeys._ckKeys.createSession();
ckSession._ck = true;
try {
await ckSession.generateRequest('cenc', parsed.pssh);
} catch {
const pssh = makeCkInitData(parsed.keys);
await ckSession.generateRequest('cenc', pssh);
}
const encoder = new TextEncoder();
const encodedLicense = encoder.encode(ckLicense);
await ckSession.update(encodedLicense);
// DSNP Fix
const fakeKeyId = parsed.keys[0].kid;
const keyIdBuffer = base64toUint8Array(hexToBase64(fakeKeyId)).buffer;
Object.defineProperty(_this, "keyStatuses", {
value: new Map([[keyIdBuffer, "usable"]]),
writable: false
});
const keyStatusEvent = new Event("keystatuseschange");
_this.dispatchEvent(keyStatusEvent);
return;
}
} catch (e) {
console.error("[Vineless] update FAILED,", e);
// If parsing failed, fall through to original Widevine path
}
return await _target.apply(_this, _args);
});
}
})();
const originalFetch = window.fetch;
window.fetch = function() {
return new Promise(async (resolve, reject) => {
originalFetch.apply(this, arguments).then((response) => {
if (response) {
response.clone().text().then((text) => {
const manifest_type = Evaluator.getManifestType(text);
if (manifest_type) {
if (arguments.length === 1) {
emitAndWaitForResponse("MANIFEST", JSON.stringify({
"url": arguments[0].url,
"type": manifest_type,
}));
} else if (arguments.length === 2) {
emitAndWaitForResponse("MANIFEST", JSON.stringify({
"url": arguments[0],
"type": manifest_type,
}));
}
}
resolve(response);
}).catch(() => {
resolve(response);
})
} else {
resolve(response);
}
}).catch(() => {
resolve();
})
})
}
const open = XMLHttpRequest.prototype.open;
XMLHttpRequest.prototype.open = function(method, url) {
this._method = method;
return open.apply(this, arguments);
};
const send = XMLHttpRequest.prototype.send;
XMLHttpRequest.prototype.send = function(postData) {
this.addEventListener('load', async function() {
if (this._method === "GET") {
let body = void 0;
switch (this.responseType) {
case "":
case "text":
body = this.responseText ?? this.response;
break;
case "json":
// TODO: untested
body = JSON.stringify(this.response);
break;
case "arraybuffer":
// TODO: untested
if (this.response.byteLength) {
const response = new Uint8Array(this.response);
body = uint8ArrayToString(new Uint8Array([...response.slice(0, 2000), ...response.slice(-2000)]));
}
break;
case "document":
// todo
break;
case "blob":
body = await this.response.text();
break;
}
if (body) {
const manifest_type = Evaluator.getManifestType(body);
if (manifest_type) {
emitAndWaitForResponse("MANIFEST", JSON.stringify({
"url": this.responseURL,
"type": manifest_type,
}));
}
}
}
});
return send.apply(this, arguments);
};