feat: add local variant name

Author: HerrCai0907

src/builtins.ts

@@ -10198,6 +10198,7 @@ export function compileVisitGlobals(compiler: Compiler): void {
     TypeRef.I32,  // cookie
     TypeRef.None, // => void
     [ sizeTypeRef ],
+    null,
     exprs.length
       ? module.block(null, exprs)
       : module.nop()
@@ -10304,6 +10305,7 @@ function ensureVisitMembersOf(compiler: Compiler, instance: Class): void {
     createType([sizeTypeRef, TypeRef.I32]),
     TypeRef.None,
     needsTempValue ? [ sizeTypeRef ] : null,
+    null,
     module.flatten(body, TypeRef.None)
   );
 
@@ -10386,6 +10388,7 @@ export function compileVisitMembers(compiler: Compiler): void {
     createType([ sizeTypeRef, TypeRef.I32 ]), // this, cookie
     TypeRef.None, // => void
     null,
+    null,
     module.flatten([
       current,
       module.unreachable()
@@ -10522,6 +10525,7 @@ export function compileClassInstanceOf(compiler: Compiler, prototype: ClassProto
     sizeTypeRef,
     TypeRef.I32,
     null,
+    null,
     module.flatten(stmts)
   );
 }

src/compiler.ts

@@ -749,7 +749,7 @@ export class Compiler extends DiagnosticEmitter {
 
     // expose the arguments length helper if there are varargs exports
     if (this.runtimeFeatures & RuntimeFeatures.setArgumentsLength) {
-      module.addFunction(BuiltinNames.setArgumentsLength, TypeRef.I32, TypeRef.None, null,
+      module.addFunction(BuiltinNames.setArgumentsLength, TypeRef.I32, TypeRef.None, null, null,
         module.global_set(this.ensureArgumentsLength(), module.local_get(0, TypeRef.I32))
       );
       module.addFunctionExport(BuiltinNames.setArgumentsLength, ExportNames.setArgumentsLength);
@@ -784,6 +784,7 @@ export class Compiler extends DiagnosticEmitter {
         signature.paramRefs,
         signature.resultRefs,
         typesToRefs(startFunctionInstance.additionalLocals),
+        startFunctionInstance.localsByIndex.map<string>(local => local.internalName),
         module.flatten(startFunctionBody)
       );
       startFunctionInstance.finalize(module, funcRef);
@@ -1024,6 +1025,7 @@ export class Compiler extends DiagnosticEmitter {
         startSignature.paramRefs,
         startSignature.resultRefs,
         varTypes,
+        startFunction.localsByIndex.map<string>(locals => locals.internalName),
         module.flatten(startFunctionBody)
       );
       previousBody.push(
@@ -1475,6 +1477,7 @@ export class Compiler extends DiagnosticEmitter {
         signature.paramRefs,
         signature.resultRefs,
         typesToRefs(instance.additionalLocals),
+        instance.localsByIndex.map<string>(local => local.internalName),
         module.flatten(stmts, instance.signature.returnType.toRef())
       );
 
@@ -1515,6 +1518,7 @@ export class Compiler extends DiagnosticEmitter {
         signature.paramRefs,
         signature.resultRefs,
         null,
+        null,
         module.unreachable()
       );
     } else {
@@ -1680,7 +1684,7 @@ export class Compiler extends DiagnosticEmitter {
     var valueTypeRef = valueType.toRef();
     var thisTypeRef = this.options.sizeTypeRef;
     // return this.field
-    instance.getterRef = module.addFunction(instance.internalGetterName, thisTypeRef, valueTypeRef, null,
+    instance.getterRef = module.addFunction(instance.internalGetterName, thisTypeRef, valueTypeRef, null, null,
       module.load(valueType.byteSize, valueType.isSignedIntegerValue,
         module.local_get(0, thisTypeRef),
         valueTypeRef, instance.memoryOffset
@@ -1724,7 +1728,10 @@ export class Compiler extends DiagnosticEmitter {
         ], TypeRef.None);
       }
     }
-    instance.setterRef = module.addFunction(instance.internalSetterName, createType([ thisTypeRef, valueTypeRef ]), TypeRef.None, null,
+    instance.setterRef = module.addFunction(
+      instance.internalSetterName,
+      createType([ thisTypeRef, valueTypeRef ]),
+      TypeRef.None, null, null,
       bodyExpr
     );
     if (instance.getterRef) {
@@ -6565,6 +6572,7 @@ export class Compiler extends DiagnosticEmitter {
       stub.signature.paramRefs,
       stub.signature.resultRefs,
       typesToRefs(stub.additionalLocals),
+      null,
       module.flatten(stmts, returnType.toRef())
     );
     stub.set(CommonFlags.COMPILED);
@@ -6588,7 +6596,7 @@ export class Compiler extends DiagnosticEmitter {
       stub.internalName,
       stub.signature.paramRefs,
       stub.signature.resultRefs,
-      null,
+      null, null,
       module.unreachable()
     );
     this.virtualStubs.add(original);
@@ -6713,6 +6721,7 @@ export class Compiler extends DiagnosticEmitter {
       stub.signature.paramRefs,
       stub.signature.resultRefs,
       [ TypeRef.I32 ],
+      null, 
       module.block(null, [
         builder.render(tempIndex),
         body
@@ -8610,6 +8619,7 @@ export class Compiler extends DiagnosticEmitter {
         signature.paramRefs,
         signature.resultRefs,
         varTypes,
+        instance.localsByIndex.map<string>(local => local.internalName),
         module.flatten(stmts, sizeTypeRef)
       );
       instance.finalize(module, funcRef);

src/module.ts

@@ -1896,6 +1896,7 @@ export class Module {
     params: TypeRef,
     results: TypeRef,
     varTypes: TypeRef[] | null,
+    localNames: string[] | null,
     body: ExpressionRef
   ): FunctionRef {
     var cStr = this.allocStringCached(name);
@@ -1910,6 +1911,22 @@ export class Module {
       body
     );
     binaryen._free(cArr);
+    if (localNames) {
+      let localNameMap = new Set<string>();
+      for (let i = 0, k = localNames.length; i < k; i++) {
+        let localNameParts = localNames[i].split("~");
+        let localName = localNameParts[localNameParts.length - 1];
+        if (localNameMap.has(localName)) {
+          let repeat = 0;
+          while (localNameMap.has(`${localName}_${repeat}`)) {
+            repeat++;
+          }
+          localName = `${localName}_${repeat}`;
+        }
+        localNameMap.add(localName);
+        binaryen._BinaryenFunctionSetLocalName(ret, i, this.allocStringCached(localName));
+      }
+    }
     return ret;
   }
 

src/passes/ministack.ts

@@ -144,7 +144,7 @@ export class MiniStack extends Pass {
           module.local_get(numParams, results)
         );
       }
-      module.addFunction(wrapperName, params, results, vars,
+      module.addFunction(wrapperName, params, results, vars, null,
         module.block(null, stmts, results)
       );
     }

src/passes/shadowstack.ts

@@ -317,7 +317,7 @@ export class ShadowStackPass extends Pass {
     var module = this.module;
     if (!this.hasStackCheckFunction) {
       this.hasStackCheckFunction = true;
-      module.addFunction("~stack_check", TypeRef.None, TypeRef.None, null,
+      module.addFunction("~stack_check", TypeRef.None, TypeRef.None, null, null,
         module.if(
           module.binary(BinaryOp.LtI32,
             module.global_get(BuiltinNames.stack_pointer, this.ptrType),