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

return real window.parent

Browse source
This commit is contained in:
velzie 2024-07-27 18:55:59 -04:00
parent 396170e4ea
commit b0f5166916
No known key found for this signature in database
GPG key ID: 048413F95F0DDE1F
4 changed files with 62 additions and 38 deletions
Download patch file Download diff file Expand all files Collapse all files

81
src/client/shared/wrap.ts
View file

@ -14,40 +14,65 @@ export default function (client: ScramjetClient, self: typeof globalThis) {
// the main magic of the proxy. all attempts to access any "banned objects" will be redirected here, and instead served a proxy object
// this contrasts from how other proxies will leave the root object alone and instead attempt to catch every member access
// this presents some issues (see element.ts), but makes us a good bit faster at runtime!
Object.defineProperty(self, wrapfn, {
value: function (identifier: any) {
if (iswindow && identifier instanceof self.Window) {
return client.windowProxy;
} else if (iswindow && identifier instanceof self.parent.self.Window) {
if (ScramjetClient.SCRAMJET in self.parent.self) {
// ... then we're in a subframe, and the parent frame is also in a proxy context, so we should return its proxy
return self.parent.self[ScramjetClient.SCRAMJET].windowProxy;
} else {
// ... then we should pretend we aren't nested and return the current window
return client.windowProxy;
}
} else if (iswindow && identifier instanceof self.top.self.Window) {
// instead of returning top, we need to return the uppermost parent that's inside a scramjet context
let current = self.self;
self[wrapfn] = function (identifier: any) {
if (
iswindow &&
(identifier instanceof self.Window ||
identifier instanceof self.top.window.Window ||
identifier instanceof self.parent.window.Window)
) {
// this will break iframe postmessage!
return client.windowProxy;
} else if (
(iswindow && identifier instanceof Location) ||
(isworker && identifier instanceof WorkerLocation)
) {
return client.locationProxy;
} else if (iswindow && identifier instanceof Document) {
return client.documentProxy;
} else if (isworker && identifier instanceof WorkerGlobalScope) {
return client.windowProxy;
}
for (;;) {
const test = current.parent.self;
if (test === current) break; // there is no parent, actual or emulated.
return identifier;
};
// ... then `test` represents a window outside of the proxy context, and therefore `current` is the topmost window in the proxy context
if (!(ScramjetClient.SCRAMJET in test)) break;
// test is also insde a proxy, so we should continue up the chain
current = test;
}
return current[ScramjetClient.SCRAMJET].windowProxy;
} else if (
(iswindow && identifier instanceof self.Location) ||
(isworker && identifier instanceof self.WorkerLocation)
) {
return client.locationProxy;
} else if (iswindow && identifier instanceof self.Document) {
return client.documentProxy;
} else if (isworker && identifier instanceof self.WorkerGlobalScope) {
return client.windowProxy;
}
return identifier;
},
writable: false,
configurable: false,
});
// location = "..." can't be rewritten as wrapfn(location) = ..., so instead it will actually be rewritten as
// ((t)=>$scramjet$tryset(location,"+=",t)||location+=t)(...);
// it has to be a discrete function because there's always the possibility that "location" is a local variable
// we have to use an IIFE to avoid duplicating side-effects in the getter
self[trysetfn] = function (lhs: any, op: string, rhs: any) {
if (lhs instanceof Location) {
// @ts-ignore
locationProxy.href = rhs;
Object.defineProperty(self, trysetfn, {
value: function (lhs: any, op: string, rhs: any) {
if (lhs instanceof Location) {
// @ts-ignore
locationProxy.href = rhs;
return true;
}
};
return true;
}
},
writable: false,
configurable: false,
});
}