1 //===--- Stmt.cpp - Statement AST Node Implementation ---------------------===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 // This file implements the Stmt class and statement subclasses.
12 //===----------------------------------------------------------------------===//
14 #include "clang/AST/ASTContext.h"
15 #include "clang/AST/ASTDiagnostic.h"
16 #include "clang/AST/ExprCXX.h"
17 #include "clang/AST/ExprObjC.h"
18 #include "clang/AST/ExprOpenMP.h"
19 #include "clang/AST/Stmt.h"
20 #include "clang/AST/StmtCXX.h"
21 #include "clang/AST/StmtObjC.h"
22 #include "clang/AST/StmtOpenMP.h"
23 #include "clang/AST/Type.h"
24 #include "clang/Basic/CharInfo.h"
25 #include "clang/Basic/TargetInfo.h"
26 #include "clang/Lex/Token.h"
27 #include "llvm/ADT/StringExtras.h"
28 #include "llvm/Support/raw_ostream.h"
29 using namespace clang;
31 static struct StmtClassNameTable {
35 } StmtClassInfo[Stmt::lastStmtConstant+1];
37 static StmtClassNameTable &getStmtInfoTableEntry(Stmt::StmtClass E) {
38 static bool Initialized = false;
40 return StmtClassInfo[E];
42 // Intialize the table on the first use.
44 #define ABSTRACT_STMT(STMT)
45 #define STMT(CLASS, PARENT) \
46 StmtClassInfo[(unsigned)Stmt::CLASS##Class].Name = #CLASS; \
47 StmtClassInfo[(unsigned)Stmt::CLASS##Class].Size = sizeof(CLASS);
48 #include "clang/AST/StmtNodes.inc"
50 return StmtClassInfo[E];
53 void *Stmt::operator new(size_t bytes, const ASTContext& C,
55 return ::operator new(bytes, C, alignment);
58 const char *Stmt::getStmtClassName() const {
59 return getStmtInfoTableEntry((StmtClass) StmtBits.sClass).Name;
62 void Stmt::PrintStats() {
63 // Ensure the table is primed.
64 getStmtInfoTableEntry(Stmt::NullStmtClass);
67 llvm::errs() << "\n*** Stmt/Expr Stats:\n";
68 for (int i = 0; i != Stmt::lastStmtConstant+1; i++) {
69 if (StmtClassInfo[i].Name == nullptr) continue;
70 sum += StmtClassInfo[i].Counter;
72 llvm::errs() << " " << sum << " stmts/exprs total.\n";
74 for (int i = 0; i != Stmt::lastStmtConstant+1; i++) {
75 if (StmtClassInfo[i].Name == nullptr) continue;
76 if (StmtClassInfo[i].Counter == 0) continue;
77 llvm::errs() << " " << StmtClassInfo[i].Counter << " "
78 << StmtClassInfo[i].Name << ", " << StmtClassInfo[i].Size
79 << " each (" << StmtClassInfo[i].Counter*StmtClassInfo[i].Size
81 sum += StmtClassInfo[i].Counter*StmtClassInfo[i].Size;
84 llvm::errs() << "Total bytes = " << sum << "\n";
87 void Stmt::addStmtClass(StmtClass s) {
88 ++getStmtInfoTableEntry(s).Counter;
91 bool Stmt::StatisticsEnabled = false;
92 void Stmt::EnableStatistics() {
93 StatisticsEnabled = true;
96 Stmt *Stmt::IgnoreImplicit() {
99 if (auto *ewc = dyn_cast<ExprWithCleanups>(s))
100 s = ewc->getSubExpr();
102 if (auto *mte = dyn_cast<MaterializeTemporaryExpr>(s))
103 s = mte->GetTemporaryExpr();
105 if (auto *bte = dyn_cast<CXXBindTemporaryExpr>(s))
106 s = bte->getSubExpr();
108 while (auto *ice = dyn_cast<ImplicitCastExpr>(s))
109 s = ice->getSubExpr();
114 /// \brief Skip no-op (attributed, compound) container stmts and skip captured
115 /// stmt at the top, if \a IgnoreCaptured is true.
116 Stmt *Stmt::IgnoreContainers(bool IgnoreCaptured) {
119 if (auto CapS = dyn_cast_or_null<CapturedStmt>(S))
120 S = CapS->getCapturedStmt();
122 if (auto AS = dyn_cast_or_null<AttributedStmt>(S))
123 S = AS->getSubStmt();
124 else if (auto CS = dyn_cast_or_null<CompoundStmt>(S)) {
134 /// \brief Strip off all label-like statements.
136 /// This will strip off label statements, case statements, attributed
137 /// statements and default statements recursively.
138 const Stmt *Stmt::stripLabelLikeStatements() const {
139 const Stmt *S = this;
141 if (const LabelStmt *LS = dyn_cast<LabelStmt>(S))
142 S = LS->getSubStmt();
143 else if (const SwitchCase *SC = dyn_cast<SwitchCase>(S))
144 S = SC->getSubStmt();
145 else if (const AttributedStmt *AS = dyn_cast<AttributedStmt>(S))
146 S = AS->getSubStmt();
156 // These silly little functions have to be static inline to suppress
157 // unused warnings, and they have to be defined to suppress other
159 static inline good is_good(good) { return good(); }
161 typedef Stmt::child_range children_t();
162 template <class T> good implements_children(children_t T::*) {
165 LLVM_ATTRIBUTE_UNUSED
166 static inline bad implements_children(children_t Stmt::*) {
170 typedef SourceLocation getLocStart_t() const;
171 template <class T> good implements_getLocStart(getLocStart_t T::*) {
174 LLVM_ATTRIBUTE_UNUSED
175 static inline bad implements_getLocStart(getLocStart_t Stmt::*) {
179 typedef SourceLocation getLocEnd_t() const;
180 template <class T> good implements_getLocEnd(getLocEnd_t T::*) {
183 LLVM_ATTRIBUTE_UNUSED
184 static inline bad implements_getLocEnd(getLocEnd_t Stmt::*) {
188 #define ASSERT_IMPLEMENTS_children(type) \
189 (void) is_good(implements_children(&type::children))
190 #define ASSERT_IMPLEMENTS_getLocStart(type) \
191 (void) is_good(implements_getLocStart(&type::getLocStart))
192 #define ASSERT_IMPLEMENTS_getLocEnd(type) \
193 (void) is_good(implements_getLocEnd(&type::getLocEnd))
196 /// Check whether the various Stmt classes implement their member
198 LLVM_ATTRIBUTE_UNUSED
199 static inline void check_implementations() {
200 #define ABSTRACT_STMT(type)
201 #define STMT(type, base) \
202 ASSERT_IMPLEMENTS_children(type); \
203 ASSERT_IMPLEMENTS_getLocStart(type); \
204 ASSERT_IMPLEMENTS_getLocEnd(type);
205 #include "clang/AST/StmtNodes.inc"
208 Stmt::child_range Stmt::children() {
209 switch (getStmtClass()) {
210 case Stmt::NoStmtClass: llvm_unreachable("statement without class");
211 #define ABSTRACT_STMT(type)
212 #define STMT(type, base) \
213 case Stmt::type##Class: \
214 return static_cast<type*>(this)->children();
215 #include "clang/AST/StmtNodes.inc"
217 llvm_unreachable("unknown statement kind!");
220 // Amusing macro metaprogramming hack: check whether a class provides
221 // a more specific implementation of getSourceRange.
223 // See also Expr.cpp:getExprLoc().
225 /// This implementation is used when a class provides a custom
226 /// implementation of getSourceRange.
227 template <class S, class T>
228 SourceRange getSourceRangeImpl(const Stmt *stmt,
229 SourceRange (T::*v)() const) {
230 return static_cast<const S*>(stmt)->getSourceRange();
233 /// This implementation is used when a class doesn't provide a custom
234 /// implementation of getSourceRange. Overload resolution should pick it over
235 /// the implementation above because it's more specialized according to
236 /// function template partial ordering.
238 SourceRange getSourceRangeImpl(const Stmt *stmt,
239 SourceRange (Stmt::*v)() const) {
240 return SourceRange(static_cast<const S*>(stmt)->getLocStart(),
241 static_cast<const S*>(stmt)->getLocEnd());
245 SourceRange Stmt::getSourceRange() const {
246 switch (getStmtClass()) {
247 case Stmt::NoStmtClass: llvm_unreachable("statement without class");
248 #define ABSTRACT_STMT(type)
249 #define STMT(type, base) \
250 case Stmt::type##Class: \
251 return getSourceRangeImpl<type>(this, &type::getSourceRange);
252 #include "clang/AST/StmtNodes.inc"
254 llvm_unreachable("unknown statement kind!");
257 SourceLocation Stmt::getLocStart() const {
258 // llvm::errs() << "getLocStart() for " << getStmtClassName() << "\n";
259 switch (getStmtClass()) {
260 case Stmt::NoStmtClass: llvm_unreachable("statement without class");
261 #define ABSTRACT_STMT(type)
262 #define STMT(type, base) \
263 case Stmt::type##Class: \
264 return static_cast<const type*>(this)->getLocStart();
265 #include "clang/AST/StmtNodes.inc"
267 llvm_unreachable("unknown statement kind");
270 SourceLocation Stmt::getLocEnd() const {
271 switch (getStmtClass()) {
272 case Stmt::NoStmtClass: llvm_unreachable("statement without class");
273 #define ABSTRACT_STMT(type)
274 #define STMT(type, base) \
275 case Stmt::type##Class: \
276 return static_cast<const type*>(this)->getLocEnd();
277 #include "clang/AST/StmtNodes.inc"
279 llvm_unreachable("unknown statement kind");
282 CompoundStmt::CompoundStmt(const ASTContext &C, ArrayRef<Stmt*> Stmts,
283 SourceLocation LB, SourceLocation RB)
284 : Stmt(CompoundStmtClass), LBraceLoc(LB), RBraceLoc(RB) {
285 CompoundStmtBits.NumStmts = Stmts.size();
286 assert(CompoundStmtBits.NumStmts == Stmts.size() &&
287 "NumStmts doesn't fit in bits of CompoundStmtBits.NumStmts!");
289 if (Stmts.size() == 0) {
294 Body = new (C) Stmt*[Stmts.size()];
295 std::copy(Stmts.begin(), Stmts.end(), Body);
298 void CompoundStmt::setStmts(const ASTContext &C, ArrayRef<Stmt *> Stmts) {
301 CompoundStmtBits.NumStmts = Stmts.size();
302 assert(CompoundStmtBits.NumStmts == Stmts.size() &&
303 "NumStmts doesn't fit in bits of CompoundStmtBits.NumStmts!");
305 Body = new (C) Stmt*[Stmts.size()];
306 std::copy(Stmts.begin(), Stmts.end(), Body);
309 const char *LabelStmt::getName() const {
310 return getDecl()->getIdentifier()->getNameStart();
313 AttributedStmt *AttributedStmt::Create(const ASTContext &C, SourceLocation Loc,
314 ArrayRef<const Attr*> Attrs,
316 assert(!Attrs.empty() && "Attrs should not be empty");
317 void *Mem = C.Allocate(sizeof(AttributedStmt) + sizeof(Attr *) * Attrs.size(),
318 alignof(AttributedStmt));
319 return new (Mem) AttributedStmt(Loc, Attrs, SubStmt);
322 AttributedStmt *AttributedStmt::CreateEmpty(const ASTContext &C,
324 assert(NumAttrs > 0 && "NumAttrs should be greater than zero");
325 void *Mem = C.Allocate(sizeof(AttributedStmt) + sizeof(Attr *) * NumAttrs,
326 alignof(AttributedStmt));
327 return new (Mem) AttributedStmt(EmptyShell(), NumAttrs);
330 std::string AsmStmt::generateAsmString(const ASTContext &C) const {
331 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
332 return gccAsmStmt->generateAsmString(C);
333 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
334 return msAsmStmt->generateAsmString(C);
335 llvm_unreachable("unknown asm statement kind!");
338 StringRef AsmStmt::getOutputConstraint(unsigned i) const {
339 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
340 return gccAsmStmt->getOutputConstraint(i);
341 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
342 return msAsmStmt->getOutputConstraint(i);
343 llvm_unreachable("unknown asm statement kind!");
346 const Expr *AsmStmt::getOutputExpr(unsigned i) const {
347 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
348 return gccAsmStmt->getOutputExpr(i);
349 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
350 return msAsmStmt->getOutputExpr(i);
351 llvm_unreachable("unknown asm statement kind!");
354 StringRef AsmStmt::getInputConstraint(unsigned i) const {
355 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
356 return gccAsmStmt->getInputConstraint(i);
357 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
358 return msAsmStmt->getInputConstraint(i);
359 llvm_unreachable("unknown asm statement kind!");
362 const Expr *AsmStmt::getInputExpr(unsigned i) const {
363 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
364 return gccAsmStmt->getInputExpr(i);
365 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
366 return msAsmStmt->getInputExpr(i);
367 llvm_unreachable("unknown asm statement kind!");
370 StringRef AsmStmt::getClobber(unsigned i) const {
371 if (const GCCAsmStmt *gccAsmStmt = dyn_cast<GCCAsmStmt>(this))
372 return gccAsmStmt->getClobber(i);
373 if (const MSAsmStmt *msAsmStmt = dyn_cast<MSAsmStmt>(this))
374 return msAsmStmt->getClobber(i);
375 llvm_unreachable("unknown asm statement kind!");
378 /// getNumPlusOperands - Return the number of output operands that have a "+"
380 unsigned AsmStmt::getNumPlusOperands() const {
382 for (unsigned i = 0, e = getNumOutputs(); i != e; ++i)
383 if (isOutputPlusConstraint(i))
388 char GCCAsmStmt::AsmStringPiece::getModifier() const {
389 assert(isOperand() && "Only Operands can have modifiers.");
390 return isLetter(Str[0]) ? Str[0] : '\0';
393 StringRef GCCAsmStmt::getClobber(unsigned i) const {
394 return getClobberStringLiteral(i)->getString();
397 Expr *GCCAsmStmt::getOutputExpr(unsigned i) {
398 return cast<Expr>(Exprs[i]);
401 /// getOutputConstraint - Return the constraint string for the specified
402 /// output operand. All output constraints are known to be non-empty (either
404 StringRef GCCAsmStmt::getOutputConstraint(unsigned i) const {
405 return getOutputConstraintLiteral(i)->getString();
408 Expr *GCCAsmStmt::getInputExpr(unsigned i) {
409 return cast<Expr>(Exprs[i + NumOutputs]);
411 void GCCAsmStmt::setInputExpr(unsigned i, Expr *E) {
412 Exprs[i + NumOutputs] = E;
415 /// getInputConstraint - Return the specified input constraint. Unlike output
416 /// constraints, these can be empty.
417 StringRef GCCAsmStmt::getInputConstraint(unsigned i) const {
418 return getInputConstraintLiteral(i)->getString();
421 void GCCAsmStmt::setOutputsAndInputsAndClobbers(const ASTContext &C,
422 IdentifierInfo **Names,
423 StringLiteral **Constraints,
427 StringLiteral **Clobbers,
428 unsigned NumClobbers) {
429 this->NumOutputs = NumOutputs;
430 this->NumInputs = NumInputs;
431 this->NumClobbers = NumClobbers;
433 unsigned NumExprs = NumOutputs + NumInputs;
435 C.Deallocate(this->Names);
436 this->Names = new (C) IdentifierInfo*[NumExprs];
437 std::copy(Names, Names + NumExprs, this->Names);
439 C.Deallocate(this->Exprs);
440 this->Exprs = new (C) Stmt*[NumExprs];
441 std::copy(Exprs, Exprs + NumExprs, this->Exprs);
443 C.Deallocate(this->Constraints);
444 this->Constraints = new (C) StringLiteral*[NumExprs];
445 std::copy(Constraints, Constraints + NumExprs, this->Constraints);
447 C.Deallocate(this->Clobbers);
448 this->Clobbers = new (C) StringLiteral*[NumClobbers];
449 std::copy(Clobbers, Clobbers + NumClobbers, this->Clobbers);
452 /// getNamedOperand - Given a symbolic operand reference like %[foo],
453 /// translate this into a numeric value needed to reference the same operand.
454 /// This returns -1 if the operand name is invalid.
455 int GCCAsmStmt::getNamedOperand(StringRef SymbolicName) const {
456 unsigned NumPlusOperands = 0;
458 // Check if this is an output operand.
459 for (unsigned i = 0, e = getNumOutputs(); i != e; ++i) {
460 if (getOutputName(i) == SymbolicName)
464 for (unsigned i = 0, e = getNumInputs(); i != e; ++i)
465 if (getInputName(i) == SymbolicName)
466 return getNumOutputs() + NumPlusOperands + i;
472 /// AnalyzeAsmString - Analyze the asm string of the current asm, decomposing
473 /// it into pieces. If the asm string is erroneous, emit errors and return
474 /// true, otherwise return false.
475 unsigned GCCAsmStmt::AnalyzeAsmString(SmallVectorImpl<AsmStringPiece>&Pieces,
476 const ASTContext &C, unsigned &DiagOffs) const {
477 StringRef Str = getAsmString()->getString();
478 const char *StrStart = Str.begin();
479 const char *StrEnd = Str.end();
480 const char *CurPtr = StrStart;
482 // "Simple" inline asms have no constraints or operands, just convert the asm
483 // string to escape $'s.
486 for (; CurPtr != StrEnd; ++CurPtr) {
496 Pieces.push_back(AsmStringPiece(Result));
500 // CurStringPiece - The current string that we are building up as we scan the
502 std::string CurStringPiece;
504 bool HasVariants = !C.getTargetInfo().hasNoAsmVariants();
506 unsigned LastAsmStringToken = 0;
507 unsigned LastAsmStringOffset = 0;
510 // Done with the string?
511 if (CurPtr == StrEnd) {
512 if (!CurStringPiece.empty())
513 Pieces.push_back(AsmStringPiece(CurStringPiece));
517 char CurChar = *CurPtr++;
519 case '$': CurStringPiece += "$$"; continue;
520 case '{': CurStringPiece += (HasVariants ? "$(" : "{"); continue;
521 case '|': CurStringPiece += (HasVariants ? "$|" : "|"); continue;
522 case '}': CurStringPiece += (HasVariants ? "$)" : "}"); continue;
526 CurStringPiece += CurChar;
530 // Escaped "%" character in asm string.
531 if (CurPtr == StrEnd) {
532 // % at end of string is invalid (no escape).
533 DiagOffs = CurPtr-StrStart-1;
534 return diag::err_asm_invalid_escape;
536 // Handle escaped char and continue looping over the asm string.
537 char EscapedChar = *CurPtr++;
538 switch (EscapedChar) {
544 CurStringPiece += EscapedChar;
546 case '=': // %= -> Generate a unique ID.
547 CurStringPiece += "${:uid}";
551 // Otherwise, we have an operand. If we have accumulated a string so far,
552 // add it to the Pieces list.
553 if (!CurStringPiece.empty()) {
554 Pieces.push_back(AsmStringPiece(CurStringPiece));
555 CurStringPiece.clear();
558 // Handle operands that have asmSymbolicName (e.g., %x[foo]) and those that
559 // don't (e.g., %x4). 'x' following the '%' is the constraint modifier.
561 const char *Begin = CurPtr - 1; // Points to the character following '%'.
562 const char *Percent = Begin - 1; // Points to '%'.
564 if (isLetter(EscapedChar)) {
565 if (CurPtr == StrEnd) { // Premature end.
566 DiagOffs = CurPtr-StrStart-1;
567 return diag::err_asm_invalid_escape;
569 EscapedChar = *CurPtr++;
572 const TargetInfo &TI = C.getTargetInfo();
573 const SourceManager &SM = C.getSourceManager();
574 const LangOptions &LO = C.getLangOpts();
576 // Handle operands that don't have asmSymbolicName (e.g., %x4).
577 if (isDigit(EscapedChar)) {
578 // %n - Assembler operand n
582 while (CurPtr != StrEnd && isDigit(*CurPtr))
583 N = N*10 + ((*CurPtr++)-'0');
585 unsigned NumOperands =
586 getNumOutputs() + getNumPlusOperands() + getNumInputs();
587 if (N >= NumOperands) {
588 DiagOffs = CurPtr-StrStart-1;
589 return diag::err_asm_invalid_operand_number;
592 // Str contains "x4" (Operand without the leading %).
593 std::string Str(Begin, CurPtr - Begin);
595 // (BeginLoc, EndLoc) represents the range of the operand we are currently
596 // processing. Unlike Str, the range includes the leading '%'.
597 SourceLocation BeginLoc = getAsmString()->getLocationOfByte(
598 Percent - StrStart, SM, LO, TI, &LastAsmStringToken,
599 &LastAsmStringOffset);
600 SourceLocation EndLoc = getAsmString()->getLocationOfByte(
601 CurPtr - StrStart, SM, LO, TI, &LastAsmStringToken,
602 &LastAsmStringOffset);
604 Pieces.emplace_back(N, std::move(Str), BeginLoc, EndLoc);
608 // Handle operands that have asmSymbolicName (e.g., %x[foo]).
609 if (EscapedChar == '[') {
610 DiagOffs = CurPtr-StrStart-1;
613 const char *NameEnd = (const char*)memchr(CurPtr, ']', StrEnd-CurPtr);
614 if (NameEnd == nullptr)
615 return diag::err_asm_unterminated_symbolic_operand_name;
616 if (NameEnd == CurPtr)
617 return diag::err_asm_empty_symbolic_operand_name;
619 StringRef SymbolicName(CurPtr, NameEnd - CurPtr);
621 int N = getNamedOperand(SymbolicName);
623 // Verify that an operand with that name exists.
624 DiagOffs = CurPtr-StrStart;
625 return diag::err_asm_unknown_symbolic_operand_name;
628 // Str contains "x[foo]" (Operand without the leading %).
629 std::string Str(Begin, NameEnd + 1 - Begin);
631 // (BeginLoc, EndLoc) represents the range of the operand we are currently
632 // processing. Unlike Str, the range includes the leading '%'.
633 SourceLocation BeginLoc = getAsmString()->getLocationOfByte(
634 Percent - StrStart, SM, LO, TI, &LastAsmStringToken,
635 &LastAsmStringOffset);
636 SourceLocation EndLoc = getAsmString()->getLocationOfByte(
637 NameEnd + 1 - StrStart, SM, LO, TI, &LastAsmStringToken,
638 &LastAsmStringOffset);
640 Pieces.emplace_back(N, std::move(Str), BeginLoc, EndLoc);
646 DiagOffs = CurPtr-StrStart-1;
647 return diag::err_asm_invalid_escape;
651 /// Assemble final IR asm string (GCC-style).
652 std::string GCCAsmStmt::generateAsmString(const ASTContext &C) const {
653 // Analyze the asm string to decompose it into its pieces. We know that Sema
654 // has already done this, so it is guaranteed to be successful.
655 SmallVector<GCCAsmStmt::AsmStringPiece, 4> Pieces;
657 AnalyzeAsmString(Pieces, C, DiagOffs);
659 std::string AsmString;
660 for (unsigned i = 0, e = Pieces.size(); i != e; ++i) {
661 if (Pieces[i].isString())
662 AsmString += Pieces[i].getString();
663 else if (Pieces[i].getModifier() == '\0')
664 AsmString += '$' + llvm::utostr(Pieces[i].getOperandNo());
666 AsmString += "${" + llvm::utostr(Pieces[i].getOperandNo()) + ':' +
667 Pieces[i].getModifier() + '}';
672 /// Assemble final IR asm string (MS-style).
673 std::string MSAsmStmt::generateAsmString(const ASTContext &C) const {
674 // FIXME: This needs to be translated into the IR string representation.
678 Expr *MSAsmStmt::getOutputExpr(unsigned i) {
679 return cast<Expr>(Exprs[i]);
682 Expr *MSAsmStmt::getInputExpr(unsigned i) {
683 return cast<Expr>(Exprs[i + NumOutputs]);
685 void MSAsmStmt::setInputExpr(unsigned i, Expr *E) {
686 Exprs[i + NumOutputs] = E;
689 //===----------------------------------------------------------------------===//
691 //===----------------------------------------------------------------------===//
693 GCCAsmStmt::GCCAsmStmt(const ASTContext &C, SourceLocation asmloc,
694 bool issimple, bool isvolatile, unsigned numoutputs,
695 unsigned numinputs, IdentifierInfo **names,
696 StringLiteral **constraints, Expr **exprs,
697 StringLiteral *asmstr, unsigned numclobbers,
698 StringLiteral **clobbers, SourceLocation rparenloc)
699 : AsmStmt(GCCAsmStmtClass, asmloc, issimple, isvolatile, numoutputs,
700 numinputs, numclobbers), RParenLoc(rparenloc), AsmStr(asmstr) {
702 unsigned NumExprs = NumOutputs + NumInputs;
704 Names = new (C) IdentifierInfo*[NumExprs];
705 std::copy(names, names + NumExprs, Names);
707 Exprs = new (C) Stmt*[NumExprs];
708 std::copy(exprs, exprs + NumExprs, Exprs);
710 Constraints = new (C) StringLiteral*[NumExprs];
711 std::copy(constraints, constraints + NumExprs, Constraints);
713 Clobbers = new (C) StringLiteral*[NumClobbers];
714 std::copy(clobbers, clobbers + NumClobbers, Clobbers);
717 MSAsmStmt::MSAsmStmt(const ASTContext &C, SourceLocation asmloc,
718 SourceLocation lbraceloc, bool issimple, bool isvolatile,
719 ArrayRef<Token> asmtoks, unsigned numoutputs,
721 ArrayRef<StringRef> constraints, ArrayRef<Expr*> exprs,
722 StringRef asmstr, ArrayRef<StringRef> clobbers,
723 SourceLocation endloc)
724 : AsmStmt(MSAsmStmtClass, asmloc, issimple, isvolatile, numoutputs,
725 numinputs, clobbers.size()), LBraceLoc(lbraceloc),
726 EndLoc(endloc), NumAsmToks(asmtoks.size()) {
728 initialize(C, asmstr, asmtoks, constraints, exprs, clobbers);
731 static StringRef copyIntoContext(const ASTContext &C, StringRef str) {
735 void MSAsmStmt::initialize(const ASTContext &C, StringRef asmstr,
736 ArrayRef<Token> asmtoks,
737 ArrayRef<StringRef> constraints,
738 ArrayRef<Expr*> exprs,
739 ArrayRef<StringRef> clobbers) {
740 assert(NumAsmToks == asmtoks.size());
741 assert(NumClobbers == clobbers.size());
743 assert(exprs.size() == NumOutputs + NumInputs);
744 assert(exprs.size() == constraints.size());
746 AsmStr = copyIntoContext(C, asmstr);
748 Exprs = new (C) Stmt*[exprs.size()];
749 std::copy(exprs.begin(), exprs.end(), Exprs);
751 AsmToks = new (C) Token[asmtoks.size()];
752 std::copy(asmtoks.begin(), asmtoks.end(), AsmToks);
754 Constraints = new (C) StringRef[exprs.size()];
755 std::transform(constraints.begin(), constraints.end(), Constraints,
756 [&](StringRef Constraint) {
757 return copyIntoContext(C, Constraint);
760 Clobbers = new (C) StringRef[NumClobbers];
761 // FIXME: Avoid the allocation/copy if at all possible.
762 std::transform(clobbers.begin(), clobbers.end(), Clobbers,
763 [&](StringRef Clobber) {
764 return copyIntoContext(C, Clobber);
768 IfStmt::IfStmt(const ASTContext &C, SourceLocation IL, bool IsConstexpr,
769 Stmt *init, VarDecl *var, Expr *cond, Stmt *then,
770 SourceLocation EL, Stmt *elsev)
771 : Stmt(IfStmtClass), IfLoc(IL), ElseLoc(EL) {
772 setConstexpr(IsConstexpr);
773 setConditionVariable(C, var);
774 SubExprs[INIT] = init;
775 SubExprs[COND] = cond;
776 SubExprs[THEN] = then;
777 SubExprs[ELSE] = elsev;
780 VarDecl *IfStmt::getConditionVariable() const {
784 DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
785 return cast<VarDecl>(DS->getSingleDecl());
788 void IfStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
790 SubExprs[VAR] = nullptr;
794 SourceRange VarRange = V->getSourceRange();
795 SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
799 bool IfStmt::isObjCAvailabilityCheck() const {
800 return isa<ObjCAvailabilityCheckExpr>(SubExprs[COND]);
803 ForStmt::ForStmt(const ASTContext &C, Stmt *Init, Expr *Cond, VarDecl *condVar,
804 Expr *Inc, Stmt *Body, SourceLocation FL, SourceLocation LP,
806 : Stmt(ForStmtClass), ForLoc(FL), LParenLoc(LP), RParenLoc(RP)
808 SubExprs[INIT] = Init;
809 setConditionVariable(C, condVar);
810 SubExprs[COND] = Cond;
812 SubExprs[BODY] = Body;
815 VarDecl *ForStmt::getConditionVariable() const {
816 if (!SubExprs[CONDVAR])
819 DeclStmt *DS = cast<DeclStmt>(SubExprs[CONDVAR]);
820 return cast<VarDecl>(DS->getSingleDecl());
823 void ForStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
825 SubExprs[CONDVAR] = nullptr;
829 SourceRange VarRange = V->getSourceRange();
830 SubExprs[CONDVAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
834 SwitchStmt::SwitchStmt(const ASTContext &C, Stmt *init, VarDecl *Var,
836 : Stmt(SwitchStmtClass), FirstCase(nullptr, false) {
837 setConditionVariable(C, Var);
838 SubExprs[INIT] = init;
839 SubExprs[COND] = cond;
840 SubExprs[BODY] = nullptr;
843 VarDecl *SwitchStmt::getConditionVariable() const {
847 DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
848 return cast<VarDecl>(DS->getSingleDecl());
851 void SwitchStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
853 SubExprs[VAR] = nullptr;
857 SourceRange VarRange = V->getSourceRange();
858 SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
862 Stmt *SwitchCase::getSubStmt() {
863 if (isa<CaseStmt>(this))
864 return cast<CaseStmt>(this)->getSubStmt();
865 return cast<DefaultStmt>(this)->getSubStmt();
868 WhileStmt::WhileStmt(const ASTContext &C, VarDecl *Var, Expr *cond, Stmt *body,
870 : Stmt(WhileStmtClass) {
871 setConditionVariable(C, Var);
872 SubExprs[COND] = cond;
873 SubExprs[BODY] = body;
877 VarDecl *WhileStmt::getConditionVariable() const {
881 DeclStmt *DS = cast<DeclStmt>(SubExprs[VAR]);
882 return cast<VarDecl>(DS->getSingleDecl());
885 void WhileStmt::setConditionVariable(const ASTContext &C, VarDecl *V) {
887 SubExprs[VAR] = nullptr;
891 SourceRange VarRange = V->getSourceRange();
892 SubExprs[VAR] = new (C) DeclStmt(DeclGroupRef(V), VarRange.getBegin(),
897 LabelDecl *IndirectGotoStmt::getConstantTarget() {
898 if (AddrLabelExpr *E =
899 dyn_cast<AddrLabelExpr>(getTarget()->IgnoreParenImpCasts()))
900 return E->getLabel();
905 const Expr* ReturnStmt::getRetValue() const {
906 return cast_or_null<Expr>(RetExpr);
908 Expr* ReturnStmt::getRetValue() {
909 return cast_or_null<Expr>(RetExpr);
912 SEHTryStmt::SEHTryStmt(bool IsCXXTry,
913 SourceLocation TryLoc,
916 : Stmt(SEHTryStmtClass),
920 Children[TRY] = TryBlock;
921 Children[HANDLER] = Handler;
924 SEHTryStmt* SEHTryStmt::Create(const ASTContext &C, bool IsCXXTry,
925 SourceLocation TryLoc, Stmt *TryBlock,
927 return new(C) SEHTryStmt(IsCXXTry,TryLoc,TryBlock,Handler);
930 SEHExceptStmt* SEHTryStmt::getExceptHandler() const {
931 return dyn_cast<SEHExceptStmt>(getHandler());
934 SEHFinallyStmt* SEHTryStmt::getFinallyHandler() const {
935 return dyn_cast<SEHFinallyStmt>(getHandler());
938 SEHExceptStmt::SEHExceptStmt(SourceLocation Loc,
941 : Stmt(SEHExceptStmtClass),
944 Children[FILTER_EXPR] = FilterExpr;
945 Children[BLOCK] = Block;
948 SEHExceptStmt* SEHExceptStmt::Create(const ASTContext &C, SourceLocation Loc,
949 Expr *FilterExpr, Stmt *Block) {
950 return new(C) SEHExceptStmt(Loc,FilterExpr,Block);
953 SEHFinallyStmt::SEHFinallyStmt(SourceLocation Loc,
955 : Stmt(SEHFinallyStmtClass),
960 SEHFinallyStmt* SEHFinallyStmt::Create(const ASTContext &C, SourceLocation Loc,
962 return new(C)SEHFinallyStmt(Loc,Block);
965 CapturedStmt::Capture::Capture(SourceLocation Loc, VariableCaptureKind Kind,
967 : VarAndKind(Var, Kind), Loc(Loc) {
970 assert(!Var && "'this' capture cannot have a variable!");
973 assert(Var && "capturing by reference must have a variable!");
976 assert(Var && "capturing by copy must have a variable!");
978 (Var->getType()->isScalarType() || (Var->getType()->isReferenceType() &&
980 ->castAs<ReferenceType>()
982 ->isScalarType())) &&
983 "captures by copy are expected to have a scalar type!");
987 "Variable-length array type capture cannot have a variable!");
992 CapturedStmt::VariableCaptureKind
993 CapturedStmt::Capture::getCaptureKind() const {
994 return VarAndKind.getInt();
997 VarDecl *CapturedStmt::Capture::getCapturedVar() const {
998 assert((capturesVariable() || capturesVariableByCopy()) &&
999 "No variable available for 'this' or VAT capture");
1000 return VarAndKind.getPointer();
1003 CapturedStmt::Capture *CapturedStmt::getStoredCaptures() const {
1004 unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (NumCaptures + 1);
1006 // Offset of the first Capture object.
1007 unsigned FirstCaptureOffset = llvm::alignTo(Size, alignof(Capture));
1009 return reinterpret_cast<Capture *>(
1010 reinterpret_cast<char *>(const_cast<CapturedStmt *>(this))
1011 + FirstCaptureOffset);
1014 CapturedStmt::CapturedStmt(Stmt *S, CapturedRegionKind Kind,
1015 ArrayRef<Capture> Captures,
1016 ArrayRef<Expr *> CaptureInits,
1019 : Stmt(CapturedStmtClass), NumCaptures(Captures.size()),
1020 CapDeclAndKind(CD, Kind), TheRecordDecl(RD) {
1021 assert( S && "null captured statement");
1022 assert(CD && "null captured declaration for captured statement");
1023 assert(RD && "null record declaration for captured statement");
1025 // Copy initialization expressions.
1026 Stmt **Stored = getStoredStmts();
1027 for (unsigned I = 0, N = NumCaptures; I != N; ++I)
1028 *Stored++ = CaptureInits[I];
1030 // Copy the statement being captured.
1033 // Copy all Capture objects.
1034 Capture *Buffer = getStoredCaptures();
1035 std::copy(Captures.begin(), Captures.end(), Buffer);
1038 CapturedStmt::CapturedStmt(EmptyShell Empty, unsigned NumCaptures)
1039 : Stmt(CapturedStmtClass, Empty), NumCaptures(NumCaptures),
1040 CapDeclAndKind(nullptr, CR_Default), TheRecordDecl(nullptr) {
1041 getStoredStmts()[NumCaptures] = nullptr;
1044 CapturedStmt *CapturedStmt::Create(const ASTContext &Context, Stmt *S,
1045 CapturedRegionKind Kind,
1046 ArrayRef<Capture> Captures,
1047 ArrayRef<Expr *> CaptureInits,
1052 // -----------------------------------------------------------
1053 // | CapturedStmt, Init, ..., Init, S, Capture, ..., Capture |
1054 // ----------------^-------------------^----------------------
1055 // getStoredStmts() getStoredCaptures()
1057 // where S is the statement being captured.
1059 assert(CaptureInits.size() == Captures.size() && "wrong number of arguments");
1061 unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (Captures.size() + 1);
1062 if (!Captures.empty()) {
1063 // Realign for the following Capture array.
1064 Size = llvm::alignTo(Size, alignof(Capture));
1065 Size += sizeof(Capture) * Captures.size();
1068 void *Mem = Context.Allocate(Size);
1069 return new (Mem) CapturedStmt(S, Kind, Captures, CaptureInits, CD, RD);
1072 CapturedStmt *CapturedStmt::CreateDeserialized(const ASTContext &Context,
1073 unsigned NumCaptures) {
1074 unsigned Size = sizeof(CapturedStmt) + sizeof(Stmt *) * (NumCaptures + 1);
1075 if (NumCaptures > 0) {
1076 // Realign for the following Capture array.
1077 Size = llvm::alignTo(Size, alignof(Capture));
1078 Size += sizeof(Capture) * NumCaptures;
1081 void *Mem = Context.Allocate(Size);
1082 return new (Mem) CapturedStmt(EmptyShell(), NumCaptures);
1085 Stmt::child_range CapturedStmt::children() {
1086 // Children are captured field initializers.
1087 return child_range(getStoredStmts(), getStoredStmts() + NumCaptures);
1090 CapturedDecl *CapturedStmt::getCapturedDecl() {
1091 return CapDeclAndKind.getPointer();
1093 const CapturedDecl *CapturedStmt::getCapturedDecl() const {
1094 return CapDeclAndKind.getPointer();
1097 /// \brief Set the outlined function declaration.
1098 void CapturedStmt::setCapturedDecl(CapturedDecl *D) {
1099 assert(D && "null CapturedDecl");
1100 CapDeclAndKind.setPointer(D);
1103 /// \brief Retrieve the captured region kind.
1104 CapturedRegionKind CapturedStmt::getCapturedRegionKind() const {
1105 return CapDeclAndKind.getInt();
1108 /// \brief Set the captured region kind.
1109 void CapturedStmt::setCapturedRegionKind(CapturedRegionKind Kind) {
1110 CapDeclAndKind.setInt(Kind);
1113 bool CapturedStmt::capturesVariable(const VarDecl *Var) const {
1114 for (const auto &I : captures()) {
1115 if (!I.capturesVariable())
1118 // This does not handle variable redeclarations. This should be
1119 // extended to capture variables with redeclarations, for example
1120 // a thread-private variable in OpenMP.
1121 if (I.getCapturedVar() == Var)