icd/rentgen
icd
/
rentgen
2
1
Fork 0
Code Issues 70 Pull Requests Projects Releases 6 Wiki Activity

Further change to make the origin -> originalURL transition work

This commit is contained in:
Kuba Orlik 2021-11-07 18:02:01 +01:00
parent dc28bd5fa2
commit e8075b384d
2 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
memory.ts
View File

@ -7,19 +7,19 @@ export default class Memory extends EventEmitter {
origin_to_history = {} as Record<string, Record<string, RequestCluster>>;
async register(request: ExtendedRequest) {
await request.init();
console.log("registering request for", request.origin);
console.log("registering request for", request.originalURL);
if (!request.isThirdParty()) {
return;
}
if (!this.origin_to_history[request.origin]) {
this.origin_to_history[request.origin] = {};
if (!this.origin_to_history[request.originalURL]) {
this.origin_to_history[request.originalURL] = {};
}
const shorthost = getshorthost(new URL(request.url).host);
if (!this.origin_to_history[request.origin][shorthost]) {
if (!this.origin_to_history[request.originalURL][shorthost]) {
const cluster = new RequestCluster(shorthost);
this.origin_to_history[request.origin][shorthost] = cluster;
this.origin_to_history[request.originalURL][shorthost] = cluster;
}
this.origin_to_history[request.origin][shorthost].add(request);
this.origin_to_history[request.originalURL][shorthost].add(request);
this.emit("change");
}

4
stolen-data-entry.ts
View File

@ -49,7 +49,7 @@ export class StolenDataEntry {
getPriority() {
let priority = 0;
priority += Math.min(this.value.length, 50);
const url = new URL(this.request.getOrigin());
const url = new URL(this.request.originalURL);
if (this.value.includes(url.host)) {
priority += 100;
}
@ -127,7 +127,7 @@ export class StolenDataEntry {
}
private classify(): keyof typeof Classifications {
if (this.value.includes(this.request.origin)) {
if (this.value.includes(this.request.originalURL)) {
return "history";
} else {
return "id";