blob: 43c86495f3d3c11dafc2f787d3dc447d1e7fcd4e [file] [log] [blame]
Chris Lattner2da04b32007-08-24 05:35:26 +00001//===--- CGExprScalar.cpp - Emit LLVM Code for Scalar Exprs ---------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner5b12ab82007-12-29 19:59:25 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattner2da04b32007-08-24 05:35:26 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This contains code to emit Expr nodes with scalar LLVM types as LLVM code.
11//
12//===----------------------------------------------------------------------===//
13
14#include "CodeGenFunction.h"
Reid Kleckner092d0652017-03-06 22:18:34 +000015#include "CGCleanup.h"
John McCall5d865c322010-08-31 07:33:07 +000016#include "CGCXXABI.h"
Chandler Carruth3a022472012-12-04 09:13:33 +000017#include "CGDebugInfo.h"
Fariborz Jahanian07ca7272009-10-10 20:07:56 +000018#include "CGObjCRuntime.h"
Chris Lattner2da04b32007-08-24 05:35:26 +000019#include "CodeGenModule.h"
Alexey Bataev00396512015-07-02 03:40:19 +000020#include "TargetInfo.h"
Daniel Dunbarad319a72008-08-11 05:00:27 +000021#include "clang/AST/ASTContext.h"
Daniel Dunbar6630e102008-08-12 05:08:18 +000022#include "clang/AST/DeclObjC.h"
Yaxun Liu402804b2016-12-15 08:09:08 +000023#include "clang/AST/Expr.h"
Anders Carlsson15b73de2009-07-18 19:43:29 +000024#include "clang/AST/RecordLayout.h"
Daniel Dunbarad319a72008-08-11 05:00:27 +000025#include "clang/AST/StmtVisitor.h"
Chris Lattnerff2367c2008-04-20 00:50:39 +000026#include "clang/Basic/TargetInfo.h"
Saleem Abdulrasool10a49722016-04-08 16:52:00 +000027#include "clang/Frontend/CodeGenOptions.h"
Vedant Kumar82ee16b2017-02-25 00:43:36 +000028#include "llvm/ADT/Optional.h"
Chandler Carruth735e6d82014-03-04 11:46:22 +000029#include "llvm/IR/CFG.h"
Chandler Carruthffd55512013-01-02 11:45:17 +000030#include "llvm/IR/Constants.h"
31#include "llvm/IR/DataLayout.h"
32#include "llvm/IR/Function.h"
Vedant Kumara125eb52017-06-01 19:22:18 +000033#include "llvm/IR/GetElementPtrTypeIterator.h"
Chandler Carruthffd55512013-01-02 11:45:17 +000034#include "llvm/IR/GlobalVariable.h"
35#include "llvm/IR/Intrinsics.h"
36#include "llvm/IR/Module.h"
Chris Lattner1800c182008-01-03 07:05:49 +000037#include <cstdarg>
Ted Kremenekf182e812007-12-10 23:44:32 +000038
Chris Lattner2da04b32007-08-24 05:35:26 +000039using namespace clang;
40using namespace CodeGen;
41using llvm::Value;
42
43//===----------------------------------------------------------------------===//
44// Scalar Expression Emitter
45//===----------------------------------------------------------------------===//
46
Benjamin Kramerfb5e5842010-10-22 16:48:22 +000047namespace {
Vedant Kumara125eb52017-06-01 19:22:18 +000048
49/// Determine whether the given binary operation may overflow.
50/// Sets \p Result to the value of the operation for BO_Add, BO_Sub, BO_Mul,
51/// and signed BO_{Div,Rem}. For these opcodes, and for unsigned BO_{Div,Rem},
52/// the returned overflow check is precise. The returned value is 'true' for
53/// all other opcodes, to be conservative.
54bool mayHaveIntegerOverflow(llvm::ConstantInt *LHS, llvm::ConstantInt *RHS,
55 BinaryOperator::Opcode Opcode, bool Signed,
56 llvm::APInt &Result) {
57 // Assume overflow is possible, unless we can prove otherwise.
58 bool Overflow = true;
59 const auto &LHSAP = LHS->getValue();
60 const auto &RHSAP = RHS->getValue();
61 if (Opcode == BO_Add) {
62 if (Signed)
63 Result = LHSAP.sadd_ov(RHSAP, Overflow);
64 else
65 Result = LHSAP.uadd_ov(RHSAP, Overflow);
66 } else if (Opcode == BO_Sub) {
67 if (Signed)
68 Result = LHSAP.ssub_ov(RHSAP, Overflow);
69 else
70 Result = LHSAP.usub_ov(RHSAP, Overflow);
71 } else if (Opcode == BO_Mul) {
72 if (Signed)
73 Result = LHSAP.smul_ov(RHSAP, Overflow);
74 else
75 Result = LHSAP.umul_ov(RHSAP, Overflow);
76 } else if (Opcode == BO_Div || Opcode == BO_Rem) {
77 if (Signed && !RHS->isZero())
78 Result = LHSAP.sdiv_ov(RHSAP, Overflow);
79 else
80 return false;
81 }
82 return Overflow;
83}
84
Chris Lattner2da04b32007-08-24 05:35:26 +000085struct BinOpInfo {
86 Value *LHS;
87 Value *RHS;
Chris Lattner3d966d62007-08-24 21:00:35 +000088 QualType Ty; // Computation Type.
Chris Lattner0bf27622010-06-26 21:48:21 +000089 BinaryOperator::Opcode Opcode; // Opcode of BinOp to perform
Adam Nemet484aa452017-03-27 19:17:25 +000090 FPOptions FPFeatures;
Chris Lattner0bf27622010-06-26 21:48:21 +000091 const Expr *E; // Entire expr, for error unsupported. May not be binop.
Vedant Kumard9191152017-05-02 23:46:56 +000092
93 /// Check if the binop can result in integer overflow.
94 bool mayHaveIntegerOverflow() const {
95 // Without constant input, we can't rule out overflow.
Vedant Kumara125eb52017-06-01 19:22:18 +000096 auto *LHSCI = dyn_cast<llvm::ConstantInt>(LHS);
97 auto *RHSCI = dyn_cast<llvm::ConstantInt>(RHS);
Vedant Kumard9191152017-05-02 23:46:56 +000098 if (!LHSCI || !RHSCI)
99 return true;
100
Vedant Kumara125eb52017-06-01 19:22:18 +0000101 llvm::APInt Result;
102 return ::mayHaveIntegerOverflow(
103 LHSCI, RHSCI, Opcode, Ty->hasSignedIntegerRepresentation(), Result);
Vedant Kumard9191152017-05-02 23:46:56 +0000104 }
105
106 /// Check if the binop computes a division or a remainder.
Vedant Kumar94cb34b2017-05-09 00:12:33 +0000107 bool isDivremOp() const {
Vedant Kumard9191152017-05-02 23:46:56 +0000108 return Opcode == BO_Div || Opcode == BO_Rem || Opcode == BO_DivAssign ||
109 Opcode == BO_RemAssign;
110 }
111
112 /// Check if the binop can result in an integer division by zero.
113 bool mayHaveIntegerDivisionByZero() const {
Vedant Kumar94cb34b2017-05-09 00:12:33 +0000114 if (isDivremOp())
Vedant Kumard9191152017-05-02 23:46:56 +0000115 if (auto *CI = dyn_cast<llvm::ConstantInt>(RHS))
116 return CI->isZero();
117 return true;
118 }
119
120 /// Check if the binop can result in a float division by zero.
121 bool mayHaveFloatDivisionByZero() const {
Vedant Kumar94cb34b2017-05-09 00:12:33 +0000122 if (isDivremOp())
Vedant Kumard9191152017-05-02 23:46:56 +0000123 if (auto *CFP = dyn_cast<llvm::ConstantFP>(RHS))
124 return CFP->isZero();
125 return true;
126 }
Chris Lattner2da04b32007-08-24 05:35:26 +0000127};
128
John McCalle84af4e2010-11-13 01:35:44 +0000129static bool MustVisitNullValue(const Expr *E) {
130 // If a null pointer expression's type is the C++0x nullptr_t, then
131 // it's not necessarily a simple constant and it must be evaluated
132 // for its potential side effects.
133 return E->getType()->isNullPtrType();
134}
135
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000136/// If \p E is a widened promoted integer, get its base (unpromoted) type.
137static llvm::Optional<QualType> getUnwidenedIntegerType(const ASTContext &Ctx,
138 const Expr *E) {
139 const Expr *Base = E->IgnoreImpCasts();
140 if (E == Base)
141 return llvm::None;
142
143 QualType BaseTy = Base->getType();
144 if (!BaseTy->isPromotableIntegerType() ||
145 Ctx.getTypeSize(BaseTy) >= Ctx.getTypeSize(E->getType()))
146 return llvm::None;
147
148 return BaseTy;
149}
150
151/// Check if \p E is a widened promoted integer.
152static bool IsWidenedIntegerOp(const ASTContext &Ctx, const Expr *E) {
153 return getUnwidenedIntegerType(Ctx, E).hasValue();
154}
155
156/// Check if we can skip the overflow check for \p Op.
157static bool CanElideOverflowCheck(const ASTContext &Ctx, const BinOpInfo &Op) {
Vedant Kumar66c00cc2017-02-25 06:47:00 +0000158 assert((isa<UnaryOperator>(Op.E) || isa<BinaryOperator>(Op.E)) &&
159 "Expected a unary or binary operator");
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000160
Vedant Kumard9191152017-05-02 23:46:56 +0000161 // If the binop has constant inputs and we can prove there is no overflow,
162 // we can elide the overflow check.
163 if (!Op.mayHaveIntegerOverflow())
164 return true;
165
166 // If a unary op has a widened operand, the op cannot overflow.
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000167 if (const auto *UO = dyn_cast<UnaryOperator>(Op.E))
168 return IsWidenedIntegerOp(Ctx, UO->getSubExpr());
169
Vedant Kumard9191152017-05-02 23:46:56 +0000170 // We usually don't need overflow checks for binops with widened operands.
171 // Multiplication with promoted unsigned operands is a special case.
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000172 const auto *BO = cast<BinaryOperator>(Op.E);
173 auto OptionalLHSTy = getUnwidenedIntegerType(Ctx, BO->getLHS());
174 if (!OptionalLHSTy)
175 return false;
176
177 auto OptionalRHSTy = getUnwidenedIntegerType(Ctx, BO->getRHS());
178 if (!OptionalRHSTy)
179 return false;
180
181 QualType LHSTy = *OptionalLHSTy;
182 QualType RHSTy = *OptionalRHSTy;
183
Vedant Kumard9191152017-05-02 23:46:56 +0000184 // This is the simple case: binops without unsigned multiplication, and with
185 // widened operands. No overflow check is needed here.
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000186 if ((Op.Opcode != BO_Mul && Op.Opcode != BO_MulAssign) ||
187 !LHSTy->isUnsignedIntegerType() || !RHSTy->isUnsignedIntegerType())
188 return true;
189
Vedant Kumard9191152017-05-02 23:46:56 +0000190 // For unsigned multiplication the overflow check can be elided if either one
191 // of the unpromoted types are less than half the size of the promoted type.
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000192 unsigned PromotedSize = Ctx.getTypeSize(Op.E->getType());
193 return (2 * Ctx.getTypeSize(LHSTy)) < PromotedSize ||
194 (2 * Ctx.getTypeSize(RHSTy)) < PromotedSize;
195}
196
Adam Nemet370d0872017-04-04 21:18:30 +0000197/// Update the FastMathFlags of LLVM IR from the FPOptions in LangOptions.
198static void updateFastMathFlags(llvm::FastMathFlags &FMF,
199 FPOptions FPFeatures) {
200 FMF.setAllowContract(FPFeatures.allowFPContractAcrossStatement());
201}
202
203/// Propagate fast-math flags from \p Op to the instruction in \p V.
204static Value *propagateFMFlags(Value *V, const BinOpInfo &Op) {
205 if (auto *I = dyn_cast<llvm::Instruction>(V)) {
206 llvm::FastMathFlags FMF = I->getFastMathFlags();
207 updateFastMathFlags(FMF, Op.FPFeatures);
208 I->setFastMathFlags(FMF);
209 }
210 return V;
211}
212
Benjamin Kramer337e3a52009-11-28 19:45:26 +0000213class ScalarExprEmitter
Chris Lattner2da04b32007-08-24 05:35:26 +0000214 : public StmtVisitor<ScalarExprEmitter, Value*> {
215 CodeGenFunction &CGF;
Daniel Dunbarcb463852008-11-01 01:53:16 +0000216 CGBuilderTy &Builder;
Mike Stumpdf0fe272009-05-29 15:46:01 +0000217 bool IgnoreResultAssign;
Owen Anderson170229f2009-07-14 23:10:40 +0000218 llvm::LLVMContext &VMContext;
Chris Lattner2da04b32007-08-24 05:35:26 +0000219public:
220
Mike Stumpdf0fe272009-05-29 15:46:01 +0000221 ScalarExprEmitter(CodeGenFunction &cgf, bool ira=false)
Mike Stump4a3999f2009-09-09 13:00:44 +0000222 : CGF(cgf), Builder(CGF.Builder), IgnoreResultAssign(ira),
Owen Anderson170229f2009-07-14 23:10:40 +0000223 VMContext(cgf.getLLVMContext()) {
Chris Lattner2da04b32007-08-24 05:35:26 +0000224 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000225
Chris Lattner2da04b32007-08-24 05:35:26 +0000226 //===--------------------------------------------------------------------===//
227 // Utilities
228 //===--------------------------------------------------------------------===//
229
Mike Stumpdf0fe272009-05-29 15:46:01 +0000230 bool TestAndClearIgnoreResultAssign() {
Chris Lattner2a7deb62009-07-08 01:08:03 +0000231 bool I = IgnoreResultAssign;
232 IgnoreResultAssign = false;
233 return I;
234 }
Mike Stumpdf0fe272009-05-29 15:46:01 +0000235
Chris Lattner2192fe52011-07-18 04:24:23 +0000236 llvm::Type *ConvertType(QualType T) { return CGF.ConvertType(T); }
Chris Lattner2da04b32007-08-24 05:35:26 +0000237 LValue EmitLValue(const Expr *E) { return CGF.EmitLValue(E); }
Richard Smith4d1458e2012-09-08 02:08:36 +0000238 LValue EmitCheckedLValue(const Expr *E, CodeGenFunction::TypeCheckKind TCK) {
239 return CGF.EmitCheckedLValue(E, TCK);
Richard Smith69d0d262012-08-24 00:54:33 +0000240 }
Chris Lattner2da04b32007-08-24 05:35:26 +0000241
Peter Collingbourne3eea6772015-05-11 21:39:14 +0000242 void EmitBinOpCheck(ArrayRef<std::pair<Value *, SanitizerMask>> Checks,
Alexey Samsonove396bfc2014-11-11 22:03:54 +0000243 const BinOpInfo &Info);
Richard Smithe30752c2012-10-09 19:52:38 +0000244
Nick Lewycky2d84e842013-10-02 02:29:49 +0000245 Value *EmitLoadOfLValue(LValue LV, SourceLocation Loc) {
246 return CGF.EmitLoadOfLValue(LV, Loc).getScalarVal();
Chris Lattner2da04b32007-08-24 05:35:26 +0000247 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000248
Hal Finkel64567a82014-10-04 15:26:49 +0000249 void EmitLValueAlignmentAssumption(const Expr *E, Value *V) {
250 const AlignValueAttr *AVAttr = nullptr;
251 if (const auto *DRE = dyn_cast<DeclRefExpr>(E)) {
252 const ValueDecl *VD = DRE->getDecl();
253
254 if (VD->getType()->isReferenceType()) {
255 if (const auto *TTy =
256 dyn_cast<TypedefType>(VD->getType().getNonReferenceType()))
257 AVAttr = TTy->getDecl()->getAttr<AlignValueAttr>();
258 } else {
259 // Assumptions for function parameters are emitted at the start of the
260 // function, so there is no need to repeat that here.
261 if (isa<ParmVarDecl>(VD))
262 return;
263
264 AVAttr = VD->getAttr<AlignValueAttr>();
265 }
266 }
267
268 if (!AVAttr)
269 if (const auto *TTy =
270 dyn_cast<TypedefType>(E->getType()))
271 AVAttr = TTy->getDecl()->getAttr<AlignValueAttr>();
272
273 if (!AVAttr)
274 return;
275
276 Value *AlignmentValue = CGF.EmitScalarExpr(AVAttr->getAlignment());
277 llvm::ConstantInt *AlignmentCI = cast<llvm::ConstantInt>(AlignmentValue);
278 CGF.EmitAlignmentAssumption(V, AlignmentCI->getZExtValue());
279 }
280
Chris Lattner2da04b32007-08-24 05:35:26 +0000281 /// EmitLoadOfLValue - Given an expression with complex type that represents a
282 /// value l-value, this method emits the address of the l-value, then loads
283 /// and returns the result.
284 Value *EmitLoadOfLValue(const Expr *E) {
Hal Finkel64567a82014-10-04 15:26:49 +0000285 Value *V = EmitLoadOfLValue(EmitCheckedLValue(E, CodeGenFunction::TCK_Load),
286 E->getExprLoc());
287
288 EmitLValueAlignmentAssumption(E, V);
289 return V;
Chris Lattner2da04b32007-08-24 05:35:26 +0000290 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000291
Chris Lattnere0044382007-08-26 16:42:57 +0000292 /// EmitConversionToBool - Convert the specified expression value to a
Chris Lattner2e928882007-08-26 17:25:57 +0000293 /// boolean (i1) truth value. This is equivalent to "Val != 0".
Chris Lattnere0044382007-08-26 16:42:57 +0000294 Value *EmitConversionToBool(Value *Src, QualType DstTy);
Mike Stump4a3999f2009-09-09 13:00:44 +0000295
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +0000296 /// Emit a check that a conversion to or from a floating-point type does not
297 /// overflow.
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000298 void EmitFloatConversionCheck(Value *OrigSrc, QualType OrigSrcType,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +0000299 Value *Src, QualType SrcType, QualType DstType,
300 llvm::Type *DstTy, SourceLocation Loc);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000301
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +0000302 /// Emit a conversion from the specified type to the specified destination
303 /// type, both of which are LLVM scalar types.
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +0000304 Value *EmitScalarConversion(Value *Src, QualType SrcTy, QualType DstTy,
305 SourceLocation Loc);
Chris Lattner42e6b812007-08-26 16:34:22 +0000306
Anastasia Stulovab02e7832015-10-05 11:27:41 +0000307 Value *EmitScalarConversion(Value *Src, QualType SrcTy, QualType DstTy,
308 SourceLocation Loc, bool TreatBooleanAsSigned);
309
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +0000310 /// Emit a conversion from the specified complex type to the specified
311 /// destination type, where the destination type is an LLVM scalar type.
Chris Lattner42e6b812007-08-26 16:34:22 +0000312 Value *EmitComplexToScalarConversion(CodeGenFunction::ComplexPairTy Src,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +0000313 QualType SrcTy, QualType DstTy,
314 SourceLocation Loc);
Mike Stumpab3afd82009-02-12 18:29:15 +0000315
Anders Carlsson5b944432010-05-22 17:45:10 +0000316 /// EmitNullValue - Emit a value that corresponds to null for the given type.
317 Value *EmitNullValue(QualType Ty);
318
John McCall8cb679e2010-11-15 09:13:47 +0000319 /// EmitFloatToBoolConversion - Perform an FP to boolean conversion.
320 Value *EmitFloatToBoolConversion(Value *V) {
321 // Compare against 0.0 for fp scalars.
322 llvm::Value *Zero = llvm::Constant::getNullValue(V->getType());
323 return Builder.CreateFCmpUNE(V, Zero, "tobool");
324 }
325
326 /// EmitPointerToBoolConversion - Perform a pointer to boolean conversion.
Yaxun Liu402804b2016-12-15 08:09:08 +0000327 Value *EmitPointerToBoolConversion(Value *V, QualType QT) {
328 Value *Zero = CGF.CGM.getNullPointer(cast<llvm::PointerType>(V->getType()), QT);
329
John McCall8cb679e2010-11-15 09:13:47 +0000330 return Builder.CreateICmpNE(V, Zero, "tobool");
331 }
332
333 Value *EmitIntToBoolConversion(Value *V) {
334 // Because of the type rules of C, we often end up computing a
335 // logical value, then zero extending it to int, then wanting it
336 // as a logical value again. Optimize this common case.
337 if (llvm::ZExtInst *ZI = dyn_cast<llvm::ZExtInst>(V)) {
338 if (ZI->getOperand(0)->getType() == Builder.getInt1Ty()) {
339 Value *Result = ZI->getOperand(0);
340 // If there aren't any more uses, zap the instruction to save space.
341 // Note that there can be more uses, for example if this
342 // is the result of an assignment.
343 if (ZI->use_empty())
344 ZI->eraseFromParent();
345 return Result;
346 }
347 }
348
Chris Lattner2531eb42011-04-19 22:55:03 +0000349 return Builder.CreateIsNotNull(V, "tobool");
John McCall8cb679e2010-11-15 09:13:47 +0000350 }
351
Chris Lattner2da04b32007-08-24 05:35:26 +0000352 //===--------------------------------------------------------------------===//
353 // Visitor Methods
354 //===--------------------------------------------------------------------===//
355
Fariborz Jahanian5bbd1b02010-09-17 15:51:28 +0000356 Value *Visit(Expr *E) {
David Blaikie9b479662015-01-25 01:19:10 +0000357 ApplyDebugLocation DL(CGF, E);
Fariborz Jahanian5bbd1b02010-09-17 15:51:28 +0000358 return StmtVisitor<ScalarExprEmitter, Value*>::Visit(E);
359 }
Craig Toppera97d7e72013-07-26 06:16:11 +0000360
Chris Lattner2da04b32007-08-24 05:35:26 +0000361 Value *VisitStmt(Stmt *S) {
Ted Kremenekd4e5fba2007-12-11 21:27:55 +0000362 S->dump(CGF.getContext().getSourceManager());
David Blaikie83d382b2011-09-23 05:06:16 +0000363 llvm_unreachable("Stmt can't have complex result type!");
Chris Lattner2da04b32007-08-24 05:35:26 +0000364 }
365 Value *VisitExpr(Expr *S);
Craig Toppera97d7e72013-07-26 06:16:11 +0000366
Fariborz Jahanian5bbd1b02010-09-17 15:51:28 +0000367 Value *VisitParenExpr(ParenExpr *PE) {
Craig Toppera97d7e72013-07-26 06:16:11 +0000368 return Visit(PE->getSubExpr());
Fariborz Jahanian5bbd1b02010-09-17 15:51:28 +0000369 }
John McCall7c454bb2011-07-15 05:09:51 +0000370 Value *VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *E) {
Craig Toppera97d7e72013-07-26 06:16:11 +0000371 return Visit(E->getReplacement());
John McCall7c454bb2011-07-15 05:09:51 +0000372 }
Peter Collingbourne91147592011-04-15 00:35:48 +0000373 Value *VisitGenericSelectionExpr(GenericSelectionExpr *GE) {
374 return Visit(GE->getResultExpr());
375 }
Gor Nishanov5eb58582017-03-26 02:18:05 +0000376 Value *VisitCoawaitExpr(CoawaitExpr *S) {
377 return CGF.EmitCoawaitExpr(*S).getScalarVal();
378 }
379 Value *VisitCoyieldExpr(CoyieldExpr *S) {
380 return CGF.EmitCoyieldExpr(*S).getScalarVal();
381 }
382 Value *VisitUnaryCoawait(const UnaryOperator *E) {
383 return Visit(E->getSubExpr());
384 }
Chris Lattner2da04b32007-08-24 05:35:26 +0000385
386 // Leaves.
387 Value *VisitIntegerLiteral(const IntegerLiteral *E) {
Chris Lattner2531eb42011-04-19 22:55:03 +0000388 return Builder.getInt(E->getValue());
Chris Lattner2da04b32007-08-24 05:35:26 +0000389 }
390 Value *VisitFloatingLiteral(const FloatingLiteral *E) {
Owen Andersone05f2ed2009-07-27 21:00:51 +0000391 return llvm::ConstantFP::get(VMContext, E->getValue());
Chris Lattner2da04b32007-08-24 05:35:26 +0000392 }
393 Value *VisitCharacterLiteral(const CharacterLiteral *E) {
Owen Andersonb7a2fe62009-07-24 23:12:58 +0000394 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
Chris Lattner2da04b32007-08-24 05:35:26 +0000395 }
Ted Kremeneke65b0862012-03-06 20:05:56 +0000396 Value *VisitObjCBoolLiteralExpr(const ObjCBoolLiteralExpr *E) {
397 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
398 }
Nate Begeman4c18c232007-11-15 05:40:03 +0000399 Value *VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
Owen Andersonb7a2fe62009-07-24 23:12:58 +0000400 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
Nate Begeman4c18c232007-11-15 05:40:03 +0000401 }
Douglas Gregor747eb782010-07-08 06:14:04 +0000402 Value *VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E) {
Anders Carlsson5b944432010-05-22 17:45:10 +0000403 return EmitNullValue(E->getType());
Argyrios Kyrtzidisce4528f2008-08-23 19:35:47 +0000404 }
Anders Carlsson39def3a2008-12-21 22:39:40 +0000405 Value *VisitGNUNullExpr(const GNUNullExpr *E) {
Anders Carlsson5b944432010-05-22 17:45:10 +0000406 return EmitNullValue(E->getType());
Anders Carlsson39def3a2008-12-21 22:39:40 +0000407 }
Eli Friedmand7c72322010-08-05 09:58:49 +0000408 Value *VisitOffsetOfExpr(OffsetOfExpr *E);
Peter Collingbournee190dee2011-03-11 19:24:49 +0000409 Value *VisitUnaryExprOrTypeTraitExpr(const UnaryExprOrTypeTraitExpr *E);
Daniel Dunbar88402ce2008-08-04 16:51:22 +0000410 Value *VisitAddrLabelExpr(const AddrLabelExpr *E) {
Chris Lattner6c4d2552009-10-28 23:59:40 +0000411 llvm::Value *V = CGF.GetAddrOfLabel(E->getLabel());
412 return Builder.CreateBitCast(V, ConvertType(E->getType()));
Daniel Dunbar88402ce2008-08-04 16:51:22 +0000413 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000414
Douglas Gregorbe7b5482011-01-12 22:11:34 +0000415 Value *VisitSizeOfPackExpr(SizeOfPackExpr *E) {
Chris Lattner2531eb42011-04-19 22:55:03 +0000416 return llvm::ConstantInt::get(ConvertType(E->getType()),E->getPackLength());
Douglas Gregorbe7b5482011-01-12 22:11:34 +0000417 }
John McCall1bf58462011-02-16 08:02:54 +0000418
John McCallfe96e0b2011-11-06 09:01:30 +0000419 Value *VisitPseudoObjectExpr(PseudoObjectExpr *E) {
420 return CGF.EmitPseudoObjectRValue(E).getScalarVal();
421 }
422
John McCall1bf58462011-02-16 08:02:54 +0000423 Value *VisitOpaqueValueExpr(OpaqueValueExpr *E) {
John McCallc07a0c72011-02-17 10:25:35 +0000424 if (E->isGLValue())
Nick Lewycky2d84e842013-10-02 02:29:49 +0000425 return EmitLoadOfLValue(CGF.getOpaqueLValueMapping(E), E->getExprLoc());
John McCall1bf58462011-02-16 08:02:54 +0000426
427 // Otherwise, assume the mapping is the scalar directly.
John McCallc07a0c72011-02-17 10:25:35 +0000428 return CGF.getOpaqueRValueMapping(E).getScalarVal();
John McCall1bf58462011-02-16 08:02:54 +0000429 }
John McCall71335052012-03-10 03:05:10 +0000430
Chris Lattner2da04b32007-08-24 05:35:26 +0000431 // l-values.
John McCall113bee02012-03-10 09:33:50 +0000432 Value *VisitDeclRefExpr(DeclRefExpr *E) {
433 if (CodeGenFunction::ConstantEmission result = CGF.tryEmitAsConstant(E)) {
John McCall71335052012-03-10 03:05:10 +0000434 if (result.isReference())
Nick Lewycky2d84e842013-10-02 02:29:49 +0000435 return EmitLoadOfLValue(result.getReferenceLValue(CGF, E),
436 E->getExprLoc());
John McCall71335052012-03-10 03:05:10 +0000437 return result.getValue();
Richard Smithbc6387672012-03-02 23:27:11 +0000438 }
John McCall113bee02012-03-10 09:33:50 +0000439 return EmitLoadOfLValue(E);
John McCall71335052012-03-10 03:05:10 +0000440 }
441
Mike Stump4a3999f2009-09-09 13:00:44 +0000442 Value *VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
443 return CGF.EmitObjCSelectorExpr(E);
Daniel Dunbar55310df2008-08-27 06:57:25 +0000444 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000445 Value *VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
446 return CGF.EmitObjCProtocolExpr(E);
Daniel Dunbar55310df2008-08-27 06:57:25 +0000447 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000448 Value *VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
Daniel Dunbar55310df2008-08-27 06:57:25 +0000449 return EmitLoadOfLValue(E);
450 }
Daniel Dunbar55310df2008-08-27 06:57:25 +0000451 Value *VisitObjCMessageExpr(ObjCMessageExpr *E) {
Craig Toppera97d7e72013-07-26 06:16:11 +0000452 if (E->getMethodDecl() &&
Alp Toker314cc812014-01-25 16:55:45 +0000453 E->getMethodDecl()->getReturnType()->isReferenceType())
Fariborz Jahanianff989032011-03-02 20:09:49 +0000454 return EmitLoadOfLValue(E);
Daniel Dunbar55310df2008-08-27 06:57:25 +0000455 return CGF.EmitObjCMessageExpr(E).getScalarVal();
Daniel Dunbarc8317a42008-08-23 10:51:21 +0000456 }
457
Fariborz Jahaniana5fee262009-12-09 19:05:56 +0000458 Value *VisitObjCIsaExpr(ObjCIsaExpr *E) {
Fariborz Jahanian531c16f2009-12-09 23:35:29 +0000459 LValue LV = CGF.EmitObjCIsaExpr(E);
Nick Lewycky2d84e842013-10-02 02:29:49 +0000460 Value *V = CGF.EmitLoadOfLValue(LV, E->getExprLoc()).getScalarVal();
Fariborz Jahaniana5fee262009-12-09 19:05:56 +0000461 return V;
462 }
463
Erik Pilkington9c42a8d2017-02-23 21:08:08 +0000464 Value *VisitObjCAvailabilityCheckExpr(ObjCAvailabilityCheckExpr *E) {
465 VersionTuple Version = E->getVersion();
466
467 // If we're checking for a platform older than our minimum deployment
468 // target, we can fold the check away.
469 if (Version <= CGF.CGM.getTarget().getPlatformMinVersion())
470 return llvm::ConstantInt::get(Builder.getInt1Ty(), 1);
471
472 Optional<unsigned> Min = Version.getMinor(), SMin = Version.getSubminor();
473 llvm::Value *Args[] = {
474 llvm::ConstantInt::get(CGF.CGM.Int32Ty, Version.getMajor()),
475 llvm::ConstantInt::get(CGF.CGM.Int32Ty, Min ? *Min : 0),
476 llvm::ConstantInt::get(CGF.CGM.Int32Ty, SMin ? *SMin : 0),
477 };
478
479 return CGF.EmitBuiltinAvailable(Args);
480 }
481
Chris Lattner2da04b32007-08-24 05:35:26 +0000482 Value *VisitArraySubscriptExpr(ArraySubscriptExpr *E);
Eli Friedmana1b4ed82008-05-14 19:38:39 +0000483 Value *VisitShuffleVectorExpr(ShuffleVectorExpr *E);
Hal Finkelc4d7c822013-09-18 03:29:45 +0000484 Value *VisitConvertVectorExpr(ConvertVectorExpr *E);
Eli Friedmancb422f12009-11-26 03:22:21 +0000485 Value *VisitMemberExpr(MemberExpr *E);
Nate Begemance4d7fc2008-04-18 23:10:10 +0000486 Value *VisitExtVectorElementExpr(Expr *E) { return EmitLoadOfLValue(E); }
Chris Lattner084bc322008-10-26 23:53:12 +0000487 Value *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
488 return EmitLoadOfLValue(E);
489 }
Devang Patel43fc86d2007-10-24 17:18:43 +0000490
Nate Begeman19351632009-10-18 20:10:40 +0000491 Value *VisitInitListExpr(InitListExpr *E);
Mike Stump4a3999f2009-09-09 13:00:44 +0000492
Richard Smith410306b2016-12-12 02:53:20 +0000493 Value *VisitArrayInitIndexExpr(ArrayInitIndexExpr *E) {
494 assert(CGF.getArrayInitIndex() &&
495 "ArrayInitIndexExpr not inside an ArrayInitLoopExpr?");
496 return CGF.getArrayInitIndex();
497 }
498
Douglas Gregor0202cb42009-01-29 17:44:32 +0000499 Value *VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E) {
Richard Smithd82a2ce2012-12-21 03:17:28 +0000500 return EmitNullValue(E->getType());
Douglas Gregor0202cb42009-01-29 17:44:32 +0000501 }
John McCall23c29fe2011-06-24 21:55:10 +0000502 Value *VisitExplicitCastExpr(ExplicitCastExpr *E) {
Alexey Bataev2bf9b4c2015-10-20 04:24:12 +0000503 CGF.CGM.EmitExplicitCastExprType(E, &CGF);
John McCall23c29fe2011-06-24 21:55:10 +0000504 return VisitCastExpr(E);
Chris Lattner2da04b32007-08-24 05:35:26 +0000505 }
John McCall23c29fe2011-06-24 21:55:10 +0000506 Value *VisitCastExpr(CastExpr *E);
Chris Lattner2da04b32007-08-24 05:35:26 +0000507
508 Value *VisitCallExpr(const CallExpr *E) {
David Majnemerced8bdf2015-02-25 17:36:15 +0000509 if (E->getCallReturnType(CGF.getContext())->isReferenceType())
Anders Carlssond8b7ae22009-05-27 03:37:57 +0000510 return EmitLoadOfLValue(E);
Mike Stump4a3999f2009-09-09 13:00:44 +0000511
Hal Finkel64567a82014-10-04 15:26:49 +0000512 Value *V = CGF.EmitCallExpr(E).getScalarVal();
513
514 EmitLValueAlignmentAssumption(E, V);
515 return V;
Chris Lattner2da04b32007-08-24 05:35:26 +0000516 }
Daniel Dunbar97db84c2008-08-23 03:46:30 +0000517
Chris Lattner04a913b2007-08-31 22:09:40 +0000518 Value *VisitStmtExpr(const StmtExpr *E);
Mike Stumpcb2fbcb2009-02-21 20:00:35 +0000519
Chris Lattner2da04b32007-08-24 05:35:26 +0000520 // Unary Operators.
Chris Lattner2da04b32007-08-24 05:35:26 +0000521 Value *VisitUnaryPostDec(const UnaryOperator *E) {
Chris Lattner05dc78c2010-06-26 22:09:34 +0000522 LValue LV = EmitLValue(E->getSubExpr());
523 return EmitScalarPrePostIncDec(E, LV, false, false);
Chris Lattner2da04b32007-08-24 05:35:26 +0000524 }
525 Value *VisitUnaryPostInc(const UnaryOperator *E) {
Chris Lattner05dc78c2010-06-26 22:09:34 +0000526 LValue LV = EmitLValue(E->getSubExpr());
527 return EmitScalarPrePostIncDec(E, LV, true, false);
Chris Lattner2da04b32007-08-24 05:35:26 +0000528 }
529 Value *VisitUnaryPreDec(const UnaryOperator *E) {
Chris Lattner05dc78c2010-06-26 22:09:34 +0000530 LValue LV = EmitLValue(E->getSubExpr());
531 return EmitScalarPrePostIncDec(E, LV, false, true);
Chris Lattner2da04b32007-08-24 05:35:26 +0000532 }
533 Value *VisitUnaryPreInc(const UnaryOperator *E) {
Chris Lattner05dc78c2010-06-26 22:09:34 +0000534 LValue LV = EmitLValue(E->getSubExpr());
535 return EmitScalarPrePostIncDec(E, LV, true, true);
Chris Lattner2da04b32007-08-24 05:35:26 +0000536 }
Chris Lattner05dc78c2010-06-26 22:09:34 +0000537
Alexey Samsonovf6246502015-04-23 01:50:45 +0000538 llvm::Value *EmitIncDecConsiderOverflowBehavior(const UnaryOperator *E,
539 llvm::Value *InVal,
540 bool IsInc);
Anton Yartsev85129b82011-02-07 02:17:30 +0000541
Chris Lattner05dc78c2010-06-26 22:09:34 +0000542 llvm::Value *EmitScalarPrePostIncDec(const UnaryOperator *E, LValue LV,
543 bool isInc, bool isPre);
544
Craig Toppera97d7e72013-07-26 06:16:11 +0000545
Chris Lattner2da04b32007-08-24 05:35:26 +0000546 Value *VisitUnaryAddrOf(const UnaryOperator *E) {
John McCallf3a88602011-02-03 08:15:49 +0000547 if (isa<MemberPointerType>(E->getType())) // never sugared
548 return CGF.CGM.getMemberPointerConstant(E);
549
John McCall7f416cc2015-09-08 08:05:57 +0000550 return EmitLValue(E->getSubExpr()).getPointer();
Chris Lattner2da04b32007-08-24 05:35:26 +0000551 }
John McCall59482722010-12-04 12:43:24 +0000552 Value *VisitUnaryDeref(const UnaryOperator *E) {
553 if (E->getType()->isVoidType())
554 return Visit(E->getSubExpr()); // the actual value should be unused
555 return EmitLoadOfLValue(E);
556 }
Chris Lattner2da04b32007-08-24 05:35:26 +0000557 Value *VisitUnaryPlus(const UnaryOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +0000558 // This differs from gcc, though, most likely due to a bug in gcc.
559 TestAndClearIgnoreResultAssign();
Chris Lattner2da04b32007-08-24 05:35:26 +0000560 return Visit(E->getSubExpr());
561 }
562 Value *VisitUnaryMinus (const UnaryOperator *E);
563 Value *VisitUnaryNot (const UnaryOperator *E);
564 Value *VisitUnaryLNot (const UnaryOperator *E);
Chris Lattner9f0ad962007-08-24 21:20:17 +0000565 Value *VisitUnaryReal (const UnaryOperator *E);
566 Value *VisitUnaryImag (const UnaryOperator *E);
Chris Lattner2da04b32007-08-24 05:35:26 +0000567 Value *VisitUnaryExtension(const UnaryOperator *E) {
568 return Visit(E->getSubExpr());
569 }
Craig Toppera97d7e72013-07-26 06:16:11 +0000570
Anders Carlssona5d077d2009-04-14 16:58:56 +0000571 // C++
Douglas Gregor34f6c6d2011-08-09 00:37:14 +0000572 Value *VisitMaterializeTemporaryExpr(const MaterializeTemporaryExpr *E) {
Eli Friedman0be39702011-08-14 04:50:34 +0000573 return EmitLoadOfLValue(E);
Douglas Gregor34f6c6d2011-08-09 00:37:14 +0000574 }
Craig Toppera97d7e72013-07-26 06:16:11 +0000575
Chris Lattneraa9c7ae2008-04-08 04:40:51 +0000576 Value *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
577 return Visit(DAE->getExpr());
578 }
Richard Smith852c9db2013-04-20 22:23:05 +0000579 Value *VisitCXXDefaultInitExpr(CXXDefaultInitExpr *DIE) {
580 CodeGenFunction::CXXDefaultInitExprScope Scope(CGF);
581 return Visit(DIE->getExpr());
582 }
Anders Carlssona5d077d2009-04-14 16:58:56 +0000583 Value *VisitCXXThisExpr(CXXThisExpr *TE) {
584 return CGF.LoadCXXThis();
Mike Stump4a3999f2009-09-09 13:00:44 +0000585 }
586
Reid Kleckner092d0652017-03-06 22:18:34 +0000587 Value *VisitExprWithCleanups(ExprWithCleanups *E);
Anders Carlsson4a7b49b2009-05-31 01:40:14 +0000588 Value *VisitCXXNewExpr(const CXXNewExpr *E) {
589 return CGF.EmitCXXNewExpr(E);
590 }
Anders Carlsson81f0df92009-08-16 21:13:42 +0000591 Value *VisitCXXDeleteExpr(const CXXDeleteExpr *E) {
592 CGF.EmitCXXDeleteExpr(E);
Craig Topper8a13c412014-05-21 05:09:00 +0000593 return nullptr;
Anders Carlsson81f0df92009-08-16 21:13:42 +0000594 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000595
Alp Tokercbb90342013-12-13 20:49:58 +0000596 Value *VisitTypeTraitExpr(const TypeTraitExpr *E) {
Francois Pichet34b21132010-12-08 22:35:30 +0000597 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
Francois Pichet9dfa3ce2010-12-07 00:08:36 +0000598 }
599
John Wiegley6242b6a2011-04-28 00:16:57 +0000600 Value *VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *E) {
601 return llvm::ConstantInt::get(Builder.getInt32Ty(), E->getValue());
602 }
603
John Wiegleyf9f65842011-04-25 06:54:41 +0000604 Value *VisitExpressionTraitExpr(const ExpressionTraitExpr *E) {
605 return llvm::ConstantInt::get(Builder.getInt1Ty(), E->getValue());
606 }
607
Douglas Gregorad8a3362009-09-04 17:36:40 +0000608 Value *VisitCXXPseudoDestructorExpr(const CXXPseudoDestructorExpr *E) {
609 // C++ [expr.pseudo]p1:
Mike Stump4a3999f2009-09-09 13:00:44 +0000610 // The result shall only be used as the operand for the function call
Douglas Gregorad8a3362009-09-04 17:36:40 +0000611 // operator (), and the result of such a call has type void. The only
612 // effect is the evaluation of the postfix-expression before the dot or
613 // arrow.
614 CGF.EmitScalarExpr(E->getBase());
Craig Topper8a13c412014-05-21 05:09:00 +0000615 return nullptr;
Douglas Gregorad8a3362009-09-04 17:36:40 +0000616 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000617
Anders Carlsson04c3bf42009-09-15 04:39:46 +0000618 Value *VisitCXXNullPtrLiteralExpr(const CXXNullPtrLiteralExpr *E) {
Anders Carlsson5b944432010-05-22 17:45:10 +0000619 return EmitNullValue(E->getType());
Anders Carlsson04c3bf42009-09-15 04:39:46 +0000620 }
Anders Carlsson4b08db72009-10-30 01:42:31 +0000621
622 Value *VisitCXXThrowExpr(const CXXThrowExpr *E) {
623 CGF.EmitCXXThrowExpr(E);
Craig Topper8a13c412014-05-21 05:09:00 +0000624 return nullptr;
Anders Carlsson4b08db72009-10-30 01:42:31 +0000625 }
626
Sebastian Redlb67655f2010-09-10 21:04:00 +0000627 Value *VisitCXXNoexceptExpr(const CXXNoexceptExpr *E) {
Chris Lattner2531eb42011-04-19 22:55:03 +0000628 return Builder.getInt1(E->getValue());
Sebastian Redlb67655f2010-09-10 21:04:00 +0000629 }
630
Chris Lattner2da04b32007-08-24 05:35:26 +0000631 // Binary Operators.
Chris Lattner2da04b32007-08-24 05:35:26 +0000632 Value *EmitMul(const BinOpInfo &Ops) {
Douglas Gregor6ab2fa82011-05-20 16:38:50 +0000633 if (Ops.Ty->isSignedIntegerOrEnumerationType()) {
Richard Smith9c6890a2012-11-01 22:30:59 +0000634 switch (CGF.getLangOpts().getSignedOverflowBehavior()) {
Chris Lattner51924e512010-06-26 21:25:03 +0000635 case LangOptions::SOB_Defined:
636 return Builder.CreateMul(Ops.LHS, Ops.RHS, "mul");
Richard Smith3e056de2012-08-25 00:32:28 +0000637 case LangOptions::SOB_Undefined:
Alexey Samsonovedf99a92014-11-07 22:29:38 +0000638 if (!CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow))
Richard Smith3e056de2012-08-25 00:32:28 +0000639 return Builder.CreateNSWMul(Ops.LHS, Ops.RHS, "mul");
640 // Fall through.
Chris Lattner51924e512010-06-26 21:25:03 +0000641 case LangOptions::SOB_Trapping:
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000642 if (CanElideOverflowCheck(CGF.getContext(), Ops))
643 return Builder.CreateNSWMul(Ops.LHS, Ops.RHS, "mul");
Chris Lattner51924e512010-06-26 21:25:03 +0000644 return EmitOverflowCheckedBinOp(Ops);
645 }
646 }
Richard Smithb1b0ab42012-11-05 22:21:05 +0000647
Alexey Samsonovedf99a92014-11-07 22:29:38 +0000648 if (Ops.Ty->isUnsignedIntegerType() &&
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000649 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow) &&
650 !CanElideOverflowCheck(CGF.getContext(), Ops))
Will Dietz1897cb32012-11-27 15:01:55 +0000651 return EmitOverflowCheckedBinOp(Ops);
652
Adam Nemet370d0872017-04-04 21:18:30 +0000653 if (Ops.LHS->getType()->isFPOrFPVectorTy()) {
654 Value *V = Builder.CreateFMul(Ops.LHS, Ops.RHS, "mul");
655 return propagateFMFlags(V, Ops);
656 }
Chris Lattner2da04b32007-08-24 05:35:26 +0000657 return Builder.CreateMul(Ops.LHS, Ops.RHS, "mul");
658 }
Mike Stump0c61b732009-04-01 20:28:16 +0000659 /// Create a binary op that checks for overflow.
660 /// Currently only supports +, - and *.
661 Value *EmitOverflowCheckedBinOp(const BinOpInfo &Ops);
Richard Smith4d1458e2012-09-08 02:08:36 +0000662
Chris Lattner8ee6a412010-09-11 21:47:09 +0000663 // Check for undefined division and modulus behaviors.
Craig Toppera97d7e72013-07-26 06:16:11 +0000664 void EmitUndefinedBehaviorIntegerDivAndRemCheck(const BinOpInfo &Ops,
Chris Lattner8ee6a412010-09-11 21:47:09 +0000665 llvm::Value *Zero,bool isDiv);
David Tweed042e0882013-01-07 16:43:27 +0000666 // Common helper for getting how wide LHS of shift is.
667 static Value *GetWidthMinusOneValue(Value* LHS,Value* RHS);
Chris Lattner2da04b32007-08-24 05:35:26 +0000668 Value *EmitDiv(const BinOpInfo &Ops);
669 Value *EmitRem(const BinOpInfo &Ops);
670 Value *EmitAdd(const BinOpInfo &Ops);
671 Value *EmitSub(const BinOpInfo &Ops);
672 Value *EmitShl(const BinOpInfo &Ops);
673 Value *EmitShr(const BinOpInfo &Ops);
674 Value *EmitAnd(const BinOpInfo &Ops) {
675 return Builder.CreateAnd(Ops.LHS, Ops.RHS, "and");
676 }
677 Value *EmitXor(const BinOpInfo &Ops) {
678 return Builder.CreateXor(Ops.LHS, Ops.RHS, "xor");
679 }
680 Value *EmitOr (const BinOpInfo &Ops) {
681 return Builder.CreateOr(Ops.LHS, Ops.RHS, "or");
682 }
683
Chris Lattner3d966d62007-08-24 21:00:35 +0000684 BinOpInfo EmitBinOps(const BinaryOperator *E);
Douglas Gregor914af212010-04-23 04:16:32 +0000685 LValue EmitCompoundAssignLValue(const CompoundAssignOperator *E,
686 Value *(ScalarExprEmitter::*F)(const BinOpInfo &),
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +0000687 Value *&Result);
Douglas Gregor914af212010-04-23 04:16:32 +0000688
Chris Lattnerb6334692007-08-26 21:41:21 +0000689 Value *EmitCompoundAssign(const CompoundAssignOperator *E,
Chris Lattner3d966d62007-08-24 21:00:35 +0000690 Value *(ScalarExprEmitter::*F)(const BinOpInfo &));
691
692 // Binary operators and binary compound assignment operators.
693#define HANDLEBINOP(OP) \
Chris Lattnerb6334692007-08-26 21:41:21 +0000694 Value *VisitBin ## OP(const BinaryOperator *E) { \
695 return Emit ## OP(EmitBinOps(E)); \
696 } \
697 Value *VisitBin ## OP ## Assign(const CompoundAssignOperator *E) { \
698 return EmitCompoundAssign(E, &ScalarExprEmitter::Emit ## OP); \
Chris Lattner3d966d62007-08-24 21:00:35 +0000699 }
Daniel Dunbare017ecc2009-12-19 17:50:07 +0000700 HANDLEBINOP(Mul)
701 HANDLEBINOP(Div)
702 HANDLEBINOP(Rem)
703 HANDLEBINOP(Add)
704 HANDLEBINOP(Sub)
705 HANDLEBINOP(Shl)
706 HANDLEBINOP(Shr)
707 HANDLEBINOP(And)
708 HANDLEBINOP(Xor)
709 HANDLEBINOP(Or)
Chris Lattner3d966d62007-08-24 21:00:35 +0000710#undef HANDLEBINOP
Daniel Dunbarbfb1cd72008-08-06 02:00:38 +0000711
Chris Lattner2da04b32007-08-24 05:35:26 +0000712 // Comparisons.
Craig Topperc82f8962015-12-16 06:24:28 +0000713 Value *EmitCompare(const BinaryOperator *E, llvm::CmpInst::Predicate UICmpOpc,
714 llvm::CmpInst::Predicate SICmpOpc,
715 llvm::CmpInst::Predicate FCmpOpc);
Chris Lattner2da04b32007-08-24 05:35:26 +0000716#define VISITCOMP(CODE, UI, SI, FP) \
717 Value *VisitBin##CODE(const BinaryOperator *E) { \
718 return EmitCompare(E, llvm::ICmpInst::UI, llvm::ICmpInst::SI, \
719 llvm::FCmpInst::FP); }
Daniel Dunbare017ecc2009-12-19 17:50:07 +0000720 VISITCOMP(LT, ICMP_ULT, ICMP_SLT, FCMP_OLT)
721 VISITCOMP(GT, ICMP_UGT, ICMP_SGT, FCMP_OGT)
722 VISITCOMP(LE, ICMP_ULE, ICMP_SLE, FCMP_OLE)
723 VISITCOMP(GE, ICMP_UGE, ICMP_SGE, FCMP_OGE)
724 VISITCOMP(EQ, ICMP_EQ , ICMP_EQ , FCMP_OEQ)
725 VISITCOMP(NE, ICMP_NE , ICMP_NE , FCMP_UNE)
Chris Lattner2da04b32007-08-24 05:35:26 +0000726#undef VISITCOMP
Mike Stump4a3999f2009-09-09 13:00:44 +0000727
Chris Lattner2da04b32007-08-24 05:35:26 +0000728 Value *VisitBinAssign (const BinaryOperator *E);
729
730 Value *VisitBinLAnd (const BinaryOperator *E);
731 Value *VisitBinLOr (const BinaryOperator *E);
Chris Lattner2da04b32007-08-24 05:35:26 +0000732 Value *VisitBinComma (const BinaryOperator *E);
733
Eli Friedmanacfb1df2009-11-18 09:41:26 +0000734 Value *VisitBinPtrMemD(const Expr *E) { return EmitLoadOfLValue(E); }
735 Value *VisitBinPtrMemI(const Expr *E) { return EmitLoadOfLValue(E); }
736
Chris Lattner2da04b32007-08-24 05:35:26 +0000737 // Other Operators.
Mike Stumpab3afd82009-02-12 18:29:15 +0000738 Value *VisitBlockExpr(const BlockExpr *BE);
John McCallc07a0c72011-02-17 10:25:35 +0000739 Value *VisitAbstractConditionalOperator(const AbstractConditionalOperator *);
Chris Lattner2da04b32007-08-24 05:35:26 +0000740 Value *VisitChooseExpr(ChooseExpr *CE);
Anders Carlsson7e13ab82007-10-15 20:28:48 +0000741 Value *VisitVAArgExpr(VAArgExpr *VE);
Chris Lattner2da04b32007-08-24 05:35:26 +0000742 Value *VisitObjCStringLiteral(const ObjCStringLiteral *E) {
743 return CGF.EmitObjCStringLiteral(E);
744 }
Patrick Beard0caa3942012-04-19 00:25:12 +0000745 Value *VisitObjCBoxedExpr(ObjCBoxedExpr *E) {
746 return CGF.EmitObjCBoxedExpr(E);
Ted Kremeneke65b0862012-03-06 20:05:56 +0000747 }
748 Value *VisitObjCArrayLiteral(ObjCArrayLiteral *E) {
749 return CGF.EmitObjCArrayLiteral(E);
750 }
751 Value *VisitObjCDictionaryLiteral(ObjCDictionaryLiteral *E) {
752 return CGF.EmitObjCDictionaryLiteral(E);
753 }
Tanya Lattner55808c12011-06-04 00:47:47 +0000754 Value *VisitAsTypeExpr(AsTypeExpr *CE);
Eli Friedmandf14b3a2011-10-11 02:20:01 +0000755 Value *VisitAtomicExpr(AtomicExpr *AE);
Chris Lattner2da04b32007-08-24 05:35:26 +0000756};
757} // end anonymous namespace.
758
759//===----------------------------------------------------------------------===//
760// Utilities
761//===----------------------------------------------------------------------===//
762
Chris Lattnere0044382007-08-26 16:42:57 +0000763/// EmitConversionToBool - Convert the specified expression value to a
Chris Lattner2e928882007-08-26 17:25:57 +0000764/// boolean (i1) truth value. This is equivalent to "Val != 0".
Chris Lattnere0044382007-08-26 16:42:57 +0000765Value *ScalarExprEmitter::EmitConversionToBool(Value *Src, QualType SrcType) {
John McCallb692a092009-10-22 20:10:53 +0000766 assert(SrcType.isCanonical() && "EmitScalarConversion strips typedefs");
Mike Stump4a3999f2009-09-09 13:00:44 +0000767
John McCall8cb679e2010-11-15 09:13:47 +0000768 if (SrcType->isRealFloatingType())
769 return EmitFloatToBoolConversion(Src);
Mike Stump4a3999f2009-09-09 13:00:44 +0000770
John McCall7a9aac22010-08-23 01:21:21 +0000771 if (const MemberPointerType *MPT = dyn_cast<MemberPointerType>(SrcType))
772 return CGF.CGM.getCXXABI().EmitMemberPointerIsNotNull(CGF, Src, MPT);
Mike Stump4a3999f2009-09-09 13:00:44 +0000773
Daniel Dunbaref957f32008-08-25 10:38:11 +0000774 assert((SrcType->isIntegerType() || isa<llvm::PointerType>(Src->getType())) &&
Chris Lattnere0044382007-08-26 16:42:57 +0000775 "Unknown scalar type to convert");
Mike Stump4a3999f2009-09-09 13:00:44 +0000776
John McCall8cb679e2010-11-15 09:13:47 +0000777 if (isa<llvm::IntegerType>(Src->getType()))
778 return EmitIntToBoolConversion(Src);
Mike Stump4a3999f2009-09-09 13:00:44 +0000779
John McCall8cb679e2010-11-15 09:13:47 +0000780 assert(isa<llvm::PointerType>(Src->getType()));
Yaxun Liu402804b2016-12-15 08:09:08 +0000781 return EmitPointerToBoolConversion(Src, SrcType);
Chris Lattnere0044382007-08-26 16:42:57 +0000782}
783
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +0000784void ScalarExprEmitter::EmitFloatConversionCheck(
785 Value *OrigSrc, QualType OrigSrcType, Value *Src, QualType SrcType,
786 QualType DstType, llvm::Type *DstTy, SourceLocation Loc) {
Alexey Samsonov24cad992014-07-17 18:46:27 +0000787 CodeGenFunction::SanitizerScope SanScope(&CGF);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000788 using llvm::APFloat;
789 using llvm::APSInt;
790
791 llvm::Type *SrcTy = Src->getType();
792
Craig Topper8a13c412014-05-21 05:09:00 +0000793 llvm::Value *Check = nullptr;
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000794 if (llvm::IntegerType *IntTy = dyn_cast<llvm::IntegerType>(SrcTy)) {
795 // Integer to floating-point. This can fail for unsigned short -> __half
796 // or unsigned __int128 -> float.
797 assert(DstType->isFloatingType());
798 bool SrcIsUnsigned = OrigSrcType->isUnsignedIntegerOrEnumerationType();
799
800 APFloat LargestFloat =
801 APFloat::getLargest(CGF.getContext().getFloatTypeSemantics(DstType));
802 APSInt LargestInt(IntTy->getBitWidth(), SrcIsUnsigned);
803
804 bool IsExact;
805 if (LargestFloat.convertToInteger(LargestInt, APFloat::rmTowardZero,
806 &IsExact) != APFloat::opOK)
807 // The range of representable values of this floating point type includes
808 // all values of this integer type. Don't need an overflow check.
809 return;
810
811 llvm::Value *Max = llvm::ConstantInt::get(VMContext, LargestInt);
812 if (SrcIsUnsigned)
813 Check = Builder.CreateICmpULE(Src, Max);
814 else {
815 llvm::Value *Min = llvm::ConstantInt::get(VMContext, -LargestInt);
816 llvm::Value *GE = Builder.CreateICmpSGE(Src, Min);
817 llvm::Value *LE = Builder.CreateICmpSLE(Src, Max);
818 Check = Builder.CreateAnd(GE, LE);
819 }
820 } else {
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000821 const llvm::fltSemantics &SrcSema =
822 CGF.getContext().getFloatTypeSemantics(OrigSrcType);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000823 if (isa<llvm::IntegerType>(DstTy)) {
Richard Smith2b01d502013-03-27 23:20:25 +0000824 // Floating-point to integer. This has undefined behavior if the source is
825 // +-Inf, NaN, or doesn't fit into the destination type (after truncation
826 // to an integer).
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000827 unsigned Width = CGF.getContext().getIntWidth(DstType);
828 bool Unsigned = DstType->isUnsignedIntegerOrEnumerationType();
829
830 APSInt Min = APSInt::getMinValue(Width, Unsigned);
Richard Smith2b01d502013-03-27 23:20:25 +0000831 APFloat MinSrc(SrcSema, APFloat::uninitialized);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000832 if (MinSrc.convertFromAPInt(Min, !Unsigned, APFloat::rmTowardZero) &
833 APFloat::opOverflow)
834 // Don't need an overflow check for lower bound. Just check for
835 // -Inf/NaN.
Richard Smith4af40c42013-03-19 00:01:12 +0000836 MinSrc = APFloat::getInf(SrcSema, true);
837 else
838 // Find the largest value which is too small to represent (before
839 // truncation toward zero).
840 MinSrc.subtract(APFloat(SrcSema, 1), APFloat::rmTowardNegative);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000841
842 APSInt Max = APSInt::getMaxValue(Width, Unsigned);
Richard Smith2b01d502013-03-27 23:20:25 +0000843 APFloat MaxSrc(SrcSema, APFloat::uninitialized);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000844 if (MaxSrc.convertFromAPInt(Max, !Unsigned, APFloat::rmTowardZero) &
845 APFloat::opOverflow)
846 // Don't need an overflow check for upper bound. Just check for
847 // +Inf/NaN.
Richard Smith4af40c42013-03-19 00:01:12 +0000848 MaxSrc = APFloat::getInf(SrcSema, false);
849 else
850 // Find the smallest value which is too large to represent (before
851 // truncation toward zero).
852 MaxSrc.add(APFloat(SrcSema, 1), APFloat::rmTowardPositive);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000853
Richard Smith2b01d502013-03-27 23:20:25 +0000854 // If we're converting from __half, convert the range to float to match
855 // the type of src.
856 if (OrigSrcType->isHalfType()) {
857 const llvm::fltSemantics &Sema =
858 CGF.getContext().getFloatTypeSemantics(SrcType);
859 bool IsInexact;
860 MinSrc.convert(Sema, APFloat::rmTowardZero, &IsInexact);
861 MaxSrc.convert(Sema, APFloat::rmTowardZero, &IsInexact);
862 }
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000863
Richard Smith4af40c42013-03-19 00:01:12 +0000864 llvm::Value *GE =
865 Builder.CreateFCmpOGT(Src, llvm::ConstantFP::get(VMContext, MinSrc));
866 llvm::Value *LE =
867 Builder.CreateFCmpOLT(Src, llvm::ConstantFP::get(VMContext, MaxSrc));
868 Check = Builder.CreateAnd(GE, LE);
869 } else {
Richard Smith2b01d502013-03-27 23:20:25 +0000870 // FIXME: Maybe split this sanitizer out from float-cast-overflow.
871 //
872 // Floating-point to floating-point. This has undefined behavior if the
873 // source is not in the range of representable values of the destination
874 // type. The C and C++ standards are spectacularly unclear here. We
875 // diagnose finite out-of-range conversions, but allow infinities and NaNs
876 // to convert to the corresponding value in the smaller type.
877 //
878 // C11 Annex F gives all such conversions defined behavior for IEC 60559
879 // conforming implementations. Unfortunately, LLVM's fptrunc instruction
880 // does not.
881
882 // Converting from a lower rank to a higher rank can never have
883 // undefined behavior, since higher-rank types must have a superset
884 // of values of lower-rank types.
885 if (CGF.getContext().getFloatingTypeOrder(OrigSrcType, DstType) != 1)
886 return;
887
888 assert(!OrigSrcType->isHalfType() &&
889 "should not check conversion from __half, it has the lowest rank");
890
891 const llvm::fltSemantics &DstSema =
892 CGF.getContext().getFloatTypeSemantics(DstType);
893 APFloat MinBad = APFloat::getLargest(DstSema, false);
894 APFloat MaxBad = APFloat::getInf(DstSema, false);
895
896 bool IsInexact;
897 MinBad.convert(SrcSema, APFloat::rmTowardZero, &IsInexact);
898 MaxBad.convert(SrcSema, APFloat::rmTowardZero, &IsInexact);
899
900 Value *AbsSrc = CGF.EmitNounwindRuntimeCall(
901 CGF.CGM.getIntrinsic(llvm::Intrinsic::fabs, Src->getType()), Src);
Richard Smith4af40c42013-03-19 00:01:12 +0000902 llvm::Value *GE =
Richard Smith2b01d502013-03-27 23:20:25 +0000903 Builder.CreateFCmpOGT(AbsSrc, llvm::ConstantFP::get(VMContext, MinBad));
Richard Smith4af40c42013-03-19 00:01:12 +0000904 llvm::Value *LE =
Richard Smith2b01d502013-03-27 23:20:25 +0000905 Builder.CreateFCmpOLT(AbsSrc, llvm::ConstantFP::get(VMContext, MaxBad));
906 Check = Builder.CreateNot(Builder.CreateAnd(GE, LE));
Richard Smith4af40c42013-03-19 00:01:12 +0000907 }
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000908 }
909
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +0000910 llvm::Constant *StaticArgs[] = {CGF.EmitCheckSourceLocation(Loc),
911 CGF.EmitCheckTypeDescriptor(OrigSrcType),
912 CGF.EmitCheckTypeDescriptor(DstType)};
Alexey Samsonove396bfc2014-11-11 22:03:54 +0000913 CGF.EmitCheck(std::make_pair(Check, SanitizerKind::FloatCastOverflow),
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +0000914 SanitizerHandler::FloatCastOverflow, StaticArgs, OrigSrc);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000915}
916
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +0000917/// Emit a conversion from the specified type to the specified destination type,
918/// both of which are LLVM scalar types.
Chris Lattner42e6b812007-08-26 16:34:22 +0000919Value *ScalarExprEmitter::EmitScalarConversion(Value *Src, QualType SrcType,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +0000920 QualType DstType,
921 SourceLocation Loc) {
Anastasia Stulovab02e7832015-10-05 11:27:41 +0000922 return EmitScalarConversion(Src, SrcType, DstType, Loc, false);
923}
924
925Value *ScalarExprEmitter::EmitScalarConversion(Value *Src, QualType SrcType,
926 QualType DstType,
927 SourceLocation Loc,
928 bool TreatBooleanAsSigned) {
Chris Lattner0f398c42008-07-26 22:37:01 +0000929 SrcType = CGF.getContext().getCanonicalType(SrcType);
930 DstType = CGF.getContext().getCanonicalType(DstType);
Chris Lattner3474c202007-08-26 06:48:56 +0000931 if (SrcType == DstType) return Src;
Mike Stump4a3999f2009-09-09 13:00:44 +0000932
Craig Topper8a13c412014-05-21 05:09:00 +0000933 if (DstType->isVoidType()) return nullptr;
Mike Stump4a3999f2009-09-09 13:00:44 +0000934
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000935 llvm::Value *OrigSrc = Src;
936 QualType OrigSrcType = SrcType;
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +0000937 llvm::Type *SrcTy = Src->getType();
938
Ahmed Bougacha47ec2c72015-03-23 17:48:07 +0000939 // Handle conversions to bool first, they are special: comparisons against 0.
940 if (DstType->isBooleanType())
941 return EmitConversionToBool(Src, SrcType);
942
943 llvm::Type *DstTy = ConvertType(DstType);
944
Ahmed Bougachad1801af2015-03-23 17:54:16 +0000945 // Cast from half through float if half isn't a native type.
946 if (SrcType->isHalfType() && !CGF.getContext().getLangOpts().NativeHalfType) {
947 // Cast to FP using the intrinsic if the half type itself isn't supported.
948 if (DstTy->isFloatingPointTy()) {
949 if (!CGF.getContext().getLangOpts().HalfArgsAndReturns)
950 return Builder.CreateCall(
951 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_from_fp16, DstTy),
952 Src);
953 } else {
954 // Cast to other types through float, using either the intrinsic or FPExt,
955 // depending on whether the half type itself is supported
956 // (as opposed to operations on half, available with NativeHalfType).
957 if (!CGF.getContext().getLangOpts().HalfArgsAndReturns) {
958 Src = Builder.CreateCall(
959 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_from_fp16,
960 CGF.CGM.FloatTy),
961 Src);
962 } else {
963 Src = Builder.CreateFPExt(Src, CGF.CGM.FloatTy, "conv");
964 }
965 SrcType = CGF.getContext().FloatTy;
966 SrcTy = CGF.FloatTy;
967 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +0000968 }
969
Chris Lattner3474c202007-08-26 06:48:56 +0000970 // Ignore conversions like int -> uint.
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +0000971 if (SrcTy == DstTy)
Chris Lattner3474c202007-08-26 06:48:56 +0000972 return Src;
973
Mike Stump4a3999f2009-09-09 13:00:44 +0000974 // Handle pointer conversions next: pointers can only be converted to/from
975 // other pointers and integers. Check for pointer types in terms of LLVM, as
976 // some native types (like Obj-C id) may map to a pointer type.
Yaxun Liu26f75662016-08-19 05:17:25 +0000977 if (auto DstPT = dyn_cast<llvm::PointerType>(DstTy)) {
Chris Lattner3474c202007-08-26 06:48:56 +0000978 // The source value may be an integer, or a pointer.
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +0000979 if (isa<llvm::PointerType>(SrcTy))
Chris Lattner3474c202007-08-26 06:48:56 +0000980 return Builder.CreateBitCast(Src, DstTy, "conv");
Anders Carlsson12f5a252009-09-12 04:57:16 +0000981
Chris Lattner3474c202007-08-26 06:48:56 +0000982 assert(SrcType->isIntegerType() && "Not ptr->ptr or int->ptr conversion?");
Eli Friedman42d2a3a2009-03-04 04:02:35 +0000983 // First, convert to the correct width so that we control the kind of
984 // extension.
Yaxun Liu26f75662016-08-19 05:17:25 +0000985 llvm::Type *MiddleTy = CGF.CGM.getDataLayout().getIntPtrType(DstPT);
Douglas Gregor6ab2fa82011-05-20 16:38:50 +0000986 bool InputSigned = SrcType->isSignedIntegerOrEnumerationType();
Eli Friedman42d2a3a2009-03-04 04:02:35 +0000987 llvm::Value* IntResult =
988 Builder.CreateIntCast(Src, MiddleTy, InputSigned, "conv");
989 // Then, cast to pointer.
990 return Builder.CreateIntToPtr(IntResult, DstTy, "conv");
Chris Lattner3474c202007-08-26 06:48:56 +0000991 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000992
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +0000993 if (isa<llvm::PointerType>(SrcTy)) {
Chris Lattner3474c202007-08-26 06:48:56 +0000994 // Must be an ptr to int cast.
995 assert(isa<llvm::IntegerType>(DstTy) && "not ptr->int?");
Anders Carlssone89b84a2007-10-31 23:18:02 +0000996 return Builder.CreatePtrToInt(Src, DstTy, "conv");
Chris Lattner3474c202007-08-26 06:48:56 +0000997 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000998
Nate Begemance4d7fc2008-04-18 23:10:10 +0000999 // A scalar can be splatted to an extended vector of the same element type
Nate Begeman5ec4b312009-08-10 23:49:36 +00001000 if (DstType->isExtVectorType() && !SrcType->isVectorType()) {
George Burgess IVdf1ed002016-01-13 01:52:39 +00001001 // Sema should add casts to make sure that the source expression's type is
1002 // the same as the vector's element type (sans qualifiers)
1003 assert(DstType->castAs<ExtVectorType>()->getElementType().getTypePtr() ==
1004 SrcType.getTypePtr() &&
1005 "Splatted expr doesn't match with vector element type?");
Nate Begemanb699c9b2009-01-18 06:42:49 +00001006
Nate Begemanb699c9b2009-01-18 06:42:49 +00001007 // Splat the element across to all elements
Craig Topperf2f1a092016-07-08 02:17:35 +00001008 unsigned NumElements = DstTy->getVectorNumElements();
George Burgess IVdf1ed002016-01-13 01:52:39 +00001009 return Builder.CreateVectorSplat(NumElements, Src, "splat");
Nate Begemanb699c9b2009-01-18 06:42:49 +00001010 }
Nate Begeman330aaa72007-12-30 02:59:45 +00001011
Chris Lattner6cba8e92008-02-02 04:51:41 +00001012 // Allow bitcast from vector to integer/fp of the same size.
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001013 if (isa<llvm::VectorType>(SrcTy) ||
Chris Lattner6cba8e92008-02-02 04:51:41 +00001014 isa<llvm::VectorType>(DstTy))
Anders Carlssona297e7a2007-12-05 07:36:10 +00001015 return Builder.CreateBitCast(Src, DstTy, "conv");
Mike Stump4a3999f2009-09-09 13:00:44 +00001016
Chris Lattner3474c202007-08-26 06:48:56 +00001017 // Finally, we have the arithmetic types: real int/float.
Craig Topper8a13c412014-05-21 05:09:00 +00001018 Value *Res = nullptr;
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001019 llvm::Type *ResTy = DstTy;
1020
Richard Smithf9a1e4a2012-10-12 22:57:06 +00001021 // An overflowing conversion has undefined behavior if either the source type
1022 // or the destination type is a floating-point type.
Alexey Samsonovedf99a92014-11-07 22:29:38 +00001023 if (CGF.SanOpts.has(SanitizerKind::FloatCastOverflow) &&
Richard Smithf9a1e4a2012-10-12 22:57:06 +00001024 (OrigSrcType->isFloatingType() || DstType->isFloatingType()))
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001025 EmitFloatConversionCheck(OrigSrc, OrigSrcType, Src, SrcType, DstType, DstTy,
1026 Loc);
Richard Smithf9a1e4a2012-10-12 22:57:06 +00001027
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001028 // Cast to half through float if half isn't a native type.
1029 if (DstType->isHalfType() && !CGF.getContext().getLangOpts().NativeHalfType) {
1030 // Make sure we cast in a single step if from another FP type.
1031 if (SrcTy->isFloatingPointTy()) {
1032 // Use the intrinsic if the half type itself isn't supported
1033 // (as opposed to operations on half, available with NativeHalfType).
1034 if (!CGF.getContext().getLangOpts().HalfArgsAndReturns)
1035 return Builder.CreateCall(
1036 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_to_fp16, SrcTy), Src);
1037 // If the half type is supported, just use an fptrunc.
1038 return Builder.CreateFPTrunc(Src, DstTy);
1039 }
Chris Lattnerece04092012-02-07 00:39:47 +00001040 DstTy = CGF.FloatTy;
Ahmed Bougacha47ec2c72015-03-23 17:48:07 +00001041 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001042
1043 if (isa<llvm::IntegerType>(SrcTy)) {
Douglas Gregor6ab2fa82011-05-20 16:38:50 +00001044 bool InputSigned = SrcType->isSignedIntegerOrEnumerationType();
Anastasia Stulovab02e7832015-10-05 11:27:41 +00001045 if (SrcType->isBooleanType() && TreatBooleanAsSigned) {
1046 InputSigned = true;
1047 }
Anders Carlssonc9d41e72007-12-26 18:20:19 +00001048 if (isa<llvm::IntegerType>(DstTy))
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001049 Res = Builder.CreateIntCast(Src, DstTy, InputSigned, "conv");
Anders Carlssonc9d41e72007-12-26 18:20:19 +00001050 else if (InputSigned)
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001051 Res = Builder.CreateSIToFP(Src, DstTy, "conv");
Anders Carlssonc9d41e72007-12-26 18:20:19 +00001052 else
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001053 Res = Builder.CreateUIToFP(Src, DstTy, "conv");
1054 } else if (isa<llvm::IntegerType>(DstTy)) {
1055 assert(SrcTy->isFloatingPointTy() && "Unknown real conversion");
Douglas Gregor6ab2fa82011-05-20 16:38:50 +00001056 if (DstType->isSignedIntegerOrEnumerationType())
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001057 Res = Builder.CreateFPToSI(Src, DstTy, "conv");
Anders Carlssonc9d41e72007-12-26 18:20:19 +00001058 else
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001059 Res = Builder.CreateFPToUI(Src, DstTy, "conv");
1060 } else {
1061 assert(SrcTy->isFloatingPointTy() && DstTy->isFloatingPointTy() &&
1062 "Unknown real conversion");
1063 if (DstTy->getTypeID() < SrcTy->getTypeID())
1064 Res = Builder.CreateFPTrunc(Src, DstTy, "conv");
1065 else
1066 Res = Builder.CreateFPExt(Src, DstTy, "conv");
Chris Lattner3474c202007-08-26 06:48:56 +00001067 }
1068
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001069 if (DstTy != ResTy) {
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001070 if (!CGF.getContext().getLangOpts().HalfArgsAndReturns) {
1071 assert(ResTy->isIntegerTy(16) && "Only half FP requires extra conversion");
1072 Res = Builder.CreateCall(
Tim Northover6dbcbac2014-07-17 10:51:31 +00001073 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_to_fp16, CGF.CGM.FloatTy),
1074 Res);
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001075 } else {
1076 Res = Builder.CreateFPTrunc(Res, ResTy, "conv");
1077 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001078 }
1079
1080 return Res;
Chris Lattner3474c202007-08-26 06:48:56 +00001081}
1082
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +00001083/// Emit a conversion from the specified complex type to the specified
1084/// destination type, where the destination type is an LLVM scalar type.
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001085Value *ScalarExprEmitter::EmitComplexToScalarConversion(
1086 CodeGenFunction::ComplexPairTy Src, QualType SrcTy, QualType DstTy,
1087 SourceLocation Loc) {
Chris Lattnerc141c1b2007-08-26 16:52:28 +00001088 // Get the source element type.
John McCall47fb9502013-03-07 21:37:08 +00001089 SrcTy = SrcTy->castAs<ComplexType>()->getElementType();
Mike Stump4a3999f2009-09-09 13:00:44 +00001090
Chris Lattnerc141c1b2007-08-26 16:52:28 +00001091 // Handle conversions to bool first, they are special: comparisons against 0.
1092 if (DstTy->isBooleanType()) {
1093 // Complex != 0 -> (Real != 0) | (Imag != 0)
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001094 Src.first = EmitScalarConversion(Src.first, SrcTy, DstTy, Loc);
1095 Src.second = EmitScalarConversion(Src.second, SrcTy, DstTy, Loc);
Chris Lattnerc141c1b2007-08-26 16:52:28 +00001096 return Builder.CreateOr(Src.first, Src.second, "tobool");
1097 }
Mike Stump4a3999f2009-09-09 13:00:44 +00001098
Chris Lattner42e6b812007-08-26 16:34:22 +00001099 // C99 6.3.1.7p2: "When a value of complex type is converted to a real type,
1100 // the imaginary part of the complex value is discarded and the value of the
1101 // real part is converted according to the conversion rules for the
Mike Stump4a3999f2009-09-09 13:00:44 +00001102 // corresponding real type.
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001103 return EmitScalarConversion(Src.first, SrcTy, DstTy, Loc);
Chris Lattner42e6b812007-08-26 16:34:22 +00001104}
1105
Anders Carlsson5b944432010-05-22 17:45:10 +00001106Value *ScalarExprEmitter::EmitNullValue(QualType Ty) {
Richard Smithd82a2ce2012-12-21 03:17:28 +00001107 return CGF.EmitFromMemory(CGF.CGM.EmitNullConstant(Ty), Ty);
Anders Carlsson5b944432010-05-22 17:45:10 +00001108}
Chris Lattner42e6b812007-08-26 16:34:22 +00001109
Richard Smithe30752c2012-10-09 19:52:38 +00001110/// \brief Emit a sanitization check for the given "binary" operation (which
1111/// might actually be a unary increment which has been lowered to a binary
Alexey Samsonove396bfc2014-11-11 22:03:54 +00001112/// operation). The check passes if all values in \p Checks (which are \c i1),
1113/// are \c true.
1114void ScalarExprEmitter::EmitBinOpCheck(
Peter Collingbourne3eea6772015-05-11 21:39:14 +00001115 ArrayRef<std::pair<Value *, SanitizerMask>> Checks, const BinOpInfo &Info) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00001116 assert(CGF.IsSanitizerScope);
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001117 SanitizerHandler Check;
Dmitri Gribenkof8579502013-01-12 19:30:44 +00001118 SmallVector<llvm::Constant *, 4> StaticData;
1119 SmallVector<llvm::Value *, 2> DynamicData;
Richard Smithe30752c2012-10-09 19:52:38 +00001120
1121 BinaryOperatorKind Opcode = Info.Opcode;
1122 if (BinaryOperator::isCompoundAssignmentOp(Opcode))
1123 Opcode = BinaryOperator::getOpForCompoundAssignment(Opcode);
1124
1125 StaticData.push_back(CGF.EmitCheckSourceLocation(Info.E->getExprLoc()));
1126 const UnaryOperator *UO = dyn_cast<UnaryOperator>(Info.E);
1127 if (UO && UO->getOpcode() == UO_Minus) {
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001128 Check = SanitizerHandler::NegateOverflow;
Richard Smithe30752c2012-10-09 19:52:38 +00001129 StaticData.push_back(CGF.EmitCheckTypeDescriptor(UO->getType()));
1130 DynamicData.push_back(Info.RHS);
1131 } else {
1132 if (BinaryOperator::isShiftOp(Opcode)) {
1133 // Shift LHS negative or too large, or RHS out of bounds.
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001134 Check = SanitizerHandler::ShiftOutOfBounds;
Richard Smithe30752c2012-10-09 19:52:38 +00001135 const BinaryOperator *BO = cast<BinaryOperator>(Info.E);
1136 StaticData.push_back(
1137 CGF.EmitCheckTypeDescriptor(BO->getLHS()->getType()));
1138 StaticData.push_back(
1139 CGF.EmitCheckTypeDescriptor(BO->getRHS()->getType()));
1140 } else if (Opcode == BO_Div || Opcode == BO_Rem) {
1141 // Divide or modulo by zero, or signed overflow (eg INT_MAX / -1).
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001142 Check = SanitizerHandler::DivremOverflow;
Will Dietzcefb4482013-01-07 22:25:52 +00001143 StaticData.push_back(CGF.EmitCheckTypeDescriptor(Info.Ty));
Richard Smithe30752c2012-10-09 19:52:38 +00001144 } else {
Alexey Samsonov4c1a96f2014-11-10 22:27:30 +00001145 // Arithmetic overflow (+, -, *).
Richard Smithe30752c2012-10-09 19:52:38 +00001146 switch (Opcode) {
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001147 case BO_Add: Check = SanitizerHandler::AddOverflow; break;
1148 case BO_Sub: Check = SanitizerHandler::SubOverflow; break;
1149 case BO_Mul: Check = SanitizerHandler::MulOverflow; break;
Richard Smithe30752c2012-10-09 19:52:38 +00001150 default: llvm_unreachable("unexpected opcode for bin op check");
1151 }
Will Dietzcefb4482013-01-07 22:25:52 +00001152 StaticData.push_back(CGF.EmitCheckTypeDescriptor(Info.Ty));
Richard Smithe30752c2012-10-09 19:52:38 +00001153 }
1154 DynamicData.push_back(Info.LHS);
1155 DynamicData.push_back(Info.RHS);
1156 }
1157
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001158 CGF.EmitCheck(Checks, Check, StaticData, DynamicData);
Richard Smithe30752c2012-10-09 19:52:38 +00001159}
1160
Chris Lattner2da04b32007-08-24 05:35:26 +00001161//===----------------------------------------------------------------------===//
1162// Visitor Methods
1163//===----------------------------------------------------------------------===//
1164
1165Value *ScalarExprEmitter::VisitExpr(Expr *E) {
Daniel Dunbara7c8cf62008-08-16 00:56:44 +00001166 CGF.ErrorUnsupported(E, "scalar expression");
Chris Lattner2da04b32007-08-24 05:35:26 +00001167 if (E->getType()->isVoidType())
Craig Topper8a13c412014-05-21 05:09:00 +00001168 return nullptr;
Owen Anderson7ec07a52009-07-30 23:11:26 +00001169 return llvm::UndefValue::get(CGF.ConvertType(E->getType()));
Chris Lattner2da04b32007-08-24 05:35:26 +00001170}
1171
Eli Friedmana1b4ed82008-05-14 19:38:39 +00001172Value *ScalarExprEmitter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
Nate Begemana0110022010-06-08 00:16:34 +00001173 // Vector Mask Case
Craig Topperb3174a82016-05-18 04:11:25 +00001174 if (E->getNumSubExprs() == 2) {
Chris Lattner5e016ae2010-06-27 07:15:29 +00001175 Value *LHS = CGF.EmitScalarExpr(E->getExpr(0));
1176 Value *RHS = CGF.EmitScalarExpr(E->getExpr(1));
1177 Value *Mask;
Craig Toppera97d7e72013-07-26 06:16:11 +00001178
Chris Lattner2192fe52011-07-18 04:24:23 +00001179 llvm::VectorType *LTy = cast<llvm::VectorType>(LHS->getType());
Nate Begemana0110022010-06-08 00:16:34 +00001180 unsigned LHSElts = LTy->getNumElements();
1181
Craig Topperb3174a82016-05-18 04:11:25 +00001182 Mask = RHS;
Craig Toppera97d7e72013-07-26 06:16:11 +00001183
Chris Lattner2192fe52011-07-18 04:24:23 +00001184 llvm::VectorType *MTy = cast<llvm::VectorType>(Mask->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00001185
Nate Begemana0110022010-06-08 00:16:34 +00001186 // Mask off the high bits of each shuffle index.
Benjamin Kramer99383102015-07-28 16:25:32 +00001187 Value *MaskBits =
1188 llvm::ConstantInt::get(MTy, llvm::NextPowerOf2(LHSElts - 1) - 1);
Nate Begemana0110022010-06-08 00:16:34 +00001189 Mask = Builder.CreateAnd(Mask, MaskBits, "mask");
Craig Toppera97d7e72013-07-26 06:16:11 +00001190
Nate Begemana0110022010-06-08 00:16:34 +00001191 // newv = undef
1192 // mask = mask & maskbits
1193 // for each elt
1194 // n = extract mask i
1195 // x = extract val n
1196 // newv = insert newv, x, i
Chris Lattner2192fe52011-07-18 04:24:23 +00001197 llvm::VectorType *RTy = llvm::VectorType::get(LTy->getElementType(),
Craig Topper18243fb2013-07-27 05:00:42 +00001198 MTy->getNumElements());
Nate Begemana0110022010-06-08 00:16:34 +00001199 Value* NewV = llvm::UndefValue::get(RTy);
1200 for (unsigned i = 0, e = MTy->getNumElements(); i != e; ++i) {
Michael J. Spencerdd597752014-05-31 00:22:12 +00001201 Value *IIndx = llvm::ConstantInt::get(CGF.SizeTy, i);
Eli Friedman1fa36052012-04-05 21:48:40 +00001202 Value *Indx = Builder.CreateExtractElement(Mask, IIndx, "shuf_idx");
Craig Toppera97d7e72013-07-26 06:16:11 +00001203
Nate Begemana0110022010-06-08 00:16:34 +00001204 Value *VExt = Builder.CreateExtractElement(LHS, Indx, "shuf_elt");
Eli Friedman1fa36052012-04-05 21:48:40 +00001205 NewV = Builder.CreateInsertElement(NewV, VExt, IIndx, "shuf_ins");
Nate Begemana0110022010-06-08 00:16:34 +00001206 }
1207 return NewV;
Eli Friedmana1b4ed82008-05-14 19:38:39 +00001208 }
Craig Toppera97d7e72013-07-26 06:16:11 +00001209
Eli Friedmana1b4ed82008-05-14 19:38:39 +00001210 Value* V1 = CGF.EmitScalarExpr(E->getExpr(0));
1211 Value* V2 = CGF.EmitScalarExpr(E->getExpr(1));
Craig Toppera97d7e72013-07-26 06:16:11 +00001212
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001213 SmallVector<llvm::Constant*, 32> indices;
Craig Topper0ed37bd2013-08-01 04:51:48 +00001214 for (unsigned i = 2; i < E->getNumSubExprs(); ++i) {
Craig Topper50ad5b72013-08-03 17:40:38 +00001215 llvm::APSInt Idx = E->getShuffleMaskIdx(CGF.getContext(), i-2);
1216 // Check for -1 and output it as undef in the IR.
1217 if (Idx.isSigned() && Idx.isAllOnesValue())
1218 indices.push_back(llvm::UndefValue::get(CGF.Int32Ty));
1219 else
1220 indices.push_back(Builder.getInt32(Idx.getZExtValue()));
Nate Begemana0110022010-06-08 00:16:34 +00001221 }
1222
Chris Lattner91c08ad2011-02-15 00:14:06 +00001223 Value *SV = llvm::ConstantVector::get(indices);
Eli Friedmana1b4ed82008-05-14 19:38:39 +00001224 return Builder.CreateShuffleVector(V1, V2, SV, "shuffle");
1225}
Hal Finkelc4d7c822013-09-18 03:29:45 +00001226
1227Value *ScalarExprEmitter::VisitConvertVectorExpr(ConvertVectorExpr *E) {
1228 QualType SrcType = E->getSrcExpr()->getType(),
1229 DstType = E->getType();
1230
1231 Value *Src = CGF.EmitScalarExpr(E->getSrcExpr());
1232
1233 SrcType = CGF.getContext().getCanonicalType(SrcType);
1234 DstType = CGF.getContext().getCanonicalType(DstType);
1235 if (SrcType == DstType) return Src;
1236
1237 assert(SrcType->isVectorType() &&
1238 "ConvertVector source type must be a vector");
1239 assert(DstType->isVectorType() &&
1240 "ConvertVector destination type must be a vector");
1241
1242 llvm::Type *SrcTy = Src->getType();
1243 llvm::Type *DstTy = ConvertType(DstType);
1244
1245 // Ignore conversions like int -> uint.
1246 if (SrcTy == DstTy)
1247 return Src;
1248
1249 QualType SrcEltType = SrcType->getAs<VectorType>()->getElementType(),
1250 DstEltType = DstType->getAs<VectorType>()->getElementType();
1251
1252 assert(SrcTy->isVectorTy() &&
1253 "ConvertVector source IR type must be a vector");
1254 assert(DstTy->isVectorTy() &&
1255 "ConvertVector destination IR type must be a vector");
1256
1257 llvm::Type *SrcEltTy = SrcTy->getVectorElementType(),
1258 *DstEltTy = DstTy->getVectorElementType();
1259
1260 if (DstEltType->isBooleanType()) {
1261 assert((SrcEltTy->isFloatingPointTy() ||
1262 isa<llvm::IntegerType>(SrcEltTy)) && "Unknown boolean conversion");
1263
1264 llvm::Value *Zero = llvm::Constant::getNullValue(SrcTy);
1265 if (SrcEltTy->isFloatingPointTy()) {
1266 return Builder.CreateFCmpUNE(Src, Zero, "tobool");
1267 } else {
1268 return Builder.CreateICmpNE(Src, Zero, "tobool");
1269 }
1270 }
1271
1272 // We have the arithmetic types: real int/float.
Craig Topper8a13c412014-05-21 05:09:00 +00001273 Value *Res = nullptr;
Hal Finkelc4d7c822013-09-18 03:29:45 +00001274
1275 if (isa<llvm::IntegerType>(SrcEltTy)) {
1276 bool InputSigned = SrcEltType->isSignedIntegerOrEnumerationType();
1277 if (isa<llvm::IntegerType>(DstEltTy))
1278 Res = Builder.CreateIntCast(Src, DstTy, InputSigned, "conv");
1279 else if (InputSigned)
1280 Res = Builder.CreateSIToFP(Src, DstTy, "conv");
1281 else
1282 Res = Builder.CreateUIToFP(Src, DstTy, "conv");
1283 } else if (isa<llvm::IntegerType>(DstEltTy)) {
1284 assert(SrcEltTy->isFloatingPointTy() && "Unknown real conversion");
1285 if (DstEltType->isSignedIntegerOrEnumerationType())
1286 Res = Builder.CreateFPToSI(Src, DstTy, "conv");
1287 else
1288 Res = Builder.CreateFPToUI(Src, DstTy, "conv");
1289 } else {
1290 assert(SrcEltTy->isFloatingPointTy() && DstEltTy->isFloatingPointTy() &&
1291 "Unknown real conversion");
1292 if (DstEltTy->getTypeID() < SrcEltTy->getTypeID())
1293 Res = Builder.CreateFPTrunc(Src, DstTy, "conv");
1294 else
1295 Res = Builder.CreateFPExt(Src, DstTy, "conv");
1296 }
1297
1298 return Res;
1299}
1300
Eli Friedmancb422f12009-11-26 03:22:21 +00001301Value *ScalarExprEmitter::VisitMemberExpr(MemberExpr *E) {
Richard Smith5fab0c92011-12-28 19:48:30 +00001302 llvm::APSInt Value;
1303 if (E->EvaluateAsInt(Value, CGF.getContext(), Expr::SE_AllowSideEffects)) {
Eli Friedmancb422f12009-11-26 03:22:21 +00001304 if (E->isArrow())
1305 CGF.EmitScalarExpr(E->getBase());
1306 else
1307 EmitLValue(E->getBase());
Richard Smith5fab0c92011-12-28 19:48:30 +00001308 return Builder.getInt(Value);
Eli Friedmancb422f12009-11-26 03:22:21 +00001309 }
Devang Patel44b8bf02010-10-04 21:46:04 +00001310
Eli Friedmancb422f12009-11-26 03:22:21 +00001311 return EmitLoadOfLValue(E);
1312}
Eli Friedmana1b4ed82008-05-14 19:38:39 +00001313
Chris Lattner2da04b32007-08-24 05:35:26 +00001314Value *ScalarExprEmitter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00001315 TestAndClearIgnoreResultAssign();
1316
Chris Lattner2da04b32007-08-24 05:35:26 +00001317 // Emit subscript expressions in rvalue context's. For most cases, this just
1318 // loads the lvalue formed by the subscript expr. However, we have to be
1319 // careful, because the base of a vector subscript is occasionally an rvalue,
1320 // so we can't get it as an lvalue.
1321 if (!E->getBase()->getType()->isVectorType())
1322 return EmitLoadOfLValue(E);
Mike Stump4a3999f2009-09-09 13:00:44 +00001323
Chris Lattner2da04b32007-08-24 05:35:26 +00001324 // Handle the vector case. The base must be a vector, the index must be an
1325 // integer value.
1326 Value *Base = Visit(E->getBase());
1327 Value *Idx = Visit(E->getIdx());
Richard Smith539e4a72013-02-23 02:53:19 +00001328 QualType IdxTy = E->getIdx()->getType();
1329
Alexey Samsonovedf99a92014-11-07 22:29:38 +00001330 if (CGF.SanOpts.has(SanitizerKind::ArrayBounds))
Richard Smith539e4a72013-02-23 02:53:19 +00001331 CGF.EmitBoundsCheck(E, E->getBase(), Idx, IdxTy, /*Accessed*/true);
1332
Chris Lattner2da04b32007-08-24 05:35:26 +00001333 return Builder.CreateExtractElement(Base, Idx, "vecext");
1334}
1335
Nate Begeman19351632009-10-18 20:10:40 +00001336static llvm::Constant *getMaskElt(llvm::ShuffleVectorInst *SVI, unsigned Idx,
Chris Lattner2192fe52011-07-18 04:24:23 +00001337 unsigned Off, llvm::Type *I32Ty) {
Nate Begeman19351632009-10-18 20:10:40 +00001338 int MV = SVI->getMaskValue(Idx);
Craig Toppera97d7e72013-07-26 06:16:11 +00001339 if (MV == -1)
Nate Begeman19351632009-10-18 20:10:40 +00001340 return llvm::UndefValue::get(I32Ty);
1341 return llvm::ConstantInt::get(I32Ty, Off+MV);
1342}
1343
Simon Pilgrim4034b9f2015-08-02 15:28:10 +00001344static llvm::Constant *getAsInt32(llvm::ConstantInt *C, llvm::Type *I32Ty) {
1345 if (C->getBitWidth() != 32) {
1346 assert(llvm::ConstantInt::isValueValidForType(I32Ty,
1347 C->getZExtValue()) &&
1348 "Index operand too large for shufflevector mask!");
1349 return llvm::ConstantInt::get(I32Ty, C->getZExtValue());
1350 }
1351 return C;
1352}
1353
Nate Begeman19351632009-10-18 20:10:40 +00001354Value *ScalarExprEmitter::VisitInitListExpr(InitListExpr *E) {
1355 bool Ignore = TestAndClearIgnoreResultAssign();
1356 (void)Ignore;
1357 assert (Ignore == false && "init list ignored");
1358 unsigned NumInitElements = E->getNumInits();
Craig Toppera97d7e72013-07-26 06:16:11 +00001359
Nate Begeman19351632009-10-18 20:10:40 +00001360 if (E->hadArrayRangeDesignator())
1361 CGF.ErrorUnsupported(E, "GNU array range designator extension");
Craig Toppera97d7e72013-07-26 06:16:11 +00001362
Chris Lattner2192fe52011-07-18 04:24:23 +00001363 llvm::VectorType *VType =
Nate Begeman19351632009-10-18 20:10:40 +00001364 dyn_cast<llvm::VectorType>(ConvertType(E->getType()));
Craig Toppera97d7e72013-07-26 06:16:11 +00001365
Sebastian Redl12757ab2011-09-24 17:48:14 +00001366 if (!VType) {
1367 if (NumInitElements == 0) {
1368 // C++11 value-initialization for the scalar.
1369 return EmitNullValue(E->getType());
1370 }
1371 // We have a scalar in braces. Just use the first element.
Nate Begeman19351632009-10-18 20:10:40 +00001372 return Visit(E->getInit(0));
Sebastian Redl12757ab2011-09-24 17:48:14 +00001373 }
Craig Toppera97d7e72013-07-26 06:16:11 +00001374
Nate Begeman19351632009-10-18 20:10:40 +00001375 unsigned ResElts = VType->getNumElements();
Craig Toppera97d7e72013-07-26 06:16:11 +00001376
1377 // Loop over initializers collecting the Value for each, and remembering
Nate Begeman19351632009-10-18 20:10:40 +00001378 // whether the source was swizzle (ExtVectorElementExpr). This will allow
1379 // us to fold the shuffle for the swizzle into the shuffle for the vector
1380 // initializer, since LLVM optimizers generally do not want to touch
1381 // shuffles.
1382 unsigned CurIdx = 0;
1383 bool VIsUndefShuffle = false;
1384 llvm::Value *V = llvm::UndefValue::get(VType);
1385 for (unsigned i = 0; i != NumInitElements; ++i) {
1386 Expr *IE = E->getInit(i);
1387 Value *Init = Visit(IE);
Chris Lattner0e62c1c2011-07-23 10:55:15 +00001388 SmallVector<llvm::Constant*, 16> Args;
Craig Toppera97d7e72013-07-26 06:16:11 +00001389
Chris Lattner2192fe52011-07-18 04:24:23 +00001390 llvm::VectorType *VVT = dyn_cast<llvm::VectorType>(Init->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00001391
Nate Begeman19351632009-10-18 20:10:40 +00001392 // Handle scalar elements. If the scalar initializer is actually one
Craig Toppera97d7e72013-07-26 06:16:11 +00001393 // element of a different vector of the same width, use shuffle instead of
Nate Begeman19351632009-10-18 20:10:40 +00001394 // extract+insert.
1395 if (!VVT) {
1396 if (isa<ExtVectorElementExpr>(IE)) {
1397 llvm::ExtractElementInst *EI = cast<llvm::ExtractElementInst>(Init);
1398
1399 if (EI->getVectorOperandType()->getNumElements() == ResElts) {
1400 llvm::ConstantInt *C = cast<llvm::ConstantInt>(EI->getIndexOperand());
Craig Topper8a13c412014-05-21 05:09:00 +00001401 Value *LHS = nullptr, *RHS = nullptr;
Nate Begeman19351632009-10-18 20:10:40 +00001402 if (CurIdx == 0) {
1403 // insert into undef -> shuffle (src, undef)
Simon Pilgrim4034b9f2015-08-02 15:28:10 +00001404 // shufflemask must use an i32
1405 Args.push_back(getAsInt32(C, CGF.Int32Ty));
Benjamin Kramer8001f742012-02-14 12:06:21 +00001406 Args.resize(ResElts, llvm::UndefValue::get(CGF.Int32Ty));
Nate Begeman19351632009-10-18 20:10:40 +00001407
1408 LHS = EI->getVectorOperand();
1409 RHS = V;
1410 VIsUndefShuffle = true;
1411 } else if (VIsUndefShuffle) {
1412 // insert into undefshuffle && size match -> shuffle (v, src)
1413 llvm::ShuffleVectorInst *SVV = cast<llvm::ShuffleVectorInst>(V);
1414 for (unsigned j = 0; j != CurIdx; ++j)
Chris Lattner5e016ae2010-06-27 07:15:29 +00001415 Args.push_back(getMaskElt(SVV, j, 0, CGF.Int32Ty));
Chris Lattner2531eb42011-04-19 22:55:03 +00001416 Args.push_back(Builder.getInt32(ResElts + C->getZExtValue()));
Benjamin Kramer8001f742012-02-14 12:06:21 +00001417 Args.resize(ResElts, llvm::UndefValue::get(CGF.Int32Ty));
1418
Nate Begeman19351632009-10-18 20:10:40 +00001419 LHS = cast<llvm::ShuffleVectorInst>(V)->getOperand(0);
1420 RHS = EI->getVectorOperand();
1421 VIsUndefShuffle = false;
1422 }
1423 if (!Args.empty()) {
Chris Lattner91c08ad2011-02-15 00:14:06 +00001424 llvm::Constant *Mask = llvm::ConstantVector::get(Args);
Nate Begeman19351632009-10-18 20:10:40 +00001425 V = Builder.CreateShuffleVector(LHS, RHS, Mask);
1426 ++CurIdx;
1427 continue;
1428 }
1429 }
1430 }
Chris Lattner2531eb42011-04-19 22:55:03 +00001431 V = Builder.CreateInsertElement(V, Init, Builder.getInt32(CurIdx),
1432 "vecinit");
Nate Begeman19351632009-10-18 20:10:40 +00001433 VIsUndefShuffle = false;
1434 ++CurIdx;
1435 continue;
1436 }
Craig Toppera97d7e72013-07-26 06:16:11 +00001437
Nate Begeman19351632009-10-18 20:10:40 +00001438 unsigned InitElts = VVT->getNumElements();
1439
Craig Toppera97d7e72013-07-26 06:16:11 +00001440 // If the initializer is an ExtVecEltExpr (a swizzle), and the swizzle's
Nate Begeman19351632009-10-18 20:10:40 +00001441 // input is the same width as the vector being constructed, generate an
1442 // optimized shuffle of the swizzle input into the result.
Nate Begemanb8326be2009-10-25 02:26:01 +00001443 unsigned Offset = (CurIdx == 0) ? 0 : ResElts;
Nate Begeman19351632009-10-18 20:10:40 +00001444 if (isa<ExtVectorElementExpr>(IE)) {
1445 llvm::ShuffleVectorInst *SVI = cast<llvm::ShuffleVectorInst>(Init);
1446 Value *SVOp = SVI->getOperand(0);
Chris Lattner2192fe52011-07-18 04:24:23 +00001447 llvm::VectorType *OpTy = cast<llvm::VectorType>(SVOp->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00001448
Nate Begeman19351632009-10-18 20:10:40 +00001449 if (OpTy->getNumElements() == ResElts) {
Nate Begeman19351632009-10-18 20:10:40 +00001450 for (unsigned j = 0; j != CurIdx; ++j) {
1451 // If the current vector initializer is a shuffle with undef, merge
1452 // this shuffle directly into it.
1453 if (VIsUndefShuffle) {
1454 Args.push_back(getMaskElt(cast<llvm::ShuffleVectorInst>(V), j, 0,
Chris Lattner5e016ae2010-06-27 07:15:29 +00001455 CGF.Int32Ty));
Nate Begeman19351632009-10-18 20:10:40 +00001456 } else {
Chris Lattner2531eb42011-04-19 22:55:03 +00001457 Args.push_back(Builder.getInt32(j));
Nate Begeman19351632009-10-18 20:10:40 +00001458 }
1459 }
1460 for (unsigned j = 0, je = InitElts; j != je; ++j)
Chris Lattner5e016ae2010-06-27 07:15:29 +00001461 Args.push_back(getMaskElt(SVI, j, Offset, CGF.Int32Ty));
Benjamin Kramer8001f742012-02-14 12:06:21 +00001462 Args.resize(ResElts, llvm::UndefValue::get(CGF.Int32Ty));
Nate Begeman19351632009-10-18 20:10:40 +00001463
1464 if (VIsUndefShuffle)
1465 V = cast<llvm::ShuffleVectorInst>(V)->getOperand(0);
1466
1467 Init = SVOp;
1468 }
1469 }
1470
1471 // Extend init to result vector length, and then shuffle its contribution
1472 // to the vector initializer into V.
1473 if (Args.empty()) {
1474 for (unsigned j = 0; j != InitElts; ++j)
Chris Lattner2531eb42011-04-19 22:55:03 +00001475 Args.push_back(Builder.getInt32(j));
Benjamin Kramer8001f742012-02-14 12:06:21 +00001476 Args.resize(ResElts, llvm::UndefValue::get(CGF.Int32Ty));
Chris Lattner91c08ad2011-02-15 00:14:06 +00001477 llvm::Constant *Mask = llvm::ConstantVector::get(Args);
Nate Begeman19351632009-10-18 20:10:40 +00001478 Init = Builder.CreateShuffleVector(Init, llvm::UndefValue::get(VVT),
Nate Begemanb8326be2009-10-25 02:26:01 +00001479 Mask, "vext");
Nate Begeman19351632009-10-18 20:10:40 +00001480
1481 Args.clear();
1482 for (unsigned j = 0; j != CurIdx; ++j)
Chris Lattner2531eb42011-04-19 22:55:03 +00001483 Args.push_back(Builder.getInt32(j));
Nate Begeman19351632009-10-18 20:10:40 +00001484 for (unsigned j = 0; j != InitElts; ++j)
Chris Lattner2531eb42011-04-19 22:55:03 +00001485 Args.push_back(Builder.getInt32(j+Offset));
Benjamin Kramer8001f742012-02-14 12:06:21 +00001486 Args.resize(ResElts, llvm::UndefValue::get(CGF.Int32Ty));
Nate Begeman19351632009-10-18 20:10:40 +00001487 }
1488
1489 // If V is undef, make sure it ends up on the RHS of the shuffle to aid
1490 // merging subsequent shuffles into this one.
1491 if (CurIdx == 0)
1492 std::swap(V, Init);
Chris Lattner91c08ad2011-02-15 00:14:06 +00001493 llvm::Constant *Mask = llvm::ConstantVector::get(Args);
Nate Begeman19351632009-10-18 20:10:40 +00001494 V = Builder.CreateShuffleVector(V, Init, Mask, "vecinit");
1495 VIsUndefShuffle = isa<llvm::UndefValue>(Init);
1496 CurIdx += InitElts;
1497 }
Craig Toppera97d7e72013-07-26 06:16:11 +00001498
Nate Begeman19351632009-10-18 20:10:40 +00001499 // FIXME: evaluate codegen vs. shuffling against constant null vector.
1500 // Emit remaining default initializers.
Chris Lattner2192fe52011-07-18 04:24:23 +00001501 llvm::Type *EltTy = VType->getElementType();
Craig Toppera97d7e72013-07-26 06:16:11 +00001502
Nate Begeman19351632009-10-18 20:10:40 +00001503 // Emit remaining default initializers
1504 for (/* Do not initialize i*/; CurIdx < ResElts; ++CurIdx) {
Chris Lattner2531eb42011-04-19 22:55:03 +00001505 Value *Idx = Builder.getInt32(CurIdx);
Nate Begeman19351632009-10-18 20:10:40 +00001506 llvm::Value *Init = llvm::Constant::getNullValue(EltTy);
1507 V = Builder.CreateInsertElement(V, Init, Idx, "vecinit");
1508 }
1509 return V;
1510}
1511
John McCall7f416cc2015-09-08 08:05:57 +00001512bool CodeGenFunction::ShouldNullCheckClassCastValue(const CastExpr *CE) {
Anders Carlsson8c793172009-11-23 17:57:54 +00001513 const Expr *E = CE->getSubExpr();
John McCalld9c7c6562010-03-30 23:58:03 +00001514
John McCalle3027922010-08-25 11:45:40 +00001515 if (CE->getCastKind() == CK_UncheckedDerivedToBase)
John McCalld9c7c6562010-03-30 23:58:03 +00001516 return false;
Craig Toppera97d7e72013-07-26 06:16:11 +00001517
John McCall7f416cc2015-09-08 08:05:57 +00001518 if (isa<CXXThisExpr>(E->IgnoreParens())) {
Anders Carlsson8c793172009-11-23 17:57:54 +00001519 // We always assume that 'this' is never null.
1520 return false;
1521 }
Craig Toppera97d7e72013-07-26 06:16:11 +00001522
Anders Carlsson8c793172009-11-23 17:57:54 +00001523 if (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(CE)) {
Sebastian Redlc57d34b2010-07-20 04:20:21 +00001524 // And that glvalue casts are never null.
John McCall2536c6d2010-08-25 10:28:54 +00001525 if (ICE->getValueKind() != VK_RValue)
Anders Carlsson8c793172009-11-23 17:57:54 +00001526 return false;
1527 }
1528
1529 return true;
1530}
1531
Chris Lattner2da04b32007-08-24 05:35:26 +00001532// VisitCastExpr - Emit code for an explicit or implicit cast. Implicit casts
1533// have to handle a more broad range of conversions than explicit casts, as they
1534// handle things like function to ptr-to-function decay etc.
John McCall23c29fe2011-06-24 21:55:10 +00001535Value *ScalarExprEmitter::VisitCastExpr(CastExpr *CE) {
Eli Friedmane96f1d32009-11-27 04:41:50 +00001536 Expr *E = CE->getSubExpr();
Anders Carlsson8c978b42009-09-22 22:00:46 +00001537 QualType DestTy = CE->getType();
John McCalle3027922010-08-25 11:45:40 +00001538 CastKind Kind = CE->getCastKind();
Craig Toppera97d7e72013-07-26 06:16:11 +00001539
John McCalle399e5b2016-01-27 18:32:30 +00001540 // These cases are generally not written to ignore the result of
1541 // evaluating their sub-expressions, so we clear this now.
1542 bool Ignored = TestAndClearIgnoreResultAssign();
Mike Stump4a3999f2009-09-09 13:00:44 +00001543
Eli Friedman0dfc6802009-11-27 02:07:44 +00001544 // Since almost all cast kinds apply to scalars, this switch doesn't have
1545 // a default case, so the compiler will warn on a missing case. The cases
1546 // are in the same order as in the CastKind enum.
Anders Carlsson3df53bc2009-08-24 18:26:39 +00001547 switch (Kind) {
John McCall8cb679e2010-11-15 09:13:47 +00001548 case CK_Dependent: llvm_unreachable("dependent cast kind in IR gen!");
Eli Friedman34866c72012-08-31 00:14:07 +00001549 case CK_BuiltinFnToFnPtr:
1550 llvm_unreachable("builtin functions are handled elsewhere");
1551
Craig Toppera97d7e72013-07-26 06:16:11 +00001552 case CK_LValueBitCast:
John McCalle3027922010-08-25 11:45:40 +00001553 case CK_ObjCObjectLValueCast: {
John McCall7f416cc2015-09-08 08:05:57 +00001554 Address Addr = EmitLValue(E).getAddress();
Alexey Bataevf2440332015-10-07 10:22:08 +00001555 Addr = Builder.CreateElementBitCast(Addr, CGF.ConvertTypeForMem(DestTy));
John McCall7f416cc2015-09-08 08:05:57 +00001556 LValue LV = CGF.MakeAddrLValue(Addr, DestTy);
1557 return EmitLoadOfLValue(LV, CE->getExprLoc());
Douglas Gregor51954272010-07-13 23:17:26 +00001558 }
John McCallcd78e802011-09-10 01:16:55 +00001559
John McCall9320b872011-09-09 05:25:32 +00001560 case CK_CPointerToObjCPointerCast:
1561 case CK_BlockPointerToObjCPointerCast:
John McCalle3027922010-08-25 11:45:40 +00001562 case CK_AnyPointerToBlockPointerCast:
1563 case CK_BitCast: {
Anders Carlssonf1ae6d42009-09-01 20:52:42 +00001564 Value *Src = Visit(const_cast<Expr*>(E));
David Tweede1468322013-12-11 13:39:46 +00001565 llvm::Type *SrcTy = Src->getType();
1566 llvm::Type *DstTy = ConvertType(DestTy);
Bob Wilson95a27b02014-02-17 19:20:59 +00001567 if (SrcTy->isPtrOrPtrVectorTy() && DstTy->isPtrOrPtrVectorTy() &&
David Tweede1468322013-12-11 13:39:46 +00001568 SrcTy->getPointerAddressSpace() != DstTy->getPointerAddressSpace()) {
Anastasia Stulova5d8ad8a2014-11-26 15:36:41 +00001569 llvm_unreachable("wrong cast for pointers in different address spaces"
1570 "(must be an address space cast)!");
David Tweede1468322013-12-11 13:39:46 +00001571 }
Peter Collingbourned2926c92015-03-14 02:42:25 +00001572
1573 if (CGF.SanOpts.has(SanitizerKind::CFIUnrelatedCast)) {
1574 if (auto PT = DestTy->getAs<PointerType>())
1575 CGF.EmitVTablePtrCheckForCast(PT->getPointeeType(), Src,
Peter Collingbourne6708c4a2015-06-19 01:51:54 +00001576 /*MayBeNull=*/true,
1577 CodeGenFunction::CFITCK_UnrelatedCast,
1578 CE->getLocStart());
Peter Collingbourned2926c92015-03-14 02:42:25 +00001579 }
1580
David Tweede1468322013-12-11 13:39:46 +00001581 return Builder.CreateBitCast(Src, DstTy);
1582 }
1583 case CK_AddressSpaceConversion: {
Yaxun Liu402804b2016-12-15 08:09:08 +00001584 Expr::EvalResult Result;
1585 if (E->EvaluateAsRValue(Result, CGF.getContext()) &&
1586 Result.Val.isNullPointer()) {
1587 // If E has side effect, it is emitted even if its final result is a
1588 // null pointer. In that case, a DCE pass should be able to
1589 // eliminate the useless instructions emitted during translating E.
1590 if (Result.HasSideEffects)
1591 Visit(E);
1592 return CGF.CGM.getNullPointer(cast<llvm::PointerType>(
1593 ConvertType(DestTy)), DestTy);
1594 }
Yaxun Liub7b6d0f2016-04-12 19:03:49 +00001595 // Since target may map different address spaces in AST to the same address
1596 // space, an address space conversion may end up as a bitcast.
Yaxun Liu6d96f1632017-05-18 18:51:09 +00001597 return CGF.CGM.getTargetCodeGenInfo().performAddrSpaceCast(
1598 CGF, Visit(E), E->getType()->getPointeeType().getAddressSpace(),
1599 DestTy->getPointeeType().getAddressSpace(), ConvertType(DestTy));
Anders Carlssonf1ae6d42009-09-01 20:52:42 +00001600 }
David Chisnallfa35df62012-01-16 17:27:18 +00001601 case CK_AtomicToNonAtomic:
1602 case CK_NonAtomicToAtomic:
John McCalle3027922010-08-25 11:45:40 +00001603 case CK_NoOp:
1604 case CK_UserDefinedConversion:
Eli Friedmanc08bdea2009-11-16 21:33:53 +00001605 return Visit(const_cast<Expr*>(E));
Mike Stump4a3999f2009-09-09 13:00:44 +00001606
John McCalle3027922010-08-25 11:45:40 +00001607 case CK_BaseToDerived: {
Jordan Rose7bb26112012-10-03 01:08:28 +00001608 const CXXRecordDecl *DerivedClassDecl = DestTy->getPointeeCXXRecordDecl();
1609 assert(DerivedClassDecl && "BaseToDerived arg isn't a C++ object pointer!");
1610
John McCall7f416cc2015-09-08 08:05:57 +00001611 Address Base = CGF.EmitPointerWithAlignment(E);
1612 Address Derived =
1613 CGF.GetAddressOfDerivedClass(Base, DerivedClassDecl,
Filipe Cabecinhas178a8df2013-08-08 01:08:17 +00001614 CE->path_begin(), CE->path_end(),
John McCall7f416cc2015-09-08 08:05:57 +00001615 CGF.ShouldNullCheckClassCastValue(CE));
Filipe Cabecinhas178a8df2013-08-08 01:08:17 +00001616
Richard Smith2c5868c2013-02-13 21:18:23 +00001617 // C++11 [expr.static.cast]p11: Behavior is undefined if a downcast is
1618 // performed and the object is not of the derived type.
Alexey Samsonovac4afe42014-07-07 23:59:57 +00001619 if (CGF.sanitizePerformTypeCheck())
Richard Smith2c5868c2013-02-13 21:18:23 +00001620 CGF.EmitTypeCheck(CodeGenFunction::TCK_DowncastPointer, CE->getExprLoc(),
John McCall7f416cc2015-09-08 08:05:57 +00001621 Derived.getPointer(), DestTy->getPointeeType());
Richard Smith2c5868c2013-02-13 21:18:23 +00001622
Peter Collingbourned2926c92015-03-14 02:42:25 +00001623 if (CGF.SanOpts.has(SanitizerKind::CFIDerivedCast))
John McCall7f416cc2015-09-08 08:05:57 +00001624 CGF.EmitVTablePtrCheckForCast(DestTy->getPointeeType(),
1625 Derived.getPointer(),
Peter Collingbourne6708c4a2015-06-19 01:51:54 +00001626 /*MayBeNull=*/true,
1627 CodeGenFunction::CFITCK_DerivedCast,
1628 CE->getLocStart());
Peter Collingbourned2926c92015-03-14 02:42:25 +00001629
John McCall7f416cc2015-09-08 08:05:57 +00001630 return Derived.getPointer();
Anders Carlsson8c793172009-11-23 17:57:54 +00001631 }
John McCalle3027922010-08-25 11:45:40 +00001632 case CK_UncheckedDerivedToBase:
1633 case CK_DerivedToBase: {
John McCall7f416cc2015-09-08 08:05:57 +00001634 // The EmitPointerWithAlignment path does this fine; just discard
1635 // the alignment.
1636 return CGF.EmitPointerWithAlignment(CE).getPointer();
Anders Carlsson12f5a252009-09-12 04:57:16 +00001637 }
John McCall7f416cc2015-09-08 08:05:57 +00001638
Anders Carlsson8a01a752011-04-11 02:03:26 +00001639 case CK_Dynamic: {
John McCall7f416cc2015-09-08 08:05:57 +00001640 Address V = CGF.EmitPointerWithAlignment(E);
Eli Friedman0dfc6802009-11-27 02:07:44 +00001641 const CXXDynamicCastExpr *DCE = cast<CXXDynamicCastExpr>(CE);
1642 return CGF.EmitDynamicCast(V, DCE);
1643 }
Eli Friedmane96f1d32009-11-27 04:41:50 +00001644
John McCall7f416cc2015-09-08 08:05:57 +00001645 case CK_ArrayToPointerDecay:
1646 return CGF.EmitArrayToPointerDecay(E).getPointer();
John McCalle3027922010-08-25 11:45:40 +00001647 case CK_FunctionToPointerDecay:
John McCall7f416cc2015-09-08 08:05:57 +00001648 return EmitLValue(E).getPointer();
Eli Friedmanc08bdea2009-11-16 21:33:53 +00001649
John McCalle84af4e2010-11-13 01:35:44 +00001650 case CK_NullToPointer:
1651 if (MustVisitNullValue(E))
1652 (void) Visit(E);
1653
Yaxun Liu402804b2016-12-15 08:09:08 +00001654 return CGF.CGM.getNullPointer(cast<llvm::PointerType>(ConvertType(DestTy)),
1655 DestTy);
John McCalle84af4e2010-11-13 01:35:44 +00001656
John McCalle3027922010-08-25 11:45:40 +00001657 case CK_NullToMemberPointer: {
John McCalle84af4e2010-11-13 01:35:44 +00001658 if (MustVisitNullValue(E))
John McCalla1dee5302010-08-22 10:59:02 +00001659 (void) Visit(E);
1660
John McCall7a9aac22010-08-23 01:21:21 +00001661 const MemberPointerType *MPT = CE->getType()->getAs<MemberPointerType>();
1662 return CGF.CGM.getCXXABI().EmitNullMemberPointer(MPT);
1663 }
Anders Carlsson12f5a252009-09-12 04:57:16 +00001664
John McCallc62bb392012-02-15 01:22:51 +00001665 case CK_ReinterpretMemberPointer:
John McCalle3027922010-08-25 11:45:40 +00001666 case CK_BaseToDerivedMemberPointer:
1667 case CK_DerivedToBaseMemberPointer: {
Eli Friedmane96f1d32009-11-27 04:41:50 +00001668 Value *Src = Visit(E);
Craig Toppera97d7e72013-07-26 06:16:11 +00001669
John McCalla1dee5302010-08-22 10:59:02 +00001670 // Note that the AST doesn't distinguish between checked and
1671 // unchecked member pointer conversions, so we always have to
1672 // implement checked conversions here. This is inefficient when
1673 // actual control flow may be required in order to perform the
1674 // check, which it is for data member pointers (but not member
1675 // function pointers on Itanium and ARM).
John McCall7a9aac22010-08-23 01:21:21 +00001676 return CGF.CGM.getCXXABI().EmitMemberPointerConversion(CGF, CE, Src);
Eli Friedmane96f1d32009-11-27 04:41:50 +00001677 }
John McCall31168b02011-06-15 23:02:42 +00001678
John McCall2d637d22011-09-10 06:18:15 +00001679 case CK_ARCProduceObject:
John McCall31168b02011-06-15 23:02:42 +00001680 return CGF.EmitARCRetainScalarExpr(E);
John McCall2d637d22011-09-10 06:18:15 +00001681 case CK_ARCConsumeObject:
John McCall31168b02011-06-15 23:02:42 +00001682 return CGF.EmitObjCConsumeObject(E->getType(), Visit(E));
John McCalle399e5b2016-01-27 18:32:30 +00001683 case CK_ARCReclaimReturnedObject:
1684 return CGF.EmitARCReclaimReturnedObject(E, /*allowUnsafe*/ Ignored);
John McCallff613032011-10-04 06:23:45 +00001685 case CK_ARCExtendBlockObject:
1686 return CGF.EmitARCExtendBlockObject(E);
John McCall31168b02011-06-15 23:02:42 +00001687
Douglas Gregored90df32012-02-22 05:02:47 +00001688 case CK_CopyAndAutoreleaseBlockObject:
Eli Friedmanec75fec2012-02-28 01:08:45 +00001689 return CGF.EmitBlockCopyAndAutorelease(Visit(E), E->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00001690
John McCallc5e62b42010-11-13 09:02:35 +00001691 case CK_FloatingRealToComplex:
1692 case CK_FloatingComplexCast:
1693 case CK_IntegralRealToComplex:
1694 case CK_IntegralComplexCast:
John McCalld7646252010-11-14 08:17:51 +00001695 case CK_IntegralComplexToFloatingComplex:
1696 case CK_FloatingComplexToIntegralComplex:
John McCalle3027922010-08-25 11:45:40 +00001697 case CK_ConstructorConversion:
John McCall3eba6e62010-11-16 06:21:14 +00001698 case CK_ToUnion:
1699 llvm_unreachable("scalar cast to non-scalar value");
John McCall34376a62010-12-04 03:47:34 +00001700
John McCallf3735e02010-12-01 04:43:34 +00001701 case CK_LValueToRValue:
1702 assert(CGF.getContext().hasSameUnqualifiedType(E->getType(), DestTy));
John McCall34376a62010-12-04 03:47:34 +00001703 assert(E->isGLValue() && "lvalue-to-rvalue applied to r-value!");
John McCallf3735e02010-12-01 04:43:34 +00001704 return Visit(const_cast<Expr*>(E));
Eli Friedman0dfc6802009-11-27 02:07:44 +00001705
John McCalle3027922010-08-25 11:45:40 +00001706 case CK_IntegralToPointer: {
Anders Carlsson7cd39e02009-09-15 04:48:33 +00001707 Value *Src = Visit(const_cast<Expr*>(E));
Daniel Dunbaread6824c2010-08-25 03:32:38 +00001708
Anders Carlsson094c4592009-10-18 18:12:03 +00001709 // First, convert to the correct width so that we control the kind of
1710 // extension.
Yaxun Liu26f75662016-08-19 05:17:25 +00001711 auto DestLLVMTy = ConvertType(DestTy);
1712 llvm::Type *MiddleTy = CGF.CGM.getDataLayout().getIntPtrType(DestLLVMTy);
Douglas Gregor6ab2fa82011-05-20 16:38:50 +00001713 bool InputSigned = E->getType()->isSignedIntegerOrEnumerationType();
Anders Carlsson094c4592009-10-18 18:12:03 +00001714 llvm::Value* IntResult =
1715 Builder.CreateIntCast(Src, MiddleTy, InputSigned, "conv");
Daniel Dunbaread6824c2010-08-25 03:32:38 +00001716
Yaxun Liu26f75662016-08-19 05:17:25 +00001717 return Builder.CreateIntToPtr(IntResult, DestLLVMTy);
Anders Carlsson7cd39e02009-09-15 04:48:33 +00001718 }
Eli Friedman58368522011-06-25 02:58:47 +00001719 case CK_PointerToIntegral:
1720 assert(!DestTy->isBooleanType() && "bool should use PointerToBool");
1721 return Builder.CreatePtrToInt(Visit(E), ConvertType(DestTy));
Daniel Dunbaread6824c2010-08-25 03:32:38 +00001722
John McCalle3027922010-08-25 11:45:40 +00001723 case CK_ToVoid: {
John McCalla2342eb2010-12-05 02:00:02 +00001724 CGF.EmitIgnoredExpr(E);
Craig Topper8a13c412014-05-21 05:09:00 +00001725 return nullptr;
Eli Friedmanc08bdea2009-11-16 21:33:53 +00001726 }
John McCalle3027922010-08-25 11:45:40 +00001727 case CK_VectorSplat: {
Chris Lattner2192fe52011-07-18 04:24:23 +00001728 llvm::Type *DstTy = ConvertType(DestTy);
George Burgess IVdf1ed002016-01-13 01:52:39 +00001729 Value *Elt = Visit(const_cast<Expr*>(E));
Eli Friedmanc08bdea2009-11-16 21:33:53 +00001730 // Splat the element across to all elements
Craig Topperf2f1a092016-07-08 02:17:35 +00001731 unsigned NumElements = DstTy->getVectorNumElements();
Alp Toker5f072d82014-04-19 23:55:49 +00001732 return Builder.CreateVectorSplat(NumElements, Elt, "splat");
Eli Friedmanc08bdea2009-11-16 21:33:53 +00001733 }
John McCall8cb679e2010-11-15 09:13:47 +00001734
John McCalle3027922010-08-25 11:45:40 +00001735 case CK_IntegralCast:
1736 case CK_IntegralToFloating:
1737 case CK_FloatingToIntegral:
1738 case CK_FloatingCast:
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001739 return EmitScalarConversion(Visit(E), E->getType(), DestTy,
1740 CE->getExprLoc());
George Burgess IVdf1ed002016-01-13 01:52:39 +00001741 case CK_BooleanToSignedIntegral:
1742 return EmitScalarConversion(Visit(E), E->getType(), DestTy,
1743 CE->getExprLoc(),
1744 /*TreatBooleanAsSigned=*/true);
John McCall8cb679e2010-11-15 09:13:47 +00001745 case CK_IntegralToBoolean:
1746 return EmitIntToBoolConversion(Visit(E));
1747 case CK_PointerToBoolean:
Yaxun Liu402804b2016-12-15 08:09:08 +00001748 return EmitPointerToBoolConversion(Visit(E), E->getType());
John McCall8cb679e2010-11-15 09:13:47 +00001749 case CK_FloatingToBoolean:
1750 return EmitFloatToBoolConversion(Visit(E));
John McCalle3027922010-08-25 11:45:40 +00001751 case CK_MemberPointerToBoolean: {
John McCall7a9aac22010-08-23 01:21:21 +00001752 llvm::Value *MemPtr = Visit(E);
1753 const MemberPointerType *MPT = E->getType()->getAs<MemberPointerType>();
1754 return CGF.CGM.getCXXABI().EmitMemberPointerIsNotNull(CGF, MemPtr, MPT);
Anders Carlsson3df53bc2009-08-24 18:26:39 +00001755 }
John McCalld7646252010-11-14 08:17:51 +00001756
1757 case CK_FloatingComplexToReal:
1758 case CK_IntegralComplexToReal:
John McCall07bb1962010-11-16 10:08:07 +00001759 return CGF.EmitComplexExpr(E, false, true).first;
John McCalld7646252010-11-14 08:17:51 +00001760
1761 case CK_FloatingComplexToBoolean:
1762 case CK_IntegralComplexToBoolean: {
John McCall07bb1962010-11-16 10:08:07 +00001763 CodeGenFunction::ComplexPairTy V = CGF.EmitComplexExpr(E);
John McCalld7646252010-11-14 08:17:51 +00001764
1765 // TODO: kill this function off, inline appropriate case here
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001766 return EmitComplexToScalarConversion(V, E->getType(), DestTy,
1767 CE->getExprLoc());
John McCalld7646252010-11-14 08:17:51 +00001768 }
1769
Guy Benyei1b4fb3e2013-01-20 12:31:11 +00001770 case CK_ZeroToOCLEvent: {
Alp Tokerd4733632013-12-05 04:47:09 +00001771 assert(DestTy->isEventT() && "CK_ZeroToOCLEvent cast on non-event type");
Guy Benyei1b4fb3e2013-01-20 12:31:11 +00001772 return llvm::Constant::getNullValue(ConvertType(DestTy));
1773 }
1774
Egor Churaev89831422016-12-23 14:55:49 +00001775 case CK_ZeroToOCLQueue: {
1776 assert(DestTy->isQueueT() && "CK_ZeroToOCLQueue cast on non queue_t type");
1777 return llvm::Constant::getNullValue(ConvertType(DestTy));
1778 }
1779
Yaxun Liu0bc4b2d2016-07-28 19:26:30 +00001780 case CK_IntToOCLSampler:
1781 return CGF.CGM.createOpenCLIntToSamplerConversion(E, CGF);
1782
1783 } // end of switch
Mike Stump4a3999f2009-09-09 13:00:44 +00001784
John McCall3eba6e62010-11-16 06:21:14 +00001785 llvm_unreachable("unknown scalar cast");
Chris Lattner2da04b32007-08-24 05:35:26 +00001786}
1787
Chris Lattner04a913b2007-08-31 22:09:40 +00001788Value *ScalarExprEmitter::VisitStmtExpr(const StmtExpr *E) {
John McCallce1de612011-01-26 04:00:11 +00001789 CodeGenFunction::StmtExprEvaluation eval(CGF);
John McCall7f416cc2015-09-08 08:05:57 +00001790 Address RetAlloca = CGF.EmitCompoundStmt(*E->getSubStmt(),
1791 !E->getType()->isVoidType());
1792 if (!RetAlloca.isValid())
Craig Topper8a13c412014-05-21 05:09:00 +00001793 return nullptr;
Nick Lewycky2d84e842013-10-02 02:29:49 +00001794 return CGF.EmitLoadOfScalar(CGF.MakeAddrLValue(RetAlloca, E->getType()),
1795 E->getExprLoc());
Chris Lattner04a913b2007-08-31 22:09:40 +00001796}
1797
Reid Kleckner092d0652017-03-06 22:18:34 +00001798Value *ScalarExprEmitter::VisitExprWithCleanups(ExprWithCleanups *E) {
1799 CGF.enterFullExpression(E);
1800 CodeGenFunction::RunCleanupsScope Scope(CGF);
1801 Value *V = Visit(E->getSubExpr());
1802 // Defend against dominance problems caused by jumps out of expression
1803 // evaluation through the shared cleanup block.
1804 Scope.ForceCleanup({&V});
1805 return V;
1806}
1807
Chris Lattner2da04b32007-08-24 05:35:26 +00001808//===----------------------------------------------------------------------===//
1809// Unary Operators
1810//===----------------------------------------------------------------------===//
1811
Alexey Samsonovf6246502015-04-23 01:50:45 +00001812static BinOpInfo createBinOpInfoFromIncDec(const UnaryOperator *E,
1813 llvm::Value *InVal, bool IsInc) {
1814 BinOpInfo BinOp;
1815 BinOp.LHS = InVal;
1816 BinOp.RHS = llvm::ConstantInt::get(InVal->getType(), 1, false);
1817 BinOp.Ty = E->getType();
1818 BinOp.Opcode = IsInc ? BO_Add : BO_Sub;
Adam Nemet484aa452017-03-27 19:17:25 +00001819 // FIXME: once UnaryOperator carries FPFeatures, copy it here.
Alexey Samsonovf6246502015-04-23 01:50:45 +00001820 BinOp.E = E;
1821 return BinOp;
1822}
1823
1824llvm::Value *ScalarExprEmitter::EmitIncDecConsiderOverflowBehavior(
1825 const UnaryOperator *E, llvm::Value *InVal, bool IsInc) {
1826 llvm::Value *Amount =
1827 llvm::ConstantInt::get(InVal->getType(), IsInc ? 1 : -1, true);
1828 StringRef Name = IsInc ? "inc" : "dec";
Richard Smith9c6890a2012-11-01 22:30:59 +00001829 switch (CGF.getLangOpts().getSignedOverflowBehavior()) {
Anton Yartsev85129b82011-02-07 02:17:30 +00001830 case LangOptions::SOB_Defined:
Alexey Samsonovf6246502015-04-23 01:50:45 +00001831 return Builder.CreateAdd(InVal, Amount, Name);
Richard Smith3e056de2012-08-25 00:32:28 +00001832 case LangOptions::SOB_Undefined:
Alexey Samsonovedf99a92014-11-07 22:29:38 +00001833 if (!CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow))
Alexey Samsonovf6246502015-04-23 01:50:45 +00001834 return Builder.CreateNSWAdd(InVal, Amount, Name);
Richard Smith3e056de2012-08-25 00:32:28 +00001835 // Fall through.
Anton Yartsev85129b82011-02-07 02:17:30 +00001836 case LangOptions::SOB_Trapping:
Vedant Kumar82ee16b2017-02-25 00:43:36 +00001837 if (IsWidenedIntegerOp(CGF.getContext(), E->getSubExpr()))
1838 return Builder.CreateNSWAdd(InVal, Amount, Name);
Alexey Samsonovf6246502015-04-23 01:50:45 +00001839 return EmitOverflowCheckedBinOp(createBinOpInfoFromIncDec(E, InVal, IsInc));
Anton Yartsev85129b82011-02-07 02:17:30 +00001840 }
David Blaikie83d382b2011-09-23 05:06:16 +00001841 llvm_unreachable("Unknown SignedOverflowBehaviorTy");
Anton Yartsev85129b82011-02-07 02:17:30 +00001842}
1843
John McCalle3dc1702011-02-15 09:22:45 +00001844llvm::Value *
1845ScalarExprEmitter::EmitScalarPrePostIncDec(const UnaryOperator *E, LValue LV,
1846 bool isInc, bool isPre) {
Craig Toppera97d7e72013-07-26 06:16:11 +00001847
John McCalle3dc1702011-02-15 09:22:45 +00001848 QualType type = E->getSubExpr()->getType();
Craig Topper8a13c412014-05-21 05:09:00 +00001849 llvm::PHINode *atomicPHI = nullptr;
David Chisnallef78c302013-03-03 16:02:42 +00001850 llvm::Value *value;
1851 llvm::Value *input;
Anton Yartsev85129b82011-02-07 02:17:30 +00001852
John McCalle3dc1702011-02-15 09:22:45 +00001853 int amount = (isInc ? 1 : -1);
Vedant Kumar6dbf4272017-06-12 18:42:51 +00001854 bool signedIndex = !isInc;
John McCalle3dc1702011-02-15 09:22:45 +00001855
David Chisnallfa35df62012-01-16 17:27:18 +00001856 if (const AtomicType *atomicTy = type->getAs<AtomicType>()) {
David Chisnallef78c302013-03-03 16:02:42 +00001857 type = atomicTy->getValueType();
1858 if (isInc && type->isBooleanType()) {
1859 llvm::Value *True = CGF.EmitToMemory(Builder.getTrue(), type);
1860 if (isPre) {
John McCall7f416cc2015-09-08 08:05:57 +00001861 Builder.CreateStore(True, LV.getAddress(), LV.isVolatileQualified())
JF Bastien92f4ef12016-04-06 17:26:42 +00001862 ->setAtomic(llvm::AtomicOrdering::SequentiallyConsistent);
David Chisnallef78c302013-03-03 16:02:42 +00001863 return Builder.getTrue();
1864 }
1865 // For atomic bool increment, we just store true and return it for
1866 // preincrement, do an atomic swap with true for postincrement
JF Bastien92f4ef12016-04-06 17:26:42 +00001867 return Builder.CreateAtomicRMW(
1868 llvm::AtomicRMWInst::Xchg, LV.getPointer(), True,
1869 llvm::AtomicOrdering::SequentiallyConsistent);
David Chisnallef78c302013-03-03 16:02:42 +00001870 }
1871 // Special case for atomic increment / decrement on integers, emit
1872 // atomicrmw instructions. We skip this if we want to be doing overflow
Craig Toppera97d7e72013-07-26 06:16:11 +00001873 // checking, and fall into the slow path with the atomic cmpxchg loop.
David Chisnallef78c302013-03-03 16:02:42 +00001874 if (!type->isBooleanType() && type->isIntegerType() &&
1875 !(type->isUnsignedIntegerType() &&
Alexey Samsonovedf99a92014-11-07 22:29:38 +00001876 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow)) &&
David Chisnallef78c302013-03-03 16:02:42 +00001877 CGF.getLangOpts().getSignedOverflowBehavior() !=
Alexey Samsonovedf99a92014-11-07 22:29:38 +00001878 LangOptions::SOB_Trapping) {
David Chisnallef78c302013-03-03 16:02:42 +00001879 llvm::AtomicRMWInst::BinOp aop = isInc ? llvm::AtomicRMWInst::Add :
1880 llvm::AtomicRMWInst::Sub;
1881 llvm::Instruction::BinaryOps op = isInc ? llvm::Instruction::Add :
1882 llvm::Instruction::Sub;
1883 llvm::Value *amt = CGF.EmitToMemory(
1884 llvm::ConstantInt::get(ConvertType(type), 1, true), type);
1885 llvm::Value *old = Builder.CreateAtomicRMW(aop,
JF Bastien92f4ef12016-04-06 17:26:42 +00001886 LV.getPointer(), amt, llvm::AtomicOrdering::SequentiallyConsistent);
David Chisnallef78c302013-03-03 16:02:42 +00001887 return isPre ? Builder.CreateBinOp(op, old, amt) : old;
1888 }
Nick Lewycky2d84e842013-10-02 02:29:49 +00001889 value = EmitLoadOfLValue(LV, E->getExprLoc());
David Chisnallef78c302013-03-03 16:02:42 +00001890 input = value;
1891 // For every other atomic operation, we need to emit a load-op-cmpxchg loop
David Chisnallfa35df62012-01-16 17:27:18 +00001892 llvm::BasicBlock *startBB = Builder.GetInsertBlock();
1893 llvm::BasicBlock *opBB = CGF.createBasicBlock("atomic_op", CGF.CurFn);
David Chisnallef78c302013-03-03 16:02:42 +00001894 value = CGF.EmitToMemory(value, type);
David Chisnallfa35df62012-01-16 17:27:18 +00001895 Builder.CreateBr(opBB);
1896 Builder.SetInsertPoint(opBB);
1897 atomicPHI = Builder.CreatePHI(value->getType(), 2);
1898 atomicPHI->addIncoming(value, startBB);
David Chisnallfa35df62012-01-16 17:27:18 +00001899 value = atomicPHI;
David Chisnallef78c302013-03-03 16:02:42 +00001900 } else {
Nick Lewycky2d84e842013-10-02 02:29:49 +00001901 value = EmitLoadOfLValue(LV, E->getExprLoc());
David Chisnallef78c302013-03-03 16:02:42 +00001902 input = value;
David Chisnallfa35df62012-01-16 17:27:18 +00001903 }
1904
John McCalle3dc1702011-02-15 09:22:45 +00001905 // Special case of integer increment that we have to check first: bool++.
1906 // Due to promotion rules, we get:
1907 // bool++ -> bool = bool + 1
1908 // -> bool = (int)bool + 1
1909 // -> bool = ((int)bool + 1 != 0)
1910 // An interesting aspect of this is that increment is always true.
1911 // Decrement does not have this property.
1912 if (isInc && type->isBooleanType()) {
1913 value = Builder.getTrue();
1914
1915 // Most common case by far: integer increment.
1916 } else if (type->isIntegerType()) {
Eli Friedman846ded22011-03-02 01:49:12 +00001917 // Note that signed integer inc/dec with width less than int can't
1918 // overflow because of promotion rules; we're just eliding a few steps here.
Richard Smith45d099b2014-07-07 05:36:14 +00001919 bool CanOverflow = value->getType()->getIntegerBitWidth() >=
1920 CGF.IntTy->getIntegerBitWidth();
1921 if (CanOverflow && type->isSignedIntegerOrEnumerationType()) {
Alexey Samsonovf6246502015-04-23 01:50:45 +00001922 value = EmitIncDecConsiderOverflowBehavior(E, value, isInc);
Richard Smith45d099b2014-07-07 05:36:14 +00001923 } else if (CanOverflow && type->isUnsignedIntegerType() &&
Alexey Samsonovedf99a92014-11-07 22:29:38 +00001924 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow)) {
Alexey Samsonovf6246502015-04-23 01:50:45 +00001925 value =
1926 EmitOverflowCheckedBinOp(createBinOpInfoFromIncDec(E, value, isInc));
1927 } else {
1928 llvm::Value *amt = llvm::ConstantInt::get(value->getType(), amount, true);
John McCalle3dc1702011-02-15 09:22:45 +00001929 value = Builder.CreateAdd(value, amt, isInc ? "inc" : "dec");
Alexey Samsonovf6246502015-04-23 01:50:45 +00001930 }
Craig Toppera97d7e72013-07-26 06:16:11 +00001931
John McCalle3dc1702011-02-15 09:22:45 +00001932 // Next most common: pointer increment.
1933 } else if (const PointerType *ptr = type->getAs<PointerType>()) {
1934 QualType type = ptr->getPointeeType();
1935
1936 // VLA types don't have constant size.
John McCall77527a82011-06-25 01:32:37 +00001937 if (const VariableArrayType *vla
1938 = CGF.getContext().getAsVariableArrayType(type)) {
1939 llvm::Value *numElts = CGF.getVLASize(vla).first;
John McCall23c29fe2011-06-24 21:55:10 +00001940 if (!isInc) numElts = Builder.CreateNSWNeg(numElts, "vla.negsize");
Richard Smith9c6890a2012-11-01 22:30:59 +00001941 if (CGF.getLangOpts().isSignedOverflowDefined())
John McCall23c29fe2011-06-24 21:55:10 +00001942 value = Builder.CreateGEP(value, numElts, "vla.inc");
Chris Lattner2e72da942011-03-01 00:03:48 +00001943 else
Vedant Kumar6dbf4272017-06-12 18:42:51 +00001944 value = CGF.EmitCheckedInBoundsGEP(value, numElts, signedIndex,
1945 E->getExprLoc(), "vla.inc");
Craig Toppera97d7e72013-07-26 06:16:11 +00001946
John McCalle3dc1702011-02-15 09:22:45 +00001947 // Arithmetic on function pointers (!) is just +-1.
1948 } else if (type->isFunctionType()) {
Chris Lattner2531eb42011-04-19 22:55:03 +00001949 llvm::Value *amt = Builder.getInt32(amount);
John McCalle3dc1702011-02-15 09:22:45 +00001950
1951 value = CGF.EmitCastToVoidPtr(value);
Richard Smith9c6890a2012-11-01 22:30:59 +00001952 if (CGF.getLangOpts().isSignedOverflowDefined())
Chris Lattner2e72da942011-03-01 00:03:48 +00001953 value = Builder.CreateGEP(value, amt, "incdec.funcptr");
1954 else
Vedant Kumar6dbf4272017-06-12 18:42:51 +00001955 value = CGF.EmitCheckedInBoundsGEP(value, amt, signedIndex,
1956 E->getExprLoc(), "incdec.funcptr");
John McCalle3dc1702011-02-15 09:22:45 +00001957 value = Builder.CreateBitCast(value, input->getType());
1958
1959 // For everything else, we can just do a simple increment.
Anton Yartsev85129b82011-02-07 02:17:30 +00001960 } else {
Chris Lattner2531eb42011-04-19 22:55:03 +00001961 llvm::Value *amt = Builder.getInt32(amount);
Richard Smith9c6890a2012-11-01 22:30:59 +00001962 if (CGF.getLangOpts().isSignedOverflowDefined())
Chris Lattner2e72da942011-03-01 00:03:48 +00001963 value = Builder.CreateGEP(value, amt, "incdec.ptr");
1964 else
Vedant Kumar6dbf4272017-06-12 18:42:51 +00001965 value = CGF.EmitCheckedInBoundsGEP(value, amt, signedIndex,
1966 E->getExprLoc(), "incdec.ptr");
John McCalle3dc1702011-02-15 09:22:45 +00001967 }
1968
1969 // Vector increment/decrement.
1970 } else if (type->isVectorType()) {
1971 if (type->hasIntegerRepresentation()) {
1972 llvm::Value *amt = llvm::ConstantInt::get(value->getType(), amount);
1973
Eli Friedman409943e2011-05-06 18:04:18 +00001974 value = Builder.CreateAdd(value, amt, isInc ? "inc" : "dec");
John McCalle3dc1702011-02-15 09:22:45 +00001975 } else {
1976 value = Builder.CreateFAdd(
1977 value,
1978 llvm::ConstantFP::get(value->getType(), amount),
Anton Yartsev85129b82011-02-07 02:17:30 +00001979 isInc ? "inc" : "dec");
1980 }
Anton Yartsev85129b82011-02-07 02:17:30 +00001981
John McCalle3dc1702011-02-15 09:22:45 +00001982 // Floating point.
1983 } else if (type->isRealFloatingType()) {
Chris Lattner05dc78c2010-06-26 22:09:34 +00001984 // Add the inc/dec to the real part.
John McCalle3dc1702011-02-15 09:22:45 +00001985 llvm::Value *amt;
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001986
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001987 if (type->isHalfType() && !CGF.getContext().getLangOpts().NativeHalfType) {
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001988 // Another special case: half FP increment should be done via float
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001989 if (!CGF.getContext().getLangOpts().HalfArgsAndReturns) {
1990 value = Builder.CreateCall(
1991 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_from_fp16,
1992 CGF.CGM.FloatTy),
1993 input, "incdec.conv");
1994 } else {
1995 value = Builder.CreateFPExt(input, CGF.CGM.FloatTy, "incdec.conv");
1996 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001997 }
1998
John McCalle3dc1702011-02-15 09:22:45 +00001999 if (value->getType()->isFloatTy())
2000 amt = llvm::ConstantFP::get(VMContext,
2001 llvm::APFloat(static_cast<float>(amount)));
2002 else if (value->getType()->isDoubleTy())
2003 amt = llvm::ConstantFP::get(VMContext,
2004 llvm::APFloat(static_cast<double>(amount)));
Chris Lattner05dc78c2010-06-26 22:09:34 +00002005 else {
Nemanja Ivanovicbb1ea2d2016-05-09 08:52:33 +00002006 // Remaining types are Half, LongDouble or __float128. Convert from float.
John McCalle3dc1702011-02-15 09:22:45 +00002007 llvm::APFloat F(static_cast<float>(amount));
Chris Lattner05dc78c2010-06-26 22:09:34 +00002008 bool ignored;
Nemanja Ivanovicbb1ea2d2016-05-09 08:52:33 +00002009 const llvm::fltSemantics *FS;
Ahmed Bougacha6ba38312015-03-24 23:44:42 +00002010 // Don't use getFloatTypeSemantics because Half isn't
2011 // necessarily represented using the "half" LLVM type.
Nemanja Ivanovicbb1ea2d2016-05-09 08:52:33 +00002012 if (value->getType()->isFP128Ty())
2013 FS = &CGF.getTarget().getFloat128Format();
2014 else if (value->getType()->isHalfTy())
2015 FS = &CGF.getTarget().getHalfFormat();
2016 else
2017 FS = &CGF.getTarget().getLongDoubleFormat();
2018 F.convert(*FS, llvm::APFloat::rmTowardZero, &ignored);
John McCalle3dc1702011-02-15 09:22:45 +00002019 amt = llvm::ConstantFP::get(VMContext, F);
Chris Lattner05dc78c2010-06-26 22:09:34 +00002020 }
John McCalle3dc1702011-02-15 09:22:45 +00002021 value = Builder.CreateFAdd(value, amt, isInc ? "inc" : "dec");
2022
Ahmed Bougachad1801af2015-03-23 17:54:16 +00002023 if (type->isHalfType() && !CGF.getContext().getLangOpts().NativeHalfType) {
2024 if (!CGF.getContext().getLangOpts().HalfArgsAndReturns) {
2025 value = Builder.CreateCall(
2026 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_to_fp16,
2027 CGF.CGM.FloatTy),
2028 value, "incdec.conv");
2029 } else {
2030 value = Builder.CreateFPTrunc(value, input->getType(), "incdec.conv");
2031 }
2032 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00002033
John McCalle3dc1702011-02-15 09:22:45 +00002034 // Objective-C pointer types.
2035 } else {
2036 const ObjCObjectPointerType *OPT = type->castAs<ObjCObjectPointerType>();
2037 value = CGF.EmitCastToVoidPtr(value);
2038
2039 CharUnits size = CGF.getContext().getTypeSizeInChars(OPT->getObjectType());
2040 if (!isInc) size = -size;
2041 llvm::Value *sizeValue =
2042 llvm::ConstantInt::get(CGF.SizeTy, size.getQuantity());
2043
Richard Smith9c6890a2012-11-01 22:30:59 +00002044 if (CGF.getLangOpts().isSignedOverflowDefined())
Chris Lattner2e72da942011-03-01 00:03:48 +00002045 value = Builder.CreateGEP(value, sizeValue, "incdec.objptr");
2046 else
Vedant Kumar6dbf4272017-06-12 18:42:51 +00002047 value = CGF.EmitCheckedInBoundsGEP(value, sizeValue, signedIndex,
2048 E->getExprLoc(), "incdec.objptr");
John McCalle3dc1702011-02-15 09:22:45 +00002049 value = Builder.CreateBitCast(value, input->getType());
Chris Lattner05dc78c2010-06-26 22:09:34 +00002050 }
Craig Toppera97d7e72013-07-26 06:16:11 +00002051
David Chisnallfa35df62012-01-16 17:27:18 +00002052 if (atomicPHI) {
2053 llvm::BasicBlock *opBB = Builder.GetInsertBlock();
2054 llvm::BasicBlock *contBB = CGF.createBasicBlock("atomic_cont", CGF.CurFn);
Alexey Bataev452d8e12014-12-15 05:25:25 +00002055 auto Pair = CGF.EmitAtomicCompareExchange(
Alexey Bataevb4505a72015-03-30 05:20:59 +00002056 LV, RValue::get(atomicPHI), RValue::get(value), E->getExprLoc());
2057 llvm::Value *old = CGF.EmitToMemory(Pair.first.getScalarVal(), type);
2058 llvm::Value *success = Pair.second;
David Chisnallfa35df62012-01-16 17:27:18 +00002059 atomicPHI->addIncoming(old, opBB);
David Chisnallfa35df62012-01-16 17:27:18 +00002060 Builder.CreateCondBr(success, contBB, opBB);
2061 Builder.SetInsertPoint(contBB);
2062 return isPre ? value : input;
2063 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00002064
Chris Lattner05dc78c2010-06-26 22:09:34 +00002065 // Store the updated result through the lvalue.
2066 if (LV.isBitField())
John McCall55e1fbc2011-06-25 02:11:03 +00002067 CGF.EmitStoreThroughBitfieldLValue(RValue::get(value), LV, &value);
Chris Lattner05dc78c2010-06-26 22:09:34 +00002068 else
John McCall55e1fbc2011-06-25 02:11:03 +00002069 CGF.EmitStoreThroughLValue(RValue::get(value), LV);
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00002070
Chris Lattner05dc78c2010-06-26 22:09:34 +00002071 // If this is a postinc, return the value read from memory, otherwise use the
2072 // updated value.
John McCalle3dc1702011-02-15 09:22:45 +00002073 return isPre ? value : input;
Chris Lattner05dc78c2010-06-26 22:09:34 +00002074}
2075
2076
2077
Chris Lattner2da04b32007-08-24 05:35:26 +00002078Value *ScalarExprEmitter::VisitUnaryMinus(const UnaryOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00002079 TestAndClearIgnoreResultAssign();
Chris Lattner0bf27622010-06-26 21:48:21 +00002080 // Emit unary minus with EmitSub so we handle overflow cases etc.
2081 BinOpInfo BinOp;
Chris Lattnerc1028f62010-06-28 17:12:37 +00002082 BinOp.RHS = Visit(E->getSubExpr());
Craig Toppera97d7e72013-07-26 06:16:11 +00002083
Chris Lattnerc1028f62010-06-28 17:12:37 +00002084 if (BinOp.RHS->getType()->isFPOrFPVectorTy())
2085 BinOp.LHS = llvm::ConstantFP::getZeroValueForNegation(BinOp.RHS->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00002086 else
Chris Lattnerc1028f62010-06-28 17:12:37 +00002087 BinOp.LHS = llvm::Constant::getNullValue(BinOp.RHS->getType());
Chris Lattner0bf27622010-06-26 21:48:21 +00002088 BinOp.Ty = E->getType();
John McCalle3027922010-08-25 11:45:40 +00002089 BinOp.Opcode = BO_Sub;
Adam Nemet484aa452017-03-27 19:17:25 +00002090 // FIXME: once UnaryOperator carries FPFeatures, copy it here.
Chris Lattner0bf27622010-06-26 21:48:21 +00002091 BinOp.E = E;
2092 return EmitSub(BinOp);
Chris Lattner2da04b32007-08-24 05:35:26 +00002093}
2094
2095Value *ScalarExprEmitter::VisitUnaryNot(const UnaryOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00002096 TestAndClearIgnoreResultAssign();
Chris Lattner2da04b32007-08-24 05:35:26 +00002097 Value *Op = Visit(E->getSubExpr());
2098 return Builder.CreateNot(Op, "neg");
2099}
2100
2101Value *ScalarExprEmitter::VisitUnaryLNot(const UnaryOperator *E) {
Tanya Lattner20248222012-01-16 21:02:28 +00002102 // Perform vector logical not on comparison with zero vector.
2103 if (E->getType()->isExtVectorType()) {
2104 Value *Oper = Visit(E->getSubExpr());
2105 Value *Zero = llvm::Constant::getNullValue(Oper->getType());
Joey Gouly7d00f002013-02-21 11:49:56 +00002106 Value *Result;
2107 if (Oper->getType()->isFPOrFPVectorTy())
2108 Result = Builder.CreateFCmp(llvm::CmpInst::FCMP_OEQ, Oper, Zero, "cmp");
2109 else
2110 Result = Builder.CreateICmp(llvm::CmpInst::ICMP_EQ, Oper, Zero, "cmp");
Tanya Lattner20248222012-01-16 21:02:28 +00002111 return Builder.CreateSExt(Result, ConvertType(E->getType()), "sext");
2112 }
Craig Toppera97d7e72013-07-26 06:16:11 +00002113
Chris Lattner2da04b32007-08-24 05:35:26 +00002114 // Compare operand to zero.
2115 Value *BoolVal = CGF.EvaluateExprAsBool(E->getSubExpr());
Mike Stump4a3999f2009-09-09 13:00:44 +00002116
Chris Lattner2da04b32007-08-24 05:35:26 +00002117 // Invert value.
2118 // TODO: Could dynamically modify easy computations here. For example, if
2119 // the operand is an icmp ne, turn into icmp eq.
2120 BoolVal = Builder.CreateNot(BoolVal, "lnot");
Mike Stump4a3999f2009-09-09 13:00:44 +00002121
Anders Carlsson775640d2009-05-19 18:44:53 +00002122 // ZExt result to the expr type.
2123 return Builder.CreateZExt(BoolVal, ConvertType(E->getType()), "lnot.ext");
Chris Lattner2da04b32007-08-24 05:35:26 +00002124}
2125
Eli Friedmand7c72322010-08-05 09:58:49 +00002126Value *ScalarExprEmitter::VisitOffsetOfExpr(OffsetOfExpr *E) {
2127 // Try folding the offsetof to a constant.
Richard Smith5fab0c92011-12-28 19:48:30 +00002128 llvm::APSInt Value;
2129 if (E->EvaluateAsInt(Value, CGF.getContext()))
2130 return Builder.getInt(Value);
Eli Friedmand7c72322010-08-05 09:58:49 +00002131
2132 // Loop over the components of the offsetof to compute the value.
2133 unsigned n = E->getNumComponents();
Chris Lattner2192fe52011-07-18 04:24:23 +00002134 llvm::Type* ResultType = ConvertType(E->getType());
Eli Friedmand7c72322010-08-05 09:58:49 +00002135 llvm::Value* Result = llvm::Constant::getNullValue(ResultType);
2136 QualType CurrentType = E->getTypeSourceInfo()->getType();
2137 for (unsigned i = 0; i != n; ++i) {
James Y Knight7281c352015-12-29 22:31:18 +00002138 OffsetOfNode ON = E->getComponent(i);
Craig Topper8a13c412014-05-21 05:09:00 +00002139 llvm::Value *Offset = nullptr;
Eli Friedmand7c72322010-08-05 09:58:49 +00002140 switch (ON.getKind()) {
James Y Knight7281c352015-12-29 22:31:18 +00002141 case OffsetOfNode::Array: {
Eli Friedmand7c72322010-08-05 09:58:49 +00002142 // Compute the index
2143 Expr *IdxExpr = E->getIndexExpr(ON.getArrayExprIndex());
2144 llvm::Value* Idx = CGF.EmitScalarExpr(IdxExpr);
Douglas Gregor6ab2fa82011-05-20 16:38:50 +00002145 bool IdxSigned = IdxExpr->getType()->isSignedIntegerOrEnumerationType();
Eli Friedmand7c72322010-08-05 09:58:49 +00002146 Idx = Builder.CreateIntCast(Idx, ResultType, IdxSigned, "conv");
2147
2148 // Save the element type
2149 CurrentType =
2150 CGF.getContext().getAsArrayType(CurrentType)->getElementType();
2151
2152 // Compute the element size
2153 llvm::Value* ElemSize = llvm::ConstantInt::get(ResultType,
2154 CGF.getContext().getTypeSizeInChars(CurrentType).getQuantity());
2155
2156 // Multiply out to compute the result
2157 Offset = Builder.CreateMul(Idx, ElemSize);
2158 break;
2159 }
2160
James Y Knight7281c352015-12-29 22:31:18 +00002161 case OffsetOfNode::Field: {
Eli Friedmand7c72322010-08-05 09:58:49 +00002162 FieldDecl *MemberDecl = ON.getField();
2163 RecordDecl *RD = CurrentType->getAs<RecordType>()->getDecl();
2164 const ASTRecordLayout &RL = CGF.getContext().getASTRecordLayout(RD);
2165
2166 // Compute the index of the field in its parent.
2167 unsigned i = 0;
2168 // FIXME: It would be nice if we didn't have to loop here!
2169 for (RecordDecl::field_iterator Field = RD->field_begin(),
2170 FieldEnd = RD->field_end();
David Blaikie2d7c57e2012-04-30 02:36:29 +00002171 Field != FieldEnd; ++Field, ++i) {
David Blaikie40ed2972012-06-06 20:45:41 +00002172 if (*Field == MemberDecl)
Eli Friedmand7c72322010-08-05 09:58:49 +00002173 break;
2174 }
2175 assert(i < RL.getFieldCount() && "offsetof field in wrong type");
2176
2177 // Compute the offset to the field
2178 int64_t OffsetInt = RL.getFieldOffset(i) /
2179 CGF.getContext().getCharWidth();
2180 Offset = llvm::ConstantInt::get(ResultType, OffsetInt);
2181
2182 // Save the element type.
2183 CurrentType = MemberDecl->getType();
2184 break;
2185 }
Eli Friedman165301d2010-08-06 16:37:05 +00002186
James Y Knight7281c352015-12-29 22:31:18 +00002187 case OffsetOfNode::Identifier:
Eli Friedmane83d2b762010-08-06 01:17:25 +00002188 llvm_unreachable("dependent __builtin_offsetof");
Eli Friedman165301d2010-08-06 16:37:05 +00002189
James Y Knight7281c352015-12-29 22:31:18 +00002190 case OffsetOfNode::Base: {
Eli Friedmand7c72322010-08-05 09:58:49 +00002191 if (ON.getBase()->isVirtual()) {
2192 CGF.ErrorUnsupported(E, "virtual base in offsetof");
2193 continue;
2194 }
2195
2196 RecordDecl *RD = CurrentType->getAs<RecordType>()->getDecl();
2197 const ASTRecordLayout &RL = CGF.getContext().getASTRecordLayout(RD);
2198
2199 // Save the element type.
2200 CurrentType = ON.getBase()->getType();
Craig Toppera97d7e72013-07-26 06:16:11 +00002201
Eli Friedmand7c72322010-08-05 09:58:49 +00002202 // Compute the offset to the base.
2203 const RecordType *BaseRT = CurrentType->getAs<RecordType>();
2204 CXXRecordDecl *BaseRD = cast<CXXRecordDecl>(BaseRT->getDecl());
Benjamin Kramer2ef30312012-07-04 18:45:14 +00002205 CharUnits OffsetInt = RL.getBaseClassOffset(BaseRD);
2206 Offset = llvm::ConstantInt::get(ResultType, OffsetInt.getQuantity());
Eli Friedmand7c72322010-08-05 09:58:49 +00002207 break;
2208 }
2209 }
2210 Result = Builder.CreateAdd(Result, Offset);
2211 }
2212 return Result;
Douglas Gregor882211c2010-04-28 22:16:22 +00002213}
2214
Peter Collingbournee190dee2011-03-11 19:24:49 +00002215/// VisitUnaryExprOrTypeTraitExpr - Return the size or alignment of the type of
Sebastian Redl6f282892008-11-11 17:56:53 +00002216/// argument of the sizeof expression as an integer.
2217Value *
Peter Collingbournee190dee2011-03-11 19:24:49 +00002218ScalarExprEmitter::VisitUnaryExprOrTypeTraitExpr(
2219 const UnaryExprOrTypeTraitExpr *E) {
Sebastian Redl6f282892008-11-11 17:56:53 +00002220 QualType TypeToSize = E->getTypeOfArgument();
Peter Collingbournee190dee2011-03-11 19:24:49 +00002221 if (E->getKind() == UETT_SizeOf) {
Mike Stump4a3999f2009-09-09 13:00:44 +00002222 if (const VariableArrayType *VAT =
Eli Friedman2aa38fe2009-01-24 22:19:05 +00002223 CGF.getContext().getAsVariableArrayType(TypeToSize)) {
2224 if (E->isArgumentType()) {
2225 // sizeof(type) - make sure to emit the VLA size.
John McCall23c29fe2011-06-24 21:55:10 +00002226 CGF.EmitVariablyModifiedType(TypeToSize);
Eli Friedman3253e182009-04-20 03:21:44 +00002227 } else {
2228 // C99 6.5.3.4p2: If the argument is an expression of type
2229 // VLA, it is evaluated.
John McCalla2342eb2010-12-05 02:00:02 +00002230 CGF.EmitIgnoredExpr(E->getArgumentExpr());
Eli Friedman2aa38fe2009-01-24 22:19:05 +00002231 }
Mike Stump4a3999f2009-09-09 13:00:44 +00002232
John McCall23c29fe2011-06-24 21:55:10 +00002233 QualType eltType;
2234 llvm::Value *numElts;
Benjamin Kramer867ea1d2014-03-02 13:01:17 +00002235 std::tie(numElts, eltType) = CGF.getVLASize(VAT);
John McCall23c29fe2011-06-24 21:55:10 +00002236
2237 llvm::Value *size = numElts;
2238
2239 // Scale the number of non-VLA elements by the non-VLA element size.
2240 CharUnits eltSize = CGF.getContext().getTypeSizeInChars(eltType);
2241 if (!eltSize.isOne())
2242 size = CGF.Builder.CreateNUWMul(CGF.CGM.getSize(eltSize), numElts);
2243
2244 return size;
Anders Carlsson76dbc042008-12-21 03:33:21 +00002245 }
Alexey Bataev00396512015-07-02 03:40:19 +00002246 } else if (E->getKind() == UETT_OpenMPRequiredSimdAlign) {
2247 auto Alignment =
2248 CGF.getContext()
2249 .toCharUnitsFromBits(CGF.getContext().getOpenMPDefaultSimdAlign(
2250 E->getTypeOfArgument()->getPointeeType()))
2251 .getQuantity();
2252 return llvm::ConstantInt::get(CGF.SizeTy, Alignment);
Anders Carlsson30032882008-12-12 07:38:43 +00002253 }
Eli Friedman2aa38fe2009-01-24 22:19:05 +00002254
Mike Stump4a3999f2009-09-09 13:00:44 +00002255 // If this isn't sizeof(vla), the result must be constant; use the constant
2256 // folding logic so we don't have to duplicate it here.
Richard Smith5fab0c92011-12-28 19:48:30 +00002257 return Builder.getInt(E->EvaluateKnownConstInt(CGF.getContext()));
Chris Lattner2da04b32007-08-24 05:35:26 +00002258}
2259
Chris Lattner9f0ad962007-08-24 21:20:17 +00002260Value *ScalarExprEmitter::VisitUnaryReal(const UnaryOperator *E) {
2261 Expr *Op = E->getSubExpr();
John McCall07bb1962010-11-16 10:08:07 +00002262 if (Op->getType()->isAnyComplexType()) {
2263 // If it's an l-value, load through the appropriate subobject l-value.
2264 // Note that we have to ask E because Op might be an l-value that
2265 // this won't work for, e.g. an Obj-C property.
John McCall086a4642010-11-24 05:12:34 +00002266 if (E->isGLValue())
Nick Lewycky2d84e842013-10-02 02:29:49 +00002267 return CGF.EmitLoadOfLValue(CGF.EmitLValue(E),
2268 E->getExprLoc()).getScalarVal();
John McCall07bb1962010-11-16 10:08:07 +00002269
2270 // Otherwise, calculate and project.
2271 return CGF.EmitComplexExpr(Op, false, true).first;
2272 }
2273
Chris Lattner9f0ad962007-08-24 21:20:17 +00002274 return Visit(Op);
2275}
John McCall07bb1962010-11-16 10:08:07 +00002276
Chris Lattner9f0ad962007-08-24 21:20:17 +00002277Value *ScalarExprEmitter::VisitUnaryImag(const UnaryOperator *E) {
2278 Expr *Op = E->getSubExpr();
John McCall07bb1962010-11-16 10:08:07 +00002279 if (Op->getType()->isAnyComplexType()) {
2280 // If it's an l-value, load through the appropriate subobject l-value.
2281 // Note that we have to ask E because Op might be an l-value that
2282 // this won't work for, e.g. an Obj-C property.
John McCall086a4642010-11-24 05:12:34 +00002283 if (Op->isGLValue())
Nick Lewycky2d84e842013-10-02 02:29:49 +00002284 return CGF.EmitLoadOfLValue(CGF.EmitLValue(E),
2285 E->getExprLoc()).getScalarVal();
John McCall07bb1962010-11-16 10:08:07 +00002286
2287 // Otherwise, calculate and project.
2288 return CGF.EmitComplexExpr(Op, true, false).second;
2289 }
Mike Stump4a3999f2009-09-09 13:00:44 +00002290
Mike Stumpdf0fe272009-05-29 15:46:01 +00002291 // __imag on a scalar returns zero. Emit the subexpr to ensure side
2292 // effects are evaluated, but not the actual value.
Richard Smith0b6b8e42012-02-18 20:53:32 +00002293 if (Op->isGLValue())
2294 CGF.EmitLValue(Op);
2295 else
2296 CGF.EmitScalarExpr(Op, true);
Owen Anderson0b75f232009-07-31 20:28:54 +00002297 return llvm::Constant::getNullValue(ConvertType(E->getType()));
Chris Lattner9f0ad962007-08-24 21:20:17 +00002298}
2299
Chris Lattner2da04b32007-08-24 05:35:26 +00002300//===----------------------------------------------------------------------===//
2301// Binary Operators
2302//===----------------------------------------------------------------------===//
2303
2304BinOpInfo ScalarExprEmitter::EmitBinOps(const BinaryOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00002305 TestAndClearIgnoreResultAssign();
Chris Lattner2da04b32007-08-24 05:35:26 +00002306 BinOpInfo Result;
2307 Result.LHS = Visit(E->getLHS());
2308 Result.RHS = Visit(E->getRHS());
Chris Lattner3d966d62007-08-24 21:00:35 +00002309 Result.Ty = E->getType();
Chris Lattner0bf27622010-06-26 21:48:21 +00002310 Result.Opcode = E->getOpcode();
Adam Nemet484aa452017-03-27 19:17:25 +00002311 Result.FPFeatures = E->getFPFeatures();
Chris Lattner2da04b32007-08-24 05:35:26 +00002312 Result.E = E;
2313 return Result;
2314}
2315
Douglas Gregor914af212010-04-23 04:16:32 +00002316LValue ScalarExprEmitter::EmitCompoundAssignLValue(
2317 const CompoundAssignOperator *E,
2318 Value *(ScalarExprEmitter::*Func)(const BinOpInfo &),
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00002319 Value *&Result) {
Benjamin Kramerd20ef752009-12-25 15:43:36 +00002320 QualType LHSTy = E->getLHS()->getType();
Chris Lattner3d966d62007-08-24 21:00:35 +00002321 BinOpInfo OpInfo;
Craig Toppera97d7e72013-07-26 06:16:11 +00002322
Eli Friedmanf0450072013-06-12 01:40:06 +00002323 if (E->getComputationResultType()->isAnyComplexType())
Richard Smith527473d2015-02-12 21:23:20 +00002324 return CGF.EmitScalarCompoundAssignWithComplex(E, Result);
Craig Toppera97d7e72013-07-26 06:16:11 +00002325
Mike Stumpc63428b2009-05-22 19:07:20 +00002326 // Emit the RHS first. __block variables need to have the rhs evaluated
2327 // first, plus this should improve codegen a little.
2328 OpInfo.RHS = Visit(E->getRHS());
2329 OpInfo.Ty = E->getComputationResultType();
Chris Lattner0bf27622010-06-26 21:48:21 +00002330 OpInfo.Opcode = E->getOpcode();
Adam Nemet484aa452017-03-27 19:17:25 +00002331 OpInfo.FPFeatures = E->getFPFeatures();
Mike Stumpc63428b2009-05-22 19:07:20 +00002332 OpInfo.E = E;
Eli Friedman8b7b1b12009-03-28 01:22:36 +00002333 // Load/convert the LHS.
Richard Smith4d1458e2012-09-08 02:08:36 +00002334 LValue LHSLV = EmitCheckedLValue(E->getLHS(), CodeGenFunction::TCK_Store);
David Chisnallfa35df62012-01-16 17:27:18 +00002335
Craig Topper8a13c412014-05-21 05:09:00 +00002336 llvm::PHINode *atomicPHI = nullptr;
David Chisnallef78c302013-03-03 16:02:42 +00002337 if (const AtomicType *atomicTy = LHSTy->getAs<AtomicType>()) {
2338 QualType type = atomicTy->getValueType();
2339 if (!type->isBooleanType() && type->isIntegerType() &&
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002340 !(type->isUnsignedIntegerType() &&
2341 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow)) &&
2342 CGF.getLangOpts().getSignedOverflowBehavior() !=
2343 LangOptions::SOB_Trapping) {
David Chisnallef78c302013-03-03 16:02:42 +00002344 llvm::AtomicRMWInst::BinOp aop = llvm::AtomicRMWInst::BAD_BINOP;
2345 switch (OpInfo.Opcode) {
2346 // We don't have atomicrmw operands for *, %, /, <<, >>
2347 case BO_MulAssign: case BO_DivAssign:
2348 case BO_RemAssign:
2349 case BO_ShlAssign:
2350 case BO_ShrAssign:
2351 break;
2352 case BO_AddAssign:
2353 aop = llvm::AtomicRMWInst::Add;
2354 break;
2355 case BO_SubAssign:
2356 aop = llvm::AtomicRMWInst::Sub;
2357 break;
2358 case BO_AndAssign:
2359 aop = llvm::AtomicRMWInst::And;
2360 break;
2361 case BO_XorAssign:
2362 aop = llvm::AtomicRMWInst::Xor;
2363 break;
2364 case BO_OrAssign:
2365 aop = llvm::AtomicRMWInst::Or;
2366 break;
2367 default:
2368 llvm_unreachable("Invalid compound assignment type");
2369 }
2370 if (aop != llvm::AtomicRMWInst::BAD_BINOP) {
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00002371 llvm::Value *amt = CGF.EmitToMemory(
2372 EmitScalarConversion(OpInfo.RHS, E->getRHS()->getType(), LHSTy,
2373 E->getExprLoc()),
2374 LHSTy);
John McCall7f416cc2015-09-08 08:05:57 +00002375 Builder.CreateAtomicRMW(aop, LHSLV.getPointer(), amt,
JF Bastien92f4ef12016-04-06 17:26:42 +00002376 llvm::AtomicOrdering::SequentiallyConsistent);
David Chisnallef78c302013-03-03 16:02:42 +00002377 return LHSLV;
2378 }
2379 }
David Chisnallfa35df62012-01-16 17:27:18 +00002380 // FIXME: For floating point types, we should be saving and restoring the
2381 // floating point environment in the loop.
2382 llvm::BasicBlock *startBB = Builder.GetInsertBlock();
2383 llvm::BasicBlock *opBB = CGF.createBasicBlock("atomic_op", CGF.CurFn);
Nick Lewycky2d84e842013-10-02 02:29:49 +00002384 OpInfo.LHS = EmitLoadOfLValue(LHSLV, E->getExprLoc());
David Chisnallef78c302013-03-03 16:02:42 +00002385 OpInfo.LHS = CGF.EmitToMemory(OpInfo.LHS, type);
David Chisnallfa35df62012-01-16 17:27:18 +00002386 Builder.CreateBr(opBB);
2387 Builder.SetInsertPoint(opBB);
2388 atomicPHI = Builder.CreatePHI(OpInfo.LHS->getType(), 2);
2389 atomicPHI->addIncoming(OpInfo.LHS, startBB);
David Chisnallfa35df62012-01-16 17:27:18 +00002390 OpInfo.LHS = atomicPHI;
2391 }
David Chisnallef78c302013-03-03 16:02:42 +00002392 else
Nick Lewycky2d84e842013-10-02 02:29:49 +00002393 OpInfo.LHS = EmitLoadOfLValue(LHSLV, E->getExprLoc());
Eli Friedman93ee5ca2012-06-16 02:19:17 +00002394
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00002395 SourceLocation Loc = E->getExprLoc();
2396 OpInfo.LHS =
2397 EmitScalarConversion(OpInfo.LHS, LHSTy, E->getComputationLHSType(), Loc);
Eli Friedman93ee5ca2012-06-16 02:19:17 +00002398
Chris Lattner3d966d62007-08-24 21:00:35 +00002399 // Expand the binary operator.
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00002400 Result = (this->*Func)(OpInfo);
Craig Toppera97d7e72013-07-26 06:16:11 +00002401
Daniel Dunbarbfb1cd72008-08-06 02:00:38 +00002402 // Convert the result back to the LHS type.
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00002403 Result =
2404 EmitScalarConversion(Result, E->getComputationResultType(), LHSTy, Loc);
David Chisnallfa35df62012-01-16 17:27:18 +00002405
2406 if (atomicPHI) {
2407 llvm::BasicBlock *opBB = Builder.GetInsertBlock();
2408 llvm::BasicBlock *contBB = CGF.createBasicBlock("atomic_cont", CGF.CurFn);
Alexey Bataev452d8e12014-12-15 05:25:25 +00002409 auto Pair = CGF.EmitAtomicCompareExchange(
Alexey Bataevb4505a72015-03-30 05:20:59 +00002410 LHSLV, RValue::get(atomicPHI), RValue::get(Result), E->getExprLoc());
2411 llvm::Value *old = CGF.EmitToMemory(Pair.first.getScalarVal(), LHSTy);
2412 llvm::Value *success = Pair.second;
David Chisnallfa35df62012-01-16 17:27:18 +00002413 atomicPHI->addIncoming(old, opBB);
David Chisnallfa35df62012-01-16 17:27:18 +00002414 Builder.CreateCondBr(success, contBB, opBB);
2415 Builder.SetInsertPoint(contBB);
2416 return LHSLV;
2417 }
Craig Toppera97d7e72013-07-26 06:16:11 +00002418
Mike Stump4a3999f2009-09-09 13:00:44 +00002419 // Store the result value into the LHS lvalue. Bit-fields are handled
2420 // specially because the result is altered by the store, i.e., [C99 6.5.16p1]
2421 // 'An assignment expression has the value of the left operand after the
2422 // assignment...'.
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00002423 if (LHSLV.isBitField())
John McCall55e1fbc2011-06-25 02:11:03 +00002424 CGF.EmitStoreThroughBitfieldLValue(RValue::get(Result), LHSLV, &Result);
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00002425 else
John McCall55e1fbc2011-06-25 02:11:03 +00002426 CGF.EmitStoreThroughLValue(RValue::get(Result), LHSLV);
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00002427
Douglas Gregor914af212010-04-23 04:16:32 +00002428 return LHSLV;
2429}
2430
2431Value *ScalarExprEmitter::EmitCompoundAssign(const CompoundAssignOperator *E,
2432 Value *(ScalarExprEmitter::*Func)(const BinOpInfo &)) {
2433 bool Ignore = TestAndClearIgnoreResultAssign();
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00002434 Value *RHS;
2435 LValue LHS = EmitCompoundAssignLValue(E, Func, RHS);
2436
2437 // If the result is clearly ignored, return now.
Mike Stumpdf0fe272009-05-29 15:46:01 +00002438 if (Ignore)
Craig Topper8a13c412014-05-21 05:09:00 +00002439 return nullptr;
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00002440
John McCall07bb1962010-11-16 10:08:07 +00002441 // The result of an assignment in C is the assigned r-value.
Richard Smith9c6890a2012-11-01 22:30:59 +00002442 if (!CGF.getLangOpts().CPlusPlus)
John McCall07bb1962010-11-16 10:08:07 +00002443 return RHS;
2444
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00002445 // If the lvalue is non-volatile, return the computed value of the assignment.
2446 if (!LHS.isVolatileQualified())
2447 return RHS;
2448
2449 // Otherwise, reload the value.
Nick Lewycky2d84e842013-10-02 02:29:49 +00002450 return EmitLoadOfLValue(LHS, E->getExprLoc());
Chris Lattner3d966d62007-08-24 21:00:35 +00002451}
2452
Chris Lattner8ee6a412010-09-11 21:47:09 +00002453void ScalarExprEmitter::EmitUndefinedBehaviorIntegerDivAndRemCheck(
Richard Smith4d1458e2012-09-08 02:08:36 +00002454 const BinOpInfo &Ops, llvm::Value *Zero, bool isDiv) {
Peter Collingbourne3eea6772015-05-11 21:39:14 +00002455 SmallVector<std::pair<llvm::Value *, SanitizerMask>, 2> Checks;
Chris Lattner8ee6a412010-09-11 21:47:09 +00002456
Alexey Samsonov4c1a96f2014-11-10 22:27:30 +00002457 if (CGF.SanOpts.has(SanitizerKind::IntegerDivideByZero)) {
Alexey Samsonove396bfc2014-11-11 22:03:54 +00002458 Checks.push_back(std::make_pair(Builder.CreateICmpNE(Ops.RHS, Zero),
2459 SanitizerKind::IntegerDivideByZero));
Alexey Samsonov4c1a96f2014-11-10 22:27:30 +00002460 }
Richard Smithc86a1142012-11-06 02:30:30 +00002461
Vedant Kumar82ee16b2017-02-25 00:43:36 +00002462 const auto *BO = cast<BinaryOperator>(Ops.E);
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002463 if (CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow) &&
Vedant Kumar82ee16b2017-02-25 00:43:36 +00002464 Ops.Ty->hasSignedIntegerRepresentation() &&
Vedant Kumard9191152017-05-02 23:46:56 +00002465 !IsWidenedIntegerOp(CGF.getContext(), BO->getLHS()) &&
2466 Ops.mayHaveIntegerOverflow()) {
Richard Smithc86a1142012-11-06 02:30:30 +00002467 llvm::IntegerType *Ty = cast<llvm::IntegerType>(Zero->getType());
2468
Chris Lattner8ee6a412010-09-11 21:47:09 +00002469 llvm::Value *IntMin =
Chris Lattner2531eb42011-04-19 22:55:03 +00002470 Builder.getInt(llvm::APInt::getSignedMinValue(Ty->getBitWidth()));
Chris Lattner8ee6a412010-09-11 21:47:09 +00002471 llvm::Value *NegOne = llvm::ConstantInt::get(Ty, -1ULL);
2472
Richard Smith4d1458e2012-09-08 02:08:36 +00002473 llvm::Value *LHSCmp = Builder.CreateICmpNE(Ops.LHS, IntMin);
2474 llvm::Value *RHSCmp = Builder.CreateICmpNE(Ops.RHS, NegOne);
Alexey Samsonove396bfc2014-11-11 22:03:54 +00002475 llvm::Value *NotOverflow = Builder.CreateOr(LHSCmp, RHSCmp, "or");
2476 Checks.push_back(
2477 std::make_pair(NotOverflow, SanitizerKind::SignedIntegerOverflow));
Chris Lattner8ee6a412010-09-11 21:47:09 +00002478 }
Richard Smithc86a1142012-11-06 02:30:30 +00002479
Alexey Samsonove396bfc2014-11-11 22:03:54 +00002480 if (Checks.size() > 0)
2481 EmitBinOpCheck(Checks, Ops);
Chris Lattner8ee6a412010-09-11 21:47:09 +00002482}
Chris Lattner3d966d62007-08-24 21:00:35 +00002483
Chris Lattner2da04b32007-08-24 05:35:26 +00002484Value *ScalarExprEmitter::EmitDiv(const BinOpInfo &Ops) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00002485 {
2486 CodeGenFunction::SanitizerScope SanScope(&CGF);
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002487 if ((CGF.SanOpts.has(SanitizerKind::IntegerDivideByZero) ||
2488 CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow)) &&
Vedant Kumard9191152017-05-02 23:46:56 +00002489 Ops.Ty->isIntegerType() &&
2490 (Ops.mayHaveIntegerDivisionByZero() || Ops.mayHaveIntegerOverflow())) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00002491 llvm::Value *Zero = llvm::Constant::getNullValue(ConvertType(Ops.Ty));
2492 EmitUndefinedBehaviorIntegerDivAndRemCheck(Ops, Zero, true);
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002493 } else if (CGF.SanOpts.has(SanitizerKind::FloatDivideByZero) &&
Vedant Kumard9191152017-05-02 23:46:56 +00002494 Ops.Ty->isRealFloatingType() &&
2495 Ops.mayHaveFloatDivisionByZero()) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00002496 llvm::Value *Zero = llvm::Constant::getNullValue(ConvertType(Ops.Ty));
Alexey Samsonove396bfc2014-11-11 22:03:54 +00002497 llvm::Value *NonZero = Builder.CreateFCmpUNE(Ops.RHS, Zero);
2498 EmitBinOpCheck(std::make_pair(NonZero, SanitizerKind::FloatDivideByZero),
2499 Ops);
Alexey Samsonov24cad992014-07-17 18:46:27 +00002500 }
Chris Lattner8ee6a412010-09-11 21:47:09 +00002501 }
Will Dietz1897cb32012-11-27 15:01:55 +00002502
Peter Collingbourne95fd2ca2011-10-27 19:19:51 +00002503 if (Ops.LHS->getType()->isFPOrFPVectorTy()) {
2504 llvm::Value *Val = Builder.CreateFDiv(Ops.LHS, Ops.RHS, "div");
Yaxun Liuffb60902016-08-09 20:10:18 +00002505 if (CGF.getLangOpts().OpenCL &&
2506 !CGF.CGM.getCodeGenOpts().CorrectlyRoundedDivSqrt) {
2507 // OpenCL v1.1 s7.4: minimum accuracy of single precision / is 2.5ulp
2508 // OpenCL v1.2 s5.6.4.2: The -cl-fp32-correctly-rounded-divide-sqrt
2509 // build option allows an application to specify that single precision
2510 // floating-point divide (x/y and 1/x) and sqrt used in the program
2511 // source are correctly rounded.
Peter Collingbourne95fd2ca2011-10-27 19:19:51 +00002512 llvm::Type *ValTy = Val->getType();
2513 if (ValTy->isFloatTy() ||
2514 (isa<llvm::VectorType>(ValTy) &&
2515 cast<llvm::VectorType>(ValTy)->getElementType()->isFloatTy()))
Duncan Sandse81111c2012-04-10 08:23:07 +00002516 CGF.SetFPAccuracy(Val, 2.5);
Peter Collingbourne95fd2ca2011-10-27 19:19:51 +00002517 }
2518 return Val;
2519 }
Douglas Gregor5cc2c8b2010-07-23 15:58:24 +00002520 else if (Ops.Ty->hasUnsignedIntegerRepresentation())
Chris Lattner2da04b32007-08-24 05:35:26 +00002521 return Builder.CreateUDiv(Ops.LHS, Ops.RHS, "div");
2522 else
2523 return Builder.CreateSDiv(Ops.LHS, Ops.RHS, "div");
2524}
2525
2526Value *ScalarExprEmitter::EmitRem(const BinOpInfo &Ops) {
2527 // Rem in C can't be a floating point type: C99 6.5.5p2.
Vedant Kumar42de3802017-02-25 00:43:39 +00002528 if ((CGF.SanOpts.has(SanitizerKind::IntegerDivideByZero) ||
2529 CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow)) &&
Vedant Kumard9191152017-05-02 23:46:56 +00002530 Ops.Ty->isIntegerType() &&
2531 (Ops.mayHaveIntegerDivisionByZero() || Ops.mayHaveIntegerOverflow())) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00002532 CodeGenFunction::SanitizerScope SanScope(&CGF);
Chris Lattner8ee6a412010-09-11 21:47:09 +00002533 llvm::Value *Zero = llvm::Constant::getNullValue(ConvertType(Ops.Ty));
Vedant Kumar42de3802017-02-25 00:43:39 +00002534 EmitUndefinedBehaviorIntegerDivAndRemCheck(Ops, Zero, false);
Chris Lattner8ee6a412010-09-11 21:47:09 +00002535 }
2536
Eli Friedman493c34a2011-04-10 04:44:11 +00002537 if (Ops.Ty->hasUnsignedIntegerRepresentation())
Chris Lattner2da04b32007-08-24 05:35:26 +00002538 return Builder.CreateURem(Ops.LHS, Ops.RHS, "rem");
2539 else
2540 return Builder.CreateSRem(Ops.LHS, Ops.RHS, "rem");
2541}
2542
Mike Stump0c61b732009-04-01 20:28:16 +00002543Value *ScalarExprEmitter::EmitOverflowCheckedBinOp(const BinOpInfo &Ops) {
2544 unsigned IID;
2545 unsigned OpID = 0;
Mike Stump40968592009-04-02 01:03:55 +00002546
Will Dietz1897cb32012-11-27 15:01:55 +00002547 bool isSigned = Ops.Ty->isSignedIntegerOrEnumerationType();
Chris Lattner0bf27622010-06-26 21:48:21 +00002548 switch (Ops.Opcode) {
John McCalle3027922010-08-25 11:45:40 +00002549 case BO_Add:
2550 case BO_AddAssign:
Mike Stumpd3e38852009-04-02 18:15:54 +00002551 OpID = 1;
Will Dietz1897cb32012-11-27 15:01:55 +00002552 IID = isSigned ? llvm::Intrinsic::sadd_with_overflow :
2553 llvm::Intrinsic::uadd_with_overflow;
Mike Stumpd3e38852009-04-02 18:15:54 +00002554 break;
John McCalle3027922010-08-25 11:45:40 +00002555 case BO_Sub:
2556 case BO_SubAssign:
Mike Stumpd3e38852009-04-02 18:15:54 +00002557 OpID = 2;
Will Dietz1897cb32012-11-27 15:01:55 +00002558 IID = isSigned ? llvm::Intrinsic::ssub_with_overflow :
2559 llvm::Intrinsic::usub_with_overflow;
Mike Stumpd3e38852009-04-02 18:15:54 +00002560 break;
John McCalle3027922010-08-25 11:45:40 +00002561 case BO_Mul:
2562 case BO_MulAssign:
Mike Stumpd3e38852009-04-02 18:15:54 +00002563 OpID = 3;
Will Dietz1897cb32012-11-27 15:01:55 +00002564 IID = isSigned ? llvm::Intrinsic::smul_with_overflow :
2565 llvm::Intrinsic::umul_with_overflow;
Mike Stumpd3e38852009-04-02 18:15:54 +00002566 break;
2567 default:
David Blaikie83d382b2011-09-23 05:06:16 +00002568 llvm_unreachable("Unsupported operation for overflow detection");
Mike Stump0c61b732009-04-01 20:28:16 +00002569 }
Mike Stumpd3e38852009-04-02 18:15:54 +00002570 OpID <<= 1;
Will Dietz1897cb32012-11-27 15:01:55 +00002571 if (isSigned)
2572 OpID |= 1;
Mike Stumpd3e38852009-04-02 18:15:54 +00002573
Vedant Kumar4b62b5c2017-05-09 23:34:49 +00002574 CodeGenFunction::SanitizerScope SanScope(&CGF);
Chris Lattnera5f58b02011-07-09 17:41:47 +00002575 llvm::Type *opTy = CGF.CGM.getTypes().ConvertType(Ops.Ty);
Mike Stump0c61b732009-04-01 20:28:16 +00002576
Benjamin Kramer8d375ce2011-07-14 17:45:50 +00002577 llvm::Function *intrinsic = CGF.CGM.getIntrinsic(IID, opTy);
Mike Stump0c61b732009-04-01 20:28:16 +00002578
David Blaikie43f9bb72015-05-18 22:14:03 +00002579 Value *resultAndOverflow = Builder.CreateCall(intrinsic, {Ops.LHS, Ops.RHS});
Mike Stump0c61b732009-04-01 20:28:16 +00002580 Value *result = Builder.CreateExtractValue(resultAndOverflow, 0);
2581 Value *overflow = Builder.CreateExtractValue(resultAndOverflow, 1);
2582
Richard Smith4d1458e2012-09-08 02:08:36 +00002583 // Handle overflow with llvm.trap if no custom handler has been specified.
2584 const std::string *handlerName =
Richard Smith9c6890a2012-11-01 22:30:59 +00002585 &CGF.getLangOpts().OverflowHandler;
Richard Smith4d1458e2012-09-08 02:08:36 +00002586 if (handlerName->empty()) {
Richard Smithb1b0ab42012-11-05 22:21:05 +00002587 // If the signed-integer-overflow sanitizer is enabled, emit a call to its
Richard Smithde670682012-11-01 22:15:34 +00002588 // runtime. Otherwise, this is a -ftrapv check, so just emit a trap.
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002589 if (!isSigned || CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow)) {
Alexey Samsonove396bfc2014-11-11 22:03:54 +00002590 llvm::Value *NotOverflow = Builder.CreateNot(overflow);
Peter Collingbourne3eea6772015-05-11 21:39:14 +00002591 SanitizerMask Kind = isSigned ? SanitizerKind::SignedIntegerOverflow
Alexey Samsonove396bfc2014-11-11 22:03:54 +00002592 : SanitizerKind::UnsignedIntegerOverflow;
2593 EmitBinOpCheck(std::make_pair(NotOverflow, Kind), Ops);
Alexey Samsonov24cad992014-07-17 18:46:27 +00002594 } else
Chad Rosierae229d52013-01-29 23:31:22 +00002595 CGF.EmitTrapCheck(Builder.CreateNot(overflow));
Richard Smith4d1458e2012-09-08 02:08:36 +00002596 return result;
2597 }
2598
Mike Stump0c61b732009-04-01 20:28:16 +00002599 // Branch in case of overflow.
David Chisnalldd84ef12010-09-17 18:29:54 +00002600 llvm::BasicBlock *initialBB = Builder.GetInsertBlock();
Duncan P. N. Exon Smith01f574c2016-08-17 03:15:29 +00002601 llvm::BasicBlock *continueBB =
2602 CGF.createBasicBlock("nooverflow", CGF.CurFn, initialBB->getNextNode());
Chris Lattner8139c982010-08-07 00:20:46 +00002603 llvm::BasicBlock *overflowBB = CGF.createBasicBlock("overflow", CGF.CurFn);
Mike Stump0c61b732009-04-01 20:28:16 +00002604
2605 Builder.CreateCondBr(overflow, overflowBB, continueBB);
2606
David Chisnalldd84ef12010-09-17 18:29:54 +00002607 // If an overflow handler is set, then we want to call it and then use its
2608 // result, if it returns.
2609 Builder.SetInsertPoint(overflowBB);
2610
2611 // Get the overflow handler.
Chris Lattnerece04092012-02-07 00:39:47 +00002612 llvm::Type *Int8Ty = CGF.Int8Ty;
Chris Lattnera5f58b02011-07-09 17:41:47 +00002613 llvm::Type *argTypes[] = { CGF.Int64Ty, CGF.Int64Ty, Int8Ty, Int8Ty };
David Chisnalldd84ef12010-09-17 18:29:54 +00002614 llvm::FunctionType *handlerTy =
2615 llvm::FunctionType::get(CGF.Int64Ty, argTypes, true);
2616 llvm::Value *handler = CGF.CGM.CreateRuntimeFunction(handlerTy, *handlerName);
2617
2618 // Sign extend the args to 64-bit, so that we can use the same handler for
2619 // all types of overflow.
2620 llvm::Value *lhs = Builder.CreateSExt(Ops.LHS, CGF.Int64Ty);
2621 llvm::Value *rhs = Builder.CreateSExt(Ops.RHS, CGF.Int64Ty);
2622
2623 // Call the handler with the two arguments, the operation, and the size of
2624 // the result.
John McCall882987f2013-02-28 19:01:20 +00002625 llvm::Value *handlerArgs[] = {
2626 lhs,
2627 rhs,
2628 Builder.getInt8(OpID),
2629 Builder.getInt8(cast<llvm::IntegerType>(opTy)->getBitWidth())
2630 };
2631 llvm::Value *handlerResult =
2632 CGF.EmitNounwindRuntimeCall(handler, handlerArgs);
David Chisnalldd84ef12010-09-17 18:29:54 +00002633
2634 // Truncate the result back to the desired size.
2635 handlerResult = Builder.CreateTrunc(handlerResult, opTy);
2636 Builder.CreateBr(continueBB);
2637
Mike Stump0c61b732009-04-01 20:28:16 +00002638 Builder.SetInsertPoint(continueBB);
Jay Foad20c0f022011-03-30 11:28:58 +00002639 llvm::PHINode *phi = Builder.CreatePHI(opTy, 2);
David Chisnalldd84ef12010-09-17 18:29:54 +00002640 phi->addIncoming(result, initialBB);
2641 phi->addIncoming(handlerResult, overflowBB);
2642
2643 return phi;
Mike Stump0c61b732009-04-01 20:28:16 +00002644}
Chris Lattner2da04b32007-08-24 05:35:26 +00002645
John McCall77527a82011-06-25 01:32:37 +00002646/// Emit pointer + index arithmetic.
2647static Value *emitPointerArithmetic(CodeGenFunction &CGF,
2648 const BinOpInfo &op,
2649 bool isSubtraction) {
2650 // Must have binary (not unary) expr here. Unary pointer
2651 // increment/decrement doesn't use this path.
2652 const BinaryOperator *expr = cast<BinaryOperator>(op.E);
Craig Toppera97d7e72013-07-26 06:16:11 +00002653
John McCall77527a82011-06-25 01:32:37 +00002654 Value *pointer = op.LHS;
2655 Expr *pointerOperand = expr->getLHS();
2656 Value *index = op.RHS;
2657 Expr *indexOperand = expr->getRHS();
2658
2659 // In a subtraction, the LHS is always the pointer.
2660 if (!isSubtraction && !pointer->getType()->isPointerTy()) {
2661 std::swap(pointer, index);
2662 std::swap(pointerOperand, indexOperand);
2663 }
2664
Vedant Kumar6dbf4272017-06-12 18:42:51 +00002665 bool isSigned = indexOperand->getType()->isSignedIntegerOrEnumerationType();
2666 bool mayHaveNegativeGEPIndex = isSigned || isSubtraction;
2667
John McCall77527a82011-06-25 01:32:37 +00002668 unsigned width = cast<llvm::IntegerType>(index->getType())->getBitWidth();
Yaxun Liu26f75662016-08-19 05:17:25 +00002669 auto &DL = CGF.CGM.getDataLayout();
2670 auto PtrTy = cast<llvm::PointerType>(pointer->getType());
2671 if (width != DL.getTypeSizeInBits(PtrTy)) {
John McCall77527a82011-06-25 01:32:37 +00002672 // Zero-extend or sign-extend the pointer value according to
2673 // whether the index is signed or not.
Yaxun Liu26f75662016-08-19 05:17:25 +00002674 index = CGF.Builder.CreateIntCast(index, DL.getIntPtrType(PtrTy), isSigned,
John McCall77527a82011-06-25 01:32:37 +00002675 "idx.ext");
2676 }
2677
2678 // If this is subtraction, negate the index.
2679 if (isSubtraction)
2680 index = CGF.Builder.CreateNeg(index, "idx.neg");
2681
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002682 if (CGF.SanOpts.has(SanitizerKind::ArrayBounds))
Richard Smith539e4a72013-02-23 02:53:19 +00002683 CGF.EmitBoundsCheck(op.E, pointerOperand, index, indexOperand->getType(),
2684 /*Accessed*/ false);
2685
John McCall77527a82011-06-25 01:32:37 +00002686 const PointerType *pointerType
2687 = pointerOperand->getType()->getAs<PointerType>();
2688 if (!pointerType) {
2689 QualType objectType = pointerOperand->getType()
2690 ->castAs<ObjCObjectPointerType>()
2691 ->getPointeeType();
2692 llvm::Value *objectSize
2693 = CGF.CGM.getSize(CGF.getContext().getTypeSizeInChars(objectType));
2694
2695 index = CGF.Builder.CreateMul(index, objectSize);
2696
2697 Value *result = CGF.Builder.CreateBitCast(pointer, CGF.VoidPtrTy);
2698 result = CGF.Builder.CreateGEP(result, index, "add.ptr");
2699 return CGF.Builder.CreateBitCast(result, pointer->getType());
2700 }
2701
2702 QualType elementType = pointerType->getPointeeType();
2703 if (const VariableArrayType *vla
2704 = CGF.getContext().getAsVariableArrayType(elementType)) {
2705 // The element count here is the total number of non-VLA elements.
2706 llvm::Value *numElements = CGF.getVLASize(vla).first;
2707
2708 // Effectively, the multiply by the VLA size is part of the GEP.
2709 // GEP indexes are signed, and scaling an index isn't permitted to
2710 // signed-overflow, so we use the same semantics for our explicit
2711 // multiply. We suppress this if overflow is not undefined behavior.
David Blaikiebbafb8a2012-03-11 07:00:24 +00002712 if (CGF.getLangOpts().isSignedOverflowDefined()) {
John McCall77527a82011-06-25 01:32:37 +00002713 index = CGF.Builder.CreateMul(index, numElements, "vla.index");
2714 pointer = CGF.Builder.CreateGEP(pointer, index, "add.ptr");
2715 } else {
2716 index = CGF.Builder.CreateNSWMul(index, numElements, "vla.index");
Vedant Kumar6dbf4272017-06-12 18:42:51 +00002717 pointer =
2718 CGF.EmitCheckedInBoundsGEP(pointer, index, mayHaveNegativeGEPIndex,
2719 op.E->getExprLoc(), "add.ptr");
Chris Lattner51924e512010-06-26 21:25:03 +00002720 }
John McCall77527a82011-06-25 01:32:37 +00002721 return pointer;
Mike Stump4a3999f2009-09-09 13:00:44 +00002722 }
Daniel Dunbaref2ffbc2009-04-25 05:08:32 +00002723
Mike Stump4a3999f2009-09-09 13:00:44 +00002724 // Explicitly handle GNU void* and function pointer arithmetic extensions. The
2725 // GNU void* casts amount to no-ops since our void* type is i8*, but this is
2726 // future proof.
John McCall77527a82011-06-25 01:32:37 +00002727 if (elementType->isVoidType() || elementType->isFunctionType()) {
2728 Value *result = CGF.Builder.CreateBitCast(pointer, CGF.VoidPtrTy);
2729 result = CGF.Builder.CreateGEP(result, index, "add.ptr");
2730 return CGF.Builder.CreateBitCast(result, pointer->getType());
Mike Stump4a3999f2009-09-09 13:00:44 +00002731 }
2732
David Blaikiebbafb8a2012-03-11 07:00:24 +00002733 if (CGF.getLangOpts().isSignedOverflowDefined())
John McCall77527a82011-06-25 01:32:37 +00002734 return CGF.Builder.CreateGEP(pointer, index, "add.ptr");
2735
Vedant Kumar6dbf4272017-06-12 18:42:51 +00002736 return CGF.EmitCheckedInBoundsGEP(pointer, index, mayHaveNegativeGEPIndex,
2737 op.E->getExprLoc(), "add.ptr");
Chris Lattner2da04b32007-08-24 05:35:26 +00002738}
2739
Lang Hames5de91cc2012-10-02 04:45:10 +00002740// Construct an fmuladd intrinsic to represent a fused mul-add of MulOp and
2741// Addend. Use negMul and negAdd to negate the first operand of the Mul or
2742// the add operand respectively. This allows fmuladd to represent a*b-c, or
2743// c-a*b. Patterns in LLVM should catch the negated forms and translate them to
2744// efficient operations.
2745static Value* buildFMulAdd(llvm::BinaryOperator *MulOp, Value *Addend,
2746 const CodeGenFunction &CGF, CGBuilderTy &Builder,
2747 bool negMul, bool negAdd) {
2748 assert(!(negMul && negAdd) && "Only one of negMul and negAdd should be set.");
Craig Toppera97d7e72013-07-26 06:16:11 +00002749
Lang Hames5de91cc2012-10-02 04:45:10 +00002750 Value *MulOp0 = MulOp->getOperand(0);
2751 Value *MulOp1 = MulOp->getOperand(1);
2752 if (negMul) {
2753 MulOp0 =
2754 Builder.CreateFSub(
2755 llvm::ConstantFP::getZeroValueForNegation(MulOp0->getType()), MulOp0,
2756 "neg");
2757 } else if (negAdd) {
2758 Addend =
2759 Builder.CreateFSub(
2760 llvm::ConstantFP::getZeroValueForNegation(Addend->getType()), Addend,
2761 "neg");
2762 }
2763
David Blaikie43f9bb72015-05-18 22:14:03 +00002764 Value *FMulAdd = Builder.CreateCall(
Lang Hames5de91cc2012-10-02 04:45:10 +00002765 CGF.CGM.getIntrinsic(llvm::Intrinsic::fmuladd, Addend->getType()),
David Blaikie43f9bb72015-05-18 22:14:03 +00002766 {MulOp0, MulOp1, Addend});
Lang Hames5de91cc2012-10-02 04:45:10 +00002767 MulOp->eraseFromParent();
2768
2769 return FMulAdd;
2770}
2771
2772// Check whether it would be legal to emit an fmuladd intrinsic call to
2773// represent op and if so, build the fmuladd.
2774//
2775// Checks that (a) the operation is fusable, and (b) -ffp-contract=on.
2776// Does NOT check the type of the operation - it's assumed that this function
2777// will be called from contexts where it's known that the type is contractable.
Craig Toppera97d7e72013-07-26 06:16:11 +00002778static Value* tryEmitFMulAdd(const BinOpInfo &op,
Lang Hames5de91cc2012-10-02 04:45:10 +00002779 const CodeGenFunction &CGF, CGBuilderTy &Builder,
2780 bool isSub=false) {
2781
2782 assert((op.Opcode == BO_Add || op.Opcode == BO_AddAssign ||
2783 op.Opcode == BO_Sub || op.Opcode == BO_SubAssign) &&
2784 "Only fadd/fsub can be the root of an fmuladd.");
2785
2786 // Check whether this op is marked as fusable.
Adam Nemet049a31d2017-03-29 21:54:24 +00002787 if (!op.FPFeatures.allowFPContractWithinStatement())
Craig Topper8a13c412014-05-21 05:09:00 +00002788 return nullptr;
Lang Hames5de91cc2012-10-02 04:45:10 +00002789
2790 // We have a potentially fusable op. Look for a mul on one of the operands.
Sanjay Patela30cee62015-12-03 01:25:12 +00002791 // Also, make sure that the mul result isn't used directly. In that case,
2792 // there's no point creating a muladd operation.
2793 if (auto *LHSBinOp = dyn_cast<llvm::BinaryOperator>(op.LHS)) {
2794 if (LHSBinOp->getOpcode() == llvm::Instruction::FMul &&
2795 LHSBinOp->use_empty())
Lang Hames5de91cc2012-10-02 04:45:10 +00002796 return buildFMulAdd(LHSBinOp, op.RHS, CGF, Builder, false, isSub);
Sanjay Patela30cee62015-12-03 01:25:12 +00002797 }
2798 if (auto *RHSBinOp = dyn_cast<llvm::BinaryOperator>(op.RHS)) {
2799 if (RHSBinOp->getOpcode() == llvm::Instruction::FMul &&
2800 RHSBinOp->use_empty())
Lang Hames5de91cc2012-10-02 04:45:10 +00002801 return buildFMulAdd(RHSBinOp, op.LHS, CGF, Builder, isSub, false);
Lang Hames5de91cc2012-10-02 04:45:10 +00002802 }
2803
Craig Topper8a13c412014-05-21 05:09:00 +00002804 return nullptr;
Lang Hames5de91cc2012-10-02 04:45:10 +00002805}
2806
John McCall77527a82011-06-25 01:32:37 +00002807Value *ScalarExprEmitter::EmitAdd(const BinOpInfo &op) {
2808 if (op.LHS->getType()->isPointerTy() ||
2809 op.RHS->getType()->isPointerTy())
2810 return emitPointerArithmetic(CGF, op, /*subtraction*/ false);
2811
2812 if (op.Ty->isSignedIntegerOrEnumerationType()) {
Richard Smith9c6890a2012-11-01 22:30:59 +00002813 switch (CGF.getLangOpts().getSignedOverflowBehavior()) {
John McCall77527a82011-06-25 01:32:37 +00002814 case LangOptions::SOB_Defined:
2815 return Builder.CreateAdd(op.LHS, op.RHS, "add");
Richard Smith3e056de2012-08-25 00:32:28 +00002816 case LangOptions::SOB_Undefined:
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002817 if (!CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow))
Richard Smith3e056de2012-08-25 00:32:28 +00002818 return Builder.CreateNSWAdd(op.LHS, op.RHS, "add");
2819 // Fall through.
John McCall77527a82011-06-25 01:32:37 +00002820 case LangOptions::SOB_Trapping:
Vedant Kumar82ee16b2017-02-25 00:43:36 +00002821 if (CanElideOverflowCheck(CGF.getContext(), op))
2822 return Builder.CreateNSWAdd(op.LHS, op.RHS, "add");
John McCall77527a82011-06-25 01:32:37 +00002823 return EmitOverflowCheckedBinOp(op);
2824 }
2825 }
Will Dietz1897cb32012-11-27 15:01:55 +00002826
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002827 if (op.Ty->isUnsignedIntegerType() &&
Vedant Kumar82ee16b2017-02-25 00:43:36 +00002828 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow) &&
2829 !CanElideOverflowCheck(CGF.getContext(), op))
Will Dietz1897cb32012-11-27 15:01:55 +00002830 return EmitOverflowCheckedBinOp(op);
2831
Lang Hames5de91cc2012-10-02 04:45:10 +00002832 if (op.LHS->getType()->isFPOrFPVectorTy()) {
2833 // Try to form an fmuladd.
2834 if (Value *FMulAdd = tryEmitFMulAdd(op, CGF, Builder))
2835 return FMulAdd;
2836
Adam Nemet370d0872017-04-04 21:18:30 +00002837 Value *V = Builder.CreateFAdd(op.LHS, op.RHS, "add");
2838 return propagateFMFlags(V, op);
Lang Hames5de91cc2012-10-02 04:45:10 +00002839 }
John McCall77527a82011-06-25 01:32:37 +00002840
2841 return Builder.CreateAdd(op.LHS, op.RHS, "add");
2842}
2843
2844Value *ScalarExprEmitter::EmitSub(const BinOpInfo &op) {
2845 // The LHS is always a pointer if either side is.
2846 if (!op.LHS->getType()->isPointerTy()) {
2847 if (op.Ty->isSignedIntegerOrEnumerationType()) {
Richard Smith9c6890a2012-11-01 22:30:59 +00002848 switch (CGF.getLangOpts().getSignedOverflowBehavior()) {
Chris Lattner51924e512010-06-26 21:25:03 +00002849 case LangOptions::SOB_Defined:
John McCall77527a82011-06-25 01:32:37 +00002850 return Builder.CreateSub(op.LHS, op.RHS, "sub");
Richard Smith3e056de2012-08-25 00:32:28 +00002851 case LangOptions::SOB_Undefined:
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002852 if (!CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow))
Richard Smith3e056de2012-08-25 00:32:28 +00002853 return Builder.CreateNSWSub(op.LHS, op.RHS, "sub");
2854 // Fall through.
Chris Lattner51924e512010-06-26 21:25:03 +00002855 case LangOptions::SOB_Trapping:
Vedant Kumar82ee16b2017-02-25 00:43:36 +00002856 if (CanElideOverflowCheck(CGF.getContext(), op))
2857 return Builder.CreateNSWSub(op.LHS, op.RHS, "sub");
John McCall77527a82011-06-25 01:32:37 +00002858 return EmitOverflowCheckedBinOp(op);
Chris Lattner51924e512010-06-26 21:25:03 +00002859 }
2860 }
Will Dietz1897cb32012-11-27 15:01:55 +00002861
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002862 if (op.Ty->isUnsignedIntegerType() &&
Vedant Kumar82ee16b2017-02-25 00:43:36 +00002863 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow) &&
2864 !CanElideOverflowCheck(CGF.getContext(), op))
Will Dietz1897cb32012-11-27 15:01:55 +00002865 return EmitOverflowCheckedBinOp(op);
2866
Lang Hames5de91cc2012-10-02 04:45:10 +00002867 if (op.LHS->getType()->isFPOrFPVectorTy()) {
2868 // Try to form an fmuladd.
2869 if (Value *FMulAdd = tryEmitFMulAdd(op, CGF, Builder, true))
2870 return FMulAdd;
Adam Nemet370d0872017-04-04 21:18:30 +00002871 Value *V = Builder.CreateFSub(op.LHS, op.RHS, "sub");
2872 return propagateFMFlags(V, op);
Lang Hames5de91cc2012-10-02 04:45:10 +00002873 }
Chris Lattner5902e7b2010-03-29 17:28:16 +00002874
John McCall77527a82011-06-25 01:32:37 +00002875 return Builder.CreateSub(op.LHS, op.RHS, "sub");
Mike Stump0c61b732009-04-01 20:28:16 +00002876 }
Chris Lattner3d966d62007-08-24 21:00:35 +00002877
John McCall77527a82011-06-25 01:32:37 +00002878 // If the RHS is not a pointer, then we have normal pointer
2879 // arithmetic.
2880 if (!op.RHS->getType()->isPointerTy())
2881 return emitPointerArithmetic(CGF, op, /*subtraction*/ true);
Eli Friedmane381f7e2009-03-28 02:45:41 +00002882
John McCall77527a82011-06-25 01:32:37 +00002883 // Otherwise, this is a pointer subtraction.
Daniel Dunbar42a8cd32009-01-23 18:51:09 +00002884
John McCall77527a82011-06-25 01:32:37 +00002885 // Do the raw subtraction part.
2886 llvm::Value *LHS
2887 = Builder.CreatePtrToInt(op.LHS, CGF.PtrDiffTy, "sub.ptr.lhs.cast");
2888 llvm::Value *RHS
2889 = Builder.CreatePtrToInt(op.RHS, CGF.PtrDiffTy, "sub.ptr.rhs.cast");
2890 Value *diffInChars = Builder.CreateSub(LHS, RHS, "sub.ptr.sub");
Daniel Dunbaref2ffbc2009-04-25 05:08:32 +00002891
John McCall77527a82011-06-25 01:32:37 +00002892 // Okay, figure out the element size.
2893 const BinaryOperator *expr = cast<BinaryOperator>(op.E);
2894 QualType elementType = expr->getLHS()->getType()->getPointeeType();
Mike Stump4a3999f2009-09-09 13:00:44 +00002895
Craig Topper8a13c412014-05-21 05:09:00 +00002896 llvm::Value *divisor = nullptr;
John McCall77527a82011-06-25 01:32:37 +00002897
2898 // For a variable-length array, this is going to be non-constant.
2899 if (const VariableArrayType *vla
2900 = CGF.getContext().getAsVariableArrayType(elementType)) {
2901 llvm::Value *numElements;
Benjamin Kramer867ea1d2014-03-02 13:01:17 +00002902 std::tie(numElements, elementType) = CGF.getVLASize(vla);
John McCall77527a82011-06-25 01:32:37 +00002903
2904 divisor = numElements;
2905
2906 // Scale the number of non-VLA elements by the non-VLA element size.
2907 CharUnits eltSize = CGF.getContext().getTypeSizeInChars(elementType);
2908 if (!eltSize.isOne())
2909 divisor = CGF.Builder.CreateNUWMul(CGF.CGM.getSize(eltSize), divisor);
2910
2911 // For everything elese, we can just compute it, safe in the
2912 // assumption that Sema won't let anything through that we can't
2913 // safely compute the size of.
2914 } else {
2915 CharUnits elementSize;
2916 // Handle GCC extension for pointer arithmetic on void* and
2917 // function pointer types.
2918 if (elementType->isVoidType() || elementType->isFunctionType())
2919 elementSize = CharUnits::One();
2920 else
2921 elementSize = CGF.getContext().getTypeSizeInChars(elementType);
2922
2923 // Don't even emit the divide for element size of 1.
2924 if (elementSize.isOne())
2925 return diffInChars;
2926
2927 divisor = CGF.CGM.getSize(elementSize);
Chris Lattner2da04b32007-08-24 05:35:26 +00002928 }
Craig Toppera97d7e72013-07-26 06:16:11 +00002929
Chris Lattner2e72da942011-03-01 00:03:48 +00002930 // Otherwise, do a full sdiv. This uses the "exact" form of sdiv, since
2931 // pointer difference in C is only defined in the case where both operands
2932 // are pointing to elements of an array.
John McCall77527a82011-06-25 01:32:37 +00002933 return Builder.CreateExactSDiv(diffInChars, divisor, "sub.ptr.div");
Chris Lattner2da04b32007-08-24 05:35:26 +00002934}
2935
David Tweed042e0882013-01-07 16:43:27 +00002936Value *ScalarExprEmitter::GetWidthMinusOneValue(Value* LHS,Value* RHS) {
David Tweed9fb566c2013-01-10 09:11:33 +00002937 llvm::IntegerType *Ty;
2938 if (llvm::VectorType *VT = dyn_cast<llvm::VectorType>(LHS->getType()))
2939 Ty = cast<llvm::IntegerType>(VT->getElementType());
2940 else
2941 Ty = cast<llvm::IntegerType>(LHS->getType());
2942 return llvm::ConstantInt::get(RHS->getType(), Ty->getBitWidth() - 1);
David Tweed042e0882013-01-07 16:43:27 +00002943}
2944
Chris Lattner2da04b32007-08-24 05:35:26 +00002945Value *ScalarExprEmitter::EmitShl(const BinOpInfo &Ops) {
2946 // LLVM requires the LHS and RHS to be the same type: promote or truncate the
2947 // RHS to the same size as the LHS.
2948 Value *RHS = Ops.RHS;
2949 if (Ops.LHS->getType() != RHS->getType())
2950 RHS = Builder.CreateIntCast(RHS, Ops.LHS->getType(), false, "sh_prom");
Mike Stump4a3999f2009-09-09 13:00:44 +00002951
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00002952 bool SanitizeBase = CGF.SanOpts.has(SanitizerKind::ShiftBase) &&
James Molloy59802322016-08-16 09:45:36 +00002953 Ops.Ty->hasSignedIntegerRepresentation() &&
2954 !CGF.getLangOpts().isSignedOverflowDefined();
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00002955 bool SanitizeExponent = CGF.SanOpts.has(SanitizerKind::ShiftExponent);
2956 // OpenCL 6.3j: shift values are effectively % word size of LHS.
2957 if (CGF.getLangOpts().OpenCL)
2958 RHS =
2959 Builder.CreateAnd(RHS, GetWidthMinusOneValue(Ops.LHS, RHS), "shl.mask");
2960 else if ((SanitizeBase || SanitizeExponent) &&
2961 isa<llvm::IntegerType>(Ops.LHS->getType())) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00002962 CodeGenFunction::SanitizerScope SanScope(&CGF);
Peter Collingbourne3eea6772015-05-11 21:39:14 +00002963 SmallVector<std::pair<Value *, SanitizerMask>, 2> Checks;
Vedant Kumard3a601b2017-01-30 23:38:54 +00002964 llvm::Value *WidthMinusOne = GetWidthMinusOneValue(Ops.LHS, Ops.RHS);
2965 llvm::Value *ValidExponent = Builder.CreateICmpULE(Ops.RHS, WidthMinusOne);
Richard Smith3e056de2012-08-25 00:32:28 +00002966
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00002967 if (SanitizeExponent) {
2968 Checks.push_back(
2969 std::make_pair(ValidExponent, SanitizerKind::ShiftExponent));
2970 }
2971
2972 if (SanitizeBase) {
2973 // Check whether we are shifting any non-zero bits off the top of the
2974 // integer. We only emit this check if exponent is valid - otherwise
2975 // instructions below will have undefined behavior themselves.
Alexey Samsonov48a9db02015-03-05 21:57:35 +00002976 llvm::BasicBlock *Orig = Builder.GetInsertBlock();
2977 llvm::BasicBlock *Cont = CGF.createBasicBlock("cont");
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00002978 llvm::BasicBlock *CheckShiftBase = CGF.createBasicBlock("check");
2979 Builder.CreateCondBr(ValidExponent, CheckShiftBase, Cont);
Vedant Kumard3a601b2017-01-30 23:38:54 +00002980 llvm::Value *PromotedWidthMinusOne =
2981 (RHS == Ops.RHS) ? WidthMinusOne
2982 : GetWidthMinusOneValue(Ops.LHS, RHS);
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00002983 CGF.EmitBlock(CheckShiftBase);
Vedant Kumard3a601b2017-01-30 23:38:54 +00002984 llvm::Value *BitsShiftedOff = Builder.CreateLShr(
2985 Ops.LHS, Builder.CreateSub(PromotedWidthMinusOne, RHS, "shl.zeros",
2986 /*NUW*/ true, /*NSW*/ true),
2987 "shl.check");
Richard Smith3e056de2012-08-25 00:32:28 +00002988 if (CGF.getLangOpts().CPlusPlus) {
2989 // In C99, we are not permitted to shift a 1 bit into the sign bit.
2990 // Under C++11's rules, shifting a 1 bit into the sign bit is
2991 // OK, but shifting a 1 bit out of it is not. (C89 and C++03 don't
2992 // define signed left shifts, so we use the C99 and C++11 rules there).
2993 llvm::Value *One = llvm::ConstantInt::get(BitsShiftedOff->getType(), 1);
2994 BitsShiftedOff = Builder.CreateLShr(BitsShiftedOff, One);
2995 }
2996 llvm::Value *Zero = llvm::ConstantInt::get(BitsShiftedOff->getType(), 0);
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00002997 llvm::Value *ValidBase = Builder.CreateICmpEQ(BitsShiftedOff, Zero);
Alexey Samsonov48a9db02015-03-05 21:57:35 +00002998 CGF.EmitBlock(Cont);
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00002999 llvm::PHINode *BaseCheck = Builder.CreatePHI(ValidBase->getType(), 2);
3000 BaseCheck->addIncoming(Builder.getTrue(), Orig);
3001 BaseCheck->addIncoming(ValidBase, CheckShiftBase);
3002 Checks.push_back(std::make_pair(BaseCheck, SanitizerKind::ShiftBase));
Richard Smith3e056de2012-08-25 00:32:28 +00003003 }
Will Dietz11d0a9f2013-02-25 22:37:49 +00003004
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003005 assert(!Checks.empty());
3006 EmitBinOpCheck(Checks, Ops);
Mike Stumpba6a0c42009-12-14 21:58:14 +00003007 }
3008
Chris Lattner2da04b32007-08-24 05:35:26 +00003009 return Builder.CreateShl(Ops.LHS, RHS, "shl");
3010}
3011
3012Value *ScalarExprEmitter::EmitShr(const BinOpInfo &Ops) {
3013 // LLVM requires the LHS and RHS to be the same type: promote or truncate the
3014 // RHS to the same size as the LHS.
3015 Value *RHS = Ops.RHS;
3016 if (Ops.LHS->getType() != RHS->getType())
3017 RHS = Builder.CreateIntCast(RHS, Ops.LHS->getType(), false, "sh_prom");
Mike Stump4a3999f2009-09-09 13:00:44 +00003018
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003019 // OpenCL 6.3j: shift values are effectively % word size of LHS.
3020 if (CGF.getLangOpts().OpenCL)
3021 RHS =
3022 Builder.CreateAnd(RHS, GetWidthMinusOneValue(Ops.LHS, RHS), "shr.mask");
3023 else if (CGF.SanOpts.has(SanitizerKind::ShiftExponent) &&
3024 isa<llvm::IntegerType>(Ops.LHS->getType())) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00003025 CodeGenFunction::SanitizerScope SanScope(&CGF);
Alexey Samsonove396bfc2014-11-11 22:03:54 +00003026 llvm::Value *Valid =
3027 Builder.CreateICmpULE(RHS, GetWidthMinusOneValue(Ops.LHS, RHS));
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003028 EmitBinOpCheck(std::make_pair(Valid, SanitizerKind::ShiftExponent), Ops);
Alexey Samsonov24cad992014-07-17 18:46:27 +00003029 }
David Tweed042e0882013-01-07 16:43:27 +00003030
Douglas Gregor5cc2c8b2010-07-23 15:58:24 +00003031 if (Ops.Ty->hasUnsignedIntegerRepresentation())
Chris Lattner2da04b32007-08-24 05:35:26 +00003032 return Builder.CreateLShr(Ops.LHS, RHS, "shr");
3033 return Builder.CreateAShr(Ops.LHS, RHS, "shr");
3034}
3035
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003036enum IntrinsicType { VCMPEQ, VCMPGT };
3037// return corresponding comparison intrinsic for given vector type
3038static llvm::Intrinsic::ID GetIntrinsic(IntrinsicType IT,
3039 BuiltinType::Kind ElemKind) {
3040 switch (ElemKind) {
David Blaikie83d382b2011-09-23 05:06:16 +00003041 default: llvm_unreachable("unexpected element type");
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003042 case BuiltinType::Char_U:
3043 case BuiltinType::UChar:
3044 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequb_p :
3045 llvm::Intrinsic::ppc_altivec_vcmpgtub_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003046 case BuiltinType::Char_S:
3047 case BuiltinType::SChar:
3048 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequb_p :
3049 llvm::Intrinsic::ppc_altivec_vcmpgtsb_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003050 case BuiltinType::UShort:
3051 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequh_p :
3052 llvm::Intrinsic::ppc_altivec_vcmpgtuh_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003053 case BuiltinType::Short:
3054 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequh_p :
3055 llvm::Intrinsic::ppc_altivec_vcmpgtsh_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003056 case BuiltinType::UInt:
3057 case BuiltinType::ULong:
3058 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequw_p :
3059 llvm::Intrinsic::ppc_altivec_vcmpgtuw_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003060 case BuiltinType::Int:
3061 case BuiltinType::Long:
3062 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequw_p :
3063 llvm::Intrinsic::ppc_altivec_vcmpgtsw_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003064 case BuiltinType::Float:
3065 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpeqfp_p :
3066 llvm::Intrinsic::ppc_altivec_vcmpgtfp_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003067 }
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003068}
3069
Craig Topperc82f8962015-12-16 06:24:28 +00003070Value *ScalarExprEmitter::EmitCompare(const BinaryOperator *E,
3071 llvm::CmpInst::Predicate UICmpOpc,
3072 llvm::CmpInst::Predicate SICmpOpc,
3073 llvm::CmpInst::Predicate FCmpOpc) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00003074 TestAndClearIgnoreResultAssign();
Chris Lattner42e6b812007-08-26 16:34:22 +00003075 Value *Result;
Chris Lattner2da04b32007-08-24 05:35:26 +00003076 QualType LHSTy = E->getLHS()->getType();
Chandler Carruthb29a7432014-10-11 11:03:30 +00003077 QualType RHSTy = E->getRHS()->getType();
John McCall7a9aac22010-08-23 01:21:21 +00003078 if (const MemberPointerType *MPT = LHSTy->getAs<MemberPointerType>()) {
John McCalle3027922010-08-25 11:45:40 +00003079 assert(E->getOpcode() == BO_EQ ||
3080 E->getOpcode() == BO_NE);
John McCalla1dee5302010-08-22 10:59:02 +00003081 Value *LHS = CGF.EmitScalarExpr(E->getLHS());
3082 Value *RHS = CGF.EmitScalarExpr(E->getRHS());
John McCall7a9aac22010-08-23 01:21:21 +00003083 Result = CGF.CGM.getCXXABI().EmitMemberPointerComparison(
John McCalle3027922010-08-25 11:45:40 +00003084 CGF, LHS, RHS, MPT, E->getOpcode() == BO_NE);
Chandler Carruthb29a7432014-10-11 11:03:30 +00003085 } else if (!LHSTy->isAnyComplexType() && !RHSTy->isAnyComplexType()) {
Chris Lattner2da04b32007-08-24 05:35:26 +00003086 Value *LHS = Visit(E->getLHS());
3087 Value *RHS = Visit(E->getRHS());
Mike Stump4a3999f2009-09-09 13:00:44 +00003088
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003089 // If AltiVec, the comparison results in a numeric type, so we use
3090 // intrinsics comparing vectors and giving 0 or 1 as a result
Anton Yartsev93900c72011-03-28 21:00:05 +00003091 if (LHSTy->isVectorType() && !E->getType()->isVectorType()) {
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003092 // constants for mapping CR6 register bits to predicate result
3093 enum { CR6_EQ=0, CR6_EQ_REV, CR6_LT, CR6_LT_REV } CR6;
3094
3095 llvm::Intrinsic::ID ID = llvm::Intrinsic::not_intrinsic;
3096
3097 // in several cases vector arguments order will be reversed
3098 Value *FirstVecArg = LHS,
3099 *SecondVecArg = RHS;
3100
3101 QualType ElTy = LHSTy->getAs<VectorType>()->getElementType();
John McCall424cec92011-01-19 06:33:43 +00003102 const BuiltinType *BTy = ElTy->getAs<BuiltinType>();
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003103 BuiltinType::Kind ElementKind = BTy->getKind();
3104
3105 switch(E->getOpcode()) {
David Blaikie83d382b2011-09-23 05:06:16 +00003106 default: llvm_unreachable("is not a comparison operation");
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003107 case BO_EQ:
3108 CR6 = CR6_LT;
3109 ID = GetIntrinsic(VCMPEQ, ElementKind);
3110 break;
3111 case BO_NE:
3112 CR6 = CR6_EQ;
3113 ID = GetIntrinsic(VCMPEQ, ElementKind);
3114 break;
3115 case BO_LT:
3116 CR6 = CR6_LT;
3117 ID = GetIntrinsic(VCMPGT, ElementKind);
3118 std::swap(FirstVecArg, SecondVecArg);
3119 break;
3120 case BO_GT:
3121 CR6 = CR6_LT;
3122 ID = GetIntrinsic(VCMPGT, ElementKind);
3123 break;
3124 case BO_LE:
3125 if (ElementKind == BuiltinType::Float) {
3126 CR6 = CR6_LT;
3127 ID = llvm::Intrinsic::ppc_altivec_vcmpgefp_p;
3128 std::swap(FirstVecArg, SecondVecArg);
3129 }
3130 else {
3131 CR6 = CR6_EQ;
3132 ID = GetIntrinsic(VCMPGT, ElementKind);
3133 }
3134 break;
3135 case BO_GE:
3136 if (ElementKind == BuiltinType::Float) {
3137 CR6 = CR6_LT;
3138 ID = llvm::Intrinsic::ppc_altivec_vcmpgefp_p;
3139 }
3140 else {
3141 CR6 = CR6_EQ;
3142 ID = GetIntrinsic(VCMPGT, ElementKind);
3143 std::swap(FirstVecArg, SecondVecArg);
3144 }
3145 break;
3146 }
3147
Chris Lattner2531eb42011-04-19 22:55:03 +00003148 Value *CR6Param = Builder.getInt32(CR6);
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003149 llvm::Function *F = CGF.CGM.getIntrinsic(ID);
David Blaikie43f9bb72015-05-18 22:14:03 +00003150 Result = Builder.CreateCall(F, {CR6Param, FirstVecArg, SecondVecArg});
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00003151 return EmitScalarConversion(Result, CGF.getContext().BoolTy, E->getType(),
3152 E->getExprLoc());
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003153 }
3154
Duncan Sands998f9d92010-02-15 16:14:01 +00003155 if (LHS->getType()->isFPOrFPVectorTy()) {
Craig Topperc82f8962015-12-16 06:24:28 +00003156 Result = Builder.CreateFCmp(FCmpOpc, LHS, RHS, "cmp");
Douglas Gregor5cc2c8b2010-07-23 15:58:24 +00003157 } else if (LHSTy->hasSignedIntegerRepresentation()) {
Craig Topperc82f8962015-12-16 06:24:28 +00003158 Result = Builder.CreateICmp(SICmpOpc, LHS, RHS, "cmp");
Chris Lattner2da04b32007-08-24 05:35:26 +00003159 } else {
Eli Friedman3c285242008-05-29 15:09:15 +00003160 // Unsigned integers and pointers.
Craig Topperc82f8962015-12-16 06:24:28 +00003161 Result = Builder.CreateICmp(UICmpOpc, LHS, RHS, "cmp");
Chris Lattner2da04b32007-08-24 05:35:26 +00003162 }
Chris Lattner2a7deb62009-07-08 01:08:03 +00003163
3164 // If this is a vector comparison, sign extend the result to the appropriate
3165 // vector integer type and return it (don't convert to bool).
3166 if (LHSTy->isVectorType())
3167 return Builder.CreateSExt(Result, ConvertType(E->getType()), "sext");
Mike Stump4a3999f2009-09-09 13:00:44 +00003168
Chris Lattner2da04b32007-08-24 05:35:26 +00003169 } else {
3170 // Complex Comparison: can only be an equality comparison.
Chandler Carruthb29a7432014-10-11 11:03:30 +00003171 CodeGenFunction::ComplexPairTy LHS, RHS;
3172 QualType CETy;
3173 if (auto *CTy = LHSTy->getAs<ComplexType>()) {
3174 LHS = CGF.EmitComplexExpr(E->getLHS());
3175 CETy = CTy->getElementType();
3176 } else {
3177 LHS.first = Visit(E->getLHS());
3178 LHS.second = llvm::Constant::getNullValue(LHS.first->getType());
3179 CETy = LHSTy;
3180 }
3181 if (auto *CTy = RHSTy->getAs<ComplexType>()) {
3182 RHS = CGF.EmitComplexExpr(E->getRHS());
3183 assert(CGF.getContext().hasSameUnqualifiedType(CETy,
3184 CTy->getElementType()) &&
3185 "The element types must always match.");
Chandler Carruth60fdc412014-10-11 11:29:26 +00003186 (void)CTy;
Chandler Carruthb29a7432014-10-11 11:03:30 +00003187 } else {
3188 RHS.first = Visit(E->getRHS());
3189 RHS.second = llvm::Constant::getNullValue(RHS.first->getType());
3190 assert(CGF.getContext().hasSameUnqualifiedType(CETy, RHSTy) &&
3191 "The element types must always match.");
3192 }
Mike Stump4a3999f2009-09-09 13:00:44 +00003193
Chris Lattner42e6b812007-08-26 16:34:22 +00003194 Value *ResultR, *ResultI;
Chris Lattner2da04b32007-08-24 05:35:26 +00003195 if (CETy->isRealFloatingType()) {
Craig Topperc82f8962015-12-16 06:24:28 +00003196 ResultR = Builder.CreateFCmp(FCmpOpc, LHS.first, RHS.first, "cmp.r");
3197 ResultI = Builder.CreateFCmp(FCmpOpc, LHS.second, RHS.second, "cmp.i");
Chris Lattner2da04b32007-08-24 05:35:26 +00003198 } else {
3199 // Complex comparisons can only be equality comparisons. As such, signed
3200 // and unsigned opcodes are the same.
Craig Topperc82f8962015-12-16 06:24:28 +00003201 ResultR = Builder.CreateICmp(UICmpOpc, LHS.first, RHS.first, "cmp.r");
3202 ResultI = Builder.CreateICmp(UICmpOpc, LHS.second, RHS.second, "cmp.i");
Chris Lattner2da04b32007-08-24 05:35:26 +00003203 }
Mike Stump4a3999f2009-09-09 13:00:44 +00003204
John McCalle3027922010-08-25 11:45:40 +00003205 if (E->getOpcode() == BO_EQ) {
Chris Lattner2da04b32007-08-24 05:35:26 +00003206 Result = Builder.CreateAnd(ResultR, ResultI, "and.ri");
3207 } else {
John McCalle3027922010-08-25 11:45:40 +00003208 assert(E->getOpcode() == BO_NE &&
Chris Lattner2da04b32007-08-24 05:35:26 +00003209 "Complex comparison other than == or != ?");
3210 Result = Builder.CreateOr(ResultR, ResultI, "or.ri");
3211 }
3212 }
Nuno Lopesa0abe622009-01-11 23:22:37 +00003213
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00003214 return EmitScalarConversion(Result, CGF.getContext().BoolTy, E->getType(),
3215 E->getExprLoc());
Chris Lattner2da04b32007-08-24 05:35:26 +00003216}
3217
3218Value *ScalarExprEmitter::VisitBinAssign(const BinaryOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00003219 bool Ignore = TestAndClearIgnoreResultAssign();
3220
John McCall31168b02011-06-15 23:02:42 +00003221 Value *RHS;
3222 LValue LHS;
Mike Stump4a3999f2009-09-09 13:00:44 +00003223
John McCall31168b02011-06-15 23:02:42 +00003224 switch (E->getLHS()->getType().getObjCLifetime()) {
3225 case Qualifiers::OCL_Strong:
Benjamin Kramer867ea1d2014-03-02 13:01:17 +00003226 std::tie(LHS, RHS) = CGF.EmitARCStoreStrong(E, Ignore);
John McCall31168b02011-06-15 23:02:42 +00003227 break;
3228
3229 case Qualifiers::OCL_Autoreleasing:
Benjamin Kramer867ea1d2014-03-02 13:01:17 +00003230 std::tie(LHS, RHS) = CGF.EmitARCStoreAutoreleasing(E);
John McCall31168b02011-06-15 23:02:42 +00003231 break;
3232
John McCalle399e5b2016-01-27 18:32:30 +00003233 case Qualifiers::OCL_ExplicitNone:
3234 std::tie(LHS, RHS) = CGF.EmitARCStoreUnsafeUnretained(E, Ignore);
3235 break;
3236
John McCall31168b02011-06-15 23:02:42 +00003237 case Qualifiers::OCL_Weak:
3238 RHS = Visit(E->getRHS());
Richard Smith4d1458e2012-09-08 02:08:36 +00003239 LHS = EmitCheckedLValue(E->getLHS(), CodeGenFunction::TCK_Store);
John McCall31168b02011-06-15 23:02:42 +00003240 RHS = CGF.EmitARCStoreWeak(LHS.getAddress(), RHS, Ignore);
3241 break;
3242
John McCall31168b02011-06-15 23:02:42 +00003243 case Qualifiers::OCL_None:
John McCall31168b02011-06-15 23:02:42 +00003244 // __block variables need to have the rhs evaluated first, plus
3245 // this should improve codegen just a little.
3246 RHS = Visit(E->getRHS());
Richard Smith4d1458e2012-09-08 02:08:36 +00003247 LHS = EmitCheckedLValue(E->getLHS(), CodeGenFunction::TCK_Store);
John McCall31168b02011-06-15 23:02:42 +00003248
3249 // Store the value into the LHS. Bit-fields are handled specially
3250 // because the result is altered by the store, i.e., [C99 6.5.16p1]
3251 // 'An assignment expression has the value of the left operand after
3252 // the assignment...'.
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003253 if (LHS.isBitField()) {
John McCall55e1fbc2011-06-25 02:11:03 +00003254 CGF.EmitStoreThroughBitfieldLValue(RValue::get(RHS), LHS, &RHS);
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003255 } else {
3256 CGF.EmitNullabilityCheck(LHS, RHS, E->getExprLoc());
John McCall55e1fbc2011-06-25 02:11:03 +00003257 CGF.EmitStoreThroughLValue(RValue::get(RHS), LHS);
Vedant Kumar42c17ec2017-03-14 01:56:34 +00003258 }
John McCall31168b02011-06-15 23:02:42 +00003259 }
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00003260
3261 // If the result is clearly ignored, return now.
Mike Stumpdf0fe272009-05-29 15:46:01 +00003262 if (Ignore)
Craig Topper8a13c412014-05-21 05:09:00 +00003263 return nullptr;
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00003264
John McCall07bb1962010-11-16 10:08:07 +00003265 // The result of an assignment in C is the assigned r-value.
Richard Smith9c6890a2012-11-01 22:30:59 +00003266 if (!CGF.getLangOpts().CPlusPlus)
John McCall07bb1962010-11-16 10:08:07 +00003267 return RHS;
3268
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00003269 // If the lvalue is non-volatile, return the computed value of the assignment.
3270 if (!LHS.isVolatileQualified())
3271 return RHS;
3272
3273 // Otherwise, reload the value.
Nick Lewycky2d84e842013-10-02 02:29:49 +00003274 return EmitLoadOfLValue(LHS, E->getExprLoc());
Chris Lattner2da04b32007-08-24 05:35:26 +00003275}
3276
3277Value *ScalarExprEmitter::VisitBinLAnd(const BinaryOperator *E) {
Tanya Lattner20248222012-01-16 21:02:28 +00003278 // Perform vector logical and on comparisons with zero vectors.
3279 if (E->getType()->isVectorType()) {
Justin Bogner66242d62015-04-23 23:06:47 +00003280 CGF.incrementProfileCounter(E);
Justin Bogneref512b92014-01-06 22:27:43 +00003281
Tanya Lattner20248222012-01-16 21:02:28 +00003282 Value *LHS = Visit(E->getLHS());
3283 Value *RHS = Visit(E->getRHS());
3284 Value *Zero = llvm::ConstantAggregateZero::get(LHS->getType());
Joey Gouly7d00f002013-02-21 11:49:56 +00003285 if (LHS->getType()->isFPOrFPVectorTy()) {
3286 LHS = Builder.CreateFCmp(llvm::CmpInst::FCMP_UNE, LHS, Zero, "cmp");
3287 RHS = Builder.CreateFCmp(llvm::CmpInst::FCMP_UNE, RHS, Zero, "cmp");
3288 } else {
3289 LHS = Builder.CreateICmp(llvm::CmpInst::ICMP_NE, LHS, Zero, "cmp");
3290 RHS = Builder.CreateICmp(llvm::CmpInst::ICMP_NE, RHS, Zero, "cmp");
3291 }
Tanya Lattner20248222012-01-16 21:02:28 +00003292 Value *And = Builder.CreateAnd(LHS, RHS);
Joey Gouly7d00f002013-02-21 11:49:56 +00003293 return Builder.CreateSExt(And, ConvertType(E->getType()), "sext");
Tanya Lattner20248222012-01-16 21:02:28 +00003294 }
Craig Toppera97d7e72013-07-26 06:16:11 +00003295
Chris Lattner2192fe52011-07-18 04:24:23 +00003296 llvm::Type *ResTy = ConvertType(E->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00003297
Chris Lattner8b084582008-11-12 08:26:50 +00003298 // If we have 0 && RHS, see if we can elide RHS, if so, just return 0.
3299 // If we have 1 && X, just emit X without inserting the control flow.
Chris Lattner41c6ab52011-02-27 23:02:32 +00003300 bool LHSCondVal;
3301 if (CGF.ConstantFoldsToSimpleInteger(E->getLHS(), LHSCondVal)) {
3302 if (LHSCondVal) { // If we have 1 && X, just emit X.
Justin Bogner66242d62015-04-23 23:06:47 +00003303 CGF.incrementProfileCounter(E);
Justin Bogneref512b92014-01-06 22:27:43 +00003304
Chris Lattner5b1964b2008-11-11 07:41:27 +00003305 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
Chris Lattner671fec82009-10-17 04:24:20 +00003306 // ZExt result to int or bool.
3307 return Builder.CreateZExtOrBitCast(RHSCond, ResTy, "land.ext");
Chris Lattner5b1964b2008-11-11 07:41:27 +00003308 }
Mike Stump4a3999f2009-09-09 13:00:44 +00003309
Chris Lattner671fec82009-10-17 04:24:20 +00003310 // 0 && RHS: If it is safe, just elide the RHS, and return 0/false.
Chris Lattner8b084582008-11-12 08:26:50 +00003311 if (!CGF.ContainsLabel(E->getRHS()))
Chris Lattner671fec82009-10-17 04:24:20 +00003312 return llvm::Constant::getNullValue(ResTy);
Chris Lattner5b1964b2008-11-11 07:41:27 +00003313 }
Mike Stump4a3999f2009-09-09 13:00:44 +00003314
Daniel Dunbara612e792008-11-13 01:38:36 +00003315 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("land.end");
3316 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("land.rhs");
Chris Lattner8b084582008-11-12 08:26:50 +00003317
John McCallce1de612011-01-26 04:00:11 +00003318 CodeGenFunction::ConditionalEvaluation eval(CGF);
3319
Chris Lattner35710d182008-11-12 08:38:24 +00003320 // Branch on the LHS first. If it is false, go to the failure (cont) block.
Justin Bogner66242d62015-04-23 23:06:47 +00003321 CGF.EmitBranchOnBoolExpr(E->getLHS(), RHSBlock, ContBlock,
3322 CGF.getProfileCount(E->getRHS()));
Chris Lattner35710d182008-11-12 08:38:24 +00003323
3324 // Any edges into the ContBlock are now from an (indeterminate number of)
3325 // edges from this first condition. All of these values will be false. Start
3326 // setting up the PHI node in the Cont Block for this.
Jay Foad20c0f022011-03-30 11:28:58 +00003327 llvm::PHINode *PN = llvm::PHINode::Create(llvm::Type::getInt1Ty(VMContext), 2,
Owen Anderson41a75022009-08-13 21:57:51 +00003328 "", ContBlock);
Chris Lattner35710d182008-11-12 08:38:24 +00003329 for (llvm::pred_iterator PI = pred_begin(ContBlock), PE = pred_end(ContBlock);
3330 PI != PE; ++PI)
Owen Andersonfe4e3472009-07-31 17:39:36 +00003331 PN->addIncoming(llvm::ConstantInt::getFalse(VMContext), *PI);
Mike Stump4a3999f2009-09-09 13:00:44 +00003332
John McCallce1de612011-01-26 04:00:11 +00003333 eval.begin(CGF);
Chris Lattner2da04b32007-08-24 05:35:26 +00003334 CGF.EmitBlock(RHSBlock);
Justin Bogner66242d62015-04-23 23:06:47 +00003335 CGF.incrementProfileCounter(E);
Chris Lattner2da04b32007-08-24 05:35:26 +00003336 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
John McCallce1de612011-01-26 04:00:11 +00003337 eval.end(CGF);
Mike Stump4a3999f2009-09-09 13:00:44 +00003338
Chris Lattner2da04b32007-08-24 05:35:26 +00003339 // Reaquire the RHS block, as there may be subblocks inserted.
3340 RHSBlock = Builder.GetInsertBlock();
Chris Lattner35710d182008-11-12 08:38:24 +00003341
David Blaikie1b5adb82014-07-10 20:42:59 +00003342 // Emit an unconditional branch from this block to ContBlock.
3343 {
Devang Patel4d761272011-03-30 00:08:31 +00003344 // There is no need to emit line number for unconditional branch.
Adrian Prantl95b24e92015-02-03 20:00:54 +00003345 auto NL = ApplyDebugLocation::CreateEmpty(CGF);
David Blaikie1b5adb82014-07-10 20:42:59 +00003346 CGF.EmitBlock(ContBlock);
3347 }
3348 // Insert an entry into the phi node for the edge with the value of RHSCond.
Chris Lattner2da04b32007-08-24 05:35:26 +00003349 PN->addIncoming(RHSCond, RHSBlock);
Mike Stump4a3999f2009-09-09 13:00:44 +00003350
Chris Lattner2da04b32007-08-24 05:35:26 +00003351 // ZExt result to int.
Chris Lattner671fec82009-10-17 04:24:20 +00003352 return Builder.CreateZExtOrBitCast(PN, ResTy, "land.ext");
Chris Lattner2da04b32007-08-24 05:35:26 +00003353}
3354
3355Value *ScalarExprEmitter::VisitBinLOr(const BinaryOperator *E) {
Tanya Lattner20248222012-01-16 21:02:28 +00003356 // Perform vector logical or on comparisons with zero vectors.
3357 if (E->getType()->isVectorType()) {
Justin Bogner66242d62015-04-23 23:06:47 +00003358 CGF.incrementProfileCounter(E);
Justin Bogneref512b92014-01-06 22:27:43 +00003359
Tanya Lattner20248222012-01-16 21:02:28 +00003360 Value *LHS = Visit(E->getLHS());
3361 Value *RHS = Visit(E->getRHS());
3362 Value *Zero = llvm::ConstantAggregateZero::get(LHS->getType());
Joey Gouly7d00f002013-02-21 11:49:56 +00003363 if (LHS->getType()->isFPOrFPVectorTy()) {
3364 LHS = Builder.CreateFCmp(llvm::CmpInst::FCMP_UNE, LHS, Zero, "cmp");
3365 RHS = Builder.CreateFCmp(llvm::CmpInst::FCMP_UNE, RHS, Zero, "cmp");
3366 } else {
3367 LHS = Builder.CreateICmp(llvm::CmpInst::ICMP_NE, LHS, Zero, "cmp");
3368 RHS = Builder.CreateICmp(llvm::CmpInst::ICMP_NE, RHS, Zero, "cmp");
3369 }
Tanya Lattner20248222012-01-16 21:02:28 +00003370 Value *Or = Builder.CreateOr(LHS, RHS);
Joey Gouly7d00f002013-02-21 11:49:56 +00003371 return Builder.CreateSExt(Or, ConvertType(E->getType()), "sext");
Tanya Lattner20248222012-01-16 21:02:28 +00003372 }
Craig Toppera97d7e72013-07-26 06:16:11 +00003373
Chris Lattner2192fe52011-07-18 04:24:23 +00003374 llvm::Type *ResTy = ConvertType(E->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00003375
Chris Lattner8b084582008-11-12 08:26:50 +00003376 // If we have 1 || RHS, see if we can elide RHS, if so, just return 1.
3377 // If we have 0 || X, just emit X without inserting the control flow.
Chris Lattner41c6ab52011-02-27 23:02:32 +00003378 bool LHSCondVal;
3379 if (CGF.ConstantFoldsToSimpleInteger(E->getLHS(), LHSCondVal)) {
3380 if (!LHSCondVal) { // If we have 0 || X, just emit X.
Justin Bogner66242d62015-04-23 23:06:47 +00003381 CGF.incrementProfileCounter(E);
Justin Bogneref512b92014-01-06 22:27:43 +00003382
Chris Lattner5b1964b2008-11-11 07:41:27 +00003383 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
Chris Lattner671fec82009-10-17 04:24:20 +00003384 // ZExt result to int or bool.
3385 return Builder.CreateZExtOrBitCast(RHSCond, ResTy, "lor.ext");
Chris Lattner5b1964b2008-11-11 07:41:27 +00003386 }
Mike Stump4a3999f2009-09-09 13:00:44 +00003387
Chris Lattner671fec82009-10-17 04:24:20 +00003388 // 1 || RHS: If it is safe, just elide the RHS, and return 1/true.
Chris Lattner8b084582008-11-12 08:26:50 +00003389 if (!CGF.ContainsLabel(E->getRHS()))
Chris Lattner671fec82009-10-17 04:24:20 +00003390 return llvm::ConstantInt::get(ResTy, 1);
Chris Lattner5b1964b2008-11-11 07:41:27 +00003391 }
Mike Stump4a3999f2009-09-09 13:00:44 +00003392
Daniel Dunbara612e792008-11-13 01:38:36 +00003393 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("lor.end");
3394 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("lor.rhs");
Mike Stump4a3999f2009-09-09 13:00:44 +00003395
John McCallce1de612011-01-26 04:00:11 +00003396 CodeGenFunction::ConditionalEvaluation eval(CGF);
3397
Chris Lattner35710d182008-11-12 08:38:24 +00003398 // Branch on the LHS first. If it is true, go to the success (cont) block.
Justin Bogneref512b92014-01-06 22:27:43 +00003399 CGF.EmitBranchOnBoolExpr(E->getLHS(), ContBlock, RHSBlock,
Justin Bogner66242d62015-04-23 23:06:47 +00003400 CGF.getCurrentProfileCount() -
3401 CGF.getProfileCount(E->getRHS()));
Chris Lattner35710d182008-11-12 08:38:24 +00003402
3403 // Any edges into the ContBlock are now from an (indeterminate number of)
3404 // edges from this first condition. All of these values will be true. Start
3405 // setting up the PHI node in the Cont Block for this.
Jay Foad20c0f022011-03-30 11:28:58 +00003406 llvm::PHINode *PN = llvm::PHINode::Create(llvm::Type::getInt1Ty(VMContext), 2,
Owen Anderson41a75022009-08-13 21:57:51 +00003407 "", ContBlock);
Chris Lattner35710d182008-11-12 08:38:24 +00003408 for (llvm::pred_iterator PI = pred_begin(ContBlock), PE = pred_end(ContBlock);
3409 PI != PE; ++PI)
Owen Andersonfe4e3472009-07-31 17:39:36 +00003410 PN->addIncoming(llvm::ConstantInt::getTrue(VMContext), *PI);
Chris Lattner35710d182008-11-12 08:38:24 +00003411
John McCallce1de612011-01-26 04:00:11 +00003412 eval.begin(CGF);
Anders Carlssonf47a3de2009-06-04 02:53:13 +00003413
Chris Lattner35710d182008-11-12 08:38:24 +00003414 // Emit the RHS condition as a bool value.
Chris Lattner2da04b32007-08-24 05:35:26 +00003415 CGF.EmitBlock(RHSBlock);
Justin Bogner66242d62015-04-23 23:06:47 +00003416 CGF.incrementProfileCounter(E);
Chris Lattner2da04b32007-08-24 05:35:26 +00003417 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
Mike Stump4a3999f2009-09-09 13:00:44 +00003418
John McCallce1de612011-01-26 04:00:11 +00003419 eval.end(CGF);
Mike Stump4a3999f2009-09-09 13:00:44 +00003420
Chris Lattner2da04b32007-08-24 05:35:26 +00003421 // Reaquire the RHS block, as there may be subblocks inserted.
3422 RHSBlock = Builder.GetInsertBlock();
Mike Stump4a3999f2009-09-09 13:00:44 +00003423
Chris Lattner35710d182008-11-12 08:38:24 +00003424 // Emit an unconditional branch from this block to ContBlock. Insert an entry
3425 // into the phi node for the edge with the value of RHSCond.
3426 CGF.EmitBlock(ContBlock);
Chris Lattner2da04b32007-08-24 05:35:26 +00003427 PN->addIncoming(RHSCond, RHSBlock);
Mike Stump4a3999f2009-09-09 13:00:44 +00003428
Chris Lattner2da04b32007-08-24 05:35:26 +00003429 // ZExt result to int.
Chris Lattner671fec82009-10-17 04:24:20 +00003430 return Builder.CreateZExtOrBitCast(PN, ResTy, "lor.ext");
Chris Lattner2da04b32007-08-24 05:35:26 +00003431}
3432
3433Value *ScalarExprEmitter::VisitBinComma(const BinaryOperator *E) {
John McCalla2342eb2010-12-05 02:00:02 +00003434 CGF.EmitIgnoredExpr(E->getLHS());
Daniel Dunbar5c7e3932008-11-11 23:11:34 +00003435 CGF.EnsureInsertPoint();
Chris Lattner2da04b32007-08-24 05:35:26 +00003436 return Visit(E->getRHS());
3437}
3438
3439//===----------------------------------------------------------------------===//
3440// Other Operators
3441//===----------------------------------------------------------------------===//
3442
Chris Lattner3fd91f832008-11-12 08:55:54 +00003443/// isCheapEnoughToEvaluateUnconditionally - Return true if the specified
3444/// expression is cheap enough and side-effect-free enough to evaluate
3445/// unconditionally instead of conditionally. This is used to convert control
3446/// flow into selects in some cases.
Mike Stump53f9ded2009-11-03 23:25:48 +00003447static bool isCheapEnoughToEvaluateUnconditionally(const Expr *E,
3448 CodeGenFunction &CGF) {
Chris Lattner56784f92011-04-16 23:15:35 +00003449 // Anything that is an integer or floating point constant is fine.
Nick Lewycky22e55a02013-11-08 23:00:12 +00003450 return E->IgnoreParens()->isEvaluatable(CGF.getContext());
Mike Stump4a3999f2009-09-09 13:00:44 +00003451
Nick Lewycky22e55a02013-11-08 23:00:12 +00003452 // Even non-volatile automatic variables can't be evaluated unconditionally.
3453 // Referencing a thread_local may cause non-trivial initialization work to
3454 // occur. If we're inside a lambda and one of the variables is from the scope
3455 // outside the lambda, that function may have returned already. Reading its
3456 // locals is a bad idea. Also, these reads may introduce races there didn't
3457 // exist in the source-level program.
Chris Lattner3fd91f832008-11-12 08:55:54 +00003458}
3459
3460
Chris Lattner2da04b32007-08-24 05:35:26 +00003461Value *ScalarExprEmitter::
John McCallc07a0c72011-02-17 10:25:35 +00003462VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00003463 TestAndClearIgnoreResultAssign();
John McCallc07a0c72011-02-17 10:25:35 +00003464
3465 // Bind the common expression if necessary.
Eli Friedman48fd89a2012-01-06 20:42:20 +00003466 CodeGenFunction::OpaqueValueMapping binding(CGF, E);
John McCallc07a0c72011-02-17 10:25:35 +00003467
3468 Expr *condExpr = E->getCond();
3469 Expr *lhsExpr = E->getTrueExpr();
3470 Expr *rhsExpr = E->getFalseExpr();
3471
Chris Lattnercd439292008-11-12 08:04:58 +00003472 // If the condition constant folds and can be elided, try to avoid emitting
3473 // the condition and the dead arm.
Chris Lattner41c6ab52011-02-27 23:02:32 +00003474 bool CondExprBool;
3475 if (CGF.ConstantFoldsToSimpleInteger(condExpr, CondExprBool)) {
John McCallc07a0c72011-02-17 10:25:35 +00003476 Expr *live = lhsExpr, *dead = rhsExpr;
Chris Lattner41c6ab52011-02-27 23:02:32 +00003477 if (!CondExprBool) std::swap(live, dead);
Mike Stump4a3999f2009-09-09 13:00:44 +00003478
Eli Friedman27ef75b2011-10-15 02:10:40 +00003479 // If the dead side doesn't have labels we need, just emit the Live part.
3480 if (!CGF.ContainsLabel(dead)) {
Justin Bogneref512b92014-01-06 22:27:43 +00003481 if (CondExprBool)
Justin Bogner66242d62015-04-23 23:06:47 +00003482 CGF.incrementProfileCounter(E);
Eli Friedman27ef75b2011-10-15 02:10:40 +00003483 Value *Result = Visit(live);
3484
3485 // If the live part is a throw expression, it acts like it has a void
3486 // type, so evaluating it returns a null Value*. However, a conditional
3487 // with non-void type must return a non-null Value*.
3488 if (!Result && !E->getType()->isVoidType())
3489 Result = llvm::UndefValue::get(CGF.ConvertType(E->getType()));
3490
3491 return Result;
3492 }
Chris Lattnerd53e2332008-11-11 18:56:45 +00003493 }
Mike Stump4a3999f2009-09-09 13:00:44 +00003494
Nate Begemanabb5a732010-09-20 22:41:17 +00003495 // OpenCL: If the condition is a vector, we can treat this condition like
3496 // the select function.
Craig Toppera97d7e72013-07-26 06:16:11 +00003497 if (CGF.getLangOpts().OpenCL
John McCallc07a0c72011-02-17 10:25:35 +00003498 && condExpr->getType()->isVectorType()) {
Justin Bogner66242d62015-04-23 23:06:47 +00003499 CGF.incrementProfileCounter(E);
Justin Bogneref512b92014-01-06 22:27:43 +00003500
John McCallc07a0c72011-02-17 10:25:35 +00003501 llvm::Value *CondV = CGF.EmitScalarExpr(condExpr);
3502 llvm::Value *LHS = Visit(lhsExpr);
3503 llvm::Value *RHS = Visit(rhsExpr);
Craig Toppera97d7e72013-07-26 06:16:11 +00003504
Chris Lattner2192fe52011-07-18 04:24:23 +00003505 llvm::Type *condType = ConvertType(condExpr->getType());
3506 llvm::VectorType *vecTy = cast<llvm::VectorType>(condType);
Craig Toppera97d7e72013-07-26 06:16:11 +00003507
3508 unsigned numElem = vecTy->getNumElements();
Chris Lattner2192fe52011-07-18 04:24:23 +00003509 llvm::Type *elemType = vecTy->getElementType();
Craig Toppera97d7e72013-07-26 06:16:11 +00003510
Chris Lattner2d6b7b92012-01-25 05:34:41 +00003511 llvm::Value *zeroVec = llvm::Constant::getNullValue(vecTy);
Nate Begemanabb5a732010-09-20 22:41:17 +00003512 llvm::Value *TestMSB = Builder.CreateICmpSLT(CondV, zeroVec);
Craig Toppera97d7e72013-07-26 06:16:11 +00003513 llvm::Value *tmp = Builder.CreateSExt(TestMSB,
Nate Begemanabb5a732010-09-20 22:41:17 +00003514 llvm::VectorType::get(elemType,
Craig Toppera97d7e72013-07-26 06:16:11 +00003515 numElem),
Nate Begemanabb5a732010-09-20 22:41:17 +00003516 "sext");
3517 llvm::Value *tmp2 = Builder.CreateNot(tmp);
Craig Toppera97d7e72013-07-26 06:16:11 +00003518
Nate Begemanabb5a732010-09-20 22:41:17 +00003519 // Cast float to int to perform ANDs if necessary.
3520 llvm::Value *RHSTmp = RHS;
3521 llvm::Value *LHSTmp = LHS;
3522 bool wasCast = false;
Chris Lattner2192fe52011-07-18 04:24:23 +00003523 llvm::VectorType *rhsVTy = cast<llvm::VectorType>(RHS->getType());
Peter Collingbourneaac265c2012-05-29 00:35:18 +00003524 if (rhsVTy->getElementType()->isFloatingPointTy()) {
Nate Begemanabb5a732010-09-20 22:41:17 +00003525 RHSTmp = Builder.CreateBitCast(RHS, tmp2->getType());
3526 LHSTmp = Builder.CreateBitCast(LHS, tmp->getType());
3527 wasCast = true;
3528 }
Craig Toppera97d7e72013-07-26 06:16:11 +00003529
Nate Begemanabb5a732010-09-20 22:41:17 +00003530 llvm::Value *tmp3 = Builder.CreateAnd(RHSTmp, tmp2);
3531 llvm::Value *tmp4 = Builder.CreateAnd(LHSTmp, tmp);
3532 llvm::Value *tmp5 = Builder.CreateOr(tmp3, tmp4, "cond");
3533 if (wasCast)
3534 tmp5 = Builder.CreateBitCast(tmp5, RHS->getType());
3535
3536 return tmp5;
3537 }
Craig Toppera97d7e72013-07-26 06:16:11 +00003538
Chris Lattner3fd91f832008-11-12 08:55:54 +00003539 // If this is a really simple expression (like x ? 4 : 5), emit this as a
3540 // select instead of as control flow. We can only do this if it is cheap and
Chris Lattner9ce8a532008-11-16 06:16:27 +00003541 // safe to evaluate the LHS and RHS unconditionally.
John McCallc07a0c72011-02-17 10:25:35 +00003542 if (isCheapEnoughToEvaluateUnconditionally(lhsExpr, CGF) &&
3543 isCheapEnoughToEvaluateUnconditionally(rhsExpr, CGF)) {
3544 llvm::Value *CondV = CGF.EvaluateExprAsBool(condExpr);
Vedant Kumar502bbfa2017-02-25 06:35:45 +00003545 llvm::Value *StepV = Builder.CreateZExtOrBitCast(CondV, CGF.Int64Ty);
3546
3547 CGF.incrementProfileCounter(E, StepV);
3548
John McCallc07a0c72011-02-17 10:25:35 +00003549 llvm::Value *LHS = Visit(lhsExpr);
3550 llvm::Value *RHS = Visit(rhsExpr);
Eli Friedman516c2ad2011-12-08 22:01:56 +00003551 if (!LHS) {
3552 // If the conditional has void type, make sure we return a null Value*.
3553 assert(!RHS && "LHS and RHS types must match");
Craig Topper8a13c412014-05-21 05:09:00 +00003554 return nullptr;
Eli Friedman516c2ad2011-12-08 22:01:56 +00003555 }
Chris Lattner3fd91f832008-11-12 08:55:54 +00003556 return Builder.CreateSelect(CondV, LHS, RHS, "cond");
3557 }
Mike Stump4a3999f2009-09-09 13:00:44 +00003558
Daniel Dunbard2a53a72008-11-12 10:13:37 +00003559 llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true");
3560 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false");
Daniel Dunbara612e792008-11-13 01:38:36 +00003561 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end");
John McCallce1de612011-01-26 04:00:11 +00003562
3563 CodeGenFunction::ConditionalEvaluation eval(CGF);
Justin Bogner66242d62015-04-23 23:06:47 +00003564 CGF.EmitBranchOnBoolExpr(condExpr, LHSBlock, RHSBlock,
3565 CGF.getProfileCount(lhsExpr));
Anders Carlsson43c52cd2009-06-04 03:00:32 +00003566
Chris Lattner2da04b32007-08-24 05:35:26 +00003567 CGF.EmitBlock(LHSBlock);
Justin Bogner66242d62015-04-23 23:06:47 +00003568 CGF.incrementProfileCounter(E);
John McCallce1de612011-01-26 04:00:11 +00003569 eval.begin(CGF);
John McCallc07a0c72011-02-17 10:25:35 +00003570 Value *LHS = Visit(lhsExpr);
John McCallce1de612011-01-26 04:00:11 +00003571 eval.end(CGF);
Mike Stump4a3999f2009-09-09 13:00:44 +00003572
Chris Lattner2da04b32007-08-24 05:35:26 +00003573 LHSBlock = Builder.GetInsertBlock();
John McCallce1de612011-01-26 04:00:11 +00003574 Builder.CreateBr(ContBlock);
Mike Stump4a3999f2009-09-09 13:00:44 +00003575
Chris Lattner2da04b32007-08-24 05:35:26 +00003576 CGF.EmitBlock(RHSBlock);
John McCallce1de612011-01-26 04:00:11 +00003577 eval.begin(CGF);
John McCallc07a0c72011-02-17 10:25:35 +00003578 Value *RHS = Visit(rhsExpr);
John McCallce1de612011-01-26 04:00:11 +00003579 eval.end(CGF);
Mike Stump4a3999f2009-09-09 13:00:44 +00003580
John McCallce1de612011-01-26 04:00:11 +00003581 RHSBlock = Builder.GetInsertBlock();
Chris Lattner2da04b32007-08-24 05:35:26 +00003582 CGF.EmitBlock(ContBlock);
Mike Stump4a3999f2009-09-09 13:00:44 +00003583
Eli Friedmanf6c175b2009-12-07 20:25:53 +00003584 // If the LHS or RHS is a throw expression, it will be legitimately null.
3585 if (!LHS)
3586 return RHS;
3587 if (!RHS)
3588 return LHS;
Mike Stump4a3999f2009-09-09 13:00:44 +00003589
Chris Lattner2da04b32007-08-24 05:35:26 +00003590 // Create a PHI node for the real part.
Jay Foad20c0f022011-03-30 11:28:58 +00003591 llvm::PHINode *PN = Builder.CreatePHI(LHS->getType(), 2, "cond");
Chris Lattner2da04b32007-08-24 05:35:26 +00003592 PN->addIncoming(LHS, LHSBlock);
3593 PN->addIncoming(RHS, RHSBlock);
3594 return PN;
3595}
3596
3597Value *ScalarExprEmitter::VisitChooseExpr(ChooseExpr *E) {
Eli Friedman75807f22013-07-20 00:40:58 +00003598 return Visit(E->getChosenSubExpr());
Chris Lattner2da04b32007-08-24 05:35:26 +00003599}
3600
Chris Lattnerb6a7b582007-11-30 17:56:23 +00003601Value *ScalarExprEmitter::VisitVAArgExpr(VAArgExpr *VE) {
Richard Smitha1a808c2014-04-14 23:47:48 +00003602 QualType Ty = VE->getType();
Daniel Sanders59229dc2014-11-19 10:01:35 +00003603
Richard Smitha1a808c2014-04-14 23:47:48 +00003604 if (Ty->isVariablyModifiedType())
3605 CGF.EmitVariablyModifiedType(Ty);
3606
Charles Davisc7d5c942015-09-17 20:55:33 +00003607 Address ArgValue = Address::invalid();
3608 Address ArgPtr = CGF.EmitVAArg(VE, ArgValue);
3609
Daniel Sanders59229dc2014-11-19 10:01:35 +00003610 llvm::Type *ArgTy = ConvertType(VE->getType());
Anders Carlsson13abd7e2008-11-04 05:30:00 +00003611
James Y Knight29b5f082016-02-24 02:59:33 +00003612 // If EmitVAArg fails, emit an error.
3613 if (!ArgPtr.isValid()) {
3614 CGF.ErrorUnsupported(VE, "va_arg expression");
3615 return llvm::UndefValue::get(ArgTy);
3616 }
Anders Carlsson13abd7e2008-11-04 05:30:00 +00003617
Mike Stumpdf0fe272009-05-29 15:46:01 +00003618 // FIXME Volatility.
Daniel Sanders59229dc2014-11-19 10:01:35 +00003619 llvm::Value *Val = Builder.CreateLoad(ArgPtr);
3620
3621 // If EmitVAArg promoted the type, we must truncate it.
Daniel Sanderscdcb5802015-01-13 10:47:00 +00003622 if (ArgTy != Val->getType()) {
3623 if (ArgTy->isPointerTy() && !Val->getType()->isPointerTy())
3624 Val = Builder.CreateIntToPtr(Val, ArgTy);
3625 else
3626 Val = Builder.CreateTrunc(Val, ArgTy);
3627 }
Daniel Sanders59229dc2014-11-19 10:01:35 +00003628
3629 return Val;
Anders Carlsson7e13ab82007-10-15 20:28:48 +00003630}
3631
John McCall351762c2011-02-07 10:33:21 +00003632Value *ScalarExprEmitter::VisitBlockExpr(const BlockExpr *block) {
3633 return CGF.EmitBlockLiteral(block);
Mike Stumpab3afd82009-02-12 18:29:15 +00003634}
3635
Yaxun Liuc5647012016-06-08 15:11:21 +00003636// Convert a vec3 to vec4, or vice versa.
3637static Value *ConvertVec3AndVec4(CGBuilderTy &Builder, CodeGenFunction &CGF,
3638 Value *Src, unsigned NumElementsDst) {
3639 llvm::Value *UnV = llvm::UndefValue::get(Src->getType());
3640 SmallVector<llvm::Constant*, 4> Args;
3641 Args.push_back(Builder.getInt32(0));
3642 Args.push_back(Builder.getInt32(1));
3643 Args.push_back(Builder.getInt32(2));
3644 if (NumElementsDst == 4)
3645 Args.push_back(llvm::UndefValue::get(CGF.Int32Ty));
3646 llvm::Constant *Mask = llvm::ConstantVector::get(Args);
3647 return Builder.CreateShuffleVector(Src, UnV, Mask);
3648}
3649
Yaxun Liuea6b7962016-10-03 14:41:50 +00003650// Create cast instructions for converting LLVM value \p Src to LLVM type \p
3651// DstTy. \p Src has the same size as \p DstTy. Both are single value types
3652// but could be scalar or vectors of different lengths, and either can be
3653// pointer.
3654// There are 4 cases:
3655// 1. non-pointer -> non-pointer : needs 1 bitcast
3656// 2. pointer -> pointer : needs 1 bitcast or addrspacecast
3657// 3. pointer -> non-pointer
3658// a) pointer -> intptr_t : needs 1 ptrtoint
3659// b) pointer -> non-intptr_t : needs 1 ptrtoint then 1 bitcast
3660// 4. non-pointer -> pointer
3661// a) intptr_t -> pointer : needs 1 inttoptr
3662// b) non-intptr_t -> pointer : needs 1 bitcast then 1 inttoptr
3663// Note: for cases 3b and 4b two casts are required since LLVM casts do not
3664// allow casting directly between pointer types and non-integer non-pointer
3665// types.
3666static Value *createCastsForTypeOfSameSize(CGBuilderTy &Builder,
3667 const llvm::DataLayout &DL,
3668 Value *Src, llvm::Type *DstTy,
3669 StringRef Name = "") {
3670 auto SrcTy = Src->getType();
3671
3672 // Case 1.
3673 if (!SrcTy->isPointerTy() && !DstTy->isPointerTy())
3674 return Builder.CreateBitCast(Src, DstTy, Name);
3675
3676 // Case 2.
3677 if (SrcTy->isPointerTy() && DstTy->isPointerTy())
3678 return Builder.CreatePointerBitCastOrAddrSpaceCast(Src, DstTy, Name);
3679
3680 // Case 3.
3681 if (SrcTy->isPointerTy() && !DstTy->isPointerTy()) {
3682 // Case 3b.
3683 if (!DstTy->isIntegerTy())
3684 Src = Builder.CreatePtrToInt(Src, DL.getIntPtrType(SrcTy));
3685 // Cases 3a and 3b.
3686 return Builder.CreateBitOrPointerCast(Src, DstTy, Name);
3687 }
3688
3689 // Case 4b.
3690 if (!SrcTy->isIntegerTy())
3691 Src = Builder.CreateBitCast(Src, DL.getIntPtrType(DstTy));
3692 // Cases 4a and 4b.
3693 return Builder.CreateIntToPtr(Src, DstTy, Name);
3694}
3695
Tanya Lattner55808c12011-06-04 00:47:47 +00003696Value *ScalarExprEmitter::VisitAsTypeExpr(AsTypeExpr *E) {
3697 Value *Src = CGF.EmitScalarExpr(E->getSrcExpr());
Chris Lattner2192fe52011-07-18 04:24:23 +00003698 llvm::Type *DstTy = ConvertType(E->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00003699
Chris Lattner2192fe52011-07-18 04:24:23 +00003700 llvm::Type *SrcTy = Src->getType();
Yaxun Liuc5647012016-06-08 15:11:21 +00003701 unsigned NumElementsSrc = isa<llvm::VectorType>(SrcTy) ?
3702 cast<llvm::VectorType>(SrcTy)->getNumElements() : 0;
3703 unsigned NumElementsDst = isa<llvm::VectorType>(DstTy) ?
3704 cast<llvm::VectorType>(DstTy)->getNumElements() : 0;
Craig Toppera97d7e72013-07-26 06:16:11 +00003705
Yaxun Liuc5647012016-06-08 15:11:21 +00003706 // Going from vec3 to non-vec3 is a special case and requires a shuffle
3707 // vector to get a vec4, then a bitcast if the target type is different.
3708 if (NumElementsSrc == 3 && NumElementsDst != 3) {
3709 Src = ConvertVec3AndVec4(Builder, CGF, Src, 4);
Jin-Gu Kange7cdcde2017-04-04 16:40:25 +00003710
3711 if (!CGF.CGM.getCodeGenOpts().PreserveVec3Type) {
3712 Src = createCastsForTypeOfSameSize(Builder, CGF.CGM.getDataLayout(), Src,
3713 DstTy);
3714 }
3715
Yaxun Liuc5647012016-06-08 15:11:21 +00003716 Src->setName("astype");
3717 return Src;
3718 }
Craig Toppera97d7e72013-07-26 06:16:11 +00003719
Yaxun Liuc5647012016-06-08 15:11:21 +00003720 // Going from non-vec3 to vec3 is a special case and requires a bitcast
3721 // to vec4 if the original type is not vec4, then a shuffle vector to
3722 // get a vec3.
3723 if (NumElementsSrc != 3 && NumElementsDst == 3) {
Jin-Gu Kange7cdcde2017-04-04 16:40:25 +00003724 if (!CGF.CGM.getCodeGenOpts().PreserveVec3Type) {
3725 auto Vec4Ty = llvm::VectorType::get(DstTy->getVectorElementType(), 4);
3726 Src = createCastsForTypeOfSameSize(Builder, CGF.CGM.getDataLayout(), Src,
3727 Vec4Ty);
3728 }
3729
Yaxun Liuc5647012016-06-08 15:11:21 +00003730 Src = ConvertVec3AndVec4(Builder, CGF, Src, 3);
3731 Src->setName("astype");
3732 return Src;
Tanya Lattner55808c12011-06-04 00:47:47 +00003733 }
Craig Toppera97d7e72013-07-26 06:16:11 +00003734
Yaxun Liuea6b7962016-10-03 14:41:50 +00003735 return Src = createCastsForTypeOfSameSize(Builder, CGF.CGM.getDataLayout(),
3736 Src, DstTy, "astype");
Tanya Lattner55808c12011-06-04 00:47:47 +00003737}
3738
Eli Friedmandf14b3a2011-10-11 02:20:01 +00003739Value *ScalarExprEmitter::VisitAtomicExpr(AtomicExpr *E) {
3740 return CGF.EmitAtomicExpr(E).getScalarVal();
3741}
3742
Chris Lattner2da04b32007-08-24 05:35:26 +00003743//===----------------------------------------------------------------------===//
3744// Entry Point into this File
3745//===----------------------------------------------------------------------===//
3746
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +00003747/// Emit the computation of the specified expression of scalar type, ignoring
3748/// the result.
Mike Stumpdf0fe272009-05-29 15:46:01 +00003749Value *CodeGenFunction::EmitScalarExpr(const Expr *E, bool IgnoreResultAssign) {
John McCall47fb9502013-03-07 21:37:08 +00003750 assert(E && hasScalarEvaluationKind(E->getType()) &&
Chris Lattner2da04b32007-08-24 05:35:26 +00003751 "Invalid scalar expression to emit");
Mike Stump4a3999f2009-09-09 13:00:44 +00003752
David Blaikie38b25912015-02-09 19:13:51 +00003753 return ScalarExprEmitter(*this, IgnoreResultAssign)
3754 .Visit(const_cast<Expr *>(E));
Chris Lattner2da04b32007-08-24 05:35:26 +00003755}
Chris Lattner3474c202007-08-26 06:48:56 +00003756
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +00003757/// Emit a conversion from the specified type to the specified destination type,
3758/// both of which are LLVM scalar types.
Chris Lattner42e6b812007-08-26 16:34:22 +00003759Value *CodeGenFunction::EmitScalarConversion(Value *Src, QualType SrcTy,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00003760 QualType DstTy,
3761 SourceLocation Loc) {
John McCall47fb9502013-03-07 21:37:08 +00003762 assert(hasScalarEvaluationKind(SrcTy) && hasScalarEvaluationKind(DstTy) &&
Chris Lattner3474c202007-08-26 06:48:56 +00003763 "Invalid scalar expression to emit");
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00003764 return ScalarExprEmitter(*this).EmitScalarConversion(Src, SrcTy, DstTy, Loc);
Chris Lattner3474c202007-08-26 06:48:56 +00003765}
Chris Lattner42e6b812007-08-26 16:34:22 +00003766
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +00003767/// Emit a conversion from the specified complex type to the specified
3768/// destination type, where the destination type is an LLVM scalar type.
Chris Lattner42e6b812007-08-26 16:34:22 +00003769Value *CodeGenFunction::EmitComplexToScalarConversion(ComplexPairTy Src,
3770 QualType SrcTy,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00003771 QualType DstTy,
3772 SourceLocation Loc) {
John McCall47fb9502013-03-07 21:37:08 +00003773 assert(SrcTy->isAnyComplexType() && hasScalarEvaluationKind(DstTy) &&
Chris Lattner42e6b812007-08-26 16:34:22 +00003774 "Invalid complex -> scalar conversion");
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00003775 return ScalarExprEmitter(*this)
3776 .EmitComplexToScalarConversion(Src, SrcTy, DstTy, Loc);
Chris Lattner42e6b812007-08-26 16:34:22 +00003777}
Anders Carlssonb9eb82c2007-12-10 19:35:18 +00003778
Chris Lattner05dc78c2010-06-26 22:09:34 +00003779
3780llvm::Value *CodeGenFunction::
3781EmitScalarPrePostIncDec(const UnaryOperator *E, LValue LV,
3782 bool isInc, bool isPre) {
3783 return ScalarExprEmitter(*this).EmitScalarPrePostIncDec(E, LV, isInc, isPre);
3784}
3785
Fariborz Jahanian531c16f2009-12-09 23:35:29 +00003786LValue CodeGenFunction::EmitObjCIsaExpr(const ObjCIsaExpr *E) {
Fariborz Jahanian531c16f2009-12-09 23:35:29 +00003787 // object->isa or (*object).isa
3788 // Generate code as for: *(Class*)object
Fariborz Jahaniandf506b92010-02-05 19:18:30 +00003789
3790 Expr *BaseExpr = E->getBase();
John McCall7f416cc2015-09-08 08:05:57 +00003791 Address Addr = Address::invalid();
John McCall086a4642010-11-24 05:12:34 +00003792 if (BaseExpr->isRValue()) {
John McCall7f416cc2015-09-08 08:05:57 +00003793 Addr = Address(EmitScalarExpr(BaseExpr), getPointerAlign());
Daniel Dunbarf6fb7e22010-08-21 03:08:16 +00003794 } else {
John McCall7f416cc2015-09-08 08:05:57 +00003795 Addr = EmitLValue(BaseExpr).getAddress();
Fariborz Jahaniandf506b92010-02-05 19:18:30 +00003796 }
Craig Toppera97d7e72013-07-26 06:16:11 +00003797
John McCall7f416cc2015-09-08 08:05:57 +00003798 // Cast the address to Class*.
3799 Addr = Builder.CreateElementBitCast(Addr, ConvertType(E->getType()));
3800 return MakeAddrLValue(Addr, E->getType());
Fariborz Jahanian531c16f2009-12-09 23:35:29 +00003801}
3802
Douglas Gregor914af212010-04-23 04:16:32 +00003803
John McCalla2342eb2010-12-05 02:00:02 +00003804LValue CodeGenFunction::EmitCompoundAssignmentLValue(
Douglas Gregor914af212010-04-23 04:16:32 +00003805 const CompoundAssignOperator *E) {
3806 ScalarExprEmitter Scalar(*this);
Craig Topper8a13c412014-05-21 05:09:00 +00003807 Value *Result = nullptr;
Douglas Gregor914af212010-04-23 04:16:32 +00003808 switch (E->getOpcode()) {
3809#define COMPOUND_OP(Op) \
John McCalle3027922010-08-25 11:45:40 +00003810 case BO_##Op##Assign: \
Douglas Gregor914af212010-04-23 04:16:32 +00003811 return Scalar.EmitCompoundAssignLValue(E, &ScalarExprEmitter::Emit##Op, \
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00003812 Result)
Douglas Gregor914af212010-04-23 04:16:32 +00003813 COMPOUND_OP(Mul);
3814 COMPOUND_OP(Div);
3815 COMPOUND_OP(Rem);
3816 COMPOUND_OP(Add);
3817 COMPOUND_OP(Sub);
3818 COMPOUND_OP(Shl);
3819 COMPOUND_OP(Shr);
3820 COMPOUND_OP(And);
3821 COMPOUND_OP(Xor);
3822 COMPOUND_OP(Or);
3823#undef COMPOUND_OP
Craig Toppera97d7e72013-07-26 06:16:11 +00003824
John McCalle3027922010-08-25 11:45:40 +00003825 case BO_PtrMemD:
3826 case BO_PtrMemI:
3827 case BO_Mul:
3828 case BO_Div:
3829 case BO_Rem:
3830 case BO_Add:
3831 case BO_Sub:
3832 case BO_Shl:
3833 case BO_Shr:
3834 case BO_LT:
3835 case BO_GT:
3836 case BO_LE:
3837 case BO_GE:
3838 case BO_EQ:
3839 case BO_NE:
3840 case BO_And:
3841 case BO_Xor:
3842 case BO_Or:
3843 case BO_LAnd:
3844 case BO_LOr:
3845 case BO_Assign:
3846 case BO_Comma:
David Blaikie83d382b2011-09-23 05:06:16 +00003847 llvm_unreachable("Not valid compound assignment operators");
Douglas Gregor914af212010-04-23 04:16:32 +00003848 }
Craig Toppera97d7e72013-07-26 06:16:11 +00003849
Douglas Gregor914af212010-04-23 04:16:32 +00003850 llvm_unreachable("Unhandled compound assignment operator");
3851}
Vedant Kumara125eb52017-06-01 19:22:18 +00003852
3853Value *CodeGenFunction::EmitCheckedInBoundsGEP(Value *Ptr,
3854 ArrayRef<Value *> IdxList,
Vedant Kumar6dbf4272017-06-12 18:42:51 +00003855 bool SignedIndices,
Vedant Kumara125eb52017-06-01 19:22:18 +00003856 SourceLocation Loc,
3857 const Twine &Name) {
3858 Value *GEPVal = Builder.CreateInBoundsGEP(Ptr, IdxList, Name);
3859
3860 // If the pointer overflow sanitizer isn't enabled, do nothing.
3861 if (!SanOpts.has(SanitizerKind::PointerOverflow))
3862 return GEPVal;
3863
3864 // If the GEP has already been reduced to a constant, leave it be.
3865 if (isa<llvm::Constant>(GEPVal))
3866 return GEPVal;
3867
3868 // Only check for overflows in the default address space.
3869 if (GEPVal->getType()->getPointerAddressSpace())
3870 return GEPVal;
3871
3872 auto *GEP = cast<llvm::GEPOperator>(GEPVal);
3873 assert(GEP->isInBounds() && "Expected inbounds GEP");
3874
3875 SanitizerScope SanScope(this);
3876 auto &VMContext = getLLVMContext();
3877 const auto &DL = CGM.getDataLayout();
3878 auto *IntPtrTy = DL.getIntPtrType(GEP->getPointerOperandType());
3879
3880 // Grab references to the signed add/mul overflow intrinsics for intptr_t.
3881 auto *Zero = llvm::ConstantInt::getNullValue(IntPtrTy);
3882 auto *SAddIntrinsic =
3883 CGM.getIntrinsic(llvm::Intrinsic::sadd_with_overflow, IntPtrTy);
3884 auto *SMulIntrinsic =
3885 CGM.getIntrinsic(llvm::Intrinsic::smul_with_overflow, IntPtrTy);
3886
3887 // The total (signed) byte offset for the GEP.
3888 llvm::Value *TotalOffset = nullptr;
3889 // The offset overflow flag - true if the total offset overflows.
3890 llvm::Value *OffsetOverflows = Builder.getFalse();
3891
3892 /// Return the result of the given binary operation.
3893 auto eval = [&](BinaryOperator::Opcode Opcode, llvm::Value *LHS,
3894 llvm::Value *RHS) -> llvm::Value * {
Davide Italiano77378e42017-06-01 23:55:18 +00003895 assert((Opcode == BO_Add || Opcode == BO_Mul) && "Can't eval binop");
Vedant Kumara125eb52017-06-01 19:22:18 +00003896
3897 // If the operands are constants, return a constant result.
3898 if (auto *LHSCI = dyn_cast<llvm::ConstantInt>(LHS)) {
3899 if (auto *RHSCI = dyn_cast<llvm::ConstantInt>(RHS)) {
3900 llvm::APInt N;
3901 bool HasOverflow = mayHaveIntegerOverflow(LHSCI, RHSCI, Opcode,
3902 /*Signed=*/true, N);
3903 if (HasOverflow)
3904 OffsetOverflows = Builder.getTrue();
3905 return llvm::ConstantInt::get(VMContext, N);
3906 }
3907 }
3908
3909 // Otherwise, compute the result with checked arithmetic.
3910 auto *ResultAndOverflow = Builder.CreateCall(
3911 (Opcode == BO_Add) ? SAddIntrinsic : SMulIntrinsic, {LHS, RHS});
3912 OffsetOverflows = Builder.CreateOr(
Vedant Kumar6dbf4272017-06-12 18:42:51 +00003913 Builder.CreateExtractValue(ResultAndOverflow, 1), OffsetOverflows);
Vedant Kumara125eb52017-06-01 19:22:18 +00003914 return Builder.CreateExtractValue(ResultAndOverflow, 0);
3915 };
3916
3917 // Determine the total byte offset by looking at each GEP operand.
3918 for (auto GTI = llvm::gep_type_begin(GEP), GTE = llvm::gep_type_end(GEP);
3919 GTI != GTE; ++GTI) {
3920 llvm::Value *LocalOffset;
3921 auto *Index = GTI.getOperand();
3922 // Compute the local offset contributed by this indexing step:
3923 if (auto *STy = GTI.getStructTypeOrNull()) {
3924 // For struct indexing, the local offset is the byte position of the
3925 // specified field.
3926 unsigned FieldNo = cast<llvm::ConstantInt>(Index)->getZExtValue();
3927 LocalOffset = llvm::ConstantInt::get(
3928 IntPtrTy, DL.getStructLayout(STy)->getElementOffset(FieldNo));
3929 } else {
3930 // Otherwise this is array-like indexing. The local offset is the index
3931 // multiplied by the element size.
3932 auto *ElementSize = llvm::ConstantInt::get(
3933 IntPtrTy, DL.getTypeAllocSize(GTI.getIndexedType()));
3934 auto *IndexS = Builder.CreateIntCast(Index, IntPtrTy, /*isSigned=*/true);
3935 LocalOffset = eval(BO_Mul, ElementSize, IndexS);
3936 }
3937
3938 // If this is the first offset, set it as the total offset. Otherwise, add
3939 // the local offset into the running total.
3940 if (!TotalOffset || TotalOffset == Zero)
3941 TotalOffset = LocalOffset;
3942 else
3943 TotalOffset = eval(BO_Add, TotalOffset, LocalOffset);
3944 }
3945
3946 // Common case: if the total offset is zero, don't emit a check.
3947 if (TotalOffset == Zero)
3948 return GEPVal;
3949
3950 // Now that we've computed the total offset, add it to the base pointer (with
3951 // wrapping semantics).
3952 auto *IntPtr = Builder.CreatePtrToInt(GEP->getPointerOperand(), IntPtrTy);
3953 auto *ComputedGEP = Builder.CreateAdd(IntPtr, TotalOffset);
3954
3955 // The GEP is valid if:
3956 // 1) The total offset doesn't overflow, and
3957 // 2) The sign of the difference between the computed address and the base
3958 // pointer matches the sign of the total offset.
Vedant Kumar6dbf4272017-06-12 18:42:51 +00003959 llvm::Value *ValidGEP;
3960 auto *NoOffsetOverflow = Builder.CreateNot(OffsetOverflows);
3961 auto *PosOrZeroValid = Builder.CreateICmpUGE(ComputedGEP, IntPtr);
3962 if (SignedIndices) {
3963 auto *PosOrZeroOffset = Builder.CreateICmpSGE(TotalOffset, Zero);
3964 llvm::Value *NegValid = Builder.CreateICmpULT(ComputedGEP, IntPtr);
3965 ValidGEP = Builder.CreateAnd(
3966 Builder.CreateSelect(PosOrZeroOffset, PosOrZeroValid, NegValid),
3967 NoOffsetOverflow);
3968 } else {
3969 ValidGEP = Builder.CreateAnd(PosOrZeroValid, NoOffsetOverflow);
3970 }
Vedant Kumara125eb52017-06-01 19:22:18 +00003971
3972 llvm::Constant *StaticArgs[] = {EmitCheckSourceLocation(Loc)};
3973 // Pass the computed GEP to the runtime to avoid emitting poisoned arguments.
3974 llvm::Value *DynamicArgs[] = {IntPtr, ComputedGEP};
3975 EmitCheck(std::make_pair(ValidGEP, SanitizerKind::PointerOverflow),
3976 SanitizerHandler::PointerOverflow, StaticArgs, DynamicArgs);
3977
3978 return GEPVal;
3979}