diff options
| -rw-r--r-- | src/wasm-s-parser.h | 22 |
1 files changed, 21 insertions, 1 deletions
diff --git a/src/wasm-s-parser.h b/src/wasm-s-parser.h index 09c30ce92..9f9e98a1a 100644 --- a/src/wasm-s-parser.h +++ b/src/wasm-s-parser.h @@ -22,7 +22,8 @@ IString MODULE("module"), MEMORY("memory"), EXPORT("export"), TABLE("table"), - LOCAL("local"); + LOCAL("local"), + TYPE("type"); // // An element in an S-Expression: a list or a string @@ -191,6 +192,7 @@ private: if (id == MEMORY) return parseMemory(curr); if (id == EXPORT) return parseExport(curr); if (id == TABLE) return parseTable(curr); + if (id == TYPE) return parseType(curr); std::cerr << "bad module element " << id.str << '\n'; abort(); } @@ -553,6 +555,24 @@ private: wasm.table.names.push_back(s[i]->str()); } } + + void parseType(Element& s) { + auto type = allocator.alloc<FunctionType>(); + type->name = s[1]->str(); + Element& func = *s[2]; + assert(func.isList()); + for (size_t i = 1; i < func.size(); i++) { + Element& curr = *func[i]; + if (curr[0]->str() == PARAM) { + for (size_t j = 1; j < curr.size(); j++) { + type->params.push_back(stringToWasmType(curr[j]->str())); + } + } else if (curr[0]->str() == RESULT) { + type->result = stringToWasmType(curr[1]->str()); + } + } + wasm.functionTypes[type->name] = type; + } }; } // namespace wasm |