Index: include/llvm/Transforms/IPO/PassManagerBuilder.h =================================================================== --- include/llvm/Transforms/IPO/PassManagerBuilder.h +++ include/llvm/Transforms/IPO/PassManagerBuilder.h @@ -60,9 +60,10 @@ public: /// Extensions are passed the builder itself (so they can see how it is /// configured) as well as the pass manager to add stuff to. - typedef std::function - ExtensionFn; + typedef void ExtensionProc (const PassManagerBuilder &Builder, + legacy::PassManagerBase &PM); + typedef std::function ExtensionFn; + enum ExtensionPointTy { /// EP_EarlyAsPossible - This extension point allows adding passes before /// any other transformations, allowing them to see the code as it is coming @@ -179,7 +180,7 @@ /// Adds an extension that will be used by all PassManagerBuilder instances. /// This is intended to be used by plugins, to register a set of /// optimisations to run automatically. - static void addGlobalExtension(ExtensionPointTy Ty, ExtensionFn Fn); + static void addGlobalExtension(ExtensionPointTy Ty, ExtensionProc Fn); void addExtension(ExtensionPointTy Ty, ExtensionFn Fn); private: @@ -210,8 +211,8 @@ /// private function that the PassManagerBuilder can use to add your passes. struct RegisterStandardPasses { RegisterStandardPasses(PassManagerBuilder::ExtensionPointTy Ty, - PassManagerBuilder::ExtensionFn Fn) { - PassManagerBuilder::addGlobalExtension(Ty, std::move(Fn)); + PassManagerBuilder::ExtensionProc Fn) { + PassManagerBuilder::addGlobalExtension(Ty, Fn); } }; Index: lib/Transforms/IPO/PassManagerBuilder.cpp =================================================================== --- lib/Transforms/IPO/PassManagerBuilder.cpp +++ lib/Transforms/IPO/PassManagerBuilder.cpp @@ -196,8 +196,8 @@ void PassManagerBuilder::addGlobalExtension( PassManagerBuilder::ExtensionPointTy Ty, - PassManagerBuilder::ExtensionFn Fn) { - GlobalExtensions->push_back(std::make_pair(Ty, std::move(Fn))); + PassManagerBuilder::ExtensionProc Fn) { + GlobalExtensions->push_back(std::make_pair(Ty, Fn)); } void PassManagerBuilder::addExtension(ExtensionPointTy Ty, ExtensionFn Fn) { Index: unittests/Support/DynamicLibrary/CMakeLists.txt =================================================================== --- unittests/Support/DynamicLibrary/CMakeLists.txt +++ unittests/Support/DynamicLibrary/CMakeLists.txt @@ -1,4 +1,4 @@ -set(LLVM_LINK_COMPONENTS Support) +set(LLVM_LINK_COMPONENTS Support IPO) add_llvm_unittest(DynamicLibraryTests DynamicLibraryTest.cpp) Index: unittests/Support/DynamicLibrary/DynamicLibraryTest.cpp =================================================================== --- unittests/Support/DynamicLibrary/DynamicLibraryTest.cpp +++ unittests/Support/DynamicLibrary/DynamicLibraryTest.cpp @@ -12,6 +12,7 @@ #include "llvm/Support/FileSystem.h" #include "llvm/Support/ManagedStatic.h" #include "llvm/Support/Path.h" +#include "llvm/Transforms/IPO/PassManagerBuilder.h" #include "gtest/gtest.h" #include "PipSqueak.h" @@ -54,6 +55,28 @@ return Tmp.P; } +static void +NoPassRegistration(const PassManagerBuilder &, legacy::PassManagerBase &) { +} + +static RegisterStandardPasses TestOrder(PassManagerBuilder::EP_LoopOptimizerEnd, + NoPassRegistration); + +typedef void (*TestPassReg)(void (*)(PassManagerBuilder::ExtensionPointTy, + PassManagerBuilder::ExtensionProc)); + +TEST(DynamicLibrary, PassRegistration) { + std::string Err; + llvm_shutdown_obj Shutdown; + DynamicLibrary DL = + DynamicLibrary::getPermanentLibrary(LibPath().c_str(), &Err); + EXPECT_TRUE(DL.isValid()); + EXPECT_TRUE(Err.empty()); + + TestPassReg RP = FuncPtr(DL.getAddressOfSymbol("TestPassReg")); + RP(&PassManagerBuilder::addGlobalExtension); +} + static const char *OverloadTestA() { return "OverloadCall"; } std::string StdString(const char *Ptr) { return Ptr ? Ptr : ""; } Index: unittests/Support/DynamicLibrary/PipSqueak.cxx =================================================================== --- unittests/Support/DynamicLibrary/PipSqueak.cxx +++ unittests/Support/DynamicLibrary/PipSqueak.cxx @@ -8,6 +8,7 @@ //===----------------------------------------------------------------------===// #include "PipSqueak.h" +#include "llvm/Transforms/IPO/PassManagerBuilder.h" #if defined(_WIN32) && !defined(__GNUC__) // Disable warnings from inclusion of xlocale & exception @@ -44,3 +45,13 @@ } extern "C" PIPSQUEAK_EXPORT const char *TestA() { return "LibCall"; } + +static void LibPassRegistration(const llvm::PassManagerBuilder &, + llvm::legacy::PassManagerBase &) {} + +extern "C" PIPSQUEAK_EXPORT void TestPassReg( + void (*addGlobalExtension)(llvm::PassManagerBuilder::ExtensionPointTy, + llvm::PassManagerBuilder::ExtensionProc)) { + addGlobalExtension(llvm::PassManagerBuilder::EP_EarlyAsPossible, + LibPassRegistration); +}