sam_oneal/fx_cast
1
0
Fork 0
mirror of https://github.com/hensm/fx_cast.git synced 2026-06-12 02:29:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rewrite User-Agent header for child resources on white-listed URLs

Browse Source
This commit is contained in:
hensm
2020-02-26 15:15:35 +00:00
parent e645e535dd
commit b0e2e546f1
1 changed files with 43 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
ext/src/background/background.ts
View File

@@ -472,6 +472,10 @@ async function initWhitelist () {
type OnBeforeSendHeadersDetails = Parameters<Parameters< type OnBeforeSendHeadersDetails = Parameters<Parameters<
typeof browser.webRequest.onBeforeSendHeaders.addListener>[0]>[0]; typeof browser.webRequest.onBeforeSendHeaders.addListener>[0]>[0];
const originUrlCache: string[] = [];
const chromeUserAgent = getChromeUserAgent(
(await browser.runtime.getPlatformInfo()).os, true);
/** /**
* Web apps usually only load the sender library and * Web apps usually only load the sender library and
* provide cast functionality if the browser is detected * provide cast functionality if the browser is detected
@@ -483,25 +487,18 @@ async function initWhitelist () {
throw logger.error("OnBeforeSendHeaders handler details missing requestHeaders."); throw logger.error("OnBeforeSendHeaders handler details missing requestHeaders.");
} }
if (details.originUrl && !originUrlCache.includes(details.originUrl)) {
originUrlCache.push(details.originUrl);
}
const host = details.requestHeaders.find( const host = details.requestHeaders.find(
header => header.name === "Host"); header => header.name === "Host");
for (const header of details.requestHeaders) { for (const header of details.requestHeaders) {
const { os } = await browser.runtime.getPlatformInfo(); const { os } = await browser.runtime.getPlatformInfo();
if (header.name.toLowerCase() === "user-agent") { if (header.name === "User-Agent") {
/** header.value = chromeUserAgent;
* New YouTube breaks without the default user agent string,
* so use hybrid Firefox/Chrome version.
*
* TODO: Test this on other sites
*/
if (host?.value === "www.youtube.com") {
header.value = getChromeUserAgent(os, true);
break;
}
header.value = getChromeUserAgent(os);
break; break;
} }
} }
@@ -511,6 +508,27 @@ async function initWhitelist () {
}; };
} }
function handleResourceRequests (details: OnBeforeSendHeadersDetails) {
if (!details.requestHeaders) {
return;
}
for (const ancestor of details.frameAncestors) {
if (originUrlCache.includes(ancestor.url)) {
console.log(details);
for (const header of details.requestHeaders) {
if (header.name === "User-Agent") {
header.value = chromeUserAgent;
}
}
return {
requestHeaders: details.requestHeaders
};
}
}
}
async function registerUserAgentWhitelist () { async function registerUserAgentWhitelist () {
const { userAgentWhitelist const { userAgentWhitelist
, userAgentWhitelistEnabled } = await options.getAll(); , userAgentWhitelistEnabled } = await options.getAll();
@@ -518,6 +536,12 @@ async function initWhitelist () {
if (!userAgentWhitelistEnabled) { if (!userAgentWhitelistEnabled) {
return; return;
} }
browser.webRequest.onBeforeSendHeaders.addListener(
handleResourceRequests
, { urls: [ "<all_urls>" ] }
, [ "blocking", "requestHeaders" ]);
browser.webRequest.onBeforeSendHeaders.addListener( browser.webRequest.onBeforeSendHeaders.addListener(
onBeforeSendHeaders onBeforeSendHeaders
, { urls: userAgentWhitelist } , { urls: userAgentWhitelist }
@@ -525,8 +549,12 @@ async function initWhitelist () {
} }
function unregisterUserAgentWhitelist () { function unregisterUserAgentWhitelist () {
browser.webRequest.onBeforeSendHeaders.removeListener( originUrlCache.length = 0;
onBeforeSendHeaders);
browser.webRequest.onBeforeSendHeaders
.removeListener(handleResourceRequests);
browser.webRequest.onBeforeSendHeaders
.removeListener(onBeforeSendHeaders);
} }
// Register on first run // Register on first run