129 lines
3.3 KiB
TypeScript
129 lines
3.3 KiB
TypeScript
import { StolenDataEntry } from "./request-cluster";
|
|
import { getshorthost, parseCookie, Request } from "./util";
|
|
|
|
export default class ExtendedRequest {
|
|
public tabId: number;
|
|
public url: string;
|
|
public requestHeaders: Request["requestHeaders"];
|
|
public origin: string;
|
|
public initialized = false;
|
|
|
|
async init() {
|
|
await this.cacheOrigin();
|
|
this.initialized = true;
|
|
}
|
|
|
|
async cacheOrigin(): Promise<void> {
|
|
let url: string;
|
|
if (this.data.tabId && this.data.tabId >= 0) {
|
|
const tab = await browser.tabs.get(this.data.tabId);
|
|
url = tab.url;
|
|
} else {
|
|
url = (this.data as any).frameAncestors[0].url;
|
|
}
|
|
this.origin = url;
|
|
}
|
|
|
|
getOrigin(): string {
|
|
if (!this.initialized) {
|
|
throw new Error("initialize first!!");
|
|
}
|
|
return this.origin;
|
|
}
|
|
|
|
isThirdParty() {
|
|
const request_url = new URL(this.data.url);
|
|
const origin_url = new URL(this.getOrigin());
|
|
if (request_url.host.includes(origin_url.host)) {
|
|
return false;
|
|
}
|
|
if (getshorthost(request_url.host) == getshorthost(origin_url.host)) {
|
|
return false;
|
|
}
|
|
return (
|
|
request_url.origin != origin_url.origin ||
|
|
(this.data as any).urlClassification.thirdParty.length > 0
|
|
);
|
|
}
|
|
|
|
getReferer() {
|
|
return this.data.requestHeaders.filter((h) => h.name === "Referer")[0]
|
|
.value;
|
|
}
|
|
|
|
exposesOrigin() {
|
|
const url = new URL(this.getOrigin());
|
|
const host = url.host;
|
|
const path = url.pathname;
|
|
return (
|
|
this.getReferer().includes(host) ||
|
|
this.getAllStolenData().filter(
|
|
(entry) => entry.value.includes(host) || entry.value.includes(path)
|
|
).length > 0
|
|
);
|
|
}
|
|
|
|
getAllStolenData(): StolenDataEntry[] {
|
|
return [
|
|
...this.getPathParams(),
|
|
...this.getCookieData(),
|
|
...this.getQueryParams(),
|
|
];
|
|
}
|
|
|
|
getCookieData(): StolenDataEntry[] {
|
|
if (!this.hasCookie() || this.getCookie() === undefined) {
|
|
return [];
|
|
}
|
|
return Object.entries(parseCookie(this.getCookie()))
|
|
.map(([key, value]) => [key, value || ""])
|
|
.map(([key, value]) => new StolenDataEntry(this, "cookie", key, value));
|
|
}
|
|
|
|
hasReferer() {
|
|
return this.data.requestHeaders.some((h) => h.name === "Referer");
|
|
}
|
|
|
|
hasCookie() {
|
|
return this.data.requestHeaders.some((h) => h.name === "Cookie");
|
|
}
|
|
|
|
getCookie(): string {
|
|
return this.requestHeaders.find((h) => h.name == "Cookie")?.value;
|
|
}
|
|
|
|
getPathParams(): StolenDataEntry[] {
|
|
const url = new URL(this.data.url);
|
|
const path = url.pathname;
|
|
if (!path.includes(";")) {
|
|
return [];
|
|
}
|
|
return path
|
|
.split(";")
|
|
.map((e) => e.split("="))
|
|
.map(([key, value]) => [key, value || ""])
|
|
.map(
|
|
([key, value]) =>
|
|
new StolenDataEntry(this, "pathname", key, decodeURIComponent(value))
|
|
);
|
|
}
|
|
|
|
getQueryParams(): StolenDataEntry[] {
|
|
const url = new URL(this.data.url);
|
|
return Array.from((url.searchParams as any).entries())
|
|
.map(([key, value]) => [key, value || ""])
|
|
.map(([key, value]) => {
|
|
try {
|
|
value = decodeURIComponent(value);
|
|
} catch (e) {}
|
|
return new StolenDataEntry(this, "queryparams", key, value);
|
|
});
|
|
}
|
|
|
|
constructor(public data: Request) {
|
|
this.tabId = data.tabId;
|
|
this.url = data.url;
|
|
this.requestHeaders = data.requestHeaders;
|
|
}
|
|
}
|