CIndex.cpp revision df5faf5e7ae6823d0af0b801c4ac26d47f2cee97
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
185b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Recurse:
186b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return VisitChildren(Cursor);
187b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
1880d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
1897530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid CXChildVisitResult!");
190b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
1910d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
192f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidisstatic bool visitPreprocessedEntitiesInRange(SourceRange R,
193f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                             PreprocessingRecord &PPRec,
194f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                             CursorVisitor &Visitor) {
195f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  SourceManager &SM = Visitor.getASTUnit()->getSourceManager();
196f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  FileID FID;
197f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
198e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis  if (!Visitor.shouldVisitIncludedEntities()) {
199f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    // If the begin/end of the range lie in the same FileID, do the optimization
200f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    // where we skip preprocessed entities that do not come from the same FileID.
201acca41167ce78bb032906f6b1d2ced62efbe059aArgyrios Kyrtzidis    FID = SM.getFileID(SM.getFileLoc(R.getBegin()));
202acca41167ce78bb032906f6b1d2ced62efbe059aArgyrios Kyrtzidis    if (FID != SM.getFileID(SM.getFileLoc(R.getEnd())))
203f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      FID = FileID();
204f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  }
205f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
206f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  std::pair<PreprocessingRecord::iterator, PreprocessingRecord::iterator>
207f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    Entities = PPRec.getPreprocessedEntitiesInRange(R);
208f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  return Visitor.visitPreprocessedEntities(Entities.first, Entities.second,
209f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                           PPRec, FID);
210f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis}
211f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
212dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidisvoid CursorVisitor::visitFileRegion() {
213dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (RegionOfInterest.isInvalid())
214dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
215dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
216dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  ASTUnit *Unit = static_cast<ASTUnit *>(TU->TUData);
217dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceManager &SM = Unit->getSourceManager();
218dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
219dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  std::pair<FileID, unsigned>
220dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Begin = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getBegin())),
221dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getEnd()));
222dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
223dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (End.first != Begin.first) {
224dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    // If the end does not reside in the same file, try to recover by
225dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    // picking the end of the file of begin location.
226dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End.first = Begin.first;
227dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End.second = SM.getFileIDSize(Begin.first);
228dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
229dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
230dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  assert(Begin.first == End.first);
231dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (Begin.second > End.second)
232dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
233dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
234dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  FileID File = Begin.first;
235dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  unsigned Offset = Begin.second;
236dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  unsigned Length = End.second - Begin.second;
237dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
238b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis  if (!VisitDeclsOnly && !VisitPreprocessorLast)
239b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis    if (visitPreprocessedEntitiesInRegion())
240b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis      return; // visitation break.
241dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
242dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  visitDeclsFromFileRegion(File, Offset, Length);
243dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
244b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis  if (!VisitDeclsOnly && VisitPreprocessorLast)
245dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    visitPreprocessedEntitiesInRegion();
246dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis}
247dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
248e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidisstatic bool isInLexicalContext(Decl *D, DeclContext *DC) {
249e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  if (!DC)
250e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    return false;
251e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
252e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  for (DeclContext *DeclDC = D->getLexicalDeclContext();
253e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis         DeclDC; DeclDC = DeclDC->getLexicalParent()) {
254e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    if (DeclDC == DC)
255e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis      return true;
256e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  }
257e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  return false;
258e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis}
259e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
260dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidisvoid CursorVisitor::visitDeclsFromFileRegion(FileID File,
261dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis                                             unsigned Offset, unsigned Length) {
262dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  ASTUnit *Unit = static_cast<ASTUnit *>(TU->TUData);
263dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceManager &SM = Unit->getSourceManager();
264dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceRange Range = RegionOfInterest;
265dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
266dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SmallVector<Decl *, 16> Decls;
267dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  Unit->findFileRegionDecls(File, Offset, Length, Decls);
268dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
269dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // If we didn't find any file level decls for the file, try looking at the
270dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // file that it was included from.
271c14a03dffff69b5e1c55cc118fc52d8fd9f3a28dArgyrios Kyrtzidis  while (Decls.empty() || Decls.front()->isTopLevelDeclInObjCContainer()) {
272dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    bool Invalid = false;
273dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    const SrcMgr::SLocEntry &SLEntry = SM.getSLocEntry(File, &Invalid);
274dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (Invalid)
275dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      return;
276dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
277dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    SourceLocation Outer;
278dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (SLEntry.isFile())
279dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      Outer = SLEntry.getFile().getIncludeLoc();
280dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    else
281dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      Outer = SLEntry.getExpansion().getExpansionLocStart();
282dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (Outer.isInvalid())
283dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      return;
284dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
285dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    llvm::tie(File, Offset) = SM.getDecomposedExpansionLoc(Outer);
286dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Length = 0;
287dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Unit->findFileRegionDecls(File, Offset, Length, Decls);
288dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
289dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
290dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  assert(!Decls.empty());
291dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
292dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  bool VisitedAtLeastOnce = false;
293e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  DeclContext *CurDC = 0;
294dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SmallVector<Decl *, 16>::iterator DIt = Decls.begin();
295dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  for (SmallVector<Decl *, 16>::iterator DE = Decls.end(); DIt != DE; ++DIt) {
296dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Decl *D = *DIt;
297ed8bef44c0545fd55a78715606f8d733f6498b21Argyrios Kyrtzidis    if (D->getSourceRange().isInvalid())
298ed8bef44c0545fd55a78715606f8d733f6498b21Argyrios Kyrtzidis      continue;
299dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
300e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    if (isInLexicalContext(D, CurDC))
301e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis      continue;
302e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
303e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    CurDC = dyn_cast<DeclContext>(D);
304e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
305e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    if (TagDecl *TD = dyn_cast<TagDecl>(D))
306e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis      if (!TD->isFreeStanding())
307e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis        continue;
308e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
309dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    RangeComparisonResult CompRes = RangeCompare(SM, D->getSourceRange(),Range);
310dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CompRes == RangeBefore)
311dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      continue;
312dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CompRes == RangeAfter)
313dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
314dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
315dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    assert(CompRes == RangeOverlap);
316dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    VisitedAtLeastOnce = true;
31703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
31803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (isa<ObjCContainerDecl>(D)) {
31903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      FileDI_current = &DIt;
32003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      FileDE_current = DE;
32103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    } else {
32203ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      FileDI_current = 0;
32303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    }
32403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
325ba98617b994864b7554ff75445983ad02a962f45Argyrios Kyrtzidis    if (Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true))
326dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
327dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
328dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
329dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (VisitedAtLeastOnce)
330dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
331dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
332dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // No Decls overlapped with the range. Move up the lexical context until there
333dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // is a context that contains the range or we reach the translation unit
334dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // level.
335dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  DeclContext *DC = DIt == Decls.begin() ? (*DIt)->getLexicalDeclContext()
336dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis                                         : (*(DIt-1))->getLexicalDeclContext();
337dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
338dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  while (DC && !DC->isTranslationUnit()) {
339dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Decl *D = cast<Decl>(DC);
340dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    SourceRange CurDeclRange = D->getSourceRange();
341dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CurDeclRange.isInvalid())
342dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
343dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
344dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (RangeCompare(SM, CurDeclRange, Range) == RangeOverlap) {
345ba98617b994864b7554ff75445983ad02a962f45Argyrios Kyrtzidis      Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true);
346dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
347dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    }
348dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
349dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    DC = D->getLexicalDeclContext();
350dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
351dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis}
352dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
3534c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregorbool CursorVisitor::visitPreprocessedEntitiesInRegion() {
354b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis  if (!AU->getPreprocessor().getPreprocessingRecord())
355b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis    return false;
356b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis
357788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  PreprocessingRecord &PPRec
358a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    = *AU->getPreprocessor().getPreprocessingRecord();
359f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  SourceManager &SM = AU->getSourceManager();
360788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
36192ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  if (RegionOfInterest.isValid()) {
362ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange MappedRange = AU->mapRangeToPreamble(RegionOfInterest);
363f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    SourceLocation B = MappedRange.getBegin();
364f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    SourceLocation E = MappedRange.getEnd();
365f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
366f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (AU->isInPreambleFileID(B)) {
367f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      if (SM.isLoadedSourceLocation(E))
368f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis        return visitPreprocessedEntitiesInRange(SourceRange(B, E),
369f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                                 PPRec, *this);
370f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
371f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // Beginning of range lies in the preamble but it also extends beyond
372f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // it into the main file. Split the range into 2 parts, one covering
373f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // the preamble and another covering the main file. This allows subsequent
374f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // calls to visitPreprocessedEntitiesInRange to accept a source range that
375f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // lies in the same FileID, allowing it to skip preprocessed entities that
376f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // do not come from the same FileID.
377f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      bool breaked =
378f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis        visitPreprocessedEntitiesInRange(
379f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                   SourceRange(B, AU->getEndOfPreambleFileID()),
380f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                          PPRec, *this);
381f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      if (breaked) return true;
382f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      return visitPreprocessedEntitiesInRange(
383f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                    SourceRange(AU->getStartOfMainFileID(), E),
384f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                        PPRec, *this);
385f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    }
386f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
387f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    return visitPreprocessedEntitiesInRange(SourceRange(B, E), PPRec, *this);
38892ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  }
38992ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis
390788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  bool OnlyLocalDecls
39132038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    = !AU->isMainFileAST() && AU->getOnlyLocalDecls();
39232038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor
39392ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  if (OnlyLocalDecls)
394f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    return visitPreprocessedEntities(PPRec.local_begin(), PPRec.local_end(),
395f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                     PPRec);
3964c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
397f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  return visitPreprocessedEntities(PPRec.begin(), PPRec.end(), PPRec);
3984c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor}
3994c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4004c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregortemplate<typename InputIterator>
4014c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregorbool CursorVisitor::visitPreprocessedEntities(InputIterator First,
402f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              InputIterator Last,
403f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              PreprocessingRecord &PPRec,
404f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              FileID FID) {
4054c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  for (; First != Last; ++First) {
406f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (!FID.isInvalid() && !PPRec.isEntityInFileID(First, FID))
407f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      continue;
408f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
409f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    PreprocessedEntity *PPE = *First;
410f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (MacroExpansion *ME = dyn_cast<MacroExpansion>(PPE)) {
4114c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeMacroExpansionCursor(ME, TU)))
4124c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
4134c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4144c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
4154c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    }
4164c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
417f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (MacroDefinition *MD = dyn_cast<MacroDefinition>(PPE)) {
4184c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeMacroDefinitionCursor(MD, TU)))
4194c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
42089d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
4214c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
4224c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    }
4234c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
424f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (InclusionDirective *ID = dyn_cast<InclusionDirective>(PPE)) {
4254c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeInclusionDirectiveCursor(ID, TU)))
4264c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
4274c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4284c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
429788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    }
430788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  }
431788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
4324c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  return false;
433788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor}
434788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
435b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the children of the given cursor.
436a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek///
437b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
438b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
439f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekbool CursorVisitor::VisitChildren(CXCursor Cursor) {
440c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (clang_isReference(Cursor.kind) &&
441c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      Cursor.kind != CXCursor_CXXBaseSpecifier) {
442a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    // By definition, references have no children.
443a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return false;
444a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  }
445f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
446f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Set the Parent field to Cursor, then back to its old value once we're
447b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // done.
4480f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  SetParentRAII SetParent(Parent, StmtParent, Cursor);
449f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
450b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
451b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
45206d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (!D)
45306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return false;
45406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
455539311e0221df256c70c1c3080c8af847cd29dffTed Kremenek    return VisitAttributes(D) || Visit(D);
456b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
457f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
45806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isStatement(Cursor.kind)) {
45906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Stmt *S = getCursorStmt(Cursor))
46006d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(S);
46106d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
46206d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
46306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
46406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
46506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isExpression(Cursor.kind)) {
46606d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Expr *E = getCursorExpr(Cursor))
46706d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(E);
46806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
46906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
47006d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
471f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
472b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isTranslationUnit(Cursor.kind)) {
473a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CXTranslationUnit tu = getCursorTU(Cursor);
474a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    ASTUnit *CXXUnit = static_cast<ASTUnit*>(tu->TUData);
47504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
47604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    int VisitOrder[2] = { VisitPreprocessorLast, !VisitPreprocessorLast };
47704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    for (unsigned I = 0; I != 2; ++I) {
47804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      if (VisitOrder[I]) {
47904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
48004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            RegionOfInterest.isInvalid()) {
48104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          for (ASTUnit::top_level_iterator TL = CXXUnit->top_level_begin(),
48204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                        TLEnd = CXXUnit->top_level_end();
48304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor               TL != TLEnd; ++TL) {
484aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis            if (Visit(MakeCXCursor(*TL, tu, RegionOfInterest), true))
48504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
48604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
48704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        } else if (VisitDeclContext(
48804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                CXXUnit->getASTContext().getTranslationUnitDecl()))
4897b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor          return true;
49004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        continue;
4917b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor      }
4923178cb674ac8c3b59e1791e14d38d48619a1b621Bob Wilson
49304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      // Walk the preprocessing record.
4944c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (CXXUnit->getPreprocessor().getPreprocessingRecord())
4954c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        visitPreprocessedEntitiesInRegion();
4960396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    }
49704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
4987b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor    return false;
499b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
500f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
501c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (Cursor.kind == CXCursor_CXXBaseSpecifier) {
502c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    if (CXXBaseSpecifier *Base = getCursorCXXBaseSpecifier(Cursor)) {
503c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      if (TypeSourceInfo *BaseTSInfo = Base->getTypeSourceInfo()) {
504c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor        return Visit(BaseTSInfo->getTypeLoc());
505c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      }
506c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    }
507c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  }
508221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis
509221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis  if (Cursor.kind == CXCursor_IBOutletCollectionAttr) {
510221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis    IBOutletCollectionAttr *A =
511221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis      cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(Cursor));
512221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis    if (const ObjCInterfaceType *InterT = A->getInterface()->getAs<ObjCInterfaceType>())
513221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis      return Visit(cxcursor::MakeCursorObjCClassRef(InterT->getInterface(),
514221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis                                                    A->getInterfaceLoc(), TU));
515221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis  }
516221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis
517b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // Nothing to visit at the moment.
518b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
519dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
520dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5211ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenekbool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
52213c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor  if (TypeSourceInfo *TSInfo = B->getSignatureAsWritten())
52313c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor    if (Visit(TSInfo->getTypeLoc()))
52413c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor        return true;
5251ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
526664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  if (Stmt *Body = B->getBody())
527aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Body, StmtParent, TU, RegionOfInterest));
528664cffd330611d78fc0286f539589920a37ca328Ted Kremenek
529664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  return false;
5301ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek}
5311ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
532d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekllvm::Optional<bool> CursorVisitor::shouldVisitCursor(CXCursor Cursor) {
533d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (RegionOfInterest.isValid()) {
5346653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange Range = getFullCursorExtent(Cursor, AU->getSourceManager());
535d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Range.isInvalid())
536d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
5376653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
538d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    switch (CompareRegionOfInterest(Range)) {
539d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeBefore:
540d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes before the region of interest; skip it.
541d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
54223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
543d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeAfter:
544d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes after the region of interest; we're done.
545d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
546d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
547d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeOverlap:
548d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration overlaps the region of interest; visit it.
549d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
550d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
551d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
552d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return true;
553d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
554f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
555d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekbool CursorVisitor::VisitDeclContext(DeclContext *DC) {
556d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator I = DC->decls_begin(), E = DC->decls_end();
557f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
558d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually remove.  This part of a hack to support proper
559d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // iteration over all Decls contained lexically within an ObjC container.
560d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator*> DI_saved(DI_current, &I);
561d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator> DE_saved(DE_current, E);
562f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
563d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for ( ; I != E; ++I) {
564d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *D = *I;
565d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (D->getLexicalDeclContext() != DC)
566d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
567aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    CXCursor Cursor = MakeCXCursor(D, TU, RegionOfInterest);
5681836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis
5691836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    // FIXME: ObjCClassRef/ObjCProtocolRef for forward class/protocol
5701836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    // declarations is a mismatch with the compiler semantics.
5711836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    if (Cursor.kind == CXCursor_ObjCInterfaceDecl) {
5721836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      ObjCInterfaceDecl *ID = cast<ObjCInterfaceDecl>(D);
5731836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      if (!ID->isThisDeclarationADefinition())
5741836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis        Cursor = MakeCursorObjCClassRef(ID, ID->getLocation(), TU);
5751836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis
5761836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    } else if (Cursor.kind == CXCursor_ObjCProtocolDecl) {
5771836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      ObjCProtocolDecl *PD = cast<ObjCProtocolDecl>(D);
5781836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      if (!PD->isThisDeclarationADefinition())
5791836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis        Cursor = MakeCursorObjCProtocolRef(PD, PD->getLocation(), TU);
5801836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    }
5811836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis
582d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
583d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
584d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
585d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
586d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
587d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar    if (Visit(Cursor, true))
588b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
589b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
590b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
591dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
592dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
5941ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  llvm_unreachable("Translation units are visited directly by Visit()");
5951ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
5961ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
597162e1c1b487352434552147967c3dd296ebee2f7Richard Smithbool CursorVisitor::VisitTypeAliasDecl(TypeAliasDecl *D) {
598162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
599162e1c1b487352434552147967c3dd296ebee2f7Richard Smith    return Visit(TSInfo->getTypeLoc());
600162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
601162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return false;
602162e1c1b487352434552147967c3dd296ebee2f7Richard Smith}
603162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
6041ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
6051ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
6061ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(TSInfo->getTypeLoc());
607f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6081ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6091ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6101ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6111ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTagDecl(TagDecl *D) {
6121ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitDeclContext(D);
6131ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6141ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6150ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregorbool CursorVisitor::VisitClassTemplateSpecializationDecl(
6160ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor                                          ClassTemplateSpecializationDecl *D) {
6170ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool ShouldVisitBody = false;
6180ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  switch (D->getSpecializationKind()) {
6190ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_Undeclared:
6200ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ImplicitInstantiation:
6210ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    // Nothing to visit
6220ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return false;
6230ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6240ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDeclaration:
6250ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDefinition:
6260ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6270ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6280ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitSpecialization:
6290ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    ShouldVisitBody = true;
6300ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6310ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6320ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6330ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  // Visit the template arguments used in the specialization.
6340ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (TypeSourceInfo *SpecType = D->getTypeAsWritten()) {
6350ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    TypeLoc TL = SpecType->getTypeLoc();
6360ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    if (TemplateSpecializationTypeLoc *TSTLoc
6370ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          = dyn_cast<TemplateSpecializationTypeLoc>(&TL)) {
6380ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor      for (unsigned I = 0, N = TSTLoc->getNumArgs(); I != N; ++I)
6390ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor        if (VisitTemplateArgumentLoc(TSTLoc->getArgLoc(I)))
6400ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          return true;
6410ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    }
6420ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6430ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6440ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (ShouldVisitBody && VisitCXXRecordDecl(D))
6450ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return true;
6460ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6470ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  return false;
6480ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor}
6490ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
65074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregorbool CursorVisitor::VisitClassTemplatePartialSpecializationDecl(
65174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                   ClassTemplatePartialSpecializationDecl *D) {
65274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
65374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // before visiting these template parameters.
65474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
65574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return true;
65674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
65774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // Visit the partial specialization arguments.
65874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  const TemplateArgumentLoc *TemplateArgs = D->getTemplateArgsAsWritten();
65974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  for (unsigned I = 0, N = D->getNumTemplateArgsAsWritten(); I != N; ++I)
66074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    if (VisitTemplateArgumentLoc(TemplateArgs[I]))
66174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor      return true;
66274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
66374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  return VisitCXXRecordDecl(D);
66474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor}
66574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
666fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
66784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  // Visit the default argument.
66884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
66984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (TypeSourceInfo *DefArg = D->getDefaultArgumentInfo())
67084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      if (Visit(DefArg->getTypeLoc()))
67184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor        return true;
67284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
673fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
674fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
675fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
6761ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
6771ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInitExpr())
678aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
6791ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6801ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6811ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6827d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
6837d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
6847d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
6857d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      return true;
6867d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
687c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
688c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = DD->getQualifierLoc())
689c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
690c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
691c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
6927d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return false;
6937d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
6947d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
695a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor/// \brief Compare two base or member initializers based on their source order.
696cbb67480094b3bcb5b715acd827cbad55e2a204cSean Huntstatic int CompareCXXCtorInitializers(const void* Xp, const void *Yp) {
697cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *X
698cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Xp);
699cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *Y
700cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Yp);
701a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
702a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  if ((*X)->getSourceOrder() < (*Y)->getSourceOrder())
703a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return -1;
704a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else if ((*X)->getSourceOrder() > (*Y)->getSourceOrder())
705a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 1;
706a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else
707a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 0;
708a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor}
709a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
710b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
71101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (TypeSourceInfo *TSInfo = ND->getTypeSourceInfo()) {
71201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function declaration's syntactic components in the order
71301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // written. This requires a bit of work.
714723df245307a530da5433dfb43accf187dc3e243Abramo Bagnara    TypeLoc TL = TSInfo->getTypeLoc().IgnoreParens();
71501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    FunctionTypeLoc *FTL = dyn_cast<FunctionTypeLoc>(&TL);
71601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
71701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // If we have a function declared directly (without the use of a typedef),
71801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // visit just the return type. Otherwise, just visit the function's type
71901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // now.
72001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if ((FTL && !isa<CXXConversionDecl>(ND) && Visit(FTL->getResultLoc())) ||
72101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor        (!FTL && Visit(TL)))
72201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
72301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
724c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // Visit the nested-name-specifier, if present.
725c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (NestedNameSpecifierLoc QualifierLoc = ND->getQualifierLoc())
726c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      if (VisitNestedNameSpecifierLoc(QualifierLoc))
727c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor        return true;
72801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
72901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the declaration name.
73001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (VisitDeclarationNameInfo(ND->getNameInfo()))
73101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
73201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
73301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Visit explicitly-specified template arguments!
73401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
73501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function parameters, if we have a function type.
73601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (FTL && VisitFunctionTypeLoc(*FTL, true))
73701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
73801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
73901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Attributes?
74001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
74101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
74210620eb5164e31208fcbf0437cd79ae535ed0559Sean Hunt  if (ND->doesThisDeclarationHaveABody() && !ND->isLateTemplateParsed()) {
743a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    if (CXXConstructorDecl *Constructor = dyn_cast<CXXConstructorDecl>(ND)) {
744a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Find the initializers that were written in the source.
7455f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      SmallVector<CXXCtorInitializer *, 4> WrittenInits;
746a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (CXXConstructorDecl::init_iterator I = Constructor->init_begin(),
747a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                          IEnd = Constructor->init_end();
748a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor           I != IEnd; ++I) {
749a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (!(*I)->isWritten())
750a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          continue;
751a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
752a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        WrittenInits.push_back(*I);
753a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
754a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
755a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Sort the initializers in source order
756a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      llvm::array_pod_sort(WrittenInits.begin(), WrittenInits.end(),
757cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt                           &CompareCXXCtorInitializers);
758a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
759a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Visit the initializers in source order
760a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (unsigned I = 0, N = WrittenInits.size(); I != N; ++I) {
761cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt        CXXCtorInitializer *Init = WrittenInits[I];
76200eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet        if (Init->isAnyMemberInitializer()) {
76300eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet          if (Visit(MakeCursorMemberRef(Init->getAnyMember(),
764a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                        Init->getMemberLocation(), TU)))
765a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
76676852c218a207ef43583515cb835b6e855353a0fDouglas Gregor        } else if (TypeSourceInfo *TInfo = Init->getTypeSourceInfo()) {
76776852c218a207ef43583515cb835b6e855353a0fDouglas Gregor          if (Visit(TInfo->getTypeLoc()))
768a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
769a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        }
770a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
771a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        // Visit the initializer value.
772a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (Expr *Initializer = Init->getInit())
773aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis          if (Visit(MakeCXCursor(Initializer, ND, TU, RegionOfInterest)))
774a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
775a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
776a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
777a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
778aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
779a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return true;
780a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  }
781f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
782b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
783b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
784dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
7851ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
7861ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
7871ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
788f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7891ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *BitWidth = D->getBitWidth())
790aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(BitWidth, StmtParent, TU, RegionOfInterest));
791f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
7931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
7941ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
7951ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitVarDecl(VarDecl *D) {
7961ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
7971ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
798f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7991ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInit())
800aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
801f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8021ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8031ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8041ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
80584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D) {
80684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitDeclaratorDecl(D))
80784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
80884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
80984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
81084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (Expr *DefArg = D->getDefaultArgument())
811aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(DefArg, StmtParent, TU, RegionOfInterest));
81284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
81384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
81484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
81584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
816fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
817fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the FunctionDecl
818fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // before visiting these template parameters.
819fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
820fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return true;
821fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
822fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return VisitFunctionDecl(D->getTemplatedDecl());
823fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
824fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
82539d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregorbool CursorVisitor::VisitClassTemplateDecl(ClassTemplateDecl *D) {
82639d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
82739d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // before visiting these template parameters.
82839d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
82939d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return true;
83039d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
83139d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  return VisitCXXRecordDecl(D->getTemplatedDecl());
83239d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor}
83339d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
83484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D) {
83584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
83684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
83784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
83884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited() &&
83984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      VisitTemplateArgumentLoc(D->getDefaultArgument()))
84084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
84184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
84284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
84384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
84484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
8451ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
8464bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor  if (TypeSourceInfo *TSInfo = ND->getResultTypeSourceInfo())
8474bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
8484bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor      return true;
8494bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor
850f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  for (ObjCMethodDecl::param_iterator P = ND->param_begin(),
8511ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       PEnd = ND->param_end();
8521ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       P != PEnd; ++P) {
853aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
8541ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
8551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  }
856f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8571ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (ND->isThisDeclarationADefinition() &&
858aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
8591ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
860f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8611ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8621ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8631ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
86403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidistemplate <typename DeclIt>
86503ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidisstatic void addRangedDeclsInContainer(DeclIt *DI_current, DeclIt DE_current,
86603ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                      SourceManager &SM, SourceLocation EndLoc,
86703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                      SmallVectorImpl<Decl *> &Decls) {
86803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  DeclIt next = *DI_current;
86903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  while (++next != DE_current) {
87003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    Decl *D_next = *next;
87103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (!D_next)
87203ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      break;
87303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    SourceLocation L = D_next->getLocStart();
87403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (!L.isValid())
87503ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      break;
87603ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (SM.isBeforeInTranslationUnit(L, EndLoc)) {
87703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      *DI_current = next;
87803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      Decls.push_back(D_next);
87903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      continue;
88003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    }
88103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    break;
88203ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  }
88303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis}
88403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
885d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremeneknamespace {
886d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  struct ContainerDeclsSort {
887d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    SourceManager &SM;
888d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    ContainerDeclsSort(SourceManager &sm) : SM(sm) {}
889d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    bool operator()(Decl *A, Decl *B) {
890d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_A = A->getLocStart();
891d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_B = B->getLocStart();
892d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      assert(L_A.isValid() && L_B.isValid());
893d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return SM.isBeforeInTranslationUnit(L_A, L_B);
894d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
895d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  };
896d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
897d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
898a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregorbool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
899d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually convert back to just 'VisitDeclContext()'.  Essentially
900d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // an @implementation can lexically contain Decls that are not properly
901d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // nested in the AST.  When we identify such cases, we need to retrofit
902d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // this nesting here.
90303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  if (!DI_current && !FileDI_current)
904d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
905d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
906d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Scan the Decls that immediately come after the container
907d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // in the current DeclContext.  If any fall within the
908d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // container's lexical region, stash them into a vector
909d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // for later processing.
9105f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<Decl *, 24> DeclsInContainer;
911d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SourceLocation EndLoc = D->getSourceRange().getEnd();
912a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  SourceManager &SM = AU->getSourceManager();
913d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (EndLoc.isValid()) {
91403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (DI_current) {
91503ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      addRangedDeclsInContainer(DI_current, DE_current, SM, EndLoc,
91603ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                DeclsInContainer);
91703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    } else {
91803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      addRangedDeclsInContainer(FileDI_current, FileDE_current, SM, EndLoc,
91903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                DeclsInContainer);
920d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
921d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
922d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
923d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // The common case.
924d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (DeclsInContainer.empty())
925d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
926d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
927d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Get all the Decls in the DeclContext, and sort them with the
928d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // additional ones we've collected.  Then visit them.
929d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for (DeclContext::decl_iterator I = D->decls_begin(), E = D->decls_end();
930d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek       I!=E; ++I) {
931d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *subDecl = *I;
9320582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek    if (!subDecl || subDecl->getLexicalDeclContext() != D ||
9330582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek        subDecl->getLocStart().isInvalid())
934d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
935d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclsInContainer.push_back(subDecl);
936d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
937d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
938d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now sort the Decls so that they appear in lexical order.
939d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  std::sort(DeclsInContainer.begin(), DeclsInContainer.end(),
940d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek            ContainerDeclsSort(SM));
941d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
942d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now visit the decls.
9435f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  for (SmallVectorImpl<Decl*>::iterator I = DeclsInContainer.begin(),
944d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek         E = DeclsInContainer.end(); I != E; ++I) {
945aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    CXCursor Cursor = MakeCXCursor(*I, TU, RegionOfInterest);
946d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
947d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
948d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
949d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
950d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
951d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Visit(Cursor, true))
952d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return true;
953d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
954d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return false;
955a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor}
956a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor
957b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
958b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
959b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                   TU)))
960b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
961f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
96278db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
96378db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
96478db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = ND->protocol_end(); I != E; ++I, ++PL)
965b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
966b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
967f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
968a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(ND);
969dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
970dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
9711ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
972bd9482d859a74bf2c45ef8b8aedec61c0e1c8374Douglas Gregor  if (!PID->isThisDeclarationADefinition())
973bd9482d859a74bf2c45ef8b8aedec61c0e1c8374Douglas Gregor    return Visit(MakeCursorObjCProtocolRef(PID, PID->getLocation(), TU));
974bd9482d859a74bf2c45ef8b8aedec61c0e1c8374Douglas Gregor
9751ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
9761ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
9771ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       E = PID->protocol_end(); I != E; ++I, ++PL)
9781ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
9791ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
980f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9811ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(PID);
9821ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
9831ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
98423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenekbool CursorVisitor::VisitObjCPropertyDecl(ObjCPropertyDecl *PD) {
98583cb94269015bf2770ade71e616c5322ea7e76e1Douglas Gregor  if (PD->getTypeSourceInfo() && Visit(PD->getTypeSourceInfo()->getTypeLoc()))
986fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall    return true;
987fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall
98823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // FIXME: This implements a workaround with @property declarations also being
98923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // installed in the DeclContext for the @interface.  Eventually this code
99023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // should be removed.
99123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCCategoryDecl *CDecl = dyn_cast<ObjCCategoryDecl>(PD->getDeclContext());
99223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!CDecl || !CDecl->IsClassExtension())
99323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
99423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
99523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCInterfaceDecl *ID = CDecl->getClassInterface();
99623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!ID)
99723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
99823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
99923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  IdentifierInfo *PropertyId = PD->getIdentifier();
100023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCPropertyDecl *prevDecl =
100123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(ID), PropertyId);
100223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
100323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!prevDecl)
100423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
100523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
100623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // Visit synthesized methods since they will be skipped when visiting
100723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // the @interface.
100823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getGetterMethodDecl())
1009a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
1010aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
101123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
101223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
101323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getSetterMethodDecl())
1014a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
1015aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
101623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
101723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
101823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  return false;
101923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek}
102023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
1021b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
1022375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  if (!D->isThisDeclarationADefinition()) {
1023375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    // Forward declaration is treated like a reference.
1024375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    return Visit(MakeCursorObjCClassRef(D, D->getLocation(), TU));
1025375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  }
1026375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor
1027dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek  // Issue callbacks for super class.
1028b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (D->getSuperClass() &&
1029b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1030f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
1031b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                        TU)))
1032b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
1033f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
103478db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
103578db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
103678db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = D->protocol_end(); I != E; ++I, ++PL)
1037b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1038b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1039f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1040a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(D);
1041dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1042dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10431ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
10441ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(D);
10451ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10461ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10471ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
1048ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  // 'ID' could be null when dealing with invalid code.
1049ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  if (ObjCInterfaceDecl *ID = D->getClassInterface())
1050ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek    if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
1051ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek      return true;
1052f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10531ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
10541ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10561ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
10571ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#if 0
10581ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // Issue callbacks for super class.
10591ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // FIXME: No source location information!
10601ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (D->getSuperClass() &&
10611ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1062f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
10631ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                        TU)))
1064a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return true;
10651ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#endif
1066f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10671ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
1068dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1069dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
1070a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregorbool CursorVisitor::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD) {
1071a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  if (ObjCIvarDecl *Ivar = PD->getPropertyIvarDecl())
1072135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis    if (PD->isIvarNameSpecified())
1073135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      return Visit(MakeCursorMemberRef(Ivar, PD->getPropertyIvarDeclLoc(), TU));
1074a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
1075a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  return false;
1076a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor}
1077a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
10788f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenekbool CursorVisitor::VisitNamespaceDecl(NamespaceDecl *D) {
10798f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  return VisitDeclContext(D);
10808f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek}
10818f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek
10826931900f43cea558c6974075256c07728dbfecc6Douglas Gregorbool CursorVisitor::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
1083c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
10840cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
10850cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1086c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
10876931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
10886931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  return Visit(MakeCursorNamespaceRef(D->getAliasedNamespace(),
10896931900f43cea558c6974075256c07728dbfecc6Douglas Gregor                                      D->getTargetNameLoc(), TU));
10906931900f43cea558c6974075256c07728dbfecc6Douglas Gregor}
10916931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
10927e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUsingDecl(UsingDecl *D) {
1093c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1094dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1095dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1096c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1097dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
10987e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
10991f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (Visit(MakeCursorOverloadedDeclRef(D, D->getLocation(), TU)))
11001f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return true;
11011f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
11027e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11037e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11047e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11050a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregorbool CursorVisitor::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
1106c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1107db9924191092b4d426cc066637d81698211846aaDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1108db9924191092b4d426cc066637d81698211846aaDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1109c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
11100a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11110a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  return Visit(MakeCursorNamespaceRef(D->getNominatedNamespaceAsWritten(),
11120a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor                                      D->getIdentLocation(), TU));
11130a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor}
11140a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11157e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1116c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1117dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1118dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1119c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1120dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1121c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11227e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11237e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11247e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11257e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypenameDecl(
11267e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor                                               UnresolvedUsingTypenameDecl *D) {
1127c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1128dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1129dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1130c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1131c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11327e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return false;
11337e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11347e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
113501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitDeclarationNameInfo(DeclarationNameInfo Name) {
113601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  switch (Name.getName().getNameKind()) {
113701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::Identifier:
113801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXLiteralOperatorName:
113901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXOperatorName:
114001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXUsingDirective:
114101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
114201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
114301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConstructorName:
114401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXDestructorName:
114501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConversionFunctionName:
114601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (TypeSourceInfo *TSInfo = Name.getNamedTypeInfo())
114701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return Visit(TSInfo->getTypeLoc());
114801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
114901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
115001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCZeroArgSelector:
115101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCOneArgSelector:
115201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCMultiArgSelector:
115301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Per-identifier location info?
115401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
115501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
11567530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie
11577530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid DeclarationName::Kind!");
115801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
115901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1160c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregorbool CursorVisitor::VisitNestedNameSpecifier(NestedNameSpecifier *NNS,
1161c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                             SourceRange Range) {
1162c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // FIXME: This whole routine is a hack to work around the lack of proper
1163c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // source information in nested-name-specifiers (PR5791). Since we do have
1164c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // a beginning source location, we can visit the first component of the
1165c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // nested-name-specifier, if it's a single-token component.
1166c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  if (!NNS)
1167c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return false;
1168c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1169c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Get the first component in the nested-name-specifier.
1170c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  while (NestedNameSpecifier *Prefix = NNS->getPrefix())
1171c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    NNS = Prefix;
1172c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1173c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  switch (NNS->getKind()) {
1174c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Namespace:
1175c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(), Range.getBegin(),
1176c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                        TU));
1177c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
117814aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor  case NestedNameSpecifier::NamespaceAlias:
117914aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
118014aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor                                        Range.getBegin(), TU));
118114aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor
1182c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpec: {
1183c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // If the type has a form where we know that the beginning of the source
1184c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // range matches up with a reference cursor. Visit the appropriate reference
1185c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // cursor.
1186f4c7371fb1d3cebcfb40abad4537bb82515704eaJohn McCall    const Type *T = NNS->getAsType();
1187c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TypedefType *Typedef = dyn_cast<TypedefType>(T))
1188c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Typedef->getDecl(), Range.getBegin(), TU));
1189c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TagType *Tag = dyn_cast<TagType>(T))
1190c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Tag->getDecl(), Range.getBegin(), TU));
1191c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TemplateSpecializationType *TST
1192c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                      = dyn_cast<TemplateSpecializationType>(T))
1193c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return VisitTemplateName(TST->getTemplateName(), Range.getBegin());
1194c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1195c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1196c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1197c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpecWithTemplate:
1198c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Global:
1199c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Identifier:
1200c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1201c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1202c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1203c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  return false;
1204c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor}
1205c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1206dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregorbool
1207dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas GregorCursorVisitor::VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
12085f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<NestedNameSpecifierLoc, 4> Qualifiers;
1209dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  for (; Qualifier; Qualifier = Qualifier.getPrefix())
1210dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    Qualifiers.push_back(Qualifier);
1211dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1212dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  while (!Qualifiers.empty()) {
1213dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifierLoc Q = Qualifiers.pop_back_val();
1214dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifier *NNS = Q.getNestedNameSpecifier();
1215dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    switch (NNS->getKind()) {
1216dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Namespace:
1217dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(),
1218c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1219dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1220dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1221dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1222dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1223dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1224dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::NamespaceAlias:
1225dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1226c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1227dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1228dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1229dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1230dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1231dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1232dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpec:
1233dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpecWithTemplate:
1234dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(Q.getTypeLoc()))
1235dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1236dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1237dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1238dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1239dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Global:
1240dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Identifier:
1241dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1242dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    }
1243dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1244dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1245dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  return false;
1246dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor}
1247dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1248fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateParameters(
1249fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          const TemplateParameterList *Params) {
1250fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (!Params)
1251fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1252fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1253fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (TemplateParameterList::const_iterator P = Params->begin(),
1254fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          PEnd = Params->end();
1255fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor       P != PEnd; ++P) {
1256aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
1257fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1258fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1259fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1260fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1261fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1262fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
12630b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregorbool CursorVisitor::VisitTemplateName(TemplateName Name, SourceLocation Loc) {
12640b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  switch (Name.getKind()) {
12650b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::Template:
12660b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(Name.getAsTemplateDecl(), Loc, TU));
12670b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12680b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::OverloadedTemplate:
12691f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // Visit the overloaded template set.
12701f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (Visit(MakeCursorOverloadedDeclRef(Name, Loc, TU)))
12711f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return true;
12721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
12730b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
12740b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12750b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::DependentTemplate:
12760b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
12770b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
12780b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12790b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::QualifiedTemplate:
12800b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
12810b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(
12820b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                  Name.getAsQualifiedTemplateName()->getDecl(),
12830b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                       Loc, TU));
1284146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall
1285146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall  case TemplateName::SubstTemplateTemplateParm:
1286146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall    return Visit(MakeCursorTemplateRef(
1287146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall                         Name.getAsSubstTemplateTemplateParm()->getParameter(),
1288146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall                                       Loc, TU));
12891aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor
12901aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor  case TemplateName::SubstTemplateTemplateParmPack:
12911aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor    return Visit(MakeCursorTemplateRef(
12921aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                  Name.getAsSubstTemplateTemplateParmPack()->getParameterPack(),
12931aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                                       Loc, TU));
12940b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  }
12957530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie
12967530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid TemplateName::Kind!");
12970b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor}
12980b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
1299fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL) {
1300fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  switch (TAL.getArgument().getKind()) {
1301fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Null:
1302fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Integral:
1303fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Pack:
1304fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
130587dd697dcc8ecb64df73ae64d61b8c80ff0c157cDouglas Gregor
1306fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Type:
1307fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (TypeSourceInfo *TSInfo = TAL.getTypeSourceInfo())
1308fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(TSInfo->getTypeLoc());
1309fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1310fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1311fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Declaration:
1312fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceDeclExpression())
1313aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1314fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1315fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1316fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Expression:
1317fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceExpression())
1318aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1319fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1320fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1321fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Template:
1322a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor  case TemplateArgument::TemplateExpansion:
1323b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor    if (VisitNestedNameSpecifierLoc(TAL.getTemplateQualifierLoc()))
1324b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor      return true;
1325b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor
1326a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor    return VisitTemplateName(TAL.getArgument().getAsTemplateOrTemplatePattern(),
13270b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                             TAL.getTemplateNameLoc());
1328fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
13297530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie
13307530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid TemplateArgument::Kind!");
1331fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1332fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1333a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenekbool CursorVisitor::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
1334a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  return VisitDeclContext(D);
1335a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek}
1336a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek
133701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
133801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return Visit(TL.getUnqualifiedLoc());
133901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
134001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1341f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
1342a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTContext &Context = AU->getASTContext();
1343f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1344f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // Some builtin types (such as Objective-C's "id", "sel", and
1345f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // "Class") have associated declarations. Create cursors for those.
1346f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  QualType VisitType;
1347e0a22d06888c13989b3f72db319f1d498bf69153John McCall  switch (TL.getTypePtr()->getKind()) {
13482dde35bc626153492f5f58202506c88a27fbff5bJohn McCall
13496b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Void:
1350f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::NullPtr:
13516b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Dependent:
13522dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define BUILTIN_TYPE(Id, SingletonId)
13532dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define SIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
13542dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define UNSIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
13552dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define FLOATING_TYPE(Id, SingletonId) case BuiltinType::Id:
13562dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define PLACEHOLDER_TYPE(Id, SingletonId) case BuiltinType::Id:
13572dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#include "clang/AST/BuiltinTypes.def"
1358f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
13596b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1360f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCId:
1361f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCIdType();
1362f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
13636b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
13646b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ObjCClass:
13656b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    VisitType = Context.getObjCClassType();
13666b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    break;
13676b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1368f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCSel:
1369f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCSelType();
1370f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
1371f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1372f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1373f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (!VisitType.isNull()) {
1374f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
1375f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
1376f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                     TU));
1377f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1378f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1379f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1380f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1381f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
13827d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
1383162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return Visit(MakeCursorTypeRef(TL.getTypedefNameDecl(), TL.getNameLoc(), TU));
13847d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
13857d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1386f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
1387f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1388f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1389f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1390f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
13916f155de99c59af890817146ec8526bafb6560f1fArgyrios Kyrtzidis  if (TL.isDefinition())
1392aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(TL.getDecl(), TU, RegionOfInterest));
13936f155de99c59af890817146ec8526bafb6560f1fArgyrios Kyrtzidis
1394f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1395f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1396f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1397fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
1398960d13dde337a59dacc9dc3936c26d4aa8478986Chandler Carruth  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1399fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1400fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1401f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
1402f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU)))
1403f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1404f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1405c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return false;
1406c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall}
1407c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1408c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCallbool CursorVisitor::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
1409c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseLoc()))
1410c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall    return true;
1411c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1412f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1413f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1414f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                        TU)))
1415f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor      return true;
1416f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1417f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1418f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1419f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1420f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1421f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
1422c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return Visit(TL.getPointeeLoc());
1423f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1424f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1425075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnarabool CursorVisitor::VisitParenTypeLoc(ParenTypeLoc TL) {
1426075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara  return Visit(TL.getInnerLoc());
1427075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara}
1428075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara
1429f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
1430f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1431f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1432f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1433f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
1434f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1435f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1436f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1437f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
1438f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1439f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1440f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1441f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
1442f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1443f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1444f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1445f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
1446f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1447f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1448f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
14493422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidisbool CursorVisitor::VisitAttributedTypeLoc(AttributedTypeLoc TL) {
14503422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis  return Visit(TL.getModifiedLoc());
14513422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis}
14523422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis
145301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL,
145401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor                                         bool SkipResultType) {
145501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (!SkipResultType && Visit(TL.getResultLoc()))
1456f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1457f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1458f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
14595dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek    if (Decl *D = TL.getArg(I))
1460aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(D, TU, RegionOfInterest)))
14615dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek        return true;
1462f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1463f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1464f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1465f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1466f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
1467f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(TL.getElementLoc()))
1468f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1469f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1470f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Expr *Size = TL.getSizeExpr())
1471aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Size, StmtParent, TU, RegionOfInterest));
1472f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1473f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1474f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1475f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1476fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateSpecializationTypeLoc(
1477fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                             TemplateSpecializationTypeLoc TL) {
14780b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  // Visit the template name.
14790b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
14800b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                        TL.getTemplateNameLoc()))
14810b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return true;
1482fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1483fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Visit the template arguments.
1484fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1485fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1486fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1487fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1488fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1489fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1490fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
14912332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
14922332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
14932332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
14942332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
14952332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
14962332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
1497ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt    return Visit(TSInfo->getTypeLoc());
1498ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt
1499ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt  return false;
1500ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt}
1501ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt
1502ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Huntbool CursorVisitor::VisitUnaryTransformTypeLoc(UnaryTransformTypeLoc TL) {
1503ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
15042332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor    return Visit(TSInfo->getTypeLoc());
15052332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15062332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return false;
15072332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15082332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15092494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregorbool CursorVisitor::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
15102494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15112494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    return true;
15122494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
15132494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  return false;
15142494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor}
15152494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
151694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregorbool CursorVisitor::VisitDependentTemplateSpecializationTypeLoc(
151794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor                                    DependentTemplateSpecializationTypeLoc TL) {
151894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the nested-name-specifier, if there is one.
151994fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  if (TL.getQualifierLoc() &&
152094fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
152194fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    return true;
152294fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
152394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the template arguments.
152494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
152594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
152694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      return true;
152794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
152894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  return false;
152994fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor}
153094fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
15319e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregorbool CursorVisitor::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
15329e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15339e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor    return true;
15349e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15359e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  return Visit(TL.getNamedTypeLoc());
15369e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor}
15379e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15387536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregorbool CursorVisitor::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
15397536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor  return Visit(TL.getPatternLoc());
15407536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor}
15417536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor
1542427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::VisitDecltypeTypeLoc(DecltypeTypeLoc TL) {
1543427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  if (Expr *E = TL.getUnderlyingExpr())
1544427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis    return Visit(MakeCXCursor(E, StmtParent, TU));
1545427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1546427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return false;
1547427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1548427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1549427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::VisitInjectedClassNameTypeLoc(InjectedClassNameTypeLoc TL) {
1550427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1551427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1552427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1553b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedmanbool CursorVisitor::VisitAtomicTypeLoc(AtomicTypeLoc TL) {
1554b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman  return Visit(TL.getValueLoc());
1555b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman}
1556b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman
1557427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis#define DEFAULT_TYPELOC_IMPL(CLASS, PARENT) \
1558427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::Visit##CLASS##TypeLoc(CLASS##TypeLoc TL) { \
1559427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return Visit##PARENT##Loc(TL); \
1560427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1561427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1562427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Complex, Type)
1563427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(ConstantArray, ArrayType)
1564427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(IncompleteArray, ArrayType)
1565427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(VariableArray, ArrayType)
1566427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(DependentSizedArray, ArrayType)
1567427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(DependentSizedExtVector, Type)
1568427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Vector, Type)
1569427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(ExtVector, VectorType)
1570427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(FunctionProto, FunctionType)
1571427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(FunctionNoProto, FunctionType)
1572427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Record, TagType)
1573427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Enum, TagType)
1574427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(SubstTemplateTypeParm, Type)
1575427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(SubstTemplateTypeParmPack, Type)
1576427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Auto, Type)
1577427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
15783064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenekbool CursorVisitor::VisitCXXRecordDecl(CXXRecordDecl *D) {
1579c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
1580c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1581c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1582c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
1583c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
15845e1cdac63c3d9c9b32fa41fa0b2d242a58a20d49John McCall  if (D->isCompleteDefinition()) {
15853064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    for (CXXRecordDecl::base_class_iterator I = D->bases_begin(),
15863064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek         E = D->bases_end(); I != E; ++I) {
15873064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      if (Visit(cxcursor::MakeCursorCXXBaseSpecifier(I, TU)))
15883064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek        return true;
15893064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
15903064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  }
15913064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
15923064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  return VisitTagDecl(D);
15933064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek}
15943064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
159509dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenekbool CursorVisitor::VisitAttributes(Decl *D) {
1596cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt  for (AttrVec::const_iterator i = D->attr_begin(), e = D->attr_end();
1597cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt       i != e; ++i)
1598cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt    if (Visit(MakeCXCursor(*i, D, TU)))
159909dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek        return true;
160009dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
160109dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  return false;
160209dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek}
160309dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
1604c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1605c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Data-recursive visitor methods.
1606c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1607c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
160828a719433411ef782b582946823bc648ddcc4533Ted Kremeneknamespace {
1609035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#define DEF_JOB(NAME, DATA, KIND)\
1610035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass NAME : public VisitorJob {\
1611035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:\
1612035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  NAME(DATA *d, CXCursor parent) : VisitorJob(parent, VisitorJob::KIND, d) {} \
1613035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) { return VJ->getKind() == KIND; }\
1614f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DATA *get() const { return static_cast<DATA*>(data[0]); }\
1615035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1616035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1617035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(StmtVisit, Stmt, StmtVisitKind)
1618035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(MemberExprParts, MemberExpr, MemberExprPartsKind)
1619e4979ccb5960608edce73f3b274eb7c2de15dac5Ted KremenekDEF_JOB(DeclRefExprParts, DeclRefExpr, DeclRefExprPartsKind)
1620035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(OverloadExprParts, OverloadExpr, OverloadExprPartsKind)
1621b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios KyrtzidisDEF_JOB(ExplicitTemplateArgsVisit, ASTTemplateArgumentListInfo,
162260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        ExplicitTemplateArgsVisitKind)
162394d96291cd041adc5731a2294828a9c20e450b74Douglas GregorDEF_JOB(SizeOfPackExprParts, SizeOfPackExpr, SizeOfPackExprPartsKind)
1624011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas GregorDEF_JOB(LambdaExprParts, LambdaExpr, LambdaExprPartsKind)
1625035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#undef DEF_JOB
1626035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1627035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass DeclVisit : public VisitorJob {
1628035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1629035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  DeclVisit(Decl *d, CXCursor parent, bool isFirst) :
1630035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::DeclVisitKind,
1631035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               d, isFirst ? (void*) 1 : (void*) 0) {}
1632035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
163382f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    return VJ->getKind() == DeclVisitKind;
1634035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1635f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  Decl *get() const { return static_cast<Decl*>(data[0]); }
1636f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  bool isFirst() const { return data[1] ? true : false; }
1637035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1638035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass TypeLocVisit : public VisitorJob {
1639035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1640035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  TypeLocVisit(TypeLoc tl, CXCursor parent) :
1641035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::TypeLocVisitKind,
1642035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               tl.getType().getAsOpaquePtr(), tl.getOpaqueData()) {}
1643035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1644035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1645035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return VJ->getKind() == TypeLocVisitKind;
1646035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1647035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
164882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek  TypeLoc get() const {
1649f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    QualType T = QualType::getFromOpaquePtr(data[0]);
1650f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return TypeLoc(T, data[1]);
1651035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1652035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1653035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1654ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekclass LabelRefVisit : public VisitorJob {
1655ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekpublic:
1656ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelRefVisit(LabelDecl *LD, SourceLocation labelLoc, CXCursor parent)
1657ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner    : VisitorJob(parent, VisitorJob::LabelRefVisitKind, LD,
1658dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 labelLoc.getPtrEncoding()) {}
1659ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek
1660ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1661ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek    return VJ->getKind() == VisitorJob::LabelRefVisitKind;
1662ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  }
1663ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelDecl *get() const { return static_cast<LabelDecl*>(data[0]); }
1664ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  SourceLocation getLoc() const {
1665dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin    return SourceLocation::getFromPtrEncoding(data[1]); }
1666f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek};
1667f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1668f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorclass NestedNameSpecifierLocVisit : public VisitorJob {
1669f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorpublic:
1670f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLocVisit(NestedNameSpecifierLoc Qualifier, CXCursor parent)
1671f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    : VisitorJob(parent, VisitorJob::NestedNameSpecifierLocVisitKind,
1672f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getNestedNameSpecifier(),
1673f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getOpaqueData()) { }
1674f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1675f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  static bool classof(const VisitorJob *VJ) {
1676f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return VJ->getKind() == VisitorJob::NestedNameSpecifierLocVisitKind;
1677f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1678f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1679f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLoc get() const {
1680f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return NestedNameSpecifierLoc(static_cast<NestedNameSpecifier*>(data[0]),
1681f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                                  data[1]);
1682f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1683f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor};
1684f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1685f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekclass DeclarationNameInfoVisit : public VisitorJob {
1686f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekpublic:
1687f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfoVisit(Stmt *S, CXCursor parent)
1688f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : VisitorJob(parent, VisitorJob::DeclarationNameInfoVisitKind, S) {}
1689f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1690f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return VJ->getKind() == VisitorJob::DeclarationNameInfoVisitKind;
1691f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1692f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfo get() const {
1693f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    Stmt *S = static_cast<Stmt*>(data[0]);
1694f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    switch (S->getStmtClass()) {
1695f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    default:
1696f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      llvm_unreachable("Unhandled Stmt");
1697ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor    case clang::Stmt::MSDependentExistsStmtClass:
1698ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor      return cast<MSDependentExistsStmt>(S)->getNameInfo();
1699f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::CXXDependentScopeMemberExprClass:
1700f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<CXXDependentScopeMemberExpr>(S)->getMemberNameInfo();
1701f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::DependentScopeDeclRefExprClass:
1702f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<DependentScopeDeclRefExpr>(S)->getNameInfo();
1703f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    }
1704f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1705ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek};
1706cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekclass MemberRefVisit : public VisitorJob {
1707cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekpublic:
1708cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  MemberRefVisit(FieldDecl *D, SourceLocation L, CXCursor parent)
1709cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    : VisitorJob(parent, VisitorJob::MemberRefVisitKind, D,
1710dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 L.getPtrEncoding()) {}
1711cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  static bool classof(const VisitorJob *VJ) {
1712cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return VJ->getKind() == VisitorJob::MemberRefVisitKind;
1713cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1714cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  FieldDecl *get() const {
1715cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return static_cast<FieldDecl*>(data[0]);
1716cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1717cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  SourceLocation getLoc() const {
1718cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1719cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1720cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek};
172128a719433411ef782b582946823bc648ddcc4533Ted Kremenekclass EnqueueVisitor : public StmtVisitor<EnqueueVisitor, void> {
172228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitorWorkList &WL;
172328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  CXCursor Parent;
172428a719433411ef782b582946823bc648ddcc4533Ted Kremenekpublic:
172528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueVisitor(VisitorWorkList &wl, CXCursor parent)
172628a719433411ef782b582946823bc648ddcc4533Ted Kremenek    : WL(wl), Parent(parent) {}
172728a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1728ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitAddrLabelExpr(AddrLabelExpr *E);
172973d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitBlockExpr(BlockExpr *B);
173028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1731083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  void VisitCompoundStmt(CompoundStmt *S);
173211b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) { /* Do nothing. */ }
1733ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  void VisitMSDependentExistsStmt(MSDependentExistsStmt *S);
1734f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
173511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXNewExpr(CXXNewExpr *E);
17366d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
173728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
1738cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E);
173973d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E);
1740b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  void VisitCXXTypeidExpr(CXXTypeidExpr *E);
174155b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
17421e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  void VisitCXXUuidofExpr(CXXUuidofExpr *E);
1743dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis  void VisitCXXCatchStmt(CXXCatchStmt *S);
1744e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  void VisitDeclRefExpr(DeclRefExpr *D);
1745035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void VisitDeclStmt(DeclStmt *S);
1746f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E);
1747cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitDesignatedInitExpr(DesignatedInitExpr *E);
174828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitExplicitCastExpr(ExplicitCastExpr *E);
174928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitForStmt(ForStmt *FS);
1750ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitGotoStmt(GotoStmt *GS);
175128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitIfStmt(IfStmt *If);
175228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitInitListExpr(InitListExpr *IE);
175328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitMemberExpr(MemberExpr *M);
1754cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitOffsetOfExpr(OffsetOfExpr *E);
175573d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
175628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitObjCMessageExpr(ObjCMessageExpr *M);
175728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitOverloadExpr(OverloadExpr *E);
1758f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne  void VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E);
175928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitStmt(Stmt *S);
176028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitSwitchStmt(SwitchStmt *S);
176128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitWhileStmt(WhileStmt *W);
17622939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  void VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E);
17636ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  void VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E);
17644ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor  void VisitTypeTraitExpr(TypeTraitExpr *E);
176521ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley  void VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E);
1766552622067dc45013d240f73952fece703f5e63bdJohn Wiegley  void VisitExpressionTraitExpr(ExpressionTraitExpr *E);
176728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U);
17689d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  void VisitVAArgExpr(VAArgExpr *E);
176994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  void VisitSizeOfPackExpr(SizeOfPackExpr *E);
17704b9c2d235fb9449e249d74f48ecfec601650de93John McCall  void VisitPseudoObjectExpr(PseudoObjectExpr *E);
17714b9c2d235fb9449e249d74f48ecfec601650de93John McCall  void VisitOpaqueValueExpr(OpaqueValueExpr *E);
1772011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  void VisitLambdaExpr(LambdaExpr *E);
1773ee8aff06f6a96214731de17b2cb6df407c6c1820Douglas Gregor
177428a719433411ef782b582946823bc648ddcc4533Ted Kremenekprivate:
1775f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void AddDeclarationNameInfo(Stmt *S);
1776f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  void AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier);
1777b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis  void AddExplicitTemplateArgs(const ASTTemplateArgumentListInfo *A);
1778cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void AddMemberRef(FieldDecl *D, SourceLocation L);
177928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddStmt(Stmt *S);
1780035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void AddDecl(Decl *D, bool isFirst = true);
178128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddTypeLoc(TypeSourceInfo *TI);
178228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void EnqueueChildren(Stmt *S);
178328a719433411ef782b582946823bc648ddcc4533Ted Kremenek};
178428a719433411ef782b582946823bc648ddcc4533Ted Kremenek} // end anonyous namespace
178528a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1786f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::AddDeclarationNameInfo(Stmt *S) {
1787f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // 'S' should always be non-null, since it comes from the
1788f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // statement we are visiting.
1789f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  WL.push_back(DeclarationNameInfoVisit(S, Parent));
1790f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1791f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1792f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorvoid
1793f3db29fff6a583ecda823cf909ab7737d8d30129Douglas GregorEnqueueVisitor::AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1794f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (Qualifier)
1795f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    WL.push_back(NestedNameSpecifierLocVisit(Qualifier, Parent));
1796f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor}
1797f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
179828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddStmt(Stmt *S) {
179928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (S)
180028a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(StmtVisit(S, Parent));
180128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1802035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::AddDecl(Decl *D, bool isFirst) {
180328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (D)
1804035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    WL.push_back(DeclVisit(D, Parent, isFirst));
180528a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
180660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenekvoid EnqueueVisitor::
1807b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis  AddExplicitTemplateArgs(const ASTTemplateArgumentListInfo *A) {
180860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (A)
180960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    WL.push_back(ExplicitTemplateArgsVisit(
1810b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis                        const_cast<ASTTemplateArgumentListInfo*>(A), Parent));
181160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek}
1812cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::AddMemberRef(FieldDecl *D, SourceLocation L) {
1813cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  if (D)
1814cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    WL.push_back(MemberRefVisit(D, L, Parent));
1815cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
181628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddTypeLoc(TypeSourceInfo *TI) {
181728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (TI)
181828a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(TypeLocVisit(TI->getTypeLoc(), Parent));
181928a719433411ef782b582946823bc648ddcc4533Ted Kremenek }
182028a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::EnqueueChildren(Stmt *S) {
1821a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  unsigned size = WL.size();
18227502c1d3ce8bb97bcc4f7bebef507040bd93b26fJohn McCall  for (Stmt::child_range Child = S->children(); Child; ++Child) {
182328a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(*Child);
1824a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  }
1825a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  if (size == WL.size())
1826a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek    return;
1827a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // Now reverse the entries we just added.  This will match the DFS
1828a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // ordering performed by the worklist.
1829a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1830a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  std::reverse(I, E);
1831a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek}
1832ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitAddrLabelExpr(AddrLabelExpr *E) {
1833ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(E->getLabel(), E->getLabelLoc(), Parent));
1834ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
183573d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitBlockExpr(BlockExpr *B) {
183673d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddDecl(B->getBlockDecl());
183773d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
183828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
183928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
184028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeSourceInfo());
184128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1842083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenekvoid EnqueueVisitor::VisitCompoundStmt(CompoundStmt *S) {
1843083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  for (CompoundStmt::reverse_body_iterator I = S->body_rbegin(),
1844083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek        E = S->body_rend(); I != E; ++I) {
1845083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek    AddStmt(*I);
1846083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  }
184711b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
1848f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::
1849ba0513de93d2fab6db5ab30b6927209fcc883078Douglas GregorVisitMSDependentExistsStmt(MSDependentExistsStmt *S) {
1850ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  AddStmt(S->getSubStmt());
1851ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  AddDeclarationNameInfo(S);
1852ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = S->getQualifierLoc())
1853ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1854ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor}
1855ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor
1856ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregorvoid EnqueueVisitor::
1857f64d80306144f978148ba92f36f7cea7b671dd34Ted KremenekVisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E) {
1858f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1859f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
18607c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
18617c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1862f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  if (!E->isImplicitAccess())
1863f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    AddStmt(E->getBase());
1864f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
186511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenekvoid EnqueueVisitor::VisitCXXNewExpr(CXXNewExpr *E) {
18662aed8b88613863f3c439cdfb205bdf8b608fb205Sebastian Redl  // Enqueue the initializer , if any.
18672aed8b88613863f3c439cdfb205bdf8b608fb205Sebastian Redl  AddStmt(E->getInitializer());
186811b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the array size, if any.
186911b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddStmt(E->getArraySize());
187011b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the allocated type.
187111b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddTypeLoc(E->getAllocatedTypeSourceInfo());
187211b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the placement arguments.
187311b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumPlacementArgs(); I > 0; --I)
187411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getPlacementArg(I-1));
187511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
187628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *CE) {
18778b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek  for (unsigned I = CE->getNumArgs(); I > 1 /* Yes, this is 1 */; --I)
18788b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek    AddStmt(CE->getArg(I-1));
187928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getCallee());
188028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getArg(0));
188128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1882cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1883cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the name of the type being destroyed.
1884cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getDestroyedTypeInfo());
1885cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the scope type that looks disturbingly like the nested-name-specifier
1886cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // but isn't.
1887cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getScopeTypeInfo());
1888cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the nested-name-specifier.
1889f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
1890f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1891cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit base expression.
1892cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getBase());
1893cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
18946d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenekvoid EnqueueVisitor::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
18956d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
18966d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
189773d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
189873d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  EnqueueChildren(E);
189973d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
190073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
1901b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenekvoid EnqueueVisitor::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1902b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  EnqueueChildren(E);
1903b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  if (E->isTypeOperand())
1904b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
1905b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek}
190655b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek
190755b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenekvoid EnqueueVisitor::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr
190855b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek                                                     *E) {
190955b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  EnqueueChildren(E);
191055b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
191155b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek}
19121e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenekvoid EnqueueVisitor::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
19131e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  EnqueueChildren(E);
19141e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  if (E->isTypeOperand())
19151e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
19161e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek}
1917dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis
1918dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidisvoid EnqueueVisitor::VisitCXXCatchStmt(CXXCatchStmt *S) {
1919dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis  EnqueueChildren(S);
1920dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis  AddDecl(S->getExceptionDecl());
1921dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis}
1922dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis
1923e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenekvoid EnqueueVisitor::VisitDeclRefExpr(DeclRefExpr *DR) {
192460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (DR->hasExplicitTemplateArgs()) {
192560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    AddExplicitTemplateArgs(&DR->getExplicitTemplateArgs());
192660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  }
1927e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  WL.push_back(DeclRefExprParts(DR, Parent));
1928e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek}
1929f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1930f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1931f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
193200cf3cc2718671aa48e8da264a523b0058a8591eDouglas Gregor  AddNestedNameSpecifierLoc(E->getQualifierLoc());
1933f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1934035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::VisitDeclStmt(DeclStmt *S) {
1935035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  unsigned size = WL.size();
1936035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  bool isFirst = true;
1937035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  for (DeclStmt::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
1938035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek       D != DEnd; ++D) {
1939035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    AddDecl(*D, isFirst);
1940035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    isFirst = false;
1941035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1942035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  if (size == WL.size())
1943035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return;
1944035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // Now reverse the entries we just added.  This will match the DFS
1945035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // ordering performed by the worklist.
1946035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1947035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  std::reverse(I, E);
1948035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek}
1949cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
1950cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getInit());
1951cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  typedef DesignatedInitExpr::Designator Designator;
1952cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (DesignatedInitExpr::reverse_designators_iterator
1953cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D = E->designators_rbegin(), DEnd = E->designators_rend();
1954cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D != DEnd; ++D) {
1955cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isFieldDesignator()) {
1956cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      if (FieldDecl *Field = D->getField())
1957cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        AddMemberRef(Field, D->getFieldLoc());
1958cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1959cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1960cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isArrayDesignator()) {
1961cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getArrayIndex(*D));
1962cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1963cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1964cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    assert(D->isArrayRangeDesignator() && "Unknown designator kind");
1965cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeEnd(*D));
1966cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeStart(*D));
1967cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1968cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
196928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitExplicitCastExpr(ExplicitCastExpr *E) {
197028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
197128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeInfoAsWritten());
197228a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
197328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitForStmt(ForStmt *FS) {
197428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getBody());
197528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInc());
197628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getCond());
197728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(FS->getConditionVariable());
197828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInit());
197928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1980ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitGotoStmt(GotoStmt *GS) {
1981ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(GS->getLabel(), GS->getLabelLoc(), Parent));
1982ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
198328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitIfStmt(IfStmt *If) {
198428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getElse());
198528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getThen());
198628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getCond());
198728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(If->getConditionVariable());
198828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
198928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitInitListExpr(InitListExpr *IE) {
199028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  // We care about the syntactic form of the initializer list, only.
199128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (InitListExpr *Syntactic = IE->getSyntacticForm())
199228a719433411ef782b582946823bc648ddcc4533Ted Kremenek    IE = Syntactic;
199328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(IE);
199428a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
199528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitMemberExpr(MemberExpr *M) {
199689629a746019a42797495b091711a1d68467e88aDouglas Gregor  WL.push_back(MemberExprParts(M, Parent));
199789629a746019a42797495b091711a1d68467e88aDouglas Gregor
199889629a746019a42797495b091711a1d68467e88aDouglas Gregor  // If the base of the member access expression is an implicit 'this', don't
199989629a746019a42797495b091711a1d68467e88aDouglas Gregor  // visit it.
200089629a746019a42797495b091711a1d68467e88aDouglas Gregor  // FIXME: If we ever want to show these implicit accesses, this will be
200189629a746019a42797495b091711a1d68467e88aDouglas Gregor  // unfortunate. However, clang_getCursor() relies on this behavior.
200275e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor  if (!M->isImplicitAccess())
200375e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor    AddStmt(M->getBase());
200428a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
200573d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
200673d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getEncodedTypeSourceInfo());
200773d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
200828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitObjCMessageExpr(ObjCMessageExpr *M) {
200928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(M);
201028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(M->getClassReceiverTypeInfo());
201128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2012cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitOffsetOfExpr(OffsetOfExpr *E) {
2013cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the components of the offsetof expression.
2014cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (unsigned N = E->getNumComponents(), I = N; I > 0; --I) {
2015cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    typedef OffsetOfExpr::OffsetOfNode OffsetOfNode;
2016cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    const OffsetOfNode &Node = E->getComponent(I-1);
2017cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    switch (Node.getKind()) {
2018cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Array:
2019cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getIndexExpr(Node.getArrayExprIndex()));
2020cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2021cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Field:
202206dec892b5300b43263d25c5476b506c9d6cfbadAbramo Bagnara      AddMemberRef(Node.getField(), Node.getSourceRange().getEnd());
2023cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2024cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Identifier:
2025cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Base:
2026cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
2027cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
2028cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
2029cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the type into which we're computing the offset.
2030cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
2031cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
203228a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitOverloadExpr(OverloadExpr *E) {
203360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
20346045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek  WL.push_back(OverloadExprParts(E, Parent));
20356045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek}
2036f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbournevoid EnqueueVisitor::VisitUnaryExprOrTypeTraitExpr(
2037f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne                                              UnaryExprOrTypeTraitExpr *E) {
20386d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  EnqueueChildren(E);
20396d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  if (E->isArgumentType())
20406d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek    AddTypeLoc(E->getArgumentTypeInfo());
20416d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
204228a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitStmt(Stmt *S) {
204328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(S);
204428a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
204528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitSwitchStmt(SwitchStmt *S) {
204628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getBody());
204728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getCond());
204828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(S->getConditionVariable());
204928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2050fafa75aebadef8d6b44a920e3f40529f150a5574Ted Kremenek
205128a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitWhileStmt(WhileStmt *W) {
205228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getBody());
205328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getCond());
205428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(W->getConditionVariable());
205528a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
205621ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley
20572939b6f356161f572712d4d6310b65f9599e3675Ted Kremenekvoid EnqueueVisitor::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
20582939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  AddTypeLoc(E->getQueriedTypeSourceInfo());
20592939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek}
20606ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
20616ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichetvoid EnqueueVisitor::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
20626ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  AddTypeLoc(E->getRhsTypeSourceInfo());
20630a03a3f98b14006a54bcac9e8908a7c9f50e519fFrancois Pichet  AddTypeLoc(E->getLhsTypeSourceInfo());
20646ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet}
20656ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
20664ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregorvoid EnqueueVisitor::VisitTypeTraitExpr(TypeTraitExpr *E) {
20674ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor  for (unsigned I = E->getNumArgs(); I > 0; --I)
20684ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor    AddTypeLoc(E->getArg(I-1));
20694ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor}
20704ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor
207121ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegleyvoid EnqueueVisitor::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
207221ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley  AddTypeLoc(E->getQueriedTypeSourceInfo());
207321ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley}
207421ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley
2075552622067dc45013d240f73952fece703f5e63bdJohn Wiegleyvoid EnqueueVisitor::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
2076552622067dc45013d240f73952fece703f5e63bdJohn Wiegley  EnqueueChildren(E);
2077552622067dc45013d240f73952fece703f5e63bdJohn Wiegley}
2078552622067dc45013d240f73952fece703f5e63bdJohn Wiegley
207928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U) {
208028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitOverloadExpr(U);
208128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (!U->isImplicitAccess())
208228a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(U->getBase());
208328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
20849d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenekvoid EnqueueVisitor::VisitVAArgExpr(VAArgExpr *E) {
20859d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddStmt(E->getSubExpr());
20869d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddTypeLoc(E->getWrittenTypeInfo());
20879d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek}
208894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregorvoid EnqueueVisitor::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
208994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  WL.push_back(SizeOfPackExprParts(E, Parent));
209094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor}
20914b9c2d235fb9449e249d74f48ecfec601650de93John McCallvoid EnqueueVisitor::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
20924b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // If the opaque value has a source expression, just transparently
20934b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // visit that.  This is useful for (e.g.) pseudo-object expressions.
20944b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (Expr *SourceExpr = E->getSourceExpr())
20954b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return Visit(SourceExpr);
20964b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
2097011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregorvoid EnqueueVisitor::VisitLambdaExpr(LambdaExpr *E) {
2098011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  AddStmt(E->getBody());
2099011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  WL.push_back(LambdaExprParts(E, Parent));
2100011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor}
21014b9c2d235fb9449e249d74f48ecfec601650de93John McCallvoid EnqueueVisitor::VisitPseudoObjectExpr(PseudoObjectExpr *E) {
21024b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Treat the expression like its syntactic form.
21034b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Visit(E->getSyntacticForm());
21044b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
21056045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek
2106c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekvoid CursorVisitor::EnqueueWorkList(VisitorWorkList &WL, Stmt *S) {
2107aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  EnqueueVisitor(WL, MakeCXCursor(S, StmtParent, TU,RegionOfInterest)).Visit(S);
2108c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2109c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2110c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::IsInRegionOfInterest(CXCursor C) {
2111c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  if (RegionOfInterest.isValid()) {
2112c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SourceRange Range = getRawCursorExtent(C);
2113c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    if (Range.isInvalid() || CompareRegionOfInterest(Range))
2114c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      return false;
2115c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2116c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return true;
2117c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2118c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2119c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::RunVisitorWorkList(VisitorWorkList &WL) {
2120c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  while (!WL.empty()) {
2121c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Dequeue the worklist item.
212282f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    VisitorJob LI = WL.back();
212382f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    WL.pop_back();
212482f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek
2125c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Set the Parent field, then back to its old value once we're done.
2126c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SetParentRAII SetParent(Parent, StmtParent, LI.getParent());
2127c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2128c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    switch (LI.getKind()) {
2129f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      case VisitorJob::DeclVisitKind: {
213082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Decl *D = cast<DeclVisit>(&LI)->get();
2131f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        if (!D)
2132f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek          continue;
2133f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2134f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // For now, perform default visitation for Decls.
2135aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis        if (Visit(MakeCXCursor(D, TU, RegionOfInterest,
2136aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis                               cast<DeclVisit>(&LI)->isFirst())))
2137f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek            return true;
2138f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2139f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        continue;
2140f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      }
214160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      case VisitorJob::ExplicitTemplateArgsVisitKind: {
2142b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis        const ASTTemplateArgumentListInfo *ArgList =
214360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          cast<ExplicitTemplateArgsVisit>(&LI)->get();
214460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        for (const TemplateArgumentLoc *Arg = ArgList->getTemplateArgs(),
214560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               *ArgEnd = Arg + ArgList->NumTemplateArgs;
214660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               Arg != ArgEnd; ++Arg) {
214760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          if (VisitTemplateArgumentLoc(*Arg))
214860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek            return true;
214960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        }
215060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        continue;
215160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      }
2152cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      case VisitorJob::TypeLocVisitKind: {
2153cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        // Perform default visitation for TypeLocs.
215482f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        if (Visit(cast<TypeLocVisit>(&LI)->get()))
2155cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek          return true;
2156cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        continue;
2157cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      }
2158ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      case VisitorJob::LabelRefVisitKind: {
2159ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner        LabelDecl *LS = cast<LabelRefVisit>(&LI)->get();
2160e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        if (LabelStmt *stmt = LS->getStmt()) {
2161e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          if (Visit(MakeCursorLabelRef(stmt, cast<LabelRefVisit>(&LI)->getLoc(),
2162e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek                                       TU))) {
2163e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek            return true;
2164e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          }
2165e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        }
2166ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek        continue;
2167ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      }
216847695c8ad8424851f62e0d4a983b45b15daee1c5Ted Kremenek
2169f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      case VisitorJob::NestedNameSpecifierLocVisitKind: {
2170f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        NestedNameSpecifierLocVisit *V = cast<NestedNameSpecifierLocVisit>(&LI);
2171f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        if (VisitNestedNameSpecifierLoc(V->get()))
2172f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor          return true;
2173f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        continue;
2174f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      }
2175f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2176f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      case VisitorJob::DeclarationNameInfoVisitKind: {
2177f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        if (VisitDeclarationNameInfo(cast<DeclarationNameInfoVisit>(&LI)
2178f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                                     ->get()))
2179f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek          return true;
2180f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        continue;
2181f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      }
2182cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      case VisitorJob::MemberRefVisitKind: {
2183cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        MemberRefVisit *V = cast<MemberRefVisit>(&LI);
2184cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (Visit(MakeCursorMemberRef(V->get(), V->getLoc(), TU)))
2185cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          return true;
2186cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        continue;
2187cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      }
2188c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::StmtVisitKind: {
218982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Stmt *S = cast<StmtVisit>(&LI)->get();
21908c269ac75569454a049385b1246140db5f2b6faaTed Kremenek        if (!S)
21918c269ac75569454a049385b1246140db5f2b6faaTed Kremenek          continue;
21928c269ac75569454a049385b1246140db5f2b6faaTed Kremenek
2193f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // Update the current cursor.
2194aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis        CXCursor Cursor = MakeCXCursor(S, StmtParent, TU, RegionOfInterest);
2195cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (!IsInRegionOfInterest(Cursor))
2196cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          continue;
2197cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        switch (Visitor(Cursor, Parent, ClientData)) {
2198cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Break: return true;
2199cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Continue: break;
2200cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Recurse:
2201cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek            EnqueueWorkList(WL, S);
220282f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek            break;
2203c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
220482f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        continue;
2205c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2206c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::MemberExprPartsKind: {
2207c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Handle the other pieces in the MemberExpr besides the base.
220882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        MemberExpr *M = cast<MemberExprParts>(&LI)->get();
2209c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2210c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the nested-name-specifier
221140d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = M->getQualifierLoc())
221240d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2213c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            return true;
2214c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2215c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the declaration name.
2216c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (VisitDeclarationNameInfo(M->getMemberNameInfo()))
2217c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          return true;
2218c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2219c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the explicitly-specified template arguments, if any.
2220c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (M->hasExplicitTemplateArgs()) {
2221c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          for (const TemplateArgumentLoc *Arg = M->getTemplateArgs(),
2222c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               *ArgEnd = Arg + M->getNumTemplateArgs();
2223c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               Arg != ArgEnd; ++Arg) {
2224c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            if (VisitTemplateArgumentLoc(*Arg))
2225c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek              return true;
2226c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          }
2227c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
2228c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        continue;
2229c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2230e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      case VisitorJob::DeclRefExprPartsKind: {
223182f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        DeclRefExpr *DR = cast<DeclRefExprParts>(&LI)->get();
2232e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit nested-name-specifier, if present.
223340d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = DR->getQualifierLoc())
223440d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2235e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek            return true;
2236e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit declaration name.
2237e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        if (VisitDeclarationNameInfo(DR->getNameInfo()))
2238e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek          return true;
2239e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        continue;
2240e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      }
22416045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      case VisitorJob::OverloadExprPartsKind: {
224282f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        OverloadExpr *O = cast<OverloadExprParts>(&LI)->get();
22436045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the nested-name-specifier.
22444c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = O->getQualifierLoc())
22454c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
22466045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek            return true;
22476045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the declaration name.
22486045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (VisitDeclarationNameInfo(O->getNameInfo()))
22496045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22506045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the overloaded declaration reference.
22516045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (Visit(MakeCursorOverloadedDeclRef(O, TU)))
22526045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22536045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        continue;
22546045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      }
225594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      case VisitorJob::SizeOfPackExprPartsKind: {
225694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        SizeOfPackExpr *E = cast<SizeOfPackExprParts>(&LI)->get();
225794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        NamedDecl *Pack = E->getPack();
225894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTypeParmDecl>(Pack)) {
225994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTypeRef(cast<TemplateTypeParmDecl>(Pack),
226094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                      E->getPackLoc(), TU)))
226194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
226294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
226394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
226494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
226594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
226694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTemplateParmDecl>(Pack)) {
226794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTemplateRef(cast<TemplateTemplateParmDecl>(Pack),
226894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                          E->getPackLoc(), TU)))
226994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
227094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
227194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
227294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
227394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
227494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // Non-type template parameter packs and function parameter packs are
227594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // treated like DeclRefExpr cursors.
227694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        continue;
227794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      }
2278011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
2279011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      case VisitorJob::LambdaExprPartsKind: {
2280011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        // Visit captures.
2281011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        LambdaExpr *E = cast<LambdaExprParts>(&LI)->get();
2282011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        for (LambdaExpr::capture_iterator C = E->explicit_capture_begin(),
2283011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                                       CEnd = E->explicit_capture_end();
2284011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor             C != CEnd; ++C) {
2285011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          if (C->capturesThis())
2286011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            continue;
2287011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
2288011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          if (Visit(MakeCursorVariableRef(C->getCapturedVar(),
2289011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                                          C->getLocation(),
2290011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                                          TU)))
2291011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            return true;
2292011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        }
2293011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
2294011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        // Visit parameters and return type, if present.
2295011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        if (E->hasExplicitParameters() || E->hasExplicitResultType()) {
2296011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          TypeLoc TL = E->getCallOperator()->getTypeSourceInfo()->getTypeLoc();
2297011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          if (E->hasExplicitParameters() && E->hasExplicitResultType()) {
2298011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            // Visit the whole type.
2299011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            if (Visit(TL))
2300011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              return true;
2301011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          } else if (isa<FunctionProtoTypeLoc>(TL)) {
2302011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            FunctionProtoTypeLoc Proto = cast<FunctionProtoTypeLoc>(TL);
2303011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            if (E->hasExplicitParameters()) {
2304011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              // Visit parameters.
2305011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              for (unsigned I = 0, N = Proto.getNumArgs(); I != N; ++I)
2306011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                if (Visit(MakeCXCursor(Proto.getArg(I), TU)))
2307011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                  return true;
2308011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            } else {
2309011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              // Visit result type.
2310011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              if (Visit(Proto.getResultLoc()))
2311011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                return true;
2312011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            }
2313011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          }
2314011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        }
2315011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        break;
2316011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      }
2317c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    }
2318c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2319c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return false;
2320c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2321c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2322cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekbool CursorVisitor::Visit(Stmt *S) {
2323d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  VisitorWorkList *WL = 0;
2324d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  if (!WorkListFreeList.empty()) {
2325d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = WorkListFreeList.back();
2326d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL->clear();
2327d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListFreeList.pop_back();
2328d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2329d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  else {
2330d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = new VisitorWorkList();
2331d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListCache.push_back(WL);
2332d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2333d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  EnqueueWorkList(*WL, S);
2334d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  bool result = RunVisitorWorkList(*WL);
2335d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  WorkListFreeList.push_back(WL);
2336d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  return result;
2337c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2338c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
233948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichetnamespace {
234048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichettypedef llvm::SmallVector<SourceRange, 4> RefNamePieces;
234148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois PichetRefNamePieces buildPieces(unsigned NameFlags, bool IsMemberRefExpr,
234248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                          const DeclarationNameInfo &NI,
234348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                          const SourceRange &QLoc,
2344b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis                          const ASTTemplateArgumentListInfo *TemplateArgs = 0){
234548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantQualifier = NameFlags & CXNameRange_WantQualifier;
234648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantTemplateArgs = NameFlags & CXNameRange_WantTemplateArgs;
234748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantSinglePiece = NameFlags & CXNameRange_WantSinglePiece;
234848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
234948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const DeclarationName::NameKind Kind = NI.getName().getNameKind();
235048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
235148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  RefNamePieces Pieces;
235248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
235348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantQualifier && QLoc.isValid())
235448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(QLoc);
235548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
235648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (Kind != DeclarationName::CXXOperatorName || IsMemberRefExpr)
235748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(NI.getLoc());
235848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
235948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantTemplateArgs && TemplateArgs)
236048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceRange(TemplateArgs->LAngleLoc,
236148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                                 TemplateArgs->RAngleLoc));
236248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
236348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (Kind == DeclarationName::CXXOperatorName) {
236448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceLocation::getFromRawEncoding(
236548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                       NI.getInfo().CXXOperatorName.BeginOpNameLoc));
236648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceLocation::getFromRawEncoding(
236748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                       NI.getInfo().CXXOperatorName.EndOpNameLoc));
236848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  }
236948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
237048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantSinglePiece) {
237148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    SourceRange R(Pieces.front().getBegin(), Pieces.back().getEnd());
237248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.clear();
237348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(R);
237448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  }
237548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
237648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  return Pieces;
237748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet}
237848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet}
237948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
2380c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2381c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Misc. API hooks.
2382c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2383c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
23848c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic llvm::sys::Mutex EnableMultithreadingMutex;
23858c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic bool EnabledMultithreading;
23868c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
2387fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidisstatic void fatal_error_handler(void *user_data, const std::string& reason) {
2388fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis  // Write the result out to stderr avoiding errs() because raw_ostreams can
2389fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis  // call report_fatal_error.
2390db7a800e0b76036d0faa7123f2e05e45ee3294e5Argyrios Kyrtzidis  fprintf(stderr, "LIBCLANG FATAL ERROR: %s\n", reason.c_str());
2391fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis  ::abort();
2392fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis}
2393fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis
23945e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramerextern "C" {
23950a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas GregorCXIndex clang_createIndex(int excludeDeclarationsFromPCH,
23960a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                          int displayDiagnostics) {
239748615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // Disable pretty stack trace functionality, which will otherwise be a very
239848615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // poor citizen of the world and set up all sorts of signal handlers.
239948615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  llvm::DisablePrettyStackTrace = true;
240048615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar
2401c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // We use crash recovery to make some of our APIs more reliable, implicitly
2402c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // enable it.
2403c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  llvm::CrashRecoveryContext::Enable();
2404c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar
24058c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  // Enable support for multithreading in LLVM.
24068c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  {
24078c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    llvm::sys::ScopedLock L(EnableMultithreadingMutex);
24088c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    if (!EnabledMultithreading) {
2409fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis      llvm::install_fatal_error_handler(fatal_error_handler, 0);
24108c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      llvm::llvm_start_multithreaded();
24118c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      EnabledMultithreading = true;
24128c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    }
24138c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  }
24148c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
2415a030b7cf5e6aad5889b1b662b6979840bc75f87fDouglas Gregor  CIndexer *CIdxr = new CIndexer();
2416e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  if (excludeDeclarationsFromPCH)
2417e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    CIdxr->setOnlyLocalDecls();
24180a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  if (displayDiagnostics)
24190a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    CIdxr->setDisplayDiagnostics();
2420fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2421fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (getenv("LIBCLANG_BGPRIO_INDEX"))
2422fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
2423fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis                               CXGlobalOpt_ThreadBackgroundPriorityForIndexing);
2424fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (getenv("LIBCLANG_BGPRIO_EDIT"))
2425fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
2426fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis                               CXGlobalOpt_ThreadBackgroundPriorityForEditing);
2427fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2428e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  return CIdxr;
2429600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2430600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
24319ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeIndex(CXIndex CIdx) {
24322b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (CIdx)
24332b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    delete static_cast<CIndexer *>(CIdx);
24342bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
24352bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff
2436fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidisvoid clang_CXIndex_setGlobalOptions(CXIndex CIdx, unsigned options) {
2437fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CIdx)
2438fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    static_cast<CIndexer *>(CIdx)->setCXGlobalOptFlags(options);
2439fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis}
2440fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2441fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidisunsigned clang_CXIndex_getGlobalOptions(CXIndex CIdx) {
2442fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CIdx)
2443fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    return static_cast<CIndexer *>(CIdx)->getCXGlobalOptFlags();
2444fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  return 0;
2445fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis}
2446fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2447d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenekvoid clang_toggleCrashRecovery(unsigned isEnabled) {
2448d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  if (isEnabled)
2449d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Enable();
2450d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  else
2451d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Disable();
2452d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek}
2453d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek
24549ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
2455a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                              const char *ast_filename) {
24562b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
24572b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    return 0;
2458f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
24597d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2460389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOptions FileSystemOpts;
2461389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOpts.WorkingDir = CXXIdx->getWorkingDirectory();
24620d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2463c93dc7889644293e318e19d82830ea2acc45b678Dylan Noblesmith  IntrusiveRefCntPtr<DiagnosticsEngine> Diags;
2464a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *TU = ASTUnit::LoadFromASTFile(ast_filename, Diags, FileSystemOpts,
2465a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  CXXIdx->getOnlyLocalDecls(),
2466bef35c91b594f66216f4aab303b71a6c5ab7abcfArgyrios Kyrtzidis                                  0, 0,
2467bef35c91b594f66216f4aab303b71a6c5ab7abcfArgyrios Kyrtzidis                                  /*CaptureDiagnostics=*/true,
2468ff398965a5abfaf5bc47bc022876f56a28e5b9a7Argyrios Kyrtzidis                                  /*AllowPCHWithCompilerErrors=*/true,
2469ff398965a5abfaf5bc47bc022876f56a28e5b9a7Argyrios Kyrtzidis                                  /*UserFilesAreVolatile=*/true);
2470fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  return MakeCXTranslationUnit(CXXIdx, TU);
2471600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2472600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2473b1c031be513705d924038f497279b9b599868ba1Douglas Gregorunsigned clang_defaultEditingTranslationUnitOptions() {
24742a2c50b330e7754499f42173616a36865b5f313bDouglas Gregor  return CXTranslationUnit_PrecompiledPreamble |
2475b5af843a20e237ad1a13ad66a867e200695b8c8eDouglas Gregor         CXTranslationUnit_CacheCompletionResults;
2476b1c031be513705d924038f497279b9b599868ba1Douglas Gregor}
2477b1c031be513705d924038f497279b9b599868ba1Douglas Gregor
24789ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit
24799ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarclang_createTranslationUnitFromSourceFile(CXIndex CIdx,
24809ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          const char *source_filename,
24819ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          int num_command_line_args,
24822ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                          const char * const *command_line_args,
24834db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          unsigned num_unsaved_files,
2484a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                          struct CXUnsavedFile *unsaved_files) {
2485e1d4330adaaa7faf093e725c9c993207eb2d778aArgyrios Kyrtzidis  unsigned Options = CXTranslationUnit_DetailedPreprocessingRecord;
24865a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor  return clang_parseTranslationUnit(CIdx, source_filename,
24875a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    command_line_args, num_command_line_args,
24885a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    unsaved_files, num_unsaved_files,
2489dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor                                    Options);
24905a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor}
2491e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis
249219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbarstruct ParseTranslationUnitInfo {
249319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx;
249419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename;
24952ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char *const *command_line_args;
249619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args;
249719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
249819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files;
249919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options;
250019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXTranslationUnit result;
250119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar};
2502b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_parseTranslationUnit_Impl(void *UserData) {
250319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo *PTUI =
250419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    static_cast<ParseTranslationUnitInfo*>(UserData);
250519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx = PTUI->CIdx;
250619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename = PTUI->source_filename;
25072ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char * const *command_line_args = PTUI->command_line_args;
250819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args = PTUI->num_command_line_args;
250919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files = PTUI->unsaved_files;
251019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files = PTUI->num_unsaved_files;
251119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options = PTUI->options;
251219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  PTUI->result = 0;
25135a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor
25142b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
251519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return;
2516f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2517e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2518e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff
2519fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
252081b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
2521fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
252244c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  bool PrecompilePreamble = options & CXTranslationUnit_PrecompiledPreamble;
2523467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor  // FIXME: Add a flag for modules.
2524467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor  TranslationUnitKind TUKind
2525467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor    = (options & CXTranslationUnit_Incomplete)? TU_Prefix : TU_Complete;
252687c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  bool CacheCodeCompetionResults
252787c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    = options & CXTranslationUnit_CacheCompletionResults;
2528d99ef536b241071b6f4c01db6525dc03242ac30bDmitri Gribenko  bool IncludeBriefCommentsInCodeCompletion
2529d99ef536b241071b6f4c01db6525dc03242ac30bDmitri Gribenko    = options & CXTranslationUnit_IncludeBriefCommentsInCodeCompletion;
25306a91d385618ea4d28236c496f540a26877c95525Erik Verbruggen  bool SkipFunctionBodies = options & CXTranslationUnit_SkipFunctionBodies;
25316a91d385618ea4d28236c496f540a26877c95525Erik Verbruggen
25325352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  // Configure the diagnostics.
25335352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  DiagnosticOptions DiagOpts;
2534c93dc7889644293e318e19d82830ea2acc45b678Dylan Noblesmith  IntrusiveRefCntPtr<DiagnosticsEngine>
253525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Diags(CompilerInstance::createDiagnostics(DiagOpts, num_command_line_args,
253625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                                command_line_args));
253725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
253825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
2539d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie  llvm::CrashRecoveryContextCleanupRegistrar<DiagnosticsEngine,
2540d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie    llvm::CrashRecoveryContextReleaseRefCleanup<DiagnosticsEngine> >
254125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    DiagCleanup(Diags.getPtr());
254225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
25431e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<std::vector<ASTUnit::RemappedFile> >
254425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
254525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
254625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
254725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
254825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
2549f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
25504db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
25515f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2552f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    const llvm::MemoryBuffer *Buffer
2553a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
255425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
255525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
25564db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  }
2557f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
25581e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<std::vector<const char *> >
255925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args(new std::vector<const char*>());
256025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
256125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this method.
256225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<std::vector<const char*> >
256325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    ArgsCleanup(Args.get());
256425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
256552ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // Since the Clang C library is primarily used by batch tools dealing with
256652ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // (often very broken) source code, where spell-checking can have a
256752ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // significant negative impact on performance (particularly when
256852ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // precompiled headers are involved), we disable it by default.
2569b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  // Only do this if we haven't found a spell-checking-related argument.
2570b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  bool FoundSpellCheckingArgument = false;
2571b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  for (int I = 0; I != num_command_line_args; ++I) {
2572b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (strcmp(command_line_args[I], "-fno-spell-checking") == 0 ||
2573b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        strcmp(command_line_args[I], "-fspell-checking") == 0) {
2574b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      FoundSpellCheckingArgument = true;
2575b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      break;
2576e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    }
2577b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  }
2578b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  if (!FoundSpellCheckingArgument)
257925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-fno-spell-checking");
2580b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
258125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  Args->insert(Args->end(), command_line_args,
258225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek               command_line_args + num_command_line_args);
2583d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2584c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // The 'source_filename' argument is optional.  If the caller does not
2585c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // specify it then it is assumed that the source file is specified
2586c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // in the actual argument list.
2587c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // Put the source file after command_line_args otherwise if '-x' flag is
2588c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // present it will be unused.
2589c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  if (source_filename)
259025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back(source_filename);
2591c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis
259244c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  // Do we need the detailed preprocessing record?
259344c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  if (options & CXTranslationUnit_DetailedPreprocessingRecord) {
259425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-Xclang");
259525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-detailed-preprocessing-record");
259644c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  }
259744c181aec37789f25f6c15543c164416f72e562aDouglas Gregor
2598026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  unsigned NumErrors = Diags->getClient()->getNumErrors();
2599e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis  OwningPtr<ASTUnit> ErrUnit;
26001e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<ASTUnit> Unit(
26014ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek    ASTUnit::LoadFromCommandLine(Args->size() ? &(*Args)[0] : 0
26024ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 /* vector::data() not portable */,
26034ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 Args->size() ? (&(*Args)[0] + Args->size()) :0,
2604b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 Diags,
2605b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getClangResourcesPath(),
2606b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getOnlyLocalDecls(),
2607e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                                 /*CaptureDiagnostics=*/true,
26084ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 RemappedFiles->size() ? &(*RemappedFiles)[0]:0,
260925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                 RemappedFiles->size(),
2610299a4a967b02c9f0d0d94ad8560e3ced893f9116Argyrios Kyrtzidis                                 /*RemappedFilesKeepOriginalName=*/true,
2611b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 PrecompilePreamble,
2612467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor                                 TUKind,
2613bef35c91b594f66216f4aab303b71a6c5ab7abcfArgyrios Kyrtzidis                                 CacheCodeCompetionResults,
2614d99ef536b241071b6f4c01db6525dc03242ac30bDmitri Gribenko                                 IncludeBriefCommentsInCodeCompletion,
2615e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis                                 /*AllowPCHWithCompilerErrors=*/true,
26166a91d385618ea4d28236c496f540a26877c95525Erik Verbruggen                                 SkipFunctionBodies,
2617ff398965a5abfaf5bc47bc022876f56a28e5b9a7Argyrios Kyrtzidis                                 /*UserFilesAreVolatile=*/true,
2618e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis                                 &ErrUnit));
2619b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
2620026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  if (NumErrors != Diags->getClient()->getNumErrors()) {
2621b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    // Make sure to check that 'Unit' is non-NULL.
2622e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis    if (CXXIdx->getDisplayDiagnostics())
2623e722ed6f5464232e23be52f4976312ef526fae99Argyrios Kyrtzidis      printDiagsToStderr(Unit ? Unit.get() : ErrUnit.get());
2624a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  }
2625d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2626fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  PTUI->result = MakeCXTranslationUnit(CXXIdx, Unit.take());
262719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar}
262819ffd492a31a25fb691098bf79f317e5f3edf177Daniel DunbarCXTranslationUnit clang_parseTranslationUnit(CXIndex CIdx,
262919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             const char *source_filename,
26302ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                         const char * const *command_line_args,
263119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             int num_command_line_args,
26329e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                            struct CXUnsavedFile *unsaved_files,
263319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned num_unsaved_files,
263419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned options) {
263519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo PTUI = { CIdx, source_filename, command_line_args,
26369e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_command_line_args, unsaved_files,
26379e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_unsaved_files, options, 0 };
263819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  llvm::CrashRecoveryContext CRC;
263919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
2640bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_parseTranslationUnit_Impl, &PTUI)) {
264160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "libclang: crash detected during parsing: {\n");
264260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'source_filename' : '%s'\n", source_filename);
264360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'command_line_args' : [");
264460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (int i = 0; i != num_command_line_args; ++i) {
264560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
264660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
264760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "'%s'", command_line_args[i]);
264860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
264960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
265060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'unsaved_files' : [");
265160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (unsigned i = 0; i != num_unsaved_files; ++i) {
265260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
265360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
265460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "('%s', '...', %ld)", unsaved_files[i].Filename,
265560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar              unsaved_files[i].Length);
265660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
265760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
265860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'options' : %d,\n", options);
265960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "}\n");
266060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar
266119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return 0;
26626df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
26636df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(PTUI.result);
266419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  }
26656df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
266619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  return PTUI.result;
26675b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff}
26685b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff
26691999844e7a18786e61e619e1dc6c789827541863Douglas Gregorunsigned clang_defaultSaveOptions(CXTranslationUnit TU) {
26701999844e7a18786e61e619e1dc6c789827541863Douglas Gregor  return CXSaveTranslationUnit_None;
26711999844e7a18786e61e619e1dc6c789827541863Douglas Gregor}
2672142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2673142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidisnamespace {
2674142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2675142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidisstruct SaveTranslationUnitInfo {
2676142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  CXTranslationUnit TU;
2677142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  const char *FileName;
2678142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  unsigned options;
2679142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  CXSaveError result;
2680142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis};
2681142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2682142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis}
2683142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2684142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidisstatic void clang_saveTranslationUnit_Impl(void *UserData) {
2685142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  SaveTranslationUnitInfo *STUI =
2686142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    static_cast<SaveTranslationUnitInfo*>(UserData);
2687142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2688fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  CIndexer *CXXIdx = (CIndexer*)STUI->TU->CIdx;
2689fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
269081b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
2691fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2692142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  STUI->result = static_cast<ASTUnit *>(STUI->TU->TUData)->Save(STUI->FileName);
2693142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis}
2694142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
26951999844e7a18786e61e619e1dc6c789827541863Douglas Gregorint clang_saveTranslationUnit(CXTranslationUnit TU, const char *FileName,
26961999844e7a18786e61e619e1dc6c789827541863Douglas Gregor                              unsigned options) {
26977ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  if (!TU)
269839c411fa229b2a6747b92f945d1702ee674d3470Douglas Gregor    return CXSaveError_InvalidTU;
2699142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2700142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
2701142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2702374a00bcc6e26b4fc3cd1d378a5d056c4c7d618eArgyrios Kyrtzidis  if (!CXXUnit->hasSema())
2703374a00bcc6e26b4fc3cd1d378a5d056c4c7d618eArgyrios Kyrtzidis    return CXSaveError_InvalidTU;
2704142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2705142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  SaveTranslationUnitInfo STUI = { TU, FileName, options, CXSaveError_None };
2706142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2707142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  if (!CXXUnit->getDiagnostics().hasUnrecoverableErrorOccurred() ||
2708142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis      getenv("LIBCLANG_NOTHREADS")) {
2709142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    clang_saveTranslationUnit_Impl(&STUI);
2710142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2711142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    if (getenv("LIBCLANG_RESOURCE_USAGE"))
2712142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis      PrintLibclangResourceUsage(TU);
2713142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2714142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    return STUI.result;
2715142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  }
2716142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2717142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  // We have an AST that has invalid nodes due to compiler errors.
2718142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  // Use a crash recovery thread for protection.
2719142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2720142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  llvm::CrashRecoveryContext CRC;
2721142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2722142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  if (!RunSafely(CRC, clang_saveTranslationUnit_Impl, &STUI)) {
2723142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "libclang: crash detected during AST saving: {\n");
2724142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "  'filename' : '%s'\n", FileName);
2725142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "  'options' : %d,\n", options);
2726142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "}\n");
2727142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2728142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    return CXSaveError_Unknown;
2729142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2730142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
27316df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(TU);
2732142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  }
2733142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2734142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  return STUI.result;
27357ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor}
273619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
27379ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
2738ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  if (CTUnit) {
2739ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // If the translation unit has been marked as unsafe to free, just discard
2740ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // it.
2741a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    if (static_cast<ASTUnit *>(CTUnit->TUData)->isUnsafeToFree())
2742ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar      return;
2743ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2744a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete static_cast<ASTUnit *>(CTUnit->TUData);
2745a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    disposeCXStringPool(CTUnit->StringPool);
2746153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek    delete static_cast<CXDiagnosticSetImpl *>(CTUnit->Diagnostics);
2747bbf66ca1dad17773cc682d69b8482c4e179aeaebTed Kremenek    disposeOverridenCXCursorsPool(CTUnit->OverridenCursorsPool);
2748a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete CTUnit;
2749ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  }
27502bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
27510d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2752e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregorunsigned clang_defaultReparseOptions(CXTranslationUnit TU) {
2753e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor  return CXReparse_None;
2754e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor}
2755e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor
2756ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarstruct ReparseTranslationUnitInfo {
2757ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU;
2758ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files;
2759ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
2760ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options;
2761ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  int result;
2762ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar};
2763593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2764b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_reparseTranslationUnit_Impl(void *UserData) {
2765ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo *RTUI =
2766ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    static_cast<ReparseTranslationUnitInfo*>(UserData);
2767ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU = RTUI->TU;
2768153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek
2769153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek  // Reset the associated diagnostics.
2770153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek  delete static_cast<CXDiagnosticSetImpl*>(TU->Diagnostics);
2771153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek  TU->Diagnostics = 0;
2772153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek
2773ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files = RTUI->num_unsaved_files;
2774ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files = RTUI->unsaved_files;
2775ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options = RTUI->options;
2776ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  (void) options;
2777ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  RTUI->result = 1;
2778ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2779abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  if (!TU)
2780ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return;
2781593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2782fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  CIndexer *CXXIdx = (CIndexer*)TU->CIdx;
2783fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
278481b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
2785fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2786a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
2787593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2788abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
27891e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<std::vector<ASTUnit::RemappedFile> >
279025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
279125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
279225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
279325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
279425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
279525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
2796abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
27975f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2798abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor    const llvm::MemoryBuffer *Buffer
27991abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
280025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
280125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
2802abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  }
2803abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
28044ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek  if (!CXXUnit->Reparse(RemappedFiles->size() ? &(*RemappedFiles)[0] : 0,
28054ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                        RemappedFiles->size()))
2806593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor    RTUI->result = 0;
2807abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor}
2808593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2809ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarint clang_reparseTranslationUnit(CXTranslationUnit TU,
2810ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned num_unsaved_files,
2811ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 struct CXUnsavedFile *unsaved_files,
2812ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned options) {
2813ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo RTUI = { TU, num_unsaved_files, unsaved_files,
2814ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                      options, 0 };
28158c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis
2816e7de9b4a1f4a15620ab15bc8159018df7d54080aArgyrios Kyrtzidis  if (getenv("LIBCLANG_NOTHREADS")) {
28178c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis    clang_reparseTranslationUnit_Impl(&RTUI);
28188c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis    return RTUI.result;
28198c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis  }
28208c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis
2821ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  llvm::CrashRecoveryContext CRC;
2822ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2823bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_reparseTranslationUnit_Impl, &RTUI)) {
2824b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbar    fprintf(stderr, "libclang: crash detected during reparsing\n");
2825a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    static_cast<ASTUnit *>(TU->TUData)->setUnsafeToFree(true);
2826ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return 1;
28276df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  } else if (getenv("LIBCLANG_RESOURCE_USAGE"))
28286df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(TU);
28291dfb26af4d6aa4f7818e256659a79f1ec2cba784Ted Kremenek
2830ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  return RTUI.result;
2831ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar}
2832ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2833df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor
28349ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
28352b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CTUnit)
2836ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2837f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2838a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit->TUData);
2839ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(CXXUnit->getOriginalSourceFileName(), true);
2840af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff}
28411eb79b58e56b99cf557d5d353586a10c5360364dDaniel Dunbar
28427eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas GregorCXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
28438e5900c8e9bd32bcc385124f564f3d38a11d0e28Douglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit*>(TU->TUData);
28448e5900c8e9bd32bcc385124f564f3d38a11d0e28Douglas Gregor  return MakeCXCursor(CXXUnit->getASTContext().getTranslationUnitDecl(), TU);
28457eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
28467eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
2847fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2848600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2849fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2850fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXFile Operations.
2851fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2852fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2853fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
285474844072411bae91d5dbb89955d200cbe1e0a1c8Ted KremenekCXString clang_getFileName(CXFile SFile) {
285598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
2856a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return createCXString((const char*)NULL);
2857f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
285888145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
285974844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek  return createCXString(FEnt->getName());
286088145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
286188145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff
286288145034694ed5267fa6fa5febc54fadc02bd479Steve Narofftime_t clang_getFileTime(CXFile SFile) {
286398258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
286498258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor    return 0;
2865f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
286688145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
286788145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  return FEnt->getModificationTime();
2868ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff}
2869f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2870b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXFile clang_getFile(CXTranslationUnit tu, const char *file_name) {
2871b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!tu)
2872b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return 0;
2873f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2874a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2875f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2876b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  FileManager &FMgr = CXXUnit->getFileManager();
287739b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattner  return const_cast<FileEntry *>(FMgr.getFile(file_name));
2878b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2879f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2880dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregorunsigned clang_isFileMultipleIncludeGuarded(CXTranslationUnit tu, CXFile file) {
2881dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  if (!tu || !file)
2882dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor    return 0;
2883dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor
2884dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2885dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  FileEntry *FEnt = static_cast<FileEntry *>(file);
2886dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  return CXXUnit->getPreprocessor().getHeaderSearchInfo()
2887dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor                                          .isFileMultipleIncludeGuarded(FEnt);
2888dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor}
2889dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor
2890fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2891fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2892fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2893fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXCursor Operations.
2894fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2895fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2896fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekstatic Decl *getDeclFromExpr(Stmt *E) {
2897c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis  if (ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
2898db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return getDeclFromExpr(CE->getSubExpr());
2899db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2900fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
2901fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RefExpr->getDecl();
2902fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (MemberExpr *ME = dyn_cast<MemberExpr>(E))
2903fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return ME->getMemberDecl();
2904fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
2905fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RE->getDecl();
2906b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis  if (ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(E)) {
2907b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    if (PRE->isExplicitProperty())
2908b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis      return PRE->getExplicitProperty();
2909b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // It could be messaging both getter and setter as in:
2910b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // ++myobj.myprop;
2911b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // in which case prefer to associate the setter since it is less obvious
2912b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // from inspecting the source that the setter is going to get called.
2913b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    if (PRE->isMessagingSetter())
2914b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis      return PRE->getImplicitPropertySetter();
2915b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    return PRE->getImplicitPropertyGetter();
2916b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis  }
29174b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (PseudoObjectExpr *POE = dyn_cast<PseudoObjectExpr>(E))
29184b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return getDeclFromExpr(POE->getSyntacticForm());
29194b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (OpaqueValueExpr *OVE = dyn_cast<OpaqueValueExpr>(E))
29204b9c2d235fb9449e249d74f48ecfec601650de93John McCall    if (Expr *Src = OVE->getSourceExpr())
29214b9c2d235fb9449e249d74f48ecfec601650de93John McCall      return getDeclFromExpr(Src);
2922db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2923fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (CallExpr *CE = dyn_cast<CallExpr>(E))
2924fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return getDeclFromExpr(CE->getCallee());
29255f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  if (CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E))
292693798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    if (!CE->isElidable())
292793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CE->getConstructor();
2928fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
2929fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return OME->getMethodDecl();
2930f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2931db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCProtocolExpr *PE = dyn_cast<ObjCProtocolExpr>(E))
2932db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return PE->getProtocol();
2933c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor  if (SubstNonTypeTemplateParmPackExpr *NTTP
2934c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor                              = dyn_cast<SubstNonTypeTemplateParmPackExpr>(E))
2935c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor    return NTTP->getParameterPack();
293694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
293794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    if (isa<NonTypeTemplateParmDecl>(SizeOfPack->getPack()) ||
293894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        isa<ParmVarDecl>(SizeOfPack->getPack()))
293994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      return SizeOfPack->getPack();
2940db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2941fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  return 0;
2942fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek}
2943ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff
2944c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbarstatic SourceLocation getLocationFromExpr(Expr *E) {
2945c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis  if (ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
2946c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis    return getLocationFromExpr(CE->getSubExpr());
2947c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis
2948c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
2949c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return /*FIXME:*/Msg->getLeftLoc();
2950c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
2951c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return DRE->getLocation();
2952c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (MemberExpr *Member = dyn_cast<MemberExpr>(E))
2953c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Member->getMemberLoc();
2954c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
2955c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Ivar->getLocation();
295694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
295794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    return SizeOfPack->getPackLoc();
2958d0469525581a851b68f5b4f960ee4190dcc7c932Argyrios Kyrtzidis  if (ObjCPropertyRefExpr *PropRef = dyn_cast<ObjCPropertyRefExpr>(E))
2959d0469525581a851b68f5b4f960ee4190dcc7c932Argyrios Kyrtzidis    return PropRef->getLocation();
296094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
2961c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  return E->getLocStart();
2962c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar}
2963c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar
2964fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
2965f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2966f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekunsigned clang_visitChildren(CXCursor parent,
2967b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXCursorVisitor visitor,
2968b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXClientData client_data) {
2969f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  CursorVisitor CursorVis(getCursorTU(parent), visitor, client_data,
2970f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                          /*VisitPreprocessorLast=*/false);
2971b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return CursorVis.VisitChildren(parent);
2972b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
2973b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
29743387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#ifndef __has_feature
29753387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#define __has_feature(x) 0
29763387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
29773387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#if __has_feature(blocks)
29783387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef enum CXChildVisitResult
29793387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall     (^CXCursorVisitorBlock)(CXCursor cursor, CXCursor parent);
29803387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29813387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
29823387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
29833387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
29843387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block(cursor, parent);
29853387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
29863387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#else
29873387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// If we are compiled with a compiler that doesn't have native blocks support,
29883387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// define and call the block manually, so the
29893387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef struct _CXChildVisitResult
29903387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall{
29913387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	void *isa;
29923387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int flags;
29933387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int reserved;
29949e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar	enum CXChildVisitResult(*invoke)(struct _CXChildVisitResult*, CXCursor,
29959e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                         CXCursor);
29963387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall} *CXCursorVisitorBlock;
29973387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29983387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
29993387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
30003387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
30013387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block->invoke(block, cursor, parent);
30023387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30033387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
30043387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30053387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30069e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbarunsigned clang_visitChildrenWithBlock(CXCursor parent,
30079e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                      CXCursorVisitorBlock block) {
30083387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return clang_visitChildren(parent, visitWithBlock, block);
30093387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30103387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
301178205d4bada39d95097e766af9eb30cdd0159461Douglas Gregorstatic CXString getDeclSpelling(Decl *D) {
301216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  if (!D)
301316ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    return createCXString("");
301416ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
301516ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  NamedDecl *ND = dyn_cast<NamedDecl>(D);
3016e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  if (!ND) {
30175f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    if (ObjCPropertyImplDecl *PropImpl =dyn_cast<ObjCPropertyImplDecl>(D))
3018e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
3019e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return createCXString(Property->getIdentifier()->getName());
3020e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
3021ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
3022e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  }
3023e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
302478205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
3025ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(OMD->getSelector().getAsString());
3026f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
302778205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
302878205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // No, this isn't the same as the code below. getIdentifier() is non-virtual
302978205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // and returns different names. NamedDecl returns the class name and
303078205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // ObjCCategoryImplDecl returns the category name.
3031ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(CIMP->getIdentifier()->getNameStart());
3032f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30330a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  if (isa<UsingDirectiveDecl>(D))
30340a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("");
30350a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
303636d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith  SmallString<1024> S;
303750aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::raw_svector_ostream os(S);
303850aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  ND->printName(os);
303950aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek
304050aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  return createCXString(os.str());
304178205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor}
3042f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30439ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getCursorSpelling(CXCursor C) {
30447eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  if (clang_isTranslationUnit(C.kind))
3045a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return clang_getTranslationUnitSpelling(
3046a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                            static_cast<CXTranslationUnit>(C.data[2]));
30477eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
3048f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  if (clang_isReference(C.kind)) {
3049f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    switch (C.kind) {
3050acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCSuperClassRef: {
30512e331b938b38057e333fab0ba841130ea8467794Douglas Gregor      ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
3052ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Super->getIdentifier()->getNameStart());
3053acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3054acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCClassRef: {
30551adb082a709f7b588f03672999294e061234b2cfDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
3056ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Class->getIdentifier()->getNameStart());
3057acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3058acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCProtocolRef: {
305978db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor      ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
3060f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      assert(OID && "getCursorSpelling(): Missing protocol decl");
3061ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(OID->getIdentifier()->getNameStart());
3062acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
30633064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
30643064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = getCursorCXXBaseSpecifier(C);
30653064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return createCXString(B->getType().getAsString());
30663064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
30677d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    case CXCursor_TypeRef: {
30687d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      TypeDecl *Type = getCursorTypeRef(C).first;
30697d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      assert(Type && "Missing type decl");
30707d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3071ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(getCursorContext(C).getTypeDeclType(Type).
3072ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                              getAsString());
30737d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
30740b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
30750b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      TemplateDecl *Template = getCursorTemplateRef(C).first;
30766931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(Template && "Missing template decl");
30770b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
30780b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString(Template->getNameAsString());
30790b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
30806931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
30816931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
30826931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      NamedDecl *NS = getCursorNamespaceRef(C).first;
30836931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(NS && "Missing namespace decl");
30846931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
30856931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return createCXString(NS->getNameAsString());
30866931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
30877d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3088a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3089a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      FieldDecl *Field = getCursorMemberRef(C).first;
3090a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      assert(Field && "Missing member decl");
3091a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3092a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return createCXString(Field->getNameAsString());
3093a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3094a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
309536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
309636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      LabelStmt *Label = getCursorLabelRef(C).first;
309736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      assert(Label && "Missing label");
309836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
3099ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
310036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
310136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31021f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef: {
31031f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
31041f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Decl *D = Storage.dyn_cast<Decl *>()) {
31051f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        if (NamedDecl *ND = dyn_cast<NamedDecl>(D))
31061f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor          return createCXString(ND->getNameAsString());
31071f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
31081f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      }
31091f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
31101f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString(E->getName().getAsString());
31111f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedTemplateStorage *Ovl
31121f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        = Storage.get<OverloadedTemplateStorage*>();
31131f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Ovl->size() == 0)
31141f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
31151f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return createCXString((*Ovl->begin())->getNameAsString());
31161f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    }
31171f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3118011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef: {
3119011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      VarDecl *Var = getCursorVariableRef(C).first;
3120011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      assert(Var && "Missing variable decl");
3121011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
3122011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return createCXString(Var->getNameAsString());
3123011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    }
3124011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
3125acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    default:
3126ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString("<not implemented>");
3127f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    }
3128f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  }
312997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
313097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
313197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    Decl *D = getDeclFromExpr(getCursorExpr(C));
313297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
313378205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor      return getDeclSpelling(D);
3134ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
313597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
313697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
313736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
313836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
313936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
3140ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
314136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
314236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("");
314336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
314436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31459b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion)
31469e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth    return createCXString(getCursorMacroExpansion(C)->getName()
31474ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor                                                           ->getNameStart());
31484ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor
3149572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
3150572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString(getCursorMacroDefinition(C)->getName()
3151572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor                                                           ->getNameStart());
3152572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3153ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective)
3154ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString(getCursorInclusionDirective(C)->getFileName());
3155ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
315660cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor  if (clang_isDeclaration(C.kind))
315760cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor    return getDeclSpelling(getCursorDecl(C));
3158e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
31595f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  if (C.kind == CXCursor_AnnotateAttr) {
31605f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    AnnotateAttr *AA = cast<AnnotateAttr>(cxcursor::getCursorAttr(C));
31615f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    return createCXString(AA->getAnnotation());
31625f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  }
31635f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen
316484b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis  if (C.kind == CXCursor_AsmLabelAttr) {
316584b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis    AsmLabelAttr *AA = cast<AsmLabelAttr>(cxcursor::getCursorAttr(C));
316684b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis    return createCXString(AA->getLabel());
316784b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis  }
316884b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis
3169ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString("");
3170f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3171f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
3172ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios KyrtzidisCXSourceRange clang_Cursor_getSpellingNameRange(CXCursor C,
3173ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis                                                unsigned pieceIndex,
3174ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis                                                unsigned options) {
3175ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  if (clang_Cursor_isNull(C))
3176ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    return clang_getNullRange();
3177ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3178ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  ASTContext &Ctx = getCursorContext(C);
3179ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3180ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  if (clang_isStatement(C.kind)) {
3181ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    Stmt *S = getCursorStmt(C);
3182ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    if (LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S)) {
3183ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      if (pieceIndex > 0)
3184ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis        return clang_getNullRange();
3185ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      return cxloc::translateSourceRange(Ctx, Label->getIdentLoc());
3186ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    }
3187ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3188ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    return clang_getNullRange();
3189ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  }
3190ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3191ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  if (C.kind == CXCursor_ObjCMessageExpr) {
3192ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    if (ObjCMessageExpr *
3193ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis          ME = dyn_cast_or_null<ObjCMessageExpr>(getCursorExpr(C))) {
3194ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      if (pieceIndex >= ME->getNumSelectorLocs())
3195ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis        return clang_getNullRange();
3196ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      return cxloc::translateSourceRange(Ctx, ME->getSelectorLoc(pieceIndex));
3197ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    }
3198ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  }
3199ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3200ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  if (C.kind == CXCursor_ObjCInstanceMethodDecl ||
3201ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      C.kind == CXCursor_ObjCClassMethodDecl) {
3202ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    if (ObjCMethodDecl *
3203ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis          MD = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(C))) {
3204ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      if (pieceIndex >= MD->getNumSelectorLocs())
3205ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis        return clang_getNullRange();
3206ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis      return cxloc::translateSourceRange(Ctx, MD->getSelectorLoc(pieceIndex));
3207ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    }
3208ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  }
3209ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
32109482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis  if (C.kind == CXCursor_ObjCCategoryDecl ||
32119482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis      C.kind == CXCursor_ObjCCategoryImplDecl) {
32129482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis    if (pieceIndex > 0)
32139482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis      return clang_getNullRange();
32149482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis    if (ObjCCategoryDecl *
32159482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis          CD = dyn_cast_or_null<ObjCCategoryDecl>(getCursorDecl(C)))
32169482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis      return cxloc::translateSourceRange(Ctx, CD->getCategoryNameLoc());
32179482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis    if (ObjCCategoryImplDecl *
32189482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis          CID = dyn_cast_or_null<ObjCCategoryImplDecl>(getCursorDecl(C)))
32199482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis      return cxloc::translateSourceRange(Ctx, CID->getCategoryNameLoc());
32209482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis  }
32219482a185d34d8b0f6d788c44e2c128991622c0adArgyrios Kyrtzidis
3222ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // FIXME: A CXCursor_InclusionDirective should give the location of the
3223ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // filename, but we don't keep track of this.
3224ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3225ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // FIXME: A CXCursor_AnnotateAttr should give the location of the annotation
3226ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // but we don't keep track of this.
3227ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3228ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // FIXME: A CXCursor_AsmLabelAttr should give the location of the label
3229ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // but we don't keep track of this.
3230ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3231ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  // Default handling, give the location of the cursor.
3232ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3233ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  if (pieceIndex > 0)
3234ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis    return clang_getNullRange();
3235ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3236ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  CXSourceLocation CXLoc = clang_getCursorLocation(C);
3237ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  SourceLocation Loc = cxloc::translateSourceLocation(CXLoc);
3238ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis  return cxloc::translateSourceRange(Ctx, Loc);
3239ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis}
3240ba1da14e79c1ecd49306e5175b69baa1e7ed4293Argyrios Kyrtzidis
3241358559d8d7b458c5f64941842383a16e61f0828dDouglas GregorCXString clang_getCursorDisplayName(CXCursor C) {
3242358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!clang_isDeclaration(C.kind))
3243358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return clang_getCursorSpelling(C);
3244358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3245358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  Decl *D = getCursorDecl(C);
3246358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!D)
3247358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString("");
3248358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
324930c42404202d2e2512e51efc6066bd614cfdb5a4Douglas Gregor  PrintingPolicy Policy = getCursorContext(C).getPrintingPolicy();
3250358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(D))
3251358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    D = FunTmpl->getTemplatedDecl();
3252358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3253358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
325436d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith    SmallString<64> Str;
3255358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3256a59d20b135bfde058a5a69045bab5ec4e2553f74Benjamin Kramer    OS << *Function;
3257358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->getPrimaryTemplate())
3258358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "<>";
3259358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "(";
3260358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Function->getNumParams(); I != N; ++I) {
3261358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3262358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3263358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << Function->getParamDecl(I)->getType().getAsString(Policy);
3264358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3265358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3266358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->isVariadic()) {
3267358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Function->getNumParams())
3268358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3269358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "...";
3270358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3271358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ")";
3272358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3273358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3274358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3275358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) {
327636d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith    SmallString<64> Str;
3277358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3278a59d20b135bfde058a5a69045bab5ec4e2553f74Benjamin Kramer    OS << *ClassTemplate;
3279358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "<";
3280358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    TemplateParameterList *Params = ClassTemplate->getTemplateParameters();
3281358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Params->size(); I != N; ++I) {
3282358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3283358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3284358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3285358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      NamedDecl *Param = Params->getParam(I);
3286358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Param->getIdentifier()) {
3287358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << Param->getIdentifier()->getName();
3288358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        continue;
3289358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      }
3290358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3291358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // There is no parameter name, which makes this tricky. Try to come up
3292358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // with something useful that isn't too long.
3293358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
3294358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << (TTP->wasDeclaredWithTypename()? "typename" : "class");
3295358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else if (NonTypeTemplateParmDecl *NTTP
3296358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                    = dyn_cast<NonTypeTemplateParmDecl>(Param))
3297358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << NTTP->getType().getAsString(Policy);
3298358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else
3299358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << "template<...> class";
3300358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3301358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3302358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ">";
3303358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3304358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3305358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3306358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateSpecializationDecl *ClassSpec
3307358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                              = dyn_cast<ClassTemplateSpecializationDecl>(D)) {
3308358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    // If the type was explicitly written, use that.
3309358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (TypeSourceInfo *TSInfo = ClassSpec->getTypeAsWritten())
3310358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      return createCXString(TSInfo->getType().getAsString(Policy));
3311358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
331236d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith    SmallString<64> Str;
3313358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3314a59d20b135bfde058a5a69045bab5ec4e2553f74Benjamin Kramer    OS << *ClassSpec;
3315358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << TemplateSpecializationType::PrintTemplateArgumentList(
3316910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().data(),
3317910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().size(),
3318358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                                                Policy);
3319358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3320358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3321358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3322358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  return clang_getCursorSpelling(C);
3323358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor}
3324358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3325e68fff6fc083c6270d835216a3de0b82c6ef0310Ted KremenekCXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
332689922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  switch (Kind) {
3327e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FunctionDecl:
3328e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FunctionDecl");
3329e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypedefDecl:
3330e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypedefDecl");
3331e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumDecl:
3332e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumDecl");
3333e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumConstantDecl:
3334e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumConstantDecl");
3335e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_StructDecl:
3336e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("StructDecl");
3337e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnionDecl:
3338e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnionDecl");
3339e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ClassDecl:
3340e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ClassDecl");
3341e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FieldDecl:
3342e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FieldDecl");
3343e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_VarDecl:
3344e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("VarDecl");
3345e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ParmDecl:
3346e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ParmDecl");
3347e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInterfaceDecl:
3348e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInterfaceDecl");
3349e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryDecl:
3350e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryDecl");
3351e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolDecl:
3352e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolDecl");
3353e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCPropertyDecl:
3354e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCPropertyDecl");
3355e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCIvarDecl:
3356e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCIvarDecl");
3357e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInstanceMethodDecl:
3358e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInstanceMethodDecl");
3359e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassMethodDecl:
3360e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassMethodDecl");
3361e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCImplementationDecl:
3362e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCImplementationDecl");
3363e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryImplDecl:
3364e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryImplDecl");
33658bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek  case CXCursor_CXXMethod:
33668bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek      return createCXString("CXXMethod");
3367e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedDecl:
3368e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedDecl");
3369e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCSuperClassRef:
3370e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCSuperClassRef");
3371e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolRef:
3372e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolRef");
3373e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassRef:
3374e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassRef");
3375e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypeRef:
3376e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypeRef");
33770b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case CXCursor_TemplateRef:
33780b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString("TemplateRef");
33796931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceRef:
33806931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceRef");
3381a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  case CXCursor_MemberRef:
3382a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return createCXString("MemberRef");
338336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelRef:
338436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("LabelRef");
33851f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case CXCursor_OverloadedDeclRef:
33861f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return createCXString("OverloadedDeclRef");
3387011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  case CXCursor_VariableRef:
3388011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    return createCXString("VariableRef");
338942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IntegerLiteral:
339042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IntegerLiteral");
339142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_FloatingLiteral:
339242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("FloatingLiteral");
339342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ImaginaryLiteral:
339442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ImaginaryLiteral");
339542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_StringLiteral:
339642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("StringLiteral");
339742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CharacterLiteral:
339842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CharacterLiteral");
339942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ParenExpr:
340042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ParenExpr");
340142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnaryOperator:
340242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnaryOperator");
340342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ArraySubscriptExpr:
340442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ArraySubscriptExpr");
340542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_BinaryOperator:
340642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("BinaryOperator");
340742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundAssignOperator:
340842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundAssignOperator");
340942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ConditionalOperator:
341042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ConditionalOperator");
341142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CStyleCastExpr:
341242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CStyleCastExpr");
341342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundLiteralExpr:
341442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundLiteralExpr");
341542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_InitListExpr:
341642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("InitListExpr");
341742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_AddrLabelExpr:
341842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("AddrLabelExpr");
341942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_StmtExpr:
342042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("StmtExpr");
342142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GenericSelectionExpr:
342242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GenericSelectionExpr");
342342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GNUNullExpr:
342442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GNUNullExpr");
342542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXStaticCastExpr:
342642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXStaticCastExpr");
342742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXDynamicCastExpr:
342842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXDynamicCastExpr");
342942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXReinterpretCastExpr:
343042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXReinterpretCastExpr");
343142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXConstCastExpr:
343242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXConstCastExpr");
343342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXFunctionalCastExpr:
343442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXFunctionalCastExpr");
343542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXTypeidExpr:
343642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXTypeidExpr");
343742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXBoolLiteralExpr:
343842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXBoolLiteralExpr");
343942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXNullPtrLiteralExpr:
344042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXNullPtrLiteralExpr");
344142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXThisExpr:
344242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXThisExpr");
344342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXThrowExpr:
344442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXThrowExpr");
344542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXNewExpr:
344642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXNewExpr");
344742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXDeleteExpr:
344842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXDeleteExpr");
344942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnaryExpr:
345042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnaryExpr");
345142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCStringLiteral:
345242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCStringLiteral");
3453b3f7542a950ac0d585a7783e825cfe670e05c553Ted Kremenek  case CXCursor_ObjCBoolLiteralExpr:
3454b3f7542a950ac0d585a7783e825cfe670e05c553Ted Kremenek      return createCXString("ObjCBoolLiteralExpr");
345542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCEncodeExpr:
345642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCEncodeExpr");
345742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCSelectorExpr:
345842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCSelectorExpr");
345942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCProtocolExpr:
346042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCProtocolExpr");
346142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCBridgedCastExpr:
346242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCBridgedCastExpr");
34631ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  case CXCursor_BlockExpr:
34641ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek      return createCXString("BlockExpr");
346542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_PackExpansionExpr:
346642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("PackExpansionExpr");
346742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SizeOfPackExpr:
346842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SizeOfPackExpr");
3469011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  case CXCursor_LambdaExpr:
3470011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    return createCXString("LambdaExpr");
347142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnexposedExpr:
347242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnexposedExpr");
3473e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_DeclRefExpr:
3474e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("DeclRefExpr");
3475e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_MemberRefExpr:
3476e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("MemberRefExpr");
3477e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_CallExpr:
3478e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("CallExpr");
3479e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCMessageExpr:
3480e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCMessageExpr");
3481e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedStmt:
3482e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedStmt");
348342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DeclStmt:
348442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DeclStmt");
348536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelStmt:
348636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return createCXString("LabelStmt");
348742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundStmt:
348842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundStmt");
348942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CaseStmt:
349042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CaseStmt");
349142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DefaultStmt:
349242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DefaultStmt");
349342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IfStmt:
349442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IfStmt");
349542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SwitchStmt:
349642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SwitchStmt");
349742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_WhileStmt:
349842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("WhileStmt");
349942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DoStmt:
350042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DoStmt");
350142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ForStmt:
350242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ForStmt");
350342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GotoStmt:
350442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GotoStmt");
350542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IndirectGotoStmt:
350642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IndirectGotoStmt");
350742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ContinueStmt:
350842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ContinueStmt");
350942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_BreakStmt:
351042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("BreakStmt");
351142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ReturnStmt:
351242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ReturnStmt");
3513df5faf5e7ae6823d0af0b801c4ac26d47f2cee97Chad Rosier  case CXCursor_GCCAsmStmt:
3514df5faf5e7ae6823d0af0b801c4ac26d47f2cee97Chad Rosier      return createCXString("GCCAsmStmt");
35158cd64b4c5553fa6284d248336cb7c82dc960a394Chad Rosier  case CXCursor_MSAsmStmt:
35168cd64b4c5553fa6284d248336cb7c82dc960a394Chad Rosier      return createCXString("MSAsmStmt");
351742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtTryStmt:
351842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtTryStmt");
351942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtCatchStmt:
352042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtCatchStmt");
352142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtFinallyStmt:
352242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtFinallyStmt");
352342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtThrowStmt:
352442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtThrowStmt");
352542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtSynchronizedStmt:
352642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtSynchronizedStmt");
352742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAutoreleasePoolStmt:
352842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAutoreleasePoolStmt");
352942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCForCollectionStmt:
353042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCForCollectionStmt");
353142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXCatchStmt:
353242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXCatchStmt");
353342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXTryStmt:
353442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXTryStmt");
353542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXForRangeStmt:
353642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXForRangeStmt");
353742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHTryStmt:
353842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHTryStmt");
353942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHExceptStmt:
354042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHExceptStmt");
354142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHFinallyStmt:
354242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHFinallyStmt");
354342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_NullStmt:
354442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("NullStmt");
3545e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_InvalidFile:
3546e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("InvalidFile");
3547292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek  case CXCursor_InvalidCode:
3548292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek    return createCXString("InvalidCode");
3549e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NoDeclFound:
3550e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NoDeclFound");
3551e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NotImplemented:
3552e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NotImplemented");
3553e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TranslationUnit:
3554e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TranslationUnit");
3555e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_UnexposedAttr:
3556e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("UnexposedAttr");
3557e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_IBActionAttr:
3558e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("attribute(ibaction)");
35599f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_IBOutletAttr:
35609f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor     return createCXString("attribute(iboutlet)");
3561857e918a8a40deb128840308a318bf623d68295fTed Kremenek  case CXCursor_IBOutletCollectionAttr:
3562857e918a8a40deb128840308a318bf623d68295fTed Kremenek      return createCXString("attribute(iboutletcollection)");
35636639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  case CXCursor_CXXFinalAttr:
35646639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      return createCXString("attribute(final)");
35656639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  case CXCursor_CXXOverrideAttr:
35666639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      return createCXString("attribute(override)");
35675f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  case CXCursor_AnnotateAttr:
35685f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    return createCXString("attribute(annotate)");
356984b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis  case CXCursor_AsmLabelAttr:
357084b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis    return createCXString("asm label");
35719f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_PreprocessingDirective:
35729f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return createCXString("preprocessing directive");
3573572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  case CXCursor_MacroDefinition:
3574572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString("macro definition");
35759b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  case CXCursor_MacroExpansion:
35769b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth    return createCXString("macro expansion");
3577ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  case CXCursor_InclusionDirective:
3578ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString("inclusion directive");
35798f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  case CXCursor_Namespace:
35808f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek    return createCXString("Namespace");
3581a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  case CXCursor_LinkageSpec:
3582a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek    return createCXString("LinkageSpec");
35833064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  case CXCursor_CXXBaseSpecifier:
35843064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    return createCXString("C++ base class specifier");
358501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Constructor:
358601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConstructor");
358701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Destructor:
358801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXDestructor");
358901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_ConversionFunction:
359001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConversion");
3591fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTypeParameter:
3592fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTypeParameter");
3593fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_NonTypeTemplateParameter:
3594fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("NonTypeTemplateParameter");
3595fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTemplateParameter:
3596fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTemplateParameter");
3597fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_FunctionTemplate:
3598fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("FunctionTemplate");
359939d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  case CXCursor_ClassTemplate:
360039d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return createCXString("ClassTemplate");
360174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  case CXCursor_ClassTemplatePartialSpecialization:
360274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return createCXString("ClassTemplatePartialSpecialization");
36036931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceAlias:
36046931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceAlias");
36050a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  case CXCursor_UsingDirective:
36060a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("UsingDirective");
36077e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  case CXCursor_UsingDeclaration:
36087e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor    return createCXString("UsingDeclaration");
3609162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  case CXCursor_TypeAliasDecl:
3610352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("TypeAliasDecl");
3611352697a87bca664356f21a838b162084013625eaDouglas Gregor  case CXCursor_ObjCSynthesizeDecl:
3612352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("ObjCSynthesizeDecl");
3613352697a87bca664356f21a838b162084013625eaDouglas Gregor  case CXCursor_ObjCDynamicDecl:
3614352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("ObjCDynamicDecl");
36152dfdb948bef51a601e763191e4becfe59880d382Argyrios Kyrtzidis  case CXCursor_CXXAccessSpecifier:
36162dfdb948bef51a601e763191e4becfe59880d382Argyrios Kyrtzidis    return createCXString("CXXAccessSpecifier");
361789922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  }
3618e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3619deb06bd3566e18f677e76bc435d478b033fe328bTed Kremenek  llvm_unreachable("Unhandled CXCursorKind");
3620600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
362189922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff
3622064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidisstruct GetCursorData {
3623064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  SourceLocation TokenBeginLoc;
36244b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  bool PointsAtMacroArgExpansion;
3625135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis  bool VisitedObjCPropertyImplDecl;
3626135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis  SourceLocation VisitedDeclaratorDeclStartLoc;
3627064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  CXCursor &BestCursor;
3628064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
36294b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  GetCursorData(SourceManager &SM,
36304b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                SourceLocation tokenBegin, CXCursor &outputCursor)
36314b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    : TokenBeginLoc(tokenBegin), BestCursor(outputCursor) {
36324b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    PointsAtMacroArgExpansion = SM.isMacroArgExpansion(tokenBegin);
3633135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis    VisitedObjCPropertyImplDecl = false;
36344b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  }
3635064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis};
3636064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
36374b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidisstatic enum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
36384b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                                                CXCursor parent,
36394b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                                                CXClientData client_data) {
3640064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  GetCursorData *Data = static_cast<GetCursorData *>(client_data);
3641064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  CXCursor *BestCursor = &Data->BestCursor;
36424b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis
36434b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // If we point inside a macro argument we should provide info of what the
36444b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // token is so use the actual cursor, don't replace it with a macro expansion
36454b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // cursor.
36464b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  if (cursor.kind == CXCursor_MacroExpansion && Data->PointsAtMacroArgExpansion)
36474b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    return CXChildVisit_Recurse;
364865ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis
364965ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis  if (clang_isDeclaration(cursor.kind)) {
365065ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // Avoid having the implicit methods override the property decls.
3651a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis    if (ObjCMethodDecl *MD
3652a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis          = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
365365ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis      if (MD->isImplicit())
365465ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis        return CXChildVisit_Break;
3655a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis
3656a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis    } else if (ObjCInterfaceDecl *ID
3657a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis                 = dyn_cast_or_null<ObjCInterfaceDecl>(getCursorDecl(cursor))) {
3658a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // Check that when we have multiple @class references in the same line,
3659a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // that later ones do not override the previous ones.
3660a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // If we have:
3661a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // @class Foo, Bar;
3662a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // source ranges for both start at '@', so 'Bar' will end up overriding
3663a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      // 'Foo' even though the cursor location was at 'Foo'.
3664a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis      if (BestCursor->kind == CXCursor_ObjCInterfaceDecl ||
3665a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis          BestCursor->kind == CXCursor_ObjCClassRef)
3666a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis        if (ObjCInterfaceDecl *PrevID
3667a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis             = dyn_cast_or_null<ObjCInterfaceDecl>(getCursorDecl(*BestCursor))){
3668a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis         if (PrevID != ID &&
3669a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis             !PrevID->isThisDeclarationADefinition() &&
3670a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis             !ID->isThisDeclarationADefinition())
3671a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis           return CXChildVisit_Break;
3672a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis        }
3673135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis
3674135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis    } else if (DeclaratorDecl *DD
3675135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis                    = dyn_cast_or_null<DeclaratorDecl>(getCursorDecl(cursor))) {
3676135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      SourceLocation StartLoc = DD->getSourceRange().getBegin();
3677135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // Check that when we have multiple declarators in the same line,
3678135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // that later ones do not override the previous ones.
3679135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // If we have:
3680135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // int Foo, Bar;
3681135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // source ranges for both start at 'int', so 'Bar' will end up overriding
3682135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // 'Foo' even though the cursor location was at 'Foo'.
3683135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      if (Data->VisitedDeclaratorDeclStartLoc == StartLoc)
3684135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis        return CXChildVisit_Break;
3685135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      Data->VisitedDeclaratorDeclStartLoc = StartLoc;
3686135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis
3687135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis    } else if (ObjCPropertyImplDecl *PropImp
3688135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis              = dyn_cast_or_null<ObjCPropertyImplDecl>(getCursorDecl(cursor))) {
3689135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      (void)PropImp;
3690135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // Check that when we have multiple @synthesize in the same line,
3691135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // that later ones do not override the previous ones.
3692135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // If we have:
3693135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // @synthesize Foo, Bar;
3694135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // source ranges for both start at '@', so 'Bar' will end up overriding
3695135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      // 'Foo' even though the cursor location was at 'Foo'.
3696135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      if (Data->VisitedObjCPropertyImplDecl)
3697135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis        return CXChildVisit_Break;
3698135bf8ee69e2ae2daea4f713381995028c41e264Argyrios Kyrtzidis      Data->VisitedObjCPropertyImplDecl = true;
3699a9d45a3f423f60fa5d1f977acb5c8df50198f27eArgyrios Kyrtzidis    }
370065ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis  }
3701064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
3702064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  if (clang_isExpression(cursor.kind) &&
3703064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis      clang_isDeclaration(BestCursor->kind)) {
370416ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (Decl *D = getCursorDecl(*BestCursor)) {
370516ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // Avoid having the cursor of an expression replace the declaration cursor
370616ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // when the expression source range overlaps the declaration range.
370716ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // This can happen for C++ constructor expressions whose range generally
370816ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // include the variable declaration, e.g.:
370916ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      //  MyCXXClass foo; // Make sure pointing at 'foo' returns a VarDecl cursor.
371016ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      if (D->getLocation().isValid() && Data->TokenBeginLoc.isValid() &&
371116ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis          D->getLocation() == Data->TokenBeginLoc)
371216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis        return CXChildVisit_Break;
371316ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    }
3714064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  }
3715064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
371693798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // If our current best cursor is the construction of a temporary object,
371793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // don't replace that cursor with a type reference, because we want
371893798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // clang_getCursor() to point at the constructor.
371993798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  if (clang_isExpression(BestCursor->kind) &&
372093798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor      isa<CXXTemporaryObjectExpr>(getCursorExpr(*BestCursor)) &&
3721aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      cursor.kind == CXCursor_TypeRef) {
3722aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    // Keep the cursor pointing at CXXTemporaryObjectExpr but also mark it
3723aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    // as having the actual point on the type reference.
3724aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    *BestCursor = getTypeRefedCallExprCursor(*BestCursor);
372593798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CXChildVisit_Recurse;
3726aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  }
372793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor
372833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  *BestCursor = cursor;
372933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return CXChildVisit_Recurse;
373033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
3731e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3732b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
3733b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!TU)
3734f462989fe8d6f59ab2d7d0fe2b4b96292ce706eaTed Kremenek    return clang_getNullCursor();
3735e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3736a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3737bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
3738bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
3739a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek  SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
3740671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  CXCursor Result = cxcursor::getCursor(TU, SLoc);
3741a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek
374240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  bool Logging = getenv("LIBCLANG_LOGGING");
374340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  if (Logging) {
374440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile SearchFile;
374540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned SearchLine, SearchColumn;
374640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile ResultFile;
374740749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned ResultLine, ResultColumn;
37486653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    CXString SearchFileName, ResultFileName, KindSpelling, USR;
37496653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    const char *IsDef = clang_isCursorDefinition(Result)? " (Definition)" : "";
375040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXSourceLocation ResultLoc = clang_getCursorLocation(Result);
375140749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
375220174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    clang_getExpansionLocation(Loc, &SearchFile, &SearchLine, &SearchColumn, 0);
375320174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    clang_getExpansionLocation(ResultLoc, &ResultFile, &ResultLine,
375420174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                               &ResultColumn, 0);
375540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    SearchFileName = clang_getFileName(SearchFile);
375640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    ResultFileName = clang_getFileName(ResultFile);
375740749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    KindSpelling = clang_getCursorKindSpelling(Result.kind);
37586653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    USR = clang_getCursorUSR(Result);
37596653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    fprintf(stderr, "clang_getCursor(%s:%d:%d) = %s(%s:%d:%d):%s%s\n",
376040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(SearchFileName), SearchLine, SearchColumn,
376140749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(KindSpelling),
37626653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(ResultFileName), ResultLine, ResultColumn,
37636653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(USR), IsDef);
376440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(SearchFileName);
376540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(ResultFileName);
376640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(KindSpelling);
37676653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    clang_disposeString(USR);
37680aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor
37690aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    CXCursor Definition = clang_getCursorDefinition(Result);
37700aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    if (!clang_equalCursors(Definition, clang_getNullCursor())) {
37710aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXSourceLocation DefinitionLoc = clang_getCursorLocation(Definition);
37720aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionKindSpelling
37730aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor                                = clang_getCursorKindSpelling(Definition.kind);
37740aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXFile DefinitionFile;
37750aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      unsigned DefinitionLine, DefinitionColumn;
377620174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth      clang_getExpansionLocation(DefinitionLoc, &DefinitionFile,
377720174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                 &DefinitionLine, &DefinitionColumn, 0);
37780aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionFileName = clang_getFileName(DefinitionFile);
37790aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      fprintf(stderr, "  -> %s(%s:%d:%d)\n",
37800aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionKindSpelling),
37810aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionFileName),
37820aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              DefinitionLine, DefinitionColumn);
37830aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionFileName);
37840aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionKindSpelling);
37850aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    }
378640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  }
378740749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
3788e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  return Result;
378977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
379077128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
3791738855554394a6afcf39cc8345fd22c3756b8dd0Ted KremenekCXCursor clang_getNullCursor(void) {
37925bfb8c128c2ac8eb4032afc180cdc400a0f953caDouglas Gregor  return MakeCXCursorInvalid(CXCursor_InvalidFile);
3793738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
3794738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek
3795738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenekunsigned clang_equalCursors(CXCursor X, CXCursor Y) {
3796283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  return X == Y;
3797738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
37980d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
37999ce5584553054d0cb934940586aca0186e87fa57Douglas Gregorunsigned clang_hashCursor(CXCursor C) {
38009ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  unsigned Index = 0;
38019ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  if (clang_isExpression(C.kind) || clang_isStatement(C.kind))
38029ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor    Index = 1;
38039ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
38049ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  return llvm::DenseMapInfo<std::pair<unsigned, void*> >::getHashValue(
38059ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor                                        std::make_pair(C.kind, C.data[Index]));
38069ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor}
38079ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
38089ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isInvalid(enum CXCursorKind K) {
380977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
381077128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
381177128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
38129ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isDeclaration(enum CXCursorKind K) {
381389922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl;
381489922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff}
38152d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff
38169ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isReference(enum CXCursorKind K) {
3817f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
3818f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3819f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
382097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isExpression(enum CXCursorKind K) {
382197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
382297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
382397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
382497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isStatement(enum CXCursorKind K) {
382597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
382697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
382797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
38288be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregorunsigned clang_isAttribute(enum CXCursorKind K) {
38298be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor    return K >= CXCursor_FirstAttr && K <= CXCursor_LastAttr;
38308be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor}
38318be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor
38327eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregorunsigned clang_isTranslationUnit(enum CXCursorKind K) {
38337eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return K == CXCursor_TranslationUnit;
38347eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
38357eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
38369f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregorunsigned clang_isPreprocessing(enum CXCursorKind K) {
38379f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
38389f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor}
38399f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor
3840ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenekunsigned clang_isUnexposed(enum CXCursorKind K) {
3841ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  switch (K) {
3842ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedDecl:
3843ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedExpr:
3844ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedStmt:
3845ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedAttr:
3846ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return true;
3847ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    default:
3848ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return false;
3849ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  }
3850ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek}
3851ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek
38529ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXCursorKind clang_getCursorKind(CXCursor C) {
38539efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff  return C.kind;
38549efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff}
38559efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff
385698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas GregorCXSourceLocation clang_getCursorLocation(CXCursor C) {
385798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (clang_isReference(C.kind)) {
3858f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    switch (C.kind) {
3859f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCSuperClassRef: {
3860f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3861f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCSuperClassRef(C);
3862a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3863f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3864f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3865f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
3866f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCProtocolDecl *, SourceLocation> P
3867f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCProtocolRef(C);
3868a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3869f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3870f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3871f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef: {
3872f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3873f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCClassRef(C);
3874a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3875f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
38767d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3877f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef: {
38787d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      std::pair<TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
3879a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
38807d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
38810b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
38820b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
38830b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      std::pair<TemplateDecl *, SourceLocation> P = getCursorTemplateRef(C);
38840b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
38850b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
38860b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
38876931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
38886931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      std::pair<NamedDecl *, SourceLocation> P = getCursorNamespaceRef(C);
38896931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
38906931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
38916931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
3892a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3893a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      std::pair<FieldDecl *, SourceLocation> P = getCursorMemberRef(C);
3894a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3895a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3896a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3897011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef: {
3898011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      std::pair<VarDecl *, SourceLocation> P = getCursorVariableRef(C);
3899011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3900011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    }
3901011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
39023064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
39031b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      CXXBaseSpecifier *BaseSpec = getCursorCXXBaseSpecifier(C);
39041b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (!BaseSpec)
39051b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return clang_getNullLocation();
39061b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
39071b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (TypeSourceInfo *TSInfo = BaseSpec->getTypeSourceInfo())
39081b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return cxloc::translateSourceLocation(getCursorContext(C),
39091b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                            TSInfo->getTypeLoc().getBeginLoc());
39101b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
39111b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
391296a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar                                        BaseSpec->getLocStart());
39133064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
3914f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
391536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
391636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      std::pair<LabelStmt *, SourceLocation> P = getCursorLabelRef(C);
391736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C), P.second);
391836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
391936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
39201f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
39211f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
39221f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                          getCursorOverloadedDeclRef(C).second);
39231f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3924f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    default:
3925f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3926f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      llvm_unreachable("Missed a reference kind");
3927f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
392898258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  }
392997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
393097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3931f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    return cxloc::translateSourceLocation(getCursorContext(C),
393297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor                                   getLocationFromExpr(getCursorExpr(C)));
393397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
393436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind))
393536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C),
393636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor                                          getCursorStmt(C)->getLocStart());
393736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
39389f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  if (C.kind == CXCursor_PreprocessingDirective) {
39399f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
39409f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
39419f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  }
39424807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
39439b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion) {
39444ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    SourceLocation L
39459e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth      = cxcursor::getCursorMacroExpansion(C)->getSourceRange().getBegin();
39464807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
39474807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  }
3948572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3949572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition) {
3950572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
3951572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3952572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  }
3953ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3954ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective) {
3955ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    SourceLocation L
3956ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      = cxcursor::getCursorInclusionDirective(C)->getSourceRange().getBegin();
3957ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3958ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  }
3959ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
39609a700d277c38d9afaa7cb3fe93a714bfe9b62eecTed Kremenek  if (C.kind < CXCursor_FirstDecl || C.kind > CXCursor_LastDecl)
39615352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullLocation();
396298258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor
3963f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  Decl *D = getCursorDecl(C);
396416ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  if (!D)
396516ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    return clang_getNullLocation();
396616ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
3967f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  SourceLocation Loc = D->getLocation();
3968007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // FIXME: Multiple variables declared in a single declaration
3969007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // currently lack the information needed to correctly determine their
3970007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // ranges when accounting for the type-specifier.  We use context
3971007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3972007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // and if so, whether it is the first decl.
3973007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3974007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (!cxcursor::isFirstInDeclGroup(C))
3975007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      Loc = VD->getLocation();
3976007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
3977007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek
3978ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis  // For ObjC methods, give the start location of the method name.
3979ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis  if (ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
3980ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis    Loc = MD->getSelectorStartLoc();
3981ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis
39822ca54feee89d7277fb967e3247a64f40ef155a82Douglas Gregor  return cxloc::translateSourceLocation(getCursorContext(C), Loc);
398388145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
3984a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor
3985a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor} // end extern "C"
3986a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3987671436e9e2794c56f3c2e62739d225571493af37Argyrios KyrtzidisCXCursor cxcursor::getCursor(CXTranslationUnit TU, SourceLocation SLoc) {
3988671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  assert(TU);
3989671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3990671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // Guard against an invalid SourceLocation, or we may assert in one
3991671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // of the following calls.
3992671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  if (SLoc.isInvalid())
3993671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    return clang_getNullCursor();
3994671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3995671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3996671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3997671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // Translate the given source location to make it point at the beginning of
3998671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // the token under the cursor.
3999671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  SLoc = Lexer::GetBeginningOfToken(SLoc, CXXUnit->getSourceManager(),
40004e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie                                    CXXUnit->getASTContext().getLangOpts());
4001671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
4002671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
4003671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  if (SLoc.isValid()) {
4004671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    GetCursorData ResultData(CXXUnit->getSourceManager(), SLoc, Result);
4005671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    CursorVisitor CursorVis(TU, GetCursorVisitor, &ResultData,
4006671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                            /*VisitPreprocessorLast=*/true,
4007e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                            /*VisitIncludedEntities=*/false,
4008671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                            SourceLocation(SLoc));
4009dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    CursorVis.visitFileRegion();
4010671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  }
4011671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
4012671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  return Result;
4013671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis}
4014671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
4015a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C) {
4016a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  if (clang_isReference(C.kind)) {
4017a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    switch (C.kind) {
4018a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCSuperClassRef:
4019a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return  getCursorObjCSuperClassRef(C).second;
4020f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4021a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCProtocolRef:
4022a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCProtocolRef(C).second;
4023f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4024a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCClassRef:
4025a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCClassRef(C).second;
40267d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
4027a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_TypeRef:
4028a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorTypeRef(C).second;
40290b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
40300b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
40310b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return getCursorTemplateRef(C).second;
40320b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
40336931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
40346931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return getCursorNamespaceRef(C).second;
4035a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
4036a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
4037a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return getCursorMemberRef(C).second;
4038a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
40393064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier:
40401b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return getCursorCXXBaseSpecifier(C)->getSourceRange();
4041f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
404236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
404336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return getCursorLabelRef(C).second;
404436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
40451f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
40461f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return getCursorOverloadedDeclRef(C).second;
40471f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4048011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef:
4049011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return getCursorVariableRef(C).second;
4050011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
4051a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    default:
4052a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
4053a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      llvm_unreachable("Missed a reference kind");
4054a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    }
4055a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  }
405697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
405797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
4058a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorExpr(C)->getSourceRange();
405933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
406033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (clang_isStatement(C.kind))
4061a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorStmt(C)->getSourceRange();
4062f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
40636639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  if (clang_isAttribute(C.kind))
40646639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis    return getCursorAttr(C)->getRange();
40656639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis
4066a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_PreprocessingDirective)
4067a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorPreprocessingDirective(C);
40684807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
4069ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_MacroExpansion) {
4070ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
4071ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorMacroExpansion(C)->getSourceRange();
4072ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
4073ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
4074572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
4075ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_MacroDefinition) {
4076ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
4077ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorMacroDefinition(C)->getSourceRange();
4078ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
4079ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
4080ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
4081ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_InclusionDirective) {
4082ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
4083ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorInclusionDirective(C)->getSourceRange();
4084ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
4085ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
4086ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
40870822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis  if (C.kind == CXCursor_TranslationUnit) {
40880822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
40890822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    FileID MainID = TU->getSourceManager().getMainFileID();
40900822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    SourceLocation Start = TU->getSourceManager().getLocForStartOfFile(MainID);
40910822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    SourceLocation End = TU->getSourceManager().getLocForEndOfFile(MainID);
40920822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    return SourceRange(Start, End);
40930822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis  }
40940822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis
4095007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
4096007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    Decl *D = cxcursor::getCursorDecl(C);
409716ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (!D)
409816ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      return SourceRange();
409916ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
4100007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    SourceRange R = D->getSourceRange();
4101007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // FIXME: Multiple variables declared in a single declaration
4102007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // currently lack the information needed to correctly determine their
4103007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // ranges when accounting for the type-specifier.  We use context
4104007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
4105007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // and if so, whether it is the first decl.
4106007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
4107007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      if (!cxcursor::isFirstInDeclGroup(C))
4108007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek        R.setBegin(VD->getLocation());
4109007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    }
4110007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    return R;
4111007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
41126653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return SourceRange();
41136653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
41146653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
41156653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// \brief Retrieves the "raw" cursor extent, which is then extended to include
41166653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// the decl-specifier-seq for declarations.
41176653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr) {
41186653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
41196653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    Decl *D = cxcursor::getCursorDecl(C);
412016ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (!D)
412116ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      return SourceRange();
412216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
41236653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange R = D->getSourceRange();
41242494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
41252494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // Adjust the start of the location for declarations preceded by
41262494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // declaration specifiers.
41272494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
41286653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
41292494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
413096a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
41312494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
41322494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
413396a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
41342494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    }
41356653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
41362494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && R.getBegin().isValid() &&
41372494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, R.getBegin()))
41382494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      R.setBegin(StartLoc);
41392494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
41402494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // FIXME: Multiple variables declared in a single declaration
41412494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // currently lack the information needed to correctly determine their
41422494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // ranges when accounting for the type-specifier.  We use context
41432494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
41442494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // and if so, whether it is the first decl.
41452494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
41462494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (!cxcursor::isFirstInDeclGroup(C))
41472494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        R.setBegin(VD->getLocation());
41486653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    }
41496653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
41506653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    return R;
41516653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  }
41526653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
41536653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return getRawCursorExtent(C);
41546653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
4155a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
4156a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorextern "C" {
4157a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
4158a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas GregorCXSourceRange clang_getCursorExtent(CXCursor C) {
4159a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  SourceRange R = getRawCursorExtent(C);
4160a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R.isInvalid())
41615352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4162f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4163a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  return cxloc::translateSourceRange(getCursorContext(C), R);
4164a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor}
4165c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
4166c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas GregorCXCursor clang_getCursorReferenced(CXCursor C) {
4167b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
4168b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
4169f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4170a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit tu = getCursorTU(C);
41711f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (clang_isDeclaration(C.kind)) {
41721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getCursorDecl(C);
417316ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (!D)
417416ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      return clang_getNullCursor();
41751f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4176a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Using, D->getLocation(), tu);
41775f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    if (ObjCPropertyImplDecl *PropImpl =dyn_cast<ObjCPropertyImplDecl>(D))
4178e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
4179e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return MakeCXCursor(Property, tu);
4180e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
4181c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return C;
41821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
41831f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
418497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
41851f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Expr *E = getCursorExpr(C);
41861f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getDeclFromExpr(E);
4187aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (D) {
4188aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      CXCursor declCursor = MakeCXCursor(D, tu);
4189aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      declCursor = getSelectorIdentifierCursor(getSelectorIdentifierIndex(C),
4190aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis                                               declCursor);
4191aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return declCursor;
4192aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    }
41931f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
41941f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (OverloadExpr *Ovl = dyn_cast_or_null<OverloadExpr>(E))
4195a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Ovl, tu);
41961f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
419797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    return clang_getNullCursor();
419897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
419997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
420036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
420136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
420236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (GotoStmt *Goto = dyn_cast_or_null<GotoStmt>(S))
420337c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek      if (LabelDecl *label = Goto->getLabel())
420437c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        if (LabelStmt *labelS = label->getStmt())
420537c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        return MakeCXCursor(labelS, getCursorDecl(C), tu);
420636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
420736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return clang_getNullCursor();
420836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
420936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
42109b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion) {
42119e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth    if (MacroDefinition *Def = getCursorMacroExpansion(C)->getDefinition())
4212a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeMacroDefinitionCursor(Def, tu);
4213bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  }
4214bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
4215c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  if (!clang_isReference(C.kind))
4216c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return clang_getNullCursor();
4217f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4218c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  switch (C.kind) {
4219c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    case CXCursor_ObjCSuperClassRef:
4220a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCSuperClassRef(C).first, tu);
4221f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4222f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
422398c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis      ObjCProtocolDecl *Prot = getCursorObjCProtocolRef(C).first;
422498c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis      if (ObjCProtocolDecl *Def = Prot->getDefinition())
422598c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis        return MakeCXCursor(Def, tu);
422698c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis
4227c15707d8da08df2eb22f6ed047743fa3f7c9831bArgyrios Kyrtzidis      return MakeCXCursor(Prot, tu);
422898c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis    }
4229f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
42307723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor    case CXCursor_ObjCClassRef: {
42317723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
42327723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor      if (ObjCInterfaceDecl *Def = Class->getDefinition())
42337723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor        return MakeCXCursor(Def, tu);
42347723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor
4235c15707d8da08df2eb22f6ed047743fa3f7c9831bArgyrios Kyrtzidis      return MakeCXCursor(Class, tu);
42367723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor    }
42377d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
4238f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef:
4239a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTypeRef(C).first, tu );
42400b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
42410b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
4242a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTemplateRef(C).first, tu );
42430b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
42446931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
4245a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorNamespaceRef(C).first, tu );
42466931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
4247a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
4248a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorMemberRef(C).first, tu );
4249a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
42503064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
42513064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = cxcursor::getCursorCXXBaseSpecifier(C);
42523064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return clang_getTypeDeclaration(cxtype::MakeCXType(B->getType(),
4253a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                                         tu ));
42543064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
4255f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
425636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
425736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // FIXME: We end up faking the "parent" declaration here because we
425836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // don't want to make CXCursor larger.
425936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return MakeCXCursor(getCursorLabelRef(C).first,
4260a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek               static_cast<ASTUnit*>(tu->TUData)->getASTContext()
4261a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          .getTranslationUnitDecl(),
4262a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          tu);
426336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
42641f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
42651f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return C;
4266011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
4267011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef:
4268011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return MakeCXCursor(getCursorVariableRef(C).first, tu);
42691f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4270c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    default:
4271c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      // We would prefer to enumerate all non-reference cursor kinds here.
4272c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      llvm_unreachable("Unhandled reference cursor kind");
4273c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  }
4274c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor}
4275c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
4276b699866820102a69d83d6ac6941985c5ef4e8c40Douglas GregorCXCursor clang_getCursorDefinition(CXCursor C) {
4277b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
4278b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
4279f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4280a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(C);
4281f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4282b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  bool WasReference = false;
428397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
4284b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    C = clang_getCursorReferenced(C);
4285b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    WasReference = true;
4286b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4287b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
42889b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion)
4289bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    return clang_getCursorReferenced(C);
4290bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
4291b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4292b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4293b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4294b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  Decl *D = getCursorDecl(C);
4295b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!D)
4296b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4297f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4298b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  switch (D->getKind()) {
4299b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't really separate the notions of
4300b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // declaration and definition.
4301b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Namespace:
4302b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Typedef:
4303162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  case Decl::TypeAlias:
43043e4c6c4c79a03f5cb0c4671d7c282d623c6dc35eRichard Smith  case Decl::TypeAliasTemplate:
4305b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTypeParm:
4306b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::EnumConstant:
4307b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Field:
4308d98114647e16796a976b04af79975b4f0eacf22bBenjamin Kramer  case Decl::IndirectField:
4309b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCIvar:
4310b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCAtDefsField:
4311b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ImplicitParam:
4312b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ParmVar:
4313b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NonTypeTemplateParm:
4314b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTemplateParm:
4315b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategoryImpl:
4316b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCImplementation:
43176206d53f67613958ae1b023aba337ebb46f11a8bAbramo Bagnara  case Decl::AccessSpec:
4318b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::LinkageSpec:
4319b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCPropertyImpl:
4320b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FileScopeAsm:
4321b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::StaticAssert:
4322b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Block:
4323ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  case Decl::Label:  // FIXME: Is this right??
4324af0f4d0b2e38c810effc8b024ad2fb6604eec5d3Francois Pichet  case Decl::ClassScopeFunctionSpecialization:
432515de72cf580840c61e5704c2f8a2b56f9d0638e1Douglas Gregor  case Decl::Import:
4326b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return C;
4327b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4328b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't make any sense here, but are
4329b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // nonetheless harmless.
4330b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TranslationUnit:
4331b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
4332b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4333b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds for which the definition is not resolvable.
4334b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingTypename:
4335b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingValue:
4336b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
4337b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4338b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingDirective:
4339b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
4340a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                        TU);
4341b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4342b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NamespaceAlias:
4343a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), TU);
4344b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4345b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Enum:
4346b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Record:
4347b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXRecord:
4348b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplateSpecialization:
4349b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplatePartialSpecialization:
4350952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor    if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
4351a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
4352b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4353b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4354b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Function:
4355b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXMethod:
4356b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConstructor:
4357b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXDestructor:
4358b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConversion: {
4359b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
4360b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionDecl>(D)->getBody(Def))
4361a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(const_cast<FunctionDecl *>(Def), TU);
4362b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4363b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4364b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4365b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Var: {
436631310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    // Ask the variable if it has a definition.
436731310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (VarDecl *Def = cast<VarDecl>(D)->getDefinition())
4368a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
436931310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    return clang_getNullCursor();
4370b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4371f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4372b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FunctionTemplate: {
4373b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
4374b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
4375a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def->getDescribedFunctionTemplate(), TU);
4376b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4377b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4378f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4379b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplate: {
4380b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
4381952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor                                                            ->getDefinition())
43820b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return MakeCXCursor(cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
4383a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          TU);
4384b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4385b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4386b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
43871f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::Using:
43881f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<UsingDecl>(D),
4389a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
4390b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4391b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingShadow:
4392b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getCursorDefinition(
4393f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                       MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
4394a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                    TU));
4395b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4396b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCMethod: {
4397b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
4398b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (Method->isThisDeclarationADefinition())
4399b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
4400b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4401b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // Dig out the method definition in the associated
4402b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // @implementation, if we have it.
4403b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: The ASTs should make finding the definition easier.
4404b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4405b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                       = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
4406b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
4407b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
4408b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                                                  Method->isInstanceMethod()))
4409b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          if (Def->isThisDeclarationADefinition())
4410a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek            return MakeCXCursor(Def, TU);
4411b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4412b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4413b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4414b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4415b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategory:
4416b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCCategoryImplDecl *Impl
4417b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                               = cast<ObjCCategoryDecl>(D)->getImplementation())
4418a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4419b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4420b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4421b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProtocol:
44225e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregor    if (ObjCProtocolDecl *Def = cast<ObjCProtocolDecl>(D)->getDefinition())
44235e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregor      return MakeCXCursor(Def, TU);
4424b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4425b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4426375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  case Decl::ObjCInterface: {
4427b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // There are two notions of a "definition" for an Objective-C
4428b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // class: the interface and its implementation. When we resolved a
4429b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // reference to an Objective-C class, produce the @interface as
4430b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // the definition; when we were provided with the interface,
4431b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // produce the @implementation as the definition.
4432375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    ObjCInterfaceDecl *IFace = cast<ObjCInterfaceDecl>(D);
4433b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (WasReference) {
4434375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor      if (ObjCInterfaceDecl *Def = IFace->getDefinition())
44357723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor        return MakeCXCursor(Def, TU);
4436375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    } else if (ObjCImplementationDecl *Impl = IFace->getImplementation())
4437a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4438b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4439375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  }
4440f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4441b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProperty:
4442b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: We don't really know where to find the
4443b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // ObjCPropertyImplDecls that implement this property.
4444b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4445b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4446b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCompatibleAlias:
4447b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4448b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
44497723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor      if (ObjCInterfaceDecl *Def = Class->getDefinition())
44507723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor        return MakeCXCursor(Def, TU);
4451f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4452b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4453b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4454b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Friend:
4455b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
4456a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4457b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4458b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4459b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FriendTemplate:
4460b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
4461a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4462b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4463b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4464b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4465b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getNullCursor();
4466b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4467b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4468b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregorunsigned clang_isCursorDefinition(CXCursor C) {
4469b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4470b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return 0;
4471b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4472b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getCursorDefinition(C) == C;
4473b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4474b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
44751a9d0503b67a499797141af0fd6d315d5045f0eaDouglas GregorCXCursor clang_getCanonicalCursor(CXCursor C) {
44761a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  if (!clang_isDeclaration(C.kind))
44771a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return C;
44781a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
4479e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis  if (Decl *D = getCursorDecl(C)) {
4480debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis    if (ObjCCategoryImplDecl *CatImplD = dyn_cast<ObjCCategoryImplDecl>(D))
4481debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis      if (ObjCCategoryDecl *CatD = CatImplD->getCategoryDecl())
4482debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis        return MakeCXCursor(CatD, getCursorTU(C));
4483debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis
4484e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis    if (ObjCImplDecl *ImplD = dyn_cast<ObjCImplDecl>(D))
4485e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis      if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
4486e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis        return MakeCXCursor(IFD, getCursorTU(C));
4487e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis
44881a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return MakeCXCursor(D->getCanonicalDecl(), getCursorTU(C));
4489e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis  }
44901a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
44911a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  return C;
44921a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor}
449334ebe1e1b0779bcea2f277bc6b4e9dd98bf70b7bArgyrios Kyrtzidis
449434ebe1e1b0779bcea2f277bc6b4e9dd98bf70b7bArgyrios Kyrtzidisint clang_Cursor_getObjCSelectorIndex(CXCursor cursor) {
449534ebe1e1b0779bcea2f277bc6b4e9dd98bf70b7bArgyrios Kyrtzidis  return cxcursor::getSelectorIdentifierIndexAndLoc(cursor).first;
449634ebe1e1b0779bcea2f277bc6b4e9dd98bf70b7bArgyrios Kyrtzidis}
44971a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
44981f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregorunsigned clang_getNumOverloadedDecls(CXCursor C) {
44997c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (C.kind != CXCursor_OverloadedDeclRef)
45001f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return 0;
45011f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45021f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
45031f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
45041f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return E->getNumDecls();
45051f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45061f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
45071f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
45081f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return S->size();
45091f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45101f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
45111f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4512826faa22bae112e01293a58534a40711043cce65Argyrios Kyrtzidis    return Using->shadow_size();
45131f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45141f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return 0;
45151f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
45161f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45171f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas GregorCXCursor clang_getOverloadedDecl(CXCursor cursor, unsigned index) {
45187c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (cursor.kind != CXCursor_OverloadedDeclRef)
45191f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
45201f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45211f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (index >= clang_getNumOverloadedDecls(cursor))
45221f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
45231f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4524a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(cursor);
45251f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(cursor).first;
45261f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
4527a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(E->decls_begin()[index], TU);
45281f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45291f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
45301f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
4531a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(S->begin()[index], TU);
45321f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45331f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
45341f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D)) {
45351f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // FIXME: This is, unfortunately, linear time.
45361f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    UsingDecl::shadow_iterator Pos = Using->shadow_begin();
45371f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    std::advance(Pos, index);
4538a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<UsingShadowDecl>(*Pos)->getTargetDecl(), TU);
45391f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
45401f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45411f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return clang_getNullCursor();
45421f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
45431f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
45440d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbarvoid clang_getDefinitionSpellingAndExtent(CXCursor C,
45454ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **startBuf,
45464ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **endBuf,
45474ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startLine,
45484ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startColumn,
45494ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *endLine,
45509ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          unsigned *endColumn) {
4551283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  assert(getCursorDecl(C) && "CXCursor has null decl");
4552283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  NamedDecl *ND = static_cast<NamedDecl *>(getCursorDecl(C));
45534ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  FunctionDecl *FD = dyn_cast<FunctionDecl>(ND);
45544ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
4555f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
45564ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  SourceManager &SM = FD->getASTContext().getSourceManager();
45574ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startBuf = SM.getCharacterData(Body->getLBracLoc());
45584ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endBuf = SM.getCharacterData(Body->getRBracLoc());
45594ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
45604ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
45614ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
45624ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
45634ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff}
4564f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4565430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4566430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas GregorCXSourceRange clang_getCursorReferenceNameRange(CXCursor C, unsigned NameFlags,
4567430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                                                unsigned PieceIndex) {
4568430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  RefNamePieces Pieces;
4569430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4570430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  switch (C.kind) {
4571430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_MemberRefExpr:
4572430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (MemberExpr *E = dyn_cast<MemberExpr>(getCursorExpr(C)))
4573430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Pieces = buildPieces(NameFlags, true, E->getMemberNameInfo(),
4574430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getQualifierLoc().getSourceRange());
4575430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4576430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4577430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_DeclRefExpr:
4578430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (DeclRefExpr *E = dyn_cast<DeclRefExpr>(getCursorExpr(C)))
4579430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Pieces = buildPieces(NameFlags, false, E->getNameInfo(),
4580430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getQualifierLoc().getSourceRange(),
4581e4b92761b43ced611c417ae478568610f1ad7b1eAbramo Bagnara                           E->getOptionalExplicitTemplateArgs());
4582430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4583430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4584430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_CallExpr:
4585430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (CXXOperatorCallExpr *OCE =
4586430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        dyn_cast<CXXOperatorCallExpr>(getCursorExpr(C))) {
4587430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Expr *Callee = OCE->getCallee();
4588430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Callee))
4589430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        Callee = ICE->getSubExpr();
4590430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4591430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Callee))
4592430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        Pieces = buildPieces(NameFlags, false, DRE->getNameInfo(),
4593430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                             DRE->getQualifierLoc().getSourceRange());
4594430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    }
4595430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4596430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4597430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  default:
4598430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4599430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  }
4600430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4601430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  if (Pieces.empty()) {
4602430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (PieceIndex == 0)
4603430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      return clang_getCursorExtent(C);
4604430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  } else if (PieceIndex < Pieces.size()) {
4605430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      SourceRange R = Pieces[PieceIndex];
4606430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (R.isValid())
4607430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        return cxloc::translateSourceRange(getCursorContext(C), R);
4608430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  }
4609430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4610430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  return clang_getNullRange();
4611430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor}
4612430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
46130a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregorvoid clang_enableStackTraces(void) {
46140a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  llvm::sys::PrintStackTraceOnErrorSignal();
46150a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor}
46160a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor
4617995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbarvoid clang_executeOnThread(void (*fn)(void*), void *user_data,
4618995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar                           unsigned stack_size) {
4619995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar  llvm::llvm_execute_on_thread(fn, user_data, stack_size);
4620995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar}
4621995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar
4622fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
4623fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
4624fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
4625fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor// Token-based Operations.
4626fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
4627fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4628fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor/* CXToken layout:
4629fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[0]: a CXTokenKind
4630fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[1]: starting token location
4631fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[2]: token length
4632fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[3]: reserved
4633f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek *   ptr_data: for identifiers and keywords, an IdentifierInfo*.
4634fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   otherwise unused.
4635fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor */
4636fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorextern "C" {
4637fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4638fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXTokenKind clang_getTokenKind(CXToken CXTok) {
4639fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return static_cast<CXTokenKind>(CXTok.int_data[0]);
4640fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4641fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4642fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
4643fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  switch (clang_getTokenKind(CXTok)) {
4644fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Identifier:
4645fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Keyword:
4646fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We know we have an IdentifierInfo*, so use that.
4647ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(static_cast<IdentifierInfo *>(CXTok.ptr_data)
4648ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                            ->getNameStart());
4649fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4650fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Literal: {
4651fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We have stashed the starting pointer in the ptr_data field. Use it.
4652fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    const char *Text = static_cast<const char *>(CXTok.ptr_data);
46535f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    return createCXString(StringRef(Text, CXTok.int_data[2]));
4654fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4655f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4656fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Punctuation:
4657fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Comment:
4658fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    break;
4659fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4660f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4661f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // We have to find the starting buffer pointer the hard way, by
4662fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // deconstructing the source location.
4663a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4664fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4665ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
4666f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4667fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
4668fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> LocInfo
4669a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    = CXXUnit->getSourceManager().getDecomposedSpellingLoc(Loc);
4670f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
46715f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  StringRef Buffer
4672f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
4673f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  if (Invalid)
4674aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor    return createCXString("");
4675fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4676f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  return createCXString(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
4677fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4678f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4679fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
4680a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4681fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4682fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return clang_getNullLocation();
4683f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4684fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
4685fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4686fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4687fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4688fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
4689a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
46905352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (!CXXUnit)
46915352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4692f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4693f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return cxloc::translateSourceRange(CXXUnit->getASTContext(),
4694fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4695fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4696f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4697ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisstatic void getTokens(ASTUnit *CXXUnit, SourceRange Range,
4698ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                      SmallVectorImpl<CXToken> &CXTokens) {
4699fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
4700fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
4701ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(Range.getBegin());
4702fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
4703ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(Range.getEnd());
4704f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4705fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Cannot tokenize across files.
4706fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (BeginLocInfo.first != EndLocInfo.first)
4707fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4708f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4709f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Create a lexer
4710f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
47115f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  StringRef Buffer
4712f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
471347a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor  if (Invalid)
471447a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor    return;
4715aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor
4716fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
47174e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie            CXXUnit->getASTContext().getLangOpts(),
4718f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer            Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
4719fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lex.SetCommentRetentionState(true);
4720f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4721fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Lex tokens until we hit the end of the range.
4722f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
4723fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Token Tok;
4724096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall  bool previousWasAt = false;
4725fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  do {
4726fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Lex the next token
4727fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    Lex.LexFromRawLexer(Tok);
4728fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.is(tok::eof))
4729fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      break;
4730f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4731fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Initialize the CXToken.
4732fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXToken CXTok;
4733f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4734fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Common fields
4735fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
4736fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[2] = Tok.getLength();
4737fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[3] = 0;
4738f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4739fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Kind-specific fields
4740fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.isLiteral()) {
4741fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Literal;
4742fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = (void *)Tok.getLiteralData();
4743c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara    } else if (Tok.is(tok::raw_identifier)) {
4744aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor      // Lookup the identifier to determine whether we have a keyword.
4745fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      IdentifierInfo *II
4746c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok);
4747aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek
4748096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall      if ((II->getObjCKeywordID() != tok::objc_not_keyword) && previousWasAt) {
4749aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek        CXTok.int_data[0] = CXToken_Keyword;
4750aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4751aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      else {
4752c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        CXTok.int_data[0] = Tok.is(tok::identifier)
4753c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          ? CXToken_Identifier
4754c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          : CXToken_Keyword;
4755aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4756fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = II;
4757fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else if (Tok.is(tok::comment)) {
4758fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Comment;
4759fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4760fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else {
4761fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Punctuation;
4762fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4763fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    }
4764fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTokens.push_back(CXTok);
4765096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall    previousWasAt = Tok.is(tok::at);
4766fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  } while (Lex.getBufferLocation() <= EffectiveBufferEnd);
4767ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis}
4768ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4769ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisvoid clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
4770ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                    CXToken **Tokens, unsigned *NumTokens) {
4771ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (Tokens)
4772ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    *Tokens = 0;
4773ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (NumTokens)
4774ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    *NumTokens = 0;
4775ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4776ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4777ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (!CXXUnit || !Tokens || !NumTokens)
4778ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
4779ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4780ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
4781ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4782ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SourceRange R = cxloc::translateCXSourceRange(Range);
4783ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (R.isInvalid())
4784ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
4785ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4786ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SmallVector<CXToken, 32> CXTokens;
4787ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  getTokens(CXXUnit, R, CXTokens);
4788f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4789fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (CXTokens.empty())
4790fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4791f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4792fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
4793fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
4794fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *NumTokens = CXTokens.size();
4795fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
47960045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
47976db610934bedc6896393c1e1099525b35380acd6Ted Kremenekvoid clang_disposeTokens(CXTranslationUnit TU,
47986db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                         CXToken *Tokens, unsigned NumTokens) {
47996db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  free(Tokens);
48006db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
48016db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
48026db610934bedc6896393c1e1099525b35380acd6Ted Kremenek} // end: extern "C"
48036db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
48046db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
48056db610934bedc6896393c1e1099525b35380acd6Ted Kremenek// Token annotation APIs.
48066db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
48076db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
48080045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregortypedef llvm::DenseMap<unsigned, CXCursor> AnnotateTokensData;
4809fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
4810fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXCursor parent,
4811fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXClientData client_data);
48126db610934bedc6896393c1e1099525b35380acd6Ted Kremeneknamespace {
48136db610934bedc6896393c1e1099525b35380acd6Ted Kremenekclass AnnotateTokensWorker {
48146db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  AnnotateTokensData &Annotated;
481511949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXToken *Tokens;
481611949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXCursor *Cursors;
481711949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  unsigned NumTokens;
4818fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned TokIdx;
48194419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  unsigned PreprocessingTokIdx;
4820fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CursorVisitor AnnotateVis;
4821fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceManager &SrcMgr;
4822f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool HasContextSensitiveKeywords;
4823f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4824fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  bool MoreTokens() const { return TokIdx < NumTokens; }
4825fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned NextToken() const { return TokIdx; }
4826fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AdvanceToken() { ++TokIdx; }
4827fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation GetTokenLoc(unsigned tokI) {
4828fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
4829fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
48305f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  bool isFunctionMacroToken(unsigned tokI) const {
4831a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return Tokens[tokI].int_data[3] != 0;
4832a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
48335f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  SourceLocation getFunctionMacroTokenLoc(unsigned tokI) const {
4834a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[3]);
4835a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4836a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4837a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  void annotateAndAdvanceTokens(CXCursor, RangeComparisonResult, SourceRange);
48385f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  void annotateAndAdvanceFunctionMacroTokens(CXCursor, RangeComparisonResult,
48395f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis                                             SourceRange);
4840fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
48416db610934bedc6896393c1e1099525b35380acd6Ted Kremenekpublic:
484211949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  AnnotateTokensWorker(AnnotateTokensData &annotated,
4843fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                       CXToken *tokens, CXCursor *cursors, unsigned numTokens,
4844a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                       CXTranslationUnit tu, SourceRange RegionOfInterest)
484511949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    : Annotated(annotated), Tokens(tokens), Cursors(cursors),
48464419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      NumTokens(numTokens), TokIdx(0), PreprocessingTokIdx(0),
4847a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      AnnotateVis(tu,
4848f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                  AnnotateTokensVisitor, this,
4849f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                  /*VisitPreprocessorLast=*/true,
4850e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                  /*VisitIncludedEntities=*/false,
4851f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                  RegionOfInterest),
4852f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      SrcMgr(static_cast<ASTUnit*>(tu->TUData)->getSourceManager()),
4853f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      HasContextSensitiveKeywords(false) { }
485411949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4855fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
48566db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
485703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  void AnnotateTokens();
4858f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4859f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// \brief Determine whether the annotator saw any cursors that have
4860f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// context-sensitive keywords.
4861f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool hasContextSensitiveKeywords() const {
4862f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    return HasContextSensitiveKeywords;
4863f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
48646db610934bedc6896393c1e1099525b35380acd6Ted Kremenek};
48656db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
48660045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
486703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidisvoid AnnotateTokensWorker::AnnotateTokens() {
4868fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Walk the AST within the region of interest, annotating tokens
4869fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // along the way.
487003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  AnnotateVis.visitFileRegion();
4871fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4872fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = 0 ; I < TokIdx ; ++I) {
487311949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
48744419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    if (Pos != Annotated.end() &&
48754419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        (clang_isInvalid(Cursors[I].kind) ||
48764419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor         Pos->second.kind != CXCursor_PreprocessingDirective))
4877fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      Cursors[I] = Pos->second;
4878fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4879fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4880fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Finish up annotating any tokens left.
4881fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (!MoreTokens())
4882fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return;
488311949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4884fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor &C = clang_getNullCursor();
4885fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = TokIdx ; I < NumTokens ; ++I) {
488603ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (I < PreprocessingTokIdx && clang_isPreprocessing(Cursors[I].kind))
488703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      continue;
488803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
4889fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
4890fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = (Pos == Annotated.end()) ? C : Pos->second;
489111949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  }
489211949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek}
489311949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4894a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief It annotates and advances tokens with a cursor until the comparison
4895a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis//// between the cursor location and the source range is the same as
4896a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \arg compResult.
4897a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis///
4898a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// Pass RangeBefore to annotate tokens with a cursor until a range is reached.
4899a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// Pass RangeOverlap to annotate tokens inside a range.
4900a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisvoid AnnotateTokensWorker::annotateAndAdvanceTokens(CXCursor updateC,
4901a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               RangeComparisonResult compResult,
4902a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               SourceRange range) {
4903a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  while (MoreTokens()) {
4904a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    const unsigned I = NextToken();
49055f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis    if (isFunctionMacroToken(I))
49065f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis      return annotateAndAdvanceFunctionMacroTokens(updateC, compResult, range);
4907a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4908a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    SourceLocation TokLoc = GetTokenLoc(I);
4909a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
4910a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      Cursors[I] = updateC;
4911a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      AdvanceToken();
4912a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      continue;
4913a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
4914a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    break;
4915a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4916a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
4917a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4918a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief Special annotation handling for macro argument tokens.
49195f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidisvoid AnnotateTokensWorker::annotateAndAdvanceFunctionMacroTokens(
49205f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis                                               CXCursor updateC,
4921a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               RangeComparisonResult compResult,
4922a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               SourceRange range) {
49235f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  assert(MoreTokens());
49245f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  assert(isFunctionMacroToken(NextToken()) &&
4925a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis         "Should be called only for macro arg tokens");
4926a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4927a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // This works differently than annotateAndAdvanceTokens; because expanded
4928a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // macro arguments can have arbitrary translation-unit source order, we do not
4929a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // advance the token index one by one until a token fails the range test.
4930a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // We only advance once past all of the macro arg tokens if all of them
4931a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // pass the range test. If one of them fails we keep the token index pointing
4932a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // at the start of the macro arg tokens so that the failing token will be
4933a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // annotated by a subsequent annotation try.
4934a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4935a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  bool atLeastOneCompFail = false;
4936a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4937a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned I = NextToken();
49385f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  for (; I < NumTokens && isFunctionMacroToken(I); ++I) {
49395f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis    SourceLocation TokLoc = getFunctionMacroTokenLoc(I);
4940a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (TokLoc.isFileID())
4941a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      continue; // not macro arg token, it's parens or comma.
4942a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
4943a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (clang_isInvalid(clang_getCursorKind(Cursors[I])))
4944a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        Cursors[I] = updateC;
4945a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    } else
4946a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      atLeastOneCompFail = true;
4947a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4948a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4949a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  if (!atLeastOneCompFail)
4950a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    TokIdx = I; // All of the tokens were handled, advance beyond all of them.
4951a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
4952a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
49536db610934bedc6896393c1e1099525b35380acd6Ted Kremenekenum CXChildVisitResult
49544419b675577d7c281a659fab1fec10e1bfbe04c5Douglas GregorAnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
4955fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CXSourceLocation Loc = clang_getCursorLocation(cursor);
49564419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  SourceRange cursorRange = getRawCursorExtent(cursor);
495781d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor  if (cursorRange.isInvalid())
495881d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor    return CXChildVisit_Recurse;
4959f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4960f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (!HasContextSensitiveKeywords) {
4961f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C properties can have context-sensitive keywords.
4962f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    if (cursor.kind == CXCursor_ObjCPropertyDecl) {
4963f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCPropertyDecl *Property
4964f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(cursor)))
4965f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        HasContextSensitiveKeywords = Property->getPropertyAttributesAsWritten() != 0;
4966f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4967f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C methods can have context-sensitive keywords.
4968f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_ObjCInstanceMethodDecl ||
4969f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ObjCClassMethodDecl) {
4970f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCMethodDecl *Method
4971f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
4972f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->getObjCDeclQualifier())
4973f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4974f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        else {
4975f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
4976f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                                           PEnd = Method->param_end();
4977f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor               P != PEnd; ++P) {
4978f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((*P)->getObjCDeclQualifier()) {
4979f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              HasContextSensitiveKeywords = true;
4980f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              break;
4981f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            }
4982f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          }
4983f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4984f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4985f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4986f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ methods can have context-sensitive keywords.
4987f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_CXXMethod) {
4988f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (CXXMethodDecl *Method
4989f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(cursor))) {
4990f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->hasAttr<FinalAttr>() || Method->hasAttr<OverrideAttr>())
4991f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4992f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4993f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4994f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ classes can have context-sensitive keywords.
4995f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_StructDecl ||
4996f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassDecl ||
4997f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplate ||
4998f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplatePartialSpecialization) {
4999f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Decl *D = getCursorDecl(cursor))
5000f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (D->hasAttr<FinalAttr>())
5001f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
5002f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
5003f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
5004f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
50054419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  if (clang_isPreprocessing(cursor.kind)) {
5006cea731a9cb7de3f473d60e5ea544e25621cebd76Chandler Carruth    // For macro expansions, just note where the beginning of the macro
5007cea731a9cb7de3f473d60e5ea544e25621cebd76Chandler Carruth    // expansion occurs.
50089b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth    if (cursor.kind == CXCursor_MacroExpansion) {
50094419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      Annotated[Loc.int_data] = cursor;
50104419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      return CXChildVisit_Recurse;
50114419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
50124419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
50134419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Items in the preprocessing record are kept separate from items in
50144419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // declarations, so we keep a separate token index.
50154419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    unsigned SavedTokIdx = TokIdx;
50164419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = PreprocessingTokIdx;
50174419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
50184419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Skip tokens up until we catch up to the beginning of the preprocessing
50194419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // entry.
50204419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
50214419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
50224419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
50234419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
50244419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
50254419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
50264419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
50274419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
50284419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
50294419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
50304419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
50314419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
50324419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
50334419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
50344419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Look at all of the tokens within this range.
50354419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
50364419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
50374419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
50384419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
50394419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
5040b219cfc4d75f0a03630b7c4509ef791b7e97b2c8David Blaikie        llvm_unreachable("Infeasible");
50414419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
50424419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
50434419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
50444419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        Cursors[I] = cursor;
50454419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
50464419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
50474419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
50484419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
50494419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
50504419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
50514419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Save the preprocessing token index; restore the non-preprocessing
50524419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // token index.
50534419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    PreprocessingTokIdx = TokIdx;
50544419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = SavedTokIdx;
50550045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor    return CXChildVisit_Recurse;
50560045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  }
5057fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5058fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (cursorRange.isInvalid())
5059fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return CXChildVisit_Continue;
5060a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek
5061fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation L = SourceLocation::getFromRawEncoding(Loc.int_data);
5062fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5063a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  // Adjust the annotated range based specific declarations.
5064a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  const enum CXCursorKind cursorK = clang_getCursorKind(cursor);
5065a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  if (cursorK >= CXCursor_FirstDecl && cursorK <= CXCursor_LastDecl) {
506623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    Decl *D = cxcursor::getCursorDecl(cursor);
50672494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
50682494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
506916ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (const DeclaratorDecl *DD = dyn_cast_or_null<DeclaratorDecl>(D)) {
50702494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
507196a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
507216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    } else if (TypedefDecl *Typedef = dyn_cast_or_null<TypedefDecl>(D)) {
50732494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
507496a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
5075a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek    }
50762494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
50772494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && L.isValid() &&
50782494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, L))
50792494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      cursorRange.setBegin(StartLoc);
5080a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  }
508181d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor
50823f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // If the location of the cursor occurs within a macro instantiation, record
50833f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // the spelling location of the cursor in our annotation map.  We can then
50843f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // paper over the token labelings during a post-processing step to try and
50853f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // get cursor mappings for tokens that are the *arguments* of a macro
50863f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // instantiation.
50873f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  if (L.isMacroID()) {
50883f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    unsigned rawEncoding = SrcMgr.getSpellingLoc(L).getRawEncoding();
50893f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // Only invalidate the old annotation if it isn't part of a preprocessing
50903f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // directive.  Here we assume that the default construction of CXCursor
50913f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // results in CXCursor.kind being an initialized value (i.e., 0).  If
50923f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // this isn't the case, we can fix by doing lookup + insertion.
50934419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
50943f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    CXCursor &oldC = Annotated[rawEncoding];
50953f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    if (!clang_isPreprocessing(oldC.kind))
50963f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek      oldC = cursor;
50973f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  }
50983f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek
5099fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const enum CXCursorKind K = clang_getCursorKind(parent);
5100fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor updateC =
5101d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek    (clang_isInvalid(K) || K == CXCursor_TranslationUnit)
5102d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek     ? clang_getNullCursor() : parent;
5103fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5104a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  annotateAndAdvanceTokens(updateC, RangeBefore, cursorRange);
5105fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
51065517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // Avoid having the cursor of an expression "overwrite" the annotation of the
51075517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // variable declaration that it belongs to.
51085517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // This can happen for C++ constructor expressions whose range generally
51095517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // include the variable declaration, e.g.:
51105517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  //  MyCXXClass foo; // Make sure we don't annotate 'foo' as a CallExpr cursor.
51115517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  if (clang_isExpression(cursorK)) {
51125517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis    Expr *E = getCursorExpr(cursor);
51138ccac3de1335f1cfd7cea56ba1cefcf0b724ce3fArgyrios Kyrtzidis    if (Decl *D = getCursorParentDecl(cursor)) {
51145517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      const unsigned I = NextToken();
51155517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      if (E->getLocStart().isValid() && D->getLocation().isValid() &&
51165517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis          E->getLocStart() == D->getLocation() &&
51175517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis          E->getLocStart() == GetTokenLoc(I)) {
51185517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis        Cursors[I] = updateC;
51195517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis        AdvanceToken();
51205517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      }
51215517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis    }
51225517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  }
51235517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis
5124fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Visit children to get their cursor information.
5125fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned BeforeChildren = NextToken();
5126fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(cursor);
5127fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned AfterChildren = NextToken();
5128fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5129a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // Scan the tokens that are at the end of the cursor, but are not captured
5130a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // but the child cursors.
5131a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  annotateAndAdvanceTokens(cursor, RangeOverlap, cursorRange);
51326db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
5133fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the beginning of the cursor, but are not
5134fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // capture by the child cursors.
5135fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
5136fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
5137fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      break;
51384419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
5139fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
5140fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
5141fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5142fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return CXChildVisit_Continue;
51430045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor}
51440045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
51456db610934bedc6896393c1e1099525b35380acd6Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
51466db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXCursor parent,
51476db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXClientData client_data) {
51486db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
51496db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
51506db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
51516628a614c504263ae539462f049d523dd07ac1baTed Kremeneknamespace {
5152a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5153a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief Uses the macro expansions in the preprocessing record to find
5154a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// and mark tokens that are macro arguments. This info is used by the
5155a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// AnnotateTokensWorker.
5156a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisclass MarkMacroArgTokensVisitor {
5157a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  SourceManager &SM;
5158a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  CXToken *Tokens;
5159a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned NumTokens;
5160a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned CurIdx;
5161a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5162a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidispublic:
5163a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  MarkMacroArgTokensVisitor(SourceManager &SM,
5164a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                            CXToken *tokens, unsigned numTokens)
5165a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    : SM(SM), Tokens(tokens), NumTokens(numTokens), CurIdx(0) { }
5166a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5167a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  CXChildVisitResult visit(CXCursor cursor, CXCursor parent) {
5168a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (cursor.kind != CXCursor_MacroExpansion)
5169a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Continue;
5170a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5171a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    SourceRange macroRange = getCursorMacroExpansion(cursor)->getSourceRange();
5172a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (macroRange.getBegin() == macroRange.getEnd())
5173a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Continue; // it's not a function macro.
5174a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5175a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    for (; CurIdx < NumTokens; ++CurIdx) {
5176a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (!SM.isBeforeInTranslationUnit(getTokenLoc(CurIdx),
5177a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                        macroRange.getBegin()))
5178a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        break;
5179a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
5180a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5181a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (CurIdx == NumTokens)
5182a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Break;
5183a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5184a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    for (; CurIdx < NumTokens; ++CurIdx) {
5185a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      SourceLocation tokLoc = getTokenLoc(CurIdx);
5186a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (!SM.isBeforeInTranslationUnit(tokLoc, macroRange.getEnd()))
5187a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        break;
5188a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
51895f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis      setFunctionMacroTokenLoc(CurIdx, SM.getMacroArgExpandedLocation(tokLoc));
5190a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
5191a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5192a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (CurIdx == NumTokens)
5193a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Break;
5194a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5195a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return CXChildVisit_Continue;
5196a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5197a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5198a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisprivate:
5199a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  SourceLocation getTokenLoc(unsigned tokI) {
5200a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
5201a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5202a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
52035f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  void setFunctionMacroTokenLoc(unsigned tokI, SourceLocation loc) {
5204a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // The third field is reserved and currently not used. Use it here
5205a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // to mark macro arg expanded tokens with their expanded locations.
5206a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    Tokens[tokI].int_data[3] = loc.getRawEncoding();
5207a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5208a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis};
5209a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5210a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis} // end anonymous namespace
5211a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5212a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisstatic CXChildVisitResult
5213a676379b26edc959193f9f919ba9c6d296a57824Argyrios KyrtzidisMarkMacroArgTokensVisitorDelegate(CXCursor cursor, CXCursor parent,
5214a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                  CXClientData client_data) {
5215a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  return static_cast<MarkMacroArgTokensVisitor*>(client_data)->visit(cursor,
5216a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                                                     parent);
5217a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
5218a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5219a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisnamespace {
52206628a614c504263ae539462f049d523dd07ac1baTed Kremenek  struct clang_annotateTokens_Data {
52216628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXTranslationUnit TU;
52226628a614c504263ae539462f049d523dd07ac1baTed Kremenek    ASTUnit *CXXUnit;
52236628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXToken *Tokens;
52246628a614c504263ae539462f049d523dd07ac1baTed Kremenek    unsigned NumTokens;
52256628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXCursor *Cursors;
52266628a614c504263ae539462f049d523dd07ac1baTed Kremenek  };
5227ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek}
5228ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek
5229ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisstatic void annotatePreprocessorTokens(CXTranslationUnit TU,
5230ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                       SourceRange RegionOfInterest,
5231ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                       AnnotateTokensData &Annotated) {
5232ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
5233ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5234ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SourceManager &SourceMgr = CXXUnit->getSourceManager();
5235ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  std::pair<FileID, unsigned> BeginLocInfo
5236ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(RegionOfInterest.getBegin());
5237ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  std::pair<FileID, unsigned> EndLocInfo
5238ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(RegionOfInterest.getEnd());
5239ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5240ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (BeginLocInfo.first != EndLocInfo.first)
5241ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
5242ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5243ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  StringRef Buffer;
5244ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  bool Invalid = false;
5245ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
5246ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (Buffer.empty() || Invalid)
5247ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
5248ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5249ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
52504e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie            CXXUnit->getASTContext().getLangOpts(),
5251ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            Buffer.begin(), Buffer.data() + BeginLocInfo.second,
5252ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            Buffer.end());
5253ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Lex.SetCommentRetentionState(true);
5254ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5255ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  // Lex tokens in raw mode until we hit the end of the range, to avoid
5256ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  // entering #includes or expanding macros.
5257ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  while (true) {
5258ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    Token Tok;
5259ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    Lex.LexFromRawLexer(Tok);
5260ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5261ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  reprocess:
5262ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
5263ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // We have found a preprocessing directive. Gobble it up so that we
5264ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // don't see it while preprocessing these tokens later, but keep track
5265ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // of all of the token locations inside this preprocessing directive so
5266ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // that we can annotate them appropriately.
5267ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      //
5268ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // FIXME: Some simple tests here could identify macro definitions and
5269ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // #undefs, to provide specific cursor kinds for those.
5270ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      SmallVector<SourceLocation, 32> Locations;
5271ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      do {
5272ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Locations.push_back(Tok.getLocation());
5273ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Lex.LexFromRawLexer(Tok);
5274ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      } while (!Tok.isAtStartOfLine() && !Tok.is(tok::eof));
5275ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5276ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      using namespace cxcursor;
5277ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      CXCursor Cursor
5278ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      = MakePreprocessingDirectiveCursor(SourceRange(Locations.front(),
5279ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                                     Locations.back()),
5280ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                         TU);
5281ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      for (unsigned I = 0, N = Locations.size(); I != N; ++I) {
5282ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Annotated[Locations[I].getRawEncoding()] = Cursor;
5283ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      }
5284ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5285ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      if (Tok.isAtStartOfLine())
5286ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        goto reprocess;
5287ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5288ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      continue;
5289ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    }
5290ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5291ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    if (Tok.is(tok::eof))
5292ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      break;
5293ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
5294ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis}
5295ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
52966628a614c504263ae539462f049d523dd07ac1baTed Kremenek// This gets run a separate thread to avoid stack blowout.
52976628a614c504263ae539462f049d523dd07ac1baTed Kremenekstatic void clang_annotateTokensImpl(void *UserData) {
52986628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXTranslationUnit TU = ((clang_annotateTokens_Data*)UserData)->TU;
52996628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = ((clang_annotateTokens_Data*)UserData)->CXXUnit;
53006628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXToken *Tokens = ((clang_annotateTokens_Data*)UserData)->Tokens;
53016628a614c504263ae539462f049d523dd07ac1baTed Kremenek  const unsigned NumTokens = ((clang_annotateTokens_Data*)UserData)->NumTokens;
53026628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor *Cursors = ((clang_annotateTokens_Data*)UserData)->Cursors;
5303fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5304fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  CIndexer *CXXIdx = (CIndexer*)TU->CIdx;
5305fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
530681b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
5307fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
53080396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Determine the region of interest, which contains all of the tokens.
53090045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  SourceRange RegionOfInterest;
53106628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setBegin(
53116628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU, Tokens[0])));
53126628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setEnd(
53136628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU,
53146628a614c504263ae539462f049d523dd07ac1baTed Kremenek                                                         Tokens[NumTokens-1])));
5315fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
53160396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // A mapping from the source locations found when re-lexing or traversing the
53170396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // region of interest to the corresponding cursors.
53180045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  AnnotateTokensData Annotated;
5319ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5320fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Relex the tokens within the source range to look for preprocessing
53210396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // directives.
5322ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  annotatePreprocessorTokens(TU, RegionOfInterest, Annotated);
53236628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5324a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
5325a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // Search and mark tokens that are macro argument expansions.
5326a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    MarkMacroArgTokensVisitor Visitor(CXXUnit->getSourceManager(),
5327a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                      Tokens, NumTokens);
5328a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    CursorVisitor MacroArgMarker(TU,
5329a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                 MarkMacroArgTokensVisitorDelegate, &Visitor,
5330f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                 /*VisitPreprocessorLast=*/true,
5331e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                                 /*VisitIncludedEntities=*/false,
5332f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                 RegionOfInterest);
5333a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    MacroArgMarker.visitPreprocessedEntitiesInRegion();
5334a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5335a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
53360396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Annotate all of the source locations in the region of interest that map to
5337fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // a specific cursor.
5338fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  AnnotateTokensWorker W(Annotated, Tokens, Cursors, NumTokens,
5339a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                         TU, RegionOfInterest);
53406628a614c504263ae539462f049d523dd07ac1baTed Kremenek
53416c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // FIXME: We use a ridiculous stack size here because the data-recursion
53426c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm uses a large stack frame than the non-data recursive version,
53436c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // and AnnotationTokensWorker currently transforms the data-recursion
53446c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm back into a traditional recursion by explicitly calling
53456c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // VisitChildren().  We will need to remove this explicit recursive call.
53466628a614c504263ae539462f049d523dd07ac1baTed Kremenek  W.AnnotateTokens();
53476628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5348f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // If we ran into any entities that involve context-sensitive keywords,
5349f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // take another pass through the tokens to mark them as such.
5350f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (W.hasContextSensitiveKeywords()) {
5351f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    for (unsigned I = 0; I != NumTokens; ++I) {
5352f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (clang_getTokenKind(Tokens[I]) != CXToken_Identifier)
5353f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5354f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
5355f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCPropertyDecl) {
5356f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
5357f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (ObjCPropertyDecl *Property
53586628a614c504263ae539462f049d523dd07ac1baTed Kremenek            = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(Cursors[I]))) {
5359f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if (Property->getPropertyAttributesAsWritten() != 0 &&
5360f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              llvm::StringSwitch<bool>(II->getName())
53616628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readonly", true)
53626628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("assign", true)
5363f85e193739c953358c865005855253af4f68a497John McCall              .Case("unsafe_unretained", true)
53646628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readwrite", true)
53656628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("retain", true)
53666628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("copy", true)
53676628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("nonatomic", true)
53686628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("atomic", true)
53696628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("getter", true)
53706628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("setter", true)
5371f85e193739c953358c865005855253af4f68a497John McCall              .Case("strong", true)
5372f85e193739c953358c865005855253af4f68a497John McCall              .Case("weak", true)
53736628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Default(false))
5374f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            Tokens[I].int_data[0] = CXToken_Keyword;
5375f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
5376f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5377f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5378f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
5379f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCInstanceMethodDecl ||
5380f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_ObjCClassMethodDecl) {
5381f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
5382f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (llvm::StringSwitch<bool>(II->getName())
53836628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("in", true)
53846628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("out", true)
53856628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("inout", true)
53866628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("oneway", true)
53876628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("bycopy", true)
53886628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("byref", true)
53896628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Default(false))
5390f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Tokens[I].int_data[0] = CXToken_Keyword;
5391f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5392f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
53936639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis
53946639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      if (Cursors[I].kind == CXCursor_CXXFinalAttr ||
53956639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis          Cursors[I].kind == CXCursor_CXXOverrideAttr) {
53966639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis        Tokens[I].int_data[0] = CXToken_Keyword;
53976628a614c504263ae539462f049d523dd07ac1baTed Kremenek        continue;
5398f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5399f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
5400f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
5401fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
54026628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54036628a614c504263ae539462f049d523dd07ac1baTed Kremenekextern "C" {
54046628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54056628a614c504263ae539462f049d523dd07ac1baTed Kremenekvoid clang_annotateTokens(CXTranslationUnit TU,
54066628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXToken *Tokens, unsigned NumTokens,
54076628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXCursor *Cursors) {
54086628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54096628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (NumTokens == 0 || !Tokens || !Cursors)
54106628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
54116628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54126628a614c504263ae539462f049d523dd07ac1baTed Kremenek  // Any token we don't specifically annotate will have a NULL cursor.
54136628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor C = clang_getNullCursor();
54146628a614c504263ae539462f049d523dd07ac1baTed Kremenek  for (unsigned I = 0; I != NumTokens; ++I)
54156628a614c504263ae539462f049d523dd07ac1baTed Kremenek    Cursors[I] = C;
54166628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54176628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
54186628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!CXXUnit)
54196628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
54206628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54216628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
54226628a614c504263ae539462f049d523dd07ac1baTed Kremenek
54236628a614c504263ae539462f049d523dd07ac1baTed Kremenek  clang_annotateTokens_Data data = { TU, CXXUnit, Tokens, NumTokens, Cursors };
54246628a614c504263ae539462f049d523dd07ac1baTed Kremenek  llvm::CrashRecoveryContext CRC;
54256628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!RunSafely(CRC, clang_annotateTokensImpl, &data,
54266628a614c504263ae539462f049d523dd07ac1baTed Kremenek                 GetSafetyThreadStackSize() * 2)) {
54276628a614c504263ae539462f049d523dd07ac1baTed Kremenek    fprintf(stderr, "libclang: crash detected while annotating tokens\n");
54286628a614c504263ae539462f049d523dd07ac1baTed Kremenek  }
54296628a614c504263ae539462f049d523dd07ac1baTed Kremenek}
54306628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5431fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} // end: extern "C"
5432fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
5433fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
543416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek// Operations for querying linkage of a cursor.
543516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
543616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
543716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenekextern "C" {
543816b4259aecaa22b642d35d36fd89965ed700c1e0Ted KremenekCXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
54390396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
54400396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    return CXLinkage_Invalid;
54410396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
544216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  Decl *D = cxcursor::getCursorDecl(cursor);
544316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  if (NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
544416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    switch (ND->getLinkage()) {
544516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case NoLinkage: return CXLinkage_NoLinkage;
544616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case InternalLinkage: return CXLinkage_Internal;
544716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
544816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case ExternalLinkage: return CXLinkage_External;
544916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    };
545016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
545116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  return CXLinkage_Invalid;
545216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek}
545316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek} // end: extern "C"
545416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
545516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
545645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek// Operations for querying language of a cursor.
545745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
545845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
545945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekstatic CXLanguageKind getDeclLanguage(const Decl *D) {
546016ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  if (!D)
546116ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    return CXLanguage_C;
546216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
546345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  switch (D->getKind()) {
546445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    default:
546545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      break;
546645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ImplicitParam:
546745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCAtDefsField:
546845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategory:
546945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategoryImpl:
547045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCompatibleAlias:
547145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCImplementation:
547245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCInterface:
547345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCIvar:
547445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCMethod:
547545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProperty:
547645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCPropertyImpl:
547745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProtocol:
547845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_ObjC;
547945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConstructor:
548045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConversion:
548145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXDestructor:
548245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXMethod:
548345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXRecord:
548445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplate:
548545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplatePartialSpecialization:
548645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplateSpecialization:
548745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Friend:
548845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FriendTemplate:
548945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FunctionTemplate:
549045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::LinkageSpec:
549145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Namespace:
549245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NamespaceAlias:
549345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NonTypeTemplateParm:
549445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::StaticAssert:
549545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTemplateParm:
549645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTypeParm:
549745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingTypename:
549845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingValue:
549945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Using:
550045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingDirective:
550145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingShadow:
550245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_CPlusPlus;
550345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  }
550445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
550545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_C;
550645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
550745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
550845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekextern "C" {
550958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
551058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregorenum CXAvailabilityKind clang_getCursorAvailability(CXCursor cursor) {
551158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  if (clang_isDeclaration(cursor.kind))
551258ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    if (Decl *D = cxcursor::getCursorDecl(cursor)) {
55130a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      if (isa<FunctionDecl>(D) && cast<FunctionDecl>(D)->isDeleted())
551458ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Available;
551558ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
55160a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      switch (D->getAvailability()) {
55170a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Available:
55180a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_NotYetIntroduced:
55190a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_Available;
55200a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
55210a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Deprecated:
552258ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Deprecated;
55230a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
55240a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Unavailable:
55250a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_NotAvailable;
55260a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      }
552758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    }
55280a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
552958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  return CXAvailability_Available;
553058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor}
553158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
5532cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregorstatic CXVersion convertVersion(VersionTuple In) {
5533cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  CXVersion Out = { -1, -1, -1 };
5534cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (In.empty())
5535cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    return Out;
5536cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5537cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  Out.Major = In.getMajor();
5538cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5539cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (llvm::Optional<unsigned> Minor = In.getMinor())
5540cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    Out.Minor = *Minor;
5541cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  else
5542cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    return Out;
5543cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5544cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (llvm::Optional<unsigned> Subminor = In.getSubminor())
5545cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    Out.Subminor = *Subminor;
5546cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5547cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  return Out;
5548cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor}
5549cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5550cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregorint clang_getCursorPlatformAvailability(CXCursor cursor,
5551cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        int *always_deprecated,
5552cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        CXString *deprecated_message,
5553cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        int *always_unavailable,
5554cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        CXString *unavailable_message,
5555cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        CXPlatformAvailability *availability,
5556cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor                                        int availability_size) {
5557cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (always_deprecated)
5558cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    *always_deprecated = 0;
5559cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (deprecated_message)
5560cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    *deprecated_message = cxstring::createCXString("", /*DupString=*/false);
5561cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (always_unavailable)
5562cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    *always_unavailable = 0;
5563cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (unavailable_message)
5564cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    *unavailable_message = cxstring::createCXString("", /*DupString=*/false);
5565cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5566cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
5567cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    return 0;
5568cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5569cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  Decl *D = cxcursor::getCursorDecl(cursor);
5570cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  if (!D)
5571cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    return 0;
5572cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5573cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  int N = 0;
5574cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  for (Decl::attr_iterator A = D->attr_begin(), AEnd = D->attr_end(); A != AEnd;
5575cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor       ++A) {
5576cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    if (DeprecatedAttr *Deprecated = dyn_cast<DeprecatedAttr>(*A)) {
5577cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      if (always_deprecated)
5578cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        *always_deprecated = 1;
5579cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      if (deprecated_message)
5580cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        *deprecated_message = cxstring::createCXString(Deprecated->getMessage());
5581cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      continue;
5582cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    }
5583cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5584cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    if (UnavailableAttr *Unavailable = dyn_cast<UnavailableAttr>(*A)) {
5585cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      if (always_unavailable)
5586cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        *always_unavailable = 1;
5587cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      if (unavailable_message) {
5588cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        *unavailable_message
5589cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor          = cxstring::createCXString(Unavailable->getMessage());
5590cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      }
5591cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      continue;
5592cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    }
5593cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5594cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    if (AvailabilityAttr *Avail = dyn_cast<AvailabilityAttr>(*A)) {
5595cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      if (N < availability_size) {
5596cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Platform
5597cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor          = cxstring::createCXString(Avail->getPlatform()->getName());
5598cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Introduced = convertVersion(Avail->getIntroduced());
5599cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Deprecated = convertVersion(Avail->getDeprecated());
5600cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Obsoleted = convertVersion(Avail->getObsoleted());
5601cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Unavailable = Avail->getUnavailable();
5602cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor        availability[N].Message = cxstring::createCXString(Avail->getMessage());
5603cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      }
5604cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor      ++N;
5605cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor    }
5606cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  }
5607cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5608cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  return N;
5609cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor}
5610cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
5611cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregorvoid clang_disposeCXPlatformAvailability(CXPlatformAvailability *availability) {
5612cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  clang_disposeString(availability->Platform);
5613cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor  clang_disposeString(availability->Message);
5614cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor}
5615cc889664dec7776ebb598e4584e7df5ba2f59ab4Douglas Gregor
561645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted KremenekCXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
561745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  if (clang_isDeclaration(cursor.kind))
561845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    return getDeclLanguage(cxcursor::getCursorDecl(cursor));
561945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
562045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_Invalid;
562145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
56223910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
56233910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// \brief If the given cursor is the "templated" declaration
56243910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// descibing a class or function template, return the class or
56253910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// function template.
56263910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregorstatic Decl *maybeGetTemplateCursor(Decl *D) {
56273910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (!D)
56283910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    return 0;
56293910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
56303910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
56313910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (FunctionTemplateDecl *FunTmpl = FD->getDescribedFunctionTemplate())
56323910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return FunTmpl;
56333910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
56343910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D))
56353910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (ClassTemplateDecl *ClassTmpl = RD->getDescribedClassTemplate())
56363910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return ClassTmpl;
56373910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
56383910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  return D;
56393910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor}
56403910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
56412be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorSemanticParent(CXCursor cursor) {
56422be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
56432be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
56442be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getDeclContext();
56453910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
56463910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
56473910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
56483910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
56493910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
56502be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
56512be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
56522be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
56532be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isStatement(cursor.kind) || clang_isExpression(cursor.kind)) {
56542be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor))
5655a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(D, getCursorTU(cursor));
56562be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
56572be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
56582be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
56592be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
56602be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
56612be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorLexicalParent(CXCursor cursor) {
56622be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
56632be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
56642be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getLexicalDeclContext();
56653910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
56663910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
56673910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
56683910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
56693910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
56702be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
56712be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
56722be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
56732be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // FIXME: Note that we can't easily compute the lexical context of a
56742be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // statement or expression, so we return nothing.
56752be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
56762be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
56772be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
5678ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas GregorCXFile clang_getIncludedFile(CXCursor cursor) {
5679ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (cursor.kind != CXCursor_InclusionDirective)
5680ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return 0;
5681ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
5682ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  InclusionDirective *ID = getCursorInclusionDirective(cursor);
5683ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  return (void *)ID->getFile();
5684ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor}
5685aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5686aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri GribenkoCXSourceRange clang_Cursor_getCommentRange(CXCursor C) {
5687aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  if (!clang_isDeclaration(C.kind))
5688aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko    return clang_getNullRange();
5689aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5690aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  const Decl *D = getCursorDecl(C);
5691aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  ASTContext &Context = getCursorContext(C);
5692f50555eedef33fd5a67d369aa0ae8a6f1d201543Dmitri Gribenko  const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
5693aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  if (!RC)
5694aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko    return clang_getNullRange();
5695aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5696aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  return cxloc::translateSourceRange(Context, RC->getSourceRange());
5697aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko}
5698aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5699aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri GribenkoCXString clang_Cursor_getRawCommentText(CXCursor C) {
5700aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  if (!clang_isDeclaration(C.kind))
5701aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko    return createCXString((const char *) NULL);
5702aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5703aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  const Decl *D = getCursorDecl(C);
5704aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  ASTContext &Context = getCursorContext(C);
5705f50555eedef33fd5a67d369aa0ae8a6f1d201543Dmitri Gribenko  const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
5706aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  StringRef RawText = RC ? RC->getRawText(Context.getSourceManager()) :
5707aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko                           StringRef();
5708aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
5709aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  // Don't duplicate the string because RawText points directly into source
5710aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  // code.
5711aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko  return createCXString(RawText, false);
5712aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko}
5713aa0cd85838f2a024e589ea4e8c2094130065af21Dmitri Gribenko
57142d44d77fed3200e2eff289f55493317e90d3398cDmitri GribenkoCXString clang_Cursor_getBriefCommentText(CXCursor C) {
57152d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko  if (!clang_isDeclaration(C.kind))
57162d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko    return createCXString((const char *) NULL);
57172d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko
57182d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko  const Decl *D = getCursorDecl(C);
57192d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko  const ASTContext &Context = getCursorContext(C);
5720f50555eedef33fd5a67d369aa0ae8a6f1d201543Dmitri Gribenko  const RawComment *RC = Context.getRawCommentForAnyRedecl(D);
57212d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko
57228376f5934a18b950ac7323d8a38ed231623010faDmitri Gribenko  if (RC) {
57232d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko    StringRef BriefText = RC->getBriefText(Context);
57242d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko
57252d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko    // Don't duplicate the string because RawComment ensures that this memory
57262d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko    // will not go away.
57272d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko    return createCXString(BriefText, false);
57282d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko  }
57292d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko
57302d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko  return createCXString((const char *) NULL);
57312d44d77fed3200e2eff289f55493317e90d3398cDmitri Gribenko}
57329ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
5733ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri GribenkoCXComment clang_Cursor_getParsedComment(CXCursor C) {
5734ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko  if (!clang_isDeclaration(C.kind))
5735ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko    return cxcomment::createCXComment(NULL);
5736ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko
5737ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko  const Decl *D = getCursorDecl(C);
5738ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko  const ASTContext &Context = getCursorContext(C);
5739ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko  const comments::FullComment *FC = Context.getCommentForDecl(D);
5740ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko
5741ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko  return cxcomment::createCXComment(FC);
5742ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko}
5743ae99b75fbbac1deaccdcc1b326b8fb6b07a1e72dDmitri Gribenko
5744b619e7877f09f2984d06629fd8653f3f333d8ea2Dmitri Gribenko} // end: extern "C"
5745b619e7877f09f2984d06629fd8653f3f333d8ea2Dmitri Gribenko
57469ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
57479ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek// C++ AST instrospection.
57489ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
57499ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
57509ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekextern "C" {
57519ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekunsigned clang_CXXMethod_isStatic(CXCursor C) {
57529ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek  if (!clang_isDeclaration(C.kind))
57539ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek    return 0;
575449f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor
575549f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  CXXMethodDecl *Method = 0;
575649f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
575749f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
575849f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
575949f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  else
576049f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
576149f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  return (Method && Method->isStatic()) ? 1 : 0;
576240b492a43bac3ed0c465772aa6921d011cfc273fTed Kremenek}
5763b12903e1a4b8d1b611b8c7e4f910665d628e68cdTed Kremenek
5764211924b563aa31421836cee7655be729ad02733fDouglas Gregorunsigned clang_CXXMethod_isVirtual(CXCursor C) {
5765211924b563aa31421836cee7655be729ad02733fDouglas Gregor  if (!clang_isDeclaration(C.kind))
5766211924b563aa31421836cee7655be729ad02733fDouglas Gregor    return 0;
5767211924b563aa31421836cee7655be729ad02733fDouglas Gregor
5768211924b563aa31421836cee7655be729ad02733fDouglas Gregor  CXXMethodDecl *Method = 0;
5769211924b563aa31421836cee7655be729ad02733fDouglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
5770211924b563aa31421836cee7655be729ad02733fDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
5771211924b563aa31421836cee7655be729ad02733fDouglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
5772211924b563aa31421836cee7655be729ad02733fDouglas Gregor  else
5773211924b563aa31421836cee7655be729ad02733fDouglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
5774211924b563aa31421836cee7655be729ad02733fDouglas Gregor  return (Method && Method->isVirtual()) ? 1 : 0;
5775211924b563aa31421836cee7655be729ad02733fDouglas Gregor}
57769ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek} // end: extern "C"
57779ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
577845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
577995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek// Attribute introspection.
578095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
578195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
578295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenekextern "C" {
578395f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted KremenekCXType clang_getIBOutletCollectionType(CXCursor C) {
578495f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  if (C.kind != CXCursor_IBOutletCollectionAttr)
5785a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return cxtype::MakeCXType(QualType(), cxcursor::getCursorTU(C));
578695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
578795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  IBOutletCollectionAttr *A =
578895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek    cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(C));
578995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
579018aa2ff4641847d7f8866e8c5912d4d0ddb858ceArgyrios Kyrtzidis  return cxtype::MakeCXType(A->getInterface(), cxcursor::getCursorTU(C));
579195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek}
579295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek} // end: extern "C"
579395f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
579495f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
579559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek// Inspecting memory usage.
579659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek//===----------------------------------------------------------------------===//
579759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5798f787002478f09af1741fb0f82a562002e6799c49Ted Kremenektypedef std::vector<CXTUResourceUsageEntry> MemUsageEntries;
579959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5800f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekstatic inline void createCXTUResourceUsageEntry(MemUsageEntries &entries,
5801f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek                                              enum CXTUResourceUsageKind k,
5802ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek                                              unsigned long amount) {
5803f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  CXTUResourceUsageEntry entry = { k, amount };
580459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  entries.push_back(entry);
580559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
580659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
580759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenekextern "C" {
580859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5809f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekconst char *clang_getTUResourceUsageName(CXTUResourceUsageKind kind) {
581059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  const char *str = "";
581159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  switch (kind) {
5812f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_AST:
581359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: expressions, declarations, and types";
581459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      break;
5815f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_Identifiers:
581659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: identifiers";
581759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      break;
5818f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_Selectors:
581959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: selectors";
5820e294ab717fc9535429ca5d8f575d41ae4441d822Ted Kremenek      break;
5821f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_GlobalCompletionResults:
58224e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek      str = "Code completion: cached global results";
5823e294ab717fc9535429ca5d8f575d41ae4441d822Ted Kremenek      break;
5824457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek    case CXTUResourceUsage_SourceManagerContentCache:
5825457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek      str = "SourceManager: content cache allocator";
5826457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek      break;
5827ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    case CXTUResourceUsage_AST_SideTables:
5828ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek      str = "ASTContext: side tables";
5829ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek      break;
5830f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    case CXTUResourceUsage_SourceManager_Membuffer_Malloc:
5831f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      str = "SourceManager: malloc'ed memory buffers";
5832f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      break;
5833f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    case CXTUResourceUsage_SourceManager_Membuffer_MMap:
5834f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      str = "SourceManager: mmap'ed memory buffers";
5835f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      break;
5836e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    case CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc:
5837e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      str = "ExternalASTSource: malloc'ed memory buffers";
5838e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      break;
5839e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    case CXTUResourceUsage_ExternalASTSource_Membuffer_MMap:
5840e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      str = "ExternalASTSource: mmap'ed memory buffers";
5841e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      break;
58425e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    case CXTUResourceUsage_Preprocessor:
58435e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      str = "Preprocessor: malloc'ed memory";
58445e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      break;
58455e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    case CXTUResourceUsage_PreprocessingRecord:
58465e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      str = "Preprocessor: PreprocessingRecord";
58475e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      break;
5848ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek    case CXTUResourceUsage_SourceManager_DataStructures:
5849ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek      str = "SourceManager: data structures and tables";
5850ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek      break;
5851d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek    case CXTUResourceUsage_Preprocessor_HeaderSearch:
5852d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek      str = "Preprocessor: header search tables";
5853d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek      break;
585459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  }
585559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  return str;
585659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
585759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5858f787002478f09af1741fb0f82a562002e6799c49Ted KremenekCXTUResourceUsage clang_getCXTUResourceUsage(CXTranslationUnit TU) {
585959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  if (!TU) {
5860f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    CXTUResourceUsage usage = { (void*) 0, 0, 0 };
586159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    return usage;
586259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  }
586359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
586459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  ASTUnit *astUnit = static_cast<ASTUnit*>(TU->TUData);
58651e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<MemUsageEntries> entries(new MemUsageEntries());
586659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  ASTContext &astContext = astUnit->getASTContext();
586759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
586859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used by AST nodes and types?
5869f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST,
5870ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    (unsigned long) astContext.getASTAllocatedMemory());
587159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
587259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used by identifiers?
5873f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Identifiers,
587459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    (unsigned long) astContext.Idents.getAllocator().getTotalMemory());
587559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
587659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used for selectors?
5877f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Selectors,
587859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    (unsigned long) astContext.Selectors.getTotalMemory());
587959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5880ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek  // How much memory is used by ASTContext's side tables?
5881ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST_SideTables,
5882ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    (unsigned long) astContext.getSideTableAllocatedMemory());
5883ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek
58844e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  // How much memory is used for caching global code completion results?
58854e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  unsigned long completionBytes = 0;
58864e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  if (GlobalCodeCompletionAllocator *completionAllocator =
58874e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek      astUnit->getCachedCompletionAllocator().getPtr()) {
58885e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    completionBytes = completionAllocator->getTotalMemory();
58894e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  }
5890457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  createCXTUResourceUsageEntry(*entries,
5891457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek                               CXTUResourceUsage_GlobalCompletionResults,
5892457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek                               completionBytes);
5893457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek
5894457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  // How much memory is being used by SourceManager's content cache?
5895457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  createCXTUResourceUsageEntry(*entries,
5896457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek          CXTUResourceUsage_SourceManagerContentCache,
5897457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek          (unsigned long) astContext.getSourceManager().getContentCacheSize());
5898f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek
5899f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  // How much memory is being used by the MemoryBuffer's in SourceManager?
5900f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  const SourceManager::MemoryBufferSizes &srcBufs =
5901f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    astUnit->getSourceManager().getMemoryBufferSizes();
5902f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek
5903f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5904f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               CXTUResourceUsage_SourceManager_Membuffer_Malloc,
5905f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               (unsigned long) srcBufs.malloc_bytes);
5906ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5907f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               CXTUResourceUsage_SourceManager_Membuffer_MMap,
5908f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               (unsigned long) srcBufs.mmap_bytes);
5909ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5910ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                               CXTUResourceUsage_SourceManager_DataStructures,
5911ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                               (unsigned long) astContext.getSourceManager()
5912ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                                .getDataStructureSizes());
5913e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek
5914e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  // How much memory is being used by the ExternalASTSource?
5915e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  if (ExternalASTSource *esrc = astContext.getExternalSource()) {
5916e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    const ExternalASTSource::MemoryBufferSizes &sizes =
5917e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      esrc->getMemoryBufferSizes();
5918e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek
5919e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    createCXTUResourceUsageEntry(*entries,
5920e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc,
5921e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek                                 (unsigned long) sizes.malloc_bytes);
5922e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    createCXTUResourceUsageEntry(*entries,
5923e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      CXTUResourceUsage_ExternalASTSource_Membuffer_MMap,
5924e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek                                 (unsigned long) sizes.mmap_bytes);
5925e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  }
59265e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
59275e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  // How much memory is being used by the Preprocessor?
59285e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  Preprocessor &pp = astUnit->getPreprocessor();
59295e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  createCXTUResourceUsageEntry(*entries,
59305e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                               CXTUResourceUsage_Preprocessor,
5931c5c5e92ec53f7e6ac7ebbbf77c6d8e4b7d88daecArgyrios Kyrtzidis                               pp.getTotalMemory());
59325e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
59335e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  if (PreprocessingRecord *pRec = pp.getPreprocessingRecord()) {
59345e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    createCXTUResourceUsageEntry(*entries,
59355e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                                 CXTUResourceUsage_PreprocessingRecord,
59365e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                                 pRec->getTotalMemory());
59375e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  }
59385e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
5939d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5940d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek                               CXTUResourceUsage_Preprocessor_HeaderSearch,
5941d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek                               pp.getHeaderSearchInfo().getTotalMemory());
59425e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
5943f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  CXTUResourceUsage usage = { (void*) entries.get(),
594459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek                            (unsigned) entries->size(),
594559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek                            entries->size() ? &(*entries)[0] : 0 };
594659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  entries.take();
594759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  return usage;
594859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
594959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5950f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekvoid clang_disposeCXTUResourceUsage(CXTUResourceUsage usage) {
595159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  if (usage.data)
595259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    delete (MemUsageEntries*) usage.data;
595359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
595459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
595559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek} // end extern "C"
595659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
59576df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregorvoid clang::PrintLibclangResourceUsage(CXTranslationUnit TU) {
59586df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  CXTUResourceUsage Usage = clang_getCXTUResourceUsage(TU);
59596df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  for (unsigned I = 0; I != Usage.numEntries; ++I)
59606df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    fprintf(stderr, "  %s: %lu\n",
59616df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor            clang_getTUResourceUsageName(Usage.entries[I].kind),
59626df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor            Usage.entries[I].amount);
59636df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
59646df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  clang_disposeCXTUResourceUsage(Usage);
59656df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor}
59666df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
596759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek//===----------------------------------------------------------------------===//
596804bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek// Misc. utility functions.
596904bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek//===----------------------------------------------------------------------===//
5970f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
5971abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbar/// Default to using an 8 MB stack size on "safety" threads.
5972abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbarstatic unsigned SafetyStackThreadSize = 8 << 20;
5973bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5974bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarnamespace clang {
5975bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5976bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarbool RunSafely(llvm::CrashRecoveryContext &CRC,
59776c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               void (*Fn)(void*), void *UserData,
59786c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               unsigned Size) {
59796c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (!Size)
59806c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek    Size = GetSafetyThreadStackSize();
59816c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (Size)
5982bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar    return CRC.RunSafelyOnThread(Fn, UserData, Size);
5983bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return CRC.RunSafely(Fn, UserData);
5984bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5985bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5986bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarunsigned GetSafetyThreadStackSize() {
5987bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return SafetyStackThreadSize;
5988bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5989bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5990bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarvoid SetSafetyThreadStackSize(unsigned Value) {
5991bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  SafetyStackThreadSize = Value;
5992bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5993bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
59948e7c48a54b4325925f5edda1738a3537ab2c3c5eArgyrios Kyrtzidis}
59958e7c48a54b4325925f5edda1738a3537ab2c3c5eArgyrios Kyrtzidis
599681b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidisvoid clang::setThreadBackgroundPriority() {
5997fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  // FIXME: Move to llvm/Support and make it cross-platform.
5998fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis#ifdef __APPLE__
5999fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  setpriority(PRIO_DARWIN_THREAD, 0, PRIO_DARWIN_BG);
6000fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis#endif
6001fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis}
6002fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
60039793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidisvoid cxindex::printDiagsToStderr(ASTUnit *Unit) {
60049793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  if (!Unit)
60059793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis    return;
60069793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis
60079793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
60089793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis                                  DEnd = Unit->stored_diag_end();
60099793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis       D != DEnd; ++D) {
60109793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis    CXStoredDiagnostic Diag(*D, Unit->getASTContext().getLangOpts());
60119793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis    CXString Msg = clang_formatDiagnostic(&Diag,
60129793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis                                clang_defaultDiagnosticDisplayOptions());
60139793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis    fprintf(stderr, "%s\n", clang_getCString(Msg));
60149793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis    clang_disposeString(Msg);
60159793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  }
60169793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis#ifdef LLVM_ON_WIN32
60179793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  // On Windows, force a flush, since there may be multiple copies of
60189793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  // stderr and stdout in the file system, all with different buffers
60199793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  // but writing to the same device.
60209793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis  fflush(stderr);
60219793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis#endif
60229793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis}
60239793428697f01b139443b7a0a882ea5dd1967e56Argyrios Kyrtzidis
602404bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenekextern "C" {
602504bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
6026a2a9d6e4e5b6001b86b7dfc5db1ea296ce29a3d3Ted KremenekCXString clang_getClangVersion() {
6027ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(getClangFullVersion());
602804bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek}
602904bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
603004bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek} // end: extern "C"
603159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
6032