diff options
author | Devang Patel <dpatel@apple.com> | 2008-01-16 03:33:05 +0000 |
---|---|---|
committer | Devang Patel <dpatel@apple.com> | 2008-01-16 03:33:05 +0000 |
commit | 8c231e5dda26b790ff388fe2f70eeeda621c9261 (patch) | |
tree | 9922a3ec23e77d151ff2f28ff41a957f8cdf6ac6 | |
parent | 316e98447122fc3f1e99930583a305ba481e848c (diff) | |
download | llvm-8c231e5dda26b790ff388fe2f70eeeda621c9261.tar.gz llvm-8c231e5dda26b790ff388fe2f70eeeda621c9261.tar.bz2 llvm-8c231e5dda26b790ff388fe2f70eeeda621c9261.tar.xz |
Do not strip llvm.used values.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@46045 91177308-0d34-0410-b5e6-96231b3b80d8
-rw-r--r-- | lib/Transforms/IPO/StripSymbols.cpp | 28 | ||||
-rw-r--r-- | test/Transforms/StripSymbols/2007-01-15-llvm.used.ll | 15 | ||||
-rw-r--r-- | test/Transforms/StripSymbols/dg.exp | 3 |
3 files changed, 43 insertions, 3 deletions
diff --git a/lib/Transforms/IPO/StripSymbols.cpp b/lib/Transforms/IPO/StripSymbols.cpp index 9427eb6782..297c2763df 100644 --- a/lib/Transforms/IPO/StripSymbols.cpp +++ b/lib/Transforms/IPO/StripSymbols.cpp @@ -29,6 +29,7 @@ #include "llvm/ValueSymbolTable.h" #include "llvm/TypeSymbolTable.h" #include "llvm/Support/Compiler.h" +#include "llvm/ADT/SmallPtrSet.h" using namespace llvm; namespace { @@ -100,13 +101,34 @@ bool StripSymbols::runOnModule(Module &M) { // If we're not just stripping debug info, strip all symbols from the // functions and the names from any internal globals. if (!OnlyDebugInfo) { + SmallPtrSet<const Constant *, 8> llvmUsedValues; + Value *LLVMUsed = M.getValueSymbolTable().lookup("llvm.used"); + if (LLVMUsed) { + // Collect values that are preserved as per explicit request. + // llvm.used is used to list these values. + if (GlobalVariable *GV = dyn_cast<GlobalVariable>(LLVMUsed)) { + if (ConstantArray *InitList = + dyn_cast<ConstantArray>(GV->getInitializer())) { + for (unsigned i = 0, e = InitList->getNumOperands(); i != e; ++i) { + if (ConstantExpr *CE = + dyn_cast<ConstantExpr>(InitList->getOperand(i))) + if (CE->isCast()) + llvmUsedValues.insert(CE->getOperand(0)); + } + } + } + } + for (Module::global_iterator I = M.global_begin(), E = M.global_end(); - I != E; ++I) - if (I->hasInternalLinkage()) + I != E; ++I) { + if (I->hasInternalLinkage() && llvmUsedValues.count(I) == 0) I->setName(""); // Internal symbols can't participate in linkage + else if (I->getName() == "llvm.used") { + } + } for (Module::iterator I = M.begin(), E = M.end(); I != E; ++I) { - if (I->hasInternalLinkage()) + if (I->hasInternalLinkage() && llvmUsedValues.count(I) == 0) I->setName(""); // Internal symbols can't participate in linkage StripSymtab(I->getValueSymbolTable()); } diff --git a/test/Transforms/StripSymbols/2007-01-15-llvm.used.ll b/test/Transforms/StripSymbols/2007-01-15-llvm.used.ll new file mode 100644 index 0000000000..16cee15ee1 --- /dev/null +++ b/test/Transforms/StripSymbols/2007-01-15-llvm.used.ll @@ -0,0 +1,15 @@ +; RUN: llvm-as < %s | opt -strip | llvm-dis | grep foo | count 2 +; RUN: llvm-as < %s | opt -strip | llvm-dis | grep bar | count 2 +@llvm.used = appending global [2 x i8*] [ i8* bitcast (i32* @foo to i8*), i8* bitcast (i32 ()* @bar to i8*) ], section "llvm.metadata" ; <[2 x i8*]*> [#uses=0] +@foo = internal constant i32 41 ; <i32*> [#uses=1] + +define internal i32 @bar() nounwind { +entry: + ret i32 42 +} + +define i32 @main() nounwind { +entry: + ret i32 0 +} + diff --git a/test/Transforms/StripSymbols/dg.exp b/test/Transforms/StripSymbols/dg.exp new file mode 100644 index 0000000000..879685ca87 --- /dev/null +++ b/test/Transforms/StripSymbols/dg.exp @@ -0,0 +1,3 @@ +load_lib llvm.exp + +RunLLVMTests [lsort [glob -nocomplain $srcdir/$subdir/*.{ll,llx,c,cpp,tr}]] |