• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 //===- unittest/Tooling/RecursiveASTVisitorTest.cpp -----------------------===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "TestVisitor.h"
11 #include <stack>
12 
13 using namespace clang;
14 
15 namespace {
16 
17 class TypeLocVisitor : public ExpectedLocationVisitor<TypeLocVisitor> {
18 public:
VisitTypeLoc(TypeLoc TypeLocation)19   bool VisitTypeLoc(TypeLoc TypeLocation) {
20     Match(TypeLocation.getType().getAsString(), TypeLocation.getBeginLoc());
21     return true;
22   }
23 };
24 
25 class DeclRefExprVisitor : public ExpectedLocationVisitor<DeclRefExprVisitor> {
26 public:
VisitDeclRefExpr(DeclRefExpr * Reference)27   bool VisitDeclRefExpr(DeclRefExpr *Reference) {
28     Match(Reference->getNameInfo().getAsString(), Reference->getLocation());
29     return true;
30   }
31 };
32 
33 class VarDeclVisitor : public ExpectedLocationVisitor<VarDeclVisitor> {
34 public:
VisitVarDecl(VarDecl * Variable)35  bool VisitVarDecl(VarDecl *Variable) {
36    Match(Variable->getNameAsString(), Variable->getLocStart());
37    return true;
38  }
39 };
40 
41 class ParmVarDeclVisitorForImplicitCode :
42   public ExpectedLocationVisitor<ParmVarDeclVisitorForImplicitCode> {
43 public:
shouldVisitImplicitCode() const44   bool shouldVisitImplicitCode() const { return true; }
45 
VisitParmVarDecl(ParmVarDecl * ParamVar)46   bool VisitParmVarDecl(ParmVarDecl *ParamVar) {
47     Match(ParamVar->getNameAsString(), ParamVar->getLocStart());
48     return true;
49   }
50 };
51 
52 class CXXMemberCallVisitor
53   : public ExpectedLocationVisitor<CXXMemberCallVisitor> {
54 public:
VisitCXXMemberCallExpr(CXXMemberCallExpr * Call)55   bool VisitCXXMemberCallExpr(CXXMemberCallExpr *Call) {
56     Match(Call->getMethodDecl()->getQualifiedNameAsString(),
57           Call->getLocStart());
58     return true;
59   }
60 };
61 
62 class NamedDeclVisitor
63   : public ExpectedLocationVisitor<NamedDeclVisitor> {
64 public:
VisitNamedDecl(NamedDecl * Decl)65   bool VisitNamedDecl(NamedDecl *Decl) {
66     std::string NameWithTemplateArgs;
67     llvm::raw_string_ostream OS(NameWithTemplateArgs);
68     Decl->getNameForDiagnostic(OS,
69                                Decl->getASTContext().getPrintingPolicy(),
70                                true);
71     Match(OS.str(), Decl->getLocation());
72     return true;
73   }
74 };
75 
76 class CXXOperatorCallExprTraverser
77   : public ExpectedLocationVisitor<CXXOperatorCallExprTraverser> {
78 public:
79   // Use Traverse, not Visit, to check that data recursion optimization isn't
80   // bypassing the call of this function.
TraverseCXXOperatorCallExpr(CXXOperatorCallExpr * CE)81   bool TraverseCXXOperatorCallExpr(CXXOperatorCallExpr *CE) {
82     Match(getOperatorSpelling(CE->getOperator()), CE->getExprLoc());
83     return ExpectedLocationVisitor<CXXOperatorCallExprTraverser>::
84         TraverseCXXOperatorCallExpr(CE);
85   }
86 };
87 
88 class ParenExprVisitor : public ExpectedLocationVisitor<ParenExprVisitor> {
89 public:
VisitParenExpr(ParenExpr * Parens)90   bool VisitParenExpr(ParenExpr *Parens) {
91     Match("", Parens->getExprLoc());
92     return true;
93   }
94 };
95 
96 class LambdaExprVisitor : public ExpectedLocationVisitor<LambdaExprVisitor> {
97 public:
VisitLambdaExpr(LambdaExpr * Lambda)98   bool VisitLambdaExpr(LambdaExpr *Lambda) {
99     PendingBodies.push(Lambda);
100     Match("", Lambda->getIntroducerRange().getBegin());
101     return true;
102   }
103   /// For each call to VisitLambdaExpr, we expect a subsequent call (with
104   /// proper nesting) to TraverseLambdaBody.
TraverseLambdaBody(LambdaExpr * Lambda)105   bool TraverseLambdaBody(LambdaExpr *Lambda) {
106     EXPECT_FALSE(PendingBodies.empty());
107     EXPECT_EQ(PendingBodies.top(), Lambda);
108     PendingBodies.pop();
109     return TraverseStmt(Lambda->getBody());
110   }
111   /// Determine whether TraverseLambdaBody has been called for every call to
112   /// VisitLambdaExpr.
allBodiesHaveBeenTraversed() const113   bool allBodiesHaveBeenTraversed() const {
114     return PendingBodies.empty();
115   }
116 private:
117   std::stack<LambdaExpr *> PendingBodies;
118 };
119 
120 // Matches the (optional) capture-default of a lambda-introducer.
121 class LambdaDefaultCaptureVisitor
122   : public ExpectedLocationVisitor<LambdaDefaultCaptureVisitor> {
123 public:
VisitLambdaExpr(LambdaExpr * Lambda)124   bool VisitLambdaExpr(LambdaExpr *Lambda) {
125     if (Lambda->getCaptureDefault() != LCD_None) {
126       Match("", Lambda->getCaptureDefaultLoc());
127     }
128     return true;
129   }
130 };
131 
132 class TemplateArgumentLocTraverser
133   : public ExpectedLocationVisitor<TemplateArgumentLocTraverser> {
134 public:
TraverseTemplateArgumentLoc(const TemplateArgumentLoc & ArgLoc)135   bool TraverseTemplateArgumentLoc(const TemplateArgumentLoc &ArgLoc) {
136     std::string ArgStr;
137     llvm::raw_string_ostream Stream(ArgStr);
138     const TemplateArgument &Arg = ArgLoc.getArgument();
139 
140     Arg.print(Context->getPrintingPolicy(), Stream);
141     Match(Stream.str(), ArgLoc.getLocation());
142     return ExpectedLocationVisitor<TemplateArgumentLocTraverser>::
143       TraverseTemplateArgumentLoc(ArgLoc);
144   }
145 };
146 
147 class CXXBoolLiteralExprVisitor
148   : public ExpectedLocationVisitor<CXXBoolLiteralExprVisitor> {
149 public:
VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr * BE)150   bool VisitCXXBoolLiteralExpr(CXXBoolLiteralExpr *BE) {
151     if (BE->getValue())
152       Match("true", BE->getLocation());
153     else
154       Match("false", BE->getLocation());
155     return true;
156   }
157 };
158 
159 // Test RAV visits parameter variable declaration of the implicit
160 // copy assignment operator and implicit copy constructor.
TEST(RecursiveASTVisitor,VisitsParmVarDeclForImplicitCode)161 TEST(RecursiveASTVisitor, VisitsParmVarDeclForImplicitCode) {
162   ParmVarDeclVisitorForImplicitCode Visitor;
163   // Match parameter variable name of implicit copy assignment operator and
164   // implicit copy constructor.
165   // This parameter name does not have a valid IdentifierInfo, and shares
166   // same SourceLocation with its class declaration, so we match an empty name
167   // with the class' source location.
168   Visitor.ExpectMatch("", 1, 7);
169   Visitor.ExpectMatch("", 3, 7);
170   EXPECT_TRUE(Visitor.runOver(
171     "class X {};\n"
172     "void foo(X a, X b) {a = b;}\n"
173     "class Y {};\n"
174     "void bar(Y a) {Y b = a;}"));
175 }
176 
TEST(RecursiveASTVisitor,VisitsBaseClassDeclarations)177 TEST(RecursiveASTVisitor, VisitsBaseClassDeclarations) {
178   TypeLocVisitor Visitor;
179   Visitor.ExpectMatch("class X", 1, 30);
180   EXPECT_TRUE(Visitor.runOver("class X {}; class Y : public X {};"));
181 }
182 
TEST(RecursiveASTVisitor,VisitsCXXBaseSpecifiersOfForwardDeclaredClass)183 TEST(RecursiveASTVisitor, VisitsCXXBaseSpecifiersOfForwardDeclaredClass) {
184   TypeLocVisitor Visitor;
185   Visitor.ExpectMatch("class X", 3, 18);
186   EXPECT_TRUE(Visitor.runOver(
187     "class Y;\n"
188     "class X {};\n"
189     "class Y : public X {};"));
190 }
191 
TEST(RecursiveASTVisitor,VisitsCXXBaseSpecifiersWithIncompleteInnerClass)192 TEST(RecursiveASTVisitor, VisitsCXXBaseSpecifiersWithIncompleteInnerClass) {
193   TypeLocVisitor Visitor;
194   Visitor.ExpectMatch("class X", 2, 18);
195   EXPECT_TRUE(Visitor.runOver(
196     "class X {};\n"
197     "class Y : public X { class Z; };"));
198 }
199 
TEST(RecursiveASTVisitor,VisitsCXXBaseSpecifiersOfSelfReferentialType)200 TEST(RecursiveASTVisitor, VisitsCXXBaseSpecifiersOfSelfReferentialType) {
201   TypeLocVisitor Visitor;
202   Visitor.ExpectMatch("X<class Y>", 2, 18);
203   EXPECT_TRUE(Visitor.runOver(
204     "template<typename T> class X {};\n"
205     "class Y : public X<Y> {};"));
206 }
207 
TEST(RecursiveASTVisitor,VisitsBaseClassTemplateArguments)208 TEST(RecursiveASTVisitor, VisitsBaseClassTemplateArguments) {
209   DeclRefExprVisitor Visitor;
210   Visitor.ExpectMatch("x", 2, 3);
211   EXPECT_TRUE(Visitor.runOver(
212     "void x(); template <void (*T)()> class X {};\nX<x> y;"));
213 }
214 
TEST(RecursiveASTVisitor,VisitsCXXForRangeStmtRange)215 TEST(RecursiveASTVisitor, VisitsCXXForRangeStmtRange) {
216   DeclRefExprVisitor Visitor;
217   Visitor.ExpectMatch("x", 2, 25);
218   Visitor.ExpectMatch("x", 2, 30);
219   EXPECT_TRUE(Visitor.runOver(
220     "int x[5];\n"
221     "void f() { for (int i : x) { x[0] = 1; } }",
222     DeclRefExprVisitor::Lang_CXX11));
223 }
224 
TEST(RecursiveASTVisitor,VisitsCXXForRangeStmtLoopVariable)225 TEST(RecursiveASTVisitor, VisitsCXXForRangeStmtLoopVariable) {
226   VarDeclVisitor Visitor;
227   Visitor.ExpectMatch("i", 2, 17);
228   EXPECT_TRUE(Visitor.runOver(
229     "int x[5];\n"
230     "void f() { for (int i : x) {} }",
231     VarDeclVisitor::Lang_CXX11));
232 }
233 
TEST(RecursiveASTVisitor,VisitsCallExpr)234 TEST(RecursiveASTVisitor, VisitsCallExpr) {
235   DeclRefExprVisitor Visitor;
236   Visitor.ExpectMatch("x", 1, 22);
237   EXPECT_TRUE(Visitor.runOver(
238     "void x(); void y() { x(); }"));
239 }
240 
TEST(RecursiveASTVisitor,VisitsCallInTemplateInstantiation)241 TEST(RecursiveASTVisitor, VisitsCallInTemplateInstantiation) {
242   CXXMemberCallVisitor Visitor;
243   Visitor.ExpectMatch("Y::x", 3, 3);
244   EXPECT_TRUE(Visitor.runOver(
245     "struct Y { void x(); };\n"
246     "template<typename T> void y(T t) {\n"
247     "  t.x();\n"
248     "}\n"
249     "void foo() { y<Y>(Y()); }"));
250 }
251 
TEST(RecursiveASTVisitor,VisitsCallInNestedFunctionTemplateInstantiation)252 TEST(RecursiveASTVisitor, VisitsCallInNestedFunctionTemplateInstantiation) {
253   CXXMemberCallVisitor Visitor;
254   Visitor.ExpectMatch("Y::x", 4, 5);
255   EXPECT_TRUE(Visitor.runOver(
256     "struct Y { void x(); };\n"
257     "template<typename T> struct Z {\n"
258     "  template<typename U> static void f() {\n"
259     "    T().x();\n"
260     "  }\n"
261     "};\n"
262     "void foo() { Z<Y>::f<int>(); }"));
263 }
264 
TEST(RecursiveASTVisitor,VisitsCallInNestedClassTemplateInstantiation)265 TEST(RecursiveASTVisitor, VisitsCallInNestedClassTemplateInstantiation) {
266   CXXMemberCallVisitor Visitor;
267   Visitor.ExpectMatch("A::x", 5, 7);
268   EXPECT_TRUE(Visitor.runOver(
269     "template <typename T1> struct X {\n"
270     "  template <typename T2> struct Y {\n"
271     "    void f() {\n"
272     "      T2 y;\n"
273     "      y.x();\n"
274     "    }\n"
275     "  };\n"
276     "};\n"
277     "struct A { void x(); };\n"
278     "int main() {\n"
279     "  (new X<A>::Y<A>())->f();\n"
280     "}"));
281 }
282 
283 /* FIXME: According to Richard Smith this is a bug in the AST.
284 TEST(RecursiveASTVisitor, VisitsBaseClassTemplateArgumentsInInstantiation) {
285   DeclRefExprVisitor Visitor;
286   Visitor.ExpectMatch("x", 3, 43);
287   EXPECT_TRUE(Visitor.runOver(
288     "template <typename T> void x();\n"
289     "template <void (*T)()> class X {};\n"
290     "template <typename T> class Y : public X< x<T> > {};\n"
291     "Y<int> y;"));
292 }
293 */
294 
TEST(RecursiveASTVisitor,VisitsCallInPartialTemplateSpecialization)295 TEST(RecursiveASTVisitor, VisitsCallInPartialTemplateSpecialization) {
296   CXXMemberCallVisitor Visitor;
297   Visitor.ExpectMatch("A::x", 6, 20);
298   EXPECT_TRUE(Visitor.runOver(
299     "template <typename T1> struct X {\n"
300     "  template <typename T2, bool B> struct Y { void g(); };\n"
301     "};\n"
302     "template <typename T1> template <typename T2>\n"
303     "struct X<T1>::Y<T2, true> {\n"
304     "  void f() { T2 y; y.x(); }\n"
305     "};\n"
306     "struct A { void x(); };\n"
307     "int main() {\n"
308     "  (new X<A>::Y<A, true>())->f();\n"
309     "}\n"));
310 }
311 
TEST(RecursiveASTVisitor,VisitsExplicitTemplateSpecialization)312 TEST(RecursiveASTVisitor, VisitsExplicitTemplateSpecialization) {
313   CXXMemberCallVisitor Visitor;
314   Visitor.ExpectMatch("A::f", 4, 5);
315   EXPECT_TRUE(Visitor.runOver(
316     "struct A {\n"
317     "  void f() const {}\n"
318     "  template<class T> void g(const T& t) const {\n"
319     "    t.f();\n"
320     "  }\n"
321     "};\n"
322     "template void A::g(const A& a) const;\n"));
323 }
324 
TEST(RecursiveASTVisitor,VisitsPartialTemplateSpecialization)325 TEST(RecursiveASTVisitor, VisitsPartialTemplateSpecialization) {
326   // From cfe-commits/Week-of-Mon-20100830/033998.html
327   // Contrary to the approach suggested in that email, we visit all
328   // specializations when we visit the primary template.  Visiting them when we
329   // visit the associated specialization is problematic for specializations of
330   // template members of class templates.
331   NamedDeclVisitor Visitor;
332   Visitor.ExpectMatch("A<bool>", 1, 26);
333   Visitor.ExpectMatch("A<char *>", 2, 26);
334   EXPECT_TRUE(Visitor.runOver(
335     "template <class T> class A {};\n"
336     "template <class T> class A<T*> {};\n"
337     "A<bool> ab;\n"
338     "A<char*> acp;\n"));
339 }
340 
TEST(RecursiveASTVisitor,VisitsUndefinedClassTemplateSpecialization)341 TEST(RecursiveASTVisitor, VisitsUndefinedClassTemplateSpecialization) {
342   NamedDeclVisitor Visitor;
343   Visitor.ExpectMatch("A<int>", 1, 29);
344   EXPECT_TRUE(Visitor.runOver(
345     "template<typename T> struct A;\n"
346     "A<int> *p;\n"));
347 }
348 
TEST(RecursiveASTVisitor,VisitsNestedUndefinedClassTemplateSpecialization)349 TEST(RecursiveASTVisitor, VisitsNestedUndefinedClassTemplateSpecialization) {
350   NamedDeclVisitor Visitor;
351   Visitor.ExpectMatch("A<int>::B<char>", 2, 31);
352   EXPECT_TRUE(Visitor.runOver(
353     "template<typename T> struct A {\n"
354     "  template<typename U> struct B;\n"
355     "};\n"
356     "A<int>::B<char> *p;\n"));
357 }
358 
TEST(RecursiveASTVisitor,VisitsUndefinedFunctionTemplateSpecialization)359 TEST(RecursiveASTVisitor, VisitsUndefinedFunctionTemplateSpecialization) {
360   NamedDeclVisitor Visitor;
361   Visitor.ExpectMatch("A<int>", 1, 26);
362   EXPECT_TRUE(Visitor.runOver(
363     "template<typename T> int A();\n"
364     "int k = A<int>();\n"));
365 }
366 
TEST(RecursiveASTVisitor,VisitsNestedUndefinedFunctionTemplateSpecialization)367 TEST(RecursiveASTVisitor, VisitsNestedUndefinedFunctionTemplateSpecialization) {
368   NamedDeclVisitor Visitor;
369   Visitor.ExpectMatch("A<int>::B<char>", 2, 35);
370   EXPECT_TRUE(Visitor.runOver(
371     "template<typename T> struct A {\n"
372     "  template<typename U> static int B();\n"
373     "};\n"
374     "int k = A<int>::B<char>();\n"));
375 }
376 
TEST(RecursiveASTVisitor,NoRecursionInSelfFriend)377 TEST(RecursiveASTVisitor, NoRecursionInSelfFriend) {
378   // From cfe-commits/Week-of-Mon-20100830/033977.html
379   NamedDeclVisitor Visitor;
380   Visitor.ExpectMatch("vector_iterator<int>", 2, 7);
381   EXPECT_TRUE(Visitor.runOver(
382     "template<typename Container>\n"
383     "class vector_iterator {\n"
384     "    template <typename C> friend class vector_iterator;\n"
385     "};\n"
386     "vector_iterator<int> it_int;\n"));
387 }
388 
TEST(RecursiveASTVisitor,TraversesOverloadedOperator)389 TEST(RecursiveASTVisitor, TraversesOverloadedOperator) {
390   CXXOperatorCallExprTraverser Visitor;
391   Visitor.ExpectMatch("()", 4, 9);
392   EXPECT_TRUE(Visitor.runOver(
393     "struct A {\n"
394     "  int operator()();\n"
395     "} a;\n"
396     "int k = a();\n"));
397 }
398 
TEST(RecursiveASTVisitor,VisitsParensDuringDataRecursion)399 TEST(RecursiveASTVisitor, VisitsParensDuringDataRecursion) {
400   ParenExprVisitor Visitor;
401   Visitor.ExpectMatch("", 1, 9);
402   EXPECT_TRUE(Visitor.runOver("int k = (4) + 9;\n"));
403 }
404 
TEST(RecursiveASTVisitor,VisitsClassTemplateNonTypeParmDefaultArgument)405 TEST(RecursiveASTVisitor, VisitsClassTemplateNonTypeParmDefaultArgument) {
406   CXXBoolLiteralExprVisitor Visitor;
407   Visitor.ExpectMatch("true", 2, 19);
408   EXPECT_TRUE(Visitor.runOver(
409     "template<bool B> class X;\n"
410     "template<bool B = true> class Y;\n"
411     "template<bool B> class Y {};\n"));
412 }
413 
TEST(RecursiveASTVisitor,VisitsClassTemplateTypeParmDefaultArgument)414 TEST(RecursiveASTVisitor, VisitsClassTemplateTypeParmDefaultArgument) {
415   TypeLocVisitor Visitor;
416   Visitor.ExpectMatch("class X", 2, 23);
417   EXPECT_TRUE(Visitor.runOver(
418     "class X;\n"
419     "template<typename T = X> class Y;\n"
420     "template<typename T> class Y {};\n"));
421 }
422 
TEST(RecursiveASTVisitor,VisitsClassTemplateTemplateParmDefaultArgument)423 TEST(RecursiveASTVisitor, VisitsClassTemplateTemplateParmDefaultArgument) {
424   TemplateArgumentLocTraverser Visitor;
425   Visitor.ExpectMatch("X", 2, 40);
426   EXPECT_TRUE(Visitor.runOver(
427     "template<typename T> class X;\n"
428     "template<template <typename> class T = X> class Y;\n"
429     "template<template <typename> class T> class Y {};\n"));
430 }
431 
432 // A visitor that visits implicit declarations and matches constructors.
433 class ImplicitCtorVisitor
434     : public ExpectedLocationVisitor<ImplicitCtorVisitor> {
435 public:
shouldVisitImplicitCode() const436   bool shouldVisitImplicitCode() const { return true; }
437 
VisitCXXConstructorDecl(CXXConstructorDecl * Ctor)438   bool VisitCXXConstructorDecl(CXXConstructorDecl* Ctor) {
439     if (Ctor->isImplicit()) {  // Was not written in source code
440       if (const CXXRecordDecl* Class = Ctor->getParent()) {
441         Match(Class->getName(), Ctor->getLocation());
442       }
443     }
444     return true;
445   }
446 };
447 
TEST(RecursiveASTVisitor,VisitsImplicitCopyConstructors)448 TEST(RecursiveASTVisitor, VisitsImplicitCopyConstructors) {
449   ImplicitCtorVisitor Visitor;
450   Visitor.ExpectMatch("Simple", 2, 8);
451   // Note: Clang lazily instantiates implicit declarations, so we need
452   // to use them in order to force them to appear in the AST.
453   EXPECT_TRUE(Visitor.runOver(
454       "struct WithCtor { WithCtor(); }; \n"
455       "struct Simple { Simple(); WithCtor w; }; \n"
456       "int main() { Simple s; Simple t(s); }\n"));
457 }
458 
459 /// \brief A visitor that optionally includes implicit code and matches
460 /// CXXConstructExpr.
461 ///
462 /// The name recorded for the match is the name of the class whose constructor
463 /// is invoked by the CXXConstructExpr, not the name of the class whose
464 /// constructor the CXXConstructExpr is contained in.
465 class ConstructExprVisitor
466     : public ExpectedLocationVisitor<ConstructExprVisitor> {
467 public:
ConstructExprVisitor()468   ConstructExprVisitor() : ShouldVisitImplicitCode(false) {}
469 
shouldVisitImplicitCode() const470   bool shouldVisitImplicitCode() const { return ShouldVisitImplicitCode; }
471 
setShouldVisitImplicitCode(bool NewValue)472   void setShouldVisitImplicitCode(bool NewValue) {
473     ShouldVisitImplicitCode = NewValue;
474   }
475 
VisitCXXConstructExpr(CXXConstructExpr * Expr)476   bool VisitCXXConstructExpr(CXXConstructExpr* Expr) {
477     if (const CXXConstructorDecl* Ctor = Expr->getConstructor()) {
478       if (const CXXRecordDecl* Class = Ctor->getParent()) {
479         Match(Class->getName(), Expr->getLocation());
480       }
481     }
482     return true;
483   }
484 
485  private:
486   bool ShouldVisitImplicitCode;
487 };
488 
TEST(RecursiveASTVisitor,CanVisitImplicitMemberInitializations)489 TEST(RecursiveASTVisitor, CanVisitImplicitMemberInitializations) {
490   ConstructExprVisitor Visitor;
491   Visitor.setShouldVisitImplicitCode(true);
492   Visitor.ExpectMatch("WithCtor", 2, 8);
493   // Simple has a constructor that implicitly initializes 'w'.  Test
494   // that a visitor that visits implicit code visits that initialization.
495   // Note: Clang lazily instantiates implicit declarations, so we need
496   // to use them in order to force them to appear in the AST.
497   EXPECT_TRUE(Visitor.runOver(
498       "struct WithCtor { WithCtor(); }; \n"
499       "struct Simple { WithCtor w; }; \n"
500       "int main() { Simple s; }\n"));
501 }
502 
503 // The same as CanVisitImplicitMemberInitializations, but checking that the
504 // visits are omitted when the visitor does not include implicit code.
TEST(RecursiveASTVisitor,CanSkipImplicitMemberInitializations)505 TEST(RecursiveASTVisitor, CanSkipImplicitMemberInitializations) {
506   ConstructExprVisitor Visitor;
507   Visitor.setShouldVisitImplicitCode(false);
508   Visitor.DisallowMatch("WithCtor", 2, 8);
509   // Simple has a constructor that implicitly initializes 'w'.  Test
510   // that a visitor that skips implicit code skips that initialization.
511   // Note: Clang lazily instantiates implicit declarations, so we need
512   // to use them in order to force them to appear in the AST.
513   EXPECT_TRUE(Visitor.runOver(
514       "struct WithCtor { WithCtor(); }; \n"
515       "struct Simple { WithCtor w; }; \n"
516       "int main() { Simple s; }\n"));
517 }
518 
TEST(RecursiveASTVisitor,VisitsExtension)519 TEST(RecursiveASTVisitor, VisitsExtension) {
520   DeclRefExprVisitor Visitor;
521   Visitor.ExpectMatch("s", 1, 24);
522   EXPECT_TRUE(Visitor.runOver(
523     "int s = __extension__ (s);\n"));
524 }
525 
TEST(RecursiveASTVisitor,VisitsCompoundLiteralType)526 TEST(RecursiveASTVisitor, VisitsCompoundLiteralType) {
527   TypeLocVisitor Visitor;
528   Visitor.ExpectMatch("struct S", 1, 26);
529   EXPECT_TRUE(Visitor.runOver(
530       "int f() { return (struct S { int a; }){.a = 0}.a; }",
531       TypeLocVisitor::Lang_C));
532 }
533 
TEST(RecursiveASTVisitor,VisitsObjCPropertyType)534 TEST(RecursiveASTVisitor, VisitsObjCPropertyType) {
535   TypeLocVisitor Visitor;
536   Visitor.ExpectMatch("NSNumber", 2, 33);
537   EXPECT_TRUE(Visitor.runOver(
538       "@class NSNumber; \n"
539       "@interface A @property (retain) NSNumber *x; @end\n",
540       TypeLocVisitor::Lang_OBJC));
541 }
542 
TEST(RecursiveASTVisitor,VisitsLambdaExpr)543 TEST(RecursiveASTVisitor, VisitsLambdaExpr) {
544   LambdaExprVisitor Visitor;
545   Visitor.ExpectMatch("", 1, 12);
546   EXPECT_TRUE(Visitor.runOver("void f() { []{ return; }(); }",
547 			      LambdaExprVisitor::Lang_CXX11));
548 }
549 
TEST(RecursiveASTVisitor,TraverseLambdaBodyCanBeOverridden)550 TEST(RecursiveASTVisitor, TraverseLambdaBodyCanBeOverridden) {
551   LambdaExprVisitor Visitor;
552   EXPECT_TRUE(Visitor.runOver("void f() { []{ return; }(); }",
553 			      LambdaExprVisitor::Lang_CXX11));
554   EXPECT_TRUE(Visitor.allBodiesHaveBeenTraversed());
555 }
556 
TEST(RecursiveASTVisitor,HasCaptureDefaultLoc)557 TEST(RecursiveASTVisitor, HasCaptureDefaultLoc) {
558   LambdaDefaultCaptureVisitor Visitor;
559   Visitor.ExpectMatch("", 1, 20);
560   EXPECT_TRUE(Visitor.runOver("void f() { int a; [=]{a;}; }",
561                               LambdaDefaultCaptureVisitor::Lang_CXX11));
562 }
563 
TEST(RecursiveASTVisitor,VisitsCopyExprOfBlockDeclCapture)564 TEST(RecursiveASTVisitor, VisitsCopyExprOfBlockDeclCapture) {
565   DeclRefExprVisitor Visitor;
566   Visitor.ExpectMatch("x", 3, 24);
567   EXPECT_TRUE(Visitor.runOver("void f(int(^)(int)); \n"
568                               "void g() { \n"
569                               "  f([&](int x){ return x; }); \n"
570                               "}",
571                               DeclRefExprVisitor::Lang_OBJCXX11));
572 }
573 
574 // Checks for lambda classes that are not marked as implicitly-generated.
575 // (There should be none.)
576 class ClassVisitor : public ExpectedLocationVisitor<ClassVisitor> {
577 public:
ClassVisitor()578   ClassVisitor() : SawNonImplicitLambdaClass(false) {}
VisitCXXRecordDecl(CXXRecordDecl * record)579   bool VisitCXXRecordDecl(CXXRecordDecl* record) {
580     if (record->isLambda() && !record->isImplicit())
581       SawNonImplicitLambdaClass = true;
582     return true;
583   }
584 
sawOnlyImplicitLambdaClasses() const585   bool sawOnlyImplicitLambdaClasses() const {
586     return !SawNonImplicitLambdaClass;
587   }
588 
589 private:
590   bool SawNonImplicitLambdaClass;
591 };
592 
TEST(RecursiveASTVisitor,LambdaClosureTypesAreImplicit)593 TEST(RecursiveASTVisitor, LambdaClosureTypesAreImplicit) {
594   ClassVisitor Visitor;
595   EXPECT_TRUE(Visitor.runOver("auto lambda = []{};",
596 			      ClassVisitor::Lang_CXX11));
597   EXPECT_TRUE(Visitor.sawOnlyImplicitLambdaClasses());
598 }
599 
600 
601 
602 // Check to ensure that attributes and expressions within them are being
603 // visited.
604 class AttrVisitor : public ExpectedLocationVisitor<AttrVisitor> {
605 public:
VisitMemberExpr(MemberExpr * ME)606   bool VisitMemberExpr(MemberExpr *ME) {
607     Match(ME->getMemberDecl()->getNameAsString(), ME->getLocStart());
608     return true;
609   }
VisitAttr(Attr * A)610   bool VisitAttr(Attr *A) {
611     Match("Attr", A->getLocation());
612     return true;
613   }
VisitGuardedByAttr(GuardedByAttr * A)614   bool VisitGuardedByAttr(GuardedByAttr *A) {
615     Match("guarded_by", A->getLocation());
616     return true;
617   }
618 };
619 
620 
TEST(RecursiveASTVisitor,AttributesAreVisited)621 TEST(RecursiveASTVisitor, AttributesAreVisited) {
622   AttrVisitor Visitor;
623   Visitor.ExpectMatch("Attr", 4, 24);
624   Visitor.ExpectMatch("guarded_by", 4, 24);
625   Visitor.ExpectMatch("mu1",  4, 35);
626   Visitor.ExpectMatch("Attr", 5, 29);
627   Visitor.ExpectMatch("mu1",  5, 54);
628   Visitor.ExpectMatch("mu2",  5, 59);
629   EXPECT_TRUE(Visitor.runOver(
630     "class Foo {\n"
631     "  int mu1;\n"
632     "  int mu2;\n"
633     "  int a __attribute__((guarded_by(mu1)));\n"
634     "  void bar() __attribute__((exclusive_locks_required(mu1, mu2)));\n"
635     "};\n"));
636 }
637 
638 } // end anonymous namespace
639