OpenSource/reverse-engineering-anything
1
0 Fork 0
mirror of https://github.com/morluto/rea.git synced 2026-11-15 11:13:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(javascript): yield during receiver and reference classification (#1557)

Browse Source
This commit is contained in:
Ysopwnable authored and GitHub committed 2026-10-10 21:40:16 +08:00
1 parent de619bd334
commit 8c47a9ce4b
5 files changed
+109 -39
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/domain/javascript/javascriptSemanticAnalysis.ts
+42 -28
View File
@@ -11,6 +11,7 @@ import {
collectSemanticModuleLink,
collectSemanticCallable,
collectSemanticReferences,
collectSemanticReferencesSteps,
immutableSemanticBindings,
immutableSemanticScopes,
} from "./javascriptSemanticProjection.js";
@@ -31,7 +32,10 @@ import {
semanticScopeId,
semanticVariableScope,
} from "./javascriptSemanticState.js";
import { traverseJavaScriptAst } from "./javascriptSemanticTraversal.js";
import {
completeSemanticSteps,
traverseJavaScriptAstSteps,
} from "./javascriptSemanticTraversal.js";
import { semanticRequireOrigin } from "./javascriptSemanticRequireOrigin.js";
import { collectSemanticMemberMutationsSteps } from "./javascriptSemanticMemberMutations.js";
import {
@@ -99,20 +103,15 @@ export const analyzeParsedJavaScriptReferences = (
name?: string,
): readonly JavaScriptSemanticReference[] => {
const state = createState(file.program);
collectDefinitions(file.program, state);
completeSemanticSteps(collectDefinitionsSteps(file.program, state));
return collectSemanticReferences(file.program, state, name);
};
/** Recover semantics from an already parsed JavaScript artifact. */
export const analyzeParsedJavaScriptSemantics = (
file: ParsedJavaScriptSource,
): JavaScriptSemanticIr => {
const steps = analyzeParsedJavaScriptSemanticsSteps(file);
for (;;) {
const step = steps.next();
if (step.done === true) return step.value;
}
};
): JavaScriptSemanticIr =>
completeSemanticSteps(analyzeParsedJavaScriptSemanticsSteps(file));
/**
* Recover semantics from a parsed artifact with a yield between analysis
@@ -124,15 +123,13 @@ export function* analyzeParsedJavaScriptSemanticsSteps(
file: ParsedJavaScriptSource,
): Generator<void, JavaScriptSemanticIr> {
const state = createState(file.program);
collectDefinitions(file.program, state);
yield;
yield* collectDefinitionsSteps(file.program, state);
yield* collectSemanticMemberMutationsSteps(file.program, state);
traverseJavaScriptAst(file.program, {
yield* traverseJavaScriptAstSteps(file.program, {
enter: (node) => collectSemanticModuleLink(node, state),
});
yield;
const references = collectSemanticReferences(file.program, state);
yield;
const references = yield* collectSemanticReferencesSteps(file.program, state);
const parserPartial = file.errors.length > 0;
const bindings = immutableSemanticBindings(state);
yield;
@@ -193,11 +190,17 @@ export type JavaScriptOpenReceiverFact =
*/
export const classifyParsedJavaScriptOpenReceivers = (
file: ParsedJavaScriptSource,
): ReadonlyMap<number, JavaScriptOpenReceiverFact> => {
): ReadonlyMap<number, JavaScriptOpenReceiverFact> =>
completeSemanticSteps(classifyParsedJavaScriptOpenReceiversSteps(file));
/** Classify `.open` receivers while yielding during lexical and AST scans. */
export function* classifyParsedJavaScriptOpenReceiversSteps(
file: ParsedJavaScriptSource,
): Generator<void, ReadonlyMap<number, JavaScriptOpenReceiverFact>> {
const state = createState(file.program);
collectDefinitions(file.program, state);
yield* collectDefinitionsSteps(file.program, state);
const facts = new Map<number, JavaScriptOpenReceiverFact>();
traverseJavaScriptAst(file.program, {
yield* traverseJavaScriptAstSteps(file.program, {
enter: (node) => {
if (
(!t.isCallExpression(node) && !t.isOptionalCallExpression(node)) ||
@@ -273,7 +276,7 @@ export const classifyParsedJavaScriptOpenReceivers = (
},
});
return facts;
};
}
const openGlobalFact = (
name: string,
@@ -316,12 +319,18 @@ const ELECTRON_MODULE = /^electron(?:\/(?:common|main|renderer|utility))?$/u;
*/
export const classifyParsedJavaScriptElectronBindings = (
file: ParsedJavaScriptSource,
): ReadonlyMap<number, string> => {
): ReadonlyMap<number, string> =>
completeSemanticSteps(classifyParsedJavaScriptElectronBindingsSteps(file));
/** Resolve Electron aliases while yielding during lexical and AST scans. */
export function* classifyParsedJavaScriptElectronBindingsSteps(
file: ParsedJavaScriptSource,
): Generator<void, ReadonlyMap<number, string>> {
const facts = new Map<number, string>();
const state = createState(file.program);
collectDefinitions(file.program, state);
yield* collectDefinitionsSteps(file.program, state);
const exportsByBinding = new Map<string, string | null>();
traverseJavaScriptAst(file.program, {
yield* traverseJavaScriptAstSteps(file.program, {
enter: (node) => {
if (
!t.isCallExpression(node) &&
@@ -343,7 +352,7 @@ export const classifyParsedJavaScriptElectronBindings = (
},
});
return facts;
};
}
const calleeRoot = (node: t.Node): t.Identifier | undefined => {
while (t.isMemberExpression(node) || t.isOptionalMemberExpression(node)) {
@@ -416,10 +425,10 @@ const createState = (program: t.Program): JavaScriptSemanticAnalysisState => {
};
};
const collectDefinitions = (
function* collectDefinitionsSteps(
program: t.Program,
state: JavaScriptSemanticAnalysisState,
): void => {
): Generator<void, void> {
const stack: JavaScriptSemanticScopeState[] = [
currentSemanticScope(state.scopes),
];
@@ -433,7 +442,7 @@ const collectDefinitions = (
readonly scope: JavaScriptSemanticScopeState;
readonly parameters: t.Function["params"];
}[] = [];
traverseJavaScriptAst(program, {
yield* traverseJavaScriptAstSteps(program, {
enter: (node, parent, readAncestors) => {
let parentScope = currentSemanticScope(stack);
if (
@@ -516,14 +525,19 @@ const collectDefinitions = (
});
// All declarations must exist before resolving writes, including hoisted
// functions and local declarations that shadow an outer binding.
for (const { node, scope } of assignments) bindAssignment(node, scope, state);
for (const { body, scope, parameters } of functionBodies)
for (const [index, { node, scope }] of assignments.entries()) {
if (index % 64 === 0) yield;
bindAssignment(node, scope, state);
}
for (const [index, { body, scope, parameters }] of functionBodies.entries()) {
if (index % 64 === 0) yield;
for (const parameter of parameters)
for (const identifier of assignedPatternIdentifiers(
t.isTSParameterProperty(parameter) ? parameter.parameter : parameter,
))
copyParameterToBody({ identifier, body, scope, state });
};
}
}
const copyParameterToBody = (input: {
readonly identifier: t.Identifier;