diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/passes/RemoveImports.cpp | 13 |
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); + } } } }; |