Merge pull request #435 from WebAssembly/more-relooper

Add optimization to C API, and with that, more relooper tests

9 files changed, 30 insertions, 14 deletions

diff --git a/src/asm2wasm.h b/src/asm2wasm.h
index fa35211a4..0f4852250 100644
--- a/src/asm2wasm.h
+++ b/src/asm2wasm.h
@@ -1725,12 +1725,12 @@ Function* Asm2WasmBuilder::processFunction(Ref ast) {
 }
 
 void Asm2WasmBuilder::optimize() {
-  PassRunner passRunner(&allocator);
+  PassRunner passRunner(&wasm);
   passRunner.addDefaultOptimizationPasses();
   if (maxGlobal < 1024) {
     passRunner.add("post-emscripten");
   }
-  passRunner.run(&wasm);
+  passRunner.run();
 }
 
 } // namespace wasm
diff --git a/src/binaryen-c.cpp b/src/binaryen-c.cpp
index b2c695ec0..bbafdf49d 100644
--- a/src/binaryen-c.cpp
+++ b/src/binaryen-c.cpp
@@ -19,6 +19,7 @@
 //===============================
 
 #include "binaryen-c.h"
+#include "pass.h"
 #include "wasm.h"
 #include "wasm-builder.h"
 #include "wasm-printing.h"
@@ -384,6 +385,13 @@ void BinaryenModulePrint(BinaryenModuleRef module) {
   WasmPrinter::printModule((Module*)module);
 }
 
+void BinaryenModuleOptimize(BinaryenModuleRef module) {
+  Module* wasm = (Module*)module;
+  PassRunner passRunner(wasm);
+  passRunner.addDefaultOptimizationPasses();
+  passRunner.run();
+}
+
 //
 // ========== CFG / Relooper ==========
 //
diff --git a/src/binaryen-c.h b/src/binaryen-c.h
index 2dbf27076..2e0a7b996 100644
--- a/src/binaryen-c.h
+++ b/src/binaryen-c.h
@@ -266,6 +266,9 @@ void BinaryenSetStart(BinaryenModuleRef module, const char* name);
 // Print a module to stdout.
 void BinaryenModulePrint(BinaryenModuleRef module);
 
+// Run the standard optimization passes on the module.
+void BinaryenModuleOptimize(BinaryenModuleRef module);
+
 //
 // ========== CFG / Relooper ==========
 //
diff --git a/src/binaryen-shell.cpp b/src/binaryen-shell.cpp
index f988379c4..8c0390d15 100644
--- a/src/binaryen-shell.cpp
+++ b/src/binaryen-shell.cpp
@@ -223,7 +223,7 @@ int main(int argc, const char* argv[]) {
 
       if (passes.size() > 0) {
         if (options.debug) std::cerr << "running passes...\n";
-        PassRunner passRunner(&moreModuleAllocations);
+        PassRunner passRunner(&wasm);
         if (options.debug) passRunner.setDebug(true);
         for (auto& passName : passes) {
           if (passName == "O") {
@@ -232,7 +232,7 @@ int main(int argc, const char* argv[]) {
             passRunner.add(passName);
           }
         }
-        passRunner.run(&wasm);
+        passRunner.run();
       }
 
       run_asserts(&i, &checked, &wasm, &root, &builder, entry);
diff --git a/src/pass.cpp b/src/pass.cpp
index 5c4f510f8..e4cc35554 100644
--- a/src/pass.cpp
+++ b/src/pass.cpp
@@ -68,7 +68,7 @@ void PassRunner::addDefaultOptimizationPasses() {
   add("optimize-instructions");
 }
 
-void PassRunner::run(Module* module) {
+void PassRunner::run() {
   std::chrono::high_resolution_clock::time_point beforeEverything;
   size_t padding = 0;
   if (debug) {
@@ -88,7 +88,7 @@ void PassRunner::run(Module* module) {
       }
       before = std::chrono::high_resolution_clock::now();
     }
-    pass->run(this, module);
+    pass->run(this, wasm);
     if (debug) {
       auto after = std::chrono::high_resolution_clock::now();
       std::chrono::duration<double> diff = after - before;
diff --git a/src/pass.h b/src/pass.h
index 85aec624f..bfbcb26d8 100644
--- a/src/pass.h
+++ b/src/pass.h
@@ -22,6 +22,7 @@
 #include "wasm.h"
 #include "wasm-traversal.h"
 #include "mixed_arena.h"
+#include "support/utilities.h"
 
 namespace wasm {
 
@@ -66,18 +67,19 @@ struct RegisterPass {
 // Runs a set of passes, in order
 //
 struct PassRunner {
+  Module* wasm;
   MixedArena* allocator;
   std::vector<Pass*> passes;
   Pass* currPass;
   bool debug = false;
 
-  PassRunner(MixedArena* allocator) : allocator(allocator) {}
+  PassRunner(Module* wasm) : wasm(wasm), allocator(&wasm->allocator) {}
 
   void setDebug(bool debug_) { debug = debug_; }
 
   void add(std::string passName) {
     auto pass = PassRegistry::get()->createPass(passName);
-    assert(pass);
+    if (!pass) Fatal() << "Could not find pass: " << passName << "\n";
     passes.push_back(pass);
   }
 
@@ -95,7 +97,7 @@ struct PassRunner {
   // what -O does.
   void addDefaultOptimizationPasses();
 
-  void run(Module* module);
+  void run();
 
   // Get the last pass that was already executed of a certain type.
   template<class P>
diff --git a/src/passes/ReorderLocals.cpp b/src/passes/ReorderLocals.cpp
index 1a3fbf09a..0e626e0a2 100644
--- a/src/passes/ReorderLocals.cpp
+++ b/src/passes/ReorderLocals.cpp
@@ -106,8 +106,11 @@ struct ReorderLocals : public WalkerPass<PostWalker<ReorderLocals, Visitor<Reord
     curr->localNames.resize(newToOld.size());
     curr->localIndices.clear();
     for (size_t i = 0; i < newToOld.size(); i++) {
-      curr->localNames[i] = oldLocalNames[newToOld[i]];
-      curr->localIndices[oldLocalNames[newToOld[i]]] = i;
+      if (newToOld[i] < oldLocalNames.size()) {
+        auto old = oldLocalNames[newToOld[i]];
+        curr->localNames[i] = old;
+        curr->localIndices[old] = i;
+      }
     }
   }
 
diff --git a/src/support/archive.cpp b/src/support/archive.cpp
index 3879f72a7..c0351bc05 100644
--- a/src/support/archive.cpp
+++ b/src/support/archive.cpp
@@ -211,7 +211,7 @@ static uint32_t read32be(const uint8_t* buf) {
 
 void Archive::dump() const {
   printf("Archive data %p len %lu, firstRegularData %p\n", data.data(),
-         data.size(), firstRegularData);
+         (long unsigned)data.size(), firstRegularData);
   printf("Symbol table %p, len %u\n", symbolTable.data, symbolTable.len);
   printf("string table %p, len %u\n", stringTable.data, stringTable.len);
   const uint8_t* buf = symbolTable.data;
diff --git a/src/wasm-printing.h b/src/wasm-printing.h
index 2fab99938..d1f1c42a5 100644
--- a/src/wasm-printing.h
+++ b/src/wasm-printing.h
@@ -26,9 +26,9 @@ namespace wasm {
 
 struct WasmPrinter {
   static std::ostream& printModule(Module* module, std::ostream& o) {
-    PassRunner passRunner(nullptr);
+    PassRunner passRunner(module);
     passRunner.add<Printer>(o);
-    passRunner.run(module);
+    passRunner.run();
     return o;
   }