1 //===--- ASTWriterStmt.cpp - Statement and Expression Serialization -------===//
2 //
3 // 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
6 //
7 //===----------------------------------------------------------------------===//
8 ///
9 /// \file
10 /// Implements serialization for Statements and Expressions.
11 ///
12 //===----------------------------------------------------------------------===//
13 
14 #include "clang/AST/ExprOpenMP.h"
15 #include "clang/Serialization/ASTRecordWriter.h"
16 #include "clang/Sema/DeclSpec.h"
17 #include "clang/AST/ASTContext.h"
18 #include "clang/AST/DeclCXX.h"
19 #include "clang/AST/DeclObjC.h"
20 #include "clang/AST/DeclTemplate.h"
21 #include "clang/AST/StmtVisitor.h"
22 #include "clang/Lex/Token.h"
23 #include "llvm/Bitstream/BitstreamWriter.h"
24 using namespace clang;
25 
26 //===----------------------------------------------------------------------===//
27 // Statement/expression serialization
28 //===----------------------------------------------------------------------===//
29 
30 namespace clang {
31 
32   class ASTStmtWriter : public StmtVisitor<ASTStmtWriter, void> {
33     ASTWriter &Writer;
34     ASTRecordWriter Record;
35 
36     serialization::StmtCode Code;
37     unsigned AbbrevToUse;
38 
39   public:
ASTStmtWriter(ASTWriter & Writer,ASTWriter::RecordData & Record)40     ASTStmtWriter(ASTWriter &Writer, ASTWriter::RecordData &Record)
41         : Writer(Writer), Record(Writer, Record),
42           Code(serialization::STMT_NULL_PTR), AbbrevToUse(0) {}
43 
44     ASTStmtWriter(const ASTStmtWriter&) = delete;
45 
Emit()46     uint64_t Emit() {
47       assert(Code != serialization::STMT_NULL_PTR &&
48              "unhandled sub-statement writing AST file");
49       return Record.EmitStmt(Code, AbbrevToUse);
50     }
51 
52     void AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo &ArgInfo,
53                                   const TemplateArgumentLoc *Args);
54 
55     void VisitStmt(Stmt *S);
56 #define STMT(Type, Base) \
57     void Visit##Type(Type *);
58 #include "clang/AST/StmtNodes.inc"
59   };
60 }
61 
AddTemplateKWAndArgsInfo(const ASTTemplateKWAndArgsInfo & ArgInfo,const TemplateArgumentLoc * Args)62 void ASTStmtWriter::AddTemplateKWAndArgsInfo(
63     const ASTTemplateKWAndArgsInfo &ArgInfo, const TemplateArgumentLoc *Args) {
64   Record.AddSourceLocation(ArgInfo.TemplateKWLoc);
65   Record.AddSourceLocation(ArgInfo.LAngleLoc);
66   Record.AddSourceLocation(ArgInfo.RAngleLoc);
67   for (unsigned i = 0; i != ArgInfo.NumTemplateArgs; ++i)
68     Record.AddTemplateArgumentLoc(Args[i]);
69 }
70 
VisitStmt(Stmt * S)71 void ASTStmtWriter::VisitStmt(Stmt *S) {
72 }
73 
VisitNullStmt(NullStmt * S)74 void ASTStmtWriter::VisitNullStmt(NullStmt *S) {
75   VisitStmt(S);
76   Record.AddSourceLocation(S->getSemiLoc());
77   Record.push_back(S->NullStmtBits.HasLeadingEmptyMacro);
78   Code = serialization::STMT_NULL;
79 }
80 
VisitCompoundStmt(CompoundStmt * S)81 void ASTStmtWriter::VisitCompoundStmt(CompoundStmt *S) {
82   VisitStmt(S);
83   Record.push_back(S->size());
84   for (auto *CS : S->body())
85     Record.AddStmt(CS);
86   Record.AddSourceLocation(S->getLBracLoc());
87   Record.AddSourceLocation(S->getRBracLoc());
88   Code = serialization::STMT_COMPOUND;
89 }
90 
VisitSwitchCase(SwitchCase * S)91 void ASTStmtWriter::VisitSwitchCase(SwitchCase *S) {
92   VisitStmt(S);
93   Record.push_back(Writer.getSwitchCaseID(S));
94   Record.AddSourceLocation(S->getKeywordLoc());
95   Record.AddSourceLocation(S->getColonLoc());
96 }
97 
VisitCaseStmt(CaseStmt * S)98 void ASTStmtWriter::VisitCaseStmt(CaseStmt *S) {
99   VisitSwitchCase(S);
100   Record.push_back(S->caseStmtIsGNURange());
101   Record.AddStmt(S->getLHS());
102   Record.AddStmt(S->getSubStmt());
103   if (S->caseStmtIsGNURange()) {
104     Record.AddStmt(S->getRHS());
105     Record.AddSourceLocation(S->getEllipsisLoc());
106   }
107   Code = serialization::STMT_CASE;
108 }
109 
VisitDefaultStmt(DefaultStmt * S)110 void ASTStmtWriter::VisitDefaultStmt(DefaultStmt *S) {
111   VisitSwitchCase(S);
112   Record.AddStmt(S->getSubStmt());
113   Code = serialization::STMT_DEFAULT;
114 }
115 
VisitLabelStmt(LabelStmt * S)116 void ASTStmtWriter::VisitLabelStmt(LabelStmt *S) {
117   VisitStmt(S);
118   Record.push_back(S->isSideEntry());
119   Record.AddDeclRef(S->getDecl());
120   Record.AddStmt(S->getSubStmt());
121   Record.AddSourceLocation(S->getIdentLoc());
122   Code = serialization::STMT_LABEL;
123 }
124 
VisitAttributedStmt(AttributedStmt * S)125 void ASTStmtWriter::VisitAttributedStmt(AttributedStmt *S) {
126   VisitStmt(S);
127   Record.push_back(S->getAttrs().size());
128   Record.AddAttributes(S->getAttrs());
129   Record.AddStmt(S->getSubStmt());
130   Record.AddSourceLocation(S->getAttrLoc());
131   Code = serialization::STMT_ATTRIBUTED;
132 }
133 
VisitIfStmt(IfStmt * S)134 void ASTStmtWriter::VisitIfStmt(IfStmt *S) {
135   VisitStmt(S);
136 
137   bool HasElse = S->getElse() != nullptr;
138   bool HasVar = S->getConditionVariableDeclStmt() != nullptr;
139   bool HasInit = S->getInit() != nullptr;
140 
141   Record.push_back(S->isConstexpr());
142   Record.push_back(HasElse);
143   Record.push_back(HasVar);
144   Record.push_back(HasInit);
145 
146   Record.AddStmt(S->getCond());
147   Record.AddStmt(S->getThen());
148   if (HasElse)
149     Record.AddStmt(S->getElse());
150   if (HasVar)
151     Record.AddDeclRef(S->getConditionVariable());
152   if (HasInit)
153     Record.AddStmt(S->getInit());
154 
155   Record.AddSourceLocation(S->getIfLoc());
156   Record.AddSourceLocation(S->getLParenLoc());
157   Record.AddSourceLocation(S->getRParenLoc());
158   if (HasElse)
159     Record.AddSourceLocation(S->getElseLoc());
160 
161   Code = serialization::STMT_IF;
162 }
163 
VisitSwitchStmt(SwitchStmt * S)164 void ASTStmtWriter::VisitSwitchStmt(SwitchStmt *S) {
165   VisitStmt(S);
166 
167   bool HasInit = S->getInit() != nullptr;
168   bool HasVar = S->getConditionVariableDeclStmt() != nullptr;
169   Record.push_back(HasInit);
170   Record.push_back(HasVar);
171   Record.push_back(S->isAllEnumCasesCovered());
172 
173   Record.AddStmt(S->getCond());
174   Record.AddStmt(S->getBody());
175   if (HasInit)
176     Record.AddStmt(S->getInit());
177   if (HasVar)
178     Record.AddDeclRef(S->getConditionVariable());
179 
180   Record.AddSourceLocation(S->getSwitchLoc());
181   Record.AddSourceLocation(S->getLParenLoc());
182   Record.AddSourceLocation(S->getRParenLoc());
183 
184   for (SwitchCase *SC = S->getSwitchCaseList(); SC;
185        SC = SC->getNextSwitchCase())
186     Record.push_back(Writer.RecordSwitchCaseID(SC));
187   Code = serialization::STMT_SWITCH;
188 }
189 
VisitWhileStmt(WhileStmt * S)190 void ASTStmtWriter::VisitWhileStmt(WhileStmt *S) {
191   VisitStmt(S);
192 
193   bool HasVar = S->getConditionVariableDeclStmt() != nullptr;
194   Record.push_back(HasVar);
195 
196   Record.AddStmt(S->getCond());
197   Record.AddStmt(S->getBody());
198   if (HasVar)
199     Record.AddDeclRef(S->getConditionVariable());
200 
201   Record.AddSourceLocation(S->getWhileLoc());
202   Record.AddSourceLocation(S->getLParenLoc());
203   Record.AddSourceLocation(S->getRParenLoc());
204   Code = serialization::STMT_WHILE;
205 }
206 
VisitDoStmt(DoStmt * S)207 void ASTStmtWriter::VisitDoStmt(DoStmt *S) {
208   VisitStmt(S);
209   Record.AddStmt(S->getCond());
210   Record.AddStmt(S->getBody());
211   Record.AddSourceLocation(S->getDoLoc());
212   Record.AddSourceLocation(S->getWhileLoc());
213   Record.AddSourceLocation(S->getRParenLoc());
214   Code = serialization::STMT_DO;
215 }
216 
VisitForStmt(ForStmt * S)217 void ASTStmtWriter::VisitForStmt(ForStmt *S) {
218   VisitStmt(S);
219   Record.AddStmt(S->getInit());
220   Record.AddStmt(S->getCond());
221   Record.AddDeclRef(S->getConditionVariable());
222   Record.AddStmt(S->getInc());
223   Record.AddStmt(S->getBody());
224   Record.AddSourceLocation(S->getForLoc());
225   Record.AddSourceLocation(S->getLParenLoc());
226   Record.AddSourceLocation(S->getRParenLoc());
227   Code = serialization::STMT_FOR;
228 }
229 
VisitGotoStmt(GotoStmt * S)230 void ASTStmtWriter::VisitGotoStmt(GotoStmt *S) {
231   VisitStmt(S);
232   Record.AddDeclRef(S->getLabel());
233   Record.AddSourceLocation(S->getGotoLoc());
234   Record.AddSourceLocation(S->getLabelLoc());
235   Code = serialization::STMT_GOTO;
236 }
237 
VisitIndirectGotoStmt(IndirectGotoStmt * S)238 void ASTStmtWriter::VisitIndirectGotoStmt(IndirectGotoStmt *S) {
239   VisitStmt(S);
240   Record.AddSourceLocation(S->getGotoLoc());
241   Record.AddSourceLocation(S->getStarLoc());
242   Record.AddStmt(S->getTarget());
243   Code = serialization::STMT_INDIRECT_GOTO;
244 }
245 
VisitContinueStmt(ContinueStmt * S)246 void ASTStmtWriter::VisitContinueStmt(ContinueStmt *S) {
247   VisitStmt(S);
248   Record.AddSourceLocation(S->getContinueLoc());
249   Code = serialization::STMT_CONTINUE;
250 }
251 
VisitBreakStmt(BreakStmt * S)252 void ASTStmtWriter::VisitBreakStmt(BreakStmt *S) {
253   VisitStmt(S);
254   Record.AddSourceLocation(S->getBreakLoc());
255   Code = serialization::STMT_BREAK;
256 }
257 
VisitReturnStmt(ReturnStmt * S)258 void ASTStmtWriter::VisitReturnStmt(ReturnStmt *S) {
259   VisitStmt(S);
260 
261   bool HasNRVOCandidate = S->getNRVOCandidate() != nullptr;
262   Record.push_back(HasNRVOCandidate);
263 
264   Record.AddStmt(S->getRetValue());
265   if (HasNRVOCandidate)
266     Record.AddDeclRef(S->getNRVOCandidate());
267 
268   Record.AddSourceLocation(S->getReturnLoc());
269   Code = serialization::STMT_RETURN;
270 }
271 
VisitDeclStmt(DeclStmt * S)272 void ASTStmtWriter::VisitDeclStmt(DeclStmt *S) {
273   VisitStmt(S);
274   Record.AddSourceLocation(S->getBeginLoc());
275   Record.AddSourceLocation(S->getEndLoc());
276   DeclGroupRef DG = S->getDeclGroup();
277   for (DeclGroupRef::iterator D = DG.begin(), DEnd = DG.end(); D != DEnd; ++D)
278     Record.AddDeclRef(*D);
279   Code = serialization::STMT_DECL;
280 }
281 
VisitAsmStmt(AsmStmt * S)282 void ASTStmtWriter::VisitAsmStmt(AsmStmt *S) {
283   VisitStmt(S);
284   Record.push_back(S->getNumOutputs());
285   Record.push_back(S->getNumInputs());
286   Record.push_back(S->getNumClobbers());
287   Record.AddSourceLocation(S->getAsmLoc());
288   Record.push_back(S->isVolatile());
289   Record.push_back(S->isSimple());
290 }
291 
VisitGCCAsmStmt(GCCAsmStmt * S)292 void ASTStmtWriter::VisitGCCAsmStmt(GCCAsmStmt *S) {
293   VisitAsmStmt(S);
294   Record.push_back(S->getNumLabels());
295   Record.AddSourceLocation(S->getRParenLoc());
296   Record.AddStmt(S->getAsmString());
297 
298   // Outputs
299   for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
300     Record.AddIdentifierRef(S->getOutputIdentifier(I));
301     Record.AddStmt(S->getOutputConstraintLiteral(I));
302     Record.AddStmt(S->getOutputExpr(I));
303   }
304 
305   // Inputs
306   for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
307     Record.AddIdentifierRef(S->getInputIdentifier(I));
308     Record.AddStmt(S->getInputConstraintLiteral(I));
309     Record.AddStmt(S->getInputExpr(I));
310   }
311 
312   // Clobbers
313   for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I)
314     Record.AddStmt(S->getClobberStringLiteral(I));
315 
316   // Labels
317   for (auto *E : S->labels()) Record.AddStmt(E);
318 
319   Code = serialization::STMT_GCCASM;
320 }
321 
VisitMSAsmStmt(MSAsmStmt * S)322 void ASTStmtWriter::VisitMSAsmStmt(MSAsmStmt *S) {
323   VisitAsmStmt(S);
324   Record.AddSourceLocation(S->getLBraceLoc());
325   Record.AddSourceLocation(S->getEndLoc());
326   Record.push_back(S->getNumAsmToks());
327   Record.AddString(S->getAsmString());
328 
329   // Tokens
330   for (unsigned I = 0, N = S->getNumAsmToks(); I != N; ++I) {
331     // FIXME: Move this to ASTRecordWriter?
332     Writer.AddToken(S->getAsmToks()[I], Record.getRecordData());
333   }
334 
335   // Clobbers
336   for (unsigned I = 0, N = S->getNumClobbers(); I != N; ++I) {
337     Record.AddString(S->getClobber(I));
338   }
339 
340   // Outputs
341   for (unsigned I = 0, N = S->getNumOutputs(); I != N; ++I) {
342     Record.AddStmt(S->getOutputExpr(I));
343     Record.AddString(S->getOutputConstraint(I));
344   }
345 
346   // Inputs
347   for (unsigned I = 0, N = S->getNumInputs(); I != N; ++I) {
348     Record.AddStmt(S->getInputExpr(I));
349     Record.AddString(S->getInputConstraint(I));
350   }
351 
352   Code = serialization::STMT_MSASM;
353 }
354 
VisitCoroutineBodyStmt(CoroutineBodyStmt * CoroStmt)355 void ASTStmtWriter::VisitCoroutineBodyStmt(CoroutineBodyStmt *CoroStmt) {
356   VisitStmt(CoroStmt);
357   Record.push_back(CoroStmt->getParamMoves().size());
358   for (Stmt *S : CoroStmt->children())
359     Record.AddStmt(S);
360   Code = serialization::STMT_COROUTINE_BODY;
361 }
362 
VisitCoreturnStmt(CoreturnStmt * S)363 void ASTStmtWriter::VisitCoreturnStmt(CoreturnStmt *S) {
364   VisitStmt(S);
365   Record.AddSourceLocation(S->getKeywordLoc());
366   Record.AddStmt(S->getOperand());
367   Record.AddStmt(S->getPromiseCall());
368   Record.push_back(S->isImplicit());
369   Code = serialization::STMT_CORETURN;
370 }
371 
VisitCoroutineSuspendExpr(CoroutineSuspendExpr * E)372 void ASTStmtWriter::VisitCoroutineSuspendExpr(CoroutineSuspendExpr *E) {
373   VisitExpr(E);
374   Record.AddSourceLocation(E->getKeywordLoc());
375   for (Stmt *S : E->children())
376     Record.AddStmt(S);
377   Record.AddStmt(E->getOpaqueValue());
378 }
379 
VisitCoawaitExpr(CoawaitExpr * E)380 void ASTStmtWriter::VisitCoawaitExpr(CoawaitExpr *E) {
381   VisitCoroutineSuspendExpr(E);
382   Record.push_back(E->isImplicit());
383   Code = serialization::EXPR_COAWAIT;
384 }
385 
VisitCoyieldExpr(CoyieldExpr * E)386 void ASTStmtWriter::VisitCoyieldExpr(CoyieldExpr *E) {
387   VisitCoroutineSuspendExpr(E);
388   Code = serialization::EXPR_COYIELD;
389 }
390 
VisitDependentCoawaitExpr(DependentCoawaitExpr * E)391 void ASTStmtWriter::VisitDependentCoawaitExpr(DependentCoawaitExpr *E) {
392   VisitExpr(E);
393   Record.AddSourceLocation(E->getKeywordLoc());
394   for (Stmt *S : E->children())
395     Record.AddStmt(S);
396   Code = serialization::EXPR_DEPENDENT_COAWAIT;
397 }
398 
399 static void
addConstraintSatisfaction(ASTRecordWriter & Record,const ASTConstraintSatisfaction & Satisfaction)400 addConstraintSatisfaction(ASTRecordWriter &Record,
401                           const ASTConstraintSatisfaction &Satisfaction) {
402   Record.push_back(Satisfaction.IsSatisfied);
403   if (!Satisfaction.IsSatisfied) {
404     Record.push_back(Satisfaction.NumRecords);
405     for (const auto &DetailRecord : Satisfaction) {
406       Record.AddStmt(const_cast<Expr *>(DetailRecord.first));
407       auto *E = DetailRecord.second.dyn_cast<Expr *>();
408       Record.push_back(E == nullptr);
409       if (E)
410         Record.AddStmt(E);
411       else {
412         auto *Diag = DetailRecord.second.get<std::pair<SourceLocation,
413                                                        StringRef> *>();
414         Record.AddSourceLocation(Diag->first);
415         Record.AddString(Diag->second);
416       }
417     }
418   }
419 }
420 
421 static void
addSubstitutionDiagnostic(ASTRecordWriter & Record,const concepts::Requirement::SubstitutionDiagnostic * D)422 addSubstitutionDiagnostic(
423     ASTRecordWriter &Record,
424     const concepts::Requirement::SubstitutionDiagnostic *D) {
425   Record.AddString(D->SubstitutedEntity);
426   Record.AddSourceLocation(D->DiagLoc);
427   Record.AddString(D->DiagMessage);
428 }
429 
VisitConceptSpecializationExpr(ConceptSpecializationExpr * E)430 void ASTStmtWriter::VisitConceptSpecializationExpr(
431         ConceptSpecializationExpr *E) {
432   VisitExpr(E);
433   ArrayRef<TemplateArgument> TemplateArgs = E->getTemplateArguments();
434   Record.push_back(TemplateArgs.size());
435   Record.AddNestedNameSpecifierLoc(E->getNestedNameSpecifierLoc());
436   Record.AddSourceLocation(E->getTemplateKWLoc());
437   Record.AddDeclarationNameInfo(E->getConceptNameInfo());
438   Record.AddDeclRef(E->getNamedConcept());
439   Record.AddDeclRef(E->getFoundDecl());
440   Record.AddASTTemplateArgumentListInfo(E->getTemplateArgsAsWritten());
441   for (const TemplateArgument &Arg : TemplateArgs)
442     Record.AddTemplateArgument(Arg);
443   if (!E->isValueDependent())
444     addConstraintSatisfaction(Record, E->getSatisfaction());
445 
446   Code = serialization::EXPR_CONCEPT_SPECIALIZATION;
447 }
448 
VisitRequiresExpr(RequiresExpr * E)449 void ASTStmtWriter::VisitRequiresExpr(RequiresExpr *E) {
450   VisitExpr(E);
451   Record.push_back(E->getLocalParameters().size());
452   Record.push_back(E->getRequirements().size());
453   Record.AddSourceLocation(E->RequiresExprBits.RequiresKWLoc);
454   Record.push_back(E->RequiresExprBits.IsSatisfied);
455   Record.AddDeclRef(E->getBody());
456   for (ParmVarDecl *P : E->getLocalParameters())
457     Record.AddDeclRef(P);
458   for (concepts::Requirement *R : E->getRequirements()) {
459     if (auto *TypeReq = dyn_cast<concepts::TypeRequirement>(R)) {
460       Record.push_back(concepts::Requirement::RK_Type);
461       Record.push_back(TypeReq->Status);
462       if (TypeReq->Status == concepts::TypeRequirement::SS_SubstitutionFailure)
463         addSubstitutionDiagnostic(Record, TypeReq->getSubstitutionDiagnostic());
464       else
465         Record.AddTypeSourceInfo(TypeReq->getType());
466     } else if (auto *ExprReq = dyn_cast<concepts::ExprRequirement>(R)) {
467       Record.push_back(ExprReq->getKind());
468       Record.push_back(ExprReq->Status);
469       if (ExprReq->isExprSubstitutionFailure()) {
470         addSubstitutionDiagnostic(Record,
471          ExprReq->Value.get<concepts::Requirement::SubstitutionDiagnostic *>());
472       } else
473         Record.AddStmt(ExprReq->Value.get<Expr *>());
474       if (ExprReq->getKind() == concepts::Requirement::RK_Compound) {
475         Record.AddSourceLocation(ExprReq->NoexceptLoc);
476         const auto &RetReq = ExprReq->getReturnTypeRequirement();
477         if (RetReq.isSubstitutionFailure()) {
478           Record.push_back(2);
479           addSubstitutionDiagnostic(Record, RetReq.getSubstitutionDiagnostic());
480         } else if (RetReq.isTypeConstraint()) {
481           Record.push_back(1);
482           Record.AddTemplateParameterList(
483               RetReq.getTypeConstraintTemplateParameterList());
484           if (ExprReq->Status >=
485               concepts::ExprRequirement::SS_ConstraintsNotSatisfied)
486             Record.AddStmt(
487                 ExprReq->getReturnTypeRequirementSubstitutedConstraintExpr());
488         } else {
489           assert(RetReq.isEmpty());
490           Record.push_back(0);
491         }
492       }
493     } else {
494       auto *NestedReq = cast<concepts::NestedRequirement>(R);
495       Record.push_back(concepts::Requirement::RK_Nested);
496       Record.push_back(NestedReq->isSubstitutionFailure());
497       if (NestedReq->isSubstitutionFailure()){
498         addSubstitutionDiagnostic(Record,
499                                   NestedReq->getSubstitutionDiagnostic());
500       } else {
501         Record.AddStmt(NestedReq->Value.get<Expr *>());
502         if (!NestedReq->isDependent())
503           addConstraintSatisfaction(Record, *NestedReq->Satisfaction);
504       }
505     }
506   }
507   Record.AddSourceLocation(E->getEndLoc());
508 
509   Code = serialization::EXPR_REQUIRES;
510 }
511 
512 
VisitCapturedStmt(CapturedStmt * S)513 void ASTStmtWriter::VisitCapturedStmt(CapturedStmt *S) {
514   VisitStmt(S);
515   // NumCaptures
516   Record.push_back(std::distance(S->capture_begin(), S->capture_end()));
517 
518   // CapturedDecl and captured region kind
519   Record.AddDeclRef(S->getCapturedDecl());
520   Record.push_back(S->getCapturedRegionKind());
521 
522   Record.AddDeclRef(S->getCapturedRecordDecl());
523 
524   // Capture inits
525   for (auto *I : S->capture_inits())
526     Record.AddStmt(I);
527 
528   // Body
529   Record.AddStmt(S->getCapturedStmt());
530 
531   // Captures
532   for (const auto &I : S->captures()) {
533     if (I.capturesThis() || I.capturesVariableArrayType())
534       Record.AddDeclRef(nullptr);
535     else
536       Record.AddDeclRef(I.getCapturedVar());
537     Record.push_back(I.getCaptureKind());
538     Record.AddSourceLocation(I.getLocation());
539   }
540 
541   Code = serialization::STMT_CAPTURED;
542 }
543 
VisitExpr(Expr * E)544 void ASTStmtWriter::VisitExpr(Expr *E) {
545   VisitStmt(E);
546   Record.AddTypeRef(E->getType());
547   Record.push_back(E->isTypeDependent());
548   Record.push_back(E->isValueDependent());
549   Record.push_back(E->isInstantiationDependent());
550   Record.push_back(E->containsUnexpandedParameterPack());
551   Record.push_back(E->containsErrors());
552   Record.push_back(E->getValueKind());
553   Record.push_back(E->getObjectKind());
554 }
555 
VisitConstantExpr(ConstantExpr * E)556 void ASTStmtWriter::VisitConstantExpr(ConstantExpr *E) {
557   VisitExpr(E);
558   Record.push_back(E->ConstantExprBits.ResultKind);
559 
560   Record.push_back(E->ConstantExprBits.APValueKind);
561   Record.push_back(E->ConstantExprBits.IsUnsigned);
562   Record.push_back(E->ConstantExprBits.BitWidth);
563   // HasCleanup not serialized since we can just query the APValue.
564   Record.push_back(E->ConstantExprBits.IsImmediateInvocation);
565 
566   switch (E->ConstantExprBits.ResultKind) {
567   case ConstantExpr::RSK_None:
568     break;
569   case ConstantExpr::RSK_Int64:
570     Record.push_back(E->Int64Result());
571     break;
572   case ConstantExpr::RSK_APValue:
573     Record.AddAPValue(E->APValueResult());
574     break;
575   default:
576     llvm_unreachable("unexpected ResultKind!");
577   }
578 
579   Record.AddStmt(E->getSubExpr());
580   Code = serialization::EXPR_CONSTANT;
581 }
582 
VisitPredefinedExpr(PredefinedExpr * E)583 void ASTStmtWriter::VisitPredefinedExpr(PredefinedExpr *E) {
584   VisitExpr(E);
585 
586   bool HasFunctionName = E->getFunctionName() != nullptr;
587   Record.push_back(HasFunctionName);
588   Record.push_back(E->getIdentKind()); // FIXME: stable encoding
589   Record.AddSourceLocation(E->getLocation());
590   if (HasFunctionName)
591     Record.AddStmt(E->getFunctionName());
592   Code = serialization::EXPR_PREDEFINED;
593 }
594 
VisitDeclRefExpr(DeclRefExpr * E)595 void ASTStmtWriter::VisitDeclRefExpr(DeclRefExpr *E) {
596   VisitExpr(E);
597 
598   Record.push_back(E->hasQualifier());
599   Record.push_back(E->getDecl() != E->getFoundDecl());
600   Record.push_back(E->hasTemplateKWAndArgsInfo());
601   Record.push_back(E->hadMultipleCandidates());
602   Record.push_back(E->refersToEnclosingVariableOrCapture());
603   Record.push_back(E->isNonOdrUse());
604 
605   if (E->hasTemplateKWAndArgsInfo()) {
606     unsigned NumTemplateArgs = E->getNumTemplateArgs();
607     Record.push_back(NumTemplateArgs);
608   }
609 
610   DeclarationName::NameKind nk = (E->getDecl()->getDeclName().getNameKind());
611 
612   if ((!E->hasTemplateKWAndArgsInfo()) && (!E->hasQualifier()) &&
613       (E->getDecl() == E->getFoundDecl()) &&
614       nk == DeclarationName::Identifier &&
615       !E->refersToEnclosingVariableOrCapture() && !E->isNonOdrUse()) {
616     AbbrevToUse = Writer.getDeclRefExprAbbrev();
617   }
618 
619   if (E->hasQualifier())
620     Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
621 
622   if (E->getDecl() != E->getFoundDecl())
623     Record.AddDeclRef(E->getFoundDecl());
624 
625   if (E->hasTemplateKWAndArgsInfo())
626     AddTemplateKWAndArgsInfo(*E->getTrailingObjects<ASTTemplateKWAndArgsInfo>(),
627                              E->getTrailingObjects<TemplateArgumentLoc>());
628 
629   Record.AddDeclRef(E->getDecl());
630   Record.AddSourceLocation(E->getLocation());
631   Record.AddDeclarationNameLoc(E->DNLoc, E->getDecl()->getDeclName());
632   Code = serialization::EXPR_DECL_REF;
633 }
634 
VisitIntegerLiteral(IntegerLiteral * E)635 void ASTStmtWriter::VisitIntegerLiteral(IntegerLiteral *E) {
636   VisitExpr(E);
637   Record.AddSourceLocation(E->getLocation());
638   Record.AddAPInt(E->getValue());
639 
640   if (E->getValue().getBitWidth() == 32) {
641     AbbrevToUse = Writer.getIntegerLiteralAbbrev();
642   }
643 
644   Code = serialization::EXPR_INTEGER_LITERAL;
645 }
646 
VisitFixedPointLiteral(FixedPointLiteral * E)647 void ASTStmtWriter::VisitFixedPointLiteral(FixedPointLiteral *E) {
648   VisitExpr(E);
649   Record.AddSourceLocation(E->getLocation());
650   Record.push_back(E->getScale());
651   Record.AddAPInt(E->getValue());
652   Code = serialization::EXPR_FIXEDPOINT_LITERAL;
653 }
654 
VisitFloatingLiteral(FloatingLiteral * E)655 void ASTStmtWriter::VisitFloatingLiteral(FloatingLiteral *E) {
656   VisitExpr(E);
657   Record.push_back(E->getRawSemantics());
658   Record.push_back(E->isExact());
659   Record.AddAPFloat(E->getValue());
660   Record.AddSourceLocation(E->getLocation());
661   Code = serialization::EXPR_FLOATING_LITERAL;
662 }
663 
VisitImaginaryLiteral(ImaginaryLiteral * E)664 void ASTStmtWriter::VisitImaginaryLiteral(ImaginaryLiteral *E) {
665   VisitExpr(E);
666   Record.AddStmt(E->getSubExpr());
667   Code = serialization::EXPR_IMAGINARY_LITERAL;
668 }
669 
VisitStringLiteral(StringLiteral * E)670 void ASTStmtWriter::VisitStringLiteral(StringLiteral *E) {
671   VisitExpr(E);
672 
673   // Store the various bits of data of StringLiteral.
674   Record.push_back(E->getNumConcatenated());
675   Record.push_back(E->getLength());
676   Record.push_back(E->getCharByteWidth());
677   Record.push_back(E->getKind());
678   Record.push_back(E->isPascal());
679 
680   // Store the trailing array of SourceLocation.
681   for (unsigned I = 0, N = E->getNumConcatenated(); I != N; ++I)
682     Record.AddSourceLocation(E->getStrTokenLoc(I));
683 
684   // Store the trailing array of char holding the string data.
685   StringRef StrData = E->getBytes();
686   for (unsigned I = 0, N = E->getByteLength(); I != N; ++I)
687     Record.push_back(StrData[I]);
688 
689   Code = serialization::EXPR_STRING_LITERAL;
690 }
691 
VisitCharacterLiteral(CharacterLiteral * E)692 void ASTStmtWriter::VisitCharacterLiteral(CharacterLiteral *E) {
693   VisitExpr(E);
694   Record.push_back(E->getValue());
695   Record.AddSourceLocation(E->getLocation());
696   Record.push_back(E->getKind());
697 
698   AbbrevToUse = Writer.getCharacterLiteralAbbrev();
699 
700   Code = serialization::EXPR_CHARACTER_LITERAL;
701 }
702 
VisitParenExpr(ParenExpr * E)703 void ASTStmtWriter::VisitParenExpr(ParenExpr *E) {
704   VisitExpr(E);
705   Record.AddSourceLocation(E->getLParen());
706   Record.AddSourceLocation(E->getRParen());
707   Record.AddStmt(E->getSubExpr());
708   Code = serialization::EXPR_PAREN;
709 }
710 
VisitParenListExpr(ParenListExpr * E)711 void ASTStmtWriter::VisitParenListExpr(ParenListExpr *E) {
712   VisitExpr(E);
713   Record.push_back(E->getNumExprs());
714   for (auto *SubStmt : E->exprs())
715     Record.AddStmt(SubStmt);
716   Record.AddSourceLocation(E->getLParenLoc());
717   Record.AddSourceLocation(E->getRParenLoc());
718   Code = serialization::EXPR_PAREN_LIST;
719 }
720 
VisitUnaryOperator(UnaryOperator * E)721 void ASTStmtWriter::VisitUnaryOperator(UnaryOperator *E) {
722   VisitExpr(E);
723   bool HasFPFeatures = E->hasStoredFPFeatures();
724   // Write this first for easy access when deserializing, as they affect the
725   // size of the UnaryOperator.
726   Record.push_back(HasFPFeatures);
727   Record.AddStmt(E->getSubExpr());
728   Record.push_back(E->getOpcode()); // FIXME: stable encoding
729   Record.AddSourceLocation(E->getOperatorLoc());
730   Record.push_back(E->canOverflow());
731   if (HasFPFeatures)
732     Record.push_back(E->getStoredFPFeatures().getAsOpaqueInt());
733   Code = serialization::EXPR_UNARY_OPERATOR;
734 }
735 
VisitOffsetOfExpr(OffsetOfExpr * E)736 void ASTStmtWriter::VisitOffsetOfExpr(OffsetOfExpr *E) {
737   VisitExpr(E);
738   Record.push_back(E->getNumComponents());
739   Record.push_back(E->getNumExpressions());
740   Record.AddSourceLocation(E->getOperatorLoc());
741   Record.AddSourceLocation(E->getRParenLoc());
742   Record.AddTypeSourceInfo(E->getTypeSourceInfo());
743   for (unsigned I = 0, N = E->getNumComponents(); I != N; ++I) {
744     const OffsetOfNode &ON = E->getComponent(I);
745     Record.push_back(ON.getKind()); // FIXME: Stable encoding
746     Record.AddSourceLocation(ON.getSourceRange().getBegin());
747     Record.AddSourceLocation(ON.getSourceRange().getEnd());
748     switch (ON.getKind()) {
749     case OffsetOfNode::Array:
750       Record.push_back(ON.getArrayExprIndex());
751       break;
752 
753     case OffsetOfNode::Field:
754       Record.AddDeclRef(ON.getField());
755       break;
756 
757     case OffsetOfNode::Identifier:
758       Record.AddIdentifierRef(ON.getFieldName());
759       break;
760 
761     case OffsetOfNode::Base:
762       Record.AddCXXBaseSpecifier(*ON.getBase());
763       break;
764     }
765   }
766   for (unsigned I = 0, N = E->getNumExpressions(); I != N; ++I)
767     Record.AddStmt(E->getIndexExpr(I));
768   Code = serialization::EXPR_OFFSETOF;
769 }
770 
VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr * E)771 void ASTStmtWriter::VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E) {
772   VisitExpr(E);
773   Record.push_back(E->getKind());
774   if (E->isArgumentType())
775     Record.AddTypeSourceInfo(E->getArgumentTypeInfo());
776   else {
777     Record.push_back(0);
778     Record.AddStmt(E->getArgumentExpr());
779   }
780   Record.AddSourceLocation(E->getOperatorLoc());
781   Record.AddSourceLocation(E->getRParenLoc());
782   Code = serialization::EXPR_SIZEOF_ALIGN_OF;
783 }
784 
VisitArraySubscriptExpr(ArraySubscriptExpr * E)785 void ASTStmtWriter::VisitArraySubscriptExpr(ArraySubscriptExpr *E) {
786   VisitExpr(E);
787   Record.AddStmt(E->getLHS());
788   Record.AddStmt(E->getRHS());
789   Record.AddSourceLocation(E->getRBracketLoc());
790   Code = serialization::EXPR_ARRAY_SUBSCRIPT;
791 }
792 
VisitMatrixSubscriptExpr(MatrixSubscriptExpr * E)793 void ASTStmtWriter::VisitMatrixSubscriptExpr(MatrixSubscriptExpr *E) {
794   VisitExpr(E);
795   Record.AddStmt(E->getBase());
796   Record.AddStmt(E->getRowIdx());
797   Record.AddStmt(E->getColumnIdx());
798   Record.AddSourceLocation(E->getRBracketLoc());
799   Code = serialization::EXPR_ARRAY_SUBSCRIPT;
800 }
801 
VisitOMPArraySectionExpr(OMPArraySectionExpr * E)802 void ASTStmtWriter::VisitOMPArraySectionExpr(OMPArraySectionExpr *E) {
803   VisitExpr(E);
804   Record.AddStmt(E->getBase());
805   Record.AddStmt(E->getLowerBound());
806   Record.AddStmt(E->getLength());
807   Record.AddStmt(E->getStride());
808   Record.AddSourceLocation(E->getColonLocFirst());
809   Record.AddSourceLocation(E->getColonLocSecond());
810   Record.AddSourceLocation(E->getRBracketLoc());
811   Code = serialization::EXPR_OMP_ARRAY_SECTION;
812 }
813 
VisitOMPArrayShapingExpr(OMPArrayShapingExpr * E)814 void ASTStmtWriter::VisitOMPArrayShapingExpr(OMPArrayShapingExpr *E) {
815   VisitExpr(E);
816   Record.push_back(E->getDimensions().size());
817   Record.AddStmt(E->getBase());
818   for (Expr *Dim : E->getDimensions())
819     Record.AddStmt(Dim);
820   for (SourceRange SR : E->getBracketsRanges())
821     Record.AddSourceRange(SR);
822   Record.AddSourceLocation(E->getLParenLoc());
823   Record.AddSourceLocation(E->getRParenLoc());
824   Code = serialization::EXPR_OMP_ARRAY_SHAPING;
825 }
826 
VisitOMPIteratorExpr(OMPIteratorExpr * E)827 void ASTStmtWriter::VisitOMPIteratorExpr(OMPIteratorExpr *E) {
828   VisitExpr(E);
829   Record.push_back(E->numOfIterators());
830   Record.AddSourceLocation(E->getIteratorKwLoc());
831   Record.AddSourceLocation(E->getLParenLoc());
832   Record.AddSourceLocation(E->getRParenLoc());
833   for (unsigned I = 0, End = E->numOfIterators(); I < End; ++I) {
834     Record.AddDeclRef(E->getIteratorDecl(I));
835     Record.AddSourceLocation(E->getAssignLoc(I));
836     OMPIteratorExpr::IteratorRange Range = E->getIteratorRange(I);
837     Record.AddStmt(Range.Begin);
838     Record.AddStmt(Range.End);
839     Record.AddStmt(Range.Step);
840     Record.AddSourceLocation(E->getColonLoc(I));
841     if (Range.Step)
842       Record.AddSourceLocation(E->getSecondColonLoc(I));
843     // Serialize helpers
844     OMPIteratorHelperData &HD = E->getHelper(I);
845     Record.AddDeclRef(HD.CounterVD);
846     Record.AddStmt(HD.Upper);
847     Record.AddStmt(HD.Update);
848     Record.AddStmt(HD.CounterUpdate);
849   }
850   Code = serialization::EXPR_OMP_ITERATOR;
851 }
852 
VisitCallExpr(CallExpr * E)853 void ASTStmtWriter::VisitCallExpr(CallExpr *E) {
854   VisitExpr(E);
855   Record.push_back(E->getNumArgs());
856   Record.push_back(E->hasStoredFPFeatures());
857   Record.AddSourceLocation(E->getRParenLoc());
858   Record.AddStmt(E->getCallee());
859   for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
860        Arg != ArgEnd; ++Arg)
861     Record.AddStmt(*Arg);
862   Record.push_back(static_cast<unsigned>(E->getADLCallKind()));
863   if (E->hasStoredFPFeatures())
864     Record.push_back(E->getFPFeatures().getAsOpaqueInt());
865   Code = serialization::EXPR_CALL;
866 }
867 
VisitRecoveryExpr(RecoveryExpr * E)868 void ASTStmtWriter::VisitRecoveryExpr(RecoveryExpr *E) {
869   VisitExpr(E);
870   Record.push_back(std::distance(E->children().begin(), E->children().end()));
871   Record.AddSourceLocation(E->getBeginLoc());
872   Record.AddSourceLocation(E->getEndLoc());
873   for (Stmt *Child : E->children())
874     Record.AddStmt(Child);
875   Code = serialization::EXPR_RECOVERY;
876 }
877 
VisitMemberExpr(MemberExpr * E)878 void ASTStmtWriter::VisitMemberExpr(MemberExpr *E) {
879   VisitExpr(E);
880 
881   bool HasQualifier = E->hasQualifier();
882   bool HasFoundDecl =
883       E->hasQualifierOrFoundDecl() &&
884       (E->getFoundDecl().getDecl() != E->getMemberDecl() ||
885        E->getFoundDecl().getAccess() != E->getMemberDecl()->getAccess());
886   bool HasTemplateInfo = E->hasTemplateKWAndArgsInfo();
887   unsigned NumTemplateArgs = E->getNumTemplateArgs();
888 
889   // Write these first for easy access when deserializing, as they affect the
890   // size of the MemberExpr.
891   Record.push_back(HasQualifier);
892   Record.push_back(HasFoundDecl);
893   Record.push_back(HasTemplateInfo);
894   Record.push_back(NumTemplateArgs);
895 
896   Record.AddStmt(E->getBase());
897   Record.AddDeclRef(E->getMemberDecl());
898   Record.AddDeclarationNameLoc(E->MemberDNLoc,
899                                E->getMemberDecl()->getDeclName());
900   Record.AddSourceLocation(E->getMemberLoc());
901   Record.push_back(E->isArrow());
902   Record.push_back(E->hadMultipleCandidates());
903   Record.push_back(E->isNonOdrUse());
904   Record.AddSourceLocation(E->getOperatorLoc());
905 
906   if (HasFoundDecl) {
907     DeclAccessPair FoundDecl = E->getFoundDecl();
908     Record.AddDeclRef(FoundDecl.getDecl());
909     Record.push_back(FoundDecl.getAccess());
910   }
911 
912   if (HasQualifier)
913     Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
914 
915   if (HasTemplateInfo)
916     AddTemplateKWAndArgsInfo(*E->getTrailingObjects<ASTTemplateKWAndArgsInfo>(),
917                              E->getTrailingObjects<TemplateArgumentLoc>());
918 
919   Code = serialization::EXPR_MEMBER;
920 }
921 
VisitObjCIsaExpr(ObjCIsaExpr * E)922 void ASTStmtWriter::VisitObjCIsaExpr(ObjCIsaExpr *E) {
923   VisitExpr(E);
924   Record.AddStmt(E->getBase());
925   Record.AddSourceLocation(E->getIsaMemberLoc());
926   Record.AddSourceLocation(E->getOpLoc());
927   Record.push_back(E->isArrow());
928   Code = serialization::EXPR_OBJC_ISA;
929 }
930 
931 void ASTStmtWriter::
VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr * E)932 VisitObjCIndirectCopyRestoreExpr(ObjCIndirectCopyRestoreExpr *E) {
933   VisitExpr(E);
934   Record.AddStmt(E->getSubExpr());
935   Record.push_back(E->shouldCopy());
936   Code = serialization::EXPR_OBJC_INDIRECT_COPY_RESTORE;
937 }
938 
VisitObjCBridgedCastExpr(ObjCBridgedCastExpr * E)939 void ASTStmtWriter::VisitObjCBridgedCastExpr(ObjCBridgedCastExpr *E) {
940   VisitExplicitCastExpr(E);
941   Record.AddSourceLocation(E->getLParenLoc());
942   Record.AddSourceLocation(E->getBridgeKeywordLoc());
943   Record.push_back(E->getBridgeKind()); // FIXME: Stable encoding
944   Code = serialization::EXPR_OBJC_BRIDGED_CAST;
945 }
946 
VisitCastExpr(CastExpr * E)947 void ASTStmtWriter::VisitCastExpr(CastExpr *E) {
948   VisitExpr(E);
949   Record.push_back(E->path_size());
950   Record.push_back(E->hasStoredFPFeatures());
951   Record.AddStmt(E->getSubExpr());
952   Record.push_back(E->getCastKind()); // FIXME: stable encoding
953 
954   for (CastExpr::path_iterator
955          PI = E->path_begin(), PE = E->path_end(); PI != PE; ++PI)
956     Record.AddCXXBaseSpecifier(**PI);
957 
958   if (E->hasStoredFPFeatures())
959     Record.push_back(E->getFPFeatures().getAsOpaqueInt());
960 }
961 
VisitBinaryOperator(BinaryOperator * E)962 void ASTStmtWriter::VisitBinaryOperator(BinaryOperator *E) {
963   VisitExpr(E);
964   bool HasFPFeatures = E->hasStoredFPFeatures();
965   // Write this first for easy access when deserializing, as they affect the
966   // size of the UnaryOperator.
967   Record.push_back(HasFPFeatures);
968   Record.push_back(E->getOpcode()); // FIXME: stable encoding
969   Record.AddStmt(E->getLHS());
970   Record.AddStmt(E->getRHS());
971   Record.AddSourceLocation(E->getOperatorLoc());
972   if (HasFPFeatures)
973     Record.push_back(E->getStoredFPFeatures().getAsOpaqueInt());
974   Code = serialization::EXPR_BINARY_OPERATOR;
975 }
976 
VisitCompoundAssignOperator(CompoundAssignOperator * E)977 void ASTStmtWriter::VisitCompoundAssignOperator(CompoundAssignOperator *E) {
978   VisitBinaryOperator(E);
979   Record.AddTypeRef(E->getComputationLHSType());
980   Record.AddTypeRef(E->getComputationResultType());
981   Code = serialization::EXPR_COMPOUND_ASSIGN_OPERATOR;
982 }
983 
VisitConditionalOperator(ConditionalOperator * E)984 void ASTStmtWriter::VisitConditionalOperator(ConditionalOperator *E) {
985   VisitExpr(E);
986   Record.AddStmt(E->getCond());
987   Record.AddStmt(E->getLHS());
988   Record.AddStmt(E->getRHS());
989   Record.AddSourceLocation(E->getQuestionLoc());
990   Record.AddSourceLocation(E->getColonLoc());
991   Code = serialization::EXPR_CONDITIONAL_OPERATOR;
992 }
993 
994 void
VisitBinaryConditionalOperator(BinaryConditionalOperator * E)995 ASTStmtWriter::VisitBinaryConditionalOperator(BinaryConditionalOperator *E) {
996   VisitExpr(E);
997   Record.AddStmt(E->getOpaqueValue());
998   Record.AddStmt(E->getCommon());
999   Record.AddStmt(E->getCond());
1000   Record.AddStmt(E->getTrueExpr());
1001   Record.AddStmt(E->getFalseExpr());
1002   Record.AddSourceLocation(E->getQuestionLoc());
1003   Record.AddSourceLocation(E->getColonLoc());
1004   Code = serialization::EXPR_BINARY_CONDITIONAL_OPERATOR;
1005 }
1006 
VisitImplicitCastExpr(ImplicitCastExpr * E)1007 void ASTStmtWriter::VisitImplicitCastExpr(ImplicitCastExpr *E) {
1008   VisitCastExpr(E);
1009   Record.push_back(E->isPartOfExplicitCast());
1010 
1011   if (E->path_size() == 0 && !E->hasStoredFPFeatures())
1012     AbbrevToUse = Writer.getExprImplicitCastAbbrev();
1013 
1014   Code = serialization::EXPR_IMPLICIT_CAST;
1015 }
1016 
VisitExplicitCastExpr(ExplicitCastExpr * E)1017 void ASTStmtWriter::VisitExplicitCastExpr(ExplicitCastExpr *E) {
1018   VisitCastExpr(E);
1019   Record.AddTypeSourceInfo(E->getTypeInfoAsWritten());
1020 }
1021 
VisitCStyleCastExpr(CStyleCastExpr * E)1022 void ASTStmtWriter::VisitCStyleCastExpr(CStyleCastExpr *E) {
1023   VisitExplicitCastExpr(E);
1024   Record.AddSourceLocation(E->getLParenLoc());
1025   Record.AddSourceLocation(E->getRParenLoc());
1026   Code = serialization::EXPR_CSTYLE_CAST;
1027 }
1028 
VisitCompoundLiteralExpr(CompoundLiteralExpr * E)1029 void ASTStmtWriter::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
1030   VisitExpr(E);
1031   Record.AddSourceLocation(E->getLParenLoc());
1032   Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1033   Record.AddStmt(E->getInitializer());
1034   Record.push_back(E->isFileScope());
1035   Code = serialization::EXPR_COMPOUND_LITERAL;
1036 }
1037 
VisitExtVectorElementExpr(ExtVectorElementExpr * E)1038 void ASTStmtWriter::VisitExtVectorElementExpr(ExtVectorElementExpr *E) {
1039   VisitExpr(E);
1040   Record.AddStmt(E->getBase());
1041   Record.AddIdentifierRef(&E->getAccessor());
1042   Record.AddSourceLocation(E->getAccessorLoc());
1043   Code = serialization::EXPR_EXT_VECTOR_ELEMENT;
1044 }
1045 
VisitInitListExpr(InitListExpr * E)1046 void ASTStmtWriter::VisitInitListExpr(InitListExpr *E) {
1047   VisitExpr(E);
1048   // NOTE: only add the (possibly null) syntactic form.
1049   // No need to serialize the isSemanticForm flag and the semantic form.
1050   Record.AddStmt(E->getSyntacticForm());
1051   Record.AddSourceLocation(E->getLBraceLoc());
1052   Record.AddSourceLocation(E->getRBraceLoc());
1053   bool isArrayFiller = E->ArrayFillerOrUnionFieldInit.is<Expr*>();
1054   Record.push_back(isArrayFiller);
1055   if (isArrayFiller)
1056     Record.AddStmt(E->getArrayFiller());
1057   else
1058     Record.AddDeclRef(E->getInitializedFieldInUnion());
1059   Record.push_back(E->hadArrayRangeDesignator());
1060   Record.push_back(E->getNumInits());
1061   if (isArrayFiller) {
1062     // ArrayFiller may have filled "holes" due to designated initializer.
1063     // Replace them by 0 to indicate that the filler goes in that place.
1064     Expr *filler = E->getArrayFiller();
1065     for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
1066       Record.AddStmt(E->getInit(I) != filler ? E->getInit(I) : nullptr);
1067   } else {
1068     for (unsigned I = 0, N = E->getNumInits(); I != N; ++I)
1069       Record.AddStmt(E->getInit(I));
1070   }
1071   Code = serialization::EXPR_INIT_LIST;
1072 }
1073 
VisitDesignatedInitExpr(DesignatedInitExpr * E)1074 void ASTStmtWriter::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
1075   VisitExpr(E);
1076   Record.push_back(E->getNumSubExprs());
1077   for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
1078     Record.AddStmt(E->getSubExpr(I));
1079   Record.AddSourceLocation(E->getEqualOrColonLoc());
1080   Record.push_back(E->usesGNUSyntax());
1081   for (const DesignatedInitExpr::Designator &D : E->designators()) {
1082     if (D.isFieldDesignator()) {
1083       if (FieldDecl *Field = D.getField()) {
1084         Record.push_back(serialization::DESIG_FIELD_DECL);
1085         Record.AddDeclRef(Field);
1086       } else {
1087         Record.push_back(serialization::DESIG_FIELD_NAME);
1088         Record.AddIdentifierRef(D.getFieldName());
1089       }
1090       Record.AddSourceLocation(D.getDotLoc());
1091       Record.AddSourceLocation(D.getFieldLoc());
1092     } else if (D.isArrayDesignator()) {
1093       Record.push_back(serialization::DESIG_ARRAY);
1094       Record.push_back(D.getFirstExprIndex());
1095       Record.AddSourceLocation(D.getLBracketLoc());
1096       Record.AddSourceLocation(D.getRBracketLoc());
1097     } else {
1098       assert(D.isArrayRangeDesignator() && "Unknown designator");
1099       Record.push_back(serialization::DESIG_ARRAY_RANGE);
1100       Record.push_back(D.getFirstExprIndex());
1101       Record.AddSourceLocation(D.getLBracketLoc());
1102       Record.AddSourceLocation(D.getEllipsisLoc());
1103       Record.AddSourceLocation(D.getRBracketLoc());
1104     }
1105   }
1106   Code = serialization::EXPR_DESIGNATED_INIT;
1107 }
1108 
VisitDesignatedInitUpdateExpr(DesignatedInitUpdateExpr * E)1109 void ASTStmtWriter::VisitDesignatedInitUpdateExpr(DesignatedInitUpdateExpr *E) {
1110   VisitExpr(E);
1111   Record.AddStmt(E->getBase());
1112   Record.AddStmt(E->getUpdater());
1113   Code = serialization::EXPR_DESIGNATED_INIT_UPDATE;
1114 }
1115 
VisitNoInitExpr(NoInitExpr * E)1116 void ASTStmtWriter::VisitNoInitExpr(NoInitExpr *E) {
1117   VisitExpr(E);
1118   Code = serialization::EXPR_NO_INIT;
1119 }
1120 
VisitArrayInitLoopExpr(ArrayInitLoopExpr * E)1121 void ASTStmtWriter::VisitArrayInitLoopExpr(ArrayInitLoopExpr *E) {
1122   VisitExpr(E);
1123   Record.AddStmt(E->SubExprs[0]);
1124   Record.AddStmt(E->SubExprs[1]);
1125   Code = serialization::EXPR_ARRAY_INIT_LOOP;
1126 }
1127 
VisitArrayInitIndexExpr(ArrayInitIndexExpr * E)1128 void ASTStmtWriter::VisitArrayInitIndexExpr(ArrayInitIndexExpr *E) {
1129   VisitExpr(E);
1130   Code = serialization::EXPR_ARRAY_INIT_INDEX;
1131 }
1132 
VisitImplicitValueInitExpr(ImplicitValueInitExpr * E)1133 void ASTStmtWriter::VisitImplicitValueInitExpr(ImplicitValueInitExpr *E) {
1134   VisitExpr(E);
1135   Code = serialization::EXPR_IMPLICIT_VALUE_INIT;
1136 }
1137 
VisitVAArgExpr(VAArgExpr * E)1138 void ASTStmtWriter::VisitVAArgExpr(VAArgExpr *E) {
1139   VisitExpr(E);
1140   Record.AddStmt(E->getSubExpr());
1141   Record.AddTypeSourceInfo(E->getWrittenTypeInfo());
1142   Record.AddSourceLocation(E->getBuiltinLoc());
1143   Record.AddSourceLocation(E->getRParenLoc());
1144   Record.push_back(E->isMicrosoftABI());
1145   Code = serialization::EXPR_VA_ARG;
1146 }
1147 
VisitSourceLocExpr(SourceLocExpr * E)1148 void ASTStmtWriter::VisitSourceLocExpr(SourceLocExpr *E) {
1149   VisitExpr(E);
1150   Record.AddDeclRef(cast_or_null<Decl>(E->getParentContext()));
1151   Record.AddSourceLocation(E->getBeginLoc());
1152   Record.AddSourceLocation(E->getEndLoc());
1153   Record.push_back(E->getIdentKind());
1154   Code = serialization::EXPR_SOURCE_LOC;
1155 }
1156 
VisitAddrLabelExpr(AddrLabelExpr * E)1157 void ASTStmtWriter::VisitAddrLabelExpr(AddrLabelExpr *E) {
1158   VisitExpr(E);
1159   Record.AddSourceLocation(E->getAmpAmpLoc());
1160   Record.AddSourceLocation(E->getLabelLoc());
1161   Record.AddDeclRef(E->getLabel());
1162   Code = serialization::EXPR_ADDR_LABEL;
1163 }
1164 
VisitStmtExpr(StmtExpr * E)1165 void ASTStmtWriter::VisitStmtExpr(StmtExpr *E) {
1166   VisitExpr(E);
1167   Record.AddStmt(E->getSubStmt());
1168   Record.AddSourceLocation(E->getLParenLoc());
1169   Record.AddSourceLocation(E->getRParenLoc());
1170   Record.push_back(E->getTemplateDepth());
1171   Code = serialization::EXPR_STMT;
1172 }
1173 
VisitChooseExpr(ChooseExpr * E)1174 void ASTStmtWriter::VisitChooseExpr(ChooseExpr *E) {
1175   VisitExpr(E);
1176   Record.AddStmt(E->getCond());
1177   Record.AddStmt(E->getLHS());
1178   Record.AddStmt(E->getRHS());
1179   Record.AddSourceLocation(E->getBuiltinLoc());
1180   Record.AddSourceLocation(E->getRParenLoc());
1181   Record.push_back(E->isConditionDependent() ? false : E->isConditionTrue());
1182   Code = serialization::EXPR_CHOOSE;
1183 }
1184 
VisitGNUNullExpr(GNUNullExpr * E)1185 void ASTStmtWriter::VisitGNUNullExpr(GNUNullExpr *E) {
1186   VisitExpr(E);
1187   Record.AddSourceLocation(E->getTokenLocation());
1188   Code = serialization::EXPR_GNU_NULL;
1189 }
1190 
VisitShuffleVectorExpr(ShuffleVectorExpr * E)1191 void ASTStmtWriter::VisitShuffleVectorExpr(ShuffleVectorExpr *E) {
1192   VisitExpr(E);
1193   Record.push_back(E->getNumSubExprs());
1194   for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
1195     Record.AddStmt(E->getExpr(I));
1196   Record.AddSourceLocation(E->getBuiltinLoc());
1197   Record.AddSourceLocation(E->getRParenLoc());
1198   Code = serialization::EXPR_SHUFFLE_VECTOR;
1199 }
1200 
VisitConvertVectorExpr(ConvertVectorExpr * E)1201 void ASTStmtWriter::VisitConvertVectorExpr(ConvertVectorExpr *E) {
1202   VisitExpr(E);
1203   Record.AddSourceLocation(E->getBuiltinLoc());
1204   Record.AddSourceLocation(E->getRParenLoc());
1205   Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1206   Record.AddStmt(E->getSrcExpr());
1207   Code = serialization::EXPR_CONVERT_VECTOR;
1208 }
1209 
VisitBlockExpr(BlockExpr * E)1210 void ASTStmtWriter::VisitBlockExpr(BlockExpr *E) {
1211   VisitExpr(E);
1212   Record.AddDeclRef(E->getBlockDecl());
1213   Code = serialization::EXPR_BLOCK;
1214 }
1215 
VisitGenericSelectionExpr(GenericSelectionExpr * E)1216 void ASTStmtWriter::VisitGenericSelectionExpr(GenericSelectionExpr *E) {
1217   VisitExpr(E);
1218 
1219   Record.push_back(E->getNumAssocs());
1220   Record.push_back(E->ResultIndex);
1221   Record.AddSourceLocation(E->getGenericLoc());
1222   Record.AddSourceLocation(E->getDefaultLoc());
1223   Record.AddSourceLocation(E->getRParenLoc());
1224 
1225   Stmt **Stmts = E->getTrailingObjects<Stmt *>();
1226   // Add 1 to account for the controlling expression which is the first
1227   // expression in the trailing array of Stmt *. This is not needed for
1228   // the trailing array of TypeSourceInfo *.
1229   for (unsigned I = 0, N = E->getNumAssocs() + 1; I < N; ++I)
1230     Record.AddStmt(Stmts[I]);
1231 
1232   TypeSourceInfo **TSIs = E->getTrailingObjects<TypeSourceInfo *>();
1233   for (unsigned I = 0, N = E->getNumAssocs(); I < N; ++I)
1234     Record.AddTypeSourceInfo(TSIs[I]);
1235 
1236   Code = serialization::EXPR_GENERIC_SELECTION;
1237 }
1238 
VisitPseudoObjectExpr(PseudoObjectExpr * E)1239 void ASTStmtWriter::VisitPseudoObjectExpr(PseudoObjectExpr *E) {
1240   VisitExpr(E);
1241   Record.push_back(E->getNumSemanticExprs());
1242 
1243   // Push the result index.  Currently, this needs to exactly match
1244   // the encoding used internally for ResultIndex.
1245   unsigned result = E->getResultExprIndex();
1246   result = (result == PseudoObjectExpr::NoResult ? 0 : result + 1);
1247   Record.push_back(result);
1248 
1249   Record.AddStmt(E->getSyntacticForm());
1250   for (PseudoObjectExpr::semantics_iterator
1251          i = E->semantics_begin(), e = E->semantics_end(); i != e; ++i) {
1252     Record.AddStmt(*i);
1253   }
1254   Code = serialization::EXPR_PSEUDO_OBJECT;
1255 }
1256 
VisitAtomicExpr(AtomicExpr * E)1257 void ASTStmtWriter::VisitAtomicExpr(AtomicExpr *E) {
1258   VisitExpr(E);
1259   Record.push_back(E->getOp());
1260   for (unsigned I = 0, N = E->getNumSubExprs(); I != N; ++I)
1261     Record.AddStmt(E->getSubExprs()[I]);
1262   Record.AddSourceLocation(E->getBuiltinLoc());
1263   Record.AddSourceLocation(E->getRParenLoc());
1264   Code = serialization::EXPR_ATOMIC;
1265 }
1266 
1267 //===----------------------------------------------------------------------===//
1268 // Objective-C Expressions and Statements.
1269 //===----------------------------------------------------------------------===//
1270 
VisitObjCStringLiteral(ObjCStringLiteral * E)1271 void ASTStmtWriter::VisitObjCStringLiteral(ObjCStringLiteral *E) {
1272   VisitExpr(E);
1273   Record.AddStmt(E->getString());
1274   Record.AddSourceLocation(E->getAtLoc());
1275   Code = serialization::EXPR_OBJC_STRING_LITERAL;
1276 }
1277 
VisitObjCBoxedExpr(ObjCBoxedExpr * E)1278 void ASTStmtWriter::VisitObjCBoxedExpr(ObjCBoxedExpr *E) {
1279   VisitExpr(E);
1280   Record.AddStmt(E->getSubExpr());
1281   Record.AddDeclRef(E->getBoxingMethod());
1282   Record.AddSourceRange(E->getSourceRange());
1283   Code = serialization::EXPR_OBJC_BOXED_EXPRESSION;
1284 }
1285 
VisitObjCArrayLiteral(ObjCArrayLiteral * E)1286 void ASTStmtWriter::VisitObjCArrayLiteral(ObjCArrayLiteral *E) {
1287   VisitExpr(E);
1288   Record.push_back(E->getNumElements());
1289   for (unsigned i = 0; i < E->getNumElements(); i++)
1290     Record.AddStmt(E->getElement(i));
1291   Record.AddDeclRef(E->getArrayWithObjectsMethod());
1292   Record.AddSourceRange(E->getSourceRange());
1293   Code = serialization::EXPR_OBJC_ARRAY_LITERAL;
1294 }
1295 
VisitObjCDictionaryLiteral(ObjCDictionaryLiteral * E)1296 void ASTStmtWriter::VisitObjCDictionaryLiteral(ObjCDictionaryLiteral *E) {
1297   VisitExpr(E);
1298   Record.push_back(E->getNumElements());
1299   Record.push_back(E->HasPackExpansions);
1300   for (unsigned i = 0; i < E->getNumElements(); i++) {
1301     ObjCDictionaryElement Element = E->getKeyValueElement(i);
1302     Record.AddStmt(Element.Key);
1303     Record.AddStmt(Element.Value);
1304     if (E->HasPackExpansions) {
1305       Record.AddSourceLocation(Element.EllipsisLoc);
1306       unsigned NumExpansions = 0;
1307       if (Element.NumExpansions)
1308         NumExpansions = *Element.NumExpansions + 1;
1309       Record.push_back(NumExpansions);
1310     }
1311   }
1312 
1313   Record.AddDeclRef(E->getDictWithObjectsMethod());
1314   Record.AddSourceRange(E->getSourceRange());
1315   Code = serialization::EXPR_OBJC_DICTIONARY_LITERAL;
1316 }
1317 
VisitObjCEncodeExpr(ObjCEncodeExpr * E)1318 void ASTStmtWriter::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
1319   VisitExpr(E);
1320   Record.AddTypeSourceInfo(E->getEncodedTypeSourceInfo());
1321   Record.AddSourceLocation(E->getAtLoc());
1322   Record.AddSourceLocation(E->getRParenLoc());
1323   Code = serialization::EXPR_OBJC_ENCODE;
1324 }
1325 
VisitObjCSelectorExpr(ObjCSelectorExpr * E)1326 void ASTStmtWriter::VisitObjCSelectorExpr(ObjCSelectorExpr *E) {
1327   VisitExpr(E);
1328   Record.AddSelectorRef(E->getSelector());
1329   Record.AddSourceLocation(E->getAtLoc());
1330   Record.AddSourceLocation(E->getRParenLoc());
1331   Code = serialization::EXPR_OBJC_SELECTOR_EXPR;
1332 }
1333 
VisitObjCProtocolExpr(ObjCProtocolExpr * E)1334 void ASTStmtWriter::VisitObjCProtocolExpr(ObjCProtocolExpr *E) {
1335   VisitExpr(E);
1336   Record.AddDeclRef(E->getProtocol());
1337   Record.AddSourceLocation(E->getAtLoc());
1338   Record.AddSourceLocation(E->ProtoLoc);
1339   Record.AddSourceLocation(E->getRParenLoc());
1340   Code = serialization::EXPR_OBJC_PROTOCOL_EXPR;
1341 }
1342 
VisitObjCIvarRefExpr(ObjCIvarRefExpr * E)1343 void ASTStmtWriter::VisitObjCIvarRefExpr(ObjCIvarRefExpr *E) {
1344   VisitExpr(E);
1345   Record.AddDeclRef(E->getDecl());
1346   Record.AddSourceLocation(E->getLocation());
1347   Record.AddSourceLocation(E->getOpLoc());
1348   Record.AddStmt(E->getBase());
1349   Record.push_back(E->isArrow());
1350   Record.push_back(E->isFreeIvar());
1351   Code = serialization::EXPR_OBJC_IVAR_REF_EXPR;
1352 }
1353 
VisitObjCPropertyRefExpr(ObjCPropertyRefExpr * E)1354 void ASTStmtWriter::VisitObjCPropertyRefExpr(ObjCPropertyRefExpr *E) {
1355   VisitExpr(E);
1356   Record.push_back(E->SetterAndMethodRefFlags.getInt());
1357   Record.push_back(E->isImplicitProperty());
1358   if (E->isImplicitProperty()) {
1359     Record.AddDeclRef(E->getImplicitPropertyGetter());
1360     Record.AddDeclRef(E->getImplicitPropertySetter());
1361   } else {
1362     Record.AddDeclRef(E->getExplicitProperty());
1363   }
1364   Record.AddSourceLocation(E->getLocation());
1365   Record.AddSourceLocation(E->getReceiverLocation());
1366   if (E->isObjectReceiver()) {
1367     Record.push_back(0);
1368     Record.AddStmt(E->getBase());
1369   } else if (E->isSuperReceiver()) {
1370     Record.push_back(1);
1371     Record.AddTypeRef(E->getSuperReceiverType());
1372   } else {
1373     Record.push_back(2);
1374     Record.AddDeclRef(E->getClassReceiver());
1375   }
1376 
1377   Code = serialization::EXPR_OBJC_PROPERTY_REF_EXPR;
1378 }
1379 
VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr * E)1380 void ASTStmtWriter::VisitObjCSubscriptRefExpr(ObjCSubscriptRefExpr *E) {
1381   VisitExpr(E);
1382   Record.AddSourceLocation(E->getRBracket());
1383   Record.AddStmt(E->getBaseExpr());
1384   Record.AddStmt(E->getKeyExpr());
1385   Record.AddDeclRef(E->getAtIndexMethodDecl());
1386   Record.AddDeclRef(E->setAtIndexMethodDecl());
1387 
1388   Code = serialization::EXPR_OBJC_SUBSCRIPT_REF_EXPR;
1389 }
1390 
VisitObjCMessageExpr(ObjCMessageExpr * E)1391 void ASTStmtWriter::VisitObjCMessageExpr(ObjCMessageExpr *E) {
1392   VisitExpr(E);
1393   Record.push_back(E->getNumArgs());
1394   Record.push_back(E->getNumStoredSelLocs());
1395   Record.push_back(E->SelLocsKind);
1396   Record.push_back(E->isDelegateInitCall());
1397   Record.push_back(E->IsImplicit);
1398   Record.push_back((unsigned)E->getReceiverKind()); // FIXME: stable encoding
1399   switch (E->getReceiverKind()) {
1400   case ObjCMessageExpr::Instance:
1401     Record.AddStmt(E->getInstanceReceiver());
1402     break;
1403 
1404   case ObjCMessageExpr::Class:
1405     Record.AddTypeSourceInfo(E->getClassReceiverTypeInfo());
1406     break;
1407 
1408   case ObjCMessageExpr::SuperClass:
1409   case ObjCMessageExpr::SuperInstance:
1410     Record.AddTypeRef(E->getSuperType());
1411     Record.AddSourceLocation(E->getSuperLoc());
1412     break;
1413   }
1414 
1415   if (E->getMethodDecl()) {
1416     Record.push_back(1);
1417     Record.AddDeclRef(E->getMethodDecl());
1418   } else {
1419     Record.push_back(0);
1420     Record.AddSelectorRef(E->getSelector());
1421   }
1422 
1423   Record.AddSourceLocation(E->getLeftLoc());
1424   Record.AddSourceLocation(E->getRightLoc());
1425 
1426   for (CallExpr::arg_iterator Arg = E->arg_begin(), ArgEnd = E->arg_end();
1427        Arg != ArgEnd; ++Arg)
1428     Record.AddStmt(*Arg);
1429 
1430   SourceLocation *Locs = E->getStoredSelLocs();
1431   for (unsigned i = 0, e = E->getNumStoredSelLocs(); i != e; ++i)
1432     Record.AddSourceLocation(Locs[i]);
1433 
1434   Code = serialization::EXPR_OBJC_MESSAGE_EXPR;
1435 }
1436 
VisitObjCForCollectionStmt(ObjCForCollectionStmt * S)1437 void ASTStmtWriter::VisitObjCForCollectionStmt(ObjCForCollectionStmt *S) {
1438   VisitStmt(S);
1439   Record.AddStmt(S->getElement());
1440   Record.AddStmt(S->getCollection());
1441   Record.AddStmt(S->getBody());
1442   Record.AddSourceLocation(S->getForLoc());
1443   Record.AddSourceLocation(S->getRParenLoc());
1444   Code = serialization::STMT_OBJC_FOR_COLLECTION;
1445 }
1446 
VisitObjCAtCatchStmt(ObjCAtCatchStmt * S)1447 void ASTStmtWriter::VisitObjCAtCatchStmt(ObjCAtCatchStmt *S) {
1448   VisitStmt(S);
1449   Record.AddStmt(S->getCatchBody());
1450   Record.AddDeclRef(S->getCatchParamDecl());
1451   Record.AddSourceLocation(S->getAtCatchLoc());
1452   Record.AddSourceLocation(S->getRParenLoc());
1453   Code = serialization::STMT_OBJC_CATCH;
1454 }
1455 
VisitObjCAtFinallyStmt(ObjCAtFinallyStmt * S)1456 void ASTStmtWriter::VisitObjCAtFinallyStmt(ObjCAtFinallyStmt *S) {
1457   VisitStmt(S);
1458   Record.AddStmt(S->getFinallyBody());
1459   Record.AddSourceLocation(S->getAtFinallyLoc());
1460   Code = serialization::STMT_OBJC_FINALLY;
1461 }
1462 
VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt * S)1463 void ASTStmtWriter::VisitObjCAutoreleasePoolStmt(ObjCAutoreleasePoolStmt *S) {
1464   VisitStmt(S); // FIXME: no test coverage.
1465   Record.AddStmt(S->getSubStmt());
1466   Record.AddSourceLocation(S->getAtLoc());
1467   Code = serialization::STMT_OBJC_AUTORELEASE_POOL;
1468 }
1469 
VisitObjCAtTryStmt(ObjCAtTryStmt * S)1470 void ASTStmtWriter::VisitObjCAtTryStmt(ObjCAtTryStmt *S) {
1471   VisitStmt(S);
1472   Record.push_back(S->getNumCatchStmts());
1473   Record.push_back(S->getFinallyStmt() != nullptr);
1474   Record.AddStmt(S->getTryBody());
1475   for (unsigned I = 0, N = S->getNumCatchStmts(); I != N; ++I)
1476     Record.AddStmt(S->getCatchStmt(I));
1477   if (S->getFinallyStmt())
1478     Record.AddStmt(S->getFinallyStmt());
1479   Record.AddSourceLocation(S->getAtTryLoc());
1480   Code = serialization::STMT_OBJC_AT_TRY;
1481 }
1482 
VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt * S)1483 void ASTStmtWriter::VisitObjCAtSynchronizedStmt(ObjCAtSynchronizedStmt *S) {
1484   VisitStmt(S); // FIXME: no test coverage.
1485   Record.AddStmt(S->getSynchExpr());
1486   Record.AddStmt(S->getSynchBody());
1487   Record.AddSourceLocation(S->getAtSynchronizedLoc());
1488   Code = serialization::STMT_OBJC_AT_SYNCHRONIZED;
1489 }
1490 
VisitObjCAtThrowStmt(ObjCAtThrowStmt * S)1491 void ASTStmtWriter::VisitObjCAtThrowStmt(ObjCAtThrowStmt *S) {
1492   VisitStmt(S); // FIXME: no test coverage.
1493   Record.AddStmt(S->getThrowExpr());
1494   Record.AddSourceLocation(S->getThrowLoc());
1495   Code = serialization::STMT_OBJC_AT_THROW;
1496 }
1497 
VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr * E)1498 void ASTStmtWriter::VisitObjCBoolLiteralExpr(ObjCBoolLiteralExpr *E) {
1499   VisitExpr(E);
1500   Record.push_back(E->getValue());
1501   Record.AddSourceLocation(E->getLocation());
1502   Code = serialization::EXPR_OBJC_BOOL_LITERAL;
1503 }
1504 
VisitObjCAvailabilityCheckExpr(ObjCAvailabilityCheckExpr * E)1505 void ASTStmtWriter::VisitObjCAvailabilityCheckExpr(ObjCAvailabilityCheckExpr *E) {
1506   VisitExpr(E);
1507   Record.AddSourceRange(E->getSourceRange());
1508   Record.AddVersionTuple(E->getVersion());
1509   Code = serialization::EXPR_OBJC_AVAILABILITY_CHECK;
1510 }
1511 
1512 //===----------------------------------------------------------------------===//
1513 // C++ Expressions and Statements.
1514 //===----------------------------------------------------------------------===//
1515 
VisitCXXCatchStmt(CXXCatchStmt * S)1516 void ASTStmtWriter::VisitCXXCatchStmt(CXXCatchStmt *S) {
1517   VisitStmt(S);
1518   Record.AddSourceLocation(S->getCatchLoc());
1519   Record.AddDeclRef(S->getExceptionDecl());
1520   Record.AddStmt(S->getHandlerBlock());
1521   Code = serialization::STMT_CXX_CATCH;
1522 }
1523 
VisitCXXTryStmt(CXXTryStmt * S)1524 void ASTStmtWriter::VisitCXXTryStmt(CXXTryStmt *S) {
1525   VisitStmt(S);
1526   Record.push_back(S->getNumHandlers());
1527   Record.AddSourceLocation(S->getTryLoc());
1528   Record.AddStmt(S->getTryBlock());
1529   for (unsigned i = 0, e = S->getNumHandlers(); i != e; ++i)
1530     Record.AddStmt(S->getHandler(i));
1531   Code = serialization::STMT_CXX_TRY;
1532 }
1533 
VisitCXXForRangeStmt(CXXForRangeStmt * S)1534 void ASTStmtWriter::VisitCXXForRangeStmt(CXXForRangeStmt *S) {
1535   VisitStmt(S);
1536   Record.AddSourceLocation(S->getForLoc());
1537   Record.AddSourceLocation(S->getCoawaitLoc());
1538   Record.AddSourceLocation(S->getColonLoc());
1539   Record.AddSourceLocation(S->getRParenLoc());
1540   Record.AddStmt(S->getInit());
1541   Record.AddStmt(S->getRangeStmt());
1542   Record.AddStmt(S->getBeginStmt());
1543   Record.AddStmt(S->getEndStmt());
1544   Record.AddStmt(S->getCond());
1545   Record.AddStmt(S->getInc());
1546   Record.AddStmt(S->getLoopVarStmt());
1547   Record.AddStmt(S->getBody());
1548   Code = serialization::STMT_CXX_FOR_RANGE;
1549 }
1550 
VisitMSDependentExistsStmt(MSDependentExistsStmt * S)1551 void ASTStmtWriter::VisitMSDependentExistsStmt(MSDependentExistsStmt *S) {
1552   VisitStmt(S);
1553   Record.AddSourceLocation(S->getKeywordLoc());
1554   Record.push_back(S->isIfExists());
1555   Record.AddNestedNameSpecifierLoc(S->getQualifierLoc());
1556   Record.AddDeclarationNameInfo(S->getNameInfo());
1557   Record.AddStmt(S->getSubStmt());
1558   Code = serialization::STMT_MS_DEPENDENT_EXISTS;
1559 }
1560 
VisitCXXOperatorCallExpr(CXXOperatorCallExpr * E)1561 void ASTStmtWriter::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E) {
1562   VisitCallExpr(E);
1563   Record.push_back(E->getOperator());
1564   Record.AddSourceRange(E->Range);
1565   Code = serialization::EXPR_CXX_OPERATOR_CALL;
1566 }
1567 
VisitCXXMemberCallExpr(CXXMemberCallExpr * E)1568 void ASTStmtWriter::VisitCXXMemberCallExpr(CXXMemberCallExpr *E) {
1569   VisitCallExpr(E);
1570   Code = serialization::EXPR_CXX_MEMBER_CALL;
1571 }
1572 
VisitCXXRewrittenBinaryOperator(CXXRewrittenBinaryOperator * E)1573 void ASTStmtWriter::VisitCXXRewrittenBinaryOperator(
1574     CXXRewrittenBinaryOperator *E) {
1575   VisitExpr(E);
1576   Record.push_back(E->isReversed());
1577   Record.AddStmt(E->getSemanticForm());
1578   Code = serialization::EXPR_CXX_REWRITTEN_BINARY_OPERATOR;
1579 }
1580 
VisitCXXConstructExpr(CXXConstructExpr * E)1581 void ASTStmtWriter::VisitCXXConstructExpr(CXXConstructExpr *E) {
1582   VisitExpr(E);
1583 
1584   Record.push_back(E->getNumArgs());
1585   Record.push_back(E->isElidable());
1586   Record.push_back(E->hadMultipleCandidates());
1587   Record.push_back(E->isListInitialization());
1588   Record.push_back(E->isStdInitListInitialization());
1589   Record.push_back(E->requiresZeroInitialization());
1590   Record.push_back(E->getConstructionKind()); // FIXME: stable encoding
1591   Record.AddSourceLocation(E->getLocation());
1592   Record.AddDeclRef(E->getConstructor());
1593   Record.AddSourceRange(E->getParenOrBraceRange());
1594 
1595   for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
1596     Record.AddStmt(E->getArg(I));
1597 
1598   Code = serialization::EXPR_CXX_CONSTRUCT;
1599 }
1600 
VisitCXXInheritedCtorInitExpr(CXXInheritedCtorInitExpr * E)1601 void ASTStmtWriter::VisitCXXInheritedCtorInitExpr(CXXInheritedCtorInitExpr *E) {
1602   VisitExpr(E);
1603   Record.AddDeclRef(E->getConstructor());
1604   Record.AddSourceLocation(E->getLocation());
1605   Record.push_back(E->constructsVBase());
1606   Record.push_back(E->inheritedFromVBase());
1607   Code = serialization::EXPR_CXX_INHERITED_CTOR_INIT;
1608 }
1609 
VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr * E)1610 void ASTStmtWriter::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
1611   VisitCXXConstructExpr(E);
1612   Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1613   Code = serialization::EXPR_CXX_TEMPORARY_OBJECT;
1614 }
1615 
VisitLambdaExpr(LambdaExpr * E)1616 void ASTStmtWriter::VisitLambdaExpr(LambdaExpr *E) {
1617   VisitExpr(E);
1618   Record.push_back(E->LambdaExprBits.NumCaptures);
1619   Record.AddSourceRange(E->IntroducerRange);
1620   Record.push_back(E->LambdaExprBits.CaptureDefault); // FIXME: stable encoding
1621   Record.AddSourceLocation(E->CaptureDefaultLoc);
1622   Record.push_back(E->LambdaExprBits.ExplicitParams);
1623   Record.push_back(E->LambdaExprBits.ExplicitResultType);
1624   Record.AddSourceLocation(E->ClosingBrace);
1625 
1626   // Add capture initializers.
1627   for (LambdaExpr::capture_init_iterator C = E->capture_init_begin(),
1628                                       CEnd = E->capture_init_end();
1629        C != CEnd; ++C) {
1630     Record.AddStmt(*C);
1631   }
1632 
1633   // Don't serialize the body. It belongs to the call operator declaration.
1634   // LambdaExpr only stores a copy of the Stmt *.
1635 
1636   Code = serialization::EXPR_LAMBDA;
1637 }
1638 
VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr * E)1639 void ASTStmtWriter::VisitCXXStdInitializerListExpr(CXXStdInitializerListExpr *E) {
1640   VisitExpr(E);
1641   Record.AddStmt(E->getSubExpr());
1642   Code = serialization::EXPR_CXX_STD_INITIALIZER_LIST;
1643 }
1644 
VisitCXXNamedCastExpr(CXXNamedCastExpr * E)1645 void ASTStmtWriter::VisitCXXNamedCastExpr(CXXNamedCastExpr *E) {
1646   VisitExplicitCastExpr(E);
1647   Record.AddSourceRange(SourceRange(E->getOperatorLoc(), E->getRParenLoc()));
1648   Record.AddSourceRange(E->getAngleBrackets());
1649 }
1650 
VisitCXXStaticCastExpr(CXXStaticCastExpr * E)1651 void ASTStmtWriter::VisitCXXStaticCastExpr(CXXStaticCastExpr *E) {
1652   VisitCXXNamedCastExpr(E);
1653   Code = serialization::EXPR_CXX_STATIC_CAST;
1654 }
1655 
VisitCXXDynamicCastExpr(CXXDynamicCastExpr * E)1656 void ASTStmtWriter::VisitCXXDynamicCastExpr(CXXDynamicCastExpr *E) {
1657   VisitCXXNamedCastExpr(E);
1658   Code = serialization::EXPR_CXX_DYNAMIC_CAST;
1659 }
1660 
VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr * E)1661 void ASTStmtWriter::VisitCXXReinterpretCastExpr(CXXReinterpretCastExpr *E) {
1662   VisitCXXNamedCastExpr(E);
1663   Code = serialization::EXPR_CXX_REINTERPRET_CAST;
1664 }
1665 
VisitCXXConstCastExpr(CXXConstCastExpr * E)1666 void ASTStmtWriter::VisitCXXConstCastExpr(CXXConstCastExpr *E) {
1667   VisitCXXNamedCastExpr(E);
1668   Code = serialization::EXPR_CXX_CONST_CAST;
1669 }
1670 
VisitCXXAddrspaceCastExpr(CXXAddrspaceCastExpr * E)1671 void ASTStmtWriter::VisitCXXAddrspaceCastExpr(CXXAddrspaceCastExpr *E) {
1672   VisitCXXNamedCastExpr(E);
1673   Code = serialization::EXPR_CXX_ADDRSPACE_CAST;
1674 }
1675 
VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr * E)1676 void ASTStmtWriter::VisitCXXFunctionalCastExpr(CXXFunctionalCastExpr *E) {
1677   VisitExplicitCastExpr(E);
1678   Record.AddSourceLocation(E->getLParenLoc());
1679   Record.AddSourceLocation(E->getRParenLoc());
1680   Code = serialization::EXPR_CXX_FUNCTIONAL_CAST;
1681 }
1682 
VisitBuiltinBitCastExpr(BuiltinBitCastExpr * E)1683 void ASTStmtWriter::VisitBuiltinBitCastExpr(BuiltinBitCastExpr *E) {
1684   VisitExplicitCastExpr(E);
1685   Record.AddSourceLocation(E->getBeginLoc());
1686   Record.AddSourceLocation(E->getEndLoc());
1687   Code = serialization::EXPR_BUILTIN_BIT_CAST;
1688 }
1689 
VisitUserDefinedLiteral(UserDefinedLiteral * E)1690 void ASTStmtWriter::VisitUserDefinedLiteral(UserDefinedLiteral *E) {
1691   VisitCallExpr(E);
1692   Record.AddSourceLocation(E->UDSuffixLoc);
1693   Code = serialization::EXPR_USER_DEFINED_LITERAL;
1694 }
1695 
VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr * E)1696 void ASTStmtWriter::VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *E) {
1697   VisitExpr(E);
1698   Record.push_back(E->getValue());
1699   Record.AddSourceLocation(E->getLocation());
1700   Code = serialization::EXPR_CXX_BOOL_LITERAL;
1701 }
1702 
VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr * E)1703 void ASTStmtWriter::VisitCXXNullPtrLiteralExpr(CXXNullPtrLiteralExpr *E) {
1704   VisitExpr(E);
1705   Record.AddSourceLocation(E->getLocation());
1706   Code = serialization::EXPR_CXX_NULL_PTR_LITERAL;
1707 }
1708 
VisitCXXTypeidExpr(CXXTypeidExpr * E)1709 void ASTStmtWriter::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1710   VisitExpr(E);
1711   Record.AddSourceRange(E->getSourceRange());
1712   if (E->isTypeOperand()) {
1713     Record.AddTypeSourceInfo(E->getTypeOperandSourceInfo());
1714     Code = serialization::EXPR_CXX_TYPEID_TYPE;
1715   } else {
1716     Record.AddStmt(E->getExprOperand());
1717     Code = serialization::EXPR_CXX_TYPEID_EXPR;
1718   }
1719 }
1720 
VisitCXXThisExpr(CXXThisExpr * E)1721 void ASTStmtWriter::VisitCXXThisExpr(CXXThisExpr *E) {
1722   VisitExpr(E);
1723   Record.AddSourceLocation(E->getLocation());
1724   Record.push_back(E->isImplicit());
1725   Code = serialization::EXPR_CXX_THIS;
1726 }
1727 
VisitCXXThrowExpr(CXXThrowExpr * E)1728 void ASTStmtWriter::VisitCXXThrowExpr(CXXThrowExpr *E) {
1729   VisitExpr(E);
1730   Record.AddSourceLocation(E->getThrowLoc());
1731   Record.AddStmt(E->getSubExpr());
1732   Record.push_back(E->isThrownVariableInScope());
1733   Code = serialization::EXPR_CXX_THROW;
1734 }
1735 
VisitCXXDefaultArgExpr(CXXDefaultArgExpr * E)1736 void ASTStmtWriter::VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) {
1737   VisitExpr(E);
1738   Record.AddDeclRef(E->getParam());
1739   Record.AddDeclRef(cast_or_null<Decl>(E->getUsedContext()));
1740   Record.AddSourceLocation(E->getUsedLocation());
1741   Code = serialization::EXPR_CXX_DEFAULT_ARG;
1742 }
1743 
VisitCXXDefaultInitExpr(CXXDefaultInitExpr * E)1744 void ASTStmtWriter::VisitCXXDefaultInitExpr(CXXDefaultInitExpr *E) {
1745   VisitExpr(E);
1746   Record.AddDeclRef(E->getField());
1747   Record.AddDeclRef(cast_or_null<Decl>(E->getUsedContext()));
1748   Record.AddSourceLocation(E->getExprLoc());
1749   Code = serialization::EXPR_CXX_DEFAULT_INIT;
1750 }
1751 
VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr * E)1752 void ASTStmtWriter::VisitCXXBindTemporaryExpr(CXXBindTemporaryExpr *E) {
1753   VisitExpr(E);
1754   Record.AddCXXTemporary(E->getTemporary());
1755   Record.AddStmt(E->getSubExpr());
1756   Code = serialization::EXPR_CXX_BIND_TEMPORARY;
1757 }
1758 
VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr * E)1759 void ASTStmtWriter::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
1760   VisitExpr(E);
1761   Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1762   Record.AddSourceLocation(E->getRParenLoc());
1763   Code = serialization::EXPR_CXX_SCALAR_VALUE_INIT;
1764 }
1765 
VisitCXXNewExpr(CXXNewExpr * E)1766 void ASTStmtWriter::VisitCXXNewExpr(CXXNewExpr *E) {
1767   VisitExpr(E);
1768 
1769   Record.push_back(E->isArray());
1770   Record.push_back(E->hasInitializer());
1771   Record.push_back(E->getNumPlacementArgs());
1772   Record.push_back(E->isParenTypeId());
1773 
1774   Record.push_back(E->isGlobalNew());
1775   Record.push_back(E->passAlignment());
1776   Record.push_back(E->doesUsualArrayDeleteWantSize());
1777   Record.push_back(E->CXXNewExprBits.StoredInitializationStyle);
1778 
1779   Record.AddDeclRef(E->getOperatorNew());
1780   Record.AddDeclRef(E->getOperatorDelete());
1781   Record.AddTypeSourceInfo(E->getAllocatedTypeSourceInfo());
1782   if (E->isParenTypeId())
1783     Record.AddSourceRange(E->getTypeIdParens());
1784   Record.AddSourceRange(E->getSourceRange());
1785   Record.AddSourceRange(E->getDirectInitRange());
1786 
1787   for (CXXNewExpr::arg_iterator I = E->raw_arg_begin(), N = E->raw_arg_end();
1788        I != N; ++I)
1789     Record.AddStmt(*I);
1790 
1791   Code = serialization::EXPR_CXX_NEW;
1792 }
1793 
VisitCXXDeleteExpr(CXXDeleteExpr * E)1794 void ASTStmtWriter::VisitCXXDeleteExpr(CXXDeleteExpr *E) {
1795   VisitExpr(E);
1796   Record.push_back(E->isGlobalDelete());
1797   Record.push_back(E->isArrayForm());
1798   Record.push_back(E->isArrayFormAsWritten());
1799   Record.push_back(E->doesUsualArrayDeleteWantSize());
1800   Record.AddDeclRef(E->getOperatorDelete());
1801   Record.AddStmt(E->getArgument());
1802   Record.AddSourceLocation(E->getBeginLoc());
1803 
1804   Code = serialization::EXPR_CXX_DELETE;
1805 }
1806 
VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr * E)1807 void ASTStmtWriter::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1808   VisitExpr(E);
1809 
1810   Record.AddStmt(E->getBase());
1811   Record.push_back(E->isArrow());
1812   Record.AddSourceLocation(E->getOperatorLoc());
1813   Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1814   Record.AddTypeSourceInfo(E->getScopeTypeInfo());
1815   Record.AddSourceLocation(E->getColonColonLoc());
1816   Record.AddSourceLocation(E->getTildeLoc());
1817 
1818   // PseudoDestructorTypeStorage.
1819   Record.AddIdentifierRef(E->getDestroyedTypeIdentifier());
1820   if (E->getDestroyedTypeIdentifier())
1821     Record.AddSourceLocation(E->getDestroyedTypeLoc());
1822   else
1823     Record.AddTypeSourceInfo(E->getDestroyedTypeInfo());
1824 
1825   Code = serialization::EXPR_CXX_PSEUDO_DESTRUCTOR;
1826 }
1827 
VisitExprWithCleanups(ExprWithCleanups * E)1828 void ASTStmtWriter::VisitExprWithCleanups(ExprWithCleanups *E) {
1829   VisitExpr(E);
1830   Record.push_back(E->getNumObjects());
1831   for (auto &Obj : E->getObjects()) {
1832     if (auto *BD = Obj.dyn_cast<BlockDecl *>()) {
1833       Record.push_back(serialization::COK_Block);
1834       Record.AddDeclRef(BD);
1835     } else if (auto *CLE = Obj.dyn_cast<CompoundLiteralExpr *>()) {
1836       Record.push_back(serialization::COK_CompoundLiteral);
1837       Record.AddStmt(CLE);
1838     }
1839   }
1840 
1841   Record.push_back(E->cleanupsHaveSideEffects());
1842   Record.AddStmt(E->getSubExpr());
1843   Code = serialization::EXPR_EXPR_WITH_CLEANUPS;
1844 }
1845 
VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr * E)1846 void ASTStmtWriter::VisitCXXDependentScopeMemberExpr(
1847     CXXDependentScopeMemberExpr *E) {
1848   VisitExpr(E);
1849 
1850   // Don't emit anything here (or if you do you will have to update
1851   // the corresponding deserialization function).
1852 
1853   Record.push_back(E->hasTemplateKWAndArgsInfo());
1854   Record.push_back(E->getNumTemplateArgs());
1855   Record.push_back(E->hasFirstQualifierFoundInScope());
1856 
1857   if (E->hasTemplateKWAndArgsInfo()) {
1858     const ASTTemplateKWAndArgsInfo &ArgInfo =
1859         *E->getTrailingObjects<ASTTemplateKWAndArgsInfo>();
1860     AddTemplateKWAndArgsInfo(ArgInfo,
1861                              E->getTrailingObjects<TemplateArgumentLoc>());
1862   }
1863 
1864   Record.push_back(E->isArrow());
1865   Record.AddSourceLocation(E->getOperatorLoc());
1866   Record.AddTypeRef(E->getBaseType());
1867   Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1868   if (!E->isImplicitAccess())
1869     Record.AddStmt(E->getBase());
1870   else
1871     Record.AddStmt(nullptr);
1872 
1873   if (E->hasFirstQualifierFoundInScope())
1874     Record.AddDeclRef(E->getFirstQualifierFoundInScope());
1875 
1876   Record.AddDeclarationNameInfo(E->MemberNameInfo);
1877   Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_MEMBER;
1878 }
1879 
1880 void
VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr * E)1881 ASTStmtWriter::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1882   VisitExpr(E);
1883 
1884   // Don't emit anything here, HasTemplateKWAndArgsInfo must be
1885   // emitted first.
1886 
1887   Record.push_back(E->DependentScopeDeclRefExprBits.HasTemplateKWAndArgsInfo);
1888   if (E->DependentScopeDeclRefExprBits.HasTemplateKWAndArgsInfo) {
1889     const ASTTemplateKWAndArgsInfo &ArgInfo =
1890         *E->getTrailingObjects<ASTTemplateKWAndArgsInfo>();
1891     Record.push_back(ArgInfo.NumTemplateArgs);
1892     AddTemplateKWAndArgsInfo(ArgInfo,
1893                              E->getTrailingObjects<TemplateArgumentLoc>());
1894   }
1895 
1896   Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1897   Record.AddDeclarationNameInfo(E->NameInfo);
1898   Code = serialization::EXPR_CXX_DEPENDENT_SCOPE_DECL_REF;
1899 }
1900 
1901 void
VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr * E)1902 ASTStmtWriter::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E) {
1903   VisitExpr(E);
1904   Record.push_back(E->getNumArgs());
1905   for (CXXUnresolvedConstructExpr::arg_iterator
1906          ArgI = E->arg_begin(), ArgE = E->arg_end(); ArgI != ArgE; ++ArgI)
1907     Record.AddStmt(*ArgI);
1908   Record.AddTypeSourceInfo(E->getTypeSourceInfo());
1909   Record.AddSourceLocation(E->getLParenLoc());
1910   Record.AddSourceLocation(E->getRParenLoc());
1911   Code = serialization::EXPR_CXX_UNRESOLVED_CONSTRUCT;
1912 }
1913 
VisitOverloadExpr(OverloadExpr * E)1914 void ASTStmtWriter::VisitOverloadExpr(OverloadExpr *E) {
1915   VisitExpr(E);
1916 
1917   Record.push_back(E->getNumDecls());
1918   Record.push_back(E->hasTemplateKWAndArgsInfo());
1919   if (E->hasTemplateKWAndArgsInfo()) {
1920     const ASTTemplateKWAndArgsInfo &ArgInfo =
1921         *E->getTrailingASTTemplateKWAndArgsInfo();
1922     Record.push_back(ArgInfo.NumTemplateArgs);
1923     AddTemplateKWAndArgsInfo(ArgInfo, E->getTrailingTemplateArgumentLoc());
1924   }
1925 
1926   for (OverloadExpr::decls_iterator OvI = E->decls_begin(),
1927                                     OvE = E->decls_end();
1928        OvI != OvE; ++OvI) {
1929     Record.AddDeclRef(OvI.getDecl());
1930     Record.push_back(OvI.getAccess());
1931   }
1932 
1933   Record.AddDeclarationNameInfo(E->getNameInfo());
1934   Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
1935 }
1936 
VisitUnresolvedMemberExpr(UnresolvedMemberExpr * E)1937 void ASTStmtWriter::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *E) {
1938   VisitOverloadExpr(E);
1939   Record.push_back(E->isArrow());
1940   Record.push_back(E->hasUnresolvedUsing());
1941   Record.AddStmt(!E->isImplicitAccess() ? E->getBase() : nullptr);
1942   Record.AddTypeRef(E->getBaseType());
1943   Record.AddSourceLocation(E->getOperatorLoc());
1944   Code = serialization::EXPR_CXX_UNRESOLVED_MEMBER;
1945 }
1946 
VisitUnresolvedLookupExpr(UnresolvedLookupExpr * E)1947 void ASTStmtWriter::VisitUnresolvedLookupExpr(UnresolvedLookupExpr *E) {
1948   VisitOverloadExpr(E);
1949   Record.push_back(E->requiresADL());
1950   Record.push_back(E->isOverloaded());
1951   Record.AddDeclRef(E->getNamingClass());
1952   Code = serialization::EXPR_CXX_UNRESOLVED_LOOKUP;
1953 }
1954 
VisitTypeTraitExpr(TypeTraitExpr * E)1955 void ASTStmtWriter::VisitTypeTraitExpr(TypeTraitExpr *E) {
1956   VisitExpr(E);
1957   Record.push_back(E->TypeTraitExprBits.NumArgs);
1958   Record.push_back(E->TypeTraitExprBits.Kind); // FIXME: Stable encoding
1959   Record.push_back(E->TypeTraitExprBits.Value);
1960   Record.AddSourceRange(E->getSourceRange());
1961   for (unsigned I = 0, N = E->getNumArgs(); I != N; ++I)
1962     Record.AddTypeSourceInfo(E->getArg(I));
1963   Code = serialization::EXPR_TYPE_TRAIT;
1964 }
1965 
VisitArrayTypeTraitExpr(ArrayTypeTraitExpr * E)1966 void ASTStmtWriter::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
1967   VisitExpr(E);
1968   Record.push_back(E->getTrait());
1969   Record.push_back(E->getValue());
1970   Record.AddSourceRange(E->getSourceRange());
1971   Record.AddTypeSourceInfo(E->getQueriedTypeSourceInfo());
1972   Record.AddStmt(E->getDimensionExpression());
1973   Code = serialization::EXPR_ARRAY_TYPE_TRAIT;
1974 }
1975 
VisitExpressionTraitExpr(ExpressionTraitExpr * E)1976 void ASTStmtWriter::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
1977   VisitExpr(E);
1978   Record.push_back(E->getTrait());
1979   Record.push_back(E->getValue());
1980   Record.AddSourceRange(E->getSourceRange());
1981   Record.AddStmt(E->getQueriedExpression());
1982   Code = serialization::EXPR_CXX_EXPRESSION_TRAIT;
1983 }
1984 
VisitCXXNoexceptExpr(CXXNoexceptExpr * E)1985 void ASTStmtWriter::VisitCXXNoexceptExpr(CXXNoexceptExpr *E) {
1986   VisitExpr(E);
1987   Record.push_back(E->getValue());
1988   Record.AddSourceRange(E->getSourceRange());
1989   Record.AddStmt(E->getOperand());
1990   Code = serialization::EXPR_CXX_NOEXCEPT;
1991 }
1992 
VisitPackExpansionExpr(PackExpansionExpr * E)1993 void ASTStmtWriter::VisitPackExpansionExpr(PackExpansionExpr *E) {
1994   VisitExpr(E);
1995   Record.AddSourceLocation(E->getEllipsisLoc());
1996   Record.push_back(E->NumExpansions);
1997   Record.AddStmt(E->getPattern());
1998   Code = serialization::EXPR_PACK_EXPANSION;
1999 }
2000 
VisitSizeOfPackExpr(SizeOfPackExpr * E)2001 void ASTStmtWriter::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
2002   VisitExpr(E);
2003   Record.push_back(E->isPartiallySubstituted() ? E->getPartialArguments().size()
2004                                                : 0);
2005   Record.AddSourceLocation(E->OperatorLoc);
2006   Record.AddSourceLocation(E->PackLoc);
2007   Record.AddSourceLocation(E->RParenLoc);
2008   Record.AddDeclRef(E->Pack);
2009   if (E->isPartiallySubstituted()) {
2010     for (const auto &TA : E->getPartialArguments())
2011       Record.AddTemplateArgument(TA);
2012   } else if (!E->isValueDependent()) {
2013     Record.push_back(E->getPackLength());
2014   }
2015   Code = serialization::EXPR_SIZEOF_PACK;
2016 }
2017 
VisitSubstNonTypeTemplateParmExpr(SubstNonTypeTemplateParmExpr * E)2018 void ASTStmtWriter::VisitSubstNonTypeTemplateParmExpr(
2019                                               SubstNonTypeTemplateParmExpr *E) {
2020   VisitExpr(E);
2021   Record.AddDeclRef(E->getParameter());
2022   Record.push_back(E->isReferenceParameter());
2023   Record.AddSourceLocation(E->getNameLoc());
2024   Record.AddStmt(E->getReplacement());
2025   Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM;
2026 }
2027 
VisitSubstNonTypeTemplateParmPackExpr(SubstNonTypeTemplateParmPackExpr * E)2028 void ASTStmtWriter::VisitSubstNonTypeTemplateParmPackExpr(
2029                                           SubstNonTypeTemplateParmPackExpr *E) {
2030   VisitExpr(E);
2031   Record.AddDeclRef(E->getParameterPack());
2032   Record.AddTemplateArgument(E->getArgumentPack());
2033   Record.AddSourceLocation(E->getParameterPackLocation());
2034   Code = serialization::EXPR_SUBST_NON_TYPE_TEMPLATE_PARM_PACK;
2035 }
2036 
VisitFunctionParmPackExpr(FunctionParmPackExpr * E)2037 void ASTStmtWriter::VisitFunctionParmPackExpr(FunctionParmPackExpr *E) {
2038   VisitExpr(E);
2039   Record.push_back(E->getNumExpansions());
2040   Record.AddDeclRef(E->getParameterPack());
2041   Record.AddSourceLocation(E->getParameterPackLocation());
2042   for (FunctionParmPackExpr::iterator I = E->begin(), End = E->end();
2043        I != End; ++I)
2044     Record.AddDeclRef(*I);
2045   Code = serialization::EXPR_FUNCTION_PARM_PACK;
2046 }
2047 
VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr * E)2048 void ASTStmtWriter::VisitMaterializeTemporaryExpr(MaterializeTemporaryExpr *E) {
2049   VisitExpr(E);
2050   Record.push_back(static_cast<bool>(E->getLifetimeExtendedTemporaryDecl()));
2051   if (E->getLifetimeExtendedTemporaryDecl())
2052     Record.AddDeclRef(E->getLifetimeExtendedTemporaryDecl());
2053   else
2054     Record.AddStmt(E->getSubExpr());
2055   Code = serialization::EXPR_MATERIALIZE_TEMPORARY;
2056 }
2057 
VisitCXXFoldExpr(CXXFoldExpr * E)2058 void ASTStmtWriter::VisitCXXFoldExpr(CXXFoldExpr *E) {
2059   VisitExpr(E);
2060   Record.AddSourceLocation(E->LParenLoc);
2061   Record.AddSourceLocation(E->EllipsisLoc);
2062   Record.AddSourceLocation(E->RParenLoc);
2063   Record.push_back(E->NumExpansions);
2064   Record.AddStmt(E->SubExprs[0]);
2065   Record.AddStmt(E->SubExprs[1]);
2066   Record.AddStmt(E->SubExprs[2]);
2067   Record.push_back(E->Opcode);
2068   Code = serialization::EXPR_CXX_FOLD;
2069 }
2070 
VisitOpaqueValueExpr(OpaqueValueExpr * E)2071 void ASTStmtWriter::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
2072   VisitExpr(E);
2073   Record.AddStmt(E->getSourceExpr());
2074   Record.AddSourceLocation(E->getLocation());
2075   Record.push_back(E->isUnique());
2076   Code = serialization::EXPR_OPAQUE_VALUE;
2077 }
2078 
VisitTypoExpr(TypoExpr * E)2079 void ASTStmtWriter::VisitTypoExpr(TypoExpr *E) {
2080   VisitExpr(E);
2081   // TODO: Figure out sane writer behavior for a TypoExpr, if necessary
2082   llvm_unreachable("Cannot write TypoExpr nodes");
2083 }
2084 
2085 //===----------------------------------------------------------------------===//
2086 // CUDA Expressions and Statements.
2087 //===----------------------------------------------------------------------===//
2088 
VisitCUDAKernelCallExpr(CUDAKernelCallExpr * E)2089 void ASTStmtWriter::VisitCUDAKernelCallExpr(CUDAKernelCallExpr *E) {
2090   VisitCallExpr(E);
2091   Record.AddStmt(E->getConfig());
2092   Code = serialization::EXPR_CUDA_KERNEL_CALL;
2093 }
2094 
2095 //===----------------------------------------------------------------------===//
2096 // OpenCL Expressions and Statements.
2097 //===----------------------------------------------------------------------===//
VisitAsTypeExpr(AsTypeExpr * E)2098 void ASTStmtWriter::VisitAsTypeExpr(AsTypeExpr *E) {
2099   VisitExpr(E);
2100   Record.AddSourceLocation(E->getBuiltinLoc());
2101   Record.AddSourceLocation(E->getRParenLoc());
2102   Record.AddStmt(E->getSrcExpr());
2103   Code = serialization::EXPR_ASTYPE;
2104 }
2105 
2106 //===----------------------------------------------------------------------===//
2107 // Microsoft Expressions and Statements.
2108 //===----------------------------------------------------------------------===//
VisitMSPropertyRefExpr(MSPropertyRefExpr * E)2109 void ASTStmtWriter::VisitMSPropertyRefExpr(MSPropertyRefExpr *E) {
2110   VisitExpr(E);
2111   Record.push_back(E->isArrow());
2112   Record.AddStmt(E->getBaseExpr());
2113   Record.AddNestedNameSpecifierLoc(E->getQualifierLoc());
2114   Record.AddSourceLocation(E->getMemberLoc());
2115   Record.AddDeclRef(E->getPropertyDecl());
2116   Code = serialization::EXPR_CXX_PROPERTY_REF_EXPR;
2117 }
2118 
VisitMSPropertySubscriptExpr(MSPropertySubscriptExpr * E)2119 void ASTStmtWriter::VisitMSPropertySubscriptExpr(MSPropertySubscriptExpr *E) {
2120   VisitExpr(E);
2121   Record.AddStmt(E->getBase());
2122   Record.AddStmt(E->getIdx());
2123   Record.AddSourceLocation(E->getRBracketLoc());
2124   Code = serialization::EXPR_CXX_PROPERTY_SUBSCRIPT_EXPR;
2125 }
2126 
VisitCXXUuidofExpr(CXXUuidofExpr * E)2127 void ASTStmtWriter::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
2128   VisitExpr(E);
2129   Record.AddSourceRange(E->getSourceRange());
2130   Record.AddDeclRef(E->getGuidDecl());
2131   if (E->isTypeOperand()) {
2132     Record.AddTypeSourceInfo(E->getTypeOperandSourceInfo());
2133     Code = serialization::EXPR_CXX_UUIDOF_TYPE;
2134   } else {
2135     Record.AddStmt(E->getExprOperand());
2136     Code = serialization::EXPR_CXX_UUIDOF_EXPR;
2137   }
2138 }
2139 
VisitSEHExceptStmt(SEHExceptStmt * S)2140 void ASTStmtWriter::VisitSEHExceptStmt(SEHExceptStmt *S) {
2141   VisitStmt(S);
2142   Record.AddSourceLocation(S->getExceptLoc());
2143   Record.AddStmt(S->getFilterExpr());
2144   Record.AddStmt(S->getBlock());
2145   Code = serialization::STMT_SEH_EXCEPT;
2146 }
2147 
VisitSEHFinallyStmt(SEHFinallyStmt * S)2148 void ASTStmtWriter::VisitSEHFinallyStmt(SEHFinallyStmt *S) {
2149   VisitStmt(S);
2150   Record.AddSourceLocation(S->getFinallyLoc());
2151   Record.AddStmt(S->getBlock());
2152   Code = serialization::STMT_SEH_FINALLY;
2153 }
2154 
VisitSEHTryStmt(SEHTryStmt * S)2155 void ASTStmtWriter::VisitSEHTryStmt(SEHTryStmt *S) {
2156   VisitStmt(S);
2157   Record.push_back(S->getIsCXXTry());
2158   Record.AddSourceLocation(S->getTryLoc());
2159   Record.AddStmt(S->getTryBlock());
2160   Record.AddStmt(S->getHandler());
2161   Code = serialization::STMT_SEH_TRY;
2162 }
2163 
VisitSEHLeaveStmt(SEHLeaveStmt * S)2164 void ASTStmtWriter::VisitSEHLeaveStmt(SEHLeaveStmt *S) {
2165   VisitStmt(S);
2166   Record.AddSourceLocation(S->getLeaveLoc());
2167   Code = serialization::STMT_SEH_LEAVE;
2168 }
2169 
2170 //===----------------------------------------------------------------------===//
2171 // OpenMP Directives.
2172 //===----------------------------------------------------------------------===//
2173 
VisitOMPCanonicalLoop(OMPCanonicalLoop * S)2174 void ASTStmtWriter::VisitOMPCanonicalLoop(OMPCanonicalLoop *S) {
2175   VisitStmt(S);
2176   for (Stmt *SubStmt : S->SubStmts)
2177     Record.AddStmt(SubStmt);
2178   Code = serialization::STMT_OMP_CANONICAL_LOOP;
2179 }
2180 
VisitOMPExecutableDirective(OMPExecutableDirective * E)2181 void ASTStmtWriter::VisitOMPExecutableDirective(OMPExecutableDirective *E) {
2182   Record.writeOMPChildren(E->Data);
2183   Record.AddSourceLocation(E->getBeginLoc());
2184   Record.AddSourceLocation(E->getEndLoc());
2185 }
2186 
VisitOMPLoopBasedDirective(OMPLoopBasedDirective * D)2187 void ASTStmtWriter::VisitOMPLoopBasedDirective(OMPLoopBasedDirective *D) {
2188   VisitStmt(D);
2189   Record.writeUInt32(D->getLoopsNumber());
2190   VisitOMPExecutableDirective(D);
2191 }
2192 
VisitOMPLoopDirective(OMPLoopDirective * D)2193 void ASTStmtWriter::VisitOMPLoopDirective(OMPLoopDirective *D) {
2194   VisitOMPLoopBasedDirective(D);
2195 }
2196 
VisitOMPParallelDirective(OMPParallelDirective * D)2197 void ASTStmtWriter::VisitOMPParallelDirective(OMPParallelDirective *D) {
2198   VisitStmt(D);
2199   VisitOMPExecutableDirective(D);
2200   Record.writeBool(D->hasCancel());
2201   Code = serialization::STMT_OMP_PARALLEL_DIRECTIVE;
2202 }
2203 
VisitOMPSimdDirective(OMPSimdDirective * D)2204 void ASTStmtWriter::VisitOMPSimdDirective(OMPSimdDirective *D) {
2205   VisitOMPLoopDirective(D);
2206   Code = serialization::STMT_OMP_SIMD_DIRECTIVE;
2207 }
2208 
VisitOMPTileDirective(OMPTileDirective * D)2209 void ASTStmtWriter::VisitOMPTileDirective(OMPTileDirective *D) {
2210   VisitOMPLoopBasedDirective(D);
2211   Code = serialization::STMT_OMP_TILE_DIRECTIVE;
2212 }
2213 
VisitOMPForDirective(OMPForDirective * D)2214 void ASTStmtWriter::VisitOMPForDirective(OMPForDirective *D) {
2215   VisitOMPLoopDirective(D);
2216   Record.writeBool(D->hasCancel());
2217   Code = serialization::STMT_OMP_FOR_DIRECTIVE;
2218 }
2219 
VisitOMPForSimdDirective(OMPForSimdDirective * D)2220 void ASTStmtWriter::VisitOMPForSimdDirective(OMPForSimdDirective *D) {
2221   VisitOMPLoopDirective(D);
2222   Code = serialization::STMT_OMP_FOR_SIMD_DIRECTIVE;
2223 }
2224 
VisitOMPSectionsDirective(OMPSectionsDirective * D)2225 void ASTStmtWriter::VisitOMPSectionsDirective(OMPSectionsDirective *D) {
2226   VisitStmt(D);
2227   VisitOMPExecutableDirective(D);
2228   Record.writeBool(D->hasCancel());
2229   Code = serialization::STMT_OMP_SECTIONS_DIRECTIVE;
2230 }
2231 
VisitOMPSectionDirective(OMPSectionDirective * D)2232 void ASTStmtWriter::VisitOMPSectionDirective(OMPSectionDirective *D) {
2233   VisitStmt(D);
2234   VisitOMPExecutableDirective(D);
2235   Record.writeBool(D->hasCancel());
2236   Code = serialization::STMT_OMP_SECTION_DIRECTIVE;
2237 }
2238 
VisitOMPSingleDirective(OMPSingleDirective * D)2239 void ASTStmtWriter::VisitOMPSingleDirective(OMPSingleDirective *D) {
2240   VisitStmt(D);
2241   VisitOMPExecutableDirective(D);
2242   Code = serialization::STMT_OMP_SINGLE_DIRECTIVE;
2243 }
2244 
VisitOMPMasterDirective(OMPMasterDirective * D)2245 void ASTStmtWriter::VisitOMPMasterDirective(OMPMasterDirective *D) {
2246   VisitStmt(D);
2247   VisitOMPExecutableDirective(D);
2248   Code = serialization::STMT_OMP_MASTER_DIRECTIVE;
2249 }
2250 
VisitOMPCriticalDirective(OMPCriticalDirective * D)2251 void ASTStmtWriter::VisitOMPCriticalDirective(OMPCriticalDirective *D) {
2252   VisitStmt(D);
2253   VisitOMPExecutableDirective(D);
2254   Record.AddDeclarationNameInfo(D->getDirectiveName());
2255   Code = serialization::STMT_OMP_CRITICAL_DIRECTIVE;
2256 }
2257 
VisitOMPParallelForDirective(OMPParallelForDirective * D)2258 void ASTStmtWriter::VisitOMPParallelForDirective(OMPParallelForDirective *D) {
2259   VisitOMPLoopDirective(D);
2260   Record.writeBool(D->hasCancel());
2261   Code = serialization::STMT_OMP_PARALLEL_FOR_DIRECTIVE;
2262 }
2263 
VisitOMPParallelForSimdDirective(OMPParallelForSimdDirective * D)2264 void ASTStmtWriter::VisitOMPParallelForSimdDirective(
2265     OMPParallelForSimdDirective *D) {
2266   VisitOMPLoopDirective(D);
2267   Code = serialization::STMT_OMP_PARALLEL_FOR_SIMD_DIRECTIVE;
2268 }
2269 
VisitOMPParallelMasterDirective(OMPParallelMasterDirective * D)2270 void ASTStmtWriter::VisitOMPParallelMasterDirective(
2271     OMPParallelMasterDirective *D) {
2272   VisitStmt(D);
2273   VisitOMPExecutableDirective(D);
2274   Code = serialization::STMT_OMP_PARALLEL_MASTER_DIRECTIVE;
2275 }
2276 
VisitOMPParallelSectionsDirective(OMPParallelSectionsDirective * D)2277 void ASTStmtWriter::VisitOMPParallelSectionsDirective(
2278     OMPParallelSectionsDirective *D) {
2279   VisitStmt(D);
2280   VisitOMPExecutableDirective(D);
2281   Record.writeBool(D->hasCancel());
2282   Code = serialization::STMT_OMP_PARALLEL_SECTIONS_DIRECTIVE;
2283 }
2284 
VisitOMPTaskDirective(OMPTaskDirective * D)2285 void ASTStmtWriter::VisitOMPTaskDirective(OMPTaskDirective *D) {
2286   VisitStmt(D);
2287   VisitOMPExecutableDirective(D);
2288   Record.writeBool(D->hasCancel());
2289   Code = serialization::STMT_OMP_TASK_DIRECTIVE;
2290 }
2291 
VisitOMPAtomicDirective(OMPAtomicDirective * D)2292 void ASTStmtWriter::VisitOMPAtomicDirective(OMPAtomicDirective *D) {
2293   VisitStmt(D);
2294   VisitOMPExecutableDirective(D);
2295   Record.writeBool(D->isXLHSInRHSPart());
2296   Record.writeBool(D->isPostfixUpdate());
2297   Code = serialization::STMT_OMP_ATOMIC_DIRECTIVE;
2298 }
2299 
VisitOMPTargetDirective(OMPTargetDirective * D)2300 void ASTStmtWriter::VisitOMPTargetDirective(OMPTargetDirective *D) {
2301   VisitStmt(D);
2302   VisitOMPExecutableDirective(D);
2303   Code = serialization::STMT_OMP_TARGET_DIRECTIVE;
2304 }
2305 
VisitOMPTargetDataDirective(OMPTargetDataDirective * D)2306 void ASTStmtWriter::VisitOMPTargetDataDirective(OMPTargetDataDirective *D) {
2307   VisitStmt(D);
2308   VisitOMPExecutableDirective(D);
2309   Code = serialization::STMT_OMP_TARGET_DATA_DIRECTIVE;
2310 }
2311 
VisitOMPTargetEnterDataDirective(OMPTargetEnterDataDirective * D)2312 void ASTStmtWriter::VisitOMPTargetEnterDataDirective(
2313     OMPTargetEnterDataDirective *D) {
2314   VisitStmt(D);
2315   VisitOMPExecutableDirective(D);
2316   Code = serialization::STMT_OMP_TARGET_ENTER_DATA_DIRECTIVE;
2317 }
2318 
VisitOMPTargetExitDataDirective(OMPTargetExitDataDirective * D)2319 void ASTStmtWriter::VisitOMPTargetExitDataDirective(
2320     OMPTargetExitDataDirective *D) {
2321   VisitStmt(D);
2322   VisitOMPExecutableDirective(D);
2323   Code = serialization::STMT_OMP_TARGET_EXIT_DATA_DIRECTIVE;
2324 }
2325 
VisitOMPTargetParallelDirective(OMPTargetParallelDirective * D)2326 void ASTStmtWriter::VisitOMPTargetParallelDirective(
2327     OMPTargetParallelDirective *D) {
2328   VisitStmt(D);
2329   VisitOMPExecutableDirective(D);
2330   Record.writeBool(D->hasCancel());
2331   Code = serialization::STMT_OMP_TARGET_PARALLEL_DIRECTIVE;
2332 }
2333 
VisitOMPTargetParallelForDirective(OMPTargetParallelForDirective * D)2334 void ASTStmtWriter::VisitOMPTargetParallelForDirective(
2335     OMPTargetParallelForDirective *D) {
2336   VisitOMPLoopDirective(D);
2337   Record.writeBool(D->hasCancel());
2338   Code = serialization::STMT_OMP_TARGET_PARALLEL_FOR_DIRECTIVE;
2339 }
2340 
VisitOMPTaskyieldDirective(OMPTaskyieldDirective * D)2341 void ASTStmtWriter::VisitOMPTaskyieldDirective(OMPTaskyieldDirective *D) {
2342   VisitStmt(D);
2343   VisitOMPExecutableDirective(D);
2344   Code = serialization::STMT_OMP_TASKYIELD_DIRECTIVE;
2345 }
2346 
VisitOMPBarrierDirective(OMPBarrierDirective * D)2347 void ASTStmtWriter::VisitOMPBarrierDirective(OMPBarrierDirective *D) {
2348   VisitStmt(D);
2349   VisitOMPExecutableDirective(D);
2350   Code = serialization::STMT_OMP_BARRIER_DIRECTIVE;
2351 }
2352 
VisitOMPTaskwaitDirective(OMPTaskwaitDirective * D)2353 void ASTStmtWriter::VisitOMPTaskwaitDirective(OMPTaskwaitDirective *D) {
2354   VisitStmt(D);
2355   VisitOMPExecutableDirective(D);
2356   Code = serialization::STMT_OMP_TASKWAIT_DIRECTIVE;
2357 }
2358 
VisitOMPTaskgroupDirective(OMPTaskgroupDirective * D)2359 void ASTStmtWriter::VisitOMPTaskgroupDirective(OMPTaskgroupDirective *D) {
2360   VisitStmt(D);
2361   VisitOMPExecutableDirective(D);
2362   Code = serialization::STMT_OMP_TASKGROUP_DIRECTIVE;
2363 }
2364 
VisitOMPFlushDirective(OMPFlushDirective * D)2365 void ASTStmtWriter::VisitOMPFlushDirective(OMPFlushDirective *D) {
2366   VisitStmt(D);
2367   VisitOMPExecutableDirective(D);
2368   Code = serialization::STMT_OMP_FLUSH_DIRECTIVE;
2369 }
2370 
VisitOMPDepobjDirective(OMPDepobjDirective * D)2371 void ASTStmtWriter::VisitOMPDepobjDirective(OMPDepobjDirective *D) {
2372   VisitStmt(D);
2373   VisitOMPExecutableDirective(D);
2374   Code = serialization::STMT_OMP_DEPOBJ_DIRECTIVE;
2375 }
2376 
VisitOMPScanDirective(OMPScanDirective * D)2377 void ASTStmtWriter::VisitOMPScanDirective(OMPScanDirective *D) {
2378   VisitStmt(D);
2379   VisitOMPExecutableDirective(D);
2380   Code = serialization::STMT_OMP_SCAN_DIRECTIVE;
2381 }
2382 
VisitOMPOrderedDirective(OMPOrderedDirective * D)2383 void ASTStmtWriter::VisitOMPOrderedDirective(OMPOrderedDirective *D) {
2384   VisitStmt(D);
2385   VisitOMPExecutableDirective(D);
2386   Code = serialization::STMT_OMP_ORDERED_DIRECTIVE;
2387 }
2388 
VisitOMPTeamsDirective(OMPTeamsDirective * D)2389 void ASTStmtWriter::VisitOMPTeamsDirective(OMPTeamsDirective *D) {
2390   VisitStmt(D);
2391   VisitOMPExecutableDirective(D);
2392   Code = serialization::STMT_OMP_TEAMS_DIRECTIVE;
2393 }
2394 
VisitOMPCancellationPointDirective(OMPCancellationPointDirective * D)2395 void ASTStmtWriter::VisitOMPCancellationPointDirective(
2396     OMPCancellationPointDirective *D) {
2397   VisitStmt(D);
2398   VisitOMPExecutableDirective(D);
2399   Record.writeEnum(D->getCancelRegion());
2400   Code = serialization::STMT_OMP_CANCELLATION_POINT_DIRECTIVE;
2401 }
2402 
VisitOMPCancelDirective(OMPCancelDirective * D)2403 void ASTStmtWriter::VisitOMPCancelDirective(OMPCancelDirective *D) {
2404   VisitStmt(D);
2405   VisitOMPExecutableDirective(D);
2406   Record.writeEnum(D->getCancelRegion());
2407   Code = serialization::STMT_OMP_CANCEL_DIRECTIVE;
2408 }
2409 
VisitOMPTaskLoopDirective(OMPTaskLoopDirective * D)2410 void ASTStmtWriter::VisitOMPTaskLoopDirective(OMPTaskLoopDirective *D) {
2411   VisitOMPLoopDirective(D);
2412   Record.writeBool(D->hasCancel());
2413   Code = serialization::STMT_OMP_TASKLOOP_DIRECTIVE;
2414 }
2415 
VisitOMPTaskLoopSimdDirective(OMPTaskLoopSimdDirective * D)2416 void ASTStmtWriter::VisitOMPTaskLoopSimdDirective(OMPTaskLoopSimdDirective *D) {
2417   VisitOMPLoopDirective(D);
2418   Code = serialization::STMT_OMP_TASKLOOP_SIMD_DIRECTIVE;
2419 }
2420 
VisitOMPMasterTaskLoopDirective(OMPMasterTaskLoopDirective * D)2421 void ASTStmtWriter::VisitOMPMasterTaskLoopDirective(
2422     OMPMasterTaskLoopDirective *D) {
2423   VisitOMPLoopDirective(D);
2424   Record.writeBool(D->hasCancel());
2425   Code = serialization::STMT_OMP_MASTER_TASKLOOP_DIRECTIVE;
2426 }
2427 
VisitOMPMasterTaskLoopSimdDirective(OMPMasterTaskLoopSimdDirective * D)2428 void ASTStmtWriter::VisitOMPMasterTaskLoopSimdDirective(
2429     OMPMasterTaskLoopSimdDirective *D) {
2430   VisitOMPLoopDirective(D);
2431   Code = serialization::STMT_OMP_MASTER_TASKLOOP_SIMD_DIRECTIVE;
2432 }
2433 
VisitOMPParallelMasterTaskLoopDirective(OMPParallelMasterTaskLoopDirective * D)2434 void ASTStmtWriter::VisitOMPParallelMasterTaskLoopDirective(
2435     OMPParallelMasterTaskLoopDirective *D) {
2436   VisitOMPLoopDirective(D);
2437   Record.writeBool(D->hasCancel());
2438   Code = serialization::STMT_OMP_PARALLEL_MASTER_TASKLOOP_DIRECTIVE;
2439 }
2440 
VisitOMPParallelMasterTaskLoopSimdDirective(OMPParallelMasterTaskLoopSimdDirective * D)2441 void ASTStmtWriter::VisitOMPParallelMasterTaskLoopSimdDirective(
2442     OMPParallelMasterTaskLoopSimdDirective *D) {
2443   VisitOMPLoopDirective(D);
2444   Code = serialization::STMT_OMP_PARALLEL_MASTER_TASKLOOP_SIMD_DIRECTIVE;
2445 }
2446 
VisitOMPDistributeDirective(OMPDistributeDirective * D)2447 void ASTStmtWriter::VisitOMPDistributeDirective(OMPDistributeDirective *D) {
2448   VisitOMPLoopDirective(D);
2449   Code = serialization::STMT_OMP_DISTRIBUTE_DIRECTIVE;
2450 }
2451 
VisitOMPTargetUpdateDirective(OMPTargetUpdateDirective * D)2452 void ASTStmtWriter::VisitOMPTargetUpdateDirective(OMPTargetUpdateDirective *D) {
2453   VisitStmt(D);
2454   VisitOMPExecutableDirective(D);
2455   Code = serialization::STMT_OMP_TARGET_UPDATE_DIRECTIVE;
2456 }
2457 
VisitOMPDistributeParallelForDirective(OMPDistributeParallelForDirective * D)2458 void ASTStmtWriter::VisitOMPDistributeParallelForDirective(
2459     OMPDistributeParallelForDirective *D) {
2460   VisitOMPLoopDirective(D);
2461   Record.writeBool(D->hasCancel());
2462   Code = serialization::STMT_OMP_DISTRIBUTE_PARALLEL_FOR_DIRECTIVE;
2463 }
2464 
VisitOMPDistributeParallelForSimdDirective(OMPDistributeParallelForSimdDirective * D)2465 void ASTStmtWriter::VisitOMPDistributeParallelForSimdDirective(
2466     OMPDistributeParallelForSimdDirective *D) {
2467   VisitOMPLoopDirective(D);
2468   Code = serialization::STMT_OMP_DISTRIBUTE_PARALLEL_FOR_SIMD_DIRECTIVE;
2469 }
2470 
VisitOMPDistributeSimdDirective(OMPDistributeSimdDirective * D)2471 void ASTStmtWriter::VisitOMPDistributeSimdDirective(
2472     OMPDistributeSimdDirective *D) {
2473   VisitOMPLoopDirective(D);
2474   Code = serialization::STMT_OMP_DISTRIBUTE_SIMD_DIRECTIVE;
2475 }
2476 
VisitOMPTargetParallelForSimdDirective(OMPTargetParallelForSimdDirective * D)2477 void ASTStmtWriter::VisitOMPTargetParallelForSimdDirective(
2478     OMPTargetParallelForSimdDirective *D) {
2479   VisitOMPLoopDirective(D);
2480   Code = serialization::STMT_OMP_TARGET_PARALLEL_FOR_SIMD_DIRECTIVE;
2481 }
2482 
VisitOMPTargetSimdDirective(OMPTargetSimdDirective * D)2483 void ASTStmtWriter::VisitOMPTargetSimdDirective(OMPTargetSimdDirective *D) {
2484   VisitOMPLoopDirective(D);
2485   Code = serialization::STMT_OMP_TARGET_SIMD_DIRECTIVE;
2486 }
2487 
VisitOMPTeamsDistributeDirective(OMPTeamsDistributeDirective * D)2488 void ASTStmtWriter::VisitOMPTeamsDistributeDirective(
2489     OMPTeamsDistributeDirective *D) {
2490   VisitOMPLoopDirective(D);
2491   Code = serialization::STMT_OMP_TEAMS_DISTRIBUTE_DIRECTIVE;
2492 }
2493 
VisitOMPTeamsDistributeSimdDirective(OMPTeamsDistributeSimdDirective * D)2494 void ASTStmtWriter::VisitOMPTeamsDistributeSimdDirective(
2495     OMPTeamsDistributeSimdDirective *D) {
2496   VisitOMPLoopDirective(D);
2497   Code = serialization::STMT_OMP_TEAMS_DISTRIBUTE_SIMD_DIRECTIVE;
2498 }
2499 
VisitOMPTeamsDistributeParallelForSimdDirective(OMPTeamsDistributeParallelForSimdDirective * D)2500 void ASTStmtWriter::VisitOMPTeamsDistributeParallelForSimdDirective(
2501     OMPTeamsDistributeParallelForSimdDirective *D) {
2502   VisitOMPLoopDirective(D);
2503   Code = serialization::STMT_OMP_TEAMS_DISTRIBUTE_PARALLEL_FOR_SIMD_DIRECTIVE;
2504 }
2505 
VisitOMPTeamsDistributeParallelForDirective(OMPTeamsDistributeParallelForDirective * D)2506 void ASTStmtWriter::VisitOMPTeamsDistributeParallelForDirective(
2507     OMPTeamsDistributeParallelForDirective *D) {
2508   VisitOMPLoopDirective(D);
2509   Record.writeBool(D->hasCancel());
2510   Code = serialization::STMT_OMP_TEAMS_DISTRIBUTE_PARALLEL_FOR_DIRECTIVE;
2511 }
2512 
VisitOMPTargetTeamsDirective(OMPTargetTeamsDirective * D)2513 void ASTStmtWriter::VisitOMPTargetTeamsDirective(OMPTargetTeamsDirective *D) {
2514   VisitStmt(D);
2515   VisitOMPExecutableDirective(D);
2516   Code = serialization::STMT_OMP_TARGET_TEAMS_DIRECTIVE;
2517 }
2518 
VisitOMPTargetTeamsDistributeDirective(OMPTargetTeamsDistributeDirective * D)2519 void ASTStmtWriter::VisitOMPTargetTeamsDistributeDirective(
2520     OMPTargetTeamsDistributeDirective *D) {
2521   VisitOMPLoopDirective(D);
2522   Code = serialization::STMT_OMP_TARGET_TEAMS_DISTRIBUTE_DIRECTIVE;
2523 }
2524 
VisitOMPTargetTeamsDistributeParallelForDirective(OMPTargetTeamsDistributeParallelForDirective * D)2525 void ASTStmtWriter::VisitOMPTargetTeamsDistributeParallelForDirective(
2526     OMPTargetTeamsDistributeParallelForDirective *D) {
2527   VisitOMPLoopDirective(D);
2528   Record.writeBool(D->hasCancel());
2529   Code = serialization::STMT_OMP_TARGET_TEAMS_DISTRIBUTE_PARALLEL_FOR_DIRECTIVE;
2530 }
2531 
VisitOMPTargetTeamsDistributeParallelForSimdDirective(OMPTargetTeamsDistributeParallelForSimdDirective * D)2532 void ASTStmtWriter::VisitOMPTargetTeamsDistributeParallelForSimdDirective(
2533     OMPTargetTeamsDistributeParallelForSimdDirective *D) {
2534   VisitOMPLoopDirective(D);
2535   Code = serialization::
2536       STMT_OMP_TARGET_TEAMS_DISTRIBUTE_PARALLEL_FOR_SIMD_DIRECTIVE;
2537 }
2538 
VisitOMPTargetTeamsDistributeSimdDirective(OMPTargetTeamsDistributeSimdDirective * D)2539 void ASTStmtWriter::VisitOMPTargetTeamsDistributeSimdDirective(
2540     OMPTargetTeamsDistributeSimdDirective *D) {
2541   VisitOMPLoopDirective(D);
2542   Code = serialization::STMT_OMP_TARGET_TEAMS_DISTRIBUTE_SIMD_DIRECTIVE;
2543 }
2544 
VisitOMPInteropDirective(OMPInteropDirective * D)2545 void ASTStmtWriter::VisitOMPInteropDirective(OMPInteropDirective *D) {
2546   VisitStmt(D);
2547   VisitOMPExecutableDirective(D);
2548   Code = serialization::STMT_OMP_INTEROP_DIRECTIVE;
2549 }
2550 
VisitOMPDispatchDirective(OMPDispatchDirective * D)2551 void ASTStmtWriter::VisitOMPDispatchDirective(OMPDispatchDirective *D) {
2552   VisitStmt(D);
2553   VisitOMPExecutableDirective(D);
2554   Record.AddSourceLocation(D->getTargetCallLoc());
2555   Code = serialization::STMT_OMP_DISPATCH_DIRECTIVE;
2556 }
2557 
VisitOMPMaskedDirective(OMPMaskedDirective * D)2558 void ASTStmtWriter::VisitOMPMaskedDirective(OMPMaskedDirective *D) {
2559   VisitStmt(D);
2560   VisitOMPExecutableDirective(D);
2561   Code = serialization::STMT_OMP_MASKED_DIRECTIVE;
2562 }
2563 
2564 //===----------------------------------------------------------------------===//
2565 // ASTWriter Implementation
2566 //===----------------------------------------------------------------------===//
2567 
RecordSwitchCaseID(SwitchCase * S)2568 unsigned ASTWriter::RecordSwitchCaseID(SwitchCase *S) {
2569   assert(SwitchCaseIDs.find(S) == SwitchCaseIDs.end() &&
2570          "SwitchCase recorded twice");
2571   unsigned NextID = SwitchCaseIDs.size();
2572   SwitchCaseIDs[S] = NextID;
2573   return NextID;
2574 }
2575 
getSwitchCaseID(SwitchCase * S)2576 unsigned ASTWriter::getSwitchCaseID(SwitchCase *S) {
2577   assert(SwitchCaseIDs.find(S) != SwitchCaseIDs.end() &&
2578          "SwitchCase hasn't been seen yet");
2579   return SwitchCaseIDs[S];
2580 }
2581 
ClearSwitchCaseIDs()2582 void ASTWriter::ClearSwitchCaseIDs() {
2583   SwitchCaseIDs.clear();
2584 }
2585 
2586 /// Write the given substatement or subexpression to the
2587 /// bitstream.
WriteSubStmt(Stmt * S)2588 void ASTWriter::WriteSubStmt(Stmt *S) {
2589   RecordData Record;
2590   ASTStmtWriter Writer(*this, Record);
2591   ++NumStatements;
2592 
2593   if (!S) {
2594     Stream.EmitRecord(serialization::STMT_NULL_PTR, Record);
2595     return;
2596   }
2597 
2598   llvm::DenseMap<Stmt *, uint64_t>::iterator I = SubStmtEntries.find(S);
2599   if (I != SubStmtEntries.end()) {
2600     Record.push_back(I->second);
2601     Stream.EmitRecord(serialization::STMT_REF_PTR, Record);
2602     return;
2603   }
2604 
2605 #ifndef NDEBUG
2606   assert(!ParentStmts.count(S) && "There is a Stmt cycle!");
2607 
2608   struct ParentStmtInserterRAII {
2609     Stmt *S;
2610     llvm::DenseSet<Stmt *> &ParentStmts;
2611 
2612     ParentStmtInserterRAII(Stmt *S, llvm::DenseSet<Stmt *> &ParentStmts)
2613       : S(S), ParentStmts(ParentStmts) {
2614       ParentStmts.insert(S);
2615     }
2616     ~ParentStmtInserterRAII() {
2617       ParentStmts.erase(S);
2618     }
2619   };
2620 
2621   ParentStmtInserterRAII ParentStmtInserter(S, ParentStmts);
2622 #endif
2623 
2624   Writer.Visit(S);
2625 
2626   uint64_t Offset = Writer.Emit();
2627   SubStmtEntries[S] = Offset;
2628 }
2629 
2630 /// Flush all of the statements that have been added to the
2631 /// queue via AddStmt().
FlushStmts()2632 void ASTRecordWriter::FlushStmts() {
2633   // We expect to be the only consumer of the two temporary statement maps,
2634   // assert that they are empty.
2635   assert(Writer->SubStmtEntries.empty() && "unexpected entries in sub-stmt map");
2636   assert(Writer->ParentStmts.empty() && "unexpected entries in parent stmt map");
2637 
2638   for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
2639     Writer->WriteSubStmt(StmtsToEmit[I]);
2640 
2641     assert(N == StmtsToEmit.size() && "record modified while being written!");
2642 
2643     // Note that we are at the end of a full expression. Any
2644     // expression records that follow this one are part of a different
2645     // expression.
2646     Writer->Stream.EmitRecord(serialization::STMT_STOP, ArrayRef<uint32_t>());
2647 
2648     Writer->SubStmtEntries.clear();
2649     Writer->ParentStmts.clear();
2650   }
2651 
2652   StmtsToEmit.clear();
2653 }
2654 
FlushSubStmts()2655 void ASTRecordWriter::FlushSubStmts() {
2656   // For a nested statement, write out the substatements in reverse order (so
2657   // that a simple stack machine can be used when loading), and don't emit a
2658   // STMT_STOP after each one.
2659   for (unsigned I = 0, N = StmtsToEmit.size(); I != N; ++I) {
2660     Writer->WriteSubStmt(StmtsToEmit[N - I - 1]);
2661     assert(N == StmtsToEmit.size() && "record modified while being written!");
2662   }
2663 
2664   StmtsToEmit.clear();
2665 }
2666