fix asm const arg

2 files changed, 6 insertions, 2 deletions

diff --git a/src/s2wasm.h b/src/s2wasm.h
index f7597b460..20cdce0c4 100644
--- a/src/s2wasm.h
+++ b/src/s2wasm.h
@@ -882,15 +882,19 @@ public:
           auto arg = curr->operands[0]->cast<Const>();
           size_t segmentIndex = parent->addressSegments[arg->value.geti32()];
           std::string code = escape(parent->wasm.memory.segments[segmentIndex].data);
+          int32_t id;
           if (ids.count(code) == 0) {
-            size_t id = ids.size();
+            id = ids.size();
             ids[code] = id;
+          } else {
+            id = ids[code];
           }
           std::string sig = getSig(curr);
           sigsForCode[code].insert(sig);
           std::string fixedTarget = EMSCRIPTEN_ASM_CONST.str;
           fixedTarget += '_' + sig;
           curr->target = cashew::IString(fixedTarget.c_str(), false);
+          arg->value = Literal(id);
         }
       }
 
diff --git a/test/dot_s/asm_const.wast b/test/dot_s/asm_const.wast
index 55665533c..cdfd6f1c2 100644
--- a/test/dot_s/asm_const.wast
+++ b/test/dot_s/asm_const.wast
@@ -6,7 +6,7 @@
     (block $fake_return_waka123
       (block
         (call_import $emscripten_asm_const_vi
-          (i32.const 16)
+          (i32.const 0)
         )
         (br $fake_return_waka123
           (i32.const 0)