Be careful not to make "external" function internal

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@2363 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/lib/Transforms/IPO/Internalize.cpp b/lib/Transforms/IPO/Internalize.cpp
index aadde7c92cf53c5264b2c2715078871d033606a2..8bb1a9c111c59186269fe7de7a54d8f3cde913ad 100644 (file)
--- a/lib/Transforms/IPO/Internalize.cpp
+++ b/lib/Transforms/IPO/Internalize.cpp
@@ -15,7 +15,7 @@ class InternalizePass : public Pass {
   virtual bool run(Module *M) {
     bool FoundMain = false;   // Look for a function named main...
     for (Module::iterator I = M->begin(), E = M->end(); I != E; ++I)
-      if ((*I)->getName() == "main") {
+      if ((*I)->getName() == "main" && !(*I)->isExternal()) {
         FoundMain = true;
         break;
       }
@@ -26,7 +26,8 @@ class InternalizePass : public Pass {
 
     // Found a main function, mark all functions not named main as internal.
     for (Module::iterator I = M->begin(), E = M->end(); I != E; ++I)
-      if ((*I)->getName() != "main")  // Leave the main function external
+      if ((*I)->getName() != "main" &&  // Leave the main function external
+          !(*I)->isExternal())          // Function must be defined here
         (*I)->setInternalLinkage(Changed = true);
 
     return Changed;