Index: include/llvm/IR/Module.h =================================================================== --- include/llvm/IR/Module.h +++ include/llvm/IR/Module.h @@ -732,6 +732,17 @@ void setPICLevel(PICLevel::Level PL); /// @} +/// @} +/// @name Utility functions for querying and setting PIE level +/// @{ + + /// \brief Returns the PIE level (small or large model) + PIELevel::Level getPIELevel() const; + + /// \brief Set the PIE level (small or large model) + void setPIELevel(PIELevel::Level PL); +/// @} + /// @name Utility functions for querying and setting PGO summary /// @{ Index: include/llvm/Support/CodeGen.h =================================================================== --- include/llvm/Support/CodeGen.h +++ include/llvm/Support/CodeGen.h @@ -32,6 +32,10 @@ enum Level { Default=0, Small=1, Large=2 }; } + namespace PIELevel { + enum Level { Default=0, Small=1, Large=2 }; + } + // TLS models. namespace TLSModel { enum Model { Index: lib/IR/Module.cpp =================================================================== --- lib/IR/Module.cpp +++ lib/IR/Module.cpp @@ -497,6 +497,20 @@ addModuleFlag(ModFlagBehavior::Error, "PIC Level", PL); } +PIELevel::Level Module::getPIELevel() const { + auto *Val = cast_or_null(getModuleFlag("PIE Level")); + + if (!Val) + return PIELevel::Default; + + return static_cast( + cast(Val->getValue())->getZExtValue()); +} + +void Module::setPIELevel(PIELevel::Level PL) { + addModuleFlag(ModFlagBehavior::Error, "PIE Level", PL); +} + void Module::setMaximumFunctionCount(uint64_t Count) { addModuleFlag(ModFlagBehavior::Error, "MaxFunctionCount", Count); }