mercuryworkshop-mirror/scramjet
1
0
Fork 0
mirror of https://github.com/MercuryWorkshop/scramjet.git synced 2025-05-14 06:50:01 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix unrewriting stolen functions

Browse source
This commit is contained in:
Toshit Chawda 2025-03-09 13:26:15 -07:00
parent 3948171dc6
commit ad73d45b45
No known key found for this signature in database
GPG key ID: 91480ED99E2B3D9D
4 changed files with 46 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

33
src/client/shared/sourcemaps.ts
View file

@ -1,4 +1,5 @@
import { flagEnabled } from "../../scramjet";
import { SCRAMJETCLIENT, SCRAMJETCLIENTNAME } from "../../symbols";
import { ProxyCtx, ScramjetClient } from "../client";
enum RewriteType {
@ -20,6 +21,8 @@ type Rewrite = {
}
);
export type SourceMaps = Record<string, Rewrite[]>;
function getEnd(rewrite: Rewrite): number {
if (rewrite.type === RewriteType.Insert) {
return rewrite.start + rewrite.size;
@ -31,6 +34,30 @@ function getEnd(rewrite: Rewrite): number {
const scramtag_ident = "/*scramtag ";
// some sites like to steal funcs from frames and then unrewrite them
function searchRewrites(tag: string): Rewrite[] | undefined {
function searchFrame(globalThis: Self) {
const SCRAMJETCLIENT = globalThis.Symbol.for(SCRAMJETCLIENTNAME);
if (globalThis[SCRAMJETCLIENT].sourcemaps[tag])
return globalThis[SCRAMJETCLIENT].sourcemaps[tag];
// no enhanced for :frowning2:
for (let i = 0; i < globalThis.frames.length; i++) {
const rewrites = searchFrame(globalThis.frames[i].self);
if (rewrites) return rewrites;
}
}
let globalThis = self;
let rewrites = searchFrame(globalThis);
if (rewrites) return rewrites;
while (globalThis.parent && globalThis.parent !== globalThis.window) {
globalThis = globalThis.parent.self;
let rewrites = searchFrame(globalThis);
if (rewrites) return rewrites;
}
}
function registerRewrites(buf: Array<number>, tag: string) {
const sourcemap = Uint8Array.from(buf);
const view = new DataView(sourcemap.buffer);
@ -65,7 +92,7 @@ function registerRewrites(buf: Array<number>, tag: string) {
}
}
sourcemaps[tag] = rewrites;
self[SCRAMJETCLIENT].sourcemaps[tag] = rewrites;
}
function doUnrewrite(ctx: ProxyCtx) {
@ -94,7 +121,7 @@ function doUnrewrite(ctx: ProxyCtx) {
const scramtagend = stringified.indexOf("*/", scramtagstart);
const tag = stringified.substring(firstspace + 1, scramtagend);
const rewrites = sourcemaps[tag];
const rewrites = searchRewrites(tag);
if (!rewrites) {
console.warn("failed to get rewrites for tag", tag);
@ -137,8 +164,6 @@ function doUnrewrite(ctx: ProxyCtx) {
return ctx.return(newString);
}
const sourcemaps: Record<string, Rewrite[]> = {};
export const enabled = (client: ScramjetClient) =>
flagEnabled("sourcemaps", client.url);