Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 1 | //===--- CodeGenFunction.cpp - Emit LLVM Code from ASTs for a Function ----===// |
| 2 | // |
| 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 0bc735f | 2007-12-29 19:59:25 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 7 | // |
| 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This coordinates the per-function state used while generating code. |
| 11 | // |
| 12 | //===----------------------------------------------------------------------===// |
| 13 | |
| 14 | #include "CodeGenFunction.h" |
| 15 | #include "CodeGenModule.h" |
Eli Friedman | 3f2af10 | 2008-05-22 01:40:10 +0000 | [diff] [blame] | 16 | #include "CGDebugInfo.h" |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 17 | #include "clang/Basic/TargetInfo.h" |
Chris Lattner | 31a0984 | 2008-11-12 08:04:58 +0000 | [diff] [blame] | 18 | #include "clang/AST/APValue.h" |
Daniel Dunbar | de7fb84 | 2008-08-11 05:00:27 +0000 | [diff] [blame] | 19 | #include "clang/AST/ASTContext.h" |
Daniel Dunbar | c4a1dea | 2008-08-11 05:35:13 +0000 | [diff] [blame] | 20 | #include "clang/AST/Decl.h" |
Anders Carlsson | 2b77ba8 | 2009-04-04 20:47:02 +0000 | [diff] [blame] | 21 | #include "clang/AST/DeclCXX.h" |
Devang Patel | d9363c3 | 2007-09-28 21:49:18 +0000 | [diff] [blame] | 22 | #include "llvm/Support/CFG.h" |
Mike Stump | 4e7a1f7 | 2009-02-21 20:00:35 +0000 | [diff] [blame] | 23 | #include "llvm/Target/TargetData.h" |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 24 | using namespace clang; |
| 25 | using namespace CodeGen; |
| 26 | |
| 27 | CodeGenFunction::CodeGenFunction(CodeGenModule &cgm) |
Mike Stump | a4f668f | 2009-03-06 01:33:24 +0000 | [diff] [blame] | 28 | : BlockFunction(cgm, *this, Builder), CGM(cgm), |
| 29 | Target(CGM.getContext().Target), |
Owen Anderson | aac8705 | 2009-07-08 20:52:20 +0000 | [diff] [blame] | 30 | Builder(cgm.getModule().getContext()), |
Anders Carlsson | 2b77ba8 | 2009-04-04 20:47:02 +0000 | [diff] [blame] | 31 | DebugInfo(0), SwitchInsn(0), CaseRangeBlock(0), InvokeDest(0), |
| 32 | CXXThisDecl(0) { |
Mike Stump | 4e7a1f7 | 2009-02-21 20:00:35 +0000 | [diff] [blame] | 33 | LLVMIntTy = ConvertType(getContext().IntTy); |
| 34 | LLVMPointerWidth = Target.getPointerWidth(0); |
Chris Lattner | 4111024 | 2008-06-17 18:05:57 +0000 | [diff] [blame] | 35 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 36 | |
| 37 | ASTContext &CodeGenFunction::getContext() const { |
| 38 | return CGM.getContext(); |
| 39 | } |
| 40 | |
| 41 | |
| 42 | llvm::BasicBlock *CodeGenFunction::getBasicBlockForLabel(const LabelStmt *S) { |
| 43 | llvm::BasicBlock *&BB = LabelMap[S]; |
| 44 | if (BB) return BB; |
| 45 | |
| 46 | // Create, but don't insert, the new block. |
Daniel Dunbar | 55e8742 | 2008-11-11 02:29:29 +0000 | [diff] [blame] | 47 | return BB = createBasicBlock(S->getName()); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 48 | } |
| 49 | |
Daniel Dunbar | 0096acf | 2009-02-25 19:24:29 +0000 | [diff] [blame] | 50 | llvm::Value *CodeGenFunction::GetAddrOfLocalVar(const VarDecl *VD) { |
| 51 | llvm::Value *Res = LocalDeclMap[VD]; |
| 52 | assert(Res && "Invalid argument to GetAddrOfLocalVar(), no decl!"); |
| 53 | return Res; |
Lauro Ramos Venancio | 8137335 | 2008-02-26 21:41:45 +0000 | [diff] [blame] | 54 | } |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 55 | |
Daniel Dunbar | 0096acf | 2009-02-25 19:24:29 +0000 | [diff] [blame] | 56 | llvm::Constant * |
| 57 | CodeGenFunction::GetAddrOfStaticLocalVar(const VarDecl *BVD) { |
| 58 | return cast<llvm::Constant>(GetAddrOfLocalVar(BVD)); |
Anders Carlsson | dde0a94 | 2008-09-11 09:15:33 +0000 | [diff] [blame] | 59 | } |
| 60 | |
Daniel Dunbar | 8b1a343 | 2009-02-03 23:03:55 +0000 | [diff] [blame] | 61 | const llvm::Type *CodeGenFunction::ConvertTypeForMem(QualType T) { |
| 62 | return CGM.getTypes().ConvertTypeForMem(T); |
| 63 | } |
| 64 | |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 65 | const llvm::Type *CodeGenFunction::ConvertType(QualType T) { |
| 66 | return CGM.getTypes().ConvertType(T); |
| 67 | } |
| 68 | |
| 69 | bool CodeGenFunction::hasAggregateLLVMType(QualType T) { |
Mike Stump | f5408fe | 2009-05-16 07:57:57 +0000 | [diff] [blame] | 70 | // FIXME: Use positive checks instead of negative ones to be more robust in |
| 71 | // the face of extension. |
Daniel Dunbar | 8958891 | 2009-02-26 20:52:22 +0000 | [diff] [blame] | 72 | return !T->hasPointerRepresentation() &&!T->isRealType() && |
| 73 | !T->isVoidType() && !T->isVectorType() && !T->isFunctionType() && |
Daniel Dunbar | a782ca7 | 2009-01-09 02:44:18 +0000 | [diff] [blame] | 74 | !T->isBlockPointerType(); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 75 | } |
| 76 | |
Daniel Dunbar | 1c1d607 | 2009-01-26 23:27:52 +0000 | [diff] [blame] | 77 | void CodeGenFunction::EmitReturnBlock() { |
| 78 | // For cleanliness, we try to avoid emitting the return block for |
| 79 | // simple cases. |
| 80 | llvm::BasicBlock *CurBB = Builder.GetInsertBlock(); |
| 81 | |
| 82 | if (CurBB) { |
| 83 | assert(!CurBB->getTerminator() && "Unexpected terminated block."); |
| 84 | |
Daniel Dunbar | 96e18b0 | 2009-07-19 08:24:34 +0000 | [diff] [blame^] | 85 | // We have a valid insert point, reuse it if it is empty or there are no |
| 86 | // explicit jumps to the return block. |
| 87 | if (CurBB->empty() || ReturnBlock->use_empty()) { |
| 88 | ReturnBlock->replaceAllUsesWith(CurBB); |
Daniel Dunbar | 1c1d607 | 2009-01-26 23:27:52 +0000 | [diff] [blame] | 89 | delete ReturnBlock; |
Daniel Dunbar | 96e18b0 | 2009-07-19 08:24:34 +0000 | [diff] [blame^] | 90 | } else |
Daniel Dunbar | 1c1d607 | 2009-01-26 23:27:52 +0000 | [diff] [blame] | 91 | EmitBlock(ReturnBlock); |
| 92 | return; |
| 93 | } |
| 94 | |
| 95 | // Otherwise, if the return block is the target of a single direct |
| 96 | // branch then we can just put the code in that block instead. This |
| 97 | // cleans up functions which started with a unified return block. |
| 98 | if (ReturnBlock->hasOneUse()) { |
| 99 | llvm::BranchInst *BI = |
| 100 | dyn_cast<llvm::BranchInst>(*ReturnBlock->use_begin()); |
| 101 | if (BI && BI->isUnconditional() && BI->getSuccessor(0) == ReturnBlock) { |
| 102 | // Reset insertion point and delete the branch. |
| 103 | Builder.SetInsertPoint(BI->getParent()); |
| 104 | BI->eraseFromParent(); |
| 105 | delete ReturnBlock; |
| 106 | return; |
| 107 | } |
| 108 | } |
| 109 | |
Mike Stump | f5408fe | 2009-05-16 07:57:57 +0000 | [diff] [blame] | 110 | // FIXME: We are at an unreachable point, there is no reason to emit the block |
| 111 | // unless it has uses. However, we still need a place to put the debug |
| 112 | // region.end for now. |
Daniel Dunbar | 1c1d607 | 2009-01-26 23:27:52 +0000 | [diff] [blame] | 113 | |
| 114 | EmitBlock(ReturnBlock); |
| 115 | } |
| 116 | |
Daniel Dunbar | af05bb9 | 2008-08-26 08:29:31 +0000 | [diff] [blame] | 117 | void CodeGenFunction::FinishFunction(SourceLocation EndLoc) { |
Daniel Dunbar | 0ffb125 | 2008-08-04 16:51:22 +0000 | [diff] [blame] | 118 | // Finish emission of indirect switches. |
| 119 | EmitIndirectSwitches(); |
| 120 | |
Chris Lattner | da13870 | 2007-07-16 21:28:45 +0000 | [diff] [blame] | 121 | assert(BreakContinueStack.empty() && |
| 122 | "mismatched push/pop in break/continue stack!"); |
Anders Carlsson | bd6fa3d | 2009-02-08 00:16:35 +0000 | [diff] [blame] | 123 | assert(BlockScopes.empty() && |
| 124 | "did not remove all blocks from block scope map!"); |
| 125 | assert(CleanupEntries.empty() && |
| 126 | "mismatched push/pop in cleanup stack!"); |
| 127 | |
Daniel Dunbar | 1c1d607 | 2009-01-26 23:27:52 +0000 | [diff] [blame] | 128 | // Emit function epilog (to return). |
| 129 | EmitReturnBlock(); |
Daniel Dunbar | f5bd45c | 2008-11-11 20:59:54 +0000 | [diff] [blame] | 130 | |
| 131 | // Emit debug descriptor for function end. |
Anders Carlsson | e896d98 | 2009-02-13 08:11:52 +0000 | [diff] [blame] | 132 | if (CGDebugInfo *DI = getDebugInfo()) { |
Daniel Dunbar | f5bd45c | 2008-11-11 20:59:54 +0000 | [diff] [blame] | 133 | DI->setLocation(EndLoc); |
| 134 | DI->EmitRegionEnd(CurFn, Builder); |
| 135 | } |
| 136 | |
Daniel Dunbar | 88b5396 | 2009-02-02 22:03:45 +0000 | [diff] [blame] | 137 | EmitFunctionEpilog(*CurFnInfo, ReturnValue); |
Daniel Dunbar | 5ca2084 | 2008-09-09 21:00:17 +0000 | [diff] [blame] | 138 | |
Chris Lattner | 5a2fa14 | 2007-12-02 06:32:24 +0000 | [diff] [blame] | 139 | // Remove the AllocaInsertPt instruction, which is just a convenience for us. |
Chris Lattner | 481769b | 2009-03-31 22:17:44 +0000 | [diff] [blame] | 140 | llvm::Instruction *Ptr = AllocaInsertPt; |
Chris Lattner | 5a2fa14 | 2007-12-02 06:32:24 +0000 | [diff] [blame] | 141 | AllocaInsertPt = 0; |
Chris Lattner | 481769b | 2009-03-31 22:17:44 +0000 | [diff] [blame] | 142 | Ptr->eraseFromParent(); |
Reid Spencer | 5f016e2 | 2007-07-11 17:01:13 +0000 | [diff] [blame] | 143 | } |
| 144 | |
Daniel Dunbar | 7c08651 | 2008-09-09 23:14:03 +0000 | [diff] [blame] | 145 | void CodeGenFunction::StartFunction(const Decl *D, QualType RetTy, |
| 146 | llvm::Function *Fn, |
Daniel Dunbar | 2284ac9 | 2008-10-18 18:22:23 +0000 | [diff] [blame] | 147 | const FunctionArgList &Args, |
| 148 | SourceLocation StartLoc) { |
Anders Carlsson | 4cc1a47 | 2009-02-09 20:20:56 +0000 | [diff] [blame] | 149 | DidCallStackSave = false; |
Chris Lattner | b5437d2 | 2009-04-23 05:30:27 +0000 | [diff] [blame] | 150 | CurCodeDecl = CurFuncDecl = D; |
Daniel Dunbar | 7c08651 | 2008-09-09 23:14:03 +0000 | [diff] [blame] | 151 | FnRetTy = RetTy; |
Daniel Dunbar | bd012ff | 2008-07-29 23:18:29 +0000 | [diff] [blame] | 152 | CurFn = Fn; |
Chris Lattner | 4111024 | 2008-06-17 18:05:57 +0000 | [diff] [blame] | 153 | assert(CurFn->isDeclaration() && "Function already has body?"); |
| 154 | |
Daniel Dunbar | 55e8742 | 2008-11-11 02:29:29 +0000 | [diff] [blame] | 155 | llvm::BasicBlock *EntryBB = createBasicBlock("entry", CurFn); |
Daniel Dunbar | 5ca2084 | 2008-09-09 21:00:17 +0000 | [diff] [blame] | 156 | |
Chris Lattner | 4111024 | 2008-06-17 18:05:57 +0000 | [diff] [blame] | 157 | // Create a marker to make it easy to insert allocas into the entryblock |
| 158 | // later. Don't create this with the builder, because we don't want it |
| 159 | // folded. |
Owen Anderson | a1cf15f | 2009-07-14 23:10:40 +0000 | [diff] [blame] | 160 | llvm::Value *Undef = VMContext.getUndef(llvm::Type::Int32Ty); |
Chris Lattner | f146684 | 2009-03-22 00:24:14 +0000 | [diff] [blame] | 161 | AllocaInsertPt = new llvm::BitCastInst(Undef, llvm::Type::Int32Ty, "", |
Chris Lattner | 4111024 | 2008-06-17 18:05:57 +0000 | [diff] [blame] | 162 | EntryBB); |
Chris Lattner | f146684 | 2009-03-22 00:24:14 +0000 | [diff] [blame] | 163 | if (Builder.isNamePreserving()) |
| 164 | AllocaInsertPt->setName("allocapt"); |
| 165 | |
Daniel Dunbar | 55e8742 | 2008-11-11 02:29:29 +0000 | [diff] [blame] | 166 | ReturnBlock = createBasicBlock("return"); |
Daniel Dunbar | 5ca2084 | 2008-09-09 21:00:17 +0000 | [diff] [blame] | 167 | ReturnValue = 0; |
Daniel Dunbar | 7c08651 | 2008-09-09 23:14:03 +0000 | [diff] [blame] | 168 | if (!RetTy->isVoidType()) |
| 169 | ReturnValue = CreateTempAlloca(ConvertType(RetTy), "retval"); |
Daniel Dunbar | 5ca2084 | 2008-09-09 21:00:17 +0000 | [diff] [blame] | 170 | |
Chris Lattner | 4111024 | 2008-06-17 18:05:57 +0000 | [diff] [blame] | 171 | Builder.SetInsertPoint(EntryBB); |
| 172 | |
Sanjiv Gupta | af99417 | 2008-07-04 11:04:26 +0000 | [diff] [blame] | 173 | // Emit subprogram debug descriptor. |
Daniel Dunbar | 7c08651 | 2008-09-09 23:14:03 +0000 | [diff] [blame] | 174 | // FIXME: The cast here is a huge hack. |
Anders Carlsson | e896d98 | 2009-02-13 08:11:52 +0000 | [diff] [blame] | 175 | if (CGDebugInfo *DI = getDebugInfo()) { |
Daniel Dunbar | 2284ac9 | 2008-10-18 18:22:23 +0000 | [diff] [blame] | 176 | DI->setLocation(StartLoc); |
| 177 | if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) { |
Douglas Gregor | 6ec3668 | 2009-02-18 23:53:56 +0000 | [diff] [blame] | 178 | DI->EmitFunctionStart(CGM.getMangledName(FD), RetTy, CurFn, Builder); |
Daniel Dunbar | 2284ac9 | 2008-10-18 18:22:23 +0000 | [diff] [blame] | 179 | } else { |
| 180 | // Just use LLVM function name. |
| 181 | DI->EmitFunctionStart(Fn->getName().c_str(), |
| 182 | RetTy, CurFn, Builder); |
Sanjiv Gupta | af99417 | 2008-07-04 11:04:26 +0000 | [diff] [blame] | 183 | } |
Sanjiv Gupta | af99417 | 2008-07-04 11:04:26 +0000 | [diff] [blame] | 184 | } |
| 185 | |
Daniel Dunbar | 88b5396 | 2009-02-02 22:03:45 +0000 | [diff] [blame] | 186 | // FIXME: Leaked. |
Daniel Dunbar | 541b63b | 2009-02-02 23:23:47 +0000 | [diff] [blame] | 187 | CurFnInfo = &CGM.getTypes().getFunctionInfo(FnRetTy, Args); |
Daniel Dunbar | 88b5396 | 2009-02-02 22:03:45 +0000 | [diff] [blame] | 188 | EmitFunctionProlog(*CurFnInfo, CurFn, Args); |
Anders Carlsson | 751358f | 2008-12-20 21:28:43 +0000 | [diff] [blame] | 189 | |
| 190 | // If any of the arguments have a variably modified type, make sure to |
| 191 | // emit the type size. |
| 192 | for (FunctionArgList::const_iterator i = Args.begin(), e = Args.end(); |
| 193 | i != e; ++i) { |
| 194 | QualType Ty = i->second; |
| 195 | |
| 196 | if (Ty->isVariablyModifiedType()) |
| 197 | EmitVLASize(Ty); |
| 198 | } |
Daniel Dunbar | 7c08651 | 2008-09-09 23:14:03 +0000 | [diff] [blame] | 199 | } |
Eli Friedman | eb4b705 | 2008-08-25 21:31:01 +0000 | [diff] [blame] | 200 | |
Daniel Dunbar | 7c08651 | 2008-09-09 23:14:03 +0000 | [diff] [blame] | 201 | void CodeGenFunction::GenerateCode(const FunctionDecl *FD, |
| 202 | llvm::Function *Fn) { |
Anders Carlsson | e896d98 | 2009-02-13 08:11:52 +0000 | [diff] [blame] | 203 | // Check if we should generate debug info for this function. |
Argyrios Kyrtzidis | 40b598e | 2009-06-30 02:34:44 +0000 | [diff] [blame] | 204 | if (CGM.getDebugInfo() && !FD->hasAttr<NodebugAttr>()) |
Anders Carlsson | e896d98 | 2009-02-13 08:11:52 +0000 | [diff] [blame] | 205 | DebugInfo = CGM.getDebugInfo(); |
| 206 | |
Daniel Dunbar | 7c08651 | 2008-09-09 23:14:03 +0000 | [diff] [blame] | 207 | FunctionArgList Args; |
Anders Carlsson | 2b77ba8 | 2009-04-04 20:47:02 +0000 | [diff] [blame] | 208 | |
| 209 | if (const CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(FD)) { |
| 210 | if (MD->isInstance()) { |
| 211 | // Create the implicit 'this' decl. |
| 212 | // FIXME: I'm not entirely sure I like using a fake decl just for code |
| 213 | // generation. Maybe we can come up with a better way? |
| 214 | CXXThisDecl = ImplicitParamDecl::Create(getContext(), 0, SourceLocation(), |
| 215 | &getContext().Idents.get("this"), |
| 216 | MD->getThisType(getContext())); |
| 217 | Args.push_back(std::make_pair(CXXThisDecl, CXXThisDecl->getType())); |
| 218 | } |
| 219 | } |
| 220 | |
Eli Friedman | eb4b705 | 2008-08-25 21:31:01 +0000 | [diff] [blame] | 221 | if (FD->getNumParams()) { |
Douglas Gregor | 72564e7 | 2009-02-26 23:50:07 +0000 | [diff] [blame] | 222 | const FunctionProtoType* FProto = FD->getType()->getAsFunctionProtoType(); |
Eli Friedman | eb4b705 | 2008-08-25 21:31:01 +0000 | [diff] [blame] | 223 | assert(FProto && "Function def must have prototype!"); |
Daniel Dunbar | 7c08651 | 2008-09-09 23:14:03 +0000 | [diff] [blame] | 224 | |
| 225 | for (unsigned i = 0, e = FD->getNumParams(); i != e; ++i) |
| 226 | Args.push_back(std::make_pair(FD->getParamDecl(i), |
| 227 | FProto->getArgType(i))); |
Chris Lattner | 4111024 | 2008-06-17 18:05:57 +0000 | [diff] [blame] | 228 | } |
Daniel Dunbar | af05bb9 | 2008-08-26 08:29:31 +0000 | [diff] [blame] | 229 | |
Sebastian Redl | d3a413d | 2009-04-26 20:35:05 +0000 | [diff] [blame] | 230 | // FIXME: Support CXXTryStmt here, too. |
Argyrios Kyrtzidis | 6fb0aee | 2009-06-30 02:35:26 +0000 | [diff] [blame] | 231 | if (const CompoundStmt *S = FD->getCompoundBody()) { |
Sebastian Redl | d3a413d | 2009-04-26 20:35:05 +0000 | [diff] [blame] | 232 | StartFunction(FD, FD->getResultType(), Fn, Args, S->getLBracLoc()); |
| 233 | EmitStmt(S); |
| 234 | FinishFunction(S->getRBracLoc()); |
| 235 | } |
Daniel Dunbar | 7c08651 | 2008-09-09 23:14:03 +0000 | [diff] [blame] | 236 | |
Anders Carlsson | 2b77ba8 | 2009-04-04 20:47:02 +0000 | [diff] [blame] | 237 | // Destroy the 'this' declaration. |
| 238 | if (CXXThisDecl) |
| 239 | CXXThisDecl->Destroy(getContext()); |
Chris Lattner | 4111024 | 2008-06-17 18:05:57 +0000 | [diff] [blame] | 240 | } |
| 241 | |
Chris Lattner | 0946ccd | 2008-11-11 07:41:27 +0000 | [diff] [blame] | 242 | /// ContainsLabel - Return true if the statement contains a label in it. If |
| 243 | /// this statement is not executed normally, it not containing a label means |
| 244 | /// that we can just remove the code. |
| 245 | bool CodeGenFunction::ContainsLabel(const Stmt *S, bool IgnoreCaseStmts) { |
| 246 | // Null statement, not a label! |
| 247 | if (S == 0) return false; |
| 248 | |
| 249 | // If this is a label, we have to emit the code, consider something like: |
| 250 | // if (0) { ... foo: bar(); } goto foo; |
| 251 | if (isa<LabelStmt>(S)) |
| 252 | return true; |
| 253 | |
| 254 | // If this is a case/default statement, and we haven't seen a switch, we have |
| 255 | // to emit the code. |
| 256 | if (isa<SwitchCase>(S) && !IgnoreCaseStmts) |
| 257 | return true; |
| 258 | |
| 259 | // If this is a switch statement, we want to ignore cases below it. |
| 260 | if (isa<SwitchStmt>(S)) |
| 261 | IgnoreCaseStmts = true; |
| 262 | |
| 263 | // Scan subexpressions for verboten labels. |
| 264 | for (Stmt::const_child_iterator I = S->child_begin(), E = S->child_end(); |
| 265 | I != E; ++I) |
| 266 | if (ContainsLabel(*I, IgnoreCaseStmts)) |
| 267 | return true; |
| 268 | |
| 269 | return false; |
| 270 | } |
| 271 | |
Chris Lattner | 31a0984 | 2008-11-12 08:04:58 +0000 | [diff] [blame] | 272 | |
| 273 | /// ConstantFoldsToSimpleInteger - If the sepcified expression does not fold to |
| 274 | /// a constant, or if it does but contains a label, return 0. If it constant |
| 275 | /// folds to 'true' and does not contain a label, return 1, if it constant folds |
| 276 | /// to 'false' and does not contain a label, return -1. |
| 277 | int CodeGenFunction::ConstantFoldsToSimpleInteger(const Expr *Cond) { |
Daniel Dunbar | 36bc14c | 2008-11-12 22:37:10 +0000 | [diff] [blame] | 278 | // FIXME: Rename and handle conversion of other evaluatable things |
| 279 | // to bool. |
Anders Carlsson | 64712f1 | 2008-12-01 02:46:24 +0000 | [diff] [blame] | 280 | Expr::EvalResult Result; |
| 281 | if (!Cond->Evaluate(Result, getContext()) || !Result.Val.isInt() || |
| 282 | Result.HasSideEffects) |
Anders Carlsson | ef5a66d | 2008-11-22 22:32:07 +0000 | [diff] [blame] | 283 | return 0; // Not foldable, not integer or not fully evaluatable. |
Chris Lattner | 31a0984 | 2008-11-12 08:04:58 +0000 | [diff] [blame] | 284 | |
| 285 | if (CodeGenFunction::ContainsLabel(Cond)) |
| 286 | return 0; // Contains a label. |
| 287 | |
Anders Carlsson | 64712f1 | 2008-12-01 02:46:24 +0000 | [diff] [blame] | 288 | return Result.Val.getInt().getBoolValue() ? 1 : -1; |
Chris Lattner | 31a0984 | 2008-11-12 08:04:58 +0000 | [diff] [blame] | 289 | } |
| 290 | |
| 291 | |
| 292 | /// EmitBranchOnBoolExpr - Emit a branch on a boolean condition (e.g. for an if |
| 293 | /// statement) to the specified blocks. Based on the condition, this might try |
| 294 | /// to simplify the codegen of the conditional based on the branch. |
| 295 | /// |
| 296 | void CodeGenFunction::EmitBranchOnBoolExpr(const Expr *Cond, |
| 297 | llvm::BasicBlock *TrueBlock, |
| 298 | llvm::BasicBlock *FalseBlock) { |
| 299 | if (const ParenExpr *PE = dyn_cast<ParenExpr>(Cond)) |
| 300 | return EmitBranchOnBoolExpr(PE->getSubExpr(), TrueBlock, FalseBlock); |
| 301 | |
| 302 | if (const BinaryOperator *CondBOp = dyn_cast<BinaryOperator>(Cond)) { |
| 303 | // Handle X && Y in a condition. |
| 304 | if (CondBOp->getOpcode() == BinaryOperator::LAnd) { |
| 305 | // If we have "1 && X", simplify the code. "0 && X" would have constant |
| 306 | // folded if the case was simple enough. |
| 307 | if (ConstantFoldsToSimpleInteger(CondBOp->getLHS()) == 1) { |
| 308 | // br(1 && X) -> br(X). |
| 309 | return EmitBranchOnBoolExpr(CondBOp->getRHS(), TrueBlock, FalseBlock); |
| 310 | } |
| 311 | |
| 312 | // If we have "X && 1", simplify the code to use an uncond branch. |
| 313 | // "X && 0" would have been constant folded to 0. |
| 314 | if (ConstantFoldsToSimpleInteger(CondBOp->getRHS()) == 1) { |
| 315 | // br(X && 1) -> br(X). |
| 316 | return EmitBranchOnBoolExpr(CondBOp->getLHS(), TrueBlock, FalseBlock); |
| 317 | } |
| 318 | |
| 319 | // Emit the LHS as a conditional. If the LHS conditional is false, we |
| 320 | // want to jump to the FalseBlock. |
Daniel Dunbar | 9615ecb | 2008-11-13 01:38:36 +0000 | [diff] [blame] | 321 | llvm::BasicBlock *LHSTrue = createBasicBlock("land.lhs.true"); |
Chris Lattner | 31a0984 | 2008-11-12 08:04:58 +0000 | [diff] [blame] | 322 | EmitBranchOnBoolExpr(CondBOp->getLHS(), LHSTrue, FalseBlock); |
| 323 | EmitBlock(LHSTrue); |
| 324 | |
| 325 | EmitBranchOnBoolExpr(CondBOp->getRHS(), TrueBlock, FalseBlock); |
| 326 | return; |
| 327 | } else if (CondBOp->getOpcode() == BinaryOperator::LOr) { |
| 328 | // If we have "0 || X", simplify the code. "1 || X" would have constant |
| 329 | // folded if the case was simple enough. |
| 330 | if (ConstantFoldsToSimpleInteger(CondBOp->getLHS()) == -1) { |
| 331 | // br(0 || X) -> br(X). |
| 332 | return EmitBranchOnBoolExpr(CondBOp->getRHS(), TrueBlock, FalseBlock); |
| 333 | } |
| 334 | |
| 335 | // If we have "X || 0", simplify the code to use an uncond branch. |
| 336 | // "X || 1" would have been constant folded to 1. |
| 337 | if (ConstantFoldsToSimpleInteger(CondBOp->getRHS()) == -1) { |
| 338 | // br(X || 0) -> br(X). |
| 339 | return EmitBranchOnBoolExpr(CondBOp->getLHS(), TrueBlock, FalseBlock); |
| 340 | } |
| 341 | |
| 342 | // Emit the LHS as a conditional. If the LHS conditional is true, we |
| 343 | // want to jump to the TrueBlock. |
Daniel Dunbar | 9615ecb | 2008-11-13 01:38:36 +0000 | [diff] [blame] | 344 | llvm::BasicBlock *LHSFalse = createBasicBlock("lor.lhs.false"); |
Chris Lattner | 31a0984 | 2008-11-12 08:04:58 +0000 | [diff] [blame] | 345 | EmitBranchOnBoolExpr(CondBOp->getLHS(), TrueBlock, LHSFalse); |
| 346 | EmitBlock(LHSFalse); |
| 347 | |
| 348 | EmitBranchOnBoolExpr(CondBOp->getRHS(), TrueBlock, FalseBlock); |
| 349 | return; |
| 350 | } |
Chris Lattner | 552f4c4 | 2008-11-12 08:13:36 +0000 | [diff] [blame] | 351 | } |
| 352 | |
| 353 | if (const UnaryOperator *CondUOp = dyn_cast<UnaryOperator>(Cond)) { |
| 354 | // br(!x, t, f) -> br(x, f, t) |
| 355 | if (CondUOp->getOpcode() == UnaryOperator::LNot) |
| 356 | return EmitBranchOnBoolExpr(CondUOp->getSubExpr(), FalseBlock, TrueBlock); |
Chris Lattner | 31a0984 | 2008-11-12 08:04:58 +0000 | [diff] [blame] | 357 | } |
| 358 | |
Daniel Dunbar | 09b1489 | 2008-11-12 10:30:32 +0000 | [diff] [blame] | 359 | if (const ConditionalOperator *CondOp = dyn_cast<ConditionalOperator>(Cond)) { |
| 360 | // Handle ?: operator. |
| 361 | |
| 362 | // Just ignore GNU ?: extension. |
| 363 | if (CondOp->getLHS()) { |
| 364 | // br(c ? x : y, t, f) -> br(c, br(x, t, f), br(y, t, f)) |
| 365 | llvm::BasicBlock *LHSBlock = createBasicBlock("cond.true"); |
| 366 | llvm::BasicBlock *RHSBlock = createBasicBlock("cond.false"); |
| 367 | EmitBranchOnBoolExpr(CondOp->getCond(), LHSBlock, RHSBlock); |
| 368 | EmitBlock(LHSBlock); |
| 369 | EmitBranchOnBoolExpr(CondOp->getLHS(), TrueBlock, FalseBlock); |
| 370 | EmitBlock(RHSBlock); |
| 371 | EmitBranchOnBoolExpr(CondOp->getRHS(), TrueBlock, FalseBlock); |
| 372 | return; |
| 373 | } |
| 374 | } |
| 375 | |
Chris Lattner | 31a0984 | 2008-11-12 08:04:58 +0000 | [diff] [blame] | 376 | // Emit the code with the fully general case. |
| 377 | llvm::Value *CondV = EvaluateExprAsBool(Cond); |
| 378 | Builder.CreateCondBr(CondV, TrueBlock, FalseBlock); |
| 379 | } |
| 380 | |
Devang Patel | 88a981b | 2007-11-01 19:11:01 +0000 | [diff] [blame] | 381 | /// getCGRecordLayout - Return record layout info. |
| 382 | const CGRecordLayout *CodeGenFunction::getCGRecordLayout(CodeGenTypes &CGT, |
Chris Lattner | af31913 | 2008-02-05 06:55:31 +0000 | [diff] [blame] | 383 | QualType Ty) { |
Ted Kremenek | 35366a6 | 2009-07-17 17:50:17 +0000 | [diff] [blame] | 384 | const RecordType *RTy = Ty->getAsRecordType(); |
Chris Lattner | af31913 | 2008-02-05 06:55:31 +0000 | [diff] [blame] | 385 | assert (RTy && "Unexpected type. RecordType expected here."); |
Devang Patel | b84a06e | 2007-10-23 02:10:49 +0000 | [diff] [blame] | 386 | |
Chris Lattner | af31913 | 2008-02-05 06:55:31 +0000 | [diff] [blame] | 387 | return CGT.getCGRecordLayout(RTy->getDecl()); |
Devang Patel | b84a06e | 2007-10-23 02:10:49 +0000 | [diff] [blame] | 388 | } |
Chris Lattner | dc5e826 | 2007-12-02 01:43:38 +0000 | [diff] [blame] | 389 | |
Daniel Dunbar | 488e993 | 2008-08-16 00:56:44 +0000 | [diff] [blame] | 390 | /// ErrorUnsupported - Print out an error that codegen doesn't support the |
Chris Lattner | dc5e826 | 2007-12-02 01:43:38 +0000 | [diff] [blame] | 391 | /// specified stmt yet. |
Daniel Dunbar | 90df4b6 | 2008-09-04 03:43:08 +0000 | [diff] [blame] | 392 | void CodeGenFunction::ErrorUnsupported(const Stmt *S, const char *Type, |
| 393 | bool OmitOnError) { |
| 394 | CGM.ErrorUnsupported(S, Type, OmitOnError); |
Chris Lattner | dc5e826 | 2007-12-02 01:43:38 +0000 | [diff] [blame] | 395 | } |
| 396 | |
Daniel Dunbar | 0ffb125 | 2008-08-04 16:51:22 +0000 | [diff] [blame] | 397 | unsigned CodeGenFunction::GetIDForAddrOfLabel(const LabelStmt *L) { |
| 398 | // Use LabelIDs.size() as the new ID if one hasn't been assigned. |
| 399 | return LabelIDs.insert(std::make_pair(L, LabelIDs.size())).first->second; |
| 400 | } |
| 401 | |
Chris Lattner | 88207c9 | 2009-04-21 17:59:23 +0000 | [diff] [blame] | 402 | void CodeGenFunction::EmitMemSetToZero(llvm::Value *DestPtr, QualType Ty) { |
Owen Anderson | a1cf15f | 2009-07-14 23:10:40 +0000 | [diff] [blame] | 403 | const llvm::Type *BP = VMContext.getPointerTypeUnqual(llvm::Type::Int8Ty); |
Anders Carlsson | 3d8400d | 2008-08-30 19:51:14 +0000 | [diff] [blame] | 404 | if (DestPtr->getType() != BP) |
| 405 | DestPtr = Builder.CreateBitCast(DestPtr, BP, "tmp"); |
| 406 | |
| 407 | // Get size and alignment info for this aggregate. |
| 408 | std::pair<uint64_t, unsigned> TypeInfo = getContext().getTypeInfo(Ty); |
| 409 | |
Chris Lattner | 88207c9 | 2009-04-21 17:59:23 +0000 | [diff] [blame] | 410 | // Don't bother emitting a zero-byte memset. |
| 411 | if (TypeInfo.first == 0) |
| 412 | return; |
| 413 | |
Anders Carlsson | 3d8400d | 2008-08-30 19:51:14 +0000 | [diff] [blame] | 414 | // FIXME: Handle variable sized types. |
Owen Anderson | a1cf15f | 2009-07-14 23:10:40 +0000 | [diff] [blame] | 415 | const llvm::Type *IntPtr = VMContext.getIntegerType(LLVMPointerWidth); |
Anders Carlsson | 3d8400d | 2008-08-30 19:51:14 +0000 | [diff] [blame] | 416 | |
| 417 | Builder.CreateCall4(CGM.getMemSetFn(), DestPtr, |
Owen Anderson | 6924382 | 2009-07-13 04:10:07 +0000 | [diff] [blame] | 418 | getLLVMContext().getNullValue(llvm::Type::Int8Ty), |
Anders Carlsson | 3d8400d | 2008-08-30 19:51:14 +0000 | [diff] [blame] | 419 | // TypeInfo.first describes size in bits. |
Owen Anderson | a1cf15f | 2009-07-14 23:10:40 +0000 | [diff] [blame] | 420 | VMContext.getConstantInt(IntPtr, TypeInfo.first/8), |
| 421 | VMContext.getConstantInt(llvm::Type::Int32Ty, |
Anders Carlsson | 3d8400d | 2008-08-30 19:51:14 +0000 | [diff] [blame] | 422 | TypeInfo.second/8)); |
| 423 | } |
| 424 | |
Daniel Dunbar | 0ffb125 | 2008-08-04 16:51:22 +0000 | [diff] [blame] | 425 | void CodeGenFunction::EmitIndirectSwitches() { |
| 426 | llvm::BasicBlock *Default; |
| 427 | |
Daniel Dunbar | 76526a5 | 2008-08-04 17:24:44 +0000 | [diff] [blame] | 428 | if (IndirectSwitches.empty()) |
| 429 | return; |
| 430 | |
Daniel Dunbar | 0ffb125 | 2008-08-04 16:51:22 +0000 | [diff] [blame] | 431 | if (!LabelIDs.empty()) { |
| 432 | Default = getBasicBlockForLabel(LabelIDs.begin()->first); |
| 433 | } else { |
| 434 | // No possible targets for indirect goto, just emit an infinite |
| 435 | // loop. |
Daniel Dunbar | 55e8742 | 2008-11-11 02:29:29 +0000 | [diff] [blame] | 436 | Default = createBasicBlock("indirectgoto.loop", CurFn); |
Daniel Dunbar | 0ffb125 | 2008-08-04 16:51:22 +0000 | [diff] [blame] | 437 | llvm::BranchInst::Create(Default, Default); |
| 438 | } |
| 439 | |
| 440 | for (std::vector<llvm::SwitchInst*>::iterator i = IndirectSwitches.begin(), |
| 441 | e = IndirectSwitches.end(); i != e; ++i) { |
| 442 | llvm::SwitchInst *I = *i; |
| 443 | |
| 444 | I->setSuccessor(0, Default); |
| 445 | for (std::map<const LabelStmt*,unsigned>::iterator LI = LabelIDs.begin(), |
| 446 | LE = LabelIDs.end(); LI != LE; ++LI) { |
Owen Anderson | a1cf15f | 2009-07-14 23:10:40 +0000 | [diff] [blame] | 447 | I->addCase(VMContext.getConstantInt(llvm::Type::Int32Ty, |
Daniel Dunbar | 0ffb125 | 2008-08-04 16:51:22 +0000 | [diff] [blame] | 448 | LI->second), |
| 449 | getBasicBlockForLabel(LI->first)); |
| 450 | } |
| 451 | } |
| 452 | } |
Anders Carlsson | ddf7cac | 2008-11-04 05:30:00 +0000 | [diff] [blame] | 453 | |
Daniel Dunbar | d286f05 | 2009-07-19 06:58:07 +0000 | [diff] [blame] | 454 | llvm::Value *CodeGenFunction::GetVLASize(const VariableArrayType *VAT) { |
Anders Carlsson | dcc90d8 | 2008-12-12 07:19:02 +0000 | [diff] [blame] | 455 | llvm::Value *&SizeEntry = VLASizeMap[VAT]; |
Anders Carlsson | dcc90d8 | 2008-12-12 07:19:02 +0000 | [diff] [blame] | 456 | |
Anders Carlsson | f666b77 | 2008-12-20 20:27:15 +0000 | [diff] [blame] | 457 | assert(SizeEntry && "Did not emit size for type"); |
| 458 | return SizeEntry; |
| 459 | } |
Anders Carlsson | dcc90d8 | 2008-12-12 07:19:02 +0000 | [diff] [blame] | 460 | |
Daniel Dunbar | d286f05 | 2009-07-19 06:58:07 +0000 | [diff] [blame] | 461 | llvm::Value *CodeGenFunction::EmitVLASize(QualType Ty) { |
Anders Carlsson | 60d3541 | 2008-12-20 20:46:34 +0000 | [diff] [blame] | 462 | assert(Ty->isVariablyModifiedType() && |
| 463 | "Must pass variably modified type to EmitVLASizes!"); |
Anders Carlsson | f666b77 | 2008-12-20 20:27:15 +0000 | [diff] [blame] | 464 | |
Daniel Dunbar | d286f05 | 2009-07-19 06:58:07 +0000 | [diff] [blame] | 465 | EnsureInsertPoint(); |
| 466 | |
Anders Carlsson | 60d3541 | 2008-12-20 20:46:34 +0000 | [diff] [blame] | 467 | if (const VariableArrayType *VAT = getContext().getAsVariableArrayType(Ty)) { |
| 468 | llvm::Value *&SizeEntry = VLASizeMap[VAT]; |
| 469 | |
Anders Carlsson | fcdbb93 | 2008-12-20 21:51:53 +0000 | [diff] [blame] | 470 | if (!SizeEntry) { |
| 471 | // Get the element size; |
| 472 | llvm::Value *ElemSize; |
Anders Carlsson | 60d3541 | 2008-12-20 20:46:34 +0000 | [diff] [blame] | 473 | |
Anders Carlsson | fcdbb93 | 2008-12-20 21:51:53 +0000 | [diff] [blame] | 474 | QualType ElemTy = VAT->getElementType(); |
Anders Carlsson | 96f2147 | 2009-02-05 19:43:10 +0000 | [diff] [blame] | 475 | |
| 476 | const llvm::Type *SizeTy = ConvertType(getContext().getSizeType()); |
| 477 | |
Anders Carlsson | fcdbb93 | 2008-12-20 21:51:53 +0000 | [diff] [blame] | 478 | if (ElemTy->isVariableArrayType()) |
| 479 | ElemSize = EmitVLASize(ElemTy); |
| 480 | else { |
Owen Anderson | a1cf15f | 2009-07-14 23:10:40 +0000 | [diff] [blame] | 481 | ElemSize = VMContext.getConstantInt(SizeTy, |
Anders Carlsson | fcdbb93 | 2008-12-20 21:51:53 +0000 | [diff] [blame] | 482 | getContext().getTypeSize(ElemTy) / 8); |
| 483 | } |
Anders Carlsson | 60d3541 | 2008-12-20 20:46:34 +0000 | [diff] [blame] | 484 | |
Anders Carlsson | fcdbb93 | 2008-12-20 21:51:53 +0000 | [diff] [blame] | 485 | llvm::Value *NumElements = EmitScalarExpr(VAT->getSizeExpr()); |
Anders Carlsson | 96f2147 | 2009-02-05 19:43:10 +0000 | [diff] [blame] | 486 | NumElements = Builder.CreateIntCast(NumElements, SizeTy, false, "tmp"); |
| 487 | |
Anders Carlsson | fcdbb93 | 2008-12-20 21:51:53 +0000 | [diff] [blame] | 488 | SizeEntry = Builder.CreateMul(ElemSize, NumElements); |
Anders Carlsson | 60d3541 | 2008-12-20 20:46:34 +0000 | [diff] [blame] | 489 | } |
| 490 | |
Anders Carlsson | 60d3541 | 2008-12-20 20:46:34 +0000 | [diff] [blame] | 491 | return SizeEntry; |
Eli Friedman | bdad6b6 | 2009-05-29 19:23:46 +0000 | [diff] [blame] | 492 | } else if (const ArrayType *AT = dyn_cast<ArrayType>(Ty)) { |
| 493 | EmitVLASize(AT->getElementType()); |
Ted Kremenek | 35366a6 | 2009-07-17 17:50:17 +0000 | [diff] [blame] | 494 | } else if (const PointerType *PT = Ty->getAsPointerType()) |
Anders Carlsson | 60d3541 | 2008-12-20 20:46:34 +0000 | [diff] [blame] | 495 | EmitVLASize(PT->getPointeeType()); |
Anders Carlsson | f666b77 | 2008-12-20 20:27:15 +0000 | [diff] [blame] | 496 | else { |
Anders Carlsson | 60d3541 | 2008-12-20 20:46:34 +0000 | [diff] [blame] | 497 | assert(0 && "unknown VM type!"); |
Anders Carlsson | dcc90d8 | 2008-12-12 07:19:02 +0000 | [diff] [blame] | 498 | } |
Anders Carlsson | 60d3541 | 2008-12-20 20:46:34 +0000 | [diff] [blame] | 499 | |
| 500 | return 0; |
Anders Carlsson | dcc90d8 | 2008-12-12 07:19:02 +0000 | [diff] [blame] | 501 | } |
Eli Friedman | 4fd0aa5 | 2009-01-20 17:46:04 +0000 | [diff] [blame] | 502 | |
| 503 | llvm::Value* CodeGenFunction::EmitVAListRef(const Expr* E) { |
| 504 | if (CGM.getContext().getBuiltinVaListType()->isArrayType()) { |
| 505 | return EmitScalarExpr(E); |
| 506 | } |
| 507 | return EmitLValue(E).getAddress(); |
| 508 | } |
Anders Carlsson | 6ccc476 | 2009-02-07 22:53:43 +0000 | [diff] [blame] | 509 | |
Anders Carlsson | 6fc5591 | 2009-02-08 03:22:36 +0000 | [diff] [blame] | 510 | void CodeGenFunction::PushCleanupBlock(llvm::BasicBlock *CleanupBlock) |
Anders Carlsson | 6ccc476 | 2009-02-07 22:53:43 +0000 | [diff] [blame] | 511 | { |
Anders Carlsson | 6ccc476 | 2009-02-07 22:53:43 +0000 | [diff] [blame] | 512 | CleanupEntries.push_back(CleanupEntry(CleanupBlock)); |
Anders Carlsson | 6ccc476 | 2009-02-07 22:53:43 +0000 | [diff] [blame] | 513 | } |
Anders Carlsson | c71c845 | 2009-02-07 23:50:39 +0000 | [diff] [blame] | 514 | |
| 515 | void CodeGenFunction::EmitCleanupBlocks(size_t OldCleanupStackSize) |
| 516 | { |
| 517 | assert(CleanupEntries.size() >= OldCleanupStackSize && |
| 518 | "Cleanup stack mismatch!"); |
| 519 | |
| 520 | while (CleanupEntries.size() > OldCleanupStackSize) |
| 521 | EmitCleanupBlock(); |
| 522 | } |
| 523 | |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 524 | CodeGenFunction::CleanupBlockInfo CodeGenFunction::PopCleanupBlock() |
Anders Carlsson | c71c845 | 2009-02-07 23:50:39 +0000 | [diff] [blame] | 525 | { |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 526 | CleanupEntry &CE = CleanupEntries.back(); |
| 527 | |
| 528 | llvm::BasicBlock *CleanupBlock = CE.CleanupBlock; |
| 529 | |
| 530 | std::vector<llvm::BasicBlock *> Blocks; |
| 531 | std::swap(Blocks, CE.Blocks); |
| 532 | |
| 533 | std::vector<llvm::BranchInst *> BranchFixups; |
| 534 | std::swap(BranchFixups, CE.BranchFixups); |
| 535 | |
| 536 | CleanupEntries.pop_back(); |
| 537 | |
Anders Carlsson | ad9d00e | 2009-02-08 22:45:15 +0000 | [diff] [blame] | 538 | // Check if any branch fixups pointed to the scope we just popped. If so, |
| 539 | // we can remove them. |
| 540 | for (size_t i = 0, e = BranchFixups.size(); i != e; ++i) { |
| 541 | llvm::BasicBlock *Dest = BranchFixups[i]->getSuccessor(0); |
| 542 | BlockScopeMap::iterator I = BlockScopes.find(Dest); |
Anders Carlsson | d66a9f9 | 2009-02-08 03:55:35 +0000 | [diff] [blame] | 543 | |
Anders Carlsson | ad9d00e | 2009-02-08 22:45:15 +0000 | [diff] [blame] | 544 | if (I == BlockScopes.end()) |
| 545 | continue; |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 546 | |
Anders Carlsson | ad9d00e | 2009-02-08 22:45:15 +0000 | [diff] [blame] | 547 | assert(I->second <= CleanupEntries.size() && "Invalid branch fixup!"); |
Anders Carlsson | d66a9f9 | 2009-02-08 03:55:35 +0000 | [diff] [blame] | 548 | |
Anders Carlsson | ad9d00e | 2009-02-08 22:45:15 +0000 | [diff] [blame] | 549 | if (I->second == CleanupEntries.size()) { |
| 550 | // We don't need to do this branch fixup. |
| 551 | BranchFixups[i] = BranchFixups.back(); |
| 552 | BranchFixups.pop_back(); |
| 553 | i--; |
| 554 | e--; |
| 555 | continue; |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 556 | } |
| 557 | } |
Anders Carlsson | d66a9f9 | 2009-02-08 03:55:35 +0000 | [diff] [blame] | 558 | |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 559 | llvm::BasicBlock *SwitchBlock = 0; |
| 560 | llvm::BasicBlock *EndBlock = 0; |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 561 | if (!BranchFixups.empty()) { |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 562 | SwitchBlock = createBasicBlock("cleanup.switch"); |
| 563 | EndBlock = createBasicBlock("cleanup.end"); |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 564 | |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 565 | llvm::BasicBlock *CurBB = Builder.GetInsertBlock(); |
Anders Carlsson | 0ae7b2b | 2009-03-17 05:53:35 +0000 | [diff] [blame] | 566 | |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 567 | Builder.SetInsertPoint(SwitchBlock); |
| 568 | |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 569 | llvm::Value *DestCodePtr = CreateTempAlloca(llvm::Type::Int32Ty, |
| 570 | "cleanup.dst"); |
| 571 | llvm::Value *DestCode = Builder.CreateLoad(DestCodePtr, "tmp"); |
| 572 | |
| 573 | // Create a switch instruction to determine where to jump next. |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 574 | llvm::SwitchInst *SI = Builder.CreateSwitch(DestCode, EndBlock, |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 575 | BranchFixups.size()); |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 576 | |
Anders Carlsson | 46831a9 | 2009-02-08 22:13:37 +0000 | [diff] [blame] | 577 | // Restore the current basic block (if any) |
Anders Carlsson | 0ae7b2b | 2009-03-17 05:53:35 +0000 | [diff] [blame] | 578 | if (CurBB) { |
Anders Carlsson | 46831a9 | 2009-02-08 22:13:37 +0000 | [diff] [blame] | 579 | Builder.SetInsertPoint(CurBB); |
Anders Carlsson | 0ae7b2b | 2009-03-17 05:53:35 +0000 | [diff] [blame] | 580 | |
| 581 | // If we had a current basic block, we also need to emit an instruction |
| 582 | // to initialize the cleanup destination. |
Owen Anderson | 6924382 | 2009-07-13 04:10:07 +0000 | [diff] [blame] | 583 | Builder.CreateStore(getLLVMContext().getNullValue(llvm::Type::Int32Ty), |
Anders Carlsson | 0ae7b2b | 2009-03-17 05:53:35 +0000 | [diff] [blame] | 584 | DestCodePtr); |
| 585 | } else |
Anders Carlsson | 46831a9 | 2009-02-08 22:13:37 +0000 | [diff] [blame] | 586 | Builder.ClearInsertionPoint(); |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 587 | |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 588 | for (size_t i = 0, e = BranchFixups.size(); i != e; ++i) { |
| 589 | llvm::BranchInst *BI = BranchFixups[i]; |
| 590 | llvm::BasicBlock *Dest = BI->getSuccessor(0); |
| 591 | |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 592 | // Fixup the branch instruction to point to the cleanup block. |
| 593 | BI->setSuccessor(0, CleanupBlock); |
Anders Carlsson | d66a9f9 | 2009-02-08 03:55:35 +0000 | [diff] [blame] | 594 | |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 595 | if (CleanupEntries.empty()) { |
Anders Carlsson | cc89920 | 2009-02-08 22:46:50 +0000 | [diff] [blame] | 596 | llvm::ConstantInt *ID; |
| 597 | |
| 598 | // Check if we already have a destination for this block. |
| 599 | if (Dest == SI->getDefaultDest()) |
Owen Anderson | a1cf15f | 2009-07-14 23:10:40 +0000 | [diff] [blame] | 600 | ID = VMContext.getConstantInt(llvm::Type::Int32Ty, 0); |
Anders Carlsson | cc89920 | 2009-02-08 22:46:50 +0000 | [diff] [blame] | 601 | else { |
| 602 | ID = SI->findCaseDest(Dest); |
| 603 | if (!ID) { |
| 604 | // No code found, get a new unique one by using the number of |
| 605 | // switch successors. |
Owen Anderson | a1cf15f | 2009-07-14 23:10:40 +0000 | [diff] [blame] | 606 | ID = VMContext.getConstantInt(llvm::Type::Int32Ty, |
Anders Carlsson | cc89920 | 2009-02-08 22:46:50 +0000 | [diff] [blame] | 607 | SI->getNumSuccessors()); |
| 608 | SI->addCase(ID, Dest); |
| 609 | } |
| 610 | } |
| 611 | |
| 612 | // Store the jump destination before the branch instruction. |
| 613 | new llvm::StoreInst(ID, DestCodePtr, BI); |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 614 | } else { |
| 615 | // We need to jump through another cleanup block. Create a pad block |
| 616 | // with a branch instruction that jumps to the final destination and |
| 617 | // add it as a branch fixup to the current cleanup scope. |
| 618 | |
| 619 | // Create the pad block. |
| 620 | llvm::BasicBlock *CleanupPad = createBasicBlock("cleanup.pad", CurFn); |
Anders Carlsson | cc89920 | 2009-02-08 22:46:50 +0000 | [diff] [blame] | 621 | |
| 622 | // Create a unique case ID. |
Owen Anderson | a1cf15f | 2009-07-14 23:10:40 +0000 | [diff] [blame] | 623 | llvm::ConstantInt *ID = VMContext.getConstantInt(llvm::Type::Int32Ty, |
Anders Carlsson | cc89920 | 2009-02-08 22:46:50 +0000 | [diff] [blame] | 624 | SI->getNumSuccessors()); |
| 625 | |
| 626 | // Store the jump destination before the branch instruction. |
| 627 | new llvm::StoreInst(ID, DestCodePtr, BI); |
| 628 | |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 629 | // Add it as the destination. |
Anders Carlsson | cc89920 | 2009-02-08 22:46:50 +0000 | [diff] [blame] | 630 | SI->addCase(ID, CleanupPad); |
Anders Carlsson | 1093c2c | 2009-02-08 01:23:05 +0000 | [diff] [blame] | 631 | |
| 632 | // Create the branch to the final destination. |
| 633 | llvm::BranchInst *BI = llvm::BranchInst::Create(Dest); |
| 634 | CleanupPad->getInstList().push_back(BI); |
| 635 | |
| 636 | // And add it as a branch fixup. |
| 637 | CleanupEntries.back().BranchFixups.push_back(BI); |
| 638 | } |
| 639 | } |
| 640 | } |
Anders Carlsson | d66a9f9 | 2009-02-08 03:55:35 +0000 | [diff] [blame] | 641 | |
Anders Carlsson | bd6fa3d | 2009-02-08 00:16:35 +0000 | [diff] [blame] | 642 | // Remove all blocks from the block scope map. |
| 643 | for (size_t i = 0, e = Blocks.size(); i != e; ++i) { |
| 644 | assert(BlockScopes.count(Blocks[i]) && |
| 645 | "Did not find block in scope map!"); |
| 646 | |
| 647 | BlockScopes.erase(Blocks[i]); |
| 648 | } |
Anders Carlsson | d66a9f9 | 2009-02-08 03:55:35 +0000 | [diff] [blame] | 649 | |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 650 | return CleanupBlockInfo(CleanupBlock, SwitchBlock, EndBlock); |
Anders Carlsson | d66a9f9 | 2009-02-08 03:55:35 +0000 | [diff] [blame] | 651 | } |
| 652 | |
| 653 | void CodeGenFunction::EmitCleanupBlock() |
| 654 | { |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 655 | CleanupBlockInfo Info = PopCleanupBlock(); |
Anders Carlsson | d66a9f9 | 2009-02-08 03:55:35 +0000 | [diff] [blame] | 656 | |
Anders Carlsson | eb6437a | 2009-05-31 00:33:20 +0000 | [diff] [blame] | 657 | llvm::BasicBlock *CurBB = Builder.GetInsertBlock(); |
| 658 | if (CurBB && !CurBB->getTerminator() && |
| 659 | Info.CleanupBlock->getNumUses() == 0) { |
| 660 | CurBB->getInstList().splice(CurBB->end(), Info.CleanupBlock->getInstList()); |
| 661 | delete Info.CleanupBlock; |
| 662 | } else |
| 663 | EmitBlock(Info.CleanupBlock); |
Anders Carlsson | d66a9f9 | 2009-02-08 03:55:35 +0000 | [diff] [blame] | 664 | |
Anders Carlsson | bb66f9f | 2009-02-08 07:46:24 +0000 | [diff] [blame] | 665 | if (Info.SwitchBlock) |
| 666 | EmitBlock(Info.SwitchBlock); |
| 667 | if (Info.EndBlock) |
| 668 | EmitBlock(Info.EndBlock); |
Anders Carlsson | d66a9f9 | 2009-02-08 03:55:35 +0000 | [diff] [blame] | 669 | } |
| 670 | |
Anders Carlsson | 87eaf17 | 2009-02-08 00:50:42 +0000 | [diff] [blame] | 671 | void CodeGenFunction::AddBranchFixup(llvm::BranchInst *BI) |
| 672 | { |
| 673 | assert(!CleanupEntries.empty() && |
| 674 | "Trying to add branch fixup without cleanup block!"); |
| 675 | |
Mike Stump | f5408fe | 2009-05-16 07:57:57 +0000 | [diff] [blame] | 676 | // FIXME: We could be more clever here and check if there's already a branch |
| 677 | // fixup for this destination and recycle it. |
Anders Carlsson | 87eaf17 | 2009-02-08 00:50:42 +0000 | [diff] [blame] | 678 | CleanupEntries.back().BranchFixups.push_back(BI); |
| 679 | } |
| 680 | |
| 681 | void CodeGenFunction::EmitBranchThroughCleanup(llvm::BasicBlock *Dest) |
| 682 | { |
Anders Carlsson | 46831a9 | 2009-02-08 22:13:37 +0000 | [diff] [blame] | 683 | if (!HaveInsertPoint()) |
| 684 | return; |
| 685 | |
Anders Carlsson | 87eaf17 | 2009-02-08 00:50:42 +0000 | [diff] [blame] | 686 | llvm::BranchInst* BI = Builder.CreateBr(Dest); |
| 687 | |
Anders Carlsson | 46831a9 | 2009-02-08 22:13:37 +0000 | [diff] [blame] | 688 | Builder.ClearInsertionPoint(); |
| 689 | |
Anders Carlsson | 87eaf17 | 2009-02-08 00:50:42 +0000 | [diff] [blame] | 690 | // The stack is empty, no need to do any cleanup. |
| 691 | if (CleanupEntries.empty()) |
| 692 | return; |
| 693 | |
| 694 | if (!Dest->getParent()) { |
| 695 | // We are trying to branch to a block that hasn't been inserted yet. |
| 696 | AddBranchFixup(BI); |
| 697 | return; |
| 698 | } |
| 699 | |
| 700 | BlockScopeMap::iterator I = BlockScopes.find(Dest); |
| 701 | if (I == BlockScopes.end()) { |
| 702 | // We are trying to jump to a block that is outside of any cleanup scope. |
| 703 | AddBranchFixup(BI); |
| 704 | return; |
| 705 | } |
| 706 | |
| 707 | assert(I->second < CleanupEntries.size() && |
| 708 | "Trying to branch into cleanup region"); |
| 709 | |
| 710 | if (I->second == CleanupEntries.size() - 1) { |
| 711 | // We have a branch to a block in the same scope. |
| 712 | return; |
| 713 | } |
| 714 | |
| 715 | AddBranchFixup(BI); |
| 716 | } |