CIndex.cpp revision b085d898bdfe35097eba45f4072b0f6865f561dc
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"
1616c440a377b7ec8b722a2e2c7c864f75c95bd305Ted Kremenek#include "CXCursor.h"
170a90d32523bfe5fa63e11b648686c9699f786d15Ted Kremenek#include "CXTranslationUnit.h"
18ed122735639d83c10f18c28c7fd117bfcd0f62cbTed Kremenek#include "CXString.h"
1995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek#include "CXType.h"
20a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek#include "CXSourceLocation.h"
215352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor#include "CIndexDiagnostic.h"
22e397bf1bd90cfceb0166606ebcd2580b7671a828Argyrios Kyrtzidis#include "CursorVisitor.h"
23ab1889321f6f8f200f2b318ac26883ac18e49d03Ted Kremenek
2404bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek#include "clang/Basic/Version.h"
25936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor
26fb5704295c6137685a7b90b92cd6b958028740c8Steve Naroff#include "clang/AST/StmtVisitor.h"
27b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Basic/Diagnostic.h"
28b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Frontend/ASTUnit.h"
29b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Frontend/CompilerInstance.h"
30936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor#include "clang/Frontend/FrontendDiagnostic.h"
31d8210650ed948de65a08a8daf16d291b747717c4Ted Kremenek#include "clang/Lex/Lexer.h"
32dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor#include "clang/Lex/HeaderSearch.h"
33b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Lex/PreprocessingRecord.h"
3433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor#include "clang/Lex/Preprocessor.h"
35a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor#include "llvm/ADT/STLExtras.h"
36d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek#include "llvm/ADT/Optional.h"
37f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor#include "llvm/ADT/StringSwitch.h"
38b2c60b04a597cc5ba4154837cf8e0a155a376fd7Argyrios Kyrtzidis#include "llvm/Support/SaveAndRestore.h"
39c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar#include "llvm/Support/CrashRecoveryContext.h"
4048615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar#include "llvm/Support/PrettyStackTrace.h"
4102465750c8c3fa96b1e7e596b02297e24361dc4fDouglas Gregor#include "llvm/Support/MemoryBuffer.h"
42358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor#include "llvm/Support/raw_ostream.h"
437a07fcb8f10fe45ea65a0a41798eb1c40777bde4Douglas Gregor#include "llvm/Support/Timer.h"
4403013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Mutex.h"
4503013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Program.h"
4603013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Signals.h"
4703013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Threading.h"
4837f1ea0eb08a00fa90edbecb427cfbb50ca0f4d0Ted Kremenek#include "llvm/Support/Compiler.h"
49fc0622155fa61349698a8fd0053773c37d9f7ac4Ted Kremenek
5050398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroffusing namespace clang;
5116c440a377b7ec8b722a2e2c7c864f75c95bd305Ted Kremenekusing namespace clang::cxcursor;
52ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenekusing namespace clang::cxstring;
539049cf6cb468c856888e88251dab659955fa767eArgyrios Kyrtzidisusing namespace clang::cxtu;
5450398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff
55fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios KyrtzidisCXTranslationUnit cxtu::MakeCXTranslationUnit(CIndexer *CIdx, ASTUnit *TU) {
56a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  if (!TU)
57a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return 0;
58a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit D = new CXTranslationUnitImpl();
59fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  D->CIdx = CIdx;
60a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  D->TUData = TU;
61a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  D->StringPool = createCXStringPool();
62153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek  D->Diagnostics = 0;
63a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return D;
64a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek}
65a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek
664e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidiscxtu::CXTUOwner::~CXTUOwner() {
674e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis  if (TU)
684e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis    clang_disposeTranslationUnit(TU);
694e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis}
704e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis
71f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek/// \brief Compare two source ranges to determine their relative position in
7233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// the translation unit.
73f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekstatic RangeComparisonResult RangeCompare(SourceManager &SM,
74f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                          SourceRange R1,
7533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                                          SourceRange R2) {
7633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R1.isValid() && "First range is invalid?");
7733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R2.isValid() && "Second range is invalid?");
78a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R1.getEnd() != R2.getBegin() &&
79d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R1.getEnd(), R2.getBegin()))
8033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeBefore;
81a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R2.getEnd() != R1.getBegin() &&
82d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R2.getEnd(), R1.getBegin()))
8333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeAfter;
8433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return RangeOverlap;
8533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
8633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
87fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek/// \brief Determine if a source location falls within, before, or after a
88fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek///   a given source range.
89fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic RangeComparisonResult LocationCompare(SourceManager &SM,
90fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                             SourceLocation L, SourceRange R) {
91fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(R.isValid() && "First range is invalid?");
92fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(L.isValid() && "Second range is invalid?");
93a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (L == R.getBegin() || L == R.getEnd())
94fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeOverlap;
95fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(L, R.getBegin()))
96fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeBefore;
97fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(R.getEnd(), L))
98fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeAfter;
99fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return RangeOverlap;
100fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek}
101fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
10276dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// \brief Translate a Clang source range into a CIndex source range.
10376dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar///
10476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// Clang internally represents ranges where the end location points to the
10576dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// start of the token at the end. However, for external clients it is more
10676dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// useful to have a CXSourceRange be a proper half-open interval. This routine
10776dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// does the appropriate translation.
108f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed KremenekCXSourceRange cxloc::translateSourceRange(const SourceManager &SM,
10976dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                                          const LangOptions &LangOpts,
1100a76aae8c03cb7dd7bdbe683485560afaf695959Chris Lattner                                          const CharSourceRange &R) {
11176dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  // We want the last character in this location, so we will adjust the
1126a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor  // location accordingly.
11376dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  SourceLocation EndLoc = R.getEnd();
114a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (EndLoc.isValid() && EndLoc.isMacroID())
115edc3dccece244a584f8ebdb81da6c962c08e79beChandler Carruth    EndLoc = SM.getExpansionRange(EndLoc).second;
1160a76aae8c03cb7dd7bdbe683485560afaf695959Chris Lattner  if (R.isTokenRange() && !EndLoc.isInvalid() && EndLoc.isFileID()) {
1176a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor    unsigned Length = Lexer::MeasureTokenLength(EndLoc, SM, LangOpts);
118a64ccefdf0ea4e03ec88805d71b0af74950c7472Argyrios Kyrtzidis    EndLoc = EndLoc.getLocWithOffset(Length);
11976dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  }
12076dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar
12176dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  CXSourceRange Result = { { (void *)&SM, (void *)&LangOpts },
12276dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           R.getBegin().getRawEncoding(),
12376dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           EndLoc.getRawEncoding() };
12476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  return Result;
12576dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar}
1261db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
1278a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
12833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor// Cursor visitor.
1298a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
1308a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek
131a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C);
1326653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr);
1336653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
134a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
13533e9abd21083a0191a7676a04b497006d2da184dDouglas GregorRangeComparisonResult CursorVisitor::CompareRegionOfInterest(SourceRange R) {
136a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return RangeCompare(AU->getSourceManager(), R, RegionOfInterest);
13733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
13833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
139b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the given cursor and, if requested by the visitor,
140b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// its children.
141b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor///
14233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param Cursor the cursor to visit.
14333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
14433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param CheckRegionOfInterest if true, then the caller already checked that
14533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// this cursor is within the region of interest.
14633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
147b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
148b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
14933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorbool CursorVisitor::Visit(CXCursor Cursor, bool CheckedRegionOfInterest) {
150b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isInvalid(Cursor.kind))
151b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
152f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
153b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
154b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
15516ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (!D) {
15616ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      assert(0 && "Invalid declaration cursor");
15716ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      return true; // abort.
15816ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    }
15916ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
16065ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // Ignore implicit declarations, unless it's an objc method because
16165ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // currently we should report implicit methods for properties when indexing.
16265ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    if (D->isImplicit() && !isa<ObjCMethodDecl>(D))
163b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return false;
164b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
1650d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
16633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // If we have a range of interest, and this cursor doesn't intersect with it,
16733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // we're done.
16833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (RegionOfInterest.isValid() && !CheckedRegionOfInterest) {
169a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    SourceRange Range = getRawCursorExtent(Cursor);
170f408f32aa9ae3d97bc656267dc5d78fa7d03499bDaniel Dunbar    if (Range.isInvalid() || CompareRegionOfInterest(Range))
17133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor      return false;
17233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  }
173f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
174b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  switch (Visitor(Cursor, Parent, ClientData)) {
175b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Break:
176b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
1770d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
178b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Continue:
179b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
1802e331b938b38057e333fab0ba841130ea8467794Douglas Gregor
181b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Recurse:
182b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return VisitChildren(Cursor);
183b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
1840d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
1857530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid CXChildVisitResult!");
186b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
1870d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
188f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidisstatic bool visitPreprocessedEntitiesInRange(SourceRange R,
189f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                             PreprocessingRecord &PPRec,
190f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                             CursorVisitor &Visitor) {
191f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  SourceManager &SM = Visitor.getASTUnit()->getSourceManager();
192f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  FileID FID;
193f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
194e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis  if (!Visitor.shouldVisitIncludedEntities()) {
195f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    // If the begin/end of the range lie in the same FileID, do the optimization
196f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    // where we skip preprocessed entities that do not come from the same FileID.
197acca41167ce78bb032906f6b1d2ced62efbe059aArgyrios Kyrtzidis    FID = SM.getFileID(SM.getFileLoc(R.getBegin()));
198acca41167ce78bb032906f6b1d2ced62efbe059aArgyrios Kyrtzidis    if (FID != SM.getFileID(SM.getFileLoc(R.getEnd())))
199f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      FID = FileID();
200f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  }
201f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
202f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  std::pair<PreprocessingRecord::iterator, PreprocessingRecord::iterator>
203f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    Entities = PPRec.getPreprocessedEntitiesInRange(R);
204f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  return Visitor.visitPreprocessedEntities(Entities.first, Entities.second,
205f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                           PPRec, FID);
206f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis}
207f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
208dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidisvoid CursorVisitor::visitFileRegion() {
209dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (RegionOfInterest.isInvalid())
210dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
211dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
212dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  ASTUnit *Unit = static_cast<ASTUnit *>(TU->TUData);
213dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceManager &SM = Unit->getSourceManager();
214dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
215dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  std::pair<FileID, unsigned>
216dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Begin = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getBegin())),
217dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getEnd()));
218dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
219dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (End.first != Begin.first) {
220dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    // If the end does not reside in the same file, try to recover by
221dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    // picking the end of the file of begin location.
222dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End.first = Begin.first;
223dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End.second = SM.getFileIDSize(Begin.first);
224dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
225dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
226dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  assert(Begin.first == End.first);
227dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (Begin.second > End.second)
228dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
229dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
230dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  FileID File = Begin.first;
231dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  unsigned Offset = Begin.second;
232dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  unsigned Length = End.second - Begin.second;
233dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
234b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis  if (!VisitDeclsOnly && !VisitPreprocessorLast)
235b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis    if (visitPreprocessedEntitiesInRegion())
236b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis      return; // visitation break.
237dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
238dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  visitDeclsFromFileRegion(File, Offset, Length);
239dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
240b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis  if (!VisitDeclsOnly && VisitPreprocessorLast)
241dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    visitPreprocessedEntitiesInRegion();
242dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis}
243dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
244e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidisstatic bool isInLexicalContext(Decl *D, DeclContext *DC) {
245e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  if (!DC)
246e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    return false;
247e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
248e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  for (DeclContext *DeclDC = D->getLexicalDeclContext();
249e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis         DeclDC; DeclDC = DeclDC->getLexicalParent()) {
250e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    if (DeclDC == DC)
251e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis      return true;
252e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  }
253e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  return false;
254e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis}
255e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
256dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidisvoid CursorVisitor::visitDeclsFromFileRegion(FileID File,
257dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis                                             unsigned Offset, unsigned Length) {
258dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  ASTUnit *Unit = static_cast<ASTUnit *>(TU->TUData);
259dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceManager &SM = Unit->getSourceManager();
260dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceRange Range = RegionOfInterest;
261dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
262dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SmallVector<Decl *, 16> Decls;
263dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  Unit->findFileRegionDecls(File, Offset, Length, Decls);
264dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
265dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // If we didn't find any file level decls for the file, try looking at the
266dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // file that it was included from.
267c14a03dffff69b5e1c55cc118fc52d8fd9f3a28dArgyrios Kyrtzidis  while (Decls.empty() || Decls.front()->isTopLevelDeclInObjCContainer()) {
268dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    bool Invalid = false;
269dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    const SrcMgr::SLocEntry &SLEntry = SM.getSLocEntry(File, &Invalid);
270dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (Invalid)
271dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      return;
272dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
273dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    SourceLocation Outer;
274dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (SLEntry.isFile())
275dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      Outer = SLEntry.getFile().getIncludeLoc();
276dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    else
277dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      Outer = SLEntry.getExpansion().getExpansionLocStart();
278dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (Outer.isInvalid())
279dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      return;
280dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
281dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    llvm::tie(File, Offset) = SM.getDecomposedExpansionLoc(Outer);
282dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Length = 0;
283dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Unit->findFileRegionDecls(File, Offset, Length, Decls);
284dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
285dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
286dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  assert(!Decls.empty());
287dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
288dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  bool VisitedAtLeastOnce = false;
289e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis  DeclContext *CurDC = 0;
290dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SmallVector<Decl *, 16>::iterator DIt = Decls.begin();
291dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  for (SmallVector<Decl *, 16>::iterator DE = Decls.end(); DIt != DE; ++DIt) {
292dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Decl *D = *DIt;
293ed8bef44c0545fd55a78715606f8d733f6498b21Argyrios Kyrtzidis    if (D->getSourceRange().isInvalid())
294ed8bef44c0545fd55a78715606f8d733f6498b21Argyrios Kyrtzidis      continue;
295dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
296e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    if (isInLexicalContext(D, CurDC))
297e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis      continue;
298e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
299e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    CurDC = dyn_cast<DeclContext>(D);
300e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
301e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis    if (TagDecl *TD = dyn_cast<TagDecl>(D))
302e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis      if (!TD->isFreeStanding())
303e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis        continue;
304e2079cf54ded1eda9e35d215aef6628373368276Argyrios Kyrtzidis
305dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    RangeComparisonResult CompRes = RangeCompare(SM, D->getSourceRange(),Range);
306dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CompRes == RangeBefore)
307dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      continue;
308dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CompRes == RangeAfter)
309dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
310dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
311dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    assert(CompRes == RangeOverlap);
312dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    VisitedAtLeastOnce = true;
31303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
31403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (isa<ObjCContainerDecl>(D)) {
31503ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      FileDI_current = &DIt;
31603ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      FileDE_current = DE;
31703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    } else {
31803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      FileDI_current = 0;
31903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    }
32003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
321ba98617b994864b7554ff75445983ad02a962f45Argyrios Kyrtzidis    if (Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true))
322dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
323dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
324dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
325dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (VisitedAtLeastOnce)
326dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
327dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
328dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // No Decls overlapped with the range. Move up the lexical context until there
329dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // is a context that contains the range or we reach the translation unit
330dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // level.
331dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  DeclContext *DC = DIt == Decls.begin() ? (*DIt)->getLexicalDeclContext()
332dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis                                         : (*(DIt-1))->getLexicalDeclContext();
333dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
334dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  while (DC && !DC->isTranslationUnit()) {
335dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Decl *D = cast<Decl>(DC);
336dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    SourceRange CurDeclRange = D->getSourceRange();
337dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CurDeclRange.isInvalid())
338dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
339dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
340dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (RangeCompare(SM, CurDeclRange, Range) == RangeOverlap) {
341ba98617b994864b7554ff75445983ad02a962f45Argyrios Kyrtzidis      Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true);
342dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
343dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    }
344dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
345dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    DC = D->getLexicalDeclContext();
346dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
347dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis}
348dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
3494c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregorbool CursorVisitor::visitPreprocessedEntitiesInRegion() {
350b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis  if (!AU->getPreprocessor().getPreprocessingRecord())
351b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis    return false;
352b49e728a4d1a84b72f3aebf60ff494684f9cb004Argyrios Kyrtzidis
353788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  PreprocessingRecord &PPRec
354a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    = *AU->getPreprocessor().getPreprocessingRecord();
355f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  SourceManager &SM = AU->getSourceManager();
356788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
35792ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  if (RegionOfInterest.isValid()) {
358ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange MappedRange = AU->mapRangeToPreamble(RegionOfInterest);
359f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    SourceLocation B = MappedRange.getBegin();
360f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    SourceLocation E = MappedRange.getEnd();
361f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
362f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (AU->isInPreambleFileID(B)) {
363f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      if (SM.isLoadedSourceLocation(E))
364f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis        return visitPreprocessedEntitiesInRange(SourceRange(B, E),
365f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                                 PPRec, *this);
366f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
367f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // Beginning of range lies in the preamble but it also extends beyond
368f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // it into the main file. Split the range into 2 parts, one covering
369f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // the preamble and another covering the main file. This allows subsequent
370f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // calls to visitPreprocessedEntitiesInRange to accept a source range that
371f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // lies in the same FileID, allowing it to skip preprocessed entities that
372f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // do not come from the same FileID.
373f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      bool breaked =
374f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis        visitPreprocessedEntitiesInRange(
375f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                   SourceRange(B, AU->getEndOfPreambleFileID()),
376f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                          PPRec, *this);
377f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      if (breaked) return true;
378f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      return visitPreprocessedEntitiesInRange(
379f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                    SourceRange(AU->getStartOfMainFileID(), E),
380f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                        PPRec, *this);
381f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    }
382f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
383f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    return visitPreprocessedEntitiesInRange(SourceRange(B, E), PPRec, *this);
38492ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  }
38592ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis
386788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  bool OnlyLocalDecls
38732038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    = !AU->isMainFileAST() && AU->getOnlyLocalDecls();
38832038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor
38992ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  if (OnlyLocalDecls)
390f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    return visitPreprocessedEntities(PPRec.local_begin(), PPRec.local_end(),
391f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                     PPRec);
3924c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
393f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  return visitPreprocessedEntities(PPRec.begin(), PPRec.end(), PPRec);
3944c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor}
3954c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
3964c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregortemplate<typename InputIterator>
3974c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregorbool CursorVisitor::visitPreprocessedEntities(InputIterator First,
398f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              InputIterator Last,
399f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              PreprocessingRecord &PPRec,
400f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              FileID FID) {
4014c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  for (; First != Last; ++First) {
402f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (!FID.isInvalid() && !PPRec.isEntityInFileID(First, FID))
403f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      continue;
404f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
405f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    PreprocessedEntity *PPE = *First;
406f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (MacroExpansion *ME = dyn_cast<MacroExpansion>(PPE)) {
4074c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeMacroExpansionCursor(ME, TU)))
4084c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
4094c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4104c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
4114c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    }
4124c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
413f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (MacroDefinition *MD = dyn_cast<MacroDefinition>(PPE)) {
4144c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeMacroDefinitionCursor(MD, TU)))
4154c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
41689d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
4174c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
4184c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    }
4194c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
420f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (InclusionDirective *ID = dyn_cast<InclusionDirective>(PPE)) {
4214c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeInclusionDirectiveCursor(ID, TU)))
4224c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
4234c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4244c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
425788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    }
426788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  }
427788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
4284c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  return false;
429788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor}
430788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
431b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the children of the given cursor.
432a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek///
433b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
434b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
435f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekbool CursorVisitor::VisitChildren(CXCursor Cursor) {
436c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (clang_isReference(Cursor.kind) &&
437c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      Cursor.kind != CXCursor_CXXBaseSpecifier) {
438a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    // By definition, references have no children.
439a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return false;
440a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  }
441f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
442f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Set the Parent field to Cursor, then back to its old value once we're
443b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // done.
4440f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  SetParentRAII SetParent(Parent, StmtParent, Cursor);
445f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
446b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
447b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
44806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (!D)
44906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return false;
45006d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
451539311e0221df256c70c1c3080c8af847cd29dffTed Kremenek    return VisitAttributes(D) || Visit(D);
452b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
453f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
45406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isStatement(Cursor.kind)) {
45506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Stmt *S = getCursorStmt(Cursor))
45606d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(S);
45706d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
45806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
45906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
46006d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
46106d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isExpression(Cursor.kind)) {
46206d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Expr *E = getCursorExpr(Cursor))
46306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(E);
46406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
46506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
46606d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
467f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
468b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isTranslationUnit(Cursor.kind)) {
469a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CXTranslationUnit tu = getCursorTU(Cursor);
470a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    ASTUnit *CXXUnit = static_cast<ASTUnit*>(tu->TUData);
47104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
47204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    int VisitOrder[2] = { VisitPreprocessorLast, !VisitPreprocessorLast };
47304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    for (unsigned I = 0; I != 2; ++I) {
47404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      if (VisitOrder[I]) {
47504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
47604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            RegionOfInterest.isInvalid()) {
47704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          for (ASTUnit::top_level_iterator TL = CXXUnit->top_level_begin(),
47804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                        TLEnd = CXXUnit->top_level_end();
47904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor               TL != TLEnd; ++TL) {
480aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis            if (Visit(MakeCXCursor(*TL, tu, RegionOfInterest), true))
48104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
48204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
48304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        } else if (VisitDeclContext(
48404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                CXXUnit->getASTContext().getTranslationUnitDecl()))
4857b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor          return true;
48604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        continue;
4877b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor      }
4883178cb674ac8c3b59e1791e14d38d48619a1b621Bob Wilson
48904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      // Walk the preprocessing record.
4904c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (CXXUnit->getPreprocessor().getPreprocessingRecord())
4914c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        visitPreprocessedEntitiesInRegion();
4920396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    }
49304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
4947b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor    return false;
495b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
496f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
497c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (Cursor.kind == CXCursor_CXXBaseSpecifier) {
498c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    if (CXXBaseSpecifier *Base = getCursorCXXBaseSpecifier(Cursor)) {
499c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      if (TypeSourceInfo *BaseTSInfo = Base->getTypeSourceInfo()) {
500c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor        return Visit(BaseTSInfo->getTypeLoc());
501c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      }
502c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    }
503c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  }
504221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis
505221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis  if (Cursor.kind == CXCursor_IBOutletCollectionAttr) {
506221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis    IBOutletCollectionAttr *A =
507221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis      cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(Cursor));
508221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis    if (const ObjCInterfaceType *InterT = A->getInterface()->getAs<ObjCInterfaceType>())
509221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis      return Visit(cxcursor::MakeCursorObjCClassRef(InterT->getInterface(),
510221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis                                                    A->getInterfaceLoc(), TU));
511221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis  }
512221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis
513b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // Nothing to visit at the moment.
514b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
515dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
516dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5171ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenekbool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
51813c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor  if (TypeSourceInfo *TSInfo = B->getSignatureAsWritten())
51913c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor    if (Visit(TSInfo->getTypeLoc()))
52013c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor        return true;
5211ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
522664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  if (Stmt *Body = B->getBody())
523aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Body, StmtParent, TU, RegionOfInterest));
524664cffd330611d78fc0286f539589920a37ca328Ted Kremenek
525664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  return false;
5261ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek}
5271ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
528d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekllvm::Optional<bool> CursorVisitor::shouldVisitCursor(CXCursor Cursor) {
529d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (RegionOfInterest.isValid()) {
5306653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange Range = getFullCursorExtent(Cursor, AU->getSourceManager());
531d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Range.isInvalid())
532d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
5336653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
534d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    switch (CompareRegionOfInterest(Range)) {
535d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeBefore:
536d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes before the region of interest; skip it.
537d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
53823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
539d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeAfter:
540d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes after the region of interest; we're done.
541d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
542d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
543d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeOverlap:
544d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration overlaps the region of interest; visit it.
545d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
546d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
547d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
548d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return true;
549d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
550f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
551d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekbool CursorVisitor::VisitDeclContext(DeclContext *DC) {
552d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator I = DC->decls_begin(), E = DC->decls_end();
553f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
554d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually remove.  This part of a hack to support proper
555d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // iteration over all Decls contained lexically within an ObjC container.
556d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator*> DI_saved(DI_current, &I);
557d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator> DE_saved(DE_current, E);
558f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
559d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for ( ; I != E; ++I) {
560d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *D = *I;
561d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (D->getLexicalDeclContext() != DC)
562d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
563aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    CXCursor Cursor = MakeCXCursor(D, TU, RegionOfInterest);
5641836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis
5651836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    // FIXME: ObjCClassRef/ObjCProtocolRef for forward class/protocol
5661836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    // declarations is a mismatch with the compiler semantics.
5671836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    if (Cursor.kind == CXCursor_ObjCInterfaceDecl) {
5681836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      ObjCInterfaceDecl *ID = cast<ObjCInterfaceDecl>(D);
5691836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      if (!ID->isThisDeclarationADefinition())
5701836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis        Cursor = MakeCursorObjCClassRef(ID, ID->getLocation(), TU);
5711836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis
5721836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    } else if (Cursor.kind == CXCursor_ObjCProtocolDecl) {
5731836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      ObjCProtocolDecl *PD = cast<ObjCProtocolDecl>(D);
5741836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis      if (!PD->isThisDeclarationADefinition())
5751836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis        Cursor = MakeCursorObjCProtocolRef(PD, PD->getLocation(), TU);
5761836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis    }
5771836db0f2c7527ac11af0044ab89150f9aaf22e1Argyrios Kyrtzidis
578d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
579d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
580d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
581d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
582d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
583d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar    if (Visit(Cursor, true))
584b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
585b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
586b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
587dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
588dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5891ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
5901ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  llvm_unreachable("Translation units are visited directly by Visit()");
5911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
5921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
593162e1c1b487352434552147967c3dd296ebee2f7Richard Smithbool CursorVisitor::VisitTypeAliasDecl(TypeAliasDecl *D) {
594162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
595162e1c1b487352434552147967c3dd296ebee2f7Richard Smith    return Visit(TSInfo->getTypeLoc());
596162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
597162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return false;
598162e1c1b487352434552147967c3dd296ebee2f7Richard Smith}
599162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
6001ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
6011ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
6021ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(TSInfo->getTypeLoc());
603f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6041ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6051ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6061ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6071ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTagDecl(TagDecl *D) {
6081ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitDeclContext(D);
6091ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6101ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6110ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregorbool CursorVisitor::VisitClassTemplateSpecializationDecl(
6120ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor                                          ClassTemplateSpecializationDecl *D) {
6130ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool ShouldVisitBody = false;
6140ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  switch (D->getSpecializationKind()) {
6150ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_Undeclared:
6160ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ImplicitInstantiation:
6170ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    // Nothing to visit
6180ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return false;
6190ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6200ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDeclaration:
6210ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDefinition:
6220ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6230ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6240ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitSpecialization:
6250ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    ShouldVisitBody = true;
6260ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6270ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6280ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6290ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  // Visit the template arguments used in the specialization.
6300ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (TypeSourceInfo *SpecType = D->getTypeAsWritten()) {
6310ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    TypeLoc TL = SpecType->getTypeLoc();
6320ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    if (TemplateSpecializationTypeLoc *TSTLoc
6330ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          = dyn_cast<TemplateSpecializationTypeLoc>(&TL)) {
6340ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor      for (unsigned I = 0, N = TSTLoc->getNumArgs(); I != N; ++I)
6350ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor        if (VisitTemplateArgumentLoc(TSTLoc->getArgLoc(I)))
6360ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          return true;
6370ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    }
6380ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6390ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6400ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (ShouldVisitBody && VisitCXXRecordDecl(D))
6410ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return true;
6420ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6430ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  return false;
6440ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor}
6450ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
64674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregorbool CursorVisitor::VisitClassTemplatePartialSpecializationDecl(
64774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                   ClassTemplatePartialSpecializationDecl *D) {
64874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
64974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // before visiting these template parameters.
65074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
65174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return true;
65274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
65374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // Visit the partial specialization arguments.
65474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  const TemplateArgumentLoc *TemplateArgs = D->getTemplateArgsAsWritten();
65574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  for (unsigned I = 0, N = D->getNumTemplateArgsAsWritten(); I != N; ++I)
65674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    if (VisitTemplateArgumentLoc(TemplateArgs[I]))
65774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor      return true;
65874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
65974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  return VisitCXXRecordDecl(D);
66074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor}
66174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
662fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
66384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  // Visit the default argument.
66484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
66584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (TypeSourceInfo *DefArg = D->getDefaultArgumentInfo())
66684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      if (Visit(DefArg->getTypeLoc()))
66784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor        return true;
66884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
669fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
670fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
671fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
6721ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
6731ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInitExpr())
674aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
6751ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6761ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6771ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6787d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
6797d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
6807d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
6817d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      return true;
6827d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
683c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
684c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = DD->getQualifierLoc())
685c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
686c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
687c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
6887d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return false;
6897d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
6907d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
691a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor/// \brief Compare two base or member initializers based on their source order.
692cbb67480094b3bcb5b715acd827cbad55e2a204cSean Huntstatic int CompareCXXCtorInitializers(const void* Xp, const void *Yp) {
693cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *X
694cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Xp);
695cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *Y
696cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Yp);
697a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
698a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  if ((*X)->getSourceOrder() < (*Y)->getSourceOrder())
699a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return -1;
700a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else if ((*X)->getSourceOrder() > (*Y)->getSourceOrder())
701a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 1;
702a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else
703a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 0;
704a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor}
705a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
706b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
70701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (TypeSourceInfo *TSInfo = ND->getTypeSourceInfo()) {
70801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function declaration's syntactic components in the order
70901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // written. This requires a bit of work.
710723df245307a530da5433dfb43accf187dc3e243Abramo Bagnara    TypeLoc TL = TSInfo->getTypeLoc().IgnoreParens();
71101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    FunctionTypeLoc *FTL = dyn_cast<FunctionTypeLoc>(&TL);
71201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
71301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // If we have a function declared directly (without the use of a typedef),
71401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // visit just the return type. Otherwise, just visit the function's type
71501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // now.
71601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if ((FTL && !isa<CXXConversionDecl>(ND) && Visit(FTL->getResultLoc())) ||
71701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor        (!FTL && Visit(TL)))
71801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
71901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
720c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // Visit the nested-name-specifier, if present.
721c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (NestedNameSpecifierLoc QualifierLoc = ND->getQualifierLoc())
722c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      if (VisitNestedNameSpecifierLoc(QualifierLoc))
723c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor        return true;
72401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
72501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the declaration name.
72601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (VisitDeclarationNameInfo(ND->getNameInfo()))
72701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
72801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
72901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Visit explicitly-specified template arguments!
73001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
73101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function parameters, if we have a function type.
73201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (FTL && VisitFunctionTypeLoc(*FTL, true))
73301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
73401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
73501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Attributes?
73601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
73701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
73810620eb5164e31208fcbf0437cd79ae535ed0559Sean Hunt  if (ND->doesThisDeclarationHaveABody() && !ND->isLateTemplateParsed()) {
739a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    if (CXXConstructorDecl *Constructor = dyn_cast<CXXConstructorDecl>(ND)) {
740a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Find the initializers that were written in the source.
7415f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      SmallVector<CXXCtorInitializer *, 4> WrittenInits;
742a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (CXXConstructorDecl::init_iterator I = Constructor->init_begin(),
743a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                          IEnd = Constructor->init_end();
744a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor           I != IEnd; ++I) {
745a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (!(*I)->isWritten())
746a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          continue;
747a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
748a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        WrittenInits.push_back(*I);
749a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
750a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
751a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Sort the initializers in source order
752a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      llvm::array_pod_sort(WrittenInits.begin(), WrittenInits.end(),
753cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt                           &CompareCXXCtorInitializers);
754a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
755a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Visit the initializers in source order
756a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (unsigned I = 0, N = WrittenInits.size(); I != N; ++I) {
757cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt        CXXCtorInitializer *Init = WrittenInits[I];
75800eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet        if (Init->isAnyMemberInitializer()) {
75900eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet          if (Visit(MakeCursorMemberRef(Init->getAnyMember(),
760a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                        Init->getMemberLocation(), TU)))
761a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
76276852c218a207ef43583515cb835b6e855353a0fDouglas Gregor        } else if (TypeSourceInfo *TInfo = Init->getTypeSourceInfo()) {
76376852c218a207ef43583515cb835b6e855353a0fDouglas Gregor          if (Visit(TInfo->getTypeLoc()))
764a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
765a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        }
766a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
767a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        // Visit the initializer value.
768a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (Expr *Initializer = Init->getInit())
769aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis          if (Visit(MakeCXCursor(Initializer, ND, TU, RegionOfInterest)))
770a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
771a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
772a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
773a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
774aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
775a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return true;
776a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  }
777f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
778b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
779b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
780dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
7811ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
7821ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
7831ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
784f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7851ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *BitWidth = D->getBitWidth())
786aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(BitWidth, StmtParent, TU, RegionOfInterest));
787f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7881ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
7891ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
7901ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
7911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitVarDecl(VarDecl *D) {
7921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
7931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
794f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7951ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInit())
796aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
797f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7981ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
7991ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8001ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
80184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D) {
80284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitDeclaratorDecl(D))
80384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
80484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
80584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
80684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (Expr *DefArg = D->getDefaultArgument())
807aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(DefArg, StmtParent, TU, RegionOfInterest));
80884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
80984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
81084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
81184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
812fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
813fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the FunctionDecl
814fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // before visiting these template parameters.
815fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
816fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return true;
817fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
818fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return VisitFunctionDecl(D->getTemplatedDecl());
819fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
820fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
82139d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregorbool CursorVisitor::VisitClassTemplateDecl(ClassTemplateDecl *D) {
82239d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
82339d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // before visiting these template parameters.
82439d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
82539d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return true;
82639d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
82739d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  return VisitCXXRecordDecl(D->getTemplatedDecl());
82839d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor}
82939d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
83084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D) {
83184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
83284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
83384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
83484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited() &&
83584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      VisitTemplateArgumentLoc(D->getDefaultArgument()))
83684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
83784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
83884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
83984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
84084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
8411ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
8424bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor  if (TypeSourceInfo *TSInfo = ND->getResultTypeSourceInfo())
8434bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
8444bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor      return true;
8454bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor
846f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  for (ObjCMethodDecl::param_iterator P = ND->param_begin(),
8471ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       PEnd = ND->param_end();
8481ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       P != PEnd; ++P) {
849aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
8501ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
8511ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  }
852f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8531ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (ND->isThisDeclarationADefinition() &&
854aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
8551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
856f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8571ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8581ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8591ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
86003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidistemplate <typename DeclIt>
86103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidisstatic void addRangedDeclsInContainer(DeclIt *DI_current, DeclIt DE_current,
86203ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                      SourceManager &SM, SourceLocation EndLoc,
86303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                      SmallVectorImpl<Decl *> &Decls) {
86403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  DeclIt next = *DI_current;
86503ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  while (++next != DE_current) {
86603ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    Decl *D_next = *next;
86703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (!D_next)
86803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      break;
86903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    SourceLocation L = D_next->getLocStart();
87003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (!L.isValid())
87103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      break;
87203ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (SM.isBeforeInTranslationUnit(L, EndLoc)) {
87303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      *DI_current = next;
87403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      Decls.push_back(D_next);
87503ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      continue;
87603ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    }
87703ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    break;
87803ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  }
87903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis}
88003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
881d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremeneknamespace {
882d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  struct ContainerDeclsSort {
883d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    SourceManager &SM;
884d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    ContainerDeclsSort(SourceManager &sm) : SM(sm) {}
885d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    bool operator()(Decl *A, Decl *B) {
886d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_A = A->getLocStart();
887d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_B = B->getLocStart();
888d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      assert(L_A.isValid() && L_B.isValid());
889d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return SM.isBeforeInTranslationUnit(L_A, L_B);
890d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
891d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  };
892d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
893d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
894a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregorbool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
895d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually convert back to just 'VisitDeclContext()'.  Essentially
896d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // an @implementation can lexically contain Decls that are not properly
897d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // nested in the AST.  When we identify such cases, we need to retrofit
898d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // this nesting here.
89903ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  if (!DI_current && !FileDI_current)
900d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
901d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
902d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Scan the Decls that immediately come after the container
903d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // in the current DeclContext.  If any fall within the
904d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // container's lexical region, stash them into a vector
905d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // for later processing.
9065f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<Decl *, 24> DeclsInContainer;
907d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SourceLocation EndLoc = D->getSourceRange().getEnd();
908a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  SourceManager &SM = AU->getSourceManager();
909d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (EndLoc.isValid()) {
91003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (DI_current) {
91103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      addRangedDeclsInContainer(DI_current, DE_current, SM, EndLoc,
91203ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                DeclsInContainer);
91303ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    } else {
91403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      addRangedDeclsInContainer(FileDI_current, FileDE_current, SM, EndLoc,
91503ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis                                DeclsInContainer);
916d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
917d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
918d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
919d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // The common case.
920d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (DeclsInContainer.empty())
921d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
922d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
923d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Get all the Decls in the DeclContext, and sort them with the
924d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // additional ones we've collected.  Then visit them.
925d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for (DeclContext::decl_iterator I = D->decls_begin(), E = D->decls_end();
926d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek       I!=E; ++I) {
927d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *subDecl = *I;
9280582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek    if (!subDecl || subDecl->getLexicalDeclContext() != D ||
9290582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek        subDecl->getLocStart().isInvalid())
930d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
931d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclsInContainer.push_back(subDecl);
932d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
933d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
934d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now sort the Decls so that they appear in lexical order.
935d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  std::sort(DeclsInContainer.begin(), DeclsInContainer.end(),
936d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek            ContainerDeclsSort(SM));
937d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
938d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now visit the decls.
9395f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  for (SmallVectorImpl<Decl*>::iterator I = DeclsInContainer.begin(),
940d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek         E = DeclsInContainer.end(); I != E; ++I) {
941aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    CXCursor Cursor = MakeCXCursor(*I, TU, RegionOfInterest);
942d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
943d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
944d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
945d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
946d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
947d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Visit(Cursor, true))
948d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return true;
949d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
950d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return false;
951a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor}
952a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor
953b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
954b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
955b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                   TU)))
956b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
957f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
95878db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
95978db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
96078db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = ND->protocol_end(); I != E; ++I, ++PL)
961b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
962b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
963f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
964a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(ND);
965dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
966dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
9671ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
968bd9482d859a74bf2c45ef8b8aedec61c0e1c8374Douglas Gregor  if (!PID->isThisDeclarationADefinition())
969bd9482d859a74bf2c45ef8b8aedec61c0e1c8374Douglas Gregor    return Visit(MakeCursorObjCProtocolRef(PID, PID->getLocation(), TU));
970bd9482d859a74bf2c45ef8b8aedec61c0e1c8374Douglas Gregor
9711ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
9721ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
9731ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       E = PID->protocol_end(); I != E; ++I, ++PL)
9741ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
9751ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
976f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9771ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(PID);
9781ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
9791ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
98023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenekbool CursorVisitor::VisitObjCPropertyDecl(ObjCPropertyDecl *PD) {
98183cb94269015bf2770ade71e616c5322ea7e76e1Douglas Gregor  if (PD->getTypeSourceInfo() && Visit(PD->getTypeSourceInfo()->getTypeLoc()))
982fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall    return true;
983fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall
98423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // FIXME: This implements a workaround with @property declarations also being
98523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // installed in the DeclContext for the @interface.  Eventually this code
98623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // should be removed.
98723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCCategoryDecl *CDecl = dyn_cast<ObjCCategoryDecl>(PD->getDeclContext());
98823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!CDecl || !CDecl->IsClassExtension())
98923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
99023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
99123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCInterfaceDecl *ID = CDecl->getClassInterface();
99223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!ID)
99323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
99423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
99523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  IdentifierInfo *PropertyId = PD->getIdentifier();
99623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCPropertyDecl *prevDecl =
99723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(ID), PropertyId);
99823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
99923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!prevDecl)
100023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
100123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
100223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // Visit synthesized methods since they will be skipped when visiting
100323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // the @interface.
100423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getGetterMethodDecl())
1005a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
1006aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
100723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
100823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
100923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getSetterMethodDecl())
1010a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
1011aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
101223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
101323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
101423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  return false;
101523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek}
101623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
1017b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
1018375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  if (!D->isThisDeclarationADefinition()) {
1019375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    // Forward declaration is treated like a reference.
1020375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    return Visit(MakeCursorObjCClassRef(D, D->getLocation(), TU));
1021375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  }
1022375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor
1023dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek  // Issue callbacks for super class.
1024b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (D->getSuperClass() &&
1025b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1026f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
1027b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                        TU)))
1028b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
1029f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
103078db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
103178db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
103278db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = D->protocol_end(); I != E; ++I, ++PL)
1033b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1034b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1035f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1036a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(D);
1037dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1038dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10391ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
10401ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(D);
10411ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10421ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10431ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
1044ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  // 'ID' could be null when dealing with invalid code.
1045ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  if (ObjCInterfaceDecl *ID = D->getClassInterface())
1046ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek    if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
1047ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek      return true;
1048f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10491ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
10501ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10511ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10521ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
10531ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#if 0
10541ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // Issue callbacks for super class.
10551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // FIXME: No source location information!
10561ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (D->getSuperClass() &&
10571ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1058f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
10591ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                        TU)))
1060a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return true;
10611ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#endif
1062f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10631ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
1064dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1065dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
1066a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregorbool CursorVisitor::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD) {
1067a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  if (ObjCIvarDecl *Ivar = PD->getPropertyIvarDecl())
1068a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor    return Visit(MakeCursorMemberRef(Ivar, PD->getPropertyIvarDeclLoc(), TU));
1069a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
1070a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  return false;
1071a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor}
1072a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
10738f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenekbool CursorVisitor::VisitNamespaceDecl(NamespaceDecl *D) {
10748f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  return VisitDeclContext(D);
10758f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek}
10768f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek
10776931900f43cea558c6974075256c07728dbfecc6Douglas Gregorbool CursorVisitor::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
1078c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
10790cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
10800cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1081c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
10826931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
10836931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  return Visit(MakeCursorNamespaceRef(D->getAliasedNamespace(),
10846931900f43cea558c6974075256c07728dbfecc6Douglas Gregor                                      D->getTargetNameLoc(), TU));
10856931900f43cea558c6974075256c07728dbfecc6Douglas Gregor}
10866931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
10877e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUsingDecl(UsingDecl *D) {
1088c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1089dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1090dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1091c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1092dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
10937e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
10941f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (Visit(MakeCursorOverloadedDeclRef(D, D->getLocation(), TU)))
10951f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return true;
10961f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
10977e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
10987e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
10997e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11000a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregorbool CursorVisitor::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
1101c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1102db9924191092b4d426cc066637d81698211846aaDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1103db9924191092b4d426cc066637d81698211846aaDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1104c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
11050a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11060a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  return Visit(MakeCursorNamespaceRef(D->getNominatedNamespaceAsWritten(),
11070a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor                                      D->getIdentLocation(), TU));
11080a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor}
11090a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11107e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1111c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1112dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1113dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1114c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1115dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1116c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11177e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11187e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11197e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11207e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypenameDecl(
11217e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor                                               UnresolvedUsingTypenameDecl *D) {
1122c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1123dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1124dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1125c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1126c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11277e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return false;
11287e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11297e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
113001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitDeclarationNameInfo(DeclarationNameInfo Name) {
113101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  switch (Name.getName().getNameKind()) {
113201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::Identifier:
113301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXLiteralOperatorName:
113401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXOperatorName:
113501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXUsingDirective:
113601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
113701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
113801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConstructorName:
113901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXDestructorName:
114001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConversionFunctionName:
114101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (TypeSourceInfo *TSInfo = Name.getNamedTypeInfo())
114201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return Visit(TSInfo->getTypeLoc());
114301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
114401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
114501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCZeroArgSelector:
114601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCOneArgSelector:
114701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCMultiArgSelector:
114801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Per-identifier location info?
114901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
115001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
11517530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie
11527530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid DeclarationName::Kind!");
115301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
115401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1155c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregorbool CursorVisitor::VisitNestedNameSpecifier(NestedNameSpecifier *NNS,
1156c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                             SourceRange Range) {
1157c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // FIXME: This whole routine is a hack to work around the lack of proper
1158c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // source information in nested-name-specifiers (PR5791). Since we do have
1159c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // a beginning source location, we can visit the first component of the
1160c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // nested-name-specifier, if it's a single-token component.
1161c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  if (!NNS)
1162c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return false;
1163c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1164c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Get the first component in the nested-name-specifier.
1165c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  while (NestedNameSpecifier *Prefix = NNS->getPrefix())
1166c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    NNS = Prefix;
1167c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1168c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  switch (NNS->getKind()) {
1169c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Namespace:
1170c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(), Range.getBegin(),
1171c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                        TU));
1172c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
117314aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor  case NestedNameSpecifier::NamespaceAlias:
117414aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
117514aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor                                        Range.getBegin(), TU));
117614aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor
1177c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpec: {
1178c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // If the type has a form where we know that the beginning of the source
1179c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // range matches up with a reference cursor. Visit the appropriate reference
1180c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // cursor.
1181f4c7371fb1d3cebcfb40abad4537bb82515704eaJohn McCall    const Type *T = NNS->getAsType();
1182c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TypedefType *Typedef = dyn_cast<TypedefType>(T))
1183c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Typedef->getDecl(), Range.getBegin(), TU));
1184c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TagType *Tag = dyn_cast<TagType>(T))
1185c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Tag->getDecl(), Range.getBegin(), TU));
1186c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TemplateSpecializationType *TST
1187c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                      = dyn_cast<TemplateSpecializationType>(T))
1188c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return VisitTemplateName(TST->getTemplateName(), Range.getBegin());
1189c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1190c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1191c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1192c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpecWithTemplate:
1193c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Global:
1194c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Identifier:
1195c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1196c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1197c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1198c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  return false;
1199c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor}
1200c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1201dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregorbool
1202dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas GregorCursorVisitor::VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
12035f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<NestedNameSpecifierLoc, 4> Qualifiers;
1204dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  for (; Qualifier; Qualifier = Qualifier.getPrefix())
1205dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    Qualifiers.push_back(Qualifier);
1206dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1207dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  while (!Qualifiers.empty()) {
1208dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifierLoc Q = Qualifiers.pop_back_val();
1209dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifier *NNS = Q.getNestedNameSpecifier();
1210dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    switch (NNS->getKind()) {
1211dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Namespace:
1212dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(),
1213c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1214dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1215dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1216dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1217dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1218dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1219dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::NamespaceAlias:
1220dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1221c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1222dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1223dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1224dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1225dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1226dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1227dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpec:
1228dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpecWithTemplate:
1229dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(Q.getTypeLoc()))
1230dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1231dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1232dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1233dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1234dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Global:
1235dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Identifier:
1236dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1237dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    }
1238dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1239dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1240dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  return false;
1241dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor}
1242dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1243fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateParameters(
1244fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          const TemplateParameterList *Params) {
1245fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (!Params)
1246fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1247fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1248fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (TemplateParameterList::const_iterator P = Params->begin(),
1249fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          PEnd = Params->end();
1250fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor       P != PEnd; ++P) {
1251aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
1252fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1253fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1254fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1255fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1256fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1257fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
12580b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregorbool CursorVisitor::VisitTemplateName(TemplateName Name, SourceLocation Loc) {
12590b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  switch (Name.getKind()) {
12600b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::Template:
12610b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(Name.getAsTemplateDecl(), Loc, TU));
12620b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12630b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::OverloadedTemplate:
12641f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // Visit the overloaded template set.
12651f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (Visit(MakeCursorOverloadedDeclRef(Name, Loc, TU)))
12661f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return true;
12671f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
12680b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
12690b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12700b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::DependentTemplate:
12710b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
12720b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
12730b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12740b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::QualifiedTemplate:
12750b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
12760b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(
12770b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                  Name.getAsQualifiedTemplateName()->getDecl(),
12780b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                       Loc, TU));
1279146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall
1280146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall  case TemplateName::SubstTemplateTemplateParm:
1281146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall    return Visit(MakeCursorTemplateRef(
1282146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall                         Name.getAsSubstTemplateTemplateParm()->getParameter(),
1283146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall                                       Loc, TU));
12841aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor
12851aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor  case TemplateName::SubstTemplateTemplateParmPack:
12861aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor    return Visit(MakeCursorTemplateRef(
12871aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                  Name.getAsSubstTemplateTemplateParmPack()->getParameterPack(),
12881aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                                       Loc, TU));
12890b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  }
12907530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie
12917530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid TemplateName::Kind!");
12920b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor}
12930b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
1294fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL) {
1295fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  switch (TAL.getArgument().getKind()) {
1296fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Null:
1297fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Integral:
1298fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Pack:
1299fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
130087dd697dcc8ecb64df73ae64d61b8c80ff0c157cDouglas Gregor
1301fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Type:
1302fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (TypeSourceInfo *TSInfo = TAL.getTypeSourceInfo())
1303fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(TSInfo->getTypeLoc());
1304fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1305fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1306fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Declaration:
1307fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceDeclExpression())
1308aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1309fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1310fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1311fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Expression:
1312fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceExpression())
1313aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1314fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1315fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1316fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Template:
1317a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor  case TemplateArgument::TemplateExpansion:
1318b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor    if (VisitNestedNameSpecifierLoc(TAL.getTemplateQualifierLoc()))
1319b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor      return true;
1320b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor
1321a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor    return VisitTemplateName(TAL.getArgument().getAsTemplateOrTemplatePattern(),
13220b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                             TAL.getTemplateNameLoc());
1323fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
13247530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie
13257530c034c0c71a64c5a9173206d9742ae847af8bDavid Blaikie  llvm_unreachable("Invalid TemplateArgument::Kind!");
1326fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1327fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1328a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenekbool CursorVisitor::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
1329a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  return VisitDeclContext(D);
1330a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek}
1331a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek
133201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
133301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return Visit(TL.getUnqualifiedLoc());
133401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
133501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1336f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
1337a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTContext &Context = AU->getASTContext();
1338f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1339f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // Some builtin types (such as Objective-C's "id", "sel", and
1340f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // "Class") have associated declarations. Create cursors for those.
1341f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  QualType VisitType;
1342e0a22d06888c13989b3f72db319f1d498bf69153John McCall  switch (TL.getTypePtr()->getKind()) {
13432dde35bc626153492f5f58202506c88a27fbff5bJohn McCall
13446b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Void:
1345f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::NullPtr:
13466b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Dependent:
13472dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define BUILTIN_TYPE(Id, SingletonId)
13482dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define SIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
13492dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define UNSIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
13502dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define FLOATING_TYPE(Id, SingletonId) case BuiltinType::Id:
13512dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define PLACEHOLDER_TYPE(Id, SingletonId) case BuiltinType::Id:
13522dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#include "clang/AST/BuiltinTypes.def"
1353f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
13546b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1355f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCId:
1356f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCIdType();
1357f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
13586b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
13596b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ObjCClass:
13606b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    VisitType = Context.getObjCClassType();
13616b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    break;
13626b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1363f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCSel:
1364f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCSelType();
1365f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
1366f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1367f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1368f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (!VisitType.isNull()) {
1369f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
1370f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
1371f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                     TU));
1372f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1373f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1374f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1375f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1376f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
13777d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
1378162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return Visit(MakeCursorTypeRef(TL.getTypedefNameDecl(), TL.getNameLoc(), TU));
13797d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
13807d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1381f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
1382f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1383f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1384f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1385f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
13866f155de99c59af890817146ec8526bafb6560f1fArgyrios Kyrtzidis  if (TL.isDefinition())
1387aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(TL.getDecl(), TU, RegionOfInterest));
13886f155de99c59af890817146ec8526bafb6560f1fArgyrios Kyrtzidis
1389f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1390f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1391f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1392fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
1393960d13dde337a59dacc9dc3936c26d4aa8478986Chandler Carruth  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1394fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1395fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1396f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
1397f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU)))
1398f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1399f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1400c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return false;
1401c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall}
1402c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1403c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCallbool CursorVisitor::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
1404c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseLoc()))
1405c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall    return true;
1406c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1407f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1408f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1409f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                        TU)))
1410f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor      return true;
1411f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1412f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1413f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1414f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1415f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1416f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
1417c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return Visit(TL.getPointeeLoc());
1418f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1419f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1420075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnarabool CursorVisitor::VisitParenTypeLoc(ParenTypeLoc TL) {
1421075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara  return Visit(TL.getInnerLoc());
1422075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara}
1423075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara
1424f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
1425f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1426f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1427f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1428f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
1429f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1430f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1431f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1432f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
1433f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1434f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1435f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1436f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
1437f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1438f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1439f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1440f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
1441f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1442f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1443f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
14443422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidisbool CursorVisitor::VisitAttributedTypeLoc(AttributedTypeLoc TL) {
14453422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis  return Visit(TL.getModifiedLoc());
14463422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis}
14473422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis
144801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL,
144901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor                                         bool SkipResultType) {
145001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (!SkipResultType && Visit(TL.getResultLoc()))
1451f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1452f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1453f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
14545dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek    if (Decl *D = TL.getArg(I))
1455aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(D, TU, RegionOfInterest)))
14565dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek        return true;
1457f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1458f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1459f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1460f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1461f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
1462f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(TL.getElementLoc()))
1463f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1464f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1465f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Expr *Size = TL.getSizeExpr())
1466aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Size, StmtParent, TU, RegionOfInterest));
1467f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1468f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1469f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1470f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1471fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateSpecializationTypeLoc(
1472fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                             TemplateSpecializationTypeLoc TL) {
14730b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  // Visit the template name.
14740b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
14750b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                        TL.getTemplateNameLoc()))
14760b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return true;
1477fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1478fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Visit the template arguments.
1479fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1480fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1481fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1482fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1483fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1484fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1485fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
14862332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
14872332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
14882332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
14892332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
14902332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
14912332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
1492ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt    return Visit(TSInfo->getTypeLoc());
1493ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt
1494ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt  return false;
1495ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt}
1496ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt
1497ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Huntbool CursorVisitor::VisitUnaryTransformTypeLoc(UnaryTransformTypeLoc TL) {
1498ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
14992332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor    return Visit(TSInfo->getTypeLoc());
15002332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15012332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return false;
15022332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15032332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15042494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregorbool CursorVisitor::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
15052494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15062494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    return true;
15072494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
15082494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  return false;
15092494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor}
15102494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
151194fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregorbool CursorVisitor::VisitDependentTemplateSpecializationTypeLoc(
151294fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor                                    DependentTemplateSpecializationTypeLoc TL) {
151394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the nested-name-specifier, if there is one.
151494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  if (TL.getQualifierLoc() &&
151594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
151694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    return true;
151794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
151894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the template arguments.
151994fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
152094fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
152194fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      return true;
152294fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
152394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  return false;
152494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor}
152594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
15269e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregorbool CursorVisitor::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
15279e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15289e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor    return true;
15299e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15309e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  return Visit(TL.getNamedTypeLoc());
15319e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor}
15329e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15337536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregorbool CursorVisitor::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
15347536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor  return Visit(TL.getPatternLoc());
15357536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor}
15367536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor
1537427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::VisitDecltypeTypeLoc(DecltypeTypeLoc TL) {
1538427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  if (Expr *E = TL.getUnderlyingExpr())
1539427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis    return Visit(MakeCXCursor(E, StmtParent, TU));
1540427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1541427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return false;
1542427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1543427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1544427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::VisitInjectedClassNameTypeLoc(InjectedClassNameTypeLoc TL) {
1545427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1546427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1547427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1548b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedmanbool CursorVisitor::VisitAtomicTypeLoc(AtomicTypeLoc TL) {
1549b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman  return Visit(TL.getValueLoc());
1550b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman}
1551b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman
1552427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis#define DEFAULT_TYPELOC_IMPL(CLASS, PARENT) \
1553427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::Visit##CLASS##TypeLoc(CLASS##TypeLoc TL) { \
1554427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return Visit##PARENT##Loc(TL); \
1555427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1556427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1557427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Complex, Type)
1558427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(ConstantArray, ArrayType)
1559427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(IncompleteArray, ArrayType)
1560427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(VariableArray, ArrayType)
1561427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(DependentSizedArray, ArrayType)
1562427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(DependentSizedExtVector, Type)
1563427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Vector, Type)
1564427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(ExtVector, VectorType)
1565427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(FunctionProto, FunctionType)
1566427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(FunctionNoProto, FunctionType)
1567427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Record, TagType)
1568427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Enum, TagType)
1569427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(SubstTemplateTypeParm, Type)
1570427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(SubstTemplateTypeParmPack, Type)
1571427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Auto, Type)
1572427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
15733064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenekbool CursorVisitor::VisitCXXRecordDecl(CXXRecordDecl *D) {
1574c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
1575c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1576c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1577c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
1578c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
15795e1cdac63c3d9c9b32fa41fa0b2d242a58a20d49John McCall  if (D->isCompleteDefinition()) {
15803064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    for (CXXRecordDecl::base_class_iterator I = D->bases_begin(),
15813064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek         E = D->bases_end(); I != E; ++I) {
15823064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      if (Visit(cxcursor::MakeCursorCXXBaseSpecifier(I, TU)))
15833064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek        return true;
15843064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
15853064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  }
15863064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
15873064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  return VisitTagDecl(D);
15883064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek}
15893064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
159009dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenekbool CursorVisitor::VisitAttributes(Decl *D) {
1591cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt  for (AttrVec::const_iterator i = D->attr_begin(), e = D->attr_end();
1592cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt       i != e; ++i)
1593cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt    if (Visit(MakeCXCursor(*i, D, TU)))
159409dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek        return true;
159509dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
159609dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  return false;
159709dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek}
159809dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
1599c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1600c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Data-recursive visitor methods.
1601c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1602c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
160328a719433411ef782b582946823bc648ddcc4533Ted Kremeneknamespace {
1604035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#define DEF_JOB(NAME, DATA, KIND)\
1605035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass NAME : public VisitorJob {\
1606035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:\
1607035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  NAME(DATA *d, CXCursor parent) : VisitorJob(parent, VisitorJob::KIND, d) {} \
1608035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) { return VJ->getKind() == KIND; }\
1609f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DATA *get() const { return static_cast<DATA*>(data[0]); }\
1610035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1611035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1612035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(StmtVisit, Stmt, StmtVisitKind)
1613035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(MemberExprParts, MemberExpr, MemberExprPartsKind)
1614e4979ccb5960608edce73f3b274eb7c2de15dac5Ted KremenekDEF_JOB(DeclRefExprParts, DeclRefExpr, DeclRefExprPartsKind)
1615035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(OverloadExprParts, OverloadExpr, OverloadExprPartsKind)
1616b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios KyrtzidisDEF_JOB(ExplicitTemplateArgsVisit, ASTTemplateArgumentListInfo,
161760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        ExplicitTemplateArgsVisitKind)
161894d96291cd041adc5731a2294828a9c20e450b74Douglas GregorDEF_JOB(SizeOfPackExprParts, SizeOfPackExpr, SizeOfPackExprPartsKind)
1619011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas GregorDEF_JOB(LambdaExprParts, LambdaExpr, LambdaExprPartsKind)
1620035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#undef DEF_JOB
1621035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1622035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass DeclVisit : public VisitorJob {
1623035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1624035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  DeclVisit(Decl *d, CXCursor parent, bool isFirst) :
1625035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::DeclVisitKind,
1626035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               d, isFirst ? (void*) 1 : (void*) 0) {}
1627035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
162882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    return VJ->getKind() == DeclVisitKind;
1629035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1630f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  Decl *get() const { return static_cast<Decl*>(data[0]); }
1631f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  bool isFirst() const { return data[1] ? true : false; }
1632035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1633035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass TypeLocVisit : public VisitorJob {
1634035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1635035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  TypeLocVisit(TypeLoc tl, CXCursor parent) :
1636035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::TypeLocVisitKind,
1637035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               tl.getType().getAsOpaquePtr(), tl.getOpaqueData()) {}
1638035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1639035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1640035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return VJ->getKind() == TypeLocVisitKind;
1641035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1642035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
164382f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek  TypeLoc get() const {
1644f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    QualType T = QualType::getFromOpaquePtr(data[0]);
1645f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return TypeLoc(T, data[1]);
1646035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1647035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1648035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1649ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekclass LabelRefVisit : public VisitorJob {
1650ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekpublic:
1651ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelRefVisit(LabelDecl *LD, SourceLocation labelLoc, CXCursor parent)
1652ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner    : VisitorJob(parent, VisitorJob::LabelRefVisitKind, LD,
1653dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 labelLoc.getPtrEncoding()) {}
1654ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek
1655ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1656ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek    return VJ->getKind() == VisitorJob::LabelRefVisitKind;
1657ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  }
1658ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelDecl *get() const { return static_cast<LabelDecl*>(data[0]); }
1659ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  SourceLocation getLoc() const {
1660dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin    return SourceLocation::getFromPtrEncoding(data[1]); }
1661f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek};
1662f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1663f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorclass NestedNameSpecifierLocVisit : public VisitorJob {
1664f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorpublic:
1665f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLocVisit(NestedNameSpecifierLoc Qualifier, CXCursor parent)
1666f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    : VisitorJob(parent, VisitorJob::NestedNameSpecifierLocVisitKind,
1667f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getNestedNameSpecifier(),
1668f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getOpaqueData()) { }
1669f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1670f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  static bool classof(const VisitorJob *VJ) {
1671f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return VJ->getKind() == VisitorJob::NestedNameSpecifierLocVisitKind;
1672f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1673f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1674f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLoc get() const {
1675f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return NestedNameSpecifierLoc(static_cast<NestedNameSpecifier*>(data[0]),
1676f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                                  data[1]);
1677f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1678f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor};
1679f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1680f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekclass DeclarationNameInfoVisit : public VisitorJob {
1681f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekpublic:
1682f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfoVisit(Stmt *S, CXCursor parent)
1683f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : VisitorJob(parent, VisitorJob::DeclarationNameInfoVisitKind, S) {}
1684f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1685f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return VJ->getKind() == VisitorJob::DeclarationNameInfoVisitKind;
1686f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1687f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfo get() const {
1688f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    Stmt *S = static_cast<Stmt*>(data[0]);
1689f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    switch (S->getStmtClass()) {
1690f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    default:
1691f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      llvm_unreachable("Unhandled Stmt");
1692ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor    case clang::Stmt::MSDependentExistsStmtClass:
1693ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor      return cast<MSDependentExistsStmt>(S)->getNameInfo();
1694f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::CXXDependentScopeMemberExprClass:
1695f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<CXXDependentScopeMemberExpr>(S)->getMemberNameInfo();
1696f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::DependentScopeDeclRefExprClass:
1697f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<DependentScopeDeclRefExpr>(S)->getNameInfo();
1698f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    }
1699f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1700ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek};
1701cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekclass MemberRefVisit : public VisitorJob {
1702cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekpublic:
1703cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  MemberRefVisit(FieldDecl *D, SourceLocation L, CXCursor parent)
1704cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    : VisitorJob(parent, VisitorJob::MemberRefVisitKind, D,
1705dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 L.getPtrEncoding()) {}
1706cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  static bool classof(const VisitorJob *VJ) {
1707cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return VJ->getKind() == VisitorJob::MemberRefVisitKind;
1708cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1709cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  FieldDecl *get() const {
1710cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return static_cast<FieldDecl*>(data[0]);
1711cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1712cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  SourceLocation getLoc() const {
1713cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1714cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1715cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek};
171628a719433411ef782b582946823bc648ddcc4533Ted Kremenekclass EnqueueVisitor : public StmtVisitor<EnqueueVisitor, void> {
171728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitorWorkList &WL;
171828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  CXCursor Parent;
171928a719433411ef782b582946823bc648ddcc4533Ted Kremenekpublic:
172028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueVisitor(VisitorWorkList &wl, CXCursor parent)
172128a719433411ef782b582946823bc648ddcc4533Ted Kremenek    : WL(wl), Parent(parent) {}
172228a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1723ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitAddrLabelExpr(AddrLabelExpr *E);
172473d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitBlockExpr(BlockExpr *B);
172528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1726083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  void VisitCompoundStmt(CompoundStmt *S);
172711b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) { /* Do nothing. */ }
1728ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  void VisitMSDependentExistsStmt(MSDependentExistsStmt *S);
1729f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
173011b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXNewExpr(CXXNewExpr *E);
17316d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
173228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
1733cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E);
173473d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E);
1735b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  void VisitCXXTypeidExpr(CXXTypeidExpr *E);
173655b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
17371e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  void VisitCXXUuidofExpr(CXXUuidofExpr *E);
1738dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis  void VisitCXXCatchStmt(CXXCatchStmt *S);
1739e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  void VisitDeclRefExpr(DeclRefExpr *D);
1740035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void VisitDeclStmt(DeclStmt *S);
1741f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E);
1742cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitDesignatedInitExpr(DesignatedInitExpr *E);
174328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitExplicitCastExpr(ExplicitCastExpr *E);
174428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitForStmt(ForStmt *FS);
1745ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitGotoStmt(GotoStmt *GS);
174628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitIfStmt(IfStmt *If);
174728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitInitListExpr(InitListExpr *IE);
174828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitMemberExpr(MemberExpr *M);
1749cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitOffsetOfExpr(OffsetOfExpr *E);
175073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
175128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitObjCMessageExpr(ObjCMessageExpr *M);
175228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitOverloadExpr(OverloadExpr *E);
1753f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne  void VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E);
175428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitStmt(Stmt *S);
175528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitSwitchStmt(SwitchStmt *S);
175628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitWhileStmt(WhileStmt *W);
17572939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  void VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E);
17586ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  void VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E);
17594ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor  void VisitTypeTraitExpr(TypeTraitExpr *E);
176021ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley  void VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E);
1761552622067dc45013d240f73952fece703f5e63bdJohn Wiegley  void VisitExpressionTraitExpr(ExpressionTraitExpr *E);
176228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U);
17639d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  void VisitVAArgExpr(VAArgExpr *E);
176494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  void VisitSizeOfPackExpr(SizeOfPackExpr *E);
17654b9c2d235fb9449e249d74f48ecfec601650de93John McCall  void VisitPseudoObjectExpr(PseudoObjectExpr *E);
17664b9c2d235fb9449e249d74f48ecfec601650de93John McCall  void VisitOpaqueValueExpr(OpaqueValueExpr *E);
1767011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  void VisitLambdaExpr(LambdaExpr *E);
1768ee8aff06f6a96214731de17b2cb6df407c6c1820Douglas Gregor
176928a719433411ef782b582946823bc648ddcc4533Ted Kremenekprivate:
1770f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void AddDeclarationNameInfo(Stmt *S);
1771f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  void AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier);
1772b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis  void AddExplicitTemplateArgs(const ASTTemplateArgumentListInfo *A);
1773cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void AddMemberRef(FieldDecl *D, SourceLocation L);
177428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddStmt(Stmt *S);
1775035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void AddDecl(Decl *D, bool isFirst = true);
177628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddTypeLoc(TypeSourceInfo *TI);
177728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void EnqueueChildren(Stmt *S);
177828a719433411ef782b582946823bc648ddcc4533Ted Kremenek};
177928a719433411ef782b582946823bc648ddcc4533Ted Kremenek} // end anonyous namespace
178028a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1781f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::AddDeclarationNameInfo(Stmt *S) {
1782f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // 'S' should always be non-null, since it comes from the
1783f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // statement we are visiting.
1784f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  WL.push_back(DeclarationNameInfoVisit(S, Parent));
1785f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1786f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1787f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorvoid
1788f3db29fff6a583ecda823cf909ab7737d8d30129Douglas GregorEnqueueVisitor::AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1789f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (Qualifier)
1790f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    WL.push_back(NestedNameSpecifierLocVisit(Qualifier, Parent));
1791f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor}
1792f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
179328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddStmt(Stmt *S) {
179428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (S)
179528a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(StmtVisit(S, Parent));
179628a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1797035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::AddDecl(Decl *D, bool isFirst) {
179828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (D)
1799035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    WL.push_back(DeclVisit(D, Parent, isFirst));
180028a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
180160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenekvoid EnqueueVisitor::
1802b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis  AddExplicitTemplateArgs(const ASTTemplateArgumentListInfo *A) {
180360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (A)
180460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    WL.push_back(ExplicitTemplateArgsVisit(
1805b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis                        const_cast<ASTTemplateArgumentListInfo*>(A), Parent));
180660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek}
1807cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::AddMemberRef(FieldDecl *D, SourceLocation L) {
1808cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  if (D)
1809cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    WL.push_back(MemberRefVisit(D, L, Parent));
1810cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
181128a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddTypeLoc(TypeSourceInfo *TI) {
181228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (TI)
181328a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(TypeLocVisit(TI->getTypeLoc(), Parent));
181428a719433411ef782b582946823bc648ddcc4533Ted Kremenek }
181528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::EnqueueChildren(Stmt *S) {
1816a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  unsigned size = WL.size();
18177502c1d3ce8bb97bcc4f7bebef507040bd93b26fJohn McCall  for (Stmt::child_range Child = S->children(); Child; ++Child) {
181828a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(*Child);
1819a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  }
1820a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  if (size == WL.size())
1821a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek    return;
1822a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // Now reverse the entries we just added.  This will match the DFS
1823a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // ordering performed by the worklist.
1824a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1825a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  std::reverse(I, E);
1826a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek}
1827ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitAddrLabelExpr(AddrLabelExpr *E) {
1828ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(E->getLabel(), E->getLabelLoc(), Parent));
1829ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
183073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitBlockExpr(BlockExpr *B) {
183173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddDecl(B->getBlockDecl());
183273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
183328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
183428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
183528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeSourceInfo());
183628a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1837083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenekvoid EnqueueVisitor::VisitCompoundStmt(CompoundStmt *S) {
1838083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  for (CompoundStmt::reverse_body_iterator I = S->body_rbegin(),
1839083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek        E = S->body_rend(); I != E; ++I) {
1840083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek    AddStmt(*I);
1841083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  }
184211b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
1843f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::
1844ba0513de93d2fab6db5ab30b6927209fcc883078Douglas GregorVisitMSDependentExistsStmt(MSDependentExistsStmt *S) {
1845ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  AddStmt(S->getSubStmt());
1846ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  AddDeclarationNameInfo(S);
1847ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = S->getQualifierLoc())
1848ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1849ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor}
1850ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor
1851ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregorvoid EnqueueVisitor::
1852f64d80306144f978148ba92f36f7cea7b671dd34Ted KremenekVisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E) {
1853f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1854f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
18557c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
18567c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1857f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  if (!E->isImplicitAccess())
1858f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    AddStmt(E->getBase());
1859f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
186011b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenekvoid EnqueueVisitor::VisitCXXNewExpr(CXXNewExpr *E) {
18612aed8b88613863f3c439cdfb205bdf8b608fb205Sebastian Redl  // Enqueue the initializer , if any.
18622aed8b88613863f3c439cdfb205bdf8b608fb205Sebastian Redl  AddStmt(E->getInitializer());
186311b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the array size, if any.
186411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddStmt(E->getArraySize());
186511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the allocated type.
186611b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddTypeLoc(E->getAllocatedTypeSourceInfo());
186711b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the placement arguments.
186811b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumPlacementArgs(); I > 0; --I)
186911b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getPlacementArg(I-1));
187011b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
187128a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *CE) {
18728b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek  for (unsigned I = CE->getNumArgs(); I > 1 /* Yes, this is 1 */; --I)
18738b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek    AddStmt(CE->getArg(I-1));
187428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getCallee());
187528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getArg(0));
187628a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1877cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1878cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the name of the type being destroyed.
1879cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getDestroyedTypeInfo());
1880cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the scope type that looks disturbingly like the nested-name-specifier
1881cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // but isn't.
1882cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getScopeTypeInfo());
1883cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the nested-name-specifier.
1884f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
1885f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1886cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit base expression.
1887cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getBase());
1888cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
18896d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenekvoid EnqueueVisitor::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
18906d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
18916d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
189273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
189373d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  EnqueueChildren(E);
189473d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
189573d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
1896b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenekvoid EnqueueVisitor::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1897b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  EnqueueChildren(E);
1898b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  if (E->isTypeOperand())
1899b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
1900b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek}
190155b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek
190255b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenekvoid EnqueueVisitor::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr
190355b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek                                                     *E) {
190455b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  EnqueueChildren(E);
190555b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
190655b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek}
19071e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenekvoid EnqueueVisitor::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
19081e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  EnqueueChildren(E);
19091e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  if (E->isTypeOperand())
19101e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
19111e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek}
1912dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis
1913dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidisvoid EnqueueVisitor::VisitCXXCatchStmt(CXXCatchStmt *S) {
1914dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis  EnqueueChildren(S);
1915dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis  AddDecl(S->getExceptionDecl());
1916dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis}
1917dcbb2fb8710459fdc8073b76a4ef73fbbcbeac9fArgyrios Kyrtzidis
1918e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenekvoid EnqueueVisitor::VisitDeclRefExpr(DeclRefExpr *DR) {
191960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (DR->hasExplicitTemplateArgs()) {
192060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    AddExplicitTemplateArgs(&DR->getExplicitTemplateArgs());
192160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  }
1922e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  WL.push_back(DeclRefExprParts(DR, Parent));
1923e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek}
1924f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1925f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1926f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
192700cf3cc2718671aa48e8da264a523b0058a8591eDouglas Gregor  AddNestedNameSpecifierLoc(E->getQualifierLoc());
1928f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1929035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::VisitDeclStmt(DeclStmt *S) {
1930035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  unsigned size = WL.size();
1931035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  bool isFirst = true;
1932035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  for (DeclStmt::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
1933035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek       D != DEnd; ++D) {
1934035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    AddDecl(*D, isFirst);
1935035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    isFirst = false;
1936035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1937035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  if (size == WL.size())
1938035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return;
1939035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // Now reverse the entries we just added.  This will match the DFS
1940035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // ordering performed by the worklist.
1941035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1942035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  std::reverse(I, E);
1943035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek}
1944cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
1945cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getInit());
1946cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  typedef DesignatedInitExpr::Designator Designator;
1947cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (DesignatedInitExpr::reverse_designators_iterator
1948cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D = E->designators_rbegin(), DEnd = E->designators_rend();
1949cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D != DEnd; ++D) {
1950cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isFieldDesignator()) {
1951cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      if (FieldDecl *Field = D->getField())
1952cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        AddMemberRef(Field, D->getFieldLoc());
1953cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1954cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1955cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isArrayDesignator()) {
1956cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getArrayIndex(*D));
1957cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1958cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1959cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    assert(D->isArrayRangeDesignator() && "Unknown designator kind");
1960cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeEnd(*D));
1961cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeStart(*D));
1962cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1963cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
196428a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitExplicitCastExpr(ExplicitCastExpr *E) {
196528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
196628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeInfoAsWritten());
196728a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
196828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitForStmt(ForStmt *FS) {
196928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getBody());
197028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInc());
197128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getCond());
197228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(FS->getConditionVariable());
197328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInit());
197428a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1975ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitGotoStmt(GotoStmt *GS) {
1976ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(GS->getLabel(), GS->getLabelLoc(), Parent));
1977ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
197828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitIfStmt(IfStmt *If) {
197928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getElse());
198028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getThen());
198128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getCond());
198228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(If->getConditionVariable());
198328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
198428a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitInitListExpr(InitListExpr *IE) {
198528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  // We care about the syntactic form of the initializer list, only.
198628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (InitListExpr *Syntactic = IE->getSyntacticForm())
198728a719433411ef782b582946823bc648ddcc4533Ted Kremenek    IE = Syntactic;
198828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(IE);
198928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
199028a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitMemberExpr(MemberExpr *M) {
199189629a746019a42797495b091711a1d68467e88aDouglas Gregor  WL.push_back(MemberExprParts(M, Parent));
199289629a746019a42797495b091711a1d68467e88aDouglas Gregor
199389629a746019a42797495b091711a1d68467e88aDouglas Gregor  // If the base of the member access expression is an implicit 'this', don't
199489629a746019a42797495b091711a1d68467e88aDouglas Gregor  // visit it.
199589629a746019a42797495b091711a1d68467e88aDouglas Gregor  // FIXME: If we ever want to show these implicit accesses, this will be
199689629a746019a42797495b091711a1d68467e88aDouglas Gregor  // unfortunate. However, clang_getCursor() relies on this behavior.
199775e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor  if (!M->isImplicitAccess())
199875e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor    AddStmt(M->getBase());
199928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
200073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
200173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getEncodedTypeSourceInfo());
200273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
200328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitObjCMessageExpr(ObjCMessageExpr *M) {
200428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(M);
200528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(M->getClassReceiverTypeInfo());
200628a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2007cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitOffsetOfExpr(OffsetOfExpr *E) {
2008cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the components of the offsetof expression.
2009cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (unsigned N = E->getNumComponents(), I = N; I > 0; --I) {
2010cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    typedef OffsetOfExpr::OffsetOfNode OffsetOfNode;
2011cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    const OffsetOfNode &Node = E->getComponent(I-1);
2012cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    switch (Node.getKind()) {
2013cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Array:
2014cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getIndexExpr(Node.getArrayExprIndex()));
2015cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2016cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Field:
201706dec892b5300b43263d25c5476b506c9d6cfbadAbramo Bagnara      AddMemberRef(Node.getField(), Node.getSourceRange().getEnd());
2018cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2019cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Identifier:
2020cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Base:
2021cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
2022cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
2023cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
2024cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the type into which we're computing the offset.
2025cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
2026cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
202728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitOverloadExpr(OverloadExpr *E) {
202860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
20296045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek  WL.push_back(OverloadExprParts(E, Parent));
20306045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek}
2031f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbournevoid EnqueueVisitor::VisitUnaryExprOrTypeTraitExpr(
2032f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne                                              UnaryExprOrTypeTraitExpr *E) {
20336d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  EnqueueChildren(E);
20346d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  if (E->isArgumentType())
20356d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek    AddTypeLoc(E->getArgumentTypeInfo());
20366d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
203728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitStmt(Stmt *S) {
203828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(S);
203928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
204028a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitSwitchStmt(SwitchStmt *S) {
204128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getBody());
204228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getCond());
204328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(S->getConditionVariable());
204428a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2045fafa75aebadef8d6b44a920e3f40529f150a5574Ted Kremenek
204628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitWhileStmt(WhileStmt *W) {
204728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getBody());
204828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getCond());
204928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(W->getConditionVariable());
205028a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
205121ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley
20522939b6f356161f572712d4d6310b65f9599e3675Ted Kremenekvoid EnqueueVisitor::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
20532939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  AddTypeLoc(E->getQueriedTypeSourceInfo());
20542939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek}
20556ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
20566ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichetvoid EnqueueVisitor::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
20576ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  AddTypeLoc(E->getRhsTypeSourceInfo());
20580a03a3f98b14006a54bcac9e8908a7c9f50e519fFrancois Pichet  AddTypeLoc(E->getLhsTypeSourceInfo());
20596ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet}
20606ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
20614ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregorvoid EnqueueVisitor::VisitTypeTraitExpr(TypeTraitExpr *E) {
20624ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor  for (unsigned I = E->getNumArgs(); I > 0; --I)
20634ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor    AddTypeLoc(E->getArg(I-1));
20644ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor}
20654ca8ac2e61c37ddadf37024af86f3e1019af8532Douglas Gregor
206621ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegleyvoid EnqueueVisitor::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
206721ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley  AddTypeLoc(E->getQueriedTypeSourceInfo());
206821ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley}
206921ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley
2070552622067dc45013d240f73952fece703f5e63bdJohn Wiegleyvoid EnqueueVisitor::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
2071552622067dc45013d240f73952fece703f5e63bdJohn Wiegley  EnqueueChildren(E);
2072552622067dc45013d240f73952fece703f5e63bdJohn Wiegley}
2073552622067dc45013d240f73952fece703f5e63bdJohn Wiegley
207428a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U) {
207528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitOverloadExpr(U);
207628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (!U->isImplicitAccess())
207728a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(U->getBase());
207828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
20799d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenekvoid EnqueueVisitor::VisitVAArgExpr(VAArgExpr *E) {
20809d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddStmt(E->getSubExpr());
20819d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddTypeLoc(E->getWrittenTypeInfo());
20829d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek}
208394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregorvoid EnqueueVisitor::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
208494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  WL.push_back(SizeOfPackExprParts(E, Parent));
208594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor}
20864b9c2d235fb9449e249d74f48ecfec601650de93John McCallvoid EnqueueVisitor::VisitOpaqueValueExpr(OpaqueValueExpr *E) {
20874b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // If the opaque value has a source expression, just transparently
20884b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // visit that.  This is useful for (e.g.) pseudo-object expressions.
20894b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (Expr *SourceExpr = E->getSourceExpr())
20904b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return Visit(SourceExpr);
20914b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
2092011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregorvoid EnqueueVisitor::VisitLambdaExpr(LambdaExpr *E) {
2093011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  AddStmt(E->getBody());
2094011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  WL.push_back(LambdaExprParts(E, Parent));
2095011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor}
20964b9c2d235fb9449e249d74f48ecfec601650de93John McCallvoid EnqueueVisitor::VisitPseudoObjectExpr(PseudoObjectExpr *E) {
20974b9c2d235fb9449e249d74f48ecfec601650de93John McCall  // Treat the expression like its syntactic form.
20984b9c2d235fb9449e249d74f48ecfec601650de93John McCall  Visit(E->getSyntacticForm());
20994b9c2d235fb9449e249d74f48ecfec601650de93John McCall}
21006045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek
2101c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekvoid CursorVisitor::EnqueueWorkList(VisitorWorkList &WL, Stmt *S) {
2102aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  EnqueueVisitor(WL, MakeCXCursor(S, StmtParent, TU,RegionOfInterest)).Visit(S);
2103c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2104c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2105c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::IsInRegionOfInterest(CXCursor C) {
2106c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  if (RegionOfInterest.isValid()) {
2107c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SourceRange Range = getRawCursorExtent(C);
2108c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    if (Range.isInvalid() || CompareRegionOfInterest(Range))
2109c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      return false;
2110c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2111c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return true;
2112c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2113c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2114c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::RunVisitorWorkList(VisitorWorkList &WL) {
2115c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  while (!WL.empty()) {
2116c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Dequeue the worklist item.
211782f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    VisitorJob LI = WL.back();
211882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    WL.pop_back();
211982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek
2120c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Set the Parent field, then back to its old value once we're done.
2121c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SetParentRAII SetParent(Parent, StmtParent, LI.getParent());
2122c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2123c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    switch (LI.getKind()) {
2124f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      case VisitorJob::DeclVisitKind: {
212582f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Decl *D = cast<DeclVisit>(&LI)->get();
2126f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        if (!D)
2127f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek          continue;
2128f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2129f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // For now, perform default visitation for Decls.
2130aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis        if (Visit(MakeCXCursor(D, TU, RegionOfInterest,
2131aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis                               cast<DeclVisit>(&LI)->isFirst())))
2132f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek            return true;
2133f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2134f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        continue;
2135f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      }
213660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      case VisitorJob::ExplicitTemplateArgsVisitKind: {
2137b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis        const ASTTemplateArgumentListInfo *ArgList =
213860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          cast<ExplicitTemplateArgsVisit>(&LI)->get();
213960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        for (const TemplateArgumentLoc *Arg = ArgList->getTemplateArgs(),
214060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               *ArgEnd = Arg + ArgList->NumTemplateArgs;
214160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               Arg != ArgEnd; ++Arg) {
214260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          if (VisitTemplateArgumentLoc(*Arg))
214360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek            return true;
214460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        }
214560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        continue;
214660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      }
2147cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      case VisitorJob::TypeLocVisitKind: {
2148cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        // Perform default visitation for TypeLocs.
214982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        if (Visit(cast<TypeLocVisit>(&LI)->get()))
2150cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek          return true;
2151cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        continue;
2152cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      }
2153ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      case VisitorJob::LabelRefVisitKind: {
2154ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner        LabelDecl *LS = cast<LabelRefVisit>(&LI)->get();
2155e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        if (LabelStmt *stmt = LS->getStmt()) {
2156e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          if (Visit(MakeCursorLabelRef(stmt, cast<LabelRefVisit>(&LI)->getLoc(),
2157e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek                                       TU))) {
2158e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek            return true;
2159e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          }
2160e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        }
2161ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek        continue;
2162ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      }
216347695c8ad8424851f62e0d4a983b45b15daee1c5Ted Kremenek
2164f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      case VisitorJob::NestedNameSpecifierLocVisitKind: {
2165f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        NestedNameSpecifierLocVisit *V = cast<NestedNameSpecifierLocVisit>(&LI);
2166f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        if (VisitNestedNameSpecifierLoc(V->get()))
2167f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor          return true;
2168f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        continue;
2169f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      }
2170f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2171f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      case VisitorJob::DeclarationNameInfoVisitKind: {
2172f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        if (VisitDeclarationNameInfo(cast<DeclarationNameInfoVisit>(&LI)
2173f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                                     ->get()))
2174f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek          return true;
2175f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        continue;
2176f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      }
2177cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      case VisitorJob::MemberRefVisitKind: {
2178cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        MemberRefVisit *V = cast<MemberRefVisit>(&LI);
2179cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (Visit(MakeCursorMemberRef(V->get(), V->getLoc(), TU)))
2180cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          return true;
2181cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        continue;
2182cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      }
2183c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::StmtVisitKind: {
218482f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Stmt *S = cast<StmtVisit>(&LI)->get();
21858c269ac75569454a049385b1246140db5f2b6faaTed Kremenek        if (!S)
21868c269ac75569454a049385b1246140db5f2b6faaTed Kremenek          continue;
21878c269ac75569454a049385b1246140db5f2b6faaTed Kremenek
2188f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // Update the current cursor.
2189aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis        CXCursor Cursor = MakeCXCursor(S, StmtParent, TU, RegionOfInterest);
2190cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (!IsInRegionOfInterest(Cursor))
2191cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          continue;
2192cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        switch (Visitor(Cursor, Parent, ClientData)) {
2193cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Break: return true;
2194cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Continue: break;
2195cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Recurse:
2196cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek            EnqueueWorkList(WL, S);
219782f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek            break;
2198c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
219982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        continue;
2200c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2201c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::MemberExprPartsKind: {
2202c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Handle the other pieces in the MemberExpr besides the base.
220382f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        MemberExpr *M = cast<MemberExprParts>(&LI)->get();
2204c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2205c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the nested-name-specifier
220640d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = M->getQualifierLoc())
220740d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2208c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            return true;
2209c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2210c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the declaration name.
2211c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (VisitDeclarationNameInfo(M->getMemberNameInfo()))
2212c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          return true;
2213c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2214c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the explicitly-specified template arguments, if any.
2215c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (M->hasExplicitTemplateArgs()) {
2216c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          for (const TemplateArgumentLoc *Arg = M->getTemplateArgs(),
2217c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               *ArgEnd = Arg + M->getNumTemplateArgs();
2218c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               Arg != ArgEnd; ++Arg) {
2219c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            if (VisitTemplateArgumentLoc(*Arg))
2220c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek              return true;
2221c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          }
2222c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
2223c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        continue;
2224c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2225e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      case VisitorJob::DeclRefExprPartsKind: {
222682f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        DeclRefExpr *DR = cast<DeclRefExprParts>(&LI)->get();
2227e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit nested-name-specifier, if present.
222840d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = DR->getQualifierLoc())
222940d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2230e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek            return true;
2231e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit declaration name.
2232e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        if (VisitDeclarationNameInfo(DR->getNameInfo()))
2233e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek          return true;
2234e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        continue;
2235e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      }
22366045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      case VisitorJob::OverloadExprPartsKind: {
223782f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        OverloadExpr *O = cast<OverloadExprParts>(&LI)->get();
22386045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the nested-name-specifier.
22394c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = O->getQualifierLoc())
22404c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
22416045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek            return true;
22426045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the declaration name.
22436045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (VisitDeclarationNameInfo(O->getNameInfo()))
22446045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22456045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the overloaded declaration reference.
22466045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (Visit(MakeCursorOverloadedDeclRef(O, TU)))
22476045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22486045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        continue;
22496045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      }
225094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      case VisitorJob::SizeOfPackExprPartsKind: {
225194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        SizeOfPackExpr *E = cast<SizeOfPackExprParts>(&LI)->get();
225294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        NamedDecl *Pack = E->getPack();
225394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTypeParmDecl>(Pack)) {
225494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTypeRef(cast<TemplateTypeParmDecl>(Pack),
225594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                      E->getPackLoc(), TU)))
225694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
225794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
225894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
225994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
226094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
226194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTemplateParmDecl>(Pack)) {
226294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTemplateRef(cast<TemplateTemplateParmDecl>(Pack),
226394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                          E->getPackLoc(), TU)))
226494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
226594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
226694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
226794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
226894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
226994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // Non-type template parameter packs and function parameter packs are
227094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // treated like DeclRefExpr cursors.
227194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        continue;
227294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      }
2273011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
2274011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      case VisitorJob::LambdaExprPartsKind: {
2275011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        // Visit captures.
2276011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        LambdaExpr *E = cast<LambdaExprParts>(&LI)->get();
2277011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        for (LambdaExpr::capture_iterator C = E->explicit_capture_begin(),
2278011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                                       CEnd = E->explicit_capture_end();
2279011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor             C != CEnd; ++C) {
2280011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          if (C->capturesThis())
2281011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            continue;
2282011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
2283011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          if (Visit(MakeCursorVariableRef(C->getCapturedVar(),
2284011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                                          C->getLocation(),
2285011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                                          TU)))
2286011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            return true;
2287011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        }
2288011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
2289011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        // Visit parameters and return type, if present.
2290011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        if (E->hasExplicitParameters() || E->hasExplicitResultType()) {
2291011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          TypeLoc TL = E->getCallOperator()->getTypeSourceInfo()->getTypeLoc();
2292011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          if (E->hasExplicitParameters() && E->hasExplicitResultType()) {
2293011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            // Visit the whole type.
2294011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            if (Visit(TL))
2295011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              return true;
2296011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          } else if (isa<FunctionProtoTypeLoc>(TL)) {
2297011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            FunctionProtoTypeLoc Proto = cast<FunctionProtoTypeLoc>(TL);
2298011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            if (E->hasExplicitParameters()) {
2299011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              // Visit parameters.
2300011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              for (unsigned I = 0, N = Proto.getNumArgs(); I != N; ++I)
2301011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                if (Visit(MakeCXCursor(Proto.getArg(I), TU)))
2302011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                  return true;
2303011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            } else {
2304011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              // Visit result type.
2305011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor              if (Visit(Proto.getResultLoc()))
2306011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor                return true;
2307011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor            }
2308011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor          }
2309011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        }
2310011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor        break;
2311011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      }
2312c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    }
2313c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2314c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return false;
2315c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2316c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2317cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekbool CursorVisitor::Visit(Stmt *S) {
2318d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  VisitorWorkList *WL = 0;
2319d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  if (!WorkListFreeList.empty()) {
2320d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = WorkListFreeList.back();
2321d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL->clear();
2322d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListFreeList.pop_back();
2323d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2324d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  else {
2325d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = new VisitorWorkList();
2326d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListCache.push_back(WL);
2327d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2328d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  EnqueueWorkList(*WL, S);
2329d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  bool result = RunVisitorWorkList(*WL);
2330d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  WorkListFreeList.push_back(WL);
2331d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  return result;
2332c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2333c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
233448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichetnamespace {
233548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichettypedef llvm::SmallVector<SourceRange, 4> RefNamePieces;
233648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois PichetRefNamePieces buildPieces(unsigned NameFlags, bool IsMemberRefExpr,
233748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                          const DeclarationNameInfo &NI,
233848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                          const SourceRange &QLoc,
2339b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis                          const ASTTemplateArgumentListInfo *TemplateArgs = 0){
234048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantQualifier = NameFlags & CXNameRange_WantQualifier;
234148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantTemplateArgs = NameFlags & CXNameRange_WantTemplateArgs;
234248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantSinglePiece = NameFlags & CXNameRange_WantSinglePiece;
234348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
234448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const DeclarationName::NameKind Kind = NI.getName().getNameKind();
234548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
234648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  RefNamePieces Pieces;
234748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
234848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantQualifier && QLoc.isValid())
234948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(QLoc);
235048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
235148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (Kind != DeclarationName::CXXOperatorName || IsMemberRefExpr)
235248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(NI.getLoc());
235348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
235448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantTemplateArgs && TemplateArgs)
235548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceRange(TemplateArgs->LAngleLoc,
235648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                                 TemplateArgs->RAngleLoc));
235748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
235848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (Kind == DeclarationName::CXXOperatorName) {
235948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceLocation::getFromRawEncoding(
236048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                       NI.getInfo().CXXOperatorName.BeginOpNameLoc));
236148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceLocation::getFromRawEncoding(
236248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                       NI.getInfo().CXXOperatorName.EndOpNameLoc));
236348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  }
236448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
236548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantSinglePiece) {
236648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    SourceRange R(Pieces.front().getBegin(), Pieces.back().getEnd());
236748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.clear();
236848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(R);
236948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  }
237048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
237148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  return Pieces;
237248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet}
237348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet}
237448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
2375c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2376c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Misc. API hooks.
2377c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2378c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
23798c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic llvm::sys::Mutex EnableMultithreadingMutex;
23808c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic bool EnabledMultithreading;
23818c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
2382fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidisstatic void fatal_error_handler(void *user_data, const std::string& reason) {
2383fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis  // Write the result out to stderr avoiding errs() because raw_ostreams can
2384fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis  // call report_fatal_error.
2385db7a800e0b76036d0faa7123f2e05e45ee3294e5Argyrios Kyrtzidis  fprintf(stderr, "LIBCLANG FATAL ERROR: %s\n", reason.c_str());
2386fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis  ::abort();
2387fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis}
2388fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis
23895e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramerextern "C" {
23900a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas GregorCXIndex clang_createIndex(int excludeDeclarationsFromPCH,
23910a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                          int displayDiagnostics) {
239248615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // Disable pretty stack trace functionality, which will otherwise be a very
239348615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // poor citizen of the world and set up all sorts of signal handlers.
239448615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  llvm::DisablePrettyStackTrace = true;
239548615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar
2396c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // We use crash recovery to make some of our APIs more reliable, implicitly
2397c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // enable it.
2398c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  llvm::CrashRecoveryContext::Enable();
2399c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar
24008c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  // Enable support for multithreading in LLVM.
24018c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  {
24028c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    llvm::sys::ScopedLock L(EnableMultithreadingMutex);
24038c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    if (!EnabledMultithreading) {
2404fa39f5b76bafdf536c5e305f821eb1b7f11079bdArgyrios Kyrtzidis      llvm::install_fatal_error_handler(fatal_error_handler, 0);
24058c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      llvm::llvm_start_multithreaded();
24068c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      EnabledMultithreading = true;
24078c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    }
24088c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  }
24098c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
2410a030b7cf5e6aad5889b1b662b6979840bc75f87fDouglas Gregor  CIndexer *CIdxr = new CIndexer();
2411e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  if (excludeDeclarationsFromPCH)
2412e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    CIdxr->setOnlyLocalDecls();
24130a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  if (displayDiagnostics)
24140a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    CIdxr->setDisplayDiagnostics();
2415fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2416fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (getenv("LIBCLANG_BGPRIO_INDEX"))
2417fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
2418fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis                               CXGlobalOpt_ThreadBackgroundPriorityForIndexing);
2419fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (getenv("LIBCLANG_BGPRIO_EDIT"))
2420fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    CIdxr->setCXGlobalOptFlags(CIdxr->getCXGlobalOptFlags() |
2421fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis                               CXGlobalOpt_ThreadBackgroundPriorityForEditing);
2422fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2423e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  return CIdxr;
2424600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2425600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
24269ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeIndex(CXIndex CIdx) {
24272b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (CIdx)
24282b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    delete static_cast<CIndexer *>(CIdx);
24292bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
24302bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff
2431fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidisvoid clang_CXIndex_setGlobalOptions(CXIndex CIdx, unsigned options) {
2432fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CIdx)
2433fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    static_cast<CIndexer *>(CIdx)->setCXGlobalOptFlags(options);
2434fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis}
2435fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2436fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidisunsigned clang_CXIndex_getGlobalOptions(CXIndex CIdx) {
2437fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CIdx)
2438fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis    return static_cast<CIndexer *>(CIdx)->getCXGlobalOptFlags();
2439fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  return 0;
2440fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis}
2441fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2442d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenekvoid clang_toggleCrashRecovery(unsigned isEnabled) {
2443d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  if (isEnabled)
2444d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Enable();
2445d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  else
2446d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Disable();
2447d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek}
2448d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek
24499ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
2450a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                              const char *ast_filename) {
24512b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
24522b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    return 0;
2453f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
24547d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2455389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOptions FileSystemOpts;
2456389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOpts.WorkingDir = CXXIdx->getWorkingDirectory();
24570d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2458c93dc7889644293e318e19d82830ea2acc45b678Dylan Noblesmith  IntrusiveRefCntPtr<DiagnosticsEngine> Diags;
2459a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *TU = ASTUnit::LoadFromASTFile(ast_filename, Diags, FileSystemOpts,
2460a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  CXXIdx->getOnlyLocalDecls(),
2461bef35c91b594f66216f4aab303b71a6c5ab7abcfArgyrios Kyrtzidis                                  0, 0,
2462bef35c91b594f66216f4aab303b71a6c5ab7abcfArgyrios Kyrtzidis                                  /*CaptureDiagnostics=*/true,
2463bef35c91b594f66216f4aab303b71a6c5ab7abcfArgyrios Kyrtzidis                                  /*AllowPCHWithCompilerErrors=*/true);
2464fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  return MakeCXTranslationUnit(CXXIdx, TU);
2465600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2466600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2467b1c031be513705d924038f497279b9b599868ba1Douglas Gregorunsigned clang_defaultEditingTranslationUnitOptions() {
24682a2c50b330e7754499f42173616a36865b5f313bDouglas Gregor  return CXTranslationUnit_PrecompiledPreamble |
2469b5af843a20e237ad1a13ad66a867e200695b8c8eDouglas Gregor         CXTranslationUnit_CacheCompletionResults;
2470b1c031be513705d924038f497279b9b599868ba1Douglas Gregor}
2471b1c031be513705d924038f497279b9b599868ba1Douglas Gregor
24729ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit
24739ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarclang_createTranslationUnitFromSourceFile(CXIndex CIdx,
24749ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          const char *source_filename,
24759ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          int num_command_line_args,
24762ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                          const char * const *command_line_args,
24774db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          unsigned num_unsaved_files,
2478a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                          struct CXUnsavedFile *unsaved_files) {
2479e1d4330adaaa7faf093e725c9c993207eb2d778aArgyrios Kyrtzidis  unsigned Options = CXTranslationUnit_DetailedPreprocessingRecord;
24805a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor  return clang_parseTranslationUnit(CIdx, source_filename,
24815a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    command_line_args, num_command_line_args,
24825a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    unsaved_files, num_unsaved_files,
2483dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor                                    Options);
24845a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor}
248519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
248619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbarstruct ParseTranslationUnitInfo {
248719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx;
248819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename;
24892ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char *const *command_line_args;
249019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args;
249119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
249219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files;
249319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options;
249419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXTranslationUnit result;
249519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar};
2496b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_parseTranslationUnit_Impl(void *UserData) {
249719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo *PTUI =
249819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    static_cast<ParseTranslationUnitInfo*>(UserData);
249919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx = PTUI->CIdx;
250019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename = PTUI->source_filename;
25012ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char * const *command_line_args = PTUI->command_line_args;
250219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args = PTUI->num_command_line_args;
250319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files = PTUI->unsaved_files;
250419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files = PTUI->num_unsaved_files;
250519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options = PTUI->options;
250619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  PTUI->result = 0;
25075a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor
25082b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
250919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return;
2510f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2511e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2512e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff
2513fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
251481b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
2515fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
251644c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  bool PrecompilePreamble = options & CXTranslationUnit_PrecompiledPreamble;
2517467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor  // FIXME: Add a flag for modules.
2518467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor  TranslationUnitKind TUKind
2519467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor    = (options & CXTranslationUnit_Incomplete)? TU_Prefix : TU_Complete;
252087c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  bool CacheCodeCompetionResults
252187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    = options & CXTranslationUnit_CacheCompletionResults;
252287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
25235352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  // Configure the diagnostics.
25245352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  DiagnosticOptions DiagOpts;
2525c93dc7889644293e318e19d82830ea2acc45b678Dylan Noblesmith  IntrusiveRefCntPtr<DiagnosticsEngine>
252625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Diags(CompilerInstance::createDiagnostics(DiagOpts, num_command_line_args,
252725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                                command_line_args));
252825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
252925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
2530d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie  llvm::CrashRecoveryContextCleanupRegistrar<DiagnosticsEngine,
2531d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie    llvm::CrashRecoveryContextReleaseRefCleanup<DiagnosticsEngine> >
253225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    DiagCleanup(Diags.getPtr());
253325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
25341e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<std::vector<ASTUnit::RemappedFile> >
253525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
253625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
253725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
253825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
253925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
2540f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
25414db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
25425f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2543f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    const llvm::MemoryBuffer *Buffer
2544a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
254525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
254625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
25474db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  }
2548f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
25491e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<std::vector<const char *> >
255025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args(new std::vector<const char*>());
255125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
255225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this method.
255325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<std::vector<const char*> >
255425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    ArgsCleanup(Args.get());
255525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
255652ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // Since the Clang C library is primarily used by batch tools dealing with
255752ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // (often very broken) source code, where spell-checking can have a
255852ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // significant negative impact on performance (particularly when
255952ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // precompiled headers are involved), we disable it by default.
2560b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  // Only do this if we haven't found a spell-checking-related argument.
2561b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  bool FoundSpellCheckingArgument = false;
2562b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  for (int I = 0; I != num_command_line_args; ++I) {
2563b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (strcmp(command_line_args[I], "-fno-spell-checking") == 0 ||
2564b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        strcmp(command_line_args[I], "-fspell-checking") == 0) {
2565b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      FoundSpellCheckingArgument = true;
2566b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      break;
2567e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    }
2568b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  }
2569b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  if (!FoundSpellCheckingArgument)
257025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-fno-spell-checking");
2571b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
257225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  Args->insert(Args->end(), command_line_args,
257325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek               command_line_args + num_command_line_args);
2574d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2575c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // The 'source_filename' argument is optional.  If the caller does not
2576c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // specify it then it is assumed that the source file is specified
2577c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // in the actual argument list.
2578c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // Put the source file after command_line_args otherwise if '-x' flag is
2579c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // present it will be unused.
2580c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  if (source_filename)
258125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back(source_filename);
2582c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis
258344c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  // Do we need the detailed preprocessing record?
258444c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  if (options & CXTranslationUnit_DetailedPreprocessingRecord) {
258525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-Xclang");
258625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-detailed-preprocessing-record");
258744c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  }
258844c181aec37789f25f6c15543c164416f72e562aDouglas Gregor
2589026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  unsigned NumErrors = Diags->getClient()->getNumErrors();
25901e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<ASTUnit> Unit(
25914ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek    ASTUnit::LoadFromCommandLine(Args->size() ? &(*Args)[0] : 0
25924ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 /* vector::data() not portable */,
25934ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 Args->size() ? (&(*Args)[0] + Args->size()) :0,
2594b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 Diags,
2595b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getClangResourcesPath(),
2596b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getOnlyLocalDecls(),
2597e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                                 /*CaptureDiagnostics=*/true,
25984ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 RemappedFiles->size() ? &(*RemappedFiles)[0]:0,
259925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                 RemappedFiles->size(),
2600299a4a967b02c9f0d0d94ad8560e3ced893f9116Argyrios Kyrtzidis                                 /*RemappedFilesKeepOriginalName=*/true,
2601b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 PrecompilePreamble,
2602467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor                                 TUKind,
2603bef35c91b594f66216f4aab303b71a6c5ab7abcfArgyrios Kyrtzidis                                 CacheCodeCompetionResults,
2604bef35c91b594f66216f4aab303b71a6c5ab7abcfArgyrios Kyrtzidis                                 /*AllowPCHWithCompilerErrors=*/true));
2605b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
2606026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  if (NumErrors != Diags->getClient()->getNumErrors()) {
2607b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    // Make sure to check that 'Unit' is non-NULL.
2608b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (CXXIdx->getDisplayDiagnostics() && Unit.get()) {
2609b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
2610b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                      DEnd = Unit->stored_diag_end();
2611b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor           D != DEnd; ++D) {
26124e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie        CXStoredDiagnostic Diag(*D, Unit->getASTContext().getLangOpts());
2613b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        CXString Msg = clang_formatDiagnostic(&Diag,
2614b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                    clang_defaultDiagnosticDisplayOptions());
2615b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        fprintf(stderr, "%s\n", clang_getCString(Msg));
2616b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        clang_disposeString(Msg);
2617b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      }
2618274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor#ifdef LLVM_ON_WIN32
2619b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // On Windows, force a flush, since there may be multiple copies of
2620b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // stderr and stdout in the file system, all with different buffers
2621b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // but writing to the same device.
2622b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      fflush(stderr);
2623b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor#endif
2624b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    }
2625a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  }
2626d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2627fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  PTUI->result = MakeCXTranslationUnit(CXXIdx, Unit.take());
262819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar}
262919ffd492a31a25fb691098bf79f317e5f3edf177Daniel DunbarCXTranslationUnit clang_parseTranslationUnit(CXIndex CIdx,
263019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             const char *source_filename,
26312ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                         const char * const *command_line_args,
263219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             int num_command_line_args,
26339e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                            struct CXUnsavedFile *unsaved_files,
263419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned num_unsaved_files,
263519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned options) {
263619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo PTUI = { CIdx, source_filename, command_line_args,
26379e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_command_line_args, unsaved_files,
26389e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_unsaved_files, options, 0 };
263919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  llvm::CrashRecoveryContext CRC;
264019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
2641bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_parseTranslationUnit_Impl, &PTUI)) {
264260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "libclang: crash detected during parsing: {\n");
264360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'source_filename' : '%s'\n", source_filename);
264460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'command_line_args' : [");
264560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (int i = 0; i != num_command_line_args; ++i) {
264660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
264760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
264860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "'%s'", command_line_args[i]);
264960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
265060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
265160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'unsaved_files' : [");
265260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (unsigned i = 0; i != num_unsaved_files; ++i) {
265360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
265460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
265560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "('%s', '...', %ld)", unsaved_files[i].Filename,
265660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar              unsaved_files[i].Length);
265760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
265860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
265960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'options' : %d,\n", options);
266060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "}\n");
266160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar
266219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return 0;
26636df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
26646df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(PTUI.result);
266519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  }
26666df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
266719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  return PTUI.result;
26685b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff}
26695b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff
26701999844e7a18786e61e619e1dc6c789827541863Douglas Gregorunsigned clang_defaultSaveOptions(CXTranslationUnit TU) {
26711999844e7a18786e61e619e1dc6c789827541863Douglas Gregor  return CXSaveTranslationUnit_None;
26721999844e7a18786e61e619e1dc6c789827541863Douglas Gregor}
2673142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2674142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidisnamespace {
2675142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2676142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidisstruct SaveTranslationUnitInfo {
2677142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  CXTranslationUnit TU;
2678142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  const char *FileName;
2679142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  unsigned options;
2680142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  CXSaveError result;
2681142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis};
2682142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2683142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis}
2684142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2685142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidisstatic void clang_saveTranslationUnit_Impl(void *UserData) {
2686142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  SaveTranslationUnitInfo *STUI =
2687142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    static_cast<SaveTranslationUnitInfo*>(UserData);
2688142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2689fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  CIndexer *CXXIdx = (CIndexer*)STUI->TU->CIdx;
2690fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForIndexing))
269181b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
2692fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2693142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  STUI->result = static_cast<ASTUnit *>(STUI->TU->TUData)->Save(STUI->FileName);
2694142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis}
2695142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
26961999844e7a18786e61e619e1dc6c789827541863Douglas Gregorint clang_saveTranslationUnit(CXTranslationUnit TU, const char *FileName,
26971999844e7a18786e61e619e1dc6c789827541863Douglas Gregor                              unsigned options) {
26987ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  if (!TU)
269939c411fa229b2a6747b92f945d1702ee674d3470Douglas Gregor    return CXSaveError_InvalidTU;
2700142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2701142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
2702142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2703142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2704142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  SaveTranslationUnitInfo STUI = { TU, FileName, options, CXSaveError_None };
2705142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2706142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  if (!CXXUnit->getDiagnostics().hasUnrecoverableErrorOccurred() ||
2707142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis      getenv("LIBCLANG_NOTHREADS")) {
2708142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    clang_saveTranslationUnit_Impl(&STUI);
2709142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2710142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    if (getenv("LIBCLANG_RESOURCE_USAGE"))
2711142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis      PrintLibclangResourceUsage(TU);
2712142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2713142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    return STUI.result;
2714142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  }
2715142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2716142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  // We have an AST that has invalid nodes due to compiler errors.
2717142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  // Use a crash recovery thread for protection.
2718142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2719142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  llvm::CrashRecoveryContext CRC;
2720142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2721142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  if (!RunSafely(CRC, clang_saveTranslationUnit_Impl, &STUI)) {
2722142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "libclang: crash detected during AST saving: {\n");
2723142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "  'filename' : '%s'\n", FileName);
2724142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "  'options' : %d,\n", options);
2725142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    fprintf(stderr, "}\n");
2726142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2727142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis    return CXSaveError_Unknown;
2728142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2729142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
27306df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(TU);
2731142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  }
2732142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis
2733142bcb59c012504360881e8f744d9d4a8d92c2acArgyrios Kyrtzidis  return STUI.result;
27347ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor}
273519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
27369ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
2737ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  if (CTUnit) {
2738ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // If the translation unit has been marked as unsafe to free, just discard
2739ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // it.
2740a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    if (static_cast<ASTUnit *>(CTUnit->TUData)->isUnsafeToFree())
2741ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar      return;
2742ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2743a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete static_cast<ASTUnit *>(CTUnit->TUData);
2744a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    disposeCXStringPool(CTUnit->StringPool);
2745153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek    delete static_cast<CXDiagnosticSetImpl *>(CTUnit->Diagnostics);
2746a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete CTUnit;
2747ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  }
27482bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
27490d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2750e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregorunsigned clang_defaultReparseOptions(CXTranslationUnit TU) {
2751e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor  return CXReparse_None;
2752e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor}
2753e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor
2754ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarstruct ReparseTranslationUnitInfo {
2755ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU;
2756ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files;
2757ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
2758ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options;
2759ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  int result;
2760ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar};
2761593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2762b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_reparseTranslationUnit_Impl(void *UserData) {
2763ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo *RTUI =
2764ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    static_cast<ReparseTranslationUnitInfo*>(UserData);
2765ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU = RTUI->TU;
2766153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek
2767153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek  // Reset the associated diagnostics.
2768153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek  delete static_cast<CXDiagnosticSetImpl*>(TU->Diagnostics);
2769153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek  TU->Diagnostics = 0;
2770153221717e39ce41323d5bc6b8b8bf130923c1bdTed Kremenek
2771ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files = RTUI->num_unsaved_files;
2772ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files = RTUI->unsaved_files;
2773ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options = RTUI->options;
2774ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  (void) options;
2775ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  RTUI->result = 1;
2776ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2777abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  if (!TU)
2778ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return;
2779593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2780fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  CIndexer *CXXIdx = (CIndexer*)TU->CIdx;
2781fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
278281b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
2783fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
2784a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
2785593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2786abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
27871e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<std::vector<ASTUnit::RemappedFile> >
278825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
278925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
279025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
279125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
279225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
279325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
2794abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
27955f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2796abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor    const llvm::MemoryBuffer *Buffer
27971abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
279825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
279925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
2800abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  }
2801abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
28024ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek  if (!CXXUnit->Reparse(RemappedFiles->size() ? &(*RemappedFiles)[0] : 0,
28034ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                        RemappedFiles->size()))
2804593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor    RTUI->result = 0;
2805abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor}
2806593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2807ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarint clang_reparseTranslationUnit(CXTranslationUnit TU,
2808ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned num_unsaved_files,
2809ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 struct CXUnsavedFile *unsaved_files,
2810ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned options) {
2811ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo RTUI = { TU, num_unsaved_files, unsaved_files,
2812ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                      options, 0 };
28138c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis
2814e7de9b4a1f4a15620ab15bc8159018df7d54080aArgyrios Kyrtzidis  if (getenv("LIBCLANG_NOTHREADS")) {
28158c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis    clang_reparseTranslationUnit_Impl(&RTUI);
28168c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis    return RTUI.result;
28178c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis  }
28188c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis
2819ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  llvm::CrashRecoveryContext CRC;
2820ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2821bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_reparseTranslationUnit_Impl, &RTUI)) {
2822b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbar    fprintf(stderr, "libclang: crash detected during reparsing\n");
2823a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    static_cast<ASTUnit *>(TU->TUData)->setUnsafeToFree(true);
2824ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return 1;
28256df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  } else if (getenv("LIBCLANG_RESOURCE_USAGE"))
28266df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(TU);
28271dfb26af4d6aa4f7818e256659a79f1ec2cba784Ted Kremenek
2828ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  return RTUI.result;
2829ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar}
2830ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2831df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor
28329ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
28332b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CTUnit)
2834ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2835f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2836a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit->TUData);
2837ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(CXXUnit->getOriginalSourceFileName(), true);
2838af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff}
28391eb79b58e56b99cf557d5d353586a10c5360364dDaniel Dunbar
28407eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas GregorCXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
2841aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  CXCursor Result = { CXCursor_TranslationUnit, 0, { 0, 0, TU } };
28427eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return Result;
28437eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
28447eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
2845fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2846600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2847fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2848fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXFile Operations.
2849fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2850fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2851fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
285274844072411bae91d5dbb89955d200cbe1e0a1c8Ted KremenekCXString clang_getFileName(CXFile SFile) {
285398258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
2854a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return createCXString((const char*)NULL);
2855f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
285688145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
285774844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek  return createCXString(FEnt->getName());
285888145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
285988145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff
286088145034694ed5267fa6fa5febc54fadc02bd479Steve Narofftime_t clang_getFileTime(CXFile SFile) {
286198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
286298258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor    return 0;
2863f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
286488145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
286588145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  return FEnt->getModificationTime();
2866ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff}
2867f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2868b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXFile clang_getFile(CXTranslationUnit tu, const char *file_name) {
2869b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!tu)
2870b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return 0;
2871f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2872a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2873f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2874b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  FileManager &FMgr = CXXUnit->getFileManager();
287539b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattner  return const_cast<FileEntry *>(FMgr.getFile(file_name));
2876b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2877f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2878dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregorunsigned clang_isFileMultipleIncludeGuarded(CXTranslationUnit tu, CXFile file) {
2879dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  if (!tu || !file)
2880dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor    return 0;
2881dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor
2882dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2883dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  FileEntry *FEnt = static_cast<FileEntry *>(file);
2884dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  return CXXUnit->getPreprocessor().getHeaderSearchInfo()
2885dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor                                          .isFileMultipleIncludeGuarded(FEnt);
2886dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor}
2887dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor
2888fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2889fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2890fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2891fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXCursor Operations.
2892fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2893fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2894fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekstatic Decl *getDeclFromExpr(Stmt *E) {
2895c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis  if (ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
2896db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return getDeclFromExpr(CE->getSubExpr());
2897db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2898fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
2899fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RefExpr->getDecl();
2900fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (MemberExpr *ME = dyn_cast<MemberExpr>(E))
2901fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return ME->getMemberDecl();
2902fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
2903fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RE->getDecl();
2904b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis  if (ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(E)) {
2905b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    if (PRE->isExplicitProperty())
2906b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis      return PRE->getExplicitProperty();
2907b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // It could be messaging both getter and setter as in:
2908b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // ++myobj.myprop;
2909b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // in which case prefer to associate the setter since it is less obvious
2910b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    // from inspecting the source that the setter is going to get called.
2911b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    if (PRE->isMessagingSetter())
2912b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis      return PRE->getImplicitPropertySetter();
2913b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis    return PRE->getImplicitPropertyGetter();
2914b085d898bdfe35097eba45f4072b0f6865f561dcArgyrios Kyrtzidis  }
29154b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (PseudoObjectExpr *POE = dyn_cast<PseudoObjectExpr>(E))
29164b9c2d235fb9449e249d74f48ecfec601650de93John McCall    return getDeclFromExpr(POE->getSyntacticForm());
29174b9c2d235fb9449e249d74f48ecfec601650de93John McCall  if (OpaqueValueExpr *OVE = dyn_cast<OpaqueValueExpr>(E))
29184b9c2d235fb9449e249d74f48ecfec601650de93John McCall    if (Expr *Src = OVE->getSourceExpr())
29194b9c2d235fb9449e249d74f48ecfec601650de93John McCall      return getDeclFromExpr(Src);
2920db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2921fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (CallExpr *CE = dyn_cast<CallExpr>(E))
2922fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return getDeclFromExpr(CE->getCallee());
29235f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  if (CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E))
292493798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    if (!CE->isElidable())
292593798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CE->getConstructor();
2926fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
2927fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return OME->getMethodDecl();
2928f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2929db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCProtocolExpr *PE = dyn_cast<ObjCProtocolExpr>(E))
2930db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return PE->getProtocol();
2931c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor  if (SubstNonTypeTemplateParmPackExpr *NTTP
2932c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor                              = dyn_cast<SubstNonTypeTemplateParmPackExpr>(E))
2933c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor    return NTTP->getParameterPack();
293494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
293594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    if (isa<NonTypeTemplateParmDecl>(SizeOfPack->getPack()) ||
293694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        isa<ParmVarDecl>(SizeOfPack->getPack()))
293794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      return SizeOfPack->getPack();
2938db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2939fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  return 0;
2940fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek}
2941ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff
2942c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbarstatic SourceLocation getLocationFromExpr(Expr *E) {
2943c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis  if (ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
2944c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis    return getLocationFromExpr(CE->getSubExpr());
2945c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis
2946c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
2947c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return /*FIXME:*/Msg->getLeftLoc();
2948c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
2949c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return DRE->getLocation();
2950c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (MemberExpr *Member = dyn_cast<MemberExpr>(E))
2951c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Member->getMemberLoc();
2952c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
2953c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Ivar->getLocation();
295494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
295594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    return SizeOfPack->getPackLoc();
2956d0469525581a851b68f5b4f960ee4190dcc7c932Argyrios Kyrtzidis  if (ObjCPropertyRefExpr *PropRef = dyn_cast<ObjCPropertyRefExpr>(E))
2957d0469525581a851b68f5b4f960ee4190dcc7c932Argyrios Kyrtzidis    return PropRef->getLocation();
295894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
2959c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  return E->getLocStart();
2960c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar}
2961c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar
2962fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
2963f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2964f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekunsigned clang_visitChildren(CXCursor parent,
2965b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXCursorVisitor visitor,
2966b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXClientData client_data) {
2967f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  CursorVisitor CursorVis(getCursorTU(parent), visitor, client_data,
2968f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                          /*VisitPreprocessorLast=*/false);
2969b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return CursorVis.VisitChildren(parent);
2970b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
2971b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
29723387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#ifndef __has_feature
29733387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#define __has_feature(x) 0
29743387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
29753387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#if __has_feature(blocks)
29763387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef enum CXChildVisitResult
29773387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall     (^CXCursorVisitorBlock)(CXCursor cursor, CXCursor parent);
29783387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29793387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
29803387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
29813387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
29823387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block(cursor, parent);
29833387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
29843387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#else
29853387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// If we are compiled with a compiler that doesn't have native blocks support,
29863387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// define and call the block manually, so the
29873387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef struct _CXChildVisitResult
29883387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall{
29893387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	void *isa;
29903387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int flags;
29913387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int reserved;
29929e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar	enum CXChildVisitResult(*invoke)(struct _CXChildVisitResult*, CXCursor,
29939e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                         CXCursor);
29943387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall} *CXCursorVisitorBlock;
29953387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
29963387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
29973387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
29983387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
29993387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block->invoke(block, cursor, parent);
30003387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30013387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
30023387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30033387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30049e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbarunsigned clang_visitChildrenWithBlock(CXCursor parent,
30059e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                      CXCursorVisitorBlock block) {
30063387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return clang_visitChildren(parent, visitWithBlock, block);
30073387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30083387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
300978205d4bada39d95097e766af9eb30cdd0159461Douglas Gregorstatic CXString getDeclSpelling(Decl *D) {
301016ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  if (!D)
301116ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    return createCXString("");
301216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
301316ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  NamedDecl *ND = dyn_cast<NamedDecl>(D);
3014e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  if (!ND) {
30155f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    if (ObjCPropertyImplDecl *PropImpl =dyn_cast<ObjCPropertyImplDecl>(D))
3016e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
3017e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return createCXString(Property->getIdentifier()->getName());
3018e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
3019ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
3020e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  }
3021e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
302278205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
3023ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(OMD->getSelector().getAsString());
3024f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
302578205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
302678205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // No, this isn't the same as the code below. getIdentifier() is non-virtual
302778205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // and returns different names. NamedDecl returns the class name and
302878205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // ObjCCategoryImplDecl returns the category name.
3029ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(CIMP->getIdentifier()->getNameStart());
3030f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30310a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  if (isa<UsingDirectiveDecl>(D))
30320a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("");
30330a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
303436d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith  SmallString<1024> S;
303550aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::raw_svector_ostream os(S);
303650aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  ND->printName(os);
303750aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek
303850aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  return createCXString(os.str());
303978205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor}
3040f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30419ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getCursorSpelling(CXCursor C) {
30427eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  if (clang_isTranslationUnit(C.kind))
3043a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return clang_getTranslationUnitSpelling(
3044a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                            static_cast<CXTranslationUnit>(C.data[2]));
30457eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
3046f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  if (clang_isReference(C.kind)) {
3047f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    switch (C.kind) {
3048acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCSuperClassRef: {
30492e331b938b38057e333fab0ba841130ea8467794Douglas Gregor      ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
3050ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Super->getIdentifier()->getNameStart());
3051acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3052acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCClassRef: {
30531adb082a709f7b588f03672999294e061234b2cfDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
3054ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Class->getIdentifier()->getNameStart());
3055acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3056acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCProtocolRef: {
305778db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor      ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
3058f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      assert(OID && "getCursorSpelling(): Missing protocol decl");
3059ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(OID->getIdentifier()->getNameStart());
3060acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
30613064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
30623064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = getCursorCXXBaseSpecifier(C);
30633064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return createCXString(B->getType().getAsString());
30643064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
30657d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    case CXCursor_TypeRef: {
30667d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      TypeDecl *Type = getCursorTypeRef(C).first;
30677d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      assert(Type && "Missing type decl");
30687d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3069ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(getCursorContext(C).getTypeDeclType(Type).
3070ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                              getAsString());
30717d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
30720b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
30730b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      TemplateDecl *Template = getCursorTemplateRef(C).first;
30746931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(Template && "Missing template decl");
30750b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
30760b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString(Template->getNameAsString());
30770b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
30786931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
30796931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
30806931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      NamedDecl *NS = getCursorNamespaceRef(C).first;
30816931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(NS && "Missing namespace decl");
30826931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
30836931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return createCXString(NS->getNameAsString());
30846931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
30857d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3086a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3087a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      FieldDecl *Field = getCursorMemberRef(C).first;
3088a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      assert(Field && "Missing member decl");
3089a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3090a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return createCXString(Field->getNameAsString());
3091a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3092a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
309336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
309436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      LabelStmt *Label = getCursorLabelRef(C).first;
309536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      assert(Label && "Missing label");
309636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
3097ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
309836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
309936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31001f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef: {
31011f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
31021f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Decl *D = Storage.dyn_cast<Decl *>()) {
31031f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        if (NamedDecl *ND = dyn_cast<NamedDecl>(D))
31041f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor          return createCXString(ND->getNameAsString());
31051f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
31061f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      }
31071f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
31081f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString(E->getName().getAsString());
31091f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedTemplateStorage *Ovl
31101f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        = Storage.get<OverloadedTemplateStorage*>();
31111f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Ovl->size() == 0)
31121f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
31131f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return createCXString((*Ovl->begin())->getNameAsString());
31141f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    }
31151f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3116011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef: {
3117011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      VarDecl *Var = getCursorVariableRef(C).first;
3118011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      assert(Var && "Missing variable decl");
3119011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
3120011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return createCXString(Var->getNameAsString());
3121011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    }
3122011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
3123acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    default:
3124ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString("<not implemented>");
3125f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    }
3126f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  }
312797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
312897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
312997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    Decl *D = getDeclFromExpr(getCursorExpr(C));
313097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
313178205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor      return getDeclSpelling(D);
3132ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
313397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
313497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
313536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
313636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
313736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
3138ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
313936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
314036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("");
314136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
314236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31439b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion)
31449e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth    return createCXString(getCursorMacroExpansion(C)->getName()
31454ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor                                                           ->getNameStart());
31464ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor
3147572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
3148572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString(getCursorMacroDefinition(C)->getName()
3149572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor                                                           ->getNameStart());
3150572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3151ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective)
3152ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString(getCursorInclusionDirective(C)->getFileName());
3153ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
315460cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor  if (clang_isDeclaration(C.kind))
315560cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor    return getDeclSpelling(getCursorDecl(C));
3156e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
31575f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  if (C.kind == CXCursor_AnnotateAttr) {
31585f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    AnnotateAttr *AA = cast<AnnotateAttr>(cxcursor::getCursorAttr(C));
31595f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    return createCXString(AA->getAnnotation());
31605f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  }
31615f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen
316284b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis  if (C.kind == CXCursor_AsmLabelAttr) {
316384b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis    AsmLabelAttr *AA = cast<AsmLabelAttr>(cxcursor::getCursorAttr(C));
316484b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis    return createCXString(AA->getLabel());
316584b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis  }
316684b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis
3167ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString("");
3168f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3169f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
3170358559d8d7b458c5f64941842383a16e61f0828dDouglas GregorCXString clang_getCursorDisplayName(CXCursor C) {
3171358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!clang_isDeclaration(C.kind))
3172358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return clang_getCursorSpelling(C);
3173358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3174358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  Decl *D = getCursorDecl(C);
3175358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!D)
3176358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString("");
3177358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
317830c42404202d2e2512e51efc6066bd614cfdb5a4Douglas Gregor  PrintingPolicy Policy = getCursorContext(C).getPrintingPolicy();
3179358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(D))
3180358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    D = FunTmpl->getTemplatedDecl();
3181358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3182358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
318336d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith    SmallString<64> Str;
3184358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3185a59d20b135bfde058a5a69045bab5ec4e2553f74Benjamin Kramer    OS << *Function;
3186358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->getPrimaryTemplate())
3187358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "<>";
3188358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "(";
3189358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Function->getNumParams(); I != N; ++I) {
3190358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3191358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3192358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << Function->getParamDecl(I)->getType().getAsString(Policy);
3193358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3194358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3195358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->isVariadic()) {
3196358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Function->getNumParams())
3197358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3198358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "...";
3199358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3200358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ")";
3201358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3202358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3203358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3204358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) {
320536d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith    SmallString<64> Str;
3206358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3207a59d20b135bfde058a5a69045bab5ec4e2553f74Benjamin Kramer    OS << *ClassTemplate;
3208358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "<";
3209358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    TemplateParameterList *Params = ClassTemplate->getTemplateParameters();
3210358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Params->size(); I != N; ++I) {
3211358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3212358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3213358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3214358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      NamedDecl *Param = Params->getParam(I);
3215358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Param->getIdentifier()) {
3216358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << Param->getIdentifier()->getName();
3217358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        continue;
3218358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      }
3219358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3220358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // There is no parameter name, which makes this tricky. Try to come up
3221358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // with something useful that isn't too long.
3222358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
3223358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << (TTP->wasDeclaredWithTypename()? "typename" : "class");
3224358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else if (NonTypeTemplateParmDecl *NTTP
3225358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                    = dyn_cast<NonTypeTemplateParmDecl>(Param))
3226358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << NTTP->getType().getAsString(Policy);
3227358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else
3228358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << "template<...> class";
3229358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3230358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3231358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ">";
3232358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3233358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3234358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3235358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateSpecializationDecl *ClassSpec
3236358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                              = dyn_cast<ClassTemplateSpecializationDecl>(D)) {
3237358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    // If the type was explicitly written, use that.
3238358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (TypeSourceInfo *TSInfo = ClassSpec->getTypeAsWritten())
3239358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      return createCXString(TSInfo->getType().getAsString(Policy));
3240358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
324136d592718ff342f762e32cbde73d1113f88cb275Dylan Noblesmith    SmallString<64> Str;
3242358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3243a59d20b135bfde058a5a69045bab5ec4e2553f74Benjamin Kramer    OS << *ClassSpec;
3244358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << TemplateSpecializationType::PrintTemplateArgumentList(
3245910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().data(),
3246910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().size(),
3247358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                                                Policy);
3248358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3249358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3250358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3251358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  return clang_getCursorSpelling(C);
3252358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor}
3253358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3254e68fff6fc083c6270d835216a3de0b82c6ef0310Ted KremenekCXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
325589922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  switch (Kind) {
3256e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FunctionDecl:
3257e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FunctionDecl");
3258e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypedefDecl:
3259e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypedefDecl");
3260e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumDecl:
3261e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumDecl");
3262e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumConstantDecl:
3263e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumConstantDecl");
3264e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_StructDecl:
3265e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("StructDecl");
3266e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnionDecl:
3267e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnionDecl");
3268e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ClassDecl:
3269e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ClassDecl");
3270e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FieldDecl:
3271e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FieldDecl");
3272e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_VarDecl:
3273e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("VarDecl");
3274e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ParmDecl:
3275e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ParmDecl");
3276e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInterfaceDecl:
3277e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInterfaceDecl");
3278e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryDecl:
3279e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryDecl");
3280e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolDecl:
3281e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolDecl");
3282e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCPropertyDecl:
3283e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCPropertyDecl");
3284e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCIvarDecl:
3285e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCIvarDecl");
3286e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInstanceMethodDecl:
3287e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInstanceMethodDecl");
3288e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassMethodDecl:
3289e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassMethodDecl");
3290e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCImplementationDecl:
3291e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCImplementationDecl");
3292e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryImplDecl:
3293e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryImplDecl");
32948bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek  case CXCursor_CXXMethod:
32958bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek      return createCXString("CXXMethod");
3296e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedDecl:
3297e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedDecl");
3298e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCSuperClassRef:
3299e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCSuperClassRef");
3300e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolRef:
3301e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolRef");
3302e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassRef:
3303e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassRef");
3304e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypeRef:
3305e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypeRef");
33060b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case CXCursor_TemplateRef:
33070b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString("TemplateRef");
33086931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceRef:
33096931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceRef");
3310a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  case CXCursor_MemberRef:
3311a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return createCXString("MemberRef");
331236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelRef:
331336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("LabelRef");
33141f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case CXCursor_OverloadedDeclRef:
33151f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return createCXString("OverloadedDeclRef");
3316011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  case CXCursor_VariableRef:
3317011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    return createCXString("VariableRef");
331842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IntegerLiteral:
331942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IntegerLiteral");
332042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_FloatingLiteral:
332142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("FloatingLiteral");
332242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ImaginaryLiteral:
332342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ImaginaryLiteral");
332442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_StringLiteral:
332542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("StringLiteral");
332642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CharacterLiteral:
332742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CharacterLiteral");
332842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ParenExpr:
332942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ParenExpr");
333042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnaryOperator:
333142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnaryOperator");
333242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ArraySubscriptExpr:
333342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ArraySubscriptExpr");
333442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_BinaryOperator:
333542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("BinaryOperator");
333642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundAssignOperator:
333742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundAssignOperator");
333842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ConditionalOperator:
333942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ConditionalOperator");
334042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CStyleCastExpr:
334142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CStyleCastExpr");
334242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundLiteralExpr:
334342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundLiteralExpr");
334442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_InitListExpr:
334542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("InitListExpr");
334642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_AddrLabelExpr:
334742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("AddrLabelExpr");
334842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_StmtExpr:
334942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("StmtExpr");
335042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GenericSelectionExpr:
335142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GenericSelectionExpr");
335242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GNUNullExpr:
335342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GNUNullExpr");
335442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXStaticCastExpr:
335542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXStaticCastExpr");
335642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXDynamicCastExpr:
335742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXDynamicCastExpr");
335842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXReinterpretCastExpr:
335942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXReinterpretCastExpr");
336042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXConstCastExpr:
336142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXConstCastExpr");
336242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXFunctionalCastExpr:
336342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXFunctionalCastExpr");
336442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXTypeidExpr:
336542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXTypeidExpr");
336642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXBoolLiteralExpr:
336742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXBoolLiteralExpr");
336842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXNullPtrLiteralExpr:
336942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXNullPtrLiteralExpr");
337042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXThisExpr:
337142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXThisExpr");
337242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXThrowExpr:
337342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXThrowExpr");
337442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXNewExpr:
337542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXNewExpr");
337642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXDeleteExpr:
337742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXDeleteExpr");
337842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnaryExpr:
337942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnaryExpr");
338042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCStringLiteral:
338142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCStringLiteral");
3382b3f7542a950ac0d585a7783e825cfe670e05c553Ted Kremenek  case CXCursor_ObjCBoolLiteralExpr:
3383b3f7542a950ac0d585a7783e825cfe670e05c553Ted Kremenek      return createCXString("ObjCBoolLiteralExpr");
338442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCEncodeExpr:
338542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCEncodeExpr");
338642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCSelectorExpr:
338742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCSelectorExpr");
338842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCProtocolExpr:
338942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCProtocolExpr");
339042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCBridgedCastExpr:
339142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCBridgedCastExpr");
33921ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  case CXCursor_BlockExpr:
33931ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek      return createCXString("BlockExpr");
339442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_PackExpansionExpr:
339542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("PackExpansionExpr");
339642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SizeOfPackExpr:
339742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SizeOfPackExpr");
3398011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor  case CXCursor_LambdaExpr:
3399011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    return createCXString("LambdaExpr");
340042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnexposedExpr:
340142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnexposedExpr");
3402e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_DeclRefExpr:
3403e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("DeclRefExpr");
3404e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_MemberRefExpr:
3405e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("MemberRefExpr");
3406e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_CallExpr:
3407e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("CallExpr");
3408e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCMessageExpr:
3409e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCMessageExpr");
3410e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedStmt:
3411e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedStmt");
341242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DeclStmt:
341342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DeclStmt");
341436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelStmt:
341536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return createCXString("LabelStmt");
341642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundStmt:
341742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundStmt");
341842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CaseStmt:
341942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CaseStmt");
342042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DefaultStmt:
342142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DefaultStmt");
342242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IfStmt:
342342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IfStmt");
342442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SwitchStmt:
342542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SwitchStmt");
342642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_WhileStmt:
342742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("WhileStmt");
342842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DoStmt:
342942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DoStmt");
343042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ForStmt:
343142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ForStmt");
343242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GotoStmt:
343342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GotoStmt");
343442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IndirectGotoStmt:
343542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IndirectGotoStmt");
343642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ContinueStmt:
343742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ContinueStmt");
343842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_BreakStmt:
343942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("BreakStmt");
344042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ReturnStmt:
344142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ReturnStmt");
344242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_AsmStmt:
344342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("AsmStmt");
344442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtTryStmt:
344542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtTryStmt");
344642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtCatchStmt:
344742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtCatchStmt");
344842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtFinallyStmt:
344942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtFinallyStmt");
345042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtThrowStmt:
345142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtThrowStmt");
345242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtSynchronizedStmt:
345342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtSynchronizedStmt");
345442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAutoreleasePoolStmt:
345542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAutoreleasePoolStmt");
345642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCForCollectionStmt:
345742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCForCollectionStmt");
345842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXCatchStmt:
345942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXCatchStmt");
346042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXTryStmt:
346142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXTryStmt");
346242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXForRangeStmt:
346342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXForRangeStmt");
346442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHTryStmt:
346542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHTryStmt");
346642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHExceptStmt:
346742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHExceptStmt");
346842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHFinallyStmt:
346942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHFinallyStmt");
347042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_NullStmt:
347142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("NullStmt");
3472e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_InvalidFile:
3473e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("InvalidFile");
3474292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek  case CXCursor_InvalidCode:
3475292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek    return createCXString("InvalidCode");
3476e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NoDeclFound:
3477e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NoDeclFound");
3478e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NotImplemented:
3479e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NotImplemented");
3480e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TranslationUnit:
3481e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TranslationUnit");
3482e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_UnexposedAttr:
3483e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("UnexposedAttr");
3484e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_IBActionAttr:
3485e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("attribute(ibaction)");
34869f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_IBOutletAttr:
34879f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor     return createCXString("attribute(iboutlet)");
3488857e918a8a40deb128840308a318bf623d68295fTed Kremenek  case CXCursor_IBOutletCollectionAttr:
3489857e918a8a40deb128840308a318bf623d68295fTed Kremenek      return createCXString("attribute(iboutletcollection)");
34906639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  case CXCursor_CXXFinalAttr:
34916639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      return createCXString("attribute(final)");
34926639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  case CXCursor_CXXOverrideAttr:
34936639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      return createCXString("attribute(override)");
34945f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  case CXCursor_AnnotateAttr:
34955f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    return createCXString("attribute(annotate)");
349684b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis  case CXCursor_AsmLabelAttr:
349784b796492de8a708150dd3f86ae191041d42eef9Argyrios Kyrtzidis    return createCXString("asm label");
34989f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_PreprocessingDirective:
34999f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return createCXString("preprocessing directive");
3500572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  case CXCursor_MacroDefinition:
3501572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString("macro definition");
35029b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  case CXCursor_MacroExpansion:
35039b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth    return createCXString("macro expansion");
3504ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  case CXCursor_InclusionDirective:
3505ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString("inclusion directive");
35068f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  case CXCursor_Namespace:
35078f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek    return createCXString("Namespace");
3508a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  case CXCursor_LinkageSpec:
3509a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek    return createCXString("LinkageSpec");
35103064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  case CXCursor_CXXBaseSpecifier:
35113064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    return createCXString("C++ base class specifier");
351201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Constructor:
351301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConstructor");
351401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Destructor:
351501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXDestructor");
351601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_ConversionFunction:
351701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConversion");
3518fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTypeParameter:
3519fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTypeParameter");
3520fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_NonTypeTemplateParameter:
3521fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("NonTypeTemplateParameter");
3522fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTemplateParameter:
3523fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTemplateParameter");
3524fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_FunctionTemplate:
3525fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("FunctionTemplate");
352639d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  case CXCursor_ClassTemplate:
352739d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return createCXString("ClassTemplate");
352874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  case CXCursor_ClassTemplatePartialSpecialization:
352974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return createCXString("ClassTemplatePartialSpecialization");
35306931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceAlias:
35316931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceAlias");
35320a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  case CXCursor_UsingDirective:
35330a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("UsingDirective");
35347e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  case CXCursor_UsingDeclaration:
35357e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor    return createCXString("UsingDeclaration");
3536162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  case CXCursor_TypeAliasDecl:
3537352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("TypeAliasDecl");
3538352697a87bca664356f21a838b162084013625eaDouglas Gregor  case CXCursor_ObjCSynthesizeDecl:
3539352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("ObjCSynthesizeDecl");
3540352697a87bca664356f21a838b162084013625eaDouglas Gregor  case CXCursor_ObjCDynamicDecl:
3541352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("ObjCDynamicDecl");
35422dfdb948bef51a601e763191e4becfe59880d382Argyrios Kyrtzidis  case CXCursor_CXXAccessSpecifier:
35432dfdb948bef51a601e763191e4becfe59880d382Argyrios Kyrtzidis    return createCXString("CXXAccessSpecifier");
354489922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  }
3545e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3546deb06bd3566e18f677e76bc435d478b033fe328bTed Kremenek  llvm_unreachable("Unhandled CXCursorKind");
3547600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
354889922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff
3549064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidisstruct GetCursorData {
3550064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  SourceLocation TokenBeginLoc;
35514b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  bool PointsAtMacroArgExpansion;
3552064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  CXCursor &BestCursor;
3553064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
35544b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  GetCursorData(SourceManager &SM,
35554b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                SourceLocation tokenBegin, CXCursor &outputCursor)
35564b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    : TokenBeginLoc(tokenBegin), BestCursor(outputCursor) {
35574b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    PointsAtMacroArgExpansion = SM.isMacroArgExpansion(tokenBegin);
35584b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  }
3559064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis};
3560064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
35614b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidisstatic enum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
35624b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                                                CXCursor parent,
35634b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                                                CXClientData client_data) {
3564064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  GetCursorData *Data = static_cast<GetCursorData *>(client_data);
3565064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  CXCursor *BestCursor = &Data->BestCursor;
35664b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis
35674b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // If we point inside a macro argument we should provide info of what the
35684b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // token is so use the actual cursor, don't replace it with a macro expansion
35694b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // cursor.
35704b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  if (cursor.kind == CXCursor_MacroExpansion && Data->PointsAtMacroArgExpansion)
35714b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    return CXChildVisit_Recurse;
357265ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis
357365ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis  if (clang_isDeclaration(cursor.kind)) {
357465ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // Avoid having the implicit methods override the property decls.
357516ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (ObjCMethodDecl *MD = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor)))
357665ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis      if (MD->isImplicit())
357765ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis        return CXChildVisit_Break;
357865ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis  }
3579064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
3580064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  if (clang_isExpression(cursor.kind) &&
3581064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis      clang_isDeclaration(BestCursor->kind)) {
358216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (Decl *D = getCursorDecl(*BestCursor)) {
358316ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // Avoid having the cursor of an expression replace the declaration cursor
358416ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // when the expression source range overlaps the declaration range.
358516ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // This can happen for C++ constructor expressions whose range generally
358616ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      // include the variable declaration, e.g.:
358716ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      //  MyCXXClass foo; // Make sure pointing at 'foo' returns a VarDecl cursor.
358816ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      if (D->getLocation().isValid() && Data->TokenBeginLoc.isValid() &&
358916ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis          D->getLocation() == Data->TokenBeginLoc)
359016ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis        return CXChildVisit_Break;
359116ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    }
3592064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  }
3593064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
359493798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // If our current best cursor is the construction of a temporary object,
359593798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // don't replace that cursor with a type reference, because we want
359693798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // clang_getCursor() to point at the constructor.
359793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  if (clang_isExpression(BestCursor->kind) &&
359893798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor      isa<CXXTemporaryObjectExpr>(getCursorExpr(*BestCursor)) &&
3599aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      cursor.kind == CXCursor_TypeRef) {
3600aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    // Keep the cursor pointing at CXXTemporaryObjectExpr but also mark it
3601aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    // as having the actual point on the type reference.
3602aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    *BestCursor = getTypeRefedCallExprCursor(*BestCursor);
360393798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CXChildVisit_Recurse;
3604aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  }
360593798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor
360633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  *BestCursor = cursor;
360733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return CXChildVisit_Recurse;
360833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
3609e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3610b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
3611b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!TU)
3612f462989fe8d6f59ab2d7d0fe2b4b96292ce706eaTed Kremenek    return clang_getNullCursor();
3613e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3614a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3615bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
3616bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
3617a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek  SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
3618671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  CXCursor Result = cxcursor::getCursor(TU, SLoc);
3619a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek
362040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  bool Logging = getenv("LIBCLANG_LOGGING");
362140749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  if (Logging) {
362240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile SearchFile;
362340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned SearchLine, SearchColumn;
362440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile ResultFile;
362540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned ResultLine, ResultColumn;
36266653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    CXString SearchFileName, ResultFileName, KindSpelling, USR;
36276653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    const char *IsDef = clang_isCursorDefinition(Result)? " (Definition)" : "";
362840749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXSourceLocation ResultLoc = clang_getCursorLocation(Result);
362940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
363020174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    clang_getExpansionLocation(Loc, &SearchFile, &SearchLine, &SearchColumn, 0);
363120174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    clang_getExpansionLocation(ResultLoc, &ResultFile, &ResultLine,
363220174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                               &ResultColumn, 0);
363340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    SearchFileName = clang_getFileName(SearchFile);
363440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    ResultFileName = clang_getFileName(ResultFile);
363540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    KindSpelling = clang_getCursorKindSpelling(Result.kind);
36366653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    USR = clang_getCursorUSR(Result);
36376653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    fprintf(stderr, "clang_getCursor(%s:%d:%d) = %s(%s:%d:%d):%s%s\n",
363840749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(SearchFileName), SearchLine, SearchColumn,
363940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(KindSpelling),
36406653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(ResultFileName), ResultLine, ResultColumn,
36416653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(USR), IsDef);
364240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(SearchFileName);
364340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(ResultFileName);
364440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(KindSpelling);
36456653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    clang_disposeString(USR);
36460aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor
36470aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    CXCursor Definition = clang_getCursorDefinition(Result);
36480aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    if (!clang_equalCursors(Definition, clang_getNullCursor())) {
36490aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXSourceLocation DefinitionLoc = clang_getCursorLocation(Definition);
36500aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionKindSpelling
36510aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor                                = clang_getCursorKindSpelling(Definition.kind);
36520aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXFile DefinitionFile;
36530aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      unsigned DefinitionLine, DefinitionColumn;
365420174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth      clang_getExpansionLocation(DefinitionLoc, &DefinitionFile,
365520174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                 &DefinitionLine, &DefinitionColumn, 0);
36560aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionFileName = clang_getFileName(DefinitionFile);
36570aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      fprintf(stderr, "  -> %s(%s:%d:%d)\n",
36580aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionKindSpelling),
36590aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionFileName),
36600aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              DefinitionLine, DefinitionColumn);
36610aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionFileName);
36620aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionKindSpelling);
36630aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    }
366440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  }
366540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
3666e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  return Result;
366777128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
366877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
3669738855554394a6afcf39cc8345fd22c3756b8dd0Ted KremenekCXCursor clang_getNullCursor(void) {
36705bfb8c128c2ac8eb4032afc180cdc400a0f953caDouglas Gregor  return MakeCXCursorInvalid(CXCursor_InvalidFile);
3671738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
3672738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek
3673738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenekunsigned clang_equalCursors(CXCursor X, CXCursor Y) {
3674283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  return X == Y;
3675738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
36760d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
36779ce5584553054d0cb934940586aca0186e87fa57Douglas Gregorunsigned clang_hashCursor(CXCursor C) {
36789ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  unsigned Index = 0;
36799ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  if (clang_isExpression(C.kind) || clang_isStatement(C.kind))
36809ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor    Index = 1;
36819ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
36829ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  return llvm::DenseMapInfo<std::pair<unsigned, void*> >::getHashValue(
36839ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor                                        std::make_pair(C.kind, C.data[Index]));
36849ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor}
36859ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
36869ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isInvalid(enum CXCursorKind K) {
368777128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
368877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
368977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
36909ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isDeclaration(enum CXCursorKind K) {
369189922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl;
369289922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff}
36932d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff
36949ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isReference(enum CXCursorKind K) {
3695f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
3696f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3697f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
369897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isExpression(enum CXCursorKind K) {
369997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
370097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
370197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
370297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isStatement(enum CXCursorKind K) {
370397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
370497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
370597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
37068be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregorunsigned clang_isAttribute(enum CXCursorKind K) {
37078be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor    return K >= CXCursor_FirstAttr && K <= CXCursor_LastAttr;
37088be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor}
37098be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor
37107eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregorunsigned clang_isTranslationUnit(enum CXCursorKind K) {
37117eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return K == CXCursor_TranslationUnit;
37127eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
37137eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
37149f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregorunsigned clang_isPreprocessing(enum CXCursorKind K) {
37159f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
37169f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor}
37179f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor
3718ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenekunsigned clang_isUnexposed(enum CXCursorKind K) {
3719ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  switch (K) {
3720ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedDecl:
3721ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedExpr:
3722ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedStmt:
3723ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedAttr:
3724ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return true;
3725ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    default:
3726ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return false;
3727ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  }
3728ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek}
3729ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek
37309ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXCursorKind clang_getCursorKind(CXCursor C) {
37319efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff  return C.kind;
37329efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff}
37339efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff
373498258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas GregorCXSourceLocation clang_getCursorLocation(CXCursor C) {
373598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (clang_isReference(C.kind)) {
3736f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    switch (C.kind) {
3737f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCSuperClassRef: {
3738f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3739f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCSuperClassRef(C);
3740a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3741f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3742f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3743f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
3744f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCProtocolDecl *, SourceLocation> P
3745f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCProtocolRef(C);
3746a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3747f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3748f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3749f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef: {
3750f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3751f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCClassRef(C);
3752a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3753f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
37547d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3755f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef: {
37567d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      std::pair<TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
3757a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
37587d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
37590b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
37600b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
37610b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      std::pair<TemplateDecl *, SourceLocation> P = getCursorTemplateRef(C);
37620b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
37630b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
37640b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
37656931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
37666931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      std::pair<NamedDecl *, SourceLocation> P = getCursorNamespaceRef(C);
37676931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
37686931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
37696931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
3770a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3771a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      std::pair<FieldDecl *, SourceLocation> P = getCursorMemberRef(C);
3772a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3773a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3774a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3775011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef: {
3776011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      std::pair<VarDecl *, SourceLocation> P = getCursorVariableRef(C);
3777011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3778011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    }
3779011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
37803064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
37811b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      CXXBaseSpecifier *BaseSpec = getCursorCXXBaseSpecifier(C);
37821b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (!BaseSpec)
37831b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return clang_getNullLocation();
37841b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
37851b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (TypeSourceInfo *TSInfo = BaseSpec->getTypeSourceInfo())
37861b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return cxloc::translateSourceLocation(getCursorContext(C),
37871b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                            TSInfo->getTypeLoc().getBeginLoc());
37881b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
37891b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
379096a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar                                        BaseSpec->getLocStart());
37913064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
3792f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
379336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
379436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      std::pair<LabelStmt *, SourceLocation> P = getCursorLabelRef(C);
379536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C), P.second);
379636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
379736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
37981f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
37991f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
38001f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                          getCursorOverloadedDeclRef(C).second);
38011f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3802f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    default:
3803f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3804f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      llvm_unreachable("Missed a reference kind");
3805f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
380698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  }
380797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
380897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3809f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    return cxloc::translateSourceLocation(getCursorContext(C),
381097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor                                   getLocationFromExpr(getCursorExpr(C)));
381197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
381236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind))
381336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C),
381436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor                                          getCursorStmt(C)->getLocStart());
381536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
38169f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  if (C.kind == CXCursor_PreprocessingDirective) {
38179f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
38189f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
38199f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  }
38204807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
38219b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion) {
38224ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    SourceLocation L
38239e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth      = cxcursor::getCursorMacroExpansion(C)->getSourceRange().getBegin();
38244807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
38254807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  }
3826572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3827572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition) {
3828572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
3829572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3830572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  }
3831ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3832ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective) {
3833ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    SourceLocation L
3834ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      = cxcursor::getCursorInclusionDirective(C)->getSourceRange().getBegin();
3835ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3836ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  }
3837ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
38389a700d277c38d9afaa7cb3fe93a714bfe9b62eecTed Kremenek  if (C.kind < CXCursor_FirstDecl || C.kind > CXCursor_LastDecl)
38395352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullLocation();
384098258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor
3841f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  Decl *D = getCursorDecl(C);
384216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  if (!D)
384316ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    return clang_getNullLocation();
384416ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
3845f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  SourceLocation Loc = D->getLocation();
3846007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // FIXME: Multiple variables declared in a single declaration
3847007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // currently lack the information needed to correctly determine their
3848007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // ranges when accounting for the type-specifier.  We use context
3849007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3850007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // and if so, whether it is the first decl.
3851007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3852007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (!cxcursor::isFirstInDeclGroup(C))
3853007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      Loc = VD->getLocation();
3854007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
3855007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek
3856ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis  // For ObjC methods, give the start location of the method name.
3857ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis  if (ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(D))
3858ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis    Loc = MD->getSelectorStartLoc();
3859ccc6f36e53274fccae024f30ac5adb6be6f815d3Argyrios Kyrtzidis
38602ca54feee89d7277fb967e3247a64f40ef155a82Douglas Gregor  return cxloc::translateSourceLocation(getCursorContext(C), Loc);
386188145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
3862a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor
3863a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor} // end extern "C"
3864a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3865671436e9e2794c56f3c2e62739d225571493af37Argyrios KyrtzidisCXCursor cxcursor::getCursor(CXTranslationUnit TU, SourceLocation SLoc) {
3866671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  assert(TU);
3867671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3868671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // Guard against an invalid SourceLocation, or we may assert in one
3869671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // of the following calls.
3870671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  if (SLoc.isInvalid())
3871671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    return clang_getNullCursor();
3872671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3873671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3874671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3875671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // Translate the given source location to make it point at the beginning of
3876671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // the token under the cursor.
3877671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  SLoc = Lexer::GetBeginningOfToken(SLoc, CXXUnit->getSourceManager(),
38784e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie                                    CXXUnit->getASTContext().getLangOpts());
3879671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3880671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
3881671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  if (SLoc.isValid()) {
3882671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    GetCursorData ResultData(CXXUnit->getSourceManager(), SLoc, Result);
3883671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    CursorVisitor CursorVis(TU, GetCursorVisitor, &ResultData,
3884671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                            /*VisitPreprocessorLast=*/true,
3885e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                            /*VisitIncludedEntities=*/false,
3886671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                            SourceLocation(SLoc));
3887dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    CursorVis.visitFileRegion();
3888671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  }
3889671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3890671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  return Result;
3891671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis}
3892671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3893a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C) {
3894a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  if (clang_isReference(C.kind)) {
3895a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    switch (C.kind) {
3896a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCSuperClassRef:
3897a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return  getCursorObjCSuperClassRef(C).second;
3898f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3899a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCProtocolRef:
3900a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCProtocolRef(C).second;
3901f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3902a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCClassRef:
3903a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCClassRef(C).second;
39047d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3905a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_TypeRef:
3906a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorTypeRef(C).second;
39070b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
39080b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
39090b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return getCursorTemplateRef(C).second;
39100b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
39116931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
39126931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return getCursorNamespaceRef(C).second;
3913a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3914a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
3915a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return getCursorMemberRef(C).second;
3916a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
39173064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier:
39181b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return getCursorCXXBaseSpecifier(C)->getSourceRange();
3919f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
392036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
392136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return getCursorLabelRef(C).second;
392236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
39231f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
39241f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return getCursorOverloadedDeclRef(C).second;
39251f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3926011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef:
3927011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return getCursorVariableRef(C).second;
3928011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
3929a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    default:
3930a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3931a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      llvm_unreachable("Missed a reference kind");
3932a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    }
3933a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  }
393497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
393597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3936a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorExpr(C)->getSourceRange();
393733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
393833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (clang_isStatement(C.kind))
3939a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorStmt(C)->getSourceRange();
3940f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
39416639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  if (clang_isAttribute(C.kind))
39426639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis    return getCursorAttr(C)->getRange();
39436639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis
3944a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_PreprocessingDirective)
3945a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorPreprocessingDirective(C);
39464807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
3947ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_MacroExpansion) {
3948ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
3949ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorMacroExpansion(C)->getSourceRange();
3950ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
3951ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
3952572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3953ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_MacroDefinition) {
3954ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
3955ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorMacroDefinition(C)->getSourceRange();
3956ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
3957ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
3958ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3959ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_InclusionDirective) {
3960ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
3961ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorInclusionDirective(C)->getSourceRange();
3962ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
3963ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
3964ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
39650822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis  if (C.kind == CXCursor_TranslationUnit) {
39660822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
39670822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    FileID MainID = TU->getSourceManager().getMainFileID();
39680822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    SourceLocation Start = TU->getSourceManager().getLocForStartOfFile(MainID);
39690822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    SourceLocation End = TU->getSourceManager().getLocForEndOfFile(MainID);
39700822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis    return SourceRange(Start, End);
39710822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis  }
39720822c5f29d91a08697416b0d0be28dd3e71d945aArgyrios Kyrtzidis
3973007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
3974007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    Decl *D = cxcursor::getCursorDecl(C);
397516ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (!D)
397616ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      return SourceRange();
397716ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
3978007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    SourceRange R = D->getSourceRange();
3979007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // FIXME: Multiple variables declared in a single declaration
3980007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // currently lack the information needed to correctly determine their
3981007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // ranges when accounting for the type-specifier.  We use context
3982007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3983007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // and if so, whether it is the first decl.
3984007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3985007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      if (!cxcursor::isFirstInDeclGroup(C))
3986007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek        R.setBegin(VD->getLocation());
3987007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    }
3988007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    return R;
3989007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
39906653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return SourceRange();
39916653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
39926653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
39936653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// \brief Retrieves the "raw" cursor extent, which is then extended to include
39946653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// the decl-specifier-seq for declarations.
39956653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr) {
39966653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
39976653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    Decl *D = cxcursor::getCursorDecl(C);
399816ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (!D)
399916ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      return SourceRange();
400016ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
40016653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange R = D->getSourceRange();
40022494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
40032494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // Adjust the start of the location for declarations preceded by
40042494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // declaration specifiers.
40052494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
40066653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
40072494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
400896a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
40092494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
40102494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
401196a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
40122494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    }
40136653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
40142494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && R.getBegin().isValid() &&
40152494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, R.getBegin()))
40162494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      R.setBegin(StartLoc);
40172494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
40182494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // FIXME: Multiple variables declared in a single declaration
40192494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // currently lack the information needed to correctly determine their
40202494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // ranges when accounting for the type-specifier.  We use context
40212494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
40222494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // and if so, whether it is the first decl.
40232494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
40242494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (!cxcursor::isFirstInDeclGroup(C))
40252494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        R.setBegin(VD->getLocation());
40266653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    }
40276653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
40286653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    return R;
40296653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  }
40306653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
40316653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return getRawCursorExtent(C);
40326653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
4033a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
4034a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorextern "C" {
4035a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
4036a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas GregorCXSourceRange clang_getCursorExtent(CXCursor C) {
4037a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  SourceRange R = getRawCursorExtent(C);
4038a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R.isInvalid())
40395352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4040f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4041a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  return cxloc::translateSourceRange(getCursorContext(C), R);
4042a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor}
4043c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
4044c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas GregorCXCursor clang_getCursorReferenced(CXCursor C) {
4045b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
4046b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
4047f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4048a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit tu = getCursorTU(C);
40491f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (clang_isDeclaration(C.kind)) {
40501f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getCursorDecl(C);
405116ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (!D)
405216ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis      return clang_getNullCursor();
40531f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4054a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Using, D->getLocation(), tu);
40555f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    if (ObjCPropertyImplDecl *PropImpl =dyn_cast<ObjCPropertyImplDecl>(D))
4056e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
4057e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return MakeCXCursor(Property, tu);
4058e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
4059c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return C;
40601f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
40611f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
406297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
40631f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Expr *E = getCursorExpr(C);
40641f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getDeclFromExpr(E);
4065aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (D) {
4066aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      CXCursor declCursor = MakeCXCursor(D, tu);
4067aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      declCursor = getSelectorIdentifierCursor(getSelectorIdentifierIndex(C),
4068aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis                                               declCursor);
4069aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return declCursor;
4070aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    }
40711f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (OverloadExpr *Ovl = dyn_cast_or_null<OverloadExpr>(E))
4073a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Ovl, tu);
40741f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
407597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    return clang_getNullCursor();
407697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
407797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
407836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
407936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
408036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (GotoStmt *Goto = dyn_cast_or_null<GotoStmt>(S))
408137c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek      if (LabelDecl *label = Goto->getLabel())
408237c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        if (LabelStmt *labelS = label->getStmt())
408337c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        return MakeCXCursor(labelS, getCursorDecl(C), tu);
408436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
408536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return clang_getNullCursor();
408636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
408736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
40889b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion) {
40899e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth    if (MacroDefinition *Def = getCursorMacroExpansion(C)->getDefinition())
4090a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeMacroDefinitionCursor(Def, tu);
4091bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  }
4092bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
4093c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  if (!clang_isReference(C.kind))
4094c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return clang_getNullCursor();
4095f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4096c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  switch (C.kind) {
4097c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    case CXCursor_ObjCSuperClassRef:
4098a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCSuperClassRef(C).first, tu);
4099f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4100f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
410198c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis      ObjCProtocolDecl *Prot = getCursorObjCProtocolRef(C).first;
410298c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis      if (ObjCProtocolDecl *Def = Prot->getDefinition())
410398c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis        return MakeCXCursor(Def, tu);
410498c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis
4105c15707d8da08df2eb22f6ed047743fa3f7c9831bArgyrios Kyrtzidis      return MakeCXCursor(Prot, tu);
410698c16b8b4fe7bb26b17a479d6872e390816e57d4Argyrios Kyrtzidis    }
4107f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
41087723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor    case CXCursor_ObjCClassRef: {
41097723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
41107723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor      if (ObjCInterfaceDecl *Def = Class->getDefinition())
41117723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor        return MakeCXCursor(Def, tu);
41127723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor
4113c15707d8da08df2eb22f6ed047743fa3f7c9831bArgyrios Kyrtzidis      return MakeCXCursor(Class, tu);
41147723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor    }
41157d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
4116f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef:
4117a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTypeRef(C).first, tu );
41180b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
41190b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
4120a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTemplateRef(C).first, tu );
41210b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
41226931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
4123a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorNamespaceRef(C).first, tu );
41246931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
4125a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
4126a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorMemberRef(C).first, tu );
4127a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
41283064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
41293064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = cxcursor::getCursorCXXBaseSpecifier(C);
41303064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return clang_getTypeDeclaration(cxtype::MakeCXType(B->getType(),
4131a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                                         tu ));
41323064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
4133f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
413436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
413536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // FIXME: We end up faking the "parent" declaration here because we
413636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // don't want to make CXCursor larger.
413736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return MakeCXCursor(getCursorLabelRef(C).first,
4138a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek               static_cast<ASTUnit*>(tu->TUData)->getASTContext()
4139a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          .getTranslationUnitDecl(),
4140a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          tu);
414136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
41421f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
41431f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return C;
4144011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor
4145011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor    case CXCursor_VariableRef:
4146011d8b93b7cfa8492b8a9c909a850d6577e08dcaDouglas Gregor      return MakeCXCursor(getCursorVariableRef(C).first, tu);
41471f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4148c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    default:
4149c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      // We would prefer to enumerate all non-reference cursor kinds here.
4150c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      llvm_unreachable("Unhandled reference cursor kind");
4151c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  }
4152c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor}
4153c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
4154b699866820102a69d83d6ac6941985c5ef4e8c40Douglas GregorCXCursor clang_getCursorDefinition(CXCursor C) {
4155b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
4156b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
4157f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4158a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(C);
4159f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4160b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  bool WasReference = false;
416197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
4162b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    C = clang_getCursorReferenced(C);
4163b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    WasReference = true;
4164b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4165b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
41669b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion)
4167bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    return clang_getCursorReferenced(C);
4168bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
4169b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4170b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4171b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4172b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  Decl *D = getCursorDecl(C);
4173b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!D)
4174b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4175f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4176b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  switch (D->getKind()) {
4177b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't really separate the notions of
4178b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // declaration and definition.
4179b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Namespace:
4180b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Typedef:
4181162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  case Decl::TypeAlias:
41823e4c6c4c79a03f5cb0c4671d7c282d623c6dc35eRichard Smith  case Decl::TypeAliasTemplate:
4183b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTypeParm:
4184b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::EnumConstant:
4185b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Field:
4186d98114647e16796a976b04af79975b4f0eacf22bBenjamin Kramer  case Decl::IndirectField:
4187b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCIvar:
4188b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCAtDefsField:
4189b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ImplicitParam:
4190b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ParmVar:
4191b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NonTypeTemplateParm:
4192b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTemplateParm:
4193b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategoryImpl:
4194b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCImplementation:
41956206d53f67613958ae1b023aba337ebb46f11a8bAbramo Bagnara  case Decl::AccessSpec:
4196b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::LinkageSpec:
4197b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCPropertyImpl:
4198b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FileScopeAsm:
4199b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::StaticAssert:
4200b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Block:
4201ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  case Decl::Label:  // FIXME: Is this right??
4202af0f4d0b2e38c810effc8b024ad2fb6604eec5d3Francois Pichet  case Decl::ClassScopeFunctionSpecialization:
420315de72cf580840c61e5704c2f8a2b56f9d0638e1Douglas Gregor  case Decl::Import:
4204b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return C;
4205b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4206b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't make any sense here, but are
4207b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // nonetheless harmless.
4208b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TranslationUnit:
4209b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
4210b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4211b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds for which the definition is not resolvable.
4212b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingTypename:
4213b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingValue:
4214b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
4215b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4216b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingDirective:
4217b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
4218a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                        TU);
4219b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4220b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NamespaceAlias:
4221a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), TU);
4222b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4223b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Enum:
4224b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Record:
4225b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXRecord:
4226b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplateSpecialization:
4227b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplatePartialSpecialization:
4228952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor    if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
4229a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
4230b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4231b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4232b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Function:
4233b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXMethod:
4234b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConstructor:
4235b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXDestructor:
4236b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConversion: {
4237b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
4238b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionDecl>(D)->getBody(Def))
4239a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(const_cast<FunctionDecl *>(Def), TU);
4240b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4241b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4242b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4243b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Var: {
424431310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    // Ask the variable if it has a definition.
424531310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (VarDecl *Def = cast<VarDecl>(D)->getDefinition())
4246a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
424731310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    return clang_getNullCursor();
4248b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4249f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4250b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FunctionTemplate: {
4251b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
4252b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
4253a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def->getDescribedFunctionTemplate(), TU);
4254b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4255b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4256f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4257b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplate: {
4258b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
4259952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor                                                            ->getDefinition())
42600b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return MakeCXCursor(cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
4261a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          TU);
4262b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4263b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4264b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
42651f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::Using:
42661f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<UsingDecl>(D),
4267a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
4268b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4269b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingShadow:
4270b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getCursorDefinition(
4271f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                       MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
4272a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                    TU));
4273b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4274b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCMethod: {
4275b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
4276b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (Method->isThisDeclarationADefinition())
4277b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
4278b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4279b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // Dig out the method definition in the associated
4280b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // @implementation, if we have it.
4281b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: The ASTs should make finding the definition easier.
4282b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4283b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                       = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
4284b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
4285b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
4286b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                                                  Method->isInstanceMethod()))
4287b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          if (Def->isThisDeclarationADefinition())
4288a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek            return MakeCXCursor(Def, TU);
4289b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4290b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4291b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4292b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4293b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategory:
4294b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCCategoryImplDecl *Impl
4295b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                               = cast<ObjCCategoryDecl>(D)->getImplementation())
4296a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4297b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4298b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4299b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProtocol:
43005e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregor    if (ObjCProtocolDecl *Def = cast<ObjCProtocolDecl>(D)->getDefinition())
43015e2a1ff9f28d2eab256d2553e76a9c9d54693875Douglas Gregor      return MakeCXCursor(Def, TU);
4302b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4303b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4304375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  case Decl::ObjCInterface: {
4305b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // There are two notions of a "definition" for an Objective-C
4306b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // class: the interface and its implementation. When we resolved a
4307b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // reference to an Objective-C class, produce the @interface as
4308b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // the definition; when we were provided with the interface,
4309b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // produce the @implementation as the definition.
4310375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    ObjCInterfaceDecl *IFace = cast<ObjCInterfaceDecl>(D);
4311b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (WasReference) {
4312375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor      if (ObjCInterfaceDecl *Def = IFace->getDefinition())
43137723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor        return MakeCXCursor(Def, TU);
4314375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor    } else if (ObjCImplementationDecl *Impl = IFace->getImplementation())
4315a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4316b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4317375bb1413c041055262c8a416f20d10474a5eda9Douglas Gregor  }
4318f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4319b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProperty:
4320b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: We don't really know where to find the
4321b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // ObjCPropertyImplDecls that implement this property.
4322b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4323b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4324b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCompatibleAlias:
4325b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4326b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
43277723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor      if (ObjCInterfaceDecl *Def = Class->getDefinition())
43287723fec9b45b7258c0eddf4cbfd0d335348f5edcDouglas Gregor        return MakeCXCursor(Def, TU);
4329f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4330b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4331b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4332b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Friend:
4333b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
4334a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4335b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4336b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4337b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FriendTemplate:
4338b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
4339a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4340b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4341b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4342b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4343b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getNullCursor();
4344b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4345b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4346b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregorunsigned clang_isCursorDefinition(CXCursor C) {
4347b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4348b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return 0;
4349b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4350b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getCursorDefinition(C) == C;
4351b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4352b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
43531a9d0503b67a499797141af0fd6d315d5045f0eaDouglas GregorCXCursor clang_getCanonicalCursor(CXCursor C) {
43541a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  if (!clang_isDeclaration(C.kind))
43551a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return C;
43561a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
4357e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis  if (Decl *D = getCursorDecl(C)) {
4358debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis    if (ObjCCategoryImplDecl *CatImplD = dyn_cast<ObjCCategoryImplDecl>(D))
4359debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis      if (ObjCCategoryDecl *CatD = CatImplD->getCategoryDecl())
4360debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis        return MakeCXCursor(CatD, getCursorTU(C));
4361debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis
4362e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis    if (ObjCImplDecl *ImplD = dyn_cast<ObjCImplDecl>(D))
4363e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis      if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
4364e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis        return MakeCXCursor(IFD, getCursorTU(C));
4365e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis
43661a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return MakeCXCursor(D->getCanonicalDecl(), getCursorTU(C));
4367e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis  }
43681a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
43691a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  return C;
43701a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor}
43711a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
43721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregorunsigned clang_getNumOverloadedDecls(CXCursor C) {
43737c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (C.kind != CXCursor_OverloadedDeclRef)
43741f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return 0;
43751f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43761f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
43771f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
43781f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return E->getNumDecls();
43791f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43801f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
43811f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
43821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return S->size();
43831f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43841f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
43851f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4386826faa22bae112e01293a58534a40711043cce65Argyrios Kyrtzidis    return Using->shadow_size();
43871f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43881f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return 0;
43891f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
43901f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43911f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas GregorCXCursor clang_getOverloadedDecl(CXCursor cursor, unsigned index) {
43927c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (cursor.kind != CXCursor_OverloadedDeclRef)
43931f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
43941f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43951f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (index >= clang_getNumOverloadedDecls(cursor))
43961f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
43971f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4398a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(cursor);
43991f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(cursor).first;
44001f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
4401a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(E->decls_begin()[index], TU);
44021f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
44031f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
44041f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
4405a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(S->begin()[index], TU);
44061f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
44071f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
44081f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D)) {
44091f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // FIXME: This is, unfortunately, linear time.
44101f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    UsingDecl::shadow_iterator Pos = Using->shadow_begin();
44111f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    std::advance(Pos, index);
4412a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<UsingShadowDecl>(*Pos)->getTargetDecl(), TU);
44131f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
44141f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
44151f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return clang_getNullCursor();
44161f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
44171f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
44180d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbarvoid clang_getDefinitionSpellingAndExtent(CXCursor C,
44194ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **startBuf,
44204ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **endBuf,
44214ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startLine,
44224ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startColumn,
44234ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *endLine,
44249ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          unsigned *endColumn) {
4425283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  assert(getCursorDecl(C) && "CXCursor has null decl");
4426283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  NamedDecl *ND = static_cast<NamedDecl *>(getCursorDecl(C));
44274ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  FunctionDecl *FD = dyn_cast<FunctionDecl>(ND);
44284ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
4429f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
44304ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  SourceManager &SM = FD->getASTContext().getSourceManager();
44314ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startBuf = SM.getCharacterData(Body->getLBracLoc());
44324ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endBuf = SM.getCharacterData(Body->getRBracLoc());
44334ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
44344ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
44354ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
44364ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
44374ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff}
4438f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4439430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4440430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas GregorCXSourceRange clang_getCursorReferenceNameRange(CXCursor C, unsigned NameFlags,
4441430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                                                unsigned PieceIndex) {
4442430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  RefNamePieces Pieces;
4443430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4444430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  switch (C.kind) {
4445430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_MemberRefExpr:
4446430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (MemberExpr *E = dyn_cast<MemberExpr>(getCursorExpr(C)))
4447430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Pieces = buildPieces(NameFlags, true, E->getMemberNameInfo(),
4448430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getQualifierLoc().getSourceRange());
4449430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4450430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4451430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_DeclRefExpr:
4452430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (DeclRefExpr *E = dyn_cast<DeclRefExpr>(getCursorExpr(C)))
4453430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Pieces = buildPieces(NameFlags, false, E->getNameInfo(),
4454430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getQualifierLoc().getSourceRange(),
4455e4b92761b43ced611c417ae478568610f1ad7b1eAbramo Bagnara                           E->getOptionalExplicitTemplateArgs());
4456430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4457430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4458430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_CallExpr:
4459430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (CXXOperatorCallExpr *OCE =
4460430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        dyn_cast<CXXOperatorCallExpr>(getCursorExpr(C))) {
4461430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Expr *Callee = OCE->getCallee();
4462430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Callee))
4463430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        Callee = ICE->getSubExpr();
4464430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4465430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Callee))
4466430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        Pieces = buildPieces(NameFlags, false, DRE->getNameInfo(),
4467430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                             DRE->getQualifierLoc().getSourceRange());
4468430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    }
4469430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4470430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4471430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  default:
4472430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4473430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  }
4474430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4475430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  if (Pieces.empty()) {
4476430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (PieceIndex == 0)
4477430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      return clang_getCursorExtent(C);
4478430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  } else if (PieceIndex < Pieces.size()) {
4479430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      SourceRange R = Pieces[PieceIndex];
4480430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (R.isValid())
4481430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        return cxloc::translateSourceRange(getCursorContext(C), R);
4482430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  }
4483430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4484430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  return clang_getNullRange();
4485430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor}
4486430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
44870a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregorvoid clang_enableStackTraces(void) {
44880a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  llvm::sys::PrintStackTraceOnErrorSignal();
44890a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor}
44900a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor
4491995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbarvoid clang_executeOnThread(void (*fn)(void*), void *user_data,
4492995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar                           unsigned stack_size) {
4493995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar  llvm::llvm_execute_on_thread(fn, user_data, stack_size);
4494995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar}
4495995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar
4496fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
4497fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
4498fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
4499fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor// Token-based Operations.
4500fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
4501fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4502fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor/* CXToken layout:
4503fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[0]: a CXTokenKind
4504fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[1]: starting token location
4505fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[2]: token length
4506fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[3]: reserved
4507f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek *   ptr_data: for identifiers and keywords, an IdentifierInfo*.
4508fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   otherwise unused.
4509fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor */
4510fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorextern "C" {
4511fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4512fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXTokenKind clang_getTokenKind(CXToken CXTok) {
4513fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return static_cast<CXTokenKind>(CXTok.int_data[0]);
4514fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4515fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4516fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
4517fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  switch (clang_getTokenKind(CXTok)) {
4518fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Identifier:
4519fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Keyword:
4520fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We know we have an IdentifierInfo*, so use that.
4521ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(static_cast<IdentifierInfo *>(CXTok.ptr_data)
4522ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                            ->getNameStart());
4523fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4524fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Literal: {
4525fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We have stashed the starting pointer in the ptr_data field. Use it.
4526fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    const char *Text = static_cast<const char *>(CXTok.ptr_data);
45275f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    return createCXString(StringRef(Text, CXTok.int_data[2]));
4528fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4529f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4530fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Punctuation:
4531fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Comment:
4532fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    break;
4533fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4534f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4535f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // We have to find the starting buffer pointer the hard way, by
4536fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // deconstructing the source location.
4537a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4538fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4539ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
4540f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4541fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
4542fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> LocInfo
4543a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    = CXXUnit->getSourceManager().getDecomposedSpellingLoc(Loc);
4544f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
45455f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  StringRef Buffer
4546f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
4547f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  if (Invalid)
4548aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor    return createCXString("");
4549fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4550f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  return createCXString(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
4551fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4552f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4553fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
4554a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4555fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4556fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return clang_getNullLocation();
4557f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4558fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
4559fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4560fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4561fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4562fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
4563a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
45645352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (!CXXUnit)
45655352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4566f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4567f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return cxloc::translateSourceRange(CXXUnit->getASTContext(),
4568fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4569fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4570f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4571ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisstatic void getTokens(ASTUnit *CXXUnit, SourceRange Range,
4572ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                      SmallVectorImpl<CXToken> &CXTokens) {
4573fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
4574fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
4575ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(Range.getBegin());
4576fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
4577ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(Range.getEnd());
4578f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4579fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Cannot tokenize across files.
4580fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (BeginLocInfo.first != EndLocInfo.first)
4581fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4582f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4583f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Create a lexer
4584f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
45855f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  StringRef Buffer
4586f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
458747a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor  if (Invalid)
458847a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor    return;
4589aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor
4590fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
45914e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie            CXXUnit->getASTContext().getLangOpts(),
4592f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer            Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
4593fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lex.SetCommentRetentionState(true);
4594f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4595fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Lex tokens until we hit the end of the range.
4596f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
4597fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Token Tok;
4598096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall  bool previousWasAt = false;
4599fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  do {
4600fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Lex the next token
4601fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    Lex.LexFromRawLexer(Tok);
4602fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.is(tok::eof))
4603fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      break;
4604f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4605fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Initialize the CXToken.
4606fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXToken CXTok;
4607f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4608fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Common fields
4609fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
4610fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[2] = Tok.getLength();
4611fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[3] = 0;
4612f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4613fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Kind-specific fields
4614fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.isLiteral()) {
4615fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Literal;
4616fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = (void *)Tok.getLiteralData();
4617c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara    } else if (Tok.is(tok::raw_identifier)) {
4618aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor      // Lookup the identifier to determine whether we have a keyword.
4619fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      IdentifierInfo *II
4620c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok);
4621aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek
4622096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall      if ((II->getObjCKeywordID() != tok::objc_not_keyword) && previousWasAt) {
4623aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek        CXTok.int_data[0] = CXToken_Keyword;
4624aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4625aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      else {
4626c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        CXTok.int_data[0] = Tok.is(tok::identifier)
4627c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          ? CXToken_Identifier
4628c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          : CXToken_Keyword;
4629aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4630fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = II;
4631fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else if (Tok.is(tok::comment)) {
4632fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Comment;
4633fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4634fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else {
4635fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Punctuation;
4636fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4637fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    }
4638fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTokens.push_back(CXTok);
4639096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall    previousWasAt = Tok.is(tok::at);
4640fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  } while (Lex.getBufferLocation() <= EffectiveBufferEnd);
4641ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis}
4642ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4643ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisvoid clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
4644ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                    CXToken **Tokens, unsigned *NumTokens) {
4645ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (Tokens)
4646ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    *Tokens = 0;
4647ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (NumTokens)
4648ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    *NumTokens = 0;
4649ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4650ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4651ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (!CXXUnit || !Tokens || !NumTokens)
4652ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
4653ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4654ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
4655ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4656ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SourceRange R = cxloc::translateCXSourceRange(Range);
4657ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (R.isInvalid())
4658ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
4659ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4660ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SmallVector<CXToken, 32> CXTokens;
4661ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  getTokens(CXXUnit, R, CXTokens);
4662f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4663fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (CXTokens.empty())
4664fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4665f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4666fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
4667fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
4668fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *NumTokens = CXTokens.size();
4669fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
46700045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
46716db610934bedc6896393c1e1099525b35380acd6Ted Kremenekvoid clang_disposeTokens(CXTranslationUnit TU,
46726db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                         CXToken *Tokens, unsigned NumTokens) {
46736db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  free(Tokens);
46746db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
46756db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46766db610934bedc6896393c1e1099525b35380acd6Ted Kremenek} // end: extern "C"
46776db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46786db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
46796db610934bedc6896393c1e1099525b35380acd6Ted Kremenek// Token annotation APIs.
46806db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
46816db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46820045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregortypedef llvm::DenseMap<unsigned, CXCursor> AnnotateTokensData;
4683fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
4684fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXCursor parent,
4685fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXClientData client_data);
46866db610934bedc6896393c1e1099525b35380acd6Ted Kremeneknamespace {
46876db610934bedc6896393c1e1099525b35380acd6Ted Kremenekclass AnnotateTokensWorker {
46886db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  AnnotateTokensData &Annotated;
468911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXToken *Tokens;
469011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXCursor *Cursors;
469111949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  unsigned NumTokens;
4692fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned TokIdx;
46934419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  unsigned PreprocessingTokIdx;
4694fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CursorVisitor AnnotateVis;
4695fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceManager &SrcMgr;
4696f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool HasContextSensitiveKeywords;
4697f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4698fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  bool MoreTokens() const { return TokIdx < NumTokens; }
4699fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned NextToken() const { return TokIdx; }
4700fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AdvanceToken() { ++TokIdx; }
4701fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation GetTokenLoc(unsigned tokI) {
4702fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
4703fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
47045f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  bool isFunctionMacroToken(unsigned tokI) const {
4705a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return Tokens[tokI].int_data[3] != 0;
4706a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
47075f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  SourceLocation getFunctionMacroTokenLoc(unsigned tokI) const {
4708a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[3]);
4709a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4710a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4711a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  void annotateAndAdvanceTokens(CXCursor, RangeComparisonResult, SourceRange);
47125f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  void annotateAndAdvanceFunctionMacroTokens(CXCursor, RangeComparisonResult,
47135f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis                                             SourceRange);
4714fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
47156db610934bedc6896393c1e1099525b35380acd6Ted Kremenekpublic:
471611949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  AnnotateTokensWorker(AnnotateTokensData &annotated,
4717fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                       CXToken *tokens, CXCursor *cursors, unsigned numTokens,
4718a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                       CXTranslationUnit tu, SourceRange RegionOfInterest)
471911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    : Annotated(annotated), Tokens(tokens), Cursors(cursors),
47204419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      NumTokens(numTokens), TokIdx(0), PreprocessingTokIdx(0),
4721a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      AnnotateVis(tu,
4722f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                  AnnotateTokensVisitor, this,
4723f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                  /*VisitPreprocessorLast=*/true,
4724e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                  /*VisitIncludedEntities=*/false,
4725f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                  RegionOfInterest),
4726f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      SrcMgr(static_cast<ASTUnit*>(tu->TUData)->getSourceManager()),
4727f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      HasContextSensitiveKeywords(false) { }
472811949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4729fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
47306db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
473103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  void AnnotateTokens();
4732f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4733f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// \brief Determine whether the annotator saw any cursors that have
4734f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// context-sensitive keywords.
4735f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool hasContextSensitiveKeywords() const {
4736f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    return HasContextSensitiveKeywords;
4737f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
47386db610934bedc6896393c1e1099525b35380acd6Ted Kremenek};
47396db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
47400045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
474103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidisvoid AnnotateTokensWorker::AnnotateTokens() {
4742fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Walk the AST within the region of interest, annotating tokens
4743fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // along the way.
474403ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis  AnnotateVis.visitFileRegion();
4745fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4746fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = 0 ; I < TokIdx ; ++I) {
474711949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
47484419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    if (Pos != Annotated.end() &&
47494419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        (clang_isInvalid(Cursors[I].kind) ||
47504419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor         Pos->second.kind != CXCursor_PreprocessingDirective))
4751fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      Cursors[I] = Pos->second;
4752fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4753fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4754fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Finish up annotating any tokens left.
4755fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (!MoreTokens())
4756fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return;
475711949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4758fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor &C = clang_getNullCursor();
4759fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = TokIdx ; I < NumTokens ; ++I) {
476003ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis    if (I < PreprocessingTokIdx && clang_isPreprocessing(Cursors[I].kind))
476103ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis      continue;
476203ee2dd9fc5d5fc62b5eb0fb88ee56e553f8cda7Argyrios Kyrtzidis
4763fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
4764fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = (Pos == Annotated.end()) ? C : Pos->second;
476511949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  }
476611949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek}
476711949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4768a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief It annotates and advances tokens with a cursor until the comparison
4769a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis//// between the cursor location and the source range is the same as
4770a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \arg compResult.
4771a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis///
4772a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// Pass RangeBefore to annotate tokens with a cursor until a range is reached.
4773a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// Pass RangeOverlap to annotate tokens inside a range.
4774a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisvoid AnnotateTokensWorker::annotateAndAdvanceTokens(CXCursor updateC,
4775a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               RangeComparisonResult compResult,
4776a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               SourceRange range) {
4777a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  while (MoreTokens()) {
4778a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    const unsigned I = NextToken();
47795f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis    if (isFunctionMacroToken(I))
47805f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis      return annotateAndAdvanceFunctionMacroTokens(updateC, compResult, range);
4781a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4782a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    SourceLocation TokLoc = GetTokenLoc(I);
4783a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
4784a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      Cursors[I] = updateC;
4785a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      AdvanceToken();
4786a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      continue;
4787a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
4788a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    break;
4789a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4790a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
4791a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4792a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief Special annotation handling for macro argument tokens.
47935f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidisvoid AnnotateTokensWorker::annotateAndAdvanceFunctionMacroTokens(
47945f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis                                               CXCursor updateC,
4795a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               RangeComparisonResult compResult,
4796a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               SourceRange range) {
47975f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  assert(MoreTokens());
47985f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  assert(isFunctionMacroToken(NextToken()) &&
4799a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis         "Should be called only for macro arg tokens");
4800a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4801a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // This works differently than annotateAndAdvanceTokens; because expanded
4802a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // macro arguments can have arbitrary translation-unit source order, we do not
4803a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // advance the token index one by one until a token fails the range test.
4804a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // We only advance once past all of the macro arg tokens if all of them
4805a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // pass the range test. If one of them fails we keep the token index pointing
4806a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // at the start of the macro arg tokens so that the failing token will be
4807a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // annotated by a subsequent annotation try.
4808a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4809a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  bool atLeastOneCompFail = false;
4810a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4811a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned I = NextToken();
48125f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  for (; I < NumTokens && isFunctionMacroToken(I); ++I) {
48135f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis    SourceLocation TokLoc = getFunctionMacroTokenLoc(I);
4814a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (TokLoc.isFileID())
4815a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      continue; // not macro arg token, it's parens or comma.
4816a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
4817a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (clang_isInvalid(clang_getCursorKind(Cursors[I])))
4818a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        Cursors[I] = updateC;
4819a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    } else
4820a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      atLeastOneCompFail = true;
4821a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4822a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4823a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  if (!atLeastOneCompFail)
4824a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    TokIdx = I; // All of the tokens were handled, advance beyond all of them.
4825a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
4826a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
48276db610934bedc6896393c1e1099525b35380acd6Ted Kremenekenum CXChildVisitResult
48284419b675577d7c281a659fab1fec10e1bfbe04c5Douglas GregorAnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
4829fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CXSourceLocation Loc = clang_getCursorLocation(cursor);
48304419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  SourceRange cursorRange = getRawCursorExtent(cursor);
483181d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor  if (cursorRange.isInvalid())
483281d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor    return CXChildVisit_Recurse;
4833f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4834f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (!HasContextSensitiveKeywords) {
4835f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C properties can have context-sensitive keywords.
4836f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    if (cursor.kind == CXCursor_ObjCPropertyDecl) {
4837f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCPropertyDecl *Property
4838f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(cursor)))
4839f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        HasContextSensitiveKeywords = Property->getPropertyAttributesAsWritten() != 0;
4840f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4841f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C methods can have context-sensitive keywords.
4842f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_ObjCInstanceMethodDecl ||
4843f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ObjCClassMethodDecl) {
4844f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCMethodDecl *Method
4845f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
4846f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->getObjCDeclQualifier())
4847f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4848f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        else {
4849f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
4850f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                                           PEnd = Method->param_end();
4851f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor               P != PEnd; ++P) {
4852f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((*P)->getObjCDeclQualifier()) {
4853f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              HasContextSensitiveKeywords = true;
4854f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              break;
4855f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            }
4856f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          }
4857f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4858f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4859f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4860f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ methods can have context-sensitive keywords.
4861f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_CXXMethod) {
4862f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (CXXMethodDecl *Method
4863f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(cursor))) {
4864f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->hasAttr<FinalAttr>() || Method->hasAttr<OverrideAttr>())
4865f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4866f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4867f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4868f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ classes can have context-sensitive keywords.
4869f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_StructDecl ||
4870f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassDecl ||
4871f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplate ||
4872f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplatePartialSpecialization) {
4873f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Decl *D = getCursorDecl(cursor))
4874f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (D->hasAttr<FinalAttr>())
4875f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4876f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4877f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
4878f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
48794419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  if (clang_isPreprocessing(cursor.kind)) {
4880cea731a9cb7de3f473d60e5ea544e25621cebd76Chandler Carruth    // For macro expansions, just note where the beginning of the macro
4881cea731a9cb7de3f473d60e5ea544e25621cebd76Chandler Carruth    // expansion occurs.
48829b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth    if (cursor.kind == CXCursor_MacroExpansion) {
48834419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      Annotated[Loc.int_data] = cursor;
48844419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      return CXChildVisit_Recurse;
48854419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
48864419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
48874419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Items in the preprocessing record are kept separate from items in
48884419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // declarations, so we keep a separate token index.
48894419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    unsigned SavedTokIdx = TokIdx;
48904419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = PreprocessingTokIdx;
48914419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
48924419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Skip tokens up until we catch up to the beginning of the preprocessing
48934419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // entry.
48944419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
48954419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
48964419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
48974419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
48984419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
48994419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
49004419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
49014419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
49024419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
49034419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
49044419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
49054419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
49064419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
49074419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
49084419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Look at all of the tokens within this range.
49094419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
49104419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
49114419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
49124419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
49134419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
4914b219cfc4d75f0a03630b7c4509ef791b7e97b2c8David Blaikie        llvm_unreachable("Infeasible");
49154419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
49164419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
49174419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
49184419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        Cursors[I] = cursor;
49194419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
49204419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
49214419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
49224419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
49234419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
49244419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
49254419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Save the preprocessing token index; restore the non-preprocessing
49264419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // token index.
49274419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    PreprocessingTokIdx = TokIdx;
49284419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = SavedTokIdx;
49290045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor    return CXChildVisit_Recurse;
49300045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  }
4931fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4932fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (cursorRange.isInvalid())
4933fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return CXChildVisit_Continue;
4934a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek
4935fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation L = SourceLocation::getFromRawEncoding(Loc.int_data);
4936fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4937a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  // Adjust the annotated range based specific declarations.
4938a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  const enum CXCursorKind cursorK = clang_getCursorKind(cursor);
4939a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  if (cursorK >= CXCursor_FirstDecl && cursorK <= CXCursor_LastDecl) {
494023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    Decl *D = cxcursor::getCursorDecl(cursor);
49412494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
49422494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
494316ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    if (const DeclaratorDecl *DD = dyn_cast_or_null<DeclaratorDecl>(D)) {
49442494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
494596a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
494616ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    } else if (TypedefDecl *Typedef = dyn_cast_or_null<TypedefDecl>(D)) {
49472494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
494896a0014f9b963d8a987f1cccd48808a47f9c6331Daniel Dunbar        StartLoc = TI->getTypeLoc().getLocStart();
4949a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek    }
49502494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
49512494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && L.isValid() &&
49522494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, L))
49532494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      cursorRange.setBegin(StartLoc);
4954a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  }
495581d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor
49563f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // If the location of the cursor occurs within a macro instantiation, record
49573f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // the spelling location of the cursor in our annotation map.  We can then
49583f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // paper over the token labelings during a post-processing step to try and
49593f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // get cursor mappings for tokens that are the *arguments* of a macro
49603f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // instantiation.
49613f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  if (L.isMacroID()) {
49623f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    unsigned rawEncoding = SrcMgr.getSpellingLoc(L).getRawEncoding();
49633f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // Only invalidate the old annotation if it isn't part of a preprocessing
49643f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // directive.  Here we assume that the default construction of CXCursor
49653f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // results in CXCursor.kind being an initialized value (i.e., 0).  If
49663f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // this isn't the case, we can fix by doing lookup + insertion.
49674419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
49683f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    CXCursor &oldC = Annotated[rawEncoding];
49693f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    if (!clang_isPreprocessing(oldC.kind))
49703f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek      oldC = cursor;
49713f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  }
49723f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek
4973fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const enum CXCursorKind K = clang_getCursorKind(parent);
4974fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor updateC =
4975d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek    (clang_isInvalid(K) || K == CXCursor_TranslationUnit)
4976d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek     ? clang_getNullCursor() : parent;
4977fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4978a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  annotateAndAdvanceTokens(updateC, RangeBefore, cursorRange);
4979fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
49805517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // Avoid having the cursor of an expression "overwrite" the annotation of the
49815517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // variable declaration that it belongs to.
49825517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // This can happen for C++ constructor expressions whose range generally
49835517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // include the variable declaration, e.g.:
49845517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  //  MyCXXClass foo; // Make sure we don't annotate 'foo' as a CallExpr cursor.
49855517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  if (clang_isExpression(cursorK)) {
49865517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis    Expr *E = getCursorExpr(cursor);
49878ccac3de1335f1cfd7cea56ba1cefcf0b724ce3fArgyrios Kyrtzidis    if (Decl *D = getCursorParentDecl(cursor)) {
49885517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      const unsigned I = NextToken();
49895517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      if (E->getLocStart().isValid() && D->getLocation().isValid() &&
49905517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis          E->getLocStart() == D->getLocation() &&
49915517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis          E->getLocStart() == GetTokenLoc(I)) {
49925517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis        Cursors[I] = updateC;
49935517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis        AdvanceToken();
49945517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      }
49955517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis    }
49965517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  }
49975517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis
4998fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Visit children to get their cursor information.
4999fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned BeforeChildren = NextToken();
5000fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(cursor);
5001fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned AfterChildren = NextToken();
5002fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5003a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // Scan the tokens that are at the end of the cursor, but are not captured
5004a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // but the child cursors.
5005a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  annotateAndAdvanceTokens(cursor, RangeOverlap, cursorRange);
50066db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
5007fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the beginning of the cursor, but are not
5008fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // capture by the child cursors.
5009fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
5010fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
5011fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      break;
50124419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
5013fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
5014fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
5015fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5016fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return CXChildVisit_Continue;
50170045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor}
50180045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
50196db610934bedc6896393c1e1099525b35380acd6Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
50206db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXCursor parent,
50216db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXClientData client_data) {
50226db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
50236db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
50246db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
50256628a614c504263ae539462f049d523dd07ac1baTed Kremeneknamespace {
5026a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5027a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief Uses the macro expansions in the preprocessing record to find
5028a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// and mark tokens that are macro arguments. This info is used by the
5029a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// AnnotateTokensWorker.
5030a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisclass MarkMacroArgTokensVisitor {
5031a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  SourceManager &SM;
5032a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  CXToken *Tokens;
5033a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned NumTokens;
5034a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned CurIdx;
5035a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5036a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidispublic:
5037a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  MarkMacroArgTokensVisitor(SourceManager &SM,
5038a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                            CXToken *tokens, unsigned numTokens)
5039a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    : SM(SM), Tokens(tokens), NumTokens(numTokens), CurIdx(0) { }
5040a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5041a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  CXChildVisitResult visit(CXCursor cursor, CXCursor parent) {
5042a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (cursor.kind != CXCursor_MacroExpansion)
5043a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Continue;
5044a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5045a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    SourceRange macroRange = getCursorMacroExpansion(cursor)->getSourceRange();
5046a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (macroRange.getBegin() == macroRange.getEnd())
5047a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Continue; // it's not a function macro.
5048a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5049a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    for (; CurIdx < NumTokens; ++CurIdx) {
5050a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (!SM.isBeforeInTranslationUnit(getTokenLoc(CurIdx),
5051a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                        macroRange.getBegin()))
5052a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        break;
5053a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
5054a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5055a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (CurIdx == NumTokens)
5056a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Break;
5057a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5058a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    for (; CurIdx < NumTokens; ++CurIdx) {
5059a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      SourceLocation tokLoc = getTokenLoc(CurIdx);
5060a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (!SM.isBeforeInTranslationUnit(tokLoc, macroRange.getEnd()))
5061a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        break;
5062a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
50635f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis      setFunctionMacroTokenLoc(CurIdx, SM.getMacroArgExpandedLocation(tokLoc));
5064a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
5065a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5066a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (CurIdx == NumTokens)
5067a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Break;
5068a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5069a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return CXChildVisit_Continue;
5070a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5071a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5072a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisprivate:
5073a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  SourceLocation getTokenLoc(unsigned tokI) {
5074a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
5075a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5076a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
50775f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  void setFunctionMacroTokenLoc(unsigned tokI, SourceLocation loc) {
5078a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // The third field is reserved and currently not used. Use it here
5079a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // to mark macro arg expanded tokens with their expanded locations.
5080a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    Tokens[tokI].int_data[3] = loc.getRawEncoding();
5081a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5082a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis};
5083a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5084a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis} // end anonymous namespace
5085a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5086a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisstatic CXChildVisitResult
5087a676379b26edc959193f9f919ba9c6d296a57824Argyrios KyrtzidisMarkMacroArgTokensVisitorDelegate(CXCursor cursor, CXCursor parent,
5088a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                  CXClientData client_data) {
5089a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  return static_cast<MarkMacroArgTokensVisitor*>(client_data)->visit(cursor,
5090a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                                                     parent);
5091a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
5092a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5093a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisnamespace {
50946628a614c504263ae539462f049d523dd07ac1baTed Kremenek  struct clang_annotateTokens_Data {
50956628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXTranslationUnit TU;
50966628a614c504263ae539462f049d523dd07ac1baTed Kremenek    ASTUnit *CXXUnit;
50976628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXToken *Tokens;
50986628a614c504263ae539462f049d523dd07ac1baTed Kremenek    unsigned NumTokens;
50996628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXCursor *Cursors;
51006628a614c504263ae539462f049d523dd07ac1baTed Kremenek  };
5101ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek}
5102ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek
5103ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisstatic void annotatePreprocessorTokens(CXTranslationUnit TU,
5104ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                       SourceRange RegionOfInterest,
5105ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                       AnnotateTokensData &Annotated) {
5106ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
5107ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5108ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SourceManager &SourceMgr = CXXUnit->getSourceManager();
5109ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  std::pair<FileID, unsigned> BeginLocInfo
5110ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(RegionOfInterest.getBegin());
5111ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  std::pair<FileID, unsigned> EndLocInfo
5112ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(RegionOfInterest.getEnd());
5113ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5114ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (BeginLocInfo.first != EndLocInfo.first)
5115ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
5116ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5117ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  StringRef Buffer;
5118ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  bool Invalid = false;
5119ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
5120ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (Buffer.empty() || Invalid)
5121ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
5122ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5123ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
51244e4d08403ca5cfd4d558fa2936215d3a4e5a528dDavid Blaikie            CXXUnit->getASTContext().getLangOpts(),
5125ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            Buffer.begin(), Buffer.data() + BeginLocInfo.second,
5126ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            Buffer.end());
5127ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Lex.SetCommentRetentionState(true);
5128ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5129ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  // Lex tokens in raw mode until we hit the end of the range, to avoid
5130ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  // entering #includes or expanding macros.
5131ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  while (true) {
5132ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    Token Tok;
5133ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    Lex.LexFromRawLexer(Tok);
5134ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5135ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  reprocess:
5136ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
5137ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // We have found a preprocessing directive. Gobble it up so that we
5138ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // don't see it while preprocessing these tokens later, but keep track
5139ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // of all of the token locations inside this preprocessing directive so
5140ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // that we can annotate them appropriately.
5141ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      //
5142ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // FIXME: Some simple tests here could identify macro definitions and
5143ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // #undefs, to provide specific cursor kinds for those.
5144ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      SmallVector<SourceLocation, 32> Locations;
5145ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      do {
5146ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Locations.push_back(Tok.getLocation());
5147ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Lex.LexFromRawLexer(Tok);
5148ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      } while (!Tok.isAtStartOfLine() && !Tok.is(tok::eof));
5149ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5150ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      using namespace cxcursor;
5151ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      CXCursor Cursor
5152ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      = MakePreprocessingDirectiveCursor(SourceRange(Locations.front(),
5153ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                                     Locations.back()),
5154ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                         TU);
5155ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      for (unsigned I = 0, N = Locations.size(); I != N; ++I) {
5156ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Annotated[Locations[I].getRawEncoding()] = Cursor;
5157ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      }
5158ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5159ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      if (Tok.isAtStartOfLine())
5160ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        goto reprocess;
5161ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5162ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      continue;
5163ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    }
5164ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5165ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    if (Tok.is(tok::eof))
5166ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      break;
5167ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
5168ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis}
5169ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
51706628a614c504263ae539462f049d523dd07ac1baTed Kremenek// This gets run a separate thread to avoid stack blowout.
51716628a614c504263ae539462f049d523dd07ac1baTed Kremenekstatic void clang_annotateTokensImpl(void *UserData) {
51726628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXTranslationUnit TU = ((clang_annotateTokens_Data*)UserData)->TU;
51736628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = ((clang_annotateTokens_Data*)UserData)->CXXUnit;
51746628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXToken *Tokens = ((clang_annotateTokens_Data*)UserData)->Tokens;
51756628a614c504263ae539462f049d523dd07ac1baTed Kremenek  const unsigned NumTokens = ((clang_annotateTokens_Data*)UserData)->NumTokens;
51766628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor *Cursors = ((clang_annotateTokens_Data*)UserData)->Cursors;
5177fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5178fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  CIndexer *CXXIdx = (CIndexer*)TU->CIdx;
5179fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  if (CXXIdx->isOptEnabled(CXGlobalOpt_ThreadBackgroundPriorityForEditing))
518081b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidis    setThreadBackgroundPriority();
5181fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
51820396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Determine the region of interest, which contains all of the tokens.
51830045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  SourceRange RegionOfInterest;
51846628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setBegin(
51856628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU, Tokens[0])));
51866628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setEnd(
51876628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU,
51886628a614c504263ae539462f049d523dd07ac1baTed Kremenek                                                         Tokens[NumTokens-1])));
5189fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
51900396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // A mapping from the source locations found when re-lexing or traversing the
51910396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // region of interest to the corresponding cursors.
51920045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  AnnotateTokensData Annotated;
5193ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5194fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Relex the tokens within the source range to look for preprocessing
51950396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // directives.
5196ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  annotatePreprocessorTokens(TU, RegionOfInterest, Annotated);
51976628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5198a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
5199a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // Search and mark tokens that are macro argument expansions.
5200a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    MarkMacroArgTokensVisitor Visitor(CXXUnit->getSourceManager(),
5201a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                      Tokens, NumTokens);
5202a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    CursorVisitor MacroArgMarker(TU,
5203a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                 MarkMacroArgTokensVisitorDelegate, &Visitor,
5204f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                 /*VisitPreprocessorLast=*/true,
5205e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                                 /*VisitIncludedEntities=*/false,
5206f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                 RegionOfInterest);
5207a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    MacroArgMarker.visitPreprocessedEntitiesInRegion();
5208a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5209a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
52100396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Annotate all of the source locations in the region of interest that map to
5211fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // a specific cursor.
5212fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  AnnotateTokensWorker W(Annotated, Tokens, Cursors, NumTokens,
5213a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                         TU, RegionOfInterest);
52146628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52156c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // FIXME: We use a ridiculous stack size here because the data-recursion
52166c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm uses a large stack frame than the non-data recursive version,
52176c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // and AnnotationTokensWorker currently transforms the data-recursion
52186c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm back into a traditional recursion by explicitly calling
52196c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // VisitChildren().  We will need to remove this explicit recursive call.
52206628a614c504263ae539462f049d523dd07ac1baTed Kremenek  W.AnnotateTokens();
52216628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5222f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // If we ran into any entities that involve context-sensitive keywords,
5223f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // take another pass through the tokens to mark them as such.
5224f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (W.hasContextSensitiveKeywords()) {
5225f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    for (unsigned I = 0; I != NumTokens; ++I) {
5226f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (clang_getTokenKind(Tokens[I]) != CXToken_Identifier)
5227f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5228f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
5229f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCPropertyDecl) {
5230f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
5231f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (ObjCPropertyDecl *Property
52326628a614c504263ae539462f049d523dd07ac1baTed Kremenek            = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(Cursors[I]))) {
5233f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if (Property->getPropertyAttributesAsWritten() != 0 &&
5234f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              llvm::StringSwitch<bool>(II->getName())
52356628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readonly", true)
52366628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("assign", true)
5237f85e193739c953358c865005855253af4f68a497John McCall              .Case("unsafe_unretained", true)
52386628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readwrite", true)
52396628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("retain", true)
52406628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("copy", true)
52416628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("nonatomic", true)
52426628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("atomic", true)
52436628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("getter", true)
52446628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("setter", true)
5245f85e193739c953358c865005855253af4f68a497John McCall              .Case("strong", true)
5246f85e193739c953358c865005855253af4f68a497John McCall              .Case("weak", true)
52476628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Default(false))
5248f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            Tokens[I].int_data[0] = CXToken_Keyword;
5249f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
5250f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5251f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5252f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
5253f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCInstanceMethodDecl ||
5254f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_ObjCClassMethodDecl) {
5255f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
5256f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (llvm::StringSwitch<bool>(II->getName())
52576628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("in", true)
52586628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("out", true)
52596628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("inout", true)
52606628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("oneway", true)
52616628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("bycopy", true)
52626628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("byref", true)
52636628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Default(false))
5264f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Tokens[I].int_data[0] = CXToken_Keyword;
5265f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5266f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
52676639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis
52686639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      if (Cursors[I].kind == CXCursor_CXXFinalAttr ||
52696639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis          Cursors[I].kind == CXCursor_CXXOverrideAttr) {
52706639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis        Tokens[I].int_data[0] = CXToken_Keyword;
52716628a614c504263ae539462f049d523dd07ac1baTed Kremenek        continue;
5272f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5273f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
5274f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
5275fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
52766628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52776628a614c504263ae539462f049d523dd07ac1baTed Kremenekextern "C" {
52786628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52796628a614c504263ae539462f049d523dd07ac1baTed Kremenekvoid clang_annotateTokens(CXTranslationUnit TU,
52806628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXToken *Tokens, unsigned NumTokens,
52816628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXCursor *Cursors) {
52826628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52836628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (NumTokens == 0 || !Tokens || !Cursors)
52846628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
52856628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52866628a614c504263ae539462f049d523dd07ac1baTed Kremenek  // Any token we don't specifically annotate will have a NULL cursor.
52876628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor C = clang_getNullCursor();
52886628a614c504263ae539462f049d523dd07ac1baTed Kremenek  for (unsigned I = 0; I != NumTokens; ++I)
52896628a614c504263ae539462f049d523dd07ac1baTed Kremenek    Cursors[I] = C;
52906628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52916628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
52926628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!CXXUnit)
52936628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
52946628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52956628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
52966628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52976628a614c504263ae539462f049d523dd07ac1baTed Kremenek  clang_annotateTokens_Data data = { TU, CXXUnit, Tokens, NumTokens, Cursors };
52986628a614c504263ae539462f049d523dd07ac1baTed Kremenek  llvm::CrashRecoveryContext CRC;
52996628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!RunSafely(CRC, clang_annotateTokensImpl, &data,
53006628a614c504263ae539462f049d523dd07ac1baTed Kremenek                 GetSafetyThreadStackSize() * 2)) {
53016628a614c504263ae539462f049d523dd07ac1baTed Kremenek    fprintf(stderr, "libclang: crash detected while annotating tokens\n");
53026628a614c504263ae539462f049d523dd07ac1baTed Kremenek  }
53036628a614c504263ae539462f049d523dd07ac1baTed Kremenek}
53046628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5305fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} // end: extern "C"
5306fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
5307fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
530816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek// Operations for querying linkage of a cursor.
530916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
531016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
531116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenekextern "C" {
531216b4259aecaa22b642d35d36fd89965ed700c1e0Ted KremenekCXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
53130396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
53140396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    return CXLinkage_Invalid;
53150396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
531616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  Decl *D = cxcursor::getCursorDecl(cursor);
531716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  if (NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
531816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    switch (ND->getLinkage()) {
531916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case NoLinkage: return CXLinkage_NoLinkage;
532016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case InternalLinkage: return CXLinkage_Internal;
532116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
532216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case ExternalLinkage: return CXLinkage_External;
532316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    };
532416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
532516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  return CXLinkage_Invalid;
532616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek}
532716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek} // end: extern "C"
532816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
532916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
533045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek// Operations for querying language of a cursor.
533145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
533245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
533345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekstatic CXLanguageKind getDeclLanguage(const Decl *D) {
533416ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis  if (!D)
533516ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis    return CXLanguage_C;
533616ed0e6fda74800595cb98d11969fc4afcc00bacArgyrios Kyrtzidis
533745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  switch (D->getKind()) {
533845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    default:
533945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      break;
534045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ImplicitParam:
534145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCAtDefsField:
534245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategory:
534345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategoryImpl:
534445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCompatibleAlias:
534545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCImplementation:
534645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCInterface:
534745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCIvar:
534845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCMethod:
534945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProperty:
535045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCPropertyImpl:
535145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProtocol:
535245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_ObjC;
535345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConstructor:
535445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConversion:
535545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXDestructor:
535645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXMethod:
535745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXRecord:
535845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplate:
535945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplatePartialSpecialization:
536045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplateSpecialization:
536145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Friend:
536245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FriendTemplate:
536345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FunctionTemplate:
536445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::LinkageSpec:
536545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Namespace:
536645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NamespaceAlias:
536745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NonTypeTemplateParm:
536845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::StaticAssert:
536945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTemplateParm:
537045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTypeParm:
537145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingTypename:
537245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingValue:
537345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Using:
537445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingDirective:
537545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingShadow:
537645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_CPlusPlus;
537745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  }
537845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
537945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_C;
538045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
538145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
538245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekextern "C" {
538358ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
538458ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregorenum CXAvailabilityKind clang_getCursorAvailability(CXCursor cursor) {
538558ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  if (clang_isDeclaration(cursor.kind))
538658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    if (Decl *D = cxcursor::getCursorDecl(cursor)) {
53870a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      if (isa<FunctionDecl>(D) && cast<FunctionDecl>(D)->isDeleted())
538858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Available;
538958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
53900a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      switch (D->getAvailability()) {
53910a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Available:
53920a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_NotYetIntroduced:
53930a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_Available;
53940a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
53950a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Deprecated:
539658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Deprecated;
53970a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
53980a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Unavailable:
53990a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_NotAvailable;
54000a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      }
540158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    }
54020a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
540358ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  return CXAvailability_Available;
540458ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor}
540558ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
540645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted KremenekCXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
540745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  if (clang_isDeclaration(cursor.kind))
540845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    return getDeclLanguage(cxcursor::getCursorDecl(cursor));
540945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
541045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_Invalid;
541145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
54123910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54133910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// \brief If the given cursor is the "templated" declaration
54143910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// descibing a class or function template, return the class or
54153910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// function template.
54163910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregorstatic Decl *maybeGetTemplateCursor(Decl *D) {
54173910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (!D)
54183910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    return 0;
54193910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54203910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
54213910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (FunctionTemplateDecl *FunTmpl = FD->getDescribedFunctionTemplate())
54223910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return FunTmpl;
54233910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54243910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D))
54253910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (ClassTemplateDecl *ClassTmpl = RD->getDescribedClassTemplate())
54263910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return ClassTmpl;
54273910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54283910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  return D;
54293910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor}
54303910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54312be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorSemanticParent(CXCursor cursor) {
54322be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
54332be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
54342be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getDeclContext();
54353910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
54363910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
54373910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54383910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
54393910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
54402be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
54412be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
54422be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54432be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isStatement(cursor.kind) || clang_isExpression(cursor.kind)) {
54442be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor))
5445a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(D, getCursorTU(cursor));
54462be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
54472be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54482be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
54492be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
54502be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54512be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorLexicalParent(CXCursor cursor) {
54522be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
54532be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
54542be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getLexicalDeclContext();
54553910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
54563910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
54573910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54583910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
54593910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
54602be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
54612be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
54622be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54632be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // FIXME: Note that we can't easily compute the lexical context of a
54642be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // statement or expression, so we return nothing.
54652be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
54662be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
54672be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54689f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_getOverriddenCursors(CXCursor cursor,
54699f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                CXCursor **overridden,
54709f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                unsigned *num_overridden) {
54719f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (overridden)
54729f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *overridden = 0;
54739f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (num_overridden)
54749f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *num_overridden = 0;
54759f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!overridden || !num_overridden)
54769f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
54779f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
5478b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  SmallVector<CXCursor, 8> Overridden;
5479b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  cxcursor::getOverriddenCursors(cursor, Overridden);
54809f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
54812407712d90cb1cce3bb2713d342c4df8222a2a47Ted Kremenek  // Don't allocate memory if we have no overriden cursors.
54822407712d90cb1cce3bb2713d342c4df8222a2a47Ted Kremenek  if (Overridden.size() == 0)
54832407712d90cb1cce3bb2713d342c4df8222a2a47Ted Kremenek    return;
54842407712d90cb1cce3bb2713d342c4df8222a2a47Ted Kremenek
5485b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  *num_overridden = Overridden.size();
5486b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  *overridden = new CXCursor [Overridden.size()];
5487b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  std::copy(Overridden.begin(), Overridden.end(), *overridden);
54889f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
54899f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
54909f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_disposeOverriddenCursors(CXCursor *overridden) {
54919f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  delete [] overridden;
54929f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
54939f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
5494ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas GregorCXFile clang_getIncludedFile(CXCursor cursor) {
5495ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (cursor.kind != CXCursor_InclusionDirective)
5496ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return 0;
5497ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
5498ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  InclusionDirective *ID = getCursorInclusionDirective(cursor);
5499ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  return (void *)ID->getFile();
5500ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor}
5501ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
550245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek} // end: extern "C"
550345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
55049ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
55059ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
55069ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek// C++ AST instrospection.
55079ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
55089ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
55099ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekextern "C" {
55109ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekunsigned clang_CXXMethod_isStatic(CXCursor C) {
55119ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek  if (!clang_isDeclaration(C.kind))
55129ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek    return 0;
551349f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor
551449f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  CXXMethodDecl *Method = 0;
551549f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
551649f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
551749f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
551849f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  else
551949f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
552049f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  return (Method && Method->isStatic()) ? 1 : 0;
552140b492a43bac3ed0c465772aa6921d011cfc273fTed Kremenek}
5522b12903e1a4b8d1b611b8c7e4f910665d628e68cdTed Kremenek
5523211924b563aa31421836cee7655be729ad02733fDouglas Gregorunsigned clang_CXXMethod_isVirtual(CXCursor C) {
5524211924b563aa31421836cee7655be729ad02733fDouglas Gregor  if (!clang_isDeclaration(C.kind))
5525211924b563aa31421836cee7655be729ad02733fDouglas Gregor    return 0;
5526211924b563aa31421836cee7655be729ad02733fDouglas Gregor
5527211924b563aa31421836cee7655be729ad02733fDouglas Gregor  CXXMethodDecl *Method = 0;
5528211924b563aa31421836cee7655be729ad02733fDouglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
5529211924b563aa31421836cee7655be729ad02733fDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
5530211924b563aa31421836cee7655be729ad02733fDouglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
5531211924b563aa31421836cee7655be729ad02733fDouglas Gregor  else
5532211924b563aa31421836cee7655be729ad02733fDouglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
5533211924b563aa31421836cee7655be729ad02733fDouglas Gregor  return (Method && Method->isVirtual()) ? 1 : 0;
5534211924b563aa31421836cee7655be729ad02733fDouglas Gregor}
55359ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek} // end: extern "C"
55369ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
553745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
553895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek// Attribute introspection.
553995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
554095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
554195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenekextern "C" {
554295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted KremenekCXType clang_getIBOutletCollectionType(CXCursor C) {
554395f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  if (C.kind != CXCursor_IBOutletCollectionAttr)
5544a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return cxtype::MakeCXType(QualType(), cxcursor::getCursorTU(C));
554595f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
554695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  IBOutletCollectionAttr *A =
554795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek    cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(C));
554895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
554918aa2ff4641847d7f8866e8c5912d4d0ddb858ceArgyrios Kyrtzidis  return cxtype::MakeCXType(A->getInterface(), cxcursor::getCursorTU(C));
555095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek}
555195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek} // end: extern "C"
555295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
555395f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
555459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek// Inspecting memory usage.
555559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek//===----------------------------------------------------------------------===//
555659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5557f787002478f09af1741fb0f82a562002e6799c49Ted Kremenektypedef std::vector<CXTUResourceUsageEntry> MemUsageEntries;
555859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5559f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekstatic inline void createCXTUResourceUsageEntry(MemUsageEntries &entries,
5560f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek                                              enum CXTUResourceUsageKind k,
5561ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek                                              unsigned long amount) {
5562f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  CXTUResourceUsageEntry entry = { k, amount };
556359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  entries.push_back(entry);
556459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
556559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
556659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenekextern "C" {
556759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5568f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekconst char *clang_getTUResourceUsageName(CXTUResourceUsageKind kind) {
556959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  const char *str = "";
557059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  switch (kind) {
5571f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_AST:
557259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: expressions, declarations, and types";
557359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      break;
5574f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_Identifiers:
557559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: identifiers";
557659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      break;
5577f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_Selectors:
557859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: selectors";
5579e294ab717fc9535429ca5d8f575d41ae4441d822Ted Kremenek      break;
5580f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_GlobalCompletionResults:
55814e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek      str = "Code completion: cached global results";
5582e294ab717fc9535429ca5d8f575d41ae4441d822Ted Kremenek      break;
5583457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek    case CXTUResourceUsage_SourceManagerContentCache:
5584457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek      str = "SourceManager: content cache allocator";
5585457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek      break;
5586ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    case CXTUResourceUsage_AST_SideTables:
5587ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek      str = "ASTContext: side tables";
5588ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek      break;
5589f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    case CXTUResourceUsage_SourceManager_Membuffer_Malloc:
5590f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      str = "SourceManager: malloc'ed memory buffers";
5591f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      break;
5592f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    case CXTUResourceUsage_SourceManager_Membuffer_MMap:
5593f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      str = "SourceManager: mmap'ed memory buffers";
5594f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      break;
5595e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    case CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc:
5596e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      str = "ExternalASTSource: malloc'ed memory buffers";
5597e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      break;
5598e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    case CXTUResourceUsage_ExternalASTSource_Membuffer_MMap:
5599e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      str = "ExternalASTSource: mmap'ed memory buffers";
5600e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      break;
56015e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    case CXTUResourceUsage_Preprocessor:
56025e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      str = "Preprocessor: malloc'ed memory";
56035e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      break;
56045e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    case CXTUResourceUsage_PreprocessingRecord:
56055e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      str = "Preprocessor: PreprocessingRecord";
56065e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      break;
5607ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek    case CXTUResourceUsage_SourceManager_DataStructures:
5608ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek      str = "SourceManager: data structures and tables";
5609ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek      break;
5610d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek    case CXTUResourceUsage_Preprocessor_HeaderSearch:
5611d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek      str = "Preprocessor: header search tables";
5612d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek      break;
561359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  }
561459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  return str;
561559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
561659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5617f787002478f09af1741fb0f82a562002e6799c49Ted KremenekCXTUResourceUsage clang_getCXTUResourceUsage(CXTranslationUnit TU) {
561859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  if (!TU) {
5619f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    CXTUResourceUsage usage = { (void*) 0, 0, 0 };
562059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    return usage;
562159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  }
562259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
562359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  ASTUnit *astUnit = static_cast<ASTUnit*>(TU->TUData);
56241e4c01b79273b9cd4e9e9ecfd3422df3900b8356Dylan Noblesmith  OwningPtr<MemUsageEntries> entries(new MemUsageEntries());
562559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  ASTContext &astContext = astUnit->getASTContext();
562659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
562759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used by AST nodes and types?
5628f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST,
5629ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    (unsigned long) astContext.getASTAllocatedMemory());
563059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
563159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used by identifiers?
5632f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Identifiers,
563359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    (unsigned long) astContext.Idents.getAllocator().getTotalMemory());
563459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
563559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used for selectors?
5636f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Selectors,
563759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    (unsigned long) astContext.Selectors.getTotalMemory());
563859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5639ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek  // How much memory is used by ASTContext's side tables?
5640ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST_SideTables,
5641ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    (unsigned long) astContext.getSideTableAllocatedMemory());
5642ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek
56434e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  // How much memory is used for caching global code completion results?
56444e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  unsigned long completionBytes = 0;
56454e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  if (GlobalCodeCompletionAllocator *completionAllocator =
56464e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek      astUnit->getCachedCompletionAllocator().getPtr()) {
56475e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    completionBytes = completionAllocator->getTotalMemory();
56484e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  }
5649457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  createCXTUResourceUsageEntry(*entries,
5650457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek                               CXTUResourceUsage_GlobalCompletionResults,
5651457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek                               completionBytes);
5652457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek
5653457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  // How much memory is being used by SourceManager's content cache?
5654457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  createCXTUResourceUsageEntry(*entries,
5655457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek          CXTUResourceUsage_SourceManagerContentCache,
5656457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek          (unsigned long) astContext.getSourceManager().getContentCacheSize());
5657f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek
5658f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  // How much memory is being used by the MemoryBuffer's in SourceManager?
5659f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  const SourceManager::MemoryBufferSizes &srcBufs =
5660f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    astUnit->getSourceManager().getMemoryBufferSizes();
5661f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek
5662f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5663f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               CXTUResourceUsage_SourceManager_Membuffer_Malloc,
5664f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               (unsigned long) srcBufs.malloc_bytes);
5665ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5666f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               CXTUResourceUsage_SourceManager_Membuffer_MMap,
5667f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               (unsigned long) srcBufs.mmap_bytes);
5668ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5669ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                               CXTUResourceUsage_SourceManager_DataStructures,
5670ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                               (unsigned long) astContext.getSourceManager()
5671ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                                .getDataStructureSizes());
5672e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek
5673e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  // How much memory is being used by the ExternalASTSource?
5674e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  if (ExternalASTSource *esrc = astContext.getExternalSource()) {
5675e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    const ExternalASTSource::MemoryBufferSizes &sizes =
5676e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      esrc->getMemoryBufferSizes();
5677e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek
5678e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    createCXTUResourceUsageEntry(*entries,
5679e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc,
5680e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek                                 (unsigned long) sizes.malloc_bytes);
5681e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    createCXTUResourceUsageEntry(*entries,
5682e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      CXTUResourceUsage_ExternalASTSource_Membuffer_MMap,
5683e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek                                 (unsigned long) sizes.mmap_bytes);
5684e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  }
56855e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
56865e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  // How much memory is being used by the Preprocessor?
56875e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  Preprocessor &pp = astUnit->getPreprocessor();
56885e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  createCXTUResourceUsageEntry(*entries,
56895e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                               CXTUResourceUsage_Preprocessor,
5690c5c5e92ec53f7e6ac7ebbbf77c6d8e4b7d88daecArgyrios Kyrtzidis                               pp.getTotalMemory());
56915e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
56925e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  if (PreprocessingRecord *pRec = pp.getPreprocessingRecord()) {
56935e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    createCXTUResourceUsageEntry(*entries,
56945e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                                 CXTUResourceUsage_PreprocessingRecord,
56955e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                                 pRec->getTotalMemory());
56965e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  }
56975e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
5698d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5699d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek                               CXTUResourceUsage_Preprocessor_HeaderSearch,
5700d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek                               pp.getHeaderSearchInfo().getTotalMemory());
57015e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
5702f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  CXTUResourceUsage usage = { (void*) entries.get(),
570359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek                            (unsigned) entries->size(),
570459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek                            entries->size() ? &(*entries)[0] : 0 };
570559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  entries.take();
570659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  return usage;
570759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
570859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5709f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekvoid clang_disposeCXTUResourceUsage(CXTUResourceUsage usage) {
571059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  if (usage.data)
571159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    delete (MemUsageEntries*) usage.data;
571259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
571359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
571459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek} // end extern "C"
571559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
57166df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregorvoid clang::PrintLibclangResourceUsage(CXTranslationUnit TU) {
57176df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  CXTUResourceUsage Usage = clang_getCXTUResourceUsage(TU);
57186df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  for (unsigned I = 0; I != Usage.numEntries; ++I)
57196df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    fprintf(stderr, "  %s: %lu\n",
57206df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor            clang_getTUResourceUsageName(Usage.entries[I].kind),
57216df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor            Usage.entries[I].amount);
57226df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
57236df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  clang_disposeCXTUResourceUsage(Usage);
57246df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor}
57256df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
572659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek//===----------------------------------------------------------------------===//
572704bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek// Misc. utility functions.
572804bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek//===----------------------------------------------------------------------===//
5729f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
5730abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbar/// Default to using an 8 MB stack size on "safety" threads.
5731abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbarstatic unsigned SafetyStackThreadSize = 8 << 20;
5732bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5733bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarnamespace clang {
5734bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5735bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarbool RunSafely(llvm::CrashRecoveryContext &CRC,
57366c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               void (*Fn)(void*), void *UserData,
57376c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               unsigned Size) {
57386c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (!Size)
57396c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek    Size = GetSafetyThreadStackSize();
57406c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (Size)
5741bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar    return CRC.RunSafelyOnThread(Fn, UserData, Size);
5742bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return CRC.RunSafely(Fn, UserData);
5743bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5744bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5745bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarunsigned GetSafetyThreadStackSize() {
5746bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return SafetyStackThreadSize;
5747bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5748bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5749bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarvoid SetSafetyThreadStackSize(unsigned Value) {
5750bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  SafetyStackThreadSize = Value;
5751bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5752bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
57538e7c48a54b4325925f5edda1738a3537ab2c3c5eArgyrios Kyrtzidis}
57548e7c48a54b4325925f5edda1738a3537ab2c3c5eArgyrios Kyrtzidis
575581b5ac39a97cde1a54b8d0eb7105290c40eb84d7Argyrios Kyrtzidisvoid clang::setThreadBackgroundPriority() {
5756fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  // FIXME: Move to llvm/Support and make it cross-platform.
5757fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis#ifdef __APPLE__
5758fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis  setpriority(PRIO_DARWIN_THREAD, 0, PRIO_DARWIN_BG);
5759fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis#endif
5760fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis}
5761fdc1795acc9d5d73a767cc7d43ad1546e93adbbaArgyrios Kyrtzidis
576204bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenekextern "C" {
576304bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
5764a2a9d6e4e5b6001b86b7dfc5db1ea296ce29a3d3Ted KremenekCXString clang_getClangVersion() {
5765ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(getClangFullVersion());
576604bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek}
576704bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
576804bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek} // end: extern "C"
576959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5770