blob: f48f7185b55f11304e248e5540606c48db338eba [file] [log] [blame]
Chris Lattner2da04b32007-08-24 05:35:26 +00001//===--- CGExprScalar.cpp - Emit LLVM Code for Scalar Exprs ---------------===//
2//
Chandler Carruth2946cd72019-01-19 08:50:56 +00003// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
Chris Lattner2da04b32007-08-24 05:35:26 +00006//
7//===----------------------------------------------------------------------===//
8//
9// This contains code to emit Expr nodes with scalar LLVM types as LLVM code.
10//
11//===----------------------------------------------------------------------===//
12
John McCall5d865c322010-08-31 07:33:07 +000013#include "CGCXXABI.h"
Leonard Chan99bda372018-10-15 16:07:02 +000014#include "CGCleanup.h"
Chandler Carruth3a022472012-12-04 09:13:33 +000015#include "CGDebugInfo.h"
Fariborz Jahanian07ca7272009-10-10 20:07:56 +000016#include "CGObjCRuntime.h"
Alexey Bataeva58da1a2019-12-27 09:44:43 -050017#include "CGOpenMPRuntime.h"
Leonard Chan99bda372018-10-15 16:07:02 +000018#include "CodeGenFunction.h"
Chris Lattner2da04b32007-08-24 05:35:26 +000019#include "CodeGenModule.h"
Eric Fiselier708afb52019-05-16 21:04:15 +000020#include "ConstantEmitter.h"
Alexey Bataev00396512015-07-02 03:40:19 +000021#include "TargetInfo.h"
Daniel Dunbarad319a72008-08-11 05:00:27 +000022#include "clang/AST/ASTContext.h"
Reid Kleckner98031782019-12-09 16:11:56 -080023#include "clang/AST/Attr.h"
Daniel Dunbar6630e102008-08-12 05:08:18 +000024#include "clang/AST/DeclObjC.h"
Yaxun Liu402804b2016-12-15 08:09:08 +000025#include "clang/AST/Expr.h"
Anders Carlsson15b73de2009-07-18 19:43:29 +000026#include "clang/AST/RecordLayout.h"
Daniel Dunbarad319a72008-08-11 05:00:27 +000027#include "clang/AST/StmtVisitor.h"
Richard Trieu63688182018-12-11 03:18:39 +000028#include "clang/Basic/CodeGenOptions.h"
Leonard Chan99bda372018-10-15 16:07:02 +000029#include "clang/Basic/FixedPoint.h"
Chris Lattnerff2367c2008-04-20 00:50:39 +000030#include "clang/Basic/TargetInfo.h"
Vedant Kumar82ee16b2017-02-25 00:43:36 +000031#include "llvm/ADT/Optional.h"
Chandler Carruth735e6d82014-03-04 11:46:22 +000032#include "llvm/IR/CFG.h"
Chandler Carruthffd55512013-01-02 11:45:17 +000033#include "llvm/IR/Constants.h"
34#include "llvm/IR/DataLayout.h"
35#include "llvm/IR/Function.h"
Vedant Kumara125eb52017-06-01 19:22:18 +000036#include "llvm/IR/GetElementPtrTypeIterator.h"
Chandler Carruthffd55512013-01-02 11:45:17 +000037#include "llvm/IR/GlobalVariable.h"
38#include "llvm/IR/Intrinsics.h"
Reid Kleckner5d986952019-12-11 07:55:26 -080039#include "llvm/IR/IntrinsicsPowerPC.h"
Florian Hahn6f6e91d2020-05-29 20:42:22 +010040#include "llvm/IR/MatrixBuilder.h"
Chandler Carruthffd55512013-01-02 11:45:17 +000041#include "llvm/IR/Module.h"
Chris Lattner1800c182008-01-03 07:05:49 +000042#include <cstdarg>
Ted Kremenekf182e812007-12-10 23:44:32 +000043
Chris Lattner2da04b32007-08-24 05:35:26 +000044using namespace clang;
45using namespace CodeGen;
46using llvm::Value;
47
48//===----------------------------------------------------------------------===//
49// Scalar Expression Emitter
50//===----------------------------------------------------------------------===//
51
Benjamin Kramerfb5e5842010-10-22 16:48:22 +000052namespace {
Vedant Kumara125eb52017-06-01 19:22:18 +000053
54/// Determine whether the given binary operation may overflow.
55/// Sets \p Result to the value of the operation for BO_Add, BO_Sub, BO_Mul,
56/// and signed BO_{Div,Rem}. For these opcodes, and for unsigned BO_{Div,Rem},
57/// the returned overflow check is precise. The returned value is 'true' for
58/// all other opcodes, to be conservative.
59bool mayHaveIntegerOverflow(llvm::ConstantInt *LHS, llvm::ConstantInt *RHS,
60 BinaryOperator::Opcode Opcode, bool Signed,
61 llvm::APInt &Result) {
62 // Assume overflow is possible, unless we can prove otherwise.
63 bool Overflow = true;
64 const auto &LHSAP = LHS->getValue();
65 const auto &RHSAP = RHS->getValue();
66 if (Opcode == BO_Add) {
67 if (Signed)
68 Result = LHSAP.sadd_ov(RHSAP, Overflow);
69 else
70 Result = LHSAP.uadd_ov(RHSAP, Overflow);
71 } else if (Opcode == BO_Sub) {
72 if (Signed)
73 Result = LHSAP.ssub_ov(RHSAP, Overflow);
74 else
75 Result = LHSAP.usub_ov(RHSAP, Overflow);
76 } else if (Opcode == BO_Mul) {
77 if (Signed)
78 Result = LHSAP.smul_ov(RHSAP, Overflow);
79 else
80 Result = LHSAP.umul_ov(RHSAP, Overflow);
81 } else if (Opcode == BO_Div || Opcode == BO_Rem) {
82 if (Signed && !RHS->isZero())
83 Result = LHSAP.sdiv_ov(RHSAP, Overflow);
84 else
85 return false;
86 }
87 return Overflow;
88}
89
Chris Lattner2da04b32007-08-24 05:35:26 +000090struct BinOpInfo {
91 Value *LHS;
92 Value *RHS;
Chris Lattner3d966d62007-08-24 21:00:35 +000093 QualType Ty; // Computation Type.
Chris Lattner0bf27622010-06-26 21:48:21 +000094 BinaryOperator::Opcode Opcode; // Opcode of BinOp to perform
Adam Nemet484aa452017-03-27 19:17:25 +000095 FPOptions FPFeatures;
Chris Lattner0bf27622010-06-26 21:48:21 +000096 const Expr *E; // Entire expr, for error unsupported. May not be binop.
Vedant Kumard9191152017-05-02 23:46:56 +000097
98 /// Check if the binop can result in integer overflow.
99 bool mayHaveIntegerOverflow() const {
100 // Without constant input, we can't rule out overflow.
Vedant Kumara125eb52017-06-01 19:22:18 +0000101 auto *LHSCI = dyn_cast<llvm::ConstantInt>(LHS);
102 auto *RHSCI = dyn_cast<llvm::ConstantInt>(RHS);
Vedant Kumard9191152017-05-02 23:46:56 +0000103 if (!LHSCI || !RHSCI)
104 return true;
105
Vedant Kumara125eb52017-06-01 19:22:18 +0000106 llvm::APInt Result;
107 return ::mayHaveIntegerOverflow(
108 LHSCI, RHSCI, Opcode, Ty->hasSignedIntegerRepresentation(), Result);
Vedant Kumard9191152017-05-02 23:46:56 +0000109 }
110
111 /// Check if the binop computes a division or a remainder.
Vedant Kumar94cb34b2017-05-09 00:12:33 +0000112 bool isDivremOp() const {
Vedant Kumard9191152017-05-02 23:46:56 +0000113 return Opcode == BO_Div || Opcode == BO_Rem || Opcode == BO_DivAssign ||
114 Opcode == BO_RemAssign;
115 }
116
117 /// Check if the binop can result in an integer division by zero.
118 bool mayHaveIntegerDivisionByZero() const {
Vedant Kumar94cb34b2017-05-09 00:12:33 +0000119 if (isDivremOp())
Vedant Kumard9191152017-05-02 23:46:56 +0000120 if (auto *CI = dyn_cast<llvm::ConstantInt>(RHS))
121 return CI->isZero();
122 return true;
123 }
124
125 /// Check if the binop can result in a float division by zero.
126 bool mayHaveFloatDivisionByZero() const {
Vedant Kumar94cb34b2017-05-09 00:12:33 +0000127 if (isDivremOp())
Vedant Kumard9191152017-05-02 23:46:56 +0000128 if (auto *CFP = dyn_cast<llvm::ConstantFP>(RHS))
129 return CFP->isZero();
130 return true;
131 }
Leonard Chan2044ac82019-01-16 18:13:59 +0000132
Bevin Hansson39baaab2020-01-08 11:12:55 +0100133 /// Check if at least one operand is a fixed point type. In such cases, this
134 /// operation did not follow usual arithmetic conversion and both operands
135 /// might not be of the same type.
136 bool isFixedPointOp() const {
Leonard Chance1d4f12019-02-21 20:50:09 +0000137 // We cannot simply check the result type since comparison operations return
138 // an int.
139 if (const auto *BinOp = dyn_cast<BinaryOperator>(E)) {
140 QualType LHSType = BinOp->getLHS()->getType();
141 QualType RHSType = BinOp->getRHS()->getType();
142 return LHSType->isFixedPointType() || RHSType->isFixedPointType();
143 }
Bevin Hansson39baaab2020-01-08 11:12:55 +0100144 if (const auto *UnOp = dyn_cast<UnaryOperator>(E))
145 return UnOp->getSubExpr()->getType()->isFixedPointType();
Leonard Chance1d4f12019-02-21 20:50:09 +0000146 return false;
Leonard Chan2044ac82019-01-16 18:13:59 +0000147 }
Chris Lattner2da04b32007-08-24 05:35:26 +0000148};
149
John McCalle84af4e2010-11-13 01:35:44 +0000150static bool MustVisitNullValue(const Expr *E) {
151 // If a null pointer expression's type is the C++0x nullptr_t, then
152 // it's not necessarily a simple constant and it must be evaluated
153 // for its potential side effects.
154 return E->getType()->isNullPtrType();
155}
156
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000157/// If \p E is a widened promoted integer, get its base (unpromoted) type.
158static llvm::Optional<QualType> getUnwidenedIntegerType(const ASTContext &Ctx,
159 const Expr *E) {
160 const Expr *Base = E->IgnoreImpCasts();
161 if (E == Base)
162 return llvm::None;
163
164 QualType BaseTy = Base->getType();
165 if (!BaseTy->isPromotableIntegerType() ||
166 Ctx.getTypeSize(BaseTy) >= Ctx.getTypeSize(E->getType()))
167 return llvm::None;
168
169 return BaseTy;
170}
171
172/// Check if \p E is a widened promoted integer.
173static bool IsWidenedIntegerOp(const ASTContext &Ctx, const Expr *E) {
174 return getUnwidenedIntegerType(Ctx, E).hasValue();
175}
176
177/// Check if we can skip the overflow check for \p Op.
178static bool CanElideOverflowCheck(const ASTContext &Ctx, const BinOpInfo &Op) {
Vedant Kumar66c00cc2017-02-25 06:47:00 +0000179 assert((isa<UnaryOperator>(Op.E) || isa<BinaryOperator>(Op.E)) &&
180 "Expected a unary or binary operator");
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000181
Vedant Kumard9191152017-05-02 23:46:56 +0000182 // If the binop has constant inputs and we can prove there is no overflow,
183 // we can elide the overflow check.
184 if (!Op.mayHaveIntegerOverflow())
185 return true;
Malcolm Parsonsfab36802018-04-16 08:31:08 +0000186
187 // If a unary op has a widened operand, the op cannot overflow.
188 if (const auto *UO = dyn_cast<UnaryOperator>(Op.E))
189 return !UO->canOverflow();
190
191 // We usually don't need overflow checks for binops with widened operands.
192 // Multiplication with promoted unsigned operands is a special case.
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000193 const auto *BO = cast<BinaryOperator>(Op.E);
194 auto OptionalLHSTy = getUnwidenedIntegerType(Ctx, BO->getLHS());
195 if (!OptionalLHSTy)
196 return false;
197
198 auto OptionalRHSTy = getUnwidenedIntegerType(Ctx, BO->getRHS());
199 if (!OptionalRHSTy)
200 return false;
201
202 QualType LHSTy = *OptionalLHSTy;
203 QualType RHSTy = *OptionalRHSTy;
204
Vedant Kumard9191152017-05-02 23:46:56 +0000205 // This is the simple case: binops without unsigned multiplication, and with
206 // widened operands. No overflow check is needed here.
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000207 if ((Op.Opcode != BO_Mul && Op.Opcode != BO_MulAssign) ||
208 !LHSTy->isUnsignedIntegerType() || !RHSTy->isUnsignedIntegerType())
209 return true;
210
Vedant Kumard9191152017-05-02 23:46:56 +0000211 // For unsigned multiplication the overflow check can be elided if either one
212 // of the unpromoted types are less than half the size of the promoted type.
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000213 unsigned PromotedSize = Ctx.getTypeSize(Op.E->getType());
214 return (2 * Ctx.getTypeSize(LHSTy)) < PromotedSize ||
215 (2 * Ctx.getTypeSize(RHSTy)) < PromotedSize;
216}
217
Benjamin Kramer337e3a52009-11-28 19:45:26 +0000218class ScalarExprEmitter
Chris Lattner2da04b32007-08-24 05:35:26 +0000219 : public StmtVisitor<ScalarExprEmitter, Value*> {
220 CodeGenFunction &CGF;
Daniel Dunbarcb463852008-11-01 01:53:16 +0000221 CGBuilderTy &Builder;
Mike Stumpdf0fe272009-05-29 15:46:01 +0000222 bool IgnoreResultAssign;
Owen Anderson170229f2009-07-14 23:10:40 +0000223 llvm::LLVMContext &VMContext;
Chris Lattner2da04b32007-08-24 05:35:26 +0000224public:
225
Mike Stumpdf0fe272009-05-29 15:46:01 +0000226 ScalarExprEmitter(CodeGenFunction &cgf, bool ira=false)
Mike Stump4a3999f2009-09-09 13:00:44 +0000227 : CGF(cgf), Builder(CGF.Builder), IgnoreResultAssign(ira),
Owen Anderson170229f2009-07-14 23:10:40 +0000228 VMContext(cgf.getLLVMContext()) {
Chris Lattner2da04b32007-08-24 05:35:26 +0000229 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000230
Chris Lattner2da04b32007-08-24 05:35:26 +0000231 //===--------------------------------------------------------------------===//
232 // Utilities
233 //===--------------------------------------------------------------------===//
234
Mike Stumpdf0fe272009-05-29 15:46:01 +0000235 bool TestAndClearIgnoreResultAssign() {
Chris Lattner2a7deb62009-07-08 01:08:03 +0000236 bool I = IgnoreResultAssign;
237 IgnoreResultAssign = false;
238 return I;
239 }
Mike Stumpdf0fe272009-05-29 15:46:01 +0000240
Chris Lattner2192fe52011-07-18 04:24:23 +0000241 llvm::Type *ConvertType(QualType T) { return CGF.ConvertType(T); }
Chris Lattner2da04b32007-08-24 05:35:26 +0000242 LValue EmitLValue(const Expr *E) { return CGF.EmitLValue(E); }
Richard Smith4d1458e2012-09-08 02:08:36 +0000243 LValue EmitCheckedLValue(const Expr *E, CodeGenFunction::TypeCheckKind TCK) {
244 return CGF.EmitCheckedLValue(E, TCK);
Richard Smith69d0d262012-08-24 00:54:33 +0000245 }
Chris Lattner2da04b32007-08-24 05:35:26 +0000246
Peter Collingbourne3eea6772015-05-11 21:39:14 +0000247 void EmitBinOpCheck(ArrayRef<std::pair<Value *, SanitizerMask>> Checks,
Alexey Samsonove396bfc2014-11-11 22:03:54 +0000248 const BinOpInfo &Info);
Richard Smithe30752c2012-10-09 19:52:38 +0000249
Nick Lewycky2d84e842013-10-02 02:29:49 +0000250 Value *EmitLoadOfLValue(LValue LV, SourceLocation Loc) {
251 return CGF.EmitLoadOfLValue(LV, Loc).getScalarVal();
Chris Lattner2da04b32007-08-24 05:35:26 +0000252 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000253
Hal Finkel64567a82014-10-04 15:26:49 +0000254 void EmitLValueAlignmentAssumption(const Expr *E, Value *V) {
255 const AlignValueAttr *AVAttr = nullptr;
256 if (const auto *DRE = dyn_cast<DeclRefExpr>(E)) {
257 const ValueDecl *VD = DRE->getDecl();
258
259 if (VD->getType()->isReferenceType()) {
260 if (const auto *TTy =
261 dyn_cast<TypedefType>(VD->getType().getNonReferenceType()))
262 AVAttr = TTy->getDecl()->getAttr<AlignValueAttr>();
263 } else {
264 // Assumptions for function parameters are emitted at the start of the
Roman Lebedevbd1c0872019-01-15 09:44:25 +0000265 // function, so there is no need to repeat that here,
266 // unless the alignment-assumption sanitizer is enabled,
267 // then we prefer the assumption over alignment attribute
268 // on IR function param.
269 if (isa<ParmVarDecl>(VD) && !CGF.SanOpts.has(SanitizerKind::Alignment))
Hal Finkel64567a82014-10-04 15:26:49 +0000270 return;
271
272 AVAttr = VD->getAttr<AlignValueAttr>();
273 }
274 }
275
276 if (!AVAttr)
277 if (const auto *TTy =
278 dyn_cast<TypedefType>(E->getType()))
279 AVAttr = TTy->getDecl()->getAttr<AlignValueAttr>();
280
281 if (!AVAttr)
282 return;
283
284 Value *AlignmentValue = CGF.EmitScalarExpr(AVAttr->getAlignment());
285 llvm::ConstantInt *AlignmentCI = cast<llvm::ConstantInt>(AlignmentValue);
Fangrui Song1d49eb02020-02-13 16:36:27 -0800286 CGF.emitAlignmentAssumption(V, E, AVAttr->getLocation(), AlignmentCI);
Hal Finkel64567a82014-10-04 15:26:49 +0000287 }
288
Chris Lattner2da04b32007-08-24 05:35:26 +0000289 /// EmitLoadOfLValue - Given an expression with complex type that represents a
290 /// value l-value, this method emits the address of the l-value, then loads
291 /// and returns the result.
292 Value *EmitLoadOfLValue(const Expr *E) {
Hal Finkel64567a82014-10-04 15:26:49 +0000293 Value *V = EmitLoadOfLValue(EmitCheckedLValue(E, CodeGenFunction::TCK_Load),
294 E->getExprLoc());
295
296 EmitLValueAlignmentAssumption(E, V);
297 return V;
Chris Lattner2da04b32007-08-24 05:35:26 +0000298 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000299
Chris Lattnere0044382007-08-26 16:42:57 +0000300 /// EmitConversionToBool - Convert the specified expression value to a
Chris Lattner2e928882007-08-26 17:25:57 +0000301 /// boolean (i1) truth value. This is equivalent to "Val != 0".
Chris Lattnere0044382007-08-26 16:42:57 +0000302 Value *EmitConversionToBool(Value *Src, QualType DstTy);
Mike Stump4a3999f2009-09-09 13:00:44 +0000303
Richard Smith9e52c432019-07-06 21:05:52 +0000304 /// Emit a check that a conversion from a floating-point type does not
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +0000305 /// overflow.
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000306 void EmitFloatConversionCheck(Value *OrigSrc, QualType OrigSrcType,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +0000307 Value *Src, QualType SrcType, QualType DstType,
308 llvm::Type *DstTy, SourceLocation Loc);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000309
Roman Lebedevb69ba222018-07-30 18:58:30 +0000310 /// Known implicit conversion check kinds.
311 /// Keep in sync with the enum of the same name in ubsan_handlers.h
312 enum ImplicitConversionCheckKind : unsigned char {
Roman Lebedevdd403572018-10-11 09:09:50 +0000313 ICCK_IntegerTruncation = 0, // Legacy, was only used by clang 7.
314 ICCK_UnsignedIntegerTruncation = 1,
315 ICCK_SignedIntegerTruncation = 2,
Roman Lebedev62debd802018-10-30 21:58:56 +0000316 ICCK_IntegerSignChange = 3,
317 ICCK_SignedIntegerTruncationOrSignChange = 4,
Roman Lebedevb69ba222018-07-30 18:58:30 +0000318 };
319
320 /// Emit a check that an [implicit] truncation of an integer does not
321 /// discard any bits. It is not UB, so we use the value after truncation.
322 void EmitIntegerTruncationCheck(Value *Src, QualType SrcType, Value *Dst,
323 QualType DstType, SourceLocation Loc);
324
Roman Lebedev62debd802018-10-30 21:58:56 +0000325 /// Emit a check that an [implicit] conversion of an integer does not change
326 /// the sign of the value. It is not UB, so we use the value after conversion.
327 /// NOTE: Src and Dst may be the exact same value! (point to the same thing)
328 void EmitIntegerSignChangeCheck(Value *Src, QualType SrcType, Value *Dst,
329 QualType DstType, SourceLocation Loc);
330
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +0000331 /// Emit a conversion from the specified type to the specified destination
332 /// type, both of which are LLVM scalar types.
Roman Lebedevb69ba222018-07-30 18:58:30 +0000333 struct ScalarConversionOpts {
334 bool TreatBooleanAsSigned;
335 bool EmitImplicitIntegerTruncationChecks;
Roman Lebedev62debd802018-10-30 21:58:56 +0000336 bool EmitImplicitIntegerSignChangeChecks;
Chris Lattner42e6b812007-08-26 16:34:22 +0000337
Roman Lebedevb69ba222018-07-30 18:58:30 +0000338 ScalarConversionOpts()
339 : TreatBooleanAsSigned(false),
Roman Lebedev62debd802018-10-30 21:58:56 +0000340 EmitImplicitIntegerTruncationChecks(false),
341 EmitImplicitIntegerSignChangeChecks(false) {}
Roman Lebedevd677c3f2018-11-19 19:56:43 +0000342
343 ScalarConversionOpts(clang::SanitizerSet SanOpts)
344 : TreatBooleanAsSigned(false),
345 EmitImplicitIntegerTruncationChecks(
346 SanOpts.hasOneOf(SanitizerKind::ImplicitIntegerTruncation)),
347 EmitImplicitIntegerSignChangeChecks(
348 SanOpts.has(SanitizerKind::ImplicitIntegerSignChange)) {}
Roman Lebedevb69ba222018-07-30 18:58:30 +0000349 };
350 Value *
351 EmitScalarConversion(Value *Src, QualType SrcTy, QualType DstTy,
352 SourceLocation Loc,
353 ScalarConversionOpts Opts = ScalarConversionOpts());
Anastasia Stulovab02e7832015-10-05 11:27:41 +0000354
Leonard Chan8f7caae2019-03-06 00:28:43 +0000355 /// Convert between either a fixed point and other fixed point or fixed point
356 /// and an integer.
Leonard Chan99bda372018-10-15 16:07:02 +0000357 Value *EmitFixedPointConversion(Value *Src, QualType SrcTy, QualType DstTy,
358 SourceLocation Loc);
Leonard Chan2044ac82019-01-16 18:13:59 +0000359 Value *EmitFixedPointConversion(Value *Src, FixedPointSemantics &SrcFixedSema,
360 FixedPointSemantics &DstFixedSema,
Leonard Chan8f7caae2019-03-06 00:28:43 +0000361 SourceLocation Loc,
362 bool DstIsInteger = false);
Leonard Chan99bda372018-10-15 16:07:02 +0000363
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +0000364 /// Emit a conversion from the specified complex type to the specified
365 /// destination type, where the destination type is an LLVM scalar type.
Chris Lattner42e6b812007-08-26 16:34:22 +0000366 Value *EmitComplexToScalarConversion(CodeGenFunction::ComplexPairTy Src,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +0000367 QualType SrcTy, QualType DstTy,
368 SourceLocation Loc);
Mike Stumpab3afd82009-02-12 18:29:15 +0000369
Anders Carlsson5b944432010-05-22 17:45:10 +0000370 /// EmitNullValue - Emit a value that corresponds to null for the given type.
371 Value *EmitNullValue(QualType Ty);
372
John McCall8cb679e2010-11-15 09:13:47 +0000373 /// EmitFloatToBoolConversion - Perform an FP to boolean conversion.
374 Value *EmitFloatToBoolConversion(Value *V) {
375 // Compare against 0.0 for fp scalars.
376 llvm::Value *Zero = llvm::Constant::getNullValue(V->getType());
377 return Builder.CreateFCmpUNE(V, Zero, "tobool");
378 }
379
380 /// EmitPointerToBoolConversion - Perform a pointer to boolean conversion.
Yaxun Liu402804b2016-12-15 08:09:08 +0000381 Value *EmitPointerToBoolConversion(Value *V, QualType QT) {
382 Value *Zero = CGF.CGM.getNullPointer(cast<llvm::PointerType>(V->getType()), QT);
383
John McCall8cb679e2010-11-15 09:13:47 +0000384 return Builder.CreateICmpNE(V, Zero, "tobool");
385 }
386
387 Value *EmitIntToBoolConversion(Value *V) {
388 // Because of the type rules of C, we often end up computing a
389 // logical value, then zero extending it to int, then wanting it
390 // as a logical value again. Optimize this common case.
391 if (llvm::ZExtInst *ZI = dyn_cast<llvm::ZExtInst>(V)) {
392 if (ZI->getOperand(0)->getType() == Builder.getInt1Ty()) {
393 Value *Result = ZI->getOperand(0);
394 // If there aren't any more uses, zap the instruction to save space.
395 // Note that there can be more uses, for example if this
396 // is the result of an assignment.
397 if (ZI->use_empty())
398 ZI->eraseFromParent();
399 return Result;
400 }
401 }
402
Chris Lattner2531eb42011-04-19 22:55:03 +0000403 return Builder.CreateIsNotNull(V, "tobool");
John McCall8cb679e2010-11-15 09:13:47 +0000404 }
405
Chris Lattner2da04b32007-08-24 05:35:26 +0000406 //===--------------------------------------------------------------------===//
407 // Visitor Methods
408 //===--------------------------------------------------------------------===//
409
Fariborz Jahanian5bbd1b02010-09-17 15:51:28 +0000410 Value *Visit(Expr *E) {
David Blaikie9b479662015-01-25 01:19:10 +0000411 ApplyDebugLocation DL(CGF, E);
Fariborz Jahanian5bbd1b02010-09-17 15:51:28 +0000412 return StmtVisitor<ScalarExprEmitter, Value*>::Visit(E);
413 }
Craig Toppera97d7e72013-07-26 06:16:11 +0000414
Chris Lattner2da04b32007-08-24 05:35:26 +0000415 Value *VisitStmt(Stmt *S) {
Ted Kremenekd4e5fba2007-12-11 21:27:55 +0000416 S->dump(CGF.getContext().getSourceManager());
David Blaikie83d382b2011-09-23 05:06:16 +0000417 llvm_unreachable("Stmt can't have complex result type!");
Chris Lattner2da04b32007-08-24 05:35:26 +0000418 }
419 Value *VisitExpr(Expr *S);
Craig Toppera97d7e72013-07-26 06:16:11 +0000420
Bill Wendling8003edc2018-11-09 00:41:36 +0000421 Value *VisitConstantExpr(ConstantExpr *E) {
Tyker51e4aa872020-06-14 14:39:14 +0200422 if (Value *Result = ConstantEmitter(CGF).tryEmitConstantExpr(E)) {
423 if (E->isGLValue())
424 return CGF.Builder.CreateLoad(Address(
425 Result, CGF.getContext().getTypeAlignInChars(E->getType())));
426 return Result;
427 }
Bill Wendling8003edc2018-11-09 00:41:36 +0000428 return Visit(E->getSubExpr());
429 }
Fariborz Jahanian5bbd1b02010-09-17 15:51:28 +0000430 Value *VisitParenExpr(ParenExpr *PE) {
Craig Toppera97d7e72013-07-26 06:16:11 +0000431 return Visit(PE->getSubExpr());
Fariborz Jahanian5bbd1b02010-09-17 15:51:28 +0000432 }
John McCall7c454bb2011-07-15 05:09:51 +0000433 Value *VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr *E) {
Craig Toppera97d7e72013-07-26 06:16:11 +0000434 return Visit(E->getReplacement());
John McCall7c454bb2011-07-15 05:09:51 +0000435 }
Peter Collingbourne91147592011-04-15 00:35:48 +0000436 Value *VisitGenericSelectionExpr(GenericSelectionExpr *GE) {
437 return Visit(GE->getResultExpr());
438 }
Gor Nishanov5eb58582017-03-26 02:18:05 +0000439 Value *VisitCoawaitExpr(CoawaitExpr *S) {
440 return CGF.EmitCoawaitExpr(*S).getScalarVal();
441 }
442 Value *VisitCoyieldExpr(CoyieldExpr *S) {
443 return CGF.EmitCoyieldExpr(*S).getScalarVal();
444 }
445 Value *VisitUnaryCoawait(const UnaryOperator *E) {
446 return Visit(E->getSubExpr());
447 }
Chris Lattner2da04b32007-08-24 05:35:26 +0000448
449 // Leaves.
450 Value *VisitIntegerLiteral(const IntegerLiteral *E) {
Chris Lattner2531eb42011-04-19 22:55:03 +0000451 return Builder.getInt(E->getValue());
Chris Lattner2da04b32007-08-24 05:35:26 +0000452 }
Leonard Chandb01c3a2018-06-20 17:19:40 +0000453 Value *VisitFixedPointLiteral(const FixedPointLiteral *E) {
454 return Builder.getInt(E->getValue());
455 }
Chris Lattner2da04b32007-08-24 05:35:26 +0000456 Value *VisitFloatingLiteral(const FloatingLiteral *E) {
Owen Andersone05f2ed2009-07-27 21:00:51 +0000457 return llvm::ConstantFP::get(VMContext, E->getValue());
Chris Lattner2da04b32007-08-24 05:35:26 +0000458 }
459 Value *VisitCharacterLiteral(const CharacterLiteral *E) {
Owen Andersonb7a2fe62009-07-24 23:12:58 +0000460 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
Chris Lattner2da04b32007-08-24 05:35:26 +0000461 }
Ted Kremeneke65b0862012-03-06 20:05:56 +0000462 Value *VisitObjCBoolLiteralExpr(const ObjCBoolLiteralExpr *E) {
463 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
464 }
Nate Begeman4c18c232007-11-15 05:40:03 +0000465 Value *VisitCXXBoolLiteralExpr(const CXXBoolLiteralExpr *E) {
Owen Andersonb7a2fe62009-07-24 23:12:58 +0000466 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
Nate Begeman4c18c232007-11-15 05:40:03 +0000467 }
Douglas Gregor747eb782010-07-08 06:14:04 +0000468 Value *VisitCXXScalarValueInitExpr(const CXXScalarValueInitExpr *E) {
Anders Carlsson5b944432010-05-22 17:45:10 +0000469 return EmitNullValue(E->getType());
Argyrios Kyrtzidisce4528f2008-08-23 19:35:47 +0000470 }
Anders Carlsson39def3a2008-12-21 22:39:40 +0000471 Value *VisitGNUNullExpr(const GNUNullExpr *E) {
Anders Carlsson5b944432010-05-22 17:45:10 +0000472 return EmitNullValue(E->getType());
Anders Carlsson39def3a2008-12-21 22:39:40 +0000473 }
Eli Friedmand7c72322010-08-05 09:58:49 +0000474 Value *VisitOffsetOfExpr(OffsetOfExpr *E);
Peter Collingbournee190dee2011-03-11 19:24:49 +0000475 Value *VisitUnaryExprOrTypeTraitExpr(const UnaryExprOrTypeTraitExpr *E);
Daniel Dunbar88402ce2008-08-04 16:51:22 +0000476 Value *VisitAddrLabelExpr(const AddrLabelExpr *E) {
Chris Lattner6c4d2552009-10-28 23:59:40 +0000477 llvm::Value *V = CGF.GetAddrOfLabel(E->getLabel());
478 return Builder.CreateBitCast(V, ConvertType(E->getType()));
Daniel Dunbar88402ce2008-08-04 16:51:22 +0000479 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000480
Douglas Gregorbe7b5482011-01-12 22:11:34 +0000481 Value *VisitSizeOfPackExpr(SizeOfPackExpr *E) {
Chris Lattner2531eb42011-04-19 22:55:03 +0000482 return llvm::ConstantInt::get(ConvertType(E->getType()),E->getPackLength());
Douglas Gregorbe7b5482011-01-12 22:11:34 +0000483 }
John McCall1bf58462011-02-16 08:02:54 +0000484
John McCallfe96e0b2011-11-06 09:01:30 +0000485 Value *VisitPseudoObjectExpr(PseudoObjectExpr *E) {
486 return CGF.EmitPseudoObjectRValue(E).getScalarVal();
487 }
488
John McCall1bf58462011-02-16 08:02:54 +0000489 Value *VisitOpaqueValueExpr(OpaqueValueExpr *E) {
John McCallc07a0c72011-02-17 10:25:35 +0000490 if (E->isGLValue())
Akira Hatanaka797afe32018-03-20 01:47:58 +0000491 return EmitLoadOfLValue(CGF.getOrCreateOpaqueLValueMapping(E),
492 E->getExprLoc());
John McCall1bf58462011-02-16 08:02:54 +0000493
494 // Otherwise, assume the mapping is the scalar directly.
Akira Hatanaka797afe32018-03-20 01:47:58 +0000495 return CGF.getOrCreateOpaqueRValueMapping(E).getScalarVal();
John McCall1bf58462011-02-16 08:02:54 +0000496 }
John McCall71335052012-03-10 03:05:10 +0000497
Chris Lattner2da04b32007-08-24 05:35:26 +0000498 // l-values.
John McCall113bee02012-03-10 09:33:50 +0000499 Value *VisitDeclRefExpr(DeclRefExpr *E) {
Alex Lorenz6cc83172017-08-25 10:07:00 +0000500 if (CodeGenFunction::ConstantEmission Constant = CGF.tryEmitAsConstant(E))
Volodymyr Sapsaief1899b2018-11-01 21:57:05 +0000501 return CGF.emitScalarConstant(Constant, E);
John McCall113bee02012-03-10 09:33:50 +0000502 return EmitLoadOfLValue(E);
John McCall71335052012-03-10 03:05:10 +0000503 }
504
Mike Stump4a3999f2009-09-09 13:00:44 +0000505 Value *VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
506 return CGF.EmitObjCSelectorExpr(E);
Daniel Dunbar55310df2008-08-27 06:57:25 +0000507 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000508 Value *VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
509 return CGF.EmitObjCProtocolExpr(E);
Daniel Dunbar55310df2008-08-27 06:57:25 +0000510 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000511 Value *VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
Daniel Dunbar55310df2008-08-27 06:57:25 +0000512 return EmitLoadOfLValue(E);
513 }
Daniel Dunbar55310df2008-08-27 06:57:25 +0000514 Value *VisitObjCMessageExpr(ObjCMessageExpr *E) {
Craig Toppera97d7e72013-07-26 06:16:11 +0000515 if (E->getMethodDecl() &&
Alp Toker314cc812014-01-25 16:55:45 +0000516 E->getMethodDecl()->getReturnType()->isReferenceType())
Fariborz Jahanianff989032011-03-02 20:09:49 +0000517 return EmitLoadOfLValue(E);
Daniel Dunbar55310df2008-08-27 06:57:25 +0000518 return CGF.EmitObjCMessageExpr(E).getScalarVal();
Daniel Dunbarc8317a42008-08-23 10:51:21 +0000519 }
520
Fariborz Jahaniana5fee262009-12-09 19:05:56 +0000521 Value *VisitObjCIsaExpr(ObjCIsaExpr *E) {
Fariborz Jahanian531c16f2009-12-09 23:35:29 +0000522 LValue LV = CGF.EmitObjCIsaExpr(E);
Nick Lewycky2d84e842013-10-02 02:29:49 +0000523 Value *V = CGF.EmitLoadOfLValue(LV, E->getExprLoc()).getScalarVal();
Fariborz Jahaniana5fee262009-12-09 19:05:56 +0000524 return V;
525 }
526
Erik Pilkington9c42a8d2017-02-23 21:08:08 +0000527 Value *VisitObjCAvailabilityCheckExpr(ObjCAvailabilityCheckExpr *E) {
528 VersionTuple Version = E->getVersion();
529
530 // If we're checking for a platform older than our minimum deployment
531 // target, we can fold the check away.
532 if (Version <= CGF.CGM.getTarget().getPlatformMinVersion())
533 return llvm::ConstantInt::get(Builder.getInt1Ty(), 1);
534
535 Optional<unsigned> Min = Version.getMinor(), SMin = Version.getSubminor();
536 llvm::Value *Args[] = {
537 llvm::ConstantInt::get(CGF.CGM.Int32Ty, Version.getMajor()),
538 llvm::ConstantInt::get(CGF.CGM.Int32Ty, Min ? *Min : 0),
539 llvm::ConstantInt::get(CGF.CGM.Int32Ty, SMin ? *SMin : 0),
540 };
541
542 return CGF.EmitBuiltinAvailable(Args);
543 }
544
Chris Lattner2da04b32007-08-24 05:35:26 +0000545 Value *VisitArraySubscriptExpr(ArraySubscriptExpr *E);
Florian Hahn8f3f88d2020-06-01 19:42:03 +0100546 Value *VisitMatrixSubscriptExpr(MatrixSubscriptExpr *E);
Eli Friedmana1b4ed82008-05-14 19:38:39 +0000547 Value *VisitShuffleVectorExpr(ShuffleVectorExpr *E);
Hal Finkelc4d7c822013-09-18 03:29:45 +0000548 Value *VisitConvertVectorExpr(ConvertVectorExpr *E);
Eli Friedmancb422f12009-11-26 03:22:21 +0000549 Value *VisitMemberExpr(MemberExpr *E);
Nate Begemance4d7fc2008-04-18 23:10:10 +0000550 Value *VisitExtVectorElementExpr(Expr *E) { return EmitLoadOfLValue(E); }
Chris Lattner084bc322008-10-26 23:53:12 +0000551 Value *VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
Akira Hatanaka40568fe2020-03-10 14:06:25 -0700552 // Strictly speaking, we shouldn't be calling EmitLoadOfLValue, which
553 // transitively calls EmitCompoundLiteralLValue, here in C++ since compound
554 // literals aren't l-values in C++. We do so simply because that's the
555 // cleanest way to handle compound literals in C++.
556 // See the discussion here: https://reviews.llvm.org/D64464
Chris Lattner084bc322008-10-26 23:53:12 +0000557 return EmitLoadOfLValue(E);
558 }
Devang Patel43fc86d2007-10-24 17:18:43 +0000559
Nate Begeman19351632009-10-18 20:10:40 +0000560 Value *VisitInitListExpr(InitListExpr *E);
Mike Stump4a3999f2009-09-09 13:00:44 +0000561
Richard Smith410306b2016-12-12 02:53:20 +0000562 Value *VisitArrayInitIndexExpr(ArrayInitIndexExpr *E) {
563 assert(CGF.getArrayInitIndex() &&
564 "ArrayInitIndexExpr not inside an ArrayInitLoopExpr?");
565 return CGF.getArrayInitIndex();
566 }
567
Douglas Gregor0202cb42009-01-29 17:44:32 +0000568 Value *VisitImplicitValueInitExpr(const ImplicitValueInitExpr *E) {
Richard Smithd82a2ce2012-12-21 03:17:28 +0000569 return EmitNullValue(E->getType());
Douglas Gregor0202cb42009-01-29 17:44:32 +0000570 }
John McCall23c29fe2011-06-24 21:55:10 +0000571 Value *VisitExplicitCastExpr(ExplicitCastExpr *E) {
Alexey Bataev2bf9b4c2015-10-20 04:24:12 +0000572 CGF.CGM.EmitExplicitCastExprType(E, &CGF);
John McCall23c29fe2011-06-24 21:55:10 +0000573 return VisitCastExpr(E);
Chris Lattner2da04b32007-08-24 05:35:26 +0000574 }
John McCall23c29fe2011-06-24 21:55:10 +0000575 Value *VisitCastExpr(CastExpr *E);
Chris Lattner2da04b32007-08-24 05:35:26 +0000576
577 Value *VisitCallExpr(const CallExpr *E) {
David Majnemerced8bdf2015-02-25 17:36:15 +0000578 if (E->getCallReturnType(CGF.getContext())->isReferenceType())
Anders Carlssond8b7ae22009-05-27 03:37:57 +0000579 return EmitLoadOfLValue(E);
Mike Stump4a3999f2009-09-09 13:00:44 +0000580
Hal Finkel64567a82014-10-04 15:26:49 +0000581 Value *V = CGF.EmitCallExpr(E).getScalarVal();
582
583 EmitLValueAlignmentAssumption(E, V);
584 return V;
Chris Lattner2da04b32007-08-24 05:35:26 +0000585 }
Daniel Dunbar97db84c2008-08-23 03:46:30 +0000586
Chris Lattner04a913b2007-08-31 22:09:40 +0000587 Value *VisitStmtExpr(const StmtExpr *E);
Mike Stumpcb2fbcb2009-02-21 20:00:35 +0000588
Chris Lattner2da04b32007-08-24 05:35:26 +0000589 // Unary Operators.
Chris Lattner2da04b32007-08-24 05:35:26 +0000590 Value *VisitUnaryPostDec(const UnaryOperator *E) {
Chris Lattner05dc78c2010-06-26 22:09:34 +0000591 LValue LV = EmitLValue(E->getSubExpr());
592 return EmitScalarPrePostIncDec(E, LV, false, false);
Chris Lattner2da04b32007-08-24 05:35:26 +0000593 }
594 Value *VisitUnaryPostInc(const UnaryOperator *E) {
Chris Lattner05dc78c2010-06-26 22:09:34 +0000595 LValue LV = EmitLValue(E->getSubExpr());
596 return EmitScalarPrePostIncDec(E, LV, true, false);
Chris Lattner2da04b32007-08-24 05:35:26 +0000597 }
598 Value *VisitUnaryPreDec(const UnaryOperator *E) {
Chris Lattner05dc78c2010-06-26 22:09:34 +0000599 LValue LV = EmitLValue(E->getSubExpr());
600 return EmitScalarPrePostIncDec(E, LV, false, true);
Chris Lattner2da04b32007-08-24 05:35:26 +0000601 }
602 Value *VisitUnaryPreInc(const UnaryOperator *E) {
Chris Lattner05dc78c2010-06-26 22:09:34 +0000603 LValue LV = EmitLValue(E->getSubExpr());
604 return EmitScalarPrePostIncDec(E, LV, true, true);
Chris Lattner2da04b32007-08-24 05:35:26 +0000605 }
Chris Lattner05dc78c2010-06-26 22:09:34 +0000606
Alexey Samsonovf6246502015-04-23 01:50:45 +0000607 llvm::Value *EmitIncDecConsiderOverflowBehavior(const UnaryOperator *E,
608 llvm::Value *InVal,
609 bool IsInc);
Anton Yartsev85129b82011-02-07 02:17:30 +0000610
Chris Lattner05dc78c2010-06-26 22:09:34 +0000611 llvm::Value *EmitScalarPrePostIncDec(const UnaryOperator *E, LValue LV,
612 bool isInc, bool isPre);
613
Craig Toppera97d7e72013-07-26 06:16:11 +0000614
Chris Lattner2da04b32007-08-24 05:35:26 +0000615 Value *VisitUnaryAddrOf(const UnaryOperator *E) {
John McCallf3a88602011-02-03 08:15:49 +0000616 if (isa<MemberPointerType>(E->getType())) // never sugared
617 return CGF.CGM.getMemberPointerConstant(E);
618
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800619 return EmitLValue(E->getSubExpr()).getPointer(CGF);
Chris Lattner2da04b32007-08-24 05:35:26 +0000620 }
John McCall59482722010-12-04 12:43:24 +0000621 Value *VisitUnaryDeref(const UnaryOperator *E) {
622 if (E->getType()->isVoidType())
623 return Visit(E->getSubExpr()); // the actual value should be unused
624 return EmitLoadOfLValue(E);
625 }
Chris Lattner2da04b32007-08-24 05:35:26 +0000626 Value *VisitUnaryPlus(const UnaryOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +0000627 // This differs from gcc, though, most likely due to a bug in gcc.
628 TestAndClearIgnoreResultAssign();
Chris Lattner2da04b32007-08-24 05:35:26 +0000629 return Visit(E->getSubExpr());
630 }
631 Value *VisitUnaryMinus (const UnaryOperator *E);
632 Value *VisitUnaryNot (const UnaryOperator *E);
633 Value *VisitUnaryLNot (const UnaryOperator *E);
Chris Lattner9f0ad962007-08-24 21:20:17 +0000634 Value *VisitUnaryReal (const UnaryOperator *E);
635 Value *VisitUnaryImag (const UnaryOperator *E);
Chris Lattner2da04b32007-08-24 05:35:26 +0000636 Value *VisitUnaryExtension(const UnaryOperator *E) {
637 return Visit(E->getSubExpr());
638 }
Craig Toppera97d7e72013-07-26 06:16:11 +0000639
Anders Carlssona5d077d2009-04-14 16:58:56 +0000640 // C++
Douglas Gregor34f6c6d2011-08-09 00:37:14 +0000641 Value *VisitMaterializeTemporaryExpr(const MaterializeTemporaryExpr *E) {
Eli Friedman0be39702011-08-14 04:50:34 +0000642 return EmitLoadOfLValue(E);
Douglas Gregor34f6c6d2011-08-09 00:37:14 +0000643 }
Eric Fiselier708afb52019-05-16 21:04:15 +0000644 Value *VisitSourceLocExpr(SourceLocExpr *SLE) {
645 auto &Ctx = CGF.getContext();
646 APValue Evaluated =
647 SLE->EvaluateInContext(Ctx, CGF.CurSourceLocExprScope.getDefaultExpr());
Johannes Altmanninger1ac700c2019-11-15 02:12:58 +0100648 return ConstantEmitter(CGF).emitAbstract(SLE->getLocation(), Evaluated,
649 SLE->getType());
Eric Fiselier708afb52019-05-16 21:04:15 +0000650 }
Craig Toppera97d7e72013-07-26 06:16:11 +0000651
Chris Lattneraa9c7ae2008-04-08 04:40:51 +0000652 Value *VisitCXXDefaultArgExpr(CXXDefaultArgExpr *DAE) {
Eric Fiselier708afb52019-05-16 21:04:15 +0000653 CodeGenFunction::CXXDefaultArgExprScope Scope(CGF, DAE);
Chris Lattneraa9c7ae2008-04-08 04:40:51 +0000654 return Visit(DAE->getExpr());
655 }
Richard Smith852c9db2013-04-20 22:23:05 +0000656 Value *VisitCXXDefaultInitExpr(CXXDefaultInitExpr *DIE) {
Eric Fiselier708afb52019-05-16 21:04:15 +0000657 CodeGenFunction::CXXDefaultInitExprScope Scope(CGF, DIE);
Richard Smith852c9db2013-04-20 22:23:05 +0000658 return Visit(DIE->getExpr());
659 }
Anders Carlssona5d077d2009-04-14 16:58:56 +0000660 Value *VisitCXXThisExpr(CXXThisExpr *TE) {
661 return CGF.LoadCXXThis();
Mike Stump4a3999f2009-09-09 13:00:44 +0000662 }
663
Reid Kleckner092d0652017-03-06 22:18:34 +0000664 Value *VisitExprWithCleanups(ExprWithCleanups *E);
Anders Carlsson4a7b49b2009-05-31 01:40:14 +0000665 Value *VisitCXXNewExpr(const CXXNewExpr *E) {
666 return CGF.EmitCXXNewExpr(E);
667 }
Anders Carlsson81f0df92009-08-16 21:13:42 +0000668 Value *VisitCXXDeleteExpr(const CXXDeleteExpr *E) {
669 CGF.EmitCXXDeleteExpr(E);
Craig Topper8a13c412014-05-21 05:09:00 +0000670 return nullptr;
Anders Carlsson81f0df92009-08-16 21:13:42 +0000671 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000672
Alp Tokercbb90342013-12-13 20:49:58 +0000673 Value *VisitTypeTraitExpr(const TypeTraitExpr *E) {
Francois Pichet34b21132010-12-08 22:35:30 +0000674 return llvm::ConstantInt::get(ConvertType(E->getType()), E->getValue());
Francois Pichet9dfa3ce2010-12-07 00:08:36 +0000675 }
676
Saar Raz5d98ba62019-10-15 15:24:26 +0000677 Value *VisitConceptSpecializationExpr(const ConceptSpecializationExpr *E) {
678 return Builder.getInt1(E->isSatisfied());
679 }
680
Saar Raza0f50d72020-01-18 09:11:43 +0200681 Value *VisitRequiresExpr(const RequiresExpr *E) {
682 return Builder.getInt1(E->isSatisfied());
683 }
684
John Wiegley6242b6a2011-04-28 00:16:57 +0000685 Value *VisitArrayTypeTraitExpr(const ArrayTypeTraitExpr *E) {
686 return llvm::ConstantInt::get(Builder.getInt32Ty(), E->getValue());
687 }
688
John Wiegleyf9f65842011-04-25 06:54:41 +0000689 Value *VisitExpressionTraitExpr(const ExpressionTraitExpr *E) {
690 return llvm::ConstantInt::get(Builder.getInt1Ty(), E->getValue());
691 }
692
Douglas Gregorad8a3362009-09-04 17:36:40 +0000693 Value *VisitCXXPseudoDestructorExpr(const CXXPseudoDestructorExpr *E) {
694 // C++ [expr.pseudo]p1:
Mike Stump4a3999f2009-09-09 13:00:44 +0000695 // The result shall only be used as the operand for the function call
Douglas Gregorad8a3362009-09-04 17:36:40 +0000696 // operator (), and the result of such a call has type void. The only
697 // effect is the evaluation of the postfix-expression before the dot or
698 // arrow.
699 CGF.EmitScalarExpr(E->getBase());
Craig Topper8a13c412014-05-21 05:09:00 +0000700 return nullptr;
Douglas Gregorad8a3362009-09-04 17:36:40 +0000701 }
Mike Stump4a3999f2009-09-09 13:00:44 +0000702
Anders Carlsson04c3bf42009-09-15 04:39:46 +0000703 Value *VisitCXXNullPtrLiteralExpr(const CXXNullPtrLiteralExpr *E) {
Anders Carlsson5b944432010-05-22 17:45:10 +0000704 return EmitNullValue(E->getType());
Anders Carlsson04c3bf42009-09-15 04:39:46 +0000705 }
Anders Carlsson4b08db72009-10-30 01:42:31 +0000706
707 Value *VisitCXXThrowExpr(const CXXThrowExpr *E) {
708 CGF.EmitCXXThrowExpr(E);
Craig Topper8a13c412014-05-21 05:09:00 +0000709 return nullptr;
Anders Carlsson4b08db72009-10-30 01:42:31 +0000710 }
711
Sebastian Redlb67655f2010-09-10 21:04:00 +0000712 Value *VisitCXXNoexceptExpr(const CXXNoexceptExpr *E) {
Chris Lattner2531eb42011-04-19 22:55:03 +0000713 return Builder.getInt1(E->getValue());
Sebastian Redlb67655f2010-09-10 21:04:00 +0000714 }
715
Chris Lattner2da04b32007-08-24 05:35:26 +0000716 // Binary Operators.
Chris Lattner2da04b32007-08-24 05:35:26 +0000717 Value *EmitMul(const BinOpInfo &Ops) {
Douglas Gregor6ab2fa82011-05-20 16:38:50 +0000718 if (Ops.Ty->isSignedIntegerOrEnumerationType()) {
Richard Smith9c6890a2012-11-01 22:30:59 +0000719 switch (CGF.getLangOpts().getSignedOverflowBehavior()) {
Chris Lattner51924e512010-06-26 21:25:03 +0000720 case LangOptions::SOB_Defined:
721 return Builder.CreateMul(Ops.LHS, Ops.RHS, "mul");
Richard Smith3e056de2012-08-25 00:32:28 +0000722 case LangOptions::SOB_Undefined:
Alexey Samsonovedf99a92014-11-07 22:29:38 +0000723 if (!CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow))
Richard Smith3e056de2012-08-25 00:32:28 +0000724 return Builder.CreateNSWMul(Ops.LHS, Ops.RHS, "mul");
Reid Kleckner4dc0b1a2018-11-01 19:54:45 +0000725 LLVM_FALLTHROUGH;
Chris Lattner51924e512010-06-26 21:25:03 +0000726 case LangOptions::SOB_Trapping:
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000727 if (CanElideOverflowCheck(CGF.getContext(), Ops))
728 return Builder.CreateNSWMul(Ops.LHS, Ops.RHS, "mul");
Chris Lattner51924e512010-06-26 21:25:03 +0000729 return EmitOverflowCheckedBinOp(Ops);
730 }
731 }
Richard Smithb1b0ab42012-11-05 22:21:05 +0000732
Florian Hahn4affc442020-06-07 11:11:27 +0100733 if (Ops.Ty->isConstantMatrixType()) {
734 llvm::MatrixBuilder<CGBuilderTy> MB(Builder);
735 // We need to check the types of the operands of the operator to get the
736 // correct matrix dimensions.
737 auto *BO = cast<BinaryOperator>(Ops.E);
738 auto *LHSMatTy = dyn_cast<ConstantMatrixType>(
739 BO->getLHS()->getType().getCanonicalType());
740 auto *RHSMatTy = dyn_cast<ConstantMatrixType>(
741 BO->getRHS()->getType().getCanonicalType());
742 if (LHSMatTy && RHSMatTy)
743 return MB.CreateMatrixMultiply(Ops.LHS, Ops.RHS, LHSMatTy->getNumRows(),
744 LHSMatTy->getNumColumns(),
745 RHSMatTy->getNumColumns());
746 return MB.CreateScalarMultiply(Ops.LHS, Ops.RHS);
747 }
748
Alexey Samsonovedf99a92014-11-07 22:29:38 +0000749 if (Ops.Ty->isUnsignedIntegerType() &&
Vedant Kumar82ee16b2017-02-25 00:43:36 +0000750 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow) &&
751 !CanElideOverflowCheck(CGF.getContext(), Ops))
Will Dietz1897cb32012-11-27 15:01:55 +0000752 return EmitOverflowCheckedBinOp(Ops);
753
Adam Nemet370d0872017-04-04 21:18:30 +0000754 if (Ops.LHS->getType()->isFPOrFPVectorTy()) {
Melanie Blowerf5360d42020-05-01 10:32:06 -0700755 // Preserve the old values
John McCall7fac1ac2020-06-11 18:09:36 -0400756 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, Ops.FPFeatures);
John McCall8a8d7032020-06-01 21:02:02 -0400757 return Builder.CreateFMul(Ops.LHS, Ops.RHS, "mul");
Adam Nemet370d0872017-04-04 21:18:30 +0000758 }
Bevin Hansson39baaab2020-01-08 11:12:55 +0100759 if (Ops.isFixedPointOp())
Bevin Hansson0b9922e2019-11-19 13:15:06 +0100760 return EmitFixedPointBinOp(Ops);
Chris Lattner2da04b32007-08-24 05:35:26 +0000761 return Builder.CreateMul(Ops.LHS, Ops.RHS, "mul");
762 }
Mike Stump0c61b732009-04-01 20:28:16 +0000763 /// Create a binary op that checks for overflow.
764 /// Currently only supports +, - and *.
765 Value *EmitOverflowCheckedBinOp(const BinOpInfo &Ops);
Richard Smith4d1458e2012-09-08 02:08:36 +0000766
Chris Lattner8ee6a412010-09-11 21:47:09 +0000767 // Check for undefined division and modulus behaviors.
Craig Toppera97d7e72013-07-26 06:16:11 +0000768 void EmitUndefinedBehaviorIntegerDivAndRemCheck(const BinOpInfo &Ops,
Chris Lattner8ee6a412010-09-11 21:47:09 +0000769 llvm::Value *Zero,bool isDiv);
David Tweed042e0882013-01-07 16:43:27 +0000770 // Common helper for getting how wide LHS of shift is.
771 static Value *GetWidthMinusOneValue(Value* LHS,Value* RHS);
Erich Keane5f0903e2020-04-17 10:44:19 -0700772
773 // Used for shifting constraints for OpenCL, do mask for powers of 2, URem for
774 // non powers of two.
775 Value *ConstrainShiftValue(Value *LHS, Value *RHS, const Twine &Name);
776
Chris Lattner2da04b32007-08-24 05:35:26 +0000777 Value *EmitDiv(const BinOpInfo &Ops);
778 Value *EmitRem(const BinOpInfo &Ops);
779 Value *EmitAdd(const BinOpInfo &Ops);
780 Value *EmitSub(const BinOpInfo &Ops);
781 Value *EmitShl(const BinOpInfo &Ops);
782 Value *EmitShr(const BinOpInfo &Ops);
783 Value *EmitAnd(const BinOpInfo &Ops) {
784 return Builder.CreateAnd(Ops.LHS, Ops.RHS, "and");
785 }
786 Value *EmitXor(const BinOpInfo &Ops) {
787 return Builder.CreateXor(Ops.LHS, Ops.RHS, "xor");
788 }
789 Value *EmitOr (const BinOpInfo &Ops) {
790 return Builder.CreateOr(Ops.LHS, Ops.RHS, "or");
791 }
792
Leonard Chan2044ac82019-01-16 18:13:59 +0000793 // Helper functions for fixed point binary operations.
Leonard Chan837da5d2019-01-16 19:53:50 +0000794 Value *EmitFixedPointBinOp(const BinOpInfo &Ops);
Leonard Chan2044ac82019-01-16 18:13:59 +0000795
Chris Lattner3d966d62007-08-24 21:00:35 +0000796 BinOpInfo EmitBinOps(const BinaryOperator *E);
Douglas Gregor914af212010-04-23 04:16:32 +0000797 LValue EmitCompoundAssignLValue(const CompoundAssignOperator *E,
798 Value *(ScalarExprEmitter::*F)(const BinOpInfo &),
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +0000799 Value *&Result);
Douglas Gregor914af212010-04-23 04:16:32 +0000800
Chris Lattnerb6334692007-08-26 21:41:21 +0000801 Value *EmitCompoundAssign(const CompoundAssignOperator *E,
Chris Lattner3d966d62007-08-24 21:00:35 +0000802 Value *(ScalarExprEmitter::*F)(const BinOpInfo &));
803
804 // Binary operators and binary compound assignment operators.
805#define HANDLEBINOP(OP) \
Chris Lattnerb6334692007-08-26 21:41:21 +0000806 Value *VisitBin ## OP(const BinaryOperator *E) { \
807 return Emit ## OP(EmitBinOps(E)); \
808 } \
809 Value *VisitBin ## OP ## Assign(const CompoundAssignOperator *E) { \
810 return EmitCompoundAssign(E, &ScalarExprEmitter::Emit ## OP); \
Chris Lattner3d966d62007-08-24 21:00:35 +0000811 }
Daniel Dunbare017ecc2009-12-19 17:50:07 +0000812 HANDLEBINOP(Mul)
813 HANDLEBINOP(Div)
814 HANDLEBINOP(Rem)
815 HANDLEBINOP(Add)
816 HANDLEBINOP(Sub)
817 HANDLEBINOP(Shl)
818 HANDLEBINOP(Shr)
819 HANDLEBINOP(And)
820 HANDLEBINOP(Xor)
821 HANDLEBINOP(Or)
Chris Lattner3d966d62007-08-24 21:00:35 +0000822#undef HANDLEBINOP
Daniel Dunbarbfb1cd72008-08-06 02:00:38 +0000823
Chris Lattner2da04b32007-08-24 05:35:26 +0000824 // Comparisons.
Craig Topperc82f8962015-12-16 06:24:28 +0000825 Value *EmitCompare(const BinaryOperator *E, llvm::CmpInst::Predicate UICmpOpc,
826 llvm::CmpInst::Predicate SICmpOpc,
Ulrich Weigand76e9c2a2020-01-10 14:29:24 +0100827 llvm::CmpInst::Predicate FCmpOpc, bool IsSignaling);
828#define VISITCOMP(CODE, UI, SI, FP, SIG) \
Chris Lattner2da04b32007-08-24 05:35:26 +0000829 Value *VisitBin##CODE(const BinaryOperator *E) { \
830 return EmitCompare(E, llvm::ICmpInst::UI, llvm::ICmpInst::SI, \
Ulrich Weigand76e9c2a2020-01-10 14:29:24 +0100831 llvm::FCmpInst::FP, SIG); }
832 VISITCOMP(LT, ICMP_ULT, ICMP_SLT, FCMP_OLT, true)
833 VISITCOMP(GT, ICMP_UGT, ICMP_SGT, FCMP_OGT, true)
834 VISITCOMP(LE, ICMP_ULE, ICMP_SLE, FCMP_OLE, true)
835 VISITCOMP(GE, ICMP_UGE, ICMP_SGE, FCMP_OGE, true)
836 VISITCOMP(EQ, ICMP_EQ , ICMP_EQ , FCMP_OEQ, false)
837 VISITCOMP(NE, ICMP_NE , ICMP_NE , FCMP_UNE, false)
Chris Lattner2da04b32007-08-24 05:35:26 +0000838#undef VISITCOMP
Mike Stump4a3999f2009-09-09 13:00:44 +0000839
Chris Lattner2da04b32007-08-24 05:35:26 +0000840 Value *VisitBinAssign (const BinaryOperator *E);
841
842 Value *VisitBinLAnd (const BinaryOperator *E);
843 Value *VisitBinLOr (const BinaryOperator *E);
Chris Lattner2da04b32007-08-24 05:35:26 +0000844 Value *VisitBinComma (const BinaryOperator *E);
845
Eli Friedmanacfb1df2009-11-18 09:41:26 +0000846 Value *VisitBinPtrMemD(const Expr *E) { return EmitLoadOfLValue(E); }
847 Value *VisitBinPtrMemI(const Expr *E) { return EmitLoadOfLValue(E); }
848
Richard Smith778dc0f2019-10-19 00:04:38 +0000849 Value *VisitCXXRewrittenBinaryOperator(CXXRewrittenBinaryOperator *E) {
850 return Visit(E->getSemanticForm());
851 }
852
Chris Lattner2da04b32007-08-24 05:35:26 +0000853 // Other Operators.
Mike Stumpab3afd82009-02-12 18:29:15 +0000854 Value *VisitBlockExpr(const BlockExpr *BE);
John McCallc07a0c72011-02-17 10:25:35 +0000855 Value *VisitAbstractConditionalOperator(const AbstractConditionalOperator *);
Chris Lattner2da04b32007-08-24 05:35:26 +0000856 Value *VisitChooseExpr(ChooseExpr *CE);
Anders Carlsson7e13ab82007-10-15 20:28:48 +0000857 Value *VisitVAArgExpr(VAArgExpr *VE);
Chris Lattner2da04b32007-08-24 05:35:26 +0000858 Value *VisitObjCStringLiteral(const ObjCStringLiteral *E) {
859 return CGF.EmitObjCStringLiteral(E);
860 }
Patrick Beard0caa3942012-04-19 00:25:12 +0000861 Value *VisitObjCBoxedExpr(ObjCBoxedExpr *E) {
862 return CGF.EmitObjCBoxedExpr(E);
Ted Kremeneke65b0862012-03-06 20:05:56 +0000863 }
864 Value *VisitObjCArrayLiteral(ObjCArrayLiteral *E) {
865 return CGF.EmitObjCArrayLiteral(E);
866 }
867 Value *VisitObjCDictionaryLiteral(ObjCDictionaryLiteral *E) {
868 return CGF.EmitObjCDictionaryLiteral(E);
869 }
Tanya Lattner55808c12011-06-04 00:47:47 +0000870 Value *VisitAsTypeExpr(AsTypeExpr *CE);
Eli Friedmandf14b3a2011-10-11 02:20:01 +0000871 Value *VisitAtomicExpr(AtomicExpr *AE);
Chris Lattner2da04b32007-08-24 05:35:26 +0000872};
873} // end anonymous namespace.
874
875//===----------------------------------------------------------------------===//
876// Utilities
877//===----------------------------------------------------------------------===//
878
Chris Lattnere0044382007-08-26 16:42:57 +0000879/// EmitConversionToBool - Convert the specified expression value to a
Chris Lattner2e928882007-08-26 17:25:57 +0000880/// boolean (i1) truth value. This is equivalent to "Val != 0".
Chris Lattnere0044382007-08-26 16:42:57 +0000881Value *ScalarExprEmitter::EmitConversionToBool(Value *Src, QualType SrcType) {
John McCallb692a092009-10-22 20:10:53 +0000882 assert(SrcType.isCanonical() && "EmitScalarConversion strips typedefs");
Mike Stump4a3999f2009-09-09 13:00:44 +0000883
John McCall8cb679e2010-11-15 09:13:47 +0000884 if (SrcType->isRealFloatingType())
885 return EmitFloatToBoolConversion(Src);
Mike Stump4a3999f2009-09-09 13:00:44 +0000886
John McCall7a9aac22010-08-23 01:21:21 +0000887 if (const MemberPointerType *MPT = dyn_cast<MemberPointerType>(SrcType))
888 return CGF.CGM.getCXXABI().EmitMemberPointerIsNotNull(CGF, Src, MPT);
Mike Stump4a3999f2009-09-09 13:00:44 +0000889
Daniel Dunbaref957f32008-08-25 10:38:11 +0000890 assert((SrcType->isIntegerType() || isa<llvm::PointerType>(Src->getType())) &&
Chris Lattnere0044382007-08-26 16:42:57 +0000891 "Unknown scalar type to convert");
Mike Stump4a3999f2009-09-09 13:00:44 +0000892
John McCall8cb679e2010-11-15 09:13:47 +0000893 if (isa<llvm::IntegerType>(Src->getType()))
894 return EmitIntToBoolConversion(Src);
Mike Stump4a3999f2009-09-09 13:00:44 +0000895
John McCall8cb679e2010-11-15 09:13:47 +0000896 assert(isa<llvm::PointerType>(Src->getType()));
Yaxun Liu402804b2016-12-15 08:09:08 +0000897 return EmitPointerToBoolConversion(Src, SrcType);
Chris Lattnere0044382007-08-26 16:42:57 +0000898}
899
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +0000900void ScalarExprEmitter::EmitFloatConversionCheck(
901 Value *OrigSrc, QualType OrigSrcType, Value *Src, QualType SrcType,
902 QualType DstType, llvm::Type *DstTy, SourceLocation Loc) {
Richard Smith9e52c432019-07-06 21:05:52 +0000903 assert(SrcType->isFloatingType() && "not a conversion from floating point");
904 if (!isa<llvm::IntegerType>(DstTy))
905 return;
906
Alexey Samsonov24cad992014-07-17 18:46:27 +0000907 CodeGenFunction::SanitizerScope SanScope(&CGF);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000908 using llvm::APFloat;
909 using llvm::APSInt;
910
Craig Topper8a13c412014-05-21 05:09:00 +0000911 llvm::Value *Check = nullptr;
Richard Smith9e52c432019-07-06 21:05:52 +0000912 const llvm::fltSemantics &SrcSema =
913 CGF.getContext().getFloatTypeSemantics(OrigSrcType);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000914
Richard Smith9e52c432019-07-06 21:05:52 +0000915 // Floating-point to integer. This has undefined behavior if the source is
916 // +-Inf, NaN, or doesn't fit into the destination type (after truncation
917 // to an integer).
918 unsigned Width = CGF.getContext().getIntWidth(DstType);
919 bool Unsigned = DstType->isUnsignedIntegerOrEnumerationType();
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000920
Richard Smith9e52c432019-07-06 21:05:52 +0000921 APSInt Min = APSInt::getMinValue(Width, Unsigned);
922 APFloat MinSrc(SrcSema, APFloat::uninitialized);
923 if (MinSrc.convertFromAPInt(Min, !Unsigned, APFloat::rmTowardZero) &
924 APFloat::opOverflow)
925 // Don't need an overflow check for lower bound. Just check for
926 // -Inf/NaN.
927 MinSrc = APFloat::getInf(SrcSema, true);
928 else
929 // Find the largest value which is too small to represent (before
930 // truncation toward zero).
931 MinSrc.subtract(APFloat(SrcSema, 1), APFloat::rmTowardNegative);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000932
Richard Smith9e52c432019-07-06 21:05:52 +0000933 APSInt Max = APSInt::getMaxValue(Width, Unsigned);
934 APFloat MaxSrc(SrcSema, APFloat::uninitialized);
935 if (MaxSrc.convertFromAPInt(Max, !Unsigned, APFloat::rmTowardZero) &
936 APFloat::opOverflow)
937 // Don't need an overflow check for upper bound. Just check for
938 // +Inf/NaN.
939 MaxSrc = APFloat::getInf(SrcSema, false);
940 else
941 // Find the smallest value which is too large to represent (before
942 // truncation toward zero).
943 MaxSrc.add(APFloat(SrcSema, 1), APFloat::rmTowardPositive);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000944
Richard Smith9e52c432019-07-06 21:05:52 +0000945 // If we're converting from __half, convert the range to float to match
946 // the type of src.
947 if (OrigSrcType->isHalfType()) {
948 const llvm::fltSemantics &Sema =
949 CGF.getContext().getFloatTypeSemantics(SrcType);
950 bool IsInexact;
951 MinSrc.convert(Sema, APFloat::rmTowardZero, &IsInexact);
952 MaxSrc.convert(Sema, APFloat::rmTowardZero, &IsInexact);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000953 }
954
Richard Smith9e52c432019-07-06 21:05:52 +0000955 llvm::Value *GE =
956 Builder.CreateFCmpOGT(Src, llvm::ConstantFP::get(VMContext, MinSrc));
957 llvm::Value *LE =
958 Builder.CreateFCmpOLT(Src, llvm::ConstantFP::get(VMContext, MaxSrc));
959 Check = Builder.CreateAnd(GE, LE);
960
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +0000961 llvm::Constant *StaticArgs[] = {CGF.EmitCheckSourceLocation(Loc),
962 CGF.EmitCheckTypeDescriptor(OrigSrcType),
963 CGF.EmitCheckTypeDescriptor(DstType)};
Alexey Samsonove396bfc2014-11-11 22:03:54 +0000964 CGF.EmitCheck(std::make_pair(Check, SanitizerKind::FloatCastOverflow),
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +0000965 SanitizerHandler::FloatCastOverflow, StaticArgs, OrigSrc);
Richard Smithf9a1e4a2012-10-12 22:57:06 +0000966}
967
Roman Lebedev62debd802018-10-30 21:58:56 +0000968// Should be called within CodeGenFunction::SanitizerScope RAII scope.
969// Returns 'i1 false' when the truncation Src -> Dst was lossy.
970static std::pair<ScalarExprEmitter::ImplicitConversionCheckKind,
971 std::pair<llvm::Value *, SanitizerMask>>
972EmitIntegerTruncationCheckHelper(Value *Src, QualType SrcType, Value *Dst,
973 QualType DstType, CGBuilderTy &Builder) {
974 llvm::Type *SrcTy = Src->getType();
975 llvm::Type *DstTy = Dst->getType();
Richard Trieu161121f2018-10-30 23:01:15 +0000976 (void)DstTy; // Only used in assert()
Roman Lebedev62debd802018-10-30 21:58:56 +0000977
978 // This should be truncation of integral types.
979 assert(Src != Dst);
980 assert(SrcTy->getScalarSizeInBits() > Dst->getType()->getScalarSizeInBits());
981 assert(isa<llvm::IntegerType>(SrcTy) && isa<llvm::IntegerType>(DstTy) &&
982 "non-integer llvm type");
983
984 bool SrcSigned = SrcType->isSignedIntegerOrEnumerationType();
985 bool DstSigned = DstType->isSignedIntegerOrEnumerationType();
986
987 // If both (src and dst) types are unsigned, then it's an unsigned truncation.
988 // Else, it is a signed truncation.
989 ScalarExprEmitter::ImplicitConversionCheckKind Kind;
990 SanitizerMask Mask;
991 if (!SrcSigned && !DstSigned) {
992 Kind = ScalarExprEmitter::ICCK_UnsignedIntegerTruncation;
993 Mask = SanitizerKind::ImplicitUnsignedIntegerTruncation;
994 } else {
995 Kind = ScalarExprEmitter::ICCK_SignedIntegerTruncation;
996 Mask = SanitizerKind::ImplicitSignedIntegerTruncation;
997 }
998
999 llvm::Value *Check = nullptr;
1000 // 1. Extend the truncated value back to the same width as the Src.
1001 Check = Builder.CreateIntCast(Dst, SrcTy, DstSigned, "anyext");
1002 // 2. Equality-compare with the original source value
1003 Check = Builder.CreateICmpEQ(Check, Src, "truncheck");
1004 // If the comparison result is 'i1 false', then the truncation was lossy.
1005 return std::make_pair(Kind, std::make_pair(Check, Mask));
1006}
1007
Roman Lebedevb98a0c72019-11-27 17:07:06 +03001008static bool PromotionIsPotentiallyEligibleForImplicitIntegerConversionCheck(
1009 QualType SrcType, QualType DstType) {
1010 return SrcType->isIntegerType() && DstType->isIntegerType();
1011}
1012
Roman Lebedevb69ba222018-07-30 18:58:30 +00001013void ScalarExprEmitter::EmitIntegerTruncationCheck(Value *Src, QualType SrcType,
1014 Value *Dst, QualType DstType,
1015 SourceLocation Loc) {
Roman Lebedevdd403572018-10-11 09:09:50 +00001016 if (!CGF.SanOpts.hasOneOf(SanitizerKind::ImplicitIntegerTruncation))
Roman Lebedevb69ba222018-07-30 18:58:30 +00001017 return;
1018
Roman Lebedev62debd802018-10-30 21:58:56 +00001019 // We only care about int->int conversions here.
1020 // We ignore conversions to/from pointer and/or bool.
Roman Lebedevb98a0c72019-11-27 17:07:06 +03001021 if (!PromotionIsPotentiallyEligibleForImplicitIntegerConversionCheck(SrcType,
1022 DstType))
Roman Lebedev62debd802018-10-30 21:58:56 +00001023 return;
1024
1025 unsigned SrcBits = Src->getType()->getScalarSizeInBits();
1026 unsigned DstBits = Dst->getType()->getScalarSizeInBits();
1027 // This must be truncation. Else we do not care.
1028 if (SrcBits <= DstBits)
1029 return;
1030
1031 assert(!DstType->isBooleanType() && "we should not get here with booleans.");
1032
1033 // If the integer sign change sanitizer is enabled,
1034 // and we are truncating from larger unsigned type to smaller signed type,
1035 // let that next sanitizer deal with it.
1036 bool SrcSigned = SrcType->isSignedIntegerOrEnumerationType();
1037 bool DstSigned = DstType->isSignedIntegerOrEnumerationType();
1038 if (CGF.SanOpts.has(SanitizerKind::ImplicitIntegerSignChange) &&
1039 (!SrcSigned && DstSigned))
1040 return;
1041
1042 CodeGenFunction::SanitizerScope SanScope(&CGF);
1043
1044 std::pair<ScalarExprEmitter::ImplicitConversionCheckKind,
1045 std::pair<llvm::Value *, SanitizerMask>>
1046 Check =
1047 EmitIntegerTruncationCheckHelper(Src, SrcType, Dst, DstType, Builder);
1048 // If the comparison result is 'i1 false', then the truncation was lossy.
1049
1050 // Do we care about this type of truncation?
1051 if (!CGF.SanOpts.has(Check.second.second))
1052 return;
1053
1054 llvm::Constant *StaticArgs[] = {
1055 CGF.EmitCheckSourceLocation(Loc), CGF.EmitCheckTypeDescriptor(SrcType),
1056 CGF.EmitCheckTypeDescriptor(DstType),
1057 llvm::ConstantInt::get(Builder.getInt8Ty(), Check.first)};
1058 CGF.EmitCheck(Check.second, SanitizerHandler::ImplicitConversion, StaticArgs,
1059 {Src, Dst});
1060}
1061
1062// Should be called within CodeGenFunction::SanitizerScope RAII scope.
1063// Returns 'i1 false' when the conversion Src -> Dst changed the sign.
1064static std::pair<ScalarExprEmitter::ImplicitConversionCheckKind,
1065 std::pair<llvm::Value *, SanitizerMask>>
1066EmitIntegerSignChangeCheckHelper(Value *Src, QualType SrcType, Value *Dst,
1067 QualType DstType, CGBuilderTy &Builder) {
1068 llvm::Type *SrcTy = Src->getType();
1069 llvm::Type *DstTy = Dst->getType();
1070
1071 assert(isa<llvm::IntegerType>(SrcTy) && isa<llvm::IntegerType>(DstTy) &&
1072 "non-integer llvm type");
1073
1074 bool SrcSigned = SrcType->isSignedIntegerOrEnumerationType();
1075 bool DstSigned = DstType->isSignedIntegerOrEnumerationType();
Richard Trieu161121f2018-10-30 23:01:15 +00001076 (void)SrcSigned; // Only used in assert()
1077 (void)DstSigned; // Only used in assert()
Roman Lebedev62debd802018-10-30 21:58:56 +00001078 unsigned SrcBits = SrcTy->getScalarSizeInBits();
1079 unsigned DstBits = DstTy->getScalarSizeInBits();
1080 (void)SrcBits; // Only used in assert()
1081 (void)DstBits; // Only used in assert()
1082
1083 assert(((SrcBits != DstBits) || (SrcSigned != DstSigned)) &&
1084 "either the widths should be different, or the signednesses.");
1085
1086 // NOTE: zero value is considered to be non-negative.
1087 auto EmitIsNegativeTest = [&Builder](Value *V, QualType VType,
1088 const char *Name) -> Value * {
1089 // Is this value a signed type?
1090 bool VSigned = VType->isSignedIntegerOrEnumerationType();
1091 llvm::Type *VTy = V->getType();
1092 if (!VSigned) {
1093 // If the value is unsigned, then it is never negative.
1094 // FIXME: can we encounter non-scalar VTy here?
1095 return llvm::ConstantInt::getFalse(VTy->getContext());
1096 }
1097 // Get the zero of the same type with which we will be comparing.
1098 llvm::Constant *Zero = llvm::ConstantInt::get(VTy, 0);
1099 // %V.isnegative = icmp slt %V, 0
1100 // I.e is %V *strictly* less than zero, does it have negative value?
1101 return Builder.CreateICmp(llvm::ICmpInst::ICMP_SLT, V, Zero,
1102 llvm::Twine(Name) + "." + V->getName() +
1103 ".negativitycheck");
1104 };
1105
1106 // 1. Was the old Value negative?
1107 llvm::Value *SrcIsNegative = EmitIsNegativeTest(Src, SrcType, "src");
1108 // 2. Is the new Value negative?
1109 llvm::Value *DstIsNegative = EmitIsNegativeTest(Dst, DstType, "dst");
1110 // 3. Now, was the 'negativity status' preserved during the conversion?
1111 // NOTE: conversion from negative to zero is considered to change the sign.
1112 // (We want to get 'false' when the conversion changed the sign)
1113 // So we should just equality-compare the negativity statuses.
1114 llvm::Value *Check = nullptr;
1115 Check = Builder.CreateICmpEQ(SrcIsNegative, DstIsNegative, "signchangecheck");
1116 // If the comparison result is 'false', then the conversion changed the sign.
1117 return std::make_pair(
1118 ScalarExprEmitter::ICCK_IntegerSignChange,
1119 std::make_pair(Check, SanitizerKind::ImplicitIntegerSignChange));
1120}
1121
1122void ScalarExprEmitter::EmitIntegerSignChangeCheck(Value *Src, QualType SrcType,
1123 Value *Dst, QualType DstType,
1124 SourceLocation Loc) {
1125 if (!CGF.SanOpts.has(SanitizerKind::ImplicitIntegerSignChange))
1126 return;
1127
Roman Lebedevb69ba222018-07-30 18:58:30 +00001128 llvm::Type *SrcTy = Src->getType();
1129 llvm::Type *DstTy = Dst->getType();
1130
1131 // We only care about int->int conversions here.
1132 // We ignore conversions to/from pointer and/or bool.
Roman Lebedevb98a0c72019-11-27 17:07:06 +03001133 if (!PromotionIsPotentiallyEligibleForImplicitIntegerConversionCheck(SrcType,
1134 DstType))
Roman Lebedevb69ba222018-07-30 18:58:30 +00001135 return;
1136
Roman Lebedevdd403572018-10-11 09:09:50 +00001137 bool SrcSigned = SrcType->isSignedIntegerOrEnumerationType();
1138 bool DstSigned = DstType->isSignedIntegerOrEnumerationType();
Roman Lebedev62debd802018-10-30 21:58:56 +00001139 unsigned SrcBits = SrcTy->getScalarSizeInBits();
1140 unsigned DstBits = DstTy->getScalarSizeInBits();
Roman Lebedevdd403572018-10-11 09:09:50 +00001141
Roman Lebedev62debd802018-10-30 21:58:56 +00001142 // Now, we do not need to emit the check in *all* of the cases.
1143 // We can avoid emitting it in some obvious cases where it would have been
1144 // dropped by the opt passes (instcombine) always anyways.
Roman Lebedev1bb9aea2018-11-01 08:56:51 +00001145 // If it's a cast between effectively the same type, no check.
1146 // NOTE: this is *not* equivalent to checking the canonical types.
1147 if (SrcSigned == DstSigned && SrcBits == DstBits)
Roman Lebedevdd403572018-10-11 09:09:50 +00001148 return;
Roman Lebedev62debd802018-10-30 21:58:56 +00001149 // At least one of the values needs to have signed type.
1150 // If both are unsigned, then obviously, neither of them can be negative.
1151 if (!SrcSigned && !DstSigned)
1152 return;
1153 // If the conversion is to *larger* *signed* type, then no check is needed.
1154 // Because either sign-extension happens (so the sign will remain),
1155 // or zero-extension will happen (the sign bit will be zero.)
1156 if ((DstBits > SrcBits) && DstSigned)
1157 return;
1158 if (CGF.SanOpts.has(SanitizerKind::ImplicitSignedIntegerTruncation) &&
1159 (SrcBits > DstBits) && SrcSigned) {
1160 // If the signed integer truncation sanitizer is enabled,
1161 // and this is a truncation from signed type, then no check is needed.
1162 // Because here sign change check is interchangeable with truncation check.
1163 return;
1164 }
1165 // That's it. We can't rule out any more cases with the data we have.
Roman Lebedevdd403572018-10-11 09:09:50 +00001166
Roman Lebedevb69ba222018-07-30 18:58:30 +00001167 CodeGenFunction::SanitizerScope SanScope(&CGF);
1168
Roman Lebedev62debd802018-10-30 21:58:56 +00001169 std::pair<ScalarExprEmitter::ImplicitConversionCheckKind,
1170 std::pair<llvm::Value *, SanitizerMask>>
1171 Check;
Roman Lebedevb69ba222018-07-30 18:58:30 +00001172
Roman Lebedev62debd802018-10-30 21:58:56 +00001173 // Each of these checks needs to return 'false' when an issue was detected.
1174 ImplicitConversionCheckKind CheckKind;
1175 llvm::SmallVector<std::pair<llvm::Value *, SanitizerMask>, 2> Checks;
1176 // So we can 'and' all the checks together, and still get 'false',
1177 // if at least one of the checks detected an issue.
1178
1179 Check = EmitIntegerSignChangeCheckHelper(Src, SrcType, Dst, DstType, Builder);
1180 CheckKind = Check.first;
1181 Checks.emplace_back(Check.second);
1182
1183 if (CGF.SanOpts.has(SanitizerKind::ImplicitSignedIntegerTruncation) &&
1184 (SrcBits > DstBits) && !SrcSigned && DstSigned) {
1185 // If the signed integer truncation sanitizer was enabled,
1186 // and we are truncating from larger unsigned type to smaller signed type,
1187 // let's handle the case we skipped in that check.
1188 Check =
1189 EmitIntegerTruncationCheckHelper(Src, SrcType, Dst, DstType, Builder);
1190 CheckKind = ICCK_SignedIntegerTruncationOrSignChange;
1191 Checks.emplace_back(Check.second);
1192 // If the comparison result is 'i1 false', then the truncation was lossy.
1193 }
Roman Lebedevb69ba222018-07-30 18:58:30 +00001194
1195 llvm::Constant *StaticArgs[] = {
1196 CGF.EmitCheckSourceLocation(Loc), CGF.EmitCheckTypeDescriptor(SrcType),
1197 CGF.EmitCheckTypeDescriptor(DstType),
Roman Lebedev62debd802018-10-30 21:58:56 +00001198 llvm::ConstantInt::get(Builder.getInt8Ty(), CheckKind)};
1199 // EmitCheck() will 'and' all the checks together.
1200 CGF.EmitCheck(Checks, SanitizerHandler::ImplicitConversion, StaticArgs,
1201 {Src, Dst});
Roman Lebedevb69ba222018-07-30 18:58:30 +00001202}
1203
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +00001204/// Emit a conversion from the specified type to the specified destination type,
1205/// both of which are LLVM scalar types.
Chris Lattner42e6b812007-08-26 16:34:22 +00001206Value *ScalarExprEmitter::EmitScalarConversion(Value *Src, QualType SrcType,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001207 QualType DstType,
Anastasia Stulovab02e7832015-10-05 11:27:41 +00001208 SourceLocation Loc,
Roman Lebedevb69ba222018-07-30 18:58:30 +00001209 ScalarConversionOpts Opts) {
Leonard Chanb4ba4672018-10-23 17:55:35 +00001210 // All conversions involving fixed point types should be handled by the
1211 // EmitFixedPoint family functions. This is done to prevent bloating up this
1212 // function more, and although fixed point numbers are represented by
1213 // integers, we do not want to follow any logic that assumes they should be
1214 // treated as integers.
1215 // TODO(leonardchan): When necessary, add another if statement checking for
1216 // conversions to fixed point types from other types.
1217 if (SrcType->isFixedPointType()) {
Leonard Chan8f7caae2019-03-06 00:28:43 +00001218 if (DstType->isBooleanType())
1219 // It is important that we check this before checking if the dest type is
1220 // an integer because booleans are technically integer types.
Leonard Chanb4ba4672018-10-23 17:55:35 +00001221 // We do not need to check the padding bit on unsigned types if unsigned
1222 // padding is enabled because overflow into this bit is undefined
1223 // behavior.
Bjorn Petterssonb2534022018-10-26 16:12:12 +00001224 return Builder.CreateIsNotNull(Src, "tobool");
Leonard Chan8f7caae2019-03-06 00:28:43 +00001225 if (DstType->isFixedPointType() || DstType->isIntegerType())
1226 return EmitFixedPointConversion(Src, SrcType, DstType, Loc);
Leonard Chanb4ba4672018-10-23 17:55:35 +00001227
1228 llvm_unreachable(
Leonard Chan8f7caae2019-03-06 00:28:43 +00001229 "Unhandled scalar conversion from a fixed point type to another type.");
1230 } else if (DstType->isFixedPointType()) {
1231 if (SrcType->isIntegerType())
1232 // This also includes converting booleans and enums to fixed point types.
1233 return EmitFixedPointConversion(Src, SrcType, DstType, Loc);
1234
1235 llvm_unreachable(
1236 "Unhandled scalar conversion to a fixed point type from another type.");
Leonard Chanb4ba4672018-10-23 17:55:35 +00001237 }
Leonard Chan99bda372018-10-15 16:07:02 +00001238
Roman Lebedevb69ba222018-07-30 18:58:30 +00001239 QualType NoncanonicalSrcType = SrcType;
1240 QualType NoncanonicalDstType = DstType;
1241
Chris Lattner0f398c42008-07-26 22:37:01 +00001242 SrcType = CGF.getContext().getCanonicalType(SrcType);
1243 DstType = CGF.getContext().getCanonicalType(DstType);
Chris Lattner3474c202007-08-26 06:48:56 +00001244 if (SrcType == DstType) return Src;
Mike Stump4a3999f2009-09-09 13:00:44 +00001245
Craig Topper8a13c412014-05-21 05:09:00 +00001246 if (DstType->isVoidType()) return nullptr;
Mike Stump4a3999f2009-09-09 13:00:44 +00001247
Richard Smithf9a1e4a2012-10-12 22:57:06 +00001248 llvm::Value *OrigSrc = Src;
1249 QualType OrigSrcType = SrcType;
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001250 llvm::Type *SrcTy = Src->getType();
1251
Ahmed Bougacha47ec2c72015-03-23 17:48:07 +00001252 // Handle conversions to bool first, they are special: comparisons against 0.
1253 if (DstType->isBooleanType())
1254 return EmitConversionToBool(Src, SrcType);
1255
1256 llvm::Type *DstTy = ConvertType(DstType);
1257
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001258 // Cast from half through float if half isn't a native type.
1259 if (SrcType->isHalfType() && !CGF.getContext().getLangOpts().NativeHalfType) {
1260 // Cast to FP using the intrinsic if the half type itself isn't supported.
1261 if (DstTy->isFloatingPointTy()) {
Akira Hatanaka502775a2017-12-09 00:02:37 +00001262 if (CGF.getContext().getTargetInfo().useFP16ConversionIntrinsics())
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001263 return Builder.CreateCall(
1264 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_from_fp16, DstTy),
1265 Src);
1266 } else {
1267 // Cast to other types through float, using either the intrinsic or FPExt,
1268 // depending on whether the half type itself is supported
1269 // (as opposed to operations on half, available with NativeHalfType).
Akira Hatanaka502775a2017-12-09 00:02:37 +00001270 if (CGF.getContext().getTargetInfo().useFP16ConversionIntrinsics()) {
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001271 Src = Builder.CreateCall(
1272 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_from_fp16,
1273 CGF.CGM.FloatTy),
1274 Src);
1275 } else {
1276 Src = Builder.CreateFPExt(Src, CGF.CGM.FloatTy, "conv");
1277 }
1278 SrcType = CGF.getContext().FloatTy;
1279 SrcTy = CGF.FloatTy;
1280 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001281 }
1282
Chris Lattner3474c202007-08-26 06:48:56 +00001283 // Ignore conversions like int -> uint.
Roman Lebedev62debd802018-10-30 21:58:56 +00001284 if (SrcTy == DstTy) {
1285 if (Opts.EmitImplicitIntegerSignChangeChecks)
1286 EmitIntegerSignChangeCheck(Src, NoncanonicalSrcType, Src,
1287 NoncanonicalDstType, Loc);
1288
Chris Lattner3474c202007-08-26 06:48:56 +00001289 return Src;
Roman Lebedev62debd802018-10-30 21:58:56 +00001290 }
Chris Lattner3474c202007-08-26 06:48:56 +00001291
Mike Stump4a3999f2009-09-09 13:00:44 +00001292 // Handle pointer conversions next: pointers can only be converted to/from
1293 // other pointers and integers. Check for pointer types in terms of LLVM, as
1294 // some native types (like Obj-C id) may map to a pointer type.
Yaxun Liu26f75662016-08-19 05:17:25 +00001295 if (auto DstPT = dyn_cast<llvm::PointerType>(DstTy)) {
Chris Lattner3474c202007-08-26 06:48:56 +00001296 // The source value may be an integer, or a pointer.
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001297 if (isa<llvm::PointerType>(SrcTy))
Chris Lattner3474c202007-08-26 06:48:56 +00001298 return Builder.CreateBitCast(Src, DstTy, "conv");
Anders Carlsson12f5a252009-09-12 04:57:16 +00001299
Chris Lattner3474c202007-08-26 06:48:56 +00001300 assert(SrcType->isIntegerType() && "Not ptr->ptr or int->ptr conversion?");
Eli Friedman42d2a3a2009-03-04 04:02:35 +00001301 // First, convert to the correct width so that we control the kind of
1302 // extension.
Yaxun Liu26f75662016-08-19 05:17:25 +00001303 llvm::Type *MiddleTy = CGF.CGM.getDataLayout().getIntPtrType(DstPT);
Douglas Gregor6ab2fa82011-05-20 16:38:50 +00001304 bool InputSigned = SrcType->isSignedIntegerOrEnumerationType();
Eli Friedman42d2a3a2009-03-04 04:02:35 +00001305 llvm::Value* IntResult =
1306 Builder.CreateIntCast(Src, MiddleTy, InputSigned, "conv");
1307 // Then, cast to pointer.
1308 return Builder.CreateIntToPtr(IntResult, DstTy, "conv");
Chris Lattner3474c202007-08-26 06:48:56 +00001309 }
Mike Stump4a3999f2009-09-09 13:00:44 +00001310
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001311 if (isa<llvm::PointerType>(SrcTy)) {
Chris Lattner3474c202007-08-26 06:48:56 +00001312 // Must be an ptr to int cast.
1313 assert(isa<llvm::IntegerType>(DstTy) && "not ptr->int?");
Anders Carlssone89b84a2007-10-31 23:18:02 +00001314 return Builder.CreatePtrToInt(Src, DstTy, "conv");
Chris Lattner3474c202007-08-26 06:48:56 +00001315 }
Mike Stump4a3999f2009-09-09 13:00:44 +00001316
Nate Begemance4d7fc2008-04-18 23:10:10 +00001317 // A scalar can be splatted to an extended vector of the same element type
Nate Begeman5ec4b312009-08-10 23:49:36 +00001318 if (DstType->isExtVectorType() && !SrcType->isVectorType()) {
George Burgess IVdf1ed002016-01-13 01:52:39 +00001319 // Sema should add casts to make sure that the source expression's type is
1320 // the same as the vector's element type (sans qualifiers)
1321 assert(DstType->castAs<ExtVectorType>()->getElementType().getTypePtr() ==
1322 SrcType.getTypePtr() &&
1323 "Splatted expr doesn't match with vector element type?");
Nate Begemanb699c9b2009-01-18 06:42:49 +00001324
Nate Begemanb699c9b2009-01-18 06:42:49 +00001325 // Splat the element across to all elements
Christopher Tetreaultf22fbe32020-04-13 12:30:53 -07001326 unsigned NumElements = cast<llvm::VectorType>(DstTy)->getNumElements();
George Burgess IVdf1ed002016-01-13 01:52:39 +00001327 return Builder.CreateVectorSplat(NumElements, Src, "splat");
Nate Begemanb699c9b2009-01-18 06:42:49 +00001328 }
Nate Begeman330aaa72007-12-30 02:59:45 +00001329
Akira Hatanaka34b5dbc2017-09-23 05:02:02 +00001330 if (isa<llvm::VectorType>(SrcTy) || isa<llvm::VectorType>(DstTy)) {
1331 // Allow bitcast from vector to integer/fp of the same size.
1332 unsigned SrcSize = SrcTy->getPrimitiveSizeInBits();
1333 unsigned DstSize = DstTy->getPrimitiveSizeInBits();
1334 if (SrcSize == DstSize)
1335 return Builder.CreateBitCast(Src, DstTy, "conv");
1336
1337 // Conversions between vectors of different sizes are not allowed except
1338 // when vectors of half are involved. Operations on storage-only half
1339 // vectors require promoting half vector operands to float vectors and
1340 // truncating the result, which is either an int or float vector, to a
1341 // short or half vector.
1342
1343 // Source and destination are both expected to be vectors.
Christopher Tetreaultf22fbe32020-04-13 12:30:53 -07001344 llvm::Type *SrcElementTy = cast<llvm::VectorType>(SrcTy)->getElementType();
1345 llvm::Type *DstElementTy = cast<llvm::VectorType>(DstTy)->getElementType();
Benjamin Kramer5c42bcc2017-09-23 16:08:48 +00001346 (void)DstElementTy;
Akira Hatanaka34b5dbc2017-09-23 05:02:02 +00001347
1348 assert(((SrcElementTy->isIntegerTy() &&
1349 DstElementTy->isIntegerTy()) ||
1350 (SrcElementTy->isFloatingPointTy() &&
1351 DstElementTy->isFloatingPointTy())) &&
1352 "unexpected conversion between a floating-point vector and an "
1353 "integer vector");
1354
1355 // Truncate an i32 vector to an i16 vector.
1356 if (SrcElementTy->isIntegerTy())
1357 return Builder.CreateIntCast(Src, DstTy, false, "conv");
1358
1359 // Truncate a float vector to a half vector.
1360 if (SrcSize > DstSize)
1361 return Builder.CreateFPTrunc(Src, DstTy, "conv");
1362
1363 // Promote a half vector to a float vector.
1364 return Builder.CreateFPExt(Src, DstTy, "conv");
1365 }
Mike Stump4a3999f2009-09-09 13:00:44 +00001366
Chris Lattner3474c202007-08-26 06:48:56 +00001367 // Finally, we have the arithmetic types: real int/float.
Craig Topper8a13c412014-05-21 05:09:00 +00001368 Value *Res = nullptr;
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001369 llvm::Type *ResTy = DstTy;
1370
Richard Smithf9a1e4a2012-10-12 22:57:06 +00001371 // An overflowing conversion has undefined behavior if either the source type
Richard Smith9e52c432019-07-06 21:05:52 +00001372 // or the destination type is a floating-point type. However, we consider the
1373 // range of representable values for all floating-point types to be
1374 // [-inf,+inf], so no overflow can ever happen when the destination type is a
1375 // floating-point type.
Alexey Samsonovedf99a92014-11-07 22:29:38 +00001376 if (CGF.SanOpts.has(SanitizerKind::FloatCastOverflow) &&
Richard Smith9e52c432019-07-06 21:05:52 +00001377 OrigSrcType->isFloatingType())
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001378 EmitFloatConversionCheck(OrigSrc, OrigSrcType, Src, SrcType, DstType, DstTy,
1379 Loc);
Richard Smithf9a1e4a2012-10-12 22:57:06 +00001380
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001381 // Cast to half through float if half isn't a native type.
1382 if (DstType->isHalfType() && !CGF.getContext().getLangOpts().NativeHalfType) {
1383 // Make sure we cast in a single step if from another FP type.
1384 if (SrcTy->isFloatingPointTy()) {
1385 // Use the intrinsic if the half type itself isn't supported
1386 // (as opposed to operations on half, available with NativeHalfType).
Akira Hatanaka502775a2017-12-09 00:02:37 +00001387 if (CGF.getContext().getTargetInfo().useFP16ConversionIntrinsics())
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001388 return Builder.CreateCall(
1389 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_to_fp16, SrcTy), Src);
1390 // If the half type is supported, just use an fptrunc.
1391 return Builder.CreateFPTrunc(Src, DstTy);
1392 }
Chris Lattnerece04092012-02-07 00:39:47 +00001393 DstTy = CGF.FloatTy;
Ahmed Bougacha47ec2c72015-03-23 17:48:07 +00001394 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001395
1396 if (isa<llvm::IntegerType>(SrcTy)) {
Douglas Gregor6ab2fa82011-05-20 16:38:50 +00001397 bool InputSigned = SrcType->isSignedIntegerOrEnumerationType();
Roman Lebedevb69ba222018-07-30 18:58:30 +00001398 if (SrcType->isBooleanType() && Opts.TreatBooleanAsSigned) {
Anastasia Stulovab02e7832015-10-05 11:27:41 +00001399 InputSigned = true;
1400 }
Anders Carlssonc9d41e72007-12-26 18:20:19 +00001401 if (isa<llvm::IntegerType>(DstTy))
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001402 Res = Builder.CreateIntCast(Src, DstTy, InputSigned, "conv");
Anders Carlssonc9d41e72007-12-26 18:20:19 +00001403 else if (InputSigned)
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001404 Res = Builder.CreateSIToFP(Src, DstTy, "conv");
Anders Carlssonc9d41e72007-12-26 18:20:19 +00001405 else
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001406 Res = Builder.CreateUIToFP(Src, DstTy, "conv");
1407 } else if (isa<llvm::IntegerType>(DstTy)) {
1408 assert(SrcTy->isFloatingPointTy() && "Unknown real conversion");
Douglas Gregor6ab2fa82011-05-20 16:38:50 +00001409 if (DstType->isSignedIntegerOrEnumerationType())
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001410 Res = Builder.CreateFPToSI(Src, DstTy, "conv");
Anders Carlssonc9d41e72007-12-26 18:20:19 +00001411 else
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001412 Res = Builder.CreateFPToUI(Src, DstTy, "conv");
1413 } else {
1414 assert(SrcTy->isFloatingPointTy() && DstTy->isFloatingPointTy() &&
1415 "Unknown real conversion");
1416 if (DstTy->getTypeID() < SrcTy->getTypeID())
1417 Res = Builder.CreateFPTrunc(Src, DstTy, "conv");
1418 else
1419 Res = Builder.CreateFPExt(Src, DstTy, "conv");
Chris Lattner3474c202007-08-26 06:48:56 +00001420 }
1421
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001422 if (DstTy != ResTy) {
Akira Hatanaka502775a2017-12-09 00:02:37 +00001423 if (CGF.getContext().getTargetInfo().useFP16ConversionIntrinsics()) {
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001424 assert(ResTy->isIntegerTy(16) && "Only half FP requires extra conversion");
1425 Res = Builder.CreateCall(
Tim Northover6dbcbac2014-07-17 10:51:31 +00001426 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_to_fp16, CGF.CGM.FloatTy),
1427 Res);
Ahmed Bougachad1801af2015-03-23 17:54:16 +00001428 } else {
1429 Res = Builder.CreateFPTrunc(Res, ResTy, "conv");
1430 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001431 }
1432
Roman Lebedevb69ba222018-07-30 18:58:30 +00001433 if (Opts.EmitImplicitIntegerTruncationChecks)
1434 EmitIntegerTruncationCheck(Src, NoncanonicalSrcType, Res,
1435 NoncanonicalDstType, Loc);
1436
Roman Lebedev62debd802018-10-30 21:58:56 +00001437 if (Opts.EmitImplicitIntegerSignChangeChecks)
1438 EmitIntegerSignChangeCheck(Src, NoncanonicalSrcType, Res,
1439 NoncanonicalDstType, Loc);
1440
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00001441 return Res;
Chris Lattner3474c202007-08-26 06:48:56 +00001442}
1443
Leonard Chan99bda372018-10-15 16:07:02 +00001444Value *ScalarExprEmitter::EmitFixedPointConversion(Value *Src, QualType SrcTy,
1445 QualType DstTy,
1446 SourceLocation Loc) {
Leonard Chan99bda372018-10-15 16:07:02 +00001447 FixedPointSemantics SrcFPSema =
1448 CGF.getContext().getFixedPointSemantics(SrcTy);
1449 FixedPointSemantics DstFPSema =
1450 CGF.getContext().getFixedPointSemantics(DstTy);
Leonard Chan8f7caae2019-03-06 00:28:43 +00001451 return EmitFixedPointConversion(Src, SrcFPSema, DstFPSema, Loc,
1452 DstTy->isIntegerType());
Leonard Chan2044ac82019-01-16 18:13:59 +00001453}
1454
1455Value *ScalarExprEmitter::EmitFixedPointConversion(
1456 Value *Src, FixedPointSemantics &SrcFPSema, FixedPointSemantics &DstFPSema,
Leonard Chan8f7caae2019-03-06 00:28:43 +00001457 SourceLocation Loc, bool DstIsInteger) {
Leonard Chan2044ac82019-01-16 18:13:59 +00001458 using llvm::APInt;
1459 using llvm::ConstantInt;
1460 using llvm::Value;
1461
Leonard Chan99bda372018-10-15 16:07:02 +00001462 unsigned SrcWidth = SrcFPSema.getWidth();
1463 unsigned DstWidth = DstFPSema.getWidth();
1464 unsigned SrcScale = SrcFPSema.getScale();
1465 unsigned DstScale = DstFPSema.getScale();
Bjorn Petterssonb2534022018-10-26 16:12:12 +00001466 bool SrcIsSigned = SrcFPSema.isSigned();
1467 bool DstIsSigned = DstFPSema.isSigned();
1468
1469 llvm::Type *DstIntTy = Builder.getIntNTy(DstWidth);
Leonard Chan99bda372018-10-15 16:07:02 +00001470
1471 Value *Result = Src;
1472 unsigned ResultWidth = SrcWidth;
1473
Leonard Chan8f7caae2019-03-06 00:28:43 +00001474 // Downscale.
1475 if (DstScale < SrcScale) {
1476 // When converting to integers, we round towards zero. For negative numbers,
1477 // right shifting rounds towards negative infinity. In this case, we can
1478 // just round up before shifting.
1479 if (DstIsInteger && SrcIsSigned) {
1480 Value *Zero = llvm::Constant::getNullValue(Result->getType());
1481 Value *IsNegative = Builder.CreateICmpSLT(Result, Zero);
1482 Value *LowBits = ConstantInt::get(
1483 CGF.getLLVMContext(), APInt::getLowBitsSet(ResultWidth, SrcScale));
1484 Value *Rounded = Builder.CreateAdd(Result, LowBits);
1485 Result = Builder.CreateSelect(IsNegative, Rounded, Result);
1486 }
Leonard Chan99bda372018-10-15 16:07:02 +00001487
Leonard Chan8f7caae2019-03-06 00:28:43 +00001488 Result = SrcIsSigned
1489 ? Builder.CreateAShr(Result, SrcScale - DstScale, "downscale")
1490 : Builder.CreateLShr(Result, SrcScale - DstScale, "downscale");
1491 }
1492
1493 if (!DstFPSema.isSaturated()) {
Bjorn Petterssonb2534022018-10-26 16:12:12 +00001494 // Resize.
1495 Result = Builder.CreateIntCast(Result, DstIntTy, SrcIsSigned, "resize");
Leonard Chan99bda372018-10-15 16:07:02 +00001496
Bjorn Petterssonb2534022018-10-26 16:12:12 +00001497 // Upscale.
Leonard Chan99bda372018-10-15 16:07:02 +00001498 if (DstScale > SrcScale)
Bjorn Petterssonb2534022018-10-26 16:12:12 +00001499 Result = Builder.CreateShl(Result, DstScale - SrcScale, "upscale");
Leonard Chan99bda372018-10-15 16:07:02 +00001500 } else {
Bjorn Petterssonb2534022018-10-26 16:12:12 +00001501 // Adjust the number of fractional bits.
Leonard Chan99bda372018-10-15 16:07:02 +00001502 if (DstScale > SrcScale) {
Leonard Chan2044ac82019-01-16 18:13:59 +00001503 // Compare to DstWidth to prevent resizing twice.
1504 ResultWidth = std::max(SrcWidth + DstScale - SrcScale, DstWidth);
Leonard Chan99bda372018-10-15 16:07:02 +00001505 llvm::Type *UpscaledTy = Builder.getIntNTy(ResultWidth);
Bjorn Petterssonb2534022018-10-26 16:12:12 +00001506 Result = Builder.CreateIntCast(Result, UpscaledTy, SrcIsSigned, "resize");
1507 Result = Builder.CreateShl(Result, DstScale - SrcScale, "upscale");
Leonard Chan99bda372018-10-15 16:07:02 +00001508 }
1509
Bjorn Petterssonb2534022018-10-26 16:12:12 +00001510 // Handle saturation.
1511 bool LessIntBits = DstFPSema.getIntegralBits() < SrcFPSema.getIntegralBits();
1512 if (LessIntBits) {
1513 Value *Max = ConstantInt::get(
Leonard Chan99bda372018-10-15 16:07:02 +00001514 CGF.getLLVMContext(),
1515 APFixedPoint::getMax(DstFPSema).getValue().extOrTrunc(ResultWidth));
Bjorn Petterssonb2534022018-10-26 16:12:12 +00001516 Value *TooHigh = SrcIsSigned ? Builder.CreateICmpSGT(Result, Max)
1517 : Builder.CreateICmpUGT(Result, Max);
1518 Result = Builder.CreateSelect(TooHigh, Max, Result, "satmax");
1519 }
1520 // Cannot overflow min to dest type if src is unsigned since all fixed
1521 // point types can cover the unsigned min of 0.
1522 if (SrcIsSigned && (LessIntBits || !DstIsSigned)) {
1523 Value *Min = ConstantInt::get(
1524 CGF.getLLVMContext(),
1525 APFixedPoint::getMin(DstFPSema).getValue().extOrTrunc(ResultWidth));
1526 Value *TooLow = Builder.CreateICmpSLT(Result, Min);
1527 Result = Builder.CreateSelect(TooLow, Min, Result, "satmin");
Leonard Chan99bda372018-10-15 16:07:02 +00001528 }
1529
Bjorn Petterssonb2534022018-10-26 16:12:12 +00001530 // Resize the integer part to get the final destination size.
Leonard Chan2044ac82019-01-16 18:13:59 +00001531 if (ResultWidth != DstWidth)
1532 Result = Builder.CreateIntCast(Result, DstIntTy, SrcIsSigned, "resize");
Leonard Chan99bda372018-10-15 16:07:02 +00001533 }
1534 return Result;
1535}
1536
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +00001537/// Emit a conversion from the specified complex type to the specified
1538/// destination type, where the destination type is an LLVM scalar type.
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001539Value *ScalarExprEmitter::EmitComplexToScalarConversion(
1540 CodeGenFunction::ComplexPairTy Src, QualType SrcTy, QualType DstTy,
1541 SourceLocation Loc) {
Chris Lattnerc141c1b2007-08-26 16:52:28 +00001542 // Get the source element type.
John McCall47fb9502013-03-07 21:37:08 +00001543 SrcTy = SrcTy->castAs<ComplexType>()->getElementType();
Mike Stump4a3999f2009-09-09 13:00:44 +00001544
Chris Lattnerc141c1b2007-08-26 16:52:28 +00001545 // Handle conversions to bool first, they are special: comparisons against 0.
1546 if (DstTy->isBooleanType()) {
1547 // Complex != 0 -> (Real != 0) | (Imag != 0)
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001548 Src.first = EmitScalarConversion(Src.first, SrcTy, DstTy, Loc);
1549 Src.second = EmitScalarConversion(Src.second, SrcTy, DstTy, Loc);
Chris Lattnerc141c1b2007-08-26 16:52:28 +00001550 return Builder.CreateOr(Src.first, Src.second, "tobool");
1551 }
Mike Stump4a3999f2009-09-09 13:00:44 +00001552
Chris Lattner42e6b812007-08-26 16:34:22 +00001553 // C99 6.3.1.7p2: "When a value of complex type is converted to a real type,
1554 // the imaginary part of the complex value is discarded and the value of the
1555 // real part is converted according to the conversion rules for the
Mike Stump4a3999f2009-09-09 13:00:44 +00001556 // corresponding real type.
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001557 return EmitScalarConversion(Src.first, SrcTy, DstTy, Loc);
Chris Lattner42e6b812007-08-26 16:34:22 +00001558}
1559
Anders Carlsson5b944432010-05-22 17:45:10 +00001560Value *ScalarExprEmitter::EmitNullValue(QualType Ty) {
Richard Smithd82a2ce2012-12-21 03:17:28 +00001561 return CGF.EmitFromMemory(CGF.CGM.EmitNullConstant(Ty), Ty);
Anders Carlsson5b944432010-05-22 17:45:10 +00001562}
Chris Lattner42e6b812007-08-26 16:34:22 +00001563
Adrian Prantl9fc8faf2018-05-09 01:00:01 +00001564/// Emit a sanitization check for the given "binary" operation (which
Richard Smithe30752c2012-10-09 19:52:38 +00001565/// might actually be a unary increment which has been lowered to a binary
Alexey Samsonove396bfc2014-11-11 22:03:54 +00001566/// operation). The check passes if all values in \p Checks (which are \c i1),
1567/// are \c true.
1568void ScalarExprEmitter::EmitBinOpCheck(
Peter Collingbourne3eea6772015-05-11 21:39:14 +00001569 ArrayRef<std::pair<Value *, SanitizerMask>> Checks, const BinOpInfo &Info) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00001570 assert(CGF.IsSanitizerScope);
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001571 SanitizerHandler Check;
Dmitri Gribenkof8579502013-01-12 19:30:44 +00001572 SmallVector<llvm::Constant *, 4> StaticData;
1573 SmallVector<llvm::Value *, 2> DynamicData;
Richard Smithe30752c2012-10-09 19:52:38 +00001574
1575 BinaryOperatorKind Opcode = Info.Opcode;
1576 if (BinaryOperator::isCompoundAssignmentOp(Opcode))
1577 Opcode = BinaryOperator::getOpForCompoundAssignment(Opcode);
1578
1579 StaticData.push_back(CGF.EmitCheckSourceLocation(Info.E->getExprLoc()));
1580 const UnaryOperator *UO = dyn_cast<UnaryOperator>(Info.E);
1581 if (UO && UO->getOpcode() == UO_Minus) {
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001582 Check = SanitizerHandler::NegateOverflow;
Richard Smithe30752c2012-10-09 19:52:38 +00001583 StaticData.push_back(CGF.EmitCheckTypeDescriptor(UO->getType()));
1584 DynamicData.push_back(Info.RHS);
1585 } else {
1586 if (BinaryOperator::isShiftOp(Opcode)) {
1587 // Shift LHS negative or too large, or RHS out of bounds.
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001588 Check = SanitizerHandler::ShiftOutOfBounds;
Richard Smithe30752c2012-10-09 19:52:38 +00001589 const BinaryOperator *BO = cast<BinaryOperator>(Info.E);
1590 StaticData.push_back(
1591 CGF.EmitCheckTypeDescriptor(BO->getLHS()->getType()));
1592 StaticData.push_back(
1593 CGF.EmitCheckTypeDescriptor(BO->getRHS()->getType()));
1594 } else if (Opcode == BO_Div || Opcode == BO_Rem) {
1595 // Divide or modulo by zero, or signed overflow (eg INT_MAX / -1).
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001596 Check = SanitizerHandler::DivremOverflow;
Will Dietzcefb4482013-01-07 22:25:52 +00001597 StaticData.push_back(CGF.EmitCheckTypeDescriptor(Info.Ty));
Richard Smithe30752c2012-10-09 19:52:38 +00001598 } else {
Alexey Samsonov4c1a96f2014-11-10 22:27:30 +00001599 // Arithmetic overflow (+, -, *).
Richard Smithe30752c2012-10-09 19:52:38 +00001600 switch (Opcode) {
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001601 case BO_Add: Check = SanitizerHandler::AddOverflow; break;
1602 case BO_Sub: Check = SanitizerHandler::SubOverflow; break;
1603 case BO_Mul: Check = SanitizerHandler::MulOverflow; break;
Richard Smithe30752c2012-10-09 19:52:38 +00001604 default: llvm_unreachable("unexpected opcode for bin op check");
1605 }
Will Dietzcefb4482013-01-07 22:25:52 +00001606 StaticData.push_back(CGF.EmitCheckTypeDescriptor(Info.Ty));
Richard Smithe30752c2012-10-09 19:52:38 +00001607 }
1608 DynamicData.push_back(Info.LHS);
1609 DynamicData.push_back(Info.RHS);
1610 }
1611
Filipe Cabecinhas322ecd92016-12-12 16:18:40 +00001612 CGF.EmitCheck(Checks, Check, StaticData, DynamicData);
Richard Smithe30752c2012-10-09 19:52:38 +00001613}
1614
Chris Lattner2da04b32007-08-24 05:35:26 +00001615//===----------------------------------------------------------------------===//
1616// Visitor Methods
1617//===----------------------------------------------------------------------===//
1618
1619Value *ScalarExprEmitter::VisitExpr(Expr *E) {
Daniel Dunbara7c8cf62008-08-16 00:56:44 +00001620 CGF.ErrorUnsupported(E, "scalar expression");
Chris Lattner2da04b32007-08-24 05:35:26 +00001621 if (E->getType()->isVoidType())
Craig Topper8a13c412014-05-21 05:09:00 +00001622 return nullptr;
Owen Anderson7ec07a52009-07-30 23:11:26 +00001623 return llvm::UndefValue::get(CGF.ConvertType(E->getType()));
Chris Lattner2da04b32007-08-24 05:35:26 +00001624}
1625
Eli Friedmana1b4ed82008-05-14 19:38:39 +00001626Value *ScalarExprEmitter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
Nate Begemana0110022010-06-08 00:16:34 +00001627 // Vector Mask Case
Craig Topperb3174a82016-05-18 04:11:25 +00001628 if (E->getNumSubExprs() == 2) {
Chris Lattner5e016ae2010-06-27 07:15:29 +00001629 Value *LHS = CGF.EmitScalarExpr(E->getExpr(0));
1630 Value *RHS = CGF.EmitScalarExpr(E->getExpr(1));
1631 Value *Mask;
Craig Toppera97d7e72013-07-26 06:16:11 +00001632
Chris Lattner2192fe52011-07-18 04:24:23 +00001633 llvm::VectorType *LTy = cast<llvm::VectorType>(LHS->getType());
Nate Begemana0110022010-06-08 00:16:34 +00001634 unsigned LHSElts = LTy->getNumElements();
1635
Craig Topperb3174a82016-05-18 04:11:25 +00001636 Mask = RHS;
Craig Toppera97d7e72013-07-26 06:16:11 +00001637
Chris Lattner2192fe52011-07-18 04:24:23 +00001638 llvm::VectorType *MTy = cast<llvm::VectorType>(Mask->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00001639
Nate Begemana0110022010-06-08 00:16:34 +00001640 // Mask off the high bits of each shuffle index.
Benjamin Kramer99383102015-07-28 16:25:32 +00001641 Value *MaskBits =
1642 llvm::ConstantInt::get(MTy, llvm::NextPowerOf2(LHSElts - 1) - 1);
Nate Begemana0110022010-06-08 00:16:34 +00001643 Mask = Builder.CreateAnd(Mask, MaskBits, "mask");
Craig Toppera97d7e72013-07-26 06:16:11 +00001644
Nate Begemana0110022010-06-08 00:16:34 +00001645 // newv = undef
1646 // mask = mask & maskbits
1647 // for each elt
1648 // n = extract mask i
1649 // x = extract val n
1650 // newv = insert newv, x, i
Christopher Tetreault79689812020-06-01 09:55:24 -07001651 auto *RTy = llvm::FixedVectorType::get(LTy->getElementType(),
1652 MTy->getNumElements());
Nate Begemana0110022010-06-08 00:16:34 +00001653 Value* NewV = llvm::UndefValue::get(RTy);
1654 for (unsigned i = 0, e = MTy->getNumElements(); i != e; ++i) {
Michael J. Spencerdd597752014-05-31 00:22:12 +00001655 Value *IIndx = llvm::ConstantInt::get(CGF.SizeTy, i);
Eli Friedman1fa36052012-04-05 21:48:40 +00001656 Value *Indx = Builder.CreateExtractElement(Mask, IIndx, "shuf_idx");
Craig Toppera97d7e72013-07-26 06:16:11 +00001657
Nate Begemana0110022010-06-08 00:16:34 +00001658 Value *VExt = Builder.CreateExtractElement(LHS, Indx, "shuf_elt");
Eli Friedman1fa36052012-04-05 21:48:40 +00001659 NewV = Builder.CreateInsertElement(NewV, VExt, IIndx, "shuf_ins");
Nate Begemana0110022010-06-08 00:16:34 +00001660 }
1661 return NewV;
Eli Friedmana1b4ed82008-05-14 19:38:39 +00001662 }
Craig Toppera97d7e72013-07-26 06:16:11 +00001663
Eli Friedmana1b4ed82008-05-14 19:38:39 +00001664 Value* V1 = CGF.EmitScalarExpr(E->getExpr(0));
1665 Value* V2 = CGF.EmitScalarExpr(E->getExpr(1));
Craig Toppera97d7e72013-07-26 06:16:11 +00001666
Benjamin Kramer6f64dac2020-04-15 12:41:54 +02001667 SmallVector<int, 32> Indices;
Craig Topper0ed37bd2013-08-01 04:51:48 +00001668 for (unsigned i = 2; i < E->getNumSubExprs(); ++i) {
Craig Topper50ad5b72013-08-03 17:40:38 +00001669 llvm::APSInt Idx = E->getShuffleMaskIdx(CGF.getContext(), i-2);
1670 // Check for -1 and output it as undef in the IR.
1671 if (Idx.isSigned() && Idx.isAllOnesValue())
Benjamin Kramer6f64dac2020-04-15 12:41:54 +02001672 Indices.push_back(-1);
Craig Topper50ad5b72013-08-03 17:40:38 +00001673 else
Benjamin Kramer6f64dac2020-04-15 12:41:54 +02001674 Indices.push_back(Idx.getZExtValue());
Nate Begemana0110022010-06-08 00:16:34 +00001675 }
1676
Benjamin Kramer6f64dac2020-04-15 12:41:54 +02001677 return Builder.CreateShuffleVector(V1, V2, Indices, "shuffle");
Eli Friedmana1b4ed82008-05-14 19:38:39 +00001678}
Hal Finkelc4d7c822013-09-18 03:29:45 +00001679
1680Value *ScalarExprEmitter::VisitConvertVectorExpr(ConvertVectorExpr *E) {
1681 QualType SrcType = E->getSrcExpr()->getType(),
1682 DstType = E->getType();
1683
1684 Value *Src = CGF.EmitScalarExpr(E->getSrcExpr());
1685
1686 SrcType = CGF.getContext().getCanonicalType(SrcType);
1687 DstType = CGF.getContext().getCanonicalType(DstType);
1688 if (SrcType == DstType) return Src;
1689
1690 assert(SrcType->isVectorType() &&
1691 "ConvertVector source type must be a vector");
1692 assert(DstType->isVectorType() &&
1693 "ConvertVector destination type must be a vector");
1694
1695 llvm::Type *SrcTy = Src->getType();
1696 llvm::Type *DstTy = ConvertType(DstType);
1697
1698 // Ignore conversions like int -> uint.
1699 if (SrcTy == DstTy)
1700 return Src;
1701
Simon Pilgrime0712012019-10-02 15:31:25 +00001702 QualType SrcEltType = SrcType->castAs<VectorType>()->getElementType(),
1703 DstEltType = DstType->castAs<VectorType>()->getElementType();
Hal Finkelc4d7c822013-09-18 03:29:45 +00001704
1705 assert(SrcTy->isVectorTy() &&
1706 "ConvertVector source IR type must be a vector");
1707 assert(DstTy->isVectorTy() &&
1708 "ConvertVector destination IR type must be a vector");
1709
Christopher Tetreaultf22fbe32020-04-13 12:30:53 -07001710 llvm::Type *SrcEltTy = cast<llvm::VectorType>(SrcTy)->getElementType(),
1711 *DstEltTy = cast<llvm::VectorType>(DstTy)->getElementType();
Hal Finkelc4d7c822013-09-18 03:29:45 +00001712
1713 if (DstEltType->isBooleanType()) {
1714 assert((SrcEltTy->isFloatingPointTy() ||
1715 isa<llvm::IntegerType>(SrcEltTy)) && "Unknown boolean conversion");
1716
1717 llvm::Value *Zero = llvm::Constant::getNullValue(SrcTy);
1718 if (SrcEltTy->isFloatingPointTy()) {
1719 return Builder.CreateFCmpUNE(Src, Zero, "tobool");
1720 } else {
1721 return Builder.CreateICmpNE(Src, Zero, "tobool");
1722 }
1723 }
1724
1725 // We have the arithmetic types: real int/float.
Craig Topper8a13c412014-05-21 05:09:00 +00001726 Value *Res = nullptr;
Hal Finkelc4d7c822013-09-18 03:29:45 +00001727
1728 if (isa<llvm::IntegerType>(SrcEltTy)) {
1729 bool InputSigned = SrcEltType->isSignedIntegerOrEnumerationType();
1730 if (isa<llvm::IntegerType>(DstEltTy))
1731 Res = Builder.CreateIntCast(Src, DstTy, InputSigned, "conv");
1732 else if (InputSigned)
1733 Res = Builder.CreateSIToFP(Src, DstTy, "conv");
1734 else
1735 Res = Builder.CreateUIToFP(Src, DstTy, "conv");
1736 } else if (isa<llvm::IntegerType>(DstEltTy)) {
1737 assert(SrcEltTy->isFloatingPointTy() && "Unknown real conversion");
1738 if (DstEltType->isSignedIntegerOrEnumerationType())
1739 Res = Builder.CreateFPToSI(Src, DstTy, "conv");
1740 else
1741 Res = Builder.CreateFPToUI(Src, DstTy, "conv");
1742 } else {
1743 assert(SrcEltTy->isFloatingPointTy() && DstEltTy->isFloatingPointTy() &&
1744 "Unknown real conversion");
1745 if (DstEltTy->getTypeID() < SrcEltTy->getTypeID())
1746 Res = Builder.CreateFPTrunc(Src, DstTy, "conv");
1747 else
1748 Res = Builder.CreateFPExt(Src, DstTy, "conv");
1749 }
1750
1751 return Res;
1752}
1753
Eli Friedmancb422f12009-11-26 03:22:21 +00001754Value *ScalarExprEmitter::VisitMemberExpr(MemberExpr *E) {
Alex Lorenz6cc83172017-08-25 10:07:00 +00001755 if (CodeGenFunction::ConstantEmission Constant = CGF.tryEmitAsConstant(E)) {
1756 CGF.EmitIgnoredExpr(E->getBase());
Volodymyr Sapsaief1899b2018-11-01 21:57:05 +00001757 return CGF.emitScalarConstant(Constant, E);
Alex Lorenz6cc83172017-08-25 10:07:00 +00001758 } else {
Fangrui Song407659a2018-11-30 23:41:18 +00001759 Expr::EvalResult Result;
1760 if (E->EvaluateAsInt(Result, CGF.getContext(), Expr::SE_AllowSideEffects)) {
1761 llvm::APSInt Value = Result.Val.getInt();
Alex Lorenz6cc83172017-08-25 10:07:00 +00001762 CGF.EmitIgnoredExpr(E->getBase());
1763 return Builder.getInt(Value);
1764 }
Eli Friedmancb422f12009-11-26 03:22:21 +00001765 }
Devang Patel44b8bf02010-10-04 21:46:04 +00001766
Eli Friedmancb422f12009-11-26 03:22:21 +00001767 return EmitLoadOfLValue(E);
1768}
Eli Friedmana1b4ed82008-05-14 19:38:39 +00001769
Chris Lattner2da04b32007-08-24 05:35:26 +00001770Value *ScalarExprEmitter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00001771 TestAndClearIgnoreResultAssign();
1772
Chris Lattner2da04b32007-08-24 05:35:26 +00001773 // Emit subscript expressions in rvalue context's. For most cases, this just
1774 // loads the lvalue formed by the subscript expr. However, we have to be
1775 // careful, because the base of a vector subscript is occasionally an rvalue,
1776 // so we can't get it as an lvalue.
1777 if (!E->getBase()->getType()->isVectorType())
1778 return EmitLoadOfLValue(E);
Mike Stump4a3999f2009-09-09 13:00:44 +00001779
Chris Lattner2da04b32007-08-24 05:35:26 +00001780 // Handle the vector case. The base must be a vector, the index must be an
1781 // integer value.
1782 Value *Base = Visit(E->getBase());
1783 Value *Idx = Visit(E->getIdx());
Richard Smith539e4a72013-02-23 02:53:19 +00001784 QualType IdxTy = E->getIdx()->getType();
1785
Alexey Samsonovedf99a92014-11-07 22:29:38 +00001786 if (CGF.SanOpts.has(SanitizerKind::ArrayBounds))
Richard Smith539e4a72013-02-23 02:53:19 +00001787 CGF.EmitBoundsCheck(E, E->getBase(), Idx, IdxTy, /*Accessed*/true);
1788
Chris Lattner2da04b32007-08-24 05:35:26 +00001789 return Builder.CreateExtractElement(Base, Idx, "vecext");
1790}
1791
Florian Hahn8f3f88d2020-06-01 19:42:03 +01001792Value *ScalarExprEmitter::VisitMatrixSubscriptExpr(MatrixSubscriptExpr *E) {
1793 TestAndClearIgnoreResultAssign();
1794
1795 // Handle the vector case. The base must be a vector, the index must be an
1796 // integer value.
1797 Value *RowIdx = Visit(E->getRowIdx());
1798 Value *ColumnIdx = Visit(E->getColumnIdx());
1799 Value *Matrix = Visit(E->getBase());
1800
1801 // TODO: Should we emit bounds checks with SanitizerKind::ArrayBounds?
1802 llvm::MatrixBuilder<CGBuilderTy> MB(Builder);
1803 return MB.CreateExtractElement(
1804 Matrix, RowIdx, ColumnIdx,
1805 E->getBase()->getType()->getAs<ConstantMatrixType>()->getNumRows());
1806}
1807
Benjamin Kramerb6390912020-04-17 16:33:39 +02001808static int getMaskElt(llvm::ShuffleVectorInst *SVI, unsigned Idx,
1809 unsigned Off) {
Nate Begeman19351632009-10-18 20:10:40 +00001810 int MV = SVI->getMaskValue(Idx);
Craig Toppera97d7e72013-07-26 06:16:11 +00001811 if (MV == -1)
Benjamin Kramerb6390912020-04-17 16:33:39 +02001812 return -1;
1813 return Off + MV;
Nate Begeman19351632009-10-18 20:10:40 +00001814}
1815
Benjamin Kramerb6390912020-04-17 16:33:39 +02001816static int getAsInt32(llvm::ConstantInt *C, llvm::Type *I32Ty) {
1817 assert(llvm::ConstantInt::isValueValidForType(I32Ty, C->getZExtValue()) &&
1818 "Index operand too large for shufflevector mask!");
1819 return C->getZExtValue();
Simon Pilgrim4034b9f2015-08-02 15:28:10 +00001820}
1821
Nate Begeman19351632009-10-18 20:10:40 +00001822Value *ScalarExprEmitter::VisitInitListExpr(InitListExpr *E) {
1823 bool Ignore = TestAndClearIgnoreResultAssign();
1824 (void)Ignore;
1825 assert (Ignore == false && "init list ignored");
1826 unsigned NumInitElements = E->getNumInits();
Craig Toppera97d7e72013-07-26 06:16:11 +00001827
Nate Begeman19351632009-10-18 20:10:40 +00001828 if (E->hadArrayRangeDesignator())
1829 CGF.ErrorUnsupported(E, "GNU array range designator extension");
Craig Toppera97d7e72013-07-26 06:16:11 +00001830
Chris Lattner2192fe52011-07-18 04:24:23 +00001831 llvm::VectorType *VType =
Nate Begeman19351632009-10-18 20:10:40 +00001832 dyn_cast<llvm::VectorType>(ConvertType(E->getType()));
Craig Toppera97d7e72013-07-26 06:16:11 +00001833
Sebastian Redl12757ab2011-09-24 17:48:14 +00001834 if (!VType) {
1835 if (NumInitElements == 0) {
1836 // C++11 value-initialization for the scalar.
1837 return EmitNullValue(E->getType());
1838 }
1839 // We have a scalar in braces. Just use the first element.
Nate Begeman19351632009-10-18 20:10:40 +00001840 return Visit(E->getInit(0));
Sebastian Redl12757ab2011-09-24 17:48:14 +00001841 }
Craig Toppera97d7e72013-07-26 06:16:11 +00001842
Nate Begeman19351632009-10-18 20:10:40 +00001843 unsigned ResElts = VType->getNumElements();
Craig Toppera97d7e72013-07-26 06:16:11 +00001844
1845 // Loop over initializers collecting the Value for each, and remembering
Nate Begeman19351632009-10-18 20:10:40 +00001846 // whether the source was swizzle (ExtVectorElementExpr). This will allow
1847 // us to fold the shuffle for the swizzle into the shuffle for the vector
1848 // initializer, since LLVM optimizers generally do not want to touch
1849 // shuffles.
1850 unsigned CurIdx = 0;
1851 bool VIsUndefShuffle = false;
1852 llvm::Value *V = llvm::UndefValue::get(VType);
1853 for (unsigned i = 0; i != NumInitElements; ++i) {
1854 Expr *IE = E->getInit(i);
1855 Value *Init = Visit(IE);
Benjamin Kramerb6390912020-04-17 16:33:39 +02001856 SmallVector<int, 16> Args;
Craig Toppera97d7e72013-07-26 06:16:11 +00001857
Chris Lattner2192fe52011-07-18 04:24:23 +00001858 llvm::VectorType *VVT = dyn_cast<llvm::VectorType>(Init->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00001859
Nate Begeman19351632009-10-18 20:10:40 +00001860 // Handle scalar elements. If the scalar initializer is actually one
Craig Toppera97d7e72013-07-26 06:16:11 +00001861 // element of a different vector of the same width, use shuffle instead of
Nate Begeman19351632009-10-18 20:10:40 +00001862 // extract+insert.
1863 if (!VVT) {
1864 if (isa<ExtVectorElementExpr>(IE)) {
1865 llvm::ExtractElementInst *EI = cast<llvm::ExtractElementInst>(Init);
1866
1867 if (EI->getVectorOperandType()->getNumElements() == ResElts) {
1868 llvm::ConstantInt *C = cast<llvm::ConstantInt>(EI->getIndexOperand());
Craig Topper8a13c412014-05-21 05:09:00 +00001869 Value *LHS = nullptr, *RHS = nullptr;
Nate Begeman19351632009-10-18 20:10:40 +00001870 if (CurIdx == 0) {
1871 // insert into undef -> shuffle (src, undef)
Simon Pilgrim4034b9f2015-08-02 15:28:10 +00001872 // shufflemask must use an i32
1873 Args.push_back(getAsInt32(C, CGF.Int32Ty));
Benjamin Kramerb6390912020-04-17 16:33:39 +02001874 Args.resize(ResElts, -1);
Nate Begeman19351632009-10-18 20:10:40 +00001875
1876 LHS = EI->getVectorOperand();
1877 RHS = V;
1878 VIsUndefShuffle = true;
1879 } else if (VIsUndefShuffle) {
1880 // insert into undefshuffle && size match -> shuffle (v, src)
1881 llvm::ShuffleVectorInst *SVV = cast<llvm::ShuffleVectorInst>(V);
1882 for (unsigned j = 0; j != CurIdx; ++j)
Benjamin Kramerb6390912020-04-17 16:33:39 +02001883 Args.push_back(getMaskElt(SVV, j, 0));
1884 Args.push_back(ResElts + C->getZExtValue());
1885 Args.resize(ResElts, -1);
Benjamin Kramer8001f742012-02-14 12:06:21 +00001886
Nate Begeman19351632009-10-18 20:10:40 +00001887 LHS = cast<llvm::ShuffleVectorInst>(V)->getOperand(0);
1888 RHS = EI->getVectorOperand();
1889 VIsUndefShuffle = false;
1890 }
1891 if (!Args.empty()) {
Benjamin Kramerb6390912020-04-17 16:33:39 +02001892 V = Builder.CreateShuffleVector(LHS, RHS, Args);
Nate Begeman19351632009-10-18 20:10:40 +00001893 ++CurIdx;
1894 continue;
1895 }
1896 }
1897 }
Chris Lattner2531eb42011-04-19 22:55:03 +00001898 V = Builder.CreateInsertElement(V, Init, Builder.getInt32(CurIdx),
1899 "vecinit");
Nate Begeman19351632009-10-18 20:10:40 +00001900 VIsUndefShuffle = false;
1901 ++CurIdx;
1902 continue;
1903 }
Craig Toppera97d7e72013-07-26 06:16:11 +00001904
Nate Begeman19351632009-10-18 20:10:40 +00001905 unsigned InitElts = VVT->getNumElements();
1906
Craig Toppera97d7e72013-07-26 06:16:11 +00001907 // If the initializer is an ExtVecEltExpr (a swizzle), and the swizzle's
Nate Begeman19351632009-10-18 20:10:40 +00001908 // input is the same width as the vector being constructed, generate an
1909 // optimized shuffle of the swizzle input into the result.
Nate Begemanb8326be2009-10-25 02:26:01 +00001910 unsigned Offset = (CurIdx == 0) ? 0 : ResElts;
Nate Begeman19351632009-10-18 20:10:40 +00001911 if (isa<ExtVectorElementExpr>(IE)) {
1912 llvm::ShuffleVectorInst *SVI = cast<llvm::ShuffleVectorInst>(Init);
1913 Value *SVOp = SVI->getOperand(0);
Chris Lattner2192fe52011-07-18 04:24:23 +00001914 llvm::VectorType *OpTy = cast<llvm::VectorType>(SVOp->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00001915
Nate Begeman19351632009-10-18 20:10:40 +00001916 if (OpTy->getNumElements() == ResElts) {
Nate Begeman19351632009-10-18 20:10:40 +00001917 for (unsigned j = 0; j != CurIdx; ++j) {
1918 // If the current vector initializer is a shuffle with undef, merge
1919 // this shuffle directly into it.
1920 if (VIsUndefShuffle) {
Benjamin Kramerb6390912020-04-17 16:33:39 +02001921 Args.push_back(getMaskElt(cast<llvm::ShuffleVectorInst>(V), j, 0));
Nate Begeman19351632009-10-18 20:10:40 +00001922 } else {
Benjamin Kramerb6390912020-04-17 16:33:39 +02001923 Args.push_back(j);
Nate Begeman19351632009-10-18 20:10:40 +00001924 }
1925 }
1926 for (unsigned j = 0, je = InitElts; j != je; ++j)
Benjamin Kramerb6390912020-04-17 16:33:39 +02001927 Args.push_back(getMaskElt(SVI, j, Offset));
1928 Args.resize(ResElts, -1);
Nate Begeman19351632009-10-18 20:10:40 +00001929
1930 if (VIsUndefShuffle)
1931 V = cast<llvm::ShuffleVectorInst>(V)->getOperand(0);
1932
1933 Init = SVOp;
1934 }
1935 }
1936
1937 // Extend init to result vector length, and then shuffle its contribution
1938 // to the vector initializer into V.
1939 if (Args.empty()) {
1940 for (unsigned j = 0; j != InitElts; ++j)
Benjamin Kramerb6390912020-04-17 16:33:39 +02001941 Args.push_back(j);
1942 Args.resize(ResElts, -1);
1943 Init = Builder.CreateShuffleVector(Init, llvm::UndefValue::get(VVT), Args,
1944 "vext");
Nate Begeman19351632009-10-18 20:10:40 +00001945
1946 Args.clear();
1947 for (unsigned j = 0; j != CurIdx; ++j)
Benjamin Kramerb6390912020-04-17 16:33:39 +02001948 Args.push_back(j);
Nate Begeman19351632009-10-18 20:10:40 +00001949 for (unsigned j = 0; j != InitElts; ++j)
Benjamin Kramerb6390912020-04-17 16:33:39 +02001950 Args.push_back(j + Offset);
1951 Args.resize(ResElts, -1);
Nate Begeman19351632009-10-18 20:10:40 +00001952 }
1953
1954 // If V is undef, make sure it ends up on the RHS of the shuffle to aid
1955 // merging subsequent shuffles into this one.
1956 if (CurIdx == 0)
1957 std::swap(V, Init);
Benjamin Kramerb6390912020-04-17 16:33:39 +02001958 V = Builder.CreateShuffleVector(V, Init, Args, "vecinit");
Nate Begeman19351632009-10-18 20:10:40 +00001959 VIsUndefShuffle = isa<llvm::UndefValue>(Init);
1960 CurIdx += InitElts;
1961 }
Craig Toppera97d7e72013-07-26 06:16:11 +00001962
Nate Begeman19351632009-10-18 20:10:40 +00001963 // FIXME: evaluate codegen vs. shuffling against constant null vector.
1964 // Emit remaining default initializers.
Chris Lattner2192fe52011-07-18 04:24:23 +00001965 llvm::Type *EltTy = VType->getElementType();
Craig Toppera97d7e72013-07-26 06:16:11 +00001966
Nate Begeman19351632009-10-18 20:10:40 +00001967 // Emit remaining default initializers
1968 for (/* Do not initialize i*/; CurIdx < ResElts; ++CurIdx) {
Chris Lattner2531eb42011-04-19 22:55:03 +00001969 Value *Idx = Builder.getInt32(CurIdx);
Nate Begeman19351632009-10-18 20:10:40 +00001970 llvm::Value *Init = llvm::Constant::getNullValue(EltTy);
1971 V = Builder.CreateInsertElement(V, Init, Idx, "vecinit");
1972 }
1973 return V;
1974}
1975
John McCall7f416cc2015-09-08 08:05:57 +00001976bool CodeGenFunction::ShouldNullCheckClassCastValue(const CastExpr *CE) {
Anders Carlsson8c793172009-11-23 17:57:54 +00001977 const Expr *E = CE->getSubExpr();
John McCalld9c7c6562010-03-30 23:58:03 +00001978
John McCalle3027922010-08-25 11:45:40 +00001979 if (CE->getCastKind() == CK_UncheckedDerivedToBase)
John McCalld9c7c6562010-03-30 23:58:03 +00001980 return false;
Craig Toppera97d7e72013-07-26 06:16:11 +00001981
John McCall7f416cc2015-09-08 08:05:57 +00001982 if (isa<CXXThisExpr>(E->IgnoreParens())) {
Anders Carlsson8c793172009-11-23 17:57:54 +00001983 // We always assume that 'this' is never null.
1984 return false;
1985 }
Craig Toppera97d7e72013-07-26 06:16:11 +00001986
Anders Carlsson8c793172009-11-23 17:57:54 +00001987 if (const ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(CE)) {
Sebastian Redlc57d34b2010-07-20 04:20:21 +00001988 // And that glvalue casts are never null.
John McCall2536c6d2010-08-25 10:28:54 +00001989 if (ICE->getValueKind() != VK_RValue)
Anders Carlsson8c793172009-11-23 17:57:54 +00001990 return false;
1991 }
1992
1993 return true;
1994}
1995
Chris Lattner2da04b32007-08-24 05:35:26 +00001996// VisitCastExpr - Emit code for an explicit or implicit cast. Implicit casts
1997// have to handle a more broad range of conversions than explicit casts, as they
1998// handle things like function to ptr-to-function decay etc.
John McCall23c29fe2011-06-24 21:55:10 +00001999Value *ScalarExprEmitter::VisitCastExpr(CastExpr *CE) {
Eli Friedmane96f1d32009-11-27 04:41:50 +00002000 Expr *E = CE->getSubExpr();
Anders Carlsson8c978b42009-09-22 22:00:46 +00002001 QualType DestTy = CE->getType();
John McCalle3027922010-08-25 11:45:40 +00002002 CastKind Kind = CE->getCastKind();
Craig Toppera97d7e72013-07-26 06:16:11 +00002003
John McCalle399e5b2016-01-27 18:32:30 +00002004 // These cases are generally not written to ignore the result of
2005 // evaluating their sub-expressions, so we clear this now.
2006 bool Ignored = TestAndClearIgnoreResultAssign();
Mike Stump4a3999f2009-09-09 13:00:44 +00002007
Eli Friedman0dfc6802009-11-27 02:07:44 +00002008 // Since almost all cast kinds apply to scalars, this switch doesn't have
2009 // a default case, so the compiler will warn on a missing case. The cases
2010 // are in the same order as in the CastKind enum.
Anders Carlsson3df53bc2009-08-24 18:26:39 +00002011 switch (Kind) {
John McCall8cb679e2010-11-15 09:13:47 +00002012 case CK_Dependent: llvm_unreachable("dependent cast kind in IR gen!");
Eli Friedman34866c72012-08-31 00:14:07 +00002013 case CK_BuiltinFnToFnPtr:
2014 llvm_unreachable("builtin functions are handled elsewhere");
2015
Craig Toppera97d7e72013-07-26 06:16:11 +00002016 case CK_LValueBitCast:
John McCalle3027922010-08-25 11:45:40 +00002017 case CK_ObjCObjectLValueCast: {
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002018 Address Addr = EmitLValue(E).getAddress(CGF);
Alexey Bataevf2440332015-10-07 10:22:08 +00002019 Addr = Builder.CreateElementBitCast(Addr, CGF.ConvertTypeForMem(DestTy));
John McCall7f416cc2015-09-08 08:05:57 +00002020 LValue LV = CGF.MakeAddrLValue(Addr, DestTy);
2021 return EmitLoadOfLValue(LV, CE->getExprLoc());
Douglas Gregor51954272010-07-13 23:17:26 +00002022 }
John McCallcd78e802011-09-10 01:16:55 +00002023
Erik Pilkingtoneee944e2019-07-02 18:28:13 +00002024 case CK_LValueToRValueBitCast: {
2025 LValue SourceLVal = CGF.EmitLValue(E);
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002026 Address Addr = Builder.CreateElementBitCast(SourceLVal.getAddress(CGF),
Erik Pilkingtoneee944e2019-07-02 18:28:13 +00002027 CGF.ConvertTypeForMem(DestTy));
2028 LValue DestLV = CGF.MakeAddrLValue(Addr, DestTy);
2029 DestLV.setTBAAInfo(TBAAAccessInfo::getMayAliasInfo());
2030 return EmitLoadOfLValue(DestLV, CE->getExprLoc());
2031 }
2032
John McCall9320b872011-09-09 05:25:32 +00002033 case CK_CPointerToObjCPointerCast:
2034 case CK_BlockPointerToObjCPointerCast:
John McCalle3027922010-08-25 11:45:40 +00002035 case CK_AnyPointerToBlockPointerCast:
2036 case CK_BitCast: {
Anders Carlssonf1ae6d42009-09-01 20:52:42 +00002037 Value *Src = Visit(const_cast<Expr*>(E));
David Tweede1468322013-12-11 13:39:46 +00002038 llvm::Type *SrcTy = Src->getType();
2039 llvm::Type *DstTy = ConvertType(DestTy);
Bob Wilson95a27b02014-02-17 19:20:59 +00002040 if (SrcTy->isPtrOrPtrVectorTy() && DstTy->isPtrOrPtrVectorTy() &&
David Tweede1468322013-12-11 13:39:46 +00002041 SrcTy->getPointerAddressSpace() != DstTy->getPointerAddressSpace()) {
Anastasia Stulova5d8ad8a2014-11-26 15:36:41 +00002042 llvm_unreachable("wrong cast for pointers in different address spaces"
2043 "(must be an address space cast)!");
David Tweede1468322013-12-11 13:39:46 +00002044 }
Peter Collingbourned2926c92015-03-14 02:42:25 +00002045
2046 if (CGF.SanOpts.has(SanitizerKind::CFIUnrelatedCast)) {
2047 if (auto PT = DestTy->getAs<PointerType>())
2048 CGF.EmitVTablePtrCheckForCast(PT->getPointeeType(), Src,
Peter Collingbourne6708c4a2015-06-19 01:51:54 +00002049 /*MayBeNull=*/true,
2050 CodeGenFunction::CFITCK_UnrelatedCast,
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002051 CE->getBeginLoc());
Peter Collingbourned2926c92015-03-14 02:42:25 +00002052 }
2053
Piotr Padlewski07058292018-07-02 19:21:36 +00002054 if (CGF.CGM.getCodeGenOpts().StrictVTablePointers) {
2055 const QualType SrcType = E->getType();
2056
2057 if (SrcType.mayBeNotDynamicClass() && DestTy.mayBeDynamicClass()) {
2058 // Casting to pointer that could carry dynamic information (provided by
2059 // invariant.group) requires launder.
2060 Src = Builder.CreateLaunderInvariantGroup(Src);
2061 } else if (SrcType.mayBeDynamicClass() && DestTy.mayBeNotDynamicClass()) {
2062 // Casting to pointer that does not carry dynamic information (provided
2063 // by invariant.group) requires stripping it. Note that we don't do it
2064 // if the source could not be dynamic type and destination could be
2065 // dynamic because dynamic information is already laundered. It is
2066 // because launder(strip(src)) == launder(src), so there is no need to
2067 // add extra strip before launder.
2068 Src = Builder.CreateStripInvariantGroup(Src);
2069 }
2070 }
2071
Arthur Eubanksce7d3e12020-06-08 19:07:59 -07002072 // Update heapallocsite metadata when there is an explicit pointer cast.
2073 if (auto *CI = dyn_cast<llvm::CallBase>(Src)) {
2074 if (CI->getMetadata("heapallocsite") && isa<ExplicitCastExpr>(CE)) {
2075 QualType PointeeType = DestTy->getPointeeType();
2076 if (!PointeeType.isNull())
2077 CGF.getDebugInfo()->addHeapAllocSiteMetadata(CI, PointeeType,
2078 CE->getExprLoc());
2079 }
2080 }
Amy Huang301a5bb2019-05-02 20:07:35 +00002081
David Tweede1468322013-12-11 13:39:46 +00002082 return Builder.CreateBitCast(Src, DstTy);
2083 }
2084 case CK_AddressSpaceConversion: {
Yaxun Liu402804b2016-12-15 08:09:08 +00002085 Expr::EvalResult Result;
2086 if (E->EvaluateAsRValue(Result, CGF.getContext()) &&
2087 Result.Val.isNullPointer()) {
2088 // If E has side effect, it is emitted even if its final result is a
2089 // null pointer. In that case, a DCE pass should be able to
2090 // eliminate the useless instructions emitted during translating E.
2091 if (Result.HasSideEffects)
2092 Visit(E);
2093 return CGF.CGM.getNullPointer(cast<llvm::PointerType>(
2094 ConvertType(DestTy)), DestTy);
2095 }
Yaxun Liub7b6d0f2016-04-12 19:03:49 +00002096 // Since target may map different address spaces in AST to the same address
2097 // space, an address space conversion may end up as a bitcast.
Yaxun Liu6d96f1632017-05-18 18:51:09 +00002098 return CGF.CGM.getTargetCodeGenInfo().performAddrSpaceCast(
2099 CGF, Visit(E), E->getType()->getPointeeType().getAddressSpace(),
2100 DestTy->getPointeeType().getAddressSpace(), ConvertType(DestTy));
Anders Carlssonf1ae6d42009-09-01 20:52:42 +00002101 }
David Chisnallfa35df62012-01-16 17:27:18 +00002102 case CK_AtomicToNonAtomic:
2103 case CK_NonAtomicToAtomic:
John McCalle3027922010-08-25 11:45:40 +00002104 case CK_NoOp:
2105 case CK_UserDefinedConversion:
Eli Friedmanc08bdea2009-11-16 21:33:53 +00002106 return Visit(const_cast<Expr*>(E));
Mike Stump4a3999f2009-09-09 13:00:44 +00002107
John McCalle3027922010-08-25 11:45:40 +00002108 case CK_BaseToDerived: {
Jordan Rose7bb26112012-10-03 01:08:28 +00002109 const CXXRecordDecl *DerivedClassDecl = DestTy->getPointeeCXXRecordDecl();
2110 assert(DerivedClassDecl && "BaseToDerived arg isn't a C++ object pointer!");
2111
John McCall7f416cc2015-09-08 08:05:57 +00002112 Address Base = CGF.EmitPointerWithAlignment(E);
2113 Address Derived =
2114 CGF.GetAddressOfDerivedClass(Base, DerivedClassDecl,
Filipe Cabecinhas178a8df2013-08-08 01:08:17 +00002115 CE->path_begin(), CE->path_end(),
John McCall7f416cc2015-09-08 08:05:57 +00002116 CGF.ShouldNullCheckClassCastValue(CE));
Filipe Cabecinhas178a8df2013-08-08 01:08:17 +00002117
Richard Smith2c5868c2013-02-13 21:18:23 +00002118 // C++11 [expr.static.cast]p11: Behavior is undefined if a downcast is
2119 // performed and the object is not of the derived type.
Alexey Samsonovac4afe42014-07-07 23:59:57 +00002120 if (CGF.sanitizePerformTypeCheck())
Richard Smith2c5868c2013-02-13 21:18:23 +00002121 CGF.EmitTypeCheck(CodeGenFunction::TCK_DowncastPointer, CE->getExprLoc(),
John McCall7f416cc2015-09-08 08:05:57 +00002122 Derived.getPointer(), DestTy->getPointeeType());
Richard Smith2c5868c2013-02-13 21:18:23 +00002123
Peter Collingbourned2926c92015-03-14 02:42:25 +00002124 if (CGF.SanOpts.has(SanitizerKind::CFIDerivedCast))
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002125 CGF.EmitVTablePtrCheckForCast(
2126 DestTy->getPointeeType(), Derived.getPointer(),
2127 /*MayBeNull=*/true, CodeGenFunction::CFITCK_DerivedCast,
2128 CE->getBeginLoc());
Peter Collingbourned2926c92015-03-14 02:42:25 +00002129
John McCall7f416cc2015-09-08 08:05:57 +00002130 return Derived.getPointer();
Anders Carlsson8c793172009-11-23 17:57:54 +00002131 }
John McCalle3027922010-08-25 11:45:40 +00002132 case CK_UncheckedDerivedToBase:
2133 case CK_DerivedToBase: {
John McCall7f416cc2015-09-08 08:05:57 +00002134 // The EmitPointerWithAlignment path does this fine; just discard
2135 // the alignment.
2136 return CGF.EmitPointerWithAlignment(CE).getPointer();
Anders Carlsson12f5a252009-09-12 04:57:16 +00002137 }
John McCall7f416cc2015-09-08 08:05:57 +00002138
Anders Carlsson8a01a752011-04-11 02:03:26 +00002139 case CK_Dynamic: {
John McCall7f416cc2015-09-08 08:05:57 +00002140 Address V = CGF.EmitPointerWithAlignment(E);
Eli Friedman0dfc6802009-11-27 02:07:44 +00002141 const CXXDynamicCastExpr *DCE = cast<CXXDynamicCastExpr>(CE);
2142 return CGF.EmitDynamicCast(V, DCE);
2143 }
Eli Friedmane96f1d32009-11-27 04:41:50 +00002144
John McCall7f416cc2015-09-08 08:05:57 +00002145 case CK_ArrayToPointerDecay:
2146 return CGF.EmitArrayToPointerDecay(E).getPointer();
John McCalle3027922010-08-25 11:45:40 +00002147 case CK_FunctionToPointerDecay:
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002148 return EmitLValue(E).getPointer(CGF);
Eli Friedmanc08bdea2009-11-16 21:33:53 +00002149
John McCalle84af4e2010-11-13 01:35:44 +00002150 case CK_NullToPointer:
2151 if (MustVisitNullValue(E))
Richard Smith27252a12019-06-14 17:46:38 +00002152 CGF.EmitIgnoredExpr(E);
John McCalle84af4e2010-11-13 01:35:44 +00002153
Yaxun Liu402804b2016-12-15 08:09:08 +00002154 return CGF.CGM.getNullPointer(cast<llvm::PointerType>(ConvertType(DestTy)),
2155 DestTy);
John McCalle84af4e2010-11-13 01:35:44 +00002156
John McCalle3027922010-08-25 11:45:40 +00002157 case CK_NullToMemberPointer: {
John McCalle84af4e2010-11-13 01:35:44 +00002158 if (MustVisitNullValue(E))
Richard Smith27252a12019-06-14 17:46:38 +00002159 CGF.EmitIgnoredExpr(E);
John McCalla1dee5302010-08-22 10:59:02 +00002160
John McCall7a9aac22010-08-23 01:21:21 +00002161 const MemberPointerType *MPT = CE->getType()->getAs<MemberPointerType>();
2162 return CGF.CGM.getCXXABI().EmitNullMemberPointer(MPT);
2163 }
Anders Carlsson12f5a252009-09-12 04:57:16 +00002164
John McCallc62bb392012-02-15 01:22:51 +00002165 case CK_ReinterpretMemberPointer:
John McCalle3027922010-08-25 11:45:40 +00002166 case CK_BaseToDerivedMemberPointer:
2167 case CK_DerivedToBaseMemberPointer: {
Eli Friedmane96f1d32009-11-27 04:41:50 +00002168 Value *Src = Visit(E);
Craig Toppera97d7e72013-07-26 06:16:11 +00002169
John McCalla1dee5302010-08-22 10:59:02 +00002170 // Note that the AST doesn't distinguish between checked and
2171 // unchecked member pointer conversions, so we always have to
2172 // implement checked conversions here. This is inefficient when
2173 // actual control flow may be required in order to perform the
2174 // check, which it is for data member pointers (but not member
2175 // function pointers on Itanium and ARM).
John McCall7a9aac22010-08-23 01:21:21 +00002176 return CGF.CGM.getCXXABI().EmitMemberPointerConversion(CGF, CE, Src);
Eli Friedmane96f1d32009-11-27 04:41:50 +00002177 }
John McCall31168b02011-06-15 23:02:42 +00002178
John McCall2d637d22011-09-10 06:18:15 +00002179 case CK_ARCProduceObject:
John McCall31168b02011-06-15 23:02:42 +00002180 return CGF.EmitARCRetainScalarExpr(E);
John McCall2d637d22011-09-10 06:18:15 +00002181 case CK_ARCConsumeObject:
John McCall31168b02011-06-15 23:02:42 +00002182 return CGF.EmitObjCConsumeObject(E->getType(), Visit(E));
John McCalle399e5b2016-01-27 18:32:30 +00002183 case CK_ARCReclaimReturnedObject:
2184 return CGF.EmitARCReclaimReturnedObject(E, /*allowUnsafe*/ Ignored);
John McCallff613032011-10-04 06:23:45 +00002185 case CK_ARCExtendBlockObject:
2186 return CGF.EmitARCExtendBlockObject(E);
John McCall31168b02011-06-15 23:02:42 +00002187
Douglas Gregored90df32012-02-22 05:02:47 +00002188 case CK_CopyAndAutoreleaseBlockObject:
Eli Friedmanec75fec2012-02-28 01:08:45 +00002189 return CGF.EmitBlockCopyAndAutorelease(Visit(E), E->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00002190
John McCallc5e62b42010-11-13 09:02:35 +00002191 case CK_FloatingRealToComplex:
2192 case CK_FloatingComplexCast:
2193 case CK_IntegralRealToComplex:
2194 case CK_IntegralComplexCast:
John McCalld7646252010-11-14 08:17:51 +00002195 case CK_IntegralComplexToFloatingComplex:
2196 case CK_FloatingComplexToIntegralComplex:
John McCalle3027922010-08-25 11:45:40 +00002197 case CK_ConstructorConversion:
John McCall3eba6e62010-11-16 06:21:14 +00002198 case CK_ToUnion:
2199 llvm_unreachable("scalar cast to non-scalar value");
John McCall34376a62010-12-04 03:47:34 +00002200
John McCallf3735e02010-12-01 04:43:34 +00002201 case CK_LValueToRValue:
2202 assert(CGF.getContext().hasSameUnqualifiedType(E->getType(), DestTy));
John McCall34376a62010-12-04 03:47:34 +00002203 assert(E->isGLValue() && "lvalue-to-rvalue applied to r-value!");
John McCallf3735e02010-12-01 04:43:34 +00002204 return Visit(const_cast<Expr*>(E));
Eli Friedman0dfc6802009-11-27 02:07:44 +00002205
John McCalle3027922010-08-25 11:45:40 +00002206 case CK_IntegralToPointer: {
Anders Carlsson7cd39e02009-09-15 04:48:33 +00002207 Value *Src = Visit(const_cast<Expr*>(E));
Daniel Dunbaread6824c2010-08-25 03:32:38 +00002208
Anders Carlsson094c4592009-10-18 18:12:03 +00002209 // First, convert to the correct width so that we control the kind of
2210 // extension.
Yaxun Liu26f75662016-08-19 05:17:25 +00002211 auto DestLLVMTy = ConvertType(DestTy);
2212 llvm::Type *MiddleTy = CGF.CGM.getDataLayout().getIntPtrType(DestLLVMTy);
Douglas Gregor6ab2fa82011-05-20 16:38:50 +00002213 bool InputSigned = E->getType()->isSignedIntegerOrEnumerationType();
Anders Carlsson094c4592009-10-18 18:12:03 +00002214 llvm::Value* IntResult =
2215 Builder.CreateIntCast(Src, MiddleTy, InputSigned, "conv");
Daniel Dunbaread6824c2010-08-25 03:32:38 +00002216
Piotr Padlewski07058292018-07-02 19:21:36 +00002217 auto *IntToPtr = Builder.CreateIntToPtr(IntResult, DestLLVMTy);
Daniel Dunbaread6824c2010-08-25 03:32:38 +00002218
Piotr Padlewski07058292018-07-02 19:21:36 +00002219 if (CGF.CGM.getCodeGenOpts().StrictVTablePointers) {
2220 // Going from integer to pointer that could be dynamic requires reloading
2221 // dynamic information from invariant.group.
2222 if (DestTy.mayBeDynamicClass())
2223 IntToPtr = Builder.CreateLaunderInvariantGroup(IntToPtr);
2224 }
2225 return IntToPtr;
2226 }
2227 case CK_PointerToIntegral: {
2228 assert(!DestTy->isBooleanType() && "bool should use PointerToBool");
2229 auto *PtrExpr = Visit(E);
2230
2231 if (CGF.CGM.getCodeGenOpts().StrictVTablePointers) {
2232 const QualType SrcType = E->getType();
2233
2234 // Casting to integer requires stripping dynamic information as it does
2235 // not carries it.
2236 if (SrcType.mayBeDynamicClass())
2237 PtrExpr = Builder.CreateStripInvariantGroup(PtrExpr);
2238 }
2239
2240 return Builder.CreatePtrToInt(PtrExpr, ConvertType(DestTy));
2241 }
John McCalle3027922010-08-25 11:45:40 +00002242 case CK_ToVoid: {
John McCalla2342eb2010-12-05 02:00:02 +00002243 CGF.EmitIgnoredExpr(E);
Craig Topper8a13c412014-05-21 05:09:00 +00002244 return nullptr;
Eli Friedmanc08bdea2009-11-16 21:33:53 +00002245 }
John McCalle3027922010-08-25 11:45:40 +00002246 case CK_VectorSplat: {
Chris Lattner2192fe52011-07-18 04:24:23 +00002247 llvm::Type *DstTy = ConvertType(DestTy);
George Burgess IVdf1ed002016-01-13 01:52:39 +00002248 Value *Elt = Visit(const_cast<Expr*>(E));
Eli Friedmanc08bdea2009-11-16 21:33:53 +00002249 // Splat the element across to all elements
Christopher Tetreaultf22fbe32020-04-13 12:30:53 -07002250 unsigned NumElements = cast<llvm::VectorType>(DstTy)->getNumElements();
Alp Toker5f072d82014-04-19 23:55:49 +00002251 return Builder.CreateVectorSplat(NumElements, Elt, "splat");
Eli Friedmanc08bdea2009-11-16 21:33:53 +00002252 }
John McCall8cb679e2010-11-15 09:13:47 +00002253
Leonard Chan99bda372018-10-15 16:07:02 +00002254 case CK_FixedPointCast:
Leonard Chanb4ba4672018-10-23 17:55:35 +00002255 return EmitScalarConversion(Visit(E), E->getType(), DestTy,
2256 CE->getExprLoc());
2257
2258 case CK_FixedPointToBoolean:
2259 assert(E->getType()->isFixedPointType() &&
2260 "Expected src type to be fixed point type");
2261 assert(DestTy->isBooleanType() && "Expected dest type to be boolean type");
2262 return EmitScalarConversion(Visit(E), E->getType(), DestTy,
2263 CE->getExprLoc());
Leonard Chan99bda372018-10-15 16:07:02 +00002264
Leonard Chan8f7caae2019-03-06 00:28:43 +00002265 case CK_FixedPointToIntegral:
2266 assert(E->getType()->isFixedPointType() &&
2267 "Expected src type to be fixed point type");
2268 assert(DestTy->isIntegerType() && "Expected dest type to be an integer");
2269 return EmitScalarConversion(Visit(E), E->getType(), DestTy,
2270 CE->getExprLoc());
2271
2272 case CK_IntegralToFixedPoint:
2273 assert(E->getType()->isIntegerType() &&
2274 "Expected src type to be an integer");
2275 assert(DestTy->isFixedPointType() &&
2276 "Expected dest type to be fixed point type");
2277 return EmitScalarConversion(Visit(E), E->getType(), DestTy,
2278 CE->getExprLoc());
2279
Roman Lebedevb69ba222018-07-30 18:58:30 +00002280 case CK_IntegralCast: {
2281 ScalarConversionOpts Opts;
Roman Lebedev62debd802018-10-30 21:58:56 +00002282 if (auto *ICE = dyn_cast<ImplicitCastExpr>(CE)) {
Roman Lebedevd677c3f2018-11-19 19:56:43 +00002283 if (!ICE->isPartOfExplicitCast())
2284 Opts = ScalarConversionOpts(CGF.SanOpts);
Roman Lebedevb69ba222018-07-30 18:58:30 +00002285 }
2286 return EmitScalarConversion(Visit(E), E->getType(), DestTy,
2287 CE->getExprLoc(), Opts);
2288 }
John McCalle3027922010-08-25 11:45:40 +00002289 case CK_IntegralToFloating:
2290 case CK_FloatingToIntegral:
2291 case CK_FloatingCast:
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00002292 return EmitScalarConversion(Visit(E), E->getType(), DestTy,
2293 CE->getExprLoc());
Roman Lebedevb69ba222018-07-30 18:58:30 +00002294 case CK_BooleanToSignedIntegral: {
2295 ScalarConversionOpts Opts;
2296 Opts.TreatBooleanAsSigned = true;
George Burgess IVdf1ed002016-01-13 01:52:39 +00002297 return EmitScalarConversion(Visit(E), E->getType(), DestTy,
Roman Lebedevb69ba222018-07-30 18:58:30 +00002298 CE->getExprLoc(), Opts);
2299 }
John McCall8cb679e2010-11-15 09:13:47 +00002300 case CK_IntegralToBoolean:
2301 return EmitIntToBoolConversion(Visit(E));
2302 case CK_PointerToBoolean:
Yaxun Liu402804b2016-12-15 08:09:08 +00002303 return EmitPointerToBoolConversion(Visit(E), E->getType());
John McCall8cb679e2010-11-15 09:13:47 +00002304 case CK_FloatingToBoolean:
2305 return EmitFloatToBoolConversion(Visit(E));
John McCalle3027922010-08-25 11:45:40 +00002306 case CK_MemberPointerToBoolean: {
John McCall7a9aac22010-08-23 01:21:21 +00002307 llvm::Value *MemPtr = Visit(E);
2308 const MemberPointerType *MPT = E->getType()->getAs<MemberPointerType>();
2309 return CGF.CGM.getCXXABI().EmitMemberPointerIsNotNull(CGF, MemPtr, MPT);
Anders Carlsson3df53bc2009-08-24 18:26:39 +00002310 }
John McCalld7646252010-11-14 08:17:51 +00002311
2312 case CK_FloatingComplexToReal:
2313 case CK_IntegralComplexToReal:
John McCall07bb1962010-11-16 10:08:07 +00002314 return CGF.EmitComplexExpr(E, false, true).first;
John McCalld7646252010-11-14 08:17:51 +00002315
2316 case CK_FloatingComplexToBoolean:
2317 case CK_IntegralComplexToBoolean: {
John McCall07bb1962010-11-16 10:08:07 +00002318 CodeGenFunction::ComplexPairTy V = CGF.EmitComplexExpr(E);
John McCalld7646252010-11-14 08:17:51 +00002319
2320 // TODO: kill this function off, inline appropriate case here
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00002321 return EmitComplexToScalarConversion(V, E->getType(), DestTy,
2322 CE->getExprLoc());
John McCalld7646252010-11-14 08:17:51 +00002323 }
2324
Andrew Savonichevb555b762018-10-23 15:19:20 +00002325 case CK_ZeroToOCLOpaqueType: {
Andrew Savonichev3fee3512018-11-08 11:25:41 +00002326 assert((DestTy->isEventT() || DestTy->isQueueT() ||
2327 DestTy->isOCLIntelSubgroupAVCType()) &&
Andrew Savonichevb555b762018-10-23 15:19:20 +00002328 "CK_ZeroToOCLEvent cast on non-event type");
Egor Churaev89831422016-12-23 14:55:49 +00002329 return llvm::Constant::getNullValue(ConvertType(DestTy));
2330 }
2331
Yaxun Liu0bc4b2d2016-07-28 19:26:30 +00002332 case CK_IntToOCLSampler:
Sjoerd Meijercc623ad2017-09-08 15:15:00 +00002333 return CGF.CGM.createOpenCLIntToSamplerConversion(E, CGF);
Yaxun Liu0bc4b2d2016-07-28 19:26:30 +00002334
2335 } // end of switch
Mike Stump4a3999f2009-09-09 13:00:44 +00002336
John McCall3eba6e62010-11-16 06:21:14 +00002337 llvm_unreachable("unknown scalar cast");
Chris Lattner2da04b32007-08-24 05:35:26 +00002338}
2339
Chris Lattner04a913b2007-08-31 22:09:40 +00002340Value *ScalarExprEmitter::VisitStmtExpr(const StmtExpr *E) {
John McCallce1de612011-01-26 04:00:11 +00002341 CodeGenFunction::StmtExprEvaluation eval(CGF);
John McCall7f416cc2015-09-08 08:05:57 +00002342 Address RetAlloca = CGF.EmitCompoundStmt(*E->getSubStmt(),
2343 !E->getType()->isVoidType());
2344 if (!RetAlloca.isValid())
Craig Topper8a13c412014-05-21 05:09:00 +00002345 return nullptr;
Nick Lewycky2d84e842013-10-02 02:29:49 +00002346 return CGF.EmitLoadOfScalar(CGF.MakeAddrLValue(RetAlloca, E->getType()),
2347 E->getExprLoc());
Chris Lattner04a913b2007-08-31 22:09:40 +00002348}
2349
Reid Kleckner092d0652017-03-06 22:18:34 +00002350Value *ScalarExprEmitter::VisitExprWithCleanups(ExprWithCleanups *E) {
Reid Kleckner092d0652017-03-06 22:18:34 +00002351 CodeGenFunction::RunCleanupsScope Scope(CGF);
2352 Value *V = Visit(E->getSubExpr());
2353 // Defend against dominance problems caused by jumps out of expression
2354 // evaluation through the shared cleanup block.
2355 Scope.ForceCleanup({&V});
2356 return V;
2357}
2358
Chris Lattner2da04b32007-08-24 05:35:26 +00002359//===----------------------------------------------------------------------===//
2360// Unary Operators
2361//===----------------------------------------------------------------------===//
2362
Alexey Samsonovf6246502015-04-23 01:50:45 +00002363static BinOpInfo createBinOpInfoFromIncDec(const UnaryOperator *E,
Melanie Blowerf5360d42020-05-01 10:32:06 -07002364 llvm::Value *InVal, bool IsInc,
2365 FPOptions FPFeatures) {
Alexey Samsonovf6246502015-04-23 01:50:45 +00002366 BinOpInfo BinOp;
2367 BinOp.LHS = InVal;
2368 BinOp.RHS = llvm::ConstantInt::get(InVal->getType(), 1, false);
2369 BinOp.Ty = E->getType();
2370 BinOp.Opcode = IsInc ? BO_Add : BO_Sub;
Melanie Blowerf5360d42020-05-01 10:32:06 -07002371 BinOp.FPFeatures = FPFeatures;
Alexey Samsonovf6246502015-04-23 01:50:45 +00002372 BinOp.E = E;
2373 return BinOp;
2374}
2375
2376llvm::Value *ScalarExprEmitter::EmitIncDecConsiderOverflowBehavior(
2377 const UnaryOperator *E, llvm::Value *InVal, bool IsInc) {
2378 llvm::Value *Amount =
2379 llvm::ConstantInt::get(InVal->getType(), IsInc ? 1 : -1, true);
2380 StringRef Name = IsInc ? "inc" : "dec";
Richard Smith9c6890a2012-11-01 22:30:59 +00002381 switch (CGF.getLangOpts().getSignedOverflowBehavior()) {
Anton Yartsev85129b82011-02-07 02:17:30 +00002382 case LangOptions::SOB_Defined:
Alexey Samsonovf6246502015-04-23 01:50:45 +00002383 return Builder.CreateAdd(InVal, Amount, Name);
Richard Smith3e056de2012-08-25 00:32:28 +00002384 case LangOptions::SOB_Undefined:
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002385 if (!CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow))
Malcolm Parsonsfab36802018-04-16 08:31:08 +00002386 return Builder.CreateNSWAdd(InVal, Amount, Name);
Reid Kleckner4dc0b1a2018-11-01 19:54:45 +00002387 LLVM_FALLTHROUGH;
Malcolm Parsonsfab36802018-04-16 08:31:08 +00002388 case LangOptions::SOB_Trapping:
2389 if (!E->canOverflow())
2390 return Builder.CreateNSWAdd(InVal, Amount, Name);
Melanie Blowerf5360d42020-05-01 10:32:06 -07002391 return EmitOverflowCheckedBinOp(createBinOpInfoFromIncDec(
Melanie Blowerdefd43a2020-06-26 08:45:12 -07002392 E, InVal, IsInc, E->getFPFeatures(CGF.getLangOpts())));
Malcolm Parsonsfab36802018-04-16 08:31:08 +00002393 }
David Blaikie83d382b2011-09-23 05:06:16 +00002394 llvm_unreachable("Unknown SignedOverflowBehaviorTy");
Anton Yartsev85129b82011-02-07 02:17:30 +00002395}
2396
Alexey Bataev7b518dc2020-01-06 16:14:34 -05002397namespace {
2398/// Handles check and update for lastprivate conditional variables.
2399class OMPLastprivateConditionalUpdateRAII {
2400private:
2401 CodeGenFunction &CGF;
2402 const UnaryOperator *E;
2403
2404public:
2405 OMPLastprivateConditionalUpdateRAII(CodeGenFunction &CGF,
2406 const UnaryOperator *E)
2407 : CGF(CGF), E(E) {}
2408 ~OMPLastprivateConditionalUpdateRAII() {
2409 if (CGF.getLangOpts().OpenMP)
2410 CGF.CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(
2411 CGF, E->getSubExpr());
2412 }
2413};
2414} // namespace
2415
John McCalle3dc1702011-02-15 09:22:45 +00002416llvm::Value *
2417ScalarExprEmitter::EmitScalarPrePostIncDec(const UnaryOperator *E, LValue LV,
2418 bool isInc, bool isPre) {
Alexey Bataev7b518dc2020-01-06 16:14:34 -05002419 OMPLastprivateConditionalUpdateRAII OMPRegion(CGF, E);
John McCalle3dc1702011-02-15 09:22:45 +00002420 QualType type = E->getSubExpr()->getType();
Craig Topper8a13c412014-05-21 05:09:00 +00002421 llvm::PHINode *atomicPHI = nullptr;
David Chisnallef78c302013-03-03 16:02:42 +00002422 llvm::Value *value;
2423 llvm::Value *input;
Anton Yartsev85129b82011-02-07 02:17:30 +00002424
John McCalle3dc1702011-02-15 09:22:45 +00002425 int amount = (isInc ? 1 : -1);
Vedant Kumar175b6d12017-07-13 20:55:26 +00002426 bool isSubtraction = !isInc;
John McCalle3dc1702011-02-15 09:22:45 +00002427
David Chisnallfa35df62012-01-16 17:27:18 +00002428 if (const AtomicType *atomicTy = type->getAs<AtomicType>()) {
David Chisnallef78c302013-03-03 16:02:42 +00002429 type = atomicTy->getValueType();
2430 if (isInc && type->isBooleanType()) {
2431 llvm::Value *True = CGF.EmitToMemory(Builder.getTrue(), type);
2432 if (isPre) {
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002433 Builder.CreateStore(True, LV.getAddress(CGF), LV.isVolatileQualified())
2434 ->setAtomic(llvm::AtomicOrdering::SequentiallyConsistent);
David Chisnallef78c302013-03-03 16:02:42 +00002435 return Builder.getTrue();
2436 }
2437 // For atomic bool increment, we just store true and return it for
2438 // preincrement, do an atomic swap with true for postincrement
JF Bastien92f4ef12016-04-06 17:26:42 +00002439 return Builder.CreateAtomicRMW(
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002440 llvm::AtomicRMWInst::Xchg, LV.getPointer(CGF), True,
JF Bastien92f4ef12016-04-06 17:26:42 +00002441 llvm::AtomicOrdering::SequentiallyConsistent);
David Chisnallef78c302013-03-03 16:02:42 +00002442 }
2443 // Special case for atomic increment / decrement on integers, emit
2444 // atomicrmw instructions. We skip this if we want to be doing overflow
Craig Toppera97d7e72013-07-26 06:16:11 +00002445 // checking, and fall into the slow path with the atomic cmpxchg loop.
David Chisnallef78c302013-03-03 16:02:42 +00002446 if (!type->isBooleanType() && type->isIntegerType() &&
2447 !(type->isUnsignedIntegerType() &&
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002448 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow)) &&
David Chisnallef78c302013-03-03 16:02:42 +00002449 CGF.getLangOpts().getSignedOverflowBehavior() !=
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002450 LangOptions::SOB_Trapping) {
David Chisnallef78c302013-03-03 16:02:42 +00002451 llvm::AtomicRMWInst::BinOp aop = isInc ? llvm::AtomicRMWInst::Add :
2452 llvm::AtomicRMWInst::Sub;
2453 llvm::Instruction::BinaryOps op = isInc ? llvm::Instruction::Add :
2454 llvm::Instruction::Sub;
2455 llvm::Value *amt = CGF.EmitToMemory(
2456 llvm::ConstantInt::get(ConvertType(type), 1, true), type);
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002457 llvm::Value *old =
2458 Builder.CreateAtomicRMW(aop, LV.getPointer(CGF), amt,
2459 llvm::AtomicOrdering::SequentiallyConsistent);
David Chisnallef78c302013-03-03 16:02:42 +00002460 return isPre ? Builder.CreateBinOp(op, old, amt) : old;
2461 }
Nick Lewycky2d84e842013-10-02 02:29:49 +00002462 value = EmitLoadOfLValue(LV, E->getExprLoc());
David Chisnallef78c302013-03-03 16:02:42 +00002463 input = value;
2464 // For every other atomic operation, we need to emit a load-op-cmpxchg loop
David Chisnallfa35df62012-01-16 17:27:18 +00002465 llvm::BasicBlock *startBB = Builder.GetInsertBlock();
2466 llvm::BasicBlock *opBB = CGF.createBasicBlock("atomic_op", CGF.CurFn);
David Chisnallef78c302013-03-03 16:02:42 +00002467 value = CGF.EmitToMemory(value, type);
David Chisnallfa35df62012-01-16 17:27:18 +00002468 Builder.CreateBr(opBB);
2469 Builder.SetInsertPoint(opBB);
2470 atomicPHI = Builder.CreatePHI(value->getType(), 2);
2471 atomicPHI->addIncoming(value, startBB);
David Chisnallfa35df62012-01-16 17:27:18 +00002472 value = atomicPHI;
David Chisnallef78c302013-03-03 16:02:42 +00002473 } else {
Nick Lewycky2d84e842013-10-02 02:29:49 +00002474 value = EmitLoadOfLValue(LV, E->getExprLoc());
David Chisnallef78c302013-03-03 16:02:42 +00002475 input = value;
David Chisnallfa35df62012-01-16 17:27:18 +00002476 }
2477
John McCalle3dc1702011-02-15 09:22:45 +00002478 // Special case of integer increment that we have to check first: bool++.
2479 // Due to promotion rules, we get:
2480 // bool++ -> bool = bool + 1
2481 // -> bool = (int)bool + 1
2482 // -> bool = ((int)bool + 1 != 0)
2483 // An interesting aspect of this is that increment is always true.
2484 // Decrement does not have this property.
2485 if (isInc && type->isBooleanType()) {
2486 value = Builder.getTrue();
2487
2488 // Most common case by far: integer increment.
Malcolm Parsonsfab36802018-04-16 08:31:08 +00002489 } else if (type->isIntegerType()) {
Roman Lebedevb98a0c72019-11-27 17:07:06 +03002490 QualType promotedType;
2491 bool canPerformLossyDemotionCheck = false;
2492 if (type->isPromotableIntegerType()) {
2493 promotedType = CGF.getContext().getPromotedIntegerType(type);
2494 assert(promotedType != type && "Shouldn't promote to the same type.");
2495 canPerformLossyDemotionCheck = true;
2496 canPerformLossyDemotionCheck &=
2497 CGF.getContext().getCanonicalType(type) !=
2498 CGF.getContext().getCanonicalType(promotedType);
2499 canPerformLossyDemotionCheck &=
2500 PromotionIsPotentiallyEligibleForImplicitIntegerConversionCheck(
2501 type, promotedType);
2502 assert((!canPerformLossyDemotionCheck ||
2503 type->isSignedIntegerOrEnumerationType() ||
2504 promotedType->isSignedIntegerOrEnumerationType() ||
2505 ConvertType(type)->getScalarSizeInBits() ==
2506 ConvertType(promotedType)->getScalarSizeInBits()) &&
2507 "The following check expects that if we do promotion to different "
2508 "underlying canonical type, at least one of the types (either "
2509 "base or promoted) will be signed, or the bitwidths will match.");
2510 }
2511 if (CGF.SanOpts.hasOneOf(
2512 SanitizerKind::ImplicitIntegerArithmeticValueChange) &&
2513 canPerformLossyDemotionCheck) {
2514 // While `x += 1` (for `x` with width less than int) is modeled as
2515 // promotion+arithmetics+demotion, and we can catch lossy demotion with
2516 // ease; inc/dec with width less than int can't overflow because of
2517 // promotion rules, so we omit promotion+demotion, which means that we can
2518 // not catch lossy "demotion". Because we still want to catch these cases
2519 // when the sanitizer is enabled, we perform the promotion, then perform
2520 // the increment/decrement in the wider type, and finally
2521 // perform the demotion. This will catch lossy demotions.
2522
2523 value = EmitScalarConversion(value, type, promotedType, E->getExprLoc());
2524 Value *amt = llvm::ConstantInt::get(value->getType(), amount, true);
2525 value = Builder.CreateAdd(value, amt, isInc ? "inc" : "dec");
2526 // Do pass non-default ScalarConversionOpts so that sanitizer check is
2527 // emitted.
2528 value = EmitScalarConversion(value, promotedType, type, E->getExprLoc(),
2529 ScalarConversionOpts(CGF.SanOpts));
2530
2531 // Note that signed integer inc/dec with width less than int can't
2532 // overflow because of promotion rules; we're just eliding a few steps
2533 // here.
2534 } else if (E->canOverflow() && type->isSignedIntegerOrEnumerationType()) {
Malcolm Parsonsfab36802018-04-16 08:31:08 +00002535 value = EmitIncDecConsiderOverflowBehavior(E, value, isInc);
2536 } else if (E->canOverflow() && type->isUnsignedIntegerType() &&
2537 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow)) {
Melanie Blowerf5360d42020-05-01 10:32:06 -07002538 value = EmitOverflowCheckedBinOp(createBinOpInfoFromIncDec(
Melanie Blowerdefd43a2020-06-26 08:45:12 -07002539 E, value, isInc, E->getFPFeatures(CGF.getLangOpts())));
Alexey Samsonovf6246502015-04-23 01:50:45 +00002540 } else {
2541 llvm::Value *amt = llvm::ConstantInt::get(value->getType(), amount, true);
John McCalle3dc1702011-02-15 09:22:45 +00002542 value = Builder.CreateAdd(value, amt, isInc ? "inc" : "dec");
Alexey Samsonovf6246502015-04-23 01:50:45 +00002543 }
Craig Toppera97d7e72013-07-26 06:16:11 +00002544
John McCalle3dc1702011-02-15 09:22:45 +00002545 // Next most common: pointer increment.
2546 } else if (const PointerType *ptr = type->getAs<PointerType>()) {
2547 QualType type = ptr->getPointeeType();
2548
2549 // VLA types don't have constant size.
John McCall77527a82011-06-25 01:32:37 +00002550 if (const VariableArrayType *vla
2551 = CGF.getContext().getAsVariableArrayType(type)) {
Sander de Smalen891af03a2018-02-03 13:55:59 +00002552 llvm::Value *numElts = CGF.getVLASize(vla).NumElts;
John McCall23c29fe2011-06-24 21:55:10 +00002553 if (!isInc) numElts = Builder.CreateNSWNeg(numElts, "vla.negsize");
Richard Smith9c6890a2012-11-01 22:30:59 +00002554 if (CGF.getLangOpts().isSignedOverflowDefined())
John McCall23c29fe2011-06-24 21:55:10 +00002555 value = Builder.CreateGEP(value, numElts, "vla.inc");
Chris Lattner2e72da942011-03-01 00:03:48 +00002556 else
Vedant Kumar175b6d12017-07-13 20:55:26 +00002557 value = CGF.EmitCheckedInBoundsGEP(
2558 value, numElts, /*SignedIndices=*/false, isSubtraction,
2559 E->getExprLoc(), "vla.inc");
Craig Toppera97d7e72013-07-26 06:16:11 +00002560
John McCalle3dc1702011-02-15 09:22:45 +00002561 // Arithmetic on function pointers (!) is just +-1.
2562 } else if (type->isFunctionType()) {
Chris Lattner2531eb42011-04-19 22:55:03 +00002563 llvm::Value *amt = Builder.getInt32(amount);
John McCalle3dc1702011-02-15 09:22:45 +00002564
2565 value = CGF.EmitCastToVoidPtr(value);
Richard Smith9c6890a2012-11-01 22:30:59 +00002566 if (CGF.getLangOpts().isSignedOverflowDefined())
Chris Lattner2e72da942011-03-01 00:03:48 +00002567 value = Builder.CreateGEP(value, amt, "incdec.funcptr");
2568 else
Vedant Kumar175b6d12017-07-13 20:55:26 +00002569 value = CGF.EmitCheckedInBoundsGEP(value, amt, /*SignedIndices=*/false,
2570 isSubtraction, E->getExprLoc(),
2571 "incdec.funcptr");
John McCalle3dc1702011-02-15 09:22:45 +00002572 value = Builder.CreateBitCast(value, input->getType());
2573
2574 // For everything else, we can just do a simple increment.
Anton Yartsev85129b82011-02-07 02:17:30 +00002575 } else {
Chris Lattner2531eb42011-04-19 22:55:03 +00002576 llvm::Value *amt = Builder.getInt32(amount);
Richard Smith9c6890a2012-11-01 22:30:59 +00002577 if (CGF.getLangOpts().isSignedOverflowDefined())
Chris Lattner2e72da942011-03-01 00:03:48 +00002578 value = Builder.CreateGEP(value, amt, "incdec.ptr");
2579 else
Vedant Kumar175b6d12017-07-13 20:55:26 +00002580 value = CGF.EmitCheckedInBoundsGEP(value, amt, /*SignedIndices=*/false,
2581 isSubtraction, E->getExprLoc(),
2582 "incdec.ptr");
John McCalle3dc1702011-02-15 09:22:45 +00002583 }
2584
2585 // Vector increment/decrement.
2586 } else if (type->isVectorType()) {
2587 if (type->hasIntegerRepresentation()) {
2588 llvm::Value *amt = llvm::ConstantInt::get(value->getType(), amount);
2589
Eli Friedman409943e2011-05-06 18:04:18 +00002590 value = Builder.CreateAdd(value, amt, isInc ? "inc" : "dec");
John McCalle3dc1702011-02-15 09:22:45 +00002591 } else {
2592 value = Builder.CreateFAdd(
2593 value,
2594 llvm::ConstantFP::get(value->getType(), amount),
Anton Yartsev85129b82011-02-07 02:17:30 +00002595 isInc ? "inc" : "dec");
2596 }
Anton Yartsev85129b82011-02-07 02:17:30 +00002597
John McCalle3dc1702011-02-15 09:22:45 +00002598 // Floating point.
2599 } else if (type->isRealFloatingType()) {
Chris Lattner05dc78c2010-06-26 22:09:34 +00002600 // Add the inc/dec to the real part.
John McCalle3dc1702011-02-15 09:22:45 +00002601 llvm::Value *amt;
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00002602
Ahmed Bougachad1801af2015-03-23 17:54:16 +00002603 if (type->isHalfType() && !CGF.getContext().getLangOpts().NativeHalfType) {
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00002604 // Another special case: half FP increment should be done via float
Akira Hatanaka502775a2017-12-09 00:02:37 +00002605 if (CGF.getContext().getTargetInfo().useFP16ConversionIntrinsics()) {
Ahmed Bougachad1801af2015-03-23 17:54:16 +00002606 value = Builder.CreateCall(
2607 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_from_fp16,
2608 CGF.CGM.FloatTy),
2609 input, "incdec.conv");
2610 } else {
2611 value = Builder.CreateFPExt(input, CGF.CGM.FloatTy, "incdec.conv");
2612 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00002613 }
2614
John McCalle3dc1702011-02-15 09:22:45 +00002615 if (value->getType()->isFloatTy())
2616 amt = llvm::ConstantFP::get(VMContext,
2617 llvm::APFloat(static_cast<float>(amount)));
2618 else if (value->getType()->isDoubleTy())
2619 amt = llvm::ConstantFP::get(VMContext,
2620 llvm::APFloat(static_cast<double>(amount)));
Chris Lattner05dc78c2010-06-26 22:09:34 +00002621 else {
Nemanja Ivanovicbb1ea2d2016-05-09 08:52:33 +00002622 // Remaining types are Half, LongDouble or __float128. Convert from float.
John McCalle3dc1702011-02-15 09:22:45 +00002623 llvm::APFloat F(static_cast<float>(amount));
Chris Lattner05dc78c2010-06-26 22:09:34 +00002624 bool ignored;
Nemanja Ivanovicbb1ea2d2016-05-09 08:52:33 +00002625 const llvm::fltSemantics *FS;
Ahmed Bougacha6ba38312015-03-24 23:44:42 +00002626 // Don't use getFloatTypeSemantics because Half isn't
2627 // necessarily represented using the "half" LLVM type.
Nemanja Ivanovicbb1ea2d2016-05-09 08:52:33 +00002628 if (value->getType()->isFP128Ty())
2629 FS = &CGF.getTarget().getFloat128Format();
2630 else if (value->getType()->isHalfTy())
2631 FS = &CGF.getTarget().getHalfFormat();
2632 else
2633 FS = &CGF.getTarget().getLongDoubleFormat();
2634 F.convert(*FS, llvm::APFloat::rmTowardZero, &ignored);
John McCalle3dc1702011-02-15 09:22:45 +00002635 amt = llvm::ConstantFP::get(VMContext, F);
Chris Lattner05dc78c2010-06-26 22:09:34 +00002636 }
John McCalle3dc1702011-02-15 09:22:45 +00002637 value = Builder.CreateFAdd(value, amt, isInc ? "inc" : "dec");
2638
Ahmed Bougachad1801af2015-03-23 17:54:16 +00002639 if (type->isHalfType() && !CGF.getContext().getLangOpts().NativeHalfType) {
Akira Hatanaka502775a2017-12-09 00:02:37 +00002640 if (CGF.getContext().getTargetInfo().useFP16ConversionIntrinsics()) {
Ahmed Bougachad1801af2015-03-23 17:54:16 +00002641 value = Builder.CreateCall(
2642 CGF.CGM.getIntrinsic(llvm::Intrinsic::convert_to_fp16,
2643 CGF.CGM.FloatTy),
2644 value, "incdec.conv");
2645 } else {
2646 value = Builder.CreateFPTrunc(value, input->getType(), "incdec.conv");
2647 }
2648 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00002649
Bevin Hansson39baaab2020-01-08 11:12:55 +01002650 // Fixed-point types.
2651 } else if (type->isFixedPointType()) {
2652 // Fixed-point types are tricky. In some cases, it isn't possible to
2653 // represent a 1 or a -1 in the type at all. Piggyback off of
2654 // EmitFixedPointBinOp to avoid having to reimplement saturation.
2655 BinOpInfo Info;
2656 Info.E = E;
2657 Info.Ty = E->getType();
2658 Info.Opcode = isInc ? BO_Add : BO_Sub;
2659 Info.LHS = value;
2660 Info.RHS = llvm::ConstantInt::get(value->getType(), 1, false);
2661 // If the type is signed, it's better to represent this as +(-1) or -(-1),
2662 // since -1 is guaranteed to be representable.
2663 if (type->isSignedFixedPointType()) {
2664 Info.Opcode = isInc ? BO_Sub : BO_Add;
2665 Info.RHS = Builder.CreateNeg(Info.RHS);
2666 }
2667 // Now, convert from our invented integer literal to the type of the unary
2668 // op. This will upscale and saturate if necessary. This value can become
2669 // undef in some cases.
2670 FixedPointSemantics SrcSema =
2671 FixedPointSemantics::GetIntegerSemantics(value->getType()
2672 ->getScalarSizeInBits(),
2673 /*IsSigned=*/true);
2674 FixedPointSemantics DstSema =
2675 CGF.getContext().getFixedPointSemantics(Info.Ty);
2676 Info.RHS = EmitFixedPointConversion(Info.RHS, SrcSema, DstSema,
2677 E->getExprLoc());
2678 value = EmitFixedPointBinOp(Info);
2679
John McCalle3dc1702011-02-15 09:22:45 +00002680 // Objective-C pointer types.
2681 } else {
2682 const ObjCObjectPointerType *OPT = type->castAs<ObjCObjectPointerType>();
2683 value = CGF.EmitCastToVoidPtr(value);
2684
2685 CharUnits size = CGF.getContext().getTypeSizeInChars(OPT->getObjectType());
2686 if (!isInc) size = -size;
2687 llvm::Value *sizeValue =
2688 llvm::ConstantInt::get(CGF.SizeTy, size.getQuantity());
2689
Richard Smith9c6890a2012-11-01 22:30:59 +00002690 if (CGF.getLangOpts().isSignedOverflowDefined())
Chris Lattner2e72da942011-03-01 00:03:48 +00002691 value = Builder.CreateGEP(value, sizeValue, "incdec.objptr");
2692 else
Vedant Kumar175b6d12017-07-13 20:55:26 +00002693 value = CGF.EmitCheckedInBoundsGEP(value, sizeValue,
2694 /*SignedIndices=*/false, isSubtraction,
Vedant Kumar6dbf4272017-06-12 18:42:51 +00002695 E->getExprLoc(), "incdec.objptr");
John McCalle3dc1702011-02-15 09:22:45 +00002696 value = Builder.CreateBitCast(value, input->getType());
Chris Lattner05dc78c2010-06-26 22:09:34 +00002697 }
Craig Toppera97d7e72013-07-26 06:16:11 +00002698
David Chisnallfa35df62012-01-16 17:27:18 +00002699 if (atomicPHI) {
Erik Pilkington53e43f42019-02-28 00:47:55 +00002700 llvm::BasicBlock *curBlock = Builder.GetInsertBlock();
David Chisnallfa35df62012-01-16 17:27:18 +00002701 llvm::BasicBlock *contBB = CGF.createBasicBlock("atomic_cont", CGF.CurFn);
Alexey Bataev452d8e12014-12-15 05:25:25 +00002702 auto Pair = CGF.EmitAtomicCompareExchange(
Alexey Bataevb4505a72015-03-30 05:20:59 +00002703 LV, RValue::get(atomicPHI), RValue::get(value), E->getExprLoc());
2704 llvm::Value *old = CGF.EmitToMemory(Pair.first.getScalarVal(), type);
2705 llvm::Value *success = Pair.second;
Erik Pilkington53e43f42019-02-28 00:47:55 +00002706 atomicPHI->addIncoming(old, curBlock);
2707 Builder.CreateCondBr(success, contBB, atomicPHI->getParent());
David Chisnallfa35df62012-01-16 17:27:18 +00002708 Builder.SetInsertPoint(contBB);
2709 return isPre ? value : input;
2710 }
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00002711
Chris Lattner05dc78c2010-06-26 22:09:34 +00002712 // Store the updated result through the lvalue.
2713 if (LV.isBitField())
John McCall55e1fbc2011-06-25 02:11:03 +00002714 CGF.EmitStoreThroughBitfieldLValue(RValue::get(value), LV, &value);
Chris Lattner05dc78c2010-06-26 22:09:34 +00002715 else
John McCall55e1fbc2011-06-25 02:11:03 +00002716 CGF.EmitStoreThroughLValue(RValue::get(value), LV);
Anton Korobeynikovf0c267e2011-10-14 23:23:15 +00002717
Chris Lattner05dc78c2010-06-26 22:09:34 +00002718 // If this is a postinc, return the value read from memory, otherwise use the
2719 // updated value.
John McCalle3dc1702011-02-15 09:22:45 +00002720 return isPre ? value : input;
Chris Lattner05dc78c2010-06-26 22:09:34 +00002721}
2722
2723
2724
Chris Lattner2da04b32007-08-24 05:35:26 +00002725Value *ScalarExprEmitter::VisitUnaryMinus(const UnaryOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00002726 TestAndClearIgnoreResultAssign();
Cameron McInally20b8ed22019-10-14 15:35:01 +00002727 Value *Op = Visit(E->getSubExpr());
2728
2729 // Generate a unary FNeg for FP ops.
2730 if (Op->getType()->isFPOrFPVectorTy())
2731 return Builder.CreateFNeg(Op, "fneg");
2732
Chris Lattner0bf27622010-06-26 21:48:21 +00002733 // Emit unary minus with EmitSub so we handle overflow cases etc.
2734 BinOpInfo BinOp;
Cameron McInally20b8ed22019-10-14 15:35:01 +00002735 BinOp.RHS = Op;
2736 BinOp.LHS = llvm::Constant::getNullValue(BinOp.RHS->getType());
Chris Lattner0bf27622010-06-26 21:48:21 +00002737 BinOp.Ty = E->getType();
John McCalle3027922010-08-25 11:45:40 +00002738 BinOp.Opcode = BO_Sub;
Melanie Blowerdefd43a2020-06-26 08:45:12 -07002739 BinOp.FPFeatures = E->getFPFeatures(CGF.getLangOpts());
Chris Lattner0bf27622010-06-26 21:48:21 +00002740 BinOp.E = E;
2741 return EmitSub(BinOp);
Chris Lattner2da04b32007-08-24 05:35:26 +00002742}
2743
2744Value *ScalarExprEmitter::VisitUnaryNot(const UnaryOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00002745 TestAndClearIgnoreResultAssign();
Chris Lattner2da04b32007-08-24 05:35:26 +00002746 Value *Op = Visit(E->getSubExpr());
2747 return Builder.CreateNot(Op, "neg");
2748}
2749
2750Value *ScalarExprEmitter::VisitUnaryLNot(const UnaryOperator *E) {
Tanya Lattner20248222012-01-16 21:02:28 +00002751 // Perform vector logical not on comparison with zero vector.
Fangrui Songfc935fc2020-06-08 09:32:30 -07002752 if (E->getType()->isVectorType() &&
2753 E->getType()->castAs<VectorType>()->getVectorKind() ==
2754 VectorType::GenericVector) {
Tanya Lattner20248222012-01-16 21:02:28 +00002755 Value *Oper = Visit(E->getSubExpr());
2756 Value *Zero = llvm::Constant::getNullValue(Oper->getType());
Joey Gouly7d00f002013-02-21 11:49:56 +00002757 Value *Result;
Melanie Blowerf5360d42020-05-01 10:32:06 -07002758 if (Oper->getType()->isFPOrFPVectorTy()) {
John McCall7fac1ac2020-06-11 18:09:36 -04002759 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(
Melanie Blowerdefd43a2020-06-26 08:45:12 -07002760 CGF, E->getFPFeatures(CGF.getLangOpts()));
Joey Gouly7d00f002013-02-21 11:49:56 +00002761 Result = Builder.CreateFCmp(llvm::CmpInst::FCMP_OEQ, Oper, Zero, "cmp");
Melanie Blowerf5360d42020-05-01 10:32:06 -07002762 } else
Joey Gouly7d00f002013-02-21 11:49:56 +00002763 Result = Builder.CreateICmp(llvm::CmpInst::ICMP_EQ, Oper, Zero, "cmp");
Tanya Lattner20248222012-01-16 21:02:28 +00002764 return Builder.CreateSExt(Result, ConvertType(E->getType()), "sext");
2765 }
Craig Toppera97d7e72013-07-26 06:16:11 +00002766
Chris Lattner2da04b32007-08-24 05:35:26 +00002767 // Compare operand to zero.
2768 Value *BoolVal = CGF.EvaluateExprAsBool(E->getSubExpr());
Mike Stump4a3999f2009-09-09 13:00:44 +00002769
Chris Lattner2da04b32007-08-24 05:35:26 +00002770 // Invert value.
2771 // TODO: Could dynamically modify easy computations here. For example, if
2772 // the operand is an icmp ne, turn into icmp eq.
2773 BoolVal = Builder.CreateNot(BoolVal, "lnot");
Mike Stump4a3999f2009-09-09 13:00:44 +00002774
Anders Carlsson775640d2009-05-19 18:44:53 +00002775 // ZExt result to the expr type.
2776 return Builder.CreateZExt(BoolVal, ConvertType(E->getType()), "lnot.ext");
Chris Lattner2da04b32007-08-24 05:35:26 +00002777}
2778
Eli Friedmand7c72322010-08-05 09:58:49 +00002779Value *ScalarExprEmitter::VisitOffsetOfExpr(OffsetOfExpr *E) {
2780 // Try folding the offsetof to a constant.
Fangrui Song407659a2018-11-30 23:41:18 +00002781 Expr::EvalResult EVResult;
2782 if (E->EvaluateAsInt(EVResult, CGF.getContext())) {
2783 llvm::APSInt Value = EVResult.Val.getInt();
Richard Smith5fab0c92011-12-28 19:48:30 +00002784 return Builder.getInt(Value);
Fangrui Song407659a2018-11-30 23:41:18 +00002785 }
Eli Friedmand7c72322010-08-05 09:58:49 +00002786
2787 // Loop over the components of the offsetof to compute the value.
2788 unsigned n = E->getNumComponents();
Chris Lattner2192fe52011-07-18 04:24:23 +00002789 llvm::Type* ResultType = ConvertType(E->getType());
Eli Friedmand7c72322010-08-05 09:58:49 +00002790 llvm::Value* Result = llvm::Constant::getNullValue(ResultType);
2791 QualType CurrentType = E->getTypeSourceInfo()->getType();
2792 for (unsigned i = 0; i != n; ++i) {
James Y Knight7281c352015-12-29 22:31:18 +00002793 OffsetOfNode ON = E->getComponent(i);
Craig Topper8a13c412014-05-21 05:09:00 +00002794 llvm::Value *Offset = nullptr;
Eli Friedmand7c72322010-08-05 09:58:49 +00002795 switch (ON.getKind()) {
James Y Knight7281c352015-12-29 22:31:18 +00002796 case OffsetOfNode::Array: {
Eli Friedmand7c72322010-08-05 09:58:49 +00002797 // Compute the index
2798 Expr *IdxExpr = E->getIndexExpr(ON.getArrayExprIndex());
2799 llvm::Value* Idx = CGF.EmitScalarExpr(IdxExpr);
Douglas Gregor6ab2fa82011-05-20 16:38:50 +00002800 bool IdxSigned = IdxExpr->getType()->isSignedIntegerOrEnumerationType();
Eli Friedmand7c72322010-08-05 09:58:49 +00002801 Idx = Builder.CreateIntCast(Idx, ResultType, IdxSigned, "conv");
2802
2803 // Save the element type
2804 CurrentType =
2805 CGF.getContext().getAsArrayType(CurrentType)->getElementType();
2806
2807 // Compute the element size
2808 llvm::Value* ElemSize = llvm::ConstantInt::get(ResultType,
2809 CGF.getContext().getTypeSizeInChars(CurrentType).getQuantity());
2810
2811 // Multiply out to compute the result
2812 Offset = Builder.CreateMul(Idx, ElemSize);
2813 break;
2814 }
2815
James Y Knight7281c352015-12-29 22:31:18 +00002816 case OffsetOfNode::Field: {
Eli Friedmand7c72322010-08-05 09:58:49 +00002817 FieldDecl *MemberDecl = ON.getField();
Simon Pilgrim1cd399c2019-10-03 11:22:48 +00002818 RecordDecl *RD = CurrentType->castAs<RecordType>()->getDecl();
Eli Friedmand7c72322010-08-05 09:58:49 +00002819 const ASTRecordLayout &RL = CGF.getContext().getASTRecordLayout(RD);
2820
2821 // Compute the index of the field in its parent.
2822 unsigned i = 0;
2823 // FIXME: It would be nice if we didn't have to loop here!
2824 for (RecordDecl::field_iterator Field = RD->field_begin(),
2825 FieldEnd = RD->field_end();
David Blaikie2d7c57e2012-04-30 02:36:29 +00002826 Field != FieldEnd; ++Field, ++i) {
David Blaikie40ed2972012-06-06 20:45:41 +00002827 if (*Field == MemberDecl)
Eli Friedmand7c72322010-08-05 09:58:49 +00002828 break;
2829 }
2830 assert(i < RL.getFieldCount() && "offsetof field in wrong type");
2831
2832 // Compute the offset to the field
2833 int64_t OffsetInt = RL.getFieldOffset(i) /
2834 CGF.getContext().getCharWidth();
2835 Offset = llvm::ConstantInt::get(ResultType, OffsetInt);
2836
2837 // Save the element type.
2838 CurrentType = MemberDecl->getType();
2839 break;
2840 }
Eli Friedman165301d2010-08-06 16:37:05 +00002841
James Y Knight7281c352015-12-29 22:31:18 +00002842 case OffsetOfNode::Identifier:
Eli Friedmane83d2b762010-08-06 01:17:25 +00002843 llvm_unreachable("dependent __builtin_offsetof");
Eli Friedman165301d2010-08-06 16:37:05 +00002844
James Y Knight7281c352015-12-29 22:31:18 +00002845 case OffsetOfNode::Base: {
Eli Friedmand7c72322010-08-05 09:58:49 +00002846 if (ON.getBase()->isVirtual()) {
2847 CGF.ErrorUnsupported(E, "virtual base in offsetof");
2848 continue;
2849 }
2850
Simon Pilgrim1cd399c2019-10-03 11:22:48 +00002851 RecordDecl *RD = CurrentType->castAs<RecordType>()->getDecl();
Eli Friedmand7c72322010-08-05 09:58:49 +00002852 const ASTRecordLayout &RL = CGF.getContext().getASTRecordLayout(RD);
2853
2854 // Save the element type.
2855 CurrentType = ON.getBase()->getType();
Craig Toppera97d7e72013-07-26 06:16:11 +00002856
Eli Friedmand7c72322010-08-05 09:58:49 +00002857 // Compute the offset to the base.
2858 const RecordType *BaseRT = CurrentType->getAs<RecordType>();
2859 CXXRecordDecl *BaseRD = cast<CXXRecordDecl>(BaseRT->getDecl());
Benjamin Kramer2ef30312012-07-04 18:45:14 +00002860 CharUnits OffsetInt = RL.getBaseClassOffset(BaseRD);
2861 Offset = llvm::ConstantInt::get(ResultType, OffsetInt.getQuantity());
Eli Friedmand7c72322010-08-05 09:58:49 +00002862 break;
2863 }
2864 }
2865 Result = Builder.CreateAdd(Result, Offset);
2866 }
2867 return Result;
Douglas Gregor882211c2010-04-28 22:16:22 +00002868}
2869
Peter Collingbournee190dee2011-03-11 19:24:49 +00002870/// VisitUnaryExprOrTypeTraitExpr - Return the size or alignment of the type of
Sebastian Redl6f282892008-11-11 17:56:53 +00002871/// argument of the sizeof expression as an integer.
2872Value *
Peter Collingbournee190dee2011-03-11 19:24:49 +00002873ScalarExprEmitter::VisitUnaryExprOrTypeTraitExpr(
2874 const UnaryExprOrTypeTraitExpr *E) {
Sebastian Redl6f282892008-11-11 17:56:53 +00002875 QualType TypeToSize = E->getTypeOfArgument();
Peter Collingbournee190dee2011-03-11 19:24:49 +00002876 if (E->getKind() == UETT_SizeOf) {
Mike Stump4a3999f2009-09-09 13:00:44 +00002877 if (const VariableArrayType *VAT =
Eli Friedman2aa38fe2009-01-24 22:19:05 +00002878 CGF.getContext().getAsVariableArrayType(TypeToSize)) {
2879 if (E->isArgumentType()) {
2880 // sizeof(type) - make sure to emit the VLA size.
John McCall23c29fe2011-06-24 21:55:10 +00002881 CGF.EmitVariablyModifiedType(TypeToSize);
Eli Friedman3253e182009-04-20 03:21:44 +00002882 } else {
2883 // C99 6.5.3.4p2: If the argument is an expression of type
2884 // VLA, it is evaluated.
John McCalla2342eb2010-12-05 02:00:02 +00002885 CGF.EmitIgnoredExpr(E->getArgumentExpr());
Eli Friedman2aa38fe2009-01-24 22:19:05 +00002886 }
Mike Stump4a3999f2009-09-09 13:00:44 +00002887
Sander de Smalen891af03a2018-02-03 13:55:59 +00002888 auto VlaSize = CGF.getVLASize(VAT);
2889 llvm::Value *size = VlaSize.NumElts;
John McCall23c29fe2011-06-24 21:55:10 +00002890
2891 // Scale the number of non-VLA elements by the non-VLA element size.
Sander de Smalen891af03a2018-02-03 13:55:59 +00002892 CharUnits eltSize = CGF.getContext().getTypeSizeInChars(VlaSize.Type);
John McCall23c29fe2011-06-24 21:55:10 +00002893 if (!eltSize.isOne())
Sander de Smalen891af03a2018-02-03 13:55:59 +00002894 size = CGF.Builder.CreateNUWMul(CGF.CGM.getSize(eltSize), size);
John McCall23c29fe2011-06-24 21:55:10 +00002895
2896 return size;
Anders Carlsson76dbc042008-12-21 03:33:21 +00002897 }
Alexey Bataev00396512015-07-02 03:40:19 +00002898 } else if (E->getKind() == UETT_OpenMPRequiredSimdAlign) {
2899 auto Alignment =
2900 CGF.getContext()
2901 .toCharUnitsFromBits(CGF.getContext().getOpenMPDefaultSimdAlign(
2902 E->getTypeOfArgument()->getPointeeType()))
2903 .getQuantity();
2904 return llvm::ConstantInt::get(CGF.SizeTy, Alignment);
Anders Carlsson30032882008-12-12 07:38:43 +00002905 }
Eli Friedman2aa38fe2009-01-24 22:19:05 +00002906
Mike Stump4a3999f2009-09-09 13:00:44 +00002907 // If this isn't sizeof(vla), the result must be constant; use the constant
2908 // folding logic so we don't have to duplicate it here.
Richard Smith5fab0c92011-12-28 19:48:30 +00002909 return Builder.getInt(E->EvaluateKnownConstInt(CGF.getContext()));
Chris Lattner2da04b32007-08-24 05:35:26 +00002910}
2911
Chris Lattner9f0ad962007-08-24 21:20:17 +00002912Value *ScalarExprEmitter::VisitUnaryReal(const UnaryOperator *E) {
2913 Expr *Op = E->getSubExpr();
John McCall07bb1962010-11-16 10:08:07 +00002914 if (Op->getType()->isAnyComplexType()) {
2915 // If it's an l-value, load through the appropriate subobject l-value.
2916 // Note that we have to ask E because Op might be an l-value that
2917 // this won't work for, e.g. an Obj-C property.
John McCall086a4642010-11-24 05:12:34 +00002918 if (E->isGLValue())
Nick Lewycky2d84e842013-10-02 02:29:49 +00002919 return CGF.EmitLoadOfLValue(CGF.EmitLValue(E),
2920 E->getExprLoc()).getScalarVal();
John McCall07bb1962010-11-16 10:08:07 +00002921
2922 // Otherwise, calculate and project.
2923 return CGF.EmitComplexExpr(Op, false, true).first;
2924 }
2925
Chris Lattner9f0ad962007-08-24 21:20:17 +00002926 return Visit(Op);
2927}
John McCall07bb1962010-11-16 10:08:07 +00002928
Chris Lattner9f0ad962007-08-24 21:20:17 +00002929Value *ScalarExprEmitter::VisitUnaryImag(const UnaryOperator *E) {
2930 Expr *Op = E->getSubExpr();
John McCall07bb1962010-11-16 10:08:07 +00002931 if (Op->getType()->isAnyComplexType()) {
2932 // If it's an l-value, load through the appropriate subobject l-value.
2933 // Note that we have to ask E because Op might be an l-value that
2934 // this won't work for, e.g. an Obj-C property.
John McCall086a4642010-11-24 05:12:34 +00002935 if (Op->isGLValue())
Nick Lewycky2d84e842013-10-02 02:29:49 +00002936 return CGF.EmitLoadOfLValue(CGF.EmitLValue(E),
2937 E->getExprLoc()).getScalarVal();
John McCall07bb1962010-11-16 10:08:07 +00002938
2939 // Otherwise, calculate and project.
2940 return CGF.EmitComplexExpr(Op, true, false).second;
2941 }
Mike Stump4a3999f2009-09-09 13:00:44 +00002942
Mike Stumpdf0fe272009-05-29 15:46:01 +00002943 // __imag on a scalar returns zero. Emit the subexpr to ensure side
2944 // effects are evaluated, but not the actual value.
Richard Smith0b6b8e42012-02-18 20:53:32 +00002945 if (Op->isGLValue())
2946 CGF.EmitLValue(Op);
2947 else
2948 CGF.EmitScalarExpr(Op, true);
Owen Anderson0b75f232009-07-31 20:28:54 +00002949 return llvm::Constant::getNullValue(ConvertType(E->getType()));
Chris Lattner9f0ad962007-08-24 21:20:17 +00002950}
2951
Chris Lattner2da04b32007-08-24 05:35:26 +00002952//===----------------------------------------------------------------------===//
2953// Binary Operators
2954//===----------------------------------------------------------------------===//
2955
2956BinOpInfo ScalarExprEmitter::EmitBinOps(const BinaryOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00002957 TestAndClearIgnoreResultAssign();
Chris Lattner2da04b32007-08-24 05:35:26 +00002958 BinOpInfo Result;
2959 Result.LHS = Visit(E->getLHS());
2960 Result.RHS = Visit(E->getRHS());
Chris Lattner3d966d62007-08-24 21:00:35 +00002961 Result.Ty = E->getType();
Chris Lattner0bf27622010-06-26 21:48:21 +00002962 Result.Opcode = E->getOpcode();
Melanie Blowerdefd43a2020-06-26 08:45:12 -07002963 Result.FPFeatures = E->getFPFeatures(CGF.getLangOpts());
Chris Lattner2da04b32007-08-24 05:35:26 +00002964 Result.E = E;
2965 return Result;
2966}
2967
Douglas Gregor914af212010-04-23 04:16:32 +00002968LValue ScalarExprEmitter::EmitCompoundAssignLValue(
2969 const CompoundAssignOperator *E,
2970 Value *(ScalarExprEmitter::*Func)(const BinOpInfo &),
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00002971 Value *&Result) {
Benjamin Kramerd20ef752009-12-25 15:43:36 +00002972 QualType LHSTy = E->getLHS()->getType();
Chris Lattner3d966d62007-08-24 21:00:35 +00002973 BinOpInfo OpInfo;
Craig Toppera97d7e72013-07-26 06:16:11 +00002974
Eli Friedmanf0450072013-06-12 01:40:06 +00002975 if (E->getComputationResultType()->isAnyComplexType())
Richard Smith527473d2015-02-12 21:23:20 +00002976 return CGF.EmitScalarCompoundAssignWithComplex(E, Result);
Craig Toppera97d7e72013-07-26 06:16:11 +00002977
Mike Stumpc63428b2009-05-22 19:07:20 +00002978 // Emit the RHS first. __block variables need to have the rhs evaluated
2979 // first, plus this should improve codegen a little.
2980 OpInfo.RHS = Visit(E->getRHS());
2981 OpInfo.Ty = E->getComputationResultType();
Chris Lattner0bf27622010-06-26 21:48:21 +00002982 OpInfo.Opcode = E->getOpcode();
Melanie Blowerdefd43a2020-06-26 08:45:12 -07002983 OpInfo.FPFeatures = E->getFPFeatures(CGF.getLangOpts());
Mike Stumpc63428b2009-05-22 19:07:20 +00002984 OpInfo.E = E;
Eli Friedman8b7b1b12009-03-28 01:22:36 +00002985 // Load/convert the LHS.
Richard Smith4d1458e2012-09-08 02:08:36 +00002986 LValue LHSLV = EmitCheckedLValue(E->getLHS(), CodeGenFunction::TCK_Store);
David Chisnallfa35df62012-01-16 17:27:18 +00002987
Craig Topper8a13c412014-05-21 05:09:00 +00002988 llvm::PHINode *atomicPHI = nullptr;
David Chisnallef78c302013-03-03 16:02:42 +00002989 if (const AtomicType *atomicTy = LHSTy->getAs<AtomicType>()) {
2990 QualType type = atomicTy->getValueType();
2991 if (!type->isBooleanType() && type->isIntegerType() &&
Alexey Samsonovedf99a92014-11-07 22:29:38 +00002992 !(type->isUnsignedIntegerType() &&
2993 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow)) &&
2994 CGF.getLangOpts().getSignedOverflowBehavior() !=
2995 LangOptions::SOB_Trapping) {
Tim Northover10e0d642019-11-07 13:36:03 +00002996 llvm::AtomicRMWInst::BinOp AtomicOp = llvm::AtomicRMWInst::BAD_BINOP;
2997 llvm::Instruction::BinaryOps Op;
David Chisnallef78c302013-03-03 16:02:42 +00002998 switch (OpInfo.Opcode) {
2999 // We don't have atomicrmw operands for *, %, /, <<, >>
3000 case BO_MulAssign: case BO_DivAssign:
3001 case BO_RemAssign:
3002 case BO_ShlAssign:
3003 case BO_ShrAssign:
3004 break;
3005 case BO_AddAssign:
Tim Northover10e0d642019-11-07 13:36:03 +00003006 AtomicOp = llvm::AtomicRMWInst::Add;
3007 Op = llvm::Instruction::Add;
David Chisnallef78c302013-03-03 16:02:42 +00003008 break;
3009 case BO_SubAssign:
Tim Northover10e0d642019-11-07 13:36:03 +00003010 AtomicOp = llvm::AtomicRMWInst::Sub;
3011 Op = llvm::Instruction::Sub;
David Chisnallef78c302013-03-03 16:02:42 +00003012 break;
3013 case BO_AndAssign:
Tim Northover10e0d642019-11-07 13:36:03 +00003014 AtomicOp = llvm::AtomicRMWInst::And;
3015 Op = llvm::Instruction::And;
David Chisnallef78c302013-03-03 16:02:42 +00003016 break;
3017 case BO_XorAssign:
Tim Northover10e0d642019-11-07 13:36:03 +00003018 AtomicOp = llvm::AtomicRMWInst::Xor;
3019 Op = llvm::Instruction::Xor;
David Chisnallef78c302013-03-03 16:02:42 +00003020 break;
3021 case BO_OrAssign:
Tim Northover10e0d642019-11-07 13:36:03 +00003022 AtomicOp = llvm::AtomicRMWInst::Or;
3023 Op = llvm::Instruction::Or;
David Chisnallef78c302013-03-03 16:02:42 +00003024 break;
3025 default:
3026 llvm_unreachable("Invalid compound assignment type");
3027 }
Tim Northover10e0d642019-11-07 13:36:03 +00003028 if (AtomicOp != llvm::AtomicRMWInst::BAD_BINOP) {
3029 llvm::Value *Amt = CGF.EmitToMemory(
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00003030 EmitScalarConversion(OpInfo.RHS, E->getRHS()->getType(), LHSTy,
3031 E->getExprLoc()),
3032 LHSTy);
Tim Northover10e0d642019-11-07 13:36:03 +00003033 Value *OldVal = Builder.CreateAtomicRMW(
Akira Hatanakaf139ae32019-12-03 15:17:01 -08003034 AtomicOp, LHSLV.getPointer(CGF), Amt,
JF Bastien92f4ef12016-04-06 17:26:42 +00003035 llvm::AtomicOrdering::SequentiallyConsistent);
Tim Northover10e0d642019-11-07 13:36:03 +00003036
3037 // Since operation is atomic, the result type is guaranteed to be the
3038 // same as the input in LLVM terms.
3039 Result = Builder.CreateBinOp(Op, OldVal, Amt);
David Chisnallef78c302013-03-03 16:02:42 +00003040 return LHSLV;
3041 }
3042 }
David Chisnallfa35df62012-01-16 17:27:18 +00003043 // FIXME: For floating point types, we should be saving and restoring the
3044 // floating point environment in the loop.
3045 llvm::BasicBlock *startBB = Builder.GetInsertBlock();
3046 llvm::BasicBlock *opBB = CGF.createBasicBlock("atomic_op", CGF.CurFn);
Nick Lewycky2d84e842013-10-02 02:29:49 +00003047 OpInfo.LHS = EmitLoadOfLValue(LHSLV, E->getExprLoc());
David Chisnallef78c302013-03-03 16:02:42 +00003048 OpInfo.LHS = CGF.EmitToMemory(OpInfo.LHS, type);
David Chisnallfa35df62012-01-16 17:27:18 +00003049 Builder.CreateBr(opBB);
3050 Builder.SetInsertPoint(opBB);
3051 atomicPHI = Builder.CreatePHI(OpInfo.LHS->getType(), 2);
3052 atomicPHI->addIncoming(OpInfo.LHS, startBB);
David Chisnallfa35df62012-01-16 17:27:18 +00003053 OpInfo.LHS = atomicPHI;
3054 }
David Chisnallef78c302013-03-03 16:02:42 +00003055 else
Nick Lewycky2d84e842013-10-02 02:29:49 +00003056 OpInfo.LHS = EmitLoadOfLValue(LHSLV, E->getExprLoc());
Eli Friedman93ee5ca2012-06-16 02:19:17 +00003057
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00003058 SourceLocation Loc = E->getExprLoc();
3059 OpInfo.LHS =
3060 EmitScalarConversion(OpInfo.LHS, LHSTy, E->getComputationLHSType(), Loc);
Eli Friedman93ee5ca2012-06-16 02:19:17 +00003061
Chris Lattner3d966d62007-08-24 21:00:35 +00003062 // Expand the binary operator.
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00003063 Result = (this->*Func)(OpInfo);
Craig Toppera97d7e72013-07-26 06:16:11 +00003064
Roman Lebedevd677c3f2018-11-19 19:56:43 +00003065 // Convert the result back to the LHS type,
3066 // potentially with Implicit Conversion sanitizer check.
3067 Result = EmitScalarConversion(Result, E->getComputationResultType(), LHSTy,
3068 Loc, ScalarConversionOpts(CGF.SanOpts));
David Chisnallfa35df62012-01-16 17:27:18 +00003069
3070 if (atomicPHI) {
Erik Pilkington53e43f42019-02-28 00:47:55 +00003071 llvm::BasicBlock *curBlock = Builder.GetInsertBlock();
David Chisnallfa35df62012-01-16 17:27:18 +00003072 llvm::BasicBlock *contBB = CGF.createBasicBlock("atomic_cont", CGF.CurFn);
Alexey Bataev452d8e12014-12-15 05:25:25 +00003073 auto Pair = CGF.EmitAtomicCompareExchange(
Alexey Bataevb4505a72015-03-30 05:20:59 +00003074 LHSLV, RValue::get(atomicPHI), RValue::get(Result), E->getExprLoc());
3075 llvm::Value *old = CGF.EmitToMemory(Pair.first.getScalarVal(), LHSTy);
3076 llvm::Value *success = Pair.second;
Erik Pilkington53e43f42019-02-28 00:47:55 +00003077 atomicPHI->addIncoming(old, curBlock);
3078 Builder.CreateCondBr(success, contBB, atomicPHI->getParent());
David Chisnallfa35df62012-01-16 17:27:18 +00003079 Builder.SetInsertPoint(contBB);
3080 return LHSLV;
3081 }
Craig Toppera97d7e72013-07-26 06:16:11 +00003082
Mike Stump4a3999f2009-09-09 13:00:44 +00003083 // Store the result value into the LHS lvalue. Bit-fields are handled
3084 // specially because the result is altered by the store, i.e., [C99 6.5.16p1]
3085 // 'An assignment expression has the value of the left operand after the
3086 // assignment...'.
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00003087 if (LHSLV.isBitField())
John McCall55e1fbc2011-06-25 02:11:03 +00003088 CGF.EmitStoreThroughBitfieldLValue(RValue::get(Result), LHSLV, &Result);
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00003089 else
John McCall55e1fbc2011-06-25 02:11:03 +00003090 CGF.EmitStoreThroughLValue(RValue::get(Result), LHSLV);
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00003091
Alexey Bataeva58da1a2019-12-27 09:44:43 -05003092 if (CGF.getLangOpts().OpenMP)
3093 CGF.CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(CGF,
3094 E->getLHS());
Douglas Gregor914af212010-04-23 04:16:32 +00003095 return LHSLV;
3096}
3097
3098Value *ScalarExprEmitter::EmitCompoundAssign(const CompoundAssignOperator *E,
3099 Value *(ScalarExprEmitter::*Func)(const BinOpInfo &)) {
3100 bool Ignore = TestAndClearIgnoreResultAssign();
Simon Pilgrim30aa42e2019-05-18 12:17:15 +00003101 Value *RHS = nullptr;
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00003102 LValue LHS = EmitCompoundAssignLValue(E, Func, RHS);
3103
3104 // If the result is clearly ignored, return now.
Mike Stumpdf0fe272009-05-29 15:46:01 +00003105 if (Ignore)
Craig Topper8a13c412014-05-21 05:09:00 +00003106 return nullptr;
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00003107
John McCall07bb1962010-11-16 10:08:07 +00003108 // The result of an assignment in C is the assigned r-value.
Richard Smith9c6890a2012-11-01 22:30:59 +00003109 if (!CGF.getLangOpts().CPlusPlus)
John McCall07bb1962010-11-16 10:08:07 +00003110 return RHS;
3111
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00003112 // If the lvalue is non-volatile, return the computed value of the assignment.
3113 if (!LHS.isVolatileQualified())
3114 return RHS;
3115
3116 // Otherwise, reload the value.
Nick Lewycky2d84e842013-10-02 02:29:49 +00003117 return EmitLoadOfLValue(LHS, E->getExprLoc());
Chris Lattner3d966d62007-08-24 21:00:35 +00003118}
3119
Chris Lattner8ee6a412010-09-11 21:47:09 +00003120void ScalarExprEmitter::EmitUndefinedBehaviorIntegerDivAndRemCheck(
Richard Smith4d1458e2012-09-08 02:08:36 +00003121 const BinOpInfo &Ops, llvm::Value *Zero, bool isDiv) {
Peter Collingbourne3eea6772015-05-11 21:39:14 +00003122 SmallVector<std::pair<llvm::Value *, SanitizerMask>, 2> Checks;
Chris Lattner8ee6a412010-09-11 21:47:09 +00003123
Alexey Samsonov4c1a96f2014-11-10 22:27:30 +00003124 if (CGF.SanOpts.has(SanitizerKind::IntegerDivideByZero)) {
Alexey Samsonove396bfc2014-11-11 22:03:54 +00003125 Checks.push_back(std::make_pair(Builder.CreateICmpNE(Ops.RHS, Zero),
3126 SanitizerKind::IntegerDivideByZero));
Alexey Samsonov4c1a96f2014-11-10 22:27:30 +00003127 }
Richard Smithc86a1142012-11-06 02:30:30 +00003128
Vedant Kumar82ee16b2017-02-25 00:43:36 +00003129 const auto *BO = cast<BinaryOperator>(Ops.E);
Alexey Samsonovedf99a92014-11-07 22:29:38 +00003130 if (CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow) &&
Vedant Kumar82ee16b2017-02-25 00:43:36 +00003131 Ops.Ty->hasSignedIntegerRepresentation() &&
Vedant Kumard9191152017-05-02 23:46:56 +00003132 !IsWidenedIntegerOp(CGF.getContext(), BO->getLHS()) &&
3133 Ops.mayHaveIntegerOverflow()) {
Richard Smithc86a1142012-11-06 02:30:30 +00003134 llvm::IntegerType *Ty = cast<llvm::IntegerType>(Zero->getType());
3135
Chris Lattner8ee6a412010-09-11 21:47:09 +00003136 llvm::Value *IntMin =
Chris Lattner2531eb42011-04-19 22:55:03 +00003137 Builder.getInt(llvm::APInt::getSignedMinValue(Ty->getBitWidth()));
Chris Lattner8ee6a412010-09-11 21:47:09 +00003138 llvm::Value *NegOne = llvm::ConstantInt::get(Ty, -1ULL);
3139
Richard Smith4d1458e2012-09-08 02:08:36 +00003140 llvm::Value *LHSCmp = Builder.CreateICmpNE(Ops.LHS, IntMin);
3141 llvm::Value *RHSCmp = Builder.CreateICmpNE(Ops.RHS, NegOne);
Alexey Samsonove396bfc2014-11-11 22:03:54 +00003142 llvm::Value *NotOverflow = Builder.CreateOr(LHSCmp, RHSCmp, "or");
3143 Checks.push_back(
3144 std::make_pair(NotOverflow, SanitizerKind::SignedIntegerOverflow));
Chris Lattner8ee6a412010-09-11 21:47:09 +00003145 }
Richard Smithc86a1142012-11-06 02:30:30 +00003146
Alexey Samsonove396bfc2014-11-11 22:03:54 +00003147 if (Checks.size() > 0)
3148 EmitBinOpCheck(Checks, Ops);
Chris Lattner8ee6a412010-09-11 21:47:09 +00003149}
Chris Lattner3d966d62007-08-24 21:00:35 +00003150
Chris Lattner2da04b32007-08-24 05:35:26 +00003151Value *ScalarExprEmitter::EmitDiv(const BinOpInfo &Ops) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00003152 {
3153 CodeGenFunction::SanitizerScope SanScope(&CGF);
Alexey Samsonovedf99a92014-11-07 22:29:38 +00003154 if ((CGF.SanOpts.has(SanitizerKind::IntegerDivideByZero) ||
3155 CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow)) &&
Vedant Kumard9191152017-05-02 23:46:56 +00003156 Ops.Ty->isIntegerType() &&
3157 (Ops.mayHaveIntegerDivisionByZero() || Ops.mayHaveIntegerOverflow())) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00003158 llvm::Value *Zero = llvm::Constant::getNullValue(ConvertType(Ops.Ty));
3159 EmitUndefinedBehaviorIntegerDivAndRemCheck(Ops, Zero, true);
Alexey Samsonovedf99a92014-11-07 22:29:38 +00003160 } else if (CGF.SanOpts.has(SanitizerKind::FloatDivideByZero) &&
Vedant Kumard9191152017-05-02 23:46:56 +00003161 Ops.Ty->isRealFloatingType() &&
3162 Ops.mayHaveFloatDivisionByZero()) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00003163 llvm::Value *Zero = llvm::Constant::getNullValue(ConvertType(Ops.Ty));
Alexey Samsonove396bfc2014-11-11 22:03:54 +00003164 llvm::Value *NonZero = Builder.CreateFCmpUNE(Ops.RHS, Zero);
3165 EmitBinOpCheck(std::make_pair(NonZero, SanitizerKind::FloatDivideByZero),
3166 Ops);
Alexey Samsonov24cad992014-07-17 18:46:27 +00003167 }
Chris Lattner8ee6a412010-09-11 21:47:09 +00003168 }
Will Dietz1897cb32012-11-27 15:01:55 +00003169
Peter Collingbourne95fd2ca2011-10-27 19:19:51 +00003170 if (Ops.LHS->getType()->isFPOrFPVectorTy()) {
Melanie Blowerf5360d42020-05-01 10:32:06 -07003171 llvm::Value *Val;
John McCall7fac1ac2020-06-11 18:09:36 -04003172 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, Ops.FPFeatures);
Melanie Blowerf5360d42020-05-01 10:32:06 -07003173 Val = Builder.CreateFDiv(Ops.LHS, Ops.RHS, "div");
Yaxun Liuffb60902016-08-09 20:10:18 +00003174 if (CGF.getLangOpts().OpenCL &&
3175 !CGF.CGM.getCodeGenOpts().CorrectlyRoundedDivSqrt) {
3176 // OpenCL v1.1 s7.4: minimum accuracy of single precision / is 2.5ulp
3177 // OpenCL v1.2 s5.6.4.2: The -cl-fp32-correctly-rounded-divide-sqrt
3178 // build option allows an application to specify that single precision
3179 // floating-point divide (x/y and 1/x) and sqrt used in the program
3180 // source are correctly rounded.
Peter Collingbourne95fd2ca2011-10-27 19:19:51 +00003181 llvm::Type *ValTy = Val->getType();
3182 if (ValTy->isFloatTy() ||
3183 (isa<llvm::VectorType>(ValTy) &&
3184 cast<llvm::VectorType>(ValTy)->getElementType()->isFloatTy()))
Duncan Sandse81111c2012-04-10 08:23:07 +00003185 CGF.SetFPAccuracy(Val, 2.5);
Peter Collingbourne95fd2ca2011-10-27 19:19:51 +00003186 }
3187 return Val;
3188 }
Bevin Hansson39baaab2020-01-08 11:12:55 +01003189 else if (Ops.isFixedPointOp())
Bevin Hansson0b9922e2019-11-19 13:15:06 +01003190 return EmitFixedPointBinOp(Ops);
Douglas Gregor5cc2c8b2010-07-23 15:58:24 +00003191 else if (Ops.Ty->hasUnsignedIntegerRepresentation())
Chris Lattner2da04b32007-08-24 05:35:26 +00003192 return Builder.CreateUDiv(Ops.LHS, Ops.RHS, "div");
3193 else
3194 return Builder.CreateSDiv(Ops.LHS, Ops.RHS, "div");
3195}
3196
3197Value *ScalarExprEmitter::EmitRem(const BinOpInfo &Ops) {
3198 // Rem in C can't be a floating point type: C99 6.5.5p2.
Vedant Kumar42de3802017-02-25 00:43:39 +00003199 if ((CGF.SanOpts.has(SanitizerKind::IntegerDivideByZero) ||
3200 CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow)) &&
Vedant Kumard9191152017-05-02 23:46:56 +00003201 Ops.Ty->isIntegerType() &&
3202 (Ops.mayHaveIntegerDivisionByZero() || Ops.mayHaveIntegerOverflow())) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00003203 CodeGenFunction::SanitizerScope SanScope(&CGF);
Chris Lattner8ee6a412010-09-11 21:47:09 +00003204 llvm::Value *Zero = llvm::Constant::getNullValue(ConvertType(Ops.Ty));
Vedant Kumar42de3802017-02-25 00:43:39 +00003205 EmitUndefinedBehaviorIntegerDivAndRemCheck(Ops, Zero, false);
Chris Lattner8ee6a412010-09-11 21:47:09 +00003206 }
3207
Eli Friedman493c34a2011-04-10 04:44:11 +00003208 if (Ops.Ty->hasUnsignedIntegerRepresentation())
Chris Lattner2da04b32007-08-24 05:35:26 +00003209 return Builder.CreateURem(Ops.LHS, Ops.RHS, "rem");
3210 else
3211 return Builder.CreateSRem(Ops.LHS, Ops.RHS, "rem");
3212}
3213
Mike Stump0c61b732009-04-01 20:28:16 +00003214Value *ScalarExprEmitter::EmitOverflowCheckedBinOp(const BinOpInfo &Ops) {
3215 unsigned IID;
3216 unsigned OpID = 0;
Mike Stump40968592009-04-02 01:03:55 +00003217
Will Dietz1897cb32012-11-27 15:01:55 +00003218 bool isSigned = Ops.Ty->isSignedIntegerOrEnumerationType();
Chris Lattner0bf27622010-06-26 21:48:21 +00003219 switch (Ops.Opcode) {
John McCalle3027922010-08-25 11:45:40 +00003220 case BO_Add:
3221 case BO_AddAssign:
Mike Stumpd3e38852009-04-02 18:15:54 +00003222 OpID = 1;
Will Dietz1897cb32012-11-27 15:01:55 +00003223 IID = isSigned ? llvm::Intrinsic::sadd_with_overflow :
3224 llvm::Intrinsic::uadd_with_overflow;
Mike Stumpd3e38852009-04-02 18:15:54 +00003225 break;
John McCalle3027922010-08-25 11:45:40 +00003226 case BO_Sub:
3227 case BO_SubAssign:
Mike Stumpd3e38852009-04-02 18:15:54 +00003228 OpID = 2;
Will Dietz1897cb32012-11-27 15:01:55 +00003229 IID = isSigned ? llvm::Intrinsic::ssub_with_overflow :
3230 llvm::Intrinsic::usub_with_overflow;
Mike Stumpd3e38852009-04-02 18:15:54 +00003231 break;
John McCalle3027922010-08-25 11:45:40 +00003232 case BO_Mul:
3233 case BO_MulAssign:
Mike Stumpd3e38852009-04-02 18:15:54 +00003234 OpID = 3;
Will Dietz1897cb32012-11-27 15:01:55 +00003235 IID = isSigned ? llvm::Intrinsic::smul_with_overflow :
3236 llvm::Intrinsic::umul_with_overflow;
Mike Stumpd3e38852009-04-02 18:15:54 +00003237 break;
3238 default:
David Blaikie83d382b2011-09-23 05:06:16 +00003239 llvm_unreachable("Unsupported operation for overflow detection");
Mike Stump0c61b732009-04-01 20:28:16 +00003240 }
Mike Stumpd3e38852009-04-02 18:15:54 +00003241 OpID <<= 1;
Will Dietz1897cb32012-11-27 15:01:55 +00003242 if (isSigned)
3243 OpID |= 1;
Mike Stumpd3e38852009-04-02 18:15:54 +00003244
Vedant Kumar4b62b5c2017-05-09 23:34:49 +00003245 CodeGenFunction::SanitizerScope SanScope(&CGF);
Chris Lattnera5f58b02011-07-09 17:41:47 +00003246 llvm::Type *opTy = CGF.CGM.getTypes().ConvertType(Ops.Ty);
Mike Stump0c61b732009-04-01 20:28:16 +00003247
Benjamin Kramer8d375ce2011-07-14 17:45:50 +00003248 llvm::Function *intrinsic = CGF.CGM.getIntrinsic(IID, opTy);
Mike Stump0c61b732009-04-01 20:28:16 +00003249
David Blaikie43f9bb72015-05-18 22:14:03 +00003250 Value *resultAndOverflow = Builder.CreateCall(intrinsic, {Ops.LHS, Ops.RHS});
Mike Stump0c61b732009-04-01 20:28:16 +00003251 Value *result = Builder.CreateExtractValue(resultAndOverflow, 0);
3252 Value *overflow = Builder.CreateExtractValue(resultAndOverflow, 1);
3253
Richard Smith4d1458e2012-09-08 02:08:36 +00003254 // Handle overflow with llvm.trap if no custom handler has been specified.
3255 const std::string *handlerName =
Richard Smith9c6890a2012-11-01 22:30:59 +00003256 &CGF.getLangOpts().OverflowHandler;
Richard Smith4d1458e2012-09-08 02:08:36 +00003257 if (handlerName->empty()) {
Richard Smithb1b0ab42012-11-05 22:21:05 +00003258 // If the signed-integer-overflow sanitizer is enabled, emit a call to its
Richard Smithde670682012-11-01 22:15:34 +00003259 // runtime. Otherwise, this is a -ftrapv check, so just emit a trap.
Alexey Samsonovedf99a92014-11-07 22:29:38 +00003260 if (!isSigned || CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow)) {
Alexey Samsonove396bfc2014-11-11 22:03:54 +00003261 llvm::Value *NotOverflow = Builder.CreateNot(overflow);
Peter Collingbourne3eea6772015-05-11 21:39:14 +00003262 SanitizerMask Kind = isSigned ? SanitizerKind::SignedIntegerOverflow
Alexey Samsonove396bfc2014-11-11 22:03:54 +00003263 : SanitizerKind::UnsignedIntegerOverflow;
3264 EmitBinOpCheck(std::make_pair(NotOverflow, Kind), Ops);
Alexey Samsonov24cad992014-07-17 18:46:27 +00003265 } else
Chad Rosierae229d52013-01-29 23:31:22 +00003266 CGF.EmitTrapCheck(Builder.CreateNot(overflow));
Richard Smith4d1458e2012-09-08 02:08:36 +00003267 return result;
3268 }
3269
Mike Stump0c61b732009-04-01 20:28:16 +00003270 // Branch in case of overflow.
David Chisnalldd84ef12010-09-17 18:29:54 +00003271 llvm::BasicBlock *initialBB = Builder.GetInsertBlock();
Duncan P. N. Exon Smith01f574c2016-08-17 03:15:29 +00003272 llvm::BasicBlock *continueBB =
3273 CGF.createBasicBlock("nooverflow", CGF.CurFn, initialBB->getNextNode());
Chris Lattner8139c982010-08-07 00:20:46 +00003274 llvm::BasicBlock *overflowBB = CGF.createBasicBlock("overflow", CGF.CurFn);
Mike Stump0c61b732009-04-01 20:28:16 +00003275
3276 Builder.CreateCondBr(overflow, overflowBB, continueBB);
3277
David Chisnalldd84ef12010-09-17 18:29:54 +00003278 // If an overflow handler is set, then we want to call it and then use its
3279 // result, if it returns.
3280 Builder.SetInsertPoint(overflowBB);
3281
3282 // Get the overflow handler.
Chris Lattnerece04092012-02-07 00:39:47 +00003283 llvm::Type *Int8Ty = CGF.Int8Ty;
Chris Lattnera5f58b02011-07-09 17:41:47 +00003284 llvm::Type *argTypes[] = { CGF.Int64Ty, CGF.Int64Ty, Int8Ty, Int8Ty };
David Chisnalldd84ef12010-09-17 18:29:54 +00003285 llvm::FunctionType *handlerTy =
3286 llvm::FunctionType::get(CGF.Int64Ty, argTypes, true);
James Y Knight9871db02019-02-05 16:42:33 +00003287 llvm::FunctionCallee handler =
3288 CGF.CGM.CreateRuntimeFunction(handlerTy, *handlerName);
David Chisnalldd84ef12010-09-17 18:29:54 +00003289
3290 // Sign extend the args to 64-bit, so that we can use the same handler for
3291 // all types of overflow.
3292 llvm::Value *lhs = Builder.CreateSExt(Ops.LHS, CGF.Int64Ty);
3293 llvm::Value *rhs = Builder.CreateSExt(Ops.RHS, CGF.Int64Ty);
3294
3295 // Call the handler with the two arguments, the operation, and the size of
3296 // the result.
John McCall882987f2013-02-28 19:01:20 +00003297 llvm::Value *handlerArgs[] = {
3298 lhs,
3299 rhs,
3300 Builder.getInt8(OpID),
3301 Builder.getInt8(cast<llvm::IntegerType>(opTy)->getBitWidth())
3302 };
3303 llvm::Value *handlerResult =
3304 CGF.EmitNounwindRuntimeCall(handler, handlerArgs);
David Chisnalldd84ef12010-09-17 18:29:54 +00003305
3306 // Truncate the result back to the desired size.
3307 handlerResult = Builder.CreateTrunc(handlerResult, opTy);
3308 Builder.CreateBr(continueBB);
3309
Mike Stump0c61b732009-04-01 20:28:16 +00003310 Builder.SetInsertPoint(continueBB);
Jay Foad20c0f022011-03-30 11:28:58 +00003311 llvm::PHINode *phi = Builder.CreatePHI(opTy, 2);
David Chisnalldd84ef12010-09-17 18:29:54 +00003312 phi->addIncoming(result, initialBB);
3313 phi->addIncoming(handlerResult, overflowBB);
3314
3315 return phi;
Mike Stump0c61b732009-04-01 20:28:16 +00003316}
Chris Lattner2da04b32007-08-24 05:35:26 +00003317
John McCall77527a82011-06-25 01:32:37 +00003318/// Emit pointer + index arithmetic.
3319static Value *emitPointerArithmetic(CodeGenFunction &CGF,
3320 const BinOpInfo &op,
3321 bool isSubtraction) {
3322 // Must have binary (not unary) expr here. Unary pointer
3323 // increment/decrement doesn't use this path.
3324 const BinaryOperator *expr = cast<BinaryOperator>(op.E);
Craig Toppera97d7e72013-07-26 06:16:11 +00003325
John McCall77527a82011-06-25 01:32:37 +00003326 Value *pointer = op.LHS;
3327 Expr *pointerOperand = expr->getLHS();
3328 Value *index = op.RHS;
3329 Expr *indexOperand = expr->getRHS();
3330
3331 // In a subtraction, the LHS is always the pointer.
3332 if (!isSubtraction && !pointer->getType()->isPointerTy()) {
3333 std::swap(pointer, index);
3334 std::swap(pointerOperand, indexOperand);
3335 }
3336
Vedant Kumar6dbf4272017-06-12 18:42:51 +00003337 bool isSigned = indexOperand->getType()->isSignedIntegerOrEnumerationType();
Vedant Kumar6dbf4272017-06-12 18:42:51 +00003338
John McCall77527a82011-06-25 01:32:37 +00003339 unsigned width = cast<llvm::IntegerType>(index->getType())->getBitWidth();
Yaxun Liu26f75662016-08-19 05:17:25 +00003340 auto &DL = CGF.CGM.getDataLayout();
3341 auto PtrTy = cast<llvm::PointerType>(pointer->getType());
Andrew Kaylor3d0a5402017-09-19 20:26:40 +00003342
3343 // Some versions of glibc and gcc use idioms (particularly in their malloc
3344 // routines) that add a pointer-sized integer (known to be a pointer value)
3345 // to a null pointer in order to cast the value back to an integer or as
3346 // part of a pointer alignment algorithm. This is undefined behavior, but
3347 // we'd like to be able to compile programs that use it.
3348 //
3349 // Normally, we'd generate a GEP with a null-pointer base here in response
3350 // to that code, but it's also UB to dereference a pointer created that
3351 // way. Instead (as an acknowledged hack to tolerate the idiom) we will
3352 // generate a direct cast of the integer value to a pointer.
3353 //
3354 // The idiom (p = nullptr + N) is not met if any of the following are true:
3355 //
3356 // The operation is subtraction.
3357 // The index is not pointer-sized.
3358 // The pointer type is not byte-sized.
3359 //
3360 if (BinaryOperator::isNullPointerArithmeticExtension(CGF.getContext(),
3361 op.Opcode,
Fangrui Song6907ce22018-07-30 19:24:48 +00003362 expr->getLHS(),
Andrew Kaylor3d0a5402017-09-19 20:26:40 +00003363 expr->getRHS()))
3364 return CGF.Builder.CreateIntToPtr(index, pointer->getType());
3365
Nicola Zaghen97572772019-12-13 09:55:45 +00003366 if (width != DL.getIndexTypeSizeInBits(PtrTy)) {
John McCall77527a82011-06-25 01:32:37 +00003367 // Zero-extend or sign-extend the pointer value according to
3368 // whether the index is signed or not.
Nicola Zaghen97572772019-12-13 09:55:45 +00003369 index = CGF.Builder.CreateIntCast(index, DL.getIndexType(PtrTy), isSigned,
John McCall77527a82011-06-25 01:32:37 +00003370 "idx.ext");
3371 }
3372
3373 // If this is subtraction, negate the index.
3374 if (isSubtraction)
3375 index = CGF.Builder.CreateNeg(index, "idx.neg");
3376
Alexey Samsonovedf99a92014-11-07 22:29:38 +00003377 if (CGF.SanOpts.has(SanitizerKind::ArrayBounds))
Richard Smith539e4a72013-02-23 02:53:19 +00003378 CGF.EmitBoundsCheck(op.E, pointerOperand, index, indexOperand->getType(),
3379 /*Accessed*/ false);
3380
John McCall77527a82011-06-25 01:32:37 +00003381 const PointerType *pointerType
3382 = pointerOperand->getType()->getAs<PointerType>();
3383 if (!pointerType) {
3384 QualType objectType = pointerOperand->getType()
3385 ->castAs<ObjCObjectPointerType>()
3386 ->getPointeeType();
3387 llvm::Value *objectSize
3388 = CGF.CGM.getSize(CGF.getContext().getTypeSizeInChars(objectType));
3389
3390 index = CGF.Builder.CreateMul(index, objectSize);
3391
3392 Value *result = CGF.Builder.CreateBitCast(pointer, CGF.VoidPtrTy);
3393 result = CGF.Builder.CreateGEP(result, index, "add.ptr");
3394 return CGF.Builder.CreateBitCast(result, pointer->getType());
3395 }
3396
3397 QualType elementType = pointerType->getPointeeType();
3398 if (const VariableArrayType *vla
3399 = CGF.getContext().getAsVariableArrayType(elementType)) {
3400 // The element count here is the total number of non-VLA elements.
Sander de Smalen891af03a2018-02-03 13:55:59 +00003401 llvm::Value *numElements = CGF.getVLASize(vla).NumElts;
John McCall77527a82011-06-25 01:32:37 +00003402
3403 // Effectively, the multiply by the VLA size is part of the GEP.
3404 // GEP indexes are signed, and scaling an index isn't permitted to
3405 // signed-overflow, so we use the same semantics for our explicit
3406 // multiply. We suppress this if overflow is not undefined behavior.
David Blaikiebbafb8a2012-03-11 07:00:24 +00003407 if (CGF.getLangOpts().isSignedOverflowDefined()) {
John McCall77527a82011-06-25 01:32:37 +00003408 index = CGF.Builder.CreateMul(index, numElements, "vla.index");
3409 pointer = CGF.Builder.CreateGEP(pointer, index, "add.ptr");
3410 } else {
3411 index = CGF.Builder.CreateNSWMul(index, numElements, "vla.index");
Vedant Kumar6dbf4272017-06-12 18:42:51 +00003412 pointer =
Vedant Kumar175b6d12017-07-13 20:55:26 +00003413 CGF.EmitCheckedInBoundsGEP(pointer, index, isSigned, isSubtraction,
Vedant Kumar6dbf4272017-06-12 18:42:51 +00003414 op.E->getExprLoc(), "add.ptr");
Chris Lattner51924e512010-06-26 21:25:03 +00003415 }
John McCall77527a82011-06-25 01:32:37 +00003416 return pointer;
Mike Stump4a3999f2009-09-09 13:00:44 +00003417 }
Daniel Dunbaref2ffbc2009-04-25 05:08:32 +00003418
Mike Stump4a3999f2009-09-09 13:00:44 +00003419 // Explicitly handle GNU void* and function pointer arithmetic extensions. The
3420 // GNU void* casts amount to no-ops since our void* type is i8*, but this is
3421 // future proof.
John McCall77527a82011-06-25 01:32:37 +00003422 if (elementType->isVoidType() || elementType->isFunctionType()) {
Matt Arsenaultc6da9ec2019-10-31 08:41:37 -07003423 Value *result = CGF.EmitCastToVoidPtr(pointer);
John McCall77527a82011-06-25 01:32:37 +00003424 result = CGF.Builder.CreateGEP(result, index, "add.ptr");
3425 return CGF.Builder.CreateBitCast(result, pointer->getType());
Mike Stump4a3999f2009-09-09 13:00:44 +00003426 }
3427
David Blaikiebbafb8a2012-03-11 07:00:24 +00003428 if (CGF.getLangOpts().isSignedOverflowDefined())
John McCall77527a82011-06-25 01:32:37 +00003429 return CGF.Builder.CreateGEP(pointer, index, "add.ptr");
3430
Vedant Kumar175b6d12017-07-13 20:55:26 +00003431 return CGF.EmitCheckedInBoundsGEP(pointer, index, isSigned, isSubtraction,
Vedant Kumar6dbf4272017-06-12 18:42:51 +00003432 op.E->getExprLoc(), "add.ptr");
Chris Lattner2da04b32007-08-24 05:35:26 +00003433}
3434
Lang Hames5de91cc2012-10-02 04:45:10 +00003435// Construct an fmuladd intrinsic to represent a fused mul-add of MulOp and
3436// Addend. Use negMul and negAdd to negate the first operand of the Mul or
3437// the add operand respectively. This allows fmuladd to represent a*b-c, or
3438// c-a*b. Patterns in LLVM should catch the negated forms and translate them to
3439// efficient operations.
Wang, Pengfei3239b502020-01-15 19:08:38 +08003440static Value* buildFMulAdd(llvm::Instruction *MulOp, Value *Addend,
Lang Hames5de91cc2012-10-02 04:45:10 +00003441 const CodeGenFunction &CGF, CGBuilderTy &Builder,
3442 bool negMul, bool negAdd) {
3443 assert(!(negMul && negAdd) && "Only one of negMul and negAdd should be set.");
Craig Toppera97d7e72013-07-26 06:16:11 +00003444
Lang Hames5de91cc2012-10-02 04:45:10 +00003445 Value *MulOp0 = MulOp->getOperand(0);
3446 Value *MulOp1 = MulOp->getOperand(1);
Craig Topper8b23b2b2019-12-30 13:24:08 -08003447 if (negMul)
3448 MulOp0 = Builder.CreateFNeg(MulOp0, "neg");
3449 if (negAdd)
3450 Addend = Builder.CreateFNeg(Addend, "neg");
Lang Hames5de91cc2012-10-02 04:45:10 +00003451
Wang, Pengfei3239b502020-01-15 19:08:38 +08003452 Value *FMulAdd = nullptr;
3453 if (Builder.getIsFPConstrained()) {
3454 assert(isa<llvm::ConstrainedFPIntrinsic>(MulOp) &&
3455 "Only constrained operation should be created when Builder is in FP "
3456 "constrained mode");
3457 FMulAdd = Builder.CreateConstrainedFPCall(
3458 CGF.CGM.getIntrinsic(llvm::Intrinsic::experimental_constrained_fmuladd,
3459 Addend->getType()),
3460 {MulOp0, MulOp1, Addend});
3461 } else {
3462 FMulAdd = Builder.CreateCall(
3463 CGF.CGM.getIntrinsic(llvm::Intrinsic::fmuladd, Addend->getType()),
3464 {MulOp0, MulOp1, Addend});
3465 }
3466 MulOp->eraseFromParent();
Lang Hames5de91cc2012-10-02 04:45:10 +00003467
Wang, Pengfei3239b502020-01-15 19:08:38 +08003468 return FMulAdd;
Lang Hames5de91cc2012-10-02 04:45:10 +00003469}
3470
3471// Check whether it would be legal to emit an fmuladd intrinsic call to
3472// represent op and if so, build the fmuladd.
3473//
3474// Checks that (a) the operation is fusable, and (b) -ffp-contract=on.
3475// Does NOT check the type of the operation - it's assumed that this function
3476// will be called from contexts where it's known that the type is contractable.
Craig Toppera97d7e72013-07-26 06:16:11 +00003477static Value* tryEmitFMulAdd(const BinOpInfo &op,
Lang Hames5de91cc2012-10-02 04:45:10 +00003478 const CodeGenFunction &CGF, CGBuilderTy &Builder,
3479 bool isSub=false) {
3480
3481 assert((op.Opcode == BO_Add || op.Opcode == BO_AddAssign ||
3482 op.Opcode == BO_Sub || op.Opcode == BO_SubAssign) &&
3483 "Only fadd/fsub can be the root of an fmuladd.");
3484
3485 // Check whether this op is marked as fusable.
Adam Nemet049a31d2017-03-29 21:54:24 +00003486 if (!op.FPFeatures.allowFPContractWithinStatement())
Craig Topper8a13c412014-05-21 05:09:00 +00003487 return nullptr;
Lang Hames5de91cc2012-10-02 04:45:10 +00003488
3489 // We have a potentially fusable op. Look for a mul on one of the operands.
Sanjay Patela30cee62015-12-03 01:25:12 +00003490 // Also, make sure that the mul result isn't used directly. In that case,
3491 // there's no point creating a muladd operation.
3492 if (auto *LHSBinOp = dyn_cast<llvm::BinaryOperator>(op.LHS)) {
3493 if (LHSBinOp->getOpcode() == llvm::Instruction::FMul &&
3494 LHSBinOp->use_empty())
Lang Hames5de91cc2012-10-02 04:45:10 +00003495 return buildFMulAdd(LHSBinOp, op.RHS, CGF, Builder, false, isSub);
Sanjay Patela30cee62015-12-03 01:25:12 +00003496 }
3497 if (auto *RHSBinOp = dyn_cast<llvm::BinaryOperator>(op.RHS)) {
3498 if (RHSBinOp->getOpcode() == llvm::Instruction::FMul &&
3499 RHSBinOp->use_empty())
Lang Hames5de91cc2012-10-02 04:45:10 +00003500 return buildFMulAdd(RHSBinOp, op.LHS, CGF, Builder, isSub, false);
Lang Hames5de91cc2012-10-02 04:45:10 +00003501 }
3502
Wang, Pengfei3239b502020-01-15 19:08:38 +08003503 if (auto *LHSBinOp = dyn_cast<llvm::CallBase>(op.LHS)) {
3504 if (LHSBinOp->getIntrinsicID() ==
3505 llvm::Intrinsic::experimental_constrained_fmul &&
3506 LHSBinOp->use_empty())
3507 return buildFMulAdd(LHSBinOp, op.RHS, CGF, Builder, false, isSub);
3508 }
3509 if (auto *RHSBinOp = dyn_cast<llvm::CallBase>(op.RHS)) {
3510 if (RHSBinOp->getIntrinsicID() ==
3511 llvm::Intrinsic::experimental_constrained_fmul &&
3512 RHSBinOp->use_empty())
3513 return buildFMulAdd(RHSBinOp, op.LHS, CGF, Builder, isSub, false);
3514 }
3515
Craig Topper8a13c412014-05-21 05:09:00 +00003516 return nullptr;
Lang Hames5de91cc2012-10-02 04:45:10 +00003517}
3518
John McCall77527a82011-06-25 01:32:37 +00003519Value *ScalarExprEmitter::EmitAdd(const BinOpInfo &op) {
3520 if (op.LHS->getType()->isPointerTy() ||
3521 op.RHS->getType()->isPointerTy())
Vedant Kumar175b6d12017-07-13 20:55:26 +00003522 return emitPointerArithmetic(CGF, op, CodeGenFunction::NotSubtraction);
John McCall77527a82011-06-25 01:32:37 +00003523
3524 if (op.Ty->isSignedIntegerOrEnumerationType()) {
Richard Smith9c6890a2012-11-01 22:30:59 +00003525 switch (CGF.getLangOpts().getSignedOverflowBehavior()) {
John McCall77527a82011-06-25 01:32:37 +00003526 case LangOptions::SOB_Defined:
3527 return Builder.CreateAdd(op.LHS, op.RHS, "add");
Richard Smith3e056de2012-08-25 00:32:28 +00003528 case LangOptions::SOB_Undefined:
Alexey Samsonovedf99a92014-11-07 22:29:38 +00003529 if (!CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow))
Richard Smith3e056de2012-08-25 00:32:28 +00003530 return Builder.CreateNSWAdd(op.LHS, op.RHS, "add");
Reid Kleckner4dc0b1a2018-11-01 19:54:45 +00003531 LLVM_FALLTHROUGH;
John McCall77527a82011-06-25 01:32:37 +00003532 case LangOptions::SOB_Trapping:
Vedant Kumar82ee16b2017-02-25 00:43:36 +00003533 if (CanElideOverflowCheck(CGF.getContext(), op))
3534 return Builder.CreateNSWAdd(op.LHS, op.RHS, "add");
John McCall77527a82011-06-25 01:32:37 +00003535 return EmitOverflowCheckedBinOp(op);
3536 }
3537 }
Will Dietz1897cb32012-11-27 15:01:55 +00003538
Florian Hahn6f6e91d2020-05-29 20:42:22 +01003539 if (op.Ty->isConstantMatrixType()) {
3540 llvm::MatrixBuilder<CGBuilderTy> MB(Builder);
3541 return MB.CreateAdd(op.LHS, op.RHS);
3542 }
3543
Alexey Samsonovedf99a92014-11-07 22:29:38 +00003544 if (op.Ty->isUnsignedIntegerType() &&
Vedant Kumar82ee16b2017-02-25 00:43:36 +00003545 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow) &&
3546 !CanElideOverflowCheck(CGF.getContext(), op))
Will Dietz1897cb32012-11-27 15:01:55 +00003547 return EmitOverflowCheckedBinOp(op);
3548
Lang Hames5de91cc2012-10-02 04:45:10 +00003549 if (op.LHS->getType()->isFPOrFPVectorTy()) {
John McCall7fac1ac2020-06-11 18:09:36 -04003550 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, op.FPFeatures);
Lang Hames5de91cc2012-10-02 04:45:10 +00003551 // Try to form an fmuladd.
3552 if (Value *FMulAdd = tryEmitFMulAdd(op, CGF, Builder))
3553 return FMulAdd;
3554
John McCall8a8d7032020-06-01 21:02:02 -04003555 return Builder.CreateFAdd(op.LHS, op.RHS, "add");
Lang Hames5de91cc2012-10-02 04:45:10 +00003556 }
John McCall77527a82011-06-25 01:32:37 +00003557
Bevin Hansson39baaab2020-01-08 11:12:55 +01003558 if (op.isFixedPointOp())
Leonard Chan837da5d2019-01-16 19:53:50 +00003559 return EmitFixedPointBinOp(op);
Leonard Chan2044ac82019-01-16 18:13:59 +00003560
John McCall77527a82011-06-25 01:32:37 +00003561 return Builder.CreateAdd(op.LHS, op.RHS, "add");
3562}
3563
Leonard Chan2044ac82019-01-16 18:13:59 +00003564/// The resulting value must be calculated with exact precision, so the operands
3565/// may not be the same type.
Leonard Chan837da5d2019-01-16 19:53:50 +00003566Value *ScalarExprEmitter::EmitFixedPointBinOp(const BinOpInfo &op) {
Leonard Chan2044ac82019-01-16 18:13:59 +00003567 using llvm::APSInt;
3568 using llvm::ConstantInt;
3569
Bevin Hansson39baaab2020-01-08 11:12:55 +01003570 // This is either a binary operation where at least one of the operands is
3571 // a fixed-point type, or a unary operation where the operand is a fixed-point
3572 // type. The result type of a binary operation is determined by
3573 // Sema::handleFixedPointConversions().
Leonard Chan2044ac82019-01-16 18:13:59 +00003574 QualType ResultTy = op.Ty;
Bevin Hansson39baaab2020-01-08 11:12:55 +01003575 QualType LHSTy, RHSTy;
3576 if (const auto *BinOp = dyn_cast<BinaryOperator>(op.E)) {
Bevin Hansson39baaab2020-01-08 11:12:55 +01003577 RHSTy = BinOp->getRHS()->getType();
Bevin Hansson313461f2020-01-08 14:01:30 +01003578 if (const auto *CAO = dyn_cast<CompoundAssignOperator>(BinOp)) {
3579 // For compound assignment, the effective type of the LHS at this point
3580 // is the computation LHS type, not the actual LHS type, and the final
3581 // result type is not the type of the expression but rather the
3582 // computation result type.
3583 LHSTy = CAO->getComputationLHSType();
3584 ResultTy = CAO->getComputationResultType();
3585 } else
3586 LHSTy = BinOp->getLHS()->getType();
Bevin Hansson39baaab2020-01-08 11:12:55 +01003587 } else if (const auto *UnOp = dyn_cast<UnaryOperator>(op.E)) {
3588 LHSTy = UnOp->getSubExpr()->getType();
3589 RHSTy = UnOp->getSubExpr()->getType();
3590 }
Leonard Chan2044ac82019-01-16 18:13:59 +00003591 ASTContext &Ctx = CGF.getContext();
3592 Value *LHS = op.LHS;
3593 Value *RHS = op.RHS;
3594
3595 auto LHSFixedSema = Ctx.getFixedPointSemantics(LHSTy);
3596 auto RHSFixedSema = Ctx.getFixedPointSemantics(RHSTy);
3597 auto ResultFixedSema = Ctx.getFixedPointSemantics(ResultTy);
3598 auto CommonFixedSema = LHSFixedSema.getCommonSemantics(RHSFixedSema);
3599
3600 // Convert the operands to the full precision type.
3601 Value *FullLHS = EmitFixedPointConversion(LHS, LHSFixedSema, CommonFixedSema,
Bevin Hansson39baaab2020-01-08 11:12:55 +01003602 op.E->getExprLoc());
Leonard Chan2044ac82019-01-16 18:13:59 +00003603 Value *FullRHS = EmitFixedPointConversion(RHS, RHSFixedSema, CommonFixedSema,
Bevin Hansson39baaab2020-01-08 11:12:55 +01003604 op.E->getExprLoc());
Leonard Chan2044ac82019-01-16 18:13:59 +00003605
Bevin Hansson313461f2020-01-08 14:01:30 +01003606 // Perform the actual operation.
Leonard Chan2044ac82019-01-16 18:13:59 +00003607 Value *Result;
Bevin Hansson39baaab2020-01-08 11:12:55 +01003608 switch (op.Opcode) {
Bevin Hansson313461f2020-01-08 14:01:30 +01003609 case BO_AddAssign:
Leonard Chan837da5d2019-01-16 19:53:50 +00003610 case BO_Add: {
3611 if (ResultFixedSema.isSaturated()) {
3612 llvm::Intrinsic::ID IID = ResultFixedSema.isSigned()
3613 ? llvm::Intrinsic::sadd_sat
3614 : llvm::Intrinsic::uadd_sat;
3615 Result = Builder.CreateBinaryIntrinsic(IID, FullLHS, FullRHS);
3616 } else {
3617 Result = Builder.CreateAdd(FullLHS, FullRHS);
3618 }
3619 break;
3620 }
Bevin Hansson313461f2020-01-08 14:01:30 +01003621 case BO_SubAssign:
Leonard Chan837da5d2019-01-16 19:53:50 +00003622 case BO_Sub: {
3623 if (ResultFixedSema.isSaturated()) {
3624 llvm::Intrinsic::ID IID = ResultFixedSema.isSigned()
3625 ? llvm::Intrinsic::ssub_sat
3626 : llvm::Intrinsic::usub_sat;
3627 Result = Builder.CreateBinaryIntrinsic(IID, FullLHS, FullRHS);
3628 } else {
3629 Result = Builder.CreateSub(FullLHS, FullRHS);
3630 }
3631 break;
3632 }
Bevin Hansson313461f2020-01-08 14:01:30 +01003633 case BO_MulAssign:
Bevin Hansson0b9922e2019-11-19 13:15:06 +01003634 case BO_Mul: {
3635 llvm::Intrinsic::ID IID;
3636 if (ResultFixedSema.isSaturated())
3637 IID = ResultFixedSema.isSigned()
3638 ? llvm::Intrinsic::smul_fix_sat
3639 : llvm::Intrinsic::umul_fix_sat;
3640 else
3641 IID = ResultFixedSema.isSigned()
3642 ? llvm::Intrinsic::smul_fix
3643 : llvm::Intrinsic::umul_fix;
3644 Result = Builder.CreateIntrinsic(IID, {FullLHS->getType()},
3645 {FullLHS, FullRHS, Builder.getInt32(CommonFixedSema.getScale())});
3646 break;
3647 }
Bevin Hansson313461f2020-01-08 14:01:30 +01003648 case BO_DivAssign:
Bevin Hansson0b9922e2019-11-19 13:15:06 +01003649 case BO_Div: {
3650 llvm::Intrinsic::ID IID;
3651 if (ResultFixedSema.isSaturated())
3652 IID = ResultFixedSema.isSigned() ? llvm::Intrinsic::sdiv_fix_sat
3653 : llvm::Intrinsic::udiv_fix_sat;
3654 else
3655 IID = ResultFixedSema.isSigned() ? llvm::Intrinsic::sdiv_fix
3656 : llvm::Intrinsic::udiv_fix;
3657 Result = Builder.CreateIntrinsic(IID, {FullLHS->getType()},
3658 {FullLHS, FullRHS, Builder.getInt32(CommonFixedSema.getScale())});
3659 break;
3660 }
Leonard Chance1d4f12019-02-21 20:50:09 +00003661 case BO_LT:
3662 return CommonFixedSema.isSigned() ? Builder.CreateICmpSLT(FullLHS, FullRHS)
3663 : Builder.CreateICmpULT(FullLHS, FullRHS);
3664 case BO_GT:
3665 return CommonFixedSema.isSigned() ? Builder.CreateICmpSGT(FullLHS, FullRHS)
3666 : Builder.CreateICmpUGT(FullLHS, FullRHS);
3667 case BO_LE:
3668 return CommonFixedSema.isSigned() ? Builder.CreateICmpSLE(FullLHS, FullRHS)
3669 : Builder.CreateICmpULE(FullLHS, FullRHS);
3670 case BO_GE:
3671 return CommonFixedSema.isSigned() ? Builder.CreateICmpSGE(FullLHS, FullRHS)
3672 : Builder.CreateICmpUGE(FullLHS, FullRHS);
3673 case BO_EQ:
3674 // For equality operations, we assume any padding bits on unsigned types are
3675 // zero'd out. They could be overwritten through non-saturating operations
3676 // that cause overflow, but this leads to undefined behavior.
3677 return Builder.CreateICmpEQ(FullLHS, FullRHS);
3678 case BO_NE:
3679 return Builder.CreateICmpNE(FullLHS, FullRHS);
Leonard Chan837da5d2019-01-16 19:53:50 +00003680 case BO_Shl:
3681 case BO_Shr:
3682 case BO_Cmp:
Leonard Chan837da5d2019-01-16 19:53:50 +00003683 case BO_LAnd:
3684 case BO_LOr:
Leonard Chan837da5d2019-01-16 19:53:50 +00003685 case BO_ShlAssign:
3686 case BO_ShrAssign:
3687 llvm_unreachable("Found unimplemented fixed point binary operation");
3688 case BO_PtrMemD:
3689 case BO_PtrMemI:
3690 case BO_Rem:
3691 case BO_Xor:
3692 case BO_And:
3693 case BO_Or:
3694 case BO_Assign:
3695 case BO_RemAssign:
3696 case BO_AndAssign:
3697 case BO_XorAssign:
3698 case BO_OrAssign:
3699 case BO_Comma:
3700 llvm_unreachable("Found unsupported binary operation for fixed point types.");
Leonard Chan2044ac82019-01-16 18:13:59 +00003701 }
3702
3703 // Convert to the result type.
3704 return EmitFixedPointConversion(Result, CommonFixedSema, ResultFixedSema,
Bevin Hansson39baaab2020-01-08 11:12:55 +01003705 op.E->getExprLoc());
Leonard Chan2044ac82019-01-16 18:13:59 +00003706}
3707
John McCall77527a82011-06-25 01:32:37 +00003708Value *ScalarExprEmitter::EmitSub(const BinOpInfo &op) {
3709 // The LHS is always a pointer if either side is.
3710 if (!op.LHS->getType()->isPointerTy()) {
3711 if (op.Ty->isSignedIntegerOrEnumerationType()) {
Richard Smith9c6890a2012-11-01 22:30:59 +00003712 switch (CGF.getLangOpts().getSignedOverflowBehavior()) {
Chris Lattner51924e512010-06-26 21:25:03 +00003713 case LangOptions::SOB_Defined:
John McCall77527a82011-06-25 01:32:37 +00003714 return Builder.CreateSub(op.LHS, op.RHS, "sub");
Richard Smith3e056de2012-08-25 00:32:28 +00003715 case LangOptions::SOB_Undefined:
Alexey Samsonovedf99a92014-11-07 22:29:38 +00003716 if (!CGF.SanOpts.has(SanitizerKind::SignedIntegerOverflow))
Richard Smith3e056de2012-08-25 00:32:28 +00003717 return Builder.CreateNSWSub(op.LHS, op.RHS, "sub");
Reid Kleckner4dc0b1a2018-11-01 19:54:45 +00003718 LLVM_FALLTHROUGH;
Chris Lattner51924e512010-06-26 21:25:03 +00003719 case LangOptions::SOB_Trapping:
Vedant Kumar82ee16b2017-02-25 00:43:36 +00003720 if (CanElideOverflowCheck(CGF.getContext(), op))
3721 return Builder.CreateNSWSub(op.LHS, op.RHS, "sub");
John McCall77527a82011-06-25 01:32:37 +00003722 return EmitOverflowCheckedBinOp(op);
Chris Lattner51924e512010-06-26 21:25:03 +00003723 }
3724 }
Will Dietz1897cb32012-11-27 15:01:55 +00003725
Florian Hahn6f6e91d2020-05-29 20:42:22 +01003726 if (op.Ty->isConstantMatrixType()) {
3727 llvm::MatrixBuilder<CGBuilderTy> MB(Builder);
3728 return MB.CreateSub(op.LHS, op.RHS);
3729 }
3730
Alexey Samsonovedf99a92014-11-07 22:29:38 +00003731 if (op.Ty->isUnsignedIntegerType() &&
Vedant Kumar82ee16b2017-02-25 00:43:36 +00003732 CGF.SanOpts.has(SanitizerKind::UnsignedIntegerOverflow) &&
3733 !CanElideOverflowCheck(CGF.getContext(), op))
Will Dietz1897cb32012-11-27 15:01:55 +00003734 return EmitOverflowCheckedBinOp(op);
3735
Lang Hames5de91cc2012-10-02 04:45:10 +00003736 if (op.LHS->getType()->isFPOrFPVectorTy()) {
John McCall7fac1ac2020-06-11 18:09:36 -04003737 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, op.FPFeatures);
Lang Hames5de91cc2012-10-02 04:45:10 +00003738 // Try to form an fmuladd.
3739 if (Value *FMulAdd = tryEmitFMulAdd(op, CGF, Builder, true))
3740 return FMulAdd;
John McCall8a8d7032020-06-01 21:02:02 -04003741 return Builder.CreateFSub(op.LHS, op.RHS, "sub");
Lang Hames5de91cc2012-10-02 04:45:10 +00003742 }
Chris Lattner5902e7b2010-03-29 17:28:16 +00003743
Bevin Hansson39baaab2020-01-08 11:12:55 +01003744 if (op.isFixedPointOp())
Leonard Chan837da5d2019-01-16 19:53:50 +00003745 return EmitFixedPointBinOp(op);
3746
John McCall77527a82011-06-25 01:32:37 +00003747 return Builder.CreateSub(op.LHS, op.RHS, "sub");
Mike Stump0c61b732009-04-01 20:28:16 +00003748 }
Chris Lattner3d966d62007-08-24 21:00:35 +00003749
John McCall77527a82011-06-25 01:32:37 +00003750 // If the RHS is not a pointer, then we have normal pointer
3751 // arithmetic.
3752 if (!op.RHS->getType()->isPointerTy())
Vedant Kumar175b6d12017-07-13 20:55:26 +00003753 return emitPointerArithmetic(CGF, op, CodeGenFunction::IsSubtraction);
Eli Friedmane381f7e2009-03-28 02:45:41 +00003754
John McCall77527a82011-06-25 01:32:37 +00003755 // Otherwise, this is a pointer subtraction.
Daniel Dunbar42a8cd32009-01-23 18:51:09 +00003756
John McCall77527a82011-06-25 01:32:37 +00003757 // Do the raw subtraction part.
3758 llvm::Value *LHS
3759 = Builder.CreatePtrToInt(op.LHS, CGF.PtrDiffTy, "sub.ptr.lhs.cast");
3760 llvm::Value *RHS
3761 = Builder.CreatePtrToInt(op.RHS, CGF.PtrDiffTy, "sub.ptr.rhs.cast");
3762 Value *diffInChars = Builder.CreateSub(LHS, RHS, "sub.ptr.sub");
Daniel Dunbaref2ffbc2009-04-25 05:08:32 +00003763
John McCall77527a82011-06-25 01:32:37 +00003764 // Okay, figure out the element size.
3765 const BinaryOperator *expr = cast<BinaryOperator>(op.E);
3766 QualType elementType = expr->getLHS()->getType()->getPointeeType();
Mike Stump4a3999f2009-09-09 13:00:44 +00003767
Craig Topper8a13c412014-05-21 05:09:00 +00003768 llvm::Value *divisor = nullptr;
John McCall77527a82011-06-25 01:32:37 +00003769
3770 // For a variable-length array, this is going to be non-constant.
3771 if (const VariableArrayType *vla
3772 = CGF.getContext().getAsVariableArrayType(elementType)) {
Sander de Smalen891af03a2018-02-03 13:55:59 +00003773 auto VlaSize = CGF.getVLASize(vla);
3774 elementType = VlaSize.Type;
3775 divisor = VlaSize.NumElts;
John McCall77527a82011-06-25 01:32:37 +00003776
3777 // Scale the number of non-VLA elements by the non-VLA element size.
3778 CharUnits eltSize = CGF.getContext().getTypeSizeInChars(elementType);
3779 if (!eltSize.isOne())
3780 divisor = CGF.Builder.CreateNUWMul(CGF.CGM.getSize(eltSize), divisor);
3781
3782 // For everything elese, we can just compute it, safe in the
3783 // assumption that Sema won't let anything through that we can't
3784 // safely compute the size of.
3785 } else {
3786 CharUnits elementSize;
3787 // Handle GCC extension for pointer arithmetic on void* and
3788 // function pointer types.
3789 if (elementType->isVoidType() || elementType->isFunctionType())
3790 elementSize = CharUnits::One();
3791 else
3792 elementSize = CGF.getContext().getTypeSizeInChars(elementType);
3793
3794 // Don't even emit the divide for element size of 1.
3795 if (elementSize.isOne())
3796 return diffInChars;
3797
3798 divisor = CGF.CGM.getSize(elementSize);
Chris Lattner2da04b32007-08-24 05:35:26 +00003799 }
Craig Toppera97d7e72013-07-26 06:16:11 +00003800
Chris Lattner2e72da942011-03-01 00:03:48 +00003801 // Otherwise, do a full sdiv. This uses the "exact" form of sdiv, since
3802 // pointer difference in C is only defined in the case where both operands
3803 // are pointing to elements of an array.
John McCall77527a82011-06-25 01:32:37 +00003804 return Builder.CreateExactSDiv(diffInChars, divisor, "sub.ptr.div");
Chris Lattner2da04b32007-08-24 05:35:26 +00003805}
3806
David Tweed042e0882013-01-07 16:43:27 +00003807Value *ScalarExprEmitter::GetWidthMinusOneValue(Value* LHS,Value* RHS) {
David Tweed9fb566c2013-01-10 09:11:33 +00003808 llvm::IntegerType *Ty;
3809 if (llvm::VectorType *VT = dyn_cast<llvm::VectorType>(LHS->getType()))
3810 Ty = cast<llvm::IntegerType>(VT->getElementType());
3811 else
3812 Ty = cast<llvm::IntegerType>(LHS->getType());
3813 return llvm::ConstantInt::get(RHS->getType(), Ty->getBitWidth() - 1);
David Tweed042e0882013-01-07 16:43:27 +00003814}
3815
Erich Keane5f0903e2020-04-17 10:44:19 -07003816Value *ScalarExprEmitter::ConstrainShiftValue(Value *LHS, Value *RHS,
3817 const Twine &Name) {
3818 llvm::IntegerType *Ty;
3819 if (auto *VT = dyn_cast<llvm::VectorType>(LHS->getType()))
3820 Ty = cast<llvm::IntegerType>(VT->getElementType());
3821 else
3822 Ty = cast<llvm::IntegerType>(LHS->getType());
3823
3824 if (llvm::isPowerOf2_64(Ty->getBitWidth()))
3825 return Builder.CreateAnd(RHS, GetWidthMinusOneValue(LHS, RHS), Name);
3826
3827 return Builder.CreateURem(
3828 RHS, llvm::ConstantInt::get(RHS->getType(), Ty->getBitWidth()), Name);
3829}
3830
Chris Lattner2da04b32007-08-24 05:35:26 +00003831Value *ScalarExprEmitter::EmitShl(const BinOpInfo &Ops) {
3832 // LLVM requires the LHS and RHS to be the same type: promote or truncate the
3833 // RHS to the same size as the LHS.
3834 Value *RHS = Ops.RHS;
3835 if (Ops.LHS->getType() != RHS->getType())
3836 RHS = Builder.CreateIntCast(RHS, Ops.LHS->getType(), false, "sh_prom");
Mike Stump4a3999f2009-09-09 13:00:44 +00003837
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003838 bool SanitizeBase = CGF.SanOpts.has(SanitizerKind::ShiftBase) &&
James Molloy59802322016-08-16 09:45:36 +00003839 Ops.Ty->hasSignedIntegerRepresentation() &&
Richard Smith7939ba02019-06-25 01:45:26 +00003840 !CGF.getLangOpts().isSignedOverflowDefined() &&
Aaron Ballman6a308942020-04-21 15:37:19 -04003841 !CGF.getLangOpts().CPlusPlus20;
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003842 bool SanitizeExponent = CGF.SanOpts.has(SanitizerKind::ShiftExponent);
3843 // OpenCL 6.3j: shift values are effectively % word size of LHS.
3844 if (CGF.getLangOpts().OpenCL)
Erich Keane5f0903e2020-04-17 10:44:19 -07003845 RHS = ConstrainShiftValue(Ops.LHS, RHS, "shl.mask");
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003846 else if ((SanitizeBase || SanitizeExponent) &&
3847 isa<llvm::IntegerType>(Ops.LHS->getType())) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00003848 CodeGenFunction::SanitizerScope SanScope(&CGF);
Peter Collingbourne3eea6772015-05-11 21:39:14 +00003849 SmallVector<std::pair<Value *, SanitizerMask>, 2> Checks;
Vedant Kumard3a601b2017-01-30 23:38:54 +00003850 llvm::Value *WidthMinusOne = GetWidthMinusOneValue(Ops.LHS, Ops.RHS);
3851 llvm::Value *ValidExponent = Builder.CreateICmpULE(Ops.RHS, WidthMinusOne);
Richard Smith3e056de2012-08-25 00:32:28 +00003852
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003853 if (SanitizeExponent) {
3854 Checks.push_back(
3855 std::make_pair(ValidExponent, SanitizerKind::ShiftExponent));
3856 }
3857
3858 if (SanitizeBase) {
3859 // Check whether we are shifting any non-zero bits off the top of the
3860 // integer. We only emit this check if exponent is valid - otherwise
3861 // instructions below will have undefined behavior themselves.
Alexey Samsonov48a9db02015-03-05 21:57:35 +00003862 llvm::BasicBlock *Orig = Builder.GetInsertBlock();
3863 llvm::BasicBlock *Cont = CGF.createBasicBlock("cont");
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003864 llvm::BasicBlock *CheckShiftBase = CGF.createBasicBlock("check");
3865 Builder.CreateCondBr(ValidExponent, CheckShiftBase, Cont);
Vedant Kumard3a601b2017-01-30 23:38:54 +00003866 llvm::Value *PromotedWidthMinusOne =
3867 (RHS == Ops.RHS) ? WidthMinusOne
3868 : GetWidthMinusOneValue(Ops.LHS, RHS);
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003869 CGF.EmitBlock(CheckShiftBase);
Vedant Kumard3a601b2017-01-30 23:38:54 +00003870 llvm::Value *BitsShiftedOff = Builder.CreateLShr(
3871 Ops.LHS, Builder.CreateSub(PromotedWidthMinusOne, RHS, "shl.zeros",
3872 /*NUW*/ true, /*NSW*/ true),
3873 "shl.check");
Richard Smith3e056de2012-08-25 00:32:28 +00003874 if (CGF.getLangOpts().CPlusPlus) {
3875 // In C99, we are not permitted to shift a 1 bit into the sign bit.
3876 // Under C++11's rules, shifting a 1 bit into the sign bit is
3877 // OK, but shifting a 1 bit out of it is not. (C89 and C++03 don't
3878 // define signed left shifts, so we use the C99 and C++11 rules there).
3879 llvm::Value *One = llvm::ConstantInt::get(BitsShiftedOff->getType(), 1);
3880 BitsShiftedOff = Builder.CreateLShr(BitsShiftedOff, One);
3881 }
3882 llvm::Value *Zero = llvm::ConstantInt::get(BitsShiftedOff->getType(), 0);
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003883 llvm::Value *ValidBase = Builder.CreateICmpEQ(BitsShiftedOff, Zero);
Alexey Samsonov48a9db02015-03-05 21:57:35 +00003884 CGF.EmitBlock(Cont);
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003885 llvm::PHINode *BaseCheck = Builder.CreatePHI(ValidBase->getType(), 2);
3886 BaseCheck->addIncoming(Builder.getTrue(), Orig);
3887 BaseCheck->addIncoming(ValidBase, CheckShiftBase);
3888 Checks.push_back(std::make_pair(BaseCheck, SanitizerKind::ShiftBase));
Richard Smith3e056de2012-08-25 00:32:28 +00003889 }
Will Dietz11d0a9f2013-02-25 22:37:49 +00003890
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003891 assert(!Checks.empty());
3892 EmitBinOpCheck(Checks, Ops);
Mike Stumpba6a0c42009-12-14 21:58:14 +00003893 }
3894
Chris Lattner2da04b32007-08-24 05:35:26 +00003895 return Builder.CreateShl(Ops.LHS, RHS, "shl");
3896}
3897
3898Value *ScalarExprEmitter::EmitShr(const BinOpInfo &Ops) {
3899 // LLVM requires the LHS and RHS to be the same type: promote or truncate the
3900 // RHS to the same size as the LHS.
3901 Value *RHS = Ops.RHS;
3902 if (Ops.LHS->getType() != RHS->getType())
3903 RHS = Builder.CreateIntCast(RHS, Ops.LHS->getType(), false, "sh_prom");
Mike Stump4a3999f2009-09-09 13:00:44 +00003904
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003905 // OpenCL 6.3j: shift values are effectively % word size of LHS.
3906 if (CGF.getLangOpts().OpenCL)
Erich Keane5f0903e2020-04-17 10:44:19 -07003907 RHS = ConstrainShiftValue(Ops.LHS, RHS, "shr.mask");
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003908 else if (CGF.SanOpts.has(SanitizerKind::ShiftExponent) &&
3909 isa<llvm::IntegerType>(Ops.LHS->getType())) {
Alexey Samsonov24cad992014-07-17 18:46:27 +00003910 CodeGenFunction::SanitizerScope SanScope(&CGF);
Alexey Samsonove396bfc2014-11-11 22:03:54 +00003911 llvm::Value *Valid =
3912 Builder.CreateICmpULE(RHS, GetWidthMinusOneValue(Ops.LHS, RHS));
Alexey Samsonov21d2dda2015-03-09 21:50:19 +00003913 EmitBinOpCheck(std::make_pair(Valid, SanitizerKind::ShiftExponent), Ops);
Alexey Samsonov24cad992014-07-17 18:46:27 +00003914 }
David Tweed042e0882013-01-07 16:43:27 +00003915
Douglas Gregor5cc2c8b2010-07-23 15:58:24 +00003916 if (Ops.Ty->hasUnsignedIntegerRepresentation())
Chris Lattner2da04b32007-08-24 05:35:26 +00003917 return Builder.CreateLShr(Ops.LHS, RHS, "shr");
3918 return Builder.CreateAShr(Ops.LHS, RHS, "shr");
3919}
3920
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003921enum IntrinsicType { VCMPEQ, VCMPGT };
3922// return corresponding comparison intrinsic for given vector type
3923static llvm::Intrinsic::ID GetIntrinsic(IntrinsicType IT,
3924 BuiltinType::Kind ElemKind) {
3925 switch (ElemKind) {
David Blaikie83d382b2011-09-23 05:06:16 +00003926 default: llvm_unreachable("unexpected element type");
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003927 case BuiltinType::Char_U:
3928 case BuiltinType::UChar:
3929 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequb_p :
3930 llvm::Intrinsic::ppc_altivec_vcmpgtub_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003931 case BuiltinType::Char_S:
3932 case BuiltinType::SChar:
3933 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequb_p :
3934 llvm::Intrinsic::ppc_altivec_vcmpgtsb_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003935 case BuiltinType::UShort:
3936 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequh_p :
3937 llvm::Intrinsic::ppc_altivec_vcmpgtuh_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003938 case BuiltinType::Short:
3939 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequh_p :
3940 llvm::Intrinsic::ppc_altivec_vcmpgtsh_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003941 case BuiltinType::UInt:
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003942 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequw_p :
3943 llvm::Intrinsic::ppc_altivec_vcmpgtuw_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003944 case BuiltinType::Int:
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003945 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequw_p :
3946 llvm::Intrinsic::ppc_altivec_vcmpgtsw_p;
Guozhi Wei769095b2017-10-19 20:11:23 +00003947 case BuiltinType::ULong:
3948 case BuiltinType::ULongLong:
3949 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequd_p :
3950 llvm::Intrinsic::ppc_altivec_vcmpgtud_p;
3951 case BuiltinType::Long:
3952 case BuiltinType::LongLong:
3953 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpequd_p :
3954 llvm::Intrinsic::ppc_altivec_vcmpgtsd_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003955 case BuiltinType::Float:
3956 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_altivec_vcmpeqfp_p :
3957 llvm::Intrinsic::ppc_altivec_vcmpgtfp_p;
Guozhi Wei769095b2017-10-19 20:11:23 +00003958 case BuiltinType::Double:
3959 return (IT == VCMPEQ) ? llvm::Intrinsic::ppc_vsx_xvcmpeqdp_p :
3960 llvm::Intrinsic::ppc_vsx_xvcmpgtdp_p;
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003961 }
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003962}
3963
Craig Topperc82f8962015-12-16 06:24:28 +00003964Value *ScalarExprEmitter::EmitCompare(const BinaryOperator *E,
3965 llvm::CmpInst::Predicate UICmpOpc,
3966 llvm::CmpInst::Predicate SICmpOpc,
Ulrich Weigand76e9c2a2020-01-10 14:29:24 +01003967 llvm::CmpInst::Predicate FCmpOpc,
3968 bool IsSignaling) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00003969 TestAndClearIgnoreResultAssign();
Chris Lattner42e6b812007-08-26 16:34:22 +00003970 Value *Result;
Chris Lattner2da04b32007-08-24 05:35:26 +00003971 QualType LHSTy = E->getLHS()->getType();
Chandler Carruthb29a7432014-10-11 11:03:30 +00003972 QualType RHSTy = E->getRHS()->getType();
John McCall7a9aac22010-08-23 01:21:21 +00003973 if (const MemberPointerType *MPT = LHSTy->getAs<MemberPointerType>()) {
John McCalle3027922010-08-25 11:45:40 +00003974 assert(E->getOpcode() == BO_EQ ||
3975 E->getOpcode() == BO_NE);
John McCalla1dee5302010-08-22 10:59:02 +00003976 Value *LHS = CGF.EmitScalarExpr(E->getLHS());
3977 Value *RHS = CGF.EmitScalarExpr(E->getRHS());
John McCall7a9aac22010-08-23 01:21:21 +00003978 Result = CGF.CGM.getCXXABI().EmitMemberPointerComparison(
John McCalle3027922010-08-25 11:45:40 +00003979 CGF, LHS, RHS, MPT, E->getOpcode() == BO_NE);
Chandler Carruthb29a7432014-10-11 11:03:30 +00003980 } else if (!LHSTy->isAnyComplexType() && !RHSTy->isAnyComplexType()) {
Leonard Chance1d4f12019-02-21 20:50:09 +00003981 BinOpInfo BOInfo = EmitBinOps(E);
3982 Value *LHS = BOInfo.LHS;
3983 Value *RHS = BOInfo.RHS;
Mike Stump4a3999f2009-09-09 13:00:44 +00003984
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003985 // If AltiVec, the comparison results in a numeric type, so we use
3986 // intrinsics comparing vectors and giving 0 or 1 as a result
Anton Yartsev93900c72011-03-28 21:00:05 +00003987 if (LHSTy->isVectorType() && !E->getType()->isVectorType()) {
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003988 // constants for mapping CR6 register bits to predicate result
3989 enum { CR6_EQ=0, CR6_EQ_REV, CR6_LT, CR6_LT_REV } CR6;
3990
3991 llvm::Intrinsic::ID ID = llvm::Intrinsic::not_intrinsic;
3992
3993 // in several cases vector arguments order will be reversed
3994 Value *FirstVecArg = LHS,
3995 *SecondVecArg = RHS;
3996
Simon Pilgrime0712012019-10-02 15:31:25 +00003997 QualType ElTy = LHSTy->castAs<VectorType>()->getElementType();
Simon Pilgrim16c53ff2020-01-11 15:33:25 +00003998 BuiltinType::Kind ElementKind = ElTy->castAs<BuiltinType>()->getKind();
Anton Yartsev3f8f2882010-11-18 03:19:30 +00003999
4000 switch(E->getOpcode()) {
David Blaikie83d382b2011-09-23 05:06:16 +00004001 default: llvm_unreachable("is not a comparison operation");
Anton Yartsev3f8f2882010-11-18 03:19:30 +00004002 case BO_EQ:
4003 CR6 = CR6_LT;
4004 ID = GetIntrinsic(VCMPEQ, ElementKind);
4005 break;
4006 case BO_NE:
4007 CR6 = CR6_EQ;
4008 ID = GetIntrinsic(VCMPEQ, ElementKind);
4009 break;
4010 case BO_LT:
4011 CR6 = CR6_LT;
4012 ID = GetIntrinsic(VCMPGT, ElementKind);
4013 std::swap(FirstVecArg, SecondVecArg);
4014 break;
4015 case BO_GT:
4016 CR6 = CR6_LT;
4017 ID = GetIntrinsic(VCMPGT, ElementKind);
4018 break;
4019 case BO_LE:
4020 if (ElementKind == BuiltinType::Float) {
4021 CR6 = CR6_LT;
4022 ID = llvm::Intrinsic::ppc_altivec_vcmpgefp_p;
4023 std::swap(FirstVecArg, SecondVecArg);
4024 }
4025 else {
4026 CR6 = CR6_EQ;
4027 ID = GetIntrinsic(VCMPGT, ElementKind);
4028 }
4029 break;
4030 case BO_GE:
4031 if (ElementKind == BuiltinType::Float) {
4032 CR6 = CR6_LT;
4033 ID = llvm::Intrinsic::ppc_altivec_vcmpgefp_p;
4034 }
4035 else {
4036 CR6 = CR6_EQ;
4037 ID = GetIntrinsic(VCMPGT, ElementKind);
4038 std::swap(FirstVecArg, SecondVecArg);
4039 }
4040 break;
4041 }
4042
Chris Lattner2531eb42011-04-19 22:55:03 +00004043 Value *CR6Param = Builder.getInt32(CR6);
Anton Yartsev3f8f2882010-11-18 03:19:30 +00004044 llvm::Function *F = CGF.CGM.getIntrinsic(ID);
David Blaikie43f9bb72015-05-18 22:14:03 +00004045 Result = Builder.CreateCall(F, {CR6Param, FirstVecArg, SecondVecArg});
Guozhi Wei3625f3e2017-10-10 20:31:27 +00004046
4047 // The result type of intrinsic may not be same as E->getType().
4048 // If E->getType() is not BoolTy, EmitScalarConversion will do the
4049 // conversion work. If E->getType() is BoolTy, EmitScalarConversion will
4050 // do nothing, if ResultTy is not i1 at the same time, it will cause
4051 // crash later.
4052 llvm::IntegerType *ResultTy = cast<llvm::IntegerType>(Result->getType());
4053 if (ResultTy->getBitWidth() > 1 &&
4054 E->getType() == CGF.getContext().BoolTy)
4055 Result = Builder.CreateTrunc(Result, Builder.getInt1Ty());
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00004056 return EmitScalarConversion(Result, CGF.getContext().BoolTy, E->getType(),
4057 E->getExprLoc());
Anton Yartsev3f8f2882010-11-18 03:19:30 +00004058 }
4059
Bevin Hansson39baaab2020-01-08 11:12:55 +01004060 if (BOInfo.isFixedPointOp()) {
Leonard Chance1d4f12019-02-21 20:50:09 +00004061 Result = EmitFixedPointBinOp(BOInfo);
4062 } else if (LHS->getType()->isFPOrFPVectorTy()) {
John McCall7fac1ac2020-06-11 18:09:36 -04004063 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(CGF, BOInfo.FPFeatures);
Ulrich Weigand76e9c2a2020-01-10 14:29:24 +01004064 if (!IsSignaling)
4065 Result = Builder.CreateFCmp(FCmpOpc, LHS, RHS, "cmp");
4066 else
4067 Result = Builder.CreateFCmpS(FCmpOpc, LHS, RHS, "cmp");
Douglas Gregor5cc2c8b2010-07-23 15:58:24 +00004068 } else if (LHSTy->hasSignedIntegerRepresentation()) {
Craig Topperc82f8962015-12-16 06:24:28 +00004069 Result = Builder.CreateICmp(SICmpOpc, LHS, RHS, "cmp");
Chris Lattner2da04b32007-08-24 05:35:26 +00004070 } else {
Eli Friedman3c285242008-05-29 15:09:15 +00004071 // Unsigned integers and pointers.
Piotr Padlewski07058292018-07-02 19:21:36 +00004072
4073 if (CGF.CGM.getCodeGenOpts().StrictVTablePointers &&
4074 !isa<llvm::ConstantPointerNull>(LHS) &&
4075 !isa<llvm::ConstantPointerNull>(RHS)) {
4076
4077 // Dynamic information is required to be stripped for comparisons,
4078 // because it could leak the dynamic information. Based on comparisons
4079 // of pointers to dynamic objects, the optimizer can replace one pointer
4080 // with another, which might be incorrect in presence of invariant
4081 // groups. Comparison with null is safe because null does not carry any
4082 // dynamic information.
4083 if (LHSTy.mayBeDynamicClass())
4084 LHS = Builder.CreateStripInvariantGroup(LHS);
4085 if (RHSTy.mayBeDynamicClass())
4086 RHS = Builder.CreateStripInvariantGroup(RHS);
4087 }
4088
Craig Topperc82f8962015-12-16 06:24:28 +00004089 Result = Builder.CreateICmp(UICmpOpc, LHS, RHS, "cmp");
Chris Lattner2da04b32007-08-24 05:35:26 +00004090 }
Chris Lattner2a7deb62009-07-08 01:08:03 +00004091
4092 // If this is a vector comparison, sign extend the result to the appropriate
4093 // vector integer type and return it (don't convert to bool).
4094 if (LHSTy->isVectorType())
4095 return Builder.CreateSExt(Result, ConvertType(E->getType()), "sext");
Mike Stump4a3999f2009-09-09 13:00:44 +00004096
Chris Lattner2da04b32007-08-24 05:35:26 +00004097 } else {
4098 // Complex Comparison: can only be an equality comparison.
Chandler Carruthb29a7432014-10-11 11:03:30 +00004099 CodeGenFunction::ComplexPairTy LHS, RHS;
4100 QualType CETy;
4101 if (auto *CTy = LHSTy->getAs<ComplexType>()) {
4102 LHS = CGF.EmitComplexExpr(E->getLHS());
4103 CETy = CTy->getElementType();
4104 } else {
4105 LHS.first = Visit(E->getLHS());
4106 LHS.second = llvm::Constant::getNullValue(LHS.first->getType());
4107 CETy = LHSTy;
4108 }
4109 if (auto *CTy = RHSTy->getAs<ComplexType>()) {
4110 RHS = CGF.EmitComplexExpr(E->getRHS());
4111 assert(CGF.getContext().hasSameUnqualifiedType(CETy,
4112 CTy->getElementType()) &&
4113 "The element types must always match.");
Chandler Carruth60fdc412014-10-11 11:29:26 +00004114 (void)CTy;
Chandler Carruthb29a7432014-10-11 11:03:30 +00004115 } else {
4116 RHS.first = Visit(E->getRHS());
4117 RHS.second = llvm::Constant::getNullValue(RHS.first->getType());
4118 assert(CGF.getContext().hasSameUnqualifiedType(CETy, RHSTy) &&
4119 "The element types must always match.");
4120 }
Mike Stump4a3999f2009-09-09 13:00:44 +00004121
Chris Lattner42e6b812007-08-26 16:34:22 +00004122 Value *ResultR, *ResultI;
Chris Lattner2da04b32007-08-24 05:35:26 +00004123 if (CETy->isRealFloatingType()) {
Ulrich Weigand76e9c2a2020-01-10 14:29:24 +01004124 // As complex comparisons can only be equality comparisons, they
4125 // are never signaling comparisons.
Craig Topperc82f8962015-12-16 06:24:28 +00004126 ResultR = Builder.CreateFCmp(FCmpOpc, LHS.first, RHS.first, "cmp.r");
4127 ResultI = Builder.CreateFCmp(FCmpOpc, LHS.second, RHS.second, "cmp.i");
Chris Lattner2da04b32007-08-24 05:35:26 +00004128 } else {
4129 // Complex comparisons can only be equality comparisons. As such, signed
4130 // and unsigned opcodes are the same.
Craig Topperc82f8962015-12-16 06:24:28 +00004131 ResultR = Builder.CreateICmp(UICmpOpc, LHS.first, RHS.first, "cmp.r");
4132 ResultI = Builder.CreateICmp(UICmpOpc, LHS.second, RHS.second, "cmp.i");
Chris Lattner2da04b32007-08-24 05:35:26 +00004133 }
Mike Stump4a3999f2009-09-09 13:00:44 +00004134
John McCalle3027922010-08-25 11:45:40 +00004135 if (E->getOpcode() == BO_EQ) {
Chris Lattner2da04b32007-08-24 05:35:26 +00004136 Result = Builder.CreateAnd(ResultR, ResultI, "and.ri");
4137 } else {
John McCalle3027922010-08-25 11:45:40 +00004138 assert(E->getOpcode() == BO_NE &&
Chris Lattner2da04b32007-08-24 05:35:26 +00004139 "Complex comparison other than == or != ?");
4140 Result = Builder.CreateOr(ResultR, ResultI, "or.ri");
4141 }
4142 }
Nuno Lopesa0abe622009-01-11 23:22:37 +00004143
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00004144 return EmitScalarConversion(Result, CGF.getContext().BoolTy, E->getType(),
4145 E->getExprLoc());
Chris Lattner2da04b32007-08-24 05:35:26 +00004146}
4147
4148Value *ScalarExprEmitter::VisitBinAssign(const BinaryOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00004149 bool Ignore = TestAndClearIgnoreResultAssign();
4150
John McCall31168b02011-06-15 23:02:42 +00004151 Value *RHS;
4152 LValue LHS;
Mike Stump4a3999f2009-09-09 13:00:44 +00004153
John McCall31168b02011-06-15 23:02:42 +00004154 switch (E->getLHS()->getType().getObjCLifetime()) {
4155 case Qualifiers::OCL_Strong:
Benjamin Kramer867ea1d2014-03-02 13:01:17 +00004156 std::tie(LHS, RHS) = CGF.EmitARCStoreStrong(E, Ignore);
John McCall31168b02011-06-15 23:02:42 +00004157 break;
4158
4159 case Qualifiers::OCL_Autoreleasing:
Benjamin Kramer867ea1d2014-03-02 13:01:17 +00004160 std::tie(LHS, RHS) = CGF.EmitARCStoreAutoreleasing(E);
John McCall31168b02011-06-15 23:02:42 +00004161 break;
4162
John McCalle399e5b2016-01-27 18:32:30 +00004163 case Qualifiers::OCL_ExplicitNone:
4164 std::tie(LHS, RHS) = CGF.EmitARCStoreUnsafeUnretained(E, Ignore);
4165 break;
4166
John McCall31168b02011-06-15 23:02:42 +00004167 case Qualifiers::OCL_Weak:
4168 RHS = Visit(E->getRHS());
Richard Smith4d1458e2012-09-08 02:08:36 +00004169 LHS = EmitCheckedLValue(E->getLHS(), CodeGenFunction::TCK_Store);
Akira Hatanakaf139ae32019-12-03 15:17:01 -08004170 RHS = CGF.EmitARCStoreWeak(LHS.getAddress(CGF), RHS, Ignore);
John McCall31168b02011-06-15 23:02:42 +00004171 break;
4172
John McCall31168b02011-06-15 23:02:42 +00004173 case Qualifiers::OCL_None:
John McCall31168b02011-06-15 23:02:42 +00004174 // __block variables need to have the rhs evaluated first, plus
4175 // this should improve codegen just a little.
4176 RHS = Visit(E->getRHS());
Richard Smith4d1458e2012-09-08 02:08:36 +00004177 LHS = EmitCheckedLValue(E->getLHS(), CodeGenFunction::TCK_Store);
John McCall31168b02011-06-15 23:02:42 +00004178
4179 // Store the value into the LHS. Bit-fields are handled specially
4180 // because the result is altered by the store, i.e., [C99 6.5.16p1]
4181 // 'An assignment expression has the value of the left operand after
4182 // the assignment...'.
Vedant Kumar42c17ec2017-03-14 01:56:34 +00004183 if (LHS.isBitField()) {
John McCall55e1fbc2011-06-25 02:11:03 +00004184 CGF.EmitStoreThroughBitfieldLValue(RValue::get(RHS), LHS, &RHS);
Vedant Kumar42c17ec2017-03-14 01:56:34 +00004185 } else {
4186 CGF.EmitNullabilityCheck(LHS, RHS, E->getExprLoc());
John McCall55e1fbc2011-06-25 02:11:03 +00004187 CGF.EmitStoreThroughLValue(RValue::get(RHS), LHS);
Vedant Kumar42c17ec2017-03-14 01:56:34 +00004188 }
John McCall31168b02011-06-15 23:02:42 +00004189 }
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00004190
4191 // If the result is clearly ignored, return now.
Mike Stumpdf0fe272009-05-29 15:46:01 +00004192 if (Ignore)
Craig Topper8a13c412014-05-21 05:09:00 +00004193 return nullptr;
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00004194
John McCall07bb1962010-11-16 10:08:07 +00004195 // The result of an assignment in C is the assigned r-value.
Richard Smith9c6890a2012-11-01 22:30:59 +00004196 if (!CGF.getLangOpts().CPlusPlus)
John McCall07bb1962010-11-16 10:08:07 +00004197 return RHS;
4198
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00004199 // If the lvalue is non-volatile, return the computed value of the assignment.
4200 if (!LHS.isVolatileQualified())
4201 return RHS;
4202
4203 // Otherwise, reload the value.
Nick Lewycky2d84e842013-10-02 02:29:49 +00004204 return EmitLoadOfLValue(LHS, E->getExprLoc());
Chris Lattner2da04b32007-08-24 05:35:26 +00004205}
4206
4207Value *ScalarExprEmitter::VisitBinLAnd(const BinaryOperator *E) {
Tanya Lattner20248222012-01-16 21:02:28 +00004208 // Perform vector logical and on comparisons with zero vectors.
4209 if (E->getType()->isVectorType()) {
Justin Bogner66242d62015-04-23 23:06:47 +00004210 CGF.incrementProfileCounter(E);
Justin Bogneref512b92014-01-06 22:27:43 +00004211
Tanya Lattner20248222012-01-16 21:02:28 +00004212 Value *LHS = Visit(E->getLHS());
4213 Value *RHS = Visit(E->getRHS());
4214 Value *Zero = llvm::ConstantAggregateZero::get(LHS->getType());
Joey Gouly7d00f002013-02-21 11:49:56 +00004215 if (LHS->getType()->isFPOrFPVectorTy()) {
John McCall7fac1ac2020-06-11 18:09:36 -04004216 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(
Melanie Blowerdefd43a2020-06-26 08:45:12 -07004217 CGF, E->getFPFeatures(CGF.getLangOpts()));
Joey Gouly7d00f002013-02-21 11:49:56 +00004218 LHS = Builder.CreateFCmp(llvm::CmpInst::FCMP_UNE, LHS, Zero, "cmp");
4219 RHS = Builder.CreateFCmp(llvm::CmpInst::FCMP_UNE, RHS, Zero, "cmp");
4220 } else {
4221 LHS = Builder.CreateICmp(llvm::CmpInst::ICMP_NE, LHS, Zero, "cmp");
4222 RHS = Builder.CreateICmp(llvm::CmpInst::ICMP_NE, RHS, Zero, "cmp");
4223 }
Tanya Lattner20248222012-01-16 21:02:28 +00004224 Value *And = Builder.CreateAnd(LHS, RHS);
Joey Gouly7d00f002013-02-21 11:49:56 +00004225 return Builder.CreateSExt(And, ConvertType(E->getType()), "sext");
Tanya Lattner20248222012-01-16 21:02:28 +00004226 }
Craig Toppera97d7e72013-07-26 06:16:11 +00004227
Chris Lattner2192fe52011-07-18 04:24:23 +00004228 llvm::Type *ResTy = ConvertType(E->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00004229
Chris Lattner8b084582008-11-12 08:26:50 +00004230 // If we have 0 && RHS, see if we can elide RHS, if so, just return 0.
4231 // If we have 1 && X, just emit X without inserting the control flow.
Chris Lattner41c6ab52011-02-27 23:02:32 +00004232 bool LHSCondVal;
4233 if (CGF.ConstantFoldsToSimpleInteger(E->getLHS(), LHSCondVal)) {
4234 if (LHSCondVal) { // If we have 1 && X, just emit X.
Justin Bogner66242d62015-04-23 23:06:47 +00004235 CGF.incrementProfileCounter(E);
Justin Bogneref512b92014-01-06 22:27:43 +00004236
Chris Lattner5b1964b2008-11-11 07:41:27 +00004237 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
Chris Lattner671fec82009-10-17 04:24:20 +00004238 // ZExt result to int or bool.
4239 return Builder.CreateZExtOrBitCast(RHSCond, ResTy, "land.ext");
Chris Lattner5b1964b2008-11-11 07:41:27 +00004240 }
Mike Stump4a3999f2009-09-09 13:00:44 +00004241
Chris Lattner671fec82009-10-17 04:24:20 +00004242 // 0 && RHS: If it is safe, just elide the RHS, and return 0/false.
Chris Lattner8b084582008-11-12 08:26:50 +00004243 if (!CGF.ContainsLabel(E->getRHS()))
Chris Lattner671fec82009-10-17 04:24:20 +00004244 return llvm::Constant::getNullValue(ResTy);
Chris Lattner5b1964b2008-11-11 07:41:27 +00004245 }
Mike Stump4a3999f2009-09-09 13:00:44 +00004246
Daniel Dunbara612e792008-11-13 01:38:36 +00004247 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("land.end");
4248 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("land.rhs");
Chris Lattner8b084582008-11-12 08:26:50 +00004249
John McCallce1de612011-01-26 04:00:11 +00004250 CodeGenFunction::ConditionalEvaluation eval(CGF);
4251
Chris Lattner35710d182008-11-12 08:38:24 +00004252 // Branch on the LHS first. If it is false, go to the failure (cont) block.
Justin Bogner66242d62015-04-23 23:06:47 +00004253 CGF.EmitBranchOnBoolExpr(E->getLHS(), RHSBlock, ContBlock,
4254 CGF.getProfileCount(E->getRHS()));
Chris Lattner35710d182008-11-12 08:38:24 +00004255
4256 // Any edges into the ContBlock are now from an (indeterminate number of)
4257 // edges from this first condition. All of these values will be false. Start
4258 // setting up the PHI node in the Cont Block for this.
Jay Foad20c0f022011-03-30 11:28:58 +00004259 llvm::PHINode *PN = llvm::PHINode::Create(llvm::Type::getInt1Ty(VMContext), 2,
Owen Anderson41a75022009-08-13 21:57:51 +00004260 "", ContBlock);
Chris Lattner35710d182008-11-12 08:38:24 +00004261 for (llvm::pred_iterator PI = pred_begin(ContBlock), PE = pred_end(ContBlock);
4262 PI != PE; ++PI)
Owen Andersonfe4e3472009-07-31 17:39:36 +00004263 PN->addIncoming(llvm::ConstantInt::getFalse(VMContext), *PI);
Mike Stump4a3999f2009-09-09 13:00:44 +00004264
John McCallce1de612011-01-26 04:00:11 +00004265 eval.begin(CGF);
Chris Lattner2da04b32007-08-24 05:35:26 +00004266 CGF.EmitBlock(RHSBlock);
Justin Bogner66242d62015-04-23 23:06:47 +00004267 CGF.incrementProfileCounter(E);
Chris Lattner2da04b32007-08-24 05:35:26 +00004268 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
John McCallce1de612011-01-26 04:00:11 +00004269 eval.end(CGF);
Mike Stump4a3999f2009-09-09 13:00:44 +00004270
Chris Lattner2da04b32007-08-24 05:35:26 +00004271 // Reaquire the RHS block, as there may be subblocks inserted.
4272 RHSBlock = Builder.GetInsertBlock();
Chris Lattner35710d182008-11-12 08:38:24 +00004273
David Blaikie1b5adb82014-07-10 20:42:59 +00004274 // Emit an unconditional branch from this block to ContBlock.
4275 {
Devang Patel4d761272011-03-30 00:08:31 +00004276 // There is no need to emit line number for unconditional branch.
Adrian Prantl95b24e92015-02-03 20:00:54 +00004277 auto NL = ApplyDebugLocation::CreateEmpty(CGF);
David Blaikie1b5adb82014-07-10 20:42:59 +00004278 CGF.EmitBlock(ContBlock);
4279 }
4280 // Insert an entry into the phi node for the edge with the value of RHSCond.
Chris Lattner2da04b32007-08-24 05:35:26 +00004281 PN->addIncoming(RHSCond, RHSBlock);
Mike Stump4a3999f2009-09-09 13:00:44 +00004282
Anastasis Grammenosdfe8fe52018-06-21 16:53:48 +00004283 // Artificial location to preserve the scope information
4284 {
4285 auto NL = ApplyDebugLocation::CreateArtificial(CGF);
4286 PN->setDebugLoc(Builder.getCurrentDebugLocation());
4287 }
4288
Chris Lattner2da04b32007-08-24 05:35:26 +00004289 // ZExt result to int.
Chris Lattner671fec82009-10-17 04:24:20 +00004290 return Builder.CreateZExtOrBitCast(PN, ResTy, "land.ext");
Chris Lattner2da04b32007-08-24 05:35:26 +00004291}
4292
4293Value *ScalarExprEmitter::VisitBinLOr(const BinaryOperator *E) {
Tanya Lattner20248222012-01-16 21:02:28 +00004294 // Perform vector logical or on comparisons with zero vectors.
4295 if (E->getType()->isVectorType()) {
Justin Bogner66242d62015-04-23 23:06:47 +00004296 CGF.incrementProfileCounter(E);
Justin Bogneref512b92014-01-06 22:27:43 +00004297
Tanya Lattner20248222012-01-16 21:02:28 +00004298 Value *LHS = Visit(E->getLHS());
4299 Value *RHS = Visit(E->getRHS());
4300 Value *Zero = llvm::ConstantAggregateZero::get(LHS->getType());
Joey Gouly7d00f002013-02-21 11:49:56 +00004301 if (LHS->getType()->isFPOrFPVectorTy()) {
John McCall7fac1ac2020-06-11 18:09:36 -04004302 CodeGenFunction::CGFPOptionsRAII FPOptsRAII(
Melanie Blowerdefd43a2020-06-26 08:45:12 -07004303 CGF, E->getFPFeatures(CGF.getLangOpts()));
Joey Gouly7d00f002013-02-21 11:49:56 +00004304 LHS = Builder.CreateFCmp(llvm::CmpInst::FCMP_UNE, LHS, Zero, "cmp");
4305 RHS = Builder.CreateFCmp(llvm::CmpInst::FCMP_UNE, RHS, Zero, "cmp");
4306 } else {
4307 LHS = Builder.CreateICmp(llvm::CmpInst::ICMP_NE, LHS, Zero, "cmp");
4308 RHS = Builder.CreateICmp(llvm::CmpInst::ICMP_NE, RHS, Zero, "cmp");
4309 }
Tanya Lattner20248222012-01-16 21:02:28 +00004310 Value *Or = Builder.CreateOr(LHS, RHS);
Joey Gouly7d00f002013-02-21 11:49:56 +00004311 return Builder.CreateSExt(Or, ConvertType(E->getType()), "sext");
Tanya Lattner20248222012-01-16 21:02:28 +00004312 }
Craig Toppera97d7e72013-07-26 06:16:11 +00004313
Chris Lattner2192fe52011-07-18 04:24:23 +00004314 llvm::Type *ResTy = ConvertType(E->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00004315
Chris Lattner8b084582008-11-12 08:26:50 +00004316 // If we have 1 || RHS, see if we can elide RHS, if so, just return 1.
4317 // If we have 0 || X, just emit X without inserting the control flow.
Chris Lattner41c6ab52011-02-27 23:02:32 +00004318 bool LHSCondVal;
4319 if (CGF.ConstantFoldsToSimpleInteger(E->getLHS(), LHSCondVal)) {
4320 if (!LHSCondVal) { // If we have 0 || X, just emit X.
Justin Bogner66242d62015-04-23 23:06:47 +00004321 CGF.incrementProfileCounter(E);
Justin Bogneref512b92014-01-06 22:27:43 +00004322
Chris Lattner5b1964b2008-11-11 07:41:27 +00004323 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
Chris Lattner671fec82009-10-17 04:24:20 +00004324 // ZExt result to int or bool.
4325 return Builder.CreateZExtOrBitCast(RHSCond, ResTy, "lor.ext");
Chris Lattner5b1964b2008-11-11 07:41:27 +00004326 }
Mike Stump4a3999f2009-09-09 13:00:44 +00004327
Chris Lattner671fec82009-10-17 04:24:20 +00004328 // 1 || RHS: If it is safe, just elide the RHS, and return 1/true.
Chris Lattner8b084582008-11-12 08:26:50 +00004329 if (!CGF.ContainsLabel(E->getRHS()))
Chris Lattner671fec82009-10-17 04:24:20 +00004330 return llvm::ConstantInt::get(ResTy, 1);
Chris Lattner5b1964b2008-11-11 07:41:27 +00004331 }
Mike Stump4a3999f2009-09-09 13:00:44 +00004332
Daniel Dunbara612e792008-11-13 01:38:36 +00004333 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("lor.end");
4334 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("lor.rhs");
Mike Stump4a3999f2009-09-09 13:00:44 +00004335
John McCallce1de612011-01-26 04:00:11 +00004336 CodeGenFunction::ConditionalEvaluation eval(CGF);
4337
Chris Lattner35710d182008-11-12 08:38:24 +00004338 // Branch on the LHS first. If it is true, go to the success (cont) block.
Justin Bogneref512b92014-01-06 22:27:43 +00004339 CGF.EmitBranchOnBoolExpr(E->getLHS(), ContBlock, RHSBlock,
Justin Bogner66242d62015-04-23 23:06:47 +00004340 CGF.getCurrentProfileCount() -
4341 CGF.getProfileCount(E->getRHS()));
Chris Lattner35710d182008-11-12 08:38:24 +00004342
4343 // Any edges into the ContBlock are now from an (indeterminate number of)
4344 // edges from this first condition. All of these values will be true. Start
4345 // setting up the PHI node in the Cont Block for this.
Jay Foad20c0f022011-03-30 11:28:58 +00004346 llvm::PHINode *PN = llvm::PHINode::Create(llvm::Type::getInt1Ty(VMContext), 2,
Owen Anderson41a75022009-08-13 21:57:51 +00004347 "", ContBlock);
Chris Lattner35710d182008-11-12 08:38:24 +00004348 for (llvm::pred_iterator PI = pred_begin(ContBlock), PE = pred_end(ContBlock);
4349 PI != PE; ++PI)
Owen Andersonfe4e3472009-07-31 17:39:36 +00004350 PN->addIncoming(llvm::ConstantInt::getTrue(VMContext), *PI);
Chris Lattner35710d182008-11-12 08:38:24 +00004351
John McCallce1de612011-01-26 04:00:11 +00004352 eval.begin(CGF);
Anders Carlssonf47a3de2009-06-04 02:53:13 +00004353
Chris Lattner35710d182008-11-12 08:38:24 +00004354 // Emit the RHS condition as a bool value.
Chris Lattner2da04b32007-08-24 05:35:26 +00004355 CGF.EmitBlock(RHSBlock);
Justin Bogner66242d62015-04-23 23:06:47 +00004356 CGF.incrementProfileCounter(E);
Chris Lattner2da04b32007-08-24 05:35:26 +00004357 Value *RHSCond = CGF.EvaluateExprAsBool(E->getRHS());
Mike Stump4a3999f2009-09-09 13:00:44 +00004358
John McCallce1de612011-01-26 04:00:11 +00004359 eval.end(CGF);
Mike Stump4a3999f2009-09-09 13:00:44 +00004360
Chris Lattner2da04b32007-08-24 05:35:26 +00004361 // Reaquire the RHS block, as there may be subblocks inserted.
4362 RHSBlock = Builder.GetInsertBlock();
Mike Stump4a3999f2009-09-09 13:00:44 +00004363
Chris Lattner35710d182008-11-12 08:38:24 +00004364 // Emit an unconditional branch from this block to ContBlock. Insert an entry
4365 // into the phi node for the edge with the value of RHSCond.
4366 CGF.EmitBlock(ContBlock);
Chris Lattner2da04b32007-08-24 05:35:26 +00004367 PN->addIncoming(RHSCond, RHSBlock);
Mike Stump4a3999f2009-09-09 13:00:44 +00004368
Chris Lattner2da04b32007-08-24 05:35:26 +00004369 // ZExt result to int.
Chris Lattner671fec82009-10-17 04:24:20 +00004370 return Builder.CreateZExtOrBitCast(PN, ResTy, "lor.ext");
Chris Lattner2da04b32007-08-24 05:35:26 +00004371}
4372
4373Value *ScalarExprEmitter::VisitBinComma(const BinaryOperator *E) {
John McCalla2342eb2010-12-05 02:00:02 +00004374 CGF.EmitIgnoredExpr(E->getLHS());
Daniel Dunbar5c7e3932008-11-11 23:11:34 +00004375 CGF.EnsureInsertPoint();
Chris Lattner2da04b32007-08-24 05:35:26 +00004376 return Visit(E->getRHS());
4377}
4378
4379//===----------------------------------------------------------------------===//
4380// Other Operators
4381//===----------------------------------------------------------------------===//
4382
Chris Lattner3fd91f832008-11-12 08:55:54 +00004383/// isCheapEnoughToEvaluateUnconditionally - Return true if the specified
4384/// expression is cheap enough and side-effect-free enough to evaluate
4385/// unconditionally instead of conditionally. This is used to convert control
4386/// flow into selects in some cases.
Mike Stump53f9ded2009-11-03 23:25:48 +00004387static bool isCheapEnoughToEvaluateUnconditionally(const Expr *E,
4388 CodeGenFunction &CGF) {
Chris Lattner56784f92011-04-16 23:15:35 +00004389 // Anything that is an integer or floating point constant is fine.
Nick Lewycky22e55a02013-11-08 23:00:12 +00004390 return E->IgnoreParens()->isEvaluatable(CGF.getContext());
Mike Stump4a3999f2009-09-09 13:00:44 +00004391
Nick Lewycky22e55a02013-11-08 23:00:12 +00004392 // Even non-volatile automatic variables can't be evaluated unconditionally.
4393 // Referencing a thread_local may cause non-trivial initialization work to
4394 // occur. If we're inside a lambda and one of the variables is from the scope
4395 // outside the lambda, that function may have returned already. Reading its
4396 // locals is a bad idea. Also, these reads may introduce races there didn't
4397 // exist in the source-level program.
Chris Lattner3fd91f832008-11-12 08:55:54 +00004398}
4399
4400
Chris Lattner2da04b32007-08-24 05:35:26 +00004401Value *ScalarExprEmitter::
John McCallc07a0c72011-02-17 10:25:35 +00004402VisitAbstractConditionalOperator(const AbstractConditionalOperator *E) {
Mike Stumpdf0fe272009-05-29 15:46:01 +00004403 TestAndClearIgnoreResultAssign();
John McCallc07a0c72011-02-17 10:25:35 +00004404
4405 // Bind the common expression if necessary.
Eli Friedman48fd89a2012-01-06 20:42:20 +00004406 CodeGenFunction::OpaqueValueMapping binding(CGF, E);
John McCallc07a0c72011-02-17 10:25:35 +00004407
4408 Expr *condExpr = E->getCond();
4409 Expr *lhsExpr = E->getTrueExpr();
4410 Expr *rhsExpr = E->getFalseExpr();
4411
Chris Lattnercd439292008-11-12 08:04:58 +00004412 // If the condition constant folds and can be elided, try to avoid emitting
4413 // the condition and the dead arm.
Chris Lattner41c6ab52011-02-27 23:02:32 +00004414 bool CondExprBool;
4415 if (CGF.ConstantFoldsToSimpleInteger(condExpr, CondExprBool)) {
John McCallc07a0c72011-02-17 10:25:35 +00004416 Expr *live = lhsExpr, *dead = rhsExpr;
Chris Lattner41c6ab52011-02-27 23:02:32 +00004417 if (!CondExprBool) std::swap(live, dead);
Mike Stump4a3999f2009-09-09 13:00:44 +00004418
Eli Friedman27ef75b2011-10-15 02:10:40 +00004419 // If the dead side doesn't have labels we need, just emit the Live part.
4420 if (!CGF.ContainsLabel(dead)) {
Justin Bogneref512b92014-01-06 22:27:43 +00004421 if (CondExprBool)
Justin Bogner66242d62015-04-23 23:06:47 +00004422 CGF.incrementProfileCounter(E);
Eli Friedman27ef75b2011-10-15 02:10:40 +00004423 Value *Result = Visit(live);
4424
4425 // If the live part is a throw expression, it acts like it has a void
4426 // type, so evaluating it returns a null Value*. However, a conditional
4427 // with non-void type must return a non-null Value*.
4428 if (!Result && !E->getType()->isVoidType())
4429 Result = llvm::UndefValue::get(CGF.ConvertType(E->getType()));
4430
4431 return Result;
4432 }
Chris Lattnerd53e2332008-11-11 18:56:45 +00004433 }
Mike Stump4a3999f2009-09-09 13:00:44 +00004434
Nate Begemanabb5a732010-09-20 22:41:17 +00004435 // OpenCL: If the condition is a vector, we can treat this condition like
4436 // the select function.
Min-Yih Hsu4431d642020-05-26 06:27:34 +00004437 if ((CGF.getLangOpts().OpenCL && condExpr->getType()->isVectorType()) ||
4438 condExpr->getType()->isExtVectorType()) {
Justin Bogner66242d62015-04-23 23:06:47 +00004439 CGF.incrementProfileCounter(E);
Justin Bogneref512b92014-01-06 22:27:43 +00004440
John McCallc07a0c72011-02-17 10:25:35 +00004441 llvm::Value *CondV = CGF.EmitScalarExpr(condExpr);
4442 llvm::Value *LHS = Visit(lhsExpr);
4443 llvm::Value *RHS = Visit(rhsExpr);
Craig Toppera97d7e72013-07-26 06:16:11 +00004444
Chris Lattner2192fe52011-07-18 04:24:23 +00004445 llvm::Type *condType = ConvertType(condExpr->getType());
4446 llvm::VectorType *vecTy = cast<llvm::VectorType>(condType);
Craig Toppera97d7e72013-07-26 06:16:11 +00004447
4448 unsigned numElem = vecTy->getNumElements();
Chris Lattner2192fe52011-07-18 04:24:23 +00004449 llvm::Type *elemType = vecTy->getElementType();
Craig Toppera97d7e72013-07-26 06:16:11 +00004450
Chris Lattner2d6b7b92012-01-25 05:34:41 +00004451 llvm::Value *zeroVec = llvm::Constant::getNullValue(vecTy);
Nate Begemanabb5a732010-09-20 22:41:17 +00004452 llvm::Value *TestMSB = Builder.CreateICmpSLT(CondV, zeroVec);
Christopher Tetreault79689812020-06-01 09:55:24 -07004453 llvm::Value *tmp = Builder.CreateSExt(
4454 TestMSB, llvm::FixedVectorType::get(elemType, numElem), "sext");
Nate Begemanabb5a732010-09-20 22:41:17 +00004455 llvm::Value *tmp2 = Builder.CreateNot(tmp);
Craig Toppera97d7e72013-07-26 06:16:11 +00004456
Nate Begemanabb5a732010-09-20 22:41:17 +00004457 // Cast float to int to perform ANDs if necessary.
4458 llvm::Value *RHSTmp = RHS;
4459 llvm::Value *LHSTmp = LHS;
4460 bool wasCast = false;
Chris Lattner2192fe52011-07-18 04:24:23 +00004461 llvm::VectorType *rhsVTy = cast<llvm::VectorType>(RHS->getType());
Peter Collingbourneaac265c2012-05-29 00:35:18 +00004462 if (rhsVTy->getElementType()->isFloatingPointTy()) {
Nate Begemanabb5a732010-09-20 22:41:17 +00004463 RHSTmp = Builder.CreateBitCast(RHS, tmp2->getType());
4464 LHSTmp = Builder.CreateBitCast(LHS, tmp->getType());
4465 wasCast = true;
4466 }
Craig Toppera97d7e72013-07-26 06:16:11 +00004467
Nate Begemanabb5a732010-09-20 22:41:17 +00004468 llvm::Value *tmp3 = Builder.CreateAnd(RHSTmp, tmp2);
4469 llvm::Value *tmp4 = Builder.CreateAnd(LHSTmp, tmp);
4470 llvm::Value *tmp5 = Builder.CreateOr(tmp3, tmp4, "cond");
4471 if (wasCast)
4472 tmp5 = Builder.CreateBitCast(tmp5, RHS->getType());
4473
4474 return tmp5;
4475 }
Craig Toppera97d7e72013-07-26 06:16:11 +00004476
Erich Keane349636d2019-12-05 06:17:39 -08004477 if (condExpr->getType()->isVectorType()) {
4478 CGF.incrementProfileCounter(E);
4479
4480 llvm::Value *CondV = CGF.EmitScalarExpr(condExpr);
4481 llvm::Value *LHS = Visit(lhsExpr);
4482 llvm::Value *RHS = Visit(rhsExpr);
4483
4484 llvm::Type *CondType = ConvertType(condExpr->getType());
4485 auto *VecTy = cast<llvm::VectorType>(CondType);
4486 llvm::Value *ZeroVec = llvm::Constant::getNullValue(VecTy);
4487
4488 CondV = Builder.CreateICmpNE(CondV, ZeroVec, "vector_cond");
4489 return Builder.CreateSelect(CondV, LHS, RHS, "vector_select");
4490 }
4491
Chris Lattner3fd91f832008-11-12 08:55:54 +00004492 // If this is a really simple expression (like x ? 4 : 5), emit this as a
4493 // select instead of as control flow. We can only do this if it is cheap and
Chris Lattner9ce8a532008-11-16 06:16:27 +00004494 // safe to evaluate the LHS and RHS unconditionally.
John McCallc07a0c72011-02-17 10:25:35 +00004495 if (isCheapEnoughToEvaluateUnconditionally(lhsExpr, CGF) &&
4496 isCheapEnoughToEvaluateUnconditionally(rhsExpr, CGF)) {
4497 llvm::Value *CondV = CGF.EvaluateExprAsBool(condExpr);
Vedant Kumar502bbfa2017-02-25 06:35:45 +00004498 llvm::Value *StepV = Builder.CreateZExtOrBitCast(CondV, CGF.Int64Ty);
4499
4500 CGF.incrementProfileCounter(E, StepV);
4501
John McCallc07a0c72011-02-17 10:25:35 +00004502 llvm::Value *LHS = Visit(lhsExpr);
4503 llvm::Value *RHS = Visit(rhsExpr);
Eli Friedman516c2ad2011-12-08 22:01:56 +00004504 if (!LHS) {
4505 // If the conditional has void type, make sure we return a null Value*.
4506 assert(!RHS && "LHS and RHS types must match");
Craig Topper8a13c412014-05-21 05:09:00 +00004507 return nullptr;
Eli Friedman516c2ad2011-12-08 22:01:56 +00004508 }
Chris Lattner3fd91f832008-11-12 08:55:54 +00004509 return Builder.CreateSelect(CondV, LHS, RHS, "cond");
4510 }
Mike Stump4a3999f2009-09-09 13:00:44 +00004511
Daniel Dunbard2a53a72008-11-12 10:13:37 +00004512 llvm::BasicBlock *LHSBlock = CGF.createBasicBlock("cond.true");
4513 llvm::BasicBlock *RHSBlock = CGF.createBasicBlock("cond.false");
Daniel Dunbara612e792008-11-13 01:38:36 +00004514 llvm::BasicBlock *ContBlock = CGF.createBasicBlock("cond.end");
John McCallce1de612011-01-26 04:00:11 +00004515
4516 CodeGenFunction::ConditionalEvaluation eval(CGF);
Justin Bogner66242d62015-04-23 23:06:47 +00004517 CGF.EmitBranchOnBoolExpr(condExpr, LHSBlock, RHSBlock,
4518 CGF.getProfileCount(lhsExpr));
Anders Carlsson43c52cd2009-06-04 03:00:32 +00004519
Chris Lattner2da04b32007-08-24 05:35:26 +00004520 CGF.EmitBlock(LHSBlock);
Justin Bogner66242d62015-04-23 23:06:47 +00004521 CGF.incrementProfileCounter(E);
John McCallce1de612011-01-26 04:00:11 +00004522 eval.begin(CGF);
John McCallc07a0c72011-02-17 10:25:35 +00004523 Value *LHS = Visit(lhsExpr);
John McCallce1de612011-01-26 04:00:11 +00004524 eval.end(CGF);
Mike Stump4a3999f2009-09-09 13:00:44 +00004525
Chris Lattner2da04b32007-08-24 05:35:26 +00004526 LHSBlock = Builder.GetInsertBlock();
John McCallce1de612011-01-26 04:00:11 +00004527 Builder.CreateBr(ContBlock);
Mike Stump4a3999f2009-09-09 13:00:44 +00004528
Chris Lattner2da04b32007-08-24 05:35:26 +00004529 CGF.EmitBlock(RHSBlock);
John McCallce1de612011-01-26 04:00:11 +00004530 eval.begin(CGF);
John McCallc07a0c72011-02-17 10:25:35 +00004531 Value *RHS = Visit(rhsExpr);
John McCallce1de612011-01-26 04:00:11 +00004532 eval.end(CGF);
Mike Stump4a3999f2009-09-09 13:00:44 +00004533
John McCallce1de612011-01-26 04:00:11 +00004534 RHSBlock = Builder.GetInsertBlock();
Chris Lattner2da04b32007-08-24 05:35:26 +00004535 CGF.EmitBlock(ContBlock);
Mike Stump4a3999f2009-09-09 13:00:44 +00004536
Eli Friedmanf6c175b2009-12-07 20:25:53 +00004537 // If the LHS or RHS is a throw expression, it will be legitimately null.
4538 if (!LHS)
4539 return RHS;
4540 if (!RHS)
4541 return LHS;
Mike Stump4a3999f2009-09-09 13:00:44 +00004542
Chris Lattner2da04b32007-08-24 05:35:26 +00004543 // Create a PHI node for the real part.
Jay Foad20c0f022011-03-30 11:28:58 +00004544 llvm::PHINode *PN = Builder.CreatePHI(LHS->getType(), 2, "cond");
Chris Lattner2da04b32007-08-24 05:35:26 +00004545 PN->addIncoming(LHS, LHSBlock);
4546 PN->addIncoming(RHS, RHSBlock);
4547 return PN;
4548}
4549
4550Value *ScalarExprEmitter::VisitChooseExpr(ChooseExpr *E) {
Eli Friedman75807f22013-07-20 00:40:58 +00004551 return Visit(E->getChosenSubExpr());
Chris Lattner2da04b32007-08-24 05:35:26 +00004552}
4553
Chris Lattnerb6a7b582007-11-30 17:56:23 +00004554Value *ScalarExprEmitter::VisitVAArgExpr(VAArgExpr *VE) {
Richard Smitha1a808c2014-04-14 23:47:48 +00004555 QualType Ty = VE->getType();
Daniel Sanders59229dc2014-11-19 10:01:35 +00004556
Richard Smitha1a808c2014-04-14 23:47:48 +00004557 if (Ty->isVariablyModifiedType())
4558 CGF.EmitVariablyModifiedType(Ty);
4559
Charles Davisc7d5c942015-09-17 20:55:33 +00004560 Address ArgValue = Address::invalid();
4561 Address ArgPtr = CGF.EmitVAArg(VE, ArgValue);
4562
Daniel Sanders59229dc2014-11-19 10:01:35 +00004563 llvm::Type *ArgTy = ConvertType(VE->getType());
Anders Carlsson13abd7e2008-11-04 05:30:00 +00004564
James Y Knight29b5f082016-02-24 02:59:33 +00004565 // If EmitVAArg fails, emit an error.
4566 if (!ArgPtr.isValid()) {
4567 CGF.ErrorUnsupported(VE, "va_arg expression");
4568 return llvm::UndefValue::get(ArgTy);
4569 }
Anders Carlsson13abd7e2008-11-04 05:30:00 +00004570
Mike Stumpdf0fe272009-05-29 15:46:01 +00004571 // FIXME Volatility.
Daniel Sanders59229dc2014-11-19 10:01:35 +00004572 llvm::Value *Val = Builder.CreateLoad(ArgPtr);
4573
4574 // If EmitVAArg promoted the type, we must truncate it.
Daniel Sanderscdcb5802015-01-13 10:47:00 +00004575 if (ArgTy != Val->getType()) {
4576 if (ArgTy->isPointerTy() && !Val->getType()->isPointerTy())
4577 Val = Builder.CreateIntToPtr(Val, ArgTy);
4578 else
4579 Val = Builder.CreateTrunc(Val, ArgTy);
4580 }
Daniel Sanders59229dc2014-11-19 10:01:35 +00004581
4582 return Val;
Anders Carlsson7e13ab82007-10-15 20:28:48 +00004583}
4584
John McCall351762c2011-02-07 10:33:21 +00004585Value *ScalarExprEmitter::VisitBlockExpr(const BlockExpr *block) {
4586 return CGF.EmitBlockLiteral(block);
Mike Stumpab3afd82009-02-12 18:29:15 +00004587}
4588
Yaxun Liuc5647012016-06-08 15:11:21 +00004589// Convert a vec3 to vec4, or vice versa.
4590static Value *ConvertVec3AndVec4(CGBuilderTy &Builder, CodeGenFunction &CGF,
4591 Value *Src, unsigned NumElementsDst) {
4592 llvm::Value *UnV = llvm::UndefValue::get(Src->getType());
Benjamin Kramer6f64dac2020-04-15 12:41:54 +02004593 static constexpr int Mask[] = {0, 1, 2, -1};
4594 return Builder.CreateShuffleVector(Src, UnV,
4595 llvm::makeArrayRef(Mask, NumElementsDst));
Yaxun Liuc5647012016-06-08 15:11:21 +00004596}
4597
Yaxun Liuea6b7962016-10-03 14:41:50 +00004598// Create cast instructions for converting LLVM value \p Src to LLVM type \p
4599// DstTy. \p Src has the same size as \p DstTy. Both are single value types
4600// but could be scalar or vectors of different lengths, and either can be
4601// pointer.
4602// There are 4 cases:
4603// 1. non-pointer -> non-pointer : needs 1 bitcast
4604// 2. pointer -> pointer : needs 1 bitcast or addrspacecast
4605// 3. pointer -> non-pointer
4606// a) pointer -> intptr_t : needs 1 ptrtoint
4607// b) pointer -> non-intptr_t : needs 1 ptrtoint then 1 bitcast
4608// 4. non-pointer -> pointer
4609// a) intptr_t -> pointer : needs 1 inttoptr
4610// b) non-intptr_t -> pointer : needs 1 bitcast then 1 inttoptr
4611// Note: for cases 3b and 4b two casts are required since LLVM casts do not
4612// allow casting directly between pointer types and non-integer non-pointer
4613// types.
4614static Value *createCastsForTypeOfSameSize(CGBuilderTy &Builder,
4615 const llvm::DataLayout &DL,
4616 Value *Src, llvm::Type *DstTy,
4617 StringRef Name = "") {
4618 auto SrcTy = Src->getType();
4619
4620 // Case 1.
4621 if (!SrcTy->isPointerTy() && !DstTy->isPointerTy())
4622 return Builder.CreateBitCast(Src, DstTy, Name);
4623
4624 // Case 2.
4625 if (SrcTy->isPointerTy() && DstTy->isPointerTy())
4626 return Builder.CreatePointerBitCastOrAddrSpaceCast(Src, DstTy, Name);
4627
4628 // Case 3.
4629 if (SrcTy->isPointerTy() && !DstTy->isPointerTy()) {
4630 // Case 3b.
4631 if (!DstTy->isIntegerTy())
4632 Src = Builder.CreatePtrToInt(Src, DL.getIntPtrType(SrcTy));
4633 // Cases 3a and 3b.
4634 return Builder.CreateBitOrPointerCast(Src, DstTy, Name);
4635 }
4636
4637 // Case 4b.
4638 if (!SrcTy->isIntegerTy())
4639 Src = Builder.CreateBitCast(Src, DL.getIntPtrType(DstTy));
4640 // Cases 4a and 4b.
4641 return Builder.CreateIntToPtr(Src, DstTy, Name);
4642}
4643
Tanya Lattner55808c12011-06-04 00:47:47 +00004644Value *ScalarExprEmitter::VisitAsTypeExpr(AsTypeExpr *E) {
4645 Value *Src = CGF.EmitScalarExpr(E->getSrcExpr());
Chris Lattner2192fe52011-07-18 04:24:23 +00004646 llvm::Type *DstTy = ConvertType(E->getType());
Craig Toppera97d7e72013-07-26 06:16:11 +00004647
Chris Lattner2192fe52011-07-18 04:24:23 +00004648 llvm::Type *SrcTy = Src->getType();
Yaxun Liuc5647012016-06-08 15:11:21 +00004649 unsigned NumElementsSrc = isa<llvm::VectorType>(SrcTy) ?
4650 cast<llvm::VectorType>(SrcTy)->getNumElements() : 0;
4651 unsigned NumElementsDst = isa<llvm::VectorType>(DstTy) ?
4652 cast<llvm::VectorType>(DstTy)->getNumElements() : 0;
Craig Toppera97d7e72013-07-26 06:16:11 +00004653
Yaxun Liuc5647012016-06-08 15:11:21 +00004654 // Going from vec3 to non-vec3 is a special case and requires a shuffle
4655 // vector to get a vec4, then a bitcast if the target type is different.
4656 if (NumElementsSrc == 3 && NumElementsDst != 3) {
4657 Src = ConvertVec3AndVec4(Builder, CGF, Src, 4);
Jin-Gu Kange7cdcde2017-04-04 16:40:25 +00004658
4659 if (!CGF.CGM.getCodeGenOpts().PreserveVec3Type) {
4660 Src = createCastsForTypeOfSameSize(Builder, CGF.CGM.getDataLayout(), Src,
4661 DstTy);
4662 }
4663
Yaxun Liuc5647012016-06-08 15:11:21 +00004664 Src->setName("astype");
4665 return Src;
4666 }
Craig Toppera97d7e72013-07-26 06:16:11 +00004667
Yaxun Liuc5647012016-06-08 15:11:21 +00004668 // Going from non-vec3 to vec3 is a special case and requires a bitcast
4669 // to vec4 if the original type is not vec4, then a shuffle vector to
4670 // get a vec3.
4671 if (NumElementsSrc != 3 && NumElementsDst == 3) {
Jin-Gu Kange7cdcde2017-04-04 16:40:25 +00004672 if (!CGF.CGM.getCodeGenOpts().PreserveVec3Type) {
Christopher Tetreault79689812020-06-01 09:55:24 -07004673 auto *Vec4Ty = llvm::FixedVectorType::get(
Christopher Tetreaultf22fbe32020-04-13 12:30:53 -07004674 cast<llvm::VectorType>(DstTy)->getElementType(), 4);
Jin-Gu Kange7cdcde2017-04-04 16:40:25 +00004675 Src = createCastsForTypeOfSameSize(Builder, CGF.CGM.getDataLayout(), Src,
4676 Vec4Ty);
4677 }
4678
Yaxun Liuc5647012016-06-08 15:11:21 +00004679 Src = ConvertVec3AndVec4(Builder, CGF, Src, 3);
4680 Src->setName("astype");
4681 return Src;
Tanya Lattner55808c12011-06-04 00:47:47 +00004682 }
Craig Toppera97d7e72013-07-26 06:16:11 +00004683
Sylvestre Ledru4644e9a2019-10-12 15:24:00 +00004684 return createCastsForTypeOfSameSize(Builder, CGF.CGM.getDataLayout(),
4685 Src, DstTy, "astype");
Tanya Lattner55808c12011-06-04 00:47:47 +00004686}
4687
Eli Friedmandf14b3a2011-10-11 02:20:01 +00004688Value *ScalarExprEmitter::VisitAtomicExpr(AtomicExpr *E) {
4689 return CGF.EmitAtomicExpr(E).getScalarVal();
4690}
4691
Chris Lattner2da04b32007-08-24 05:35:26 +00004692//===----------------------------------------------------------------------===//
4693// Entry Point into this File
4694//===----------------------------------------------------------------------===//
4695
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +00004696/// Emit the computation of the specified expression of scalar type, ignoring
4697/// the result.
Mike Stumpdf0fe272009-05-29 15:46:01 +00004698Value *CodeGenFunction::EmitScalarExpr(const Expr *E, bool IgnoreResultAssign) {
John McCall47fb9502013-03-07 21:37:08 +00004699 assert(E && hasScalarEvaluationKind(E->getType()) &&
Chris Lattner2da04b32007-08-24 05:35:26 +00004700 "Invalid scalar expression to emit");
Mike Stump4a3999f2009-09-09 13:00:44 +00004701
David Blaikie38b25912015-02-09 19:13:51 +00004702 return ScalarExprEmitter(*this, IgnoreResultAssign)
4703 .Visit(const_cast<Expr *>(E));
Chris Lattner2da04b32007-08-24 05:35:26 +00004704}
Chris Lattner3474c202007-08-26 06:48:56 +00004705
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +00004706/// Emit a conversion from the specified type to the specified destination type,
4707/// both of which are LLVM scalar types.
Chris Lattner42e6b812007-08-26 16:34:22 +00004708Value *CodeGenFunction::EmitScalarConversion(Value *Src, QualType SrcTy,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00004709 QualType DstTy,
4710 SourceLocation Loc) {
John McCall47fb9502013-03-07 21:37:08 +00004711 assert(hasScalarEvaluationKind(SrcTy) && hasScalarEvaluationKind(DstTy) &&
Chris Lattner3474c202007-08-26 06:48:56 +00004712 "Invalid scalar expression to emit");
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00004713 return ScalarExprEmitter(*this).EmitScalarConversion(Src, SrcTy, DstTy, Loc);
Chris Lattner3474c202007-08-26 06:48:56 +00004714}
Chris Lattner42e6b812007-08-26 16:34:22 +00004715
Filipe Cabecinhas650d7f72015-08-05 06:19:26 +00004716/// Emit a conversion from the specified complex type to the specified
4717/// destination type, where the destination type is an LLVM scalar type.
Chris Lattner42e6b812007-08-26 16:34:22 +00004718Value *CodeGenFunction::EmitComplexToScalarConversion(ComplexPairTy Src,
4719 QualType SrcTy,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00004720 QualType DstTy,
4721 SourceLocation Loc) {
John McCall47fb9502013-03-07 21:37:08 +00004722 assert(SrcTy->isAnyComplexType() && hasScalarEvaluationKind(DstTy) &&
Chris Lattner42e6b812007-08-26 16:34:22 +00004723 "Invalid complex -> scalar conversion");
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00004724 return ScalarExprEmitter(*this)
4725 .EmitComplexToScalarConversion(Src, SrcTy, DstTy, Loc);
Chris Lattner42e6b812007-08-26 16:34:22 +00004726}
Anders Carlssonb9eb82c2007-12-10 19:35:18 +00004727
Chris Lattner05dc78c2010-06-26 22:09:34 +00004728
4729llvm::Value *CodeGenFunction::
4730EmitScalarPrePostIncDec(const UnaryOperator *E, LValue LV,
4731 bool isInc, bool isPre) {
4732 return ScalarExprEmitter(*this).EmitScalarPrePostIncDec(E, LV, isInc, isPre);
4733}
4734
Fariborz Jahanian531c16f2009-12-09 23:35:29 +00004735LValue CodeGenFunction::EmitObjCIsaExpr(const ObjCIsaExpr *E) {
Fariborz Jahanian531c16f2009-12-09 23:35:29 +00004736 // object->isa or (*object).isa
4737 // Generate code as for: *(Class*)object
Fariborz Jahaniandf506b92010-02-05 19:18:30 +00004738
4739 Expr *BaseExpr = E->getBase();
John McCall7f416cc2015-09-08 08:05:57 +00004740 Address Addr = Address::invalid();
John McCall086a4642010-11-24 05:12:34 +00004741 if (BaseExpr->isRValue()) {
John McCall7f416cc2015-09-08 08:05:57 +00004742 Addr = Address(EmitScalarExpr(BaseExpr), getPointerAlign());
Daniel Dunbarf6fb7e22010-08-21 03:08:16 +00004743 } else {
Akira Hatanakaf139ae32019-12-03 15:17:01 -08004744 Addr = EmitLValue(BaseExpr).getAddress(*this);
Fariborz Jahaniandf506b92010-02-05 19:18:30 +00004745 }
Craig Toppera97d7e72013-07-26 06:16:11 +00004746
John McCall7f416cc2015-09-08 08:05:57 +00004747 // Cast the address to Class*.
4748 Addr = Builder.CreateElementBitCast(Addr, ConvertType(E->getType()));
4749 return MakeAddrLValue(Addr, E->getType());
Fariborz Jahanian531c16f2009-12-09 23:35:29 +00004750}
4751
Douglas Gregor914af212010-04-23 04:16:32 +00004752
John McCalla2342eb2010-12-05 02:00:02 +00004753LValue CodeGenFunction::EmitCompoundAssignmentLValue(
Douglas Gregor914af212010-04-23 04:16:32 +00004754 const CompoundAssignOperator *E) {
4755 ScalarExprEmitter Scalar(*this);
Craig Topper8a13c412014-05-21 05:09:00 +00004756 Value *Result = nullptr;
Douglas Gregor914af212010-04-23 04:16:32 +00004757 switch (E->getOpcode()) {
4758#define COMPOUND_OP(Op) \
John McCalle3027922010-08-25 11:45:40 +00004759 case BO_##Op##Assign: \
Douglas Gregor914af212010-04-23 04:16:32 +00004760 return Scalar.EmitCompoundAssignLValue(E, &ScalarExprEmitter::Emit##Op, \
Daniel Dunbarc85ea8e2010-06-29 22:00:45 +00004761 Result)
Douglas Gregor914af212010-04-23 04:16:32 +00004762 COMPOUND_OP(Mul);
4763 COMPOUND_OP(Div);
4764 COMPOUND_OP(Rem);
4765 COMPOUND_OP(Add);
4766 COMPOUND_OP(Sub);
4767 COMPOUND_OP(Shl);
4768 COMPOUND_OP(Shr);
4769 COMPOUND_OP(And);
4770 COMPOUND_OP(Xor);
4771 COMPOUND_OP(Or);
4772#undef COMPOUND_OP
Craig Toppera97d7e72013-07-26 06:16:11 +00004773
John McCalle3027922010-08-25 11:45:40 +00004774 case BO_PtrMemD:
4775 case BO_PtrMemI:
4776 case BO_Mul:
4777 case BO_Div:
4778 case BO_Rem:
4779 case BO_Add:
4780 case BO_Sub:
4781 case BO_Shl:
4782 case BO_Shr:
4783 case BO_LT:
4784 case BO_GT:
4785 case BO_LE:
4786 case BO_GE:
4787 case BO_EQ:
4788 case BO_NE:
Richard Smithc70f1d62017-12-14 15:16:18 +00004789 case BO_Cmp:
John McCalle3027922010-08-25 11:45:40 +00004790 case BO_And:
4791 case BO_Xor:
4792 case BO_Or:
4793 case BO_LAnd:
4794 case BO_LOr:
4795 case BO_Assign:
4796 case BO_Comma:
David Blaikie83d382b2011-09-23 05:06:16 +00004797 llvm_unreachable("Not valid compound assignment operators");
Douglas Gregor914af212010-04-23 04:16:32 +00004798 }
Craig Toppera97d7e72013-07-26 06:16:11 +00004799
Douglas Gregor914af212010-04-23 04:16:32 +00004800 llvm_unreachable("Unhandled compound assignment operator");
4801}
Vedant Kumara125eb52017-06-01 19:22:18 +00004802
Roman Lebedev8f03dcd2019-09-06 14:18:57 +00004803struct GEPOffsetAndOverflow {
4804 // The total (signed) byte offset for the GEP.
4805 llvm::Value *TotalOffset;
4806 // The offset overflow flag - true if the total offset overflows.
4807 llvm::Value *OffsetOverflows;
4808};
Vedant Kumara125eb52017-06-01 19:22:18 +00004809
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004810/// Evaluate given GEPVal, which is either an inbounds GEP, or a constant,
Roman Lebedev8f03dcd2019-09-06 14:18:57 +00004811/// and compute the total offset it applies from it's base pointer BasePtr.
4812/// Returns offset in bytes and a boolean flag whether an overflow happened
4813/// during evaluation.
4814static GEPOffsetAndOverflow EmitGEPOffsetInBytes(Value *BasePtr, Value *GEPVal,
4815 llvm::LLVMContext &VMContext,
4816 CodeGenModule &CGM,
Nikita Popov7c362b22020-02-16 17:57:18 +01004817 CGBuilderTy &Builder) {
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004818 const auto &DL = CGM.getDataLayout();
4819
4820 // The total (signed) byte offset for the GEP.
4821 llvm::Value *TotalOffset = nullptr;
4822
4823 // Was the GEP already reduced to a constant?
4824 if (isa<llvm::Constant>(GEPVal)) {
4825 // Compute the offset by casting both pointers to integers and subtracting:
4826 // GEPVal = BasePtr + ptr(Offset) <--> Offset = int(GEPVal) - int(BasePtr)
4827 Value *BasePtr_int =
4828 Builder.CreatePtrToInt(BasePtr, DL.getIntPtrType(BasePtr->getType()));
4829 Value *GEPVal_int =
4830 Builder.CreatePtrToInt(GEPVal, DL.getIntPtrType(GEPVal->getType()));
4831 TotalOffset = Builder.CreateSub(GEPVal_int, BasePtr_int);
4832 return {TotalOffset, /*OffsetOverflows=*/Builder.getFalse()};
4833 }
4834
Vedant Kumara125eb52017-06-01 19:22:18 +00004835 auto *GEP = cast<llvm::GEPOperator>(GEPVal);
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004836 assert(GEP->getPointerOperand() == BasePtr &&
4837 "BasePtr must be the the base of the GEP.");
Vedant Kumara125eb52017-06-01 19:22:18 +00004838 assert(GEP->isInBounds() && "Expected inbounds GEP");
4839
Vedant Kumara125eb52017-06-01 19:22:18 +00004840 auto *IntPtrTy = DL.getIntPtrType(GEP->getPointerOperandType());
4841
4842 // Grab references to the signed add/mul overflow intrinsics for intptr_t.
4843 auto *Zero = llvm::ConstantInt::getNullValue(IntPtrTy);
4844 auto *SAddIntrinsic =
4845 CGM.getIntrinsic(llvm::Intrinsic::sadd_with_overflow, IntPtrTy);
4846 auto *SMulIntrinsic =
4847 CGM.getIntrinsic(llvm::Intrinsic::smul_with_overflow, IntPtrTy);
4848
Vedant Kumara125eb52017-06-01 19:22:18 +00004849 // The offset overflow flag - true if the total offset overflows.
4850 llvm::Value *OffsetOverflows = Builder.getFalse();
4851
4852 /// Return the result of the given binary operation.
4853 auto eval = [&](BinaryOperator::Opcode Opcode, llvm::Value *LHS,
4854 llvm::Value *RHS) -> llvm::Value * {
Davide Italiano77378e42017-06-01 23:55:18 +00004855 assert((Opcode == BO_Add || Opcode == BO_Mul) && "Can't eval binop");
Vedant Kumara125eb52017-06-01 19:22:18 +00004856
4857 // If the operands are constants, return a constant result.
4858 if (auto *LHSCI = dyn_cast<llvm::ConstantInt>(LHS)) {
4859 if (auto *RHSCI = dyn_cast<llvm::ConstantInt>(RHS)) {
4860 llvm::APInt N;
4861 bool HasOverflow = mayHaveIntegerOverflow(LHSCI, RHSCI, Opcode,
4862 /*Signed=*/true, N);
4863 if (HasOverflow)
4864 OffsetOverflows = Builder.getTrue();
4865 return llvm::ConstantInt::get(VMContext, N);
4866 }
4867 }
4868
4869 // Otherwise, compute the result with checked arithmetic.
4870 auto *ResultAndOverflow = Builder.CreateCall(
4871 (Opcode == BO_Add) ? SAddIntrinsic : SMulIntrinsic, {LHS, RHS});
4872 OffsetOverflows = Builder.CreateOr(
Vedant Kumar6dbf4272017-06-12 18:42:51 +00004873 Builder.CreateExtractValue(ResultAndOverflow, 1), OffsetOverflows);
Vedant Kumara125eb52017-06-01 19:22:18 +00004874 return Builder.CreateExtractValue(ResultAndOverflow, 0);
4875 };
4876
4877 // Determine the total byte offset by looking at each GEP operand.
4878 for (auto GTI = llvm::gep_type_begin(GEP), GTE = llvm::gep_type_end(GEP);
4879 GTI != GTE; ++GTI) {
4880 llvm::Value *LocalOffset;
4881 auto *Index = GTI.getOperand();
4882 // Compute the local offset contributed by this indexing step:
4883 if (auto *STy = GTI.getStructTypeOrNull()) {
4884 // For struct indexing, the local offset is the byte position of the
4885 // specified field.
4886 unsigned FieldNo = cast<llvm::ConstantInt>(Index)->getZExtValue();
4887 LocalOffset = llvm::ConstantInt::get(
4888 IntPtrTy, DL.getStructLayout(STy)->getElementOffset(FieldNo));
4889 } else {
4890 // Otherwise this is array-like indexing. The local offset is the index
4891 // multiplied by the element size.
4892 auto *ElementSize = llvm::ConstantInt::get(
4893 IntPtrTy, DL.getTypeAllocSize(GTI.getIndexedType()));
4894 auto *IndexS = Builder.CreateIntCast(Index, IntPtrTy, /*isSigned=*/true);
4895 LocalOffset = eval(BO_Mul, ElementSize, IndexS);
4896 }
4897
4898 // If this is the first offset, set it as the total offset. Otherwise, add
4899 // the local offset into the running total.
4900 if (!TotalOffset || TotalOffset == Zero)
4901 TotalOffset = LocalOffset;
4902 else
4903 TotalOffset = eval(BO_Add, TotalOffset, LocalOffset);
4904 }
4905
Roman Lebedev8f03dcd2019-09-06 14:18:57 +00004906 return {TotalOffset, OffsetOverflows};
4907}
4908
4909Value *
4910CodeGenFunction::EmitCheckedInBoundsGEP(Value *Ptr, ArrayRef<Value *> IdxList,
4911 bool SignedIndices, bool IsSubtraction,
4912 SourceLocation Loc, const Twine &Name) {
4913 Value *GEPVal = Builder.CreateInBoundsGEP(Ptr, IdxList, Name);
4914
4915 // If the pointer overflow sanitizer isn't enabled, do nothing.
4916 if (!SanOpts.has(SanitizerKind::PointerOverflow))
4917 return GEPVal;
4918
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004919 llvm::Type *PtrTy = Ptr->getType();
4920
4921 // Perform nullptr-and-offset check unless the nullptr is defined.
4922 bool PerformNullCheck = !NullPointerIsDefined(
4923 Builder.GetInsertBlock()->getParent(), PtrTy->getPointerAddressSpace());
4924 // Check for overflows unless the GEP got constant-folded,
4925 // and only in the default address space
4926 bool PerformOverflowCheck =
4927 !isa<llvm::Constant>(GEPVal) && PtrTy->getPointerAddressSpace() == 0;
4928
4929 if (!(PerformNullCheck || PerformOverflowCheck))
Roman Lebedev8f03dcd2019-09-06 14:18:57 +00004930 return GEPVal;
4931
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004932 const auto &DL = CGM.getDataLayout();
Roman Lebedev8f03dcd2019-09-06 14:18:57 +00004933
4934 SanitizerScope SanScope(this);
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004935 llvm::Type *IntPtrTy = DL.getIntPtrType(PtrTy);
Roman Lebedev8f03dcd2019-09-06 14:18:57 +00004936
4937 GEPOffsetAndOverflow EvaluatedGEP =
4938 EmitGEPOffsetInBytes(Ptr, GEPVal, getLLVMContext(), CGM, Builder);
4939
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004940 assert((!isa<llvm::Constant>(EvaluatedGEP.TotalOffset) ||
4941 EvaluatedGEP.OffsetOverflows == Builder.getFalse()) &&
4942 "If the offset got constant-folded, we don't expect that there was an "
4943 "overflow.");
Roman Lebedev8f03dcd2019-09-06 14:18:57 +00004944
4945 auto *Zero = llvm::ConstantInt::getNullValue(IntPtrTy);
4946
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004947 // Common case: if the total offset is zero, and we are using C++ semantics,
4948 // where nullptr+0 is defined, don't emit a check.
4949 if (EvaluatedGEP.TotalOffset == Zero && CGM.getLangOpts().CPlusPlus)
Vedant Kumara125eb52017-06-01 19:22:18 +00004950 return GEPVal;
4951
4952 // Now that we've computed the total offset, add it to the base pointer (with
4953 // wrapping semantics).
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004954 auto *IntPtr = Builder.CreatePtrToInt(Ptr, IntPtrTy);
Roman Lebedev8f03dcd2019-09-06 14:18:57 +00004955 auto *ComputedGEP = Builder.CreateAdd(IntPtr, EvaluatedGEP.TotalOffset);
Vedant Kumara125eb52017-06-01 19:22:18 +00004956
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004957 llvm::SmallVector<std::pair<llvm::Value *, SanitizerMask>, 2> Checks;
Roman Lebedevf1d33842019-09-06 14:19:04 +00004958
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004959 if (PerformNullCheck) {
4960 // In C++, if the base pointer evaluates to a null pointer value,
4961 // the only valid pointer this inbounds GEP can produce is also
4962 // a null pointer, so the offset must also evaluate to zero.
4963 // Likewise, if we have non-zero base pointer, we can not get null pointer
4964 // as a result, so the offset can not be -intptr_t(BasePtr).
4965 // In other words, both pointers are either null, or both are non-null,
4966 // or the behaviour is undefined.
4967 //
4968 // C, however, is more strict in this regard, and gives more
4969 // optimization opportunities: in C, additionally, nullptr+0 is undefined.
4970 // So both the input to the 'gep inbounds' AND the output must not be null.
4971 auto *BaseIsNotNullptr = Builder.CreateIsNotNull(Ptr);
4972 auto *ResultIsNotNullptr = Builder.CreateIsNotNull(ComputedGEP);
4973 auto *Valid =
4974 CGM.getLangOpts().CPlusPlus
4975 ? Builder.CreateICmpEQ(BaseIsNotNullptr, ResultIsNotNullptr)
4976 : Builder.CreateAnd(BaseIsNotNullptr, ResultIsNotNullptr);
4977 Checks.emplace_back(Valid, SanitizerKind::PointerOverflow);
Vedant Kumar6dbf4272017-06-12 18:42:51 +00004978 }
Roman Lebedev536b0ee2019-10-10 09:25:02 +00004979
4980 if (PerformOverflowCheck) {
4981 // The GEP is valid if:
4982 // 1) The total offset doesn't overflow, and
4983 // 2) The sign of the difference between the computed address and the base
4984 // pointer matches the sign of the total offset.
4985 llvm::Value *ValidGEP;
4986 auto *NoOffsetOverflow = Builder.CreateNot(EvaluatedGEP.OffsetOverflows);
4987 if (SignedIndices) {
4988 // GEP is computed as `unsigned base + signed offset`, therefore:
4989 // * If offset was positive, then the computed pointer can not be
4990 // [unsigned] less than the base pointer, unless it overflowed.
4991 // * If offset was negative, then the computed pointer can not be
4992 // [unsigned] greater than the bas pointere, unless it overflowed.
4993 auto *PosOrZeroValid = Builder.CreateICmpUGE(ComputedGEP, IntPtr);
4994 auto *PosOrZeroOffset =
4995 Builder.CreateICmpSGE(EvaluatedGEP.TotalOffset, Zero);
4996 llvm::Value *NegValid = Builder.CreateICmpULT(ComputedGEP, IntPtr);
4997 ValidGEP =
4998 Builder.CreateSelect(PosOrZeroOffset, PosOrZeroValid, NegValid);
4999 } else if (!IsSubtraction) {
5000 // GEP is computed as `unsigned base + unsigned offset`, therefore the
5001 // computed pointer can not be [unsigned] less than base pointer,
5002 // unless there was an overflow.
5003 // Equivalent to `@llvm.uadd.with.overflow(%base, %offset)`.
5004 ValidGEP = Builder.CreateICmpUGE(ComputedGEP, IntPtr);
5005 } else {
5006 // GEP is computed as `unsigned base - unsigned offset`, therefore the
5007 // computed pointer can not be [unsigned] greater than base pointer,
5008 // unless there was an overflow.
5009 // Equivalent to `@llvm.usub.with.overflow(%base, sub(0, %offset))`.
5010 ValidGEP = Builder.CreateICmpULE(ComputedGEP, IntPtr);
5011 }
5012 ValidGEP = Builder.CreateAnd(ValidGEP, NoOffsetOverflow);
5013 Checks.emplace_back(ValidGEP, SanitizerKind::PointerOverflow);
5014 }
Roman Lebedevf1d33842019-09-06 14:19:04 +00005015
5016 assert(!Checks.empty() && "Should have produced some checks.");
Vedant Kumara125eb52017-06-01 19:22:18 +00005017
5018 llvm::Constant *StaticArgs[] = {EmitCheckSourceLocation(Loc)};
5019 // Pass the computed GEP to the runtime to avoid emitting poisoned arguments.
5020 llvm::Value *DynamicArgs[] = {IntPtr, ComputedGEP};
Roman Lebedevf1d33842019-09-06 14:19:04 +00005021 EmitCheck(Checks, SanitizerHandler::PointerOverflow, StaticArgs, DynamicArgs);
Vedant Kumara125eb52017-06-01 19:22:18 +00005022
5023 return GEPVal;
5024}