From: Chris Lattner <sabre@nondot.org>
Date: Thu, 18 Jul 2002 00:18:01 +0000 (+0000)
Subject: * Correctly get prototype for void*malloc(size_t)
X-Git-Url: http://demsky.eecs.uci.edu/git/?a=commitdiff_plain;h=0b5909e6ae3d31c3e6c1379f61d7b038f225a620;p=oota-llvm.git

* Correctly get prototype for void*malloc(size_t)


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

diff --git a/lib/Transforms/IPO/RaiseAllocations.cpp b/lib/Transforms/IPO/RaiseAllocations.cpp
index 42e1a28af7a..9ab3d5faf94 100644
--- a/lib/Transforms/IPO/RaiseAllocations.cpp
+++ b/lib/Transforms/IPO/RaiseAllocations.cpp
@@ -61,18 +61,27 @@ bool RaiseAllocations::doInitialization(Module &M) {
   //
   const FunctionType *MallocType =   // Get the type for malloc
     FunctionType::get(PointerType::get(Type::SByteTy),
-                    std::vector<const Type*>(1, Type::UIntTy), false);
+                    std::vector<const Type*>(1, Type::ULongTy), false);
 
   const FunctionType *FreeType =     // Get the type for free
     FunctionType::get(Type::VoidTy,
                    std::vector<const Type*>(1, PointerType::get(Type::SByteTy)),
                       false);
 
+  // Get Malloc and free prototypes if they exist!
   MallocFunc = M.getFunction("malloc", MallocType);
   FreeFunc   = M.getFunction("free"  , FreeType);
 
+  // Check to see if the prototype is wrong, giving us sbyte*(uint) * malloc
+  // This handles the common declaration of: 'void *malloc(unsigned);'
+  if (MallocFunc == 0) {
+    MallocType = FunctionType::get(PointerType::get(Type::SByteTy),
+                            std::vector<const Type*>(1, Type::UIntTy), false);
+    MallocFunc = M.getFunction("malloc", MallocType);
+  }
+
   // Check to see if the prototype is missing, giving us sbyte*(...) * malloc
-  // This handles the common declaration of: 'char *malloc();'
+  // This handles the common declaration of: 'void *malloc();'
   if (MallocFunc == 0) {
     MallocType = FunctionType::get(PointerType::get(Type::SByteTy),
                                    std::vector<const Type*>(), true);