diff options
Diffstat (limited to 'dom/media/test/eme.js')
| -rw-r--r-- | dom/media/test/eme.js | 479 |
1 files changed, 479 insertions, 0 deletions
diff --git a/dom/media/test/eme.js b/dom/media/test/eme.js new file mode 100644 index 0000000000..927c99876a --- /dev/null +++ b/dom/media/test/eme.js @@ -0,0 +1,479 @@ +/* import-globals-from manifest.js */ + +const CLEARKEY_KEYSYSTEM = "org.w3.clearkey"; + +const gCencMediaKeySystemConfig = [ + { + initDataTypes: ["cenc"], + videoCapabilities: [{ contentType: "video/mp4" }], + audioCapabilities: [{ contentType: "audio/mp4" }], + }, +]; + +function bail(message) { + return function (err) { + if (err) { + message += "; " + String(err); + } + ok(false, message); + if (err) { + info(String(err)); + } + SimpleTest.finish(); + }; +} + +function ArrayBufferToString(arr) { + var str = ""; + var view = new Uint8Array(arr); + for (var i = 0; i < view.length; i++) { + str += String.fromCharCode(view[i]); + } + return str; +} + +function StringToArrayBuffer(str) { + var arr = new ArrayBuffer(str.length); + var view = new Uint8Array(arr); + for (var i = 0; i < str.length; i++) { + view[i] = str.charCodeAt(i); + } + return arr; +} + +function StringToHex(str) { + var res = ""; + for (var i = 0; i < str.length; ++i) { + res += ("0" + str.charCodeAt(i).toString(16)).slice(-2); + } + return res; +} + +function Base64ToHex(str) { + var bin = window.atob(str.replace(/-/g, "+").replace(/_/g, "/")); + var res = ""; + for (var i = 0; i < bin.length; i++) { + res += ("0" + bin.charCodeAt(i).toString(16)).substr(-2); + } + return res; +} + +function HexToBase64(hex) { + var bin = ""; + for (var i = 0; i < hex.length; i += 2) { + bin += String.fromCharCode(parseInt(hex.substr(i, 2), 16)); + } + return window + .btoa(bin) + .replace(/=/g, "") + .replace(/\+/g, "-") + .replace(/\//g, "_"); +} + +function TimeRangesToString(trs) { + var l = trs.length; + if (l === 0) { + return "-"; + } + var s = ""; + var i = 0; + for (;;) { + s += trs.start(i) + "-" + trs.end(i); + if (++i === l) { + return s; + } + s += ","; + } +} + +function SourceBufferToString(sb) { + return ( + "SourceBuffer{" + + "AppendMode=" + + (sb.AppendMode || "-") + + ", updating=" + + (sb.updating ? "true" : "false") + + ", buffered=" + + TimeRangesToString(sb.buffered) + + ", audioTracks=" + + (sb.audioTracks ? sb.audioTracks.length : "-") + + ", videoTracks=" + + (sb.videoTracks ? sb.videoTracks.length : "-") + + "}" + ); +} + +function SourceBufferListToString(sbl) { + return "SourceBufferList[" + sbl.map(SourceBufferToString).join(", ") + "]"; +} + +function GenerateClearKeyLicense(licenseRequest, keyStore) { + var msgStr = ArrayBufferToString(licenseRequest); + var msg = JSON.parse(msgStr); + + var keys = []; + for (var i = 0; i < msg.kids.length; i++) { + var id64 = msg.kids[i]; + var idHex = Base64ToHex(msg.kids[i]).toLowerCase(); + var key = keyStore[idHex]; + + if (key) { + keys.push({ + kty: "oct", + kid: id64, + k: HexToBase64(key), + }); + } + } + + return new TextEncoder().encode( + JSON.stringify({ + keys, + type: msg.type || "temporary", + }) + ); +} + +function UpdateSessionFunc(test, token, sessionType, resolve, reject) { + return function (ev) { + var license = GenerateClearKeyLicense(ev.message, test.keys); + Log( + token, + "sending update message to CDM: " + new TextDecoder().decode(license) + ); + ev.target + .update(license) + .then(function () { + Log(token, "MediaKeySession update ok!"); + resolve(ev.target); + }) + .catch(function (reason) { + reject(`${token} MediaKeySession update failed: ${reason}`); + }); + }; +} + +function MaybeCrossOriginURI(test, uri) { + if (test.crossOrigin) { + return "https://example.com:443/tests/dom/media/test/allowed.sjs?" + uri; + } + return uri; +} + +function AppendTrack(test, ms, track, token) { + return new Promise(function (resolve, reject) { + var sb; + var curFragment = 0; + var fragments = track.fragments; + var fragmentFile; + + function addNextFragment() { + if (curFragment >= fragments.length) { + Log(token, track.name + ": end of track"); + resolve(); + return; + } + + fragmentFile = MaybeCrossOriginURI(test, fragments[curFragment++]); + + var req = new XMLHttpRequest(); + req.open("GET", fragmentFile); + req.responseType = "arraybuffer"; + + req.addEventListener("load", function () { + Log( + token, + track.name + ": fetch of " + fragmentFile + " complete, appending" + ); + sb.appendBuffer(new Uint8Array(req.response)); + }); + + req.addEventListener("error", function () { + reject(`${token} - ${track.name}: error fetching ${fragmentFile}`); + }); + req.addEventListener("abort", function () { + reject(`${token} - ${track.name}: aborted fetching ${fragmentFile}`); + }); + + Log( + token, + track.name + + ": addNextFragment() fetching next fragment " + + fragmentFile + ); + req.send(null); + } + + Log(token, track.name + ": addSourceBuffer(" + track.type + ")"); + sb = ms.addSourceBuffer(track.type); + sb.addEventListener("updateend", function () { + Log( + token, + track.name + + ": updateend for " + + fragmentFile + + ", " + + SourceBufferToString(sb) + ); + addNextFragment(); + }); + + addNextFragment(); + }); +} + +//Returns a promise that is resolved when the media element is ready to have +//its play() function called; when it's loaded MSE fragments. +function LoadTest(test, elem, token, endOfStream = true) { + if (!test.tracks) { + ok(false, token + " test does not have a tracks list"); + return Promise.reject(); + } + + var ms = new MediaSource(); + elem.src = URL.createObjectURL(ms); + elem.crossOrigin = test.crossOrigin || false; + + return new Promise(function (resolve, reject) { + ms.addEventListener( + "sourceopen", + function () { + Log(token, "sourceopen"); + Promise.all( + test.tracks.map(function (track) { + return AppendTrack(test, ms, track, token); + }) + ) + .then(function () { + Log(token, "Tracks loaded, calling MediaSource.endOfStream()"); + if (endOfStream) { + ms.endOfStream(); + } + resolve(); + }) + .catch(reject); + }, + { once: true } + ); + }); +} + +function EMEPromise() { + var self = this; + self.promise = new Promise(function (resolve, reject) { + self.resolve = resolve; + self.reject = reject; + }); +} + +/* + * Create a new MediaKeys object. + * Return a promise which will be resolved with a new MediaKeys object, + * or will be rejected with a string that describes the failure. + */ +function CreateMediaKeys(v, test, token) { + let p = new EMEPromise(); + + function streamType(type) { + var x = test.tracks.find(o => o.name == type); + return x ? x.type : undefined; + } + + function onencrypted(ev) { + var options = { initDataTypes: [ev.initDataType] }; + if (streamType("video")) { + options.videoCapabilities = [{ contentType: streamType("video") }]; + } + if (streamType("audio")) { + options.audioCapabilities = [{ contentType: streamType("audio") }]; + } + navigator.requestMediaKeySystemAccess(CLEARKEY_KEYSYSTEM, [options]).then( + keySystemAccess => { + keySystemAccess + .createMediaKeys() + .then(p.resolve, () => + p.reject(`${token} Failed to create MediaKeys object.`) + ); + }, + () => p.reject(`${token} Failed to request key system access.`) + ); + } + + v.addEventListener("encrypted", onencrypted, { once: true }); + return p.promise; +} + +/* + * Create a new MediaKeys object and provide it to the media element. + * Return a promise which will be resolved if succeeded, or will be rejected + * with a string that describes the failure. + */ +function CreateAndSetMediaKeys(v, test, token) { + let p = new EMEPromise(); + + CreateMediaKeys(v, test, token).then(mediaKeys => { + v.setMediaKeys(mediaKeys).then(p.resolve, () => + p.reject(`${token} Failed to set MediaKeys on <video> element.`) + ); + }, p.reject); + + return p.promise; +} + +/* + * Collect the init data from 'encrypted' events. + * Return a promise which will be resolved with the init data when collection + * is completed (specified by test.sessionCount). + */ +function LoadInitData(v, test, token) { + let p = new EMEPromise(); + let initDataQueue = []; + + // Call SimpleTest._originalSetTimeout() to bypass the flaky timeout checker. + let timer = SimpleTest._originalSetTimeout.call( + window, + () => { + p.reject(`${token} Timed out in waiting for the init data.`); + }, + 60000 + ); + + function onencrypted(ev) { + initDataQueue.push(ev); + Log( + token, + `got encrypted(${ev.initDataType}, ` + + `${StringToHex(ArrayBufferToString(ev.initData))}) event.` + ); + if (test.sessionCount == initDataQueue.length) { + p.resolve(initDataQueue); + clearTimeout(timer); + } + } + + v.addEventListener("encrypted", onencrypted); + return p.promise; +} + +/* + * Generate a license request and update the session. + * Return a promsise which will be resolved with the updated session + * or rejected with a string that describes the failure. + */ +function MakeRequest(test, token, ev, session, sessionType) { + sessionType = sessionType || "temporary"; + let p = new EMEPromise(); + let str = + `session[${session.sessionId}].generateRequest(` + + `${ev.initDataType}, ${StringToHex(ArrayBufferToString(ev.initData))})`; + + session.addEventListener( + "message", + UpdateSessionFunc(test, token, sessionType, p.resolve, p.reject) + ); + + Log(token, str); + session.generateRequest(ev.initDataType, ev.initData).catch(reason => { + // Reject the promise if generateRequest() failed. + // Otherwise it will be resolved in UpdateSessionFunc(). + p.reject(`${token}: ${str} failed; ${reason}`); + }); + + return p.promise; +} + +/* + * Process the init data by calling MakeRequest(). + * Return a promise which will be resolved with the updated sessions + * when all init data are processed or rejected if any failure. + */ +function ProcessInitData(v, test, token, initData, sessionType) { + return Promise.all( + initData.map(ev => { + let session = v.mediaKeys.createSession(sessionType); + return MakeRequest(test, token, ev, session, sessionType); + }) + ); +} + +/* + * Clean up the |v| element. + */ +function CleanUpMedia(v) { + v.setMediaKeys(null); + v.remove(); + v.removeAttribute("src"); + v.load(); +} + +/* + * Close all sessions and clean up the |v| element. + */ +function CloseSessions(v, sessions) { + return Promise.all(sessions.map(s => s.close())).then(CleanUpMedia(v)); +} + +/* + * Set up media keys and source buffers for the media element. + * Return a promise resolved when all key sessions are updated or rejected + * if any failure. + */ +function SetupEME(v, test, token) { + let p = new EMEPromise(); + + v.onerror = function () { + p.reject(`${token} got an error event.`); + }; + + Promise.all([ + LoadInitData(v, test, token), + CreateAndSetMediaKeys(v, test, token), + LoadTest(test, v, token), + ]) + .then(values => { + let initData = values[0]; + return ProcessInitData(v, test, token, initData); + }) + .then(p.resolve, p.reject); + + return p.promise; +} + +function fetchWithXHR(uri, onLoadFunction) { + var p = new Promise(function (resolve, reject) { + var xhr = new XMLHttpRequest(); + xhr.open("GET", uri, true); + xhr.responseType = "arraybuffer"; + xhr.addEventListener("load", function () { + is( + xhr.status, + 200, + "fetchWithXHR load uri='" + uri + "' status=" + xhr.status + ); + resolve(xhr.response); + }); + xhr.send(); + }); + + if (onLoadFunction) { + p.then(onLoadFunction); + } + + return p; +} + +function once(target, name, cb) { + var p = new Promise(function (resolve, reject) { + target.addEventListener( + name, + function (arg) { + resolve(arg); + }, + { once: true } + ); + }); + if (cb) { + p.then(cb); + } + return p; +} |