Skip imports in table during RemoveImports (#2181)

This prevents RemoveImports from producing an invalid module that
references functions that no longer exist.

1 files changed, 12 insertions, 1 deletions

diff --git a/src/passes/RemoveImports.cpp b/src/passes/RemoveImports.cpp
index a3ca45b16..50f7cfa3d 100644
--- a/src/passes/RemoveImports.cpp
+++ b/src/passes/RemoveImports.cpp
@@ -48,8 +48,19 @@ struct RemoveImports : public WalkerPass<PostWalker<RemoveImports>> {
     std::vector<Name> names;
     ModuleUtils::iterImportedFunctions(
       *curr, [&](Function* func) { names.push_back(func->name); });
+    // Do not remove names referenced in a table
+    std::set<Name> indirectNames;
+    if (curr->table.exists) {
+      for (auto& segment : curr->table.segments) {
+        for (auto& name : segment.data) {
+          indirectNames.insert(name);
+        }
+      }
+    }
     for (auto& name : names) {
-      curr->removeFunction(name);
+      if (indirectNames.find(name) == indirectNames.end()) {
+        curr->removeFunction(name);
+      }
     }
   }
 };