commit 5f792c2c536721e286e983cdcde0369bdbb4a36d
author Chris Lattner <sabre@nondot.org> Thu Mar 03 01:03:43 2005 +0000
committer Chris Lattner <sabre@nondot.org> Thu Mar 03 01:03:43 2005 +0000
tree f91a8ab86b14d25c9f9e3e703971479cd939faed
parent 7dd732bf805ff0ea792054e507ac55374ba8407b

Add an optional argument to lower to a specific constant value instead of
to a "sizeof" expression.


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

lib/Transforms/Utils/LowerAllocations.cpp

diff --git a/lib/Transforms/Utils/LowerAllocations.cpp b/lib/Transforms/Utils/LowerAllocations.cpp
index d826a91..abed41ed 100644
--- a/lib/Transforms/Utils/LowerAllocations.cpp
+++ b/lib/Transforms/Utils/LowerAllocations.cpp
@@ -31,8 +31,10 @@
   class LowerAllocations : public BasicBlockPass {
     Function *MallocFunc;   // Functions in the module we are processing
     Function *FreeFunc;     // Initialized by doInitialization
+    bool LowerMallocArgToInteger;
   public:
-    LowerAllocations() : MallocFunc(0), FreeFunc(0) {}
+    LowerAllocations(bool LowerToInt = false)
+      : MallocFunc(0), FreeFunc(0), LowerMallocArgToInteger(LowerToInt) {}
 
     virtual void getAnalysisUsage(AnalysisUsage &AU) const {
       AU.addRequired<TargetData>();
@@ -59,8 +61,8 @@
 }
 
 // createLowerAllocationsPass - Interface to this file...
-FunctionPass *llvm::createLowerAllocationsPass() {
-  return new LowerAllocations();
+FunctionPass *llvm::createLowerAllocationsPass(bool LowerMallocArgToInteger) {
+  return new LowerAllocations(LowerMallocArgToInteger);
 }
 
 
@@ -95,7 +97,8 @@
 
   BasicBlock::InstListType &BBIL = BB.getInstList();
 
-  const Type *IntPtrTy = getAnalysis<TargetData>().getIntPtrType();
+  const TargetData &TD = getAnalysis<TargetData>();
+  const Type *IntPtrTy = TD.getIntPtrType();
 
   // Loop over all of the instructions, looking for malloc or free instructions
   for (BasicBlock::iterator I = BB.begin(), E = BB.end(); I != E; ++I) {
@@ -103,8 +106,13 @@
       const Type *AllocTy = MI->getType()->getElementType();
       
       // malloc(type) becomes sbyte *malloc(size)
-      Value *MallocArg = ConstantExpr::getCast(ConstantExpr::getSizeOf(AllocTy),
-                                               IntPtrTy);
+      Value *MallocArg;
+      if (LowerMallocArgToInteger)
+        MallocArg = ConstantUInt::get(Type::ULongTy, TD.getTypeSize(AllocTy));
+      else
+        MallocArg = ConstantExpr::getSizeOf(AllocTy);
+      MallocArg = ConstantExpr::getCast(cast<Constant>(MallocArg), IntPtrTy);
+
       if (MI->isArrayAllocation()) {
         if (isa<ConstantInt>(MallocArg) &&
             cast<ConstantInt>(MallocArg)->getRawValue() == 1) {