add --mem-max option to set the maximum size of memory, overriding the default (which is the normal size of memory if no growth, or infinity if growth) (#837)

1 files changed, 10 insertions, 0 deletions

diff --git a/src/tools/asm2wasm.cpp b/src/tools/asm2wasm.cpp
index c5590c5eb..db6d723db 100644
--- a/src/tools/asm2wasm.cpp
+++ b/src/tools/asm2wasm.cpp
@@ -55,6 +55,10 @@ int main(int argc, const char *argv[]) {
            [](Options *o, const std::string &argument) {
              o->extra["mem base"] = argument;
            })
+      .add("--mem-max", "-mm", "Set the maximum size of memory in the wasm module (in bytes). Without this, TOTAL_MEMORY is used (as it is used for the initial value), or if memory growth is enabled, no limit is set. This overrides both of those.", Options::Arguments::One,
+           [](Options *o, const std::string &argument) {
+             o->extra["mem max"] = argument;
+           })
       .add("--total-memory", "-m", "Total memory size", Options::Arguments::One,
            [](Options *o, const std::string &argument) {
              o->extra["total memory"] = argument;
@@ -123,6 +127,12 @@ int main(int argc, const char *argv[]) {
     }
   }
 
+  // Set the max memory size, if requested
+  const auto &memMax = options.extra.find("mem max");
+  if (memMax != options.extra.end()) {
+    wasm.memory.max = atoi(memMax->second.c_str()) / Memory::kPageSize;
+  }
+
   if (options.debug) std::cerr << "printing..." << std::endl;
   Output output(options.extra["output"], Flags::Text, options.debug ? Flags::Debug : Flags::Release);
   WasmPrinter::printModule(&wasm, output.getStream());