blob: 06ee5a8257e9ca51e617d33d17809efee7220f66 [file] [log] [blame]
Chris Lattner25db5802004-10-07 04:16:33 +00001//===- GlobalOpt.cpp - Optimize Global Variables --------------------------===//
Misha Brukmanb1c93172005-04-21 23:48:37 +00002//
Chris Lattner25db5802004-10-07 04:16:33 +00003// The LLVM Compiler Infrastructure
4//
Chris Lattnerf3ebc3f2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Misha Brukmanb1c93172005-04-21 23:48:37 +00007//
Chris Lattner25db5802004-10-07 04:16:33 +00008//===----------------------------------------------------------------------===//
9//
10// This pass transforms simple global variables that never have their address
11// taken. If obviously true, it marks read/write globals as constant, deletes
12// variables only stored to, etc.
13//
14//===----------------------------------------------------------------------===//
15
Chris Lattner25db5802004-10-07 04:16:33 +000016#include "llvm/Transforms/IPO.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000017#include "llvm/ADT/DenseMap.h"
18#include "llvm/ADT/STLExtras.h"
19#include "llvm/ADT/SmallPtrSet.h"
David Majnemerdad0a642014-06-27 18:19:56 +000020#include "llvm/ADT/SmallSet.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000021#include "llvm/ADT/SmallVector.h"
22#include "llvm/ADT/Statistic.h"
23#include "llvm/Analysis/ConstantFolding.h"
24#include "llvm/Analysis/MemoryBuiltins.h"
Chandler Carruth219b89b2014-03-04 11:01:28 +000025#include "llvm/IR/CallSite.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000026#include "llvm/IR/CallingConv.h"
27#include "llvm/IR/Constants.h"
28#include "llvm/IR/DataLayout.h"
29#include "llvm/IR/DerivedTypes.h"
Chandler Carruth03eb0de2014-03-04 10:40:04 +000030#include "llvm/IR/GetElementPtrTypeIterator.h"
Chandler Carruth9fb823b2013-01-02 11:36:10 +000031#include "llvm/IR/Instructions.h"
32#include "llvm/IR/IntrinsicInst.h"
33#include "llvm/IR/Module.h"
34#include "llvm/IR/Operator.h"
Chandler Carruth4220e9c2014-03-04 11:17:44 +000035#include "llvm/IR/ValueHandle.h"
Chris Lattner25db5802004-10-07 04:16:33 +000036#include "llvm/Pass.h"
Chris Lattner024f4ab2007-01-30 23:46:24 +000037#include "llvm/Support/Debug.h"
Torok Edwin56d06592009-07-11 20:10:48 +000038#include "llvm/Support/ErrorHandling.h"
Chris Lattner67ca6f632008-04-26 07:40:11 +000039#include "llvm/Support/MathExtras.h"
Chandler Carruth8a8cd2b2014-01-07 11:48:04 +000040#include "llvm/Support/raw_ostream.h"
Chandler Carruthed0881b2012-12-03 16:50:05 +000041#include "llvm/Target/TargetLibraryInfo.h"
Nico Weber4b2acde2014-05-02 18:35:25 +000042#include "llvm/Transforms/Utils/CtorUtils.h"
Rafael Espindola3d7fc252013-10-21 17:14:55 +000043#include "llvm/Transforms/Utils/GlobalStatus.h"
Rafael Espindola17600e22013-07-25 03:23:25 +000044#include "llvm/Transforms/Utils/ModuleUtils.h"
Chris Lattner25db5802004-10-07 04:16:33 +000045#include <algorithm>
Benjamin Kramer64425fe2014-05-03 15:50:37 +000046#include <deque>
Chris Lattner25db5802004-10-07 04:16:33 +000047using namespace llvm;
48
Chandler Carruth964daaa2014-04-22 02:55:47 +000049#define DEBUG_TYPE "globalopt"
50
Chris Lattner1631bcb2006-12-19 22:09:18 +000051STATISTIC(NumMarked , "Number of globals marked constant");
Rafael Espindolafc355bc2011-01-19 16:32:21 +000052STATISTIC(NumUnnamed , "Number of globals marked unnamed_addr");
Chris Lattner1631bcb2006-12-19 22:09:18 +000053STATISTIC(NumSRA , "Number of aggregate globals broken into scalars");
54STATISTIC(NumHeapSRA , "Number of heap objects SRA'd");
55STATISTIC(NumSubstitute,"Number of globals with initializers stored into them");
56STATISTIC(NumDeleted , "Number of globals deleted");
57STATISTIC(NumFnDeleted , "Number of functions deleted");
58STATISTIC(NumGlobUses , "Number of global uses devirtualized");
Alexey Samsonova1944e62013-10-07 19:03:24 +000059STATISTIC(NumLocalized , "Number of globals localized");
Chris Lattner1631bcb2006-12-19 22:09:18 +000060STATISTIC(NumShrunkToBool , "Number of global vars shrunk to booleans");
61STATISTIC(NumFastCallFns , "Number of functions converted to fastcc");
62STATISTIC(NumCtorsEvaluated, "Number of static ctors evaluated");
Duncan Sands573b3f82008-02-16 20:56:04 +000063STATISTIC(NumNestRemoved , "Number of nest attributes removed");
Duncan Sandsb3f27882009-02-15 09:56:08 +000064STATISTIC(NumAliasesResolved, "Number of global aliases resolved");
65STATISTIC(NumAliasesRemoved, "Number of global aliases eliminated");
Anders Carlssonee6bc702011-03-20 17:59:11 +000066STATISTIC(NumCXXDtorsRemoved, "Number of global C++ destructors removed");
Chris Lattner25db5802004-10-07 04:16:33 +000067
Chris Lattner1631bcb2006-12-19 22:09:18 +000068namespace {
Nick Lewycky02d5f772009-10-25 06:33:48 +000069 struct GlobalOpt : public ModulePass {
Craig Topper3e4c6972014-03-05 09:10:37 +000070 void getAnalysisUsage(AnalysisUsage &AU) const override {
Chad Rosiere6de63d2011-12-01 21:29:16 +000071 AU.addRequired<TargetLibraryInfo>();
Chris Lattner004e2502004-10-11 05:54:41 +000072 }
Nick Lewyckye7da2d62007-05-06 13:37:16 +000073 static char ID; // Pass identification, replacement for typeid
Owen Anderson6c18d1a2010-10-19 17:21:58 +000074 GlobalOpt() : ModulePass(ID) {
75 initializeGlobalOptPass(*PassRegistry::getPassRegistry());
76 }
Misha Brukmanb1c93172005-04-21 23:48:37 +000077
Craig Topper3e4c6972014-03-05 09:10:37 +000078 bool runOnModule(Module &M) override;
Chris Lattner004e2502004-10-11 05:54:41 +000079
80 private:
Chris Lattner41b6a5a2005-09-26 01:43:45 +000081 bool OptimizeFunctions(Module &M);
82 bool OptimizeGlobalVars(Module &M);
Duncan Sandsed722832009-03-06 10:21:56 +000083 bool OptimizeGlobalAliases(Module &M);
Rafael Espindolafc355bc2011-01-19 16:32:21 +000084 bool ProcessGlobal(GlobalVariable *GV,Module::global_iterator &GVI);
85 bool ProcessInternalGlobal(GlobalVariable *GV,Module::global_iterator &GVI,
Rafael Espindolafc355bc2011-01-19 16:32:21 +000086 const GlobalStatus &GS);
Anders Carlssonee6bc702011-03-20 17:59:11 +000087 bool OptimizeEmptyGlobalCXXDtors(Function *CXAAtExitFn);
Nick Lewyckycf6aae62012-02-12 01:13:18 +000088
Rafael Espindolaaeff8a92014-02-24 23:12:18 +000089 const DataLayout *DL;
Nick Lewyckycf6aae62012-02-12 01:13:18 +000090 TargetLibraryInfo *TLI;
Chris Lattner25db5802004-10-07 04:16:33 +000091 };
Chris Lattner25db5802004-10-07 04:16:33 +000092}
93
Dan Gohmand78c4002008-05-13 00:00:25 +000094char GlobalOpt::ID = 0;
Chad Rosiere6de63d2011-12-01 21:29:16 +000095INITIALIZE_PASS_BEGIN(GlobalOpt, "globalopt",
96 "Global Variable Optimizer", false, false)
97INITIALIZE_PASS_DEPENDENCY(TargetLibraryInfo)
98INITIALIZE_PASS_END(GlobalOpt, "globalopt",
Owen Andersondf7a4f22010-10-07 22:25:06 +000099 "Global Variable Optimizer", false, false)
Dan Gohmand78c4002008-05-13 00:00:25 +0000100
Chris Lattner25db5802004-10-07 04:16:33 +0000101ModulePass *llvm::createGlobalOptimizerPass() { return new GlobalOpt(); }
102
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000103/// isLeakCheckerRoot - Is this global variable possibly used by a leak checker
104/// as a root? If so, we might not really want to eliminate the stores to it.
105static bool isLeakCheckerRoot(GlobalVariable *GV) {
106 // A global variable is a root if it is a pointer, or could plausibly contain
107 // a pointer. There are two challenges; one is that we could have a struct
108 // the has an inner member which is a pointer. We recurse through the type to
109 // detect these (up to a point). The other is that we may actually be a union
110 // of a pointer and another type, and so our LLVM type is an integer which
111 // gets converted into a pointer, or our type is an [i8 x #] with a pointer
112 // potentially contained here.
113
114 if (GV->hasPrivateLinkage())
115 return false;
116
117 SmallVector<Type *, 4> Types;
118 Types.push_back(cast<PointerType>(GV->getType())->getElementType());
119
120 unsigned Limit = 20;
121 do {
122 Type *Ty = Types.pop_back_val();
123 switch (Ty->getTypeID()) {
124 default: break;
125 case Type::PointerTyID: return true;
126 case Type::ArrayTyID:
127 case Type::VectorTyID: {
128 SequentialType *STy = cast<SequentialType>(Ty);
129 Types.push_back(STy->getElementType());
130 break;
131 }
132 case Type::StructTyID: {
133 StructType *STy = cast<StructType>(Ty);
134 if (STy->isOpaque()) return true;
135 for (StructType::element_iterator I = STy->element_begin(),
136 E = STy->element_end(); I != E; ++I) {
137 Type *InnerTy = *I;
138 if (isa<PointerType>(InnerTy)) return true;
139 if (isa<CompositeType>(InnerTy))
140 Types.push_back(InnerTy);
141 }
142 break;
143 }
144 }
145 if (--Limit == 0) return true;
146 } while (!Types.empty());
147 return false;
148}
149
150/// Given a value that is stored to a global but never read, determine whether
151/// it's safe to remove the store and the chain of computation that feeds the
152/// store.
Benjamin Kramer8bcc9712012-08-29 15:32:21 +0000153static bool IsSafeComputationToRemove(Value *V, const TargetLibraryInfo *TLI) {
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000154 do {
155 if (isa<Constant>(V))
156 return true;
157 if (!V->hasOneUse())
158 return false;
Nick Lewycky7d0f1102012-07-25 21:19:40 +0000159 if (isa<LoadInst>(V) || isa<InvokeInst>(V) || isa<Argument>(V) ||
160 isa<GlobalValue>(V))
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000161 return false;
Benjamin Kramer8bcc9712012-08-29 15:32:21 +0000162 if (isAllocationFn(V, TLI))
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000163 return true;
164
165 Instruction *I = cast<Instruction>(V);
166 if (I->mayHaveSideEffects())
167 return false;
168 if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(I)) {
169 if (!GEP->hasAllConstantIndices())
170 return false;
171 } else if (I->getNumOperands() != 1) {
172 return false;
173 }
174
175 V = I->getOperand(0);
176 } while (1);
177}
178
179/// CleanupPointerRootUsers - This GV is a pointer root. Loop over all users
180/// of the global and clean up any that obviously don't assign the global a
181/// value that isn't dynamically allocated.
182///
Benjamin Kramer8bcc9712012-08-29 15:32:21 +0000183static bool CleanupPointerRootUsers(GlobalVariable *GV,
184 const TargetLibraryInfo *TLI) {
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000185 // A brief explanation of leak checkers. The goal is to find bugs where
186 // pointers are forgotten, causing an accumulating growth in memory
187 // usage over time. The common strategy for leak checkers is to whitelist the
188 // memory pointed to by globals at exit. This is popular because it also
189 // solves another problem where the main thread of a C++ program may shut down
190 // before other threads that are still expecting to use those globals. To
191 // handle that case, we expect the program may create a singleton and never
192 // destroy it.
193
194 bool Changed = false;
195
196 // If Dead[n].first is the only use of a malloc result, we can delete its
197 // chain of computation and the store to the global in Dead[n].second.
198 SmallVector<std::pair<Instruction *, Instruction *>, 32> Dead;
199
200 // Constants can't be pointers to dynamically allocated memory.
Chandler Carruthcdf47882014-03-09 03:16:01 +0000201 for (Value::user_iterator UI = GV->user_begin(), E = GV->user_end();
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000202 UI != E;) {
203 User *U = *UI++;
204 if (StoreInst *SI = dyn_cast<StoreInst>(U)) {
205 Value *V = SI->getValueOperand();
206 if (isa<Constant>(V)) {
207 Changed = true;
208 SI->eraseFromParent();
209 } else if (Instruction *I = dyn_cast<Instruction>(V)) {
210 if (I->hasOneUse())
211 Dead.push_back(std::make_pair(I, SI));
212 }
213 } else if (MemSetInst *MSI = dyn_cast<MemSetInst>(U)) {
214 if (isa<Constant>(MSI->getValue())) {
215 Changed = true;
216 MSI->eraseFromParent();
217 } else if (Instruction *I = dyn_cast<Instruction>(MSI->getValue())) {
218 if (I->hasOneUse())
219 Dead.push_back(std::make_pair(I, MSI));
220 }
221 } else if (MemTransferInst *MTI = dyn_cast<MemTransferInst>(U)) {
222 GlobalVariable *MemSrc = dyn_cast<GlobalVariable>(MTI->getSource());
223 if (MemSrc && MemSrc->isConstant()) {
224 Changed = true;
225 MTI->eraseFromParent();
226 } else if (Instruction *I = dyn_cast<Instruction>(MemSrc)) {
227 if (I->hasOneUse())
228 Dead.push_back(std::make_pair(I, MTI));
229 }
230 } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(U)) {
231 if (CE->use_empty()) {
232 CE->destroyConstant();
233 Changed = true;
234 }
235 } else if (Constant *C = dyn_cast<Constant>(U)) {
Rafael Espindola27797ba2013-10-17 18:06:32 +0000236 if (isSafeToDestroyConstant(C)) {
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000237 C->destroyConstant();
238 // This could have invalidated UI, start over from scratch.
239 Dead.clear();
Benjamin Kramer8bcc9712012-08-29 15:32:21 +0000240 CleanupPointerRootUsers(GV, TLI);
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000241 return true;
242 }
243 }
244 }
245
246 for (int i = 0, e = Dead.size(); i != e; ++i) {
Benjamin Kramer8bcc9712012-08-29 15:32:21 +0000247 if (IsSafeComputationToRemove(Dead[i].first, TLI)) {
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000248 Dead[i].second->eraseFromParent();
249 Instruction *I = Dead[i].first;
250 do {
Michael Gottesman2a654272013-01-11 23:08:52 +0000251 if (isAllocationFn(I, TLI))
252 break;
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000253 Instruction *J = dyn_cast<Instruction>(I->getOperand(0));
254 if (!J)
255 break;
256 I->eraseFromParent();
257 I = J;
Nick Lewycky38be9312012-07-24 21:33:00 +0000258 } while (1);
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000259 I->eraseFromParent();
260 }
261 }
262
263 return Changed;
264}
265
Chris Lattner25db5802004-10-07 04:16:33 +0000266/// CleanupConstantGlobalUsers - We just marked GV constant. Loop over all
267/// users of the global, cleaning up the obvious ones. This is largely just a
Chris Lattnercb9f1522004-10-10 16:43:46 +0000268/// quick scan over the use list to clean up the easy and obvious cruft. This
269/// returns true if it made a change.
Nick Lewyckycf6aae62012-02-12 01:13:18 +0000270static bool CleanupConstantGlobalUsers(Value *V, Constant *Init,
Rafael Espindolaaeff8a92014-02-24 23:12:18 +0000271 const DataLayout *DL,
272 TargetLibraryInfo *TLI) {
Chris Lattnercb9f1522004-10-10 16:43:46 +0000273 bool Changed = false;
Hal Finkelf59fd7d2013-12-12 20:45:24 +0000274 // Note that we need to use a weak value handle for the worklist items. When
275 // we delete a constant array, we may also be holding pointer to one of its
276 // elements (or an element of one of its elements if we're dealing with an
277 // array of arrays) in the worklist.
Chandler Carruthcdf47882014-03-09 03:16:01 +0000278 SmallVector<WeakVH, 8> WorkList(V->user_begin(), V->user_end());
Bill Wendling88d06c32013-04-02 08:16:45 +0000279 while (!WorkList.empty()) {
Hal Finkelf59fd7d2013-12-12 20:45:24 +0000280 Value *UV = WorkList.pop_back_val();
281 if (!UV)
282 continue;
283
284 User *U = cast<User>(UV);
Misha Brukmanb1c93172005-04-21 23:48:37 +0000285
Chris Lattner25db5802004-10-07 04:16:33 +0000286 if (LoadInst *LI = dyn_cast<LoadInst>(U)) {
Chris Lattner7561ca12005-02-27 18:58:52 +0000287 if (Init) {
288 // Replace the load with the initializer.
289 LI->replaceAllUsesWith(Init);
290 LI->eraseFromParent();
291 Changed = true;
292 }
Chris Lattner25db5802004-10-07 04:16:33 +0000293 } else if (StoreInst *SI = dyn_cast<StoreInst>(U)) {
294 // Store must be unreachable or storing Init into the global.
Chris Lattner8e71c6a2004-10-16 18:09:00 +0000295 SI->eraseFromParent();
Chris Lattnercb9f1522004-10-10 16:43:46 +0000296 Changed = true;
Chris Lattner25db5802004-10-07 04:16:33 +0000297 } else if (ConstantExpr *CE = dyn_cast<ConstantExpr>(U)) {
298 if (CE->getOpcode() == Instruction::GetElementPtr) {
Craig Topperf40110f2014-04-25 05:29:35 +0000299 Constant *SubInit = nullptr;
Chris Lattner46d9ff082005-09-26 07:34:35 +0000300 if (Init)
Dan Gohmane525d9d2009-10-05 16:36:26 +0000301 SubInit = ConstantFoldLoadThroughGEPConstantExpr(Init, CE);
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000302 Changed |= CleanupConstantGlobalUsers(CE, SubInit, DL, TLI);
Matt Arsenault461c8e02014-01-02 20:01:43 +0000303 } else if ((CE->getOpcode() == Instruction::BitCast &&
304 CE->getType()->isPointerTy()) ||
305 CE->getOpcode() == Instruction::AddrSpaceCast) {
Chris Lattner7561ca12005-02-27 18:58:52 +0000306 // Pointer cast, delete any stores and memsets to the global.
Craig Topperf40110f2014-04-25 05:29:35 +0000307 Changed |= CleanupConstantGlobalUsers(CE, nullptr, DL, TLI);
Chris Lattner7561ca12005-02-27 18:58:52 +0000308 }
309
310 if (CE->use_empty()) {
311 CE->destroyConstant();
312 Changed = true;
Chris Lattner25db5802004-10-07 04:16:33 +0000313 }
314 } else if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(U)) {
Chris Lattnerf9c0fd72007-11-09 17:33:02 +0000315 // Do not transform "gepinst (gep constexpr (GV))" here, because forming
316 // "gepconstexpr (gep constexpr (GV))" will cause the two gep's to fold
317 // and will invalidate our notion of what Init is.
Craig Topperf40110f2014-04-25 05:29:35 +0000318 Constant *SubInit = nullptr;
Chris Lattnerf9c0fd72007-11-09 17:33:02 +0000319 if (!isa<ConstantExpr>(GEP->getOperand(0))) {
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000320 ConstantExpr *CE =
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000321 dyn_cast_or_null<ConstantExpr>(ConstantFoldInstruction(GEP, DL, TLI));
Chris Lattnerf9c0fd72007-11-09 17:33:02 +0000322 if (Init && CE && CE->getOpcode() == Instruction::GetElementPtr)
Dan Gohmane525d9d2009-10-05 16:36:26 +0000323 SubInit = ConstantFoldLoadThroughGEPConstantExpr(Init, CE);
Benjamin Krameraa9e4a52012-03-28 14:50:09 +0000324
325 // If the initializer is an all-null value and we have an inbounds GEP,
326 // we already know what the result of any load from that GEP is.
327 // TODO: Handle splats.
328 if (Init && isa<ConstantAggregateZero>(Init) && GEP->isInBounds())
329 SubInit = Constant::getNullValue(GEP->getType()->getElementType());
Chris Lattnerf9c0fd72007-11-09 17:33:02 +0000330 }
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000331 Changed |= CleanupConstantGlobalUsers(GEP, SubInit, DL, TLI);
Chris Lattnera0e769c2004-10-10 16:47:33 +0000332
Chris Lattnercb9f1522004-10-10 16:43:46 +0000333 if (GEP->use_empty()) {
Chris Lattner8e71c6a2004-10-16 18:09:00 +0000334 GEP->eraseFromParent();
Chris Lattnercb9f1522004-10-10 16:43:46 +0000335 Changed = true;
336 }
Chris Lattner7561ca12005-02-27 18:58:52 +0000337 } else if (MemIntrinsic *MI = dyn_cast<MemIntrinsic>(U)) { // memset/cpy/mv
338 if (MI->getRawDest() == V) {
339 MI->eraseFromParent();
340 Changed = true;
341 }
342
Chris Lattner1c4bddc2004-10-08 20:59:28 +0000343 } else if (Constant *C = dyn_cast<Constant>(U)) {
344 // If we have a chain of dead constantexprs or other things dangling from
345 // us, and if they are all dead, nuke them without remorse.
Rafael Espindola27797ba2013-10-17 18:06:32 +0000346 if (isSafeToDestroyConstant(C)) {
Devang Pateld926aaa2009-03-06 01:37:41 +0000347 C->destroyConstant();
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000348 CleanupConstantGlobalUsers(V, Init, DL, TLI);
Chris Lattnercb9f1522004-10-10 16:43:46 +0000349 return true;
Chris Lattner1c4bddc2004-10-08 20:59:28 +0000350 }
Chris Lattner25db5802004-10-07 04:16:33 +0000351 }
352 }
Chris Lattnercb9f1522004-10-10 16:43:46 +0000353 return Changed;
Chris Lattner25db5802004-10-07 04:16:33 +0000354}
355
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000356/// isSafeSROAElementUse - Return true if the specified instruction is a safe
357/// user of a derived expression from a global that we want to SROA.
358static bool isSafeSROAElementUse(Value *V) {
359 // We might have a dead and dangling constant hanging off of here.
360 if (Constant *C = dyn_cast<Constant>(V))
Rafael Espindola27797ba2013-10-17 18:06:32 +0000361 return isSafeToDestroyConstant(C);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000362
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000363 Instruction *I = dyn_cast<Instruction>(V);
364 if (!I) return false;
365
366 // Loads are ok.
367 if (isa<LoadInst>(I)) return true;
368
369 // Stores *to* the pointer are ok.
370 if (StoreInst *SI = dyn_cast<StoreInst>(I))
371 return SI->getOperand(0) != V;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000372
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000373 // Otherwise, it must be a GEP.
374 GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(I);
Craig Topperf40110f2014-04-25 05:29:35 +0000375 if (!GEPI) return false;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000376
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000377 if (GEPI->getNumOperands() < 3 || !isa<Constant>(GEPI->getOperand(1)) ||
378 !cast<Constant>(GEPI->getOperand(1))->isNullValue())
379 return false;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000380
Chandler Carruthcdf47882014-03-09 03:16:01 +0000381 for (User *U : GEPI->users())
382 if (!isSafeSROAElementUse(U))
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000383 return false;
Chris Lattnerab053722008-01-14 01:31:05 +0000384 return true;
385}
386
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000387
388/// IsUserOfGlobalSafeForSRA - U is a direct user of the specified global value.
389/// Look at it and its uses and decide whether it is safe to SROA this global.
390///
391static bool IsUserOfGlobalSafeForSRA(User *U, GlobalValue *GV) {
392 // The user of the global must be a GEP Inst or a ConstantExpr GEP.
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000393 if (!isa<GetElementPtrInst>(U) &&
394 (!isa<ConstantExpr>(U) ||
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000395 cast<ConstantExpr>(U)->getOpcode() != Instruction::GetElementPtr))
396 return false;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000397
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000398 // Check to see if this ConstantExpr GEP is SRA'able. In particular, we
399 // don't like < 3 operand CE's, and we don't like non-constant integer
400 // indices. This enforces that all uses are 'gep GV, 0, C, ...' for some
401 // value of C.
402 if (U->getNumOperands() < 3 || !isa<Constant>(U->getOperand(1)) ||
403 !cast<Constant>(U->getOperand(1))->isNullValue() ||
404 !isa<ConstantInt>(U->getOperand(2)))
405 return false;
406
407 gep_type_iterator GEPI = gep_type_begin(U), E = gep_type_end(U);
408 ++GEPI; // Skip over the pointer index.
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000409
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000410 // If this is a use of an array allocation, do a bit more checking for sanity.
Chris Lattner229907c2011-07-18 04:54:35 +0000411 if (ArrayType *AT = dyn_cast<ArrayType>(*GEPI)) {
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000412 uint64_t NumElements = AT->getNumElements();
413 ConstantInt *Idx = cast<ConstantInt>(U->getOperand(2));
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000414
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000415 // Check to make sure that index falls within the array. If not,
416 // something funny is going on, so we won't do the optimization.
417 //
418 if (Idx->getZExtValue() >= NumElements)
419 return false;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000420
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000421 // We cannot scalar repl this level of the array unless any array
422 // sub-indices are in-range constants. In particular, consider:
423 // A[0][i]. We cannot know that the user isn't doing invalid things like
424 // allowing i to index an out-of-range subscript that accesses A[1].
425 //
426 // Scalar replacing *just* the outer index of the array is probably not
427 // going to be a win anyway, so just give up.
428 for (++GEPI; // Skip array index.
Dan Gohman82ac81b2009-08-18 14:58:19 +0000429 GEPI != E;
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000430 ++GEPI) {
431 uint64_t NumElements;
Chris Lattner229907c2011-07-18 04:54:35 +0000432 if (ArrayType *SubArrayTy = dyn_cast<ArrayType>(*GEPI))
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000433 NumElements = SubArrayTy->getNumElements();
Chris Lattner229907c2011-07-18 04:54:35 +0000434 else if (VectorType *SubVectorTy = dyn_cast<VectorType>(*GEPI))
Dan Gohman82ac81b2009-08-18 14:58:19 +0000435 NumElements = SubVectorTy->getNumElements();
436 else {
Duncan Sands19d0b472010-02-16 11:11:14 +0000437 assert((*GEPI)->isStructTy() &&
Dan Gohman82ac81b2009-08-18 14:58:19 +0000438 "Indexed GEP type is not array, vector, or struct!");
439 continue;
440 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000441
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000442 ConstantInt *IdxVal = dyn_cast<ConstantInt>(GEPI.getOperand());
443 if (!IdxVal || IdxVal->getZExtValue() >= NumElements)
444 return false;
445 }
446 }
447
Chandler Carruthcdf47882014-03-09 03:16:01 +0000448 for (User *UU : U->users())
449 if (!isSafeSROAElementUse(UU))
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000450 return false;
Chandler Carruthcdf47882014-03-09 03:16:01 +0000451
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000452 return true;
453}
454
455/// GlobalUsersSafeToSRA - Look at all uses of the global and decide whether it
456/// is safe for us to perform this transformation.
457///
458static bool GlobalUsersSafeToSRA(GlobalValue *GV) {
Chandler Carruthcdf47882014-03-09 03:16:01 +0000459 for (User *U : GV->users())
460 if (!IsUserOfGlobalSafeForSRA(U, GV))
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000461 return false;
Chandler Carruthcdf47882014-03-09 03:16:01 +0000462
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000463 return true;
464}
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000465
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000466
Chris Lattnerabab0712004-10-08 17:32:09 +0000467/// SRAGlobal - Perform scalar replacement of aggregates on the specified global
468/// variable. This opens the door for other optimizations by exposing the
469/// behavior of the program in a more fine-grained way. We have determined that
470/// this transformation is safe already. We return the first global variable we
471/// insert so that the caller can reprocess it.
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000472static GlobalVariable *SRAGlobal(GlobalVariable *GV, const DataLayout &DL) {
Chris Lattnerab053722008-01-14 01:31:05 +0000473 // Make sure this global only has simple uses that we can SRA.
Chris Lattner26fe7eb2008-01-14 02:09:12 +0000474 if (!GlobalUsersSafeToSRA(GV))
Craig Topperf40110f2014-04-25 05:29:35 +0000475 return nullptr;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000476
Rafael Espindola6de96a12009-01-15 20:18:42 +0000477 assert(GV->hasLocalLinkage() && !GV->isConstant());
Chris Lattnerabab0712004-10-08 17:32:09 +0000478 Constant *Init = GV->getInitializer();
Chris Lattner229907c2011-07-18 04:54:35 +0000479 Type *Ty = Init->getType();
Misha Brukmanb1c93172005-04-21 23:48:37 +0000480
Chris Lattnerabab0712004-10-08 17:32:09 +0000481 std::vector<GlobalVariable*> NewGlobals;
482 Module::GlobalListType &Globals = GV->getParent()->getGlobalList();
483
Chris Lattner67ca6f632008-04-26 07:40:11 +0000484 // Get the alignment of the global, either explicit or target-specific.
485 unsigned StartAlignment = GV->getAlignment();
486 if (StartAlignment == 0)
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000487 StartAlignment = DL.getABITypeAlignment(GV->getType());
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000488
Chris Lattner229907c2011-07-18 04:54:35 +0000489 if (StructType *STy = dyn_cast<StructType>(Ty)) {
Chris Lattnerabab0712004-10-08 17:32:09 +0000490 NewGlobals.reserve(STy->getNumElements());
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000491 const StructLayout &Layout = *DL.getStructLayout(STy);
Chris Lattnerabab0712004-10-08 17:32:09 +0000492 for (unsigned i = 0, e = STy->getNumElements(); i != e; ++i) {
Chris Lattner67058832012-01-25 06:48:06 +0000493 Constant *In = Init->getAggregateElement(i);
Chris Lattnerabab0712004-10-08 17:32:09 +0000494 assert(In && "Couldn't get element of initializer?");
Chris Lattner46b5c642009-11-06 04:27:31 +0000495 GlobalVariable *NGV = new GlobalVariable(STy->getElementType(i), false,
Chris Lattnerabab0712004-10-08 17:32:09 +0000496 GlobalVariable::InternalLinkage,
Daniel Dunbar132f7832009-07-30 17:37:43 +0000497 In, GV->getName()+"."+Twine(i),
Hans Wennborgcbe34b42012-06-23 11:37:03 +0000498 GV->getThreadLocalMode(),
Owen Anderson5948fdf2009-07-08 01:26:06 +0000499 GV->getType()->getAddressSpace());
Chris Lattnerabab0712004-10-08 17:32:09 +0000500 Globals.insert(GV, NGV);
501 NewGlobals.push_back(NGV);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000502
Chris Lattner67ca6f632008-04-26 07:40:11 +0000503 // Calculate the known alignment of the field. If the original aggregate
504 // had 256 byte alignment for example, something might depend on that:
505 // propagate info to each field.
506 uint64_t FieldOffset = Layout.getElementOffset(i);
507 unsigned NewAlign = (unsigned)MinAlign(StartAlignment, FieldOffset);
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000508 if (NewAlign > DL.getABITypeAlignment(STy->getElementType(i)))
Chris Lattner67ca6f632008-04-26 07:40:11 +0000509 NGV->setAlignment(NewAlign);
Chris Lattnerabab0712004-10-08 17:32:09 +0000510 }
Chris Lattner229907c2011-07-18 04:54:35 +0000511 } else if (SequentialType *STy = dyn_cast<SequentialType>(Ty)) {
Chris Lattnerabab0712004-10-08 17:32:09 +0000512 unsigned NumElements = 0;
Chris Lattner229907c2011-07-18 04:54:35 +0000513 if (ArrayType *ATy = dyn_cast<ArrayType>(STy))
Chris Lattnerabab0712004-10-08 17:32:09 +0000514 NumElements = ATy->getNumElements();
Chris Lattnerabab0712004-10-08 17:32:09 +0000515 else
Chris Lattner67ca6f632008-04-26 07:40:11 +0000516 NumElements = cast<VectorType>(STy)->getNumElements();
Chris Lattnerabab0712004-10-08 17:32:09 +0000517
Chris Lattner25169ca2005-02-23 16:53:04 +0000518 if (NumElements > 16 && GV->hasNUsesOrMore(16))
Craig Topperf40110f2014-04-25 05:29:35 +0000519 return nullptr; // It's not worth it.
Chris Lattnerabab0712004-10-08 17:32:09 +0000520 NewGlobals.reserve(NumElements);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000521
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000522 uint64_t EltSize = DL.getTypeAllocSize(STy->getElementType());
523 unsigned EltAlign = DL.getABITypeAlignment(STy->getElementType());
Chris Lattnerabab0712004-10-08 17:32:09 +0000524 for (unsigned i = 0, e = NumElements; i != e; ++i) {
Chris Lattner67058832012-01-25 06:48:06 +0000525 Constant *In = Init->getAggregateElement(i);
Chris Lattnerabab0712004-10-08 17:32:09 +0000526 assert(In && "Couldn't get element of initializer?");
527
Chris Lattner46b5c642009-11-06 04:27:31 +0000528 GlobalVariable *NGV = new GlobalVariable(STy->getElementType(), false,
Chris Lattnerabab0712004-10-08 17:32:09 +0000529 GlobalVariable::InternalLinkage,
Daniel Dunbar132f7832009-07-30 17:37:43 +0000530 In, GV->getName()+"."+Twine(i),
Hans Wennborgcbe34b42012-06-23 11:37:03 +0000531 GV->getThreadLocalMode(),
Owen Andersonb17f3292009-07-08 19:03:57 +0000532 GV->getType()->getAddressSpace());
Chris Lattnerabab0712004-10-08 17:32:09 +0000533 Globals.insert(GV, NGV);
534 NewGlobals.push_back(NGV);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000535
Chris Lattner67ca6f632008-04-26 07:40:11 +0000536 // Calculate the known alignment of the field. If the original aggregate
537 // had 256 byte alignment for example, something might depend on that:
538 // propagate info to each field.
539 unsigned NewAlign = (unsigned)MinAlign(StartAlignment, EltSize*i);
540 if (NewAlign > EltAlign)
541 NGV->setAlignment(NewAlign);
Chris Lattnerabab0712004-10-08 17:32:09 +0000542 }
543 }
544
545 if (NewGlobals.empty())
Craig Topperf40110f2014-04-25 05:29:35 +0000546 return nullptr;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000547
David Greene44cb8ad2010-01-05 01:28:05 +0000548 DEBUG(dbgs() << "PERFORMING GLOBAL SRA ON: " << *GV);
Chris Lattner004e2502004-10-11 05:54:41 +0000549
Chris Lattner46b5c642009-11-06 04:27:31 +0000550 Constant *NullInt =Constant::getNullValue(Type::getInt32Ty(GV->getContext()));
Chris Lattnerabab0712004-10-08 17:32:09 +0000551
552 // Loop over all of the uses of the global, replacing the constantexpr geps,
553 // with smaller constantexpr geps or direct references.
554 while (!GV->use_empty()) {
Chandler Carruthcdf47882014-03-09 03:16:01 +0000555 User *GEP = GV->user_back();
Chris Lattner004e2502004-10-11 05:54:41 +0000556 assert(((isa<ConstantExpr>(GEP) &&
557 cast<ConstantExpr>(GEP)->getOpcode()==Instruction::GetElementPtr)||
558 isa<GetElementPtrInst>(GEP)) && "NonGEP CE's are not SRAable!");
Misha Brukmanb1c93172005-04-21 23:48:37 +0000559
Chris Lattnerabab0712004-10-08 17:32:09 +0000560 // Ignore the 1th operand, which has to be zero or else the program is quite
561 // broken (undefined). Get the 2nd operand, which is the structure or array
562 // index.
Reid Spencere0fc4df2006-10-20 07:07:24 +0000563 unsigned Val = cast<ConstantInt>(GEP->getOperand(2))->getZExtValue();
Chris Lattnerabab0712004-10-08 17:32:09 +0000564 if (Val >= NewGlobals.size()) Val = 0; // Out of bound array access.
565
Chris Lattner004e2502004-10-11 05:54:41 +0000566 Value *NewPtr = NewGlobals[Val];
Chris Lattnerabab0712004-10-08 17:32:09 +0000567
568 // Form a shorter GEP if needed.
Anton Korobeynikov1bfd1212008-02-20 11:26:25 +0000569 if (GEP->getNumOperands() > 3) {
Chris Lattner004e2502004-10-11 05:54:41 +0000570 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(GEP)) {
Chris Lattnerf96f4a82007-01-31 04:40:53 +0000571 SmallVector<Constant*, 8> Idxs;
Chris Lattner004e2502004-10-11 05:54:41 +0000572 Idxs.push_back(NullInt);
573 for (unsigned i = 3, e = CE->getNumOperands(); i != e; ++i)
574 Idxs.push_back(CE->getOperand(i));
Jay Foaded8db7d2011-07-21 14:31:17 +0000575 NewPtr = ConstantExpr::getGetElementPtr(cast<Constant>(NewPtr), Idxs);
Chris Lattner004e2502004-10-11 05:54:41 +0000576 } else {
577 GetElementPtrInst *GEPI = cast<GetElementPtrInst>(GEP);
Chris Lattner927653f2007-01-31 19:59:55 +0000578 SmallVector<Value*, 8> Idxs;
Chris Lattner004e2502004-10-11 05:54:41 +0000579 Idxs.push_back(NullInt);
580 for (unsigned i = 3, e = GEPI->getNumOperands(); i != e; ++i)
581 Idxs.push_back(GEPI->getOperand(i));
Jay Foadd1b78492011-07-25 09:48:08 +0000582 NewPtr = GetElementPtrInst::Create(NewPtr, Idxs,
Daniel Dunbar132f7832009-07-30 17:37:43 +0000583 GEPI->getName()+"."+Twine(Val),GEPI);
Chris Lattner004e2502004-10-11 05:54:41 +0000584 }
Anton Korobeynikov1bfd1212008-02-20 11:26:25 +0000585 }
Chris Lattner004e2502004-10-11 05:54:41 +0000586 GEP->replaceAllUsesWith(NewPtr);
587
588 if (GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(GEP))
Chris Lattner8e71c6a2004-10-16 18:09:00 +0000589 GEPI->eraseFromParent();
Chris Lattner004e2502004-10-11 05:54:41 +0000590 else
591 cast<ConstantExpr>(GEP)->destroyConstant();
Chris Lattnerabab0712004-10-08 17:32:09 +0000592 }
593
Chris Lattner73ad73e2004-10-08 20:25:55 +0000594 // Delete the old global, now that it is dead.
595 Globals.erase(GV);
Chris Lattnerabab0712004-10-08 17:32:09 +0000596 ++NumSRA;
Chris Lattner004e2502004-10-11 05:54:41 +0000597
598 // Loop over the new globals array deleting any globals that are obviously
599 // dead. This can arise due to scalarization of a structure or an array that
600 // has elements that are dead.
601 unsigned FirstGlobal = 0;
602 for (unsigned i = 0, e = NewGlobals.size(); i != e; ++i)
603 if (NewGlobals[i]->use_empty()) {
604 Globals.erase(NewGlobals[i]);
605 if (FirstGlobal == i) ++FirstGlobal;
606 }
607
Craig Topperf40110f2014-04-25 05:29:35 +0000608 return FirstGlobal != NewGlobals.size() ? NewGlobals[FirstGlobal] : nullptr;
Chris Lattnerabab0712004-10-08 17:32:09 +0000609}
610
Chris Lattner09a52722004-10-09 21:48:45 +0000611/// AllUsesOfValueWillTrapIfNull - Return true if all users of the specified
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000612/// value will trap if the value is dynamically null. PHIs keeps track of any
Chris Lattner2d2892e2007-09-13 16:30:19 +0000613/// phi nodes we've seen to avoid reprocessing them.
Gabor Greif67972872010-04-06 19:24:18 +0000614static bool AllUsesOfValueWillTrapIfNull(const Value *V,
615 SmallPtrSet<const PHINode*, 8> &PHIs) {
Chandler Carruthcdf47882014-03-09 03:16:01 +0000616 for (const User *U : V->users())
Gabor Greif08355d62010-04-06 19:14:05 +0000617 if (isa<LoadInst>(U)) {
Chris Lattner09a52722004-10-09 21:48:45 +0000618 // Will trap.
Gabor Greif67972872010-04-06 19:24:18 +0000619 } else if (const StoreInst *SI = dyn_cast<StoreInst>(U)) {
Chris Lattner09a52722004-10-09 21:48:45 +0000620 if (SI->getOperand(0) == V) {
Gabor Greif08355d62010-04-06 19:14:05 +0000621 //cerr << "NONTRAPPING USE: " << *U;
Chris Lattner09a52722004-10-09 21:48:45 +0000622 return false; // Storing the value.
623 }
Gabor Greif67972872010-04-06 19:24:18 +0000624 } else if (const CallInst *CI = dyn_cast<CallInst>(U)) {
Gabor Greiffebf6ab2010-03-20 21:00:25 +0000625 if (CI->getCalledValue() != V) {
Gabor Greif08355d62010-04-06 19:14:05 +0000626 //cerr << "NONTRAPPING USE: " << *U;
Chris Lattner09a52722004-10-09 21:48:45 +0000627 return false; // Not calling the ptr
628 }
Gabor Greif67972872010-04-06 19:24:18 +0000629 } else if (const InvokeInst *II = dyn_cast<InvokeInst>(U)) {
Gabor Greiffebf6ab2010-03-20 21:00:25 +0000630 if (II->getCalledValue() != V) {
Gabor Greif08355d62010-04-06 19:14:05 +0000631 //cerr << "NONTRAPPING USE: " << *U;
Chris Lattner09a52722004-10-09 21:48:45 +0000632 return false; // Not calling the ptr
633 }
Gabor Greif67972872010-04-06 19:24:18 +0000634 } else if (const BitCastInst *CI = dyn_cast<BitCastInst>(U)) {
Chris Lattner2d2892e2007-09-13 16:30:19 +0000635 if (!AllUsesOfValueWillTrapIfNull(CI, PHIs)) return false;
Gabor Greif67972872010-04-06 19:24:18 +0000636 } else if (const GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(U)) {
Chris Lattner2d2892e2007-09-13 16:30:19 +0000637 if (!AllUsesOfValueWillTrapIfNull(GEPI, PHIs)) return false;
Gabor Greif67972872010-04-06 19:24:18 +0000638 } else if (const PHINode *PN = dyn_cast<PHINode>(U)) {
Chris Lattner2d2892e2007-09-13 16:30:19 +0000639 // If we've already seen this phi node, ignore it, it has already been
640 // checked.
Jakob Stoklund Olesene27dc722010-01-29 23:54:14 +0000641 if (PHIs.insert(PN) && !AllUsesOfValueWillTrapIfNull(PN, PHIs))
642 return false;
Gabor Greif08355d62010-04-06 19:14:05 +0000643 } else if (isa<ICmpInst>(U) &&
Chandler Carruthcdf47882014-03-09 03:16:01 +0000644 isa<ConstantPointerNull>(U->getOperand(1))) {
Nick Lewycky614fb942010-02-25 06:39:10 +0000645 // Ignore icmp X, null
Chris Lattner09a52722004-10-09 21:48:45 +0000646 } else {
Gabor Greif08355d62010-04-06 19:14:05 +0000647 //cerr << "NONTRAPPING USE: " << *U;
Chris Lattner09a52722004-10-09 21:48:45 +0000648 return false;
649 }
Chandler Carruthcdf47882014-03-09 03:16:01 +0000650
Chris Lattner09a52722004-10-09 21:48:45 +0000651 return true;
652}
653
654/// AllUsesOfLoadedValueWillTrapIfNull - Return true if all uses of any loads
Chris Lattnerfe9abf92004-10-22 06:43:28 +0000655/// from GV will trap if the loaded value is null. Note that this also permits
656/// comparisons of the loaded value against null, as a special case.
Gabor Greif67972872010-04-06 19:24:18 +0000657static bool AllUsesOfLoadedValueWillTrapIfNull(const GlobalVariable *GV) {
Chandler Carruthcdf47882014-03-09 03:16:01 +0000658 for (const User *U : GV->users())
Gabor Greif67972872010-04-06 19:24:18 +0000659 if (const LoadInst *LI = dyn_cast<LoadInst>(U)) {
660 SmallPtrSet<const PHINode*, 8> PHIs;
Chris Lattner2d2892e2007-09-13 16:30:19 +0000661 if (!AllUsesOfValueWillTrapIfNull(LI, PHIs))
Chris Lattner09a52722004-10-09 21:48:45 +0000662 return false;
Gabor Greif08355d62010-04-06 19:14:05 +0000663 } else if (isa<StoreInst>(U)) {
Chris Lattner09a52722004-10-09 21:48:45 +0000664 // Ignore stores to the global.
665 } else {
666 // We don't know or understand this user, bail out.
Gabor Greif08355d62010-04-06 19:14:05 +0000667 //cerr << "UNKNOWN USER OF GLOBAL!: " << *U;
Chris Lattner09a52722004-10-09 21:48:45 +0000668 return false;
669 }
Chris Lattner09a52722004-10-09 21:48:45 +0000670 return true;
671}
672
Chris Lattner46b5c642009-11-06 04:27:31 +0000673static bool OptimizeAwayTrappingUsesOfValue(Value *V, Constant *NewV) {
Chris Lattnere42eb312004-10-10 23:14:11 +0000674 bool Changed = false;
Chandler Carruthcdf47882014-03-09 03:16:01 +0000675 for (auto UI = V->user_begin(), E = V->user_end(); UI != E; ) {
Chris Lattnere42eb312004-10-10 23:14:11 +0000676 Instruction *I = cast<Instruction>(*UI++);
677 if (LoadInst *LI = dyn_cast<LoadInst>(I)) {
678 LI->setOperand(0, NewV);
679 Changed = true;
680 } else if (StoreInst *SI = dyn_cast<StoreInst>(I)) {
681 if (SI->getOperand(1) == V) {
682 SI->setOperand(1, NewV);
683 Changed = true;
684 }
685 } else if (isa<CallInst>(I) || isa<InvokeInst>(I)) {
Gabor Greif04397892010-04-06 18:45:08 +0000686 CallSite CS(I);
687 if (CS.getCalledValue() == V) {
Chris Lattnere42eb312004-10-10 23:14:11 +0000688 // Calling through the pointer! Turn into a direct call, but be careful
689 // that the pointer is not also being passed as an argument.
Gabor Greif04397892010-04-06 18:45:08 +0000690 CS.setCalledFunction(NewV);
Chris Lattnere42eb312004-10-10 23:14:11 +0000691 Changed = true;
692 bool PassedAsArg = false;
Gabor Greif04397892010-04-06 18:45:08 +0000693 for (unsigned i = 0, e = CS.arg_size(); i != e; ++i)
694 if (CS.getArgument(i) == V) {
Chris Lattnere42eb312004-10-10 23:14:11 +0000695 PassedAsArg = true;
Gabor Greif04397892010-04-06 18:45:08 +0000696 CS.setArgument(i, NewV);
Chris Lattnere42eb312004-10-10 23:14:11 +0000697 }
698
699 if (PassedAsArg) {
700 // Being passed as an argument also. Be careful to not invalidate UI!
Chandler Carruthcdf47882014-03-09 03:16:01 +0000701 UI = V->user_begin();
Chris Lattnere42eb312004-10-10 23:14:11 +0000702 }
703 }
704 } else if (CastInst *CI = dyn_cast<CastInst>(I)) {
705 Changed |= OptimizeAwayTrappingUsesOfValue(CI,
Owen Anderson487375e2009-07-29 18:55:55 +0000706 ConstantExpr::getCast(CI->getOpcode(),
Chris Lattner46b5c642009-11-06 04:27:31 +0000707 NewV, CI->getType()));
Chris Lattnere42eb312004-10-10 23:14:11 +0000708 if (CI->use_empty()) {
709 Changed = true;
Chris Lattner8e71c6a2004-10-16 18:09:00 +0000710 CI->eraseFromParent();
Chris Lattnere42eb312004-10-10 23:14:11 +0000711 }
712 } else if (GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(I)) {
713 // Should handle GEP here.
Chris Lattnerf96f4a82007-01-31 04:40:53 +0000714 SmallVector<Constant*, 8> Idxs;
715 Idxs.reserve(GEPI->getNumOperands()-1);
Gabor Greif3a9fba52008-05-29 01:59:18 +0000716 for (User::op_iterator i = GEPI->op_begin() + 1, e = GEPI->op_end();
717 i != e; ++i)
718 if (Constant *C = dyn_cast<Constant>(*i))
Chris Lattnerf96f4a82007-01-31 04:40:53 +0000719 Idxs.push_back(C);
Chris Lattnere42eb312004-10-10 23:14:11 +0000720 else
721 break;
Chris Lattnerf96f4a82007-01-31 04:40:53 +0000722 if (Idxs.size() == GEPI->getNumOperands()-1)
Chris Lattnere42eb312004-10-10 23:14:11 +0000723 Changed |= OptimizeAwayTrappingUsesOfValue(GEPI,
Jay Foaded8db7d2011-07-21 14:31:17 +0000724 ConstantExpr::getGetElementPtr(NewV, Idxs));
Chris Lattnere42eb312004-10-10 23:14:11 +0000725 if (GEPI->use_empty()) {
726 Changed = true;
Chris Lattner8e71c6a2004-10-16 18:09:00 +0000727 GEPI->eraseFromParent();
Chris Lattnere42eb312004-10-10 23:14:11 +0000728 }
729 }
730 }
731
732 return Changed;
733}
734
735
736/// OptimizeAwayTrappingUsesOfLoads - The specified global has only one non-null
737/// value stored into it. If there are uses of the loaded value that would trap
738/// if the loaded value is dynamically null, then we know that they cannot be
739/// reachable with a null optimize away the load.
Nick Lewyckycf6aae62012-02-12 01:13:18 +0000740static bool OptimizeAwayTrappingUsesOfLoads(GlobalVariable *GV, Constant *LV,
Rafael Espindolaaeff8a92014-02-24 23:12:18 +0000741 const DataLayout *DL,
Nick Lewyckycf6aae62012-02-12 01:13:18 +0000742 TargetLibraryInfo *TLI) {
Chris Lattnere42eb312004-10-10 23:14:11 +0000743 bool Changed = false;
744
Chris Lattner2538eb62009-01-14 00:12:58 +0000745 // Keep track of whether we are able to remove all the uses of the global
746 // other than the store that defines it.
747 bool AllNonStoreUsesGone = true;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000748
Chris Lattnere42eb312004-10-10 23:14:11 +0000749 // Replace all uses of loads with uses of uses of the stored value.
Chandler Carruthcdf47882014-03-09 03:16:01 +0000750 for (Value::user_iterator GUI = GV->user_begin(), E = GV->user_end(); GUI != E;){
Chris Lattner2538eb62009-01-14 00:12:58 +0000751 User *GlobalUser = *GUI++;
752 if (LoadInst *LI = dyn_cast<LoadInst>(GlobalUser)) {
Chris Lattner46b5c642009-11-06 04:27:31 +0000753 Changed |= OptimizeAwayTrappingUsesOfValue(LI, LV);
Chris Lattner2538eb62009-01-14 00:12:58 +0000754 // If we were able to delete all uses of the loads
755 if (LI->use_empty()) {
756 LI->eraseFromParent();
757 Changed = true;
758 } else {
759 AllNonStoreUsesGone = false;
760 }
761 } else if (isa<StoreInst>(GlobalUser)) {
762 // Ignore the store that stores "LV" to the global.
763 assert(GlobalUser->getOperand(1) == GV &&
764 "Must be storing *to* the global");
Chris Lattnere42eb312004-10-10 23:14:11 +0000765 } else {
Chris Lattner2538eb62009-01-14 00:12:58 +0000766 AllNonStoreUsesGone = false;
767
768 // If we get here we could have other crazy uses that are transitively
769 // loaded.
770 assert((isa<PHINode>(GlobalUser) || isa<SelectInst>(GlobalUser) ||
Benjamin Kramered843602012-09-28 10:01:27 +0000771 isa<ConstantExpr>(GlobalUser) || isa<CmpInst>(GlobalUser) ||
772 isa<BitCastInst>(GlobalUser) ||
773 isa<GetElementPtrInst>(GlobalUser)) &&
Chris Lattner1a1acc22011-05-22 07:15:13 +0000774 "Only expect load and stores!");
Chris Lattnere42eb312004-10-10 23:14:11 +0000775 }
Chris Lattner2538eb62009-01-14 00:12:58 +0000776 }
Chris Lattnere42eb312004-10-10 23:14:11 +0000777
778 if (Changed) {
David Greene44cb8ad2010-01-05 01:28:05 +0000779 DEBUG(dbgs() << "OPTIMIZED LOADS FROM STORED ONCE POINTER: " << *GV);
Chris Lattnere42eb312004-10-10 23:14:11 +0000780 ++NumGlobUses;
781 }
782
Chris Lattnere42eb312004-10-10 23:14:11 +0000783 // If we nuked all of the loads, then none of the stores are needed either,
784 // nor is the global.
Chris Lattner2538eb62009-01-14 00:12:58 +0000785 if (AllNonStoreUsesGone) {
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000786 if (isLeakCheckerRoot(GV)) {
Benjamin Kramer8bcc9712012-08-29 15:32:21 +0000787 Changed |= CleanupPointerRootUsers(GV, TLI);
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000788 } else {
789 Changed = true;
Craig Topperf40110f2014-04-25 05:29:35 +0000790 CleanupConstantGlobalUsers(GV, nullptr, DL, TLI);
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000791 }
Chris Lattnere42eb312004-10-10 23:14:11 +0000792 if (GV->use_empty()) {
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +0000793 DEBUG(dbgs() << " *** GLOBAL NOW DEAD!\n");
794 Changed = true;
Chris Lattner8e71c6a2004-10-16 18:09:00 +0000795 GV->eraseFromParent();
Chris Lattnere42eb312004-10-10 23:14:11 +0000796 ++NumDeleted;
797 }
Chris Lattnere42eb312004-10-10 23:14:11 +0000798 }
799 return Changed;
800}
801
Chris Lattner004e2502004-10-11 05:54:41 +0000802/// ConstantPropUsersOf - Walk the use list of V, constant folding all of the
803/// instructions that are foldable.
Rafael Espindolaaeff8a92014-02-24 23:12:18 +0000804static void ConstantPropUsersOf(Value *V, const DataLayout *DL,
805 TargetLibraryInfo *TLI) {
Chandler Carruthcdf47882014-03-09 03:16:01 +0000806 for (Value::user_iterator UI = V->user_begin(), E = V->user_end(); UI != E; )
Chris Lattner004e2502004-10-11 05:54:41 +0000807 if (Instruction *I = dyn_cast<Instruction>(*UI++))
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000808 if (Constant *NewC = ConstantFoldInstruction(I, DL, TLI)) {
Chris Lattner004e2502004-10-11 05:54:41 +0000809 I->replaceAllUsesWith(NewC);
810
Chris Lattnerd6a44922005-02-01 01:23:31 +0000811 // Advance UI to the next non-I use to avoid invalidating it!
812 // Instructions could multiply use V.
813 while (UI != E && *UI == I)
Chris Lattner004e2502004-10-11 05:54:41 +0000814 ++UI;
Chris Lattnerd6a44922005-02-01 01:23:31 +0000815 I->eraseFromParent();
Chris Lattner004e2502004-10-11 05:54:41 +0000816 }
817}
818
819/// OptimizeGlobalAddressOfMalloc - This function takes the specified global
820/// variable, and transforms the program as if it always contained the result of
821/// the specified malloc. Because it is always the result of the specified
822/// malloc, there is no reason to actually DO the malloc. Instead, turn the
Chris Lattner3ede00b2006-11-30 17:32:29 +0000823/// malloc into a global, and any loads of GV as uses of the new global.
Chris Lattner004e2502004-10-11 05:54:41 +0000824static GlobalVariable *OptimizeGlobalAddressOfMalloc(GlobalVariable *GV,
Victor Hernandez5d034492009-09-18 22:35:49 +0000825 CallInst *CI,
Chris Lattner229907c2011-07-18 04:54:35 +0000826 Type *AllocTy,
Chris Lattner7939f792010-02-25 22:33:52 +0000827 ConstantInt *NElements,
Rafael Espindolaaeff8a92014-02-24 23:12:18 +0000828 const DataLayout *DL,
Nick Lewyckycf6aae62012-02-12 01:13:18 +0000829 TargetLibraryInfo *TLI) {
Chris Lattner7939f792010-02-25 22:33:52 +0000830 DEBUG(errs() << "PROMOTING GLOBAL: " << *GV << " CALL = " << *CI << '\n');
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000831
Chris Lattner229907c2011-07-18 04:54:35 +0000832 Type *GlobalType;
Chris Lattner7939f792010-02-25 22:33:52 +0000833 if (NElements->getZExtValue() == 1)
834 GlobalType = AllocTy;
835 else
836 // If we have an array allocation, the global variable is of an array.
837 GlobalType = ArrayType::get(AllocTy, NElements->getZExtValue());
Victor Hernandez5d034492009-09-18 22:35:49 +0000838
839 // Create the new global variable. The contents of the malloc'd memory is
840 // undefined, so initialize with an undef value.
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000841 GlobalVariable *NewGV = new GlobalVariable(*GV->getParent(),
Chris Lattner65d3a0a2010-02-26 23:42:13 +0000842 GlobalType, false,
Chris Lattner7939f792010-02-25 22:33:52 +0000843 GlobalValue::InternalLinkage,
Chris Lattner65d3a0a2010-02-26 23:42:13 +0000844 UndefValue::get(GlobalType),
Victor Hernandez5d034492009-09-18 22:35:49 +0000845 GV->getName()+".body",
846 GV,
Hans Wennborgcbe34b42012-06-23 11:37:03 +0000847 GV->getThreadLocalMode());
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000848
Chris Lattner7939f792010-02-25 22:33:52 +0000849 // If there are bitcast users of the malloc (which is typical, usually we have
850 // a malloc + bitcast) then replace them with uses of the new global. Update
851 // other users to use the global as well.
Craig Topperf40110f2014-04-25 05:29:35 +0000852 BitCastInst *TheBC = nullptr;
Chris Lattner7939f792010-02-25 22:33:52 +0000853 while (!CI->use_empty()) {
Chandler Carruthcdf47882014-03-09 03:16:01 +0000854 Instruction *User = cast<Instruction>(CI->user_back());
Chris Lattner7939f792010-02-25 22:33:52 +0000855 if (BitCastInst *BCI = dyn_cast<BitCastInst>(User)) {
856 if (BCI->getType() == NewGV->getType()) {
857 BCI->replaceAllUsesWith(NewGV);
858 BCI->eraseFromParent();
859 } else {
860 BCI->setOperand(0, NewGV);
861 }
862 } else {
Craig Topperf40110f2014-04-25 05:29:35 +0000863 if (!TheBC)
Chris Lattner7939f792010-02-25 22:33:52 +0000864 TheBC = new BitCastInst(NewGV, CI->getType(), "newgv", CI);
865 User->replaceUsesOfWith(CI, TheBC);
866 }
867 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000868
Victor Hernandez5d034492009-09-18 22:35:49 +0000869 Constant *RepValue = NewGV;
870 if (NewGV->getType() != GV->getType()->getElementType())
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000871 RepValue = ConstantExpr::getBitCast(RepValue,
Victor Hernandez5d034492009-09-18 22:35:49 +0000872 GV->getType()->getElementType());
873
874 // If there is a comparison against null, we will insert a global bool to
875 // keep track of whether the global was initialized yet or not.
876 GlobalVariable *InitBool =
Chris Lattner46b5c642009-11-06 04:27:31 +0000877 new GlobalVariable(Type::getInt1Ty(GV->getContext()), false,
Victor Hernandez5d034492009-09-18 22:35:49 +0000878 GlobalValue::InternalLinkage,
Chris Lattner46b5c642009-11-06 04:27:31 +0000879 ConstantInt::getFalse(GV->getContext()),
Hans Wennborgcbe34b42012-06-23 11:37:03 +0000880 GV->getName()+".init", GV->getThreadLocalMode());
Victor Hernandez5d034492009-09-18 22:35:49 +0000881 bool InitBoolUsed = false;
882
883 // Loop over all uses of GV, processing them in turn.
Chris Lattner7939f792010-02-25 22:33:52 +0000884 while (!GV->use_empty()) {
Chandler Carruthcdf47882014-03-09 03:16:01 +0000885 if (StoreInst *SI = dyn_cast<StoreInst>(GV->user_back())) {
Victor Hernandez5d034492009-09-18 22:35:49 +0000886 // The global is initialized when the store to it occurs.
Nick Lewycky52da72b2012-02-05 19:56:38 +0000887 new StoreInst(ConstantInt::getTrue(GV->getContext()), InitBool, false, 0,
888 SI->getOrdering(), SI->getSynchScope(), SI);
Victor Hernandez5d034492009-09-18 22:35:49 +0000889 SI->eraseFromParent();
Chris Lattner7939f792010-02-25 22:33:52 +0000890 continue;
Victor Hernandez5d034492009-09-18 22:35:49 +0000891 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000892
Chandler Carruthcdf47882014-03-09 03:16:01 +0000893 LoadInst *LI = cast<LoadInst>(GV->user_back());
Chris Lattner7939f792010-02-25 22:33:52 +0000894 while (!LI->use_empty()) {
Chandler Carruthcdf47882014-03-09 03:16:01 +0000895 Use &LoadUse = *LI->use_begin();
896 ICmpInst *ICI = dyn_cast<ICmpInst>(LoadUse.getUser());
897 if (!ICI) {
Chris Lattner7939f792010-02-25 22:33:52 +0000898 LoadUse = RepValue;
899 continue;
900 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000901
Chris Lattner7939f792010-02-25 22:33:52 +0000902 // Replace the cmp X, 0 with a use of the bool value.
Nick Lewycky52da72b2012-02-05 19:56:38 +0000903 // Sink the load to where the compare was, if atomic rules allow us to.
904 Value *LV = new LoadInst(InitBool, InitBool->getName()+".val", false, 0,
905 LI->getOrdering(), LI->getSynchScope(),
906 LI->isUnordered() ? (Instruction*)ICI : LI);
Chris Lattner7939f792010-02-25 22:33:52 +0000907 InitBoolUsed = true;
908 switch (ICI->getPredicate()) {
909 default: llvm_unreachable("Unknown ICmp Predicate!");
910 case ICmpInst::ICMP_ULT:
911 case ICmpInst::ICMP_SLT: // X < null -> always false
912 LV = ConstantInt::getFalse(GV->getContext());
913 break;
914 case ICmpInst::ICMP_ULE:
915 case ICmpInst::ICMP_SLE:
916 case ICmpInst::ICMP_EQ:
917 LV = BinaryOperator::CreateNot(LV, "notinit", ICI);
918 break;
919 case ICmpInst::ICMP_NE:
920 case ICmpInst::ICMP_UGE:
921 case ICmpInst::ICMP_SGE:
922 case ICmpInst::ICMP_UGT:
923 case ICmpInst::ICMP_SGT:
924 break; // no change.
925 }
926 ICI->replaceAllUsesWith(LV);
927 ICI->eraseFromParent();
928 }
929 LI->eraseFromParent();
930 }
Victor Hernandez5d034492009-09-18 22:35:49 +0000931
932 // If the initialization boolean was used, insert it, otherwise delete it.
933 if (!InitBoolUsed) {
934 while (!InitBool->use_empty()) // Delete initializations
Chandler Carruthcdf47882014-03-09 03:16:01 +0000935 cast<StoreInst>(InitBool->user_back())->eraseFromParent();
Victor Hernandez5d034492009-09-18 22:35:49 +0000936 delete InitBool;
937 } else
938 GV->getParent()->getGlobalList().insert(GV, InitBool);
939
Chris Lattner7939f792010-02-25 22:33:52 +0000940 // Now the GV is dead, nuke it and the malloc..
Victor Hernandez5d034492009-09-18 22:35:49 +0000941 GV->eraseFromParent();
Victor Hernandez5d034492009-09-18 22:35:49 +0000942 CI->eraseFromParent();
943
944 // To further other optimizations, loop over all users of NewGV and try to
945 // constant prop them. This will promote GEP instructions with constant
946 // indices into GEP constant-exprs, which will allow global-opt to hack on it.
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000947 ConstantPropUsersOf(NewGV, DL, TLI);
Victor Hernandez5d034492009-09-18 22:35:49 +0000948 if (RepValue != NewGV)
Rafael Espindola37dc9e12014-02-21 00:06:31 +0000949 ConstantPropUsersOf(RepValue, DL, TLI);
Victor Hernandez5d034492009-09-18 22:35:49 +0000950
951 return NewGV;
952}
953
Chris Lattnerc0677c02004-12-02 07:11:07 +0000954/// ValueIsOnlyUsedLocallyOrStoredToOneGlobal - Scan the use-list of V checking
955/// to make sure that there are no complex uses of V. We permit simple things
956/// like dereferencing the pointer, but not storing through the address, unless
957/// it is to the specified global.
Gabor Greifa21bc0f2010-04-06 18:58:22 +0000958static bool ValueIsOnlyUsedLocallyOrStoredToOneGlobal(const Instruction *V,
959 const GlobalVariable *GV,
Gabor Greif08355d62010-04-06 19:14:05 +0000960 SmallPtrSet<const PHINode*, 8> &PHIs) {
Chandler Carruthcdf47882014-03-09 03:16:01 +0000961 for (const User *U : V->users()) {
962 const Instruction *Inst = cast<Instruction>(U);
Gabor Greif08355d62010-04-06 19:14:05 +0000963
Chris Lattnerf0eb5682008-12-15 21:08:54 +0000964 if (isa<LoadInst>(Inst) || isa<CmpInst>(Inst)) {
965 continue; // Fine, ignore.
966 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000967
Gabor Greifa21bc0f2010-04-06 18:58:22 +0000968 if (const StoreInst *SI = dyn_cast<StoreInst>(Inst)) {
Chris Lattnerc0677c02004-12-02 07:11:07 +0000969 if (SI->getOperand(0) == V && SI->getOperand(1) != GV)
970 return false; // Storing the pointer itself... bad.
Chris Lattnerf0eb5682008-12-15 21:08:54 +0000971 continue; // Otherwise, storing through it, or storing into GV... fine.
972 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000973
Chris Lattnerb9801ff2010-04-10 18:19:22 +0000974 // Must index into the array and into the struct.
975 if (isa<GetElementPtrInst>(Inst) && Inst->getNumOperands() >= 3) {
Chris Lattnerf0eb5682008-12-15 21:08:54 +0000976 if (!ValueIsOnlyUsedLocallyOrStoredToOneGlobal(Inst, GV, PHIs))
Chris Lattnerc0677c02004-12-02 07:11:07 +0000977 return false;
Chris Lattnerf0eb5682008-12-15 21:08:54 +0000978 continue;
979 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000980
Gabor Greifa21bc0f2010-04-06 18:58:22 +0000981 if (const PHINode *PN = dyn_cast<PHINode>(Inst)) {
Chris Lattner6eed0e72007-09-13 16:37:20 +0000982 // PHIs are ok if all uses are ok. Don't infinitely recurse through PHI
983 // cycles.
984 if (PHIs.insert(PN))
Chris Lattner5d13fb532007-09-14 03:41:21 +0000985 if (!ValueIsOnlyUsedLocallyOrStoredToOneGlobal(PN, GV, PHIs))
986 return false;
Chris Lattnerf0eb5682008-12-15 21:08:54 +0000987 continue;
Chris Lattnerc0677c02004-12-02 07:11:07 +0000988 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000989
Gabor Greifa21bc0f2010-04-06 18:58:22 +0000990 if (const BitCastInst *BCI = dyn_cast<BitCastInst>(Inst)) {
Chris Lattnerf0eb5682008-12-15 21:08:54 +0000991 if (!ValueIsOnlyUsedLocallyOrStoredToOneGlobal(BCI, GV, PHIs))
992 return false;
993 continue;
994 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +0000995
Chris Lattnerf0eb5682008-12-15 21:08:54 +0000996 return false;
997 }
Chris Lattnerc0677c02004-12-02 07:11:07 +0000998 return true;
Chris Lattnerc0677c02004-12-02 07:11:07 +0000999}
1000
Chris Lattner24d3d422006-09-30 23:32:09 +00001001/// ReplaceUsesOfMallocWithGlobal - The Alloc pointer is stored into GV
1002/// somewhere. Transform all uses of the allocation into loads from the
1003/// global and uses of the resultant pointer. Further, delete the store into
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001004/// GV. This assumes that these value pass the
Chris Lattner24d3d422006-09-30 23:32:09 +00001005/// 'ValueIsOnlyUsedLocallyOrStoredToOneGlobal' predicate.
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001006static void ReplaceUsesOfMallocWithGlobal(Instruction *Alloc,
Chris Lattner24d3d422006-09-30 23:32:09 +00001007 GlobalVariable *GV) {
1008 while (!Alloc->use_empty()) {
Chandler Carruthcdf47882014-03-09 03:16:01 +00001009 Instruction *U = cast<Instruction>(*Alloc->user_begin());
Chris Lattnerba98f892007-09-13 18:00:31 +00001010 Instruction *InsertPt = U;
Chris Lattner24d3d422006-09-30 23:32:09 +00001011 if (StoreInst *SI = dyn_cast<StoreInst>(U)) {
1012 // If this is the store of the allocation into the global, remove it.
1013 if (SI->getOperand(1) == GV) {
1014 SI->eraseFromParent();
1015 continue;
1016 }
Chris Lattnerba98f892007-09-13 18:00:31 +00001017 } else if (PHINode *PN = dyn_cast<PHINode>(U)) {
1018 // Insert the load in the corresponding predecessor, not right before the
1019 // PHI.
Chandler Carruthcdf47882014-03-09 03:16:01 +00001020 InsertPt = PN->getIncomingBlock(*Alloc->use_begin())->getTerminator();
Chris Lattner49e3bdc2008-12-15 21:44:34 +00001021 } else if (isa<BitCastInst>(U)) {
1022 // Must be bitcast between the malloc and store to initialize the global.
1023 ReplaceUsesOfMallocWithGlobal(U, GV);
1024 U->eraseFromParent();
1025 continue;
1026 } else if (GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(U)) {
1027 // If this is a "GEP bitcast" and the user is a store to the global, then
1028 // just process it as a bitcast.
1029 if (GEPI->hasAllZeroIndices() && GEPI->hasOneUse())
Chandler Carruthcdf47882014-03-09 03:16:01 +00001030 if (StoreInst *SI = dyn_cast<StoreInst>(GEPI->user_back()))
Chris Lattner49e3bdc2008-12-15 21:44:34 +00001031 if (SI->getOperand(1) == GV) {
1032 // Must be bitcast GEP between the malloc and store to initialize
1033 // the global.
1034 ReplaceUsesOfMallocWithGlobal(GEPI, GV);
1035 GEPI->eraseFromParent();
1036 continue;
1037 }
Chris Lattner24d3d422006-09-30 23:32:09 +00001038 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001039
Chris Lattner24d3d422006-09-30 23:32:09 +00001040 // Insert a load from the global, and use it instead of the malloc.
Chris Lattnerba98f892007-09-13 18:00:31 +00001041 Value *NL = new LoadInst(GV, GV->getName()+".val", InsertPt);
Chris Lattner24d3d422006-09-30 23:32:09 +00001042 U->replaceUsesOfWith(Alloc, NL);
1043 }
1044}
1045
Chris Lattner56b55382008-12-16 21:24:51 +00001046/// LoadUsesSimpleEnoughForHeapSRA - Verify that all uses of V (a load, or a phi
1047/// of a load) are simple enough to perform heap SRA on. This permits GEP's
1048/// that index through the array and struct field, icmps of null, and PHIs.
Gabor Greif5d5db532010-04-01 08:21:08 +00001049static bool LoadUsesSimpleEnoughForHeapSRA(const Value *V,
Gabor Greif08d85da2010-04-07 18:59:26 +00001050 SmallPtrSet<const PHINode*, 32> &LoadUsingPHIs,
1051 SmallPtrSet<const PHINode*, 32> &LoadUsingPHIsPerLoad) {
Chris Lattner56b55382008-12-16 21:24:51 +00001052 // We permit two users of the load: setcc comparing against the null
1053 // pointer, and a getelementptr of a specific form.
Chandler Carruthcdf47882014-03-09 03:16:01 +00001054 for (const User *U : V->users()) {
1055 const Instruction *UI = cast<Instruction>(U);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001056
Chris Lattner56b55382008-12-16 21:24:51 +00001057 // Comparison against null is ok.
Chandler Carruthcdf47882014-03-09 03:16:01 +00001058 if (const ICmpInst *ICI = dyn_cast<ICmpInst>(UI)) {
Chris Lattner56b55382008-12-16 21:24:51 +00001059 if (!isa<ConstantPointerNull>(ICI->getOperand(1)))
1060 return false;
1061 continue;
1062 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001063
Chris Lattner56b55382008-12-16 21:24:51 +00001064 // getelementptr is also ok, but only a simple form.
Chandler Carruthcdf47882014-03-09 03:16:01 +00001065 if (const GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(UI)) {
Chris Lattner56b55382008-12-16 21:24:51 +00001066 // Must index into the array and into the struct.
1067 if (GEPI->getNumOperands() < 3)
1068 return false;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001069
Chris Lattner56b55382008-12-16 21:24:51 +00001070 // Otherwise the GEP is ok.
1071 continue;
1072 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001073
Chandler Carruthcdf47882014-03-09 03:16:01 +00001074 if (const PHINode *PN = dyn_cast<PHINode>(UI)) {
Evan Cheng83689442009-06-02 00:56:07 +00001075 if (!LoadUsingPHIsPerLoad.insert(PN))
1076 // This means some phi nodes are dependent on each other.
1077 // Avoid infinite looping!
1078 return false;
1079 if (!LoadUsingPHIs.insert(PN))
1080 // If we have already analyzed this PHI, then it is safe.
Chris Lattner56b55382008-12-16 21:24:51 +00001081 continue;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001082
Chris Lattner222ef4c2008-12-17 05:28:49 +00001083 // Make sure all uses of the PHI are simple enough to transform.
Evan Cheng83689442009-06-02 00:56:07 +00001084 if (!LoadUsesSimpleEnoughForHeapSRA(PN,
1085 LoadUsingPHIs, LoadUsingPHIsPerLoad))
Chris Lattner56b55382008-12-16 21:24:51 +00001086 return false;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001087
Chris Lattner56b55382008-12-16 21:24:51 +00001088 continue;
Chris Lattner24d3d422006-09-30 23:32:09 +00001089 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001090
Chris Lattner56b55382008-12-16 21:24:51 +00001091 // Otherwise we don't know what this is, not ok.
1092 return false;
1093 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001094
Chris Lattner56b55382008-12-16 21:24:51 +00001095 return true;
1096}
1097
1098
1099/// AllGlobalLoadUsesSimpleEnoughForHeapSRA - If all users of values loaded from
1100/// GV are simple enough to perform HeapSRA, return true.
Gabor Greif5d5db532010-04-01 08:21:08 +00001101static bool AllGlobalLoadUsesSimpleEnoughForHeapSRA(const GlobalVariable *GV,
Victor Hernandez5d034492009-09-18 22:35:49 +00001102 Instruction *StoredVal) {
Gabor Greif5d5db532010-04-01 08:21:08 +00001103 SmallPtrSet<const PHINode*, 32> LoadUsingPHIs;
1104 SmallPtrSet<const PHINode*, 32> LoadUsingPHIsPerLoad;
Chandler Carruthcdf47882014-03-09 03:16:01 +00001105 for (const User *U : GV->users())
1106 if (const LoadInst *LI = dyn_cast<LoadInst>(U)) {
Evan Cheng83689442009-06-02 00:56:07 +00001107 if (!LoadUsesSimpleEnoughForHeapSRA(LI, LoadUsingPHIs,
1108 LoadUsingPHIsPerLoad))
Chris Lattner56b55382008-12-16 21:24:51 +00001109 return false;
Evan Cheng83689442009-06-02 00:56:07 +00001110 LoadUsingPHIsPerLoad.clear();
1111 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001112
Chris Lattner222ef4c2008-12-17 05:28:49 +00001113 // If we reach here, we know that all uses of the loads and transitive uses
1114 // (through PHI nodes) are simple enough to transform. However, we don't know
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001115 // that all inputs the to the PHI nodes are in the same equivalence sets.
Chris Lattner222ef4c2008-12-17 05:28:49 +00001116 // Check to verify that all operands of the PHIs are either PHIS that can be
1117 // transformed, loads from GV, or MI itself.
Gabor Greif08d85da2010-04-07 18:59:26 +00001118 for (SmallPtrSet<const PHINode*, 32>::const_iterator I = LoadUsingPHIs.begin()
1119 , E = LoadUsingPHIs.end(); I != E; ++I) {
Gabor Greif5d5db532010-04-01 08:21:08 +00001120 const PHINode *PN = *I;
Chris Lattner222ef4c2008-12-17 05:28:49 +00001121 for (unsigned op = 0, e = PN->getNumIncomingValues(); op != e; ++op) {
1122 Value *InVal = PN->getIncomingValue(op);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001123
Chris Lattner222ef4c2008-12-17 05:28:49 +00001124 // PHI of the stored value itself is ok.
Victor Hernandez5d034492009-09-18 22:35:49 +00001125 if (InVal == StoredVal) continue;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001126
Gabor Greif5d5db532010-04-01 08:21:08 +00001127 if (const PHINode *InPN = dyn_cast<PHINode>(InVal)) {
Chris Lattner222ef4c2008-12-17 05:28:49 +00001128 // One of the PHIs in our set is (optimistically) ok.
1129 if (LoadUsingPHIs.count(InPN))
1130 continue;
1131 return false;
1132 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001133
Chris Lattner222ef4c2008-12-17 05:28:49 +00001134 // Load from GV is ok.
Gabor Greif5d5db532010-04-01 08:21:08 +00001135 if (const LoadInst *LI = dyn_cast<LoadInst>(InVal))
Chris Lattner222ef4c2008-12-17 05:28:49 +00001136 if (LI->getOperand(0) == GV)
1137 continue;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001138
Chris Lattner222ef4c2008-12-17 05:28:49 +00001139 // UNDEF? NULL?
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001140
Chris Lattner222ef4c2008-12-17 05:28:49 +00001141 // Anything else is rejected.
1142 return false;
1143 }
1144 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001145
Chris Lattner24d3d422006-09-30 23:32:09 +00001146 return true;
1147}
1148
Chris Lattner222ef4c2008-12-17 05:28:49 +00001149static Value *GetHeapSROAValue(Value *V, unsigned FieldNo,
1150 DenseMap<Value*, std::vector<Value*> > &InsertedScalarizedValues,
Chris Lattner46b5c642009-11-06 04:27:31 +00001151 std::vector<std::pair<PHINode*, unsigned> > &PHIsToRewrite) {
Chris Lattner222ef4c2008-12-17 05:28:49 +00001152 std::vector<Value*> &FieldVals = InsertedScalarizedValues[V];
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001153
Chris Lattner222ef4c2008-12-17 05:28:49 +00001154 if (FieldNo >= FieldVals.size())
1155 FieldVals.resize(FieldNo+1);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001156
Chris Lattner222ef4c2008-12-17 05:28:49 +00001157 // If we already have this value, just reuse the previously scalarized
1158 // version.
1159 if (Value *FieldVal = FieldVals[FieldNo])
1160 return FieldVal;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001161
Chris Lattner222ef4c2008-12-17 05:28:49 +00001162 // Depending on what instruction this is, we have several cases.
1163 Value *Result;
1164 if (LoadInst *LI = dyn_cast<LoadInst>(V)) {
1165 // This is a scalarized version of the load from the global. Just create
1166 // a new Load of the scalarized global.
1167 Result = new LoadInst(GetHeapSROAValue(LI->getOperand(0), FieldNo,
1168 InsertedScalarizedValues,
Chris Lattner46b5c642009-11-06 04:27:31 +00001169 PHIsToRewrite),
Daniel Dunbar132f7832009-07-30 17:37:43 +00001170 LI->getName()+".f"+Twine(FieldNo), LI);
Chris Lattner222ef4c2008-12-17 05:28:49 +00001171 } else if (PHINode *PN = dyn_cast<PHINode>(V)) {
1172 // PN's type is pointer to struct. Make a new PHI of pointer to struct
1173 // field.
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001174
Matt Arsenaultfcd74012014-04-23 20:36:10 +00001175 PointerType *PTy = cast<PointerType>(PN->getType());
1176 StructType *ST = cast<StructType>(PTy->getElementType());
1177
1178 unsigned AS = PTy->getAddressSpace();
Jay Foade0938d82011-03-30 11:19:20 +00001179 PHINode *NewPN =
Matt Arsenaultfcd74012014-04-23 20:36:10 +00001180 PHINode::Create(PointerType::get(ST->getElementType(FieldNo), AS),
Jay Foad52131342011-03-30 11:28:46 +00001181 PN->getNumIncomingValues(),
Daniel Dunbar132f7832009-07-30 17:37:43 +00001182 PN->getName()+".f"+Twine(FieldNo), PN);
Jay Foade0938d82011-03-30 11:19:20 +00001183 Result = NewPN;
Chris Lattner222ef4c2008-12-17 05:28:49 +00001184 PHIsToRewrite.push_back(std::make_pair(PN, FieldNo));
1185 } else {
Torok Edwinfbcc6632009-07-14 16:55:14 +00001186 llvm_unreachable("Unknown usable value");
Chris Lattner222ef4c2008-12-17 05:28:49 +00001187 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001188
Chris Lattner222ef4c2008-12-17 05:28:49 +00001189 return FieldVals[FieldNo] = Result;
Chris Lattnerba98f892007-09-13 18:00:31 +00001190}
1191
Chris Lattnerf315d4f2007-09-13 17:29:05 +00001192/// RewriteHeapSROALoadUser - Given a load instruction and a value derived from
1193/// the load, rewrite the derived value to use the HeapSRoA'd load.
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001194static void RewriteHeapSROALoadUser(Instruction *LoadUser,
Chris Lattner222ef4c2008-12-17 05:28:49 +00001195 DenseMap<Value*, std::vector<Value*> > &InsertedScalarizedValues,
Chris Lattner46b5c642009-11-06 04:27:31 +00001196 std::vector<std::pair<PHINode*, unsigned> > &PHIsToRewrite) {
Chris Lattnerf315d4f2007-09-13 17:29:05 +00001197 // If this is a comparison against null, handle it.
1198 if (ICmpInst *SCI = dyn_cast<ICmpInst>(LoadUser)) {
1199 assert(isa<ConstantPointerNull>(SCI->getOperand(1)));
1200 // If we have a setcc of the loaded pointer, we can use a setcc of any
1201 // field.
Chris Lattner222ef4c2008-12-17 05:28:49 +00001202 Value *NPtr = GetHeapSROAValue(SCI->getOperand(0), 0,
Chris Lattner46b5c642009-11-06 04:27:31 +00001203 InsertedScalarizedValues, PHIsToRewrite);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001204
Owen Anderson1e5f00e2009-07-09 23:48:35 +00001205 Value *New = new ICmpInst(SCI, SCI->getPredicate(), NPtr,
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001206 Constant::getNullValue(NPtr->getType()),
Owen Anderson1e5f00e2009-07-09 23:48:35 +00001207 SCI->getName());
Chris Lattnerf315d4f2007-09-13 17:29:05 +00001208 SCI->replaceAllUsesWith(New);
1209 SCI->eraseFromParent();
1210 return;
1211 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001212
Chris Lattner222ef4c2008-12-17 05:28:49 +00001213 // Handle 'getelementptr Ptr, Idx, i32 FieldNo ...'
Chris Lattnerba98f892007-09-13 18:00:31 +00001214 if (GetElementPtrInst *GEPI = dyn_cast<GetElementPtrInst>(LoadUser)) {
1215 assert(GEPI->getNumOperands() >= 3 && isa<ConstantInt>(GEPI->getOperand(2))
1216 && "Unexpected GEPI!");
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001217
Chris Lattnerba98f892007-09-13 18:00:31 +00001218 // Load the pointer for this field.
1219 unsigned FieldNo = cast<ConstantInt>(GEPI->getOperand(2))->getZExtValue();
Chris Lattner222ef4c2008-12-17 05:28:49 +00001220 Value *NewPtr = GetHeapSROAValue(GEPI->getOperand(0), FieldNo,
Chris Lattner46b5c642009-11-06 04:27:31 +00001221 InsertedScalarizedValues, PHIsToRewrite);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001222
Chris Lattnerba98f892007-09-13 18:00:31 +00001223 // Create the new GEP idx vector.
1224 SmallVector<Value*, 8> GEPIdx;
1225 GEPIdx.push_back(GEPI->getOperand(1));
1226 GEPIdx.append(GEPI->op_begin()+3, GEPI->op_end());
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001227
Jay Foadd1b78492011-07-25 09:48:08 +00001228 Value *NGEPI = GetElementPtrInst::Create(NewPtr, GEPIdx,
Gabor Greife9ecc682008-04-06 20:25:17 +00001229 GEPI->getName(), GEPI);
Chris Lattnerba98f892007-09-13 18:00:31 +00001230 GEPI->replaceAllUsesWith(NGEPI);
1231 GEPI->eraseFromParent();
1232 return;
1233 }
Chris Lattner011f91b2007-09-13 21:31:36 +00001234
Chris Lattner222ef4c2008-12-17 05:28:49 +00001235 // Recursively transform the users of PHI nodes. This will lazily create the
1236 // PHIs that are needed for individual elements. Keep track of what PHIs we
1237 // see in InsertedScalarizedValues so that we don't get infinite loops (very
1238 // antisocial). If the PHI is already in InsertedScalarizedValues, it has
1239 // already been seen first by another load, so its uses have already been
1240 // processed.
1241 PHINode *PN = cast<PHINode>(LoadUser);
Chris Lattner5cf753c2011-07-21 06:21:31 +00001242 if (!InsertedScalarizedValues.insert(std::make_pair(PN,
1243 std::vector<Value*>())).second)
1244 return;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001245
Chris Lattner222ef4c2008-12-17 05:28:49 +00001246 // If this is the first time we've seen this PHI, recursively process all
1247 // users.
Chandler Carruthcdf47882014-03-09 03:16:01 +00001248 for (auto UI = PN->user_begin(), E = PN->user_end(); UI != E;) {
Chris Lattner0cdf5232008-12-17 05:42:08 +00001249 Instruction *User = cast<Instruction>(*UI++);
Chris Lattner46b5c642009-11-06 04:27:31 +00001250 RewriteHeapSROALoadUser(User, InsertedScalarizedValues, PHIsToRewrite);
Chris Lattner0cdf5232008-12-17 05:42:08 +00001251 }
Chris Lattnerf315d4f2007-09-13 17:29:05 +00001252}
1253
Chris Lattner24d3d422006-09-30 23:32:09 +00001254/// RewriteUsesOfLoadForHeapSRoA - We are performing Heap SRoA on a global. Ptr
1255/// is a value loaded from the global. Eliminate all uses of Ptr, making them
1256/// use FieldGlobals instead. All uses of loaded values satisfy
Chris Lattner56b55382008-12-16 21:24:51 +00001257/// AllGlobalLoadUsesSimpleEnoughForHeapSRA.
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001258static void RewriteUsesOfLoadForHeapSRoA(LoadInst *Load,
Chris Lattner222ef4c2008-12-17 05:28:49 +00001259 DenseMap<Value*, std::vector<Value*> > &InsertedScalarizedValues,
Chris Lattner46b5c642009-11-06 04:27:31 +00001260 std::vector<std::pair<PHINode*, unsigned> > &PHIsToRewrite) {
Chandler Carruthcdf47882014-03-09 03:16:01 +00001261 for (auto UI = Load->user_begin(), E = Load->user_end(); UI != E;) {
Chris Lattner0cdf5232008-12-17 05:42:08 +00001262 Instruction *User = cast<Instruction>(*UI++);
Chris Lattner46b5c642009-11-06 04:27:31 +00001263 RewriteHeapSROALoadUser(User, InsertedScalarizedValues, PHIsToRewrite);
Chris Lattner0cdf5232008-12-17 05:42:08 +00001264 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001265
Chris Lattner222ef4c2008-12-17 05:28:49 +00001266 if (Load->use_empty()) {
1267 Load->eraseFromParent();
1268 InsertedScalarizedValues.erase(Load);
1269 }
Chris Lattner24d3d422006-09-30 23:32:09 +00001270}
1271
Victor Hernandez5d034492009-09-18 22:35:49 +00001272/// PerformHeapAllocSRoA - CI is an allocation of an array of structures. Break
1273/// it up into multiple allocations of arrays of the fields.
Victor Hernandezf3db9152009-11-07 00:16:28 +00001274static GlobalVariable *PerformHeapAllocSRoA(GlobalVariable *GV, CallInst *CI,
Rafael Espindolaaeff8a92014-02-24 23:12:18 +00001275 Value *NElems, const DataLayout *DL,
Benjamin Kramer8bcc9712012-08-29 15:32:21 +00001276 const TargetLibraryInfo *TLI) {
David Greene44cb8ad2010-01-05 01:28:05 +00001277 DEBUG(dbgs() << "SROA HEAP ALLOC: " << *GV << " MALLOC = " << *CI << '\n');
Benjamin Kramer8bcc9712012-08-29 15:32:21 +00001278 Type *MAT = getMallocAllocatedType(CI, TLI);
Chris Lattner229907c2011-07-18 04:54:35 +00001279 StructType *STy = cast<StructType>(MAT);
Victor Hernandez5d034492009-09-18 22:35:49 +00001280
1281 // There is guaranteed to be at least one use of the malloc (storing
1282 // it into GV). If there are other uses, change them to be uses of
1283 // the global to simplify later code. This also deletes the store
1284 // into GV.
Victor Hernandezf3db9152009-11-07 00:16:28 +00001285 ReplaceUsesOfMallocWithGlobal(CI, GV);
1286
Victor Hernandez5d034492009-09-18 22:35:49 +00001287 // Okay, at this point, there are no users of the malloc. Insert N
1288 // new mallocs at the same place as CI, and N globals.
1289 std::vector<Value*> FieldGlobals;
1290 std::vector<Value*> FieldMallocs;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001291
Matt Arsenaultfcd74012014-04-23 20:36:10 +00001292 unsigned AS = GV->getType()->getPointerAddressSpace();
Victor Hernandez5d034492009-09-18 22:35:49 +00001293 for (unsigned FieldNo = 0, e = STy->getNumElements(); FieldNo != e;++FieldNo){
Chris Lattner229907c2011-07-18 04:54:35 +00001294 Type *FieldTy = STy->getElementType(FieldNo);
Matt Arsenaultfcd74012014-04-23 20:36:10 +00001295 PointerType *PFieldTy = PointerType::get(FieldTy, AS);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001296
Victor Hernandez5d034492009-09-18 22:35:49 +00001297 GlobalVariable *NGV =
1298 new GlobalVariable(*GV->getParent(),
1299 PFieldTy, false, GlobalValue::InternalLinkage,
1300 Constant::getNullValue(PFieldTy),
1301 GV->getName() + ".f" + Twine(FieldNo), GV,
Hans Wennborgcbe34b42012-06-23 11:37:03 +00001302 GV->getThreadLocalMode());
Victor Hernandez5d034492009-09-18 22:35:49 +00001303 FieldGlobals.push_back(NGV);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001304
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001305 unsigned TypeSize = DL->getTypeAllocSize(FieldTy);
Chris Lattner229907c2011-07-18 04:54:35 +00001306 if (StructType *ST = dyn_cast<StructType>(FieldTy))
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001307 TypeSize = DL->getStructLayout(ST)->getSizeInBytes();
1308 Type *IntPtrTy = DL->getIntPtrType(CI->getType());
Victor Hernandezf3db9152009-11-07 00:16:28 +00001309 Value *NMI = CallInst::CreateMalloc(CI, IntPtrTy, FieldTy,
1310 ConstantInt::get(IntPtrTy, TypeSize),
Craig Topperf40110f2014-04-25 05:29:35 +00001311 NElems, nullptr,
Victor Hernandezf3db9152009-11-07 00:16:28 +00001312 CI->getName() + ".f" + Twine(FieldNo));
Chris Lattner0521c092010-02-26 18:23:13 +00001313 FieldMallocs.push_back(NMI);
Victor Hernandezf3db9152009-11-07 00:16:28 +00001314 new StoreInst(NMI, NGV, CI);
Victor Hernandez5d034492009-09-18 22:35:49 +00001315 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001316
Victor Hernandez5d034492009-09-18 22:35:49 +00001317 // The tricky aspect of this transformation is handling the case when malloc
1318 // fails. In the original code, malloc failing would set the result pointer
1319 // of malloc to null. In this case, some mallocs could succeed and others
1320 // could fail. As such, we emit code that looks like this:
1321 // F0 = malloc(field0)
1322 // F1 = malloc(field1)
1323 // F2 = malloc(field2)
1324 // if (F0 == 0 || F1 == 0 || F2 == 0) {
1325 // if (F0) { free(F0); F0 = 0; }
1326 // if (F1) { free(F1); F1 = 0; }
1327 // if (F2) { free(F2); F2 = 0; }
1328 // }
Victor Hernandezfcc77b12009-11-10 08:32:25 +00001329 // The malloc can also fail if its argument is too large.
Gabor Greif218f5542010-06-24 14:42:01 +00001330 Constant *ConstantZero = ConstantInt::get(CI->getArgOperand(0)->getType(), 0);
1331 Value *RunningOr = new ICmpInst(CI, ICmpInst::ICMP_SLT, CI->getArgOperand(0),
Victor Hernandezfcc77b12009-11-10 08:32:25 +00001332 ConstantZero, "isneg");
Victor Hernandez5d034492009-09-18 22:35:49 +00001333 for (unsigned i = 0, e = FieldMallocs.size(); i != e; ++i) {
Victor Hernandezf3db9152009-11-07 00:16:28 +00001334 Value *Cond = new ICmpInst(CI, ICmpInst::ICMP_EQ, FieldMallocs[i],
1335 Constant::getNullValue(FieldMallocs[i]->getType()),
1336 "isnull");
Victor Hernandezfcc77b12009-11-10 08:32:25 +00001337 RunningOr = BinaryOperator::CreateOr(RunningOr, Cond, "tmp", CI);
Victor Hernandez5d034492009-09-18 22:35:49 +00001338 }
1339
1340 // Split the basic block at the old malloc.
Victor Hernandezf3db9152009-11-07 00:16:28 +00001341 BasicBlock *OrigBB = CI->getParent();
1342 BasicBlock *ContBB = OrigBB->splitBasicBlock(CI, "malloc_cont");
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001343
Victor Hernandez5d034492009-09-18 22:35:49 +00001344 // Create the block to check the first condition. Put all these blocks at the
1345 // end of the function as they are unlikely to be executed.
Chris Lattner46b5c642009-11-06 04:27:31 +00001346 BasicBlock *NullPtrBlock = BasicBlock::Create(OrigBB->getContext(),
1347 "malloc_ret_null",
Victor Hernandez5d034492009-09-18 22:35:49 +00001348 OrigBB->getParent());
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001349
Victor Hernandez5d034492009-09-18 22:35:49 +00001350 // Remove the uncond branch from OrigBB to ContBB, turning it into a cond
1351 // branch on RunningOr.
1352 OrigBB->getTerminator()->eraseFromParent();
1353 BranchInst::Create(NullPtrBlock, ContBB, RunningOr, OrigBB);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001354
Victor Hernandez5d034492009-09-18 22:35:49 +00001355 // Within the NullPtrBlock, we need to emit a comparison and branch for each
1356 // pointer, because some may be null while others are not.
1357 for (unsigned i = 0, e = FieldGlobals.size(); i != e; ++i) {
1358 Value *GVVal = new LoadInst(FieldGlobals[i], "tmp", NullPtrBlock);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001359 Value *Cmp = new ICmpInst(*NullPtrBlock, ICmpInst::ICMP_NE, GVVal,
Benjamin Kramer547b6c52011-09-27 20:39:19 +00001360 Constant::getNullValue(GVVal->getType()));
Chris Lattner46b5c642009-11-06 04:27:31 +00001361 BasicBlock *FreeBlock = BasicBlock::Create(Cmp->getContext(), "free_it",
Victor Hernandez5d034492009-09-18 22:35:49 +00001362 OrigBB->getParent());
Chris Lattner46b5c642009-11-06 04:27:31 +00001363 BasicBlock *NextBlock = BasicBlock::Create(Cmp->getContext(), "next",
Victor Hernandez5d034492009-09-18 22:35:49 +00001364 OrigBB->getParent());
Victor Hernandeze2971492009-10-24 04:23:03 +00001365 Instruction *BI = BranchInst::Create(FreeBlock, NextBlock,
1366 Cmp, NullPtrBlock);
Victor Hernandez5d034492009-09-18 22:35:49 +00001367
1368 // Fill in FreeBlock.
Victor Hernandeze2971492009-10-24 04:23:03 +00001369 CallInst::CreateFree(GVVal, BI);
Victor Hernandez5d034492009-09-18 22:35:49 +00001370 new StoreInst(Constant::getNullValue(GVVal->getType()), FieldGlobals[i],
1371 FreeBlock);
1372 BranchInst::Create(NextBlock, FreeBlock);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001373
Victor Hernandez5d034492009-09-18 22:35:49 +00001374 NullPtrBlock = NextBlock;
1375 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001376
Victor Hernandez5d034492009-09-18 22:35:49 +00001377 BranchInst::Create(ContBB, NullPtrBlock);
Victor Hernandezf3db9152009-11-07 00:16:28 +00001378
1379 // CI is no longer needed, remove it.
Victor Hernandez5d034492009-09-18 22:35:49 +00001380 CI->eraseFromParent();
1381
1382 /// InsertedScalarizedLoads - As we process loads, if we can't immediately
1383 /// update all uses of the load, keep track of what scalarized loads are
1384 /// inserted for a given load.
1385 DenseMap<Value*, std::vector<Value*> > InsertedScalarizedValues;
1386 InsertedScalarizedValues[GV] = FieldGlobals;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001387
Victor Hernandez5d034492009-09-18 22:35:49 +00001388 std::vector<std::pair<PHINode*, unsigned> > PHIsToRewrite;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001389
Victor Hernandez5d034492009-09-18 22:35:49 +00001390 // Okay, the malloc site is completely handled. All of the uses of GV are now
1391 // loads, and all uses of those loads are simple. Rewrite them to use loads
1392 // of the per-field globals instead.
Chandler Carruthcdf47882014-03-09 03:16:01 +00001393 for (auto UI = GV->user_begin(), E = GV->user_end(); UI != E;) {
Victor Hernandez5d034492009-09-18 22:35:49 +00001394 Instruction *User = cast<Instruction>(*UI++);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001395
Victor Hernandez5d034492009-09-18 22:35:49 +00001396 if (LoadInst *LI = dyn_cast<LoadInst>(User)) {
Chris Lattner46b5c642009-11-06 04:27:31 +00001397 RewriteUsesOfLoadForHeapSRoA(LI, InsertedScalarizedValues, PHIsToRewrite);
Victor Hernandez5d034492009-09-18 22:35:49 +00001398 continue;
1399 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001400
Victor Hernandez5d034492009-09-18 22:35:49 +00001401 // Must be a store of null.
1402 StoreInst *SI = cast<StoreInst>(User);
1403 assert(isa<ConstantPointerNull>(SI->getOperand(0)) &&
1404 "Unexpected heap-sra user!");
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001405
Victor Hernandez5d034492009-09-18 22:35:49 +00001406 // Insert a store of null into each global.
1407 for (unsigned i = 0, e = FieldGlobals.size(); i != e; ++i) {
Chris Lattner229907c2011-07-18 04:54:35 +00001408 PointerType *PT = cast<PointerType>(FieldGlobals[i]->getType());
Victor Hernandez5d034492009-09-18 22:35:49 +00001409 Constant *Null = Constant::getNullValue(PT->getElementType());
1410 new StoreInst(Null, FieldGlobals[i], SI);
1411 }
1412 // Erase the original store.
1413 SI->eraseFromParent();
1414 }
1415
1416 // While we have PHIs that are interesting to rewrite, do it.
1417 while (!PHIsToRewrite.empty()) {
1418 PHINode *PN = PHIsToRewrite.back().first;
1419 unsigned FieldNo = PHIsToRewrite.back().second;
1420 PHIsToRewrite.pop_back();
1421 PHINode *FieldPN = cast<PHINode>(InsertedScalarizedValues[PN][FieldNo]);
1422 assert(FieldPN->getNumIncomingValues() == 0 &&"Already processed this phi");
1423
1424 // Add all the incoming values. This can materialize more phis.
1425 for (unsigned i = 0, e = PN->getNumIncomingValues(); i != e; ++i) {
1426 Value *InVal = PN->getIncomingValue(i);
1427 InVal = GetHeapSROAValue(InVal, FieldNo, InsertedScalarizedValues,
Chris Lattner46b5c642009-11-06 04:27:31 +00001428 PHIsToRewrite);
Victor Hernandez5d034492009-09-18 22:35:49 +00001429 FieldPN->addIncoming(InVal, PN->getIncomingBlock(i));
1430 }
1431 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001432
Victor Hernandez5d034492009-09-18 22:35:49 +00001433 // Drop all inter-phi links and any loads that made it this far.
1434 for (DenseMap<Value*, std::vector<Value*> >::iterator
1435 I = InsertedScalarizedValues.begin(), E = InsertedScalarizedValues.end();
1436 I != E; ++I) {
1437 if (PHINode *PN = dyn_cast<PHINode>(I->first))
1438 PN->dropAllReferences();
1439 else if (LoadInst *LI = dyn_cast<LoadInst>(I->first))
1440 LI->dropAllReferences();
1441 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001442
Victor Hernandez5d034492009-09-18 22:35:49 +00001443 // Delete all the phis and loads now that inter-references are dead.
1444 for (DenseMap<Value*, std::vector<Value*> >::iterator
1445 I = InsertedScalarizedValues.begin(), E = InsertedScalarizedValues.end();
1446 I != E; ++I) {
1447 if (PHINode *PN = dyn_cast<PHINode>(I->first))
1448 PN->eraseFromParent();
1449 else if (LoadInst *LI = dyn_cast<LoadInst>(I->first))
1450 LI->eraseFromParent();
1451 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001452
Victor Hernandez5d034492009-09-18 22:35:49 +00001453 // The old global is now dead, remove it.
1454 GV->eraseFromParent();
1455
1456 ++NumHeapSRA;
1457 return cast<GlobalVariable>(FieldGlobals[0]);
1458}
1459
Chris Lattnerc4274a72008-12-15 21:02:25 +00001460/// TryToOptimizeStoreOfMallocToGlobal - This function is called when we see a
1461/// pointer global variable with a single value stored it that is a malloc or
1462/// cast of malloc.
1463static bool TryToOptimizeStoreOfMallocToGlobal(GlobalVariable *GV,
Victor Hernandez5d034492009-09-18 22:35:49 +00001464 CallInst *CI,
Chris Lattner229907c2011-07-18 04:54:35 +00001465 Type *AllocTy,
Nick Lewycky52da72b2012-02-05 19:56:38 +00001466 AtomicOrdering Ordering,
Victor Hernandez5d034492009-09-18 22:35:49 +00001467 Module::global_iterator &GVI,
Rafael Espindolaaeff8a92014-02-24 23:12:18 +00001468 const DataLayout *DL,
Nick Lewyckycf6aae62012-02-12 01:13:18 +00001469 TargetLibraryInfo *TLI) {
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001470 if (!DL)
Evan Cheng21b588b2010-04-14 20:52:55 +00001471 return false;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001472
Victor Hernandez5d034492009-09-18 22:35:49 +00001473 // If this is a malloc of an abstract type, don't touch it.
1474 if (!AllocTy->isSized())
1475 return false;
1476
1477 // We can't optimize this global unless all uses of it are *known* to be
1478 // of the malloc value, not of the null initializer value (consider a use
1479 // that compares the global's value against zero to see if the malloc has
1480 // been reached). To do this, we check to see if all uses of the global
1481 // would trap if the global were null: this proves that they must all
1482 // happen after the malloc.
1483 if (!AllUsesOfLoadedValueWillTrapIfNull(GV))
1484 return false;
1485
1486 // We can't optimize this if the malloc itself is used in a complex way,
1487 // for example, being stored into multiple globals. This allows the
Nick Lewyckybbd11562012-02-05 19:48:37 +00001488 // malloc to be stored into the specified global, loaded icmp'd, and
Victor Hernandez5d034492009-09-18 22:35:49 +00001489 // GEP'd. These are all things we could transform to using the global
1490 // for.
Evan Cheng21b588b2010-04-14 20:52:55 +00001491 SmallPtrSet<const PHINode*, 8> PHIs;
1492 if (!ValueIsOnlyUsedLocallyOrStoredToOneGlobal(CI, GV, PHIs))
1493 return false;
Victor Hernandez5d034492009-09-18 22:35:49 +00001494
1495 // If we have a global that is only initialized with a fixed size malloc,
1496 // transform the program to use global memory instead of malloc'd memory.
1497 // This eliminates dynamic allocation, avoids an indirection accessing the
1498 // data, and exposes the resultant global to further GlobalOpt.
Victor Hernandez264da322009-10-16 23:12:25 +00001499 // We cannot optimize the malloc if we cannot determine malloc array size.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001500 Value *NElems = getMallocArraySize(CI, DL, TLI, true);
Evan Cheng21b588b2010-04-14 20:52:55 +00001501 if (!NElems)
1502 return false;
Victor Hernandez5d034492009-09-18 22:35:49 +00001503
Evan Cheng21b588b2010-04-14 20:52:55 +00001504 if (ConstantInt *NElements = dyn_cast<ConstantInt>(NElems))
1505 // Restrict this transformation to only working on small allocations
1506 // (2048 bytes currently), as we don't want to introduce a 16M global or
1507 // something.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001508 if (NElements->getZExtValue() * DL->getTypeAllocSize(AllocTy) < 2048) {
1509 GVI = OptimizeGlobalAddressOfMalloc(GV, CI, AllocTy, NElements, DL, TLI);
Evan Cheng21b588b2010-04-14 20:52:55 +00001510 return true;
Victor Hernandez5d034492009-09-18 22:35:49 +00001511 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001512
Evan Cheng21b588b2010-04-14 20:52:55 +00001513 // If the allocation is an array of structures, consider transforming this
1514 // into multiple malloc'd arrays, one for each field. This is basically
1515 // SRoA for malloc'd memory.
1516
Nick Lewycky52da72b2012-02-05 19:56:38 +00001517 if (Ordering != NotAtomic)
1518 return false;
1519
Evan Cheng21b588b2010-04-14 20:52:55 +00001520 // If this is an allocation of a fixed size array of structs, analyze as a
1521 // variable size array. malloc [100 x struct],1 -> malloc struct, 100
Gabor Greif218f5542010-06-24 14:42:01 +00001522 if (NElems == ConstantInt::get(CI->getArgOperand(0)->getType(), 1))
Chris Lattner229907c2011-07-18 04:54:35 +00001523 if (ArrayType *AT = dyn_cast<ArrayType>(AllocTy))
Evan Cheng21b588b2010-04-14 20:52:55 +00001524 AllocTy = AT->getElementType();
Gabor Greif218f5542010-06-24 14:42:01 +00001525
Chris Lattner229907c2011-07-18 04:54:35 +00001526 StructType *AllocSTy = dyn_cast<StructType>(AllocTy);
Evan Cheng21b588b2010-04-14 20:52:55 +00001527 if (!AllocSTy)
1528 return false;
1529
1530 // This the structure has an unreasonable number of fields, leave it
1531 // alone.
1532 if (AllocSTy->getNumElements() <= 16 && AllocSTy->getNumElements() != 0 &&
1533 AllGlobalLoadUsesSimpleEnoughForHeapSRA(GV, CI)) {
1534
1535 // If this is a fixed size array, transform the Malloc to be an alloc of
1536 // structs. malloc [100 x struct],1 -> malloc struct, 100
Benjamin Kramer8bcc9712012-08-29 15:32:21 +00001537 if (ArrayType *AT = dyn_cast<ArrayType>(getMallocAllocatedType(CI, TLI))) {
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001538 Type *IntPtrTy = DL->getIntPtrType(CI->getType());
1539 unsigned TypeSize = DL->getStructLayout(AllocSTy)->getSizeInBytes();
Evan Cheng21b588b2010-04-14 20:52:55 +00001540 Value *AllocSize = ConstantInt::get(IntPtrTy, TypeSize);
1541 Value *NumElements = ConstantInt::get(IntPtrTy, AT->getNumElements());
1542 Instruction *Malloc = CallInst::CreateMalloc(CI, IntPtrTy, AllocSTy,
1543 AllocSize, NumElements,
Craig Topperf40110f2014-04-25 05:29:35 +00001544 nullptr, CI->getName());
Evan Cheng21b588b2010-04-14 20:52:55 +00001545 Instruction *Cast = new BitCastInst(Malloc, CI->getType(), "tmp", CI);
1546 CI->replaceAllUsesWith(Cast);
1547 CI->eraseFromParent();
Nuno Lopes9792d682012-06-22 00:25:01 +00001548 if (BitCastInst *BCI = dyn_cast<BitCastInst>(Malloc))
1549 CI = cast<CallInst>(BCI->getOperand(0));
1550 else
Nuno Lopes0b60ebb2012-06-22 00:29:58 +00001551 CI = cast<CallInst>(Malloc);
Evan Cheng21b588b2010-04-14 20:52:55 +00001552 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001553
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001554 GVI = PerformHeapAllocSRoA(GV, CI, getMallocArraySize(CI, DL, TLI, true),
1555 DL, TLI);
Evan Cheng21b588b2010-04-14 20:52:55 +00001556 return true;
Victor Hernandez5d034492009-09-18 22:35:49 +00001557 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001558
Victor Hernandez5d034492009-09-18 22:35:49 +00001559 return false;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001560}
Victor Hernandez5d034492009-09-18 22:35:49 +00001561
Chris Lattner09a52722004-10-09 21:48:45 +00001562// OptimizeOnceStoredGlobal - Try to optimize globals based on the knowledge
1563// that only one value (besides its initializer) is ever stored to the global.
Chris Lattner004e2502004-10-11 05:54:41 +00001564static bool OptimizeOnceStoredGlobal(GlobalVariable *GV, Value *StoredOnceVal,
Nick Lewycky52da72b2012-02-05 19:56:38 +00001565 AtomicOrdering Ordering,
Chris Lattnerc2d3d312006-08-27 22:42:52 +00001566 Module::global_iterator &GVI,
Rafael Espindolaaeff8a92014-02-24 23:12:18 +00001567 const DataLayout *DL,
1568 TargetLibraryInfo *TLI) {
Chris Lattner1c731fa2008-12-15 21:20:32 +00001569 // Ignore no-op GEPs and bitcasts.
1570 StoredOnceVal = StoredOnceVal->stripPointerCasts();
Chris Lattner09a52722004-10-09 21:48:45 +00001571
Chris Lattnere42eb312004-10-10 23:14:11 +00001572 // If we are dealing with a pointer global that is initialized to null and
1573 // only has one (non-null) value stored into it, then we can optimize any
1574 // users of the loaded value (often calls and loads) that would trap if the
1575 // value was null.
Duncan Sands19d0b472010-02-16 11:11:14 +00001576 if (GV->getInitializer()->getType()->isPointerTy() &&
Chris Lattner09a52722004-10-09 21:48:45 +00001577 GV->getInitializer()->isNullValue()) {
Chris Lattnere42eb312004-10-10 23:14:11 +00001578 if (Constant *SOVC = dyn_cast<Constant>(StoredOnceVal)) {
1579 if (GV->getInitializer()->getType() != SOVC->getType())
Chris Lattner1a1acc22011-05-22 07:15:13 +00001580 SOVC = ConstantExpr::getBitCast(SOVC, GV->getInitializer()->getType());
Misha Brukmanb1c93172005-04-21 23:48:37 +00001581
Chris Lattnere42eb312004-10-10 23:14:11 +00001582 // Optimize away any trapping uses of the loaded value.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001583 if (OptimizeAwayTrappingUsesOfLoads(GV, SOVC, DL, TLI))
Chris Lattner604ed7a2004-10-10 17:07:12 +00001584 return true;
Benjamin Kramer8bcc9712012-08-29 15:32:21 +00001585 } else if (CallInst *CI = extractMallocCall(StoredOnceVal, TLI)) {
1586 Type *MallocType = getMallocAllocatedType(CI, TLI);
Nick Lewyckycf6aae62012-02-12 01:13:18 +00001587 if (MallocType &&
1588 TryToOptimizeStoreOfMallocToGlobal(GV, CI, MallocType, Ordering, GVI,
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001589 DL, TLI))
Victor Hernandezf3db9152009-11-07 00:16:28 +00001590 return true;
Chris Lattnere42eb312004-10-10 23:14:11 +00001591 }
Chris Lattner09a52722004-10-09 21:48:45 +00001592 }
Chris Lattner004e2502004-10-11 05:54:41 +00001593
Chris Lattner09a52722004-10-09 21:48:45 +00001594 return false;
1595}
Chris Lattner1c4bddc2004-10-08 20:59:28 +00001596
Lang Hames459b5dc2014-03-23 04:22:31 +00001597/// TryToShrinkGlobalToBoolean - At this point, we have learned that the only
Chris Lattner20bbac32008-01-14 01:17:44 +00001598/// two values ever stored into GV are its initializer and OtherVal. See if we
Lang Hames459b5dc2014-03-23 04:22:31 +00001599/// can shrink the global into a boolean and select between the two values
1600/// whenever it is used. This exposes the values to other scalar optimizations.
1601static bool TryToShrinkGlobalToBoolean(GlobalVariable *GV, Constant *OtherVal) {
Chris Lattner229907c2011-07-18 04:54:35 +00001602 Type *GVElType = GV->getType()->getElementType();
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001603
Lang Hames459b5dc2014-03-23 04:22:31 +00001604 // If GVElType is already i1, it is already shrunk. If the type of the GV is
1605 // an FP value, pointer or vector, don't do this optimization because a select
1606 // between them is very expensive and unlikely to lead to later
1607 // simplification. In these cases, we typically end up with "cond ? v1 : v2"
1608 // where v1 and v2 both require constant pool loads, a big loss.
Chris Lattner46b5c642009-11-06 04:27:31 +00001609 if (GVElType == Type::getInt1Ty(GV->getContext()) ||
Duncan Sands9dff9be2010-02-15 16:12:20 +00001610 GVElType->isFloatingPointTy() ||
Duncan Sands19d0b472010-02-16 11:11:14 +00001611 GVElType->isPointerTy() || GVElType->isVectorTy())
Chris Lattner20bbac32008-01-14 01:17:44 +00001612 return false;
Gabor Greifa75ed762010-07-12 14:13:15 +00001613
Chris Lattner20bbac32008-01-14 01:17:44 +00001614 // Walk the use list of the global seeing if all the uses are load or store.
1615 // If there is anything else, bail out.
Chandler Carruthcdf47882014-03-09 03:16:01 +00001616 for (User *U : GV->users())
Gabor Greifa75ed762010-07-12 14:13:15 +00001617 if (!isa<LoadInst>(U) && !isa<StoreInst>(U))
Chris Lattner20bbac32008-01-14 01:17:44 +00001618 return false;
Gabor Greifa75ed762010-07-12 14:13:15 +00001619
Lang Hames459b5dc2014-03-23 04:22:31 +00001620 DEBUG(dbgs() << " *** SHRINKING TO BOOL: " << *GV);
1621
1622 // Create the new global, initializing it to false.
1623 GlobalVariable *NewGV = new GlobalVariable(Type::getInt1Ty(GV->getContext()),
1624 false,
1625 GlobalValue::InternalLinkage,
1626 ConstantInt::getFalse(GV->getContext()),
1627 GV->getName()+".b",
1628 GV->getThreadLocalMode(),
1629 GV->getType()->getAddressSpace());
1630 GV->getParent()->getGlobalList().insert(GV, NewGV);
1631
Chris Lattner40e4cec2004-12-12 05:53:50 +00001632 Constant *InitVal = GV->getInitializer();
Chris Lattner46b5c642009-11-06 04:27:31 +00001633 assert(InitVal->getType() != Type::getInt1Ty(GV->getContext()) &&
Lang Hames459b5dc2014-03-23 04:22:31 +00001634 "No reason to shrink to bool!");
Chris Lattner40e4cec2004-12-12 05:53:50 +00001635
Lang Hames459b5dc2014-03-23 04:22:31 +00001636 // If initialized to zero and storing one into the global, we can use a cast
1637 // instead of a select to synthesize the desired value.
1638 bool IsOneZero = false;
1639 if (ConstantInt *CI = dyn_cast<ConstantInt>(OtherVal))
1640 IsOneZero = InitVal->isNullValue() && CI->isOne();
Chris Lattner40e4cec2004-12-12 05:53:50 +00001641
Lang Hames459b5dc2014-03-23 04:22:31 +00001642 while (!GV->use_empty()) {
1643 Instruction *UI = cast<Instruction>(GV->user_back());
1644 if (StoreInst *SI = dyn_cast<StoreInst>(UI)) {
1645 // Change the store into a boolean store.
1646 bool StoringOther = SI->getOperand(0) == OtherVal;
1647 // Only do this if we weren't storing a loaded value.
1648 Value *StoreVal;
1649 if (StoringOther || SI->getOperand(0) == InitVal) {
1650 StoreVal = ConstantInt::get(Type::getInt1Ty(GV->getContext()),
1651 StoringOther);
Bill Wendling7297b862013-02-13 23:00:51 +00001652 } else {
Lang Hames459b5dc2014-03-23 04:22:31 +00001653 // Otherwise, we are storing a previously loaded copy. To do this,
1654 // change the copy from copying the original value to just copying the
1655 // bool.
1656 Instruction *StoredVal = cast<Instruction>(SI->getOperand(0));
1657
1658 // If we've already replaced the input, StoredVal will be a cast or
1659 // select instruction. If not, it will be a load of the original
1660 // global.
1661 if (LoadInst *LI = dyn_cast<LoadInst>(StoredVal)) {
1662 assert(LI->getOperand(0) == GV && "Not a copy!");
1663 // Insert a new load, to preserve the saved value.
1664 StoreVal = new LoadInst(NewGV, LI->getName()+".b", false, 0,
1665 LI->getOrdering(), LI->getSynchScope(), LI);
1666 } else {
1667 assert((isa<CastInst>(StoredVal) || isa<SelectInst>(StoredVal)) &&
1668 "This is not a form that we understand!");
1669 StoreVal = StoredVal->getOperand(0);
1670 assert(isa<LoadInst>(StoreVal) && "Not a load of NewGV!");
1671 }
Chris Lattner745196a2004-12-12 19:34:41 +00001672 }
Lang Hames459b5dc2014-03-23 04:22:31 +00001673 new StoreInst(StoreVal, NewGV, false, 0,
1674 SI->getOrdering(), SI->getSynchScope(), SI);
1675 } else {
1676 // Change the load into a load of bool then a select.
1677 LoadInst *LI = cast<LoadInst>(UI);
1678 LoadInst *NLI = new LoadInst(NewGV, LI->getName()+".b", false, 0,
1679 LI->getOrdering(), LI->getSynchScope(), LI);
1680 Value *NSI;
1681 if (IsOneZero)
1682 NSI = new ZExtInst(NLI, LI->getType(), "", LI);
1683 else
1684 NSI = SelectInst::Create(NLI, OtherVal, InitVal, "", LI);
1685 NSI->takeName(LI);
1686 LI->replaceAllUsesWith(NSI);
Devang Patelfc507a12009-03-06 01:39:36 +00001687 }
Lang Hames459b5dc2014-03-23 04:22:31 +00001688 UI->eraseFromParent();
Chris Lattner40e4cec2004-12-12 05:53:50 +00001689 }
1690
Lang Hames459b5dc2014-03-23 04:22:31 +00001691 // Retain the name of the old global variable. People who are debugging their
1692 // programs may expect these variables to be named the same.
1693 NewGV->takeName(GV);
1694 GV->eraseFromParent();
Chris Lattner20bbac32008-01-14 01:17:44 +00001695 return true;
Chris Lattner40e4cec2004-12-12 05:53:50 +00001696}
1697
1698
Nick Lewycky1480f1d2012-02-12 00:52:26 +00001699/// ProcessGlobal - Analyze the specified global variable and optimize it if
1700/// possible. If we make a change, return true.
Rafael Espindolafc355bc2011-01-19 16:32:21 +00001701bool GlobalOpt::ProcessGlobal(GlobalVariable *GV,
1702 Module::global_iterator &GVI) {
Rafael Espindolafc355bc2011-01-19 16:32:21 +00001703 // Do more involved optimizations if the global is internal.
Chris Lattner1c4bddc2004-10-08 20:59:28 +00001704 GV->removeDeadConstantUsers();
1705
1706 if (GV->use_empty()) {
David Greene44cb8ad2010-01-05 01:28:05 +00001707 DEBUG(dbgs() << "GLOBAL DEAD: " << *GV);
Chris Lattner8e71c6a2004-10-16 18:09:00 +00001708 GV->eraseFromParent();
Chris Lattner1c4bddc2004-10-08 20:59:28 +00001709 ++NumDeleted;
1710 return true;
1711 }
1712
Rafael Espindola1821c6c2012-06-15 18:00:24 +00001713 if (!GV->hasLocalLinkage())
1714 return false;
1715
Rafael Espindolafc355bc2011-01-19 16:32:21 +00001716 GlobalStatus GS;
1717
Rafael Espindola3d7fc252013-10-21 17:14:55 +00001718 if (GlobalStatus::analyzeGlobal(GV, GS))
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001719 return false;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00001720
Rafael Espindola045a78f2013-10-17 18:18:52 +00001721 if (!GS.IsCompared && !GV->hasUnnamedAddr()) {
Rafael Espindolafc355bc2011-01-19 16:32:21 +00001722 GV->setUnnamedAddr(true);
1723 NumUnnamed++;
1724 }
1725
1726 if (GV->isConstant() || !GV->hasInitializer())
1727 return false;
1728
Rafael Espindolad21ac192013-09-05 19:15:21 +00001729 return ProcessInternalGlobal(GV, GVI, GS);
Rafael Espindolafc355bc2011-01-19 16:32:21 +00001730}
1731
1732/// ProcessInternalGlobal - Analyze the specified global variable and optimize
1733/// it if possible. If we make a change, return true.
1734bool GlobalOpt::ProcessInternalGlobal(GlobalVariable *GV,
1735 Module::global_iterator &GVI,
Rafael Espindolafc355bc2011-01-19 16:32:21 +00001736 const GlobalStatus &GS) {
Alexey Samsonova1944e62013-10-07 19:03:24 +00001737 // If this is a first class global and has only one accessing function
1738 // and this function is main (which we know is not recursive), we replace
1739 // the global with a local alloca in this function.
1740 //
Alp Tokerf907b892013-12-05 05:44:44 +00001741 // NOTE: It doesn't make sense to promote non-single-value types since we
Alexey Samsonova1944e62013-10-07 19:03:24 +00001742 // are just replacing static memory to stack memory.
1743 //
1744 // If the global is in different address space, don't bring it to stack.
1745 if (!GS.HasMultipleAccessingFunctions &&
1746 GS.AccessingFunction && !GS.HasNonInstructionUser &&
1747 GV->getType()->getElementType()->isSingleValueType() &&
1748 GS.AccessingFunction->getName() == "main" &&
1749 GS.AccessingFunction->hasExternalLinkage() &&
1750 GV->getType()->getAddressSpace() == 0) {
1751 DEBUG(dbgs() << "LOCALIZING GLOBAL: " << *GV);
1752 Instruction &FirstI = const_cast<Instruction&>(*GS.AccessingFunction
1753 ->getEntryBlock().begin());
1754 Type *ElemTy = GV->getType()->getElementType();
1755 // FIXME: Pass Global's alignment when globals have alignment
Craig Topperf40110f2014-04-25 05:29:35 +00001756 AllocaInst *Alloca = new AllocaInst(ElemTy, nullptr,
1757 GV->getName(), &FirstI);
Alexey Samsonova1944e62013-10-07 19:03:24 +00001758 if (!isa<UndefValue>(GV->getInitializer()))
1759 new StoreInst(GV->getInitializer(), Alloca, &FirstI);
1760
1761 GV->replaceAllUsesWith(Alloca);
1762 GV->eraseFromParent();
1763 ++NumLocalized;
1764 return true;
1765 }
1766
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001767 // If the global is never loaded (but may be stored to), it is dead.
1768 // Delete it now.
Rafael Espindola045a78f2013-10-17 18:18:52 +00001769 if (!GS.IsLoaded) {
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001770 DEBUG(dbgs() << "GLOBAL NEVER LOADED: " << *GV);
1771
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +00001772 bool Changed;
1773 if (isLeakCheckerRoot(GV)) {
1774 // Delete any constant stores to the global.
Benjamin Kramer8bcc9712012-08-29 15:32:21 +00001775 Changed = CleanupPointerRootUsers(GV, TLI);
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +00001776 } else {
1777 // Delete any stores we can find to the global. We may not be able to
1778 // make it completely dead though.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001779 Changed = CleanupConstantGlobalUsers(GV, GV->getInitializer(), DL, TLI);
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +00001780 }
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001781
1782 // If the global is dead now, delete it.
1783 if (GV->use_empty()) {
1784 GV->eraseFromParent();
1785 ++NumDeleted;
1786 Changed = true;
1787 }
1788 return Changed;
1789
Rafael Espindola045a78f2013-10-17 18:18:52 +00001790 } else if (GS.StoredType <= GlobalStatus::InitializerStored) {
Michael Gottesmand1a46f22013-01-11 20:07:53 +00001791 DEBUG(dbgs() << "MARKING CONSTANT: " << *GV << "\n");
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001792 GV->setConstant(true);
1793
1794 // Clean up any obviously simplifiable users now.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001795 CleanupConstantGlobalUsers(GV, GV->getInitializer(), DL, TLI);
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001796
1797 // If the global is dead now, just nuke it.
1798 if (GV->use_empty()) {
1799 DEBUG(dbgs() << " *** Marking constant allowed us to simplify "
1800 << "all users and delete global!\n");
1801 GV->eraseFromParent();
1802 ++NumDeleted;
1803 }
1804
1805 ++NumMarked;
1806 return true;
1807 } else if (!GV->getInitializer()->getType()->isSingleValueType()) {
Rafael Espindola93512512014-02-25 17:30:31 +00001808 if (DataLayoutPass *DLP = getAnalysisIfAvailable<DataLayoutPass>()) {
1809 const DataLayout &DL = DLP->getDataLayout();
1810 if (GlobalVariable *FirstNewGV = SRAGlobal(GV, DL)) {
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001811 GVI = FirstNewGV; // Don't skip the newly produced globals!
1812 return true;
1813 }
Rafael Espindola93512512014-02-25 17:30:31 +00001814 }
Rafael Espindola045a78f2013-10-17 18:18:52 +00001815 } else if (GS.StoredType == GlobalStatus::StoredOnce) {
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001816 // If the initial value for the global was an undef value, and if only
1817 // one other value was stored into it, we can just change the
1818 // initializer to be the stored value, then delete all stores to the
1819 // global. This allows us to mark it constant.
1820 if (Constant *SOVConstant = dyn_cast<Constant>(GS.StoredOnceValue))
1821 if (isa<UndefValue>(GV->getInitializer())) {
1822 // Change the initial value here.
1823 GV->setInitializer(SOVConstant);
1824
1825 // Clean up any obviously simplifiable users now.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001826 CleanupConstantGlobalUsers(GV, GV->getInitializer(), DL, TLI);
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001827
1828 if (GV->use_empty()) {
1829 DEBUG(dbgs() << " *** Substituting initializer allowed us to "
Nick Lewyckyfaa9c3b02012-07-24 07:21:08 +00001830 << "simplify all users and delete global!\n");
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001831 GV->eraseFromParent();
1832 ++NumDeleted;
1833 } else {
1834 GVI = GV;
1835 }
1836 ++NumSubstitute;
1837 return true;
1838 }
1839
1840 // Try to optimize globals based on the knowledge that only one value
1841 // (besides its initializer) is ever stored to the global.
Nick Lewycky52da72b2012-02-05 19:56:38 +00001842 if (OptimizeOnceStoredGlobal(GV, GS.StoredOnceValue, GS.Ordering, GVI,
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001843 DL, TLI))
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001844 return true;
1845
Lang Hames459b5dc2014-03-23 04:22:31 +00001846 // Otherwise, if the global was not a boolean, we can shrink it to be a
1847 // boolean.
Eli Friedman33d37002013-09-09 22:00:13 +00001848 if (Constant *SOVConstant = dyn_cast<Constant>(GS.StoredOnceValue)) {
1849 if (GS.Ordering == NotAtomic) {
Lang Hames459b5dc2014-03-23 04:22:31 +00001850 if (TryToShrinkGlobalToBoolean(GV, SOVConstant)) {
Eli Friedman33d37002013-09-09 22:00:13 +00001851 ++NumShrunkToBool;
1852 return true;
1853 }
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001854 }
Eli Friedman33d37002013-09-09 22:00:13 +00001855 }
Rafael Espindolaecd5b9a2011-01-18 04:36:06 +00001856 }
1857
Chris Lattner1c4bddc2004-10-08 20:59:28 +00001858 return false;
1859}
1860
Chris Lattnera4c80222005-05-08 22:18:06 +00001861/// ChangeCalleesToFastCall - Walk all of the direct calls of the specified
1862/// function, changing them to FastCC.
1863static void ChangeCalleesToFastCall(Function *F) {
Chandler Carruthcdf47882014-03-09 03:16:01 +00001864 for (User *U : F->users()) {
1865 if (isa<BlockAddress>(U))
Jay Foadca0c4992012-05-12 08:30:16 +00001866 continue;
Chandler Carruthcdf47882014-03-09 03:16:01 +00001867 CallSite CS(cast<Instruction>(U));
1868 CS.setCallingConv(CallingConv::Fast);
Chris Lattnera4c80222005-05-08 22:18:06 +00001869 }
1870}
Chris Lattner1c4bddc2004-10-08 20:59:28 +00001871
Bill Wendlinge94d8432012-12-07 23:16:57 +00001872static AttributeSet StripNest(LLVMContext &C, const AttributeSet &Attrs) {
Chris Lattner8a923e72008-03-12 17:45:29 +00001873 for (unsigned i = 0, e = Attrs.getNumSlots(); i != e; ++i) {
Bill Wendling57625a42013-01-25 23:09:36 +00001874 unsigned Index = Attrs.getSlotIndex(i);
1875 if (!Attrs.getSlotAttributes(i).hasAttribute(Index, Attribute::Nest))
Duncan Sands85fab3a2008-02-18 17:32:13 +00001876 continue;
1877
Duncan Sands85fab3a2008-02-18 17:32:13 +00001878 // There can be only one.
Bill Wendling57625a42013-01-25 23:09:36 +00001879 return Attrs.removeAttribute(C, Index, Attribute::Nest);
Duncan Sands573b3f82008-02-16 20:56:04 +00001880 }
1881
1882 return Attrs;
1883}
1884
1885static void RemoveNestAttribute(Function *F) {
Bill Wendling85a64c22012-10-14 06:39:53 +00001886 F->setAttributes(StripNest(F->getContext(), F->getAttributes()));
Chandler Carruthcdf47882014-03-09 03:16:01 +00001887 for (User *U : F->users()) {
1888 if (isa<BlockAddress>(U))
Jay Foadca0c4992012-05-12 08:30:16 +00001889 continue;
Chandler Carruthcdf47882014-03-09 03:16:01 +00001890 CallSite CS(cast<Instruction>(U));
1891 CS.setAttributes(StripNest(F->getContext(), CS.getAttributes()));
Duncan Sands573b3f82008-02-16 20:56:04 +00001892 }
1893}
1894
Reid Kleckner22869372014-02-26 19:57:30 +00001895/// Return true if this is a calling convention that we'd like to change. The
1896/// idea here is that we don't want to mess with the convention if the user
1897/// explicitly requested something with performance implications like coldcc,
1898/// GHC, or anyregcc.
1899static bool isProfitableToMakeFastCC(Function *F) {
1900 CallingConv::ID CC = F->getCallingConv();
1901 // FIXME: Is it worth transforming x86_stdcallcc and x86_fastcallcc?
1902 return CC == CallingConv::C || CC == CallingConv::X86_ThisCall;
1903}
1904
Chris Lattner41b6a5a2005-09-26 01:43:45 +00001905bool GlobalOpt::OptimizeFunctions(Module &M) {
1906 bool Changed = false;
1907 // Optimize functions.
1908 for (Module::iterator FI = M.begin(), E = M.end(); FI != E; ) {
1909 Function *F = FI++;
Duncan Sandsed722832009-03-06 10:21:56 +00001910 // Functions without names cannot be referenced outside this module.
David Majnemer5c921152014-07-01 15:26:50 +00001911 if (!F->hasName() && !F->isDeclaration() && !F->hasLocalLinkage())
Duncan Sandsed722832009-03-06 10:21:56 +00001912 F->setLinkage(GlobalValue::InternalLinkage);
Chris Lattner41b6a5a2005-09-26 01:43:45 +00001913 F->removeDeadConstantUsers();
Eli Friedman1923a332011-10-20 05:23:42 +00001914 if (F->isDefTriviallyDead()) {
Chris Lattnerb5d9c8c2009-11-01 19:03:42 +00001915 F->eraseFromParent();
Chris Lattner41b6a5a2005-09-26 01:43:45 +00001916 Changed = true;
1917 ++NumFnDeleted;
Rafael Espindola6de96a12009-01-15 20:18:42 +00001918 } else if (F->hasLocalLinkage()) {
Reid Klecknere6ff5c52014-02-28 22:50:08 +00001919 if (isProfitableToMakeFastCC(F) && !F->isVarArg() &&
1920 !F->hasAddressTaken()) {
Reid Kleckner22869372014-02-26 19:57:30 +00001921 // If this function has a calling convention worth changing, is not a
1922 // varargs function, and is only called directly, promote it to use the
1923 // Fast calling convention.
Duncan Sands573b3f82008-02-16 20:56:04 +00001924 F->setCallingConv(CallingConv::Fast);
1925 ChangeCalleesToFastCall(F);
1926 ++NumFastCallFns;
1927 Changed = true;
1928 }
1929
Bill Wendling3d7b0b82012-12-19 07:18:57 +00001930 if (F->getAttributes().hasAttrSomewhere(Attribute::Nest) &&
Jay Foad557169d2009-06-10 08:41:11 +00001931 !F->hasAddressTaken()) {
Duncan Sands573b3f82008-02-16 20:56:04 +00001932 // The function is not used by a trampoline intrinsic, so it is safe
1933 // to remove the 'nest' attribute.
1934 RemoveNestAttribute(F);
1935 ++NumNestRemoved;
1936 Changed = true;
1937 }
Chris Lattner41b6a5a2005-09-26 01:43:45 +00001938 }
1939 }
1940 return Changed;
1941}
1942
1943bool GlobalOpt::OptimizeGlobalVars(Module &M) {
1944 bool Changed = false;
David Majnemerdad0a642014-06-27 18:19:56 +00001945
1946 SmallSet<const Comdat *, 8> NotDiscardableComdats;
1947 for (const GlobalVariable &GV : M.globals())
1948 if (const Comdat *C = GV.getComdat())
1949 if (!GV.isDiscardableIfUnused())
1950 NotDiscardableComdats.insert(C);
1951
Chris Lattner41b6a5a2005-09-26 01:43:45 +00001952 for (Module::global_iterator GVI = M.global_begin(), E = M.global_end();
1953 GVI != E; ) {
1954 GlobalVariable *GV = GVI++;
Duncan Sandsed722832009-03-06 10:21:56 +00001955 // Global variables without names cannot be referenced outside this module.
David Majnemer5c921152014-07-01 15:26:50 +00001956 if (!GV->hasName() && !GV->isDeclaration() && !GV->hasLocalLinkage())
Duncan Sandsed722832009-03-06 10:21:56 +00001957 GV->setLinkage(GlobalValue::InternalLinkage);
Dan Gohman580b80d2009-11-23 16:22:21 +00001958 // Simplify the initializer.
1959 if (GV->hasInitializer())
1960 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(GV->getInitializer())) {
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001961 Constant *New = ConstantFoldConstantExpression(CE, DL, TLI);
Dan Gohman580b80d2009-11-23 16:22:21 +00001962 if (New && New != CE)
1963 GV->setInitializer(New);
1964 }
Rafael Espindolafc355bc2011-01-19 16:32:21 +00001965
David Majnemerdad0a642014-06-27 18:19:56 +00001966 if (GV->isDiscardableIfUnused()) {
1967 if (const Comdat *C = GV->getComdat())
1968 if (NotDiscardableComdats.count(C))
1969 continue;
1970 Changed |= ProcessGlobal(GV, GVI);
1971 }
Chris Lattner41b6a5a2005-09-26 01:43:45 +00001972 }
1973 return Changed;
1974}
1975
Jakub Staszak9525a772012-12-06 21:57:16 +00001976static inline bool
Chris Lattner0d71c4f2010-12-07 04:33:29 +00001977isSimpleEnoughValueToCommit(Constant *C,
Eli Friedman55fa49f32012-01-05 23:03:32 +00001978 SmallPtrSet<Constant*, 8> &SimpleConstants,
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001979 const DataLayout *DL);
Chris Lattner0d71c4f2010-12-07 04:33:29 +00001980
1981
1982/// isSimpleEnoughValueToCommit - Return true if the specified constant can be
1983/// handled by the code generator. We don't want to generate something like:
1984/// void *X = &X/42;
1985/// because the code generator doesn't have a relocation that can handle that.
1986///
1987/// This function should be called if C was not found (but just got inserted)
1988/// in SimpleConstants to avoid having to rescan the same constants all the
1989/// time.
1990static bool isSimpleEnoughValueToCommitHelper(Constant *C,
Eli Friedman55fa49f32012-01-05 23:03:32 +00001991 SmallPtrSet<Constant*, 8> &SimpleConstants,
Rafael Espindola37dc9e12014-02-21 00:06:31 +00001992 const DataLayout *DL) {
David Majnemer6098b2f2014-06-26 03:02:19 +00001993 // Simple global addresses are supported, do not allow dllimport or
1994 // thread-local globals.
David Majnemer23fc9af2014-06-24 06:53:45 +00001995 if (auto *GV = dyn_cast<GlobalValue>(C))
David Majnemer6098b2f2014-06-26 03:02:19 +00001996 return !GV->hasDLLImportStorageClass() && !GV->isThreadLocal();
David Majnemer23fc9af2014-06-24 06:53:45 +00001997
1998 // Simple integer, undef, constant aggregate zero, etc are all supported.
1999 if (C->getNumOperands() == 0 || isa<BlockAddress>(C))
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002000 return true;
Jakub Staszak9525a772012-12-06 21:57:16 +00002001
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002002 // Aggregate values are safe if all their elements are.
2003 if (isa<ConstantArray>(C) || isa<ConstantStruct>(C) ||
2004 isa<ConstantVector>(C)) {
2005 for (unsigned i = 0, e = C->getNumOperands(); i != e; ++i) {
2006 Constant *Op = cast<Constant>(C->getOperand(i));
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002007 if (!isSimpleEnoughValueToCommit(Op, SimpleConstants, DL))
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002008 return false;
2009 }
2010 return true;
2011 }
Jakub Staszak9525a772012-12-06 21:57:16 +00002012
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002013 // We don't know exactly what relocations are allowed in constant expressions,
2014 // so we allow &global+constantoffset, which is safe and uniformly supported
2015 // across targets.
2016 ConstantExpr *CE = cast<ConstantExpr>(C);
2017 switch (CE->getOpcode()) {
2018 case Instruction::BitCast:
Eli Friedman55fa49f32012-01-05 23:03:32 +00002019 // Bitcast is fine if the casted value is fine.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002020 return isSimpleEnoughValueToCommit(CE->getOperand(0), SimpleConstants, DL);
Eli Friedman55fa49f32012-01-05 23:03:32 +00002021
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002022 case Instruction::IntToPtr:
2023 case Instruction::PtrToInt:
Eli Friedman55fa49f32012-01-05 23:03:32 +00002024 // int <=> ptr is fine if the int type is the same size as the
2025 // pointer type.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002026 if (!DL || DL->getTypeSizeInBits(CE->getType()) !=
2027 DL->getTypeSizeInBits(CE->getOperand(0)->getType()))
Eli Friedman55fa49f32012-01-05 23:03:32 +00002028 return false;
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002029 return isSimpleEnoughValueToCommit(CE->getOperand(0), SimpleConstants, DL);
Jakub Staszak9525a772012-12-06 21:57:16 +00002030
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002031 // GEP is fine if it is simple + constant offset.
2032 case Instruction::GetElementPtr:
2033 for (unsigned i = 1, e = CE->getNumOperands(); i != e; ++i)
2034 if (!isa<ConstantInt>(CE->getOperand(i)))
2035 return false;
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002036 return isSimpleEnoughValueToCommit(CE->getOperand(0), SimpleConstants, DL);
Jakub Staszak9525a772012-12-06 21:57:16 +00002037
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002038 case Instruction::Add:
2039 // We allow simple+cst.
2040 if (!isa<ConstantInt>(CE->getOperand(1)))
2041 return false;
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002042 return isSimpleEnoughValueToCommit(CE->getOperand(0), SimpleConstants, DL);
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002043 }
2044 return false;
2045}
2046
Jakub Staszak9525a772012-12-06 21:57:16 +00002047static inline bool
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002048isSimpleEnoughValueToCommit(Constant *C,
Eli Friedman55fa49f32012-01-05 23:03:32 +00002049 SmallPtrSet<Constant*, 8> &SimpleConstants,
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002050 const DataLayout *DL) {
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002051 // If we already checked this constant, we win.
2052 if (!SimpleConstants.insert(C)) return true;
2053 // Check the constant.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002054 return isSimpleEnoughValueToCommitHelper(C, SimpleConstants, DL);
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002055}
2056
2057
Chris Lattner99e23fa2005-09-26 04:44:35 +00002058/// isSimpleEnoughPointerToCommit - Return true if this constant is simple
Owen Anderson9eb7cb482011-01-14 22:19:20 +00002059/// enough for us to understand. In particular, if it is a cast to anything
2060/// other than from one pointer type to another pointer type, we punt.
2061/// We basically just support direct accesses to globals and GEP's of
Chris Lattner99e23fa2005-09-26 04:44:35 +00002062/// globals. This should be kept up to date with CommitValueTo.
Chris Lattner46b5c642009-11-06 04:27:31 +00002063static bool isSimpleEnoughPointerToCommit(Constant *C) {
Dan Gohman82e74752009-09-07 22:42:05 +00002064 // Conservatively, avoid aggregate types. This is because we don't
2065 // want to worry about them partially overlapping other stores.
2066 if (!cast<PointerType>(C->getType())->getElementType()->isSingleValueType())
2067 return false;
2068
Dan Gohmanf7f3fb12009-09-07 22:31:26 +00002069 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(C))
David Majnemer23fc9af2014-06-24 06:53:45 +00002070 // Do not allow weak/*_odr/linkonce linkage or external globals.
Mikhail Glushenkov2072db22010-10-19 16:47:23 +00002071 return GV->hasUniqueInitializer();
Dan Gohmanf7f3fb12009-09-07 22:31:26 +00002072
Owen Anderson3e2f6cf2011-01-14 22:31:13 +00002073 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(C)) {
Chris Lattner46af55e2005-09-26 06:52:44 +00002074 // Handle a constantexpr gep.
2075 if (CE->getOpcode() == Instruction::GetElementPtr &&
Dan Gohmanbeee35a2009-09-07 22:40:13 +00002076 isa<GlobalVariable>(CE->getOperand(0)) &&
2077 cast<GEPOperator>(CE)->isInBounds()) {
Chris Lattner46af55e2005-09-26 06:52:44 +00002078 GlobalVariable *GV = cast<GlobalVariable>(CE->getOperand(0));
Mikhail Glushenkov2072db22010-10-19 16:47:23 +00002079 // Do not allow weak/*_odr/linkonce/dllimport/dllexport linkage or
Dan Gohmanf7f3fb12009-09-07 22:31:26 +00002080 // external globals.
Mikhail Glushenkov2072db22010-10-19 16:47:23 +00002081 if (!GV->hasUniqueInitializer())
Dan Gohmanf7f3fb12009-09-07 22:31:26 +00002082 return false;
Dan Gohman161429f2009-09-07 22:44:55 +00002083
Dan Gohman161429f2009-09-07 22:44:55 +00002084 // The first index must be zero.
Benjamin Kramerb6d0bd42014-03-02 12:27:27 +00002085 ConstantInt *CI = dyn_cast<ConstantInt>(*std::next(CE->op_begin()));
Dan Gohman161429f2009-09-07 22:44:55 +00002086 if (!CI || !CI->isZero()) return false;
Dan Gohman161429f2009-09-07 22:44:55 +00002087
2088 // The remaining indices must be compile-time known integers within the
Dan Gohman7190d482009-09-10 23:37:55 +00002089 // notional bounds of the corresponding static array types.
2090 if (!CE->isGEPWithNoNotionalOverIndexing())
2091 return false;
Dan Gohman161429f2009-09-07 22:44:55 +00002092
Dan Gohmane525d9d2009-10-05 16:36:26 +00002093 return ConstantFoldLoadThroughGEPConstantExpr(GV->getInitializer(), CE);
Jakub Staszak9525a772012-12-06 21:57:16 +00002094
Owen Anderson9eb7cb482011-01-14 22:19:20 +00002095 // A constantexpr bitcast from a pointer to another pointer is a no-op,
2096 // and we know how to evaluate it by moving the bitcast from the pointer
2097 // operand to the value operand.
2098 } else if (CE->getOpcode() == Instruction::BitCast &&
Chris Lattner8b4952f2011-01-16 02:05:10 +00002099 isa<GlobalVariable>(CE->getOperand(0))) {
Owen Anderson9eb7cb482011-01-14 22:19:20 +00002100 // Do not allow weak/*_odr/linkonce/dllimport/dllexport linkage or
2101 // external globals.
Chris Lattner8b4952f2011-01-16 02:05:10 +00002102 return cast<GlobalVariable>(CE->getOperand(0))->hasUniqueInitializer();
Chris Lattner46af55e2005-09-26 06:52:44 +00002103 }
Owen Anderson3e2f6cf2011-01-14 22:31:13 +00002104 }
Jakub Staszak9525a772012-12-06 21:57:16 +00002105
Chris Lattner99e23fa2005-09-26 04:44:35 +00002106 return false;
2107}
2108
Chris Lattner46af55e2005-09-26 06:52:44 +00002109/// EvaluateStoreInto - Evaluate a piece of a constantexpr store into a global
2110/// initializer. This returns 'Init' modified to reflect 'Val' stored into it.
2111/// At this point, the GEP operands of Addr [0, OpNo) have been stepped into.
2112static Constant *EvaluateStoreInto(Constant *Init, Constant *Val,
Zhou Sheng8e6d64a2012-12-01 10:54:28 +00002113 ConstantExpr *Addr, unsigned OpNo) {
Chris Lattner46af55e2005-09-26 06:52:44 +00002114 // Base case of the recursion.
2115 if (OpNo == Addr->getNumOperands()) {
2116 assert(Val->getType() == Init->getType() && "Type mismatch!");
2117 return Val;
2118 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002119
Chris Lattner0256be92012-01-27 03:08:05 +00002120 SmallVector<Constant*, 32> Elts;
Chris Lattner229907c2011-07-18 04:54:35 +00002121 if (StructType *STy = dyn_cast<StructType>(Init->getType())) {
Chris Lattner46af55e2005-09-26 06:52:44 +00002122 // Break up the constant into its elements.
Chris Lattnerfa775002012-01-26 02:32:04 +00002123 for (unsigned i = 0, e = STy->getNumElements(); i != e; ++i)
2124 Elts.push_back(Init->getAggregateElement(i));
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002125
Chris Lattner46af55e2005-09-26 06:52:44 +00002126 // Replace the element that we are supposed to.
Reid Spencere0fc4df2006-10-20 07:07:24 +00002127 ConstantInt *CU = cast<ConstantInt>(Addr->getOperand(OpNo));
2128 unsigned Idx = CU->getZExtValue();
2129 assert(Idx < STy->getNumElements() && "Struct index out of range!");
Zhou Sheng8e6d64a2012-12-01 10:54:28 +00002130 Elts[Idx] = EvaluateStoreInto(Elts[Idx], Val, Addr, OpNo+1);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002131
Chris Lattner46af55e2005-09-26 06:52:44 +00002132 // Return the modified struct.
Chris Lattnercc19efa2011-06-20 04:01:31 +00002133 return ConstantStruct::get(STy, Elts);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002134 }
Jakub Staszak9525a772012-12-06 21:57:16 +00002135
Chris Lattnercc19efa2011-06-20 04:01:31 +00002136 ConstantInt *CI = cast<ConstantInt>(Addr->getOperand(OpNo));
Chris Lattner229907c2011-07-18 04:54:35 +00002137 SequentialType *InitTy = cast<SequentialType>(Init->getType());
Chris Lattnercc19efa2011-06-20 04:01:31 +00002138
2139 uint64_t NumElts;
Chris Lattner229907c2011-07-18 04:54:35 +00002140 if (ArrayType *ATy = dyn_cast<ArrayType>(InitTy))
Chris Lattnercc19efa2011-06-20 04:01:31 +00002141 NumElts = ATy->getNumElements();
2142 else
Chris Lattnerfa775002012-01-26 02:32:04 +00002143 NumElts = InitTy->getVectorNumElements();
Chris Lattnercc19efa2011-06-20 04:01:31 +00002144
2145 // Break up the array into elements.
Chris Lattnerfa775002012-01-26 02:32:04 +00002146 for (uint64_t i = 0, e = NumElts; i != e; ++i)
2147 Elts.push_back(Init->getAggregateElement(i));
Chris Lattnercc19efa2011-06-20 04:01:31 +00002148
2149 assert(CI->getZExtValue() < NumElts);
2150 Elts[CI->getZExtValue()] =
Zhou Sheng8e6d64a2012-12-01 10:54:28 +00002151 EvaluateStoreInto(Elts[CI->getZExtValue()], Val, Addr, OpNo+1);
Chris Lattnercc19efa2011-06-20 04:01:31 +00002152
2153 if (Init->getType()->isArrayTy())
2154 return ConstantArray::get(cast<ArrayType>(InitTy), Elts);
2155 return ConstantVector::get(Elts);
Chris Lattner46af55e2005-09-26 06:52:44 +00002156}
2157
Chris Lattner99e23fa2005-09-26 04:44:35 +00002158/// CommitValueTo - We have decided that Addr (which satisfies the predicate
2159/// isSimpleEnoughPointerToCommit) should get Val as its value. Make it happen.
Chris Lattner46b5c642009-11-06 04:27:31 +00002160static void CommitValueTo(Constant *Val, Constant *Addr) {
Chris Lattner46af55e2005-09-26 06:52:44 +00002161 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Addr)) {
2162 assert(GV->hasInitializer());
2163 GV->setInitializer(Val);
2164 return;
2165 }
Chris Lattner64ecc462010-01-07 01:16:21 +00002166
Chris Lattner46af55e2005-09-26 06:52:44 +00002167 ConstantExpr *CE = cast<ConstantExpr>(Addr);
2168 GlobalVariable *GV = cast<GlobalVariable>(CE->getOperand(0));
Zhou Sheng8e6d64a2012-12-01 10:54:28 +00002169 GV->setInitializer(EvaluateStoreInto(GV->getInitializer(), Val, CE, 2));
Chris Lattner99e23fa2005-09-26 04:44:35 +00002170}
2171
Nick Lewycky60829a582012-02-20 03:25:59 +00002172namespace {
2173
2174/// Evaluator - This class evaluates LLVM IR, producing the Constant
2175/// representing each SSA instruction. Changes to global variables are stored
2176/// in a mapping that can be iterated over after the evaluation is complete.
2177/// Once an evaluation call fails, the evaluation object should not be reused.
2178class Evaluator {
Nick Lewycky73be5e32012-02-19 23:26:27 +00002179public:
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002180 Evaluator(const DataLayout *DL, const TargetLibraryInfo *TLI)
2181 : DL(DL), TLI(TLI) {
Benjamin Kramer64425fe2014-05-03 15:50:37 +00002182 ValueStack.emplace_back();
Nick Lewycky73be5e32012-02-19 23:26:27 +00002183 }
2184
Nick Lewycky60829a582012-02-20 03:25:59 +00002185 ~Evaluator() {
David Blaikiebc442202014-04-21 20:49:36 +00002186 for (auto &Tmp : AllocaTmps)
Nick Lewycky73be5e32012-02-19 23:26:27 +00002187 // If there are still users of the alloca, the program is doing something
2188 // silly, e.g. storing the address of the alloca somewhere and using it
2189 // later. Since this is undefined, we'll just make it be null.
2190 if (!Tmp->use_empty())
2191 Tmp->replaceAllUsesWith(Constant::getNullValue(Tmp->getType()));
Nick Lewycky73be5e32012-02-19 23:26:27 +00002192 }
2193
2194 /// EvaluateFunction - Evaluate a call to function F, returning true if
2195 /// successful, false if we can't evaluate it. ActualArgs contains the formal
2196 /// arguments for the function.
2197 bool EvaluateFunction(Function *F, Constant *&RetVal,
2198 const SmallVectorImpl<Constant*> &ActualArgs);
2199
2200 /// EvaluateBlock - Evaluate all instructions in block BB, returning true if
2201 /// successful, false if we can't evaluate it. NewBB returns the next BB that
2202 /// control flows into, or null upon return.
2203 bool EvaluateBlock(BasicBlock::iterator CurInst, BasicBlock *&NextBB);
2204
2205 Constant *getVal(Value *V) {
2206 if (Constant *CV = dyn_cast<Constant>(V)) return CV;
Benjamin Kramer64425fe2014-05-03 15:50:37 +00002207 Constant *R = ValueStack.back().lookup(V);
Nick Lewycky73be5e32012-02-19 23:26:27 +00002208 assert(R && "Reference to an uncomputed value!");
2209 return R;
2210 }
2211
2212 void setVal(Value *V, Constant *C) {
Benjamin Kramer64425fe2014-05-03 15:50:37 +00002213 ValueStack.back()[V] = C;
Nick Lewycky73be5e32012-02-19 23:26:27 +00002214 }
2215
2216 const DenseMap<Constant*, Constant*> &getMutatedMemory() const {
2217 return MutatedMemory;
2218 }
2219
2220 const SmallPtrSet<GlobalVariable*, 8> &getInvariants() const {
2221 return Invariants;
2222 }
2223
2224private:
2225 Constant *ComputeLoadResult(Constant *P);
2226
2227 /// ValueStack - As we compute SSA register values, we store their contents
Benjamin Kramer64425fe2014-05-03 15:50:37 +00002228 /// here. The back of the deque contains the current function and the stack
Nick Lewycky73be5e32012-02-19 23:26:27 +00002229 /// contains the values in the calling frames.
Benjamin Kramer64425fe2014-05-03 15:50:37 +00002230 std::deque<DenseMap<Value*, Constant*>> ValueStack;
Nick Lewycky73be5e32012-02-19 23:26:27 +00002231
2232 /// CallStack - This is used to detect recursion. In pathological situations
2233 /// we could hit exponential behavior, but at least there is nothing
2234 /// unbounded.
2235 SmallVector<Function*, 4> CallStack;
2236
2237 /// MutatedMemory - For each store we execute, we update this map. Loads
2238 /// check this to get the most up-to-date value. If evaluation is successful,
2239 /// this state is committed to the process.
2240 DenseMap<Constant*, Constant*> MutatedMemory;
2241
2242 /// AllocaTmps - To 'execute' an alloca, we create a temporary global variable
2243 /// to represent its body. This vector is needed so we can delete the
2244 /// temporary globals when we are done.
David Blaikiebc442202014-04-21 20:49:36 +00002245 SmallVector<std::unique_ptr<GlobalVariable>, 32> AllocaTmps;
Nick Lewycky73be5e32012-02-19 23:26:27 +00002246
2247 /// Invariants - These global variables have been marked invariant by the
2248 /// static constructor.
2249 SmallPtrSet<GlobalVariable*, 8> Invariants;
2250
2251 /// SimpleConstants - These are constants we have checked and know to be
2252 /// simple enough to live in a static initializer of a global.
2253 SmallPtrSet<Constant*, 8> SimpleConstants;
2254
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002255 const DataLayout *DL;
Nick Lewycky73be5e32012-02-19 23:26:27 +00002256 const TargetLibraryInfo *TLI;
2257};
2258
Nick Lewycky60829a582012-02-20 03:25:59 +00002259} // anonymous namespace
2260
Chris Lattnerb0096632005-09-26 05:16:34 +00002261/// ComputeLoadResult - Return the value that would be computed by a load from
2262/// P after the stores reflected by 'memory' have been performed. If we can't
2263/// decide, return null.
Nick Lewycky60829a582012-02-20 03:25:59 +00002264Constant *Evaluator::ComputeLoadResult(Constant *P) {
Chris Lattner4b05c322005-09-26 05:15:37 +00002265 // If this memory location has been recently stored, use the stored value: it
2266 // is the most up-to-date.
Nick Lewycky73be5e32012-02-19 23:26:27 +00002267 DenseMap<Constant*, Constant*>::const_iterator I = MutatedMemory.find(P);
2268 if (I != MutatedMemory.end()) return I->second;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002269
Chris Lattner4b05c322005-09-26 05:15:37 +00002270 // Access it.
2271 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(P)) {
Dan Gohman5d5bc6d2009-08-19 18:20:44 +00002272 if (GV->hasDefinitiveInitializer())
Chris Lattner4b05c322005-09-26 05:15:37 +00002273 return GV->getInitializer();
Craig Topperf40110f2014-04-25 05:29:35 +00002274 return nullptr;
Chris Lattner4b05c322005-09-26 05:15:37 +00002275 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002276
Chris Lattner46af55e2005-09-26 06:52:44 +00002277 // Handle a constantexpr getelementptr.
2278 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(P))
2279 if (CE->getOpcode() == Instruction::GetElementPtr &&
2280 isa<GlobalVariable>(CE->getOperand(0))) {
2281 GlobalVariable *GV = cast<GlobalVariable>(CE->getOperand(0));
Dan Gohman5d5bc6d2009-08-19 18:20:44 +00002282 if (GV->hasDefinitiveInitializer())
Dan Gohmane525d9d2009-10-05 16:36:26 +00002283 return ConstantFoldLoadThroughGEPConstantExpr(GV->getInitializer(), CE);
Chris Lattner46af55e2005-09-26 06:52:44 +00002284 }
2285
Craig Topperf40110f2014-04-25 05:29:35 +00002286 return nullptr; // don't know how to evaluate.
Chris Lattner4b05c322005-09-26 05:15:37 +00002287}
2288
Nick Lewycky239fdf02012-02-06 08:24:44 +00002289/// EvaluateBlock - Evaluate all instructions in block BB, returning true if
2290/// successful, false if we can't evaluate it. NewBB returns the next BB that
2291/// control flows into, or null upon return.
Nick Lewycky60829a582012-02-20 03:25:59 +00002292bool Evaluator::EvaluateBlock(BasicBlock::iterator CurInst,
2293 BasicBlock *&NextBB) {
Chris Lattner99e23fa2005-09-26 04:44:35 +00002294 // This is the main evaluation loop.
2295 while (1) {
Craig Topperf40110f2014-04-25 05:29:35 +00002296 Constant *InstResult = nullptr;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002297
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002298 DEBUG(dbgs() << "Evaluating Instruction: " << *CurInst << "\n");
2299
Chris Lattner99e23fa2005-09-26 04:44:35 +00002300 if (StoreInst *SI = dyn_cast<StoreInst>(CurInst)) {
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002301 if (!SI->isSimple()) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002302 DEBUG(dbgs() << "Store is not simple! Can not evaluate.\n");
2303 return false; // no volatile/atomic accesses.
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002304 }
Nick Lewycky73be5e32012-02-19 23:26:27 +00002305 Constant *Ptr = getVal(SI->getOperand(1));
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002306 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ptr)) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002307 DEBUG(dbgs() << "Folding constant ptr expression: " << *Ptr);
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002308 Ptr = ConstantFoldConstantExpression(CE, DL, TLI);
Michael Gottesman2a654272013-01-11 23:08:52 +00002309 DEBUG(dbgs() << "; To: " << *Ptr << "\n");
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002310 }
2311 if (!isSimpleEnoughPointerToCommit(Ptr)) {
Chris Lattner99e23fa2005-09-26 04:44:35 +00002312 // If this is too complex for us to commit, reject it.
Michael Gottesman2a654272013-01-11 23:08:52 +00002313 DEBUG(dbgs() << "Pointer is too complex for us to evaluate store.");
Chris Lattner65a3a092005-09-27 04:45:34 +00002314 return false;
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002315 }
Jakub Staszak9525a772012-12-06 21:57:16 +00002316
Nick Lewycky73be5e32012-02-19 23:26:27 +00002317 Constant *Val = getVal(SI->getOperand(0));
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002318
2319 // If this might be too difficult for the backend to handle (e.g. the addr
2320 // of one global variable divided by another) then we can't commit it.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002321 if (!isSimpleEnoughValueToCommit(Val, SimpleConstants, DL)) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002322 DEBUG(dbgs() << "Store value is too complex to evaluate store. " << *Val
2323 << "\n");
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002324 return false;
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002325 }
Jakub Staszak9525a772012-12-06 21:57:16 +00002326
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002327 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ptr)) {
Owen Anderson9eb7cb482011-01-14 22:19:20 +00002328 if (CE->getOpcode() == Instruction::BitCast) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002329 DEBUG(dbgs() << "Attempting to resolve bitcast on constant ptr.\n");
Owen Anderson9eb7cb482011-01-14 22:19:20 +00002330 // If we're evaluating a store through a bitcast, then we need
2331 // to pull the bitcast off the pointer type and push it onto the
2332 // stored value.
Chris Lattner8b4952f2011-01-16 02:05:10 +00002333 Ptr = CE->getOperand(0);
Jakub Staszak9525a772012-12-06 21:57:16 +00002334
Nick Lewycky239fdf02012-02-06 08:24:44 +00002335 Type *NewTy = cast<PointerType>(Ptr->getType())->getElementType();
Jakub Staszak9525a772012-12-06 21:57:16 +00002336
Owen Anderson4e54efd2011-01-16 04:33:33 +00002337 // In order to push the bitcast onto the stored value, a bitcast
2338 // from NewTy to Val's type must be legal. If it's not, we can try
2339 // introspecting NewTy to find a legal conversion.
2340 while (!Val->getType()->canLosslesslyBitCastTo(NewTy)) {
2341 // If NewTy is a struct, we can convert the pointer to the struct
2342 // into a pointer to its first member.
2343 // FIXME: This could be extended to support arrays as well.
Chris Lattner229907c2011-07-18 04:54:35 +00002344 if (StructType *STy = dyn_cast<StructType>(NewTy)) {
Owen Anderson4e54efd2011-01-16 04:33:33 +00002345 NewTy = STy->getTypeAtIndex(0U);
2346
Nick Lewycky239fdf02012-02-06 08:24:44 +00002347 IntegerType *IdxTy = IntegerType::get(NewTy->getContext(), 32);
Owen Anderson4e54efd2011-01-16 04:33:33 +00002348 Constant *IdxZero = ConstantInt::get(IdxTy, 0, false);
2349 Constant * const IdxList[] = {IdxZero, IdxZero};
2350
Jay Foad17bab442011-07-22 08:52:50 +00002351 Ptr = ConstantExpr::getGetElementPtr(Ptr, IdxList);
Nick Lewycky9d0da182012-02-21 22:08:06 +00002352 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ptr))
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002353 Ptr = ConstantFoldConstantExpression(CE, DL, TLI);
Nick Lewycky9d0da182012-02-21 22:08:06 +00002354
Owen Anderson4e54efd2011-01-16 04:33:33 +00002355 // If we can't improve the situation by introspecting NewTy,
2356 // we have to give up.
2357 } else {
Michael Gottesman2a654272013-01-11 23:08:52 +00002358 DEBUG(dbgs() << "Failed to bitcast constant ptr, can not "
2359 "evaluate.\n");
Nick Lewycky239fdf02012-02-06 08:24:44 +00002360 return false;
Owen Anderson4e54efd2011-01-16 04:33:33 +00002361 }
Owen Anderson9eb7cb482011-01-14 22:19:20 +00002362 }
Jakub Staszak9525a772012-12-06 21:57:16 +00002363
Owen Anderson4e54efd2011-01-16 04:33:33 +00002364 // If we found compatible types, go ahead and push the bitcast
2365 // onto the stored value.
Owen Anderson9eb7cb482011-01-14 22:19:20 +00002366 Val = ConstantExpr::getBitCast(Val, NewTy);
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002367
Michael Gottesman2a654272013-01-11 23:08:52 +00002368 DEBUG(dbgs() << "Evaluated bitcast: " << *Val << "\n");
Owen Anderson9eb7cb482011-01-14 22:19:20 +00002369 }
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002370 }
Jakub Staszak9525a772012-12-06 21:57:16 +00002371
Chris Lattner99e23fa2005-09-26 04:44:35 +00002372 MutatedMemory[Ptr] = Val;
2373 } else if (BinaryOperator *BO = dyn_cast<BinaryOperator>(CurInst)) {
Owen Anderson487375e2009-07-29 18:55:55 +00002374 InstResult = ConstantExpr::get(BO->getOpcode(),
Nick Lewycky73be5e32012-02-19 23:26:27 +00002375 getVal(BO->getOperand(0)),
2376 getVal(BO->getOperand(1)));
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002377 DEBUG(dbgs() << "Found a BinaryOperator! Simplifying: " << *InstResult
Michael Gottesman2a654272013-01-11 23:08:52 +00002378 << "\n");
Reid Spencer266e42b2006-12-23 06:05:41 +00002379 } else if (CmpInst *CI = dyn_cast<CmpInst>(CurInst)) {
Owen Anderson487375e2009-07-29 18:55:55 +00002380 InstResult = ConstantExpr::getCompare(CI->getPredicate(),
Nick Lewycky73be5e32012-02-19 23:26:27 +00002381 getVal(CI->getOperand(0)),
2382 getVal(CI->getOperand(1)));
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002383 DEBUG(dbgs() << "Found a CmpInst! Simplifying: " << *InstResult
Michael Gottesman2a654272013-01-11 23:08:52 +00002384 << "\n");
Chris Lattner99e23fa2005-09-26 04:44:35 +00002385 } else if (CastInst *CI = dyn_cast<CastInst>(CurInst)) {
Owen Anderson487375e2009-07-29 18:55:55 +00002386 InstResult = ConstantExpr::getCast(CI->getOpcode(),
Nick Lewycky73be5e32012-02-19 23:26:27 +00002387 getVal(CI->getOperand(0)),
Chris Lattner99e23fa2005-09-26 04:44:35 +00002388 CI->getType());
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002389 DEBUG(dbgs() << "Found a Cast! Simplifying: " << *InstResult
Michael Gottesman2a654272013-01-11 23:08:52 +00002390 << "\n");
Chris Lattner99e23fa2005-09-26 04:44:35 +00002391 } else if (SelectInst *SI = dyn_cast<SelectInst>(CurInst)) {
Nick Lewycky73be5e32012-02-19 23:26:27 +00002392 InstResult = ConstantExpr::getSelect(getVal(SI->getOperand(0)),
2393 getVal(SI->getOperand(1)),
2394 getVal(SI->getOperand(2)));
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002395 DEBUG(dbgs() << "Found a Select! Simplifying: " << *InstResult
Michael Gottesman2a654272013-01-11 23:08:52 +00002396 << "\n");
David Majnemerfe8c7542014-08-08 05:50:43 +00002397 } else if (auto *EVI = dyn_cast<ExtractValueInst>(CurInst)) {
2398 InstResult = ConstantExpr::getExtractValue(
2399 getVal(EVI->getAggregateOperand()), EVI->getIndices());
2400 DEBUG(dbgs() << "Found an ExtractValueInst! Simplifying: " << *InstResult
2401 << "\n");
2402 } else if (auto *IVI = dyn_cast<InsertValueInst>(CurInst)) {
2403 InstResult = ConstantExpr::getInsertValue(
2404 getVal(IVI->getAggregateOperand()),
2405 getVal(IVI->getInsertedValueOperand()), IVI->getIndices());
2406 DEBUG(dbgs() << "Found an InsertValueInst! Simplifying: " << *InstResult
2407 << "\n");
Chris Lattner4b05c322005-09-26 05:15:37 +00002408 } else if (GetElementPtrInst *GEP = dyn_cast<GetElementPtrInst>(CurInst)) {
Nick Lewycky73be5e32012-02-19 23:26:27 +00002409 Constant *P = getVal(GEP->getOperand(0));
Chris Lattnerf96f4a82007-01-31 04:40:53 +00002410 SmallVector<Constant*, 8> GEPOps;
Gabor Greif3a9fba52008-05-29 01:59:18 +00002411 for (User::op_iterator i = GEP->op_begin() + 1, e = GEP->op_end();
2412 i != e; ++i)
Nick Lewycky73be5e32012-02-19 23:26:27 +00002413 GEPOps.push_back(getVal(*i));
Jay Foad2f5fc8c2011-07-21 15:15:37 +00002414 InstResult =
2415 ConstantExpr::getGetElementPtr(P, GEPOps,
2416 cast<GEPOperator>(GEP)->isInBounds());
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002417 DEBUG(dbgs() << "Found a GEP! Simplifying: " << *InstResult
Michael Gottesman2a654272013-01-11 23:08:52 +00002418 << "\n");
Chris Lattner4b05c322005-09-26 05:15:37 +00002419 } else if (LoadInst *LI = dyn_cast<LoadInst>(CurInst)) {
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002420
2421 if (!LI->isSimple()) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002422 DEBUG(dbgs() << "Found a Load! Not a simple load, can not evaluate.\n");
2423 return false; // no volatile/atomic accesses.
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002424 }
2425
Nick Lewycky9d0da182012-02-21 22:08:06 +00002426 Constant *Ptr = getVal(LI->getOperand(0));
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002427 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(Ptr)) {
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002428 Ptr = ConstantFoldConstantExpression(CE, DL, TLI);
Michael Gottesman2a654272013-01-11 23:08:52 +00002429 DEBUG(dbgs() << "Found a constant pointer expression, constant "
2430 "folding: " << *Ptr << "\n");
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002431 }
Nick Lewycky9d0da182012-02-21 22:08:06 +00002432 InstResult = ComputeLoadResult(Ptr);
Craig Topperf40110f2014-04-25 05:29:35 +00002433 if (!InstResult) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002434 DEBUG(dbgs() << "Failed to compute load result. Can not evaluate load."
2435 "\n");
2436 return false; // Could not evaluate load.
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002437 }
2438
2439 DEBUG(dbgs() << "Evaluated load: " << *InstResult << "\n");
Chris Lattner6bf2cd52005-09-26 17:07:09 +00002440 } else if (AllocaInst *AI = dyn_cast<AllocaInst>(CurInst)) {
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002441 if (AI->isArrayAllocation()) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002442 DEBUG(dbgs() << "Found an array alloca. Can not evaluate.\n");
2443 return false; // Cannot handle array allocs.
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002444 }
Chris Lattner229907c2011-07-18 04:54:35 +00002445 Type *Ty = AI->getType()->getElementType();
David Blaikiebc442202014-04-21 20:49:36 +00002446 AllocaTmps.push_back(
2447 make_unique<GlobalVariable>(Ty, false, GlobalValue::InternalLinkage,
2448 UndefValue::get(Ty), AI->getName()));
2449 InstResult = AllocaTmps.back().get();
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002450 DEBUG(dbgs() << "Found an alloca. Result: " << *InstResult << "\n");
Nick Lewyckyc1572e42012-02-12 05:09:35 +00002451 } else if (isa<CallInst>(CurInst) || isa<InvokeInst>(CurInst)) {
2452 CallSite CS(CurInst);
Devang Patel04852aa2009-03-09 23:04:12 +00002453
2454 // Debug info can safely be ignored here.
Nick Lewyckyc1572e42012-02-12 05:09:35 +00002455 if (isa<DbgInfoIntrinsic>(CS.getInstruction())) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002456 DEBUG(dbgs() << "Ignoring debug info.\n");
Devang Patel04852aa2009-03-09 23:04:12 +00002457 ++CurInst;
2458 continue;
2459 }
2460
Chris Lattnerfd2e13b2006-07-07 21:37:01 +00002461 // Cannot handle inline asm.
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002462 if (isa<InlineAsm>(CS.getCalledValue())) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002463 DEBUG(dbgs() << "Found inline asm, can not evaluate.\n");
2464 return false;
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002465 }
Chris Lattnerfd2e13b2006-07-07 21:37:01 +00002466
Nick Lewycky68f9f9d2012-02-17 06:59:21 +00002467 if (IntrinsicInst *II = dyn_cast<IntrinsicInst>(CS.getInstruction())) {
2468 if (MemSetInst *MSI = dyn_cast<MemSetInst>(II)) {
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002469 if (MSI->isVolatile()) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002470 DEBUG(dbgs() << "Can not optimize a volatile memset " <<
2471 "intrinsic.\n");
2472 return false;
2473 }
Nick Lewycky73be5e32012-02-19 23:26:27 +00002474 Constant *Ptr = getVal(MSI->getDest());
2475 Constant *Val = getVal(MSI->getValue());
2476 Constant *DestVal = ComputeLoadResult(getVal(Ptr));
Nick Lewycky68f9f9d2012-02-17 06:59:21 +00002477 if (Val->isNullValue() && DestVal && DestVal->isNullValue()) {
2478 // This memset is a no-op.
Michael Gottesman2a654272013-01-11 23:08:52 +00002479 DEBUG(dbgs() << "Ignoring no-op memset.\n");
Nick Lewycky68f9f9d2012-02-17 06:59:21 +00002480 ++CurInst;
2481 continue;
2482 }
2483 }
2484
2485 if (II->getIntrinsicID() == Intrinsic::lifetime_start ||
2486 II->getIntrinsicID() == Intrinsic::lifetime_end) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002487 DEBUG(dbgs() << "Ignoring lifetime intrinsic.\n");
Nick Lewycky68f9f9d2012-02-17 06:59:21 +00002488 ++CurInst;
2489 continue;
2490 }
2491
2492 if (II->getIntrinsicID() == Intrinsic::invariant_start) {
2493 // We don't insert an entry into Values, as it doesn't have a
2494 // meaningful return value.
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002495 if (!II->use_empty()) {
Alp Tokerf907b892013-12-05 05:44:44 +00002496 DEBUG(dbgs() << "Found unused invariant_start. Can't evaluate.\n");
Nick Lewycky68f9f9d2012-02-17 06:59:21 +00002497 return false;
Michael Gottesman2a654272013-01-11 23:08:52 +00002498 }
Nick Lewycky68f9f9d2012-02-17 06:59:21 +00002499 ConstantInt *Size = cast<ConstantInt>(II->getArgOperand(0));
Nick Lewycky519561f2012-02-20 23:32:26 +00002500 Value *PtrArg = getVal(II->getArgOperand(1));
2501 Value *Ptr = PtrArg->stripPointerCasts();
2502 if (GlobalVariable *GV = dyn_cast<GlobalVariable>(Ptr)) {
2503 Type *ElemTy = cast<PointerType>(GV->getType())->getElementType();
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002504 if (DL && !Size->isAllOnesValue() &&
Nick Lewycky519561f2012-02-20 23:32:26 +00002505 Size->getValue().getLimitedValue() >=
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002506 DL->getTypeStoreSize(ElemTy)) {
Nick Lewycky68f9f9d2012-02-17 06:59:21 +00002507 Invariants.insert(GV);
Michael Gottesman2a654272013-01-11 23:08:52 +00002508 DEBUG(dbgs() << "Found a global var that is an invariant: " << *GV
2509 << "\n");
2510 } else {
2511 DEBUG(dbgs() << "Found a global var, but can not treat it as an "
2512 "invariant.\n");
2513 }
Nick Lewycky68f9f9d2012-02-17 06:59:21 +00002514 }
2515 // Continue even if we do nothing.
Nick Lewyckya3bb03e2011-05-29 18:41:56 +00002516 ++CurInst;
2517 continue;
2518 }
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002519
Michael Gottesman2a654272013-01-11 23:08:52 +00002520 DEBUG(dbgs() << "Unknown intrinsic. Can not evaluate.\n");
Nick Lewyckya3bb03e2011-05-29 18:41:56 +00002521 return false;
2522 }
2523
Chris Lattner65a3a092005-09-27 04:45:34 +00002524 // Resolve function pointers.
Nick Lewycky73be5e32012-02-19 23:26:27 +00002525 Function *Callee = dyn_cast<Function>(getVal(CS.getCalledValue()));
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002526 if (!Callee || Callee->mayBeOverridden()) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002527 DEBUG(dbgs() << "Can not resolve function pointer.\n");
Nick Lewyckyc1572e42012-02-12 05:09:35 +00002528 return false; // Cannot resolve.
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002529 }
Chris Lattner3d27e7f2005-09-27 05:02:43 +00002530
Duncan Sandsc4ce58d82009-08-17 14:33:27 +00002531 SmallVector<Constant*, 8> Formals;
Nick Lewyckyc1572e42012-02-12 05:09:35 +00002532 for (User::op_iterator i = CS.arg_begin(), e = CS.arg_end(); i != e; ++i)
Nick Lewycky73be5e32012-02-19 23:26:27 +00002533 Formals.push_back(getVal(*i));
Duncan Sandsc4ce58d82009-08-17 14:33:27 +00002534
Reid Spencer5301e7c2007-01-30 20:08:39 +00002535 if (Callee->isDeclaration()) {
Chris Lattner3d27e7f2005-09-27 05:02:43 +00002536 // If this is a function we can constant fold, do it.
Chad Rosiere6de63d2011-12-01 21:29:16 +00002537 if (Constant *C = ConstantFoldCall(Callee, Formals, TLI)) {
Chris Lattner3d27e7f2005-09-27 05:02:43 +00002538 InstResult = C;
Michael Gottesman2a654272013-01-11 23:08:52 +00002539 DEBUG(dbgs() << "Constant folded function call. Result: " <<
2540 *InstResult << "\n");
Chris Lattner3d27e7f2005-09-27 05:02:43 +00002541 } else {
Michael Gottesman2a654272013-01-11 23:08:52 +00002542 DEBUG(dbgs() << "Can not constant fold function call.\n");
Chris Lattner3d27e7f2005-09-27 05:02:43 +00002543 return false;
2544 }
2545 } else {
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002546 if (Callee->getFunctionType()->isVarArg()) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002547 DEBUG(dbgs() << "Can not constant fold vararg function call.\n");
Chris Lattner3d27e7f2005-09-27 05:02:43 +00002548 return false;
Michael Gottesman2a654272013-01-11 23:08:52 +00002549 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002550
Craig Topperf40110f2014-04-25 05:29:35 +00002551 Constant *RetVal = nullptr;
Chris Lattner3d27e7f2005-09-27 05:02:43 +00002552 // Execute the call, if successful, use the return value.
Benjamin Kramer64425fe2014-05-03 15:50:37 +00002553 ValueStack.emplace_back();
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002554 if (!EvaluateFunction(Callee, RetVal, Formals)) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002555 DEBUG(dbgs() << "Failed to evaluate function.\n");
Chris Lattner3d27e7f2005-09-27 05:02:43 +00002556 return false;
Michael Gottesman2a654272013-01-11 23:08:52 +00002557 }
David Blaikiebc442202014-04-21 20:49:36 +00002558 ValueStack.pop_back();
Chris Lattner3d27e7f2005-09-27 05:02:43 +00002559 InstResult = RetVal;
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002560
Craig Topperf40110f2014-04-25 05:29:35 +00002561 if (InstResult) {
Michael Gottesman2a654272013-01-11 23:08:52 +00002562 DEBUG(dbgs() << "Successfully evaluated function. Result: " <<
2563 InstResult << "\n\n");
2564 } else {
2565 DEBUG(dbgs() << "Successfully evaluated function. Result: 0\n\n");
2566 }
Chris Lattner3d27e7f2005-09-27 05:02:43 +00002567 }
Reid Spencerde46e482006-11-02 20:25:50 +00002568 } else if (isa<TerminatorInst>(CurInst)) {
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002569 DEBUG(dbgs() << "Found a terminator instruction.\n");
2570
Chris Lattner3e9ea5f2005-09-26 04:57:38 +00002571 if (BranchInst *BI = dyn_cast<BranchInst>(CurInst)) {
2572 if (BI->isUnconditional()) {
Nick Lewycky239fdf02012-02-06 08:24:44 +00002573 NextBB = BI->getSuccessor(0);
Chris Lattner3e9ea5f2005-09-26 04:57:38 +00002574 } else {
Zhou Sheng75b871f2007-01-11 12:24:14 +00002575 ConstantInt *Cond =
Nick Lewycky73be5e32012-02-19 23:26:27 +00002576 dyn_cast<ConstantInt>(getVal(BI->getCondition()));
Chris Lattner15649082007-01-12 18:30:11 +00002577 if (!Cond) return false; // Cannot determine.
Zhou Sheng75b871f2007-01-11 12:24:14 +00002578
Nick Lewycky239fdf02012-02-06 08:24:44 +00002579 NextBB = BI->getSuccessor(!Cond->getZExtValue());
Chris Lattner3e9ea5f2005-09-26 04:57:38 +00002580 }
2581 } else if (SwitchInst *SI = dyn_cast<SwitchInst>(CurInst)) {
2582 ConstantInt *Val =
Nick Lewycky73be5e32012-02-19 23:26:27 +00002583 dyn_cast<ConstantInt>(getVal(SI->getCondition()));
Chris Lattner65a3a092005-09-27 04:45:34 +00002584 if (!Val) return false; // Cannot determine.
Stepan Dyatkovskiy5b648af2012-03-08 07:06:20 +00002585 NextBB = SI->findCaseValue(Val).getCaseSuccessor();
Chris Lattner31274882009-10-29 05:51:50 +00002586 } else if (IndirectBrInst *IBI = dyn_cast<IndirectBrInst>(CurInst)) {
Nick Lewycky73be5e32012-02-19 23:26:27 +00002587 Value *Val = getVal(IBI->getAddress())->stripPointerCasts();
Chris Lattner31274882009-10-29 05:51:50 +00002588 if (BlockAddress *BA = dyn_cast<BlockAddress>(Val))
Nick Lewycky239fdf02012-02-06 08:24:44 +00002589 NextBB = BA->getBasicBlock();
Chris Lattneraa99c942009-11-01 01:27:45 +00002590 else
2591 return false; // Cannot determine.
Nick Lewycky239fdf02012-02-06 08:24:44 +00002592 } else if (isa<ReturnInst>(CurInst)) {
Craig Topperf40110f2014-04-25 05:29:35 +00002593 NextBB = nullptr;
Chris Lattner3e9ea5f2005-09-26 04:57:38 +00002594 } else {
Bill Wendlingf891bf82011-07-31 06:30:59 +00002595 // invoke, unwind, resume, unreachable.
Michael Gottesman2a654272013-01-11 23:08:52 +00002596 DEBUG(dbgs() << "Can not handle terminator.");
Chris Lattner65a3a092005-09-27 04:45:34 +00002597 return false; // Cannot handle this terminator.
Chris Lattner3e9ea5f2005-09-26 04:57:38 +00002598 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002599
Nick Lewycky239fdf02012-02-06 08:24:44 +00002600 // We succeeded at evaluating this block!
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002601 DEBUG(dbgs() << "Successfully evaluated block.\n");
Nick Lewycky239fdf02012-02-06 08:24:44 +00002602 return true;
Chris Lattner99e23fa2005-09-26 04:44:35 +00002603 } else {
Chris Lattner99e23fa2005-09-26 04:44:35 +00002604 // Did not know how to evaluate this!
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002605 DEBUG(dbgs() << "Failed to evaluate block due to unhandled instruction."
Michael Gottesman2a654272013-01-11 23:08:52 +00002606 "\n");
Chris Lattner65a3a092005-09-27 04:45:34 +00002607 return false;
Chris Lattner99e23fa2005-09-26 04:44:35 +00002608 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002609
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002610 if (!CurInst->use_empty()) {
2611 if (ConstantExpr *CE = dyn_cast<ConstantExpr>(InstResult))
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002612 InstResult = ConstantFoldConstantExpression(CE, DL, TLI);
Jakub Staszak9525a772012-12-06 21:57:16 +00002613
Nick Lewycky73be5e32012-02-19 23:26:27 +00002614 setVal(CurInst, InstResult);
Chris Lattner0d71c4f2010-12-07 04:33:29 +00002615 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002616
Dan Gohmaneab06fa2012-03-13 18:01:37 +00002617 // If we just processed an invoke, we finished evaluating the block.
2618 if (InvokeInst *II = dyn_cast<InvokeInst>(CurInst)) {
2619 NextBB = II->getNormalDest();
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002620 DEBUG(dbgs() << "Found an invoke instruction. Finished Block.\n\n");
Dan Gohmaneab06fa2012-03-13 18:01:37 +00002621 return true;
2622 }
2623
Chris Lattner99e23fa2005-09-26 04:44:35 +00002624 // Advance program counter.
2625 ++CurInst;
2626 }
Chris Lattnerda1889b2005-09-27 04:27:01 +00002627}
2628
Nick Lewycky239fdf02012-02-06 08:24:44 +00002629/// EvaluateFunction - Evaluate a call to function F, returning true if
2630/// successful, false if we can't evaluate it. ActualArgs contains the formal
2631/// arguments for the function.
Nick Lewycky60829a582012-02-20 03:25:59 +00002632bool Evaluator::EvaluateFunction(Function *F, Constant *&RetVal,
2633 const SmallVectorImpl<Constant*> &ActualArgs) {
Nick Lewycky239fdf02012-02-06 08:24:44 +00002634 // Check to see if this function is already executing (recursion). If so,
2635 // bail out. TODO: we might want to accept limited recursion.
2636 if (std::find(CallStack.begin(), CallStack.end(), F) != CallStack.end())
2637 return false;
2638
2639 CallStack.push_back(F);
2640
Nick Lewycky239fdf02012-02-06 08:24:44 +00002641 // Initialize arguments to the incoming values specified.
2642 unsigned ArgNo = 0;
2643 for (Function::arg_iterator AI = F->arg_begin(), E = F->arg_end(); AI != E;
2644 ++AI, ++ArgNo)
Nick Lewycky73be5e32012-02-19 23:26:27 +00002645 setVal(AI, ActualArgs[ArgNo]);
Nick Lewycky239fdf02012-02-06 08:24:44 +00002646
2647 // ExecutedBlocks - We only handle non-looping, non-recursive code. As such,
2648 // we can only evaluate any one basic block at most once. This set keeps
2649 // track of what we have executed so we can detect recursive cases etc.
2650 SmallPtrSet<BasicBlock*, 32> ExecutedBlocks;
2651
2652 // CurBB - The current basic block we're evaluating.
2653 BasicBlock *CurBB = F->begin();
2654
Nick Lewycky4231c412012-02-12 00:47:24 +00002655 BasicBlock::iterator CurInst = CurBB->begin();
2656
Nick Lewycky239fdf02012-02-06 08:24:44 +00002657 while (1) {
Craig Topperf40110f2014-04-25 05:29:35 +00002658 BasicBlock *NextBB = nullptr; // Initialized to avoid compiler warnings.
Michael Gottesmand1a46f22013-01-11 20:07:53 +00002659 DEBUG(dbgs() << "Trying to evaluate BB: " << *CurBB << "\n");
2660
Nick Lewycky73be5e32012-02-19 23:26:27 +00002661 if (!EvaluateBlock(CurInst, NextBB))
Nick Lewycky239fdf02012-02-06 08:24:44 +00002662 return false;
2663
Craig Topperf40110f2014-04-25 05:29:35 +00002664 if (!NextBB) {
Nick Lewycky239fdf02012-02-06 08:24:44 +00002665 // Successfully running until there's no next block means that we found
2666 // the return. Fill it the return value and pop the call stack.
2667 ReturnInst *RI = cast<ReturnInst>(CurBB->getTerminator());
2668 if (RI->getNumOperands())
Nick Lewycky73be5e32012-02-19 23:26:27 +00002669 RetVal = getVal(RI->getOperand(0));
Nick Lewycky239fdf02012-02-06 08:24:44 +00002670 CallStack.pop_back();
2671 return true;
2672 }
2673
2674 // Okay, we succeeded in evaluating this control flow. See if we have
2675 // executed the new block before. If so, we have a looping function,
2676 // which we cannot evaluate in reasonable time.
2677 if (!ExecutedBlocks.insert(NextBB))
2678 return false; // looped!
2679
2680 // Okay, we have never been in this block before. Check to see if there
2681 // are any PHI nodes. If so, evaluate them with information about where
2682 // we came from.
Craig Topperf40110f2014-04-25 05:29:35 +00002683 PHINode *PN = nullptr;
Nick Lewycky4231c412012-02-12 00:47:24 +00002684 for (CurInst = NextBB->begin();
2685 (PN = dyn_cast<PHINode>(CurInst)); ++CurInst)
Nick Lewycky73be5e32012-02-19 23:26:27 +00002686 setVal(PN, getVal(PN->getIncomingValueForBlock(CurBB)));
Nick Lewycky239fdf02012-02-06 08:24:44 +00002687
2688 // Advance to the next block.
2689 CurBB = NextBB;
2690 }
2691}
2692
Chris Lattnerda1889b2005-09-27 04:27:01 +00002693/// EvaluateStaticConstructor - Evaluate static constructors in the function, if
2694/// we can. Return true if we can, false otherwise.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002695static bool EvaluateStaticConstructor(Function *F, const DataLayout *DL,
Chad Rosiere6de63d2011-12-01 21:29:16 +00002696 const TargetLibraryInfo *TLI) {
Chris Lattnerda1889b2005-09-27 04:27:01 +00002697 // Call the function.
Rafael Espindola37dc9e12014-02-21 00:06:31 +00002698 Evaluator Eval(DL, TLI);
Chris Lattner65a3a092005-09-27 04:45:34 +00002699 Constant *RetValDummy;
Nick Lewycky73be5e32012-02-19 23:26:27 +00002700 bool EvalSuccess = Eval.EvaluateFunction(F, RetValDummy,
2701 SmallVector<Constant*, 0>());
Jakub Staszak9525a772012-12-06 21:57:16 +00002702
Chris Lattnerda1889b2005-09-27 04:27:01 +00002703 if (EvalSuccess) {
Nico Weber4b2acde2014-05-02 18:35:25 +00002704 ++NumCtorsEvaluated;
2705
Chris Lattner6bf2cd52005-09-26 17:07:09 +00002706 // We succeeded at evaluation: commit the result.
David Greene44cb8ad2010-01-05 01:28:05 +00002707 DEBUG(dbgs() << "FULLY EVALUATED GLOBAL CTOR FUNCTION '"
Nick Lewycky73be5e32012-02-19 23:26:27 +00002708 << F->getName() << "' to " << Eval.getMutatedMemory().size()
Daniel Dunbar0dd5e1e2009-07-25 00:23:56 +00002709 << " stores.\n");
Nick Lewycky73be5e32012-02-19 23:26:27 +00002710 for (DenseMap<Constant*, Constant*>::const_iterator I =
2711 Eval.getMutatedMemory().begin(), E = Eval.getMutatedMemory().end();
Nick Lewyckyb74ae9c2012-06-24 04:07:14 +00002712 I != E; ++I)
Chris Lattner46b5c642009-11-06 04:27:31 +00002713 CommitValueTo(I->second, I->first);
Nick Lewycky73be5e32012-02-19 23:26:27 +00002714 for (SmallPtrSet<GlobalVariable*, 8>::const_iterator I =
2715 Eval.getInvariants().begin(), E = Eval.getInvariants().end();
2716 I != E; ++I)
Nick Lewycky68f9f9d2012-02-17 06:59:21 +00002717 (*I)->setConstant(true);
Chris Lattner6bf2cd52005-09-26 17:07:09 +00002718 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00002719
Chris Lattnerda1889b2005-09-27 04:27:01 +00002720 return EvalSuccess;
Chris Lattner99e23fa2005-09-26 04:44:35 +00002721}
2722
Benjamin Krameradf1ea82014-03-07 21:52:38 +00002723static int compareNames(Constant *const *A, Constant *const *B) {
2724 return (*A)->getName().compare((*B)->getName());
2725}
2726
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002727static void setUsedInitializer(GlobalVariable &V,
2728 SmallPtrSet<GlobalValue *, 8> Init) {
Rafael Espindolac2bb73f2013-07-20 23:33:15 +00002729 if (Init.empty()) {
2730 V.eraseFromParent();
2731 return;
2732 }
2733
Matt Arsenaultda1deab2014-01-02 19:53:49 +00002734 // Type of pointer to the array of pointers.
2735 PointerType *Int8PtrTy = Type::getInt8PtrTy(V.getContext(), 0);
Rafael Espindola00752162013-05-09 17:22:59 +00002736
Matt Arsenaultda1deab2014-01-02 19:53:49 +00002737 SmallVector<llvm::Constant *, 8> UsedArray;
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002738 for (SmallPtrSet<GlobalValue *, 8>::iterator I = Init.begin(), E = Init.end();
2739 I != E; ++I) {
Matt Arsenaultda1deab2014-01-02 19:53:49 +00002740 Constant *Cast
2741 = ConstantExpr::getPointerBitCastOrAddrSpaceCast(*I, Int8PtrTy);
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002742 UsedArray.push_back(Cast);
Rafael Espindola00752162013-05-09 17:22:59 +00002743 }
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002744 // Sort to get deterministic order.
Benjamin Krameradf1ea82014-03-07 21:52:38 +00002745 array_pod_sort(UsedArray.begin(), UsedArray.end(), compareNames);
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002746 ArrayType *ATy = ArrayType::get(Int8PtrTy, UsedArray.size());
Rafael Espindola00752162013-05-09 17:22:59 +00002747
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002748 Module *M = V.getParent();
2749 V.removeFromParent();
2750 GlobalVariable *NV =
2751 new GlobalVariable(*M, ATy, false, llvm::GlobalValue::AppendingLinkage,
2752 llvm::ConstantArray::get(ATy, UsedArray), "");
2753 NV->takeName(&V);
2754 NV->setSection("llvm.metadata");
2755 delete &V;
Rafael Espindola00752162013-05-09 17:22:59 +00002756}
2757
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002758namespace {
Rafael Espindola9aadcc42013-07-19 18:44:51 +00002759/// \brief An easy to access representation of llvm.used and llvm.compiler.used.
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002760class LLVMUsed {
2761 SmallPtrSet<GlobalValue *, 8> Used;
2762 SmallPtrSet<GlobalValue *, 8> CompilerUsed;
2763 GlobalVariable *UsedV;
2764 GlobalVariable *CompilerUsedV;
2765
2766public:
Rafael Espindolaec2375f2013-07-25 02:50:08 +00002767 LLVMUsed(Module &M) {
Rafael Espindola17600e22013-07-25 03:23:25 +00002768 UsedV = collectUsedGlobalVariables(M, Used, false);
2769 CompilerUsedV = collectUsedGlobalVariables(M, CompilerUsed, true);
Rafael Espindola00752162013-05-09 17:22:59 +00002770 }
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002771 typedef SmallPtrSet<GlobalValue *, 8>::iterator iterator;
2772 iterator usedBegin() { return Used.begin(); }
2773 iterator usedEnd() { return Used.end(); }
2774 iterator compilerUsedBegin() { return CompilerUsed.begin(); }
2775 iterator compilerUsedEnd() { return CompilerUsed.end(); }
2776 bool usedCount(GlobalValue *GV) const { return Used.count(GV); }
2777 bool compilerUsedCount(GlobalValue *GV) const {
2778 return CompilerUsed.count(GV);
2779 }
2780 bool usedErase(GlobalValue *GV) { return Used.erase(GV); }
2781 bool compilerUsedErase(GlobalValue *GV) { return CompilerUsed.erase(GV); }
2782 bool usedInsert(GlobalValue *GV) { return Used.insert(GV); }
2783 bool compilerUsedInsert(GlobalValue *GV) { return CompilerUsed.insert(GV); }
Rafael Espindola00752162013-05-09 17:22:59 +00002784
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002785 void syncVariablesAndSets() {
2786 if (UsedV)
2787 setUsedInitializer(*UsedV, Used);
2788 if (CompilerUsedV)
2789 setUsedInitializer(*CompilerUsedV, CompilerUsed);
2790 }
2791};
Rafael Espindola00752162013-05-09 17:22:59 +00002792}
2793
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002794static bool hasUseOtherThanLLVMUsed(GlobalAlias &GA, const LLVMUsed &U) {
2795 if (GA.use_empty()) // No use at all.
2796 return false;
2797
2798 assert((!U.usedCount(&GA) || !U.compilerUsedCount(&GA)) &&
2799 "We should have removed the duplicated "
Rafael Espindola9aadcc42013-07-19 18:44:51 +00002800 "element from llvm.compiler.used");
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002801 if (!GA.hasOneUse())
2802 // Strictly more than one use. So at least one is not in llvm.used and
Rafael Espindola9aadcc42013-07-19 18:44:51 +00002803 // llvm.compiler.used.
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002804 return true;
2805
Rafael Espindola9aadcc42013-07-19 18:44:51 +00002806 // Exactly one use. Check if it is in llvm.used or llvm.compiler.used.
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002807 return !U.usedCount(&GA) && !U.compilerUsedCount(&GA);
Rafael Espindola00752162013-05-09 17:22:59 +00002808}
2809
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002810static bool hasMoreThanOneUseOtherThanLLVMUsed(GlobalValue &V,
2811 const LLVMUsed &U) {
2812 unsigned N = 2;
2813 assert((!U.usedCount(&V) || !U.compilerUsedCount(&V)) &&
2814 "We should have removed the duplicated "
Rafael Espindola9aadcc42013-07-19 18:44:51 +00002815 "element from llvm.compiler.used");
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002816 if (U.usedCount(&V) || U.compilerUsedCount(&V))
2817 ++N;
2818 return V.hasNUsesOrMore(N);
2819}
2820
2821static bool mayHaveOtherReferences(GlobalAlias &GA, const LLVMUsed &U) {
2822 if (!GA.hasLocalLinkage())
2823 return true;
2824
2825 return U.usedCount(&GA) || U.compilerUsedCount(&GA);
2826}
2827
2828static bool hasUsesToReplace(GlobalAlias &GA, LLVMUsed &U, bool &RenameTarget) {
2829 RenameTarget = false;
Rafael Espindola00752162013-05-09 17:22:59 +00002830 bool Ret = false;
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002831 if (hasUseOtherThanLLVMUsed(GA, U))
Rafael Espindola00752162013-05-09 17:22:59 +00002832 Ret = true;
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002833
2834 // If the alias is externally visible, we may still be able to simplify it.
2835 if (!mayHaveOtherReferences(GA, U))
2836 return Ret;
2837
2838 // If the aliasee has internal linkage, give it the name and linkage
2839 // of the alias, and delete the alias. This turns:
2840 // define internal ... @f(...)
2841 // @a = alias ... @f
2842 // into:
2843 // define ... @a(...)
2844 Constant *Aliasee = GA.getAliasee();
2845 GlobalValue *Target = cast<GlobalValue>(Aliasee->stripPointerCasts());
2846 if (!Target->hasLocalLinkage())
2847 return Ret;
2848
2849 // Do not perform the transform if multiple aliases potentially target the
2850 // aliasee. This check also ensures that it is safe to replace the section
2851 // and other attributes of the aliasee with those of the alias.
2852 if (hasMoreThanOneUseOtherThanLLVMUsed(*Target, U))
2853 return Ret;
2854
2855 RenameTarget = true;
2856 return true;
Rafael Espindola00752162013-05-09 17:22:59 +00002857}
2858
Duncan Sandsed722832009-03-06 10:21:56 +00002859bool GlobalOpt::OptimizeGlobalAliases(Module &M) {
Anton Korobeynikova9b60ee2008-09-09 19:04:59 +00002860 bool Changed = false;
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002861 LLVMUsed Used(M);
2862
2863 for (SmallPtrSet<GlobalValue *, 8>::iterator I = Used.usedBegin(),
2864 E = Used.usedEnd();
2865 I != E; ++I)
2866 Used.compilerUsedErase(*I);
Anton Korobeynikova9b60ee2008-09-09 19:04:59 +00002867
Duncan Sands0bcf0852009-01-07 20:01:06 +00002868 for (Module::alias_iterator I = M.alias_begin(), E = M.alias_end();
Duncan Sandsb3f27882009-02-15 09:56:08 +00002869 I != E;) {
2870 Module::alias_iterator J = I++;
Duncan Sandsed722832009-03-06 10:21:56 +00002871 // Aliases without names cannot be referenced outside this module.
David Majnemer5c921152014-07-01 15:26:50 +00002872 if (!J->hasName() && !J->isDeclaration() && !J->hasLocalLinkage())
Duncan Sandsed722832009-03-06 10:21:56 +00002873 J->setLinkage(GlobalValue::InternalLinkage);
Duncan Sandsb3f27882009-02-15 09:56:08 +00002874 // If the aliasee may change at link time, nothing can be done - bail out.
2875 if (J->mayBeOverridden())
Anton Korobeynikova9b60ee2008-09-09 19:04:59 +00002876 continue;
2877
Duncan Sandsb3f27882009-02-15 09:56:08 +00002878 Constant *Aliasee = J->getAliasee();
David Majnemer0e2cc2a2014-07-01 00:30:56 +00002879 GlobalValue *Target = dyn_cast<GlobalValue>(Aliasee->stripPointerCasts());
2880 // We can't trivially replace the alias with the aliasee if the aliasee is
2881 // non-trivial in some way.
2882 // TODO: Try to handle non-zero GEPs of local aliasees.
2883 if (!Target)
2884 continue;
Duncan Sands7a1db332009-02-18 17:55:38 +00002885 Target->removeDeadConstantUsers();
Duncan Sandsb3f27882009-02-15 09:56:08 +00002886
2887 // Make all users of the alias use the aliasee instead.
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002888 bool RenameTarget;
2889 if (!hasUsesToReplace(*J, Used, RenameTarget))
Rafael Espindola00752162013-05-09 17:22:59 +00002890 continue;
Duncan Sandsb3f27882009-02-15 09:56:08 +00002891
Rafael Espindola6b238632014-05-16 19:35:39 +00002892 J->replaceAllUsesWith(ConstantExpr::getBitCast(Aliasee, J->getType()));
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002893 ++NumAliasesResolved;
2894 Changed = true;
Duncan Sandsb3f27882009-02-15 09:56:08 +00002895
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002896 if (RenameTarget) {
Duncan Sands6a3df7b2009-12-08 10:10:20 +00002897 // Give the aliasee the name, linkage and other attributes of the alias.
2898 Target->takeName(J);
2899 Target->setLinkage(J->getLinkage());
Reid Kleckner22b19da2014-02-13 02:18:36 +00002900 Target->setVisibility(J->getVisibility());
2901 Target->setDLLStorageClass(J->getDLLStorageClass());
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002902
2903 if (Used.usedErase(J))
2904 Used.usedInsert(Target);
2905
2906 if (Used.compilerUsedErase(J))
2907 Used.compilerUsedInsert(Target);
Rafael Espindola8d304802013-06-12 16:45:47 +00002908 } else if (mayHaveOtherReferences(*J, Used))
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002909 continue;
2910
Duncan Sandsb3f27882009-02-15 09:56:08 +00002911 // Delete the alias.
2912 M.getAliasList().erase(J);
2913 ++NumAliasesRemoved;
2914 Changed = true;
Anton Korobeynikova9b60ee2008-09-09 19:04:59 +00002915 }
2916
Rafael Espindolaa82555c2013-06-11 17:48:06 +00002917 Used.syncVariablesAndSets();
2918
Anton Korobeynikova9b60ee2008-09-09 19:04:59 +00002919 return Changed;
2920}
Chris Lattner41b6a5a2005-09-26 01:43:45 +00002921
Nick Lewycky4b273cb2012-02-12 02:15:20 +00002922static Function *FindCXAAtExit(Module &M, TargetLibraryInfo *TLI) {
2923 if (!TLI->has(LibFunc::cxa_atexit))
Craig Topperf40110f2014-04-25 05:29:35 +00002924 return nullptr;
Nick Lewycky4b273cb2012-02-12 02:15:20 +00002925
2926 Function *Fn = M.getFunction(TLI->getName(LibFunc::cxa_atexit));
Jakub Staszak9525a772012-12-06 21:57:16 +00002927
Anders Carlssonee6bc702011-03-20 17:59:11 +00002928 if (!Fn)
Craig Topperf40110f2014-04-25 05:29:35 +00002929 return nullptr;
Nick Lewycky4b273cb2012-02-12 02:15:20 +00002930
Chris Lattner229907c2011-07-18 04:54:35 +00002931 FunctionType *FTy = Fn->getFunctionType();
Jakub Staszak9525a772012-12-06 21:57:16 +00002932
2933 // Checking that the function has the right return type, the right number of
Anders Carlsson48a44912011-03-20 19:51:13 +00002934 // parameters and that they all have pointer types should be enough.
2935 if (!FTy->getReturnType()->isIntegerTy() ||
2936 FTy->getNumParams() != 3 ||
Anders Carlssonee6bc702011-03-20 17:59:11 +00002937 !FTy->getParamType(0)->isPointerTy() ||
2938 !FTy->getParamType(1)->isPointerTy() ||
2939 !FTy->getParamType(2)->isPointerTy())
Craig Topperf40110f2014-04-25 05:29:35 +00002940 return nullptr;
Anders Carlssonee6bc702011-03-20 17:59:11 +00002941
2942 return Fn;
2943}
2944
2945/// cxxDtorIsEmpty - Returns whether the given function is an empty C++
2946/// destructor and can therefore be eliminated.
2947/// Note that we assume that other optimization passes have already simplified
2948/// the code so we only look for a function with a single basic block, where
Benjamin Kramer1a4695a2012-02-09 16:28:15 +00002949/// the only allowed instructions are 'ret', 'call' to an empty C++ dtor and
2950/// other side-effect free instructions.
Anders Carlssonfcec2f52011-03-20 20:16:43 +00002951static bool cxxDtorIsEmpty(const Function &Fn,
2952 SmallPtrSet<const Function *, 8> &CalledFunctions) {
Anders Carlsson48a44912011-03-20 19:51:13 +00002953 // FIXME: We could eliminate C++ destructors if they're readonly/readnone and
Nick Lewyckyd0781832011-03-21 02:26:01 +00002954 // nounwind, but that doesn't seem worth doing.
Anders Carlsson48a44912011-03-20 19:51:13 +00002955 if (Fn.isDeclaration())
2956 return false;
Anders Carlssonee6bc702011-03-20 17:59:11 +00002957
2958 if (++Fn.begin() != Fn.end())
2959 return false;
2960
2961 const BasicBlock &EntryBlock = Fn.getEntryBlock();
2962 for (BasicBlock::const_iterator I = EntryBlock.begin(), E = EntryBlock.end();
2963 I != E; ++I) {
2964 if (const CallInst *CI = dyn_cast<CallInst>(I)) {
Anders Carlsson4dd420f2011-03-21 14:54:40 +00002965 // Ignore debug intrinsics.
2966 if (isa<DbgInfoIntrinsic>(CI))
2967 continue;
2968
Anders Carlssonee6bc702011-03-20 17:59:11 +00002969 const Function *CalledFn = CI->getCalledFunction();
2970
2971 if (!CalledFn)
2972 return false;
2973
Anders Carlsson1cc80732011-03-22 03:21:01 +00002974 SmallPtrSet<const Function *, 8> NewCalledFunctions(CalledFunctions);
2975
Anders Carlsson48a44912011-03-20 19:51:13 +00002976 // Don't treat recursive functions as empty.
Anders Carlsson1cc80732011-03-22 03:21:01 +00002977 if (!NewCalledFunctions.insert(CalledFn))
Anders Carlsson48a44912011-03-20 19:51:13 +00002978 return false;
2979
Anders Carlsson1cc80732011-03-22 03:21:01 +00002980 if (!cxxDtorIsEmpty(*CalledFn, NewCalledFunctions))
Anders Carlssonee6bc702011-03-20 17:59:11 +00002981 return false;
2982 } else if (isa<ReturnInst>(*I))
Benjamin Kramer487a3962012-02-09 14:26:06 +00002983 return true; // We're done.
2984 else if (I->mayHaveSideEffects())
2985 return false; // Destructor with side effects, bail.
Anders Carlssonee6bc702011-03-20 17:59:11 +00002986 }
2987
2988 return false;
2989}
2990
2991bool GlobalOpt::OptimizeEmptyGlobalCXXDtors(Function *CXAAtExitFn) {
2992 /// Itanium C++ ABI p3.3.5:
2993 ///
2994 /// After constructing a global (or local static) object, that will require
2995 /// destruction on exit, a termination function is registered as follows:
2996 ///
2997 /// extern "C" int __cxa_atexit ( void (*f)(void *), void *p, void *d );
2998 ///
2999 /// This registration, e.g. __cxa_atexit(f,p,d), is intended to cause the
3000 /// call f(p) when DSO d is unloaded, before all such termination calls
3001 /// registered before this one. It returns zero if registration is
Nick Lewyckyd0781832011-03-21 02:26:01 +00003002 /// successful, nonzero on failure.
Anders Carlssonee6bc702011-03-20 17:59:11 +00003003
3004 // This pass will look for calls to __cxa_atexit where the function is trivial
3005 // and remove them.
3006 bool Changed = false;
3007
Chandler Carruthcdf47882014-03-09 03:16:01 +00003008 for (auto I = CXAAtExitFn->user_begin(), E = CXAAtExitFn->user_end();
3009 I != E;) {
Anders Carlsson336fd902011-03-20 20:21:33 +00003010 // We're only interested in calls. Theoretically, we could handle invoke
3011 // instructions as well, but neither llvm-gcc nor clang generate invokes
3012 // to __cxa_atexit.
Anders Carlsson4dd420f2011-03-21 14:54:40 +00003013 CallInst *CI = dyn_cast<CallInst>(*I++);
3014 if (!CI)
Anders Carlsson336fd902011-03-20 20:21:33 +00003015 continue;
3016
Jakub Staszak9525a772012-12-06 21:57:16 +00003017 Function *DtorFn =
Anders Carlsson4dd420f2011-03-21 14:54:40 +00003018 dyn_cast<Function>(CI->getArgOperand(0)->stripPointerCasts());
Anders Carlssonee6bc702011-03-20 17:59:11 +00003019 if (!DtorFn)
3020 continue;
3021
Anders Carlssonfcec2f52011-03-20 20:16:43 +00003022 SmallPtrSet<const Function *, 8> CalledFunctions;
3023 if (!cxxDtorIsEmpty(*DtorFn, CalledFunctions))
Anders Carlssonee6bc702011-03-20 17:59:11 +00003024 continue;
3025
3026 // Just remove the call.
Anders Carlsson4dd420f2011-03-21 14:54:40 +00003027 CI->replaceAllUsesWith(Constant::getNullValue(CI->getType()));
3028 CI->eraseFromParent();
Anders Carlsson48a44912011-03-20 19:51:13 +00003029
Anders Carlssonee6bc702011-03-20 17:59:11 +00003030 ++NumCXXDtorsRemoved;
3031
3032 Changed |= true;
3033 }
3034
3035 return Changed;
3036}
3037
Chris Lattner25db5802004-10-07 04:16:33 +00003038bool GlobalOpt::runOnModule(Module &M) {
3039 bool Changed = false;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00003040
Rafael Espindola93512512014-02-25 17:30:31 +00003041 DataLayoutPass *DLP = getAnalysisIfAvailable<DataLayoutPass>();
Craig Topperf40110f2014-04-25 05:29:35 +00003042 DL = DLP ? &DLP->getDataLayout() : nullptr;
Nick Lewycky4b273cb2012-02-12 02:15:20 +00003043 TLI = &getAnalysis<TargetLibraryInfo>();
Nick Lewyckycf6aae62012-02-12 01:13:18 +00003044
Chris Lattner25db5802004-10-07 04:16:33 +00003045 bool LocalChange = true;
3046 while (LocalChange) {
3047 LocalChange = false;
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00003048
Chris Lattner41b6a5a2005-09-26 01:43:45 +00003049 // Delete functions that are trivially dead, ccc -> fastcc
3050 LocalChange |= OptimizeFunctions(M);
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00003051
Chris Lattner41b6a5a2005-09-26 01:43:45 +00003052 // Optimize global_ctors list.
Richard Smithc167d652014-05-06 01:44:26 +00003053 LocalChange |= optimizeGlobalCtorsList(M, [&](Function *F) {
3054 return EvaluateStaticConstructor(F, DL, TLI);
3055 });
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00003056
Chris Lattner41b6a5a2005-09-26 01:43:45 +00003057 // Optimize non-address-taken globals.
3058 LocalChange |= OptimizeGlobalVars(M);
Anton Korobeynikova9b60ee2008-09-09 19:04:59 +00003059
3060 // Resolve aliases, when possible.
Duncan Sandsed722832009-03-06 10:21:56 +00003061 LocalChange |= OptimizeGlobalAliases(M);
Anders Carlssonee6bc702011-03-20 17:59:11 +00003062
Manman Renb3c52fb2013-05-14 21:52:44 +00003063 // Try to remove trivial global destructors if they are not removed
3064 // already.
3065 Function *CXAAtExitFn = FindCXAAtExit(M, TLI);
Anders Carlssonee6bc702011-03-20 17:59:11 +00003066 if (CXAAtExitFn)
3067 LocalChange |= OptimizeEmptyGlobalCXXDtors(CXAAtExitFn);
3068
Anton Korobeynikova9b60ee2008-09-09 19:04:59 +00003069 Changed |= LocalChange;
Chris Lattner25db5802004-10-07 04:16:33 +00003070 }
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00003071
Chris Lattner41b6a5a2005-09-26 01:43:45 +00003072 // TODO: Move all global ctors functions to the end of the module for code
3073 // layout.
Mikhail Glushenkovcf2afe02010-10-18 21:16:00 +00003074
Chris Lattner25db5802004-10-07 04:16:33 +00003075 return Changed;
3076}