diff options
| -rw-r--r-- | llvm/lib/Transforms/IPO/StripSymbols.cpp | 28 | ||||
| -rw-r--r-- | llvm/test/Transforms/StripSymbols/2007-01-15-llvm.used.ll | 15 | ||||
| -rw-r--r-- | llvm/test/Transforms/StripSymbols/dg.exp | 3 |
3 files changed, 43 insertions, 3 deletions
diff --git a/llvm/lib/Transforms/IPO/StripSymbols.cpp b/llvm/lib/Transforms/IPO/StripSymbols.cpp index 9427eb67824..297c2763dfe 100644 --- a/llvm/lib/Transforms/IPO/StripSymbols.cpp +++ b/llvm/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/llvm/test/Transforms/StripSymbols/2007-01-15-llvm.used.ll b/llvm/test/Transforms/StripSymbols/2007-01-15-llvm.used.ll new file mode 100644 index 00000000000..16cee15ee17 --- /dev/null +++ b/llvm/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/llvm/test/Transforms/StripSymbols/dg.exp b/llvm/test/Transforms/StripSymbols/dg.exp new file mode 100644 index 00000000000..879685ca879 --- /dev/null +++ b/llvm/test/Transforms/StripSymbols/dg.exp @@ -0,0 +1,3 @@ +load_lib llvm.exp + +RunLLVMTests [lsort [glob -nocomplain $srcdir/$subdir/*.{ll,llx,c,cpp,tr}]] |
