1*89a1d03eSRichard // RUN: %check_clang_tidy %s llvm-prefer-isa-or-dyn-cast-in-conditionals %t
2*89a1d03eSRichard 
3*89a1d03eSRichard struct X;
4*89a1d03eSRichard struct Y;
5*89a1d03eSRichard struct Z {
6*89a1d03eSRichard   int foo();
7*89a1d03eSRichard   X *bar();
8*89a1d03eSRichard   X *cast(Y*);
9*89a1d03eSRichard   bool baz(Y*);
10*89a1d03eSRichard };
11*89a1d03eSRichard 
12*89a1d03eSRichard template <class X, class Y>
13*89a1d03eSRichard bool isa(Y *);
14*89a1d03eSRichard template <class X, class Y>
15*89a1d03eSRichard X *cast(Y *);
16*89a1d03eSRichard template <class X, class Y>
17*89a1d03eSRichard X *dyn_cast(Y *);
18*89a1d03eSRichard template <class X, class Y>
19*89a1d03eSRichard X *dyn_cast_or_null(Y *);
20*89a1d03eSRichard 
foo(Y * y,Z * z)21*89a1d03eSRichard bool foo(Y *y, Z *z) {
22*89a1d03eSRichard   if (auto x = cast<X>(y))
23*89a1d03eSRichard     return true;
24*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-2]]:16: warning: cast<> in conditional will assert rather than return a null pointer [llvm-prefer-isa-or-dyn-cast-in-conditionals]
25*89a1d03eSRichard   // CHECK-FIXES: if (auto x = dyn_cast<X>(y))
26*89a1d03eSRichard 
27*89a1d03eSRichard   while (auto x = cast<X>(y))
28*89a1d03eSRichard     break;
29*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-2]]:19: warning: cast<> in conditional
30*89a1d03eSRichard   // CHECK-FIXES: while (auto x = dyn_cast<X>(y))
31*89a1d03eSRichard 
32*89a1d03eSRichard   if (cast<X>(y))
33*89a1d03eSRichard     return true;
34*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-2]]:7: warning: cast<> in conditional
35*89a1d03eSRichard   // CHECK-FIXES: if (isa<X>(y))
36*89a1d03eSRichard 
37*89a1d03eSRichard   while (cast<X>(y))
38*89a1d03eSRichard     break;
39*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-2]]:10: warning: cast<> in conditional
40*89a1d03eSRichard   // CHECK-FIXES: while (isa<X>(y))
41*89a1d03eSRichard 
42*89a1d03eSRichard   do {
43*89a1d03eSRichard     break;
44*89a1d03eSRichard   } while (cast<X>(y));
45*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-1]]:12: warning: cast<> in conditional
46*89a1d03eSRichard   // CHECK-FIXES: while (isa<X>(y));
47*89a1d03eSRichard 
48*89a1d03eSRichard   if (dyn_cast<X>(y))
49*89a1d03eSRichard     return true;
50*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-2]]:7: warning: return value from dyn_cast<> not used [llvm-prefer-isa-or-dyn-cast-in-conditionals]
51*89a1d03eSRichard   // CHECK-FIXES: if (isa<X>(y))
52*89a1d03eSRichard 
53*89a1d03eSRichard   while (dyn_cast<X>(y))
54*89a1d03eSRichard     break;
55*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-2]]:10: warning: return value from dyn_cast<> not used
56*89a1d03eSRichard   // CHECK-FIXES: while (isa<X>(y))
57*89a1d03eSRichard 
58*89a1d03eSRichard   do {
59*89a1d03eSRichard     break;
60*89a1d03eSRichard   } while (dyn_cast<X>(y));
61*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-1]]:12: warning: return value from dyn_cast<> not used
62*89a1d03eSRichard   // CHECK-FIXES: while (isa<X>(y));
63*89a1d03eSRichard 
64*89a1d03eSRichard   if (y && isa<X>(y))
65*89a1d03eSRichard     return true;
66*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-2]]:7: warning: isa_and_nonnull<> is preferred over an explicit test for null followed by calling isa<> [llvm-prefer-isa-or-dyn-cast-in-conditionals]
67*89a1d03eSRichard   // CHECK-FIXES: if (isa_and_nonnull<X>(y))
68*89a1d03eSRichard 
69*89a1d03eSRichard   if (z->bar() && isa<Y>(z->bar()))
70*89a1d03eSRichard     return true;
71*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-2]]:7: warning:  isa_and_nonnull<> is preferred
72*89a1d03eSRichard   // CHECK-FIXES: if (isa_and_nonnull<Y>(z->bar()))
73*89a1d03eSRichard 
74*89a1d03eSRichard   if (z->bar() && cast<Y>(z->bar()))
75*89a1d03eSRichard     return true;
76*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-2]]:7: warning: isa_and_nonnull<> is preferred
77*89a1d03eSRichard   // CHECK-FIXES: if (isa_and_nonnull<Y>(z->bar()))
78*89a1d03eSRichard 
79*89a1d03eSRichard   if (z->bar() && dyn_cast<Y>(z->bar()))
80*89a1d03eSRichard     return true;
81*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-2]]:7: warning: isa_and_nonnull<> is preferred
82*89a1d03eSRichard   // CHECK-FIXES: if (isa_and_nonnull<Y>(z->bar()))
83*89a1d03eSRichard 
84*89a1d03eSRichard   if (z->bar() && dyn_cast_or_null<Y>(z->bar()))
85*89a1d03eSRichard     return true;
86*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-2]]:7: warning: isa_and_nonnull<> is preferred
87*89a1d03eSRichard   // CHECK-FIXES: if (isa_and_nonnull<Y>(z->bar()))
88*89a1d03eSRichard 
89*89a1d03eSRichard   bool b = z->bar() && cast<Y>(z->bar());
90*89a1d03eSRichard   // CHECK-MESSAGES: :[[@LINE-1]]:12: warning: isa_and_nonnull<> is preferred
91*89a1d03eSRichard   // CHECK-FIXES: bool b = isa_and_nonnull<Y>(z->bar());
92*89a1d03eSRichard 
93*89a1d03eSRichard   // These don't trigger a warning.
94*89a1d03eSRichard   if (auto x = cast<Z>(y)->foo())
95*89a1d03eSRichard     return true;
96*89a1d03eSRichard   if (auto x = z->cast(y))
97*89a1d03eSRichard     return true;
98*89a1d03eSRichard   while (auto x = cast<Z>(y)->foo())
99*89a1d03eSRichard     break;
100*89a1d03eSRichard   if (cast<Z>(y)->foo())
101*89a1d03eSRichard     return true;
102*89a1d03eSRichard   if (z->cast(y))
103*89a1d03eSRichard     return true;
104*89a1d03eSRichard   while (cast<Z>(y)->foo())
105*89a1d03eSRichard     break;
106*89a1d03eSRichard   if (y && cast<X>(z->bar()))
107*89a1d03eSRichard     return true;
108*89a1d03eSRichard   if (z && cast<Z>(y)->foo())
109*89a1d03eSRichard     return true;
110*89a1d03eSRichard   bool b2 = y && cast<X>(z);
111*89a1d03eSRichard   if(z->cast(y))
112*89a1d03eSRichard     return true;
113*89a1d03eSRichard   if (z->baz(cast<Y>(z)))
114*89a1d03eSRichard     return true;
115*89a1d03eSRichard 
116*89a1d03eSRichard #define CAST(T, Obj) cast<T>(Obj)
117*89a1d03eSRichard #define AUTO_VAR_CAST(X, Y, Z) auto X = cast<Y>(Z)
118*89a1d03eSRichard #define ISA(T, Obj) isa<T>(Obj)
119*89a1d03eSRichard #define ISA_OR_NULL(T, Obj) Obj &&isa<T>(Obj)
120*89a1d03eSRichard 
121*89a1d03eSRichard   // Macros don't trigger warning.
122*89a1d03eSRichard   if (auto x = CAST(X, y))
123*89a1d03eSRichard     return true;
124*89a1d03eSRichard   if (AUTO_VAR_CAST(x, X, z))
125*89a1d03eSRichard     return true;
126*89a1d03eSRichard   if (z->bar() && ISA(Y, z->bar()))
127*89a1d03eSRichard     return true;
128*89a1d03eSRichard   if (ISA_OR_NULL(Y, z->bar()))
129*89a1d03eSRichard     return true;
130*89a1d03eSRichard 
131*89a1d03eSRichard   return false;
132*89a1d03eSRichard }
133