[Loader] Fix heap type syntax checking. (WasmEdge#3395)

apepkuss · May 9, 2024 · 8442569 · 8442569
1 parent abc43dc
commit 8442569
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/loader/ast/type.cpp b/lib/loader/ast/type.cpp
@@ -12,7 +12,11 @@ namespace Loader {
 Expect<ValType> Loader::loadHeapType(TypeCode TC, ASTNodeAttr From) {
   if (auto Res = FMgr.readS33()) {
     if (*Res < 0) {
-      // FuncRef or ExternRef case.
+      if (*Res < -64) {
+        // For checking the invalid s33 value which is larger than 1 byte.
+        return logLoadError(ErrCode::Value::MalformedRefType,
+                            FMgr.getLastOffset(), From);
+      }
       TypeCode HTCode =
           static_cast<TypeCode>(static_cast<uint8_t>((*Res) & INT64_C(0x7F)));
       switch (HTCode) {