luau-obfuscator 1.0.0 → 1.0.1

package/src/passes/WrapInFunction.ts

@@ -1,32 +1,32 @@
-import type { Program } from "luau-parser"
-import {
-    block, functionBody, functionExpression, paren, call, vararg, returnStatement,
-} from "./nodeFactory"
-
-export interface WrapInFunctionOptions {}
-
-/**
- * 전체 프로그램을:
- *   return (function(...)
- *       <원래 코드>
- *   end)(...)
- * 로 감쌈.
- *
- * - `return`을 쓰는 이유: Script/LocalScript에서는 top-level return이 그냥 청크를
- *   조기 종료시킬 뿐 무해하고, ModuleScript라면 IIFE의 결과값이 그대로 require()
- *   호출자에게 전달돼야 하므로 필요함. 즉 스크립트 종류를 가리지 않고 안전.
- * - `...`을 파라미터로 받아서 다시 그대로 넘겨주는 이유: 원본 청크가 최상위에서
- *   `...`(스크립트 인자)을 참조하는 경우를 대비. 그냥 지워버리면 그런 코드가
- *   깨짐.
- *
- * 반드시 파이프라인 맨 마지막 근처에서 실행돼야 함 — 이 패스 이후에 실행되는
- * 다른 패스가 top-level 스코프를 순회/변형한다면 이미 감싸인 함수 내부까지
- * 안 보고 지나칠 수 있음.
- */
-export function runWrapInFunction(program: Program, _options: WrapInFunctionOptions): void {
-    const innerBody = program.body
-    const wrapper = functionExpression(functionBody([], innerBody, true))
-    const iife = call(paren(wrapper), [vararg()])
-
-    program.body = block([returnStatement([iife])])
+import type { Program } from "luau-parser"
+import {
+    block, functionBody, functionExpression, paren, call, vararg, returnStatement,
+} from "./nodeFactory"
+
+export interface WrapInFunctionOptions {}
+
+/**
+ * 전체 프로그램을:
+ *   return (function(...)
+ *       <원래 코드>
+ *   end)(...)
+ * 로 감쌈.
+ *
+ * - `return`을 쓰는 이유: Script/LocalScript에서는 top-level return이 그냥 청크를
+ *   조기 종료시킬 뿐 무해하고, ModuleScript라면 IIFE의 결과값이 그대로 require()
+ *   호출자에게 전달돼야 하므로 필요함. 즉 스크립트 종류를 가리지 않고 안전.
+ * - `...`을 파라미터로 받아서 다시 그대로 넘겨주는 이유: 원본 청크가 최상위에서
+ *   `...`(스크립트 인자)을 참조하는 경우를 대비. 그냥 지워버리면 그런 코드가
+ *   깨짐.
+ *
+ * 반드시 파이프라인 맨 마지막 근처에서 실행돼야 함 — 이 패스 이후에 실행되는
+ * 다른 패스가 top-level 스코프를 순회/변형한다면 이미 감싸인 함수 내부까지
+ * 안 보고 지나칠 수 있음.
+ */
+export function runWrapInFunction(program: Program, _options: WrapInFunctionOptions): void {
+    const innerBody = program.body
+    const wrapper = functionExpression(functionBody([], innerBody, true))
+    const iife = call(paren(wrapper), [vararg()])
+
+    program.body = block([returnStatement([iife])])
 }

package/src/passes/nodeFactory.ts

@@ -5,7 +5,7 @@ import type {
     LocalStatement, LocalFunctionStatement, FunctionBody, FunctionParameter,
     TypedIdentifier, Block, Statement, AssignmentStatement, NumericForStatement,
     ReturnStatement, DoStatement, IfStatement, IfClause, VarargExpression,
-    FunctionExpression,
+    FunctionExpression, BooleanLiteral, NilLiteral, WhileStatement,
 } from "luau-parser"
 
 /** 합성 노드는 실제 소스 위치가 없으므로 0으로 채움(printer는 구조만 봄). */
@@ -67,6 +67,18 @@ export function positionalField(value: Expression): TableField {
     return { type: "TableFieldPositional", value }
 }
 
+export function namedField(name: string, value: Expression): TableField {
+    return { type: "TableFieldNamed", name: identifier(name), value }
+}
+
+export function booleanLiteral(value: boolean): BooleanLiteral {
+    return { type: "BooleanLiteral", value, line: POS, column: POS }
+}
+
+export function nilLiteral(): NilLiteral {
+    return { type: "NilLiteral", line: POS, column: POS }
+}
+
 export function localStatement(name: string, init: Expression): LocalStatement {
     return {
         type: "LocalStatement",
@@ -138,6 +150,10 @@ export function ifStatement(clauses: IfClause[], alternate?: Block): IfStatement
     return { type: "IfStatement", clauses, alternate, line: POS, column: POS }
 }
 
+export function whileStatement(condition: Expression, body: Block): WhileStatement {
+    return { type: "WhileStatement", condition, body, line: POS, column: POS }
+}
+
 export function vararg(): VarargExpression {
     return { type: "VarargExpression", line: POS, column: POS }
 }

package/src/passes/vmify/chunk.ts

@@ -0,0 +1,50 @@
+import type { Instr } from "./opcodes"
+
+/** 상수 풀에 들어갈 수 있는 값. */
+export type ConstValue = string | number | boolean | null
+
+/**
+ * upvalue 하나가 부모 함수 기준으로 어디서 오는지.
+ * - "local": 부모 함수의 레지스터(박스)를 그대로 캡처
+ * - "upval": 부모 함수 자신의 upvalue를 한 단계 더 전달(중첩 클로저)
+ */
+export interface UpvalDesc {
+    kind: "local" | "upval"
+    index: number
+}
+
+/** 함수 하나(최상위 청크 포함)에 대응하는 컴파일 결과. Lua의 Proto와 동일한 역할. */
+export interface Proto {
+    id: number
+    numParams: number
+    hasVarargs: boolean
+    maxRegs: number
+    code: Instr[]
+    consts: ConstValue[]
+    /** 이 함수가 캡처하는 upvalue들이 "부모" 기준 어디서 오는지. */
+    upvalDescs: UpvalDesc[]
+    /** 중첩 함수 표현식들. CLOSURE 명령의 B는 이 배열의 인덱스. */
+    protos: Proto[]
+}
+
+export function createProto(id: number): Proto {
+    return {
+        id,
+        numParams: 0,
+        hasVarargs: false,
+        maxRegs: 0,
+        code: [],
+        consts: [],
+        upvalDescs: [],
+        protos: [],
+    }
+}
+
+/** 상수 풀에 값 추가(중복 제거) 후 인덱스 반환. */
+export function internConst(proto: Proto, value: ConstValue): number {
+    for (let i = 0; i < proto.consts.length; i++) {
+        if (proto.consts[i] === value) return i
+    }
+    proto.consts.push(value)
+    return proto.consts.length - 1
+}