Teach the internalize pass to also internalize

global aliases.

llvm-svn: 61754

Author: CunningBaldrick

llvm/lib/Transforms/IPO/Internalize.cpp

@@ -26,6 +26,7 @@
 #include <set>
 using namespace llvm;
 
+STATISTIC(NumAliases  , "Number of aliases internalized");
 STATISTIC(NumFunctions, "Number of functions internalized");
 STATISTIC(NumGlobals  , "Number of global vars internalized");
 
@@ -158,6 +159,17 @@ bool InternalizePass::runOnModule(Module &M) {
       DOUT << "Internalized gvar " << I->getName() << "\n";
     }
 
+  // Mark all aliases that are not in the api as internal as well.
+  for (Module::alias_iterator I = M.alias_begin(), E = M.alias_end();
+       I != E; ++I)
+    if (!I->isDeclaration() && !I->hasInternalLinkage() &&
+        !ExternalNames.count(I->getName())) {
+      I->setLinkage(GlobalValue::InternalLinkage);
+      Changed = true;
+      ++NumAliases;
+      DOUT << "Internalized alias " << I->getName() << "\n";
+    }
+
   return Changed;
 }
 

llvm/test/Transforms/Internalize/2009-01-05-InternalizeAliases.ll

@@ -0,0 +1,10 @@
+; RUN: llvm-as < %s | opt -internalize | llvm-dis | grep internal | count 3
+
+@A = global i32 0
+@B = alias i32* @A
+@C = alias i32* @B
+
+define i32 @main() {
+	%tmp = load i32* @C
+	ret i32 %tmp
+}