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

fix rewriter

Browse source
This commit is contained in:
velzie 2024-07-15 08:12:45 -04:00
parent f53bc623ff
commit 3462785c39
No known key found for this signature in database
GPG key ID: 048413F95F0DDE1F
3 changed files with 89 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

2
rewriter/src/main_.rs → rewriter/src/main.rs
View file

@ -25,7 +25,7 @@ fn main() -> std::io::Result<()> {
let path = Path::new(&name);
let source_text = std::fs::read_to_string(path)?;
dbg!(rewrite(&source_text));
println!("{}", rewrite(&source_text));
Ok(())
}

83
rewriter/src/rewrite.rs
View file

@ -1,6 +1,9 @@
use oxc_allocator::Allocator;
use oxc_ast::{
ast::{AssignmentTarget, Class, Function, IdentifierReference, MemberExpression, TSImportType},
ast::{
AssignmentTarget, Class, Expression, Function, IdentifierReference, MemberExpression,
TSImportType,
},
visit::walk,
Visit,
};
@ -55,19 +58,85 @@ impl<'a> Visit<'a> for Rewriter {
// }
// }
fn visit_member_expression(&mut self, it: &MemberExpression<'a>) {
match it {
MemberExpression::StaticMemberExpression(s) => {
if s.property.name.to_string() == "location" {
self.trace_member(it);
// match it {
// MemberExpression::StaticMemberExpression(s) => {
// dbg!(s);
// if s.property.name.to_string() == "location" {
// self.jschanges.push(JsChange::GenericChange {
// span: s.property.span,
// text: "$s(location)".to_string(),
// });
// }
// }
// _ => {}
// }
}
}
// js MUST not be able to get a reference to any of these because sbx
const UNSAFE_GLOBALS: [&str; 7] = [
"window",
"self",
"globalThis",
"this",
"parent",
"top",
"location",
];
impl Rewriter {
fn trace_member<'a>(&mut self, it: &MemberExpression<'a>) {
match &it {
MemberExpression::StaticMemberExpression(s) => match &s.object {
Expression::Identifier(obj) => {
dbg!(obj);
if UNSAFE_GLOBALS.contains(&obj.name.to_string().as_str()) {
self.jschanges.push(JsChange::GenericChange {
span: s.property.span,
text: "$s(location)".to_string(),
span: obj.span,
text: format!("$s({})", obj.name),
});
}
}
Expression::ThisExpression(obj) => {
self.jschanges.push(JsChange::GenericChange {
span: obj.span,
text: "$s(this)".to_string(),
});
}
_ => {
if it.object().is_member_expression() {
self.trace_member(it.object().as_member_expression().unwrap());
}
}
},
MemberExpression::ComputedMemberExpression(s) => match &s.object {
Expression::Identifier(obj) => {
dbg!(obj);
if UNSAFE_GLOBALS.contains(&obj.name.to_string().as_str()) {
self.jschanges.push(JsChange::GenericChange {
span: obj.span,
text: format!("$s({})", obj.name),
});
}
}
Expression::ThisExpression(obj) => {
self.jschanges.push(JsChange::GenericChange {
span: obj.span,
text: "$s(this)".to_string(),
});
}
_ => {
if it.object().is_member_expression() {
self.trace_member(it.object().as_member_expression().unwrap());
}
}
},
_ => {}
}
}
}
pub fn rewrite(js: &str) -> String {
let source_text = js.to_string();
let allocator = Allocator::default();
@ -81,6 +150,8 @@ pub fn rewrite(js: &str) -> String {
let program = ret.program;
dbg!(&program);
let mut ast_pass = Rewriter::default();
ast_pass.visit_program(&program);

10
rewriter/test.js Normal file
View file

@ -0,0 +1,10 @@
window.location.href = "http://example.com"
console.log(top.window.aaa)
consle.log(globalThis["win" + "dow"])
globalThis.eval("..")
let ref = { b: this.top.window, c: globalThis["win" + "dow"] }