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

I LOVE ASI!!! I LOVE ASI!!!!

Browse source
This commit is contained in:
velzie 2024-08-09 17:08:17 -04:00
parent 1f287f4a17
commit f9dd137e5b
No known key found for this signature in database
GPG key ID: 048413F95F0DDE1F
1 changed files with 57 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

79
rewriter/src/rewrite.rs
View file

@ -5,7 +5,7 @@ use oxc_ast::{
Visit, Visit,
}; };
use oxc_parser::Parser; use oxc_parser::Parser;
use oxc_span::{SourceType, Span}; use oxc_span::{Atom, SourceType, Span};
use oxc_syntax::operator::{AssignmentOperator, UnaryOperator}; use oxc_syntax::operator::{AssignmentOperator, UnaryOperator};
use url::Url; use url::Url;
@ -49,28 +49,63 @@ impl Rewriter {
format!("\"{}{}\"", self.config.prefix, urlencoded) format!("\"{}{}\"", self.config.prefix, urlencoded)
} }
fn rewrite_ident(&mut self, name: &Atom, span: Span) {
if UNSAFE_GLOBALS.contains(&name.to_string().as_str()) {
self.jschanges.push(JsChange::GenericChange {
span,
text: format!("({}({}))", self.config.wrapfn, name),
});
}
}
} }
impl<'a> Visit<'a> for Rewriter { impl<'a> Visit<'a> for Rewriter {
fn visit_identifier_reference(&mut self, it: &IdentifierReference<'a>) { fn visit_identifier_reference(&mut self, it: &IdentifierReference<'a>) {
// self.jschanges.push(JsChange::GenericChange { // self.jschanges.push(JsChange::GenericChange {
// span: it.span, // span: it.span,
// text: format!( // text: format!(
// "({}({}, (()=>{{ try {{return arguments}} catch(_){{}} }})()))", // "({}(typeof {} == 'undefined' || {}, (()=>{{ try {{return arguments}} catch(_){{}} }})()))",
// self.config.wrapfn, it.name // self.wrapfn, it.name, it.name
// ), // ),
// }); // });
if UNSAFE_GLOBALS.contains(&it.name.to_string().as_str()) { if UNSAFE_GLOBALS.contains(&it.name.to_string().as_str()) {
self.jschanges.push(JsChange::GenericChange { self.jschanges.push(JsChange::GenericChange {
span: it.span, span: it.span,
text: format!("({}({}))", self.config.wrapfn, it.name), text: format!("{}({})", self.config.wrapfn, it.name),
}); });
} }
} }
// we need to rewrite `new Something` to `new (wrapfn(Something))` instead of `new wrapfn(Something)`, that's why there's weird extra code here
fn visit_new_expression(&mut self, it: &oxc_ast::ast::NewExpression<'a>) {
match &it.callee {
Expression::Identifier(s) => {
self.rewrite_ident(&s.name, s.span);
return;
}
Expression::StaticMemberExpression(s) => match &s.object {
Expression::Identifier(s) => {
self.rewrite_ident(&s.name, s.span);
return;
}
_ => {}
},
Expression::ComputedMemberExpression(s) => match &s.object {
Expression::Identifier(s) => {
self.rewrite_ident(&s.name, s.span);
return;
}
_ => {}
},
_ => {}
}
walk::walk_new_expression(self, it);
}
fn visit_this_expression(&mut self, it: &oxc_ast::ast::ThisExpression) { fn visit_this_expression(&mut self, it: &oxc_ast::ast::ThisExpression) {
self.jschanges.push(JsChange::GenericChange { self.jschanges.push(JsChange::GenericChange {
span: it.span, span: it.span,
text: format!("({}(this))", self.config.wrapfn), text: format!("{}(this)", self.config.wrapfn),
}); });
} }
@ -184,19 +219,19 @@ impl<'a> Visit<'a> for Rewriter {
walk::walk_object_expression(self, it); walk::walk_object_expression(self, it);
} }
// fn visit_return_statement(&mut self, it: &oxc_ast::ast::ReturnStatement<'a>) { fn visit_return_statement(&mut self, it: &oxc_ast::ast::ReturnStatement<'a>) {
// if let Some(arg) = &it.argument { // if let Some(arg) = &it.argument {
// self.jschanges.push(JsChange::GenericChange { // self.jschanges.push(JsChange::GenericChange {
// span: Span::new(it.span.start + 6, it.span.start + 6), // span: Span::new(it.span.start + 6, it.span.start + 6),
// text: format!(" $scramdbg((()=>{{ try {{return arguments}} catch(_){{}} }})(),("), // text: format!(" $scramdbg((()=>{{ try {{return arguments}} catch(_){{}} }})(),("),
// }); // });
// self.jschanges.push(JsChange::GenericChange { // self.jschanges.push(JsChange::GenericChange {
// span: Span::new(expression_span(arg).end, expression_span(arg).end), // span: Span::new(expression_span(arg).end, expression_span(arg).end),
// text: format!("))"), // text: format!("))"),
// }); // });
// } // }
// // walk::walk_return_statement(self, it); walk::walk_return_statement(self, it);
// } }
fn visit_unary_expression(&mut self, it: &oxc_ast::ast::UnaryExpression<'a>) { fn visit_unary_expression(&mut self, it: &oxc_ast::ast::UnaryExpression<'a>) {
if matches!(it.operator, UnaryOperator::Typeof) { if matches!(it.operator, UnaryOperator::Typeof) {
@ -404,7 +439,7 @@ pub fn rewrite(js: &str, url: Url, config: Config) -> Vec<u8> {
buffer.extend_from_slice( buffer.extend_from_slice(
format!( format!(
"((t)=>$scramjet$tryset({},\"{}\",t)||({}{}t))({})", "((t)=>$scramjet$tryset({},\"{}\",t)||{}{}t)({})",
name, name,
fmt_op(*op), fmt_op(*op),
name, name,