feat(arc): waittill undefined param (#149)

xensik · Sep 24, 2023 · 4721aca · 4721aca
1 parent 008d545
commit 4721aca
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/src/arc/compiler.cpp b/src/arc/compiler.cpp
@@ -302,7 +302,10 @@ auto compiler::emit_stmt_waittill(stmt_waittill const& stm) -> void
 
     for (auto const& entry : stm.args->list)
     {
-        emit_opcode(opcode::OP_SafeSetWaittillVariableFieldCached, fmt::format("{}", variable_access(entry->as<expr_identifier>())));
+        if (entry->is<expr_undefined>())
+            emit_opcode(opcode::OP_SafeDecTop);
+        else
+            emit_opcode(opcode::OP_SafeSetWaittillVariableFieldCached, fmt::format("{}", variable_access(entry->as<expr_identifier>())));
     }
 
     emit_opcode(opcode::OP_ClearParams);
@@ -2003,12 +2006,15 @@ auto compiler::process_stmt_waittill(stmt_waittill const& stm) -> void
 {
     for (auto const& entry : stm.args->list)
     {
-        if (!entry->is<expr_identifier>())
+        if (!entry->is<expr_identifier>() && !entry->is<expr_undefined>())
         {
-            throw comp_error(entry->loc(), "illegal waittill param, must be a local variable");
+            throw comp_error(entry->loc(), "illegal waittill param, must be a local variable or undefined");
         }
 
-        variable_register(entry->as<expr_identifier>());
+        if (entry->is<expr_identifier>())
+        {
+            variable_register(entry->as<expr_identifier>());
+        }
     }
 }
 

diff --git a/src/arc/decompiler.cpp b/src/arc/decompiler.cpp
@@ -192,6 +192,7 @@ auto decompiler::decompile_instruction(instruction const& inst, bool last) -> vo
             break;
         }
         case opcode::OP_GetUndefined:
+        case opcode::OP_SafeDecTop:
         {
             stack_.push(expr_undefined::make(loc));
             break;
@@ -1118,7 +1119,6 @@ auto decompiler::decompile_instruction(instruction const& inst, bool last) -> vo
         case opcode::OP_CheckClearParams:
         case opcode::OP_CastFieldObject:
         case opcode::OP_CastBool:
-        case opcode::OP_SafeDecTop:
             break;
         case opcode::OP_CreateLocalVariable:
         case opcode::OP_RemoveLocalVariables: