1d2fa56687f8bd5ac6ebf9d9468d0efd714986a54Ted Kremenek//===- CIndex.cpp - Clang-C Source Indexing Library -----------------------===//
2d2fa56687f8bd5ac6ebf9d9468d0efd714986a54Ted Kremenek//
3d2fa56687f8bd5ac6ebf9d9468d0efd714986a54Ted Kremenek//                     The LLVM Compiler Infrastructure
4d2fa56687f8bd5ac6ebf9d9468d0efd714986a54Ted Kremenek//
5d2fa56687f8bd5ac6ebf9d9468d0efd714986a54Ted Kremenek// This file is distributed under the University of Illinois Open Source
6d2fa56687f8bd5ac6ebf9d9468d0efd714986a54Ted Kremenek// License. See LICENSE.TXT for details.
70d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar//
8d2fa56687f8bd5ac6ebf9d9468d0efd714986a54Ted Kremenek//===----------------------------------------------------------------------===//
9d2fa56687f8bd5ac6ebf9d9468d0efd714986a54Ted Kremenek//
10ab1889321f6f8f200f2b318ac26883ac18e49d03Ted Kremenek// This file implements the main API hooks in the Clang-C Source Indexing
11ab1889321f6f8f200f2b318ac26883ac18e49d03Ted Kremenek// library.
12d2fa56687f8bd5ac6ebf9d9468d0efd714986a54Ted Kremenek//
13d2fa56687f8bd5ac6ebf9d9468d0efd714986a54Ted Kremenek//===----------------------------------------------------------------------===//
14d2fa56687f8bd5ac6ebf9d9468d0efd714986a54Ted Kremenek
15ab1889321f6f8f200f2b318ac26883ac18e49d03Ted Kremenek#include "CIndexer.h"
16ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko#include "CXComment.h"
1716c440a377b7ec8b722a2e2c7c864f75c95bd305Ted Kremenek#include "CXCursor.h"
180a90d32523bfe5fa63e11b648686c9699f786d15Ted Kremenek#include "CXTranslationUnit.h"
19ed122735639d83c10f18c28c7fd117bfcd0f62cbTed Kremenek#include "CXString.h"
2095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek#include "CXType.h"
21a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek#include "CXSourceLocation.h"
225352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor#include "CIndexDiagnostic.h"
23e397bf1bd90cfceb0166606ebcd2580b7671a828Argyrios Kyrtzidis#include "CursorVisitor.h"
24ab1889321f6f8f200f2b318ac26883ac18e49d03Ted Kremenek
2504bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek#include "clang/Basic/Version.h"
26936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor
27fb5704295c6137685a7b90b92cd6b958028740c8Steve Naroff#include "clang/AST/StmtVisitor.h"
28b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Basic/Diagnostic.h"
29b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Frontend/ASTUnit.h"
30b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Frontend/CompilerInstance.h"
31936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor#include "clang/Frontend/FrontendDiagnostic.h"
32d8210650ed948de65a08a8daf16d291b747717c4Ted Kremenek#include "clang/Lex/Lexer.h"
33dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor#include "clang/Lex/HeaderSearch.h"
34b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Lex/PreprocessingRecord.h"
3533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor#include "clang/Lex/Preprocessor.h"
36a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor#include "llvm/ADT/STLExtras.h"
37d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek#include "llvm/ADT/Optional.h"
38f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor#include "llvm/ADT/StringSwitch.h"
39b2c60b04a597cc5ba4154837cf8e0a155a376fd7Argyrios Kyrtzidis#include "llvm/Support/SaveAndRestore.h"
40c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar#include "llvm/Support/CrashRecoveryContext.h"
4148615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar#include "llvm/Support/PrettyStackTrace.h"
4202465750c8c3fa96b1e7e596b02297e24361dc4fDouglas Gregor#include "llvm/Support/MemoryBuffer.h"
43358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor#include "llvm/Support/raw_ostream.h"
447a07fcb8f10fe45ea65a0a41798eb1c40777bde4Douglas Gregor#include "llvm/Support/Timer.h"
4503013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Mutex.h"
4603013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Program.h"
4703013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Signals.h"
4803013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Threading.h"
4937f1ea0eb08a00fa90edbecb427cfbb50ca0f4d0Ted Kremenek#include "llvm/Support/Compiler.h"
50fc0622155fa61349698a8fd0053773c37d9f7ac4Ted Kremenek
5150398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroffusing namespace clang;
5216c440a377b7ec8b722a2e2c7c864f75c95bd305Ted Kremenekusing namespace clang::cxcursor;
53ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenekusing namespace clang::cxstring;
549049cf6cb468c856888e88251dab659955fa767eArgyrios Kyrtzidisusing namespace clang::cxtu;
55e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidisusing namespace clang::cxindex;
5650398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff
57fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios KyrtzidisCXTranslationUnit cxtu::MakeCXTranslationUnit(CIndexer *CIdx, ASTUnit *TU) {
58a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  if (!TU)
59a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return 0;
60a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit D = new CXTranslationUnitImpl();
61fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  D->CIdx = CIdx;
62a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  D->TUData = TU;
63a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  D->StringPool = createCXStringPool();
64153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek  D->Diagnostics = 0;
65bbf66ca1dad17773cc682d69b8482c4e179aeaebTed Kremenek  D->OverridenCursorsPool = createOverridenCXCursorsPool();
66a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return D;
67a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek}
68a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek
694e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidiscxtu::CXTUOwner::~CXTUOwner() {
704e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis  if (TU)
714e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis    clang_disposeTranslationUnit(TU);
724e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis}
734e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis
74f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek/// \brief Compare two source ranges to determine their relative position in
7533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// the translation unit.
76f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekstatic RangeComparisonResult RangeCompare(SourceManager &SM,
77f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                          SourceRange R1,
7833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                                          SourceRange R2) {
7933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R1.isValid() && "First range is invalid?");
8033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R2.isValid() && "Second range is invalid?");
81a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R1.getEnd() != R2.getBegin() &&
82d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R1.getEnd(), R2.getBegin()))
8333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeBefore;
84a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R2.getEnd() != R1.getBegin() &&
85d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R2.getEnd(), R1.getBegin()))
8633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeAfter;
8733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return RangeOverlap;
8833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
8933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
90fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek/// \brief Determine if a source location falls within, before, or after a
91fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek///   a given source range.
92fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic RangeComparisonResult LocationCompare(SourceManager &SM,
93fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                             SourceLocation L, SourceRange R) {
94fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(R.isValid() && "First range is invalid?");
95fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(L.isValid() && "Second range is invalid?");
96a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (L == R.getBegin() || L == R.getEnd())
97fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeOverlap;
98fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(L, R.getBegin()))
99fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeBefore;
100fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(R.getEnd(), L))
101fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeAfter;
102fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return RangeOverlap;
103fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek}
104fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
10576dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// \brief Translate a Clang source range into a CIndex source range.
10676dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar///
10776dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// Clang internally represents ranges where the end location points to the
10876dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// start of the token at the end. However, for external clients it is more
10976dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// useful to have a CXSourceRange be a proper half-open interval. This routine
11076dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// does the appropriate translation.
111f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed KremenekCXSourceRange cxloc::translateSourceRange(const SourceManager &SM,
11276dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                                          const LangOptions &LangOpts,
1130a76aae8c03cb7dd7bdbe683485560afaf695959Chris Lattner                                          const CharSourceRange &R) {
11476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  // We want the last character in this location, so we will adjust the
1156a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor  // location accordingly.
11676dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  SourceLocation EndLoc = R.getEnd();
117eba8cd5967e47592285590360bde73063c9c226fArgyrios Kyrtzidis  if (EndLoc.isValid() && EndLoc.isMacroID() && !SM.isMacroArgExpansion(EndLoc))
118edc3dccece244a584f8ebdb81da6c962c08e79beChandler Carruth    EndLoc = SM.getExpansionRange(EndLoc).second;
119eba8cd5967e47592285590360bde73063c9c226fArgyrios Kyrtzidis  if (R.isTokenRange() && !EndLoc.isInvalid()) {
120eba8cd5967e47592285590360bde73063c9c226fArgyrios Kyrtzidis    unsigned Length = Lexer::MeasureTokenLength(SM.getSpellingLoc(EndLoc),
121eba8cd5967e47592285590360bde73063c9c226fArgyrios Kyrtzidis                                                SM, LangOpts);
122a64ccefdf0ea4e03ec88805d71b0af74950c7472Argyrios Kyrtzidis    EndLoc = EndLoc.getLocWithOffset(Length);
12376dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  }
12476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar
12576dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  CXSourceRange Result = { { (void *)&SM, (void *)&LangOpts },
12676dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           R.getBegin().getRawEncoding(),
12776dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           EndLoc.getRawEncoding() };
12876dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  return Result;
12976dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar}
1301db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
1318a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
13233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor// Cursor visitor.
1338a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
1348a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek
135a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C);
1366653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr);
1376653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
138a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
13933e9abd21083a0191a7676a04b497006d2da184dDouglas GregorRangeComparisonResult CursorVisitor::CompareRegionOfInterest(SourceRange R) {
140a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return RangeCompare(AU->getSourceManager(), R, RegionOfInterest);
14133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
14233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
143b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the given cursor and, if requested by the visitor,
144b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// its children.
145b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor///
14633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param Cursor the cursor to visit.
14733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
14870517ca5c07c4b41ff8662b94ee22047b0299f8cDmitri Gribenko/// \param CheckedRegionOfInterest if true, then the caller already checked
14970517ca5c07c4b41ff8662b94ee22047b0299f8cDmitri Gribenko/// that this cursor is within the region of interest.
15033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
151b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
152b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
15333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorbool CursorVisitor::Visit(CXCursor Cursor, bool CheckedRegionOfInterest) {
154b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isInvalid(Cursor.kind))
155b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
156f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
157b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
158b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
15916ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (!D) {
16016ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      assert(0 && "Invalid declaration cursor");
16116ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      return true; // abort.
16216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    }
16316ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
16465ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // Ignore implicit declarations, unless it's an objc method because
16565ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // currently we should report implicit methods for properties when indexing.
16665ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    if (D->isImplicit() && !isa<ObjCMethodDecl>(D))
167b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return false;
168b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
1690d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
17033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // If we have a range of interest, and this cursor doesn't intersect with it,
17133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // we're done.
17233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (RegionOfInterest.isValid() && !CheckedRegionOfInterest) {
173a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    SourceRange Range = getRawCursorExtent(Cursor);
174f408f32aa9ae3d97bc656267dc5d78fa7d03499bDaniel Dunbar    if (Range.isInvalid() || CompareRegionOfInterest(Range))
17533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor      return false;
17633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  }
177f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
178b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  switch (Visitor(Cursor, Parent, ClientData)) {
179b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Break:
180b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
1810d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
182b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Continue:
183b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
1842e331b938b38057e333fab0ba841130ea8467794Douglas Gregor
185d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  case CXChildVisit_Recurse: {
186d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis    bool ret = VisitChildren(Cursor);
187d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis    if (PostChildrenVisitor)
188d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis      if (PostChildrenVisitor(Cursor, ClientData))
189d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis        return true;
190d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis    return ret;
191d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  }
192b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
1930d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
1947530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid CXChildVisitResult!");
195b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
1960d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
197f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidisstatic bool visitPreprocessedEntitiesInRange(SourceRange R,
198f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                             PreprocessingRecord &PPRec,
199f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                             CursorVisitor &Visitor) {
200f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  SourceManager &SM = Visitor.getASTUnit()->getSourceManager();
201f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  FileID FID;
202f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
203e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis  if (!Visitor.shouldVisitIncludedEntities()) {
204f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    // If the begin/end of the range lie in the same FileID, do the optimization
205f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    // where we skip preprocessed entities that do not come from the same FileID.
206acca41167ce78bb032906f6b1d2ced62efbe059aArgyrios Kyrtzidis    FID = SM.getFileID(SM.getFileLoc(R.getBegin()));
207acca41167ce78bb032906f6b1d2ced62efbe059aArgyrios Kyrtzidis    if (FID != SM.getFileID(SM.getFileLoc(R.getEnd())))
208f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      FID = FileID();
209f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  }
210f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
211f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  std::pair<PreprocessingRecord::iterator, PreprocessingRecord::iterator>
212f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    Entities = PPRec.getPreprocessedEntitiesInRange(R);
213f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  return Visitor.visitPreprocessedEntities(Entities.first, Entities.second,
214f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                           PPRec, FID);
215f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis}
216f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
217dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidisvoid CursorVisitor::visitFileRegion() {
218dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (RegionOfInterest.isInvalid())
219dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
220dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
221dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  ASTUnit *Unit = static_cast<ASTUnit *>(TU->TUData);
222dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceManager &SM = Unit->getSourceManager();
223dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
224dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  std::pair<FileID, unsigned>
225dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Begin = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getBegin())),
226dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getEnd()));
227dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
228dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (End.first != Begin.first) {
229dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    // If the end does not reside in the same file, try to recover by
230dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    // picking the end of the file of begin location.
231dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End.first = Begin.first;
232dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End.second = SM.getFileIDSize(Begin.first);
233dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
234dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
235dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  assert(Begin.first == End.first);
236dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (Begin.second > End.second)
237dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
238dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
239dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  FileID File = Begin.first;
240dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  unsigned Offset = Begin.second;
241dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  unsigned Length = End.second - Begin.second;
242dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
243b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis  if (!VisitDeclsOnly && !VisitPreprocessorLast)
244b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis    if (visitPreprocessedEntitiesInRegion())
245b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis      return; // visitation break.
246dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
247dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  visitDeclsFromFileRegion(File, Offset, Length);
248dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
249b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis  if (!VisitDeclsOnly && VisitPreprocessorLast)
250dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    visitPreprocessedEntitiesInRegion();
251dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis}
252dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
253e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidisstatic bool isInLexicalContext(Decl *D, DeclContext *DC) {
254e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  if (!DC)
255e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    return false;
256e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
257e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  for (DeclContext *DeclDC = D->getLexicalDeclContext();
258e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis         DeclDC; DeclDC = DeclDC->getLexicalParent()) {
259e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    if (DeclDC == DC)
260e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis      return true;
261e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  }
262e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  return false;
263e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis}
264e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
265dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidisvoid CursorVisitor::visitDeclsFromFileRegion(FileID File,
266dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis                                             unsigned Offset, unsigned Length) {
267dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  ASTUnit *Unit = static_cast<ASTUnit *>(TU->TUData);
268dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceManager &SM = Unit->getSourceManager();
269dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceRange Range = RegionOfInterest;
270dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
271dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SmallVector<Decl *, 16> Decls;
272dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  Unit->findFileRegionDecls(File, Offset, Length, Decls);
273dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
274dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // If we didn't find any file level decls for the file, try looking at the
275dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // file that it was included from.
276c14a03dffff69b5e1c55cc118fc52d8fd9f3a28dArgyrios Kyrtzidis  while (Decls.empty() || Decls.front()->isTopLevelDeclInObjCContainer()) {
277dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    bool Invalid = false;
278dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    const SrcMgr::SLocEntry &SLEntry = SM.getSLocEntry(File, &Invalid);
279dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (Invalid)
280dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      return;
281dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
282dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    SourceLocation Outer;
283dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (SLEntry.isFile())
284dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      Outer = SLEntry.getFile().getIncludeLoc();
285dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    else
286dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      Outer = SLEntry.getExpansion().getExpansionLocStart();
287dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (Outer.isInvalid())
288dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      return;
289dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
290dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    llvm::tie(File, Offset) = SM.getDecomposedExpansionLoc(Outer);
291dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Length = 0;
292dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Unit->findFileRegionDecls(File, Offset, Length, Decls);
293dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
294dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
295dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  assert(!Decls.empty());
296dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
297dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  bool VisitedAtLeastOnce = false;
298e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  DeclContext *CurDC = 0;
299dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SmallVector<Decl *, 16>::iterator DIt = Decls.begin();
300dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  for (SmallVector<Decl *, 16>::iterator DE = Decls.end(); DIt != DE; ++DIt) {
301dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Decl *D = *DIt;
302ed8bef44c0545fd55a78715606f8d733f6498b21Argyrios Kyrtzidis    if (D->getSourceRange().isInvalid())
303ed8bef44c0545fd55a78715606f8d733f6498b21Argyrios Kyrtzidis      continue;
304dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
305e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    if (isInLexicalContext(D, CurDC))
306e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis      continue;
307e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
308e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    CurDC = dyn_cast<DeclContext>(D);
309e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
310e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    if (TagDecl *TD = dyn_cast<TagDecl>(D))
311e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis      if (!TD->isFreeStanding())
312e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis        continue;
313e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
314dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    RangeComparisonResult CompRes = RangeCompare(SM, D->getSourceRange(),Range);
315dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CompRes == RangeBefore)
316dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      continue;
317dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CompRes == RangeAfter)
318dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
319dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
320dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    assert(CompRes == RangeOverlap);
321dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    VisitedAtLeastOnce = true;
32203ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
32303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (isa<ObjCContainerDecl>(D)) {
32403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      FileDI_current = &DIt;
32503ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      FileDE_current = DE;
32603ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    } else {
32703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      FileDI_current = 0;
32803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    }
32903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
330ba98617b994864b7554ff75445983ad02a962f45Argyrios Kyrtzidis    if (Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true))
331dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
332dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
333dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
334dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (VisitedAtLeastOnce)
335dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
336dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
337dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // No Decls overlapped with the range. Move up the lexical context until there
338dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // is a context that contains the range or we reach the translation unit
339dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // level.
340dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  DeclContext *DC = DIt == Decls.begin() ? (*DIt)->getLexicalDeclContext()
341dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis                                         : (*(DIt-1))->getLexicalDeclContext();
342dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
343dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  while (DC && !DC->isTranslationUnit()) {
344dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Decl *D = cast<Decl>(DC);
345dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    SourceRange CurDeclRange = D->getSourceRange();
346dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CurDeclRange.isInvalid())
347dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
348dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
349dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (RangeCompare(SM, CurDeclRange, Range) == RangeOverlap) {
350ba98617b994864b7554ff75445983ad02a962f45Argyrios Kyrtzidis      Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true);
351dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
352dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    }
353dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
354dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    DC = D->getLexicalDeclContext();
355dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
356dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis}
357dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
3584c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregorbool CursorVisitor::visitPreprocessedEntitiesInRegion() {
359b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis  if (!AU->getPreprocessor().getPreprocessingRecord())
360b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis    return false;
361b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis
362788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  PreprocessingRecord &PPRec
363a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    = *AU->getPreprocessor().getPreprocessingRecord();
364f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  SourceManager &SM = AU->getSourceManager();
365788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
36692ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  if (RegionOfInterest.isValid()) {
367ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange MappedRange = AU->mapRangeToPreamble(RegionOfInterest);
368f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    SourceLocation B = MappedRange.getBegin();
369f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    SourceLocation E = MappedRange.getEnd();
370f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
371f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (AU->isInPreambleFileID(B)) {
372f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      if (SM.isLoadedSourceLocation(E))
373f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis        return visitPreprocessedEntitiesInRange(SourceRange(B, E),
374f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                                 PPRec, *this);
375f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
376f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // Beginning of range lies in the preamble but it also extends beyond
377f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // it into the main file. Split the range into 2 parts, one covering
378f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // the preamble and another covering the main file. This allows subsequent
379f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // calls to visitPreprocessedEntitiesInRange to accept a source range that
380f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // lies in the same FileID, allowing it to skip preprocessed entities that
381f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // do not come from the same FileID.
382f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      bool breaked =
383f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis        visitPreprocessedEntitiesInRange(
384f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                   SourceRange(B, AU->getEndOfPreambleFileID()),
385f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                          PPRec, *this);
386f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      if (breaked) return true;
387f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      return visitPreprocessedEntitiesInRange(
388f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                    SourceRange(AU->getStartOfMainFileID(), E),
389f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                        PPRec, *this);
390f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    }
391f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
392f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    return visitPreprocessedEntitiesInRange(SourceRange(B, E), PPRec, *this);
39392ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  }
39492ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis
395788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  bool OnlyLocalDecls
39632038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    = !AU->isMainFileAST() && AU->getOnlyLocalDecls();
39732038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor
39892ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  if (OnlyLocalDecls)
399f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    return visitPreprocessedEntities(PPRec.local_begin(), PPRec.local_end(),
400f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                     PPRec);
4014c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
402f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  return visitPreprocessedEntities(PPRec.begin(), PPRec.end(), PPRec);
4034c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor}
4044c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4054c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregortemplate<typename InputIterator>
4064c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregorbool CursorVisitor::visitPreprocessedEntities(InputIterator First,
407f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              InputIterator Last,
408f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              PreprocessingRecord &PPRec,
409f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              FileID FID) {
4104c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  for (; First != Last; ++First) {
411f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (!FID.isInvalid() && !PPRec.isEntityInFileID(First, FID))
412f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      continue;
413f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
414f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    PreprocessedEntity *PPE = *First;
415f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (MacroExpansion *ME = dyn_cast<MacroExpansion>(PPE)) {
4164c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeMacroExpansionCursor(ME, TU)))
4174c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
4184c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4194c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
4204c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    }
4214c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
422f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (MacroDefinition *MD = dyn_cast<MacroDefinition>(PPE)) {
4234c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeMacroDefinitionCursor(MD, TU)))
4244c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
42589d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
4264c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
4274c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    }
4284c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
429f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (InclusionDirective *ID = dyn_cast<InclusionDirective>(PPE)) {
4304c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeInclusionDirectiveCursor(ID, TU)))
4314c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
4324c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4334c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
434788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    }
435788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  }
436788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
4374c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  return false;
438788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor}
439788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
440b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the children of the given cursor.
441a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek///
442b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
443b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
444f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekbool CursorVisitor::VisitChildren(CXCursor Cursor) {
445c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (clang_isReference(Cursor.kind) &&
446c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      Cursor.kind != CXCursor_CXXBaseSpecifier) {
447a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    // By definition, references have no children.
448a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return false;
449a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  }
450f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
451f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Set the Parent field to Cursor, then back to its old value once we're
452b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // done.
4530f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  SetParentRAII SetParent(Parent, StmtParent, Cursor);
454f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
455b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
456b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
45706d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (!D)
45806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return false;
45906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
460539311e0221df256c70c1c3080c8af847cd29dffTed Kremenek    return VisitAttributes(D) || Visit(D);
461b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
462f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
46306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isStatement(Cursor.kind)) {
46406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Stmt *S = getCursorStmt(Cursor))
46506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(S);
46606d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
46706d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
46806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
46906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
47006d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isExpression(Cursor.kind)) {
47106d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Expr *E = getCursorExpr(Cursor))
47206d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(E);
47306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
47406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
47506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
476f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
477b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isTranslationUnit(Cursor.kind)) {
478a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CXTranslationUnit tu = getCursorTU(Cursor);
479a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    ASTUnit *CXXUnit = static_cast<ASTUnit*>(tu->TUData);
48004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
48104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    int VisitOrder[2] = { VisitPreprocessorLast, !VisitPreprocessorLast };
48204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    for (unsigned I = 0; I != 2; ++I) {
48304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      if (VisitOrder[I]) {
48404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
48504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            RegionOfInterest.isInvalid()) {
48604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          for (ASTUnit::top_level_iterator TL = CXXUnit->top_level_begin(),
48704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                        TLEnd = CXXUnit->top_level_end();
48804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor               TL != TLEnd; ++TL) {
489aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis            if (Visit(MakeCXCursor(*TL, tu, RegionOfInterest), true))
49004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
49104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
49204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        } else if (VisitDeclContext(
49304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                CXXUnit->getASTContext().getTranslationUnitDecl()))
4947b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor          return true;
49504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        continue;
4967b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor      }
4973178cb674ac8c3b59e1791e14d38d48619a1b621Bob Wilson
49804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      // Walk the preprocessing record.
4994c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (CXXUnit->getPreprocessor().getPreprocessingRecord())
5004c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        visitPreprocessedEntitiesInRegion();
5010396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    }
50204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
5037b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor    return false;
504b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
505f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
506c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (Cursor.kind == CXCursor_CXXBaseSpecifier) {
507c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    if (CXXBaseSpecifier *Base = getCursorCXXBaseSpecifier(Cursor)) {
508c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      if (TypeSourceInfo *BaseTSInfo = Base->getTypeSourceInfo()) {
509c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor        return Visit(BaseTSInfo->getTypeLoc());
510c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      }
511c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    }
512c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  }
513221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis
514221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis  if (Cursor.kind == CXCursor_IBOutletCollectionAttr) {
515221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis    IBOutletCollectionAttr *A =
516221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis      cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(Cursor));
517221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis    if (const ObjCInterfaceType *InterT = A->getInterface()->getAs<ObjCInterfaceType>())
518221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis      return Visit(cxcursor::MakeCursorObjCClassRef(InterT->getInterface(),
519221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis                                                    A->getInterfaceLoc(), TU));
520221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis  }
521221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis
522b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // Nothing to visit at the moment.
523b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
524dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
525dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5261ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenekbool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
52713c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor  if (TypeSourceInfo *TSInfo = B->getSignatureAsWritten())
52813c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor    if (Visit(TSInfo->getTypeLoc()))
52913c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor        return true;
5301ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
531664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  if (Stmt *Body = B->getBody())
532aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Body, StmtParent, TU, RegionOfInterest));
533664cffd330611d78fc0286f539589920a37ca328Ted Kremenek
534664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  return false;
5351ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek}
5361ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
537d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekllvm::Optional<bool> CursorVisitor::shouldVisitCursor(CXCursor Cursor) {
538d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (RegionOfInterest.isValid()) {
5396653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange Range = getFullCursorExtent(Cursor, AU->getSourceManager());
540d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Range.isInvalid())
541d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
5426653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
543d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    switch (CompareRegionOfInterest(Range)) {
544d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeBefore:
545d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes before the region of interest; skip it.
546d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
54723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
548d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeAfter:
549d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes after the region of interest; we're done.
550d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
551d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
552d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeOverlap:
553d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration overlaps the region of interest; visit it.
554d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
555d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
556d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
557d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return true;
558d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
559f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
560d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekbool CursorVisitor::VisitDeclContext(DeclContext *DC) {
561d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator I = DC->decls_begin(), E = DC->decls_end();
562f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
563d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually remove.  This part of a hack to support proper
564d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // iteration over all Decls contained lexically within an ObjC container.
565d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator*> DI_saved(DI_current, &I);
566d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator> DE_saved(DE_current, E);
567f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
568d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for ( ; I != E; ++I) {
569d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *D = *I;
570d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (D->getLexicalDeclContext() != DC)
571d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
572aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    CXCursor Cursor = MakeCXCursor(D, TU, RegionOfInterest);
5731836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis
574c178d76c027d4259c00d4b0b910ad4875c286298Argyrios Kyrtzidis    // Ignore synthesized ivars here, otherwise if we have something like:
575c178d76c027d4259c00d4b0b910ad4875c286298Argyrios Kyrtzidis    //   @synthesize prop = _prop;
576c178d76c027d4259c00d4b0b910ad4875c286298Argyrios Kyrtzidis    // and '_prop' is not declared, we will encounter a '_prop' ivar before
577c178d76c027d4259c00d4b0b910ad4875c286298Argyrios Kyrtzidis    // encountering the 'prop' synthesize declaration and we will think that
578c178d76c027d4259c00d4b0b910ad4875c286298Argyrios Kyrtzidis    // we passed the region-of-interest.
579c178d76c027d4259c00d4b0b910ad4875c286298Argyrios Kyrtzidis    if (ObjCIvarDecl *ivarD = dyn_cast<ObjCIvarDecl>(D)) {
580c178d76c027d4259c00d4b0b910ad4875c286298Argyrios Kyrtzidis      if (ivarD->getSynthesize())
581c178d76c027d4259c00d4b0b910ad4875c286298Argyrios Kyrtzidis        continue;
582c178d76c027d4259c00d4b0b910ad4875c286298Argyrios Kyrtzidis    }
583c178d76c027d4259c00d4b0b910ad4875c286298Argyrios Kyrtzidis
5841836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    // FIXME: ObjCClassRef/ObjCProtocolRef for forward class/protocol
5851836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    // declarations is a mismatch with the compiler semantics.
5861836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    if (Cursor.kind == CXCursor_ObjCInterfaceDecl) {
5871836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      ObjCInterfaceDecl *ID = cast<ObjCInterfaceDecl>(D);
5881836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      if (!ID->isThisDeclarationADefinition())
5891836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis        Cursor = MakeCursorObjCClassRef(ID, ID->getLocation(), TU);
5901836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis
5911836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    } else if (Cursor.kind == CXCursor_ObjCProtocolDecl) {
5921836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      ObjCProtocolDecl *PD = cast<ObjCProtocolDecl>(D);
5931836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      if (!PD->isThisDeclarationADefinition())
5941836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis        Cursor = MakeCursorObjCProtocolRef(PD, PD->getLocation(), TU);
5951836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    }
5961836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis
597d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
598d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
599d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
600d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
601d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
602d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar    if (Visit(Cursor, true))
603b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
604b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
605b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
606dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
607dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
6081ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
6091ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  llvm_unreachable("Translation units are visited directly by Visit()");
6101ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6111ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
612162e1c1b487352434552147967c3dd296ebee2f7Richard Smithbool CursorVisitor::VisitTypeAliasDecl(TypeAliasDecl *D) {
613162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
614162e1c1b487352434552147967c3dd296ebee2f7Richard Smith    return Visit(TSInfo->getTypeLoc());
615162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
616162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return false;
617162e1c1b487352434552147967c3dd296ebee2f7Richard Smith}
618162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
6191ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
6201ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
6211ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(TSInfo->getTypeLoc());
622f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6231ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6241ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6251ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6261ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTagDecl(TagDecl *D) {
6271ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitDeclContext(D);
6281ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6291ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6300ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregorbool CursorVisitor::VisitClassTemplateSpecializationDecl(
6310ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor                                          ClassTemplateSpecializationDecl *D) {
6320ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool ShouldVisitBody = false;
6330ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  switch (D->getSpecializationKind()) {
6340ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_Undeclared:
6350ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ImplicitInstantiation:
6360ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    // Nothing to visit
6370ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return false;
6380ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6390ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDeclaration:
6400ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDefinition:
6410ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6420ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6430ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitSpecialization:
6440ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    ShouldVisitBody = true;
6450ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6460ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6470ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6480ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  // Visit the template arguments used in the specialization.
6490ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (TypeSourceInfo *SpecType = D->getTypeAsWritten()) {
6500ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    TypeLoc TL = SpecType->getTypeLoc();
6510ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    if (TemplateSpecializationTypeLoc *TSTLoc
6520ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          = dyn_cast<TemplateSpecializationTypeLoc>(&TL)) {
6530ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor      for (unsigned I = 0, N = TSTLoc->getNumArgs(); I != N; ++I)
6540ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor        if (VisitTemplateArgumentLoc(TSTLoc->getArgLoc(I)))
6550ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          return true;
6560ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    }
6570ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6580ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6590ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (ShouldVisitBody && VisitCXXRecordDecl(D))
6600ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return true;
6610ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6620ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  return false;
6630ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor}
6640ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
66574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregorbool CursorVisitor::VisitClassTemplatePartialSpecializationDecl(
66674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                   ClassTemplatePartialSpecializationDecl *D) {
66774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
66874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // before visiting these template parameters.
66974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
67074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return true;
67174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
67274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // Visit the partial specialization arguments.
67374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  const TemplateArgumentLoc *TemplateArgs = D->getTemplateArgsAsWritten();
67474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  for (unsigned I = 0, N = D->getNumTemplateArgsAsWritten(); I != N; ++I)
67574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    if (VisitTemplateArgumentLoc(TemplateArgs[I]))
67674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor      return true;
67774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
67874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  return VisitCXXRecordDecl(D);
67974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor}
68074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
681fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
68284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  // Visit the default argument.
68384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
68484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (TypeSourceInfo *DefArg = D->getDefaultArgumentInfo())
68584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      if (Visit(DefArg->getTypeLoc()))
68684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor        return true;
68784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
688fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
689fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
690fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
6911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
6921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInitExpr())
693aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
6941ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6951ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6961ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6977d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
6987d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
6997d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
7007d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      return true;
7017d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
702c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
703c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = DD->getQualifierLoc())
704c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
705c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
706c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
7077d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return false;
7087d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
7097d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
710a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor/// \brief Compare two base or member initializers based on their source order.
711cbb67480094b3bcb5b715acd827cbad55e2a204cSean Huntstatic int CompareCXXCtorInitializers(const void* Xp, const void *Yp) {
712cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *X
713cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Xp);
714cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *Y
715cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Yp);
716a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
717a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  if ((*X)->getSourceOrder() < (*Y)->getSourceOrder())
718a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return -1;
719a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else if ((*X)->getSourceOrder() > (*Y)->getSourceOrder())
720a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 1;
721a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else
722a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 0;
723a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor}
724a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
725b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
72601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (TypeSourceInfo *TSInfo = ND->getTypeSourceInfo()) {
72701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function declaration's syntactic components in the order
72801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // written. This requires a bit of work.
729723df245307a530da5433dfb43accf187dc3e243Abramo Bagnara    TypeLoc TL = TSInfo->getTypeLoc().IgnoreParens();
73001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    FunctionTypeLoc *FTL = dyn_cast<FunctionTypeLoc>(&TL);
73101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
73201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // If we have a function declared directly (without the use of a typedef),
73301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // visit just the return type. Otherwise, just visit the function's type
73401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // now.
73501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if ((FTL && !isa<CXXConversionDecl>(ND) && Visit(FTL->getResultLoc())) ||
73601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor        (!FTL && Visit(TL)))
73701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
73801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
739c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // Visit the nested-name-specifier, if present.
740c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (NestedNameSpecifierLoc QualifierLoc = ND->getQualifierLoc())
741c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      if (VisitNestedNameSpecifierLoc(QualifierLoc))
742c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor        return true;
74301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
74401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the declaration name.
74501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (VisitDeclarationNameInfo(ND->getNameInfo()))
74601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
74701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
74801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Visit explicitly-specified template arguments!
74901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
75001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function parameters, if we have a function type.
75101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (FTL && VisitFunctionTypeLoc(*FTL, true))
75201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
75301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
75401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Attributes?
75501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
75601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
75710620eb5164e31208fcbf0437cd79ae535ed0559Sean Hunt  if (ND->doesThisDeclarationHaveABody() && !ND->isLateTemplateParsed()) {
758a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    if (CXXConstructorDecl *Constructor = dyn_cast<CXXConstructorDecl>(ND)) {
759a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Find the initializers that were written in the source.
7605f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      SmallVector<CXXCtorInitializer *, 4> WrittenInits;
761a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (CXXConstructorDecl::init_iterator I = Constructor->init_begin(),
762a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                          IEnd = Constructor->init_end();
763a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor           I != IEnd; ++I) {
764a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (!(*I)->isWritten())
765a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          continue;
766a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
767a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        WrittenInits.push_back(*I);
768a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
769a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
770a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Sort the initializers in source order
771a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      llvm::array_pod_sort(WrittenInits.begin(), WrittenInits.end(),
772cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt                           &CompareCXXCtorInitializers);
773a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
774a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Visit the initializers in source order
775a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (unsigned I = 0, N = WrittenInits.size(); I != N; ++I) {
776cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt        CXXCtorInitializer *Init = WrittenInits[I];
77700eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet        if (Init->isAnyMemberInitializer()) {
77800eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet          if (Visit(MakeCursorMemberRef(Init->getAnyMember(),
779a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                        Init->getMemberLocation(), TU)))
780a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
78176852c218a207ef43583515cb835b6e855353a0fDouglas Gregor        } else if (TypeSourceInfo *TInfo = Init->getTypeSourceInfo()) {
78276852c218a207ef43583515cb835b6e855353a0fDouglas Gregor          if (Visit(TInfo->getTypeLoc()))
783a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
784a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        }
785a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
786a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        // Visit the initializer value.
787a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (Expr *Initializer = Init->getInit())
788aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis          if (Visit(MakeCXCursor(Initializer, ND, TU, RegionOfInterest)))
789a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
790a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
791a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
792a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
793aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
794a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return true;
795a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  }
796f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
797b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
798b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
799dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
8001ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
8011ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
8021ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
803f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8041ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *BitWidth = D->getBitWidth())
805aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(BitWidth, StmtParent, TU, RegionOfInterest));
806f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8071ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8081ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8091ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
8101ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitVarDecl(VarDecl *D) {
8111ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
8121ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
813f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8141ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInit())
815aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
816f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8171ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8181ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8191ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
82084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D) {
82184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitDeclaratorDecl(D))
82284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
82384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
82484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
82584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (Expr *DefArg = D->getDefaultArgument())
826aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(DefArg, StmtParent, TU, RegionOfInterest));
82784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
82884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
82984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
83084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
831fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
832fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the FunctionDecl
833fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // before visiting these template parameters.
834fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
835fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return true;
836fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
837fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return VisitFunctionDecl(D->getTemplatedDecl());
838fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
839fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
84039d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregorbool CursorVisitor::VisitClassTemplateDecl(ClassTemplateDecl *D) {
84139d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
84239d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // before visiting these template parameters.
84339d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
84439d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return true;
84539d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
84639d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  return VisitCXXRecordDecl(D->getTemplatedDecl());
84739d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor}
84839d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
84984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D) {
85084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
85184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
85284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
85384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited() &&
85484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      VisitTemplateArgumentLoc(D->getDefaultArgument()))
85584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
85684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
85784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
85884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
85984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
8601ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
8614bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor  if (TypeSourceInfo *TSInfo = ND->getResultTypeSourceInfo())
8624bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
8634bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor      return true;
8644bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor
865f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  for (ObjCMethodDecl::param_iterator P = ND->param_begin(),
8661ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       PEnd = ND->param_end();
8671ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       P != PEnd; ++P) {
868aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
8691ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
8701ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  }
871f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8721ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (ND->isThisDeclarationADefinition() &&
873aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
8741ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
875f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8761ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8771ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8781ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
87903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidistemplate <typename DeclIt>
88003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidisstatic void addRangedDeclsInContainer(DeclIt *DI_current, DeclIt DE_current,
88103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                      SourceManager &SM, SourceLocation EndLoc,
88203ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                      SmallVectorImpl<Decl *> &Decls) {
88303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  DeclIt next = *DI_current;
88403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  while (++next != DE_current) {
88503ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    Decl *D_next = *next;
88603ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (!D_next)
88703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      break;
88803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    SourceLocation L = D_next->getLocStart();
88903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (!L.isValid())
89003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      break;
89103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (SM.isBeforeInTranslationUnit(L, EndLoc)) {
89203ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      *DI_current = next;
89303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      Decls.push_back(D_next);
89403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      continue;
89503ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    }
89603ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    break;
89703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  }
89803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis}
89903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
900d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremeneknamespace {
901d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  struct ContainerDeclsSort {
902d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    SourceManager &SM;
903d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    ContainerDeclsSort(SourceManager &sm) : SM(sm) {}
904d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    bool operator()(Decl *A, Decl *B) {
905d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_A = A->getLocStart();
906d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_B = B->getLocStart();
907d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      assert(L_A.isValid() && L_B.isValid());
908d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return SM.isBeforeInTranslationUnit(L_A, L_B);
909d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
910d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  };
911d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
912d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
913a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregorbool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
914d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually convert back to just 'VisitDeclContext()'.  Essentially
915d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // an @implementation can lexically contain Decls that are not properly
916d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // nested in the AST.  When we identify such cases, we need to retrofit
917d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // this nesting here.
91803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  if (!DI_current && !FileDI_current)
919d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
920d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
921d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Scan the Decls that immediately come after the container
922d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // in the current DeclContext.  If any fall within the
923d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // container's lexical region, stash them into a vector
924d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // for later processing.
9255f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<Decl *, 24> DeclsInContainer;
926d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SourceLocation EndLoc = D->getSourceRange().getEnd();
927a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  SourceManager &SM = AU->getSourceManager();
928d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (EndLoc.isValid()) {
92903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (DI_current) {
93003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      addRangedDeclsInContainer(DI_current, DE_current, SM, EndLoc,
93103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                DeclsInContainer);
93203ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    } else {
93303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      addRangedDeclsInContainer(FileDI_current, FileDE_current, SM, EndLoc,
93403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                DeclsInContainer);
935d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
936d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
937d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
938d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // The common case.
939d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (DeclsInContainer.empty())
940d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
941d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
942d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Get all the Decls in the DeclContext, and sort them with the
943d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // additional ones we've collected.  Then visit them.
944d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for (DeclContext::decl_iterator I = D->decls_begin(), E = D->decls_end();
945d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek       I!=E; ++I) {
946d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *subDecl = *I;
9470582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek    if (!subDecl || subDecl->getLexicalDeclContext() != D ||
9480582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek        subDecl->getLocStart().isInvalid())
949d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
950d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclsInContainer.push_back(subDecl);
951d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
952d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
953d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now sort the Decls so that they appear in lexical order.
954d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  std::sort(DeclsInContainer.begin(), DeclsInContainer.end(),
955d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek            ContainerDeclsSort(SM));
956d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
957d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now visit the decls.
9585f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  for (SmallVectorImpl<Decl*>::iterator I = DeclsInContainer.begin(),
959d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek         E = DeclsInContainer.end(); I != E; ++I) {
960aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    CXCursor Cursor = MakeCXCursor(*I, TU, RegionOfInterest);
961d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
962d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
963d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
964d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
965d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
966d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Visit(Cursor, true))
967d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return true;
968d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
969d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return false;
970a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor}
971a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor
972b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
973b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
974b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                   TU)))
975b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
976f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
97778db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
97878db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
97978db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = ND->protocol_end(); I != E; ++I, ++PL)
980b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
981b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
982f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
983a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(ND);
984dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
985dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
9861ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
987bd9482d859a74bf2c45ef8b8aedec61c0e1c8374Douglas Gregor  if (!PID->isThisDeclarationADefinition())
988bd9482d859a74bf2c45ef8b8aedec61c0e1c8374Douglas Gregor    return Visit(MakeCursorObjCProtocolRef(PID, PID->getLocation(), TU));
989bd9482d859a74bf2c45ef8b8aedec61c0e1c8374Douglas Gregor
9901ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
9911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
9921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       E = PID->protocol_end(); I != E; ++I, ++PL)
9931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
9941ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
995f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9961ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(PID);
9971ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
9981ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
99923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenekbool CursorVisitor::VisitObjCPropertyDecl(ObjCPropertyDecl *PD) {
100083cb94269015bf2770ade71e616c5322ea7e76e1Douglas Gregor  if (PD->getTypeSourceInfo() && Visit(PD->getTypeSourceInfo()->getTypeLoc()))
1001fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall    return true;
1002fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall
100323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // FIXME: This implements a workaround with @property declarations also being
100423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // installed in the DeclContext for the @interface.  Eventually this code
100523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // should be removed.
100623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCCategoryDecl *CDecl = dyn_cast<ObjCCategoryDecl>(PD->getDeclContext());
100723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!CDecl || !CDecl->IsClassExtension())
100823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
100923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
101023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCInterfaceDecl *ID = CDecl->getClassInterface();
101123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!ID)
101223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
101323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
101423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  IdentifierInfo *PropertyId = PD->getIdentifier();
101523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCPropertyDecl *prevDecl =
101623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(ID), PropertyId);
101723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
101823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!prevDecl)
101923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
102023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
102123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // Visit synthesized methods since they will be skipped when visiting
102223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // the @interface.
102323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getGetterMethodDecl())
1024a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
1025aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
102623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
102723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
102823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getSetterMethodDecl())
1029a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
1030aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
103123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
103223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
103323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  return false;
103423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek}
103523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
1036b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
1037375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  if (!D->isThisDeclarationADefinition()) {
1038375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    // Forward declaration is treated like a reference.
1039375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    return Visit(MakeCursorObjCClassRef(D, D->getLocation(), TU));
1040375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  }
1041375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor
1042dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek  // Issue callbacks for super class.
1043b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (D->getSuperClass() &&
1044b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1045f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
1046b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                        TU)))
1047b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
1048f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
104978db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
105078db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
105178db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = D->protocol_end(); I != E; ++I, ++PL)
1052b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1053b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1054f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1055a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(D);
1056dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1057dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10581ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
10591ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(D);
10601ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10611ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10621ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
1063ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  // 'ID' could be null when dealing with invalid code.
1064ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  if (ObjCInterfaceDecl *ID = D->getClassInterface())
1065ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek    if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
1066ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek      return true;
1067f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10681ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
10691ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10701ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10711ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
10721ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#if 0
10731ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // Issue callbacks for super class.
10741ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // FIXME: No source location information!
10751ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (D->getSuperClass() &&
10761ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1077f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
10781ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                        TU)))
1079a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return true;
10801ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#endif
1081f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10821ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
1083dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1084dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
1085a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregorbool CursorVisitor::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD) {
1086a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  if (ObjCIvarDecl *Ivar = PD->getPropertyIvarDecl())
1087135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis    if (PD->isIvarNameSpecified())
1088135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      return Visit(MakeCursorMemberRef(Ivar, PD->getPropertyIvarDeclLoc(), TU));
1089a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
1090a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  return false;
1091a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor}
1092a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
10938f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenekbool CursorVisitor::VisitNamespaceDecl(NamespaceDecl *D) {
10948f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  return VisitDeclContext(D);
10958f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek}
10968f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek
10976931900f43cea558c6974075256c07728dbfecc6Douglas Gregorbool CursorVisitor::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
1098c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
10990cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
11000cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1101c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
11026931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
11036931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  return Visit(MakeCursorNamespaceRef(D->getAliasedNamespace(),
11046931900f43cea558c6974075256c07728dbfecc6Douglas Gregor                                      D->getTargetNameLoc(), TU));
11056931900f43cea558c6974075256c07728dbfecc6Douglas Gregor}
11066931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
11077e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUsingDecl(UsingDecl *D) {
1108c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1109dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1110dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1111c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1112dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
11137e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11141f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (Visit(MakeCursorOverloadedDeclRef(D, D->getLocation(), TU)))
11151f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return true;
11161f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
11177e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11187e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11197e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11200a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregorbool CursorVisitor::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
1121c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1122db9924191092b4d426cc066637d81698211846aaDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1123db9924191092b4d426cc066637d81698211846aaDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1124c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
11250a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11260a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  return Visit(MakeCursorNamespaceRef(D->getNominatedNamespaceAsWritten(),
11270a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor                                      D->getIdentLocation(), TU));
11280a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor}
11290a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11307e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1131c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1132dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1133dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1134c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1135dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1136c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11377e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11387e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11397e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11407e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypenameDecl(
11417e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor                                               UnresolvedUsingTypenameDecl *D) {
1142c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1143dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1144dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1145c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1146c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11477e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return false;
11487e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11497e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
115001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitDeclarationNameInfo(DeclarationNameInfo Name) {
115101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  switch (Name.getName().getNameKind()) {
115201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::Identifier:
115301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXLiteralOperatorName:
115401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXOperatorName:
115501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXUsingDirective:
115601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
115701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
115801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConstructorName:
115901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXDestructorName:
116001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConversionFunctionName:
116101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (TypeSourceInfo *TSInfo = Name.getNamedTypeInfo())
116201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return Visit(TSInfo->getTypeLoc());
116301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
116401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
116501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCZeroArgSelector:
116601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCOneArgSelector:
116701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCMultiArgSelector:
116801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Per-identifier location info?
116901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
117001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
11717530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie
11727530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid DeclarationName::Kind!");
117301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
117401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1175c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregorbool CursorVisitor::VisitNestedNameSpecifier(NestedNameSpecifier *NNS,
1176c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                             SourceRange Range) {
1177c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // FIXME: This whole routine is a hack to work around the lack of proper
1178c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // source information in nested-name-specifiers (PR5791). Since we do have
1179c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // a beginning source location, we can visit the first component of the
1180c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // nested-name-specifier, if it's a single-token component.
1181c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  if (!NNS)
1182c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return false;
1183c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1184c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Get the first component in the nested-name-specifier.
1185c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  while (NestedNameSpecifier *Prefix = NNS->getPrefix())
1186c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    NNS = Prefix;
1187c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1188c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  switch (NNS->getKind()) {
1189c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Namespace:
1190c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(), Range.getBegin(),
1191c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                        TU));
1192c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
119314aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor  case NestedNameSpecifier::NamespaceAlias:
119414aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
119514aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor                                        Range.getBegin(), TU));
119614aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor
1197c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpec: {
1198c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // If the type has a form where we know that the beginning of the source
1199c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // range matches up with a reference cursor. Visit the appropriate reference
1200c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // cursor.
1201f4c7371fb1d3cebcfb40abad4537bb82515704eaJohn McCall    const Type *T = NNS->getAsType();
1202c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TypedefType *Typedef = dyn_cast<TypedefType>(T))
1203c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Typedef->getDecl(), Range.getBegin(), TU));
1204c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TagType *Tag = dyn_cast<TagType>(T))
1205c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Tag->getDecl(), Range.getBegin(), TU));
1206c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TemplateSpecializationType *TST
1207c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                      = dyn_cast<TemplateSpecializationType>(T))
1208c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return VisitTemplateName(TST->getTemplateName(), Range.getBegin());
1209c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1210c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1211c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1212c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpecWithTemplate:
1213c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Global:
1214c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Identifier:
1215c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1216c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1217c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1218c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  return false;
1219c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor}
1220c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1221dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregorbool
1222dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas GregorCursorVisitor::VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
12235f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<NestedNameSpecifierLoc, 4> Qualifiers;
1224dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  for (; Qualifier; Qualifier = Qualifier.getPrefix())
1225dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    Qualifiers.push_back(Qualifier);
1226dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1227dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  while (!Qualifiers.empty()) {
1228dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifierLoc Q = Qualifiers.pop_back_val();
1229dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifier *NNS = Q.getNestedNameSpecifier();
1230dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    switch (NNS->getKind()) {
1231dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Namespace:
1232dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(),
1233c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1234dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1235dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1236dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1237dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1238dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1239dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::NamespaceAlias:
1240dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1241c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1242dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1243dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1244dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1245dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1246dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1247dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpec:
1248dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpecWithTemplate:
1249dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(Q.getTypeLoc()))
1250dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1251dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1252dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1253dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1254dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Global:
1255dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Identifier:
1256dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1257dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    }
1258dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1259dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1260dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  return false;
1261dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor}
1262dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1263fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateParameters(
1264fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          const TemplateParameterList *Params) {
1265fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (!Params)
1266fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1267fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1268fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (TemplateParameterList::const_iterator P = Params->begin(),
1269fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          PEnd = Params->end();
1270fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor       P != PEnd; ++P) {
1271aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
1272fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1273fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1274fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1275fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1276fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1277fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
12780b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregorbool CursorVisitor::VisitTemplateName(TemplateName Name, SourceLocation Loc) {
12790b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  switch (Name.getKind()) {
12800b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::Template:
12810b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(Name.getAsTemplateDecl(), Loc, TU));
12820b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12830b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::OverloadedTemplate:
12841f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // Visit the overloaded template set.
12851f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (Visit(MakeCursorOverloadedDeclRef(Name, Loc, TU)))
12861f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return true;
12871f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
12880b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
12890b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12900b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::DependentTemplate:
12910b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
12920b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
12930b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12940b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::QualifiedTemplate:
12950b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
12960b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(
12970b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                  Name.getAsQualifiedTemplateName()->getDecl(),
12980b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                       Loc, TU));
1299146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall
1300146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall  case TemplateName::SubstTemplateTemplateParm:
1301146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall    return Visit(MakeCursorTemplateRef(
1302146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall                         Name.getAsSubstTemplateTemplateParm()->getParameter(),
1303146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall                                       Loc, TU));
13041aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor
13051aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor  case TemplateName::SubstTemplateTemplateParmPack:
13061aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor    return Visit(MakeCursorTemplateRef(
13071aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                  Name.getAsSubstTemplateTemplateParmPack()->getParameterPack(),
13081aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                                       Loc, TU));
13090b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  }
13107530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie
13117530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid TemplateName::Kind!");
13120b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor}
13130b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
1314fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL) {
1315fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  switch (TAL.getArgument().getKind()) {
1316fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Null:
1317fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Integral:
1318fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Pack:
1319fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
132087dd697dcc8ecb64df73ae64d61b8c80ff0c157cDouglas Gregor
1321fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Type:
1322fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (TypeSourceInfo *TSInfo = TAL.getTypeSourceInfo())
1323fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(TSInfo->getTypeLoc());
1324fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1325fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1326fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Declaration:
1327fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceDeclExpression())
1328aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1329fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1330fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1331fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Expression:
1332fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceExpression())
1333aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1334fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1335fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1336fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Template:
1337a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor  case TemplateArgument::TemplateExpansion:
1338b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor    if (VisitNestedNameSpecifierLoc(TAL.getTemplateQualifierLoc()))
1339b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor      return true;
1340b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor
1341a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor    return VisitTemplateName(TAL.getArgument().getAsTemplateOrTemplatePattern(),
13420b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                             TAL.getTemplateNameLoc());
1343fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
13447530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie
13457530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid TemplateArgument::Kind!");
1346fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1347fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1348a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenekbool CursorVisitor::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
1349a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  return VisitDeclContext(D);
1350a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek}
1351a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek
135201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
135301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return Visit(TL.getUnqualifiedLoc());
135401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
135501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1356f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
1357a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTContext &Context = AU->getASTContext();
1358f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1359f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // Some builtin types (such as Objective-C's "id", "sel", and
1360f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // "Class") have associated declarations. Create cursors for those.
1361f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  QualType VisitType;
1362e0a22d06888c13989b3f72db319f1d498bf69153John McCall  switch (TL.getTypePtr()->getKind()) {
13632dde35bc626153492f5f58202506c88a27fbff5bJohn McCall
13646b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Void:
1365f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::NullPtr:
13666b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Dependent:
13672dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define BUILTIN_TYPE(Id, SingletonId)
13682dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define SIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
13692dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define UNSIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
13702dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define FLOATING_TYPE(Id, SingletonId) case BuiltinType::Id:
13712dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define PLACEHOLDER_TYPE(Id, SingletonId) case BuiltinType::Id:
13722dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#include "clang/AST/BuiltinTypes.def"
1373f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
13746b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1375f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCId:
1376f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCIdType();
1377f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
13786b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
13796b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ObjCClass:
13806b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    VisitType = Context.getObjCClassType();
13816b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    break;
13826b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1383f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCSel:
1384f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCSelType();
1385f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
1386f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1387f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1388f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (!VisitType.isNull()) {
1389f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
1390f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
1391f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                     TU));
1392f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1393f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1394f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1395f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1396f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
13977d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
1398162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return Visit(MakeCursorTypeRef(TL.getTypedefNameDecl(), TL.getNameLoc(), TU));
13997d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
14007d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1401f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
1402f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1403f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1404f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1405f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
14066f155de99c59af890817146ec8526bafb6560f1fArgyrios Kyrtzidis  if (TL.isDefinition())
1407aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(TL.getDecl(), TU, RegionOfInterest));
14086f155de99c59af890817146ec8526bafb6560f1fArgyrios Kyrtzidis
1409f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1410f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1411f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1412fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
1413960d13dde337a59dacc9dc3936c26d4aa8478986Chandler Carruth  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1414fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1415fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1416f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
1417f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU)))
1418f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1419f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1420c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return false;
1421c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall}
1422c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1423c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCallbool CursorVisitor::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
1424c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseLoc()))
1425c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall    return true;
1426c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1427f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1428f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1429f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                        TU)))
1430f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor      return true;
1431f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1432f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1433f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1434f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1435f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1436f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
1437c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return Visit(TL.getPointeeLoc());
1438f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1439f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1440075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnarabool CursorVisitor::VisitParenTypeLoc(ParenTypeLoc TL) {
1441075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara  return Visit(TL.getInnerLoc());
1442075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara}
1443075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara
1444f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
1445f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1446f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1447f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1448f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
1449f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1450f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1451f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1452f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
1453f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1454f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1455f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1456f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
1457f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1458f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1459f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1460f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
1461f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1462f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1463f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
14643422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidisbool CursorVisitor::VisitAttributedTypeLoc(AttributedTypeLoc TL) {
14653422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis  return Visit(TL.getModifiedLoc());
14663422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis}
14673422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis
146801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL,
146901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor                                         bool SkipResultType) {
147001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (!SkipResultType && Visit(TL.getResultLoc()))
1471f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1472f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1473f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
14745dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek    if (Decl *D = TL.getArg(I))
1475aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(D, TU, RegionOfInterest)))
14765dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek        return true;
1477f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1478f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1479f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1480f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1481f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
1482f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(TL.getElementLoc()))
1483f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1484f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1485f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Expr *Size = TL.getSizeExpr())
1486aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Size, StmtParent, TU, RegionOfInterest));
1487f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1488f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1489f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1490f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1491fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateSpecializationTypeLoc(
1492fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                             TemplateSpecializationTypeLoc TL) {
14930b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  // Visit the template name.
14940b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
14950b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                        TL.getTemplateNameLoc()))
14960b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return true;
1497fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1498fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Visit the template arguments.
1499fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1500fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1501fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1502fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1503fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1504fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1505fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
15062332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
15072332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
15082332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15092332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15102332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
15112332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
1512ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt    return Visit(TSInfo->getTypeLoc());
1513ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt
1514ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt  return false;
1515ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt}
1516ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt
1517ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Huntbool CursorVisitor::VisitUnaryTransformTypeLoc(UnaryTransformTypeLoc TL) {
1518ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
15192332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor    return Visit(TSInfo->getTypeLoc());
15202332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15212332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return false;
15222332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15232332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15242494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregorbool CursorVisitor::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
15252494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15262494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    return true;
15272494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
15282494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  return false;
15292494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor}
15302494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
153194fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregorbool CursorVisitor::VisitDependentTemplateSpecializationTypeLoc(
153294fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor                                    DependentTemplateSpecializationTypeLoc TL) {
153394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the nested-name-specifier, if there is one.
153494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  if (TL.getQualifierLoc() &&
153594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
153694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    return true;
153794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
153894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the template arguments.
153994fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
154094fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
154194fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      return true;
154294fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
154394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  return false;
154494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor}
154594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
15469e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregorbool CursorVisitor::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
15479e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15489e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor    return true;
15499e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15509e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  return Visit(TL.getNamedTypeLoc());
15519e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor}
15529e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15537536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregorbool CursorVisitor::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
15547536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor  return Visit(TL.getPatternLoc());
15557536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor}
15567536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor
1557427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::VisitDecltypeTypeLoc(DecltypeTypeLoc TL) {
1558427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  if (Expr *E = TL.getUnderlyingExpr())
1559427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis    return Visit(MakeCXCursor(E, StmtParent, TU));
1560427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1561427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return false;
1562427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1563427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1564427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::VisitInjectedClassNameTypeLoc(InjectedClassNameTypeLoc TL) {
1565427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1566427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1567427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1568b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedmanbool CursorVisitor::VisitAtomicTypeLoc(AtomicTypeLoc TL) {
1569b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman  return Visit(TL.getValueLoc());
1570b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman}
1571b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman
1572427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis#define DEFAULT_TYPELOC_IMPL(CLASS, PARENT) \
1573427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::Visit##CLASS##TypeLoc(CLASS##TypeLoc TL) { \
1574427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return Visit##PARENT##Loc(TL); \
1575427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1576427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1577427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Complex, Type)
1578427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(ConstantArray, ArrayType)
1579427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(IncompleteArray, ArrayType)
1580427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(VariableArray, ArrayType)
1581427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(DependentSizedArray, ArrayType)
1582427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(DependentSizedExtVector, Type)
1583427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Vector, Type)
1584427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(ExtVector, VectorType)
1585427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(FunctionProto, FunctionType)
1586427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(FunctionNoProto, FunctionType)
1587427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Record, TagType)
1588427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Enum, TagType)
1589427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(SubstTemplateTypeParm, Type)
1590427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(SubstTemplateTypeParmPack, Type)
1591427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Auto, Type)
1592427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
15933064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenekbool CursorVisitor::VisitCXXRecordDecl(CXXRecordDecl *D) {
1594c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
1595c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1596c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1597c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
1598c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
15995e1cdac63c3d9c9b32fa41fa0b2d242a58a20d49John McCall  if (D->isCompleteDefinition()) {
16003064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    for (CXXRecordDecl::base_class_iterator I = D->bases_begin(),
16013064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek         E = D->bases_end(); I != E; ++I) {
16023064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      if (Visit(cxcursor::MakeCursorCXXBaseSpecifier(I, TU)))
16033064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek        return true;
16043064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
16053064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  }
16063064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
16073064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  return VisitTagDecl(D);
16083064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek}
16093064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
161009dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenekbool CursorVisitor::VisitAttributes(Decl *D) {
1611cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt  for (AttrVec::const_iterator i = D->attr_begin(), e = D->attr_end();
1612cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt       i != e; ++i)
1613cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt    if (Visit(MakeCXCursor(*i, D, TU)))
161409dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek        return true;
161509dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
161609dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  return false;
161709dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek}
161809dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
1619c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1620c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Data-recursive visitor methods.
1621c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1622c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
162328a719433411ef782b582946823bc648ddcc4533Ted Kremeneknamespace {
1624035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#define DEF_JOB(NAME, DATA, KIND)\
1625035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass NAME : public VisitorJob {\
1626035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:\
1627035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  NAME(DATA *d, CXCursor parent) : VisitorJob(parent, VisitorJob::KIND, d) {} \
1628035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) { return VJ->getKind() == KIND; }\
1629f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DATA *get() const { return static_cast<DATA*>(data[0]); }\
1630035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1631035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1632035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(StmtVisit, Stmt, StmtVisitKind)
1633035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(MemberExprParts, MemberExpr, MemberExprPartsKind)
1634e4979ccb5960608edce73f3b274eb7c2de15dac5Ted KremenekDEF_JOB(DeclRefExprParts, DeclRefExpr, DeclRefExprPartsKind)
1635035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(OverloadExprParts, OverloadExpr, OverloadExprPartsKind)
1636b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios KyrtzidisDEF_JOB(ExplicitTemplateArgsVisit, ASTTemplateArgumentListInfo,
163760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        ExplicitTemplateArgsVisitKind)
163894d96291cd041adc5731a2294828a9c20e450b74Douglas GregorDEF_JOB(SizeOfPackExprParts, SizeOfPackExpr, SizeOfPackExprPartsKind)
1639011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas GregorDEF_JOB(LambdaExprParts, LambdaExpr, LambdaExprPartsKind)
1640d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios KyrtzidisDEF_JOB(PostChildrenVisit, void, PostChildrenVisitKind)
1641035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#undef DEF_JOB
1642035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1643035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass DeclVisit : public VisitorJob {
1644035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1645035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  DeclVisit(Decl *d, CXCursor parent, bool isFirst) :
1646035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::DeclVisitKind,
1647035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               d, isFirst ? (void*) 1 : (void*) 0) {}
1648035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
164982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    return VJ->getKind() == DeclVisitKind;
1650035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1651f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  Decl *get() const { return static_cast<Decl*>(data[0]); }
1652f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  bool isFirst() const { return data[1] ? true : false; }
1653035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1654035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass TypeLocVisit : public VisitorJob {
1655035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1656035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  TypeLocVisit(TypeLoc tl, CXCursor parent) :
1657035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::TypeLocVisitKind,
1658035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               tl.getType().getAsOpaquePtr(), tl.getOpaqueData()) {}
1659035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1660035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1661035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return VJ->getKind() == TypeLocVisitKind;
1662035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1663035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
166482f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek  TypeLoc get() const {
1665f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    QualType T = QualType::getFromOpaquePtr(data[0]);
1666f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return TypeLoc(T, data[1]);
1667035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1668035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1669035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1670ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekclass LabelRefVisit : public VisitorJob {
1671ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekpublic:
1672ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelRefVisit(LabelDecl *LD, SourceLocation labelLoc, CXCursor parent)
1673ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner    : VisitorJob(parent, VisitorJob::LabelRefVisitKind, LD,
1674dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 labelLoc.getPtrEncoding()) {}
1675ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek
1676ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1677ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek    return VJ->getKind() == VisitorJob::LabelRefVisitKind;
1678ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  }
1679ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelDecl *get() const { return static_cast<LabelDecl*>(data[0]); }
1680ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  SourceLocation getLoc() const {
1681dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin    return SourceLocation::getFromPtrEncoding(data[1]); }
1682f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek};
1683f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1684f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorclass NestedNameSpecifierLocVisit : public VisitorJob {
1685f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorpublic:
1686f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLocVisit(NestedNameSpecifierLoc Qualifier, CXCursor parent)
1687f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    : VisitorJob(parent, VisitorJob::NestedNameSpecifierLocVisitKind,
1688f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getNestedNameSpecifier(),
1689f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getOpaqueData()) { }
1690f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1691f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  static bool classof(const VisitorJob *VJ) {
1692f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return VJ->getKind() == VisitorJob::NestedNameSpecifierLocVisitKind;
1693f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1694f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1695f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLoc get() const {
1696f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return NestedNameSpecifierLoc(static_cast<NestedNameSpecifier*>(data[0]),
1697f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                                  data[1]);
1698f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1699f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor};
1700f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1701f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekclass DeclarationNameInfoVisit : public VisitorJob {
1702f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekpublic:
1703f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfoVisit(Stmt *S, CXCursor parent)
1704f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : VisitorJob(parent, VisitorJob::DeclarationNameInfoVisitKind, S) {}
1705f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1706f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return VJ->getKind() == VisitorJob::DeclarationNameInfoVisitKind;
1707f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1708f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfo get() const {
1709f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    Stmt *S = static_cast<Stmt*>(data[0]);
1710f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    switch (S->getStmtClass()) {
1711f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    default:
1712f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      llvm_unreachable("Unhandled Stmt");
1713ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor    case clang::Stmt::MSDependentExistsStmtClass:
1714ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor      return cast<MSDependentExistsStmt>(S)->getNameInfo();
1715f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::CXXDependentScopeMemberExprClass:
1716f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<CXXDependentScopeMemberExpr>(S)->getMemberNameInfo();
1717f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::DependentScopeDeclRefExprClass:
1718f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<DependentScopeDeclRefExpr>(S)->getNameInfo();
1719f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    }
1720f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1721ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek};
1722cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekclass MemberRefVisit : public VisitorJob {
1723cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekpublic:
1724cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  MemberRefVisit(FieldDecl *D, SourceLocation L, CXCursor parent)
1725cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    : VisitorJob(parent, VisitorJob::MemberRefVisitKind, D,
1726dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 L.getPtrEncoding()) {}
1727cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  static bool classof(const VisitorJob *VJ) {
1728cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return VJ->getKind() == VisitorJob::MemberRefVisitKind;
1729cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1730cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  FieldDecl *get() const {
1731cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return static_cast<FieldDecl*>(data[0]);
1732cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1733cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  SourceLocation getLoc() const {
1734cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1735cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1736cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek};
173728a719433411ef782b582946823bc648ddcc4533Ted Kremenekclass EnqueueVisitor : public StmtVisitor<EnqueueVisitor, void> {
173828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitorWorkList &WL;
173928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  CXCursor Parent;
174028a719433411ef782b582946823bc648ddcc4533Ted Kremenekpublic:
174128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueVisitor(VisitorWorkList &wl, CXCursor parent)
174228a719433411ef782b582946823bc648ddcc4533Ted Kremenek    : WL(wl), Parent(parent) {}
174328a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1744ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitAddrLabelExpr(AddrLabelExpr *E);
174573d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitBlockExpr(BlockExpr *B);
174628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1747083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  void VisitCompoundStmt(CompoundStmt *S);
174811b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) { /* Do nothing. */ }
1749ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  void VisitMSDependentExistsStmt(MSDependentExistsStmt *S);
1750f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
175111b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXNewExpr(CXXNewExpr *E);
17526d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
175328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
1754cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E);
175573d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E);
1756b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  void VisitCXXTypeidExpr(CXXTypeidExpr *E);
175755b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
17581e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  void VisitCXXUuidofExpr(CXXUuidofExpr *E);
1759dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis  void VisitCXXCatchStmt(CXXCatchStmt *S);
1760e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  void VisitDeclRefExpr(DeclRefExpr *D);
1761035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void VisitDeclStmt(DeclStmt *S);
1762f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E);
1763cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitDesignatedInitExpr(DesignatedInitExpr *E);
176428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitExplicitCastExpr(ExplicitCastExpr *E);
176528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitForStmt(ForStmt *FS);
1766ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitGotoStmt(GotoStmt *GS);
176728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitIfStmt(IfStmt *If);
176828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitInitListExpr(InitListExpr *IE);
176928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitMemberExpr(MemberExpr *M);
1770cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitOffsetOfExpr(OffsetOfExpr *E);
177173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
177228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitObjCMessageExpr(ObjCMessageExpr *M);
177328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitOverloadExpr(OverloadExpr *E);
1774f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne  void VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E);
177528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitStmt(Stmt *S);
177628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitSwitchStmt(SwitchStmt *S);
177728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitWhileStmt(WhileStmt *W);
17782939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  void VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E);
17796ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  void VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E);
17804ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor  void VisitTypeTraitExpr(TypeTraitExpr *E);
178121ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley  void VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E);
1782552622067dc45013d240f73952fece703f5e63bdJohn Wiegley  void VisitExpressionTraitExpr(ExpressionTraitExpr *E);
178328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U);
17849d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  void VisitVAArgExpr(VAArgExpr *E);
178594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  void VisitSizeOfPackExpr(SizeOfPackExpr *E);
17864b9c2d235fb9449e249d74f48ecfec601650de93John McCall  void VisitPseudoObjectExpr(PseudoObjectExpr *E);
17874b9c2d235fb9449e249d74f48ecfec601650de93John McCall  void VisitOpaqueValueExpr(OpaqueValueExpr *E);
1788011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  void VisitLambdaExpr(LambdaExpr *E);
1789ee8aff06f6a96214731de17b2cb6df407c6c1820Douglas Gregor
179028a719433411ef782b582946823bc648ddcc4533Ted Kremenekprivate:
1791f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void AddDeclarationNameInfo(Stmt *S);
1792f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  void AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier);
1793b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis  void AddExplicitTemplateArgs(const ASTTemplateArgumentListInfo *A);
1794cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void AddMemberRef(FieldDecl *D, SourceLocation L);
179528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddStmt(Stmt *S);
1796035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void AddDecl(Decl *D, bool isFirst = true);
179728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddTypeLoc(TypeSourceInfo *TI);
179828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void EnqueueChildren(Stmt *S);
179928a719433411ef782b582946823bc648ddcc4533Ted Kremenek};
180028a719433411ef782b582946823bc648ddcc4533Ted Kremenek} // end anonyous namespace
180128a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1802f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::AddDeclarationNameInfo(Stmt *S) {
1803f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // 'S' should always be non-null, since it comes from the
1804f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // statement we are visiting.
1805f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  WL.push_back(DeclarationNameInfoVisit(S, Parent));
1806f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1807f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1808f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorvoid
1809f3db29fff6a583ecda823cf909ab7737d8d30129Douglas GregorEnqueueVisitor::AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1810f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (Qualifier)
1811f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    WL.push_back(NestedNameSpecifierLocVisit(Qualifier, Parent));
1812f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor}
1813f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
181428a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddStmt(Stmt *S) {
181528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (S)
181628a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(StmtVisit(S, Parent));
181728a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1818035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::AddDecl(Decl *D, bool isFirst) {
181928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (D)
1820035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    WL.push_back(DeclVisit(D, Parent, isFirst));
182128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
182260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenekvoid EnqueueVisitor::
1823b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis  AddExplicitTemplateArgs(const ASTTemplateArgumentListInfo *A) {
182460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (A)
182560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    WL.push_back(ExplicitTemplateArgsVisit(
1826b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis                        const_cast<ASTTemplateArgumentListInfo*>(A), Parent));
182760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek}
1828cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::AddMemberRef(FieldDecl *D, SourceLocation L) {
1829cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  if (D)
1830cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    WL.push_back(MemberRefVisit(D, L, Parent));
1831cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
183228a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddTypeLoc(TypeSourceInfo *TI) {
183328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (TI)
183428a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(TypeLocVisit(TI->getTypeLoc(), Parent));
183528a719433411ef782b582946823bc648ddcc4533Ted Kremenek }
183628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::EnqueueChildren(Stmt *S) {
1837a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  unsigned size = WL.size();
18387502c1d3ce8bb97bcc4f7bebef507040bd93b26fJohn McCall  for (Stmt::child_range Child = S->children(); Child; ++Child) {
183928a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(*Child);
1840a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  }
1841a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  if (size == WL.size())
1842a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek    return;
1843a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // Now reverse the entries we just added.  This will match the DFS
1844a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // ordering performed by the worklist.
1845a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1846a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  std::reverse(I, E);
1847a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek}
1848ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitAddrLabelExpr(AddrLabelExpr *E) {
1849ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(E->getLabel(), E->getLabelLoc(), Parent));
1850ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
185173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitBlockExpr(BlockExpr *B) {
185273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddDecl(B->getBlockDecl());
185373d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
185428a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
185528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
185628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeSourceInfo());
185728a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1858083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenekvoid EnqueueVisitor::VisitCompoundStmt(CompoundStmt *S) {
1859083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  for (CompoundStmt::reverse_body_iterator I = S->body_rbegin(),
1860083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek        E = S->body_rend(); I != E; ++I) {
1861083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek    AddStmt(*I);
1862083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  }
186311b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
1864f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::
1865ba0513de93d2fab6db5ab30b6927209fcc883078Douglas GregorVisitMSDependentExistsStmt(MSDependentExistsStmt *S) {
1866ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  AddStmt(S->getSubStmt());
1867ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  AddDeclarationNameInfo(S);
1868ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = S->getQualifierLoc())
1869ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1870ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor}
1871ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor
1872ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregorvoid EnqueueVisitor::
1873f64d80306144f978148ba92f36f7cea7b671dd34Ted KremenekVisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E) {
1874f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1875f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
18767c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
18777c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1878f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  if (!E->isImplicitAccess())
1879f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    AddStmt(E->getBase());
1880f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
188111b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenekvoid EnqueueVisitor::VisitCXXNewExpr(CXXNewExpr *E) {
18822aed8b88613863f3c439cdfb205bdf8b608fb205Sebastian Redl  // Enqueue the initializer , if any.
18832aed8b88613863f3c439cdfb205bdf8b608fb205Sebastian Redl  AddStmt(E->getInitializer());
188411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the array size, if any.
188511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddStmt(E->getArraySize());
188611b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the allocated type.
188711b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddTypeLoc(E->getAllocatedTypeSourceInfo());
188811b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the placement arguments.
188911b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumPlacementArgs(); I > 0; --I)
189011b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getPlacementArg(I-1));
189111b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
189228a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *CE) {
18938b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek  for (unsigned I = CE->getNumArgs(); I > 1 /* Yes, this is 1 */; --I)
18948b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek    AddStmt(CE->getArg(I-1));
189528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getCallee());
189628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getArg(0));
189728a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1898cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1899cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the name of the type being destroyed.
1900cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getDestroyedTypeInfo());
1901cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the scope type that looks disturbingly like the nested-name-specifier
1902cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // but isn't.
1903cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getScopeTypeInfo());
1904cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the nested-name-specifier.
1905f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
1906f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1907cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit base expression.
1908cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getBase());
1909cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
19106d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenekvoid EnqueueVisitor::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
19116d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
19126d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
191373d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
191473d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  EnqueueChildren(E);
191573d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
191673d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
1917b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenekvoid EnqueueVisitor::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1918b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  EnqueueChildren(E);
1919b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  if (E->isTypeOperand())
1920b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
1921b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek}
192255b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek
192355b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenekvoid EnqueueVisitor::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr
192455b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek                                                     *E) {
192555b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  EnqueueChildren(E);
192655b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
192755b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek}
19281e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenekvoid EnqueueVisitor::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
19291e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  EnqueueChildren(E);
19301e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  if (E->isTypeOperand())
19311e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
19321e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek}
1933dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis
1934dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidisvoid EnqueueVisitor::VisitCXXCatchStmt(CXXCatchStmt *S) {
1935dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis  EnqueueChildren(S);
1936dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis  AddDecl(S->getExceptionDecl());
1937dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis}
1938dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis
1939e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenekvoid EnqueueVisitor::VisitDeclRefExpr(DeclRefExpr *DR) {
194060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (DR->hasExplicitTemplateArgs()) {
194160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    AddExplicitTemplateArgs(&DR->getExplicitTemplateArgs());
194260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  }
1943e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  WL.push_back(DeclRefExprParts(DR, Parent));
1944e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek}
1945f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1946f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1947f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
194800cf3cc2718671aa48e8da264a523b0058a8591eDouglas Gregor  AddNestedNameSpecifierLoc(E->getQualifierLoc());
1949f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1950035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::VisitDeclStmt(DeclStmt *S) {
1951035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  unsigned size = WL.size();
1952035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  bool isFirst = true;
1953035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  for (DeclStmt::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
1954035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek       D != DEnd; ++D) {
1955035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    AddDecl(*D, isFirst);
1956035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    isFirst = false;
1957035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1958035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  if (size == WL.size())
1959035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return;
1960035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // Now reverse the entries we just added.  This will match the DFS
1961035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // ordering performed by the worklist.
1962035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1963035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  std::reverse(I, E);
1964035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek}
1965cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
1966cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getInit());
1967cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  typedef DesignatedInitExpr::Designator Designator;
1968cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (DesignatedInitExpr::reverse_designators_iterator
1969cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D = E->designators_rbegin(), DEnd = E->designators_rend();
1970cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D != DEnd; ++D) {
1971cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isFieldDesignator()) {
1972cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      if (FieldDecl *Field = D->getField())
1973cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        AddMemberRef(Field, D->getFieldLoc());
1974cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1975cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1976cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isArrayDesignator()) {
1977cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getArrayIndex(*D));
1978cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1979cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1980cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    assert(D->isArrayRangeDesignator() && "Unknown designator kind");
1981cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeEnd(*D));
1982cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeStart(*D));
1983cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1984cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
198528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitExplicitCastExpr(ExplicitCastExpr *E) {
198628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
198728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeInfoAsWritten());
198828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
198928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitForStmt(ForStmt *FS) {
199028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getBody());
199128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInc());
199228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getCond());
199328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(FS->getConditionVariable());
199428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInit());
199528a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1996ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitGotoStmt(GotoStmt *GS) {
1997ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(GS->getLabel(), GS->getLabelLoc(), Parent));
1998ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
199928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitIfStmt(IfStmt *If) {
200028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getElse());
200128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getThen());
200228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getCond());
200328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(If->getConditionVariable());
200428a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
200528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitInitListExpr(InitListExpr *IE) {
200628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  // We care about the syntactic form of the initializer list, only.
200728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (InitListExpr *Syntactic = IE->getSyntacticForm())
200828a719433411ef782b582946823bc648ddcc4533Ted Kremenek    IE = Syntactic;
200928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(IE);
201028a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
201128a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitMemberExpr(MemberExpr *M) {
201289629a746019a42797495b091711a1d68467e88aDouglas Gregor  WL.push_back(MemberExprParts(M, Parent));
201389629a746019a42797495b091711a1d68467e88aDouglas Gregor
201489629a746019a42797495b091711a1d68467e88aDouglas Gregor  // If the base of the member access expression is an implicit 'this', don't
201589629a746019a42797495b091711a1d68467e88aDouglas Gregor  // visit it.
201689629a746019a42797495b091711a1d68467e88aDouglas Gregor  // FIXME: If we ever want to show these implicit accesses, this will be
201789629a746019a42797495b091711a1d68467e88aDouglas Gregor  // unfortunate. However, clang_getCursor() relies on this behavior.
201875e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor  if (!M->isImplicitAccess())
201975e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor    AddStmt(M->getBase());
202028a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
202173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
202273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getEncodedTypeSourceInfo());
202373d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
202428a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitObjCMessageExpr(ObjCMessageExpr *M) {
202528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(M);
202628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(M->getClassReceiverTypeInfo());
202728a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2028cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitOffsetOfExpr(OffsetOfExpr *E) {
2029cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the components of the offsetof expression.
2030cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (unsigned N = E->getNumComponents(), I = N; I > 0; --I) {
2031cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    typedef OffsetOfExpr::OffsetOfNode OffsetOfNode;
2032cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    const OffsetOfNode &Node = E->getComponent(I-1);
2033cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    switch (Node.getKind()) {
2034cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Array:
2035cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getIndexExpr(Node.getArrayExprIndex()));
2036cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2037cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Field:
203806dec892b5300b43263d25c5476b506c9d6cfbadAbramo Bagnara      AddMemberRef(Node.getField(), Node.getSourceRange().getEnd());
2039cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2040cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Identifier:
2041cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Base:
2042cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
2043cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
2044cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
2045cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the type into which we're computing the offset.
2046cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
2047cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
204828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitOverloadExpr(OverloadExpr *E) {
204960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
20506045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek  WL.push_back(OverloadExprParts(E, Parent));
20516045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek}
2052f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbournevoid EnqueueVisitor::VisitUnaryExprOrTypeTraitExpr(
2053f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne                                              UnaryExprOrTypeTraitExpr *E) {
20546d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  EnqueueChildren(E);
20556d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  if (E->isArgumentType())
20566d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek    AddTypeLoc(E->getArgumentTypeInfo());
20576d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
205828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitStmt(Stmt *S) {
205928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(S);
206028a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
206128a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitSwitchStmt(SwitchStmt *S) {
206228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getBody());
206328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getCond());
206428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(S->getConditionVariable());
206528a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2066fafa75aebadef8d6b44a920e3f40529f150a5574Ted Kremenek
206728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitWhileStmt(WhileStmt *W) {
206828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getBody());
206928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getCond());
207028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(W->getConditionVariable());
207128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
207221ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley
20732939b6f356161f572712d4d6310b65f9599e3675Ted Kremenekvoid EnqueueVisitor::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
20742939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  AddTypeLoc(E->getQueriedTypeSourceInfo());
20752939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek}
20766ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
20776ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichetvoid EnqueueVisitor::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
20786ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  AddTypeLoc(E->getRhsTypeSourceInfo());
20790a03a3f98b14006a54bcac9e8908a7c9f50e519fFrancois Pichet  AddTypeLoc(E->getLhsTypeSourceInfo());
20806ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet}
20816ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
20824ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregorvoid EnqueueVisitor::VisitTypeTraitExpr(TypeTraitExpr *E) {
20834ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor  for (unsigned I = E->getNumArgs(); I > 0; --I)
20844ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor    AddTypeLoc(E->getArg(I-1));
20854ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor}
20864ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor
208721ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegleyvoid EnqueueVisitor::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
208821ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley  AddTypeLoc(E->getQueriedTypeSourceInfo());
208921ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley}
209021ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley
2091552622067dc45013d240f73952fece703f5e63bdJohn Wiegleyvoid EnqueueVisitor::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
2092552622067dc45013d240f73952fece703f5e63bdJohn Wiegley  EnqueueChildren(E);
2093552622067dc45013d240f73952fece703f5e63bdJohn Wiegley}
2094552622067dc45013d240f73952fece703f5e63bdJohn Wiegley
209528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U) {
209628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitOverloadExpr(U);
209728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (!U->isImplicitAccess())
209828a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(U->getBase());
209928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
21009d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenekvoid EnqueueVisitor::VisitVAArgExpr(VAArgExpr *E) {
21019d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddStmt(E->getSubExpr());
21029d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddTypeLoc(E->getWrittenTypeInfo());
21039d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek}
210494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregorvoid EnqueueVisitor::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
210594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  WL.push_back(SizeOfPackExprParts(E, Parent));
210694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor}
21074b9c2d235fb9449e249d74f48ecfec601650de93John McCallvoid EnqueueVisitor::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
21084b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // If the opaque value has a source expression, just transparently
21094b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // visit that.  This is useful for (e.g.) pseudo-object expressions.
21104b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (Expr *SourceExpr = E->getSourceExpr())
21114b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return Visit(SourceExpr);
21124b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
2113011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregorvoid EnqueueVisitor::VisitLambdaExpr(LambdaExpr *E) {
2114011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  AddStmt(E->getBody());
2115011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  WL.push_back(LambdaExprParts(E, Parent));
2116011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor}
21174b9c2d235fb9449e249d74f48ecfec601650de93John McCallvoid EnqueueVisitor::VisitPseudoObjectExpr(PseudoObjectExpr *E) {
21184b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Treat the expression like its syntactic form.
21194b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Visit(E->getSyntacticForm());
21204b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
21216045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek
2122c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekvoid CursorVisitor::EnqueueWorkList(VisitorWorkList &WL, Stmt *S) {
2123aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  EnqueueVisitor(WL, MakeCXCursor(S, StmtParent, TU,RegionOfInterest)).Visit(S);
2124c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2125c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2126c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::IsInRegionOfInterest(CXCursor C) {
2127c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  if (RegionOfInterest.isValid()) {
2128c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SourceRange Range = getRawCursorExtent(C);
2129c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    if (Range.isInvalid() || CompareRegionOfInterest(Range))
2130c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      return false;
2131c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2132c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return true;
2133c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2134c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2135c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::RunVisitorWorkList(VisitorWorkList &WL) {
2136c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  while (!WL.empty()) {
2137c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Dequeue the worklist item.
213882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    VisitorJob LI = WL.back();
213982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    WL.pop_back();
214082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek
2141c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Set the Parent field, then back to its old value once we're done.
2142c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SetParentRAII SetParent(Parent, StmtParent, LI.getParent());
2143c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2144c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    switch (LI.getKind()) {
2145f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      case VisitorJob::DeclVisitKind: {
214682f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Decl *D = cast<DeclVisit>(&LI)->get();
2147f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        if (!D)
2148f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek          continue;
2149f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2150f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // For now, perform default visitation for Decls.
2151aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis        if (Visit(MakeCXCursor(D, TU, RegionOfInterest,
2152aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis                               cast<DeclVisit>(&LI)->isFirst())))
2153f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek            return true;
2154f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2155f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        continue;
2156f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      }
215760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      case VisitorJob::ExplicitTemplateArgsVisitKind: {
2158b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis        const ASTTemplateArgumentListInfo *ArgList =
215960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          cast<ExplicitTemplateArgsVisit>(&LI)->get();
216060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        for (const TemplateArgumentLoc *Arg = ArgList->getTemplateArgs(),
216160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               *ArgEnd = Arg + ArgList->NumTemplateArgs;
216260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               Arg != ArgEnd; ++Arg) {
216360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          if (VisitTemplateArgumentLoc(*Arg))
216460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek            return true;
216560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        }
216660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        continue;
216760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      }
2168cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      case VisitorJob::TypeLocVisitKind: {
2169cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        // Perform default visitation for TypeLocs.
217082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        if (Visit(cast<TypeLocVisit>(&LI)->get()))
2171cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek          return true;
2172cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        continue;
2173cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      }
2174ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      case VisitorJob::LabelRefVisitKind: {
2175ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner        LabelDecl *LS = cast<LabelRefVisit>(&LI)->get();
2176e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        if (LabelStmt *stmt = LS->getStmt()) {
2177e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          if (Visit(MakeCursorLabelRef(stmt, cast<LabelRefVisit>(&LI)->getLoc(),
2178e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek                                       TU))) {
2179e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek            return true;
2180e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          }
2181e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        }
2182ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek        continue;
2183ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      }
218447695c8ad8424851f62e0d4a983b45b15daee1c5Ted Kremenek
2185f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      case VisitorJob::NestedNameSpecifierLocVisitKind: {
2186f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        NestedNameSpecifierLocVisit *V = cast<NestedNameSpecifierLocVisit>(&LI);
2187f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        if (VisitNestedNameSpecifierLoc(V->get()))
2188f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor          return true;
2189f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        continue;
2190f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      }
2191f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2192f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      case VisitorJob::DeclarationNameInfoVisitKind: {
2193f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        if (VisitDeclarationNameInfo(cast<DeclarationNameInfoVisit>(&LI)
2194f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                                     ->get()))
2195f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek          return true;
2196f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        continue;
2197f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      }
2198cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      case VisitorJob::MemberRefVisitKind: {
2199cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        MemberRefVisit *V = cast<MemberRefVisit>(&LI);
2200cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (Visit(MakeCursorMemberRef(V->get(), V->getLoc(), TU)))
2201cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          return true;
2202cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        continue;
2203cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      }
2204c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::StmtVisitKind: {
220582f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Stmt *S = cast<StmtVisit>(&LI)->get();
22068c269ac75569454a049385b1246140db5f2b6faaTed Kremenek        if (!S)
22078c269ac75569454a049385b1246140db5f2b6faaTed Kremenek          continue;
22088c269ac75569454a049385b1246140db5f2b6faaTed Kremenek
2209f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // Update the current cursor.
2210aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis        CXCursor Cursor = MakeCXCursor(S, StmtParent, TU, RegionOfInterest);
2211cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (!IsInRegionOfInterest(Cursor))
2212cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          continue;
2213cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        switch (Visitor(Cursor, Parent, ClientData)) {
2214cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Break: return true;
2215cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Continue: break;
2216cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Recurse:
2217d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis            if (PostChildrenVisitor)
2218d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis              WL.push_back(PostChildrenVisit(0, Cursor));
2219cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek            EnqueueWorkList(WL, S);
222082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek            break;
2221c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
222282f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        continue;
2223c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2224c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::MemberExprPartsKind: {
2225c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Handle the other pieces in the MemberExpr besides the base.
222682f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        MemberExpr *M = cast<MemberExprParts>(&LI)->get();
2227c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2228c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the nested-name-specifier
222940d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = M->getQualifierLoc())
223040d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2231c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            return true;
2232c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2233c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the declaration name.
2234c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (VisitDeclarationNameInfo(M->getMemberNameInfo()))
2235c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          return true;
2236c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2237c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the explicitly-specified template arguments, if any.
2238c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (M->hasExplicitTemplateArgs()) {
2239c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          for (const TemplateArgumentLoc *Arg = M->getTemplateArgs(),
2240c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               *ArgEnd = Arg + M->getNumTemplateArgs();
2241c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               Arg != ArgEnd; ++Arg) {
2242c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            if (VisitTemplateArgumentLoc(*Arg))
2243c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek              return true;
2244c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          }
2245c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
2246c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        continue;
2247c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2248e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      case VisitorJob::DeclRefExprPartsKind: {
224982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        DeclRefExpr *DR = cast<DeclRefExprParts>(&LI)->get();
2250e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit nested-name-specifier, if present.
225140d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = DR->getQualifierLoc())
225240d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2253e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek            return true;
2254e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit declaration name.
2255e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        if (VisitDeclarationNameInfo(DR->getNameInfo()))
2256e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek          return true;
2257e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        continue;
2258e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      }
22596045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      case VisitorJob::OverloadExprPartsKind: {
226082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        OverloadExpr *O = cast<OverloadExprParts>(&LI)->get();
22616045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the nested-name-specifier.
22624c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = O->getQualifierLoc())
22634c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
22646045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek            return true;
22656045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the declaration name.
22666045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (VisitDeclarationNameInfo(O->getNameInfo()))
22676045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22686045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the overloaded declaration reference.
22696045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (Visit(MakeCursorOverloadedDeclRef(O, TU)))
22706045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22716045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        continue;
22726045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      }
227394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      case VisitorJob::SizeOfPackExprPartsKind: {
227494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        SizeOfPackExpr *E = cast<SizeOfPackExprParts>(&LI)->get();
227594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        NamedDecl *Pack = E->getPack();
227694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTypeParmDecl>(Pack)) {
227794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTypeRef(cast<TemplateTypeParmDecl>(Pack),
227894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                      E->getPackLoc(), TU)))
227994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
228094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
228194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
228294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
228394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
228494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTemplateParmDecl>(Pack)) {
228594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTemplateRef(cast<TemplateTemplateParmDecl>(Pack),
228694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                          E->getPackLoc(), TU)))
228794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
228894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
228994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
229094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
229194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
229294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // Non-type template parameter packs and function parameter packs are
229394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // treated like DeclRefExpr cursors.
229494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        continue;
229594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      }
2296011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
2297011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      case VisitorJob::LambdaExprPartsKind: {
2298011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        // Visit captures.
2299011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        LambdaExpr *E = cast<LambdaExprParts>(&LI)->get();
2300011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        for (LambdaExpr::capture_iterator C = E->explicit_capture_begin(),
2301011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                                       CEnd = E->explicit_capture_end();
2302011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor             C != CEnd; ++C) {
2303011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          if (C->capturesThis())
2304011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            continue;
2305011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
2306011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          if (Visit(MakeCursorVariableRef(C->getCapturedVar(),
2307011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                                          C->getLocation(),
2308011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                                          TU)))
2309011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            return true;
2310011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        }
2311011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
2312011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        // Visit parameters and return type, if present.
2313011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        if (E->hasExplicitParameters() || E->hasExplicitResultType()) {
2314011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          TypeLoc TL = E->getCallOperator()->getTypeSourceInfo()->getTypeLoc();
2315011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          if (E->hasExplicitParameters() && E->hasExplicitResultType()) {
2316011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            // Visit the whole type.
2317011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            if (Visit(TL))
2318011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              return true;
2319011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          } else if (isa<FunctionProtoTypeLoc>(TL)) {
2320011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            FunctionProtoTypeLoc Proto = cast<FunctionProtoTypeLoc>(TL);
2321011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            if (E->hasExplicitParameters()) {
2322011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              // Visit parameters.
2323011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              for (unsigned I = 0, N = Proto.getNumArgs(); I != N; ++I)
2324011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                if (Visit(MakeCXCursor(Proto.getArg(I), TU)))
2325011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                  return true;
2326011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            } else {
2327011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              // Visit result type.
2328011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              if (Visit(Proto.getResultLoc()))
2329011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                return true;
2330011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            }
2331011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          }
2332011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        }
2333011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        break;
2334011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      }
2335d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis
2336d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis      case VisitorJob::PostChildrenVisitKind:
2337d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis        if (PostChildrenVisitor(Parent, ClientData))
2338d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis          return true;
2339d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis        break;
2340c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    }
2341c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2342c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return false;
2343c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2344c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2345cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekbool CursorVisitor::Visit(Stmt *S) {
2346d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  VisitorWorkList *WL = 0;
2347d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  if (!WorkListFreeList.empty()) {
2348d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = WorkListFreeList.back();
2349d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL->clear();
2350d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListFreeList.pop_back();
2351d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2352d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  else {
2353d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = new VisitorWorkList();
2354d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListCache.push_back(WL);
2355d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2356d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  EnqueueWorkList(*WL, S);
2357d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  bool result = RunVisitorWorkList(*WL);
2358d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  WorkListFreeList.push_back(WL);
2359d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  return result;
2360c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2361c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
236248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichetnamespace {
236348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichettypedef llvm::SmallVector<SourceRange, 4> RefNamePieces;
236448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois PichetRefNamePieces buildPieces(unsigned NameFlags, bool IsMemberRefExpr,
236548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                          const DeclarationNameInfo &NI,
236648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                          const SourceRange &QLoc,
2367b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis                          const ASTTemplateArgumentListInfo *TemplateArgs = 0){
236848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantQualifier = NameFlags & CXNameRange_WantQualifier;
236948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantTemplateArgs = NameFlags & CXNameRange_WantTemplateArgs;
237048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantSinglePiece = NameFlags & CXNameRange_WantSinglePiece;
237148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
237248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const DeclarationName::NameKind Kind = NI.getName().getNameKind();
237348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
237448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  RefNamePieces Pieces;
237548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
237648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantQualifier && QLoc.isValid())
237748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(QLoc);
237848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
237948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (Kind != DeclarationName::CXXOperatorName || IsMemberRefExpr)
238048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(NI.getLoc());
238148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
238248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantTemplateArgs && TemplateArgs)
238348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceRange(TemplateArgs->LAngleLoc,
238448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                                 TemplateArgs->RAngleLoc));
238548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
238648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (Kind == DeclarationName::CXXOperatorName) {
238748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceLocation::getFromRawEncoding(
238848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                       NI.getInfo().CXXOperatorName.BeginOpNameLoc));
238948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceLocation::getFromRawEncoding(
239048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                       NI.getInfo().CXXOperatorName.EndOpNameLoc));
239148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  }
239248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
239348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantSinglePiece) {
239448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    SourceRange R(Pieces.front().getBegin(), Pieces.back().getEnd());
239548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.clear();
239648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(R);
239748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  }
239848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
239948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  return Pieces;
240048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet}
240148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet}
240248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
2403c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2404c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Misc. API hooks.
2405c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2406c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
24078c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic llvm::sys::Mutex EnableMultithreadingMutex;
24088c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic bool EnabledMultithreading;
24098c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
2410fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidisstatic void fatal_error_handler(void *user_data, const std::string& reason) {
2411fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis  // Write the result out to stderr avoiding errs() because raw_ostreams can
2412fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis  // call report_fatal_error.
2413db7a800e0b76036d0faa7123f2e05e45ee3294e5Argyrios Kyrtzidis  fprintf(stderr, "LIBCLANG FATAL ERROR: %s\n", reason.c_str());
2414fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis  ::abort();
2415fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis}
2416fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis
24175e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramerextern "C" {
24180a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas GregorCXIndex clang_createIndex(int excludeDeclarationsFromPCH,
24190a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                          int displayDiagnostics) {
242048615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // Disable pretty stack trace functionality, which will otherwise be a very
242148615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // poor citizen of the world and set up all sorts of signal handlers.
242248615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  llvm::DisablePrettyStackTrace = true;
242348615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar
2424c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // We use crash recovery to make some of our APIs more reliable, implicitly
2425c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // enable it.
2426c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  llvm::CrashRecoveryContext::Enable();
2427c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar
24288c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  // Enable support for multithreading in LLVM.
24298c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  {
24308c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    llvm::sys::ScopedLock L(EnableMultithreadingMutex);
24318c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    if (!EnabledMultithreading) {
2432fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis      llvm::install_fatal_error_handler(fatal_error_handler, 0);
24338c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      llvm::llvm_start_multithreaded();
24348c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      EnabledMultithreading = true;
24358c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    }
24368c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  }
24378c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
2438a030b7cf5e6aad5889b1b662b6979840bc75f87fDouglas Gregor  CIndexer *CIdxr = new CIndexer();
2439e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  if (excludeDeclarationsFromPCH)
2440e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    CIdxr->setOnlyLocalDecls();
24410a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  if (displayDiagnostics)
24420a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    CIdxr->setDisplayDiagnostics();
2443fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2444fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (getenv("LIBCLANG_BGPRIO_INDEX"))
2445fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
2446fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis                               CXGlobalOpt_ThreadBackgroundPriorityForIndexing);
2447fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (getenv("LIBCLANG_BGPRIO_EDIT"))
2448fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
2449fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis                               CXGlobalOpt_ThreadBackgroundPriorityForEditing);
2450fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2451e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  return CIdxr;
2452600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2453600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
24549ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeIndex(CXIndex CIdx) {
24552b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (CIdx)
24562b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    delete static_cast<CIndexer *>(CIdx);
24572bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
24582bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff
2459fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidisvoid clang_CXIndex_setGlobalOptions(CXIndex CIdx, unsigned options) {
2460fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CIdx)
2461fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    static_cast<CIndexer *>(CIdx)->setCXGlobalOptFlags(options);
2462fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis}
2463fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2464fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidisunsigned clang_CXIndex_getGlobalOptions(CXIndex CIdx) {
2465fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CIdx)
2466fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    return static_cast<CIndexer *>(CIdx)->getCXGlobalOptFlags();
2467fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  return 0;
2468fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis}
2469fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2470d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenekvoid clang_toggleCrashRecovery(unsigned isEnabled) {
2471d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  if (isEnabled)
2472d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Enable();
2473d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  else
2474d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Disable();
2475d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek}
2476d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek
24779ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
2478a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                              const char *ast_filename) {
24792b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
24802b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    return 0;
2481f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
24827d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2483389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOptions FileSystemOpts;
2484389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOpts.WorkingDir = CXXIdx->getWorkingDirectory();
24850d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2486c93dc7889644293e318e19d82830ea2acc45b678Dylan Noblesmith  IntrusiveRefCntPtr<DiagnosticsEngine> Diags;
2487a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *TU = ASTUnit::LoadFromASTFile(ast_filename, Diags, FileSystemOpts,
2488a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  CXXIdx->getOnlyLocalDecls(),
2489bef35c91b594f66216f4aab303b71a6c5ab7abcfArgyrios Kyrtzidis                                  0, 0,
2490bef35c91b594f66216f4aab303b71a6c5ab7abcfArgyrios Kyrtzidis                                  /*CaptureDiagnostics=*/true,
2491ff398965a5abfaf5bc47bc022876f56a28e5b9a7Argyrios Kyrtzidis                                  /*AllowPCHWithCompilerErrors=*/true,
2492ff398965a5abfaf5bc47bc022876f56a28e5b9a7Argyrios Kyrtzidis                                  /*UserFilesAreVolatile=*/true);
2493fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  return MakeCXTranslationUnit(CXXIdx, TU);
2494600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2495600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2496b1c031be513705d924038f497279b9b599868ba1Douglas Gregorunsigned clang_defaultEditingTranslationUnitOptions() {
24972a2c50b330e7754499f42173616a36865b5f313bDouglas Gregor  return CXTranslationUnit_PrecompiledPreamble |
2498b5af843a20e237ad1a13ad66a867e200695b8c8eDouglas Gregor         CXTranslationUnit_CacheCompletionResults;
2499b1c031be513705d924038f497279b9b599868ba1Douglas Gregor}
2500b1c031be513705d924038f497279b9b599868ba1Douglas Gregor
25019ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit
25029ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarclang_createTranslationUnitFromSourceFile(CXIndex CIdx,
25039ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          const char *source_filename,
25049ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          int num_command_line_args,
25052ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                          const char * const *command_line_args,
25064db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          unsigned num_unsaved_files,
2507a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                          struct CXUnsavedFile *unsaved_files) {
2508e1d4330adaaa7faf093e725c9c993207eb2d778aArgyrios Kyrtzidis  unsigned Options = CXTranslationUnit_DetailedPreprocessingRecord;
25095a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor  return clang_parseTranslationUnit(CIdx, source_filename,
25105a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    command_line_args, num_command_line_args,
25115a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    unsaved_files, num_unsaved_files,
2512dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor                                    Options);
25135a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor}
2514e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis
251519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbarstruct ParseTranslationUnitInfo {
251619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx;
251719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename;
25182ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char *const *command_line_args;
251919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args;
252019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
252119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files;
252219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options;
252319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXTranslationUnit result;
252419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar};
2525b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_parseTranslationUnit_Impl(void *UserData) {
252619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo *PTUI =
252719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    static_cast<ParseTranslationUnitInfo*>(UserData);
252819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx = PTUI->CIdx;
252919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename = PTUI->source_filename;
25302ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char * const *command_line_args = PTUI->command_line_args;
253119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args = PTUI->num_command_line_args;
253219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files = PTUI->unsaved_files;
253319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files = PTUI->num_unsaved_files;
253419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options = PTUI->options;
253519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  PTUI->result = 0;
25365a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor
25372b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
253819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return;
2539f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2540e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2541e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff
2542fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
254381b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
2544fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
254544c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  bool PrecompilePreamble = options & CXTranslationUnit_PrecompiledPreamble;
2546467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor  // FIXME: Add a flag for modules.
2547467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor  TranslationUnitKind TUKind
2548467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor    = (options & CXTranslationUnit_Incomplete)? TU_Prefix : TU_Complete;
254987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  bool CacheCodeCompetionResults
255087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    = options & CXTranslationUnit_CacheCompletionResults;
2551d99ef536b241071b6f4c01db6525dc03242ac30bDmitri Gribenko  bool IncludeBriefCommentsInCodeCompletion
2552d99ef536b241071b6f4c01db6525dc03242ac30bDmitri Gribenko    = options & CXTranslationUnit_IncludeBriefCommentsInCodeCompletion;
25536a91d385618ea4d28236c496f540a26877c95525Erik Verbruggen  bool SkipFunctionBodies = options & CXTranslationUnit_SkipFunctionBodies;
25546a91d385618ea4d28236c496f540a26877c95525Erik Verbruggen
25555352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  // Configure the diagnostics.
25565352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  DiagnosticOptions DiagOpts;
2557c93dc7889644293e318e19d82830ea2acc45b678Dylan Noblesmith  IntrusiveRefCntPtr<DiagnosticsEngine>
255825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Diags(CompilerInstance::createDiagnostics(DiagOpts, num_command_line_args,
255925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                                command_line_args));
256025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
256125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
2562d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie  llvm::CrashRecoveryContextCleanupRegistrar<DiagnosticsEngine,
2563d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie    llvm::CrashRecoveryContextReleaseRefCleanup<DiagnosticsEngine> >
256425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    DiagCleanup(Diags.getPtr());
256525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
25661e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<std::vector<ASTUnit::RemappedFile> >
256725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
256825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
256925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
257025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
257125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
2572f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
25734db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
25745f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2575f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    const llvm::MemoryBuffer *Buffer
2576a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
257725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
257825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
25794db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  }
2580f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
25811e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<std::vector<const char *> >
258225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args(new std::vector<const char*>());
258325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
258425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this method.
258525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<std::vector<const char*> >
258625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    ArgsCleanup(Args.get());
258725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
258852ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // Since the Clang C library is primarily used by batch tools dealing with
258952ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // (often very broken) source code, where spell-checking can have a
259052ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // significant negative impact on performance (particularly when
259152ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // precompiled headers are involved), we disable it by default.
2592b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  // Only do this if we haven't found a spell-checking-related argument.
2593b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  bool FoundSpellCheckingArgument = false;
2594b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  for (int I = 0; I != num_command_line_args; ++I) {
2595b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (strcmp(command_line_args[I], "-fno-spell-checking") == 0 ||
2596b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        strcmp(command_line_args[I], "-fspell-checking") == 0) {
2597b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      FoundSpellCheckingArgument = true;
2598b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      break;
2599e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    }
2600b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  }
2601b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  if (!FoundSpellCheckingArgument)
260225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-fno-spell-checking");
2603b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
260425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  Args->insert(Args->end(), command_line_args,
260525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek               command_line_args + num_command_line_args);
2606d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2607c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // The 'source_filename' argument is optional.  If the caller does not
2608c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // specify it then it is assumed that the source file is specified
2609c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // in the actual argument list.
2610c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // Put the source file after command_line_args otherwise if '-x' flag is
2611c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // present it will be unused.
2612c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  if (source_filename)
261325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back(source_filename);
2614c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis
261544c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  // Do we need the detailed preprocessing record?
261644c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  if (options & CXTranslationUnit_DetailedPreprocessingRecord) {
261725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-Xclang");
261825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-detailed-preprocessing-record");
261944c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  }
262044c181aec37789f25f6c15543c164416f72e562aDouglas Gregor
2621026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  unsigned NumErrors = Diags->getClient()->getNumErrors();
2622e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis  OwningPtr<ASTUnit> ErrUnit;
26231e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<ASTUnit> Unit(
26244ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek    ASTUnit::LoadFromCommandLine(Args->size() ? &(*Args)[0] : 0
26254ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 /* vector::data() not portable */,
26264ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 Args->size() ? (&(*Args)[0] + Args->size()) :0,
2627b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 Diags,
2628b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getClangResourcesPath(),
2629b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getOnlyLocalDecls(),
2630e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                                 /*CaptureDiagnostics=*/true,
26314ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 RemappedFiles->size() ? &(*RemappedFiles)[0]:0,
263225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                 RemappedFiles->size(),
2633299a4a967b02c9f0d0d94ad8560e3ced893f9116Argyrios Kyrtzidis                                 /*RemappedFilesKeepOriginalName=*/true,
2634b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 PrecompilePreamble,
2635467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor                                 TUKind,
2636bef35c91b594f66216f4aab303b71a6c5ab7abcfArgyrios Kyrtzidis                                 CacheCodeCompetionResults,
2637d99ef536b241071b6f4c01db6525dc03242ac30bDmitri Gribenko                                 IncludeBriefCommentsInCodeCompletion,
2638e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis                                 /*AllowPCHWithCompilerErrors=*/true,
26396a91d385618ea4d28236c496f540a26877c95525Erik Verbruggen                                 SkipFunctionBodies,
2640ff398965a5abfaf5bc47bc022876f56a28e5b9a7Argyrios Kyrtzidis                                 /*UserFilesAreVolatile=*/true,
2641e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis                                 &ErrUnit));
2642b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
2643026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  if (NumErrors != Diags->getClient()->getNumErrors()) {
2644b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    // Make sure to check that 'Unit' is non-NULL.
2645e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis    if (CXXIdx->getDisplayDiagnostics())
2646e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis      printDiagsToStderr(Unit ? Unit.get() : ErrUnit.get());
2647a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  }
2648d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2649fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  PTUI->result = MakeCXTranslationUnit(CXXIdx, Unit.take());
265019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar}
265119ffd492a31a25fb691098bf79f317e5f3edf177Daniel DunbarCXTranslationUnit clang_parseTranslationUnit(CXIndex CIdx,
265219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             const char *source_filename,
26532ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                         const char * const *command_line_args,
265419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             int num_command_line_args,
26559e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                            struct CXUnsavedFile *unsaved_files,
265619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned num_unsaved_files,
265719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned options) {
265819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo PTUI = { CIdx, source_filename, command_line_args,
26599e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_command_line_args, unsaved_files,
26609e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_unsaved_files, options, 0 };
266119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  llvm::CrashRecoveryContext CRC;
266219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
2663bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_parseTranslationUnit_Impl, &PTUI)) {
266460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "libclang: crash detected during parsing: {\n");
266560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'source_filename' : '%s'\n", source_filename);
266660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'command_line_args' : [");
266760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (int i = 0; i != num_command_line_args; ++i) {
266860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
266960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
267060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "'%s'", command_line_args[i]);
267160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
267260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
267360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'unsaved_files' : [");
267460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (unsigned i = 0; i != num_unsaved_files; ++i) {
267560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
267660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
267760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "('%s', '...', %ld)", unsaved_files[i].Filename,
267860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar              unsaved_files[i].Length);
267960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
268060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
268160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'options' : %d,\n", options);
268260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "}\n");
268360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar
268419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return 0;
26856df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
26866df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(PTUI.result);
268719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  }
26886df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
268919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  return PTUI.result;
26905b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff}
26915b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff
26921999844e7a18786e61e619e1dc6c789827541863Douglas Gregorunsigned clang_defaultSaveOptions(CXTranslationUnit TU) {
26931999844e7a18786e61e619e1dc6c789827541863Douglas Gregor  return CXSaveTranslationUnit_None;
26941999844e7a18786e61e619e1dc6c789827541863Douglas Gregor}
2695142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2696142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidisnamespace {
2697142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2698142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidisstruct SaveTranslationUnitInfo {
2699142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  CXTranslationUnit TU;
2700142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  const char *FileName;
2701142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  unsigned options;
2702142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  CXSaveError result;
2703142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis};
2704142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2705142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis}
2706142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2707142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidisstatic void clang_saveTranslationUnit_Impl(void *UserData) {
2708142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  SaveTranslationUnitInfo *STUI =
2709142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    static_cast<SaveTranslationUnitInfo*>(UserData);
2710142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2711fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  CIndexer *CXXIdx = (CIndexer*)STUI->TU->CIdx;
2712fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
271381b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
2714fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2715142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  STUI->result = static_cast<ASTUnit *>(STUI->TU->TUData)->Save(STUI->FileName);
2716142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis}
2717142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
27181999844e7a18786e61e619e1dc6c789827541863Douglas Gregorint clang_saveTranslationUnit(CXTranslationUnit TU, const char *FileName,
27191999844e7a18786e61e619e1dc6c789827541863Douglas Gregor                              unsigned options) {
27207ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  if (!TU)
272139c411fa229b2a6747b92f945d1702ee674d3470Douglas Gregor    return CXSaveError_InvalidTU;
2722142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2723142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
2724142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2725374a00bcc6e26b4fc3cd1d378a5d056c4c7d618eArgyrios Kyrtzidis  if (!CXXUnit->hasSema())
2726374a00bcc6e26b4fc3cd1d378a5d056c4c7d618eArgyrios Kyrtzidis    return CXSaveError_InvalidTU;
2727142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2728142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  SaveTranslationUnitInfo STUI = { TU, FileName, options, CXSaveError_None };
2729142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2730142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  if (!CXXUnit->getDiagnostics().hasUnrecoverableErrorOccurred() ||
2731142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis      getenv("LIBCLANG_NOTHREADS")) {
2732142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    clang_saveTranslationUnit_Impl(&STUI);
2733142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2734142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    if (getenv("LIBCLANG_RESOURCE_USAGE"))
2735142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis      PrintLibclangResourceUsage(TU);
2736142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2737142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    return STUI.result;
2738142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  }
2739142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2740142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  // We have an AST that has invalid nodes due to compiler errors.
2741142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  // Use a crash recovery thread for protection.
2742142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2743142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  llvm::CrashRecoveryContext CRC;
2744142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2745142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  if (!RunSafely(CRC, clang_saveTranslationUnit_Impl, &STUI)) {
2746142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "libclang: crash detected during AST saving: {\n");
2747142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "  'filename' : '%s'\n", FileName);
2748142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "  'options' : %d,\n", options);
2749142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "}\n");
2750142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2751142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    return CXSaveError_Unknown;
2752142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2753142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
27546df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(TU);
2755142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  }
2756142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2757142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  return STUI.result;
27587ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor}
275919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
27609ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
2761ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  if (CTUnit) {
2762ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // If the translation unit has been marked as unsafe to free, just discard
2763ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // it.
2764a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    if (static_cast<ASTUnit *>(CTUnit->TUData)->isUnsafeToFree())
2765ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar      return;
2766ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2767a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete static_cast<ASTUnit *>(CTUnit->TUData);
2768a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    disposeCXStringPool(CTUnit->StringPool);
2769153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek    delete static_cast<CXDiagnosticSetImpl *>(CTUnit->Diagnostics);
2770bbf66ca1dad17773cc682d69b8482c4e179aeaebTed Kremenek    disposeOverridenCXCursorsPool(CTUnit->OverridenCursorsPool);
2771a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete CTUnit;
2772ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  }
27732bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
27740d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2775e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregorunsigned clang_defaultReparseOptions(CXTranslationUnit TU) {
2776e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor  return CXReparse_None;
2777e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor}
2778e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor
2779ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarstruct ReparseTranslationUnitInfo {
2780ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU;
2781ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files;
2782ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
2783ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options;
2784ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  int result;
2785ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar};
2786593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2787b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_reparseTranslationUnit_Impl(void *UserData) {
2788ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo *RTUI =
2789ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    static_cast<ReparseTranslationUnitInfo*>(UserData);
2790ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU = RTUI->TU;
2791153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek
2792153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek  // Reset the associated diagnostics.
2793153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek  delete static_cast<CXDiagnosticSetImpl*>(TU->Diagnostics);
2794153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek  TU->Diagnostics = 0;
2795153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek
2796ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files = RTUI->num_unsaved_files;
2797ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files = RTUI->unsaved_files;
2798ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options = RTUI->options;
2799ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  (void) options;
2800ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  RTUI->result = 1;
2801ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2802abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  if (!TU)
2803ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return;
2804593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2805fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  CIndexer *CXXIdx = (CIndexer*)TU->CIdx;
2806fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
280781b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
2808fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2809a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
2810593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2811abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
28121e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<std::vector<ASTUnit::RemappedFile> >
281325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
281425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
281525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
281625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
281725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
281825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
2819abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
28205f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2821abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor    const llvm::MemoryBuffer *Buffer
28221abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
282325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
282425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
2825abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  }
2826abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
28274ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek  if (!CXXUnit->Reparse(RemappedFiles->size() ? &(*RemappedFiles)[0] : 0,
28284ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                        RemappedFiles->size()))
2829593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor    RTUI->result = 0;
2830abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor}
2831593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2832ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarint clang_reparseTranslationUnit(CXTranslationUnit TU,
2833ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned num_unsaved_files,
2834ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 struct CXUnsavedFile *unsaved_files,
2835ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned options) {
2836ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo RTUI = { TU, num_unsaved_files, unsaved_files,
2837ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                      options, 0 };
28388c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis
2839e7de9b4a1f4a15620ab15bc8159018df7d54080aArgyrios Kyrtzidis  if (getenv("LIBCLANG_NOTHREADS")) {
28408c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis    clang_reparseTranslationUnit_Impl(&RTUI);
28418c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis    return RTUI.result;
28428c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis  }
28438c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis
2844ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  llvm::CrashRecoveryContext CRC;
2845ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2846bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_reparseTranslationUnit_Impl, &RTUI)) {
2847b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbar    fprintf(stderr, "libclang: crash detected during reparsing\n");
2848a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    static_cast<ASTUnit *>(TU->TUData)->setUnsafeToFree(true);
2849ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return 1;
28506df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  } else if (getenv("LIBCLANG_RESOURCE_USAGE"))
28516df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(TU);
28521dfb26af4d6aa4f7818e256659a79f1ec2cba784Ted Kremenek
2853ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  return RTUI.result;
2854ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar}
2855ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2856df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor
28579ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
28582b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CTUnit)
2859ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2860f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2861a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit->TUData);
2862ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(CXXUnit->getOriginalSourceFileName(), true);
2863af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff}
28641eb79b58e56b99cf557d5d353586a10c5360364dDaniel Dunbar
28657eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas GregorCXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
28668e5900c8e9bd32bcc385124f564f3d38a11d0e28Douglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit*>(TU->TUData);
28678e5900c8e9bd32bcc385124f564f3d38a11d0e28Douglas Gregor  return MakeCXCursor(CXXUnit->getASTContext().getTranslationUnitDecl(), TU);
28687eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
28697eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
2870fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2871600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2872fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2873fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXFile Operations.
2874fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2875fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2876fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
287774844072411bae91d5dbb89955d200cbe1e0a1c8Ted KremenekCXString clang_getFileName(CXFile SFile) {
287898258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
2879a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return createCXString((const char*)NULL);
2880f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
288188145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
288274844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek  return createCXString(FEnt->getName());
288388145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
288488145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff
288588145034694ed5267fa6fa5febc54fadc02bd479Steve Narofftime_t clang_getFileTime(CXFile SFile) {
288698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
288798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor    return 0;
2888f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
288988145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
289088145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  return FEnt->getModificationTime();
2891ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff}
2892f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2893b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXFile clang_getFile(CXTranslationUnit tu, const char *file_name) {
2894b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!tu)
2895b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return 0;
2896f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2897a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2898f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2899b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  FileManager &FMgr = CXXUnit->getFileManager();
290039b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattner  return const_cast<FileEntry *>(FMgr.getFile(file_name));
2901b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2902f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2903dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregorunsigned clang_isFileMultipleIncludeGuarded(CXTranslationUnit tu, CXFile file) {
2904dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  if (!tu || !file)
2905dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor    return 0;
2906dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor
2907dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2908dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  FileEntry *FEnt = static_cast<FileEntry *>(file);
2909dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  return CXXUnit->getPreprocessor().getHeaderSearchInfo()
2910dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor                                          .isFileMultipleIncludeGuarded(FEnt);
2911dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor}
2912dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor
2913fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2914fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2915fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2916fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXCursor Operations.
2917fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2918fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2919fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekstatic Decl *getDeclFromExpr(Stmt *E) {
2920c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis  if (ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
2921db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return getDeclFromExpr(CE->getSubExpr());
2922db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2923fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
2924fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RefExpr->getDecl();
2925fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (MemberExpr *ME = dyn_cast<MemberExpr>(E))
2926fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return ME->getMemberDecl();
2927fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
2928fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RE->getDecl();
2929b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis  if (ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(E)) {
2930b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    if (PRE->isExplicitProperty())
2931b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis      return PRE->getExplicitProperty();
2932b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // It could be messaging both getter and setter as in:
2933b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // ++myobj.myprop;
2934b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // in which case prefer to associate the setter since it is less obvious
2935b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // from inspecting the source that the setter is going to get called.
2936b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    if (PRE->isMessagingSetter())
2937b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis      return PRE->getImplicitPropertySetter();
2938b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    return PRE->getImplicitPropertyGetter();
2939b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis  }
29404b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (PseudoObjectExpr *POE = dyn_cast<PseudoObjectExpr>(E))
29414b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return getDeclFromExpr(POE->getSyntacticForm());
29424b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (OpaqueValueExpr *OVE = dyn_cast<OpaqueValueExpr>(E))
29434b9c2d235fb9449e249d74f48ecfec601650de93John McCall    if (Expr *Src = OVE->getSourceExpr())
29444b9c2d235fb9449e249d74f48ecfec601650de93John McCall      return getDeclFromExpr(Src);
2945db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2946fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (CallExpr *CE = dyn_cast<CallExpr>(E))
2947fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return getDeclFromExpr(CE->getCallee());
29485f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  if (CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E))
294993798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    if (!CE->isElidable())
295093798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CE->getConstructor();
2951fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
2952fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return OME->getMethodDecl();
2953f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2954db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCProtocolExpr *PE = dyn_cast<ObjCProtocolExpr>(E))
2955db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return PE->getProtocol();
2956c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor  if (SubstNonTypeTemplateParmPackExpr *NTTP
2957c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor                              = dyn_cast<SubstNonTypeTemplateParmPackExpr>(E))
2958c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor    return NTTP->getParameterPack();
295994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
296094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    if (isa<NonTypeTemplateParmDecl>(SizeOfPack->getPack()) ||
296194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        isa<ParmVarDecl>(SizeOfPack->getPack()))
296294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      return SizeOfPack->getPack();
2963db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2964fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  return 0;
2965fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek}
2966ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff
2967c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbarstatic SourceLocation getLocationFromExpr(Expr *E) {
2968c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis  if (ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
2969c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis    return getLocationFromExpr(CE->getSubExpr());
2970c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis
2971c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
2972c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return /*FIXME:*/Msg->getLeftLoc();
2973c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
2974c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return DRE->getLocation();
2975c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (MemberExpr *Member = dyn_cast<MemberExpr>(E))
2976c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Member->getMemberLoc();
2977c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
2978c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Ivar->getLocation();
297994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
298094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    return SizeOfPack->getPackLoc();
2981d0469525581a851b68f5b4f960ee4190dcc7c932Argyrios Kyrtzidis  if (ObjCPropertyRefExpr *PropRef = dyn_cast<ObjCPropertyRefExpr>(E))
2982d0469525581a851b68f5b4f960ee4190dcc7c932Argyrios Kyrtzidis    return PropRef->getLocation();
298394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
2984c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  return E->getLocStart();
2985c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar}
2986c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar
2987fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
2988f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2989f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekunsigned clang_visitChildren(CXCursor parent,
2990b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXCursorVisitor visitor,
2991b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXClientData client_data) {
2992f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  CursorVisitor CursorVis(getCursorTU(parent), visitor, client_data,
2993f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                          /*VisitPreprocessorLast=*/false);
2994b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return CursorVis.VisitChildren(parent);
2995b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
2996b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
29973387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#ifndef __has_feature
29983387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#define __has_feature(x) 0
29993387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
30003387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#if __has_feature(blocks)
30013387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef enum CXChildVisitResult
30023387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall     (^CXCursorVisitorBlock)(CXCursor cursor, CXCursor parent);
30033387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30043387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
30053387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
30063387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
30073387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block(cursor, parent);
30083387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30093387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#else
30103387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// If we are compiled with a compiler that doesn't have native blocks support,
30113387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// define and call the block manually, so the
30123387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef struct _CXChildVisitResult
30133387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall{
30143387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	void *isa;
30153387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int flags;
30163387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int reserved;
30179e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar	enum CXChildVisitResult(*invoke)(struct _CXChildVisitResult*, CXCursor,
30189e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                         CXCursor);
30193387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall} *CXCursorVisitorBlock;
30203387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30213387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
30223387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
30233387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
30243387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block->invoke(block, cursor, parent);
30253387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30263387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
30273387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30283387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30299e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbarunsigned clang_visitChildrenWithBlock(CXCursor parent,
30309e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                      CXCursorVisitorBlock block) {
30313387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return clang_visitChildren(parent, visitWithBlock, block);
30323387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30333387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
303478205d4bada39d95097e766af9eb30cdd0159461Douglas Gregorstatic CXString getDeclSpelling(Decl *D) {
303516ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  if (!D)
303616ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    return createCXString("");
303716ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
303816ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  NamedDecl *ND = dyn_cast<NamedDecl>(D);
3039e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  if (!ND) {
30405f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    if (ObjCPropertyImplDecl *PropImpl =dyn_cast<ObjCPropertyImplDecl>(D))
3041e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
3042e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return createCXString(Property->getIdentifier()->getName());
3043e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
3044ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
3045e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  }
3046e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
304778205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
3048ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(OMD->getSelector().getAsString());
3049f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
305078205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
305178205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // No, this isn't the same as the code below. getIdentifier() is non-virtual
305278205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // and returns different names. NamedDecl returns the class name and
305378205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // ObjCCategoryImplDecl returns the category name.
3054ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(CIMP->getIdentifier()->getNameStart());
3055f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30560a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  if (isa<UsingDirectiveDecl>(D))
30570a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("");
30580a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
305936d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith  SmallString<1024> S;
306050aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::raw_svector_ostream os(S);
306150aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  ND->printName(os);
306250aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek
306350aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  return createCXString(os.str());
306478205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor}
3065f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30669ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getCursorSpelling(CXCursor C) {
30677eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  if (clang_isTranslationUnit(C.kind))
3068a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return clang_getTranslationUnitSpelling(
3069a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                            static_cast<CXTranslationUnit>(C.data[2]));
30707eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
3071f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  if (clang_isReference(C.kind)) {
3072f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    switch (C.kind) {
3073acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCSuperClassRef: {
30742e331b938b38057e333fab0ba841130ea8467794Douglas Gregor      ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
3075ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Super->getIdentifier()->getNameStart());
3076acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3077acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCClassRef: {
30781adb082a709f7b588f03672999294e061234b2cfDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
3079ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Class->getIdentifier()->getNameStart());
3080acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3081acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCProtocolRef: {
308278db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor      ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
3083f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      assert(OID && "getCursorSpelling(): Missing protocol decl");
3084ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(OID->getIdentifier()->getNameStart());
3085acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
30863064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
30873064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = getCursorCXXBaseSpecifier(C);
30883064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return createCXString(B->getType().getAsString());
30893064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
30907d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    case CXCursor_TypeRef: {
30917d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      TypeDecl *Type = getCursorTypeRef(C).first;
30927d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      assert(Type && "Missing type decl");
30937d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3094ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(getCursorContext(C).getTypeDeclType(Type).
3095ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                              getAsString());
30967d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
30970b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
30980b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      TemplateDecl *Template = getCursorTemplateRef(C).first;
30996931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(Template && "Missing template decl");
31000b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
31010b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString(Template->getNameAsString());
31020b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
31036931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
31046931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
31056931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      NamedDecl *NS = getCursorNamespaceRef(C).first;
31066931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(NS && "Missing namespace decl");
31076931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
31086931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return createCXString(NS->getNameAsString());
31096931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
31107d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3111a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3112a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      FieldDecl *Field = getCursorMemberRef(C).first;
3113a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      assert(Field && "Missing member decl");
3114a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3115a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return createCXString(Field->getNameAsString());
3116a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3117a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
311836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
311936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      LabelStmt *Label = getCursorLabelRef(C).first;
312036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      assert(Label && "Missing label");
312136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
3122ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
312336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
312436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31251f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef: {
31261f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
31271f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Decl *D = Storage.dyn_cast<Decl *>()) {
31281f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        if (NamedDecl *ND = dyn_cast<NamedDecl>(D))
31291f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor          return createCXString(ND->getNameAsString());
31301f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
31311f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      }
31321f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
31331f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString(E->getName().getAsString());
31341f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedTemplateStorage *Ovl
31351f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        = Storage.get<OverloadedTemplateStorage*>();
31361f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Ovl->size() == 0)
31371f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
31381f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return createCXString((*Ovl->begin())->getNameAsString());
31391f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    }
31401f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3141011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef: {
3142011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      VarDecl *Var = getCursorVariableRef(C).first;
3143011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      assert(Var && "Missing variable decl");
3144011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
3145011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return createCXString(Var->getNameAsString());
3146011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    }
3147011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
3148acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    default:
3149ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString("<not implemented>");
3150f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    }
3151f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  }
315297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
315397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
315497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    Decl *D = getDeclFromExpr(getCursorExpr(C));
315597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
315678205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor      return getDeclSpelling(D);
3157ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
315897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
315997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
316036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
316136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
316236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
3163ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
316436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
316536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("");
316636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
316736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31689b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion)
31699e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth    return createCXString(getCursorMacroExpansion(C)->getName()
31704ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor                                                           ->getNameStart());
31714ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor
3172572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
3173572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString(getCursorMacroDefinition(C)->getName()
3174572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor                                                           ->getNameStart());
3175572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3176ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective)
3177ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString(getCursorInclusionDirective(C)->getFileName());
3178ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
317960cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor  if (clang_isDeclaration(C.kind))
318060cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor    return getDeclSpelling(getCursorDecl(C));
3181e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
31825f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  if (C.kind == CXCursor_AnnotateAttr) {
31835f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    AnnotateAttr *AA = cast<AnnotateAttr>(cxcursor::getCursorAttr(C));
31845f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    return createCXString(AA->getAnnotation());
31855f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  }
31865f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen
318784b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis  if (C.kind == CXCursor_AsmLabelAttr) {
318884b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis    AsmLabelAttr *AA = cast<AsmLabelAttr>(cxcursor::getCursorAttr(C));
318984b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis    return createCXString(AA->getLabel());
319084b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis  }
319184b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis
3192ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString("");
3193f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3194f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
3195ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios KyrtzidisCXSourceRange clang_Cursor_getSpellingNameRange(CXCursor C,
3196ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis                                                unsigned pieceIndex,
3197ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis                                                unsigned options) {
3198ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  if (clang_Cursor_isNull(C))
3199ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    return clang_getNullRange();
3200ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3201ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  ASTContext &Ctx = getCursorContext(C);
3202ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3203ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  if (clang_isStatement(C.kind)) {
3204ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    Stmt *S = getCursorStmt(C);
3205ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    if (LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S)) {
3206ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      if (pieceIndex > 0)
3207ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis        return clang_getNullRange();
3208ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      return cxloc::translateSourceRange(Ctx, Label->getIdentLoc());
3209ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    }
3210ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3211ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    return clang_getNullRange();
3212ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  }
3213ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3214ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  if (C.kind == CXCursor_ObjCMessageExpr) {
3215ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    if (ObjCMessageExpr *
3216ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis          ME = dyn_cast_or_null<ObjCMessageExpr>(getCursorExpr(C))) {
3217ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      if (pieceIndex >= ME->getNumSelectorLocs())
3218ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis        return clang_getNullRange();
3219ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      return cxloc::translateSourceRange(Ctx, ME->getSelectorLoc(pieceIndex));
3220ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    }
3221ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  }
3222ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3223ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  if (C.kind == CXCursor_ObjCInstanceMethodDecl ||
3224ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      C.kind == CXCursor_ObjCClassMethodDecl) {
3225ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    if (ObjCMethodDecl *
3226ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis          MD = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(C))) {
3227ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      if (pieceIndex >= MD->getNumSelectorLocs())
3228ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis        return clang_getNullRange();
3229ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      return cxloc::translateSourceRange(Ctx, MD->getSelectorLoc(pieceIndex));
3230ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    }
3231ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  }
3232ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
32339482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis  if (C.kind == CXCursor_ObjCCategoryDecl ||
32349482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis      C.kind == CXCursor_ObjCCategoryImplDecl) {
32359482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis    if (pieceIndex > 0)
32369482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis      return clang_getNullRange();
32379482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis    if (ObjCCategoryDecl *
32389482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis          CD = dyn_cast_or_null<ObjCCategoryDecl>(getCursorDecl(C)))
32399482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis      return cxloc::translateSourceRange(Ctx, CD->getCategoryNameLoc());
32409482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis    if (ObjCCategoryImplDecl *
32419482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis          CID = dyn_cast_or_null<ObjCCategoryImplDecl>(getCursorDecl(C)))
32429482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis      return cxloc::translateSourceRange(Ctx, CID->getCategoryNameLoc());
32439482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis  }
32449482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis
3245ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // FIXME: A CXCursor_InclusionDirective should give the location of the
3246ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // filename, but we don't keep track of this.
3247ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3248ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // FIXME: A CXCursor_AnnotateAttr should give the location of the annotation
3249ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // but we don't keep track of this.
3250ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3251ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // FIXME: A CXCursor_AsmLabelAttr should give the location of the label
3252ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // but we don't keep track of this.
3253ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3254ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // Default handling, give the location of the cursor.
3255ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3256ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  if (pieceIndex > 0)
3257ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    return clang_getNullRange();
3258ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3259ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  CXSourceLocation CXLoc = clang_getCursorLocation(C);
3260ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  SourceLocation Loc = cxloc::translateSourceLocation(CXLoc);
3261ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  return cxloc::translateSourceRange(Ctx, Loc);
3262ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis}
3263ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3264358559d8d7b458c5f64941842383a16e61f0828dDouglas GregorCXString clang_getCursorDisplayName(CXCursor C) {
3265358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!clang_isDeclaration(C.kind))
3266358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return clang_getCursorSpelling(C);
3267358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3268358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  Decl *D = getCursorDecl(C);
3269358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!D)
3270358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString("");
3271358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
327230c42404202d2e2512e51efc6066bd614cfdb5a4Douglas Gregor  PrintingPolicy Policy = getCursorContext(C).getPrintingPolicy();
3273358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(D))
3274358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    D = FunTmpl->getTemplatedDecl();
3275358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3276358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
327736d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith    SmallString<64> Str;
3278358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3279a59d20b135bfde058a5a69045bab5ec4e2553f74Benjamin Kramer    OS << *Function;
3280358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->getPrimaryTemplate())
3281358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "<>";
3282358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "(";
3283358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Function->getNumParams(); I != N; ++I) {
3284358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3285358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3286358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << Function->getParamDecl(I)->getType().getAsString(Policy);
3287358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3288358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3289358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->isVariadic()) {
3290358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Function->getNumParams())
3291358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3292358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "...";
3293358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3294358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ")";
3295358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3296358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3297358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3298358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) {
329936d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith    SmallString<64> Str;
3300358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3301a59d20b135bfde058a5a69045bab5ec4e2553f74Benjamin Kramer    OS << *ClassTemplate;
3302358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "<";
3303358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    TemplateParameterList *Params = ClassTemplate->getTemplateParameters();
3304358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Params->size(); I != N; ++I) {
3305358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3306358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3307358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3308358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      NamedDecl *Param = Params->getParam(I);
3309358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Param->getIdentifier()) {
3310358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << Param->getIdentifier()->getName();
3311358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        continue;
3312358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      }
3313358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3314358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // There is no parameter name, which makes this tricky. Try to come up
3315358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // with something useful that isn't too long.
3316358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
3317358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << (TTP->wasDeclaredWithTypename()? "typename" : "class");
3318358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else if (NonTypeTemplateParmDecl *NTTP
3319358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                    = dyn_cast<NonTypeTemplateParmDecl>(Param))
3320358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << NTTP->getType().getAsString(Policy);
3321358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else
3322358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << "template<...> class";
3323358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3324358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3325358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ">";
3326358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3327358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3328358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3329358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateSpecializationDecl *ClassSpec
3330358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                              = dyn_cast<ClassTemplateSpecializationDecl>(D)) {
3331358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    // If the type was explicitly written, use that.
3332358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (TypeSourceInfo *TSInfo = ClassSpec->getTypeAsWritten())
3333358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      return createCXString(TSInfo->getType().getAsString(Policy));
3334358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
333536d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith    SmallString<64> Str;
3336358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3337a59d20b135bfde058a5a69045bab5ec4e2553f74Benjamin Kramer    OS << *ClassSpec;
3338358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << TemplateSpecializationType::PrintTemplateArgumentList(
3339910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().data(),
3340910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().size(),
3341358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                                                Policy);
3342358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3343358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3344358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3345358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  return clang_getCursorSpelling(C);
3346358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor}
3347358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3348e68fff6fc083c6270d835216a3de0b82c6ef0310Ted KremenekCXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
334989922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  switch (Kind) {
3350e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FunctionDecl:
3351e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FunctionDecl");
3352e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypedefDecl:
3353e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypedefDecl");
3354e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumDecl:
3355e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumDecl");
3356e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumConstantDecl:
3357e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumConstantDecl");
3358e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_StructDecl:
3359e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("StructDecl");
3360e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnionDecl:
3361e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnionDecl");
3362e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ClassDecl:
3363e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ClassDecl");
3364e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FieldDecl:
3365e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FieldDecl");
3366e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_VarDecl:
3367e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("VarDecl");
3368e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ParmDecl:
3369e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ParmDecl");
3370e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInterfaceDecl:
3371e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInterfaceDecl");
3372e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryDecl:
3373e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryDecl");
3374e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolDecl:
3375e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolDecl");
3376e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCPropertyDecl:
3377e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCPropertyDecl");
3378e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCIvarDecl:
3379e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCIvarDecl");
3380e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInstanceMethodDecl:
3381e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInstanceMethodDecl");
3382e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassMethodDecl:
3383e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassMethodDecl");
3384e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCImplementationDecl:
3385e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCImplementationDecl");
3386e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryImplDecl:
3387e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryImplDecl");
33888bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek  case CXCursor_CXXMethod:
33898bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek      return createCXString("CXXMethod");
3390e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedDecl:
3391e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedDecl");
3392e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCSuperClassRef:
3393e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCSuperClassRef");
3394e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolRef:
3395e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolRef");
3396e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassRef:
3397e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassRef");
3398e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypeRef:
3399e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypeRef");
34000b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case CXCursor_TemplateRef:
34010b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString("TemplateRef");
34026931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceRef:
34036931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceRef");
3404a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  case CXCursor_MemberRef:
3405a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return createCXString("MemberRef");
340636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelRef:
340736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("LabelRef");
34081f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case CXCursor_OverloadedDeclRef:
34091f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return createCXString("OverloadedDeclRef");
3410011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  case CXCursor_VariableRef:
3411011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    return createCXString("VariableRef");
341242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IntegerLiteral:
341342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IntegerLiteral");
341442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_FloatingLiteral:
341542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("FloatingLiteral");
341642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ImaginaryLiteral:
341742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ImaginaryLiteral");
341842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_StringLiteral:
341942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("StringLiteral");
342042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CharacterLiteral:
342142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CharacterLiteral");
342242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ParenExpr:
342342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ParenExpr");
342442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnaryOperator:
342542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnaryOperator");
342642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ArraySubscriptExpr:
342742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ArraySubscriptExpr");
342842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_BinaryOperator:
342942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("BinaryOperator");
343042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundAssignOperator:
343142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundAssignOperator");
343242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ConditionalOperator:
343342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ConditionalOperator");
343442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CStyleCastExpr:
343542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CStyleCastExpr");
343642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundLiteralExpr:
343742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundLiteralExpr");
343842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_InitListExpr:
343942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("InitListExpr");
344042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_AddrLabelExpr:
344142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("AddrLabelExpr");
344242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_StmtExpr:
344342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("StmtExpr");
344442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GenericSelectionExpr:
344542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GenericSelectionExpr");
344642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GNUNullExpr:
344742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GNUNullExpr");
344842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXStaticCastExpr:
344942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXStaticCastExpr");
345042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXDynamicCastExpr:
345142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXDynamicCastExpr");
345242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXReinterpretCastExpr:
345342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXReinterpretCastExpr");
345442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXConstCastExpr:
345542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXConstCastExpr");
345642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXFunctionalCastExpr:
345742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXFunctionalCastExpr");
345842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXTypeidExpr:
345942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXTypeidExpr");
346042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXBoolLiteralExpr:
346142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXBoolLiteralExpr");
346242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXNullPtrLiteralExpr:
346342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXNullPtrLiteralExpr");
346442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXThisExpr:
346542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXThisExpr");
346642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXThrowExpr:
346742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXThrowExpr");
346842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXNewExpr:
346942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXNewExpr");
347042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXDeleteExpr:
347142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXDeleteExpr");
347242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnaryExpr:
347342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnaryExpr");
347442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCStringLiteral:
347542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCStringLiteral");
3476b3f7542a950ac0d585a7783e825cfe670e05c553Ted Kremenek  case CXCursor_ObjCBoolLiteralExpr:
3477b3f7542a950ac0d585a7783e825cfe670e05c553Ted Kremenek      return createCXString("ObjCBoolLiteralExpr");
347842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCEncodeExpr:
347942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCEncodeExpr");
348042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCSelectorExpr:
348142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCSelectorExpr");
348242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCProtocolExpr:
348342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCProtocolExpr");
348442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCBridgedCastExpr:
348542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCBridgedCastExpr");
34861ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  case CXCursor_BlockExpr:
34871ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek      return createCXString("BlockExpr");
348842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_PackExpansionExpr:
348942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("PackExpansionExpr");
349042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SizeOfPackExpr:
349142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SizeOfPackExpr");
3492011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  case CXCursor_LambdaExpr:
3493011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    return createCXString("LambdaExpr");
349442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnexposedExpr:
349542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnexposedExpr");
3496e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_DeclRefExpr:
3497e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("DeclRefExpr");
3498e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_MemberRefExpr:
3499e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("MemberRefExpr");
3500e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_CallExpr:
3501e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("CallExpr");
3502e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCMessageExpr:
3503e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCMessageExpr");
3504e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedStmt:
3505e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedStmt");
350642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DeclStmt:
350742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DeclStmt");
350836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelStmt:
350936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return createCXString("LabelStmt");
351042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundStmt:
351142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundStmt");
351242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CaseStmt:
351342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CaseStmt");
351442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DefaultStmt:
351542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DefaultStmt");
351642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IfStmt:
351742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IfStmt");
351842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SwitchStmt:
351942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SwitchStmt");
352042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_WhileStmt:
352142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("WhileStmt");
352242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DoStmt:
352342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DoStmt");
352442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ForStmt:
352542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ForStmt");
352642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GotoStmt:
352742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GotoStmt");
352842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IndirectGotoStmt:
352942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IndirectGotoStmt");
353042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ContinueStmt:
353142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ContinueStmt");
353242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_BreakStmt:
353342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("BreakStmt");
353442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ReturnStmt:
353542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ReturnStmt");
3536df5faf5e7ae6823d0af0b801c4ac26d47f2cee97Chad Rosier  case CXCursor_GCCAsmStmt:
3537df5faf5e7ae6823d0af0b801c4ac26d47f2cee97Chad Rosier      return createCXString("GCCAsmStmt");
35388cd64b4c5553fa6284d248336cb7c82dc960a394Chad Rosier  case CXCursor_MSAsmStmt:
35398cd64b4c5553fa6284d248336cb7c82dc960a394Chad Rosier      return createCXString("MSAsmStmt");
354042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtTryStmt:
354142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtTryStmt");
354242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtCatchStmt:
354342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtCatchStmt");
354442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtFinallyStmt:
354542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtFinallyStmt");
354642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtThrowStmt:
354742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtThrowStmt");
354842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtSynchronizedStmt:
354942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtSynchronizedStmt");
355042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAutoreleasePoolStmt:
355142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAutoreleasePoolStmt");
355242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCForCollectionStmt:
355342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCForCollectionStmt");
355442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXCatchStmt:
355542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXCatchStmt");
355642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXTryStmt:
355742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXTryStmt");
355842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXForRangeStmt:
355942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXForRangeStmt");
356042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHTryStmt:
356142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHTryStmt");
356242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHExceptStmt:
356342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHExceptStmt");
356442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHFinallyStmt:
356542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHFinallyStmt");
356642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_NullStmt:
356742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("NullStmt");
3568e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_InvalidFile:
3569e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("InvalidFile");
3570292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek  case CXCursor_InvalidCode:
3571292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek    return createCXString("InvalidCode");
3572e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NoDeclFound:
3573e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NoDeclFound");
3574e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NotImplemented:
3575e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NotImplemented");
3576e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TranslationUnit:
3577e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TranslationUnit");
3578e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_UnexposedAttr:
3579e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("UnexposedAttr");
3580e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_IBActionAttr:
3581e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("attribute(ibaction)");
35829f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_IBOutletAttr:
35839f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor     return createCXString("attribute(iboutlet)");
3584857e918a8a40deb128840308a318bf623d68295fTed Kremenek  case CXCursor_IBOutletCollectionAttr:
3585857e918a8a40deb128840308a318bf623d68295fTed Kremenek      return createCXString("attribute(iboutletcollection)");
35866639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  case CXCursor_CXXFinalAttr:
35876639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      return createCXString("attribute(final)");
35886639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  case CXCursor_CXXOverrideAttr:
35896639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      return createCXString("attribute(override)");
35905f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  case CXCursor_AnnotateAttr:
35915f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    return createCXString("attribute(annotate)");
359284b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis  case CXCursor_AsmLabelAttr:
359384b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis    return createCXString("asm label");
35949f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_PreprocessingDirective:
35959f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return createCXString("preprocessing directive");
3596572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  case CXCursor_MacroDefinition:
3597572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString("macro definition");
35989b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  case CXCursor_MacroExpansion:
35999b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth    return createCXString("macro expansion");
3600ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  case CXCursor_InclusionDirective:
3601ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString("inclusion directive");
36028f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  case CXCursor_Namespace:
36038f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek    return createCXString("Namespace");
3604a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  case CXCursor_LinkageSpec:
3605a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek    return createCXString("LinkageSpec");
36063064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  case CXCursor_CXXBaseSpecifier:
36073064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    return createCXString("C++ base class specifier");
360801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Constructor:
360901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConstructor");
361001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Destructor:
361101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXDestructor");
361201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_ConversionFunction:
361301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConversion");
3614fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTypeParameter:
3615fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTypeParameter");
3616fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_NonTypeTemplateParameter:
3617fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("NonTypeTemplateParameter");
3618fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTemplateParameter:
3619fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTemplateParameter");
3620fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_FunctionTemplate:
3621fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("FunctionTemplate");
362239d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  case CXCursor_ClassTemplate:
362339d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return createCXString("ClassTemplate");
362474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  case CXCursor_ClassTemplatePartialSpecialization:
362574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return createCXString("ClassTemplatePartialSpecialization");
36266931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceAlias:
36276931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceAlias");
36280a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  case CXCursor_UsingDirective:
36290a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("UsingDirective");
36307e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  case CXCursor_UsingDeclaration:
36317e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor    return createCXString("UsingDeclaration");
3632162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  case CXCursor_TypeAliasDecl:
3633352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("TypeAliasDecl");
3634352697a87bca664356f21a838b162084013625eaDouglas Gregor  case CXCursor_ObjCSynthesizeDecl:
3635352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("ObjCSynthesizeDecl");
3636352697a87bca664356f21a838b162084013625eaDouglas Gregor  case CXCursor_ObjCDynamicDecl:
3637352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("ObjCDynamicDecl");
36382dfdb948bef51a601e763191e4becfe59880d382Argyrios Kyrtzidis  case CXCursor_CXXAccessSpecifier:
36392dfdb948bef51a601e763191e4becfe59880d382Argyrios Kyrtzidis    return createCXString("CXXAccessSpecifier");
364089922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  }
3641e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3642deb06bd3566e18f677e76bc435d478b033fe328bTed Kremenek  llvm_unreachable("Unhandled CXCursorKind");
3643600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
364489922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff
3645064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidisstruct GetCursorData {
3646064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  SourceLocation TokenBeginLoc;
36474b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  bool PointsAtMacroArgExpansion;
3648135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis  bool VisitedObjCPropertyImplDecl;
3649135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis  SourceLocation VisitedDeclaratorDeclStartLoc;
3650064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  CXCursor &BestCursor;
3651064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
36524b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  GetCursorData(SourceManager &SM,
36534b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                SourceLocation tokenBegin, CXCursor &outputCursor)
36544b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    : TokenBeginLoc(tokenBegin), BestCursor(outputCursor) {
36554b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    PointsAtMacroArgExpansion = SM.isMacroArgExpansion(tokenBegin);
3656135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis    VisitedObjCPropertyImplDecl = false;
36574b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  }
3658064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis};
3659064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
36604b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidisstatic enum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
36614b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                                                CXCursor parent,
36624b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                                                CXClientData client_data) {
3663064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  GetCursorData *Data = static_cast<GetCursorData *>(client_data);
3664064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  CXCursor *BestCursor = &Data->BestCursor;
36654b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis
36664b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // If we point inside a macro argument we should provide info of what the
36674b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // token is so use the actual cursor, don't replace it with a macro expansion
36684b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // cursor.
36694b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  if (cursor.kind == CXCursor_MacroExpansion && Data->PointsAtMacroArgExpansion)
36704b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    return CXChildVisit_Recurse;
367165ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis
367265ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis  if (clang_isDeclaration(cursor.kind)) {
367365ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // Avoid having the implicit methods override the property decls.
3674a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis    if (ObjCMethodDecl *MD
3675a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis          = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
367665ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis      if (MD->isImplicit())
367765ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis        return CXChildVisit_Break;
3678a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis
3679a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis    } else if (ObjCInterfaceDecl *ID
3680a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis                 = dyn_cast_or_null<ObjCInterfaceDecl>(getCursorDecl(cursor))) {
3681a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // Check that when we have multiple @class references in the same line,
3682a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // that later ones do not override the previous ones.
3683a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // If we have:
3684a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // @class Foo, Bar;
3685a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // source ranges for both start at '@', so 'Bar' will end up overriding
3686a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // 'Foo' even though the cursor location was at 'Foo'.
3687a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      if (BestCursor->kind == CXCursor_ObjCInterfaceDecl ||
3688a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis          BestCursor->kind == CXCursor_ObjCClassRef)
3689a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis        if (ObjCInterfaceDecl *PrevID
3690a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis             = dyn_cast_or_null<ObjCInterfaceDecl>(getCursorDecl(*BestCursor))){
3691a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis         if (PrevID != ID &&
3692a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis             !PrevID->isThisDeclarationADefinition() &&
3693a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis             !ID->isThisDeclarationADefinition())
3694a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis           return CXChildVisit_Break;
3695a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis        }
3696135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis
3697135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis    } else if (DeclaratorDecl *DD
3698135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis                    = dyn_cast_or_null<DeclaratorDecl>(getCursorDecl(cursor))) {
3699135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      SourceLocation StartLoc = DD->getSourceRange().getBegin();
3700135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // Check that when we have multiple declarators in the same line,
3701135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // that later ones do not override the previous ones.
3702135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // If we have:
3703135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // int Foo, Bar;
3704135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // source ranges for both start at 'int', so 'Bar' will end up overriding
3705135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // 'Foo' even though the cursor location was at 'Foo'.
3706135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      if (Data->VisitedDeclaratorDeclStartLoc == StartLoc)
3707135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis        return CXChildVisit_Break;
3708135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      Data->VisitedDeclaratorDeclStartLoc = StartLoc;
3709135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis
3710135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis    } else if (ObjCPropertyImplDecl *PropImp
3711135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis              = dyn_cast_or_null<ObjCPropertyImplDecl>(getCursorDecl(cursor))) {
3712135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      (void)PropImp;
3713135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // Check that when we have multiple @synthesize in the same line,
3714135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // that later ones do not override the previous ones.
3715135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // If we have:
3716135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // @synthesize Foo, Bar;
3717135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // source ranges for both start at '@', so 'Bar' will end up overriding
3718135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // 'Foo' even though the cursor location was at 'Foo'.
3719135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      if (Data->VisitedObjCPropertyImplDecl)
3720135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis        return CXChildVisit_Break;
3721135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      Data->VisitedObjCPropertyImplDecl = true;
3722a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis    }
372365ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis  }
3724064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
3725064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  if (clang_isExpression(cursor.kind) &&
3726064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis      clang_isDeclaration(BestCursor->kind)) {
372716ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (Decl *D = getCursorDecl(*BestCursor)) {
372816ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // Avoid having the cursor of an expression replace the declaration cursor
372916ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // when the expression source range overlaps the declaration range.
373016ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // This can happen for C++ constructor expressions whose range generally
373116ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // include the variable declaration, e.g.:
373216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      //  MyCXXClass foo; // Make sure pointing at 'foo' returns a VarDecl cursor.
373316ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      if (D->getLocation().isValid() && Data->TokenBeginLoc.isValid() &&
373416ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis          D->getLocation() == Data->TokenBeginLoc)
373516ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis        return CXChildVisit_Break;
373616ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    }
3737064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  }
3738064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
373993798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // If our current best cursor is the construction of a temporary object,
374093798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // don't replace that cursor with a type reference, because we want
374193798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // clang_getCursor() to point at the constructor.
374293798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  if (clang_isExpression(BestCursor->kind) &&
374393798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor      isa<CXXTemporaryObjectExpr>(getCursorExpr(*BestCursor)) &&
3744aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      cursor.kind == CXCursor_TypeRef) {
3745aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    // Keep the cursor pointing at CXXTemporaryObjectExpr but also mark it
3746aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    // as having the actual point on the type reference.
3747aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    *BestCursor = getTypeRefedCallExprCursor(*BestCursor);
374893798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CXChildVisit_Recurse;
3749aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  }
375093798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor
375133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  *BestCursor = cursor;
375233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return CXChildVisit_Recurse;
375333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
3754e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3755b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
3756b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!TU)
3757f462989fe8d6f59ab2d7d0fe2b4b96292ce706eaTed Kremenek    return clang_getNullCursor();
3758e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3759a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3760bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
3761bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
3762a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek  SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
3763671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  CXCursor Result = cxcursor::getCursor(TU, SLoc);
3764a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek
376540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  bool Logging = getenv("LIBCLANG_LOGGING");
376640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  if (Logging) {
376740749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile SearchFile;
376840749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned SearchLine, SearchColumn;
376940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile ResultFile;
377040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned ResultLine, ResultColumn;
37716653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    CXString SearchFileName, ResultFileName, KindSpelling, USR;
37726653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    const char *IsDef = clang_isCursorDefinition(Result)? " (Definition)" : "";
377340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXSourceLocation ResultLoc = clang_getCursorLocation(Result);
377440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
377520174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    clang_getExpansionLocation(Loc, &SearchFile, &SearchLine, &SearchColumn, 0);
377620174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    clang_getExpansionLocation(ResultLoc, &ResultFile, &ResultLine,
377720174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                               &ResultColumn, 0);
377840749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    SearchFileName = clang_getFileName(SearchFile);
377940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    ResultFileName = clang_getFileName(ResultFile);
378040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    KindSpelling = clang_getCursorKindSpelling(Result.kind);
37816653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    USR = clang_getCursorUSR(Result);
37826653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    fprintf(stderr, "clang_getCursor(%s:%d:%d) = %s(%s:%d:%d):%s%s\n",
378340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(SearchFileName), SearchLine, SearchColumn,
378440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(KindSpelling),
37856653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(ResultFileName), ResultLine, ResultColumn,
37866653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(USR), IsDef);
378740749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(SearchFileName);
378840749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(ResultFileName);
378940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(KindSpelling);
37906653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    clang_disposeString(USR);
37910aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor
37920aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    CXCursor Definition = clang_getCursorDefinition(Result);
37930aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    if (!clang_equalCursors(Definition, clang_getNullCursor())) {
37940aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXSourceLocation DefinitionLoc = clang_getCursorLocation(Definition);
37950aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionKindSpelling
37960aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor                                = clang_getCursorKindSpelling(Definition.kind);
37970aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXFile DefinitionFile;
37980aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      unsigned DefinitionLine, DefinitionColumn;
379920174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth      clang_getExpansionLocation(DefinitionLoc, &DefinitionFile,
380020174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                 &DefinitionLine, &DefinitionColumn, 0);
38010aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionFileName = clang_getFileName(DefinitionFile);
38020aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      fprintf(stderr, "  -> %s(%s:%d:%d)\n",
38030aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionKindSpelling),
38040aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionFileName),
38050aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              DefinitionLine, DefinitionColumn);
38060aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionFileName);
38070aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionKindSpelling);
38080aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    }
380940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  }
381040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
3811e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  return Result;
381277128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
381377128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
3814738855554394a6afcf39cc8345fd22c3756b8dd0Ted KremenekCXCursor clang_getNullCursor(void) {
38155bfb8c128c2ac8eb4032afc180cdc400a0f953caDouglas Gregor  return MakeCXCursorInvalid(CXCursor_InvalidFile);
3816738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
3817738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek
3818738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenekunsigned clang_equalCursors(CXCursor X, CXCursor Y) {
3819283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  return X == Y;
3820738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
38210d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
38229ce5584553054d0cb934940586aca0186e87fa57Douglas Gregorunsigned clang_hashCursor(CXCursor C) {
38239ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  unsigned Index = 0;
38249ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  if (clang_isExpression(C.kind) || clang_isStatement(C.kind))
38259ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor    Index = 1;
38269ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
38279ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  return llvm::DenseMapInfo<std::pair<unsigned, void*> >::getHashValue(
38289ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor                                        std::make_pair(C.kind, C.data[Index]));
38299ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor}
38309ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
38319ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isInvalid(enum CXCursorKind K) {
383277128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
383377128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
383477128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
38359ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isDeclaration(enum CXCursorKind K) {
383689922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl;
383789922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff}
38382d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff
38399ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isReference(enum CXCursorKind K) {
3840f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
3841f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3842f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
384397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isExpression(enum CXCursorKind K) {
384497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
384597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
384697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
384797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isStatement(enum CXCursorKind K) {
384897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
384997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
385097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
38518be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregorunsigned clang_isAttribute(enum CXCursorKind K) {
38528be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor    return K >= CXCursor_FirstAttr && K <= CXCursor_LastAttr;
38538be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor}
38548be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor
38557eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregorunsigned clang_isTranslationUnit(enum CXCursorKind K) {
38567eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return K == CXCursor_TranslationUnit;
38577eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
38587eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
38599f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregorunsigned clang_isPreprocessing(enum CXCursorKind K) {
38609f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
38619f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor}
38629f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor
3863ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenekunsigned clang_isUnexposed(enum CXCursorKind K) {
3864ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  switch (K) {
3865ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedDecl:
3866ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedExpr:
3867ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedStmt:
3868ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedAttr:
3869ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return true;
3870ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    default:
3871ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return false;
3872ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  }
3873ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek}
3874ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek
38759ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXCursorKind clang_getCursorKind(CXCursor C) {
38769efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff  return C.kind;
38779efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff}
38789efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff
387998258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas GregorCXSourceLocation clang_getCursorLocation(CXCursor C) {
388098258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (clang_isReference(C.kind)) {
3881f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    switch (C.kind) {
3882f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCSuperClassRef: {
3883f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3884f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCSuperClassRef(C);
3885a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3886f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3887f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3888f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
3889f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCProtocolDecl *, SourceLocation> P
3890f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCProtocolRef(C);
3891a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3892f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3893f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3894f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef: {
3895f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3896f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCClassRef(C);
3897a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3898f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
38997d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3900f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef: {
39017d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      std::pair<TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
3902a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
39037d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
39040b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
39050b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
39060b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      std::pair<TemplateDecl *, SourceLocation> P = getCursorTemplateRef(C);
39070b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
39080b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
39090b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
39106931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
39116931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      std::pair<NamedDecl *, SourceLocation> P = getCursorNamespaceRef(C);
39126931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
39136931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
39146931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
3915a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3916a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      std::pair<FieldDecl *, SourceLocation> P = getCursorMemberRef(C);
3917a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3918a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3919a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3920011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef: {
3921011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      std::pair<VarDecl *, SourceLocation> P = getCursorVariableRef(C);
3922011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3923011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    }
3924011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
39253064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
39261b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      CXXBaseSpecifier *BaseSpec = getCursorCXXBaseSpecifier(C);
39271b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (!BaseSpec)
39281b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return clang_getNullLocation();
39291b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
39301b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (TypeSourceInfo *TSInfo = BaseSpec->getTypeSourceInfo())
39311b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return cxloc::translateSourceLocation(getCursorContext(C),
39321b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                            TSInfo->getTypeLoc().getBeginLoc());
39331b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
39341b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
393596a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar                                        BaseSpec->getLocStart());
39363064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
3937f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
393836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
393936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      std::pair<LabelStmt *, SourceLocation> P = getCursorLabelRef(C);
394036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C), P.second);
394136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
394236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
39431f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
39441f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
39451f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                          getCursorOverloadedDeclRef(C).second);
39461f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3947f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    default:
3948f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3949f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      llvm_unreachable("Missed a reference kind");
3950f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
395198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  }
395297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
395397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3954f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    return cxloc::translateSourceLocation(getCursorContext(C),
395597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor                                   getLocationFromExpr(getCursorExpr(C)));
395697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
395736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind))
395836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C),
395936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor                                          getCursorStmt(C)->getLocStart());
396036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
39619f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  if (C.kind == CXCursor_PreprocessingDirective) {
39629f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
39639f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
39649f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  }
39654807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
39669b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion) {
39674ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    SourceLocation L
39689e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth      = cxcursor::getCursorMacroExpansion(C)->getSourceRange().getBegin();
39694807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
39704807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  }
3971572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3972572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition) {
3973572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
3974572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3975572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  }
3976ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3977ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective) {
3978ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    SourceLocation L
3979ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      = cxcursor::getCursorInclusionDirective(C)->getSourceRange().getBegin();
3980ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3981ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  }
3982ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
39839a700d277c38d9afaa7cb3fe93a714bfe9b62eecTed Kremenek  if (C.kind < CXCursor_FirstDecl || C.kind > CXCursor_LastDecl)
39845352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullLocation();
398598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor
3986f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  Decl *D = getCursorDecl(C);
398716ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  if (!D)
398816ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    return clang_getNullLocation();
398916ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
3990f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  SourceLocation Loc = D->getLocation();
3991007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // FIXME: Multiple variables declared in a single declaration
3992007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // currently lack the information needed to correctly determine their
3993007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // ranges when accounting for the type-specifier.  We use context
3994007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3995007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // and if so, whether it is the first decl.
3996007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3997007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (!cxcursor::isFirstInDeclGroup(C))
3998007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      Loc = VD->getLocation();
3999007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
4000007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek
4001ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis  // For ObjC methods, give the start location of the method name.
4002ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis  if (ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
4003ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis    Loc = MD->getSelectorStartLoc();
4004ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis
40052ca54feee89d7277fb967e3247a64f40ef155a82Douglas Gregor  return cxloc::translateSourceLocation(getCursorContext(C), Loc);
400688145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
4007a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor
4008a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor} // end extern "C"
4009a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
4010671436e9e2794c56f3c2e62739d225571493af37Argyrios KyrtzidisCXCursor cxcursor::getCursor(CXTranslationUnit TU, SourceLocation SLoc) {
4011671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  assert(TU);
4012671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
4013671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // Guard against an invalid SourceLocation, or we may assert in one
4014671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // of the following calls.
4015671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  if (SLoc.isInvalid())
4016671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    return clang_getNullCursor();
4017671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
4018671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4019671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
4020671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // Translate the given source location to make it point at the beginning of
4021671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // the token under the cursor.
4022671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  SLoc = Lexer::GetBeginningOfToken(SLoc, CXXUnit->getSourceManager(),
40234e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie                                    CXXUnit->getASTContext().getLangOpts());
4024671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
4025671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
4026671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  if (SLoc.isValid()) {
4027671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    GetCursorData ResultData(CXXUnit->getSourceManager(), SLoc, Result);
4028671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    CursorVisitor CursorVis(TU, GetCursorVisitor, &ResultData,
4029671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                            /*VisitPreprocessorLast=*/true,
4030e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                            /*VisitIncludedEntities=*/false,
4031671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                            SourceLocation(SLoc));
4032dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    CursorVis.visitFileRegion();
4033671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  }
4034671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
4035671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  return Result;
4036671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis}
4037671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
4038a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C) {
4039a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  if (clang_isReference(C.kind)) {
4040a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    switch (C.kind) {
4041a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCSuperClassRef:
4042a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return  getCursorObjCSuperClassRef(C).second;
4043f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4044a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCProtocolRef:
4045a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCProtocolRef(C).second;
4046f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4047a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCClassRef:
4048a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCClassRef(C).second;
40497d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
4050a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_TypeRef:
4051a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorTypeRef(C).second;
40520b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
40530b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
40540b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return getCursorTemplateRef(C).second;
40550b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
40566931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
40576931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return getCursorNamespaceRef(C).second;
4058a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
4059a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
4060a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return getCursorMemberRef(C).second;
4061a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
40623064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier:
40631b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return getCursorCXXBaseSpecifier(C)->getSourceRange();
4064f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
406536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
406636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return getCursorLabelRef(C).second;
406736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
40681f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
40691f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return getCursorOverloadedDeclRef(C).second;
40701f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4071011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef:
4072011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return getCursorVariableRef(C).second;
4073011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
4074a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    default:
4075a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
4076a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      llvm_unreachable("Missed a reference kind");
4077a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    }
4078a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  }
407997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
408097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
4081a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorExpr(C)->getSourceRange();
408233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
408333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (clang_isStatement(C.kind))
4084a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorStmt(C)->getSourceRange();
4085f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
40866639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  if (clang_isAttribute(C.kind))
40876639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis    return getCursorAttr(C)->getRange();
40886639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis
4089a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_PreprocessingDirective)
4090a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorPreprocessingDirective(C);
40914807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
4092ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_MacroExpansion) {
4093ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
4094ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorMacroExpansion(C)->getSourceRange();
4095ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
4096ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
4097572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
4098ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_MacroDefinition) {
4099ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
4100ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorMacroDefinition(C)->getSourceRange();
4101ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
4102ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
4103ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
4104ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_InclusionDirective) {
4105ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
4106ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorInclusionDirective(C)->getSourceRange();
4107ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
4108ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
4109ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
41100822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis  if (C.kind == CXCursor_TranslationUnit) {
41110822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
41120822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    FileID MainID = TU->getSourceManager().getMainFileID();
41130822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    SourceLocation Start = TU->getSourceManager().getLocForStartOfFile(MainID);
41140822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    SourceLocation End = TU->getSourceManager().getLocForEndOfFile(MainID);
41150822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    return SourceRange(Start, End);
41160822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis  }
41170822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis
4118007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
4119007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    Decl *D = cxcursor::getCursorDecl(C);
412016ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (!D)
412116ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      return SourceRange();
412216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
4123007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    SourceRange R = D->getSourceRange();
4124007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // FIXME: Multiple variables declared in a single declaration
4125007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // currently lack the information needed to correctly determine their
4126007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // ranges when accounting for the type-specifier.  We use context
4127007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
4128007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // and if so, whether it is the first decl.
4129007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
4130007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      if (!cxcursor::isFirstInDeclGroup(C))
4131007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek        R.setBegin(VD->getLocation());
4132007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    }
4133007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    return R;
4134007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
41356653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return SourceRange();
41366653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
41376653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
41386653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// \brief Retrieves the "raw" cursor extent, which is then extended to include
41396653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// the decl-specifier-seq for declarations.
41406653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr) {
41416653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
41426653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    Decl *D = cxcursor::getCursorDecl(C);
414316ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (!D)
414416ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      return SourceRange();
414516ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
41466653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange R = D->getSourceRange();
41472494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
41482494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // Adjust the start of the location for declarations preceded by
41492494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // declaration specifiers.
41502494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
41516653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
41522494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
415396a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
41542494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
41552494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
415696a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
41572494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    }
41586653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
41592494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && R.getBegin().isValid() &&
41602494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, R.getBegin()))
41612494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      R.setBegin(StartLoc);
41622494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
41632494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // FIXME: Multiple variables declared in a single declaration
41642494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // currently lack the information needed to correctly determine their
41652494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // ranges when accounting for the type-specifier.  We use context
41662494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
41672494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // and if so, whether it is the first decl.
41682494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
41692494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (!cxcursor::isFirstInDeclGroup(C))
41702494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        R.setBegin(VD->getLocation());
41716653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    }
41726653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
41736653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    return R;
41746653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  }
41756653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
41766653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return getRawCursorExtent(C);
41776653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
4178a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
4179a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorextern "C" {
4180a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
4181a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas GregorCXSourceRange clang_getCursorExtent(CXCursor C) {
4182a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  SourceRange R = getRawCursorExtent(C);
4183a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R.isInvalid())
41845352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4185f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4186a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  return cxloc::translateSourceRange(getCursorContext(C), R);
4187a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor}
4188c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
4189c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas GregorCXCursor clang_getCursorReferenced(CXCursor C) {
4190b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
4191b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
4192f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4193a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit tu = getCursorTU(C);
41941f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (clang_isDeclaration(C.kind)) {
41951f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getCursorDecl(C);
419616ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (!D)
419716ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      return clang_getNullCursor();
41981f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4199a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Using, D->getLocation(), tu);
42005f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    if (ObjCPropertyImplDecl *PropImpl =dyn_cast<ObjCPropertyImplDecl>(D))
4201e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
4202e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return MakeCXCursor(Property, tu);
4203e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
4204c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return C;
42051f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
42061f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
420797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
42081f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Expr *E = getCursorExpr(C);
42091f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getDeclFromExpr(E);
4210aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (D) {
4211aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      CXCursor declCursor = MakeCXCursor(D, tu);
4212aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      declCursor = getSelectorIdentifierCursor(getSelectorIdentifierIndex(C),
4213aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis                                               declCursor);
4214aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return declCursor;
4215aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    }
42161f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
42171f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (OverloadExpr *Ovl = dyn_cast_or_null<OverloadExpr>(E))
4218a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Ovl, tu);
42191f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
422097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    return clang_getNullCursor();
422197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
422297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
422336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
422436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
422536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (GotoStmt *Goto = dyn_cast_or_null<GotoStmt>(S))
422637c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek      if (LabelDecl *label = Goto->getLabel())
422737c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        if (LabelStmt *labelS = label->getStmt())
422837c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        return MakeCXCursor(labelS, getCursorDecl(C), tu);
422936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
423036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return clang_getNullCursor();
423136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
423236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
42339b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion) {
42349e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth    if (MacroDefinition *Def = getCursorMacroExpansion(C)->getDefinition())
4235a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeMacroDefinitionCursor(Def, tu);
4236bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  }
4237bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
4238c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  if (!clang_isReference(C.kind))
4239c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return clang_getNullCursor();
4240f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4241c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  switch (C.kind) {
4242c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    case CXCursor_ObjCSuperClassRef:
4243a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCSuperClassRef(C).first, tu);
4244f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4245f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
424698c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis      ObjCProtocolDecl *Prot = getCursorObjCProtocolRef(C).first;
424798c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis      if (ObjCProtocolDecl *Def = Prot->getDefinition())
424898c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis        return MakeCXCursor(Def, tu);
424998c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis
4250c15707d8da08df2eb22f6ed047743fa3f7c9831bArgyrios Kyrtzidis      return MakeCXCursor(Prot, tu);
425198c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis    }
4252f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
42537723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor    case CXCursor_ObjCClassRef: {
42547723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
42557723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor      if (ObjCInterfaceDecl *Def = Class->getDefinition())
42567723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor        return MakeCXCursor(Def, tu);
42577723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor
4258c15707d8da08df2eb22f6ed047743fa3f7c9831bArgyrios Kyrtzidis      return MakeCXCursor(Class, tu);
42597723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor    }
42607d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
4261f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef:
4262a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTypeRef(C).first, tu );
42630b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
42640b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
4265a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTemplateRef(C).first, tu );
42660b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
42676931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
4268a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorNamespaceRef(C).first, tu );
42696931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
4270a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
4271a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorMemberRef(C).first, tu );
4272a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
42733064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
42743064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = cxcursor::getCursorCXXBaseSpecifier(C);
42753064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return clang_getTypeDeclaration(cxtype::MakeCXType(B->getType(),
4276a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                                         tu ));
42773064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
4278f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
427936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
428036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // FIXME: We end up faking the "parent" declaration here because we
428136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // don't want to make CXCursor larger.
428236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return MakeCXCursor(getCursorLabelRef(C).first,
4283a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek               static_cast<ASTUnit*>(tu->TUData)->getASTContext()
4284a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          .getTranslationUnitDecl(),
4285a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          tu);
428636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
42871f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
42881f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return C;
4289011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
4290011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef:
4291011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return MakeCXCursor(getCursorVariableRef(C).first, tu);
42921f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4293c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    default:
4294c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      // We would prefer to enumerate all non-reference cursor kinds here.
4295c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      llvm_unreachable("Unhandled reference cursor kind");
4296c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  }
4297c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor}
4298c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
4299b699866820102a69d83d6ac6941985c5ef4e8c40Douglas GregorCXCursor clang_getCursorDefinition(CXCursor C) {
4300b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
4301b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
4302f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4303a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(C);
4304f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4305b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  bool WasReference = false;
430697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
4307b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    C = clang_getCursorReferenced(C);
4308b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    WasReference = true;
4309b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4310b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
43119b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion)
4312bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    return clang_getCursorReferenced(C);
4313bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
4314b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4315b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4316b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4317b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  Decl *D = getCursorDecl(C);
4318b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!D)
4319b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4320f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4321b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  switch (D->getKind()) {
4322b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't really separate the notions of
4323b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // declaration and definition.
4324b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Namespace:
4325b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Typedef:
4326162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  case Decl::TypeAlias:
43273e4c6c4c79a03f5cb0c4671d7c282d623c6dc35eRichard Smith  case Decl::TypeAliasTemplate:
4328b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTypeParm:
4329b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::EnumConstant:
4330b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Field:
4331d98114647e16796a976b04af79975b4f0eacf22bBenjamin Kramer  case Decl::IndirectField:
4332b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCIvar:
4333b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCAtDefsField:
4334b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ImplicitParam:
4335b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ParmVar:
4336b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NonTypeTemplateParm:
4337b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTemplateParm:
4338b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategoryImpl:
4339b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCImplementation:
43406206d53f67613958ae1b023aba337ebb46f11a8bAbramo Bagnara  case Decl::AccessSpec:
4341b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::LinkageSpec:
4342b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCPropertyImpl:
4343b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FileScopeAsm:
4344b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::StaticAssert:
4345b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Block:
4346ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  case Decl::Label:  // FIXME: Is this right??
4347af0f4d0b2e38c810effc8b024ad2fb6604eec5d3Francois Pichet  case Decl::ClassScopeFunctionSpecialization:
434815de72cf580840c61e5704c2f8a2b56f9d0638e1Douglas Gregor  case Decl::Import:
4349b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return C;
4350b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4351b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't make any sense here, but are
4352b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // nonetheless harmless.
4353b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TranslationUnit:
4354b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
4355b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4356b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds for which the definition is not resolvable.
4357b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingTypename:
4358b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingValue:
4359b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
4360b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4361b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingDirective:
4362b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
4363a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                        TU);
4364b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4365b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NamespaceAlias:
4366a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), TU);
4367b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4368b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Enum:
4369b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Record:
4370b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXRecord:
4371b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplateSpecialization:
4372b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplatePartialSpecialization:
4373952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor    if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
4374a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
4375b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4376b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4377b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Function:
4378b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXMethod:
4379b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConstructor:
4380b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXDestructor:
4381b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConversion: {
4382b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
4383b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionDecl>(D)->getBody(Def))
4384a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(const_cast<FunctionDecl *>(Def), TU);
4385b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4386b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4387b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4388b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Var: {
438931310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    // Ask the variable if it has a definition.
439031310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (VarDecl *Def = cast<VarDecl>(D)->getDefinition())
4391a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
439231310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    return clang_getNullCursor();
4393b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4394f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4395b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FunctionTemplate: {
4396b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
4397b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
4398a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def->getDescribedFunctionTemplate(), TU);
4399b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4400b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4401f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4402b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplate: {
4403b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
4404952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor                                                            ->getDefinition())
44050b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return MakeCXCursor(cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
4406a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          TU);
4407b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4408b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4409b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
44101f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::Using:
44111f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<UsingDecl>(D),
4412a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
4413b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4414b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingShadow:
4415b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getCursorDefinition(
4416f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                       MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
4417a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                    TU));
4418b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4419b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCMethod: {
4420b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
4421b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (Method->isThisDeclarationADefinition())
4422b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
4423b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4424b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // Dig out the method definition in the associated
4425b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // @implementation, if we have it.
4426b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: The ASTs should make finding the definition easier.
4427b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4428b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                       = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
4429b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
4430b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
4431b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                                                  Method->isInstanceMethod()))
4432b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          if (Def->isThisDeclarationADefinition())
4433a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek            return MakeCXCursor(Def, TU);
4434b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4435b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4436b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4437b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4438b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategory:
4439b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCCategoryImplDecl *Impl
4440b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                               = cast<ObjCCategoryDecl>(D)->getImplementation())
4441a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4442b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4443b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4444b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProtocol:
44455e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregor    if (ObjCProtocolDecl *Def = cast<ObjCProtocolDecl>(D)->getDefinition())
44465e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregor      return MakeCXCursor(Def, TU);
4447b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4448b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4449375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  case Decl::ObjCInterface: {
4450b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // There are two notions of a "definition" for an Objective-C
4451b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // class: the interface and its implementation. When we resolved a
4452b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // reference to an Objective-C class, produce the @interface as
4453b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // the definition; when we were provided with the interface,
4454b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // produce the @implementation as the definition.
4455375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    ObjCInterfaceDecl *IFace = cast<ObjCInterfaceDecl>(D);
4456b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (WasReference) {
4457375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor      if (ObjCInterfaceDecl *Def = IFace->getDefinition())
44587723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor        return MakeCXCursor(Def, TU);
4459375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    } else if (ObjCImplementationDecl *Impl = IFace->getImplementation())
4460a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4461b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4462375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  }
4463f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4464b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProperty:
4465b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: We don't really know where to find the
4466b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // ObjCPropertyImplDecls that implement this property.
4467b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4468b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4469b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCompatibleAlias:
4470b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4471b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
44727723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor      if (ObjCInterfaceDecl *Def = Class->getDefinition())
44737723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor        return MakeCXCursor(Def, TU);
4474f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4475b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4476b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4477b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Friend:
4478b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
4479a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4480b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4481b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4482b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FriendTemplate:
4483b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
4484a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4485b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4486b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4487b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4488b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getNullCursor();
4489b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4490b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4491b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregorunsigned clang_isCursorDefinition(CXCursor C) {
4492b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4493b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return 0;
4494b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4495b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getCursorDefinition(C) == C;
4496b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4497b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
44981a9d0503b67a499797141af0fd6d315d5045f0eaDouglas GregorCXCursor clang_getCanonicalCursor(CXCursor C) {
44991a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  if (!clang_isDeclaration(C.kind))
45001a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return C;
45011a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
4502e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis  if (Decl *D = getCursorDecl(C)) {
4503debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis    if (ObjCCategoryImplDecl *CatImplD = dyn_cast<ObjCCategoryImplDecl>(D))
4504debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis      if (ObjCCategoryDecl *CatD = CatImplD->getCategoryDecl())
4505debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis        return MakeCXCursor(CatD, getCursorTU(C));
4506debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis
4507e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis    if (ObjCImplDecl *ImplD = dyn_cast<ObjCImplDecl>(D))
4508e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis      if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
4509e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis        return MakeCXCursor(IFD, getCursorTU(C));
4510e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis
45111a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return MakeCXCursor(D->getCanonicalDecl(), getCursorTU(C));
4512e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis  }
45131a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
45141a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  return C;
45151a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor}
451634ebe1e1b0779bcea2f277bc6b4e9dd98bf70b7bArgyrios Kyrtzidis
451734ebe1e1b0779bcea2f277bc6b4e9dd98bf70b7bArgyrios Kyrtzidisint clang_Cursor_getObjCSelectorIndex(CXCursor cursor) {
451834ebe1e1b0779bcea2f277bc6b4e9dd98bf70b7bArgyrios Kyrtzidis  return cxcursor::getSelectorIdentifierIndexAndLoc(cursor).first;
451934ebe1e1b0779bcea2f277bc6b4e9dd98bf70b7bArgyrios Kyrtzidis}
45201a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
45211f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregorunsigned clang_getNumOverloadedDecls(CXCursor C) {
45227c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (C.kind != CXCursor_OverloadedDeclRef)
45231f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return 0;
45241f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45251f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
45261f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
45271f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return E->getNumDecls();
45281f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45291f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
45301f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
45311f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return S->size();
45321f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45331f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
45341f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4535826faa22bae112e01293a58534a40711043cce65Argyrios Kyrtzidis    return Using->shadow_size();
45361f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45371f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return 0;
45381f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
45391f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45401f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas GregorCXCursor clang_getOverloadedDecl(CXCursor cursor, unsigned index) {
45417c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (cursor.kind != CXCursor_OverloadedDeclRef)
45421f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
45431f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45441f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (index >= clang_getNumOverloadedDecls(cursor))
45451f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
45461f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4547a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(cursor);
45481f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(cursor).first;
45491f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
4550a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(E->decls_begin()[index], TU);
45511f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45521f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
45531f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
4554a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(S->begin()[index], TU);
45551f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45561f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
45571f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D)) {
45581f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // FIXME: This is, unfortunately, linear time.
45591f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    UsingDecl::shadow_iterator Pos = Using->shadow_begin();
45601f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    std::advance(Pos, index);
4561a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<UsingShadowDecl>(*Pos)->getTargetDecl(), TU);
45621f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
45631f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45641f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return clang_getNullCursor();
45651f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
45661f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45670d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbarvoid clang_getDefinitionSpellingAndExtent(CXCursor C,
45684ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **startBuf,
45694ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **endBuf,
45704ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startLine,
45714ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startColumn,
45724ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *endLine,
45739ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          unsigned *endColumn) {
4574283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  assert(getCursorDecl(C) && "CXCursor has null decl");
4575283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  NamedDecl *ND = static_cast<NamedDecl *>(getCursorDecl(C));
45764ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  FunctionDecl *FD = dyn_cast<FunctionDecl>(ND);
45774ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
4578f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
45794ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  SourceManager &SM = FD->getASTContext().getSourceManager();
45804ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startBuf = SM.getCharacterData(Body->getLBracLoc());
45814ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endBuf = SM.getCharacterData(Body->getRBracLoc());
45824ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
45834ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
45844ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
45854ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
45864ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff}
4587f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4588430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4589430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas GregorCXSourceRange clang_getCursorReferenceNameRange(CXCursor C, unsigned NameFlags,
4590430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                                                unsigned PieceIndex) {
4591430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  RefNamePieces Pieces;
4592430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4593430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  switch (C.kind) {
4594430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_MemberRefExpr:
4595430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (MemberExpr *E = dyn_cast<MemberExpr>(getCursorExpr(C)))
4596430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Pieces = buildPieces(NameFlags, true, E->getMemberNameInfo(),
4597430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getQualifierLoc().getSourceRange());
4598430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4599430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4600430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_DeclRefExpr:
4601430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (DeclRefExpr *E = dyn_cast<DeclRefExpr>(getCursorExpr(C)))
4602430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Pieces = buildPieces(NameFlags, false, E->getNameInfo(),
4603430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getQualifierLoc().getSourceRange(),
4604e4b92761b43ced611c417ae478568610f1ad7b1eAbramo Bagnara                           E->getOptionalExplicitTemplateArgs());
4605430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4606430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4607430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_CallExpr:
4608430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (CXXOperatorCallExpr *OCE =
4609430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        dyn_cast<CXXOperatorCallExpr>(getCursorExpr(C))) {
4610430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Expr *Callee = OCE->getCallee();
4611430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Callee))
4612430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        Callee = ICE->getSubExpr();
4613430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4614430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Callee))
4615430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        Pieces = buildPieces(NameFlags, false, DRE->getNameInfo(),
4616430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                             DRE->getQualifierLoc().getSourceRange());
4617430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    }
4618430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4619430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4620430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  default:
4621430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4622430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  }
4623430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4624430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  if (Pieces.empty()) {
4625430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (PieceIndex == 0)
4626430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      return clang_getCursorExtent(C);
4627430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  } else if (PieceIndex < Pieces.size()) {
4628430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      SourceRange R = Pieces[PieceIndex];
4629430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (R.isValid())
4630430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        return cxloc::translateSourceRange(getCursorContext(C), R);
4631430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  }
4632430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4633430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  return clang_getNullRange();
4634430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor}
4635430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
46360a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregorvoid clang_enableStackTraces(void) {
46370a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  llvm::sys::PrintStackTraceOnErrorSignal();
46380a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor}
46390a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor
4640995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbarvoid clang_executeOnThread(void (*fn)(void*), void *user_data,
4641995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar                           unsigned stack_size) {
4642995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar  llvm::llvm_execute_on_thread(fn, user_data, stack_size);
4643995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar}
4644995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar
4645fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
4646fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
4647fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
4648fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor// Token-based Operations.
4649fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
4650fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4651fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor/* CXToken layout:
4652fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[0]: a CXTokenKind
4653fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[1]: starting token location
4654fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[2]: token length
4655fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[3]: reserved
4656f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek *   ptr_data: for identifiers and keywords, an IdentifierInfo*.
4657fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   otherwise unused.
4658fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor */
4659fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorextern "C" {
4660fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4661fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXTokenKind clang_getTokenKind(CXToken CXTok) {
4662fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return static_cast<CXTokenKind>(CXTok.int_data[0]);
4663fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4664fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4665fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
4666fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  switch (clang_getTokenKind(CXTok)) {
4667fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Identifier:
4668fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Keyword:
4669fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We know we have an IdentifierInfo*, so use that.
4670ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(static_cast<IdentifierInfo *>(CXTok.ptr_data)
4671ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                            ->getNameStart());
4672fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4673fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Literal: {
4674fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We have stashed the starting pointer in the ptr_data field. Use it.
4675fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    const char *Text = static_cast<const char *>(CXTok.ptr_data);
46765f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    return createCXString(StringRef(Text, CXTok.int_data[2]));
4677fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4678f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4679fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Punctuation:
4680fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Comment:
4681fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    break;
4682fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4683f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4684f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // We have to find the starting buffer pointer the hard way, by
4685fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // deconstructing the source location.
4686a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4687fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4688ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
4689f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4690fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
4691fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> LocInfo
4692a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    = CXXUnit->getSourceManager().getDecomposedSpellingLoc(Loc);
4693f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
46945f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  StringRef Buffer
4695f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
4696f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  if (Invalid)
4697aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor    return createCXString("");
4698fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4699f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  return createCXString(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
4700fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4701f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4702fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
4703a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4704fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4705fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return clang_getNullLocation();
4706f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4707fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
4708fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4709fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4710fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4711fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
4712a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
47135352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (!CXXUnit)
47145352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4715f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4716f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return cxloc::translateSourceRange(CXXUnit->getASTContext(),
4717fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4718fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4719f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4720ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisstatic void getTokens(ASTUnit *CXXUnit, SourceRange Range,
4721ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                      SmallVectorImpl<CXToken> &CXTokens) {
4722fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
4723fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
4724ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(Range.getBegin());
4725fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
4726ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(Range.getEnd());
4727f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4728fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Cannot tokenize across files.
4729fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (BeginLocInfo.first != EndLocInfo.first)
4730fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4731f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4732f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Create a lexer
4733f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
47345f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  StringRef Buffer
4735f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
473647a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor  if (Invalid)
473747a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor    return;
4738aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor
4739fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
47404e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie            CXXUnit->getASTContext().getLangOpts(),
4741f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer            Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
4742fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lex.SetCommentRetentionState(true);
4743f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4744fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Lex tokens until we hit the end of the range.
4745f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
4746fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Token Tok;
4747096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall  bool previousWasAt = false;
4748fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  do {
4749fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Lex the next token
4750fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    Lex.LexFromRawLexer(Tok);
4751fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.is(tok::eof))
4752fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      break;
4753f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4754fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Initialize the CXToken.
4755fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXToken CXTok;
4756f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4757fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Common fields
4758fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
4759fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[2] = Tok.getLength();
4760fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[3] = 0;
4761f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4762fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Kind-specific fields
4763fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.isLiteral()) {
4764fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Literal;
4765fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = (void *)Tok.getLiteralData();
4766c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara    } else if (Tok.is(tok::raw_identifier)) {
4767aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor      // Lookup the identifier to determine whether we have a keyword.
4768fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      IdentifierInfo *II
4769c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok);
4770aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek
4771096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall      if ((II->getObjCKeywordID() != tok::objc_not_keyword) && previousWasAt) {
4772aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek        CXTok.int_data[0] = CXToken_Keyword;
4773aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4774aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      else {
4775c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        CXTok.int_data[0] = Tok.is(tok::identifier)
4776c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          ? CXToken_Identifier
4777c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          : CXToken_Keyword;
4778aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4779fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = II;
4780fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else if (Tok.is(tok::comment)) {
4781fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Comment;
4782fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4783fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else {
4784fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Punctuation;
4785fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4786fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    }
4787fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTokens.push_back(CXTok);
4788096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall    previousWasAt = Tok.is(tok::at);
4789fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  } while (Lex.getBufferLocation() <= EffectiveBufferEnd);
4790ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis}
4791ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4792ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisvoid clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
4793ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                    CXToken **Tokens, unsigned *NumTokens) {
4794ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (Tokens)
4795ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    *Tokens = 0;
4796ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (NumTokens)
4797ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    *NumTokens = 0;
4798ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4799ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4800ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (!CXXUnit || !Tokens || !NumTokens)
4801ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
4802ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4803ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
4804ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4805ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SourceRange R = cxloc::translateCXSourceRange(Range);
4806ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (R.isInvalid())
4807ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
4808ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4809ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SmallVector<CXToken, 32> CXTokens;
4810ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  getTokens(CXXUnit, R, CXTokens);
4811f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4812fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (CXTokens.empty())
4813fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4814f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4815fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
4816fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
4817fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *NumTokens = CXTokens.size();
4818fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
48190045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
48206db610934bedc6896393c1e1099525b35380acd6Ted Kremenekvoid clang_disposeTokens(CXTranslationUnit TU,
48216db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                         CXToken *Tokens, unsigned NumTokens) {
48226db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  free(Tokens);
48236db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
48246db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
48256db610934bedc6896393c1e1099525b35380acd6Ted Kremenek} // end: extern "C"
48266db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
48276db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
48286db610934bedc6896393c1e1099525b35380acd6Ted Kremenek// Token annotation APIs.
48296db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
48306db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
48310045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregortypedef llvm::DenseMap<unsigned, CXCursor> AnnotateTokensData;
4832fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
4833fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXCursor parent,
4834fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXClientData client_data);
4835d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidisstatic bool AnnotateTokensPostChildrenVisitor(CXCursor cursor,
4836d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis                                              CXClientData client_data);
4837d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis
48386db610934bedc6896393c1e1099525b35380acd6Ted Kremeneknamespace {
48396db610934bedc6896393c1e1099525b35380acd6Ted Kremenekclass AnnotateTokensWorker {
48406db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  AnnotateTokensData &Annotated;
484111949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXToken *Tokens;
484211949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXCursor *Cursors;
484311949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  unsigned NumTokens;
4844fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned TokIdx;
48454419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  unsigned PreprocessingTokIdx;
4846fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CursorVisitor AnnotateVis;
4847fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceManager &SrcMgr;
4848f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool HasContextSensitiveKeywords;
4849d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis
4850d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  struct PostChildrenInfo {
4851d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis    CXCursor Cursor;
4852d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis    SourceRange CursorRange;
4853d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis    unsigned BeforeChildrenTokenIdx;
4854d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  };
4855d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  llvm::SmallVector<PostChildrenInfo, 8> PostChildrenInfos;
4856f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4857fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  bool MoreTokens() const { return TokIdx < NumTokens; }
4858fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned NextToken() const { return TokIdx; }
4859fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AdvanceToken() { ++TokIdx; }
4860fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation GetTokenLoc(unsigned tokI) {
4861fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
4862fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
48635f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  bool isFunctionMacroToken(unsigned tokI) const {
4864a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return Tokens[tokI].int_data[3] != 0;
4865a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
48665f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  SourceLocation getFunctionMacroTokenLoc(unsigned tokI) const {
4867a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[3]);
4868a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4869a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4870a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  void annotateAndAdvanceTokens(CXCursor, RangeComparisonResult, SourceRange);
48715f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  void annotateAndAdvanceFunctionMacroTokens(CXCursor, RangeComparisonResult,
48725f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis                                             SourceRange);
4873fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
48746db610934bedc6896393c1e1099525b35380acd6Ted Kremenekpublic:
487511949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  AnnotateTokensWorker(AnnotateTokensData &annotated,
4876fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                       CXToken *tokens, CXCursor *cursors, unsigned numTokens,
4877a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                       CXTranslationUnit tu, SourceRange RegionOfInterest)
487811949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    : Annotated(annotated), Tokens(tokens), Cursors(cursors),
48794419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      NumTokens(numTokens), TokIdx(0), PreprocessingTokIdx(0),
4880a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      AnnotateVis(tu,
4881f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                  AnnotateTokensVisitor, this,
4882f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                  /*VisitPreprocessorLast=*/true,
4883e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                  /*VisitIncludedEntities=*/false,
4884d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis                  RegionOfInterest,
4885d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis                  /*VisitDeclsOnly=*/false,
4886d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis                  AnnotateTokensPostChildrenVisitor),
4887f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      SrcMgr(static_cast<ASTUnit*>(tu->TUData)->getSourceManager()),
4888f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      HasContextSensitiveKeywords(false) { }
488911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4890fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
48916db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
4892d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  bool postVisitChildren(CXCursor cursor);
489303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  void AnnotateTokens();
4894f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4895f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// \brief Determine whether the annotator saw any cursors that have
4896f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// context-sensitive keywords.
4897f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool hasContextSensitiveKeywords() const {
4898f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    return HasContextSensitiveKeywords;
4899f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
4900d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis
4901d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  ~AnnotateTokensWorker() {
4902d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis    assert(PostChildrenInfos.empty());
4903d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  }
49046db610934bedc6896393c1e1099525b35380acd6Ted Kremenek};
49056db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
49060045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
490703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidisvoid AnnotateTokensWorker::AnnotateTokens() {
4908fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Walk the AST within the region of interest, annotating tokens
4909fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // along the way.
491003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  AnnotateVis.visitFileRegion();
4911fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4912fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = 0 ; I < TokIdx ; ++I) {
491311949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
49144419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    if (Pos != Annotated.end() &&
49154419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        (clang_isInvalid(Cursors[I].kind) ||
49164419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor         Pos->second.kind != CXCursor_PreprocessingDirective))
4917fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      Cursors[I] = Pos->second;
4918fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4919fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4920fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Finish up annotating any tokens left.
4921fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (!MoreTokens())
4922fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return;
492311949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4924fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor &C = clang_getNullCursor();
4925fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = TokIdx ; I < NumTokens ; ++I) {
492603ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (I < PreprocessingTokIdx && clang_isPreprocessing(Cursors[I].kind))
492703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      continue;
492803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
4929fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
4930fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = (Pos == Annotated.end()) ? C : Pos->second;
493111949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  }
493211949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek}
493311949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4934a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief It annotates and advances tokens with a cursor until the comparison
4935a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis//// between the cursor location and the source range is the same as
4936a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \arg compResult.
4937a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis///
4938a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// Pass RangeBefore to annotate tokens with a cursor until a range is reached.
4939a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// Pass RangeOverlap to annotate tokens inside a range.
4940a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisvoid AnnotateTokensWorker::annotateAndAdvanceTokens(CXCursor updateC,
4941a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               RangeComparisonResult compResult,
4942a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               SourceRange range) {
4943a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  while (MoreTokens()) {
4944a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    const unsigned I = NextToken();
49455f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis    if (isFunctionMacroToken(I))
49465f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis      return annotateAndAdvanceFunctionMacroTokens(updateC, compResult, range);
4947a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4948a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    SourceLocation TokLoc = GetTokenLoc(I);
4949a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
4950a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      Cursors[I] = updateC;
4951a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      AdvanceToken();
4952a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      continue;
4953a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
4954a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    break;
4955a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4956a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
4957a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4958a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief Special annotation handling for macro argument tokens.
49595f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidisvoid AnnotateTokensWorker::annotateAndAdvanceFunctionMacroTokens(
49605f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis                                               CXCursor updateC,
4961a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               RangeComparisonResult compResult,
4962a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               SourceRange range) {
49635f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  assert(MoreTokens());
49645f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  assert(isFunctionMacroToken(NextToken()) &&
4965a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis         "Should be called only for macro arg tokens");
4966a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4967a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // This works differently than annotateAndAdvanceTokens; because expanded
4968a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // macro arguments can have arbitrary translation-unit source order, we do not
4969a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // advance the token index one by one until a token fails the range test.
4970a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // We only advance once past all of the macro arg tokens if all of them
4971a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // pass the range test. If one of them fails we keep the token index pointing
4972a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // at the start of the macro arg tokens so that the failing token will be
4973a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // annotated by a subsequent annotation try.
4974a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4975a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  bool atLeastOneCompFail = false;
4976a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4977a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned I = NextToken();
49785f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  for (; I < NumTokens && isFunctionMacroToken(I); ++I) {
49795f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis    SourceLocation TokLoc = getFunctionMacroTokenLoc(I);
4980a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (TokLoc.isFileID())
4981a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      continue; // not macro arg token, it's parens or comma.
4982a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
4983a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (clang_isInvalid(clang_getCursorKind(Cursors[I])))
4984a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        Cursors[I] = updateC;
4985a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    } else
4986a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      atLeastOneCompFail = true;
4987a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4988a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4989a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  if (!atLeastOneCompFail)
4990a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    TokIdx = I; // All of the tokens were handled, advance beyond all of them.
4991a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
4992a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
49936db610934bedc6896393c1e1099525b35380acd6Ted Kremenekenum CXChildVisitResult
49944419b675577d7c281a659fab1fec10e1bfbe04c5Douglas GregorAnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
4995fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CXSourceLocation Loc = clang_getCursorLocation(cursor);
49964419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  SourceRange cursorRange = getRawCursorExtent(cursor);
499781d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor  if (cursorRange.isInvalid())
499881d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor    return CXChildVisit_Recurse;
4999f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
5000f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (!HasContextSensitiveKeywords) {
5001f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C properties can have context-sensitive keywords.
5002f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    if (cursor.kind == CXCursor_ObjCPropertyDecl) {
5003f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCPropertyDecl *Property
5004f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(cursor)))
5005f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        HasContextSensitiveKeywords = Property->getPropertyAttributesAsWritten() != 0;
5006f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
5007f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C methods can have context-sensitive keywords.
5008f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_ObjCInstanceMethodDecl ||
5009f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ObjCClassMethodDecl) {
5010f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCMethodDecl *Method
5011f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
5012f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->getObjCDeclQualifier())
5013f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
5014f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        else {
5015f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
5016f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                                           PEnd = Method->param_end();
5017f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor               P != PEnd; ++P) {
5018f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((*P)->getObjCDeclQualifier()) {
5019f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              HasContextSensitiveKeywords = true;
5020f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              break;
5021f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            }
5022f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          }
5023f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
5024f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5025f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
5026f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ methods can have context-sensitive keywords.
5027f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_CXXMethod) {
5028f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (CXXMethodDecl *Method
5029f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(cursor))) {
5030f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->hasAttr<FinalAttr>() || Method->hasAttr<OverrideAttr>())
5031f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
5032f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5033f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
5034f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ classes can have context-sensitive keywords.
5035f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_StructDecl ||
5036f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassDecl ||
5037f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplate ||
5038f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplatePartialSpecialization) {
5039f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Decl *D = getCursorDecl(cursor))
5040f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (D->hasAttr<FinalAttr>())
5041f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
5042f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
5043f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
5044f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
50454419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  if (clang_isPreprocessing(cursor.kind)) {
5046cea731a9cb7de3f473d60e5ea544e25621cebd76Chandler Carruth    // For macro expansions, just note where the beginning of the macro
5047cea731a9cb7de3f473d60e5ea544e25621cebd76Chandler Carruth    // expansion occurs.
50489b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth    if (cursor.kind == CXCursor_MacroExpansion) {
50494419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      Annotated[Loc.int_data] = cursor;
50504419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      return CXChildVisit_Recurse;
50514419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
50524419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
50534419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Items in the preprocessing record are kept separate from items in
50544419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // declarations, so we keep a separate token index.
50554419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    unsigned SavedTokIdx = TokIdx;
50564419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = PreprocessingTokIdx;
50574419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
50584419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Skip tokens up until we catch up to the beginning of the preprocessing
50594419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // entry.
50604419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
50614419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
50624419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
50634419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
50644419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
50654419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
50664419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
50674419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
50684419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
50694419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
50704419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
50714419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
50724419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
50734419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
50744419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Look at all of the tokens within this range.
50754419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
50764419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
50774419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
50784419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
50794419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
5080b219cfc4d75f0a03630b7c4509ef791b7e97b2c8David Blaikie        llvm_unreachable("Infeasible");
50814419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
50824419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
50834419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
50844419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        Cursors[I] = cursor;
50854419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
50864419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
50874419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
50884419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
50894419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
50904419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
50914419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Save the preprocessing token index; restore the non-preprocessing
50924419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // token index.
50934419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    PreprocessingTokIdx = TokIdx;
50944419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = SavedTokIdx;
50950045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor    return CXChildVisit_Recurse;
50960045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  }
5097fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5098fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (cursorRange.isInvalid())
5099fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return CXChildVisit_Continue;
5100a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek
5101fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation L = SourceLocation::getFromRawEncoding(Loc.int_data);
5102fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5103a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  // Adjust the annotated range based specific declarations.
5104a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  const enum CXCursorKind cursorK = clang_getCursorKind(cursor);
5105a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  if (cursorK >= CXCursor_FirstDecl && cursorK <= CXCursor_LastDecl) {
510623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    Decl *D = cxcursor::getCursorDecl(cursor);
51072494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
51082494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
510916ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (const DeclaratorDecl *DD = dyn_cast_or_null<DeclaratorDecl>(D)) {
51102494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
511196a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
511216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    } else if (TypedefDecl *Typedef = dyn_cast_or_null<TypedefDecl>(D)) {
51132494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
511496a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
5115a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek    }
51162494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
51172494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && L.isValid() &&
51182494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, L))
51192494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      cursorRange.setBegin(StartLoc);
5120a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  }
512181d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor
51223f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // If the location of the cursor occurs within a macro instantiation, record
51233f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // the spelling location of the cursor in our annotation map.  We can then
51243f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // paper over the token labelings during a post-processing step to try and
51253f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // get cursor mappings for tokens that are the *arguments* of a macro
51263f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // instantiation.
51273f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  if (L.isMacroID()) {
51283f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    unsigned rawEncoding = SrcMgr.getSpellingLoc(L).getRawEncoding();
51293f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // Only invalidate the old annotation if it isn't part of a preprocessing
51303f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // directive.  Here we assume that the default construction of CXCursor
51313f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // results in CXCursor.kind being an initialized value (i.e., 0).  If
51323f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // this isn't the case, we can fix by doing lookup + insertion.
51334419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
51343f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    CXCursor &oldC = Annotated[rawEncoding];
51353f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    if (!clang_isPreprocessing(oldC.kind))
51363f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek      oldC = cursor;
51373f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  }
51383f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek
5139fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const enum CXCursorKind K = clang_getCursorKind(parent);
5140fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor updateC =
5141d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek    (clang_isInvalid(K) || K == CXCursor_TranslationUnit)
5142d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek     ? clang_getNullCursor() : parent;
5143fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5144a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  annotateAndAdvanceTokens(updateC, RangeBefore, cursorRange);
5145fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
51465517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // Avoid having the cursor of an expression "overwrite" the annotation of the
51475517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // variable declaration that it belongs to.
51485517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // This can happen for C++ constructor expressions whose range generally
51495517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // include the variable declaration, e.g.:
51505517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  //  MyCXXClass foo; // Make sure we don't annotate 'foo' as a CallExpr cursor.
51515517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  if (clang_isExpression(cursorK)) {
51525517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis    Expr *E = getCursorExpr(cursor);
51538ccac3de1335f1cfd7cea56ba1cefcf0b724ce3fArgyrios Kyrtzidis    if (Decl *D = getCursorParentDecl(cursor)) {
51545517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      const unsigned I = NextToken();
51555517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      if (E->getLocStart().isValid() && D->getLocation().isValid() &&
51565517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis          E->getLocStart() == D->getLocation() &&
51575517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis          E->getLocStart() == GetTokenLoc(I)) {
51585517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis        Cursors[I] = updateC;
51595517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis        AdvanceToken();
51605517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      }
51615517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis    }
51625517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  }
51635517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis
5164d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  // Before recursing into the children keep some state that we are going
5165d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  // to use in the AnnotateTokensWorker::postVisitChildren callback to do some
5166d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  // extra work after the child nodes are visited.
5167d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  // Note that we don't call VisitChildren here to avoid traversing statements
5168d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  // code-recursively which can blow the stack.
5169d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis
5170d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  PostChildrenInfo Info;
5171d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  Info.Cursor = cursor;
5172d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  Info.CursorRange = cursorRange;
5173d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  Info.BeforeChildrenTokenIdx = NextToken();
5174d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  PostChildrenInfos.push_back(Info);
5175d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis
5176d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  return CXChildVisit_Recurse;
5177d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis}
5178d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis
5179d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidisbool AnnotateTokensWorker::postVisitChildren(CXCursor cursor) {
5180d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  if (PostChildrenInfos.empty())
5181d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis    return false;
5182d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  const PostChildrenInfo &Info = PostChildrenInfos.back();
5183d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  if (!clang_equalCursors(Info.Cursor, cursor))
5184d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis    return false;
5185d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis
5186d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  const unsigned BeforeChildren = Info.BeforeChildrenTokenIdx;
5187fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned AfterChildren = NextToken();
5188d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  SourceRange cursorRange = Info.CursorRange;
5189fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5190a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // Scan the tokens that are at the end of the cursor, but are not captured
5191a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // but the child cursors.
5192a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  annotateAndAdvanceTokens(cursor, RangeOverlap, cursorRange);
5193d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis
5194fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the beginning of the cursor, but are not
5195fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // capture by the child cursors.
5196fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
5197fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
5198fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      break;
5199d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis
5200fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
5201fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
5202fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5203d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  PostChildrenInfos.pop_back();
5204d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  return false;
52050045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor}
52060045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
52076db610934bedc6896393c1e1099525b35380acd6Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
52086db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXCursor parent,
52096db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXClientData client_data) {
52106db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
52116db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
52126db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
5213d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidisstatic bool AnnotateTokensPostChildrenVisitor(CXCursor cursor,
5214d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis                                              CXClientData client_data) {
5215d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis  return static_cast<AnnotateTokensWorker*>(client_data)->
5216d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis                                                      postVisitChildren(cursor);
5217d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis}
5218d579dd5cc43e657647f38be5dd47c8974cd43108Argyrios Kyrtzidis
52196628a614c504263ae539462f049d523dd07ac1baTed Kremeneknamespace {
5220a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5221a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief Uses the macro expansions in the preprocessing record to find
5222a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// and mark tokens that are macro arguments. This info is used by the
5223a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// AnnotateTokensWorker.
5224a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisclass MarkMacroArgTokensVisitor {
5225a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  SourceManager &SM;
5226a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  CXToken *Tokens;
5227a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned NumTokens;
5228a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned CurIdx;
5229a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5230a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidispublic:
5231a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  MarkMacroArgTokensVisitor(SourceManager &SM,
5232a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                            CXToken *tokens, unsigned numTokens)
5233a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    : SM(SM), Tokens(tokens), NumTokens(numTokens), CurIdx(0) { }
5234a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5235a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  CXChildVisitResult visit(CXCursor cursor, CXCursor parent) {
5236a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (cursor.kind != CXCursor_MacroExpansion)
5237a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Continue;
5238a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5239a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    SourceRange macroRange = getCursorMacroExpansion(cursor)->getSourceRange();
5240a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (macroRange.getBegin() == macroRange.getEnd())
5241a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Continue; // it's not a function macro.
5242a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5243a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    for (; CurIdx < NumTokens; ++CurIdx) {
5244a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (!SM.isBeforeInTranslationUnit(getTokenLoc(CurIdx),
5245a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                        macroRange.getBegin()))
5246a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        break;
5247a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
5248a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5249a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (CurIdx == NumTokens)
5250a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Break;
5251a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5252a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    for (; CurIdx < NumTokens; ++CurIdx) {
5253a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      SourceLocation tokLoc = getTokenLoc(CurIdx);
5254a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (!SM.isBeforeInTranslationUnit(tokLoc, macroRange.getEnd()))
5255a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        break;
5256a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
52575f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis      setFunctionMacroTokenLoc(CurIdx, SM.getMacroArgExpandedLocation(tokLoc));
5258a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
5259a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5260a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (CurIdx == NumTokens)
5261a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Break;
5262a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5263a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return CXChildVisit_Continue;
5264a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5265a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5266a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisprivate:
5267a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  SourceLocation getTokenLoc(unsigned tokI) {
5268a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
5269a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5270a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
52715f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  void setFunctionMacroTokenLoc(unsigned tokI, SourceLocation loc) {
5272a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // The third field is reserved and currently not used. Use it here
5273a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // to mark macro arg expanded tokens with their expanded locations.
5274a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    Tokens[tokI].int_data[3] = loc.getRawEncoding();
5275a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5276a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis};
5277a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5278a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis} // end anonymous namespace
5279a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5280a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisstatic CXChildVisitResult
5281a676379b26edc959193f9f919ba9c6d296a57824Argyrios KyrtzidisMarkMacroArgTokensVisitorDelegate(CXCursor cursor, CXCursor parent,
5282a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                  CXClientData client_data) {
5283a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  return static_cast<MarkMacroArgTokensVisitor*>(client_data)->visit(cursor,
5284a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                                                     parent);
5285a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
5286a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5287a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisnamespace {
52886628a614c504263ae539462f049d523dd07ac1baTed Kremenek  struct clang_annotateTokens_Data {
52896628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXTranslationUnit TU;
52906628a614c504263ae539462f049d523dd07ac1baTed Kremenek    ASTUnit *CXXUnit;
52916628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXToken *Tokens;
52926628a614c504263ae539462f049d523dd07ac1baTed Kremenek    unsigned NumTokens;
52936628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXCursor *Cursors;
52946628a614c504263ae539462f049d523dd07ac1baTed Kremenek  };
5295ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek}
5296ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek
5297ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisstatic void annotatePreprocessorTokens(CXTranslationUnit TU,
5298ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                       SourceRange RegionOfInterest,
5299ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                       AnnotateTokensData &Annotated) {
5300ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
5301ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5302ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SourceManager &SourceMgr = CXXUnit->getSourceManager();
5303ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  std::pair<FileID, unsigned> BeginLocInfo
5304ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(RegionOfInterest.getBegin());
5305ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  std::pair<FileID, unsigned> EndLocInfo
5306ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(RegionOfInterest.getEnd());
5307ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5308ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (BeginLocInfo.first != EndLocInfo.first)
5309ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
5310ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5311ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  StringRef Buffer;
5312ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  bool Invalid = false;
5313ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
5314ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (Buffer.empty() || Invalid)
5315ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
5316ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5317ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
53184e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie            CXXUnit->getASTContext().getLangOpts(),
5319ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            Buffer.begin(), Buffer.data() + BeginLocInfo.second,
5320ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            Buffer.end());
5321ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Lex.SetCommentRetentionState(true);
5322ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5323ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  // Lex tokens in raw mode until we hit the end of the range, to avoid
5324ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  // entering #includes or expanding macros.
5325ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  while (true) {
5326ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    Token Tok;
5327ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    Lex.LexFromRawLexer(Tok);
5328ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5329ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  reprocess:
5330ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
5331ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // We have found a preprocessing directive. Gobble it up so that we
5332ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // don't see it while preprocessing these tokens later, but keep track
5333ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // of all of the token locations inside this preprocessing directive so
5334ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // that we can annotate them appropriately.
5335ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      //
5336ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // FIXME: Some simple tests here could identify macro definitions and
5337ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // #undefs, to provide specific cursor kinds for those.
5338ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      SmallVector<SourceLocation, 32> Locations;
5339ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      do {
5340ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Locations.push_back(Tok.getLocation());
5341ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Lex.LexFromRawLexer(Tok);
5342ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      } while (!Tok.isAtStartOfLine() && !Tok.is(tok::eof));
5343ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5344ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      using namespace cxcursor;
5345ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      CXCursor Cursor
5346ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      = MakePreprocessingDirectiveCursor(SourceRange(Locations.front(),
5347ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                                     Locations.back()),
5348ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                         TU);
5349ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      for (unsigned I = 0, N = Locations.size(); I != N; ++I) {
5350ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Annotated[Locations[I].getRawEncoding()] = Cursor;
5351ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      }
5352ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5353ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      if (Tok.isAtStartOfLine())
5354ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        goto reprocess;
5355ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5356ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      continue;
5357ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    }
5358ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5359ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    if (Tok.is(tok::eof))
5360ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      break;
5361ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
5362ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis}
5363ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
53646628a614c504263ae539462f049d523dd07ac1baTed Kremenek// This gets run a separate thread to avoid stack blowout.
53656628a614c504263ae539462f049d523dd07ac1baTed Kremenekstatic void clang_annotateTokensImpl(void *UserData) {
53666628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXTranslationUnit TU = ((clang_annotateTokens_Data*)UserData)->TU;
53676628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = ((clang_annotateTokens_Data*)UserData)->CXXUnit;
53686628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXToken *Tokens = ((clang_annotateTokens_Data*)UserData)->Tokens;
53696628a614c504263ae539462f049d523dd07ac1baTed Kremenek  const unsigned NumTokens = ((clang_annotateTokens_Data*)UserData)->NumTokens;
53706628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor *Cursors = ((clang_annotateTokens_Data*)UserData)->Cursors;
5371fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5372fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  CIndexer *CXXIdx = (CIndexer*)TU->CIdx;
5373fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
537481b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
5375fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
53760396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Determine the region of interest, which contains all of the tokens.
53770045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  SourceRange RegionOfInterest;
53786628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setBegin(
53796628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU, Tokens[0])));
53806628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setEnd(
53816628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU,
53826628a614c504263ae539462f049d523dd07ac1baTed Kremenek                                                         Tokens[NumTokens-1])));
5383fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
53840396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // A mapping from the source locations found when re-lexing or traversing the
53850396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // region of interest to the corresponding cursors.
53860045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  AnnotateTokensData Annotated;
5387ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5388fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Relex the tokens within the source range to look for preprocessing
53890396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // directives.
5390ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  annotatePreprocessorTokens(TU, RegionOfInterest, Annotated);
53916628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5392a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
5393a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // Search and mark tokens that are macro argument expansions.
5394a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    MarkMacroArgTokensVisitor Visitor(CXXUnit->getSourceManager(),
5395a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                      Tokens, NumTokens);
5396a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    CursorVisitor MacroArgMarker(TU,
5397a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                 MarkMacroArgTokensVisitorDelegate, &Visitor,
5398f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                 /*VisitPreprocessorLast=*/true,
5399e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                                 /*VisitIncludedEntities=*/false,
5400f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                 RegionOfInterest);
5401a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    MacroArgMarker.visitPreprocessedEntitiesInRegion();
5402a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5403a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
54040396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Annotate all of the source locations in the region of interest that map to
5405fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // a specific cursor.
5406fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  AnnotateTokensWorker W(Annotated, Tokens, Cursors, NumTokens,
5407a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                         TU, RegionOfInterest);
54086628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54096c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // FIXME: We use a ridiculous stack size here because the data-recursion
54106c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm uses a large stack frame than the non-data recursive version,
54116c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // and AnnotationTokensWorker currently transforms the data-recursion
54126c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm back into a traditional recursion by explicitly calling
54136c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // VisitChildren().  We will need to remove this explicit recursive call.
54146628a614c504263ae539462f049d523dd07ac1baTed Kremenek  W.AnnotateTokens();
54156628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5416f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // If we ran into any entities that involve context-sensitive keywords,
5417f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // take another pass through the tokens to mark them as such.
5418f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (W.hasContextSensitiveKeywords()) {
5419f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    for (unsigned I = 0; I != NumTokens; ++I) {
5420f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (clang_getTokenKind(Tokens[I]) != CXToken_Identifier)
5421f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5422f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
5423f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCPropertyDecl) {
5424f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
5425f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (ObjCPropertyDecl *Property
54266628a614c504263ae539462f049d523dd07ac1baTed Kremenek            = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(Cursors[I]))) {
5427f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if (Property->getPropertyAttributesAsWritten() != 0 &&
5428f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              llvm::StringSwitch<bool>(II->getName())
54296628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readonly", true)
54306628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("assign", true)
5431f85e193739c953358c865005855253af4f68a497John McCall              .Case("unsafe_unretained", true)
54326628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readwrite", true)
54336628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("retain", true)
54346628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("copy", true)
54356628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("nonatomic", true)
54366628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("atomic", true)
54376628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("getter", true)
54386628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("setter", true)
5439f85e193739c953358c865005855253af4f68a497John McCall              .Case("strong", true)
5440f85e193739c953358c865005855253af4f68a497John McCall              .Case("weak", true)
54416628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Default(false))
5442f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            Tokens[I].int_data[0] = CXToken_Keyword;
5443f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
5444f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5445f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5446f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
5447f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCInstanceMethodDecl ||
5448f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_ObjCClassMethodDecl) {
5449f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
5450f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (llvm::StringSwitch<bool>(II->getName())
54516628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("in", true)
54526628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("out", true)
54536628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("inout", true)
54546628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("oneway", true)
54556628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("bycopy", true)
54566628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("byref", true)
54576628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Default(false))
5458f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Tokens[I].int_data[0] = CXToken_Keyword;
5459f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5460f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
54616639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis
54626639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      if (Cursors[I].kind == CXCursor_CXXFinalAttr ||
54636639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis          Cursors[I].kind == CXCursor_CXXOverrideAttr) {
54646639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis        Tokens[I].int_data[0] = CXToken_Keyword;
54656628a614c504263ae539462f049d523dd07ac1baTed Kremenek        continue;
5466f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5467f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
5468f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
5469fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
54706628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54716628a614c504263ae539462f049d523dd07ac1baTed Kremenekextern "C" {
54726628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54736628a614c504263ae539462f049d523dd07ac1baTed Kremenekvoid clang_annotateTokens(CXTranslationUnit TU,
54746628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXToken *Tokens, unsigned NumTokens,
54756628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXCursor *Cursors) {
54766628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54776628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (NumTokens == 0 || !Tokens || !Cursors)
54786628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
54796628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54806628a614c504263ae539462f049d523dd07ac1baTed Kremenek  // Any token we don't specifically annotate will have a NULL cursor.
54816628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor C = clang_getNullCursor();
54826628a614c504263ae539462f049d523dd07ac1baTed Kremenek  for (unsigned I = 0; I != NumTokens; ++I)
54836628a614c504263ae539462f049d523dd07ac1baTed Kremenek    Cursors[I] = C;
54846628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54856628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
54866628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!CXXUnit)
54876628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
54886628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54896628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
54906628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54916628a614c504263ae539462f049d523dd07ac1baTed Kremenek  clang_annotateTokens_Data data = { TU, CXXUnit, Tokens, NumTokens, Cursors };
54926628a614c504263ae539462f049d523dd07ac1baTed Kremenek  llvm::CrashRecoveryContext CRC;
54936628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!RunSafely(CRC, clang_annotateTokensImpl, &data,
54946628a614c504263ae539462f049d523dd07ac1baTed Kremenek                 GetSafetyThreadStackSize() * 2)) {
54956628a614c504263ae539462f049d523dd07ac1baTed Kremenek    fprintf(stderr, "libclang: crash detected while annotating tokens\n");
54966628a614c504263ae539462f049d523dd07ac1baTed Kremenek  }
54976628a614c504263ae539462f049d523dd07ac1baTed Kremenek}
54986628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5499fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} // end: extern "C"
5500fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
5501fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
550216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek// Operations for querying linkage of a cursor.
550316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
550416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
550516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenekextern "C" {
550616b4259aecaa22b642d35d36fd89965ed700c1e0Ted KremenekCXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
55070396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
55080396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    return CXLinkage_Invalid;
55090396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
551016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  Decl *D = cxcursor::getCursorDecl(cursor);
551116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  if (NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
551216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    switch (ND->getLinkage()) {
551316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case NoLinkage: return CXLinkage_NoLinkage;
551416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case InternalLinkage: return CXLinkage_Internal;
551516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
551616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case ExternalLinkage: return CXLinkage_External;
551716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    };
551816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
551916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  return CXLinkage_Invalid;
552016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek}
552116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek} // end: extern "C"
552216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
552316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
552445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek// Operations for querying language of a cursor.
552545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
552645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
552745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekstatic CXLanguageKind getDeclLanguage(const Decl *D) {
552816ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  if (!D)
552916ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    return CXLanguage_C;
553016ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
553145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  switch (D->getKind()) {
553245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    default:
553345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      break;
553445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ImplicitParam:
553545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCAtDefsField:
553645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategory:
553745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategoryImpl:
553845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCompatibleAlias:
553945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCImplementation:
554045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCInterface:
554145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCIvar:
554245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCMethod:
554345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProperty:
554445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCPropertyImpl:
554545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProtocol:
554645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_ObjC;
554745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConstructor:
554845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConversion:
554945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXDestructor:
555045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXMethod:
555145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXRecord:
555245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplate:
555345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplatePartialSpecialization:
555445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplateSpecialization:
555545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Friend:
555645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FriendTemplate:
555745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FunctionTemplate:
555845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::LinkageSpec:
555945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Namespace:
556045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NamespaceAlias:
556145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NonTypeTemplateParm:
556245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::StaticAssert:
556345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTemplateParm:
556445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTypeParm:
556545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingTypename:
556645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingValue:
556745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Using:
556845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingDirective:
556945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingShadow:
557045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_CPlusPlus;
557145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  }
557245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
557345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_C;
557445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
557545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
557645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekextern "C" {
557758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
557858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregorenum CXAvailabilityKind clang_getCursorAvailability(CXCursor cursor) {
557958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  if (clang_isDeclaration(cursor.kind))
558058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    if (Decl *D = cxcursor::getCursorDecl(cursor)) {
55810a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      if (isa<FunctionDecl>(D) && cast<FunctionDecl>(D)->isDeleted())
558258ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Available;
558358ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
55840a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      switch (D->getAvailability()) {
55850a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Available:
55860a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_NotYetIntroduced:
55870a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_Available;
55880a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
55890a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Deprecated:
559058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Deprecated;
55910a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
55920a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Unavailable:
55930a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_NotAvailable;
55940a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      }
559558ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    }
55960a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
559758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  return CXAvailability_Available;
559858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor}
559958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
5600cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregorstatic CXVersion convertVersion(VersionTuple In) {
5601cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  CXVersion Out = { -1, -1, -1 };
5602cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (In.empty())
5603cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    return Out;
5604cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5605cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  Out.Major = In.getMajor();
5606cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5607cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (llvm::Optional<unsigned> Minor = In.getMinor())
5608cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    Out.Minor = *Minor;
5609cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  else
5610cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    return Out;
5611cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5612cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (llvm::Optional<unsigned> Subminor = In.getSubminor())
5613cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    Out.Subminor = *Subminor;
5614cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5615cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  return Out;
5616cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor}
5617cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5618cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregorint clang_getCursorPlatformAvailability(CXCursor cursor,
5619cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        int *always_deprecated,
5620cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        CXString *deprecated_message,
5621cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        int *always_unavailable,
5622cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        CXString *unavailable_message,
5623cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        CXPlatformAvailability *availability,
5624cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        int availability_size) {
5625cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (always_deprecated)
5626cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    *always_deprecated = 0;
5627cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (deprecated_message)
5628cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    *deprecated_message = cxstring::createCXString("", /*DupString=*/false);
5629cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (always_unavailable)
5630cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    *always_unavailable = 0;
5631cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (unavailable_message)
5632cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    *unavailable_message = cxstring::createCXString("", /*DupString=*/false);
5633cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5634cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
5635cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    return 0;
5636cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5637cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  Decl *D = cxcursor::getCursorDecl(cursor);
5638cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (!D)
5639cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    return 0;
5640cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5641cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  int N = 0;
5642cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  for (Decl::attr_iterator A = D->attr_begin(), AEnd = D->attr_end(); A != AEnd;
5643cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor       ++A) {
5644cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    if (DeprecatedAttr *Deprecated = dyn_cast<DeprecatedAttr>(*A)) {
5645cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      if (always_deprecated)
5646cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        *always_deprecated = 1;
5647cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      if (deprecated_message)
5648cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        *deprecated_message = cxstring::createCXString(Deprecated->getMessage());
5649cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      continue;
5650cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    }
5651cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5652cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    if (UnavailableAttr *Unavailable = dyn_cast<UnavailableAttr>(*A)) {
5653cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      if (always_unavailable)
5654cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        *always_unavailable = 1;
5655cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      if (unavailable_message) {
5656cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        *unavailable_message
5657cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor          = cxstring::createCXString(Unavailable->getMessage());
5658cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      }
5659cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      continue;
5660cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    }
5661cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5662cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    if (AvailabilityAttr *Avail = dyn_cast<AvailabilityAttr>(*A)) {
5663cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      if (N < availability_size) {
5664cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Platform
5665cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor          = cxstring::createCXString(Avail->getPlatform()->getName());
5666cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Introduced = convertVersion(Avail->getIntroduced());
5667cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Deprecated = convertVersion(Avail->getDeprecated());
5668cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Obsoleted = convertVersion(Avail->getObsoleted());
5669cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Unavailable = Avail->getUnavailable();
5670cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Message = cxstring::createCXString(Avail->getMessage());
5671cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      }
5672cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      ++N;
5673cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    }
5674cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  }
5675cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5676cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  return N;
5677cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor}
5678cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5679cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregorvoid clang_disposeCXPlatformAvailability(CXPlatformAvailability *availability) {
5680cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  clang_disposeString(availability->Platform);
5681cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  clang_disposeString(availability->Message);
5682cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor}
5683cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
568445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted KremenekCXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
568545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  if (clang_isDeclaration(cursor.kind))
568645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    return getDeclLanguage(cxcursor::getCursorDecl(cursor));
568745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
568845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_Invalid;
568945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
56903910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
56913910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// \brief If the given cursor is the "templated" declaration
56923910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// descibing a class or function template, return the class or
56933910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// function template.
56943910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregorstatic Decl *maybeGetTemplateCursor(Decl *D) {
56953910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (!D)
56963910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    return 0;
56973910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
56983910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
56993910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (FunctionTemplateDecl *FunTmpl = FD->getDescribedFunctionTemplate())
57003910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return FunTmpl;
57013910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
57023910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D))
57033910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (ClassTemplateDecl *ClassTmpl = RD->getDescribedClassTemplate())
57043910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return ClassTmpl;
57053910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
57063910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  return D;
57073910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor}
57083910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
57092be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorSemanticParent(CXCursor cursor) {
57102be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
57112be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
57122be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getDeclContext();
57133910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
57143910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
57153910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
57163910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
57173910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
57182be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
57192be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
57202be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
57212be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isStatement(cursor.kind) || clang_isExpression(cursor.kind)) {
57222be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor))
5723a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(D, getCursorTU(cursor));
57242be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
57252be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
57262be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
57272be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
57282be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
57292be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorLexicalParent(CXCursor cursor) {
57302be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
57312be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
57322be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getLexicalDeclContext();
57333910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
57343910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
57353910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
57363910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
57373910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
57382be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
57392be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
57402be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
57412be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // FIXME: Note that we can't easily compute the lexical context of a
57422be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // statement or expression, so we return nothing.
57432be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
57442be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
57452be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
5746ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas GregorCXFile clang_getIncludedFile(CXCursor cursor) {
5747ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (cursor.kind != CXCursor_InclusionDirective)
5748ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return 0;
5749ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
5750ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  InclusionDirective *ID = getCursorInclusionDirective(cursor);
5751ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  return (void *)ID->getFile();
5752ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor}
5753aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5754aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri GribenkoCXSourceRange clang_Cursor_getCommentRange(CXCursor C) {
5755aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  if (!clang_isDeclaration(C.kind))
5756aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko    return clang_getNullRange();
5757aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5758aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  const Decl *D = getCursorDecl(C);
5759aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  ASTContext &Context = getCursorContext(C);
5760f50555eedef33fd5a67d369aa0ae8a6f1d201543Dmitri Gribenko  const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
5761aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  if (!RC)
5762aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko    return clang_getNullRange();
5763aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5764aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  return cxloc::translateSourceRange(Context, RC->getSourceRange());
5765aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko}
5766aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5767aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri GribenkoCXString clang_Cursor_getRawCommentText(CXCursor C) {
5768aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  if (!clang_isDeclaration(C.kind))
5769aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko    return createCXString((const char *) NULL);
5770aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5771aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  const Decl *D = getCursorDecl(C);
5772aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  ASTContext &Context = getCursorContext(C);
5773f50555eedef33fd5a67d369aa0ae8a6f1d201543Dmitri Gribenko  const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
5774aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  StringRef RawText = RC ? RC->getRawText(Context.getSourceManager()) :
5775aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko                           StringRef();
5776aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5777aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  // Don't duplicate the string because RawText points directly into source
5778aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  // code.
5779aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  return createCXString(RawText, false);
5780aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko}
5781aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
57822d44d77fed3200e2eff289f55493317e90d3398cDmitri GribenkoCXString clang_Cursor_getBriefCommentText(CXCursor C) {
57832d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko  if (!clang_isDeclaration(C.kind))
57842d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko    return createCXString((const char *) NULL);
57852d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko
57862d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko  const Decl *D = getCursorDecl(C);
57872d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko  const ASTContext &Context = getCursorContext(C);
5788f50555eedef33fd5a67d369aa0ae8a6f1d201543Dmitri Gribenko  const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
57892d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko
57908376f5934a18b950ac7323d8a38ed231623010faDmitri Gribenko  if (RC) {
57912d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko    StringRef BriefText = RC->getBriefText(Context);
57922d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko
57932d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko    // Don't duplicate the string because RawComment ensures that this memory
57942d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko    // will not go away.
57952d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko    return createCXString(BriefText, false);
57962d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko  }
57972d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko
57982d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko  return createCXString((const char *) NULL);
57992d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko}
58009ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
5801ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri GribenkoCXComment clang_Cursor_getParsedComment(CXCursor C) {
5802ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko  if (!clang_isDeclaration(C.kind))
5803e4330a302ac20b41b9800267ebd4b5b01f8553f8Dmitri Gribenko    return cxcomment::createCXComment(NULL, NULL);
5804ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko
5805ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko  const Decl *D = getCursorDecl(C);
5806ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko  const ASTContext &Context = getCursorContext(C);
5807ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko  const comments::FullComment *FC = Context.getCommentForDecl(D);
5808ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko
5809e4330a302ac20b41b9800267ebd4b5b01f8553f8Dmitri Gribenko  return cxcomment::createCXComment(FC, getCursorTU(C));
5810ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko}
5811ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko
5812b619e7877f09f2984d06629fd8653f3f333d8ea2Dmitri Gribenko} // end: extern "C"
5813b619e7877f09f2984d06629fd8653f3f333d8ea2Dmitri Gribenko
58149ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
58159ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek// C++ AST instrospection.
58169ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
58179ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
58189ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekextern "C" {
58199ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekunsigned clang_CXXMethod_isStatic(CXCursor C) {
58209ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek  if (!clang_isDeclaration(C.kind))
58219ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek    return 0;
582249f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor
582349f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  CXXMethodDecl *Method = 0;
582449f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
582549f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
582649f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
582749f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  else
582849f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
582949f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  return (Method && Method->isStatic()) ? 1 : 0;
583040b492a43bac3ed0c465772aa6921d011cfc273fTed Kremenek}
5831b12903e1a4b8d1b611b8c7e4f910665d628e68cdTed Kremenek
5832211924b563aa31421836cee7655be729ad02733fDouglas Gregorunsigned clang_CXXMethod_isVirtual(CXCursor C) {
5833211924b563aa31421836cee7655be729ad02733fDouglas Gregor  if (!clang_isDeclaration(C.kind))
5834211924b563aa31421836cee7655be729ad02733fDouglas Gregor    return 0;
5835211924b563aa31421836cee7655be729ad02733fDouglas Gregor
5836211924b563aa31421836cee7655be729ad02733fDouglas Gregor  CXXMethodDecl *Method = 0;
5837211924b563aa31421836cee7655be729ad02733fDouglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
5838211924b563aa31421836cee7655be729ad02733fDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
5839211924b563aa31421836cee7655be729ad02733fDouglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
5840211924b563aa31421836cee7655be729ad02733fDouglas Gregor  else
5841211924b563aa31421836cee7655be729ad02733fDouglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
5842211924b563aa31421836cee7655be729ad02733fDouglas Gregor  return (Method && Method->isVirtual()) ? 1 : 0;
5843211924b563aa31421836cee7655be729ad02733fDouglas Gregor}
58449ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek} // end: extern "C"
58459ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
584645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
584795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek// Attribute introspection.
584895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
584995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
585095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenekextern "C" {
585195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted KremenekCXType clang_getIBOutletCollectionType(CXCursor C) {
585295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  if (C.kind != CXCursor_IBOutletCollectionAttr)
5853a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return cxtype::MakeCXType(QualType(), cxcursor::getCursorTU(C));
585495f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
585595f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  IBOutletCollectionAttr *A =
585695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek    cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(C));
585795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
585818aa2ff4641847d7f8866e8c5912d4d0ddb858ceArgyrios Kyrtzidis  return cxtype::MakeCXType(A->getInterface(), cxcursor::getCursorTU(C));
585995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek}
586095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek} // end: extern "C"
586195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
586295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
586359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek// Inspecting memory usage.
586459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek//===----------------------------------------------------------------------===//
586559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5866f787002478f09af1741fb0f82a562002e6799c49Ted Kremenektypedef std::vector<CXTUResourceUsageEntry> MemUsageEntries;
586759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5868f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekstatic inline void createCXTUResourceUsageEntry(MemUsageEntries &entries,
5869f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek                                              enum CXTUResourceUsageKind k,
5870ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek                                              unsigned long amount) {
5871f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  CXTUResourceUsageEntry entry = { k, amount };
587259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  entries.push_back(entry);
587359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
587459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
587559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenekextern "C" {
587659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5877f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekconst char *clang_getTUResourceUsageName(CXTUResourceUsageKind kind) {
587859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  const char *str = "";
587959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  switch (kind) {
5880f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_AST:
588159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: expressions, declarations, and types";
588259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      break;
5883f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_Identifiers:
588459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: identifiers";
588559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      break;
5886f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_Selectors:
588759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: selectors";
5888e294ab717fc9535429ca5d8f575d41ae4441d822Ted Kremenek      break;
5889f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_GlobalCompletionResults:
58904e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek      str = "Code completion: cached global results";
5891e294ab717fc9535429ca5d8f575d41ae4441d822Ted Kremenek      break;
5892457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek    case CXTUResourceUsage_SourceManagerContentCache:
5893457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek      str = "SourceManager: content cache allocator";
5894457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek      break;
5895ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    case CXTUResourceUsage_AST_SideTables:
5896ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek      str = "ASTContext: side tables";
5897ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek      break;
5898f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    case CXTUResourceUsage_SourceManager_Membuffer_Malloc:
5899f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      str = "SourceManager: malloc'ed memory buffers";
5900f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      break;
5901f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    case CXTUResourceUsage_SourceManager_Membuffer_MMap:
5902f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      str = "SourceManager: mmap'ed memory buffers";
5903f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      break;
5904e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    case CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc:
5905e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      str = "ExternalASTSource: malloc'ed memory buffers";
5906e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      break;
5907e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    case CXTUResourceUsage_ExternalASTSource_Membuffer_MMap:
5908e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      str = "ExternalASTSource: mmap'ed memory buffers";
5909e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      break;
59105e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    case CXTUResourceUsage_Preprocessor:
59115e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      str = "Preprocessor: malloc'ed memory";
59125e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      break;
59135e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    case CXTUResourceUsage_PreprocessingRecord:
59145e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      str = "Preprocessor: PreprocessingRecord";
59155e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      break;
5916ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek    case CXTUResourceUsage_SourceManager_DataStructures:
5917ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek      str = "SourceManager: data structures and tables";
5918ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek      break;
5919d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek    case CXTUResourceUsage_Preprocessor_HeaderSearch:
5920d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek      str = "Preprocessor: header search tables";
5921d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek      break;
592259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  }
592359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  return str;
592459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
592559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5926f787002478f09af1741fb0f82a562002e6799c49Ted KremenekCXTUResourceUsage clang_getCXTUResourceUsage(CXTranslationUnit TU) {
592759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  if (!TU) {
5928f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    CXTUResourceUsage usage = { (void*) 0, 0, 0 };
592959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    return usage;
593059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  }
593159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
593259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  ASTUnit *astUnit = static_cast<ASTUnit*>(TU->TUData);
59331e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<MemUsageEntries> entries(new MemUsageEntries());
593459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  ASTContext &astContext = astUnit->getASTContext();
593559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
593659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used by AST nodes and types?
5937f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST,
5938ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    (unsigned long) astContext.getASTAllocatedMemory());
593959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
594059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used by identifiers?
5941f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Identifiers,
594259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    (unsigned long) astContext.Idents.getAllocator().getTotalMemory());
594359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
594459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used for selectors?
5945f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Selectors,
594659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    (unsigned long) astContext.Selectors.getTotalMemory());
594759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5948ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek  // How much memory is used by ASTContext's side tables?
5949ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST_SideTables,
5950ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    (unsigned long) astContext.getSideTableAllocatedMemory());
5951ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek
59524e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  // How much memory is used for caching global code completion results?
59534e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  unsigned long completionBytes = 0;
59544e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  if (GlobalCodeCompletionAllocator *completionAllocator =
59554e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek      astUnit->getCachedCompletionAllocator().getPtr()) {
59565e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    completionBytes = completionAllocator->getTotalMemory();
59574e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  }
5958457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  createCXTUResourceUsageEntry(*entries,
5959457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek                               CXTUResourceUsage_GlobalCompletionResults,
5960457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek                               completionBytes);
5961457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek
5962457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  // How much memory is being used by SourceManager's content cache?
5963457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  createCXTUResourceUsageEntry(*entries,
5964457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek          CXTUResourceUsage_SourceManagerContentCache,
5965457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek          (unsigned long) astContext.getSourceManager().getContentCacheSize());
5966f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek
5967f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  // How much memory is being used by the MemoryBuffer's in SourceManager?
5968f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  const SourceManager::MemoryBufferSizes &srcBufs =
5969f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    astUnit->getSourceManager().getMemoryBufferSizes();
5970f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek
5971f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5972f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               CXTUResourceUsage_SourceManager_Membuffer_Malloc,
5973f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               (unsigned long) srcBufs.malloc_bytes);
5974ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5975f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               CXTUResourceUsage_SourceManager_Membuffer_MMap,
5976f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               (unsigned long) srcBufs.mmap_bytes);
5977ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5978ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                               CXTUResourceUsage_SourceManager_DataStructures,
5979ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                               (unsigned long) astContext.getSourceManager()
5980ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                                .getDataStructureSizes());
5981e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek
5982e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  // How much memory is being used by the ExternalASTSource?
5983e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  if (ExternalASTSource *esrc = astContext.getExternalSource()) {
5984e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    const ExternalASTSource::MemoryBufferSizes &sizes =
5985e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      esrc->getMemoryBufferSizes();
5986e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek
5987e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    createCXTUResourceUsageEntry(*entries,
5988e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc,
5989e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek                                 (unsigned long) sizes.malloc_bytes);
5990e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    createCXTUResourceUsageEntry(*entries,
5991e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      CXTUResourceUsage_ExternalASTSource_Membuffer_MMap,
5992e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek                                 (unsigned long) sizes.mmap_bytes);
5993e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  }
59945e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
59955e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  // How much memory is being used by the Preprocessor?
59965e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  Preprocessor &pp = astUnit->getPreprocessor();
59975e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  createCXTUResourceUsageEntry(*entries,
59985e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                               CXTUResourceUsage_Preprocessor,
5999c5c5e92ec53f7e6ac7ebbbf77c6d8e4b7d88daecArgyrios Kyrtzidis                               pp.getTotalMemory());
60005e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
60015e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  if (PreprocessingRecord *pRec = pp.getPreprocessingRecord()) {
60025e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    createCXTUResourceUsageEntry(*entries,
60035e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                                 CXTUResourceUsage_PreprocessingRecord,
60045e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                                 pRec->getTotalMemory());
60055e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  }
60065e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
6007d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek  createCXTUResourceUsageEntry(*entries,
6008d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek                               CXTUResourceUsage_Preprocessor_HeaderSearch,
6009d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek                               pp.getHeaderSearchInfo().getTotalMemory());
60105e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
6011f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  CXTUResourceUsage usage = { (void*) entries.get(),
601259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek                            (unsigned) entries->size(),
601359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek                            entries->size() ? &(*entries)[0] : 0 };
601459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  entries.take();
601559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  return usage;
601659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
601759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
6018f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekvoid clang_disposeCXTUResourceUsage(CXTUResourceUsage usage) {
601959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  if (usage.data)
602059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    delete (MemUsageEntries*) usage.data;
602159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
602259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
602359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek} // end extern "C"
602459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
60256df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregorvoid clang::PrintLibclangResourceUsage(CXTranslationUnit TU) {
60266df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  CXTUResourceUsage Usage = clang_getCXTUResourceUsage(TU);
60276df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  for (unsigned I = 0; I != Usage.numEntries; ++I)
60286df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    fprintf(stderr, "  %s: %lu\n",
60296df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor            clang_getTUResourceUsageName(Usage.entries[I].kind),
60306df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor            Usage.entries[I].amount);
60316df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
60326df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  clang_disposeCXTUResourceUsage(Usage);
60336df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor}
60346df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
603559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek//===----------------------------------------------------------------------===//
603604bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek// Misc. utility functions.
603704bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek//===----------------------------------------------------------------------===//
6038f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6039abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbar/// Default to using an 8 MB stack size on "safety" threads.
6040abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbarstatic unsigned SafetyStackThreadSize = 8 << 20;
6041bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
6042bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarnamespace clang {
6043bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
6044bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarbool RunSafely(llvm::CrashRecoveryContext &CRC,
60456c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               void (*Fn)(void*), void *UserData,
60466c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               unsigned Size) {
60476c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (!Size)
60486c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek    Size = GetSafetyThreadStackSize();
60496c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (Size)
6050bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar    return CRC.RunSafelyOnThread(Fn, UserData, Size);
6051bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return CRC.RunSafely(Fn, UserData);
6052bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
6053bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
6054bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarunsigned GetSafetyThreadStackSize() {
6055bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return SafetyStackThreadSize;
6056bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
6057bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
6058bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarvoid SetSafetyThreadStackSize(unsigned Value) {
6059bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  SafetyStackThreadSize = Value;
6060bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
6061bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
60628e7c48a54b4325925f5edda1738a3537ab2c3c5eArgyrios Kyrtzidis}
60638e7c48a54b4325925f5edda1738a3537ab2c3c5eArgyrios Kyrtzidis
606481b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidisvoid clang::setThreadBackgroundPriority() {
6065fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  // FIXME: Move to llvm/Support and make it cross-platform.
6066fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis#ifdef __APPLE__
6067fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  setpriority(PRIO_DARWIN_THREAD, 0, PRIO_DARWIN_BG);
6068fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis#endif
6069fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis}
6070fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
60719793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidisvoid cxindex::printDiagsToStderr(ASTUnit *Unit) {
60729793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  if (!Unit)
60739793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis    return;
60749793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis
60759793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
60769793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis                                  DEnd = Unit->stored_diag_end();
60779793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis       D != DEnd; ++D) {
60789793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis    CXStoredDiagnostic Diag(*D, Unit->getASTContext().getLangOpts());
60799793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis    CXString Msg = clang_formatDiagnostic(&Diag,
60809793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis                                clang_defaultDiagnosticDisplayOptions());
60819793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis    fprintf(stderr, "%s\n", clang_getCString(Msg));
60829793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis    clang_disposeString(Msg);
60839793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  }
60849793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis#ifdef LLVM_ON_WIN32
60859793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  // On Windows, force a flush, since there may be multiple copies of
60869793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  // stderr and stdout in the file system, all with different buffers
60879793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  // but writing to the same device.
60889793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  fflush(stderr);
60899793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis#endif
60909793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis}
60919793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis
609204bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenekextern "C" {
609304bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
6094a2a9d6e4e5b6001b86b7dfc5db1ea296ce29a3d3Ted KremenekCXString clang_getClangVersion() {
6095ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(getClangFullVersion());
609604bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek}
609704bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
609804bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek} // end: extern "C"
609959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
6100