1 //===--------------------- SemaLookup.cpp - Name Lookup  ------------------===//
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 //  This file implements name lookup for C, C++, Objective-C, and
11 //  Objective-C++.
12 //
13 //===----------------------------------------------------------------------===//
14 
15 #include "clang/Sema/Lookup.h"
16 #include "clang/AST/ASTContext.h"
17 #include "clang/AST/ASTMutationListener.h"
18 #include "clang/AST/CXXInheritance.h"
19 #include "clang/AST/Decl.h"
20 #include "clang/AST/DeclCXX.h"
21 #include "clang/AST/DeclLookups.h"
22 #include "clang/AST/DeclObjC.h"
23 #include "clang/AST/DeclTemplate.h"
24 #include "clang/AST/Expr.h"
25 #include "clang/AST/ExprCXX.h"
26 #include "clang/Basic/Builtins.h"
27 #include "clang/Basic/LangOptions.h"
28 #include "clang/Lex/HeaderSearch.h"
29 #include "clang/Lex/ModuleLoader.h"
30 #include "clang/Lex/Preprocessor.h"
31 #include "clang/Sema/DeclSpec.h"
32 #include "clang/Sema/ExternalSemaSource.h"
33 #include "clang/Sema/Overload.h"
34 #include "clang/Sema/Scope.h"
35 #include "clang/Sema/ScopeInfo.h"
36 #include "clang/Sema/Sema.h"
37 #include "clang/Sema/SemaInternal.h"
38 #include "clang/Sema/TemplateDeduction.h"
39 #include "clang/Sema/TypoCorrection.h"
40 #include "llvm/ADT/STLExtras.h"
41 #include "llvm/ADT/SetVector.h"
42 #include "llvm/ADT/SmallPtrSet.h"
43 #include "llvm/ADT/StringMap.h"
44 #include "llvm/ADT/TinyPtrVector.h"
45 #include "llvm/ADT/edit_distance.h"
46 #include "llvm/Support/ErrorHandling.h"
47 #include <algorithm>
48 #include <iterator>
49 #include <limits>
50 #include <list>
51 #include <map>
52 #include <set>
53 #include <utility>
54 #include <vector>
55 
56 using namespace clang;
57 using namespace sema;
58 
59 namespace {
60   class UnqualUsingEntry {
61     const DeclContext *Nominated;
62     const DeclContext *CommonAncestor;
63 
64   public:
65     UnqualUsingEntry(const DeclContext *Nominated,
66                      const DeclContext *CommonAncestor)
67       : Nominated(Nominated), CommonAncestor(CommonAncestor) {
68     }
69 
70     const DeclContext *getCommonAncestor() const {
71       return CommonAncestor;
72     }
73 
74     const DeclContext *getNominatedNamespace() const {
75       return Nominated;
76     }
77 
78     // Sort by the pointer value of the common ancestor.
79     struct Comparator {
80       bool operator()(const UnqualUsingEntry &L, const UnqualUsingEntry &R) {
81         return L.getCommonAncestor() < R.getCommonAncestor();
82       }
83 
84       bool operator()(const UnqualUsingEntry &E, const DeclContext *DC) {
85         return E.getCommonAncestor() < DC;
86       }
87 
88       bool operator()(const DeclContext *DC, const UnqualUsingEntry &E) {
89         return DC < E.getCommonAncestor();
90       }
91     };
92   };
93 
94   /// A collection of using directives, as used by C++ unqualified
95   /// lookup.
96   class UnqualUsingDirectiveSet {
97     typedef SmallVector<UnqualUsingEntry, 8> ListTy;
98 
99     ListTy list;
100     llvm::SmallPtrSet<DeclContext*, 8> visited;
101 
102   public:
103     UnqualUsingDirectiveSet() {}
104 
105     void visitScopeChain(Scope *S, Scope *InnermostFileScope) {
106       // C++ [namespace.udir]p1:
107       //   During unqualified name lookup, the names appear as if they
108       //   were declared in the nearest enclosing namespace which contains
109       //   both the using-directive and the nominated namespace.
110       DeclContext *InnermostFileDC = InnermostFileScope->getEntity();
111       assert(InnermostFileDC && InnermostFileDC->isFileContext());
112 
113       for (; S; S = S->getParent()) {
114         // C++ [namespace.udir]p1:
115         //   A using-directive shall not appear in class scope, but may
116         //   appear in namespace scope or in block scope.
117         DeclContext *Ctx = S->getEntity();
118         if (Ctx && Ctx->isFileContext()) {
119           visit(Ctx, Ctx);
120         } else if (!Ctx || Ctx->isFunctionOrMethod()) {
121           for (auto *I : S->using_directives())
122             visit(I, InnermostFileDC);
123         }
124       }
125     }
126 
127     // Visits a context and collect all of its using directives
128     // recursively.  Treats all using directives as if they were
129     // declared in the context.
130     //
131     // A given context is only every visited once, so it is important
132     // that contexts be visited from the inside out in order to get
133     // the effective DCs right.
134     void visit(DeclContext *DC, DeclContext *EffectiveDC) {
135       if (!visited.insert(DC).second)
136         return;
137 
138       addUsingDirectives(DC, EffectiveDC);
139     }
140 
141     // Visits a using directive and collects all of its using
142     // directives recursively.  Treats all using directives as if they
143     // were declared in the effective DC.
144     void visit(UsingDirectiveDecl *UD, DeclContext *EffectiveDC) {
145       DeclContext *NS = UD->getNominatedNamespace();
146       if (!visited.insert(NS).second)
147         return;
148 
149       addUsingDirective(UD, EffectiveDC);
150       addUsingDirectives(NS, EffectiveDC);
151     }
152 
153     // Adds all the using directives in a context (and those nominated
154     // by its using directives, transitively) as if they appeared in
155     // the given effective context.
156     void addUsingDirectives(DeclContext *DC, DeclContext *EffectiveDC) {
157       SmallVector<DeclContext*, 4> queue;
158       while (true) {
159         for (auto UD : DC->using_directives()) {
160           DeclContext *NS = UD->getNominatedNamespace();
161           if (visited.insert(NS).second) {
162             addUsingDirective(UD, EffectiveDC);
163             queue.push_back(NS);
164           }
165         }
166 
167         if (queue.empty())
168           return;
169 
170         DC = queue.pop_back_val();
171       }
172     }
173 
174     // Add a using directive as if it had been declared in the given
175     // context.  This helps implement C++ [namespace.udir]p3:
176     //   The using-directive is transitive: if a scope contains a
177     //   using-directive that nominates a second namespace that itself
178     //   contains using-directives, the effect is as if the
179     //   using-directives from the second namespace also appeared in
180     //   the first.
181     void addUsingDirective(UsingDirectiveDecl *UD, DeclContext *EffectiveDC) {
182       // Find the common ancestor between the effective context and
183       // the nominated namespace.
184       DeclContext *Common = UD->getNominatedNamespace();
185       while (!Common->Encloses(EffectiveDC))
186         Common = Common->getParent();
187       Common = Common->getPrimaryContext();
188 
189       list.push_back(UnqualUsingEntry(UD->getNominatedNamespace(), Common));
190     }
191 
192     void done() {
193       std::sort(list.begin(), list.end(), UnqualUsingEntry::Comparator());
194     }
195 
196     typedef ListTy::const_iterator const_iterator;
197 
198     const_iterator begin() const { return list.begin(); }
199     const_iterator end() const { return list.end(); }
200 
201     llvm::iterator_range<const_iterator>
202     getNamespacesFor(DeclContext *DC) const {
203       return llvm::make_range(std::equal_range(begin(), end(),
204                                                DC->getPrimaryContext(),
205                                                UnqualUsingEntry::Comparator()));
206     }
207   };
208 } // end anonymous namespace
209 
210 // Retrieve the set of identifier namespaces that correspond to a
211 // specific kind of name lookup.
212 static inline unsigned getIDNS(Sema::LookupNameKind NameKind,
213                                bool CPlusPlus,
214                                bool Redeclaration) {
215   unsigned IDNS = 0;
216   switch (NameKind) {
217   case Sema::LookupObjCImplicitSelfParam:
218   case Sema::LookupOrdinaryName:
219   case Sema::LookupRedeclarationWithLinkage:
220   case Sema::LookupLocalFriendName:
221     IDNS = Decl::IDNS_Ordinary;
222     if (CPlusPlus) {
223       IDNS |= Decl::IDNS_Tag | Decl::IDNS_Member | Decl::IDNS_Namespace;
224       if (Redeclaration)
225         IDNS |= Decl::IDNS_TagFriend | Decl::IDNS_OrdinaryFriend;
226     }
227     if (Redeclaration)
228       IDNS |= Decl::IDNS_LocalExtern;
229     break;
230 
231   case Sema::LookupOperatorName:
232     // Operator lookup is its own crazy thing;  it is not the same
233     // as (e.g.) looking up an operator name for redeclaration.
234     assert(!Redeclaration && "cannot do redeclaration operator lookup");
235     IDNS = Decl::IDNS_NonMemberOperator;
236     break;
237 
238   case Sema::LookupTagName:
239     if (CPlusPlus) {
240       IDNS = Decl::IDNS_Type;
241 
242       // When looking for a redeclaration of a tag name, we add:
243       // 1) TagFriend to find undeclared friend decls
244       // 2) Namespace because they can't "overload" with tag decls.
245       // 3) Tag because it includes class templates, which can't
246       //    "overload" with tag decls.
247       if (Redeclaration)
248         IDNS |= Decl::IDNS_Tag | Decl::IDNS_TagFriend | Decl::IDNS_Namespace;
249     } else {
250       IDNS = Decl::IDNS_Tag;
251     }
252     break;
253 
254   case Sema::LookupLabel:
255     IDNS = Decl::IDNS_Label;
256     break;
257 
258   case Sema::LookupMemberName:
259     IDNS = Decl::IDNS_Member;
260     if (CPlusPlus)
261       IDNS |= Decl::IDNS_Tag | Decl::IDNS_Ordinary;
262     break;
263 
264   case Sema::LookupNestedNameSpecifierName:
265     IDNS = Decl::IDNS_Type | Decl::IDNS_Namespace;
266     break;
267 
268   case Sema::LookupNamespaceName:
269     IDNS = Decl::IDNS_Namespace;
270     break;
271 
272   case Sema::LookupUsingDeclName:
273     assert(Redeclaration && "should only be used for redecl lookup");
274     IDNS = Decl::IDNS_Ordinary | Decl::IDNS_Tag | Decl::IDNS_Member |
275            Decl::IDNS_Using | Decl::IDNS_TagFriend | Decl::IDNS_OrdinaryFriend |
276            Decl::IDNS_LocalExtern;
277     break;
278 
279   case Sema::LookupObjCProtocolName:
280     IDNS = Decl::IDNS_ObjCProtocol;
281     break;
282 
283   case Sema::LookupAnyName:
284     IDNS = Decl::IDNS_Ordinary | Decl::IDNS_Tag | Decl::IDNS_Member
285       | Decl::IDNS_Using | Decl::IDNS_Namespace | Decl::IDNS_ObjCProtocol
286       | Decl::IDNS_Type;
287     break;
288   }
289   return IDNS;
290 }
291 
292 void LookupResult::configure() {
293   IDNS = getIDNS(LookupKind, getSema().getLangOpts().CPlusPlus,
294                  isForRedeclaration());
295 
296   // If we're looking for one of the allocation or deallocation
297   // operators, make sure that the implicitly-declared new and delete
298   // operators can be found.
299   switch (NameInfo.getName().getCXXOverloadedOperator()) {
300   case OO_New:
301   case OO_Delete:
302   case OO_Array_New:
303   case OO_Array_Delete:
304     getSema().DeclareGlobalNewDelete();
305     break;
306 
307   default:
308     break;
309   }
310 
311   // Compiler builtins are always visible, regardless of where they end
312   // up being declared.
313   if (IdentifierInfo *Id = NameInfo.getName().getAsIdentifierInfo()) {
314     if (unsigned BuiltinID = Id->getBuiltinID()) {
315       if (!getSema().Context.BuiltinInfo.isPredefinedLibFunction(BuiltinID))
316         AllowHidden = true;
317     }
318   }
319 }
320 
321 bool LookupResult::sanity() const {
322   // This function is never called by NDEBUG builds.
323   assert(ResultKind != NotFound || Decls.size() == 0);
324   assert(ResultKind != Found || Decls.size() == 1);
325   assert(ResultKind != FoundOverloaded || Decls.size() > 1 ||
326          (Decls.size() == 1 &&
327           isa<FunctionTemplateDecl>((*begin())->getUnderlyingDecl())));
328   assert(ResultKind != FoundUnresolvedValue || sanityCheckUnresolved());
329   assert(ResultKind != Ambiguous || Decls.size() > 1 ||
330          (Decls.size() == 1 && (Ambiguity == AmbiguousBaseSubobjects ||
331                                 Ambiguity == AmbiguousBaseSubobjectTypes)));
332   assert((Paths != nullptr) == (ResultKind == Ambiguous &&
333                                 (Ambiguity == AmbiguousBaseSubobjectTypes ||
334                                  Ambiguity == AmbiguousBaseSubobjects)));
335   return true;
336 }
337 
338 // Necessary because CXXBasePaths is not complete in Sema.h
339 void LookupResult::deletePaths(CXXBasePaths *Paths) {
340   delete Paths;
341 }
342 
343 /// Get a representative context for a declaration such that two declarations
344 /// will have the same context if they were found within the same scope.
345 static DeclContext *getContextForScopeMatching(Decl *D) {
346   // For function-local declarations, use that function as the context. This
347   // doesn't account for scopes within the function; the caller must deal with
348   // those.
349   DeclContext *DC = D->getLexicalDeclContext();
350   if (DC->isFunctionOrMethod())
351     return DC;
352 
353   // Otherwise, look at the semantic context of the declaration. The
354   // declaration must have been found there.
355   return D->getDeclContext()->getRedeclContext();
356 }
357 
358 /// \brief Determine whether \p D is a better lookup result than \p Existing,
359 /// given that they declare the same entity.
360 static bool isPreferredLookupResult(Sema &S, Sema::LookupNameKind Kind,
361                                     NamedDecl *D, NamedDecl *Existing) {
362   // When looking up redeclarations of a using declaration, prefer a using
363   // shadow declaration over any other declaration of the same entity.
364   if (Kind == Sema::LookupUsingDeclName && isa<UsingShadowDecl>(D) &&
365       !isa<UsingShadowDecl>(Existing))
366     return true;
367 
368   auto *DUnderlying = D->getUnderlyingDecl();
369   auto *EUnderlying = Existing->getUnderlyingDecl();
370 
371   // If they have different underlying declarations, prefer a typedef over the
372   // original type (this happens when two type declarations denote the same
373   // type), per a generous reading of C++ [dcl.typedef]p3 and p4. The typedef
374   // might carry additional semantic information, such as an alignment override.
375   // However, per C++ [dcl.typedef]p5, when looking up a tag name, prefer a tag
376   // declaration over a typedef.
377   if (DUnderlying->getCanonicalDecl() != EUnderlying->getCanonicalDecl()) {
378     assert(isa<TypeDecl>(DUnderlying) && isa<TypeDecl>(EUnderlying));
379     bool HaveTag = isa<TagDecl>(EUnderlying);
380     bool WantTag = Kind == Sema::LookupTagName;
381     return HaveTag != WantTag;
382   }
383 
384   // Pick the function with more default arguments.
385   // FIXME: In the presence of ambiguous default arguments, we should keep both,
386   //        so we can diagnose the ambiguity if the default argument is needed.
387   //        See C++ [over.match.best]p3.
388   if (auto *DFD = dyn_cast<FunctionDecl>(DUnderlying)) {
389     auto *EFD = cast<FunctionDecl>(EUnderlying);
390     unsigned DMin = DFD->getMinRequiredArguments();
391     unsigned EMin = EFD->getMinRequiredArguments();
392     // If D has more default arguments, it is preferred.
393     if (DMin != EMin)
394       return DMin < EMin;
395     // FIXME: When we track visibility for default function arguments, check
396     // that we pick the declaration with more visible default arguments.
397   }
398 
399   // Pick the template with more default template arguments.
400   if (auto *DTD = dyn_cast<TemplateDecl>(DUnderlying)) {
401     auto *ETD = cast<TemplateDecl>(EUnderlying);
402     unsigned DMin = DTD->getTemplateParameters()->getMinRequiredArguments();
403     unsigned EMin = ETD->getTemplateParameters()->getMinRequiredArguments();
404     // If D has more default arguments, it is preferred. Note that default
405     // arguments (and their visibility) is monotonically increasing across the
406     // redeclaration chain, so this is a quick proxy for "is more recent".
407     if (DMin != EMin)
408       return DMin < EMin;
409     // If D has more *visible* default arguments, it is preferred. Note, an
410     // earlier default argument being visible does not imply that a later
411     // default argument is visible, so we can't just check the first one.
412     for (unsigned I = DMin, N = DTD->getTemplateParameters()->size();
413         I != N; ++I) {
414       if (!S.hasVisibleDefaultArgument(
415               ETD->getTemplateParameters()->getParam(I)) &&
416           S.hasVisibleDefaultArgument(
417               DTD->getTemplateParameters()->getParam(I)))
418         return true;
419     }
420   }
421 
422   // For most kinds of declaration, it doesn't really matter which one we pick.
423   if (!isa<FunctionDecl>(DUnderlying) && !isa<VarDecl>(DUnderlying)) {
424     // If the existing declaration is hidden, prefer the new one. Otherwise,
425     // keep what we've got.
426     return !S.isVisible(Existing);
427   }
428 
429   // Pick the newer declaration; it might have a more precise type.
430   for (Decl *Prev = DUnderlying->getPreviousDecl(); Prev;
431        Prev = Prev->getPreviousDecl())
432     if (Prev == EUnderlying)
433       return true;
434   return false;
435 }
436 
437 /// Determine whether \p D can hide a tag declaration.
438 static bool canHideTag(NamedDecl *D) {
439   // C++ [basic.scope.declarative]p4:
440   //   Given a set of declarations in a single declarative region [...]
441   //   exactly one declaration shall declare a class name or enumeration name
442   //   that is not a typedef name and the other declarations shall all refer to
443   //   the same variable or enumerator, or all refer to functions and function
444   //   templates; in this case the class name or enumeration name is hidden.
445   // C++ [basic.scope.hiding]p2:
446   //   A class name or enumeration name can be hidden by the name of a
447   //   variable, data member, function, or enumerator declared in the same
448   //   scope.
449   D = D->getUnderlyingDecl();
450   return isa<VarDecl>(D) || isa<EnumConstantDecl>(D) || isa<FunctionDecl>(D) ||
451          isa<FunctionTemplateDecl>(D) || isa<FieldDecl>(D);
452 }
453 
454 /// Resolves the result kind of this lookup.
455 void LookupResult::resolveKind() {
456   unsigned N = Decls.size();
457 
458   // Fast case: no possible ambiguity.
459   if (N == 0) {
460     assert(ResultKind == NotFound ||
461            ResultKind == NotFoundInCurrentInstantiation);
462     return;
463   }
464 
465   // If there's a single decl, we need to examine it to decide what
466   // kind of lookup this is.
467   if (N == 1) {
468     NamedDecl *D = (*Decls.begin())->getUnderlyingDecl();
469     if (isa<FunctionTemplateDecl>(D))
470       ResultKind = FoundOverloaded;
471     else if (isa<UnresolvedUsingValueDecl>(D))
472       ResultKind = FoundUnresolvedValue;
473     return;
474   }
475 
476   // Don't do any extra resolution if we've already resolved as ambiguous.
477   if (ResultKind == Ambiguous) return;
478 
479   llvm::SmallDenseMap<NamedDecl*, unsigned, 16> Unique;
480   llvm::SmallDenseMap<QualType, unsigned, 16> UniqueTypes;
481 
482   bool Ambiguous = false;
483   bool HasTag = false, HasFunction = false;
484   bool HasFunctionTemplate = false, HasUnresolved = false;
485   NamedDecl *HasNonFunction = nullptr;
486 
487   llvm::SmallVector<NamedDecl*, 4> EquivalentNonFunctions;
488 
489   unsigned UniqueTagIndex = 0;
490 
491   unsigned I = 0;
492   while (I < N) {
493     NamedDecl *D = Decls[I]->getUnderlyingDecl();
494     D = cast<NamedDecl>(D->getCanonicalDecl());
495 
496     // Ignore an invalid declaration unless it's the only one left.
497     if (D->isInvalidDecl() && !(I == 0 && N == 1)) {
498       Decls[I] = Decls[--N];
499       continue;
500     }
501 
502     llvm::Optional<unsigned> ExistingI;
503 
504     // Redeclarations of types via typedef can occur both within a scope
505     // and, through using declarations and directives, across scopes. There is
506     // no ambiguity if they all refer to the same type, so unique based on the
507     // canonical type.
508     if (TypeDecl *TD = dyn_cast<TypeDecl>(D)) {
509       QualType T = getSema().Context.getTypeDeclType(TD);
510       auto UniqueResult = UniqueTypes.insert(
511           std::make_pair(getSema().Context.getCanonicalType(T), I));
512       if (!UniqueResult.second) {
513         // The type is not unique.
514         ExistingI = UniqueResult.first->second;
515       }
516     }
517 
518     // For non-type declarations, check for a prior lookup result naming this
519     // canonical declaration.
520     if (!ExistingI) {
521       auto UniqueResult = Unique.insert(std::make_pair(D, I));
522       if (!UniqueResult.second) {
523         // We've seen this entity before.
524         ExistingI = UniqueResult.first->second;
525       }
526     }
527 
528     if (ExistingI) {
529       // This is not a unique lookup result. Pick one of the results and
530       // discard the other.
531       if (isPreferredLookupResult(getSema(), getLookupKind(), Decls[I],
532                                   Decls[*ExistingI]))
533         Decls[*ExistingI] = Decls[I];
534       Decls[I] = Decls[--N];
535       continue;
536     }
537 
538     // Otherwise, do some decl type analysis and then continue.
539 
540     if (isa<UnresolvedUsingValueDecl>(D)) {
541       HasUnresolved = true;
542     } else if (isa<TagDecl>(D)) {
543       if (HasTag)
544         Ambiguous = true;
545       UniqueTagIndex = I;
546       HasTag = true;
547     } else if (isa<FunctionTemplateDecl>(D)) {
548       HasFunction = true;
549       HasFunctionTemplate = true;
550     } else if (isa<FunctionDecl>(D)) {
551       HasFunction = true;
552     } else {
553       if (HasNonFunction) {
554         // If we're about to create an ambiguity between two declarations that
555         // are equivalent, but one is an internal linkage declaration from one
556         // module and the other is an internal linkage declaration from another
557         // module, just skip it.
558         if (getSema().isEquivalentInternalLinkageDeclaration(HasNonFunction,
559                                                              D)) {
560           EquivalentNonFunctions.push_back(D);
561           Decls[I] = Decls[--N];
562           continue;
563         }
564 
565         Ambiguous = true;
566       }
567       HasNonFunction = D;
568     }
569     I++;
570   }
571 
572   // C++ [basic.scope.hiding]p2:
573   //   A class name or enumeration name can be hidden by the name of
574   //   an object, function, or enumerator declared in the same
575   //   scope. If a class or enumeration name and an object, function,
576   //   or enumerator are declared in the same scope (in any order)
577   //   with the same name, the class or enumeration name is hidden
578   //   wherever the object, function, or enumerator name is visible.
579   // But it's still an error if there are distinct tag types found,
580   // even if they're not visible. (ref?)
581   if (N > 1 && HideTags && HasTag && !Ambiguous &&
582       (HasFunction || HasNonFunction || HasUnresolved)) {
583     NamedDecl *OtherDecl = Decls[UniqueTagIndex ? 0 : N - 1];
584     if (isa<TagDecl>(Decls[UniqueTagIndex]->getUnderlyingDecl()) &&
585         getContextForScopeMatching(Decls[UniqueTagIndex])->Equals(
586             getContextForScopeMatching(OtherDecl)) &&
587         canHideTag(OtherDecl))
588       Decls[UniqueTagIndex] = Decls[--N];
589     else
590       Ambiguous = true;
591   }
592 
593   // FIXME: This diagnostic should really be delayed until we're done with
594   // the lookup result, in case the ambiguity is resolved by the caller.
595   if (!EquivalentNonFunctions.empty() && !Ambiguous)
596     getSema().diagnoseEquivalentInternalLinkageDeclarations(
597         getNameLoc(), HasNonFunction, EquivalentNonFunctions);
598 
599   Decls.set_size(N);
600 
601   if (HasNonFunction && (HasFunction || HasUnresolved))
602     Ambiguous = true;
603 
604   if (Ambiguous)
605     setAmbiguous(LookupResult::AmbiguousReference);
606   else if (HasUnresolved)
607     ResultKind = LookupResult::FoundUnresolvedValue;
608   else if (N > 1 || HasFunctionTemplate)
609     ResultKind = LookupResult::FoundOverloaded;
610   else
611     ResultKind = LookupResult::Found;
612 }
613 
614 void LookupResult::addDeclsFromBasePaths(const CXXBasePaths &P) {
615   CXXBasePaths::const_paths_iterator I, E;
616   for (I = P.begin(), E = P.end(); I != E; ++I)
617     for (DeclContext::lookup_iterator DI = I->Decls.begin(),
618          DE = I->Decls.end(); DI != DE; ++DI)
619       addDecl(*DI);
620 }
621 
622 void LookupResult::setAmbiguousBaseSubobjects(CXXBasePaths &P) {
623   Paths = new CXXBasePaths;
624   Paths->swap(P);
625   addDeclsFromBasePaths(*Paths);
626   resolveKind();
627   setAmbiguous(AmbiguousBaseSubobjects);
628 }
629 
630 void LookupResult::setAmbiguousBaseSubobjectTypes(CXXBasePaths &P) {
631   Paths = new CXXBasePaths;
632   Paths->swap(P);
633   addDeclsFromBasePaths(*Paths);
634   resolveKind();
635   setAmbiguous(AmbiguousBaseSubobjectTypes);
636 }
637 
638 void LookupResult::print(raw_ostream &Out) {
639   Out << Decls.size() << " result(s)";
640   if (isAmbiguous()) Out << ", ambiguous";
641   if (Paths) Out << ", base paths present";
642 
643   for (iterator I = begin(), E = end(); I != E; ++I) {
644     Out << "\n";
645     (*I)->print(Out, 2);
646   }
647 }
648 
649 LLVM_DUMP_METHOD void LookupResult::dump() {
650   llvm::errs() << "lookup results for " << getLookupName().getAsString()
651                << ":\n";
652   for (NamedDecl *D : *this)
653     D->dump();
654 }
655 
656 /// \brief Lookup a builtin function, when name lookup would otherwise
657 /// fail.
658 static bool LookupBuiltin(Sema &S, LookupResult &R) {
659   Sema::LookupNameKind NameKind = R.getLookupKind();
660 
661   // If we didn't find a use of this identifier, and if the identifier
662   // corresponds to a compiler builtin, create the decl object for the builtin
663   // now, injecting it into translation unit scope, and return it.
664   if (NameKind == Sema::LookupOrdinaryName ||
665       NameKind == Sema::LookupRedeclarationWithLinkage) {
666     IdentifierInfo *II = R.getLookupName().getAsIdentifierInfo();
667     if (II) {
668       if (S.getLangOpts().CPlusPlus11 && S.getLangOpts().GNUMode &&
669           II == S.getFloat128Identifier()) {
670         // libstdc++4.7's type_traits expects type __float128 to exist, so
671         // insert a dummy type to make that header build in gnu++11 mode.
672         R.addDecl(S.getASTContext().getFloat128StubType());
673         return true;
674       }
675       if (S.getLangOpts().CPlusPlus && NameKind == Sema::LookupOrdinaryName &&
676           II == S.getASTContext().getMakeIntegerSeqName()) {
677         R.addDecl(S.getASTContext().getMakeIntegerSeqDecl());
678         return true;
679       }
680 
681       // If this is a builtin on this (or all) targets, create the decl.
682       if (unsigned BuiltinID = II->getBuiltinID()) {
683         // In C++ and OpenCL (spec v1.2 s6.9.f), we don't have any predefined
684         // library functions like 'malloc'. Instead, we'll just error.
685         if ((S.getLangOpts().CPlusPlus || S.getLangOpts().OpenCL) &&
686             S.Context.BuiltinInfo.isPredefinedLibFunction(BuiltinID))
687           return false;
688 
689         if (NamedDecl *D = S.LazilyCreateBuiltin((IdentifierInfo *)II,
690                                                  BuiltinID, S.TUScope,
691                                                  R.isForRedeclaration(),
692                                                  R.getNameLoc())) {
693           R.addDecl(D);
694           return true;
695         }
696       }
697     }
698   }
699 
700   return false;
701 }
702 
703 /// \brief Determine whether we can declare a special member function within
704 /// the class at this point.
705 static bool CanDeclareSpecialMemberFunction(const CXXRecordDecl *Class) {
706   // We need to have a definition for the class.
707   if (!Class->getDefinition() || Class->isDependentContext())
708     return false;
709 
710   // We can't be in the middle of defining the class.
711   return !Class->isBeingDefined();
712 }
713 
714 void Sema::ForceDeclarationOfImplicitMembers(CXXRecordDecl *Class) {
715   if (!CanDeclareSpecialMemberFunction(Class))
716     return;
717 
718   // If the default constructor has not yet been declared, do so now.
719   if (Class->needsImplicitDefaultConstructor())
720     DeclareImplicitDefaultConstructor(Class);
721 
722   // If the copy constructor has not yet been declared, do so now.
723   if (Class->needsImplicitCopyConstructor())
724     DeclareImplicitCopyConstructor(Class);
725 
726   // If the copy assignment operator has not yet been declared, do so now.
727   if (Class->needsImplicitCopyAssignment())
728     DeclareImplicitCopyAssignment(Class);
729 
730   if (getLangOpts().CPlusPlus11) {
731     // If the move constructor has not yet been declared, do so now.
732     if (Class->needsImplicitMoveConstructor())
733       DeclareImplicitMoveConstructor(Class); // might not actually do it
734 
735     // If the move assignment operator has not yet been declared, do so now.
736     if (Class->needsImplicitMoveAssignment())
737       DeclareImplicitMoveAssignment(Class); // might not actually do it
738   }
739 
740   // If the destructor has not yet been declared, do so now.
741   if (Class->needsImplicitDestructor())
742     DeclareImplicitDestructor(Class);
743 }
744 
745 /// \brief Determine whether this is the name of an implicitly-declared
746 /// special member function.
747 static bool isImplicitlyDeclaredMemberFunctionName(DeclarationName Name) {
748   switch (Name.getNameKind()) {
749   case DeclarationName::CXXConstructorName:
750   case DeclarationName::CXXDestructorName:
751     return true;
752 
753   case DeclarationName::CXXOperatorName:
754     return Name.getCXXOverloadedOperator() == OO_Equal;
755 
756   default:
757     break;
758   }
759 
760   return false;
761 }
762 
763 /// \brief If there are any implicit member functions with the given name
764 /// that need to be declared in the given declaration context, do so.
765 static void DeclareImplicitMemberFunctionsWithName(Sema &S,
766                                                    DeclarationName Name,
767                                                    const DeclContext *DC) {
768   if (!DC)
769     return;
770 
771   switch (Name.getNameKind()) {
772   case DeclarationName::CXXConstructorName:
773     if (const CXXRecordDecl *Record = dyn_cast<CXXRecordDecl>(DC))
774       if (Record->getDefinition() && CanDeclareSpecialMemberFunction(Record)) {
775         CXXRecordDecl *Class = const_cast<CXXRecordDecl *>(Record);
776         if (Record->needsImplicitDefaultConstructor())
777           S.DeclareImplicitDefaultConstructor(Class);
778         if (Record->needsImplicitCopyConstructor())
779           S.DeclareImplicitCopyConstructor(Class);
780         if (S.getLangOpts().CPlusPlus11 &&
781             Record->needsImplicitMoveConstructor())
782           S.DeclareImplicitMoveConstructor(Class);
783       }
784     break;
785 
786   case DeclarationName::CXXDestructorName:
787     if (const CXXRecordDecl *Record = dyn_cast<CXXRecordDecl>(DC))
788       if (Record->getDefinition() && Record->needsImplicitDestructor() &&
789           CanDeclareSpecialMemberFunction(Record))
790         S.DeclareImplicitDestructor(const_cast<CXXRecordDecl *>(Record));
791     break;
792 
793   case DeclarationName::CXXOperatorName:
794     if (Name.getCXXOverloadedOperator() != OO_Equal)
795       break;
796 
797     if (const CXXRecordDecl *Record = dyn_cast<CXXRecordDecl>(DC)) {
798       if (Record->getDefinition() && CanDeclareSpecialMemberFunction(Record)) {
799         CXXRecordDecl *Class = const_cast<CXXRecordDecl *>(Record);
800         if (Record->needsImplicitCopyAssignment())
801           S.DeclareImplicitCopyAssignment(Class);
802         if (S.getLangOpts().CPlusPlus11 &&
803             Record->needsImplicitMoveAssignment())
804           S.DeclareImplicitMoveAssignment(Class);
805       }
806     }
807     break;
808 
809   default:
810     break;
811   }
812 }
813 
814 // Adds all qualifying matches for a name within a decl context to the
815 // given lookup result.  Returns true if any matches were found.
816 static bool LookupDirect(Sema &S, LookupResult &R, const DeclContext *DC) {
817   bool Found = false;
818 
819   // Lazily declare C++ special member functions.
820   if (S.getLangOpts().CPlusPlus)
821     DeclareImplicitMemberFunctionsWithName(S, R.getLookupName(), DC);
822 
823   // Perform lookup into this declaration context.
824   DeclContext::lookup_result DR = DC->lookup(R.getLookupName());
825   for (DeclContext::lookup_iterator I = DR.begin(), E = DR.end(); I != E;
826        ++I) {
827     NamedDecl *D = *I;
828     if ((D = R.getAcceptableDecl(D))) {
829       R.addDecl(D);
830       Found = true;
831     }
832   }
833 
834   if (!Found && DC->isTranslationUnit() && LookupBuiltin(S, R))
835     return true;
836 
837   if (R.getLookupName().getNameKind()
838         != DeclarationName::CXXConversionFunctionName ||
839       R.getLookupName().getCXXNameType()->isDependentType() ||
840       !isa<CXXRecordDecl>(DC))
841     return Found;
842 
843   // C++ [temp.mem]p6:
844   //   A specialization of a conversion function template is not found by
845   //   name lookup. Instead, any conversion function templates visible in the
846   //   context of the use are considered. [...]
847   const CXXRecordDecl *Record = cast<CXXRecordDecl>(DC);
848   if (!Record->isCompleteDefinition())
849     return Found;
850 
851   for (CXXRecordDecl::conversion_iterator U = Record->conversion_begin(),
852          UEnd = Record->conversion_end(); U != UEnd; ++U) {
853     FunctionTemplateDecl *ConvTemplate = dyn_cast<FunctionTemplateDecl>(*U);
854     if (!ConvTemplate)
855       continue;
856 
857     // When we're performing lookup for the purposes of redeclaration, just
858     // add the conversion function template. When we deduce template
859     // arguments for specializations, we'll end up unifying the return
860     // type of the new declaration with the type of the function template.
861     if (R.isForRedeclaration()) {
862       R.addDecl(ConvTemplate);
863       Found = true;
864       continue;
865     }
866 
867     // C++ [temp.mem]p6:
868     //   [...] For each such operator, if argument deduction succeeds
869     //   (14.9.2.3), the resulting specialization is used as if found by
870     //   name lookup.
871     //
872     // When referencing a conversion function for any purpose other than
873     // a redeclaration (such that we'll be building an expression with the
874     // result), perform template argument deduction and place the
875     // specialization into the result set. We do this to avoid forcing all
876     // callers to perform special deduction for conversion functions.
877     TemplateDeductionInfo Info(R.getNameLoc());
878     FunctionDecl *Specialization = nullptr;
879 
880     const FunctionProtoType *ConvProto
881       = ConvTemplate->getTemplatedDecl()->getType()->getAs<FunctionProtoType>();
882     assert(ConvProto && "Nonsensical conversion function template type");
883 
884     // Compute the type of the function that we would expect the conversion
885     // function to have, if it were to match the name given.
886     // FIXME: Calling convention!
887     FunctionProtoType::ExtProtoInfo EPI = ConvProto->getExtProtoInfo();
888     EPI.ExtInfo = EPI.ExtInfo.withCallingConv(CC_C);
889     EPI.ExceptionSpec = EST_None;
890     QualType ExpectedType
891       = R.getSema().Context.getFunctionType(R.getLookupName().getCXXNameType(),
892                                             None, EPI);
893 
894     // Perform template argument deduction against the type that we would
895     // expect the function to have.
896     if (R.getSema().DeduceTemplateArguments(ConvTemplate, nullptr, ExpectedType,
897                                             Specialization, Info)
898           == Sema::TDK_Success) {
899       R.addDecl(Specialization);
900       Found = true;
901     }
902   }
903 
904   return Found;
905 }
906 
907 // Performs C++ unqualified lookup into the given file context.
908 static bool
909 CppNamespaceLookup(Sema &S, LookupResult &R, ASTContext &Context,
910                    DeclContext *NS, UnqualUsingDirectiveSet &UDirs) {
911 
912   assert(NS && NS->isFileContext() && "CppNamespaceLookup() requires namespace!");
913 
914   // Perform direct name lookup into the LookupCtx.
915   bool Found = LookupDirect(S, R, NS);
916 
917   // Perform direct name lookup into the namespaces nominated by the
918   // using directives whose common ancestor is this namespace.
919   for (const UnqualUsingEntry &UUE : UDirs.getNamespacesFor(NS))
920     if (LookupDirect(S, R, UUE.getNominatedNamespace()))
921       Found = true;
922 
923   R.resolveKind();
924 
925   return Found;
926 }
927 
928 static bool isNamespaceOrTranslationUnitScope(Scope *S) {
929   if (DeclContext *Ctx = S->getEntity())
930     return Ctx->isFileContext();
931   return false;
932 }
933 
934 // Find the next outer declaration context from this scope. This
935 // routine actually returns the semantic outer context, which may
936 // differ from the lexical context (encoded directly in the Scope
937 // stack) when we are parsing a member of a class template. In this
938 // case, the second element of the pair will be true, to indicate that
939 // name lookup should continue searching in this semantic context when
940 // it leaves the current template parameter scope.
941 static std::pair<DeclContext *, bool> findOuterContext(Scope *S) {
942   DeclContext *DC = S->getEntity();
943   DeclContext *Lexical = nullptr;
944   for (Scope *OuterS = S->getParent(); OuterS;
945        OuterS = OuterS->getParent()) {
946     if (OuterS->getEntity()) {
947       Lexical = OuterS->getEntity();
948       break;
949     }
950   }
951 
952   // C++ [temp.local]p8:
953   //   In the definition of a member of a class template that appears
954   //   outside of the namespace containing the class template
955   //   definition, the name of a template-parameter hides the name of
956   //   a member of this namespace.
957   //
958   // Example:
959   //
960   //   namespace N {
961   //     class C { };
962   //
963   //     template<class T> class B {
964   //       void f(T);
965   //     };
966   //   }
967   //
968   //   template<class C> void N::B<C>::f(C) {
969   //     C b;  // C is the template parameter, not N::C
970   //   }
971   //
972   // In this example, the lexical context we return is the
973   // TranslationUnit, while the semantic context is the namespace N.
974   if (!Lexical || !DC || !S->getParent() ||
975       !S->getParent()->isTemplateParamScope())
976     return std::make_pair(Lexical, false);
977 
978   // Find the outermost template parameter scope.
979   // For the example, this is the scope for the template parameters of
980   // template<class C>.
981   Scope *OutermostTemplateScope = S->getParent();
982   while (OutermostTemplateScope->getParent() &&
983          OutermostTemplateScope->getParent()->isTemplateParamScope())
984     OutermostTemplateScope = OutermostTemplateScope->getParent();
985 
986   // Find the namespace context in which the original scope occurs. In
987   // the example, this is namespace N.
988   DeclContext *Semantic = DC;
989   while (!Semantic->isFileContext())
990     Semantic = Semantic->getParent();
991 
992   // Find the declaration context just outside of the template
993   // parameter scope. This is the context in which the template is
994   // being lexically declaration (a namespace context). In the
995   // example, this is the global scope.
996   if (Lexical->isFileContext() && !Lexical->Equals(Semantic) &&
997       Lexical->Encloses(Semantic))
998     return std::make_pair(Semantic, true);
999 
1000   return std::make_pair(Lexical, false);
1001 }
1002 
1003 namespace {
1004 /// An RAII object to specify that we want to find block scope extern
1005 /// declarations.
1006 struct FindLocalExternScope {
1007   FindLocalExternScope(LookupResult &R)
1008       : R(R), OldFindLocalExtern(R.getIdentifierNamespace() &
1009                                  Decl::IDNS_LocalExtern) {
1010     R.setFindLocalExtern(R.getIdentifierNamespace() & Decl::IDNS_Ordinary);
1011   }
1012   void restore() {
1013     R.setFindLocalExtern(OldFindLocalExtern);
1014   }
1015   ~FindLocalExternScope() {
1016     restore();
1017   }
1018   LookupResult &R;
1019   bool OldFindLocalExtern;
1020 };
1021 } // end anonymous namespace
1022 
1023 bool Sema::CppLookupName(LookupResult &R, Scope *S) {
1024   assert(getLangOpts().CPlusPlus && "Can perform only C++ lookup");
1025 
1026   DeclarationName Name = R.getLookupName();
1027   Sema::LookupNameKind NameKind = R.getLookupKind();
1028 
1029   // If this is the name of an implicitly-declared special member function,
1030   // go through the scope stack to implicitly declare
1031   if (isImplicitlyDeclaredMemberFunctionName(Name)) {
1032     for (Scope *PreS = S; PreS; PreS = PreS->getParent())
1033       if (DeclContext *DC = PreS->getEntity())
1034         DeclareImplicitMemberFunctionsWithName(*this, Name, DC);
1035   }
1036 
1037   // Implicitly declare member functions with the name we're looking for, if in
1038   // fact we are in a scope where it matters.
1039 
1040   Scope *Initial = S;
1041   IdentifierResolver::iterator
1042     I = IdResolver.begin(Name),
1043     IEnd = IdResolver.end();
1044 
1045   // First we lookup local scope.
1046   // We don't consider using-directives, as per 7.3.4.p1 [namespace.udir]
1047   // ...During unqualified name lookup (3.4.1), the names appear as if
1048   // they were declared in the nearest enclosing namespace which contains
1049   // both the using-directive and the nominated namespace.
1050   // [Note: in this context, "contains" means "contains directly or
1051   // indirectly".
1052   //
1053   // For example:
1054   // namespace A { int i; }
1055   // void foo() {
1056   //   int i;
1057   //   {
1058   //     using namespace A;
1059   //     ++i; // finds local 'i', A::i appears at global scope
1060   //   }
1061   // }
1062   //
1063   UnqualUsingDirectiveSet UDirs;
1064   bool VisitedUsingDirectives = false;
1065   bool LeftStartingScope = false;
1066   DeclContext *OutsideOfTemplateParamDC = nullptr;
1067 
1068   // When performing a scope lookup, we want to find local extern decls.
1069   FindLocalExternScope FindLocals(R);
1070 
1071   for (; S && !isNamespaceOrTranslationUnitScope(S); S = S->getParent()) {
1072     DeclContext *Ctx = S->getEntity();
1073 
1074     // Check whether the IdResolver has anything in this scope.
1075     bool Found = false;
1076     for (; I != IEnd && S->isDeclScope(*I); ++I) {
1077       if (NamedDecl *ND = R.getAcceptableDecl(*I)) {
1078         if (NameKind == LookupRedeclarationWithLinkage) {
1079           // Determine whether this (or a previous) declaration is
1080           // out-of-scope.
1081           if (!LeftStartingScope && !Initial->isDeclScope(*I))
1082             LeftStartingScope = true;
1083 
1084           // If we found something outside of our starting scope that
1085           // does not have linkage, skip it. If it's a template parameter,
1086           // we still find it, so we can diagnose the invalid redeclaration.
1087           if (LeftStartingScope && !((*I)->hasLinkage()) &&
1088               !(*I)->isTemplateParameter()) {
1089             R.setShadowed();
1090             continue;
1091           }
1092         }
1093 
1094         Found = true;
1095         R.addDecl(ND);
1096       }
1097     }
1098     if (Found) {
1099       R.resolveKind();
1100       if (S->isClassScope())
1101         if (CXXRecordDecl *Record = dyn_cast_or_null<CXXRecordDecl>(Ctx))
1102           R.setNamingClass(Record);
1103       return true;
1104     }
1105 
1106     if (NameKind == LookupLocalFriendName && !S->isClassScope()) {
1107       // C++11 [class.friend]p11:
1108       //   If a friend declaration appears in a local class and the name
1109       //   specified is an unqualified name, a prior declaration is
1110       //   looked up without considering scopes that are outside the
1111       //   innermost enclosing non-class scope.
1112       return false;
1113     }
1114 
1115     if (!Ctx && S->isTemplateParamScope() && OutsideOfTemplateParamDC &&
1116         S->getParent() && !S->getParent()->isTemplateParamScope()) {
1117       // We've just searched the last template parameter scope and
1118       // found nothing, so look into the contexts between the
1119       // lexical and semantic declaration contexts returned by
1120       // findOuterContext(). This implements the name lookup behavior
1121       // of C++ [temp.local]p8.
1122       Ctx = OutsideOfTemplateParamDC;
1123       OutsideOfTemplateParamDC = nullptr;
1124     }
1125 
1126     if (Ctx) {
1127       DeclContext *OuterCtx;
1128       bool SearchAfterTemplateScope;
1129       std::tie(OuterCtx, SearchAfterTemplateScope) = findOuterContext(S);
1130       if (SearchAfterTemplateScope)
1131         OutsideOfTemplateParamDC = OuterCtx;
1132 
1133       for (; Ctx && !Ctx->Equals(OuterCtx); Ctx = Ctx->getLookupParent()) {
1134         // We do not directly look into transparent contexts, since
1135         // those entities will be found in the nearest enclosing
1136         // non-transparent context.
1137         if (Ctx->isTransparentContext())
1138           continue;
1139 
1140         // We do not look directly into function or method contexts,
1141         // since all of the local variables and parameters of the
1142         // function/method are present within the Scope.
1143         if (Ctx->isFunctionOrMethod()) {
1144           // If we have an Objective-C instance method, look for ivars
1145           // in the corresponding interface.
1146           if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(Ctx)) {
1147             if (Method->isInstanceMethod() && Name.getAsIdentifierInfo())
1148               if (ObjCInterfaceDecl *Class = Method->getClassInterface()) {
1149                 ObjCInterfaceDecl *ClassDeclared;
1150                 if (ObjCIvarDecl *Ivar = Class->lookupInstanceVariable(
1151                                                  Name.getAsIdentifierInfo(),
1152                                                              ClassDeclared)) {
1153                   if (NamedDecl *ND = R.getAcceptableDecl(Ivar)) {
1154                     R.addDecl(ND);
1155                     R.resolveKind();
1156                     return true;
1157                   }
1158                 }
1159               }
1160           }
1161 
1162           continue;
1163         }
1164 
1165         // If this is a file context, we need to perform unqualified name
1166         // lookup considering using directives.
1167         if (Ctx->isFileContext()) {
1168           // If we haven't handled using directives yet, do so now.
1169           if (!VisitedUsingDirectives) {
1170             // Add using directives from this context up to the top level.
1171             for (DeclContext *UCtx = Ctx; UCtx; UCtx = UCtx->getParent()) {
1172               if (UCtx->isTransparentContext())
1173                 continue;
1174 
1175               UDirs.visit(UCtx, UCtx);
1176             }
1177 
1178             // Find the innermost file scope, so we can add using directives
1179             // from local scopes.
1180             Scope *InnermostFileScope = S;
1181             while (InnermostFileScope &&
1182                    !isNamespaceOrTranslationUnitScope(InnermostFileScope))
1183               InnermostFileScope = InnermostFileScope->getParent();
1184             UDirs.visitScopeChain(Initial, InnermostFileScope);
1185 
1186             UDirs.done();
1187 
1188             VisitedUsingDirectives = true;
1189           }
1190 
1191           if (CppNamespaceLookup(*this, R, Context, Ctx, UDirs)) {
1192             R.resolveKind();
1193             return true;
1194           }
1195 
1196           continue;
1197         }
1198 
1199         // Perform qualified name lookup into this context.
1200         // FIXME: In some cases, we know that every name that could be found by
1201         // this qualified name lookup will also be on the identifier chain. For
1202         // example, inside a class without any base classes, we never need to
1203         // perform qualified lookup because all of the members are on top of the
1204         // identifier chain.
1205         if (LookupQualifiedName(R, Ctx, /*InUnqualifiedLookup=*/true))
1206           return true;
1207       }
1208     }
1209   }
1210 
1211   // Stop if we ran out of scopes.
1212   // FIXME:  This really, really shouldn't be happening.
1213   if (!S) return false;
1214 
1215   // If we are looking for members, no need to look into global/namespace scope.
1216   if (NameKind == LookupMemberName)
1217     return false;
1218 
1219   // Collect UsingDirectiveDecls in all scopes, and recursively all
1220   // nominated namespaces by those using-directives.
1221   //
1222   // FIXME: Cache this sorted list in Scope structure, and DeclContext, so we
1223   // don't build it for each lookup!
1224   if (!VisitedUsingDirectives) {
1225     UDirs.visitScopeChain(Initial, S);
1226     UDirs.done();
1227   }
1228 
1229   // If we're not performing redeclaration lookup, do not look for local
1230   // extern declarations outside of a function scope.
1231   if (!R.isForRedeclaration())
1232     FindLocals.restore();
1233 
1234   // Lookup namespace scope, and global scope.
1235   // Unqualified name lookup in C++ requires looking into scopes
1236   // that aren't strictly lexical, and therefore we walk through the
1237   // context as well as walking through the scopes.
1238   for (; S; S = S->getParent()) {
1239     // Check whether the IdResolver has anything in this scope.
1240     bool Found = false;
1241     for (; I != IEnd && S->isDeclScope(*I); ++I) {
1242       if (NamedDecl *ND = R.getAcceptableDecl(*I)) {
1243         // We found something.  Look for anything else in our scope
1244         // with this same name and in an acceptable identifier
1245         // namespace, so that we can construct an overload set if we
1246         // need to.
1247         Found = true;
1248         R.addDecl(ND);
1249       }
1250     }
1251 
1252     if (Found && S->isTemplateParamScope()) {
1253       R.resolveKind();
1254       return true;
1255     }
1256 
1257     DeclContext *Ctx = S->getEntity();
1258     if (!Ctx && S->isTemplateParamScope() && OutsideOfTemplateParamDC &&
1259         S->getParent() && !S->getParent()->isTemplateParamScope()) {
1260       // We've just searched the last template parameter scope and
1261       // found nothing, so look into the contexts between the
1262       // lexical and semantic declaration contexts returned by
1263       // findOuterContext(). This implements the name lookup behavior
1264       // of C++ [temp.local]p8.
1265       Ctx = OutsideOfTemplateParamDC;
1266       OutsideOfTemplateParamDC = nullptr;
1267     }
1268 
1269     if (Ctx) {
1270       DeclContext *OuterCtx;
1271       bool SearchAfterTemplateScope;
1272       std::tie(OuterCtx, SearchAfterTemplateScope) = findOuterContext(S);
1273       if (SearchAfterTemplateScope)
1274         OutsideOfTemplateParamDC = OuterCtx;
1275 
1276       for (; Ctx && !Ctx->Equals(OuterCtx); Ctx = Ctx->getLookupParent()) {
1277         // We do not directly look into transparent contexts, since
1278         // those entities will be found in the nearest enclosing
1279         // non-transparent context.
1280         if (Ctx->isTransparentContext())
1281           continue;
1282 
1283         // If we have a context, and it's not a context stashed in the
1284         // template parameter scope for an out-of-line definition, also
1285         // look into that context.
1286         if (!(Found && S && S->isTemplateParamScope())) {
1287           assert(Ctx->isFileContext() &&
1288               "We should have been looking only at file context here already.");
1289 
1290           // Look into context considering using-directives.
1291           if (CppNamespaceLookup(*this, R, Context, Ctx, UDirs))
1292             Found = true;
1293         }
1294 
1295         if (Found) {
1296           R.resolveKind();
1297           return true;
1298         }
1299 
1300         if (R.isForRedeclaration() && !Ctx->isTransparentContext())
1301           return false;
1302       }
1303     }
1304 
1305     if (R.isForRedeclaration() && Ctx && !Ctx->isTransparentContext())
1306       return false;
1307   }
1308 
1309   return !R.empty();
1310 }
1311 
1312 /// \brief Find the declaration that a class temploid member specialization was
1313 /// instantiated from, or the member itself if it is an explicit specialization.
1314 static Decl *getInstantiatedFrom(Decl *D, MemberSpecializationInfo *MSInfo) {
1315   return MSInfo->isExplicitSpecialization() ? D : MSInfo->getInstantiatedFrom();
1316 }
1317 
1318 Module *Sema::getOwningModule(Decl *Entity) {
1319   // If it's imported, grab its owning module.
1320   Module *M = Entity->getImportedOwningModule();
1321   if (M || !isa<NamedDecl>(Entity) || !cast<NamedDecl>(Entity)->isHidden())
1322     return M;
1323   assert(!Entity->isFromASTFile() &&
1324          "hidden entity from AST file has no owning module");
1325 
1326   if (!getLangOpts().ModulesLocalVisibility) {
1327     // If we're not tracking visibility locally, the only way a declaration
1328     // can be hidden and local is if it's hidden because it's parent is (for
1329     // instance, maybe this is a lazily-declared special member of an imported
1330     // class).
1331     auto *Parent = cast<NamedDecl>(Entity->getDeclContext());
1332     assert(Parent->isHidden() && "unexpectedly hidden decl");
1333     return getOwningModule(Parent);
1334   }
1335 
1336   // It's local and hidden; grab or compute its owning module.
1337   M = Entity->getLocalOwningModule();
1338   if (M)
1339     return M;
1340 
1341   if (auto *Containing =
1342           PP.getModuleContainingLocation(Entity->getLocation())) {
1343     M = Containing;
1344   } else if (Entity->isInvalidDecl() || Entity->getLocation().isInvalid()) {
1345     // Don't bother tracking visibility for invalid declarations with broken
1346     // locations.
1347     cast<NamedDecl>(Entity)->setHidden(false);
1348   } else {
1349     // We need to assign a module to an entity that exists outside of any
1350     // module, so that we can hide it from modules that we textually enter.
1351     // Invent a fake module for all such entities.
1352     if (!CachedFakeTopLevelModule) {
1353       CachedFakeTopLevelModule =
1354           PP.getHeaderSearchInfo().getModuleMap().findOrCreateModule(
1355               "<top-level>", nullptr, false, false).first;
1356 
1357       auto &SrcMgr = PP.getSourceManager();
1358       SourceLocation StartLoc =
1359           SrcMgr.getLocForStartOfFile(SrcMgr.getMainFileID());
1360       auto &TopLevel =
1361           VisibleModulesStack.empty() ? VisibleModules : VisibleModulesStack[0];
1362       TopLevel.setVisible(CachedFakeTopLevelModule, StartLoc);
1363     }
1364 
1365     M = CachedFakeTopLevelModule;
1366   }
1367 
1368   if (M)
1369     Entity->setLocalOwningModule(M);
1370   return M;
1371 }
1372 
1373 void Sema::makeMergedDefinitionVisible(NamedDecl *ND, SourceLocation Loc) {
1374   if (auto *M = PP.getModuleContainingLocation(Loc))
1375     Context.mergeDefinitionIntoModule(ND, M);
1376   else
1377     // We're not building a module; just make the definition visible.
1378     ND->setHidden(false);
1379 
1380   // If ND is a template declaration, make the template parameters
1381   // visible too. They're not (necessarily) within a mergeable DeclContext.
1382   if (auto *TD = dyn_cast<TemplateDecl>(ND))
1383     for (auto *Param : *TD->getTemplateParameters())
1384       makeMergedDefinitionVisible(Param, Loc);
1385 }
1386 
1387 /// \brief Find the module in which the given declaration was defined.
1388 static Module *getDefiningModule(Sema &S, Decl *Entity) {
1389   if (FunctionDecl *FD = dyn_cast<FunctionDecl>(Entity)) {
1390     // If this function was instantiated from a template, the defining module is
1391     // the module containing the pattern.
1392     if (FunctionDecl *Pattern = FD->getTemplateInstantiationPattern())
1393       Entity = Pattern;
1394   } else if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(Entity)) {
1395     if (CXXRecordDecl *Pattern = RD->getTemplateInstantiationPattern())
1396       Entity = Pattern;
1397   } else if (EnumDecl *ED = dyn_cast<EnumDecl>(Entity)) {
1398     if (MemberSpecializationInfo *MSInfo = ED->getMemberSpecializationInfo())
1399       Entity = getInstantiatedFrom(ED, MSInfo);
1400   } else if (VarDecl *VD = dyn_cast<VarDecl>(Entity)) {
1401     // FIXME: Map from variable template specializations back to the template.
1402     if (MemberSpecializationInfo *MSInfo = VD->getMemberSpecializationInfo())
1403       Entity = getInstantiatedFrom(VD, MSInfo);
1404   }
1405 
1406   // Walk up to the containing context. That might also have been instantiated
1407   // from a template.
1408   DeclContext *Context = Entity->getDeclContext();
1409   if (Context->isFileContext())
1410     return S.getOwningModule(Entity);
1411   return getDefiningModule(S, cast<Decl>(Context));
1412 }
1413 
1414 llvm::DenseSet<Module*> &Sema::getLookupModules() {
1415   unsigned N = ActiveTemplateInstantiations.size();
1416   for (unsigned I = ActiveTemplateInstantiationLookupModules.size();
1417        I != N; ++I) {
1418     Module *M =
1419         getDefiningModule(*this, ActiveTemplateInstantiations[I].Entity);
1420     if (M && !LookupModulesCache.insert(M).second)
1421       M = nullptr;
1422     ActiveTemplateInstantiationLookupModules.push_back(M);
1423   }
1424   return LookupModulesCache;
1425 }
1426 
1427 bool Sema::hasVisibleMergedDefinition(NamedDecl *Def) {
1428   for (Module *Merged : Context.getModulesWithMergedDefinition(Def))
1429     if (isModuleVisible(Merged))
1430       return true;
1431   return false;
1432 }
1433 
1434 template<typename ParmDecl>
1435 static bool
1436 hasVisibleDefaultArgument(Sema &S, const ParmDecl *D,
1437                           llvm::SmallVectorImpl<Module *> *Modules) {
1438   if (!D->hasDefaultArgument())
1439     return false;
1440 
1441   while (D) {
1442     auto &DefaultArg = D->getDefaultArgStorage();
1443     if (!DefaultArg.isInherited() && S.isVisible(D))
1444       return true;
1445 
1446     if (!DefaultArg.isInherited() && Modules) {
1447       auto *NonConstD = const_cast<ParmDecl*>(D);
1448       Modules->push_back(S.getOwningModule(NonConstD));
1449       const auto &Merged = S.Context.getModulesWithMergedDefinition(NonConstD);
1450       Modules->insert(Modules->end(), Merged.begin(), Merged.end());
1451     }
1452 
1453     // If there was a previous default argument, maybe its parameter is visible.
1454     D = DefaultArg.getInheritedFrom();
1455   }
1456   return false;
1457 }
1458 
1459 bool Sema::hasVisibleDefaultArgument(const NamedDecl *D,
1460                                      llvm::SmallVectorImpl<Module *> *Modules) {
1461   if (auto *P = dyn_cast<TemplateTypeParmDecl>(D))
1462     return ::hasVisibleDefaultArgument(*this, P, Modules);
1463   if (auto *P = dyn_cast<NonTypeTemplateParmDecl>(D))
1464     return ::hasVisibleDefaultArgument(*this, P, Modules);
1465   return ::hasVisibleDefaultArgument(*this, cast<TemplateTemplateParmDecl>(D),
1466                                      Modules);
1467 }
1468 
1469 /// \brief Determine whether a declaration is visible to name lookup.
1470 ///
1471 /// This routine determines whether the declaration D is visible in the current
1472 /// lookup context, taking into account the current template instantiation
1473 /// stack. During template instantiation, a declaration is visible if it is
1474 /// visible from a module containing any entity on the template instantiation
1475 /// path (by instantiating a template, you allow it to see the declarations that
1476 /// your module can see, including those later on in your module).
1477 bool LookupResult::isVisibleSlow(Sema &SemaRef, NamedDecl *D) {
1478   assert(D->isHidden() && "should not call this: not in slow case");
1479   Module *DeclModule = nullptr;
1480 
1481   if (SemaRef.getLangOpts().ModulesLocalVisibility) {
1482     DeclModule = SemaRef.getOwningModule(D);
1483     if (!DeclModule) {
1484       // getOwningModule() may have decided the declaration should not be hidden.
1485       assert(!D->isHidden() && "hidden decl not from a module");
1486       return true;
1487     }
1488 
1489     // If the owning module is visible, and the decl is not module private,
1490     // then the decl is visible too. (Module private is ignored within the same
1491     // top-level module.)
1492     if ((!D->isFromASTFile() || !D->isModulePrivate()) &&
1493         (SemaRef.isModuleVisible(DeclModule) ||
1494          SemaRef.hasVisibleMergedDefinition(D)))
1495       return true;
1496   }
1497 
1498   // If this declaration is not at namespace scope nor module-private,
1499   // then it is visible if its lexical parent has a visible definition.
1500   DeclContext *DC = D->getLexicalDeclContext();
1501   if (!D->isModulePrivate() &&
1502       DC && !DC->isFileContext() && !isa<LinkageSpecDecl>(DC)) {
1503     // For a parameter, check whether our current template declaration's
1504     // lexical context is visible, not whether there's some other visible
1505     // definition of it, because parameters aren't "within" the definition.
1506     if ((D->isTemplateParameter() || isa<ParmVarDecl>(D))
1507             ? isVisible(SemaRef, cast<NamedDecl>(DC))
1508             : SemaRef.hasVisibleDefinition(cast<NamedDecl>(DC))) {
1509       if (SemaRef.ActiveTemplateInstantiations.empty() &&
1510           // FIXME: Do something better in this case.
1511           !SemaRef.getLangOpts().ModulesLocalVisibility) {
1512         // Cache the fact that this declaration is implicitly visible because
1513         // its parent has a visible definition.
1514         D->setHidden(false);
1515       }
1516       return true;
1517     }
1518     return false;
1519   }
1520 
1521   // Find the extra places where we need to look.
1522   llvm::DenseSet<Module*> &LookupModules = SemaRef.getLookupModules();
1523   if (LookupModules.empty())
1524     return false;
1525 
1526   if (!DeclModule) {
1527     DeclModule = SemaRef.getOwningModule(D);
1528     assert(DeclModule && "hidden decl not from a module");
1529   }
1530 
1531   // If our lookup set contains the decl's module, it's visible.
1532   if (LookupModules.count(DeclModule))
1533     return true;
1534 
1535   // If the declaration isn't exported, it's not visible in any other module.
1536   if (D->isModulePrivate())
1537     return false;
1538 
1539   // Check whether DeclModule is transitively exported to an import of
1540   // the lookup set.
1541   return std::any_of(LookupModules.begin(), LookupModules.end(),
1542                      [&](Module *M) { return M->isModuleVisible(DeclModule); });
1543 }
1544 
1545 bool Sema::isVisibleSlow(const NamedDecl *D) {
1546   return LookupResult::isVisible(*this, const_cast<NamedDecl*>(D));
1547 }
1548 
1549 bool Sema::shouldLinkPossiblyHiddenDecl(LookupResult &R, const NamedDecl *New) {
1550   for (auto *D : R) {
1551     if (isVisible(D))
1552       return true;
1553   }
1554   return New->isExternallyVisible();
1555 }
1556 
1557 /// \brief Retrieve the visible declaration corresponding to D, if any.
1558 ///
1559 /// This routine determines whether the declaration D is visible in the current
1560 /// module, with the current imports. If not, it checks whether any
1561 /// redeclaration of D is visible, and if so, returns that declaration.
1562 ///
1563 /// \returns D, or a visible previous declaration of D, whichever is more recent
1564 /// and visible. If no declaration of D is visible, returns null.
1565 static NamedDecl *findAcceptableDecl(Sema &SemaRef, NamedDecl *D) {
1566   assert(!LookupResult::isVisible(SemaRef, D) && "not in slow case");
1567 
1568   for (auto RD : D->redecls()) {
1569     // Don't bother with extra checks if we already know this one isn't visible.
1570     if (RD == D)
1571       continue;
1572 
1573     if (auto ND = dyn_cast<NamedDecl>(RD)) {
1574       // FIXME: This is wrong in the case where the previous declaration is not
1575       // visible in the same scope as D. This needs to be done much more
1576       // carefully.
1577       if (LookupResult::isVisible(SemaRef, ND))
1578         return ND;
1579     }
1580   }
1581 
1582   return nullptr;
1583 }
1584 
1585 NamedDecl *LookupResult::getAcceptableDeclSlow(NamedDecl *D) const {
1586   if (auto *ND = dyn_cast<NamespaceDecl>(D)) {
1587     // Namespaces are a bit of a special case: we expect there to be a lot of
1588     // redeclarations of some namespaces, all declarations of a namespace are
1589     // essentially interchangeable, all declarations are found by name lookup
1590     // if any is, and namespaces are never looked up during template
1591     // instantiation. So we benefit from caching the check in this case, and
1592     // it is correct to do so.
1593     auto *Key = ND->getCanonicalDecl();
1594     if (auto *Acceptable = getSema().VisibleNamespaceCache.lookup(Key))
1595       return Acceptable;
1596     auto *Acceptable =
1597         isVisible(getSema(), Key) ? Key : findAcceptableDecl(getSema(), Key);
1598     if (Acceptable)
1599       getSema().VisibleNamespaceCache.insert(std::make_pair(Key, Acceptable));
1600     return Acceptable;
1601   }
1602 
1603   return findAcceptableDecl(getSema(), D);
1604 }
1605 
1606 /// @brief Perform unqualified name lookup starting from a given
1607 /// scope.
1608 ///
1609 /// Unqualified name lookup (C++ [basic.lookup.unqual], C99 6.2.1) is
1610 /// used to find names within the current scope. For example, 'x' in
1611 /// @code
1612 /// int x;
1613 /// int f() {
1614 ///   return x; // unqualified name look finds 'x' in the global scope
1615 /// }
1616 /// @endcode
1617 ///
1618 /// Different lookup criteria can find different names. For example, a
1619 /// particular scope can have both a struct and a function of the same
1620 /// name, and each can be found by certain lookup criteria. For more
1621 /// information about lookup criteria, see the documentation for the
1622 /// class LookupCriteria.
1623 ///
1624 /// @param S        The scope from which unqualified name lookup will
1625 /// begin. If the lookup criteria permits, name lookup may also search
1626 /// in the parent scopes.
1627 ///
1628 /// @param [in,out] R Specifies the lookup to perform (e.g., the name to
1629 /// look up and the lookup kind), and is updated with the results of lookup
1630 /// including zero or more declarations and possibly additional information
1631 /// used to diagnose ambiguities.
1632 ///
1633 /// @returns \c true if lookup succeeded and false otherwise.
1634 bool Sema::LookupName(LookupResult &R, Scope *S, bool AllowBuiltinCreation) {
1635   DeclarationName Name = R.getLookupName();
1636   if (!Name) return false;
1637 
1638   LookupNameKind NameKind = R.getLookupKind();
1639 
1640   if (!getLangOpts().CPlusPlus) {
1641     // Unqualified name lookup in C/Objective-C is purely lexical, so
1642     // search in the declarations attached to the name.
1643     if (NameKind == Sema::LookupRedeclarationWithLinkage) {
1644       // Find the nearest non-transparent declaration scope.
1645       while (!(S->getFlags() & Scope::DeclScope) ||
1646              (S->getEntity() && S->getEntity()->isTransparentContext()))
1647         S = S->getParent();
1648     }
1649 
1650     // When performing a scope lookup, we want to find local extern decls.
1651     FindLocalExternScope FindLocals(R);
1652 
1653     // Scan up the scope chain looking for a decl that matches this
1654     // identifier that is in the appropriate namespace.  This search
1655     // should not take long, as shadowing of names is uncommon, and
1656     // deep shadowing is extremely uncommon.
1657     bool LeftStartingScope = false;
1658 
1659     for (IdentifierResolver::iterator I = IdResolver.begin(Name),
1660                                    IEnd = IdResolver.end();
1661          I != IEnd; ++I)
1662       if (NamedDecl *D = R.getAcceptableDecl(*I)) {
1663         if (NameKind == LookupRedeclarationWithLinkage) {
1664           // Determine whether this (or a previous) declaration is
1665           // out-of-scope.
1666           if (!LeftStartingScope && !S->isDeclScope(*I))
1667             LeftStartingScope = true;
1668 
1669           // If we found something outside of our starting scope that
1670           // does not have linkage, skip it.
1671           if (LeftStartingScope && !((*I)->hasLinkage())) {
1672             R.setShadowed();
1673             continue;
1674           }
1675         }
1676         else if (NameKind == LookupObjCImplicitSelfParam &&
1677                  !isa<ImplicitParamDecl>(*I))
1678           continue;
1679 
1680         R.addDecl(D);
1681 
1682         // Check whether there are any other declarations with the same name
1683         // and in the same scope.
1684         if (I != IEnd) {
1685           // Find the scope in which this declaration was declared (if it
1686           // actually exists in a Scope).
1687           while (S && !S->isDeclScope(D))
1688             S = S->getParent();
1689 
1690           // If the scope containing the declaration is the translation unit,
1691           // then we'll need to perform our checks based on the matching
1692           // DeclContexts rather than matching scopes.
1693           if (S && isNamespaceOrTranslationUnitScope(S))
1694             S = nullptr;
1695 
1696           // Compute the DeclContext, if we need it.
1697           DeclContext *DC = nullptr;
1698           if (!S)
1699             DC = (*I)->getDeclContext()->getRedeclContext();
1700 
1701           IdentifierResolver::iterator LastI = I;
1702           for (++LastI; LastI != IEnd; ++LastI) {
1703             if (S) {
1704               // Match based on scope.
1705               if (!S->isDeclScope(*LastI))
1706                 break;
1707             } else {
1708               // Match based on DeclContext.
1709               DeclContext *LastDC
1710                 = (*LastI)->getDeclContext()->getRedeclContext();
1711               if (!LastDC->Equals(DC))
1712                 break;
1713             }
1714 
1715             // If the declaration is in the right namespace and visible, add it.
1716             if (NamedDecl *LastD = R.getAcceptableDecl(*LastI))
1717               R.addDecl(LastD);
1718           }
1719 
1720           R.resolveKind();
1721         }
1722 
1723         return true;
1724       }
1725   } else {
1726     // Perform C++ unqualified name lookup.
1727     if (CppLookupName(R, S))
1728       return true;
1729   }
1730 
1731   // If we didn't find a use of this identifier, and if the identifier
1732   // corresponds to a compiler builtin, create the decl object for the builtin
1733   // now, injecting it into translation unit scope, and return it.
1734   if (AllowBuiltinCreation && LookupBuiltin(*this, R))
1735     return true;
1736 
1737   // If we didn't find a use of this identifier, the ExternalSource
1738   // may be able to handle the situation.
1739   // Note: some lookup failures are expected!
1740   // See e.g. R.isForRedeclaration().
1741   return (ExternalSource && ExternalSource->LookupUnqualified(R, S));
1742 }
1743 
1744 /// @brief Perform qualified name lookup in the namespaces nominated by
1745 /// using directives by the given context.
1746 ///
1747 /// C++98 [namespace.qual]p2:
1748 ///   Given X::m (where X is a user-declared namespace), or given \::m
1749 ///   (where X is the global namespace), let S be the set of all
1750 ///   declarations of m in X and in the transitive closure of all
1751 ///   namespaces nominated by using-directives in X and its used
1752 ///   namespaces, except that using-directives are ignored in any
1753 ///   namespace, including X, directly containing one or more
1754 ///   declarations of m. No namespace is searched more than once in
1755 ///   the lookup of a name. If S is the empty set, the program is
1756 ///   ill-formed. Otherwise, if S has exactly one member, or if the
1757 ///   context of the reference is a using-declaration
1758 ///   (namespace.udecl), S is the required set of declarations of
1759 ///   m. Otherwise if the use of m is not one that allows a unique
1760 ///   declaration to be chosen from S, the program is ill-formed.
1761 ///
1762 /// C++98 [namespace.qual]p5:
1763 ///   During the lookup of a qualified namespace member name, if the
1764 ///   lookup finds more than one declaration of the member, and if one
1765 ///   declaration introduces a class name or enumeration name and the
1766 ///   other declarations either introduce the same object, the same
1767 ///   enumerator or a set of functions, the non-type name hides the
1768 ///   class or enumeration name if and only if the declarations are
1769 ///   from the same namespace; otherwise (the declarations are from
1770 ///   different namespaces), the program is ill-formed.
1771 static bool LookupQualifiedNameInUsingDirectives(Sema &S, LookupResult &R,
1772                                                  DeclContext *StartDC) {
1773   assert(StartDC->isFileContext() && "start context is not a file context");
1774 
1775   DeclContext::udir_range UsingDirectives = StartDC->using_directives();
1776   if (UsingDirectives.begin() == UsingDirectives.end()) return false;
1777 
1778   // We have at least added all these contexts to the queue.
1779   llvm::SmallPtrSet<DeclContext*, 8> Visited;
1780   Visited.insert(StartDC);
1781 
1782   // We have not yet looked into these namespaces, much less added
1783   // their "using-children" to the queue.
1784   SmallVector<NamespaceDecl*, 8> Queue;
1785 
1786   // We have already looked into the initial namespace; seed the queue
1787   // with its using-children.
1788   for (auto *I : UsingDirectives) {
1789     NamespaceDecl *ND = I->getNominatedNamespace()->getOriginalNamespace();
1790     if (Visited.insert(ND).second)
1791       Queue.push_back(ND);
1792   }
1793 
1794   // The easiest way to implement the restriction in [namespace.qual]p5
1795   // is to check whether any of the individual results found a tag
1796   // and, if so, to declare an ambiguity if the final result is not
1797   // a tag.
1798   bool FoundTag = false;
1799   bool FoundNonTag = false;
1800 
1801   LookupResult LocalR(LookupResult::Temporary, R);
1802 
1803   bool Found = false;
1804   while (!Queue.empty()) {
1805     NamespaceDecl *ND = Queue.pop_back_val();
1806 
1807     // We go through some convolutions here to avoid copying results
1808     // between LookupResults.
1809     bool UseLocal = !R.empty();
1810     LookupResult &DirectR = UseLocal ? LocalR : R;
1811     bool FoundDirect = LookupDirect(S, DirectR, ND);
1812 
1813     if (FoundDirect) {
1814       // First do any local hiding.
1815       DirectR.resolveKind();
1816 
1817       // If the local result is a tag, remember that.
1818       if (DirectR.isSingleTagDecl())
1819         FoundTag = true;
1820       else
1821         FoundNonTag = true;
1822 
1823       // Append the local results to the total results if necessary.
1824       if (UseLocal) {
1825         R.addAllDecls(LocalR);
1826         LocalR.clear();
1827       }
1828     }
1829 
1830     // If we find names in this namespace, ignore its using directives.
1831     if (FoundDirect) {
1832       Found = true;
1833       continue;
1834     }
1835 
1836     for (auto I : ND->using_directives()) {
1837       NamespaceDecl *Nom = I->getNominatedNamespace();
1838       if (Visited.insert(Nom).second)
1839         Queue.push_back(Nom);
1840     }
1841   }
1842 
1843   if (Found) {
1844     if (FoundTag && FoundNonTag)
1845       R.setAmbiguousQualifiedTagHiding();
1846     else
1847       R.resolveKind();
1848   }
1849 
1850   return Found;
1851 }
1852 
1853 /// \brief Callback that looks for any member of a class with the given name.
1854 static bool LookupAnyMember(const CXXBaseSpecifier *Specifier,
1855                             CXXBasePath &Path, DeclarationName Name) {
1856   RecordDecl *BaseRecord = Specifier->getType()->getAs<RecordType>()->getDecl();
1857 
1858   Path.Decls = BaseRecord->lookup(Name);
1859   return !Path.Decls.empty();
1860 }
1861 
1862 /// \brief Determine whether the given set of member declarations contains only
1863 /// static members, nested types, and enumerators.
1864 template<typename InputIterator>
1865 static bool HasOnlyStaticMembers(InputIterator First, InputIterator Last) {
1866   Decl *D = (*First)->getUnderlyingDecl();
1867   if (isa<VarDecl>(D) || isa<TypeDecl>(D) || isa<EnumConstantDecl>(D))
1868     return true;
1869 
1870   if (isa<CXXMethodDecl>(D)) {
1871     // Determine whether all of the methods are static.
1872     bool AllMethodsAreStatic = true;
1873     for(; First != Last; ++First) {
1874       D = (*First)->getUnderlyingDecl();
1875 
1876       if (!isa<CXXMethodDecl>(D)) {
1877         assert(isa<TagDecl>(D) && "Non-function must be a tag decl");
1878         break;
1879       }
1880 
1881       if (!cast<CXXMethodDecl>(D)->isStatic()) {
1882         AllMethodsAreStatic = false;
1883         break;
1884       }
1885     }
1886 
1887     if (AllMethodsAreStatic)
1888       return true;
1889   }
1890 
1891   return false;
1892 }
1893 
1894 /// \brief Perform qualified name lookup into a given context.
1895 ///
1896 /// Qualified name lookup (C++ [basic.lookup.qual]) is used to find
1897 /// names when the context of those names is explicit specified, e.g.,
1898 /// "std::vector" or "x->member", or as part of unqualified name lookup.
1899 ///
1900 /// Different lookup criteria can find different names. For example, a
1901 /// particular scope can have both a struct and a function of the same
1902 /// name, and each can be found by certain lookup criteria. For more
1903 /// information about lookup criteria, see the documentation for the
1904 /// class LookupCriteria.
1905 ///
1906 /// \param R captures both the lookup criteria and any lookup results found.
1907 ///
1908 /// \param LookupCtx The context in which qualified name lookup will
1909 /// search. If the lookup criteria permits, name lookup may also search
1910 /// in the parent contexts or (for C++ classes) base classes.
1911 ///
1912 /// \param InUnqualifiedLookup true if this is qualified name lookup that
1913 /// occurs as part of unqualified name lookup.
1914 ///
1915 /// \returns true if lookup succeeded, false if it failed.
1916 bool Sema::LookupQualifiedName(LookupResult &R, DeclContext *LookupCtx,
1917                                bool InUnqualifiedLookup) {
1918   assert(LookupCtx && "Sema::LookupQualifiedName requires a lookup context");
1919 
1920   if (!R.getLookupName())
1921     return false;
1922 
1923   // Make sure that the declaration context is complete.
1924   assert((!isa<TagDecl>(LookupCtx) ||
1925           LookupCtx->isDependentContext() ||
1926           cast<TagDecl>(LookupCtx)->isCompleteDefinition() ||
1927           cast<TagDecl>(LookupCtx)->isBeingDefined()) &&
1928          "Declaration context must already be complete!");
1929 
1930   struct QualifiedLookupInScope {
1931     bool oldVal;
1932     DeclContext *Context;
1933     // Set flag in DeclContext informing debugger that we're looking for qualified name
1934     QualifiedLookupInScope(DeclContext *ctx) : Context(ctx) {
1935       oldVal = ctx->setUseQualifiedLookup();
1936     }
1937     ~QualifiedLookupInScope() {
1938       Context->setUseQualifiedLookup(oldVal);
1939     }
1940   } QL(LookupCtx);
1941 
1942   if (LookupDirect(*this, R, LookupCtx)) {
1943     R.resolveKind();
1944     if (isa<CXXRecordDecl>(LookupCtx))
1945       R.setNamingClass(cast<CXXRecordDecl>(LookupCtx));
1946     return true;
1947   }
1948 
1949   // Don't descend into implied contexts for redeclarations.
1950   // C++98 [namespace.qual]p6:
1951   //   In a declaration for a namespace member in which the
1952   //   declarator-id is a qualified-id, given that the qualified-id
1953   //   for the namespace member has the form
1954   //     nested-name-specifier unqualified-id
1955   //   the unqualified-id shall name a member of the namespace
1956   //   designated by the nested-name-specifier.
1957   // See also [class.mfct]p5 and [class.static.data]p2.
1958   if (R.isForRedeclaration())
1959     return false;
1960 
1961   // If this is a namespace, look it up in the implied namespaces.
1962   if (LookupCtx->isFileContext())
1963     return LookupQualifiedNameInUsingDirectives(*this, R, LookupCtx);
1964 
1965   // If this isn't a C++ class, we aren't allowed to look into base
1966   // classes, we're done.
1967   CXXRecordDecl *LookupRec = dyn_cast<CXXRecordDecl>(LookupCtx);
1968   if (!LookupRec || !LookupRec->getDefinition())
1969     return false;
1970 
1971   // If we're performing qualified name lookup into a dependent class,
1972   // then we are actually looking into a current instantiation. If we have any
1973   // dependent base classes, then we either have to delay lookup until
1974   // template instantiation time (at which point all bases will be available)
1975   // or we have to fail.
1976   if (!InUnqualifiedLookup && LookupRec->isDependentContext() &&
1977       LookupRec->hasAnyDependentBases()) {
1978     R.setNotFoundInCurrentInstantiation();
1979     return false;
1980   }
1981 
1982   // Perform lookup into our base classes.
1983   CXXBasePaths Paths;
1984   Paths.setOrigin(LookupRec);
1985 
1986   // Look for this member in our base classes
1987   bool (*BaseCallback)(const CXXBaseSpecifier *Specifier, CXXBasePath &Path,
1988                        DeclarationName Name) = nullptr;
1989   switch (R.getLookupKind()) {
1990     case LookupObjCImplicitSelfParam:
1991     case LookupOrdinaryName:
1992     case LookupMemberName:
1993     case LookupRedeclarationWithLinkage:
1994     case LookupLocalFriendName:
1995       BaseCallback = &CXXRecordDecl::FindOrdinaryMember;
1996       break;
1997 
1998     case LookupTagName:
1999       BaseCallback = &CXXRecordDecl::FindTagMember;
2000       break;
2001 
2002     case LookupAnyName:
2003       BaseCallback = &LookupAnyMember;
2004       break;
2005 
2006     case LookupUsingDeclName:
2007       // This lookup is for redeclarations only.
2008 
2009     case LookupOperatorName:
2010     case LookupNamespaceName:
2011     case LookupObjCProtocolName:
2012     case LookupLabel:
2013       // These lookups will never find a member in a C++ class (or base class).
2014       return false;
2015 
2016     case LookupNestedNameSpecifierName:
2017       BaseCallback = &CXXRecordDecl::FindNestedNameSpecifierMember;
2018       break;
2019   }
2020 
2021   DeclarationName Name = R.getLookupName();
2022   if (!LookupRec->lookupInBases(
2023           [=](const CXXBaseSpecifier *Specifier, CXXBasePath &Path) {
2024             return BaseCallback(Specifier, Path, Name);
2025           },
2026           Paths))
2027     return false;
2028 
2029   R.setNamingClass(LookupRec);
2030 
2031   // C++ [class.member.lookup]p2:
2032   //   [...] If the resulting set of declarations are not all from
2033   //   sub-objects of the same type, or the set has a nonstatic member
2034   //   and includes members from distinct sub-objects, there is an
2035   //   ambiguity and the program is ill-formed. Otherwise that set is
2036   //   the result of the lookup.
2037   QualType SubobjectType;
2038   int SubobjectNumber = 0;
2039   AccessSpecifier SubobjectAccess = AS_none;
2040 
2041   for (CXXBasePaths::paths_iterator Path = Paths.begin(), PathEnd = Paths.end();
2042        Path != PathEnd; ++Path) {
2043     const CXXBasePathElement &PathElement = Path->back();
2044 
2045     // Pick the best (i.e. most permissive i.e. numerically lowest) access
2046     // across all paths.
2047     SubobjectAccess = std::min(SubobjectAccess, Path->Access);
2048 
2049     // Determine whether we're looking at a distinct sub-object or not.
2050     if (SubobjectType.isNull()) {
2051       // This is the first subobject we've looked at. Record its type.
2052       SubobjectType = Context.getCanonicalType(PathElement.Base->getType());
2053       SubobjectNumber = PathElement.SubobjectNumber;
2054       continue;
2055     }
2056 
2057     if (SubobjectType
2058                  != Context.getCanonicalType(PathElement.Base->getType())) {
2059       // We found members of the given name in two subobjects of
2060       // different types. If the declaration sets aren't the same, this
2061       // lookup is ambiguous.
2062       if (HasOnlyStaticMembers(Path->Decls.begin(), Path->Decls.end())) {
2063         CXXBasePaths::paths_iterator FirstPath = Paths.begin();
2064         DeclContext::lookup_iterator FirstD = FirstPath->Decls.begin();
2065         DeclContext::lookup_iterator CurrentD = Path->Decls.begin();
2066 
2067         while (FirstD != FirstPath->Decls.end() &&
2068                CurrentD != Path->Decls.end()) {
2069          if ((*FirstD)->getUnderlyingDecl()->getCanonicalDecl() !=
2070              (*CurrentD)->getUnderlyingDecl()->getCanonicalDecl())
2071            break;
2072 
2073           ++FirstD;
2074           ++CurrentD;
2075         }
2076 
2077         if (FirstD == FirstPath->Decls.end() &&
2078             CurrentD == Path->Decls.end())
2079           continue;
2080       }
2081 
2082       R.setAmbiguousBaseSubobjectTypes(Paths);
2083       return true;
2084     }
2085 
2086     if (SubobjectNumber != PathElement.SubobjectNumber) {
2087       // We have a different subobject of the same type.
2088 
2089       // C++ [class.member.lookup]p5:
2090       //   A static member, a nested type or an enumerator defined in
2091       //   a base class T can unambiguously be found even if an object
2092       //   has more than one base class subobject of type T.
2093       if (HasOnlyStaticMembers(Path->Decls.begin(), Path->Decls.end()))
2094         continue;
2095 
2096       // We have found a nonstatic member name in multiple, distinct
2097       // subobjects. Name lookup is ambiguous.
2098       R.setAmbiguousBaseSubobjects(Paths);
2099       return true;
2100     }
2101   }
2102 
2103   // Lookup in a base class succeeded; return these results.
2104 
2105   for (auto *D : Paths.front().Decls) {
2106     AccessSpecifier AS = CXXRecordDecl::MergeAccess(SubobjectAccess,
2107                                                     D->getAccess());
2108     R.addDecl(D, AS);
2109   }
2110   R.resolveKind();
2111   return true;
2112 }
2113 
2114 /// \brief Performs qualified name lookup or special type of lookup for
2115 /// "__super::" scope specifier.
2116 ///
2117 /// This routine is a convenience overload meant to be called from contexts
2118 /// that need to perform a qualified name lookup with an optional C++ scope
2119 /// specifier that might require special kind of lookup.
2120 ///
2121 /// \param R captures both the lookup criteria and any lookup results found.
2122 ///
2123 /// \param LookupCtx The context in which qualified name lookup will
2124 /// search.
2125 ///
2126 /// \param SS An optional C++ scope-specifier.
2127 ///
2128 /// \returns true if lookup succeeded, false if it failed.
2129 bool Sema::LookupQualifiedName(LookupResult &R, DeclContext *LookupCtx,
2130                                CXXScopeSpec &SS) {
2131   auto *NNS = SS.getScopeRep();
2132   if (NNS && NNS->getKind() == NestedNameSpecifier::Super)
2133     return LookupInSuper(R, NNS->getAsRecordDecl());
2134   else
2135 
2136     return LookupQualifiedName(R, LookupCtx);
2137 }
2138 
2139 /// @brief Performs name lookup for a name that was parsed in the
2140 /// source code, and may contain a C++ scope specifier.
2141 ///
2142 /// This routine is a convenience routine meant to be called from
2143 /// contexts that receive a name and an optional C++ scope specifier
2144 /// (e.g., "N::M::x"). It will then perform either qualified or
2145 /// unqualified name lookup (with LookupQualifiedName or LookupName,
2146 /// respectively) on the given name and return those results. It will
2147 /// perform a special type of lookup for "__super::" scope specifier.
2148 ///
2149 /// @param S        The scope from which unqualified name lookup will
2150 /// begin.
2151 ///
2152 /// @param SS       An optional C++ scope-specifier, e.g., "::N::M".
2153 ///
2154 /// @param EnteringContext Indicates whether we are going to enter the
2155 /// context of the scope-specifier SS (if present).
2156 ///
2157 /// @returns True if any decls were found (but possibly ambiguous)
2158 bool Sema::LookupParsedName(LookupResult &R, Scope *S, CXXScopeSpec *SS,
2159                             bool AllowBuiltinCreation, bool EnteringContext) {
2160   if (SS && SS->isInvalid()) {
2161     // When the scope specifier is invalid, don't even look for
2162     // anything.
2163     return false;
2164   }
2165 
2166   if (SS && SS->isSet()) {
2167     NestedNameSpecifier *NNS = SS->getScopeRep();
2168     if (NNS->getKind() == NestedNameSpecifier::Super)
2169       return LookupInSuper(R, NNS->getAsRecordDecl());
2170 
2171     if (DeclContext *DC = computeDeclContext(*SS, EnteringContext)) {
2172       // We have resolved the scope specifier to a particular declaration
2173       // contex, and will perform name lookup in that context.
2174       if (!DC->isDependentContext() && RequireCompleteDeclContext(*SS, DC))
2175         return false;
2176 
2177       R.setContextRange(SS->getRange());
2178       return LookupQualifiedName(R, DC);
2179     }
2180 
2181     // We could not resolve the scope specified to a specific declaration
2182     // context, which means that SS refers to an unknown specialization.
2183     // Name lookup can't find anything in this case.
2184     R.setNotFoundInCurrentInstantiation();
2185     R.setContextRange(SS->getRange());
2186     return false;
2187   }
2188 
2189   // Perform unqualified name lookup starting in the given scope.
2190   return LookupName(R, S, AllowBuiltinCreation);
2191 }
2192 
2193 /// \brief Perform qualified name lookup into all base classes of the given
2194 /// class.
2195 ///
2196 /// \param R captures both the lookup criteria and any lookup results found.
2197 ///
2198 /// \param Class The context in which qualified name lookup will
2199 /// search. Name lookup will search in all base classes merging the results.
2200 ///
2201 /// @returns True if any decls were found (but possibly ambiguous)
2202 bool Sema::LookupInSuper(LookupResult &R, CXXRecordDecl *Class) {
2203   // The access-control rules we use here are essentially the rules for
2204   // doing a lookup in Class that just magically skipped the direct
2205   // members of Class itself.  That is, the naming class is Class, and the
2206   // access includes the access of the base.
2207   for (const auto &BaseSpec : Class->bases()) {
2208     CXXRecordDecl *RD = cast<CXXRecordDecl>(
2209         BaseSpec.getType()->castAs<RecordType>()->getDecl());
2210     LookupResult Result(*this, R.getLookupNameInfo(), R.getLookupKind());
2211 	Result.setBaseObjectType(Context.getRecordType(Class));
2212     LookupQualifiedName(Result, RD);
2213 
2214     // Copy the lookup results into the target, merging the base's access into
2215     // the path access.
2216     for (auto I = Result.begin(), E = Result.end(); I != E; ++I) {
2217       R.addDecl(I.getDecl(),
2218                 CXXRecordDecl::MergeAccess(BaseSpec.getAccessSpecifier(),
2219                                            I.getAccess()));
2220     }
2221 
2222     Result.suppressDiagnostics();
2223   }
2224 
2225   R.resolveKind();
2226   R.setNamingClass(Class);
2227 
2228   return !R.empty();
2229 }
2230 
2231 /// \brief Produce a diagnostic describing the ambiguity that resulted
2232 /// from name lookup.
2233 ///
2234 /// \param Result The result of the ambiguous lookup to be diagnosed.
2235 void Sema::DiagnoseAmbiguousLookup(LookupResult &Result) {
2236   assert(Result.isAmbiguous() && "Lookup result must be ambiguous");
2237 
2238   DeclarationName Name = Result.getLookupName();
2239   SourceLocation NameLoc = Result.getNameLoc();
2240   SourceRange LookupRange = Result.getContextRange();
2241 
2242   switch (Result.getAmbiguityKind()) {
2243   case LookupResult::AmbiguousBaseSubobjects: {
2244     CXXBasePaths *Paths = Result.getBasePaths();
2245     QualType SubobjectType = Paths->front().back().Base->getType();
2246     Diag(NameLoc, diag::err_ambiguous_member_multiple_subobjects)
2247       << Name << SubobjectType << getAmbiguousPathsDisplayString(*Paths)
2248       << LookupRange;
2249 
2250     DeclContext::lookup_iterator Found = Paths->front().Decls.begin();
2251     while (isa<CXXMethodDecl>(*Found) &&
2252            cast<CXXMethodDecl>(*Found)->isStatic())
2253       ++Found;
2254 
2255     Diag((*Found)->getLocation(), diag::note_ambiguous_member_found);
2256     break;
2257   }
2258 
2259   case LookupResult::AmbiguousBaseSubobjectTypes: {
2260     Diag(NameLoc, diag::err_ambiguous_member_multiple_subobject_types)
2261       << Name << LookupRange;
2262 
2263     CXXBasePaths *Paths = Result.getBasePaths();
2264     std::set<Decl *> DeclsPrinted;
2265     for (CXXBasePaths::paths_iterator Path = Paths->begin(),
2266                                       PathEnd = Paths->end();
2267          Path != PathEnd; ++Path) {
2268       Decl *D = Path->Decls.front();
2269       if (DeclsPrinted.insert(D).second)
2270         Diag(D->getLocation(), diag::note_ambiguous_member_found);
2271     }
2272     break;
2273   }
2274 
2275   case LookupResult::AmbiguousTagHiding: {
2276     Diag(NameLoc, diag::err_ambiguous_tag_hiding) << Name << LookupRange;
2277 
2278     llvm::SmallPtrSet<NamedDecl*, 8> TagDecls;
2279 
2280     for (auto *D : Result)
2281       if (TagDecl *TD = dyn_cast<TagDecl>(D)) {
2282         TagDecls.insert(TD);
2283         Diag(TD->getLocation(), diag::note_hidden_tag);
2284       }
2285 
2286     for (auto *D : Result)
2287       if (!isa<TagDecl>(D))
2288         Diag(D->getLocation(), diag::note_hiding_object);
2289 
2290     // For recovery purposes, go ahead and implement the hiding.
2291     LookupResult::Filter F = Result.makeFilter();
2292     while (F.hasNext()) {
2293       if (TagDecls.count(F.next()))
2294         F.erase();
2295     }
2296     F.done();
2297     break;
2298   }
2299 
2300   case LookupResult::AmbiguousReference: {
2301     Diag(NameLoc, diag::err_ambiguous_reference) << Name << LookupRange;
2302 
2303     for (auto *D : Result)
2304       Diag(D->getLocation(), diag::note_ambiguous_candidate) << D;
2305     break;
2306   }
2307   }
2308 }
2309 
2310 namespace {
2311   struct AssociatedLookup {
2312     AssociatedLookup(Sema &S, SourceLocation InstantiationLoc,
2313                      Sema::AssociatedNamespaceSet &Namespaces,
2314                      Sema::AssociatedClassSet &Classes)
2315       : S(S), Namespaces(Namespaces), Classes(Classes),
2316         InstantiationLoc(InstantiationLoc) {
2317     }
2318 
2319     Sema &S;
2320     Sema::AssociatedNamespaceSet &Namespaces;
2321     Sema::AssociatedClassSet &Classes;
2322     SourceLocation InstantiationLoc;
2323   };
2324 } // end anonymous namespace
2325 
2326 static void
2327 addAssociatedClassesAndNamespaces(AssociatedLookup &Result, QualType T);
2328 
2329 static void CollectEnclosingNamespace(Sema::AssociatedNamespaceSet &Namespaces,
2330                                       DeclContext *Ctx) {
2331   // Add the associated namespace for this class.
2332 
2333   // We don't use DeclContext::getEnclosingNamespaceContext() as this may
2334   // be a locally scoped record.
2335 
2336   // We skip out of inline namespaces. The innermost non-inline namespace
2337   // contains all names of all its nested inline namespaces anyway, so we can
2338   // replace the entire inline namespace tree with its root.
2339   while (Ctx->isRecord() || Ctx->isTransparentContext() ||
2340          Ctx->isInlineNamespace())
2341     Ctx = Ctx->getParent();
2342 
2343   if (Ctx->isFileContext())
2344     Namespaces.insert(Ctx->getPrimaryContext());
2345 }
2346 
2347 // \brief Add the associated classes and namespaces for argument-dependent
2348 // lookup that involves a template argument (C++ [basic.lookup.koenig]p2).
2349 static void
2350 addAssociatedClassesAndNamespaces(AssociatedLookup &Result,
2351                                   const TemplateArgument &Arg) {
2352   // C++ [basic.lookup.koenig]p2, last bullet:
2353   //   -- [...] ;
2354   switch (Arg.getKind()) {
2355     case TemplateArgument::Null:
2356       break;
2357 
2358     case TemplateArgument::Type:
2359       // [...] the namespaces and classes associated with the types of the
2360       // template arguments provided for template type parameters (excluding
2361       // template template parameters)
2362       addAssociatedClassesAndNamespaces(Result, Arg.getAsType());
2363       break;
2364 
2365     case TemplateArgument::Template:
2366     case TemplateArgument::TemplateExpansion: {
2367       // [...] the namespaces in which any template template arguments are
2368       // defined; and the classes in which any member templates used as
2369       // template template arguments are defined.
2370       TemplateName Template = Arg.getAsTemplateOrTemplatePattern();
2371       if (ClassTemplateDecl *ClassTemplate
2372                  = dyn_cast<ClassTemplateDecl>(Template.getAsTemplateDecl())) {
2373         DeclContext *Ctx = ClassTemplate->getDeclContext();
2374         if (CXXRecordDecl *EnclosingClass = dyn_cast<CXXRecordDecl>(Ctx))
2375           Result.Classes.insert(EnclosingClass);
2376         // Add the associated namespace for this class.
2377         CollectEnclosingNamespace(Result.Namespaces, Ctx);
2378       }
2379       break;
2380     }
2381 
2382     case TemplateArgument::Declaration:
2383     case TemplateArgument::Integral:
2384     case TemplateArgument::Expression:
2385     case TemplateArgument::NullPtr:
2386       // [Note: non-type template arguments do not contribute to the set of
2387       //  associated namespaces. ]
2388       break;
2389 
2390     case TemplateArgument::Pack:
2391       for (const auto &P : Arg.pack_elements())
2392         addAssociatedClassesAndNamespaces(Result, P);
2393       break;
2394   }
2395 }
2396 
2397 // \brief Add the associated classes and namespaces for
2398 // argument-dependent lookup with an argument of class type
2399 // (C++ [basic.lookup.koenig]p2).
2400 static void
2401 addAssociatedClassesAndNamespaces(AssociatedLookup &Result,
2402                                   CXXRecordDecl *Class) {
2403 
2404   // Just silently ignore anything whose name is __va_list_tag.
2405   if (Class->getDeclName() == Result.S.VAListTagName)
2406     return;
2407 
2408   // C++ [basic.lookup.koenig]p2:
2409   //   [...]
2410   //     -- If T is a class type (including unions), its associated
2411   //        classes are: the class itself; the class of which it is a
2412   //        member, if any; and its direct and indirect base
2413   //        classes. Its associated namespaces are the namespaces in
2414   //        which its associated classes are defined.
2415 
2416   // Add the class of which it is a member, if any.
2417   DeclContext *Ctx = Class->getDeclContext();
2418   if (CXXRecordDecl *EnclosingClass = dyn_cast<CXXRecordDecl>(Ctx))
2419     Result.Classes.insert(EnclosingClass);
2420   // Add the associated namespace for this class.
2421   CollectEnclosingNamespace(Result.Namespaces, Ctx);
2422 
2423   // Add the class itself. If we've already seen this class, we don't
2424   // need to visit base classes.
2425   //
2426   // FIXME: That's not correct, we may have added this class only because it
2427   // was the enclosing class of another class, and in that case we won't have
2428   // added its base classes yet.
2429   if (!Result.Classes.insert(Class).second)
2430     return;
2431 
2432   // -- If T is a template-id, its associated namespaces and classes are
2433   //    the namespace in which the template is defined; for member
2434   //    templates, the member template's class; the namespaces and classes
2435   //    associated with the types of the template arguments provided for
2436   //    template type parameters (excluding template template parameters); the
2437   //    namespaces in which any template template arguments are defined; and
2438   //    the classes in which any member templates used as template template
2439   //    arguments are defined. [Note: non-type template arguments do not
2440   //    contribute to the set of associated namespaces. ]
2441   if (ClassTemplateSpecializationDecl *Spec
2442         = dyn_cast<ClassTemplateSpecializationDecl>(Class)) {
2443     DeclContext *Ctx = Spec->getSpecializedTemplate()->getDeclContext();
2444     if (CXXRecordDecl *EnclosingClass = dyn_cast<CXXRecordDecl>(Ctx))
2445       Result.Classes.insert(EnclosingClass);
2446     // Add the associated namespace for this class.
2447     CollectEnclosingNamespace(Result.Namespaces, Ctx);
2448 
2449     const TemplateArgumentList &TemplateArgs = Spec->getTemplateArgs();
2450     for (unsigned I = 0, N = TemplateArgs.size(); I != N; ++I)
2451       addAssociatedClassesAndNamespaces(Result, TemplateArgs[I]);
2452   }
2453 
2454   // Only recurse into base classes for complete types.
2455   if (!Result.S.isCompleteType(Result.InstantiationLoc,
2456                                Result.S.Context.getRecordType(Class)))
2457     return;
2458 
2459   // Add direct and indirect base classes along with their associated
2460   // namespaces.
2461   SmallVector<CXXRecordDecl *, 32> Bases;
2462   Bases.push_back(Class);
2463   while (!Bases.empty()) {
2464     // Pop this class off the stack.
2465     Class = Bases.pop_back_val();
2466 
2467     // Visit the base classes.
2468     for (const auto &Base : Class->bases()) {
2469       const RecordType *BaseType = Base.getType()->getAs<RecordType>();
2470       // In dependent contexts, we do ADL twice, and the first time around,
2471       // the base type might be a dependent TemplateSpecializationType, or a
2472       // TemplateTypeParmType. If that happens, simply ignore it.
2473       // FIXME: If we want to support export, we probably need to add the
2474       // namespace of the template in a TemplateSpecializationType, or even
2475       // the classes and namespaces of known non-dependent arguments.
2476       if (!BaseType)
2477         continue;
2478       CXXRecordDecl *BaseDecl = cast<CXXRecordDecl>(BaseType->getDecl());
2479       if (Result.Classes.insert(BaseDecl).second) {
2480         // Find the associated namespace for this base class.
2481         DeclContext *BaseCtx = BaseDecl->getDeclContext();
2482         CollectEnclosingNamespace(Result.Namespaces, BaseCtx);
2483 
2484         // Make sure we visit the bases of this base class.
2485         if (BaseDecl->bases_begin() != BaseDecl->bases_end())
2486           Bases.push_back(BaseDecl);
2487       }
2488     }
2489   }
2490 }
2491 
2492 // \brief Add the associated classes and namespaces for
2493 // argument-dependent lookup with an argument of type T
2494 // (C++ [basic.lookup.koenig]p2).
2495 static void
2496 addAssociatedClassesAndNamespaces(AssociatedLookup &Result, QualType Ty) {
2497   // C++ [basic.lookup.koenig]p2:
2498   //
2499   //   For each argument type T in the function call, there is a set
2500   //   of zero or more associated namespaces and a set of zero or more
2501   //   associated classes to be considered. The sets of namespaces and
2502   //   classes is determined entirely by the types of the function
2503   //   arguments (and the namespace of any template template
2504   //   argument). Typedef names and using-declarations used to specify
2505   //   the types do not contribute to this set. The sets of namespaces
2506   //   and classes are determined in the following way:
2507 
2508   SmallVector<const Type *, 16> Queue;
2509   const Type *T = Ty->getCanonicalTypeInternal().getTypePtr();
2510 
2511   while (true) {
2512     switch (T->getTypeClass()) {
2513 
2514 #define TYPE(Class, Base)
2515 #define DEPENDENT_TYPE(Class, Base) case Type::Class:
2516 #define NON_CANONICAL_TYPE(Class, Base) case Type::Class:
2517 #define NON_CANONICAL_UNLESS_DEPENDENT_TYPE(Class, Base) case Type::Class:
2518 #define ABSTRACT_TYPE(Class, Base)
2519 #include "clang/AST/TypeNodes.def"
2520       // T is canonical.  We can also ignore dependent types because
2521       // we don't need to do ADL at the definition point, but if we
2522       // wanted to implement template export (or if we find some other
2523       // use for associated classes and namespaces...) this would be
2524       // wrong.
2525       break;
2526 
2527     //    -- If T is a pointer to U or an array of U, its associated
2528     //       namespaces and classes are those associated with U.
2529     case Type::Pointer:
2530       T = cast<PointerType>(T)->getPointeeType().getTypePtr();
2531       continue;
2532     case Type::ConstantArray:
2533     case Type::IncompleteArray:
2534     case Type::VariableArray:
2535       T = cast<ArrayType>(T)->getElementType().getTypePtr();
2536       continue;
2537 
2538     //     -- If T is a fundamental type, its associated sets of
2539     //        namespaces and classes are both empty.
2540     case Type::Builtin:
2541       break;
2542 
2543     //     -- If T is a class type (including unions), its associated
2544     //        classes are: the class itself; the class of which it is a
2545     //        member, if any; and its direct and indirect base
2546     //        classes. Its associated namespaces are the namespaces in
2547     //        which its associated classes are defined.
2548     case Type::Record: {
2549       CXXRecordDecl *Class =
2550           cast<CXXRecordDecl>(cast<RecordType>(T)->getDecl());
2551       addAssociatedClassesAndNamespaces(Result, Class);
2552       break;
2553     }
2554 
2555     //     -- If T is an enumeration type, its associated namespace is
2556     //        the namespace in which it is defined. If it is class
2557     //        member, its associated class is the member's class; else
2558     //        it has no associated class.
2559     case Type::Enum: {
2560       EnumDecl *Enum = cast<EnumType>(T)->getDecl();
2561 
2562       DeclContext *Ctx = Enum->getDeclContext();
2563       if (CXXRecordDecl *EnclosingClass = dyn_cast<CXXRecordDecl>(Ctx))
2564         Result.Classes.insert(EnclosingClass);
2565 
2566       // Add the associated namespace for this class.
2567       CollectEnclosingNamespace(Result.Namespaces, Ctx);
2568 
2569       break;
2570     }
2571 
2572     //     -- If T is a function type, its associated namespaces and
2573     //        classes are those associated with the function parameter
2574     //        types and those associated with the return type.
2575     case Type::FunctionProto: {
2576       const FunctionProtoType *Proto = cast<FunctionProtoType>(T);
2577       for (const auto &Arg : Proto->param_types())
2578         Queue.push_back(Arg.getTypePtr());
2579       // fallthrough
2580     }
2581     case Type::FunctionNoProto: {
2582       const FunctionType *FnType = cast<FunctionType>(T);
2583       T = FnType->getReturnType().getTypePtr();
2584       continue;
2585     }
2586 
2587     //     -- If T is a pointer to a member function of a class X, its
2588     //        associated namespaces and classes are those associated
2589     //        with the function parameter types and return type,
2590     //        together with those associated with X.
2591     //
2592     //     -- If T is a pointer to a data member of class X, its
2593     //        associated namespaces and classes are those associated
2594     //        with the member type together with those associated with
2595     //        X.
2596     case Type::MemberPointer: {
2597       const MemberPointerType *MemberPtr = cast<MemberPointerType>(T);
2598 
2599       // Queue up the class type into which this points.
2600       Queue.push_back(MemberPtr->getClass());
2601 
2602       // And directly continue with the pointee type.
2603       T = MemberPtr->getPointeeType().getTypePtr();
2604       continue;
2605     }
2606 
2607     // As an extension, treat this like a normal pointer.
2608     case Type::BlockPointer:
2609       T = cast<BlockPointerType>(T)->getPointeeType().getTypePtr();
2610       continue;
2611 
2612     // References aren't covered by the standard, but that's such an
2613     // obvious defect that we cover them anyway.
2614     case Type::LValueReference:
2615     case Type::RValueReference:
2616       T = cast<ReferenceType>(T)->getPointeeType().getTypePtr();
2617       continue;
2618 
2619     // These are fundamental types.
2620     case Type::Vector:
2621     case Type::ExtVector:
2622     case Type::Complex:
2623       break;
2624 
2625     // Non-deduced auto types only get here for error cases.
2626     case Type::Auto:
2627       break;
2628 
2629     // If T is an Objective-C object or interface type, or a pointer to an
2630     // object or interface type, the associated namespace is the global
2631     // namespace.
2632     case Type::ObjCObject:
2633     case Type::ObjCInterface:
2634     case Type::ObjCObjectPointer:
2635       Result.Namespaces.insert(Result.S.Context.getTranslationUnitDecl());
2636       break;
2637 
2638     // Atomic types are just wrappers; use the associations of the
2639     // contained type.
2640     case Type::Atomic:
2641       T = cast<AtomicType>(T)->getValueType().getTypePtr();
2642       continue;
2643     case Type::Pipe:
2644       T = cast<PipeType>(T)->getElementType().getTypePtr();
2645       continue;
2646     }
2647 
2648     if (Queue.empty())
2649       break;
2650     T = Queue.pop_back_val();
2651   }
2652 }
2653 
2654 /// \brief Find the associated classes and namespaces for
2655 /// argument-dependent lookup for a call with the given set of
2656 /// arguments.
2657 ///
2658 /// This routine computes the sets of associated classes and associated
2659 /// namespaces searched by argument-dependent lookup
2660 /// (C++ [basic.lookup.argdep]) for a given set of arguments.
2661 void Sema::FindAssociatedClassesAndNamespaces(
2662     SourceLocation InstantiationLoc, ArrayRef<Expr *> Args,
2663     AssociatedNamespaceSet &AssociatedNamespaces,
2664     AssociatedClassSet &AssociatedClasses) {
2665   AssociatedNamespaces.clear();
2666   AssociatedClasses.clear();
2667 
2668   AssociatedLookup Result(*this, InstantiationLoc,
2669                           AssociatedNamespaces, AssociatedClasses);
2670 
2671   // C++ [basic.lookup.koenig]p2:
2672   //   For each argument type T in the function call, there is a set
2673   //   of zero or more associated namespaces and a set of zero or more
2674   //   associated classes to be considered. The sets of namespaces and
2675   //   classes is determined entirely by the types of the function
2676   //   arguments (and the namespace of any template template
2677   //   argument).
2678   for (unsigned ArgIdx = 0; ArgIdx != Args.size(); ++ArgIdx) {
2679     Expr *Arg = Args[ArgIdx];
2680 
2681     if (Arg->getType() != Context.OverloadTy) {
2682       addAssociatedClassesAndNamespaces(Result, Arg->getType());
2683       continue;
2684     }
2685 
2686     // [...] In addition, if the argument is the name or address of a
2687     // set of overloaded functions and/or function templates, its
2688     // associated classes and namespaces are the union of those
2689     // associated with each of the members of the set: the namespace
2690     // in which the function or function template is defined and the
2691     // classes and namespaces associated with its (non-dependent)
2692     // parameter types and return type.
2693     Arg = Arg->IgnoreParens();
2694     if (UnaryOperator *unaryOp = dyn_cast<UnaryOperator>(Arg))
2695       if (unaryOp->getOpcode() == UO_AddrOf)
2696         Arg = unaryOp->getSubExpr();
2697 
2698     UnresolvedLookupExpr *ULE = dyn_cast<UnresolvedLookupExpr>(Arg);
2699     if (!ULE) continue;
2700 
2701     for (const auto *D : ULE->decls()) {
2702       // Look through any using declarations to find the underlying function.
2703       const FunctionDecl *FDecl = D->getUnderlyingDecl()->getAsFunction();
2704 
2705       // Add the classes and namespaces associated with the parameter
2706       // types and return type of this function.
2707       addAssociatedClassesAndNamespaces(Result, FDecl->getType());
2708     }
2709   }
2710 }
2711 
2712 NamedDecl *Sema::LookupSingleName(Scope *S, DeclarationName Name,
2713                                   SourceLocation Loc,
2714                                   LookupNameKind NameKind,
2715                                   RedeclarationKind Redecl) {
2716   LookupResult R(*this, Name, Loc, NameKind, Redecl);
2717   LookupName(R, S);
2718   return R.getAsSingle<NamedDecl>();
2719 }
2720 
2721 /// \brief Find the protocol with the given name, if any.
2722 ObjCProtocolDecl *Sema::LookupProtocol(IdentifierInfo *II,
2723                                        SourceLocation IdLoc,
2724                                        RedeclarationKind Redecl) {
2725   Decl *D = LookupSingleName(TUScope, II, IdLoc,
2726                              LookupObjCProtocolName, Redecl);
2727   return cast_or_null<ObjCProtocolDecl>(D);
2728 }
2729 
2730 void Sema::LookupOverloadedOperatorName(OverloadedOperatorKind Op, Scope *S,
2731                                         QualType T1, QualType T2,
2732                                         UnresolvedSetImpl &Functions) {
2733   // C++ [over.match.oper]p3:
2734   //     -- The set of non-member candidates is the result of the
2735   //        unqualified lookup of operator@ in the context of the
2736   //        expression according to the usual rules for name lookup in
2737   //        unqualified function calls (3.4.2) except that all member
2738   //        functions are ignored.
2739   DeclarationName OpName = Context.DeclarationNames.getCXXOperatorName(Op);
2740   LookupResult Operators(*this, OpName, SourceLocation(), LookupOperatorName);
2741   LookupName(Operators, S);
2742 
2743   assert(!Operators.isAmbiguous() && "Operator lookup cannot be ambiguous");
2744   Functions.append(Operators.begin(), Operators.end());
2745 }
2746 
2747 Sema::SpecialMemberOverloadResult *Sema::LookupSpecialMember(CXXRecordDecl *RD,
2748                                                             CXXSpecialMember SM,
2749                                                             bool ConstArg,
2750                                                             bool VolatileArg,
2751                                                             bool RValueThis,
2752                                                             bool ConstThis,
2753                                                             bool VolatileThis) {
2754   assert(CanDeclareSpecialMemberFunction(RD) &&
2755          "doing special member lookup into record that isn't fully complete");
2756   RD = RD->getDefinition();
2757   if (RValueThis || ConstThis || VolatileThis)
2758     assert((SM == CXXCopyAssignment || SM == CXXMoveAssignment) &&
2759            "constructors and destructors always have unqualified lvalue this");
2760   if (ConstArg || VolatileArg)
2761     assert((SM != CXXDefaultConstructor && SM != CXXDestructor) &&
2762            "parameter-less special members can't have qualified arguments");
2763 
2764   llvm::FoldingSetNodeID ID;
2765   ID.AddPointer(RD);
2766   ID.AddInteger(SM);
2767   ID.AddInteger(ConstArg);
2768   ID.AddInteger(VolatileArg);
2769   ID.AddInteger(RValueThis);
2770   ID.AddInteger(ConstThis);
2771   ID.AddInteger(VolatileThis);
2772 
2773   void *InsertPoint;
2774   SpecialMemberOverloadResult *Result =
2775     SpecialMemberCache.FindNodeOrInsertPos(ID, InsertPoint);
2776 
2777   // This was already cached
2778   if (Result)
2779     return Result;
2780 
2781   Result = BumpAlloc.Allocate<SpecialMemberOverloadResult>();
2782   Result = new (Result) SpecialMemberOverloadResult(ID);
2783   SpecialMemberCache.InsertNode(Result, InsertPoint);
2784 
2785   if (SM == CXXDestructor) {
2786     if (RD->needsImplicitDestructor())
2787       DeclareImplicitDestructor(RD);
2788     CXXDestructorDecl *DD = RD->getDestructor();
2789     assert(DD && "record without a destructor");
2790     Result->setMethod(DD);
2791     Result->setKind(DD->isDeleted() ?
2792                     SpecialMemberOverloadResult::NoMemberOrDeleted :
2793                     SpecialMemberOverloadResult::Success);
2794     return Result;
2795   }
2796 
2797   // Prepare for overload resolution. Here we construct a synthetic argument
2798   // if necessary and make sure that implicit functions are declared.
2799   CanQualType CanTy = Context.getCanonicalType(Context.getTagDeclType(RD));
2800   DeclarationName Name;
2801   Expr *Arg = nullptr;
2802   unsigned NumArgs;
2803 
2804   QualType ArgType = CanTy;
2805   ExprValueKind VK = VK_LValue;
2806 
2807   if (SM == CXXDefaultConstructor) {
2808     Name = Context.DeclarationNames.getCXXConstructorName(CanTy);
2809     NumArgs = 0;
2810     if (RD->needsImplicitDefaultConstructor())
2811       DeclareImplicitDefaultConstructor(RD);
2812   } else {
2813     if (SM == CXXCopyConstructor || SM == CXXMoveConstructor) {
2814       Name = Context.DeclarationNames.getCXXConstructorName(CanTy);
2815       if (RD->needsImplicitCopyConstructor())
2816         DeclareImplicitCopyConstructor(RD);
2817       if (getLangOpts().CPlusPlus11 && RD->needsImplicitMoveConstructor())
2818         DeclareImplicitMoveConstructor(RD);
2819     } else {
2820       Name = Context.DeclarationNames.getCXXOperatorName(OO_Equal);
2821       if (RD->needsImplicitCopyAssignment())
2822         DeclareImplicitCopyAssignment(RD);
2823       if (getLangOpts().CPlusPlus11 && RD->needsImplicitMoveAssignment())
2824         DeclareImplicitMoveAssignment(RD);
2825     }
2826 
2827     if (ConstArg)
2828       ArgType.addConst();
2829     if (VolatileArg)
2830       ArgType.addVolatile();
2831 
2832     // This isn't /really/ specified by the standard, but it's implied
2833     // we should be working from an RValue in the case of move to ensure
2834     // that we prefer to bind to rvalue references, and an LValue in the
2835     // case of copy to ensure we don't bind to rvalue references.
2836     // Possibly an XValue is actually correct in the case of move, but
2837     // there is no semantic difference for class types in this restricted
2838     // case.
2839     if (SM == CXXCopyConstructor || SM == CXXCopyAssignment)
2840       VK = VK_LValue;
2841     else
2842       VK = VK_RValue;
2843   }
2844 
2845   OpaqueValueExpr FakeArg(SourceLocation(), ArgType, VK);
2846 
2847   if (SM != CXXDefaultConstructor) {
2848     NumArgs = 1;
2849     Arg = &FakeArg;
2850   }
2851 
2852   // Create the object argument
2853   QualType ThisTy = CanTy;
2854   if (ConstThis)
2855     ThisTy.addConst();
2856   if (VolatileThis)
2857     ThisTy.addVolatile();
2858   Expr::Classification Classification =
2859     OpaqueValueExpr(SourceLocation(), ThisTy,
2860                     RValueThis ? VK_RValue : VK_LValue).Classify(Context);
2861 
2862   // Now we perform lookup on the name we computed earlier and do overload
2863   // resolution. Lookup is only performed directly into the class since there
2864   // will always be a (possibly implicit) declaration to shadow any others.
2865   OverloadCandidateSet OCS(RD->getLocation(), OverloadCandidateSet::CSK_Normal);
2866   DeclContext::lookup_result R = RD->lookup(Name);
2867 
2868   if (R.empty()) {
2869     // We might have no default constructor because we have a lambda's closure
2870     // type, rather than because there's some other declared constructor.
2871     // Every class has a copy/move constructor, copy/move assignment, and
2872     // destructor.
2873     assert(SM == CXXDefaultConstructor &&
2874            "lookup for a constructor or assignment operator was empty");
2875     Result->setMethod(nullptr);
2876     Result->setKind(SpecialMemberOverloadResult::NoMemberOrDeleted);
2877     return Result;
2878   }
2879 
2880   // Copy the candidates as our processing of them may load new declarations
2881   // from an external source and invalidate lookup_result.
2882   SmallVector<NamedDecl *, 8> Candidates(R.begin(), R.end());
2883 
2884   for (auto *Cand : Candidates) {
2885     if (Cand->isInvalidDecl())
2886       continue;
2887 
2888     if (UsingShadowDecl *U = dyn_cast<UsingShadowDecl>(Cand)) {
2889       // FIXME: [namespace.udecl]p15 says that we should only consider a
2890       // using declaration here if it does not match a declaration in the
2891       // derived class. We do not implement this correctly in other cases
2892       // either.
2893       Cand = U->getTargetDecl();
2894 
2895       if (Cand->isInvalidDecl())
2896         continue;
2897     }
2898 
2899     if (CXXMethodDecl *M = dyn_cast<CXXMethodDecl>(Cand)) {
2900       if (SM == CXXCopyAssignment || SM == CXXMoveAssignment)
2901         AddMethodCandidate(M, DeclAccessPair::make(M, AS_public), RD, ThisTy,
2902                            Classification, llvm::makeArrayRef(&Arg, NumArgs),
2903                            OCS, true);
2904       else
2905         AddOverloadCandidate(M, DeclAccessPair::make(M, AS_public),
2906                              llvm::makeArrayRef(&Arg, NumArgs), OCS, true);
2907     } else if (FunctionTemplateDecl *Tmpl =
2908                  dyn_cast<FunctionTemplateDecl>(Cand)) {
2909       if (SM == CXXCopyAssignment || SM == CXXMoveAssignment)
2910         AddMethodTemplateCandidate(Tmpl, DeclAccessPair::make(Tmpl, AS_public),
2911                                    RD, nullptr, ThisTy, Classification,
2912                                    llvm::makeArrayRef(&Arg, NumArgs),
2913                                    OCS, true);
2914       else
2915         AddTemplateOverloadCandidate(Tmpl, DeclAccessPair::make(Tmpl, AS_public),
2916                                      nullptr, llvm::makeArrayRef(&Arg, NumArgs),
2917                                      OCS, true);
2918     } else {
2919       assert(isa<UsingDecl>(Cand) && "illegal Kind of operator = Decl");
2920     }
2921   }
2922 
2923   OverloadCandidateSet::iterator Best;
2924   switch (OCS.BestViableFunction(*this, SourceLocation(), Best)) {
2925     case OR_Success:
2926       Result->setMethod(cast<CXXMethodDecl>(Best->Function));
2927       Result->setKind(SpecialMemberOverloadResult::Success);
2928       break;
2929 
2930     case OR_Deleted:
2931       Result->setMethod(cast<CXXMethodDecl>(Best->Function));
2932       Result->setKind(SpecialMemberOverloadResult::NoMemberOrDeleted);
2933       break;
2934 
2935     case OR_Ambiguous:
2936       Result->setMethod(nullptr);
2937       Result->setKind(SpecialMemberOverloadResult::Ambiguous);
2938       break;
2939 
2940     case OR_No_Viable_Function:
2941       Result->setMethod(nullptr);
2942       Result->setKind(SpecialMemberOverloadResult::NoMemberOrDeleted);
2943       break;
2944   }
2945 
2946   return Result;
2947 }
2948 
2949 /// \brief Look up the default constructor for the given class.
2950 CXXConstructorDecl *Sema::LookupDefaultConstructor(CXXRecordDecl *Class) {
2951   SpecialMemberOverloadResult *Result =
2952     LookupSpecialMember(Class, CXXDefaultConstructor, false, false, false,
2953                         false, false);
2954 
2955   return cast_or_null<CXXConstructorDecl>(Result->getMethod());
2956 }
2957 
2958 /// \brief Look up the copying constructor for the given class.
2959 CXXConstructorDecl *Sema::LookupCopyingConstructor(CXXRecordDecl *Class,
2960                                                    unsigned Quals) {
2961   assert(!(Quals & ~(Qualifiers::Const | Qualifiers::Volatile)) &&
2962          "non-const, non-volatile qualifiers for copy ctor arg");
2963   SpecialMemberOverloadResult *Result =
2964     LookupSpecialMember(Class, CXXCopyConstructor, Quals & Qualifiers::Const,
2965                         Quals & Qualifiers::Volatile, false, false, false);
2966 
2967   return cast_or_null<CXXConstructorDecl>(Result->getMethod());
2968 }
2969 
2970 /// \brief Look up the moving constructor for the given class.
2971 CXXConstructorDecl *Sema::LookupMovingConstructor(CXXRecordDecl *Class,
2972                                                   unsigned Quals) {
2973   SpecialMemberOverloadResult *Result =
2974     LookupSpecialMember(Class, CXXMoveConstructor, Quals & Qualifiers::Const,
2975                         Quals & Qualifiers::Volatile, false, false, false);
2976 
2977   return cast_or_null<CXXConstructorDecl>(Result->getMethod());
2978 }
2979 
2980 /// \brief Look up the constructors for the given class.
2981 DeclContext::lookup_result Sema::LookupConstructors(CXXRecordDecl *Class) {
2982   // If the implicit constructors have not yet been declared, do so now.
2983   if (CanDeclareSpecialMemberFunction(Class)) {
2984     if (Class->needsImplicitDefaultConstructor())
2985       DeclareImplicitDefaultConstructor(Class);
2986     if (Class->needsImplicitCopyConstructor())
2987       DeclareImplicitCopyConstructor(Class);
2988     if (getLangOpts().CPlusPlus11 && Class->needsImplicitMoveConstructor())
2989       DeclareImplicitMoveConstructor(Class);
2990   }
2991 
2992   CanQualType T = Context.getCanonicalType(Context.getTypeDeclType(Class));
2993   DeclarationName Name = Context.DeclarationNames.getCXXConstructorName(T);
2994   return Class->lookup(Name);
2995 }
2996 
2997 /// \brief Look up the copying assignment operator for the given class.
2998 CXXMethodDecl *Sema::LookupCopyingAssignment(CXXRecordDecl *Class,
2999                                              unsigned Quals, bool RValueThis,
3000                                              unsigned ThisQuals) {
3001   assert(!(Quals & ~(Qualifiers::Const | Qualifiers::Volatile)) &&
3002          "non-const, non-volatile qualifiers for copy assignment arg");
3003   assert(!(ThisQuals & ~(Qualifiers::Const | Qualifiers::Volatile)) &&
3004          "non-const, non-volatile qualifiers for copy assignment this");
3005   SpecialMemberOverloadResult *Result =
3006     LookupSpecialMember(Class, CXXCopyAssignment, Quals & Qualifiers::Const,
3007                         Quals & Qualifiers::Volatile, RValueThis,
3008                         ThisQuals & Qualifiers::Const,
3009                         ThisQuals & Qualifiers::Volatile);
3010 
3011   return Result->getMethod();
3012 }
3013 
3014 /// \brief Look up the moving assignment operator for the given class.
3015 CXXMethodDecl *Sema::LookupMovingAssignment(CXXRecordDecl *Class,
3016                                             unsigned Quals,
3017                                             bool RValueThis,
3018                                             unsigned ThisQuals) {
3019   assert(!(ThisQuals & ~(Qualifiers::Const | Qualifiers::Volatile)) &&
3020          "non-const, non-volatile qualifiers for copy assignment this");
3021   SpecialMemberOverloadResult *Result =
3022     LookupSpecialMember(Class, CXXMoveAssignment, Quals & Qualifiers::Const,
3023                         Quals & Qualifiers::Volatile, RValueThis,
3024                         ThisQuals & Qualifiers::Const,
3025                         ThisQuals & Qualifiers::Volatile);
3026 
3027   return Result->getMethod();
3028 }
3029 
3030 /// \brief Look for the destructor of the given class.
3031 ///
3032 /// During semantic analysis, this routine should be used in lieu of
3033 /// CXXRecordDecl::getDestructor().
3034 ///
3035 /// \returns The destructor for this class.
3036 CXXDestructorDecl *Sema::LookupDestructor(CXXRecordDecl *Class) {
3037   return cast<CXXDestructorDecl>(LookupSpecialMember(Class, CXXDestructor,
3038                                                      false, false, false,
3039                                                      false, false)->getMethod());
3040 }
3041 
3042 /// LookupLiteralOperator - Determine which literal operator should be used for
3043 /// a user-defined literal, per C++11 [lex.ext].
3044 ///
3045 /// Normal overload resolution is not used to select which literal operator to
3046 /// call for a user-defined literal. Look up the provided literal operator name,
3047 /// and filter the results to the appropriate set for the given argument types.
3048 Sema::LiteralOperatorLookupResult
3049 Sema::LookupLiteralOperator(Scope *S, LookupResult &R,
3050                             ArrayRef<QualType> ArgTys,
3051                             bool AllowRaw, bool AllowTemplate,
3052                             bool AllowStringTemplate) {
3053   LookupName(R, S);
3054   assert(R.getResultKind() != LookupResult::Ambiguous &&
3055          "literal operator lookup can't be ambiguous");
3056 
3057   // Filter the lookup results appropriately.
3058   LookupResult::Filter F = R.makeFilter();
3059 
3060   bool FoundRaw = false;
3061   bool FoundTemplate = false;
3062   bool FoundStringTemplate = false;
3063   bool FoundExactMatch = false;
3064 
3065   while (F.hasNext()) {
3066     Decl *D = F.next();
3067     if (UsingShadowDecl *USD = dyn_cast<UsingShadowDecl>(D))
3068       D = USD->getTargetDecl();
3069 
3070     // If the declaration we found is invalid, skip it.
3071     if (D->isInvalidDecl()) {
3072       F.erase();
3073       continue;
3074     }
3075 
3076     bool IsRaw = false;
3077     bool IsTemplate = false;
3078     bool IsStringTemplate = false;
3079     bool IsExactMatch = false;
3080 
3081     if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D)) {
3082       if (FD->getNumParams() == 1 &&
3083           FD->getParamDecl(0)->getType()->getAs<PointerType>())
3084         IsRaw = true;
3085       else if (FD->getNumParams() == ArgTys.size()) {
3086         IsExactMatch = true;
3087         for (unsigned ArgIdx = 0; ArgIdx != ArgTys.size(); ++ArgIdx) {
3088           QualType ParamTy = FD->getParamDecl(ArgIdx)->getType();
3089           if (!Context.hasSameUnqualifiedType(ArgTys[ArgIdx], ParamTy)) {
3090             IsExactMatch = false;
3091             break;
3092           }
3093         }
3094       }
3095     }
3096     if (FunctionTemplateDecl *FD = dyn_cast<FunctionTemplateDecl>(D)) {
3097       TemplateParameterList *Params = FD->getTemplateParameters();
3098       if (Params->size() == 1)
3099         IsTemplate = true;
3100       else
3101         IsStringTemplate = true;
3102     }
3103 
3104     if (IsExactMatch) {
3105       FoundExactMatch = true;
3106       AllowRaw = false;
3107       AllowTemplate = false;
3108       AllowStringTemplate = false;
3109       if (FoundRaw || FoundTemplate || FoundStringTemplate) {
3110         // Go through again and remove the raw and template decls we've
3111         // already found.
3112         F.restart();
3113         FoundRaw = FoundTemplate = FoundStringTemplate = false;
3114       }
3115     } else if (AllowRaw && IsRaw) {
3116       FoundRaw = true;
3117     } else if (AllowTemplate && IsTemplate) {
3118       FoundTemplate = true;
3119     } else if (AllowStringTemplate && IsStringTemplate) {
3120       FoundStringTemplate = true;
3121     } else {
3122       F.erase();
3123     }
3124   }
3125 
3126   F.done();
3127 
3128   // C++11 [lex.ext]p3, p4: If S contains a literal operator with a matching
3129   // parameter type, that is used in preference to a raw literal operator
3130   // or literal operator template.
3131   if (FoundExactMatch)
3132     return LOLR_Cooked;
3133 
3134   // C++11 [lex.ext]p3, p4: S shall contain a raw literal operator or a literal
3135   // operator template, but not both.
3136   if (FoundRaw && FoundTemplate) {
3137     Diag(R.getNameLoc(), diag::err_ovl_ambiguous_call) << R.getLookupName();
3138     for (LookupResult::iterator I = R.begin(), E = R.end(); I != E; ++I)
3139       NoteOverloadCandidate((*I)->getUnderlyingDecl()->getAsFunction());
3140     return LOLR_Error;
3141   }
3142 
3143   if (FoundRaw)
3144     return LOLR_Raw;
3145 
3146   if (FoundTemplate)
3147     return LOLR_Template;
3148 
3149   if (FoundStringTemplate)
3150     return LOLR_StringTemplate;
3151 
3152   // Didn't find anything we could use.
3153   Diag(R.getNameLoc(), diag::err_ovl_no_viable_literal_operator)
3154     << R.getLookupName() << (int)ArgTys.size() << ArgTys[0]
3155     << (ArgTys.size() == 2 ? ArgTys[1] : QualType()) << AllowRaw
3156     << (AllowTemplate || AllowStringTemplate);
3157   return LOLR_Error;
3158 }
3159 
3160 void ADLResult::insert(NamedDecl *New) {
3161   NamedDecl *&Old = Decls[cast<NamedDecl>(New->getCanonicalDecl())];
3162 
3163   // If we haven't yet seen a decl for this key, or the last decl
3164   // was exactly this one, we're done.
3165   if (Old == nullptr || Old == New) {
3166     Old = New;
3167     return;
3168   }
3169 
3170   // Otherwise, decide which is a more recent redeclaration.
3171   FunctionDecl *OldFD = Old->getAsFunction();
3172   FunctionDecl *NewFD = New->getAsFunction();
3173 
3174   FunctionDecl *Cursor = NewFD;
3175   while (true) {
3176     Cursor = Cursor->getPreviousDecl();
3177 
3178     // If we got to the end without finding OldFD, OldFD is the newer
3179     // declaration;  leave things as they are.
3180     if (!Cursor) return;
3181 
3182     // If we do find OldFD, then NewFD is newer.
3183     if (Cursor == OldFD) break;
3184 
3185     // Otherwise, keep looking.
3186   }
3187 
3188   Old = New;
3189 }
3190 
3191 void Sema::ArgumentDependentLookup(DeclarationName Name, SourceLocation Loc,
3192                                    ArrayRef<Expr *> Args, ADLResult &Result) {
3193   // Find all of the associated namespaces and classes based on the
3194   // arguments we have.
3195   AssociatedNamespaceSet AssociatedNamespaces;
3196   AssociatedClassSet AssociatedClasses;
3197   FindAssociatedClassesAndNamespaces(Loc, Args,
3198                                      AssociatedNamespaces,
3199                                      AssociatedClasses);
3200 
3201   // C++ [basic.lookup.argdep]p3:
3202   //   Let X be the lookup set produced by unqualified lookup (3.4.1)
3203   //   and let Y be the lookup set produced by argument dependent
3204   //   lookup (defined as follows). If X contains [...] then Y is
3205   //   empty. Otherwise Y is the set of declarations found in the
3206   //   namespaces associated with the argument types as described
3207   //   below. The set of declarations found by the lookup of the name
3208   //   is the union of X and Y.
3209   //
3210   // Here, we compute Y and add its members to the overloaded
3211   // candidate set.
3212   for (auto *NS : AssociatedNamespaces) {
3213     //   When considering an associated namespace, the lookup is the
3214     //   same as the lookup performed when the associated namespace is
3215     //   used as a qualifier (3.4.3.2) except that:
3216     //
3217     //     -- Any using-directives in the associated namespace are
3218     //        ignored.
3219     //
3220     //     -- Any namespace-scope friend functions declared in
3221     //        associated classes are visible within their respective
3222     //        namespaces even if they are not visible during an ordinary
3223     //        lookup (11.4).
3224     DeclContext::lookup_result R = NS->lookup(Name);
3225     for (auto *D : R) {
3226       // If the only declaration here is an ordinary friend, consider
3227       // it only if it was declared in an associated classes.
3228       if ((D->getIdentifierNamespace() & Decl::IDNS_Ordinary) == 0) {
3229         // If it's neither ordinarily visible nor a friend, we can't find it.
3230         if ((D->getIdentifierNamespace() & Decl::IDNS_OrdinaryFriend) == 0)
3231           continue;
3232 
3233         bool DeclaredInAssociatedClass = false;
3234         for (Decl *DI = D; DI; DI = DI->getPreviousDecl()) {
3235           DeclContext *LexDC = DI->getLexicalDeclContext();
3236           if (isa<CXXRecordDecl>(LexDC) &&
3237               AssociatedClasses.count(cast<CXXRecordDecl>(LexDC)) &&
3238               isVisible(cast<NamedDecl>(DI))) {
3239             DeclaredInAssociatedClass = true;
3240             break;
3241           }
3242         }
3243         if (!DeclaredInAssociatedClass)
3244           continue;
3245       }
3246 
3247       if (isa<UsingShadowDecl>(D))
3248         D = cast<UsingShadowDecl>(D)->getTargetDecl();
3249 
3250       if (!isa<FunctionDecl>(D) && !isa<FunctionTemplateDecl>(D))
3251         continue;
3252 
3253       if (!isVisible(D) && !(D = findAcceptableDecl(*this, D)))
3254         continue;
3255 
3256       Result.insert(D);
3257     }
3258   }
3259 }
3260 
3261 //----------------------------------------------------------------------------
3262 // Search for all visible declarations.
3263 //----------------------------------------------------------------------------
3264 VisibleDeclConsumer::~VisibleDeclConsumer() { }
3265 
3266 bool VisibleDeclConsumer::includeHiddenDecls() const { return false; }
3267 
3268 namespace {
3269 
3270 class ShadowContextRAII;
3271 
3272 class VisibleDeclsRecord {
3273 public:
3274   /// \brief An entry in the shadow map, which is optimized to store a
3275   /// single declaration (the common case) but can also store a list
3276   /// of declarations.
3277   typedef llvm::TinyPtrVector<NamedDecl*> ShadowMapEntry;
3278 
3279 private:
3280   /// \brief A mapping from declaration names to the declarations that have
3281   /// this name within a particular scope.
3282   typedef llvm::DenseMap<DeclarationName, ShadowMapEntry> ShadowMap;
3283 
3284   /// \brief A list of shadow maps, which is used to model name hiding.
3285   std::list<ShadowMap> ShadowMaps;
3286 
3287   /// \brief The declaration contexts we have already visited.
3288   llvm::SmallPtrSet<DeclContext *, 8> VisitedContexts;
3289 
3290   friend class ShadowContextRAII;
3291 
3292 public:
3293   /// \brief Determine whether we have already visited this context
3294   /// (and, if not, note that we are going to visit that context now).
3295   bool visitedContext(DeclContext *Ctx) {
3296     return !VisitedContexts.insert(Ctx).second;
3297   }
3298 
3299   bool alreadyVisitedContext(DeclContext *Ctx) {
3300     return VisitedContexts.count(Ctx);
3301   }
3302 
3303   /// \brief Determine whether the given declaration is hidden in the
3304   /// current scope.
3305   ///
3306   /// \returns the declaration that hides the given declaration, or
3307   /// NULL if no such declaration exists.
3308   NamedDecl *checkHidden(NamedDecl *ND);
3309 
3310   /// \brief Add a declaration to the current shadow map.
3311   void add(NamedDecl *ND) {
3312     ShadowMaps.back()[ND->getDeclName()].push_back(ND);
3313   }
3314 };
3315 
3316 /// \brief RAII object that records when we've entered a shadow context.
3317 class ShadowContextRAII {
3318   VisibleDeclsRecord &Visible;
3319 
3320   typedef VisibleDeclsRecord::ShadowMap ShadowMap;
3321 
3322 public:
3323   ShadowContextRAII(VisibleDeclsRecord &Visible) : Visible(Visible) {
3324     Visible.ShadowMaps.emplace_back();
3325   }
3326 
3327   ~ShadowContextRAII() {
3328     Visible.ShadowMaps.pop_back();
3329   }
3330 };
3331 
3332 } // end anonymous namespace
3333 
3334 NamedDecl *VisibleDeclsRecord::checkHidden(NamedDecl *ND) {
3335   unsigned IDNS = ND->getIdentifierNamespace();
3336   std::list<ShadowMap>::reverse_iterator SM = ShadowMaps.rbegin();
3337   for (std::list<ShadowMap>::reverse_iterator SMEnd = ShadowMaps.rend();
3338        SM != SMEnd; ++SM) {
3339     ShadowMap::iterator Pos = SM->find(ND->getDeclName());
3340     if (Pos == SM->end())
3341       continue;
3342 
3343     for (auto *D : Pos->second) {
3344       // A tag declaration does not hide a non-tag declaration.
3345       if (D->hasTagIdentifierNamespace() &&
3346           (IDNS & (Decl::IDNS_Member | Decl::IDNS_Ordinary |
3347                    Decl::IDNS_ObjCProtocol)))
3348         continue;
3349 
3350       // Protocols are in distinct namespaces from everything else.
3351       if (((D->getIdentifierNamespace() & Decl::IDNS_ObjCProtocol)
3352            || (IDNS & Decl::IDNS_ObjCProtocol)) &&
3353           D->getIdentifierNamespace() != IDNS)
3354         continue;
3355 
3356       // Functions and function templates in the same scope overload
3357       // rather than hide.  FIXME: Look for hiding based on function
3358       // signatures!
3359       if (D->getUnderlyingDecl()->isFunctionOrFunctionTemplate() &&
3360           ND->getUnderlyingDecl()->isFunctionOrFunctionTemplate() &&
3361           SM == ShadowMaps.rbegin())
3362         continue;
3363 
3364       // We've found a declaration that hides this one.
3365       return D;
3366     }
3367   }
3368 
3369   return nullptr;
3370 }
3371 
3372 static void LookupVisibleDecls(DeclContext *Ctx, LookupResult &Result,
3373                                bool QualifiedNameLookup,
3374                                bool InBaseClass,
3375                                VisibleDeclConsumer &Consumer,
3376                                VisibleDeclsRecord &Visited) {
3377   if (!Ctx)
3378     return;
3379 
3380   // Make sure we don't visit the same context twice.
3381   if (Visited.visitedContext(Ctx->getPrimaryContext()))
3382     return;
3383 
3384   // Outside C++, lookup results for the TU live on identifiers.
3385   if (isa<TranslationUnitDecl>(Ctx) &&
3386       !Result.getSema().getLangOpts().CPlusPlus) {
3387     auto &S = Result.getSema();
3388     auto &Idents = S.Context.Idents;
3389 
3390     // Ensure all external identifiers are in the identifier table.
3391     if (IdentifierInfoLookup *External = Idents.getExternalIdentifierLookup()) {
3392       std::unique_ptr<IdentifierIterator> Iter(External->getIdentifiers());
3393       for (StringRef Name = Iter->Next(); !Name.empty(); Name = Iter->Next())
3394         Idents.get(Name);
3395     }
3396 
3397     // Walk all lookup results in the TU for each identifier.
3398     for (const auto &Ident : Idents) {
3399       for (auto I = S.IdResolver.begin(Ident.getValue()),
3400                 E = S.IdResolver.end();
3401            I != E; ++I) {
3402         if (S.IdResolver.isDeclInScope(*I, Ctx)) {
3403           if (NamedDecl *ND = Result.getAcceptableDecl(*I)) {
3404             Consumer.FoundDecl(ND, Visited.checkHidden(ND), Ctx, InBaseClass);
3405             Visited.add(ND);
3406           }
3407         }
3408       }
3409     }
3410 
3411     return;
3412   }
3413 
3414   if (CXXRecordDecl *Class = dyn_cast<CXXRecordDecl>(Ctx))
3415     Result.getSema().ForceDeclarationOfImplicitMembers(Class);
3416 
3417   // Enumerate all of the results in this context.
3418   for (DeclContextLookupResult R : Ctx->lookups()) {
3419     for (auto *D : R) {
3420       if (auto *ND = Result.getAcceptableDecl(D)) {
3421         Consumer.FoundDecl(ND, Visited.checkHidden(ND), Ctx, InBaseClass);
3422         Visited.add(ND);
3423       }
3424     }
3425   }
3426 
3427   // Traverse using directives for qualified name lookup.
3428   if (QualifiedNameLookup) {
3429     ShadowContextRAII Shadow(Visited);
3430     for (auto I : Ctx->using_directives()) {
3431       LookupVisibleDecls(I->getNominatedNamespace(), Result,
3432                          QualifiedNameLookup, InBaseClass, Consumer, Visited);
3433     }
3434   }
3435 
3436   // Traverse the contexts of inherited C++ classes.
3437   if (CXXRecordDecl *Record = dyn_cast<CXXRecordDecl>(Ctx)) {
3438     if (!Record->hasDefinition())
3439       return;
3440 
3441     for (const auto &B : Record->bases()) {
3442       QualType BaseType = B.getType();
3443 
3444       // Don't look into dependent bases, because name lookup can't look
3445       // there anyway.
3446       if (BaseType->isDependentType())
3447         continue;
3448 
3449       const RecordType *Record = BaseType->getAs<RecordType>();
3450       if (!Record)
3451         continue;
3452 
3453       // FIXME: It would be nice to be able to determine whether referencing
3454       // a particular member would be ambiguous. For example, given
3455       //
3456       //   struct A { int member; };
3457       //   struct B { int member; };
3458       //   struct C : A, B { };
3459       //
3460       //   void f(C *c) { c->### }
3461       //
3462       // accessing 'member' would result in an ambiguity. However, we
3463       // could be smart enough to qualify the member with the base
3464       // class, e.g.,
3465       //
3466       //   c->B::member
3467       //
3468       // or
3469       //
3470       //   c->A::member
3471 
3472       // Find results in this base class (and its bases).
3473       ShadowContextRAII Shadow(Visited);
3474       LookupVisibleDecls(Record->getDecl(), Result, QualifiedNameLookup,
3475                          true, Consumer, Visited);
3476     }
3477   }
3478 
3479   // Traverse the contexts of Objective-C classes.
3480   if (ObjCInterfaceDecl *IFace = dyn_cast<ObjCInterfaceDecl>(Ctx)) {
3481     // Traverse categories.
3482     for (auto *Cat : IFace->visible_categories()) {
3483       ShadowContextRAII Shadow(Visited);
3484       LookupVisibleDecls(Cat, Result, QualifiedNameLookup, false,
3485                          Consumer, Visited);
3486     }
3487 
3488     // Traverse protocols.
3489     for (auto *I : IFace->all_referenced_protocols()) {
3490       ShadowContextRAII Shadow(Visited);
3491       LookupVisibleDecls(I, Result, QualifiedNameLookup, false, Consumer,
3492                          Visited);
3493     }
3494 
3495     // Traverse the superclass.
3496     if (IFace->getSuperClass()) {
3497       ShadowContextRAII Shadow(Visited);
3498       LookupVisibleDecls(IFace->getSuperClass(), Result, QualifiedNameLookup,
3499                          true, Consumer, Visited);
3500     }
3501 
3502     // If there is an implementation, traverse it. We do this to find
3503     // synthesized ivars.
3504     if (IFace->getImplementation()) {
3505       ShadowContextRAII Shadow(Visited);
3506       LookupVisibleDecls(IFace->getImplementation(), Result,
3507                          QualifiedNameLookup, InBaseClass, Consumer, Visited);
3508     }
3509   } else if (ObjCProtocolDecl *Protocol = dyn_cast<ObjCProtocolDecl>(Ctx)) {
3510     for (auto *I : Protocol->protocols()) {
3511       ShadowContextRAII Shadow(Visited);
3512       LookupVisibleDecls(I, Result, QualifiedNameLookup, false, Consumer,
3513                          Visited);
3514     }
3515   } else if (ObjCCategoryDecl *Category = dyn_cast<ObjCCategoryDecl>(Ctx)) {
3516     for (auto *I : Category->protocols()) {
3517       ShadowContextRAII Shadow(Visited);
3518       LookupVisibleDecls(I, Result, QualifiedNameLookup, false, Consumer,
3519                          Visited);
3520     }
3521 
3522     // If there is an implementation, traverse it.
3523     if (Category->getImplementation()) {
3524       ShadowContextRAII Shadow(Visited);
3525       LookupVisibleDecls(Category->getImplementation(), Result,
3526                          QualifiedNameLookup, true, Consumer, Visited);
3527     }
3528   }
3529 }
3530 
3531 static void LookupVisibleDecls(Scope *S, LookupResult &Result,
3532                                UnqualUsingDirectiveSet &UDirs,
3533                                VisibleDeclConsumer &Consumer,
3534                                VisibleDeclsRecord &Visited) {
3535   if (!S)
3536     return;
3537 
3538   if (!S->getEntity() ||
3539       (!S->getParent() &&
3540        !Visited.alreadyVisitedContext(S->getEntity())) ||
3541       (S->getEntity())->isFunctionOrMethod()) {
3542     FindLocalExternScope FindLocals(Result);
3543     // Walk through the declarations in this Scope.
3544     for (auto *D : S->decls()) {
3545       if (NamedDecl *ND = dyn_cast<NamedDecl>(D))
3546         if ((ND = Result.getAcceptableDecl(ND))) {
3547           Consumer.FoundDecl(ND, Visited.checkHidden(ND), nullptr, false);
3548           Visited.add(ND);
3549         }
3550     }
3551   }
3552 
3553   // FIXME: C++ [temp.local]p8
3554   DeclContext *Entity = nullptr;
3555   if (S->getEntity()) {
3556     // Look into this scope's declaration context, along with any of its
3557     // parent lookup contexts (e.g., enclosing classes), up to the point
3558     // where we hit the context stored in the next outer scope.
3559     Entity = S->getEntity();
3560     DeclContext *OuterCtx = findOuterContext(S).first; // FIXME
3561 
3562     for (DeclContext *Ctx = Entity; Ctx && !Ctx->Equals(OuterCtx);
3563          Ctx = Ctx->getLookupParent()) {
3564       if (ObjCMethodDecl *Method = dyn_cast<ObjCMethodDecl>(Ctx)) {
3565         if (Method->isInstanceMethod()) {
3566           // For instance methods, look for ivars in the method's interface.
3567           LookupResult IvarResult(Result.getSema(), Result.getLookupName(),
3568                                   Result.getNameLoc(), Sema::LookupMemberName);
3569           if (ObjCInterfaceDecl *IFace = Method->getClassInterface()) {
3570             LookupVisibleDecls(IFace, IvarResult, /*QualifiedNameLookup=*/false,
3571                                /*InBaseClass=*/false, Consumer, Visited);
3572           }
3573         }
3574 
3575         // We've already performed all of the name lookup that we need
3576         // to for Objective-C methods; the next context will be the
3577         // outer scope.
3578         break;
3579       }
3580 
3581       if (Ctx->isFunctionOrMethod())
3582         continue;
3583 
3584       LookupVisibleDecls(Ctx, Result, /*QualifiedNameLookup=*/false,
3585                          /*InBaseClass=*/false, Consumer, Visited);
3586     }
3587   } else if (!S->getParent()) {
3588     // Look into the translation unit scope. We walk through the translation
3589     // unit's declaration context, because the Scope itself won't have all of
3590     // the declarations if we loaded a precompiled header.
3591     // FIXME: We would like the translation unit's Scope object to point to the
3592     // translation unit, so we don't need this special "if" branch. However,
3593     // doing so would force the normal C++ name-lookup code to look into the
3594     // translation unit decl when the IdentifierInfo chains would suffice.
3595     // Once we fix that problem (which is part of a more general "don't look
3596     // in DeclContexts unless we have to" optimization), we can eliminate this.
3597     Entity = Result.getSema().Context.getTranslationUnitDecl();
3598     LookupVisibleDecls(Entity, Result, /*QualifiedNameLookup=*/false,
3599                        /*InBaseClass=*/false, Consumer, Visited);
3600   }
3601 
3602   if (Entity) {
3603     // Lookup visible declarations in any namespaces found by using
3604     // directives.
3605     for (const UnqualUsingEntry &UUE : UDirs.getNamespacesFor(Entity))
3606       LookupVisibleDecls(const_cast<DeclContext *>(UUE.getNominatedNamespace()),
3607                          Result, /*QualifiedNameLookup=*/false,
3608                          /*InBaseClass=*/false, Consumer, Visited);
3609   }
3610 
3611   // Lookup names in the parent scope.
3612   ShadowContextRAII Shadow(Visited);
3613   LookupVisibleDecls(S->getParent(), Result, UDirs, Consumer, Visited);
3614 }
3615 
3616 void Sema::LookupVisibleDecls(Scope *S, LookupNameKind Kind,
3617                               VisibleDeclConsumer &Consumer,
3618                               bool IncludeGlobalScope) {
3619   // Determine the set of using directives available during
3620   // unqualified name lookup.
3621   Scope *Initial = S;
3622   UnqualUsingDirectiveSet UDirs;
3623   if (getLangOpts().CPlusPlus) {
3624     // Find the first namespace or translation-unit scope.
3625     while (S && !isNamespaceOrTranslationUnitScope(S))
3626       S = S->getParent();
3627 
3628     UDirs.visitScopeChain(Initial, S);
3629   }
3630   UDirs.done();
3631 
3632   // Look for visible declarations.
3633   LookupResult Result(*this, DeclarationName(), SourceLocation(), Kind);
3634   Result.setAllowHidden(Consumer.includeHiddenDecls());
3635   VisibleDeclsRecord Visited;
3636   if (!IncludeGlobalScope)
3637     Visited.visitedContext(Context.getTranslationUnitDecl());
3638   ShadowContextRAII Shadow(Visited);
3639   ::LookupVisibleDecls(Initial, Result, UDirs, Consumer, Visited);
3640 }
3641 
3642 void Sema::LookupVisibleDecls(DeclContext *Ctx, LookupNameKind Kind,
3643                               VisibleDeclConsumer &Consumer,
3644                               bool IncludeGlobalScope) {
3645   LookupResult Result(*this, DeclarationName(), SourceLocation(), Kind);
3646   Result.setAllowHidden(Consumer.includeHiddenDecls());
3647   VisibleDeclsRecord Visited;
3648   if (!IncludeGlobalScope)
3649     Visited.visitedContext(Context.getTranslationUnitDecl());
3650   ShadowContextRAII Shadow(Visited);
3651   ::LookupVisibleDecls(Ctx, Result, /*QualifiedNameLookup=*/true,
3652                        /*InBaseClass=*/false, Consumer, Visited);
3653 }
3654 
3655 /// LookupOrCreateLabel - Do a name lookup of a label with the specified name.
3656 /// If GnuLabelLoc is a valid source location, then this is a definition
3657 /// of an __label__ label name, otherwise it is a normal label definition
3658 /// or use.
3659 LabelDecl *Sema::LookupOrCreateLabel(IdentifierInfo *II, SourceLocation Loc,
3660                                      SourceLocation GnuLabelLoc) {
3661   // Do a lookup to see if we have a label with this name already.
3662   NamedDecl *Res = nullptr;
3663 
3664   if (GnuLabelLoc.isValid()) {
3665     // Local label definitions always shadow existing labels.
3666     Res = LabelDecl::Create(Context, CurContext, Loc, II, GnuLabelLoc);
3667     Scope *S = CurScope;
3668     PushOnScopeChains(Res, S, true);
3669     return cast<LabelDecl>(Res);
3670   }
3671 
3672   // Not a GNU local label.
3673   Res = LookupSingleName(CurScope, II, Loc, LookupLabel, NotForRedeclaration);
3674   // If we found a label, check to see if it is in the same context as us.
3675   // When in a Block, we don't want to reuse a label in an enclosing function.
3676   if (Res && Res->getDeclContext() != CurContext)
3677     Res = nullptr;
3678   if (!Res) {
3679     // If not forward referenced or defined already, create the backing decl.
3680     Res = LabelDecl::Create(Context, CurContext, Loc, II);
3681     Scope *S = CurScope->getFnParent();
3682     assert(S && "Not in a function?");
3683     PushOnScopeChains(Res, S, true);
3684   }
3685   return cast<LabelDecl>(Res);
3686 }
3687 
3688 //===----------------------------------------------------------------------===//
3689 // Typo correction
3690 //===----------------------------------------------------------------------===//
3691 
3692 static bool isCandidateViable(CorrectionCandidateCallback &CCC,
3693                               TypoCorrection &Candidate) {
3694   Candidate.setCallbackDistance(CCC.RankCandidate(Candidate));
3695   return Candidate.getEditDistance(false) != TypoCorrection::InvalidDistance;
3696 }
3697 
3698 static void LookupPotentialTypoResult(Sema &SemaRef,
3699                                       LookupResult &Res,
3700                                       IdentifierInfo *Name,
3701                                       Scope *S, CXXScopeSpec *SS,
3702                                       DeclContext *MemberContext,
3703                                       bool EnteringContext,
3704                                       bool isObjCIvarLookup,
3705                                       bool FindHidden);
3706 
3707 /// \brief Check whether the declarations found for a typo correction are
3708 /// visible, and if none of them are, convert the correction to an 'import
3709 /// a module' correction.
3710 static void checkCorrectionVisibility(Sema &SemaRef, TypoCorrection &TC) {
3711   if (TC.begin() == TC.end())
3712     return;
3713 
3714   TypoCorrection::decl_iterator DI = TC.begin(), DE = TC.end();
3715 
3716   for (/**/; DI != DE; ++DI)
3717     if (!LookupResult::isVisible(SemaRef, *DI))
3718       break;
3719   // Nothing to do if all decls are visible.
3720   if (DI == DE)
3721     return;
3722 
3723   llvm::SmallVector<NamedDecl*, 4> NewDecls(TC.begin(), DI);
3724   bool AnyVisibleDecls = !NewDecls.empty();
3725 
3726   for (/**/; DI != DE; ++DI) {
3727     NamedDecl *VisibleDecl = *DI;
3728     if (!LookupResult::isVisible(SemaRef, *DI))
3729       VisibleDecl = findAcceptableDecl(SemaRef, *DI);
3730 
3731     if (VisibleDecl) {
3732       if (!AnyVisibleDecls) {
3733         // Found a visible decl, discard all hidden ones.
3734         AnyVisibleDecls = true;
3735         NewDecls.clear();
3736       }
3737       NewDecls.push_back(VisibleDecl);
3738     } else if (!AnyVisibleDecls && !(*DI)->isModulePrivate())
3739       NewDecls.push_back(*DI);
3740   }
3741 
3742   if (NewDecls.empty())
3743     TC = TypoCorrection();
3744   else {
3745     TC.setCorrectionDecls(NewDecls);
3746     TC.setRequiresImport(!AnyVisibleDecls);
3747   }
3748 }
3749 
3750 // Fill the supplied vector with the IdentifierInfo pointers for each piece of
3751 // the given NestedNameSpecifier (i.e. given a NestedNameSpecifier "foo::bar::",
3752 // fill the vector with the IdentifierInfo pointers for "foo" and "bar").
3753 static void getNestedNameSpecifierIdentifiers(
3754     NestedNameSpecifier *NNS,
3755     SmallVectorImpl<const IdentifierInfo*> &Identifiers) {
3756   if (NestedNameSpecifier *Prefix = NNS->getPrefix())
3757     getNestedNameSpecifierIdentifiers(Prefix, Identifiers);
3758   else
3759     Identifiers.clear();
3760 
3761   const IdentifierInfo *II = nullptr;
3762 
3763   switch (NNS->getKind()) {
3764   case NestedNameSpecifier::Identifier:
3765     II = NNS->getAsIdentifier();
3766     break;
3767 
3768   case NestedNameSpecifier::Namespace:
3769     if (NNS->getAsNamespace()->isAnonymousNamespace())
3770       return;
3771     II = NNS->getAsNamespace()->getIdentifier();
3772     break;
3773 
3774   case NestedNameSpecifier::NamespaceAlias:
3775     II = NNS->getAsNamespaceAlias()->getIdentifier();
3776     break;
3777 
3778   case NestedNameSpecifier::TypeSpecWithTemplate:
3779   case NestedNameSpecifier::TypeSpec:
3780     II = QualType(NNS->getAsType(), 0).getBaseTypeIdentifier();
3781     break;
3782 
3783   case NestedNameSpecifier::Global:
3784   case NestedNameSpecifier::Super:
3785     return;
3786   }
3787 
3788   if (II)
3789     Identifiers.push_back(II);
3790 }
3791 
3792 void TypoCorrectionConsumer::FoundDecl(NamedDecl *ND, NamedDecl *Hiding,
3793                                        DeclContext *Ctx, bool InBaseClass) {
3794   // Don't consider hidden names for typo correction.
3795   if (Hiding)
3796     return;
3797 
3798   // Only consider entities with identifiers for names, ignoring
3799   // special names (constructors, overloaded operators, selectors,
3800   // etc.).
3801   IdentifierInfo *Name = ND->getIdentifier();
3802   if (!Name)
3803     return;
3804 
3805   // Only consider visible declarations and declarations from modules with
3806   // names that exactly match.
3807   if (!LookupResult::isVisible(SemaRef, ND) && Name != Typo &&
3808       !findAcceptableDecl(SemaRef, ND))
3809     return;
3810 
3811   FoundName(Name->getName());
3812 }
3813 
3814 void TypoCorrectionConsumer::FoundName(StringRef Name) {
3815   // Compute the edit distance between the typo and the name of this
3816   // entity, and add the identifier to the list of results.
3817   addName(Name, nullptr);
3818 }
3819 
3820 void TypoCorrectionConsumer::addKeywordResult(StringRef Keyword) {
3821   // Compute the edit distance between the typo and this keyword,
3822   // and add the keyword to the list of results.
3823   addName(Keyword, nullptr, nullptr, true);
3824 }
3825 
3826 void TypoCorrectionConsumer::addName(StringRef Name, NamedDecl *ND,
3827                                      NestedNameSpecifier *NNS, bool isKeyword) {
3828   // Use a simple length-based heuristic to determine the minimum possible
3829   // edit distance. If the minimum isn't good enough, bail out early.
3830   StringRef TypoStr = Typo->getName();
3831   unsigned MinED = abs((int)Name.size() - (int)TypoStr.size());
3832   if (MinED && TypoStr.size() / MinED < 3)
3833     return;
3834 
3835   // Compute an upper bound on the allowable edit distance, so that the
3836   // edit-distance algorithm can short-circuit.
3837   unsigned UpperBound = (TypoStr.size() + 2) / 3 + 1;
3838   unsigned ED = TypoStr.edit_distance(Name, true, UpperBound);
3839   if (ED >= UpperBound) return;
3840 
3841   TypoCorrection TC(&SemaRef.Context.Idents.get(Name), ND, NNS, ED);
3842   if (isKeyword) TC.makeKeyword();
3843   TC.setCorrectionRange(nullptr, Result.getLookupNameInfo());
3844   addCorrection(TC);
3845 }
3846 
3847 static const unsigned MaxTypoDistanceResultSets = 5;
3848 
3849 void TypoCorrectionConsumer::addCorrection(TypoCorrection Correction) {
3850   StringRef TypoStr = Typo->getName();
3851   StringRef Name = Correction.getCorrectionAsIdentifierInfo()->getName();
3852 
3853   // For very short typos, ignore potential corrections that have a different
3854   // base identifier from the typo or which have a normalized edit distance
3855   // longer than the typo itself.
3856   if (TypoStr.size() < 3 &&
3857       (Name != TypoStr || Correction.getEditDistance(true) > TypoStr.size()))
3858     return;
3859 
3860   // If the correction is resolved but is not viable, ignore it.
3861   if (Correction.isResolved()) {
3862     checkCorrectionVisibility(SemaRef, Correction);
3863     if (!Correction || !isCandidateViable(*CorrectionValidator, Correction))
3864       return;
3865   }
3866 
3867   TypoResultList &CList =
3868       CorrectionResults[Correction.getEditDistance(false)][Name];
3869 
3870   if (!CList.empty() && !CList.back().isResolved())
3871     CList.pop_back();
3872   if (NamedDecl *NewND = Correction.getCorrectionDecl()) {
3873     std::string CorrectionStr = Correction.getAsString(SemaRef.getLangOpts());
3874     for (TypoResultList::iterator RI = CList.begin(), RIEnd = CList.end();
3875          RI != RIEnd; ++RI) {
3876       // If the Correction refers to a decl already in the result list,
3877       // replace the existing result if the string representation of Correction
3878       // comes before the current result alphabetically, then stop as there is
3879       // nothing more to be done to add Correction to the candidate set.
3880       if (RI->getCorrectionDecl() == NewND) {
3881         if (CorrectionStr < RI->getAsString(SemaRef.getLangOpts()))
3882           *RI = Correction;
3883         return;
3884       }
3885     }
3886   }
3887   if (CList.empty() || Correction.isResolved())
3888     CList.push_back(Correction);
3889 
3890   while (CorrectionResults.size() > MaxTypoDistanceResultSets)
3891     CorrectionResults.erase(std::prev(CorrectionResults.end()));
3892 }
3893 
3894 void TypoCorrectionConsumer::addNamespaces(
3895     const llvm::MapVector<NamespaceDecl *, bool> &KnownNamespaces) {
3896   SearchNamespaces = true;
3897 
3898   for (auto KNPair : KnownNamespaces)
3899     Namespaces.addNameSpecifier(KNPair.first);
3900 
3901   bool SSIsTemplate = false;
3902   if (NestedNameSpecifier *NNS =
3903           (SS && SS->isValid()) ? SS->getScopeRep() : nullptr) {
3904     if (const Type *T = NNS->getAsType())
3905       SSIsTemplate = T->getTypeClass() == Type::TemplateSpecialization;
3906   }
3907   // Do not transform this into an iterator-based loop. The loop body can
3908   // trigger the creation of further types (through lazy deserialization) and
3909   // invalide iterators into this list.
3910   auto &Types = SemaRef.getASTContext().getTypes();
3911   for (unsigned I = 0; I != Types.size(); ++I) {
3912     const auto *TI = Types[I];
3913     if (CXXRecordDecl *CD = TI->getAsCXXRecordDecl()) {
3914       CD = CD->getCanonicalDecl();
3915       if (!CD->isDependentType() && !CD->isAnonymousStructOrUnion() &&
3916           !CD->isUnion() && CD->getIdentifier() &&
3917           (SSIsTemplate || !isa<ClassTemplateSpecializationDecl>(CD)) &&
3918           (CD->isBeingDefined() || CD->isCompleteDefinition()))
3919         Namespaces.addNameSpecifier(CD);
3920     }
3921   }
3922 }
3923 
3924 const TypoCorrection &TypoCorrectionConsumer::getNextCorrection() {
3925   if (++CurrentTCIndex < ValidatedCorrections.size())
3926     return ValidatedCorrections[CurrentTCIndex];
3927 
3928   CurrentTCIndex = ValidatedCorrections.size();
3929   while (!CorrectionResults.empty()) {
3930     auto DI = CorrectionResults.begin();
3931     if (DI->second.empty()) {
3932       CorrectionResults.erase(DI);
3933       continue;
3934     }
3935 
3936     auto RI = DI->second.begin();
3937     if (RI->second.empty()) {
3938       DI->second.erase(RI);
3939       performQualifiedLookups();
3940       continue;
3941     }
3942 
3943     TypoCorrection TC = RI->second.pop_back_val();
3944     if (TC.isResolved() || TC.requiresImport() || resolveCorrection(TC)) {
3945       ValidatedCorrections.push_back(TC);
3946       return ValidatedCorrections[CurrentTCIndex];
3947     }
3948   }
3949   return ValidatedCorrections[0];  // The empty correction.
3950 }
3951 
3952 bool TypoCorrectionConsumer::resolveCorrection(TypoCorrection &Candidate) {
3953   IdentifierInfo *Name = Candidate.getCorrectionAsIdentifierInfo();
3954   DeclContext *TempMemberContext = MemberContext;
3955   CXXScopeSpec *TempSS = SS.get();
3956 retry_lookup:
3957   LookupPotentialTypoResult(SemaRef, Result, Name, S, TempSS, TempMemberContext,
3958                             EnteringContext,
3959                             CorrectionValidator->IsObjCIvarLookup,
3960                             Name == Typo && !Candidate.WillReplaceSpecifier());
3961   switch (Result.getResultKind()) {
3962   case LookupResult::NotFound:
3963   case LookupResult::NotFoundInCurrentInstantiation:
3964   case LookupResult::FoundUnresolvedValue:
3965     if (TempSS) {
3966       // Immediately retry the lookup without the given CXXScopeSpec
3967       TempSS = nullptr;
3968       Candidate.WillReplaceSpecifier(true);
3969       goto retry_lookup;
3970     }
3971     if (TempMemberContext) {
3972       if (SS && !TempSS)
3973         TempSS = SS.get();
3974       TempMemberContext = nullptr;
3975       goto retry_lookup;
3976     }
3977     if (SearchNamespaces)
3978       QualifiedResults.push_back(Candidate);
3979     break;
3980 
3981   case LookupResult::Ambiguous:
3982     // We don't deal with ambiguities.
3983     break;
3984 
3985   case LookupResult::Found:
3986   case LookupResult::FoundOverloaded:
3987     // Store all of the Decls for overloaded symbols
3988     for (auto *TRD : Result)
3989       Candidate.addCorrectionDecl(TRD);
3990     checkCorrectionVisibility(SemaRef, Candidate);
3991     if (!isCandidateViable(*CorrectionValidator, Candidate)) {
3992       if (SearchNamespaces)
3993         QualifiedResults.push_back(Candidate);
3994       break;
3995     }
3996     Candidate.setCorrectionRange(SS.get(), Result.getLookupNameInfo());
3997     return true;
3998   }
3999   return false;
4000 }
4001 
4002 void TypoCorrectionConsumer::performQualifiedLookups() {
4003   unsigned TypoLen = Typo->getName().size();
4004   for (auto QR : QualifiedResults) {
4005     for (auto NSI : Namespaces) {
4006       DeclContext *Ctx = NSI.DeclCtx;
4007       const Type *NSType = NSI.NameSpecifier->getAsType();
4008 
4009       // If the current NestedNameSpecifier refers to a class and the
4010       // current correction candidate is the name of that class, then skip
4011       // it as it is unlikely a qualified version of the class' constructor
4012       // is an appropriate correction.
4013       if (CXXRecordDecl *NSDecl = NSType ? NSType->getAsCXXRecordDecl() :
4014                                            nullptr) {
4015         if (NSDecl->getIdentifier() == QR.getCorrectionAsIdentifierInfo())
4016           continue;
4017       }
4018 
4019       TypoCorrection TC(QR);
4020       TC.ClearCorrectionDecls();
4021       TC.setCorrectionSpecifier(NSI.NameSpecifier);
4022       TC.setQualifierDistance(NSI.EditDistance);
4023       TC.setCallbackDistance(0); // Reset the callback distance
4024 
4025       // If the current correction candidate and namespace combination are
4026       // too far away from the original typo based on the normalized edit
4027       // distance, then skip performing a qualified name lookup.
4028       unsigned TmpED = TC.getEditDistance(true);
4029       if (QR.getCorrectionAsIdentifierInfo() != Typo && TmpED &&
4030           TypoLen / TmpED < 3)
4031         continue;
4032 
4033       Result.clear();
4034       Result.setLookupName(QR.getCorrectionAsIdentifierInfo());
4035       if (!SemaRef.LookupQualifiedName(Result, Ctx))
4036         continue;
4037 
4038       // Any corrections added below will be validated in subsequent
4039       // iterations of the main while() loop over the Consumer's contents.
4040       switch (Result.getResultKind()) {
4041       case LookupResult::Found:
4042       case LookupResult::FoundOverloaded: {
4043         if (SS && SS->isValid()) {
4044           std::string NewQualified = TC.getAsString(SemaRef.getLangOpts());
4045           std::string OldQualified;
4046           llvm::raw_string_ostream OldOStream(OldQualified);
4047           SS->getScopeRep()->print(OldOStream, SemaRef.getPrintingPolicy());
4048           OldOStream << Typo->getName();
4049           // If correction candidate would be an identical written qualified
4050           // identifer, then the existing CXXScopeSpec probably included a
4051           // typedef that didn't get accounted for properly.
4052           if (OldOStream.str() == NewQualified)
4053             break;
4054         }
4055         for (LookupResult::iterator TRD = Result.begin(), TRDEnd = Result.end();
4056              TRD != TRDEnd; ++TRD) {
4057           if (SemaRef.CheckMemberAccess(TC.getCorrectionRange().getBegin(),
4058                                         NSType ? NSType->getAsCXXRecordDecl()
4059                                                : nullptr,
4060                                         TRD.getPair()) == Sema::AR_accessible)
4061             TC.addCorrectionDecl(*TRD);
4062         }
4063         if (TC.isResolved()) {
4064           TC.setCorrectionRange(SS.get(), Result.getLookupNameInfo());
4065           addCorrection(TC);
4066         }
4067         break;
4068       }
4069       case LookupResult::NotFound:
4070       case LookupResult::NotFoundInCurrentInstantiation:
4071       case LookupResult::Ambiguous:
4072       case LookupResult::FoundUnresolvedValue:
4073         break;
4074       }
4075     }
4076   }
4077   QualifiedResults.clear();
4078 }
4079 
4080 TypoCorrectionConsumer::NamespaceSpecifierSet::NamespaceSpecifierSet(
4081     ASTContext &Context, DeclContext *CurContext, CXXScopeSpec *CurScopeSpec)
4082     : Context(Context), CurContextChain(buildContextChain(CurContext)) {
4083   if (NestedNameSpecifier *NNS =
4084           CurScopeSpec ? CurScopeSpec->getScopeRep() : nullptr) {
4085     llvm::raw_string_ostream SpecifierOStream(CurNameSpecifier);
4086     NNS->print(SpecifierOStream, Context.getPrintingPolicy());
4087 
4088     getNestedNameSpecifierIdentifiers(NNS, CurNameSpecifierIdentifiers);
4089   }
4090   // Build the list of identifiers that would be used for an absolute
4091   // (from the global context) NestedNameSpecifier referring to the current
4092   // context.
4093   for (DeclContextList::reverse_iterator C = CurContextChain.rbegin(),
4094                                          CEnd = CurContextChain.rend();
4095        C != CEnd; ++C) {
4096     if (NamespaceDecl *ND = dyn_cast_or_null<NamespaceDecl>(*C))
4097       CurContextIdentifiers.push_back(ND->getIdentifier());
4098   }
4099 
4100   // Add the global context as a NestedNameSpecifier
4101   SpecifierInfo SI = {cast<DeclContext>(Context.getTranslationUnitDecl()),
4102                       NestedNameSpecifier::GlobalSpecifier(Context), 1};
4103   DistanceMap[1].push_back(SI);
4104 }
4105 
4106 auto TypoCorrectionConsumer::NamespaceSpecifierSet::buildContextChain(
4107     DeclContext *Start) -> DeclContextList {
4108   assert(Start && "Building a context chain from a null context");
4109   DeclContextList Chain;
4110   for (DeclContext *DC = Start->getPrimaryContext(); DC != nullptr;
4111        DC = DC->getLookupParent()) {
4112     NamespaceDecl *ND = dyn_cast_or_null<NamespaceDecl>(DC);
4113     if (!DC->isInlineNamespace() && !DC->isTransparentContext() &&
4114         !(ND && ND->isAnonymousNamespace()))
4115       Chain.push_back(DC->getPrimaryContext());
4116   }
4117   return Chain;
4118 }
4119 
4120 unsigned
4121 TypoCorrectionConsumer::NamespaceSpecifierSet::buildNestedNameSpecifier(
4122     DeclContextList &DeclChain, NestedNameSpecifier *&NNS) {
4123   unsigned NumSpecifiers = 0;
4124   for (DeclContextList::reverse_iterator C = DeclChain.rbegin(),
4125                                       CEnd = DeclChain.rend();
4126        C != CEnd; ++C) {
4127     if (NamespaceDecl *ND = dyn_cast_or_null<NamespaceDecl>(*C)) {
4128       NNS = NestedNameSpecifier::Create(Context, NNS, ND);
4129       ++NumSpecifiers;
4130     } else if (RecordDecl *RD = dyn_cast_or_null<RecordDecl>(*C)) {
4131       NNS = NestedNameSpecifier::Create(Context, NNS, RD->isTemplateDecl(),
4132                                         RD->getTypeForDecl());
4133       ++NumSpecifiers;
4134     }
4135   }
4136   return NumSpecifiers;
4137 }
4138 
4139 void TypoCorrectionConsumer::NamespaceSpecifierSet::addNameSpecifier(
4140     DeclContext *Ctx) {
4141   NestedNameSpecifier *NNS = nullptr;
4142   unsigned NumSpecifiers = 0;
4143   DeclContextList NamespaceDeclChain(buildContextChain(Ctx));
4144   DeclContextList FullNamespaceDeclChain(NamespaceDeclChain);
4145 
4146   // Eliminate common elements from the two DeclContext chains.
4147   for (DeclContextList::reverse_iterator C = CurContextChain.rbegin(),
4148                                       CEnd = CurContextChain.rend();
4149        C != CEnd && !NamespaceDeclChain.empty() &&
4150        NamespaceDeclChain.back() == *C; ++C) {
4151     NamespaceDeclChain.pop_back();
4152   }
4153 
4154   // Build the NestedNameSpecifier from what is left of the NamespaceDeclChain
4155   NumSpecifiers = buildNestedNameSpecifier(NamespaceDeclChain, NNS);
4156 
4157   // Add an explicit leading '::' specifier if needed.
4158   if (NamespaceDeclChain.empty()) {
4159     // Rebuild the NestedNameSpecifier as a globally-qualified specifier.
4160     NNS = NestedNameSpecifier::GlobalSpecifier(Context);
4161     NumSpecifiers =
4162         buildNestedNameSpecifier(FullNamespaceDeclChain, NNS);
4163   } else if (NamedDecl *ND =
4164                  dyn_cast_or_null<NamedDecl>(NamespaceDeclChain.back())) {
4165     IdentifierInfo *Name = ND->getIdentifier();
4166     bool SameNameSpecifier = false;
4167     if (std::find(CurNameSpecifierIdentifiers.begin(),
4168                   CurNameSpecifierIdentifiers.end(),
4169                   Name) != CurNameSpecifierIdentifiers.end()) {
4170       std::string NewNameSpecifier;
4171       llvm::raw_string_ostream SpecifierOStream(NewNameSpecifier);
4172       SmallVector<const IdentifierInfo *, 4> NewNameSpecifierIdentifiers;
4173       getNestedNameSpecifierIdentifiers(NNS, NewNameSpecifierIdentifiers);
4174       NNS->print(SpecifierOStream, Context.getPrintingPolicy());
4175       SpecifierOStream.flush();
4176       SameNameSpecifier = NewNameSpecifier == CurNameSpecifier;
4177     }
4178     if (SameNameSpecifier ||
4179         std::find(CurContextIdentifiers.begin(), CurContextIdentifiers.end(),
4180                   Name) != CurContextIdentifiers.end()) {
4181       // Rebuild the NestedNameSpecifier as a globally-qualified specifier.
4182       NNS = NestedNameSpecifier::GlobalSpecifier(Context);
4183       NumSpecifiers =
4184           buildNestedNameSpecifier(FullNamespaceDeclChain, NNS);
4185     }
4186   }
4187 
4188   // If the built NestedNameSpecifier would be replacing an existing
4189   // NestedNameSpecifier, use the number of component identifiers that
4190   // would need to be changed as the edit distance instead of the number
4191   // of components in the built NestedNameSpecifier.
4192   if (NNS && !CurNameSpecifierIdentifiers.empty()) {
4193     SmallVector<const IdentifierInfo*, 4> NewNameSpecifierIdentifiers;
4194     getNestedNameSpecifierIdentifiers(NNS, NewNameSpecifierIdentifiers);
4195     NumSpecifiers = llvm::ComputeEditDistance(
4196         llvm::makeArrayRef(CurNameSpecifierIdentifiers),
4197         llvm::makeArrayRef(NewNameSpecifierIdentifiers));
4198   }
4199 
4200   SpecifierInfo SI = {Ctx, NNS, NumSpecifiers};
4201   DistanceMap[NumSpecifiers].push_back(SI);
4202 }
4203 
4204 /// \brief Perform name lookup for a possible result for typo correction.
4205 static void LookupPotentialTypoResult(Sema &SemaRef,
4206                                       LookupResult &Res,
4207                                       IdentifierInfo *Name,
4208                                       Scope *S, CXXScopeSpec *SS,
4209                                       DeclContext *MemberContext,
4210                                       bool EnteringContext,
4211                                       bool isObjCIvarLookup,
4212                                       bool FindHidden) {
4213   Res.suppressDiagnostics();
4214   Res.clear();
4215   Res.setLookupName(Name);
4216   Res.setAllowHidden(FindHidden);
4217   if (MemberContext) {
4218     if (ObjCInterfaceDecl *Class = dyn_cast<ObjCInterfaceDecl>(MemberContext)) {
4219       if (isObjCIvarLookup) {
4220         if (ObjCIvarDecl *Ivar = Class->lookupInstanceVariable(Name)) {
4221           Res.addDecl(Ivar);
4222           Res.resolveKind();
4223           return;
4224         }
4225       }
4226 
4227       if (ObjCPropertyDecl *Prop = Class->FindPropertyDeclaration(
4228               Name, ObjCPropertyQueryKind::OBJC_PR_query_instance)) {
4229         Res.addDecl(Prop);
4230         Res.resolveKind();
4231         return;
4232       }
4233     }
4234 
4235     SemaRef.LookupQualifiedName(Res, MemberContext);
4236     return;
4237   }
4238 
4239   SemaRef.LookupParsedName(Res, S, SS, /*AllowBuiltinCreation=*/false,
4240                            EnteringContext);
4241 
4242   // Fake ivar lookup; this should really be part of
4243   // LookupParsedName.
4244   if (ObjCMethodDecl *Method = SemaRef.getCurMethodDecl()) {
4245     if (Method->isInstanceMethod() && Method->getClassInterface() &&
4246         (Res.empty() ||
4247          (Res.isSingleResult() &&
4248           Res.getFoundDecl()->isDefinedOutsideFunctionOrMethod()))) {
4249        if (ObjCIvarDecl *IV
4250              = Method->getClassInterface()->lookupInstanceVariable(Name)) {
4251          Res.addDecl(IV);
4252          Res.resolveKind();
4253        }
4254      }
4255   }
4256 }
4257 
4258 /// \brief Add keywords to the consumer as possible typo corrections.
4259 static void AddKeywordsToConsumer(Sema &SemaRef,
4260                                   TypoCorrectionConsumer &Consumer,
4261                                   Scope *S, CorrectionCandidateCallback &CCC,
4262                                   bool AfterNestedNameSpecifier) {
4263   if (AfterNestedNameSpecifier) {
4264     // For 'X::', we know exactly which keywords can appear next.
4265     Consumer.addKeywordResult("template");
4266     if (CCC.WantExpressionKeywords)
4267       Consumer.addKeywordResult("operator");
4268     return;
4269   }
4270 
4271   if (CCC.WantObjCSuper)
4272     Consumer.addKeywordResult("super");
4273 
4274   if (CCC.WantTypeSpecifiers) {
4275     // Add type-specifier keywords to the set of results.
4276     static const char *const CTypeSpecs[] = {
4277       "char", "const", "double", "enum", "float", "int", "long", "short",
4278       "signed", "struct", "union", "unsigned", "void", "volatile",
4279       "_Complex", "_Imaginary",
4280       // storage-specifiers as well
4281       "extern", "inline", "static", "typedef"
4282     };
4283 
4284     const unsigned NumCTypeSpecs = llvm::array_lengthof(CTypeSpecs);
4285     for (unsigned I = 0; I != NumCTypeSpecs; ++I)
4286       Consumer.addKeywordResult(CTypeSpecs[I]);
4287 
4288     if (SemaRef.getLangOpts().C99)
4289       Consumer.addKeywordResult("restrict");
4290     if (SemaRef.getLangOpts().Bool || SemaRef.getLangOpts().CPlusPlus)
4291       Consumer.addKeywordResult("bool");
4292     else if (SemaRef.getLangOpts().C99)
4293       Consumer.addKeywordResult("_Bool");
4294 
4295     if (SemaRef.getLangOpts().CPlusPlus) {
4296       Consumer.addKeywordResult("class");
4297       Consumer.addKeywordResult("typename");
4298       Consumer.addKeywordResult("wchar_t");
4299 
4300       if (SemaRef.getLangOpts().CPlusPlus11) {
4301         Consumer.addKeywordResult("char16_t");
4302         Consumer.addKeywordResult("char32_t");
4303         Consumer.addKeywordResult("constexpr");
4304         Consumer.addKeywordResult("decltype");
4305         Consumer.addKeywordResult("thread_local");
4306       }
4307     }
4308 
4309     if (SemaRef.getLangOpts().GNUMode)
4310       Consumer.addKeywordResult("typeof");
4311   } else if (CCC.WantFunctionLikeCasts) {
4312     static const char *const CastableTypeSpecs[] = {
4313       "char", "double", "float", "int", "long", "short",
4314       "signed", "unsigned", "void"
4315     };
4316     for (auto *kw : CastableTypeSpecs)
4317       Consumer.addKeywordResult(kw);
4318   }
4319 
4320   if (CCC.WantCXXNamedCasts && SemaRef.getLangOpts().CPlusPlus) {
4321     Consumer.addKeywordResult("const_cast");
4322     Consumer.addKeywordResult("dynamic_cast");
4323     Consumer.addKeywordResult("reinterpret_cast");
4324     Consumer.addKeywordResult("static_cast");
4325   }
4326 
4327   if (CCC.WantExpressionKeywords) {
4328     Consumer.addKeywordResult("sizeof");
4329     if (SemaRef.getLangOpts().Bool || SemaRef.getLangOpts().CPlusPlus) {
4330       Consumer.addKeywordResult("false");
4331       Consumer.addKeywordResult("true");
4332     }
4333 
4334     if (SemaRef.getLangOpts().CPlusPlus) {
4335       static const char *const CXXExprs[] = {
4336         "delete", "new", "operator", "throw", "typeid"
4337       };
4338       const unsigned NumCXXExprs = llvm::array_lengthof(CXXExprs);
4339       for (unsigned I = 0; I != NumCXXExprs; ++I)
4340         Consumer.addKeywordResult(CXXExprs[I]);
4341 
4342       if (isa<CXXMethodDecl>(SemaRef.CurContext) &&
4343           cast<CXXMethodDecl>(SemaRef.CurContext)->isInstance())
4344         Consumer.addKeywordResult("this");
4345 
4346       if (SemaRef.getLangOpts().CPlusPlus11) {
4347         Consumer.addKeywordResult("alignof");
4348         Consumer.addKeywordResult("nullptr");
4349       }
4350     }
4351 
4352     if (SemaRef.getLangOpts().C11) {
4353       // FIXME: We should not suggest _Alignof if the alignof macro
4354       // is present.
4355       Consumer.addKeywordResult("_Alignof");
4356     }
4357   }
4358 
4359   if (CCC.WantRemainingKeywords) {
4360     if (SemaRef.getCurFunctionOrMethodDecl() || SemaRef.getCurBlock()) {
4361       // Statements.
4362       static const char *const CStmts[] = {
4363         "do", "else", "for", "goto", "if", "return", "switch", "while" };
4364       const unsigned NumCStmts = llvm::array_lengthof(CStmts);
4365       for (unsigned I = 0; I != NumCStmts; ++I)
4366         Consumer.addKeywordResult(CStmts[I]);
4367 
4368       if (SemaRef.getLangOpts().CPlusPlus) {
4369         Consumer.addKeywordResult("catch");
4370         Consumer.addKeywordResult("try");
4371       }
4372 
4373       if (S && S->getBreakParent())
4374         Consumer.addKeywordResult("break");
4375 
4376       if (S && S->getContinueParent())
4377         Consumer.addKeywordResult("continue");
4378 
4379       if (!SemaRef.getCurFunction()->SwitchStack.empty()) {
4380         Consumer.addKeywordResult("case");
4381         Consumer.addKeywordResult("default");
4382       }
4383     } else {
4384       if (SemaRef.getLangOpts().CPlusPlus) {
4385         Consumer.addKeywordResult("namespace");
4386         Consumer.addKeywordResult("template");
4387       }
4388 
4389       if (S && S->isClassScope()) {
4390         Consumer.addKeywordResult("explicit");
4391         Consumer.addKeywordResult("friend");
4392         Consumer.addKeywordResult("mutable");
4393         Consumer.addKeywordResult("private");
4394         Consumer.addKeywordResult("protected");
4395         Consumer.addKeywordResult("public");
4396         Consumer.addKeywordResult("virtual");
4397       }
4398     }
4399 
4400     if (SemaRef.getLangOpts().CPlusPlus) {
4401       Consumer.addKeywordResult("using");
4402 
4403       if (SemaRef.getLangOpts().CPlusPlus11)
4404         Consumer.addKeywordResult("static_assert");
4405     }
4406   }
4407 }
4408 
4409 std::unique_ptr<TypoCorrectionConsumer> Sema::makeTypoCorrectionConsumer(
4410     const DeclarationNameInfo &TypoName, Sema::LookupNameKind LookupKind,
4411     Scope *S, CXXScopeSpec *SS,
4412     std::unique_ptr<CorrectionCandidateCallback> CCC,
4413     DeclContext *MemberContext, bool EnteringContext,
4414     const ObjCObjectPointerType *OPT, bool ErrorRecovery) {
4415 
4416   if (Diags.hasFatalErrorOccurred() || !getLangOpts().SpellChecking ||
4417       DisableTypoCorrection)
4418     return nullptr;
4419 
4420   // In Microsoft mode, don't perform typo correction in a template member
4421   // function dependent context because it interferes with the "lookup into
4422   // dependent bases of class templates" feature.
4423   if (getLangOpts().MSVCCompat && CurContext->isDependentContext() &&
4424       isa<CXXMethodDecl>(CurContext))
4425     return nullptr;
4426 
4427   // We only attempt to correct typos for identifiers.
4428   IdentifierInfo *Typo = TypoName.getName().getAsIdentifierInfo();
4429   if (!Typo)
4430     return nullptr;
4431 
4432   // If the scope specifier itself was invalid, don't try to correct
4433   // typos.
4434   if (SS && SS->isInvalid())
4435     return nullptr;
4436 
4437   // Never try to correct typos during template deduction or
4438   // instantiation.
4439   if (!ActiveTemplateInstantiations.empty())
4440     return nullptr;
4441 
4442   // Don't try to correct 'super'.
4443   if (S && S->isInObjcMethodScope() && Typo == getSuperIdentifier())
4444     return nullptr;
4445 
4446   // Abort if typo correction already failed for this specific typo.
4447   IdentifierSourceLocations::iterator locs = TypoCorrectionFailures.find(Typo);
4448   if (locs != TypoCorrectionFailures.end() &&
4449       locs->second.count(TypoName.getLoc()))
4450     return nullptr;
4451 
4452   // Don't try to correct the identifier "vector" when in AltiVec mode.
4453   // TODO: Figure out why typo correction misbehaves in this case, fix it, and
4454   // remove this workaround.
4455   if ((getLangOpts().AltiVec || getLangOpts().ZVector) && Typo->isStr("vector"))
4456     return nullptr;
4457 
4458   // Provide a stop gap for files that are just seriously broken.  Trying
4459   // to correct all typos can turn into a HUGE performance penalty, causing
4460   // some files to take minutes to get rejected by the parser.
4461   unsigned Limit = getDiagnostics().getDiagnosticOptions().SpellCheckingLimit;
4462   if (Limit && TyposCorrected >= Limit)
4463     return nullptr;
4464   ++TyposCorrected;
4465 
4466   // If we're handling a missing symbol error, using modules, and the
4467   // special search all modules option is used, look for a missing import.
4468   if (ErrorRecovery && getLangOpts().Modules &&
4469       getLangOpts().ModulesSearchAll) {
4470     // The following has the side effect of loading the missing module.
4471     getModuleLoader().lookupMissingImports(Typo->getName(),
4472                                            TypoName.getLocStart());
4473   }
4474 
4475   CorrectionCandidateCallback &CCCRef = *CCC;
4476   auto Consumer = llvm::make_unique<TypoCorrectionConsumer>(
4477       *this, TypoName, LookupKind, S, SS, std::move(CCC), MemberContext,
4478       EnteringContext);
4479 
4480   // Perform name lookup to find visible, similarly-named entities.
4481   bool IsUnqualifiedLookup = false;
4482   DeclContext *QualifiedDC = MemberContext;
4483   if (MemberContext) {
4484     LookupVisibleDecls(MemberContext, LookupKind, *Consumer);
4485 
4486     // Look in qualified interfaces.
4487     if (OPT) {
4488       for (auto *I : OPT->quals())
4489         LookupVisibleDecls(I, LookupKind, *Consumer);
4490     }
4491   } else if (SS && SS->isSet()) {
4492     QualifiedDC = computeDeclContext(*SS, EnteringContext);
4493     if (!QualifiedDC)
4494       return nullptr;
4495 
4496     LookupVisibleDecls(QualifiedDC, LookupKind, *Consumer);
4497   } else {
4498     IsUnqualifiedLookup = true;
4499   }
4500 
4501   // Determine whether we are going to search in the various namespaces for
4502   // corrections.
4503   bool SearchNamespaces
4504     = getLangOpts().CPlusPlus &&
4505       (IsUnqualifiedLookup || (SS && SS->isSet()));
4506 
4507   if (IsUnqualifiedLookup || SearchNamespaces) {
4508     // For unqualified lookup, look through all of the names that we have
4509     // seen in this translation unit.
4510     // FIXME: Re-add the ability to skip very unlikely potential corrections.
4511     for (const auto &I : Context.Idents)
4512       Consumer->FoundName(I.getKey());
4513 
4514     // Walk through identifiers in external identifier sources.
4515     // FIXME: Re-add the ability to skip very unlikely potential corrections.
4516     if (IdentifierInfoLookup *External
4517                             = Context.Idents.getExternalIdentifierLookup()) {
4518       std::unique_ptr<IdentifierIterator> Iter(External->getIdentifiers());
4519       do {
4520         StringRef Name = Iter->Next();
4521         if (Name.empty())
4522           break;
4523 
4524         Consumer->FoundName(Name);
4525       } while (true);
4526     }
4527   }
4528 
4529   AddKeywordsToConsumer(*this, *Consumer, S, CCCRef, SS && SS->isNotEmpty());
4530 
4531   // Build the NestedNameSpecifiers for the KnownNamespaces, if we're going
4532   // to search those namespaces.
4533   if (SearchNamespaces) {
4534     // Load any externally-known namespaces.
4535     if (ExternalSource && !LoadedExternalKnownNamespaces) {
4536       SmallVector<NamespaceDecl *, 4> ExternalKnownNamespaces;
4537       LoadedExternalKnownNamespaces = true;
4538       ExternalSource->ReadKnownNamespaces(ExternalKnownNamespaces);
4539       for (auto *N : ExternalKnownNamespaces)
4540         KnownNamespaces[N] = true;
4541     }
4542 
4543     Consumer->addNamespaces(KnownNamespaces);
4544   }
4545 
4546   return Consumer;
4547 }
4548 
4549 /// \brief Try to "correct" a typo in the source code by finding
4550 /// visible declarations whose names are similar to the name that was
4551 /// present in the source code.
4552 ///
4553 /// \param TypoName the \c DeclarationNameInfo structure that contains
4554 /// the name that was present in the source code along with its location.
4555 ///
4556 /// \param LookupKind the name-lookup criteria used to search for the name.
4557 ///
4558 /// \param S the scope in which name lookup occurs.
4559 ///
4560 /// \param SS the nested-name-specifier that precedes the name we're
4561 /// looking for, if present.
4562 ///
4563 /// \param CCC A CorrectionCandidateCallback object that provides further
4564 /// validation of typo correction candidates. It also provides flags for
4565 /// determining the set of keywords permitted.
4566 ///
4567 /// \param MemberContext if non-NULL, the context in which to look for
4568 /// a member access expression.
4569 ///
4570 /// \param EnteringContext whether we're entering the context described by
4571 /// the nested-name-specifier SS.
4572 ///
4573 /// \param OPT when non-NULL, the search for visible declarations will
4574 /// also walk the protocols in the qualified interfaces of \p OPT.
4575 ///
4576 /// \returns a \c TypoCorrection containing the corrected name if the typo
4577 /// along with information such as the \c NamedDecl where the corrected name
4578 /// was declared, and any additional \c NestedNameSpecifier needed to access
4579 /// it (C++ only). The \c TypoCorrection is empty if there is no correction.
4580 TypoCorrection Sema::CorrectTypo(const DeclarationNameInfo &TypoName,
4581                                  Sema::LookupNameKind LookupKind,
4582                                  Scope *S, CXXScopeSpec *SS,
4583                                  std::unique_ptr<CorrectionCandidateCallback> CCC,
4584                                  CorrectTypoKind Mode,
4585                                  DeclContext *MemberContext,
4586                                  bool EnteringContext,
4587                                  const ObjCObjectPointerType *OPT,
4588                                  bool RecordFailure) {
4589   assert(CCC && "CorrectTypo requires a CorrectionCandidateCallback");
4590 
4591   // Always let the ExternalSource have the first chance at correction, even
4592   // if we would otherwise have given up.
4593   if (ExternalSource) {
4594     if (TypoCorrection Correction = ExternalSource->CorrectTypo(
4595         TypoName, LookupKind, S, SS, *CCC, MemberContext, EnteringContext, OPT))
4596       return Correction;
4597   }
4598 
4599   // Ugly hack equivalent to CTC == CTC_ObjCMessageReceiver;
4600   // WantObjCSuper is only true for CTC_ObjCMessageReceiver and for
4601   // some instances of CTC_Unknown, while WantRemainingKeywords is true
4602   // for CTC_Unknown but not for CTC_ObjCMessageReceiver.
4603   bool ObjCMessageReceiver = CCC->WantObjCSuper && !CCC->WantRemainingKeywords;
4604 
4605   IdentifierInfo *Typo = TypoName.getName().getAsIdentifierInfo();
4606   auto Consumer = makeTypoCorrectionConsumer(
4607       TypoName, LookupKind, S, SS, std::move(CCC), MemberContext,
4608       EnteringContext, OPT, Mode == CTK_ErrorRecovery);
4609 
4610   if (!Consumer)
4611     return TypoCorrection();
4612 
4613   // If we haven't found anything, we're done.
4614   if (Consumer->empty())
4615     return FailedCorrection(Typo, TypoName.getLoc(), RecordFailure);
4616 
4617   // Make sure the best edit distance (prior to adding any namespace qualifiers)
4618   // is not more that about a third of the length of the typo's identifier.
4619   unsigned ED = Consumer->getBestEditDistance(true);
4620   unsigned TypoLen = Typo->getName().size();
4621   if (ED > 0 && TypoLen / ED < 3)
4622     return FailedCorrection(Typo, TypoName.getLoc(), RecordFailure);
4623 
4624   TypoCorrection BestTC = Consumer->getNextCorrection();
4625   TypoCorrection SecondBestTC = Consumer->getNextCorrection();
4626   if (!BestTC)
4627     return FailedCorrection(Typo, TypoName.getLoc(), RecordFailure);
4628 
4629   ED = BestTC.getEditDistance();
4630 
4631   if (TypoLen >= 3 && ED > 0 && TypoLen / ED < 3) {
4632     // If this was an unqualified lookup and we believe the callback
4633     // object wouldn't have filtered out possible corrections, note
4634     // that no correction was found.
4635     return FailedCorrection(Typo, TypoName.getLoc(), RecordFailure);
4636   }
4637 
4638   // If only a single name remains, return that result.
4639   if (!SecondBestTC ||
4640       SecondBestTC.getEditDistance(false) > BestTC.getEditDistance(false)) {
4641     const TypoCorrection &Result = BestTC;
4642 
4643     // Don't correct to a keyword that's the same as the typo; the keyword
4644     // wasn't actually in scope.
4645     if (ED == 0 && Result.isKeyword())
4646       return FailedCorrection(Typo, TypoName.getLoc(), RecordFailure);
4647 
4648     TypoCorrection TC = Result;
4649     TC.setCorrectionRange(SS, TypoName);
4650     checkCorrectionVisibility(*this, TC);
4651     return TC;
4652   } else if (SecondBestTC && ObjCMessageReceiver) {
4653     // Prefer 'super' when we're completing in a message-receiver
4654     // context.
4655 
4656     if (BestTC.getCorrection().getAsString() != "super") {
4657       if (SecondBestTC.getCorrection().getAsString() == "super")
4658         BestTC = SecondBestTC;
4659       else if ((*Consumer)["super"].front().isKeyword())
4660         BestTC = (*Consumer)["super"].front();
4661     }
4662     // Don't correct to a keyword that's the same as the typo; the keyword
4663     // wasn't actually in scope.
4664     if (BestTC.getEditDistance() == 0 ||
4665         BestTC.getCorrection().getAsString() != "super")
4666       return FailedCorrection(Typo, TypoName.getLoc(), RecordFailure);
4667 
4668     BestTC.setCorrectionRange(SS, TypoName);
4669     return BestTC;
4670   }
4671 
4672   // Record the failure's location if needed and return an empty correction. If
4673   // this was an unqualified lookup and we believe the callback object did not
4674   // filter out possible corrections, also cache the failure for the typo.
4675   return FailedCorrection(Typo, TypoName.getLoc(), RecordFailure && !SecondBestTC);
4676 }
4677 
4678 /// \brief Try to "correct" a typo in the source code by finding
4679 /// visible declarations whose names are similar to the name that was
4680 /// present in the source code.
4681 ///
4682 /// \param TypoName the \c DeclarationNameInfo structure that contains
4683 /// the name that was present in the source code along with its location.
4684 ///
4685 /// \param LookupKind the name-lookup criteria used to search for the name.
4686 ///
4687 /// \param S the scope in which name lookup occurs.
4688 ///
4689 /// \param SS the nested-name-specifier that precedes the name we're
4690 /// looking for, if present.
4691 ///
4692 /// \param CCC A CorrectionCandidateCallback object that provides further
4693 /// validation of typo correction candidates. It also provides flags for
4694 /// determining the set of keywords permitted.
4695 ///
4696 /// \param TDG A TypoDiagnosticGenerator functor that will be used to print
4697 /// diagnostics when the actual typo correction is attempted.
4698 ///
4699 /// \param TRC A TypoRecoveryCallback functor that will be used to build an
4700 /// Expr from a typo correction candidate.
4701 ///
4702 /// \param MemberContext if non-NULL, the context in which to look for
4703 /// a member access expression.
4704 ///
4705 /// \param EnteringContext whether we're entering the context described by
4706 /// the nested-name-specifier SS.
4707 ///
4708 /// \param OPT when non-NULL, the search for visible declarations will
4709 /// also walk the protocols in the qualified interfaces of \p OPT.
4710 ///
4711 /// \returns a new \c TypoExpr that will later be replaced in the AST with an
4712 /// Expr representing the result of performing typo correction, or nullptr if
4713 /// typo correction is not possible. If nullptr is returned, no diagnostics will
4714 /// be emitted and it is the responsibility of the caller to emit any that are
4715 /// needed.
4716 TypoExpr *Sema::CorrectTypoDelayed(
4717     const DeclarationNameInfo &TypoName, Sema::LookupNameKind LookupKind,
4718     Scope *S, CXXScopeSpec *SS,
4719     std::unique_ptr<CorrectionCandidateCallback> CCC,
4720     TypoDiagnosticGenerator TDG, TypoRecoveryCallback TRC, CorrectTypoKind Mode,
4721     DeclContext *MemberContext, bool EnteringContext,
4722     const ObjCObjectPointerType *OPT) {
4723   assert(CCC && "CorrectTypoDelayed requires a CorrectionCandidateCallback");
4724 
4725   TypoCorrection Empty;
4726   auto Consumer = makeTypoCorrectionConsumer(
4727       TypoName, LookupKind, S, SS, std::move(CCC), MemberContext,
4728       EnteringContext, OPT, Mode == CTK_ErrorRecovery);
4729 
4730   if (!Consumer || Consumer->empty())
4731     return nullptr;
4732 
4733   // Make sure the best edit distance (prior to adding any namespace qualifiers)
4734   // is not more that about a third of the length of the typo's identifier.
4735   unsigned ED = Consumer->getBestEditDistance(true);
4736   IdentifierInfo *Typo = TypoName.getName().getAsIdentifierInfo();
4737   if (ED > 0 && Typo->getName().size() / ED < 3)
4738     return nullptr;
4739 
4740   ExprEvalContexts.back().NumTypos++;
4741   return createDelayedTypo(std::move(Consumer), std::move(TDG), std::move(TRC));
4742 }
4743 
4744 void TypoCorrection::addCorrectionDecl(NamedDecl *CDecl) {
4745   if (!CDecl) return;
4746 
4747   if (isKeyword())
4748     CorrectionDecls.clear();
4749 
4750   CorrectionDecls.push_back(CDecl);
4751 
4752   if (!CorrectionName)
4753     CorrectionName = CDecl->getDeclName();
4754 }
4755 
4756 std::string TypoCorrection::getAsString(const LangOptions &LO) const {
4757   if (CorrectionNameSpec) {
4758     std::string tmpBuffer;
4759     llvm::raw_string_ostream PrefixOStream(tmpBuffer);
4760     CorrectionNameSpec->print(PrefixOStream, PrintingPolicy(LO));
4761     PrefixOStream << CorrectionName;
4762     return PrefixOStream.str();
4763   }
4764 
4765   return CorrectionName.getAsString();
4766 }
4767 
4768 bool CorrectionCandidateCallback::ValidateCandidate(
4769     const TypoCorrection &candidate) {
4770   if (!candidate.isResolved())
4771     return true;
4772 
4773   if (candidate.isKeyword())
4774     return WantTypeSpecifiers || WantExpressionKeywords || WantCXXNamedCasts ||
4775            WantRemainingKeywords || WantObjCSuper;
4776 
4777   bool HasNonType = false;
4778   bool HasStaticMethod = false;
4779   bool HasNonStaticMethod = false;
4780   for (Decl *D : candidate) {
4781     if (FunctionTemplateDecl *FTD = dyn_cast<FunctionTemplateDecl>(D))
4782       D = FTD->getTemplatedDecl();
4783     if (CXXMethodDecl *Method = dyn_cast<CXXMethodDecl>(D)) {
4784       if (Method->isStatic())
4785         HasStaticMethod = true;
4786       else
4787         HasNonStaticMethod = true;
4788     }
4789     if (!isa<TypeDecl>(D))
4790       HasNonType = true;
4791   }
4792 
4793   if (IsAddressOfOperand && HasNonStaticMethod && !HasStaticMethod &&
4794       !candidate.getCorrectionSpecifier())
4795     return false;
4796 
4797   return WantTypeSpecifiers || HasNonType;
4798 }
4799 
4800 FunctionCallFilterCCC::FunctionCallFilterCCC(Sema &SemaRef, unsigned NumArgs,
4801                                              bool HasExplicitTemplateArgs,
4802                                              MemberExpr *ME)
4803     : NumArgs(NumArgs), HasExplicitTemplateArgs(HasExplicitTemplateArgs),
4804       CurContext(SemaRef.CurContext), MemberFn(ME) {
4805   WantTypeSpecifiers = false;
4806   WantFunctionLikeCasts = SemaRef.getLangOpts().CPlusPlus && NumArgs == 1;
4807   WantRemainingKeywords = false;
4808 }
4809 
4810 bool FunctionCallFilterCCC::ValidateCandidate(const TypoCorrection &candidate) {
4811   if (!candidate.getCorrectionDecl())
4812     return candidate.isKeyword();
4813 
4814   for (auto *C : candidate) {
4815     FunctionDecl *FD = nullptr;
4816     NamedDecl *ND = C->getUnderlyingDecl();
4817     if (FunctionTemplateDecl *FTD = dyn_cast<FunctionTemplateDecl>(ND))
4818       FD = FTD->getTemplatedDecl();
4819     if (!HasExplicitTemplateArgs && !FD) {
4820       if (!(FD = dyn_cast<FunctionDecl>(ND)) && isa<ValueDecl>(ND)) {
4821         // If the Decl is neither a function nor a template function,
4822         // determine if it is a pointer or reference to a function. If so,
4823         // check against the number of arguments expected for the pointee.
4824         QualType ValType = cast<ValueDecl>(ND)->getType();
4825         if (ValType->isAnyPointerType() || ValType->isReferenceType())
4826           ValType = ValType->getPointeeType();
4827         if (const FunctionProtoType *FPT = ValType->getAs<FunctionProtoType>())
4828           if (FPT->getNumParams() == NumArgs)
4829             return true;
4830       }
4831     }
4832 
4833     // Skip the current candidate if it is not a FunctionDecl or does not accept
4834     // the current number of arguments.
4835     if (!FD || !(FD->getNumParams() >= NumArgs &&
4836                  FD->getMinRequiredArguments() <= NumArgs))
4837       continue;
4838 
4839     // If the current candidate is a non-static C++ method, skip the candidate
4840     // unless the method being corrected--or the current DeclContext, if the
4841     // function being corrected is not a method--is a method in the same class
4842     // or a descendent class of the candidate's parent class.
4843     if (CXXMethodDecl *MD = dyn_cast<CXXMethodDecl>(FD)) {
4844       if (MemberFn || !MD->isStatic()) {
4845         CXXMethodDecl *CurMD =
4846             MemberFn
4847                 ? dyn_cast_or_null<CXXMethodDecl>(MemberFn->getMemberDecl())
4848                 : dyn_cast_or_null<CXXMethodDecl>(CurContext);
4849         CXXRecordDecl *CurRD =
4850             CurMD ? CurMD->getParent()->getCanonicalDecl() : nullptr;
4851         CXXRecordDecl *RD = MD->getParent()->getCanonicalDecl();
4852         if (!CurRD || (CurRD != RD && !CurRD->isDerivedFrom(RD)))
4853           continue;
4854       }
4855     }
4856     return true;
4857   }
4858   return false;
4859 }
4860 
4861 void Sema::diagnoseTypo(const TypoCorrection &Correction,
4862                         const PartialDiagnostic &TypoDiag,
4863                         bool ErrorRecovery) {
4864   diagnoseTypo(Correction, TypoDiag, PDiag(diag::note_previous_decl),
4865                ErrorRecovery);
4866 }
4867 
4868 /// Find which declaration we should import to provide the definition of
4869 /// the given declaration.
4870 static NamedDecl *getDefinitionToImport(NamedDecl *D) {
4871   if (VarDecl *VD = dyn_cast<VarDecl>(D))
4872     return VD->getDefinition();
4873   if (const FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
4874     return FD->isDefined(FD) ? const_cast<FunctionDecl*>(FD) : nullptr;
4875   if (TagDecl *TD = dyn_cast<TagDecl>(D))
4876     return TD->getDefinition();
4877   if (ObjCInterfaceDecl *ID = dyn_cast<ObjCInterfaceDecl>(D))
4878     return ID->getDefinition();
4879   if (ObjCProtocolDecl *PD = dyn_cast<ObjCProtocolDecl>(D))
4880     return PD->getDefinition();
4881   if (TemplateDecl *TD = dyn_cast<TemplateDecl>(D))
4882     return getDefinitionToImport(TD->getTemplatedDecl());
4883   return nullptr;
4884 }
4885 
4886 void Sema::diagnoseMissingImport(SourceLocation Loc, NamedDecl *Decl,
4887                                  bool NeedDefinition, bool Recover) {
4888   assert(!isVisible(Decl) && "missing import for non-hidden decl?");
4889 
4890   // Suggest importing a module providing the definition of this entity, if
4891   // possible.
4892   NamedDecl *Def = getDefinitionToImport(Decl);
4893   if (!Def)
4894     Def = Decl;
4895 
4896   // FIXME: Add a Fix-It that imports the corresponding module or includes
4897   // the header.
4898   Module *Owner = getOwningModule(Decl);
4899   assert(Owner && "definition of hidden declaration is not in a module");
4900 
4901   llvm::SmallVector<Module*, 8> OwningModules;
4902   OwningModules.push_back(Owner);
4903   auto Merged = Context.getModulesWithMergedDefinition(Decl);
4904   OwningModules.insert(OwningModules.end(), Merged.begin(), Merged.end());
4905 
4906   diagnoseMissingImport(Loc, Decl, Decl->getLocation(), OwningModules,
4907                         NeedDefinition ? MissingImportKind::Definition
4908                                        : MissingImportKind::Declaration,
4909                         Recover);
4910 }
4911 
4912 void Sema::diagnoseMissingImport(SourceLocation UseLoc, NamedDecl *Decl,
4913                                  SourceLocation DeclLoc,
4914                                  ArrayRef<Module *> Modules,
4915                                  MissingImportKind MIK, bool Recover) {
4916   assert(!Modules.empty());
4917 
4918   if (Modules.size() > 1) {
4919     std::string ModuleList;
4920     unsigned N = 0;
4921     for (Module *M : Modules) {
4922       ModuleList += "\n        ";
4923       if (++N == 5 && N != Modules.size()) {
4924         ModuleList += "[...]";
4925         break;
4926       }
4927       ModuleList += M->getFullModuleName();
4928     }
4929 
4930     Diag(UseLoc, diag::err_module_unimported_use_multiple)
4931       << (int)MIK << Decl << ModuleList;
4932   } else {
4933     Diag(UseLoc, diag::err_module_unimported_use)
4934       << (int)MIK << Decl << Modules[0]->getFullModuleName();
4935   }
4936 
4937   unsigned DiagID;
4938   switch (MIK) {
4939   case MissingImportKind::Declaration:
4940     DiagID = diag::note_previous_declaration;
4941     break;
4942   case MissingImportKind::Definition:
4943     DiagID = diag::note_previous_definition;
4944     break;
4945   case MissingImportKind::DefaultArgument:
4946     DiagID = diag::note_default_argument_declared_here;
4947     break;
4948   }
4949   Diag(DeclLoc, DiagID);
4950 
4951   // Try to recover by implicitly importing this module.
4952   if (Recover)
4953     createImplicitModuleImportForErrorRecovery(UseLoc, Modules[0]);
4954 }
4955 
4956 /// \brief Diagnose a successfully-corrected typo. Separated from the correction
4957 /// itself to allow external validation of the result, etc.
4958 ///
4959 /// \param Correction The result of performing typo correction.
4960 /// \param TypoDiag The diagnostic to produce. This will have the corrected
4961 ///        string added to it (and usually also a fixit).
4962 /// \param PrevNote A note to use when indicating the location of the entity to
4963 ///        which we are correcting. Will have the correction string added to it.
4964 /// \param ErrorRecovery If \c true (the default), the caller is going to
4965 ///        recover from the typo as if the corrected string had been typed.
4966 ///        In this case, \c PDiag must be an error, and we will attach a fixit
4967 ///        to it.
4968 void Sema::diagnoseTypo(const TypoCorrection &Correction,
4969                         const PartialDiagnostic &TypoDiag,
4970                         const PartialDiagnostic &PrevNote,
4971                         bool ErrorRecovery) {
4972   std::string CorrectedStr = Correction.getAsString(getLangOpts());
4973   std::string CorrectedQuotedStr = Correction.getQuoted(getLangOpts());
4974   FixItHint FixTypo = FixItHint::CreateReplacement(
4975       Correction.getCorrectionRange(), CorrectedStr);
4976 
4977   // Maybe we're just missing a module import.
4978   if (Correction.requiresImport()) {
4979     NamedDecl *Decl = Correction.getFoundDecl();
4980     assert(Decl && "import required but no declaration to import");
4981 
4982     diagnoseMissingImport(Correction.getCorrectionRange().getBegin(), Decl,
4983                           /*NeedDefinition*/ false, ErrorRecovery);
4984     return;
4985   }
4986 
4987   Diag(Correction.getCorrectionRange().getBegin(), TypoDiag)
4988     << CorrectedQuotedStr << (ErrorRecovery ? FixTypo : FixItHint());
4989 
4990   NamedDecl *ChosenDecl =
4991       Correction.isKeyword() ? nullptr : Correction.getFoundDecl();
4992   if (PrevNote.getDiagID() && ChosenDecl)
4993     Diag(ChosenDecl->getLocation(), PrevNote)
4994       << CorrectedQuotedStr << (ErrorRecovery ? FixItHint() : FixTypo);
4995 }
4996 
4997 TypoExpr *Sema::createDelayedTypo(std::unique_ptr<TypoCorrectionConsumer> TCC,
4998                                   TypoDiagnosticGenerator TDG,
4999                                   TypoRecoveryCallback TRC) {
5000   assert(TCC && "createDelayedTypo requires a valid TypoCorrectionConsumer");
5001   auto TE = new (Context) TypoExpr(Context.DependentTy);
5002   auto &State = DelayedTypos[TE];
5003   State.Consumer = std::move(TCC);
5004   State.DiagHandler = std::move(TDG);
5005   State.RecoveryHandler = std::move(TRC);
5006   return TE;
5007 }
5008 
5009 const Sema::TypoExprState &Sema::getTypoExprState(TypoExpr *TE) const {
5010   auto Entry = DelayedTypos.find(TE);
5011   assert(Entry != DelayedTypos.end() &&
5012          "Failed to get the state for a TypoExpr!");
5013   return Entry->second;
5014 }
5015 
5016 void Sema::clearDelayedTypo(TypoExpr *TE) {
5017   DelayedTypos.erase(TE);
5018 }
5019 
5020 void Sema::ActOnPragmaDump(Scope *S, SourceLocation IILoc, IdentifierInfo *II) {
5021   DeclarationNameInfo Name(II, IILoc);
5022   LookupResult R(*this, Name, LookupAnyName, Sema::NotForRedeclaration);
5023   R.suppressDiagnostics();
5024   R.setHideTags(false);
5025   LookupName(R, S);
5026   R.dump();
5027 }
5028