CIndex.cpp revision ba98617b994864b7554ff75445983ad02a962f45
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"
22ab1889321f6f8f200f2b318ac26883ac18e49d03Ted Kremenek
2304bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek#include "clang/Basic/Version.h"
24936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor
2550398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff#include "clang/AST/DeclVisitor.h"
26fb5704295c6137685a7b90b92cd6b958028740c8Steve Naroff#include "clang/AST/StmtVisitor.h"
277d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor#include "clang/AST/TypeLocVisitor.h"
28b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Basic/Diagnostic.h"
29b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Frontend/ASTUnit.h"
30b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Frontend/CompilerInstance.h"
31936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor#include "clang/Frontend/FrontendDiagnostic.h"
32d8210650ed948de65a08a8daf16d291b747717c4Ted Kremenek#include "clang/Lex/Lexer.h"
33dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor#include "clang/Lex/HeaderSearch.h"
34b846debc1b22a37228efe4aa87b34482d15b6a3cBenjamin Kramer#include "clang/Lex/PreprocessingRecord.h"
3533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor#include "clang/Lex/Preprocessor.h"
36a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor#include "llvm/ADT/STLExtras.h"
37d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek#include "llvm/ADT/Optional.h"
38f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor#include "llvm/ADT/StringSwitch.h"
39d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek#include "clang/Analysis/Support/SaveAndRestore.h"
40c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar#include "llvm/Support/CrashRecoveryContext.h"
4148615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar#include "llvm/Support/PrettyStackTrace.h"
4202465750c8c3fa96b1e7e596b02297e24361dc4fDouglas Gregor#include "llvm/Support/MemoryBuffer.h"
43358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor#include "llvm/Support/raw_ostream.h"
447a07fcb8f10fe45ea65a0a41798eb1c40777bde4Douglas Gregor#include "llvm/Support/Timer.h"
4503013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Mutex.h"
4603013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Program.h"
4703013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Signals.h"
4803013fa9a0bf1ef4b907f5fec006c8f4000fdd21Michael J. Spencer#include "llvm/Support/Threading.h"
4937f1ea0eb08a00fa90edbecb427cfbb50ca0f4d0Ted Kremenek#include "llvm/Support/Compiler.h"
50fc0622155fa61349698a8fd0053773c37d9f7ac4Ted Kremenek
5150398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroffusing namespace clang;
5216c440a377b7ec8b722a2e2c7c864f75c95bd305Ted Kremenekusing namespace clang::cxcursor;
53ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenekusing namespace clang::cxstring;
549049cf6cb468c856888e88251dab659955fa767eArgyrios Kyrtzidisusing namespace clang::cxtu;
5550398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff
569049cf6cb468c856888e88251dab659955fa767eArgyrios KyrtzidisCXTranslationUnit cxtu::MakeCXTranslationUnit(ASTUnit *TU) {
57a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  if (!TU)
58a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return 0;
59a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit D = new CXTranslationUnitImpl();
60a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  D->TUData = TU;
61a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  D->StringPool = createCXStringPool();
62a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return D;
63a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek}
64a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek
654e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidiscxtu::CXTUOwner::~CXTUOwner() {
664e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis  if (TU)
674e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis    clang_disposeTranslationUnit(TU);
684e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis}
694e7064fa7e344e8f87a5b8457e96dfdd252c4a9eArgyrios Kyrtzidis
7033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \brief The result of comparing two source ranges.
7133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorenum RangeComparisonResult {
7233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief Either the ranges overlap or one of the ranges is invalid.
7333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeOverlap,
74f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The first range ends before the second range starts.
7633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeBefore,
77f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The first range starts after the second range ends.
7933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeAfter
8033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor};
8133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
82f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek/// \brief Compare two source ranges to determine their relative position in
8333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// the translation unit.
84f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekstatic RangeComparisonResult RangeCompare(SourceManager &SM,
85f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                          SourceRange R1,
8633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                                          SourceRange R2) {
8733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R1.isValid() && "First range is invalid?");
8833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R2.isValid() && "Second range is invalid?");
89a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R1.getEnd() != R2.getBegin() &&
90d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R1.getEnd(), R2.getBegin()))
9133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeBefore;
92a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R2.getEnd() != R1.getBegin() &&
93d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R2.getEnd(), R1.getBegin()))
9433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeAfter;
9533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return RangeOverlap;
9633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
9733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
98fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek/// \brief Determine if a source location falls within, before, or after a
99fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek///   a given source range.
100fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic RangeComparisonResult LocationCompare(SourceManager &SM,
101fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                             SourceLocation L, SourceRange R) {
102fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(R.isValid() && "First range is invalid?");
103fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(L.isValid() && "Second range is invalid?");
104a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (L == R.getBegin() || L == R.getEnd())
105fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeOverlap;
106fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(L, R.getBegin()))
107fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeBefore;
108fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(R.getEnd(), L))
109fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeAfter;
110fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return RangeOverlap;
111fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek}
112fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
11376dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// \brief Translate a Clang source range into a CIndex source range.
11476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar///
11576dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// Clang internally represents ranges where the end location points to the
11676dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// start of the token at the end. However, for external clients it is more
11776dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// useful to have a CXSourceRange be a proper half-open interval. This routine
11876dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// does the appropriate translation.
119f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed KremenekCXSourceRange cxloc::translateSourceRange(const SourceManager &SM,
12076dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                                          const LangOptions &LangOpts,
1210a76aae8c03cb7dd7bdbe683485560afaf695959Chris Lattner                                          const CharSourceRange &R) {
12276dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  // We want the last character in this location, so we will adjust the
1236a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor  // location accordingly.
12476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  SourceLocation EndLoc = R.getEnd();
125a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (EndLoc.isValid() && EndLoc.isMacroID())
126edc3dccece244a584f8ebdb81da6c962c08e79beChandler Carruth    EndLoc = SM.getExpansionRange(EndLoc).second;
1270a76aae8c03cb7dd7bdbe683485560afaf695959Chris Lattner  if (R.isTokenRange() && !EndLoc.isInvalid() && EndLoc.isFileID()) {
1286a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor    unsigned Length = Lexer::MeasureTokenLength(EndLoc, SM, LangOpts);
129a64ccefdf0ea4e03ec88805d71b0af74950c7472Argyrios Kyrtzidis    EndLoc = EndLoc.getLocWithOffset(Length);
13076dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  }
13176dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar
13276dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  CXSourceRange Result = { { (void *)&SM, (void *)&LangOpts },
13376dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           R.getBegin().getRawEncoding(),
13476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           EndLoc.getRawEncoding() };
13576dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  return Result;
13676dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar}
1371db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
1388a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
13933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor// Cursor visitor.
1408a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
1418a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek
14289922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroffnamespace {
143c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
144c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekclass VisitorJob {
145c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekpublic:
146cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek  enum Kind { DeclVisitKind, StmtVisitKind, MemberExprPartsKind,
147e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek              TypeLocVisitKind, OverloadExprPartsKind,
14860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek              DeclRefExprPartsKind, LabelRefVisitKind,
149f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek              ExplicitTemplateArgsVisitKind,
150f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor              NestedNameSpecifierLocVisitKind,
151cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek              DeclarationNameInfoVisitKind,
15294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor              MemberRefVisitKind, SizeOfPackExprPartsKind };
153c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekprotected:
154f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void *data[3];
155c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  CXCursor parent;
156c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  Kind K;
157f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  VisitorJob(CXCursor C, Kind k, void *d1, void *d2 = 0, void *d3 = 0)
158f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : parent(C), K(k) {
159f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    data[0] = d1;
160f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    data[1] = d2;
161f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    data[2] = d3;
162f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
163c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekpublic:
164c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  Kind getKind() const { return K; }
165c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  const CXCursor &getParent() const { return parent; }
166c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  static bool classof(VisitorJob *VJ) { return true; }
167c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek};
168c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
1695f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnertypedef SmallVector<VisitorJob, 10> VisitorWorkList;
170c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
171b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor// Cursor visitor.
1727d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorclass CursorVisitor : public DeclVisitor<CursorVisitor, bool>,
173cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek                      public TypeLocVisitor<CursorVisitor, bool>
1747d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor{
17533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The translation unit we are traversing.
176a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU;
177a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *AU;
178f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
17933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The parent cursor whose children we are traversing.
180b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  CXCursor Parent;
181f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
18233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The declaration that serves at the parent of any statement or
18333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// expression nodes.
184f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  Decl *StmtParent;
185f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
18633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The visitor function.
187b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  CXCursorVisitor Visitor;
188f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
18933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The opaque client data, to be passed along to the visitor.
190b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  CXClientData ClientData;
191f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
19204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor  /// \brief Whether we should visit the preprocessing record entries last,
19304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor  /// after visiting other declarations.
19404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor  bool VisitPreprocessorLast;
195f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
196e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis  /// \brief Whether we should visit declarations or preprocessing record
197e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis  /// entries that are #included inside the \arg RegionOfInterest.
198e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis  bool VisitIncludedEntities;
19904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
20033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief When valid, a source range to which the cursor should restrict
20133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// its search.
20233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  SourceRange RegionOfInterest;
203f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
204d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually remove.  This part of a hack to support proper
205d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // iteration over all Decls contained lexically within an ObjC container.
206d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator *DI_current;
207d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator DE_current;
208d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
209d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  // Cache of pre-allocated worklists for data-recursion walk of Stmts.
2105f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<VisitorWorkList*, 5> WorkListFreeList;
2115f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<VisitorWorkList*, 5> WorkListCache;
212d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek
213b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  using DeclVisitor<CursorVisitor, bool>::Visit;
2147d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  using TypeLocVisitor<CursorVisitor, bool>::Visit;
215f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
216f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  /// \brief Determine whether this particular source range comes before, comes
217f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  /// after, or overlaps the region of interest.
21833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  ///
219d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar  /// \param R a half-open source range retrieved from the abstract syntax tree.
220f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  RangeComparisonResult CompareRegionOfInterest(SourceRange R);
221f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
222dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  void visitDeclsFromFileRegion(FileID File, unsigned Offset, unsigned Length);
223dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
2240f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  class SetParentRAII {
2250f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    CXCursor &Parent;
2260f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    Decl *&StmtParent;
2270f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    CXCursor OldParent;
2280f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek
2290f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  public:
2300f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    SetParentRAII(CXCursor &Parent, Decl *&StmtParent, CXCursor NewParent)
2310f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      : Parent(Parent), StmtParent(StmtParent), OldParent(Parent)
2320f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    {
2330f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      Parent = NewParent;
2340f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      if (clang_isDeclaration(Parent.kind))
2350f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek        StmtParent = getCursorDecl(Parent);
2360f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    }
2370f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek
2380f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    ~SetParentRAII() {
2390f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      Parent = OldParent;
2400f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      if (clang_isDeclaration(Parent.kind))
2410f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek        StmtParent = getCursorDecl(Parent);
2420f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    }
2430f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  };
2440f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek
245b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorpublic:
246a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CursorVisitor(CXTranslationUnit TU, CXCursorVisitor Visitor,
247a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                CXClientData ClientData,
24804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                bool VisitPreprocessorLast,
249f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                bool VisitIncludedPreprocessingEntries = false,
25033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                SourceRange RegionOfInterest = SourceRange())
251a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    : TU(TU), AU(static_cast<ASTUnit*>(TU->TUData)),
252a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      Visitor(Visitor), ClientData(ClientData),
25308e0bc16b3312c27e87d33be7dcf3d4fe5bdd2e2Douglas Gregor      VisitPreprocessorLast(VisitPreprocessorLast),
254e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis      VisitIncludedEntities(VisitIncludedPreprocessingEntries),
25504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      RegionOfInterest(RegionOfInterest), DI_current(0)
256b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  {
257b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.kind = CXCursor_NoDeclFound;
258b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.data[0] = 0;
259b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.data[1] = 0;
260b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.data[2] = 0;
261f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    StmtParent = 0;
262b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
263f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
264d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  ~CursorVisitor() {
265d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    // Free the pre-allocated worklists for data-recursion.
2665f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    for (SmallVectorImpl<VisitorWorkList*>::iterator
267d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek          I = WorkListCache.begin(), E = WorkListCache.end(); I != E; ++I) {
268d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek      delete *I;
269d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    }
270d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
271d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek
272a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *getASTUnit() const { return static_cast<ASTUnit*>(TU->TUData); }
273a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit getTU() const { return TU; }
274ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek
27533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  bool Visit(CXCursor Cursor, bool CheckedRegionOfInterest = false);
276dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
277dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  /// \brief Visit declarations and preprocessed entities for the file region
278dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  /// designated by \see RegionOfInterest.
279dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  void visitFileRegion();
280788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
2814c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  bool visitPreprocessedEntitiesInRegion();
2824c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
283e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis  bool shouldVisitIncludedEntities() const {
284e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis    return VisitIncludedEntities;
285f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  }
286f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
2874c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  template<typename InputIterator>
288f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  bool visitPreprocessedEntities(InputIterator First, InputIterator Last,
289f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                 PreprocessingRecord &PPRec,
290f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                 FileID FID = FileID());
291788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
292b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  bool VisitChildren(CXCursor Parent);
293f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2947d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  // Declaration visitors
295162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  bool VisitTypeAliasDecl(TypeAliasDecl *D);
29609dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  bool VisitAttributes(Decl *D);
2971ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  bool VisitBlockDecl(BlockDecl *B);
2983064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  bool VisitCXXRecordDecl(CXXRecordDecl *D);
299d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  llvm::Optional<bool> shouldVisitCursor(CXCursor C);
300b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  bool VisitDeclContext(DeclContext *DC);
30179758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTranslationUnitDecl(TranslationUnitDecl *D);
30279758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTypedefDecl(TypedefDecl *D);
30379758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTagDecl(TagDecl *D);
3040ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool VisitClassTemplateSpecializationDecl(ClassTemplateSpecializationDecl *D);
30574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  bool VisitClassTemplatePartialSpecializationDecl(
30674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                     ClassTemplatePartialSpecializationDecl *D);
307fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D);
3084540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitEnumConstantDecl(EnumConstantDecl *D);
30979758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitDeclaratorDecl(DeclaratorDecl *DD);
3104540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitFunctionDecl(FunctionDecl *ND);
31179758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitFieldDecl(FieldDecl *D);
31279758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitVarDecl(VarDecl *);
31384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  bool VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D);
314fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitFunctionTemplateDecl(FunctionTemplateDecl *D);
31539d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  bool VisitClassTemplateDecl(ClassTemplateDecl *D);
31684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  bool VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D);
31779758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCMethodDecl(ObjCMethodDecl *ND);
3184540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitObjCContainerDecl(ObjCContainerDecl *D);
31979758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCCategoryDecl(ObjCCategoryDecl *ND);
32079758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCProtocolDecl(ObjCProtocolDecl *PID);
32123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  bool VisitObjCPropertyDecl(ObjCPropertyDecl *PD);
32279758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCInterfaceDecl(ObjCInterfaceDecl *D);
3234540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitObjCImplDecl(ObjCImplDecl *D);
32479758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D);
3251ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  bool VisitObjCImplementationDecl(ObjCImplementationDecl *D);
32679758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  // FIXME: ObjCCompatibleAliasDecl requires aliased-class locations.
32779758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D);
32879758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCClassDecl(ObjCClassDecl *D);
329a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  bool VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD);
330a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  bool VisitLinkageSpecDecl(LinkageSpecDecl *D);
3318f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  bool VisitNamespaceDecl(NamespaceDecl *D);
3326931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  bool VisitNamespaceAliasDecl(NamespaceAliasDecl *D);
3330a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  bool VisitUsingDirectiveDecl(UsingDirectiveDecl *D);
3347e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUsingDecl(UsingDecl *D);
3357e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D);
3367e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUnresolvedUsingTypenameDecl(UnresolvedUsingTypenameDecl *D);
3370a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
33801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  // Name visitor
33901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  bool VisitDeclarationNameInfo(DeclarationNameInfo Name);
340c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  bool VisitNestedNameSpecifier(NestedNameSpecifier *NNS, SourceRange Range);
341dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  bool VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc NNS);
34201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
343fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Template visitors
344fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateParameters(const TemplateParameterList *Params);
3450b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  bool VisitTemplateName(TemplateName Name, SourceLocation Loc);
346fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL);
347fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
3487d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  // Type visitors
349427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis#define ABSTRACT_TYPELOC(CLASS, PARENT)
350427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis#define TYPELOC(CLASS, PARENT) \
351427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  bool Visit##CLASS##TypeLoc(CLASS##TypeLoc TyLoc);
352427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis#include "clang/AST/TypeLocNodes.def"
353427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
354f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitTagTypeLoc(TagTypeLoc TL);
355f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitArrayTypeLoc(ArrayTypeLoc TL);
356427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  bool VisitFunctionTypeLoc(FunctionTypeLoc TL, bool SkipResultType = false);
357427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
358c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  // Data-recursive visitor functions.
359c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  bool IsInRegionOfInterest(CXCursor C);
360c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  bool RunVisitorWorkList(VisitorWorkList &WL);
361c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  void EnqueueWorkList(VisitorWorkList &WL, Stmt *S);
362cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  LLVM_ATTRIBUTE_NOINLINE bool Visit(Stmt *S);
363b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor};
364f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
365b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor} // end anonymous namespace
3660d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
367a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C);
3686653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr);
3696653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
370a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
37133e9abd21083a0191a7676a04b497006d2da184dDouglas GregorRangeComparisonResult CursorVisitor::CompareRegionOfInterest(SourceRange R) {
372a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return RangeCompare(AU->getSourceManager(), R, RegionOfInterest);
37333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
37433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
375b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the given cursor and, if requested by the visitor,
376b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// its children.
377b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor///
37833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param Cursor the cursor to visit.
37933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
38033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param CheckRegionOfInterest if true, then the caller already checked that
38133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// this cursor is within the region of interest.
38233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
383b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
384b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
38533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorbool CursorVisitor::Visit(CXCursor Cursor, bool CheckedRegionOfInterest) {
386b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isInvalid(Cursor.kind))
387b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
388f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
389b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
390b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
391b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    assert(D && "Invalid declaration cursor");
39265ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // Ignore implicit declarations, unless it's an objc method because
39365ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // currently we should report implicit methods for properties when indexing.
39465ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    if (D->isImplicit() && !isa<ObjCMethodDecl>(D))
395b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return false;
396b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
3970d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
39833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // If we have a range of interest, and this cursor doesn't intersect with it,
39933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // we're done.
40033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (RegionOfInterest.isValid() && !CheckedRegionOfInterest) {
401a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    SourceRange Range = getRawCursorExtent(Cursor);
402f408f32aa9ae3d97bc656267dc5d78fa7d03499bDaniel Dunbar    if (Range.isInvalid() || CompareRegionOfInterest(Range))
40333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor      return false;
40433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  }
405f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
406b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  switch (Visitor(Cursor, Parent, ClientData)) {
407b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Break:
408b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
4090d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
410b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Continue:
411b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
4122e331b938b38057e333fab0ba841130ea8467794Douglas Gregor
413b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Recurse:
414b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return VisitChildren(Cursor);
415b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
4160d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
417fd64377225a6a140bddb3f997d52a036486f9360Douglas Gregor  return false;
418b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
4190d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
420f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidisstatic bool visitPreprocessedEntitiesInRange(SourceRange R,
421f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                             PreprocessingRecord &PPRec,
422f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                             CursorVisitor &Visitor) {
423f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  SourceManager &SM = Visitor.getASTUnit()->getSourceManager();
424f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  FileID FID;
425f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
426e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis  if (!Visitor.shouldVisitIncludedEntities()) {
427f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    // If the begin/end of the range lie in the same FileID, do the optimization
428f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    // where we skip preprocessed entities that do not come from the same FileID.
429f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    FID = SM.getFileID(R.getBegin());
430f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (FID != SM.getFileID(R.getEnd()))
431f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      FID = FileID();
432f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  }
433f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
434f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  std::pair<PreprocessingRecord::iterator, PreprocessingRecord::iterator>
435f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    Entities = PPRec.getPreprocessedEntitiesInRange(R);
436f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  return Visitor.visitPreprocessedEntities(Entities.first, Entities.second,
437f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                           PPRec, FID);
438f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis}
439f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
440dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidisvoid CursorVisitor::visitFileRegion() {
441dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (RegionOfInterest.isInvalid())
442dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
443dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
444dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  ASTUnit *Unit = static_cast<ASTUnit *>(TU->TUData);
445dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceManager &SM = Unit->getSourceManager();
446dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
447dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  std::pair<FileID, unsigned>
448dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Begin = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getBegin())),
449dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End = SM.getDecomposedLoc(SM.getFileLoc(RegionOfInterest.getEnd()));
450dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
451dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (End.first != Begin.first) {
452dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    // If the end does not reside in the same file, try to recover by
453dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    // picking the end of the file of begin location.
454dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End.first = Begin.first;
455dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    End.second = SM.getFileIDSize(Begin.first);
456dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
457dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
458dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  assert(Begin.first == End.first);
459dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (Begin.second > End.second)
460dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
461dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
462dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  FileID File = Begin.first;
463dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  unsigned Offset = Begin.second;
464dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  unsigned Length = End.second - Begin.second;
465dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
466dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (!VisitPreprocessorLast &&
467dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      Unit->getPreprocessor().getPreprocessingRecord())
468dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    visitPreprocessedEntitiesInRegion();
469dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
470dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  visitDeclsFromFileRegion(File, Offset, Length);
471dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
472dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (VisitPreprocessorLast &&
473dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      Unit->getPreprocessor().getPreprocessingRecord())
474dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    visitPreprocessedEntitiesInRegion();
475dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis}
476dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
477dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidisvoid CursorVisitor::visitDeclsFromFileRegion(FileID File,
478dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis                                             unsigned Offset, unsigned Length) {
479dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  ASTUnit *Unit = static_cast<ASTUnit *>(TU->TUData);
480dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceManager &SM = Unit->getSourceManager();
481dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SourceRange Range = RegionOfInterest;
482dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
483dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SmallVector<Decl *, 16> Decls;
484dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  Unit->findFileRegionDecls(File, Offset, Length, Decls);
485dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
486dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // If we didn't find any file level decls for the file, try looking at the
487dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // file that it was included from.
488dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  while (Decls.empty()) {
489dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    bool Invalid = false;
490dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    const SrcMgr::SLocEntry &SLEntry = SM.getSLocEntry(File, &Invalid);
491dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (Invalid)
492dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      return;
493dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
494dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    SourceLocation Outer;
495dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (SLEntry.isFile())
496dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      Outer = SLEntry.getFile().getIncludeLoc();
497dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    else
498dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      Outer = SLEntry.getExpansion().getExpansionLocStart();
499dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (Outer.isInvalid())
500dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      return;
501dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
502dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    llvm::tie(File, Offset) = SM.getDecomposedExpansionLoc(Outer);
503dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Length = 0;
504dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Unit->findFileRegionDecls(File, Offset, Length, Decls);
505dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
506dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
507dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  assert(!Decls.empty());
508dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
509dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  bool VisitedAtLeastOnce = false;
510dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  SmallVector<Decl *, 16>::iterator DIt = Decls.begin();
511dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  for (SmallVector<Decl *, 16>::iterator DE = Decls.end(); DIt != DE; ++DIt) {
512dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Decl *D = *DIt;
513dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
514dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    // We handle forward decls via ObjCClassDecl.
515dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (ObjCInterfaceDecl *InterD = dyn_cast<ObjCInterfaceDecl>(D)) {
516dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      if (InterD->isForwardDecl())
517dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis        continue;
518dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      // An interface that started as a forward decl may have changed location
519dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      // because its @interface was parsed.
520dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      if (InterD->isInitiallyForwardDecl() &&
521dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis          !SM.isInFileID(SM.getFileLoc(InterD->getLocation()), File))
522dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis        continue;
523dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    }
524dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
525dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    RangeComparisonResult CompRes = RangeCompare(SM, D->getSourceRange(),Range);
526dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CompRes == RangeBefore)
527dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      continue;
528dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CompRes == RangeAfter)
529dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
530dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
531dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    assert(CompRes == RangeOverlap);
532dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    VisitedAtLeastOnce = true;
533ba98617b994864b7554ff75445983ad02a962f45Argyrios Kyrtzidis    if (Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true))
534dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
535dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
536dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
537dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  if (VisitedAtLeastOnce)
538dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    return;
539dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
540dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // No Decls overlapped with the range. Move up the lexical context until there
541dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // is a context that contains the range or we reach the translation unit
542dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  // level.
543dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  DeclContext *DC = DIt == Decls.begin() ? (*DIt)->getLexicalDeclContext()
544dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis                                         : (*(DIt-1))->getLexicalDeclContext();
545dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
546dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  while (DC && !DC->isTranslationUnit()) {
547dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    Decl *D = cast<Decl>(DC);
548dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    SourceRange CurDeclRange = D->getSourceRange();
549dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (CurDeclRange.isInvalid())
550dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
551dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
552dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    if (RangeCompare(SM, CurDeclRange, Range) == RangeOverlap) {
553ba98617b994864b7554ff75445983ad02a962f45Argyrios Kyrtzidis      Visit(MakeCXCursor(D, TU, Range), /*CheckedRegionOfInterest=*/true);
554dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis      break;
555dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    }
556dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
557dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    DC = D->getLexicalDeclContext();
558dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis  }
559dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis}
560dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis
5614c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregorbool CursorVisitor::visitPreprocessedEntitiesInRegion() {
562788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  PreprocessingRecord &PPRec
563a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    = *AU->getPreprocessor().getPreprocessingRecord();
564f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  SourceManager &SM = AU->getSourceManager();
565788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
56692ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  if (RegionOfInterest.isValid()) {
567ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange MappedRange = AU->mapRangeToPreamble(RegionOfInterest);
568f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    SourceLocation B = MappedRange.getBegin();
569f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    SourceLocation E = MappedRange.getEnd();
570f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
571f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (AU->isInPreambleFileID(B)) {
572f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      if (SM.isLoadedSourceLocation(E))
573f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis        return visitPreprocessedEntitiesInRange(SourceRange(B, E),
574f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                                 PPRec, *this);
575f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
576f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // Beginning of range lies in the preamble but it also extends beyond
577f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // it into the main file. Split the range into 2 parts, one covering
578f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // the preamble and another covering the main file. This allows subsequent
579f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // calls to visitPreprocessedEntitiesInRange to accept a source range that
580f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // lies in the same FileID, allowing it to skip preprocessed entities that
581f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      // do not come from the same FileID.
582f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      bool breaked =
583f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis        visitPreprocessedEntitiesInRange(
584f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                   SourceRange(B, AU->getEndOfPreambleFileID()),
585f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                          PPRec, *this);
586f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      if (breaked) return true;
587f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      return visitPreprocessedEntitiesInRange(
588f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                    SourceRange(AU->getStartOfMainFileID(), E),
589f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                        PPRec, *this);
590f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    }
591f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
592f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    return visitPreprocessedEntitiesInRange(SourceRange(B, E), PPRec, *this);
59392ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  }
59492ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis
595788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  bool OnlyLocalDecls
59632038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    = !AU->isMainFileAST() && AU->getOnlyLocalDecls();
59732038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor
59892ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  if (OnlyLocalDecls)
599f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    return visitPreprocessedEntities(PPRec.local_begin(), PPRec.local_end(),
600f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                     PPRec);
6014c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
602f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  return visitPreprocessedEntities(PPRec.begin(), PPRec.end(), PPRec);
6034c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor}
6044c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
6054c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregortemplate<typename InputIterator>
6064c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregorbool CursorVisitor::visitPreprocessedEntities(InputIterator First,
607f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              InputIterator Last,
608f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              PreprocessingRecord &PPRec,
609f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                              FileID FID) {
6104c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  for (; First != Last; ++First) {
611f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (!FID.isInvalid() && !PPRec.isEntityInFileID(First, FID))
612f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis      continue;
613f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis
614f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    PreprocessedEntity *PPE = *First;
615f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (MacroExpansion *ME = dyn_cast<MacroExpansion>(PPE)) {
6164c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeMacroExpansionCursor(ME, TU)))
6174c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
6184c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
6194c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
6204c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    }
6214c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
622f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (MacroDefinition *MD = dyn_cast<MacroDefinition>(PPE)) {
6234c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeMacroDefinitionCursor(MD, TU)))
6244c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
62589d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
6264c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
6274c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    }
6284c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
629f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis    if (InclusionDirective *ID = dyn_cast<InclusionDirective>(PPE)) {
6304c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeInclusionDirectiveCursor(ID, TU)))
6314c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
6324c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
6334c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
634788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    }
635788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  }
636788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
6374c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  return false;
638788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor}
639788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
640b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the children of the given cursor.
641a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek///
642b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
643b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
644f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekbool CursorVisitor::VisitChildren(CXCursor Cursor) {
645c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (clang_isReference(Cursor.kind) &&
646c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      Cursor.kind != CXCursor_CXXBaseSpecifier) {
647a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    // By definition, references have no children.
648a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return false;
649a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  }
650f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
651f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Set the Parent field to Cursor, then back to its old value once we're
652b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // done.
6530f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  SetParentRAII SetParent(Parent, StmtParent, Cursor);
654f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
655b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
656b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
65706d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (!D)
65806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return false;
65906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
660539311e0221df256c70c1c3080c8af847cd29dffTed Kremenek    return VisitAttributes(D) || Visit(D);
661b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
662f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
66306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isStatement(Cursor.kind)) {
66406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Stmt *S = getCursorStmt(Cursor))
66506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(S);
66606d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
66706d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
66806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
66906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
67006d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isExpression(Cursor.kind)) {
67106d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Expr *E = getCursorExpr(Cursor))
67206d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(E);
67306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
67406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
67506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
676f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
677b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isTranslationUnit(Cursor.kind)) {
678a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CXTranslationUnit tu = getCursorTU(Cursor);
679a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    ASTUnit *CXXUnit = static_cast<ASTUnit*>(tu->TUData);
68004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
68104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    int VisitOrder[2] = { VisitPreprocessorLast, !VisitPreprocessorLast };
68204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    for (unsigned I = 0; I != 2; ++I) {
68304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      if (VisitOrder[I]) {
68404a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
68504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            RegionOfInterest.isInvalid()) {
68604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          for (ASTUnit::top_level_iterator TL = CXXUnit->top_level_begin(),
68704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                        TLEnd = CXXUnit->top_level_end();
68804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor               TL != TLEnd; ++TL) {
689aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis            if (Visit(MakeCXCursor(*TL, tu, RegionOfInterest), true))
69004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
69104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
69204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        } else if (VisitDeclContext(
69304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                CXXUnit->getASTContext().getTranslationUnitDecl()))
6947b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor          return true;
69504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        continue;
6967b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor      }
6973178cb674ac8c3b59e1791e14d38d48619a1b621Bob Wilson
69804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      // Walk the preprocessing record.
6994c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (CXXUnit->getPreprocessor().getPreprocessingRecord())
7004c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        visitPreprocessedEntitiesInRegion();
7010396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    }
70204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
7037b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor    return false;
704b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
705f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
706c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (Cursor.kind == CXCursor_CXXBaseSpecifier) {
707c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    if (CXXBaseSpecifier *Base = getCursorCXXBaseSpecifier(Cursor)) {
708c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      if (TypeSourceInfo *BaseTSInfo = Base->getTypeSourceInfo()) {
709c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor        return Visit(BaseTSInfo->getTypeLoc());
710c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      }
711c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    }
712c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  }
713221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis
714221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis  if (Cursor.kind == CXCursor_IBOutletCollectionAttr) {
715221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis    IBOutletCollectionAttr *A =
716221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis      cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(Cursor));
717221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis    if (const ObjCInterfaceType *InterT = A->getInterface()->getAs<ObjCInterfaceType>())
718221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis      return Visit(cxcursor::MakeCursorObjCClassRef(InterT->getInterface(),
719221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis                                                    A->getInterfaceLoc(), TU));
720221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis  }
721221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis
722b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // Nothing to visit at the moment.
723b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
724dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
725dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
7261ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenekbool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
72713c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor  if (TypeSourceInfo *TSInfo = B->getSignatureAsWritten())
72813c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor    if (Visit(TSInfo->getTypeLoc()))
72913c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor        return true;
7301ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
731664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  if (Stmt *Body = B->getBody())
732aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Body, StmtParent, TU, RegionOfInterest));
733664cffd330611d78fc0286f539589920a37ca328Ted Kremenek
734664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  return false;
7351ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek}
7361ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
737d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekllvm::Optional<bool> CursorVisitor::shouldVisitCursor(CXCursor Cursor) {
738d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (RegionOfInterest.isValid()) {
7396653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange Range = getFullCursorExtent(Cursor, AU->getSourceManager());
740d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Range.isInvalid())
741d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
7426653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
743d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    switch (CompareRegionOfInterest(Range)) {
744d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeBefore:
745d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes before the region of interest; skip it.
746d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
74723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
748d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeAfter:
749d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes after the region of interest; we're done.
750d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
751d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
752d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeOverlap:
753d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration overlaps the region of interest; visit it.
754d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
755d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
756d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
757d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return true;
758d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
759f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
760d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekbool CursorVisitor::VisitDeclContext(DeclContext *DC) {
761d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator I = DC->decls_begin(), E = DC->decls_end();
762f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
763d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually remove.  This part of a hack to support proper
764d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // iteration over all Decls contained lexically within an ObjC container.
765d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator*> DI_saved(DI_current, &I);
766d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator> DE_saved(DE_current, E);
767f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
768d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for ( ; I != E; ++I) {
769d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *D = *I;
770d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (D->getLexicalDeclContext() != DC)
771d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
772aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    CXCursor Cursor = MakeCXCursor(D, TU, RegionOfInterest);
773d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
774d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
775d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
776d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
777d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
778d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar    if (Visit(Cursor, true))
779b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
780b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
781b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
782dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
783dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
7841ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
7851ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  llvm_unreachable("Translation units are visited directly by Visit()");
7861ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
7871ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
7881ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
789162e1c1b487352434552147967c3dd296ebee2f7Richard Smithbool CursorVisitor::VisitTypeAliasDecl(TypeAliasDecl *D) {
790162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
791162e1c1b487352434552147967c3dd296ebee2f7Richard Smith    return Visit(TSInfo->getTypeLoc());
792162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
793162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return false;
794162e1c1b487352434552147967c3dd296ebee2f7Richard Smith}
795162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
7961ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
7971ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
7981ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(TSInfo->getTypeLoc());
799f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8001ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8011ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8021ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
8031ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTagDecl(TagDecl *D) {
8041ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitDeclContext(D);
8051ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8061ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
8070ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregorbool CursorVisitor::VisitClassTemplateSpecializationDecl(
8080ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor                                          ClassTemplateSpecializationDecl *D) {
8090ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool ShouldVisitBody = false;
8100ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  switch (D->getSpecializationKind()) {
8110ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_Undeclared:
8120ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ImplicitInstantiation:
8130ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    // Nothing to visit
8140ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return false;
8150ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
8160ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDeclaration:
8170ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDefinition:
8180ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
8190ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
8200ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitSpecialization:
8210ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    ShouldVisitBody = true;
8220ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
8230ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
8240ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
8250ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  // Visit the template arguments used in the specialization.
8260ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (TypeSourceInfo *SpecType = D->getTypeAsWritten()) {
8270ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    TypeLoc TL = SpecType->getTypeLoc();
8280ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    if (TemplateSpecializationTypeLoc *TSTLoc
8290ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          = dyn_cast<TemplateSpecializationTypeLoc>(&TL)) {
8300ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor      for (unsigned I = 0, N = TSTLoc->getNumArgs(); I != N; ++I)
8310ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor        if (VisitTemplateArgumentLoc(TSTLoc->getArgLoc(I)))
8320ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          return true;
8330ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    }
8340ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
8350ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
8360ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (ShouldVisitBody && VisitCXXRecordDecl(D))
8370ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return true;
8380ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
8390ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  return false;
8400ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor}
8410ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
84274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregorbool CursorVisitor::VisitClassTemplatePartialSpecializationDecl(
84374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                   ClassTemplatePartialSpecializationDecl *D) {
84474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
84574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // before visiting these template parameters.
84674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
84774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return true;
84874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
84974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // Visit the partial specialization arguments.
85074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  const TemplateArgumentLoc *TemplateArgs = D->getTemplateArgsAsWritten();
85174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  for (unsigned I = 0, N = D->getNumTemplateArgsAsWritten(); I != N; ++I)
85274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    if (VisitTemplateArgumentLoc(TemplateArgs[I]))
85374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor      return true;
85474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
85574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  return VisitCXXRecordDecl(D);
85674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor}
85774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
858fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
85984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  // Visit the default argument.
86084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
86184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (TypeSourceInfo *DefArg = D->getDefaultArgumentInfo())
86284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      if (Visit(DefArg->getTypeLoc()))
86384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor        return true;
86484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
865fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
866fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
867fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
8681ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
8691ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInitExpr())
870aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
8711ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8721ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8731ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
8747d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
8757d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
8767d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
8777d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      return true;
8787d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
879c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
880c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = DD->getQualifierLoc())
881c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
882c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
883c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
8847d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return false;
8857d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
8867d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
887a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor/// \brief Compare two base or member initializers based on their source order.
888cbb67480094b3bcb5b715acd827cbad55e2a204cSean Huntstatic int CompareCXXCtorInitializers(const void* Xp, const void *Yp) {
889cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *X
890cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Xp);
891cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *Y
892cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Yp);
893a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
894a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  if ((*X)->getSourceOrder() < (*Y)->getSourceOrder())
895a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return -1;
896a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else if ((*X)->getSourceOrder() > (*Y)->getSourceOrder())
897a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 1;
898a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else
899a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 0;
900a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor}
901a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
902b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
90301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (TypeSourceInfo *TSInfo = ND->getTypeSourceInfo()) {
90401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function declaration's syntactic components in the order
90501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // written. This requires a bit of work.
906723df245307a530da5433dfb43accf187dc3e243Abramo Bagnara    TypeLoc TL = TSInfo->getTypeLoc().IgnoreParens();
90701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    FunctionTypeLoc *FTL = dyn_cast<FunctionTypeLoc>(&TL);
90801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
90901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // If we have a function declared directly (without the use of a typedef),
91001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // visit just the return type. Otherwise, just visit the function's type
91101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // now.
91201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if ((FTL && !isa<CXXConversionDecl>(ND) && Visit(FTL->getResultLoc())) ||
91301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor        (!FTL && Visit(TL)))
91401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
91501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
916c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // Visit the nested-name-specifier, if present.
917c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (NestedNameSpecifierLoc QualifierLoc = ND->getQualifierLoc())
918c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      if (VisitNestedNameSpecifierLoc(QualifierLoc))
919c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor        return true;
92001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
92101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the declaration name.
92201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (VisitDeclarationNameInfo(ND->getNameInfo()))
92301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
92401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
92501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Visit explicitly-specified template arguments!
92601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
92701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function parameters, if we have a function type.
92801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (FTL && VisitFunctionTypeLoc(*FTL, true))
92901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
93001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
93101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Attributes?
93201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
93301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
93410620eb5164e31208fcbf0437cd79ae535ed0559Sean Hunt  if (ND->doesThisDeclarationHaveABody() && !ND->isLateTemplateParsed()) {
935a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    if (CXXConstructorDecl *Constructor = dyn_cast<CXXConstructorDecl>(ND)) {
936a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Find the initializers that were written in the source.
9375f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      SmallVector<CXXCtorInitializer *, 4> WrittenInits;
938a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (CXXConstructorDecl::init_iterator I = Constructor->init_begin(),
939a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                          IEnd = Constructor->init_end();
940a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor           I != IEnd; ++I) {
941a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (!(*I)->isWritten())
942a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          continue;
943a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
944a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        WrittenInits.push_back(*I);
945a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
946a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
947a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Sort the initializers in source order
948a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      llvm::array_pod_sort(WrittenInits.begin(), WrittenInits.end(),
949cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt                           &CompareCXXCtorInitializers);
950a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
951a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Visit the initializers in source order
952a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (unsigned I = 0, N = WrittenInits.size(); I != N; ++I) {
953cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt        CXXCtorInitializer *Init = WrittenInits[I];
95400eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet        if (Init->isAnyMemberInitializer()) {
95500eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet          if (Visit(MakeCursorMemberRef(Init->getAnyMember(),
956a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                        Init->getMemberLocation(), TU)))
957a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
95876852c218a207ef43583515cb835b6e855353a0fDouglas Gregor        } else if (TypeSourceInfo *TInfo = Init->getTypeSourceInfo()) {
95976852c218a207ef43583515cb835b6e855353a0fDouglas Gregor          if (Visit(TInfo->getTypeLoc()))
960a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
961a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        }
962a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
963a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        // Visit the initializer value.
964a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (Expr *Initializer = Init->getInit())
965aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis          if (Visit(MakeCXCursor(Initializer, ND, TU, RegionOfInterest)))
966a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
967a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
968a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
969a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
970aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
971a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return true;
972a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  }
973f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
974b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
975b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
976dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
9771ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
9781ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
9791ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
980f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9811ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *BitWidth = D->getBitWidth())
982aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(BitWidth, StmtParent, TU, RegionOfInterest));
983f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9841ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
9851ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
9861ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
9871ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitVarDecl(VarDecl *D) {
9881ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
9891ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
990f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInit())
992aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
993f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9941ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
9951ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
9961ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
99784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D) {
99884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitDeclaratorDecl(D))
99984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
100084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
100184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
100284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (Expr *DefArg = D->getDefaultArgument())
1003aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(DefArg, StmtParent, TU, RegionOfInterest));
100484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
100584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
100684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
100784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
1008fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
1009fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the FunctionDecl
1010fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // before visiting these template parameters.
1011fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
1012fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return true;
1013fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1014fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return VisitFunctionDecl(D->getTemplatedDecl());
1015fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1016fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
101739d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregorbool CursorVisitor::VisitClassTemplateDecl(ClassTemplateDecl *D) {
101839d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
101939d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // before visiting these template parameters.
102039d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
102139d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return true;
102239d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
102339d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  return VisitCXXRecordDecl(D->getTemplatedDecl());
102439d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor}
102539d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
102684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D) {
102784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
102884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
102984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
103084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited() &&
103184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      VisitTemplateArgumentLoc(D->getDefaultArgument()))
103284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
103384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
103484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
103584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
103684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
10371ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
10384bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor  if (TypeSourceInfo *TSInfo = ND->getResultTypeSourceInfo())
10394bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
10404bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor      return true;
10414bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor
1042f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  for (ObjCMethodDecl::param_iterator P = ND->param_begin(),
10431ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       PEnd = ND->param_end();
10441ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       P != PEnd; ++P) {
1045aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
10461ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
10471ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  }
1048f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10491ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (ND->isThisDeclarationADefinition() &&
1050aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
10511ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
1052f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10531ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
10541ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
1056d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremeneknamespace {
1057d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  struct ContainerDeclsSort {
1058d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    SourceManager &SM;
1059d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    ContainerDeclsSort(SourceManager &sm) : SM(sm) {}
1060d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    bool operator()(Decl *A, Decl *B) {
1061d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_A = A->getLocStart();
1062d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_B = B->getLocStart();
1063d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      assert(L_A.isValid() && L_B.isValid());
1064d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return SM.isBeforeInTranslationUnit(L_A, L_B);
1065d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
1066d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  };
1067d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
1068d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
1069a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregorbool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
1070d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually convert back to just 'VisitDeclContext()'.  Essentially
1071d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // an @implementation can lexically contain Decls that are not properly
1072d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // nested in the AST.  When we identify such cases, we need to retrofit
1073d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // this nesting here.
1074d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (!DI_current)
1075d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
1076d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
1077d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Scan the Decls that immediately come after the container
1078d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // in the current DeclContext.  If any fall within the
1079d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // container's lexical region, stash them into a vector
1080d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // for later processing.
10815f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<Decl *, 24> DeclsInContainer;
1082d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SourceLocation EndLoc = D->getSourceRange().getEnd();
1083a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  SourceManager &SM = AU->getSourceManager();
1084d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (EndLoc.isValid()) {
1085d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclContext::decl_iterator next = *DI_current;
1086d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    while (++next != DE_current) {
1087d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      Decl *D_next = *next;
1088d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (!D_next)
1089d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        break;
1090d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L = D_next->getLocStart();
1091d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (!L.isValid())
1092d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        break;
1093d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (SM.isBeforeInTranslationUnit(L, EndLoc)) {
1094d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        *DI_current = next;
1095d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        DeclsInContainer.push_back(D_next);
1096d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        continue;
1097d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      }
1098d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
1099d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
1100d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
1101d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
1102d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // The common case.
1103d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (DeclsInContainer.empty())
1104d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
1105d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
1106d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Get all the Decls in the DeclContext, and sort them with the
1107d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // additional ones we've collected.  Then visit them.
1108d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for (DeclContext::decl_iterator I = D->decls_begin(), E = D->decls_end();
1109d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek       I!=E; ++I) {
1110d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *subDecl = *I;
11110582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek    if (!subDecl || subDecl->getLexicalDeclContext() != D ||
11120582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek        subDecl->getLocStart().isInvalid())
1113d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
1114d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclsInContainer.push_back(subDecl);
1115d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
1116d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
1117d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now sort the Decls so that they appear in lexical order.
1118d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  std::sort(DeclsInContainer.begin(), DeclsInContainer.end(),
1119d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek            ContainerDeclsSort(SM));
1120d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
1121d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now visit the decls.
11225f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  for (SmallVectorImpl<Decl*>::iterator I = DeclsInContainer.begin(),
1123d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek         E = DeclsInContainer.end(); I != E; ++I) {
1124aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    CXCursor Cursor = MakeCXCursor(*I, TU, RegionOfInterest);
1125d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
1126d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
1127d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
1128d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
1129d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
1130d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Visit(Cursor, true))
1131d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return true;
1132d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
1133d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return false;
1134a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor}
1135a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor
1136b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
1137b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
1138b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                   TU)))
1139b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
1140f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
114178db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
114278db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
114378db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = ND->protocol_end(); I != E; ++I, ++PL)
1144b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1145b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1146f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1147a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(ND);
1148dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1149dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
11501ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
11511ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
11521ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
11531ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       E = PID->protocol_end(); I != E; ++I, ++PL)
11541ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
11551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
1156f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
11571ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(PID);
11581ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
11591ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
116023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenekbool CursorVisitor::VisitObjCPropertyDecl(ObjCPropertyDecl *PD) {
116183cb94269015bf2770ade71e616c5322ea7e76e1Douglas Gregor  if (PD->getTypeSourceInfo() && Visit(PD->getTypeSourceInfo()->getTypeLoc()))
1162fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall    return true;
1163fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall
116423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // FIXME: This implements a workaround with @property declarations also being
116523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // installed in the DeclContext for the @interface.  Eventually this code
116623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // should be removed.
116723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCCategoryDecl *CDecl = dyn_cast<ObjCCategoryDecl>(PD->getDeclContext());
116823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!CDecl || !CDecl->IsClassExtension())
116923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
117023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
117123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCInterfaceDecl *ID = CDecl->getClassInterface();
117223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!ID)
117323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
117423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
117523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  IdentifierInfo *PropertyId = PD->getIdentifier();
117623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCPropertyDecl *prevDecl =
117723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(ID), PropertyId);
117823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
117923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!prevDecl)
118023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
118123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
118223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // Visit synthesized methods since they will be skipped when visiting
118323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // the @interface.
118423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getGetterMethodDecl())
1185a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
1186aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
118723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
118823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
118923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getSetterMethodDecl())
1190a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
1191aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
119223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
119323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
119423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  return false;
119523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek}
119623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
1197b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
1198dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek  // Issue callbacks for super class.
1199b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (D->getSuperClass() &&
1200b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1201f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
1202b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                        TU)))
1203b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
1204f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
120578db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
120678db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
120778db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = D->protocol_end(); I != E; ++I, ++PL)
1208b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1209b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1210f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1211a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(D);
1212dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1213dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
12141ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
12151ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(D);
12161ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
12171ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
12181ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
1219ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  // 'ID' could be null when dealing with invalid code.
1220ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  if (ObjCInterfaceDecl *ID = D->getClassInterface())
1221ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek    if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
1222ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek      return true;
1223f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
12241ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
12251ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
12261ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
12271ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
12281ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#if 0
12291ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // Issue callbacks for super class.
12301ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // FIXME: No source location information!
12311ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (D->getSuperClass() &&
12321ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1233f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
12341ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                        TU)))
1235a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return true;
12361ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#endif
1237f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
12381ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
1239dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1240dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
12411ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D) {
12421ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCForwardProtocolDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
12431ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCForwardProtocolDecl::protocol_iterator I = D->protocol_begin(),
12441ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                                  E = D->protocol_end();
12451ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       I != E; ++I, ++PL)
1246b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1247b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1248f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1249f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return false;
1250dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1251dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
12521ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCClassDecl(ObjCClassDecl *D) {
125395ed7784a335aca53b0c6e952cf31a4cfb633360Fariborz Jahanian  if (Visit(MakeCursorObjCClassRef(D->getForwardInterfaceDecl(),
125495ed7784a335aca53b0c6e952cf31a4cfb633360Fariborz Jahanian                                   D->getForwardDecl()->getLocation(), TU)))
12551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
12561ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
1257dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
12585e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramer
1259a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregorbool CursorVisitor::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD) {
1260a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  if (ObjCIvarDecl *Ivar = PD->getPropertyIvarDecl())
1261a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor    return Visit(MakeCursorMemberRef(Ivar, PD->getPropertyIvarDeclLoc(), TU));
1262a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
1263a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  return false;
1264a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor}
1265a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
12668f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenekbool CursorVisitor::VisitNamespaceDecl(NamespaceDecl *D) {
12678f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  return VisitDeclContext(D);
12688f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek}
12698f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek
12706931900f43cea558c6974075256c07728dbfecc6Douglas Gregorbool CursorVisitor::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
1271c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
12720cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
12730cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1274c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
12756931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
12766931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  return Visit(MakeCursorNamespaceRef(D->getAliasedNamespace(),
12776931900f43cea558c6974075256c07728dbfecc6Douglas Gregor                                      D->getTargetNameLoc(), TU));
12786931900f43cea558c6974075256c07728dbfecc6Douglas Gregor}
12796931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
12807e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUsingDecl(UsingDecl *D) {
1281c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1282dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1283dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1284c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1285dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
12867e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
12871f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (Visit(MakeCursorOverloadedDeclRef(D, D->getLocation(), TU)))
12881f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return true;
12891f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
12907e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
12917e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
12927e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
12930a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregorbool CursorVisitor::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
1294c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1295db9924191092b4d426cc066637d81698211846aaDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1296db9924191092b4d426cc066637d81698211846aaDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1297c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
12980a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
12990a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  return Visit(MakeCursorNamespaceRef(D->getNominatedNamespaceAsWritten(),
13000a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor                                      D->getIdentLocation(), TU));
13010a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor}
13020a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
13037e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1304c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1305dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1306dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1307c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1308dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1309c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
13107e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
13117e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
13127e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
13137e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypenameDecl(
13147e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor                                               UnresolvedUsingTypenameDecl *D) {
1315c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1316dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1317dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1318c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1319c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
13207e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return false;
13217e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
13227e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
132301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitDeclarationNameInfo(DeclarationNameInfo Name) {
132401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  switch (Name.getName().getNameKind()) {
132501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::Identifier:
132601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXLiteralOperatorName:
132701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXOperatorName:
132801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXUsingDirective:
132901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
133001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
133101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConstructorName:
133201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXDestructorName:
133301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConversionFunctionName:
133401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (TypeSourceInfo *TSInfo = Name.getNamedTypeInfo())
133501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return Visit(TSInfo->getTypeLoc());
133601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
133701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
133801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCZeroArgSelector:
133901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCOneArgSelector:
134001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCMultiArgSelector:
134101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Per-identifier location info?
134201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
134301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
134401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
134501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return false;
134601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
134701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1348c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregorbool CursorVisitor::VisitNestedNameSpecifier(NestedNameSpecifier *NNS,
1349c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                             SourceRange Range) {
1350c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // FIXME: This whole routine is a hack to work around the lack of proper
1351c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // source information in nested-name-specifiers (PR5791). Since we do have
1352c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // a beginning source location, we can visit the first component of the
1353c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // nested-name-specifier, if it's a single-token component.
1354c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  if (!NNS)
1355c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return false;
1356c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1357c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Get the first component in the nested-name-specifier.
1358c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  while (NestedNameSpecifier *Prefix = NNS->getPrefix())
1359c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    NNS = Prefix;
1360c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1361c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  switch (NNS->getKind()) {
1362c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Namespace:
1363c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(), Range.getBegin(),
1364c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                        TU));
1365c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
136614aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor  case NestedNameSpecifier::NamespaceAlias:
136714aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
136814aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor                                        Range.getBegin(), TU));
136914aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor
1370c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpec: {
1371c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // If the type has a form where we know that the beginning of the source
1372c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // range matches up with a reference cursor. Visit the appropriate reference
1373c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // cursor.
1374f4c7371fb1d3cebcfb40abad4537bb82515704eaJohn McCall    const Type *T = NNS->getAsType();
1375c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TypedefType *Typedef = dyn_cast<TypedefType>(T))
1376c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Typedef->getDecl(), Range.getBegin(), TU));
1377c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TagType *Tag = dyn_cast<TagType>(T))
1378c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Tag->getDecl(), Range.getBegin(), TU));
1379c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TemplateSpecializationType *TST
1380c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                      = dyn_cast<TemplateSpecializationType>(T))
1381c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return VisitTemplateName(TST->getTemplateName(), Range.getBegin());
1382c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1383c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1384c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1385c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpecWithTemplate:
1386c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Global:
1387c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Identifier:
1388c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1389c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1390c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1391c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  return false;
1392c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor}
1393c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1394dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregorbool
1395dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas GregorCursorVisitor::VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
13965f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<NestedNameSpecifierLoc, 4> Qualifiers;
1397dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  for (; Qualifier; Qualifier = Qualifier.getPrefix())
1398dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    Qualifiers.push_back(Qualifier);
1399dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1400dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  while (!Qualifiers.empty()) {
1401dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifierLoc Q = Qualifiers.pop_back_val();
1402dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifier *NNS = Q.getNestedNameSpecifier();
1403dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    switch (NNS->getKind()) {
1404dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Namespace:
1405dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(),
1406c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1407dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1408dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1409dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1410dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1411dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1412dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::NamespaceAlias:
1413dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1414c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1415dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1416dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1417dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1418dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1419dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1420dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpec:
1421dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpecWithTemplate:
1422dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(Q.getTypeLoc()))
1423dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1424dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1425dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1426dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1427dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Global:
1428dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Identifier:
1429dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1430dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    }
1431dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1432dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1433dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  return false;
1434dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor}
1435dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1436fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateParameters(
1437fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          const TemplateParameterList *Params) {
1438fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (!Params)
1439fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1440fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1441fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (TemplateParameterList::const_iterator P = Params->begin(),
1442fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          PEnd = Params->end();
1443fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor       P != PEnd; ++P) {
1444aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
1445fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1446fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1447fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1448fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1449fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1450fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
14510b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregorbool CursorVisitor::VisitTemplateName(TemplateName Name, SourceLocation Loc) {
14520b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  switch (Name.getKind()) {
14530b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::Template:
14540b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(Name.getAsTemplateDecl(), Loc, TU));
14550b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
14560b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::OverloadedTemplate:
14571f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // Visit the overloaded template set.
14581f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (Visit(MakeCursorOverloadedDeclRef(Name, Loc, TU)))
14591f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return true;
14601f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
14610b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
14620b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
14630b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::DependentTemplate:
14640b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
14650b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
14660b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
14670b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::QualifiedTemplate:
14680b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
14690b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(
14700b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                  Name.getAsQualifiedTemplateName()->getDecl(),
14710b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                       Loc, TU));
1472146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall
1473146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall  case TemplateName::SubstTemplateTemplateParm:
1474146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall    return Visit(MakeCursorTemplateRef(
1475146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall                         Name.getAsSubstTemplateTemplateParm()->getParameter(),
1476146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall                                       Loc, TU));
14771aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor
14781aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor  case TemplateName::SubstTemplateTemplateParmPack:
14791aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor    return Visit(MakeCursorTemplateRef(
14801aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                  Name.getAsSubstTemplateTemplateParmPack()->getParameterPack(),
14811aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                                       Loc, TU));
14820b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  }
14830b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
14840b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  return false;
14850b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor}
14860b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
1487fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL) {
1488fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  switch (TAL.getArgument().getKind()) {
1489fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Null:
1490fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Integral:
1491fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Pack:
1492fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
149387dd697dcc8ecb64df73ae64d61b8c80ff0c157cDouglas Gregor
1494fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Type:
1495fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (TypeSourceInfo *TSInfo = TAL.getTypeSourceInfo())
1496fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(TSInfo->getTypeLoc());
1497fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1498fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1499fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Declaration:
1500fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceDeclExpression())
1501aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1502fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1503fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1504fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Expression:
1505fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceExpression())
1506aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1507fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1508fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1509fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Template:
1510a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor  case TemplateArgument::TemplateExpansion:
1511b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor    if (VisitNestedNameSpecifierLoc(TAL.getTemplateQualifierLoc()))
1512b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor      return true;
1513b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor
1514a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor    return VisitTemplateName(TAL.getArgument().getAsTemplateOrTemplatePattern(),
15150b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                             TAL.getTemplateNameLoc());
1516fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1517fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1518fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1519fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1520fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1521a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenekbool CursorVisitor::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
1522a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  return VisitDeclContext(D);
1523a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek}
1524a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek
152501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
152601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return Visit(TL.getUnqualifiedLoc());
152701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
152801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1529f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
1530a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTContext &Context = AU->getASTContext();
1531f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1532f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // Some builtin types (such as Objective-C's "id", "sel", and
1533f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // "Class") have associated declarations. Create cursors for those.
1534f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  QualType VisitType;
1535e0a22d06888c13989b3f72db319f1d498bf69153John McCall  switch (TL.getTypePtr()->getKind()) {
15362dde35bc626153492f5f58202506c88a27fbff5bJohn McCall
15376b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Void:
1538f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::NullPtr:
15396b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Dependent:
15402dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define BUILTIN_TYPE(Id, SingletonId)
15412dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define SIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
15422dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define UNSIGNED_TYPE(Id, SingletonId) case BuiltinType::Id:
15432dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define FLOATING_TYPE(Id, SingletonId) case BuiltinType::Id:
15442dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#define PLACEHOLDER_TYPE(Id, SingletonId) case BuiltinType::Id:
15452dde35bc626153492f5f58202506c88a27fbff5bJohn McCall#include "clang/AST/BuiltinTypes.def"
1546f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
15476b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1548f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCId:
1549f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCIdType();
1550f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
15516b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
15526b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ObjCClass:
15536b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    VisitType = Context.getObjCClassType();
15546b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    break;
15556b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1556f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCSel:
1557f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCSelType();
1558f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
1559f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1560f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1561f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (!VisitType.isNull()) {
1562f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
1563f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
1564f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                     TU));
1565f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1566f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1567f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1568f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1569f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
15707d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
1571162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return Visit(MakeCursorTypeRef(TL.getTypedefNameDecl(), TL.getNameLoc(), TU));
15727d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
15737d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1574f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
1575f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1576f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1577f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1578f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
15796f155de99c59af890817146ec8526bafb6560f1fArgyrios Kyrtzidis  if (TL.isDefinition())
1580aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(TL.getDecl(), TU, RegionOfInterest));
15816f155de99c59af890817146ec8526bafb6560f1fArgyrios Kyrtzidis
1582f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1583f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1584f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1585fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
1586960d13dde337a59dacc9dc3936c26d4aa8478986Chandler Carruth  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1587fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1588fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1589f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
1590f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU)))
1591f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1592f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1593c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return false;
1594c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall}
1595c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1596c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCallbool CursorVisitor::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
1597c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseLoc()))
1598c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall    return true;
1599c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1600f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1601f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1602f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                        TU)))
1603f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor      return true;
1604f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1605f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1606f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1607f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1608f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1609f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
1610c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return Visit(TL.getPointeeLoc());
1611f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1612f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1613075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnarabool CursorVisitor::VisitParenTypeLoc(ParenTypeLoc TL) {
1614075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara  return Visit(TL.getInnerLoc());
1615075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara}
1616075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara
1617f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
1618f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1619f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1620f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1621f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
1622f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1623f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1624f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1625f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
1626f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1627f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1628f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1629f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
1630f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1631f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1632f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1633f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
1634f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1635f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1636f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
16373422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidisbool CursorVisitor::VisitAttributedTypeLoc(AttributedTypeLoc TL) {
16383422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis  return Visit(TL.getModifiedLoc());
16393422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis}
16403422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis
164101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL,
164201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor                                         bool SkipResultType) {
164301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (!SkipResultType && Visit(TL.getResultLoc()))
1644f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1645f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1646f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
16475dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek    if (Decl *D = TL.getArg(I))
1648aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(D, TU, RegionOfInterest)))
16495dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek        return true;
1650f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1651f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1652f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1653f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1654f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
1655f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(TL.getElementLoc()))
1656f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1657f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1658f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Expr *Size = TL.getSizeExpr())
1659aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Size, StmtParent, TU, RegionOfInterest));
1660f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1661f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1662f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1663f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1664fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateSpecializationTypeLoc(
1665fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                             TemplateSpecializationTypeLoc TL) {
16660b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  // Visit the template name.
16670b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
16680b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                        TL.getTemplateNameLoc()))
16690b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return true;
1670fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1671fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Visit the template arguments.
1672fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1673fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1674fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1675fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1676fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1677fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1678fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
16792332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
16802332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
16812332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
16822332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
16832332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
16842332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
1685ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt    return Visit(TSInfo->getTypeLoc());
1686ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt
1687ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt  return false;
1688ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt}
1689ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt
1690ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Huntbool CursorVisitor::VisitUnaryTransformTypeLoc(UnaryTransformTypeLoc TL) {
1691ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
16922332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor    return Visit(TSInfo->getTypeLoc());
16932332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
16942332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return false;
16952332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
16962332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
16972494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregorbool CursorVisitor::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
16982494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
16992494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    return true;
17002494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
17012494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  return false;
17022494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor}
17032494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
170494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregorbool CursorVisitor::VisitDependentTemplateSpecializationTypeLoc(
170594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor                                    DependentTemplateSpecializationTypeLoc TL) {
170694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the nested-name-specifier, if there is one.
170794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  if (TL.getQualifierLoc() &&
170894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
170994fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    return true;
171094fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
171194fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the template arguments.
171294fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
171394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
171494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      return true;
171594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
171694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  return false;
171794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor}
171894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
17199e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregorbool CursorVisitor::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
17209e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
17219e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor    return true;
17229e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
17239e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  return Visit(TL.getNamedTypeLoc());
17249e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor}
17259e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
17267536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregorbool CursorVisitor::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
17277536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor  return Visit(TL.getPatternLoc());
17287536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor}
17297536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor
1730427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::VisitDecltypeTypeLoc(DecltypeTypeLoc TL) {
1731427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  if (Expr *E = TL.getUnderlyingExpr())
1732427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis    return Visit(MakeCXCursor(E, StmtParent, TU));
1733427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1734427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return false;
1735427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1736427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1737427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::VisitInjectedClassNameTypeLoc(InjectedClassNameTypeLoc TL) {
1738427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1739427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1740427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1741b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedmanbool CursorVisitor::VisitAtomicTypeLoc(AtomicTypeLoc TL) {
1742b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman  return Visit(TL.getValueLoc());
1743b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman}
1744b001de7458d17c17e6d8b8034c7cfcefd3b70c00Eli Friedman
1745427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis#define DEFAULT_TYPELOC_IMPL(CLASS, PARENT) \
1746427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::Visit##CLASS##TypeLoc(CLASS##TypeLoc TL) { \
1747427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return Visit##PARENT##Loc(TL); \
1748427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1749427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1750427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Complex, Type)
1751427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(ConstantArray, ArrayType)
1752427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(IncompleteArray, ArrayType)
1753427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(VariableArray, ArrayType)
1754427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(DependentSizedArray, ArrayType)
1755427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(DependentSizedExtVector, Type)
1756427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Vector, Type)
1757427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(ExtVector, VectorType)
1758427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(FunctionProto, FunctionType)
1759427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(FunctionNoProto, FunctionType)
1760427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Record, TagType)
1761427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Enum, TagType)
1762427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(SubstTemplateTypeParm, Type)
1763427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(SubstTemplateTypeParmPack, Type)
1764427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Auto, Type)
1765427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
17663064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenekbool CursorVisitor::VisitCXXRecordDecl(CXXRecordDecl *D) {
1767c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
1768c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1769c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1770c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
1771c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
17725e1cdac63c3d9c9b32fa41fa0b2d242a58a20d49John McCall  if (D->isCompleteDefinition()) {
17733064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    for (CXXRecordDecl::base_class_iterator I = D->bases_begin(),
17743064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek         E = D->bases_end(); I != E; ++I) {
17753064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      if (Visit(cxcursor::MakeCursorCXXBaseSpecifier(I, TU)))
17763064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek        return true;
17773064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
17783064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  }
17793064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
17803064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  return VisitTagDecl(D);
17813064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek}
17823064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
178309dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenekbool CursorVisitor::VisitAttributes(Decl *D) {
1784cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt  for (AttrVec::const_iterator i = D->attr_begin(), e = D->attr_end();
1785cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt       i != e; ++i)
1786cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt    if (Visit(MakeCXCursor(*i, D, TU)))
178709dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek        return true;
178809dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
178909dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  return false;
179009dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek}
179109dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
1792c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1793c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Data-recursive visitor methods.
1794c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1795c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
179628a719433411ef782b582946823bc648ddcc4533Ted Kremeneknamespace {
1797035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#define DEF_JOB(NAME, DATA, KIND)\
1798035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass NAME : public VisitorJob {\
1799035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:\
1800035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  NAME(DATA *d, CXCursor parent) : VisitorJob(parent, VisitorJob::KIND, d) {} \
1801035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) { return VJ->getKind() == KIND; }\
1802f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DATA *get() const { return static_cast<DATA*>(data[0]); }\
1803035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1804035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1805035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(StmtVisit, Stmt, StmtVisitKind)
1806035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(MemberExprParts, MemberExpr, MemberExprPartsKind)
1807e4979ccb5960608edce73f3b274eb7c2de15dac5Ted KremenekDEF_JOB(DeclRefExprParts, DeclRefExpr, DeclRefExprPartsKind)
1808035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(OverloadExprParts, OverloadExpr, OverloadExprPartsKind)
1809b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios KyrtzidisDEF_JOB(ExplicitTemplateArgsVisit, ASTTemplateArgumentListInfo,
181060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        ExplicitTemplateArgsVisitKind)
181194d96291cd041adc5731a2294828a9c20e450b74Douglas GregorDEF_JOB(SizeOfPackExprParts, SizeOfPackExpr, SizeOfPackExprPartsKind)
1812035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#undef DEF_JOB
1813035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1814035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass DeclVisit : public VisitorJob {
1815035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1816035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  DeclVisit(Decl *d, CXCursor parent, bool isFirst) :
1817035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::DeclVisitKind,
1818035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               d, isFirst ? (void*) 1 : (void*) 0) {}
1819035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
182082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    return VJ->getKind() == DeclVisitKind;
1821035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1822f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  Decl *get() const { return static_cast<Decl*>(data[0]); }
1823f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  bool isFirst() const { return data[1] ? true : false; }
1824035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1825035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass TypeLocVisit : public VisitorJob {
1826035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1827035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  TypeLocVisit(TypeLoc tl, CXCursor parent) :
1828035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::TypeLocVisitKind,
1829035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               tl.getType().getAsOpaquePtr(), tl.getOpaqueData()) {}
1830035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1831035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1832035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return VJ->getKind() == TypeLocVisitKind;
1833035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1834035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
183582f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek  TypeLoc get() const {
1836f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    QualType T = QualType::getFromOpaquePtr(data[0]);
1837f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return TypeLoc(T, data[1]);
1838035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1839035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1840035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1841ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekclass LabelRefVisit : public VisitorJob {
1842ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekpublic:
1843ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelRefVisit(LabelDecl *LD, SourceLocation labelLoc, CXCursor parent)
1844ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner    : VisitorJob(parent, VisitorJob::LabelRefVisitKind, LD,
1845dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 labelLoc.getPtrEncoding()) {}
1846ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek
1847ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1848ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek    return VJ->getKind() == VisitorJob::LabelRefVisitKind;
1849ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  }
1850ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelDecl *get() const { return static_cast<LabelDecl*>(data[0]); }
1851ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  SourceLocation getLoc() const {
1852dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin    return SourceLocation::getFromPtrEncoding(data[1]); }
1853f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek};
1854f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1855f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorclass NestedNameSpecifierLocVisit : public VisitorJob {
1856f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorpublic:
1857f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLocVisit(NestedNameSpecifierLoc Qualifier, CXCursor parent)
1858f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    : VisitorJob(parent, VisitorJob::NestedNameSpecifierLocVisitKind,
1859f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getNestedNameSpecifier(),
1860f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getOpaqueData()) { }
1861f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1862f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  static bool classof(const VisitorJob *VJ) {
1863f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return VJ->getKind() == VisitorJob::NestedNameSpecifierLocVisitKind;
1864f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1865f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1866f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLoc get() const {
1867f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return NestedNameSpecifierLoc(static_cast<NestedNameSpecifier*>(data[0]),
1868f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                                  data[1]);
1869f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1870f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor};
1871f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1872f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekclass DeclarationNameInfoVisit : public VisitorJob {
1873f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekpublic:
1874f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfoVisit(Stmt *S, CXCursor parent)
1875f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : VisitorJob(parent, VisitorJob::DeclarationNameInfoVisitKind, S) {}
1876f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1877f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return VJ->getKind() == VisitorJob::DeclarationNameInfoVisitKind;
1878f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1879f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfo get() const {
1880f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    Stmt *S = static_cast<Stmt*>(data[0]);
1881f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    switch (S->getStmtClass()) {
1882f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    default:
1883f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      llvm_unreachable("Unhandled Stmt");
1884ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor    case clang::Stmt::MSDependentExistsStmtClass:
1885ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor      return cast<MSDependentExistsStmt>(S)->getNameInfo();
1886f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::CXXDependentScopeMemberExprClass:
1887f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<CXXDependentScopeMemberExpr>(S)->getMemberNameInfo();
1888f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::DependentScopeDeclRefExprClass:
1889f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<DependentScopeDeclRefExpr>(S)->getNameInfo();
1890f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    }
1891f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1892ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek};
1893cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekclass MemberRefVisit : public VisitorJob {
1894cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekpublic:
1895cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  MemberRefVisit(FieldDecl *D, SourceLocation L, CXCursor parent)
1896cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    : VisitorJob(parent, VisitorJob::MemberRefVisitKind, D,
1897dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 L.getPtrEncoding()) {}
1898cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  static bool classof(const VisitorJob *VJ) {
1899cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return VJ->getKind() == VisitorJob::MemberRefVisitKind;
1900cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1901cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  FieldDecl *get() const {
1902cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return static_cast<FieldDecl*>(data[0]);
1903cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1904cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  SourceLocation getLoc() const {
1905cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1906cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1907cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek};
190828a719433411ef782b582946823bc648ddcc4533Ted Kremenekclass EnqueueVisitor : public StmtVisitor<EnqueueVisitor, void> {
190928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitorWorkList &WL;
191028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  CXCursor Parent;
191128a719433411ef782b582946823bc648ddcc4533Ted Kremenekpublic:
191228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueVisitor(VisitorWorkList &wl, CXCursor parent)
191328a719433411ef782b582946823bc648ddcc4533Ted Kremenek    : WL(wl), Parent(parent) {}
191428a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1915ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitAddrLabelExpr(AddrLabelExpr *E);
191673d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitBlockExpr(BlockExpr *B);
191728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1918083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  void VisitCompoundStmt(CompoundStmt *S);
191911b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) { /* Do nothing. */ }
1920ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  void VisitMSDependentExistsStmt(MSDependentExistsStmt *S);
1921f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
192211b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXNewExpr(CXXNewExpr *E);
19236d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
192428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
1925cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E);
192673d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E);
1927b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  void VisitCXXTypeidExpr(CXXTypeidExpr *E);
192855b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
19291e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  void VisitCXXUuidofExpr(CXXUuidofExpr *E);
1930e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  void VisitDeclRefExpr(DeclRefExpr *D);
1931035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void VisitDeclStmt(DeclStmt *S);
1932f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E);
1933cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitDesignatedInitExpr(DesignatedInitExpr *E);
193428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitExplicitCastExpr(ExplicitCastExpr *E);
193528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitForStmt(ForStmt *FS);
1936ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitGotoStmt(GotoStmt *GS);
193728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitIfStmt(IfStmt *If);
193828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitInitListExpr(InitListExpr *IE);
193928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitMemberExpr(MemberExpr *M);
1940cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitOffsetOfExpr(OffsetOfExpr *E);
194173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
194228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitObjCMessageExpr(ObjCMessageExpr *M);
194328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitOverloadExpr(OverloadExpr *E);
1944f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne  void VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E);
194528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitStmt(Stmt *S);
194628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitSwitchStmt(SwitchStmt *S);
194728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitWhileStmt(WhileStmt *W);
19482939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  void VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E);
19496ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  void VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E);
195021ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley  void VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E);
1951552622067dc45013d240f73952fece703f5e63bdJohn Wiegley  void VisitExpressionTraitExpr(ExpressionTraitExpr *E);
195228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U);
19539d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  void VisitVAArgExpr(VAArgExpr *E);
195494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  void VisitSizeOfPackExpr(SizeOfPackExpr *E);
1955ee8aff06f6a96214731de17b2cb6df407c6c1820Douglas Gregor
195628a719433411ef782b582946823bc648ddcc4533Ted Kremenekprivate:
1957f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void AddDeclarationNameInfo(Stmt *S);
1958f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  void AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier);
1959b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis  void AddExplicitTemplateArgs(const ASTTemplateArgumentListInfo *A);
1960cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void AddMemberRef(FieldDecl *D, SourceLocation L);
196128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddStmt(Stmt *S);
1962035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void AddDecl(Decl *D, bool isFirst = true);
196328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddTypeLoc(TypeSourceInfo *TI);
196428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void EnqueueChildren(Stmt *S);
196528a719433411ef782b582946823bc648ddcc4533Ted Kremenek};
196628a719433411ef782b582946823bc648ddcc4533Ted Kremenek} // end anonyous namespace
196728a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1968f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::AddDeclarationNameInfo(Stmt *S) {
1969f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // 'S' should always be non-null, since it comes from the
1970f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // statement we are visiting.
1971f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  WL.push_back(DeclarationNameInfoVisit(S, Parent));
1972f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1973f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1974f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorvoid
1975f3db29fff6a583ecda823cf909ab7737d8d30129Douglas GregorEnqueueVisitor::AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1976f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (Qualifier)
1977f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    WL.push_back(NestedNameSpecifierLocVisit(Qualifier, Parent));
1978f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor}
1979f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
198028a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddStmt(Stmt *S) {
198128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (S)
198228a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(StmtVisit(S, Parent));
198328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1984035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::AddDecl(Decl *D, bool isFirst) {
198528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (D)
1986035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    WL.push_back(DeclVisit(D, Parent, isFirst));
198728a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
198860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenekvoid EnqueueVisitor::
1989b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis  AddExplicitTemplateArgs(const ASTTemplateArgumentListInfo *A) {
199060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (A)
199160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    WL.push_back(ExplicitTemplateArgsVisit(
1992b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis                        const_cast<ASTTemplateArgumentListInfo*>(A), Parent));
199360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek}
1994cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::AddMemberRef(FieldDecl *D, SourceLocation L) {
1995cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  if (D)
1996cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    WL.push_back(MemberRefVisit(D, L, Parent));
1997cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
199828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddTypeLoc(TypeSourceInfo *TI) {
199928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (TI)
200028a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(TypeLocVisit(TI->getTypeLoc(), Parent));
200128a719433411ef782b582946823bc648ddcc4533Ted Kremenek }
200228a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::EnqueueChildren(Stmt *S) {
2003a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  unsigned size = WL.size();
20047502c1d3ce8bb97bcc4f7bebef507040bd93b26fJohn McCall  for (Stmt::child_range Child = S->children(); Child; ++Child) {
200528a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(*Child);
2006a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  }
2007a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  if (size == WL.size())
2008a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek    return;
2009a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // Now reverse the entries we just added.  This will match the DFS
2010a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // ordering performed by the worklist.
2011a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
2012a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  std::reverse(I, E);
2013a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek}
2014ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitAddrLabelExpr(AddrLabelExpr *E) {
2015ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(E->getLabel(), E->getLabelLoc(), Parent));
2016ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
201773d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitBlockExpr(BlockExpr *B) {
201873d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddDecl(B->getBlockDecl());
201973d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
202028a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
202128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
202228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeSourceInfo());
202328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2024083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenekvoid EnqueueVisitor::VisitCompoundStmt(CompoundStmt *S) {
2025083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  for (CompoundStmt::reverse_body_iterator I = S->body_rbegin(),
2026083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek        E = S->body_rend(); I != E; ++I) {
2027083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek    AddStmt(*I);
2028083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  }
202911b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
2030f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::
2031ba0513de93d2fab6db5ab30b6927209fcc883078Douglas GregorVisitMSDependentExistsStmt(MSDependentExistsStmt *S) {
2032ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  AddStmt(S->getSubStmt());
2033ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  AddDeclarationNameInfo(S);
2034ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = S->getQualifierLoc())
2035ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
2036ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor}
2037ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregor
2038ba0513de93d2fab6db5ab30b6927209fcc883078Douglas Gregorvoid EnqueueVisitor::
2039f64d80306144f978148ba92f36f7cea7b671dd34Ted KremenekVisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E) {
2040f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
2041f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
20427c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
20437c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
2044f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  if (!E->isImplicitAccess())
2045f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    AddStmt(E->getBase());
2046f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
204711b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenekvoid EnqueueVisitor::VisitCXXNewExpr(CXXNewExpr *E) {
204811b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the initializer or constructor arguments.
204911b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumConstructorArgs(); I > 0; --I)
205011b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getConstructorArg(I-1));
205111b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the array size, if any.
205211b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddStmt(E->getArraySize());
205311b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the allocated type.
205411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddTypeLoc(E->getAllocatedTypeSourceInfo());
205511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the placement arguments.
205611b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumPlacementArgs(); I > 0; --I)
205711b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getPlacementArg(I-1));
205811b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
205928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *CE) {
20608b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek  for (unsigned I = CE->getNumArgs(); I > 1 /* Yes, this is 1 */; --I)
20618b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek    AddStmt(CE->getArg(I-1));
206228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getCallee());
206328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getArg(0));
206428a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2065cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
2066cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the name of the type being destroyed.
2067cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getDestroyedTypeInfo());
2068cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the scope type that looks disturbingly like the nested-name-specifier
2069cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // but isn't.
2070cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getScopeTypeInfo());
2071cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the nested-name-specifier.
2072f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
2073f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
2074cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit base expression.
2075cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getBase());
2076cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
20776d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenekvoid EnqueueVisitor::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
20786d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
20796d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
208073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
208173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  EnqueueChildren(E);
208273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
208373d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
2084b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenekvoid EnqueueVisitor::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
2085b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  EnqueueChildren(E);
2086b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  if (E->isTypeOperand())
2087b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
2088b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek}
208955b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek
209055b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenekvoid EnqueueVisitor::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr
209155b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek                                                     *E) {
209255b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  EnqueueChildren(E);
209355b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
209455b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek}
20951e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenekvoid EnqueueVisitor::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
20961e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  EnqueueChildren(E);
20971e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  if (E->isTypeOperand())
20981e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
20991e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek}
2100e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenekvoid EnqueueVisitor::VisitDeclRefExpr(DeclRefExpr *DR) {
210160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (DR->hasExplicitTemplateArgs()) {
210260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    AddExplicitTemplateArgs(&DR->getExplicitTemplateArgs());
210360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  }
2104e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  WL.push_back(DeclRefExprParts(DR, Parent));
2105e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek}
2106f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
2107f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
2108f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
210900cf3cc2718671aa48e8da264a523b0058a8591eDouglas Gregor  AddNestedNameSpecifierLoc(E->getQualifierLoc());
2110f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
2111035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::VisitDeclStmt(DeclStmt *S) {
2112035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  unsigned size = WL.size();
2113035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  bool isFirst = true;
2114035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  for (DeclStmt::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
2115035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek       D != DEnd; ++D) {
2116035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    AddDecl(*D, isFirst);
2117035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    isFirst = false;
2118035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
2119035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  if (size == WL.size())
2120035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return;
2121035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // Now reverse the entries we just added.  This will match the DFS
2122035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // ordering performed by the worklist.
2123035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
2124035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  std::reverse(I, E);
2125035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek}
2126cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
2127cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getInit());
2128cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  typedef DesignatedInitExpr::Designator Designator;
2129cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (DesignatedInitExpr::reverse_designators_iterator
2130cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D = E->designators_rbegin(), DEnd = E->designators_rend();
2131cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D != DEnd; ++D) {
2132cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isFieldDesignator()) {
2133cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      if (FieldDecl *Field = D->getField())
2134cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        AddMemberRef(Field, D->getFieldLoc());
2135cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
2136cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
2137cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isArrayDesignator()) {
2138cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getArrayIndex(*D));
2139cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
2140cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
2141cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    assert(D->isArrayRangeDesignator() && "Unknown designator kind");
2142cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeEnd(*D));
2143cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeStart(*D));
2144cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
2145cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
214628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitExplicitCastExpr(ExplicitCastExpr *E) {
214728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
214828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeInfoAsWritten());
214928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
215028a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitForStmt(ForStmt *FS) {
215128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getBody());
215228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInc());
215328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getCond());
215428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(FS->getConditionVariable());
215528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInit());
215628a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2157ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitGotoStmt(GotoStmt *GS) {
2158ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(GS->getLabel(), GS->getLabelLoc(), Parent));
2159ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
216028a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitIfStmt(IfStmt *If) {
216128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getElse());
216228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getThen());
216328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getCond());
216428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(If->getConditionVariable());
216528a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
216628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitInitListExpr(InitListExpr *IE) {
216728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  // We care about the syntactic form of the initializer list, only.
216828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (InitListExpr *Syntactic = IE->getSyntacticForm())
216928a719433411ef782b582946823bc648ddcc4533Ted Kremenek    IE = Syntactic;
217028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(IE);
217128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
217228a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitMemberExpr(MemberExpr *M) {
217389629a746019a42797495b091711a1d68467e88aDouglas Gregor  WL.push_back(MemberExprParts(M, Parent));
217489629a746019a42797495b091711a1d68467e88aDouglas Gregor
217589629a746019a42797495b091711a1d68467e88aDouglas Gregor  // If the base of the member access expression is an implicit 'this', don't
217689629a746019a42797495b091711a1d68467e88aDouglas Gregor  // visit it.
217789629a746019a42797495b091711a1d68467e88aDouglas Gregor  // FIXME: If we ever want to show these implicit accesses, this will be
217889629a746019a42797495b091711a1d68467e88aDouglas Gregor  // unfortunate. However, clang_getCursor() relies on this behavior.
217975e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor  if (!M->isImplicitAccess())
218075e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor    AddStmt(M->getBase());
218128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
218273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
218373d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getEncodedTypeSourceInfo());
218473d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
218528a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitObjCMessageExpr(ObjCMessageExpr *M) {
218628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(M);
218728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(M->getClassReceiverTypeInfo());
218828a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2189cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitOffsetOfExpr(OffsetOfExpr *E) {
2190cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the components of the offsetof expression.
2191cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (unsigned N = E->getNumComponents(), I = N; I > 0; --I) {
2192cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    typedef OffsetOfExpr::OffsetOfNode OffsetOfNode;
2193cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    const OffsetOfNode &Node = E->getComponent(I-1);
2194cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    switch (Node.getKind()) {
2195cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Array:
2196cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getIndexExpr(Node.getArrayExprIndex()));
2197cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2198cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Field:
219906dec892b5300b43263d25c5476b506c9d6cfbadAbramo Bagnara      AddMemberRef(Node.getField(), Node.getSourceRange().getEnd());
2200cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2201cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Identifier:
2202cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Base:
2203cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
2204cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
2205cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
2206cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the type into which we're computing the offset.
2207cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
2208cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
220928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitOverloadExpr(OverloadExpr *E) {
221060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
22116045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek  WL.push_back(OverloadExprParts(E, Parent));
22126045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek}
2213f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbournevoid EnqueueVisitor::VisitUnaryExprOrTypeTraitExpr(
2214f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne                                              UnaryExprOrTypeTraitExpr *E) {
22156d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  EnqueueChildren(E);
22166d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  if (E->isArgumentType())
22176d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek    AddTypeLoc(E->getArgumentTypeInfo());
22186d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
221928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitStmt(Stmt *S) {
222028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(S);
222128a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
222228a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitSwitchStmt(SwitchStmt *S) {
222328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getBody());
222428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getCond());
222528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(S->getConditionVariable());
222628a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2227fafa75aebadef8d6b44a920e3f40529f150a5574Ted Kremenek
222828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitWhileStmt(WhileStmt *W) {
222928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getBody());
223028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getCond());
223128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(W->getConditionVariable());
223228a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
223321ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley
22342939b6f356161f572712d4d6310b65f9599e3675Ted Kremenekvoid EnqueueVisitor::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
22352939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  AddTypeLoc(E->getQueriedTypeSourceInfo());
22362939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek}
22376ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
22386ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichetvoid EnqueueVisitor::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
22396ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  AddTypeLoc(E->getRhsTypeSourceInfo());
22400a03a3f98b14006a54bcac9e8908a7c9f50e519fFrancois Pichet  AddTypeLoc(E->getLhsTypeSourceInfo());
22416ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet}
22426ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
224321ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegleyvoid EnqueueVisitor::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
224421ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley  AddTypeLoc(E->getQueriedTypeSourceInfo());
224521ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley}
224621ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley
2247552622067dc45013d240f73952fece703f5e63bdJohn Wiegleyvoid EnqueueVisitor::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
2248552622067dc45013d240f73952fece703f5e63bdJohn Wiegley  EnqueueChildren(E);
2249552622067dc45013d240f73952fece703f5e63bdJohn Wiegley}
2250552622067dc45013d240f73952fece703f5e63bdJohn Wiegley
225128a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U) {
225228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitOverloadExpr(U);
225328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (!U->isImplicitAccess())
225428a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(U->getBase());
225528a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
22569d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenekvoid EnqueueVisitor::VisitVAArgExpr(VAArgExpr *E) {
22579d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddStmt(E->getSubExpr());
22589d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddTypeLoc(E->getWrittenTypeInfo());
22599d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek}
226094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregorvoid EnqueueVisitor::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
226194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  WL.push_back(SizeOfPackExprParts(E, Parent));
226294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor}
22636045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek
2264c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekvoid CursorVisitor::EnqueueWorkList(VisitorWorkList &WL, Stmt *S) {
2265aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  EnqueueVisitor(WL, MakeCXCursor(S, StmtParent, TU,RegionOfInterest)).Visit(S);
2266c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2267c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2268c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::IsInRegionOfInterest(CXCursor C) {
2269c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  if (RegionOfInterest.isValid()) {
2270c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SourceRange Range = getRawCursorExtent(C);
2271c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    if (Range.isInvalid() || CompareRegionOfInterest(Range))
2272c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      return false;
2273c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2274c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return true;
2275c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2276c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2277c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::RunVisitorWorkList(VisitorWorkList &WL) {
2278c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  while (!WL.empty()) {
2279c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Dequeue the worklist item.
228082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    VisitorJob LI = WL.back();
228182f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    WL.pop_back();
228282f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek
2283c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Set the Parent field, then back to its old value once we're done.
2284c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SetParentRAII SetParent(Parent, StmtParent, LI.getParent());
2285c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2286c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    switch (LI.getKind()) {
2287f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      case VisitorJob::DeclVisitKind: {
228882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Decl *D = cast<DeclVisit>(&LI)->get();
2289f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        if (!D)
2290f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek          continue;
2291f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2292f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // For now, perform default visitation for Decls.
2293aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis        if (Visit(MakeCXCursor(D, TU, RegionOfInterest,
2294aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis                               cast<DeclVisit>(&LI)->isFirst())))
2295f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek            return true;
2296f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2297f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        continue;
2298f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      }
229960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      case VisitorJob::ExplicitTemplateArgsVisitKind: {
2300b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis        const ASTTemplateArgumentListInfo *ArgList =
230160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          cast<ExplicitTemplateArgsVisit>(&LI)->get();
230260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        for (const TemplateArgumentLoc *Arg = ArgList->getTemplateArgs(),
230360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               *ArgEnd = Arg + ArgList->NumTemplateArgs;
230460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               Arg != ArgEnd; ++Arg) {
230560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          if (VisitTemplateArgumentLoc(*Arg))
230660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek            return true;
230760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        }
230860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        continue;
230960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      }
2310cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      case VisitorJob::TypeLocVisitKind: {
2311cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        // Perform default visitation for TypeLocs.
231282f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        if (Visit(cast<TypeLocVisit>(&LI)->get()))
2313cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek          return true;
2314cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        continue;
2315cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      }
2316ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      case VisitorJob::LabelRefVisitKind: {
2317ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner        LabelDecl *LS = cast<LabelRefVisit>(&LI)->get();
2318e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        if (LabelStmt *stmt = LS->getStmt()) {
2319e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          if (Visit(MakeCursorLabelRef(stmt, cast<LabelRefVisit>(&LI)->getLoc(),
2320e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek                                       TU))) {
2321e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek            return true;
2322e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          }
2323e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        }
2324ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek        continue;
2325ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      }
232647695c8ad8424851f62e0d4a983b45b15daee1c5Ted Kremenek
2327f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      case VisitorJob::NestedNameSpecifierLocVisitKind: {
2328f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        NestedNameSpecifierLocVisit *V = cast<NestedNameSpecifierLocVisit>(&LI);
2329f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        if (VisitNestedNameSpecifierLoc(V->get()))
2330f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor          return true;
2331f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        continue;
2332f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      }
2333f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2334f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      case VisitorJob::DeclarationNameInfoVisitKind: {
2335f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        if (VisitDeclarationNameInfo(cast<DeclarationNameInfoVisit>(&LI)
2336f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                                     ->get()))
2337f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek          return true;
2338f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        continue;
2339f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      }
2340cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      case VisitorJob::MemberRefVisitKind: {
2341cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        MemberRefVisit *V = cast<MemberRefVisit>(&LI);
2342cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (Visit(MakeCursorMemberRef(V->get(), V->getLoc(), TU)))
2343cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          return true;
2344cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        continue;
2345cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      }
2346c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::StmtVisitKind: {
234782f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Stmt *S = cast<StmtVisit>(&LI)->get();
23488c269ac75569454a049385b1246140db5f2b6faaTed Kremenek        if (!S)
23498c269ac75569454a049385b1246140db5f2b6faaTed Kremenek          continue;
23508c269ac75569454a049385b1246140db5f2b6faaTed Kremenek
2351f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // Update the current cursor.
2352aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis        CXCursor Cursor = MakeCXCursor(S, StmtParent, TU, RegionOfInterest);
2353cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (!IsInRegionOfInterest(Cursor))
2354cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          continue;
2355cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        switch (Visitor(Cursor, Parent, ClientData)) {
2356cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Break: return true;
2357cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Continue: break;
2358cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Recurse:
2359cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek            EnqueueWorkList(WL, S);
236082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek            break;
2361c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
236282f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        continue;
2363c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2364c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::MemberExprPartsKind: {
2365c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Handle the other pieces in the MemberExpr besides the base.
236682f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        MemberExpr *M = cast<MemberExprParts>(&LI)->get();
2367c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2368c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the nested-name-specifier
236940d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = M->getQualifierLoc())
237040d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2371c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            return true;
2372c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2373c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the declaration name.
2374c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (VisitDeclarationNameInfo(M->getMemberNameInfo()))
2375c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          return true;
2376c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2377c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the explicitly-specified template arguments, if any.
2378c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (M->hasExplicitTemplateArgs()) {
2379c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          for (const TemplateArgumentLoc *Arg = M->getTemplateArgs(),
2380c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               *ArgEnd = Arg + M->getNumTemplateArgs();
2381c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               Arg != ArgEnd; ++Arg) {
2382c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            if (VisitTemplateArgumentLoc(*Arg))
2383c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek              return true;
2384c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          }
2385c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
2386c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        continue;
2387c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2388e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      case VisitorJob::DeclRefExprPartsKind: {
238982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        DeclRefExpr *DR = cast<DeclRefExprParts>(&LI)->get();
2390e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit nested-name-specifier, if present.
239140d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = DR->getQualifierLoc())
239240d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2393e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek            return true;
2394e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit declaration name.
2395e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        if (VisitDeclarationNameInfo(DR->getNameInfo()))
2396e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek          return true;
2397e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        continue;
2398e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      }
23996045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      case VisitorJob::OverloadExprPartsKind: {
240082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        OverloadExpr *O = cast<OverloadExprParts>(&LI)->get();
24016045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the nested-name-specifier.
24024c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = O->getQualifierLoc())
24034c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
24046045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek            return true;
24056045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the declaration name.
24066045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (VisitDeclarationNameInfo(O->getNameInfo()))
24076045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
24086045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the overloaded declaration reference.
24096045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (Visit(MakeCursorOverloadedDeclRef(O, TU)))
24106045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
24116045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        continue;
24126045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      }
241394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      case VisitorJob::SizeOfPackExprPartsKind: {
241494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        SizeOfPackExpr *E = cast<SizeOfPackExprParts>(&LI)->get();
241594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        NamedDecl *Pack = E->getPack();
241694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTypeParmDecl>(Pack)) {
241794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTypeRef(cast<TemplateTypeParmDecl>(Pack),
241894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                      E->getPackLoc(), TU)))
241994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
242094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
242194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
242294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
242394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
242494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTemplateParmDecl>(Pack)) {
242594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTemplateRef(cast<TemplateTemplateParmDecl>(Pack),
242694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                          E->getPackLoc(), TU)))
242794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
242894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
242994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
243094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
243194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
243294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // Non-type template parameter packs and function parameter packs are
243394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // treated like DeclRefExpr cursors.
243494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        continue;
243594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      }
2436c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    }
2437c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2438c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return false;
2439c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2440c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2441cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekbool CursorVisitor::Visit(Stmt *S) {
2442d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  VisitorWorkList *WL = 0;
2443d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  if (!WorkListFreeList.empty()) {
2444d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = WorkListFreeList.back();
2445d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL->clear();
2446d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListFreeList.pop_back();
2447d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2448d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  else {
2449d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = new VisitorWorkList();
2450d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListCache.push_back(WL);
2451d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2452d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  EnqueueWorkList(*WL, S);
2453d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  bool result = RunVisitorWorkList(*WL);
2454d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  WorkListFreeList.push_back(WL);
2455d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  return result;
2456c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2457c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
245848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichetnamespace {
245948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichettypedef llvm::SmallVector<SourceRange, 4> RefNamePieces;
246048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois PichetRefNamePieces buildPieces(unsigned NameFlags, bool IsMemberRefExpr,
246148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                          const DeclarationNameInfo &NI,
246248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                          const SourceRange &QLoc,
2463b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis                          const ASTTemplateArgumentListInfo *TemplateArgs = 0){
246448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantQualifier = NameFlags & CXNameRange_WantQualifier;
246548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantTemplateArgs = NameFlags & CXNameRange_WantTemplateArgs;
246648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantSinglePiece = NameFlags & CXNameRange_WantSinglePiece;
246748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
246848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const DeclarationName::NameKind Kind = NI.getName().getNameKind();
246948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
247048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  RefNamePieces Pieces;
247148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
247248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantQualifier && QLoc.isValid())
247348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(QLoc);
247448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
247548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (Kind != DeclarationName::CXXOperatorName || IsMemberRefExpr)
247648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(NI.getLoc());
247748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
247848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantTemplateArgs && TemplateArgs)
247948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceRange(TemplateArgs->LAngleLoc,
248048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                                 TemplateArgs->RAngleLoc));
248148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
248248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (Kind == DeclarationName::CXXOperatorName) {
248348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceLocation::getFromRawEncoding(
248448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                       NI.getInfo().CXXOperatorName.BeginOpNameLoc));
248548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceLocation::getFromRawEncoding(
248648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                       NI.getInfo().CXXOperatorName.EndOpNameLoc));
248748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  }
248848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
248948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantSinglePiece) {
249048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    SourceRange R(Pieces.front().getBegin(), Pieces.back().getEnd());
249148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.clear();
249248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(R);
249348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  }
249448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
249548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  return Pieces;
249648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet}
249748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet}
249848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
2499c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2500c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Misc. API hooks.
2501c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2502c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
25038c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic llvm::sys::Mutex EnableMultithreadingMutex;
25048c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic bool EnabledMultithreading;
25058c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
25065e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramerextern "C" {
25070a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas GregorCXIndex clang_createIndex(int excludeDeclarationsFromPCH,
25080a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                          int displayDiagnostics) {
250948615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // Disable pretty stack trace functionality, which will otherwise be a very
251048615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // poor citizen of the world and set up all sorts of signal handlers.
251148615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  llvm::DisablePrettyStackTrace = true;
251248615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar
2513c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // We use crash recovery to make some of our APIs more reliable, implicitly
2514c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // enable it.
2515c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  llvm::CrashRecoveryContext::Enable();
2516c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar
25178c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  // Enable support for multithreading in LLVM.
25188c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  {
25198c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    llvm::sys::ScopedLock L(EnableMultithreadingMutex);
25208c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    if (!EnabledMultithreading) {
25218c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      llvm::llvm_start_multithreaded();
25228c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      EnabledMultithreading = true;
25238c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    }
25248c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  }
25258c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
2526a030b7cf5e6aad5889b1b662b6979840bc75f87fDouglas Gregor  CIndexer *CIdxr = new CIndexer();
2527e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  if (excludeDeclarationsFromPCH)
2528e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    CIdxr->setOnlyLocalDecls();
25290a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  if (displayDiagnostics)
25300a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    CIdxr->setDisplayDiagnostics();
2531e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  return CIdxr;
2532600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2533600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
25349ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeIndex(CXIndex CIdx) {
25352b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (CIdx)
25362b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    delete static_cast<CIndexer *>(CIdx);
25372bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
25382bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff
2539d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenekvoid clang_toggleCrashRecovery(unsigned isEnabled) {
2540d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  if (isEnabled)
2541d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Enable();
2542d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  else
2543d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Disable();
2544d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek}
2545d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek
25469ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
2547a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                              const char *ast_filename) {
25482b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
25492b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    return 0;
2550f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
25517d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2552389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOptions FileSystemOpts;
2553389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOpts.WorkingDir = CXXIdx->getWorkingDirectory();
25540d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2555d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie  llvm::IntrusiveRefCntPtr<DiagnosticsEngine> Diags;
2556a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *TU = ASTUnit::LoadFromASTFile(ast_filename, Diags, FileSystemOpts,
2557a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  CXXIdx->getOnlyLocalDecls(),
2558a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  0, 0, true);
2559a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return MakeCXTranslationUnit(TU);
2560600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2561600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2562b1c031be513705d924038f497279b9b599868ba1Douglas Gregorunsigned clang_defaultEditingTranslationUnitOptions() {
25632a2c50b330e7754499f42173616a36865b5f313bDouglas Gregor  return CXTranslationUnit_PrecompiledPreamble |
2564b5af843a20e237ad1a13ad66a867e200695b8c8eDouglas Gregor         CXTranslationUnit_CacheCompletionResults;
2565b1c031be513705d924038f497279b9b599868ba1Douglas Gregor}
2566b1c031be513705d924038f497279b9b599868ba1Douglas Gregor
25679ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit
25689ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarclang_createTranslationUnitFromSourceFile(CXIndex CIdx,
25699ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          const char *source_filename,
25709ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          int num_command_line_args,
25712ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                          const char * const *command_line_args,
25724db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          unsigned num_unsaved_files,
2573a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                          struct CXUnsavedFile *unsaved_files) {
2574dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor  unsigned Options = CXTranslationUnit_DetailedPreprocessingRecord |
2575ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth                     CXTranslationUnit_NestedMacroExpansions;
25765a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor  return clang_parseTranslationUnit(CIdx, source_filename,
25775a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    command_line_args, num_command_line_args,
25785a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    unsaved_files, num_unsaved_files,
2579dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor                                    Options);
25805a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor}
258119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
258219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbarstruct ParseTranslationUnitInfo {
258319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx;
258419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename;
25852ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char *const *command_line_args;
258619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args;
258719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
258819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files;
258919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options;
259019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXTranslationUnit result;
259119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar};
2592b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_parseTranslationUnit_Impl(void *UserData) {
259319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo *PTUI =
259419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    static_cast<ParseTranslationUnitInfo*>(UserData);
259519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx = PTUI->CIdx;
259619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename = PTUI->source_filename;
25972ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char * const *command_line_args = PTUI->command_line_args;
259819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args = PTUI->num_command_line_args;
259919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files = PTUI->unsaved_files;
260019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files = PTUI->num_unsaved_files;
260119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options = PTUI->options;
260219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  PTUI->result = 0;
26035a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor
26042b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
260519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return;
2606f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2607e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2608e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff
260944c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  bool PrecompilePreamble = options & CXTranslationUnit_PrecompiledPreamble;
2610467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor  // FIXME: Add a flag for modules.
2611467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor  TranslationUnitKind TUKind
2612467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor    = (options & CXTranslationUnit_Incomplete)? TU_Prefix : TU_Complete;
261387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  bool CacheCodeCompetionResults
261487c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    = options & CXTranslationUnit_CacheCompletionResults;
261587c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
26165352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  // Configure the diagnostics.
26175352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  DiagnosticOptions DiagOpts;
2618d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie  llvm::IntrusiveRefCntPtr<DiagnosticsEngine>
261925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Diags(CompilerInstance::createDiagnostics(DiagOpts, num_command_line_args,
262025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                                command_line_args));
262125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
262225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
2623d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie  llvm::CrashRecoveryContextCleanupRegistrar<DiagnosticsEngine,
2624d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie    llvm::CrashRecoveryContextReleaseRefCleanup<DiagnosticsEngine> >
262525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    DiagCleanup(Diags.getPtr());
262625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
262725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<ASTUnit::RemappedFile> >
262825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
262925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
263025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
263125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
263225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
2633f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
26344db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
26355f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2636f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    const llvm::MemoryBuffer *Buffer
2637a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
263825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
263925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
26404db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  }
2641f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
264225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<const char *> >
264325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args(new std::vector<const char*>());
264425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
264525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this method.
264625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<std::vector<const char*> >
264725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    ArgsCleanup(Args.get());
264825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
264952ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // Since the Clang C library is primarily used by batch tools dealing with
265052ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // (often very broken) source code, where spell-checking can have a
265152ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // significant negative impact on performance (particularly when
265252ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // precompiled headers are involved), we disable it by default.
2653b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  // Only do this if we haven't found a spell-checking-related argument.
2654b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  bool FoundSpellCheckingArgument = false;
2655b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  for (int I = 0; I != num_command_line_args; ++I) {
2656b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (strcmp(command_line_args[I], "-fno-spell-checking") == 0 ||
2657b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        strcmp(command_line_args[I], "-fspell-checking") == 0) {
2658b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      FoundSpellCheckingArgument = true;
2659b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      break;
2660e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    }
2661b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  }
2662b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  if (!FoundSpellCheckingArgument)
266325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-fno-spell-checking");
2664b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
266525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  Args->insert(Args->end(), command_line_args,
266625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek               command_line_args + num_command_line_args);
2667d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2668c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // The 'source_filename' argument is optional.  If the caller does not
2669c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // specify it then it is assumed that the source file is specified
2670c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // in the actual argument list.
2671c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // Put the source file after command_line_args otherwise if '-x' flag is
2672c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // present it will be unused.
2673c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  if (source_filename)
267425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back(source_filename);
2675c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis
267644c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  // Do we need the detailed preprocessing record?
2677ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth  bool NestedMacroExpansions = false;
267844c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  if (options & CXTranslationUnit_DetailedPreprocessingRecord) {
267925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-Xclang");
268025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-detailed-preprocessing-record");
2681ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth    NestedMacroExpansions
2682ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth      = (options & CXTranslationUnit_NestedMacroExpansions);
268344c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  }
268444c181aec37789f25f6c15543c164416f72e562aDouglas Gregor
2685026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  unsigned NumErrors = Diags->getClient()->getNumErrors();
2686b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  llvm::OwningPtr<ASTUnit> Unit(
26874ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek    ASTUnit::LoadFromCommandLine(Args->size() ? &(*Args)[0] : 0
26884ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 /* vector::data() not portable */,
26894ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 Args->size() ? (&(*Args)[0] + Args->size()) :0,
2690b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 Diags,
2691b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getClangResourcesPath(),
2692b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getOnlyLocalDecls(),
2693e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                                 /*CaptureDiagnostics=*/true,
26944ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 RemappedFiles->size() ? &(*RemappedFiles)[0]:0,
269525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                 RemappedFiles->size(),
2696299a4a967b02c9f0d0d94ad8560e3ced893f9116Argyrios Kyrtzidis                                 /*RemappedFilesKeepOriginalName=*/true,
2697b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 PrecompilePreamble,
2698467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor                                 TUKind,
269999ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                 CacheCodeCompetionResults,
2700ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth                                 NestedMacroExpansions));
2701b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
2702026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  if (NumErrors != Diags->getClient()->getNumErrors()) {
2703b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    // Make sure to check that 'Unit' is non-NULL.
2704b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (CXXIdx->getDisplayDiagnostics() && Unit.get()) {
2705b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
2706b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                      DEnd = Unit->stored_diag_end();
2707b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor           D != DEnd; ++D) {
2708b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        CXStoredDiagnostic Diag(*D, Unit->getASTContext().getLangOptions());
2709b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        CXString Msg = clang_formatDiagnostic(&Diag,
2710b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                    clang_defaultDiagnosticDisplayOptions());
2711b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        fprintf(stderr, "%s\n", clang_getCString(Msg));
2712b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        clang_disposeString(Msg);
2713b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      }
2714274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor#ifdef LLVM_ON_WIN32
2715b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // On Windows, force a flush, since there may be multiple copies of
2716b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // stderr and stdout in the file system, all with different buffers
2717b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // but writing to the same device.
2718b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      fflush(stderr);
2719b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor#endif
2720b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    }
2721a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  }
2722d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2723a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  PTUI->result = MakeCXTranslationUnit(Unit.take());
272419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar}
272519ffd492a31a25fb691098bf79f317e5f3edf177Daniel DunbarCXTranslationUnit clang_parseTranslationUnit(CXIndex CIdx,
272619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             const char *source_filename,
27272ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                         const char * const *command_line_args,
272819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             int num_command_line_args,
27299e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                            struct CXUnsavedFile *unsaved_files,
273019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned num_unsaved_files,
273119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned options) {
273219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo PTUI = { CIdx, source_filename, command_line_args,
27339e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_command_line_args, unsaved_files,
27349e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_unsaved_files, options, 0 };
273519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  llvm::CrashRecoveryContext CRC;
273619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
2737bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_parseTranslationUnit_Impl, &PTUI)) {
273860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "libclang: crash detected during parsing: {\n");
273960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'source_filename' : '%s'\n", source_filename);
274060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'command_line_args' : [");
274160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (int i = 0; i != num_command_line_args; ++i) {
274260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
274360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
274460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "'%s'", command_line_args[i]);
274560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
274660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
274760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'unsaved_files' : [");
274860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (unsigned i = 0; i != num_unsaved_files; ++i) {
274960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
275060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
275160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "('%s', '...', %ld)", unsaved_files[i].Filename,
275260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar              unsaved_files[i].Length);
275360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
275460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
275560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'options' : %d,\n", options);
275660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "}\n");
275760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar
275819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return 0;
27596df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
27606df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(PTUI.result);
276119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  }
27626df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
276319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  return PTUI.result;
27645b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff}
27655b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff
27661999844e7a18786e61e619e1dc6c789827541863Douglas Gregorunsigned clang_defaultSaveOptions(CXTranslationUnit TU) {
27671999844e7a18786e61e619e1dc6c789827541863Douglas Gregor  return CXSaveTranslationUnit_None;
27681999844e7a18786e61e619e1dc6c789827541863Douglas Gregor}
27691999844e7a18786e61e619e1dc6c789827541863Douglas Gregor
27701999844e7a18786e61e619e1dc6c789827541863Douglas Gregorint clang_saveTranslationUnit(CXTranslationUnit TU, const char *FileName,
27711999844e7a18786e61e619e1dc6c789827541863Douglas Gregor                              unsigned options) {
27727ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  if (!TU)
277339c411fa229b2a6747b92f945d1702ee674d3470Douglas Gregor    return CXSaveError_InvalidTU;
27747ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor
277539c411fa229b2a6747b92f945d1702ee674d3470Douglas Gregor  CXSaveError result = static_cast<ASTUnit *>(TU->TUData)->Save(FileName);
27766df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  if (getenv("LIBCLANG_RESOURCE_USAGE"))
27776df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(TU);
27786df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  return result;
27797ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor}
278019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
27819ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
2782ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  if (CTUnit) {
2783ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // If the translation unit has been marked as unsafe to free, just discard
2784ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // it.
2785a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    if (static_cast<ASTUnit *>(CTUnit->TUData)->isUnsafeToFree())
2786ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar      return;
2787ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2788a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete static_cast<ASTUnit *>(CTUnit->TUData);
2789a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    disposeCXStringPool(CTUnit->StringPool);
2790a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete CTUnit;
2791ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  }
27922bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
27930d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2794e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregorunsigned clang_defaultReparseOptions(CXTranslationUnit TU) {
2795e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor  return CXReparse_None;
2796e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor}
2797e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor
2798ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarstruct ReparseTranslationUnitInfo {
2799ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU;
2800ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files;
2801ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
2802ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options;
2803ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  int result;
2804ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar};
2805593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2806b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_reparseTranslationUnit_Impl(void *UserData) {
2807ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo *RTUI =
2808ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    static_cast<ReparseTranslationUnitInfo*>(UserData);
2809ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU = RTUI->TU;
2810ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files = RTUI->num_unsaved_files;
2811ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files = RTUI->unsaved_files;
2812ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options = RTUI->options;
2813ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  (void) options;
2814ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  RTUI->result = 1;
2815ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2816abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  if (!TU)
2817ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return;
2818593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2819a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
2820593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2821abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
282225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<ASTUnit::RemappedFile> >
282325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
282425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
282525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
282625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
282725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
282825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
2829abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
28305f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2831abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor    const llvm::MemoryBuffer *Buffer
28321abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
283325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
283425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
2835abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  }
2836abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
28374ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek  if (!CXXUnit->Reparse(RemappedFiles->size() ? &(*RemappedFiles)[0] : 0,
28384ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                        RemappedFiles->size()))
2839593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor    RTUI->result = 0;
2840abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor}
2841593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2842ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarint clang_reparseTranslationUnit(CXTranslationUnit TU,
2843ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned num_unsaved_files,
2844ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 struct CXUnsavedFile *unsaved_files,
2845ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned options) {
2846ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo RTUI = { TU, num_unsaved_files, unsaved_files,
2847ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                      options, 0 };
28488c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis
2849e7de9b4a1f4a15620ab15bc8159018df7d54080aArgyrios Kyrtzidis  if (getenv("LIBCLANG_NOTHREADS")) {
28508c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis    clang_reparseTranslationUnit_Impl(&RTUI);
28518c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis    return RTUI.result;
28528c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis  }
28538c4b47eff0848dfd80854d70cb84b23b185828d4Argyrios Kyrtzidis
2854ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  llvm::CrashRecoveryContext CRC;
2855ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2856bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_reparseTranslationUnit_Impl, &RTUI)) {
2857b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbar    fprintf(stderr, "libclang: crash detected during reparsing\n");
2858a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    static_cast<ASTUnit *>(TU->TUData)->setUnsafeToFree(true);
2859ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return 1;
28606df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  } else if (getenv("LIBCLANG_RESOURCE_USAGE"))
28616df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(TU);
28621dfb26af4d6aa4f7818e256659a79f1ec2cba784Ted Kremenek
2863ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  return RTUI.result;
2864ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar}
2865ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2866df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor
28679ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
28682b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CTUnit)
2869ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2870f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2871a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit->TUData);
2872ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(CXXUnit->getOriginalSourceFileName(), true);
2873af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff}
28741eb79b58e56b99cf557d5d353586a10c5360364dDaniel Dunbar
28757eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas GregorCXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
2876aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  CXCursor Result = { CXCursor_TranslationUnit, 0, { 0, 0, TU } };
28777eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return Result;
28787eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
28797eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
2880fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2881600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2882fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2883fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXFile Operations.
2884fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2885fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2886fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
288774844072411bae91d5dbb89955d200cbe1e0a1c8Ted KremenekCXString clang_getFileName(CXFile SFile) {
288898258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
2889a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return createCXString((const char*)NULL);
2890f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
289188145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
289274844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek  return createCXString(FEnt->getName());
289388145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
289488145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff
289588145034694ed5267fa6fa5febc54fadc02bd479Steve Narofftime_t clang_getFileTime(CXFile SFile) {
289698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
289798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor    return 0;
2898f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
289988145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
290088145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  return FEnt->getModificationTime();
2901ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff}
2902f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2903b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXFile clang_getFile(CXTranslationUnit tu, const char *file_name) {
2904b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!tu)
2905b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return 0;
2906f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2907a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2908f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2909b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  FileManager &FMgr = CXXUnit->getFileManager();
291039b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattner  return const_cast<FileEntry *>(FMgr.getFile(file_name));
2911b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2912f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2913dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregorunsigned clang_isFileMultipleIncludeGuarded(CXTranslationUnit tu, CXFile file) {
2914dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  if (!tu || !file)
2915dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor    return 0;
2916dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor
2917dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2918dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  FileEntry *FEnt = static_cast<FileEntry *>(file);
2919dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  return CXXUnit->getPreprocessor().getHeaderSearchInfo()
2920dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor                                          .isFileMultipleIncludeGuarded(FEnt);
2921dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor}
2922dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor
2923fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2924fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2925fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2926fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXCursor Operations.
2927fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2928fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2929fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekstatic Decl *getDeclFromExpr(Stmt *E) {
2930c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis  if (ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
2931db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return getDeclFromExpr(CE->getSubExpr());
2932db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2933fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
2934fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RefExpr->getDecl();
293538f28c1189142429384996409fffbc57f67b5c60Douglas Gregor  if (BlockDeclRefExpr *RefExpr = dyn_cast<BlockDeclRefExpr>(E))
293638f28c1189142429384996409fffbc57f67b5c60Douglas Gregor    return RefExpr->getDecl();
2937fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (MemberExpr *ME = dyn_cast<MemberExpr>(E))
2938fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return ME->getMemberDecl();
2939fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
2940fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RE->getDecl();
2941db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(E))
294212f78a6741a4cb3d904340f8d3d2714568b50e7aJohn McCall    return PRE->isExplicitProperty() ? PRE->getExplicitProperty() : 0;
2943db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2944fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (CallExpr *CE = dyn_cast<CallExpr>(E))
2945fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return getDeclFromExpr(CE->getCallee());
29465f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  if (CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E))
294793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    if (!CE->isElidable())
294893798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CE->getConstructor();
2949fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
2950fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return OME->getMethodDecl();
2951f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2952db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCProtocolExpr *PE = dyn_cast<ObjCProtocolExpr>(E))
2953db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return PE->getProtocol();
2954c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor  if (SubstNonTypeTemplateParmPackExpr *NTTP
2955c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor                              = dyn_cast<SubstNonTypeTemplateParmPackExpr>(E))
2956c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor    return NTTP->getParameterPack();
295794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
295894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    if (isa<NonTypeTemplateParmDecl>(SizeOfPack->getPack()) ||
295994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        isa<ParmVarDecl>(SizeOfPack->getPack()))
296094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      return SizeOfPack->getPack();
2961db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2962fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  return 0;
2963fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek}
2964ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff
2965c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbarstatic SourceLocation getLocationFromExpr(Expr *E) {
2966c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis  if (ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
2967c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis    return getLocationFromExpr(CE->getSubExpr());
2968c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis
2969c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
2970c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return /*FIXME:*/Msg->getLeftLoc();
2971c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
2972c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return DRE->getLocation();
297338f28c1189142429384996409fffbc57f67b5c60Douglas Gregor  if (BlockDeclRefExpr *RefExpr = dyn_cast<BlockDeclRefExpr>(E))
297438f28c1189142429384996409fffbc57f67b5c60Douglas Gregor    return RefExpr->getLocation();
2975c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (MemberExpr *Member = dyn_cast<MemberExpr>(E))
2976c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Member->getMemberLoc();
2977c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
2978c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Ivar->getLocation();
297994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
298094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    return SizeOfPack->getPackLoc();
298194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
2982c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  return E->getLocStart();
2983c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar}
2984c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar
2985fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
2986f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2987f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekunsigned clang_visitChildren(CXCursor parent,
2988b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXCursorVisitor visitor,
2989b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXClientData client_data) {
2990f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis  CursorVisitor CursorVis(getCursorTU(parent), visitor, client_data,
2991f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                          /*VisitPreprocessorLast=*/false);
2992b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return CursorVis.VisitChildren(parent);
2993b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
2994b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
29953387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#ifndef __has_feature
29963387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#define __has_feature(x) 0
29973387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
29983387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#if __has_feature(blocks)
29993387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef enum CXChildVisitResult
30003387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall     (^CXCursorVisitorBlock)(CXCursor cursor, CXCursor parent);
30013387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30023387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
30033387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
30043387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
30053387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block(cursor, parent);
30063387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30073387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#else
30083387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// If we are compiled with a compiler that doesn't have native blocks support,
30093387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// define and call the block manually, so the
30103387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef struct _CXChildVisitResult
30113387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall{
30123387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	void *isa;
30133387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int flags;
30143387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int reserved;
30159e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar	enum CXChildVisitResult(*invoke)(struct _CXChildVisitResult*, CXCursor,
30169e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                         CXCursor);
30173387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall} *CXCursorVisitorBlock;
30183387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30193387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
30203387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
30213387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
30223387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block->invoke(block, cursor, parent);
30233387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30243387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
30253387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30263387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30279e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbarunsigned clang_visitChildrenWithBlock(CXCursor parent,
30289e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                      CXCursorVisitorBlock block) {
30293387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return clang_visitChildren(parent, visitWithBlock, block);
30303387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30313387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
303278205d4bada39d95097e766af9eb30cdd0159461Douglas Gregorstatic CXString getDeclSpelling(Decl *D) {
303378205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D);
3034e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  if (!ND) {
30355f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    if (ObjCPropertyImplDecl *PropImpl =dyn_cast<ObjCPropertyImplDecl>(D))
3036e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
3037e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return createCXString(Property->getIdentifier()->getName());
3038e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
3039ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
3040e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  }
3041e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
304278205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
3043ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(OMD->getSelector().getAsString());
3044f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
304578205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
304678205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // No, this isn't the same as the code below. getIdentifier() is non-virtual
304778205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // and returns different names. NamedDecl returns the class name and
304878205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // ObjCCategoryImplDecl returns the category name.
3049ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(CIMP->getIdentifier()->getNameStart());
3050f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30510a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  if (isa<UsingDirectiveDecl>(D))
30520a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("");
30530a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
305450aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::SmallString<1024> S;
305550aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::raw_svector_ostream os(S);
305650aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  ND->printName(os);
305750aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek
305850aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  return createCXString(os.str());
305978205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor}
3060f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30619ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getCursorSpelling(CXCursor C) {
30627eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  if (clang_isTranslationUnit(C.kind))
3063a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return clang_getTranslationUnitSpelling(
3064a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                            static_cast<CXTranslationUnit>(C.data[2]));
30657eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
3066f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  if (clang_isReference(C.kind)) {
3067f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    switch (C.kind) {
3068acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCSuperClassRef: {
30692e331b938b38057e333fab0ba841130ea8467794Douglas Gregor      ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
3070ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Super->getIdentifier()->getNameStart());
3071acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3072acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCClassRef: {
30731adb082a709f7b588f03672999294e061234b2cfDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
3074ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Class->getIdentifier()->getNameStart());
3075acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3076acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCProtocolRef: {
307778db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor      ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
3078f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      assert(OID && "getCursorSpelling(): Missing protocol decl");
3079ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(OID->getIdentifier()->getNameStart());
3080acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
30813064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
30823064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = getCursorCXXBaseSpecifier(C);
30833064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return createCXString(B->getType().getAsString());
30843064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
30857d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    case CXCursor_TypeRef: {
30867d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      TypeDecl *Type = getCursorTypeRef(C).first;
30877d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      assert(Type && "Missing type decl");
30887d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3089ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(getCursorContext(C).getTypeDeclType(Type).
3090ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                              getAsString());
30917d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
30920b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
30930b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      TemplateDecl *Template = getCursorTemplateRef(C).first;
30946931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(Template && "Missing template decl");
30950b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
30960b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString(Template->getNameAsString());
30970b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
30986931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
30996931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
31006931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      NamedDecl *NS = getCursorNamespaceRef(C).first;
31016931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(NS && "Missing namespace decl");
31026931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
31036931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return createCXString(NS->getNameAsString());
31046931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
31057d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3106a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3107a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      FieldDecl *Field = getCursorMemberRef(C).first;
3108a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      assert(Field && "Missing member decl");
3109a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3110a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return createCXString(Field->getNameAsString());
3111a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3112a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
311336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
311436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      LabelStmt *Label = getCursorLabelRef(C).first;
311536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      assert(Label && "Missing label");
311636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
3117ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
311836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
311936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31201f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef: {
31211f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
31221f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Decl *D = Storage.dyn_cast<Decl *>()) {
31231f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        if (NamedDecl *ND = dyn_cast<NamedDecl>(D))
31241f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor          return createCXString(ND->getNameAsString());
31251f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
31261f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      }
31271f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
31281f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString(E->getName().getAsString());
31291f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedTemplateStorage *Ovl
31301f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        = Storage.get<OverloadedTemplateStorage*>();
31311f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Ovl->size() == 0)
31321f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
31331f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return createCXString((*Ovl->begin())->getNameAsString());
31341f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    }
31351f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3136acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    default:
3137ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString("<not implemented>");
3138f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    }
3139f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  }
314097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
314197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
314297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    Decl *D = getDeclFromExpr(getCursorExpr(C));
314397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
314478205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor      return getDeclSpelling(D);
3145ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
314697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
314797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
314836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
314936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
315036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
3151ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
315236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
315336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("");
315436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
315536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31569b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion)
31579e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth    return createCXString(getCursorMacroExpansion(C)->getName()
31584ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor                                                           ->getNameStart());
31594ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor
3160572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
3161572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString(getCursorMacroDefinition(C)->getName()
3162572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor                                                           ->getNameStart());
3163572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3164ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective)
3165ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString(getCursorInclusionDirective(C)->getFileName());
3166ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
316760cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor  if (clang_isDeclaration(C.kind))
316860cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor    return getDeclSpelling(getCursorDecl(C));
3169e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
31705f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  if (C.kind == CXCursor_AnnotateAttr) {
31715f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    AnnotateAttr *AA = cast<AnnotateAttr>(cxcursor::getCursorAttr(C));
31725f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    return createCXString(AA->getAnnotation());
31735f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  }
31745f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen
3175ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString("");
3176f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3177f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
3178358559d8d7b458c5f64941842383a16e61f0828dDouglas GregorCXString clang_getCursorDisplayName(CXCursor C) {
3179358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!clang_isDeclaration(C.kind))
3180358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return clang_getCursorSpelling(C);
3181358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3182358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  Decl *D = getCursorDecl(C);
3183358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!D)
3184358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString("");
3185358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
318630c42404202d2e2512e51efc6066bd614cfdb5a4Douglas Gregor  PrintingPolicy Policy = getCursorContext(C).getPrintingPolicy();
3187358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(D))
3188358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    D = FunTmpl->getTemplatedDecl();
3189358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3190358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
3191358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3192358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3193358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << Function->getNameAsString();
3194358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->getPrimaryTemplate())
3195358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "<>";
3196358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "(";
3197358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Function->getNumParams(); I != N; ++I) {
3198358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3199358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3200358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << Function->getParamDecl(I)->getType().getAsString(Policy);
3201358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3202358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3203358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->isVariadic()) {
3204358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Function->getNumParams())
3205358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3206358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "...";
3207358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3208358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ")";
3209358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3210358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3211358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3212358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) {
3213358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3214358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3215358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ClassTemplate->getNameAsString();
3216358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "<";
3217358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    TemplateParameterList *Params = ClassTemplate->getTemplateParameters();
3218358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Params->size(); I != N; ++I) {
3219358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3220358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3221358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3222358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      NamedDecl *Param = Params->getParam(I);
3223358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Param->getIdentifier()) {
3224358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << Param->getIdentifier()->getName();
3225358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        continue;
3226358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      }
3227358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3228358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // There is no parameter name, which makes this tricky. Try to come up
3229358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // with something useful that isn't too long.
3230358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
3231358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << (TTP->wasDeclaredWithTypename()? "typename" : "class");
3232358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else if (NonTypeTemplateParmDecl *NTTP
3233358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                    = dyn_cast<NonTypeTemplateParmDecl>(Param))
3234358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << NTTP->getType().getAsString(Policy);
3235358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else
3236358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << "template<...> class";
3237358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3238358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3239358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ">";
3240358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3241358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3242358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3243358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateSpecializationDecl *ClassSpec
3244358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                              = dyn_cast<ClassTemplateSpecializationDecl>(D)) {
3245358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    // If the type was explicitly written, use that.
3246358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (TypeSourceInfo *TSInfo = ClassSpec->getTypeAsWritten())
3247358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      return createCXString(TSInfo->getType().getAsString(Policy));
3248358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3249358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3250358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3251358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ClassSpec->getNameAsString();
3252358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << TemplateSpecializationType::PrintTemplateArgumentList(
3253910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().data(),
3254910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().size(),
3255358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                                                Policy);
3256358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3257358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3258358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3259358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  return clang_getCursorSpelling(C);
3260358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor}
3261358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3262e68fff6fc083c6270d835216a3de0b82c6ef0310Ted KremenekCXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
326389922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  switch (Kind) {
3264e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FunctionDecl:
3265e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FunctionDecl");
3266e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypedefDecl:
3267e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypedefDecl");
3268e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumDecl:
3269e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumDecl");
3270e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumConstantDecl:
3271e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumConstantDecl");
3272e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_StructDecl:
3273e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("StructDecl");
3274e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnionDecl:
3275e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnionDecl");
3276e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ClassDecl:
3277e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ClassDecl");
3278e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FieldDecl:
3279e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FieldDecl");
3280e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_VarDecl:
3281e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("VarDecl");
3282e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ParmDecl:
3283e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ParmDecl");
3284e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInterfaceDecl:
3285e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInterfaceDecl");
3286e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryDecl:
3287e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryDecl");
3288e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolDecl:
3289e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolDecl");
3290e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCPropertyDecl:
3291e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCPropertyDecl");
3292e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCIvarDecl:
3293e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCIvarDecl");
3294e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInstanceMethodDecl:
3295e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInstanceMethodDecl");
3296e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassMethodDecl:
3297e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassMethodDecl");
3298e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCImplementationDecl:
3299e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCImplementationDecl");
3300e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryImplDecl:
3301e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryImplDecl");
33028bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek  case CXCursor_CXXMethod:
33038bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek      return createCXString("CXXMethod");
3304e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedDecl:
3305e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedDecl");
3306e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCSuperClassRef:
3307e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCSuperClassRef");
3308e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolRef:
3309e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolRef");
3310e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassRef:
3311e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassRef");
3312e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypeRef:
3313e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypeRef");
33140b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case CXCursor_TemplateRef:
33150b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString("TemplateRef");
33166931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceRef:
33176931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceRef");
3318a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  case CXCursor_MemberRef:
3319a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return createCXString("MemberRef");
332036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelRef:
332136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("LabelRef");
33221f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case CXCursor_OverloadedDeclRef:
33231f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return createCXString("OverloadedDeclRef");
332442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IntegerLiteral:
332542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IntegerLiteral");
332642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_FloatingLiteral:
332742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("FloatingLiteral");
332842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ImaginaryLiteral:
332942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ImaginaryLiteral");
333042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_StringLiteral:
333142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("StringLiteral");
333242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CharacterLiteral:
333342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CharacterLiteral");
333442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ParenExpr:
333542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ParenExpr");
333642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnaryOperator:
333742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnaryOperator");
333842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ArraySubscriptExpr:
333942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ArraySubscriptExpr");
334042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_BinaryOperator:
334142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("BinaryOperator");
334242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundAssignOperator:
334342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundAssignOperator");
334442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ConditionalOperator:
334542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ConditionalOperator");
334642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CStyleCastExpr:
334742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CStyleCastExpr");
334842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundLiteralExpr:
334942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundLiteralExpr");
335042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_InitListExpr:
335142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("InitListExpr");
335242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_AddrLabelExpr:
335342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("AddrLabelExpr");
335442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_StmtExpr:
335542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("StmtExpr");
335642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GenericSelectionExpr:
335742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GenericSelectionExpr");
335842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GNUNullExpr:
335942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GNUNullExpr");
336042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXStaticCastExpr:
336142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXStaticCastExpr");
336242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXDynamicCastExpr:
336342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXDynamicCastExpr");
336442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXReinterpretCastExpr:
336542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXReinterpretCastExpr");
336642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXConstCastExpr:
336742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXConstCastExpr");
336842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXFunctionalCastExpr:
336942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXFunctionalCastExpr");
337042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXTypeidExpr:
337142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXTypeidExpr");
337242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXBoolLiteralExpr:
337342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXBoolLiteralExpr");
337442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXNullPtrLiteralExpr:
337542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXNullPtrLiteralExpr");
337642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXThisExpr:
337742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXThisExpr");
337842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXThrowExpr:
337942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXThrowExpr");
338042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXNewExpr:
338142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXNewExpr");
338242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXDeleteExpr:
338342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXDeleteExpr");
338442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnaryExpr:
338542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnaryExpr");
338642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCStringLiteral:
338742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCStringLiteral");
338842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCEncodeExpr:
338942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCEncodeExpr");
339042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCSelectorExpr:
339142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCSelectorExpr");
339242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCProtocolExpr:
339342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCProtocolExpr");
339442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCBridgedCastExpr:
339542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCBridgedCastExpr");
33961ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  case CXCursor_BlockExpr:
33971ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek      return createCXString("BlockExpr");
339842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_PackExpansionExpr:
339942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("PackExpansionExpr");
340042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SizeOfPackExpr:
340142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SizeOfPackExpr");
340242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnexposedExpr:
340342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnexposedExpr");
3404e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_DeclRefExpr:
3405e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("DeclRefExpr");
3406e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_MemberRefExpr:
3407e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("MemberRefExpr");
3408e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_CallExpr:
3409e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("CallExpr");
3410e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCMessageExpr:
3411e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCMessageExpr");
3412e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedStmt:
3413e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedStmt");
341442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DeclStmt:
341542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DeclStmt");
341636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelStmt:
341736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return createCXString("LabelStmt");
341842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundStmt:
341942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundStmt");
342042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CaseStmt:
342142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CaseStmt");
342242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DefaultStmt:
342342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DefaultStmt");
342442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IfStmt:
342542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IfStmt");
342642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SwitchStmt:
342742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SwitchStmt");
342842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_WhileStmt:
342942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("WhileStmt");
343042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DoStmt:
343142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DoStmt");
343242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ForStmt:
343342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ForStmt");
343442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GotoStmt:
343542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GotoStmt");
343642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IndirectGotoStmt:
343742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IndirectGotoStmt");
343842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ContinueStmt:
343942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ContinueStmt");
344042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_BreakStmt:
344142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("BreakStmt");
344242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ReturnStmt:
344342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ReturnStmt");
344442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_AsmStmt:
344542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("AsmStmt");
344642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtTryStmt:
344742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtTryStmt");
344842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtCatchStmt:
344942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtCatchStmt");
345042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtFinallyStmt:
345142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtFinallyStmt");
345242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtThrowStmt:
345342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtThrowStmt");
345442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtSynchronizedStmt:
345542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtSynchronizedStmt");
345642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAutoreleasePoolStmt:
345742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAutoreleasePoolStmt");
345842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCForCollectionStmt:
345942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCForCollectionStmt");
346042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXCatchStmt:
346142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXCatchStmt");
346242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXTryStmt:
346342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXTryStmt");
346442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXForRangeStmt:
346542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXForRangeStmt");
346642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHTryStmt:
346742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHTryStmt");
346842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHExceptStmt:
346942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHExceptStmt");
347042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHFinallyStmt:
347142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHFinallyStmt");
347242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_NullStmt:
347342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("NullStmt");
3474e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_InvalidFile:
3475e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("InvalidFile");
3476292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek  case CXCursor_InvalidCode:
3477292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek    return createCXString("InvalidCode");
3478e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NoDeclFound:
3479e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NoDeclFound");
3480e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NotImplemented:
3481e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NotImplemented");
3482e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TranslationUnit:
3483e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TranslationUnit");
3484e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_UnexposedAttr:
3485e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("UnexposedAttr");
3486e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_IBActionAttr:
3487e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("attribute(ibaction)");
34889f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_IBOutletAttr:
34899f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor     return createCXString("attribute(iboutlet)");
3490857e918a8a40deb128840308a318bf623d68295fTed Kremenek  case CXCursor_IBOutletCollectionAttr:
3491857e918a8a40deb128840308a318bf623d68295fTed Kremenek      return createCXString("attribute(iboutletcollection)");
34926639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  case CXCursor_CXXFinalAttr:
34936639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      return createCXString("attribute(final)");
34946639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  case CXCursor_CXXOverrideAttr:
34956639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      return createCXString("attribute(override)");
34965f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen  case CXCursor_AnnotateAttr:
34975f1c822def3efffe1d8f7299fbbbc3b1cdd4833dErik Verbruggen    return createCXString("attribute(annotate)");
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");
3547a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return createCXString((const char*) 0);
3548600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
354989922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff
3550064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidisstruct GetCursorData {
3551064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  SourceLocation TokenBeginLoc;
35524b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  bool PointsAtMacroArgExpansion;
3553064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  CXCursor &BestCursor;
3554064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
35554b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  GetCursorData(SourceManager &SM,
35564b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                SourceLocation tokenBegin, CXCursor &outputCursor)
35574b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    : TokenBeginLoc(tokenBegin), BestCursor(outputCursor) {
35584b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    PointsAtMacroArgExpansion = SM.isMacroArgExpansion(tokenBegin);
35594b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  }
3560064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis};
3561064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
35624b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidisstatic enum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
35634b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                                                CXCursor parent,
35644b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                                                CXClientData client_data) {
3565064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  GetCursorData *Data = static_cast<GetCursorData *>(client_data);
3566064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  CXCursor *BestCursor = &Data->BestCursor;
35674b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis
35684b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // If we point inside a macro argument we should provide info of what the
35694b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // token is so use the actual cursor, don't replace it with a macro expansion
35704b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // cursor.
35714b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  if (cursor.kind == CXCursor_MacroExpansion && Data->PointsAtMacroArgExpansion)
35724b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    return CXChildVisit_Recurse;
357365ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis
357465ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis  if (clang_isDeclaration(cursor.kind)) {
357565ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // Avoid having the implicit methods override the property decls.
357665ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    if (ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(getCursorDecl(cursor)))
357765ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis      if (MD->isImplicit())
357865ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis        return CXChildVisit_Break;
357965ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis  }
3580064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
3581064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  if (clang_isExpression(cursor.kind) &&
3582064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis      clang_isDeclaration(BestCursor->kind)) {
3583064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    Decl *D = getCursorDecl(*BestCursor);
3584064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
3585064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    // Avoid having the cursor of an expression replace the declaration cursor
3586064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    // when the expression source range overlaps the declaration range.
3587064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    // This can happen for C++ constructor expressions whose range generally
3588064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    // include the variable declaration, e.g.:
3589064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    //  MyCXXClass foo; // Make sure pointing at 'foo' returns a VarDecl cursor.
3590064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    if (D->getLocation().isValid() && Data->TokenBeginLoc.isValid() &&
3591064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis        D->getLocation() == Data->TokenBeginLoc)
3592064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis      return CXChildVisit_Break;
3593064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  }
3594064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
359593798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // If our current best cursor is the construction of a temporary object,
359693798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // don't replace that cursor with a type reference, because we want
359793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // clang_getCursor() to point at the constructor.
359893798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  if (clang_isExpression(BestCursor->kind) &&
359993798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor      isa<CXXTemporaryObjectExpr>(getCursorExpr(*BestCursor)) &&
3600aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      cursor.kind == CXCursor_TypeRef) {
3601aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    // Keep the cursor pointing at CXXTemporaryObjectExpr but also mark it
3602aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    // as having the actual point on the type reference.
3603aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    *BestCursor = getTypeRefedCallExprCursor(*BestCursor);
360493798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CXChildVisit_Recurse;
3605aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  }
360693798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor
360733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  *BestCursor = cursor;
360833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return CXChildVisit_Recurse;
360933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
3610e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3611b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
3612b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!TU)
3613f462989fe8d6f59ab2d7d0fe2b4b96292ce706eaTed Kremenek    return clang_getNullCursor();
3614e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3615a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3616bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
3617bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
3618a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek  SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
3619671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  CXCursor Result = cxcursor::getCursor(TU, SLoc);
3620a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek
362140749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  bool Logging = getenv("LIBCLANG_LOGGING");
362240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  if (Logging) {
362340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile SearchFile;
362440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned SearchLine, SearchColumn;
362540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile ResultFile;
362640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned ResultLine, ResultColumn;
36276653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    CXString SearchFileName, ResultFileName, KindSpelling, USR;
36286653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    const char *IsDef = clang_isCursorDefinition(Result)? " (Definition)" : "";
362940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXSourceLocation ResultLoc = clang_getCursorLocation(Result);
363040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
363120174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    clang_getExpansionLocation(Loc, &SearchFile, &SearchLine, &SearchColumn, 0);
363220174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    clang_getExpansionLocation(ResultLoc, &ResultFile, &ResultLine,
363320174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                               &ResultColumn, 0);
363440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    SearchFileName = clang_getFileName(SearchFile);
363540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    ResultFileName = clang_getFileName(ResultFile);
363640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    KindSpelling = clang_getCursorKindSpelling(Result.kind);
36376653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    USR = clang_getCursorUSR(Result);
36386653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    fprintf(stderr, "clang_getCursor(%s:%d:%d) = %s(%s:%d:%d):%s%s\n",
363940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(SearchFileName), SearchLine, SearchColumn,
364040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(KindSpelling),
36416653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(ResultFileName), ResultLine, ResultColumn,
36426653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(USR), IsDef);
364340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(SearchFileName);
364440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(ResultFileName);
364540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(KindSpelling);
36466653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    clang_disposeString(USR);
36470aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor
36480aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    CXCursor Definition = clang_getCursorDefinition(Result);
36490aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    if (!clang_equalCursors(Definition, clang_getNullCursor())) {
36500aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXSourceLocation DefinitionLoc = clang_getCursorLocation(Definition);
36510aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionKindSpelling
36520aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor                                = clang_getCursorKindSpelling(Definition.kind);
36530aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXFile DefinitionFile;
36540aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      unsigned DefinitionLine, DefinitionColumn;
365520174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth      clang_getExpansionLocation(DefinitionLoc, &DefinitionFile,
365620174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                 &DefinitionLine, &DefinitionColumn, 0);
36570aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionFileName = clang_getFileName(DefinitionFile);
36580aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      fprintf(stderr, "  -> %s(%s:%d:%d)\n",
36590aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionKindSpelling),
36600aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionFileName),
36610aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              DefinitionLine, DefinitionColumn);
36620aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionFileName);
36630aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionKindSpelling);
36640aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    }
366540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  }
366640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
3667e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  return Result;
366877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
366977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
3670738855554394a6afcf39cc8345fd22c3756b8dd0Ted KremenekCXCursor clang_getNullCursor(void) {
36715bfb8c128c2ac8eb4032afc180cdc400a0f953caDouglas Gregor  return MakeCXCursorInvalid(CXCursor_InvalidFile);
3672738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
3673738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek
3674738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenekunsigned clang_equalCursors(CXCursor X, CXCursor Y) {
3675283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  return X == Y;
3676738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
36770d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
36789ce5584553054d0cb934940586aca0186e87fa57Douglas Gregorunsigned clang_hashCursor(CXCursor C) {
36799ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  unsigned Index = 0;
36809ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  if (clang_isExpression(C.kind) || clang_isStatement(C.kind))
36819ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor    Index = 1;
36829ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
36839ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  return llvm::DenseMapInfo<std::pair<unsigned, void*> >::getHashValue(
36849ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor                                        std::make_pair(C.kind, C.data[Index]));
36859ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor}
36869ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
36879ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isInvalid(enum CXCursorKind K) {
368877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
368977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
369077128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
36919ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isDeclaration(enum CXCursorKind K) {
369289922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl;
369389922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff}
36942d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff
36959ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isReference(enum CXCursorKind K) {
3696f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
3697f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3698f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
369997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isExpression(enum CXCursorKind K) {
370097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
370197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
370297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
370397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isStatement(enum CXCursorKind K) {
370497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
370597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
370697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
37078be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregorunsigned clang_isAttribute(enum CXCursorKind K) {
37088be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor    return K >= CXCursor_FirstAttr && K <= CXCursor_LastAttr;
37098be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor}
37108be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor
37117eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregorunsigned clang_isTranslationUnit(enum CXCursorKind K) {
37127eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return K == CXCursor_TranslationUnit;
37137eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
37147eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
37159f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregorunsigned clang_isPreprocessing(enum CXCursorKind K) {
37169f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
37179f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor}
37189f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor
3719ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenekunsigned clang_isUnexposed(enum CXCursorKind K) {
3720ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  switch (K) {
3721ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedDecl:
3722ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedExpr:
3723ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedStmt:
3724ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedAttr:
3725ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return true;
3726ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    default:
3727ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return false;
3728ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  }
3729ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek}
3730ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek
37319ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXCursorKind clang_getCursorKind(CXCursor C) {
37329efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff  return C.kind;
37339efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff}
37349efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff
373598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas GregorCXSourceLocation clang_getCursorLocation(CXCursor C) {
373698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (clang_isReference(C.kind)) {
3737f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    switch (C.kind) {
3738f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCSuperClassRef: {
3739f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3740f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCSuperClassRef(C);
3741a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3742f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3743f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3744f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
3745f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCProtocolDecl *, SourceLocation> P
3746f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCProtocolRef(C);
3747a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3748f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3749f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3750f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef: {
3751f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3752f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCClassRef(C);
3753a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3754f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
37557d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3756f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef: {
37577d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      std::pair<TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
3758a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
37597d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
37600b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
37610b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
37620b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      std::pair<TemplateDecl *, SourceLocation> P = getCursorTemplateRef(C);
37630b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
37640b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
37650b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
37666931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
37676931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      std::pair<NamedDecl *, SourceLocation> P = getCursorNamespaceRef(C);
37686931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
37696931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
37706931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
3771a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3772a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      std::pair<FieldDecl *, SourceLocation> P = getCursorMemberRef(C);
3773a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3774a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3775a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
37763064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
37771b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      CXXBaseSpecifier *BaseSpec = getCursorCXXBaseSpecifier(C);
37781b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (!BaseSpec)
37791b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return clang_getNullLocation();
37801b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
37811b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (TypeSourceInfo *TSInfo = BaseSpec->getTypeSourceInfo())
37821b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return cxloc::translateSourceLocation(getCursorContext(C),
37831b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                            TSInfo->getTypeLoc().getBeginLoc());
37841b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
37851b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
37861b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                        BaseSpec->getSourceRange().getBegin());
37873064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
3788f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
378936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
379036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      std::pair<LabelStmt *, SourceLocation> P = getCursorLabelRef(C);
379136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C), P.second);
379236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
379336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
37941f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
37951f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
37961f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                          getCursorOverloadedDeclRef(C).second);
37971f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3798f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    default:
3799f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3800f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      llvm_unreachable("Missed a reference kind");
3801f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
380298258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  }
380397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
380497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3805f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    return cxloc::translateSourceLocation(getCursorContext(C),
380697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor                                   getLocationFromExpr(getCursorExpr(C)));
380797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
380836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind))
380936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C),
381036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor                                          getCursorStmt(C)->getLocStart());
381136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
38129f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  if (C.kind == CXCursor_PreprocessingDirective) {
38139f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
38149f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
38159f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  }
38164807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
38179b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion) {
38184ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    SourceLocation L
38199e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth      = cxcursor::getCursorMacroExpansion(C)->getSourceRange().getBegin();
38204807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
38214807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  }
3822572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3823572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition) {
3824572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
3825572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3826572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  }
3827ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3828ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective) {
3829ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    SourceLocation L
3830ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      = cxcursor::getCursorInclusionDirective(C)->getSourceRange().getBegin();
3831ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3832ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  }
3833ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
38349a700d277c38d9afaa7cb3fe93a714bfe9b62eecTed Kremenek  if (C.kind < CXCursor_FirstDecl || C.kind > CXCursor_LastDecl)
38355352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullLocation();
383698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor
3837f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  Decl *D = getCursorDecl(C);
3838f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  SourceLocation Loc = D->getLocation();
3839007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // FIXME: Multiple variables declared in a single declaration
3840007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // currently lack the information needed to correctly determine their
3841007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // ranges when accounting for the type-specifier.  We use context
3842007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3843007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // and if so, whether it is the first decl.
3844007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3845007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (!cxcursor::isFirstInDeclGroup(C))
3846007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      Loc = VD->getLocation();
3847007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
3848007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek
38492ca54feee89d7277fb967e3247a64f40ef155a82Douglas Gregor  return cxloc::translateSourceLocation(getCursorContext(C), Loc);
385088145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
3851a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor
3852a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor} // end extern "C"
3853a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3854671436e9e2794c56f3c2e62739d225571493af37Argyrios KyrtzidisCXCursor cxcursor::getCursor(CXTranslationUnit TU, SourceLocation SLoc) {
3855671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  assert(TU);
3856671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3857671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // Guard against an invalid SourceLocation, or we may assert in one
3858671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // of the following calls.
3859671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  if (SLoc.isInvalid())
3860671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    return clang_getNullCursor();
3861671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3862671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3863671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3864671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // Translate the given source location to make it point at the beginning of
3865671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // the token under the cursor.
3866671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  SLoc = Lexer::GetBeginningOfToken(SLoc, CXXUnit->getSourceManager(),
3867671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                                    CXXUnit->getASTContext().getLangOptions());
3868671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3869671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
3870671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  if (SLoc.isValid()) {
3871671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    GetCursorData ResultData(CXXUnit->getSourceManager(), SLoc, Result);
3872671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    CursorVisitor CursorVis(TU, GetCursorVisitor, &ResultData,
3873671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                            /*VisitPreprocessorLast=*/true,
3874e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                            /*VisitIncludedEntities=*/false,
3875671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                            SourceLocation(SLoc));
3876dfb332d0081c6641d1dbae6a2aeff757c99cc740Argyrios Kyrtzidis    CursorVis.visitFileRegion();
3877671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  }
3878671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3879671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  return Result;
3880671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis}
3881671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3882a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C) {
3883a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  if (clang_isReference(C.kind)) {
3884a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    switch (C.kind) {
3885a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCSuperClassRef:
3886a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return  getCursorObjCSuperClassRef(C).second;
3887f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3888a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCProtocolRef:
3889a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCProtocolRef(C).second;
3890f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3891a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCClassRef:
3892a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCClassRef(C).second;
38937d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3894a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_TypeRef:
3895a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorTypeRef(C).second;
38960b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
38970b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
38980b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return getCursorTemplateRef(C).second;
38990b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
39006931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
39016931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return getCursorNamespaceRef(C).second;
3902a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3903a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
3904a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return getCursorMemberRef(C).second;
3905a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
39063064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier:
39071b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return getCursorCXXBaseSpecifier(C)->getSourceRange();
3908f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
390936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
391036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return getCursorLabelRef(C).second;
391136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
39121f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
39131f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return getCursorOverloadedDeclRef(C).second;
39141f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3915a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    default:
3916a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3917a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      llvm_unreachable("Missed a reference kind");
3918a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    }
3919a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  }
392097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
392197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3922a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorExpr(C)->getSourceRange();
392333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
392433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (clang_isStatement(C.kind))
3925a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorStmt(C)->getSourceRange();
3926f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
39276639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  if (clang_isAttribute(C.kind))
39286639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis    return getCursorAttr(C)->getRange();
39296639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis
3930a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_PreprocessingDirective)
3931a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorPreprocessingDirective(C);
39324807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
3933ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_MacroExpansion) {
3934ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
3935ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorMacroExpansion(C)->getSourceRange();
3936ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
3937ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
3938572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3939ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_MacroDefinition) {
3940ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
3941ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorMacroDefinition(C)->getSourceRange();
3942ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
3943ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
3944ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3945ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_InclusionDirective) {
3946ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
3947ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorInclusionDirective(C)->getSourceRange();
3948ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
3949ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
3950ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3951007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
3952007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    Decl *D = cxcursor::getCursorDecl(C);
3953007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    SourceRange R = D->getSourceRange();
3954007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // FIXME: Multiple variables declared in a single declaration
3955007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // currently lack the information needed to correctly determine their
3956007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // ranges when accounting for the type-specifier.  We use context
3957007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3958007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // and if so, whether it is the first decl.
3959007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3960007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      if (!cxcursor::isFirstInDeclGroup(C))
3961007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek        R.setBegin(VD->getLocation());
3962007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    }
3963007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    return R;
3964007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
39656653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return SourceRange();
39666653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
39676653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
39686653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// \brief Retrieves the "raw" cursor extent, which is then extended to include
39696653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// the decl-specifier-seq for declarations.
39706653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr) {
39716653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
39726653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    Decl *D = cxcursor::getCursorDecl(C);
39736653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange R = D->getSourceRange();
39742494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
39752494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // Adjust the start of the location for declarations preceded by
39762494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // declaration specifiers.
39772494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
39786653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
39792494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
39802494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
39812494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
39822494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
39832494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
39842494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    }
39856653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
39862494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && R.getBegin().isValid() &&
39872494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, R.getBegin()))
39882494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      R.setBegin(StartLoc);
39892494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
39902494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // FIXME: Multiple variables declared in a single declaration
39912494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // currently lack the information needed to correctly determine their
39922494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // ranges when accounting for the type-specifier.  We use context
39932494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
39942494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // and if so, whether it is the first decl.
39952494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
39962494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (!cxcursor::isFirstInDeclGroup(C))
39972494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        R.setBegin(VD->getLocation());
39986653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    }
39996653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
40006653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    return R;
40016653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  }
40026653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
40036653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return getRawCursorExtent(C);
40046653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
4005a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
4006a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorextern "C" {
4007a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
4008a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas GregorCXSourceRange clang_getCursorExtent(CXCursor C) {
4009a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  SourceRange R = getRawCursorExtent(C);
4010a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R.isInvalid())
40115352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4012f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4013a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  return cxloc::translateSourceRange(getCursorContext(C), R);
4014a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor}
4015c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
4016c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas GregorCXCursor clang_getCursorReferenced(CXCursor C) {
4017b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
4018b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
4019f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4020a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit tu = getCursorTU(C);
40211f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (clang_isDeclaration(C.kind)) {
40221f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getCursorDecl(C);
40231f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4024a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Using, D->getLocation(), tu);
40251f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
4026a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Classes, D->getLocation(), tu);
40271f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (ObjCForwardProtocolDecl *Protocols
40281f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                        = dyn_cast<ObjCForwardProtocolDecl>(D))
4029a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Protocols, D->getLocation(), tu);
40305f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    if (ObjCPropertyImplDecl *PropImpl =dyn_cast<ObjCPropertyImplDecl>(D))
4031e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
4032e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return MakeCXCursor(Property, tu);
4033e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
4034c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return C;
40351f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
40361f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
403797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
40381f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Expr *E = getCursorExpr(C);
40391f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getDeclFromExpr(E);
4040aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (D) {
4041aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      CXCursor declCursor = MakeCXCursor(D, tu);
4042aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      declCursor = getSelectorIdentifierCursor(getSelectorIdentifierIndex(C),
4043aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis                                               declCursor);
4044aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return declCursor;
4045aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    }
40461f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40471f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (OverloadExpr *Ovl = dyn_cast_or_null<OverloadExpr>(E))
4048a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Ovl, tu);
40491f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
405097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    return clang_getNullCursor();
405197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
405297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
405336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
405436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
405536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (GotoStmt *Goto = dyn_cast_or_null<GotoStmt>(S))
405637c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek      if (LabelDecl *label = Goto->getLabel())
405737c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        if (LabelStmt *labelS = label->getStmt())
405837c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        return MakeCXCursor(labelS, getCursorDecl(C), tu);
405936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
406036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return clang_getNullCursor();
406136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
406236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
40639b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion) {
40649e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth    if (MacroDefinition *Def = getCursorMacroExpansion(C)->getDefinition())
4065a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeMacroDefinitionCursor(Def, tu);
4066bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  }
4067bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
4068c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  if (!clang_isReference(C.kind))
4069c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return clang_getNullCursor();
4070f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4071c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  switch (C.kind) {
4072c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    case CXCursor_ObjCSuperClassRef:
4073a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCSuperClassRef(C).first, tu);
4074f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4075f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
4076a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCProtocolRef(C).first, tu);
4077f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4078f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef:
4079a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCClassRef(C).first, tu );
40807d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
4081f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef:
4082a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTypeRef(C).first, tu );
40830b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
40840b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
4085a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTemplateRef(C).first, tu );
40860b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
40876931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
4088a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorNamespaceRef(C).first, tu );
40896931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
4090a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
4091a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorMemberRef(C).first, tu );
4092a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
40933064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
40943064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = cxcursor::getCursorCXXBaseSpecifier(C);
40953064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return clang_getTypeDeclaration(cxtype::MakeCXType(B->getType(),
4096a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                                         tu ));
40973064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
4098f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
409936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
410036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // FIXME: We end up faking the "parent" declaration here because we
410136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // don't want to make CXCursor larger.
410236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return MakeCXCursor(getCursorLabelRef(C).first,
4103a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek               static_cast<ASTUnit*>(tu->TUData)->getASTContext()
4104a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          .getTranslationUnitDecl(),
4105a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          tu);
410636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
41071f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
41081f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return C;
41091f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4110c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    default:
4111c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      // We would prefer to enumerate all non-reference cursor kinds here.
4112c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      llvm_unreachable("Unhandled reference cursor kind");
4113c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      break;
4114c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    }
4115c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  }
4116f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4117c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  return clang_getNullCursor();
4118c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor}
4119c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
4120b699866820102a69d83d6ac6941985c5ef4e8c40Douglas GregorCXCursor clang_getCursorDefinition(CXCursor C) {
4121b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
4122b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
4123f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4124a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(C);
4125f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4126b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  bool WasReference = false;
412797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
4128b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    C = clang_getCursorReferenced(C);
4129b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    WasReference = true;
4130b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4131b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
41329b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion)
4133bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    return clang_getCursorReferenced(C);
4134bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
4135b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4136b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4137b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4138b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  Decl *D = getCursorDecl(C);
4139b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!D)
4140b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4141f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4142b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  switch (D->getKind()) {
4143b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't really separate the notions of
4144b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // declaration and definition.
4145b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Namespace:
4146b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Typedef:
4147162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  case Decl::TypeAlias:
41483e4c6c4c79a03f5cb0c4671d7c282d623c6dc35eRichard Smith  case Decl::TypeAliasTemplate:
4149b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTypeParm:
4150b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::EnumConstant:
4151b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Field:
4152d98114647e16796a976b04af79975b4f0eacf22bBenjamin Kramer  case Decl::IndirectField:
4153b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCIvar:
4154b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCAtDefsField:
4155b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ImplicitParam:
4156b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ParmVar:
4157b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NonTypeTemplateParm:
4158b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTemplateParm:
4159b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategoryImpl:
4160b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCImplementation:
41616206d53f67613958ae1b023aba337ebb46f11a8bAbramo Bagnara  case Decl::AccessSpec:
4162b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::LinkageSpec:
4163b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCPropertyImpl:
4164b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FileScopeAsm:
4165b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::StaticAssert:
4166b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Block:
4167ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  case Decl::Label:  // FIXME: Is this right??
4168af0f4d0b2e38c810effc8b024ad2fb6604eec5d3Francois Pichet  case Decl::ClassScopeFunctionSpecialization:
4169b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return C;
4170b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4171b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't make any sense here, but are
4172b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // nonetheless harmless.
4173b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TranslationUnit:
4174b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
4175b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4176b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds for which the definition is not resolvable.
4177b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingTypename:
4178b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingValue:
4179b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
4180b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4181b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingDirective:
4182b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
4183a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                        TU);
4184b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4185b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NamespaceAlias:
4186a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), TU);
4187b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4188b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Enum:
4189b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Record:
4190b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXRecord:
4191b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplateSpecialization:
4192b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplatePartialSpecialization:
4193952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor    if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
4194a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
4195b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4196b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4197b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Function:
4198b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXMethod:
4199b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConstructor:
4200b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXDestructor:
4201b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConversion: {
4202b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
4203b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionDecl>(D)->getBody(Def))
4204a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(const_cast<FunctionDecl *>(Def), TU);
4205b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4206b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4207b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4208b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Var: {
420931310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    // Ask the variable if it has a definition.
421031310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (VarDecl *Def = cast<VarDecl>(D)->getDefinition())
4211a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
421231310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    return clang_getNullCursor();
4213b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4214f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4215b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FunctionTemplate: {
4216b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
4217b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
4218a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def->getDescribedFunctionTemplate(), TU);
4219b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4220b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4221f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4222b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplate: {
4223b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
4224952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor                                                            ->getDefinition())
42250b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return MakeCXCursor(cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
4226a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          TU);
4227b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4228b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4229b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
42301f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::Using:
42311f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<UsingDecl>(D),
4232a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
4233b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4234b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingShadow:
4235b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getCursorDefinition(
4236f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                       MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
4237a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                    TU));
4238b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4239b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCMethod: {
4240b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
4241b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (Method->isThisDeclarationADefinition())
4242b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
4243b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4244b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // Dig out the method definition in the associated
4245b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // @implementation, if we have it.
4246b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: The ASTs should make finding the definition easier.
4247b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4248b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                       = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
4249b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
4250b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
4251b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                                                  Method->isInstanceMethod()))
4252b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          if (Def->isThisDeclarationADefinition())
4253a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek            return MakeCXCursor(Def, TU);
4254b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4255b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4256b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4257b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4258b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategory:
4259b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCCategoryImplDecl *Impl
4260b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                               = cast<ObjCCategoryDecl>(D)->getImplementation())
4261a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4262b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4263b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4264b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProtocol:
4265b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (!cast<ObjCProtocolDecl>(D)->isForwardDecl())
4266b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
4267b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4268b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4269b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCInterface:
4270b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // There are two notions of a "definition" for an Objective-C
4271b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // class: the interface and its implementation. When we resolved a
4272b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // reference to an Objective-C class, produce the @interface as
4273b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // the definition; when we were provided with the interface,
4274b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // produce the @implementation as the definition.
4275b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (WasReference) {
4276b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!cast<ObjCInterfaceDecl>(D)->isForwardDecl())
4277b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        return C;
4278b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    } else if (ObjCImplementationDecl *Impl
4279b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                              = cast<ObjCInterfaceDecl>(D)->getImplementation())
4280a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4281b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4282f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4283b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProperty:
4284b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: We don't really know where to find the
4285b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // ObjCPropertyImplDecls that implement this property.
4286b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4287b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4288b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCompatibleAlias:
4289b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4290b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
4291b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!Class->isForwardDecl())
4292a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek        return MakeCXCursor(Class, TU);
4293f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4294b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4295b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
42961f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::ObjCForwardProtocol:
42971f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<ObjCForwardProtocolDecl>(D),
4298a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
4299b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
43001f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::ObjCClass:
43019e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar    return MakeCursorOverloadedDeclRef(cast<ObjCClassDecl>(D), D->getLocation(),
4302a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       TU);
4303b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4304b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Friend:
4305b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
4306a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4307b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4308b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4309b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FriendTemplate:
4310b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
4311a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4312b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4313b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4314b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4315b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getNullCursor();
4316b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4317b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4318b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregorunsigned clang_isCursorDefinition(CXCursor C) {
4319b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4320b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return 0;
4321b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4322b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getCursorDefinition(C) == C;
4323b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4324b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
43251a9d0503b67a499797141af0fd6d315d5045f0eaDouglas GregorCXCursor clang_getCanonicalCursor(CXCursor C) {
43261a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  if (!clang_isDeclaration(C.kind))
43271a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return C;
43281a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
4329e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis  if (Decl *D = getCursorDecl(C)) {
4330debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis    if (ObjCCategoryImplDecl *CatImplD = dyn_cast<ObjCCategoryImplDecl>(D))
4331debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis      if (ObjCCategoryDecl *CatD = CatImplD->getCategoryDecl())
4332debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis        return MakeCXCursor(CatD, getCursorTU(C));
4333debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis
4334e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis    if (ObjCImplDecl *ImplD = dyn_cast<ObjCImplDecl>(D))
4335e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis      if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
4336e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis        return MakeCXCursor(IFD, getCursorTU(C));
4337e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis
43381a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return MakeCXCursor(D->getCanonicalDecl(), getCursorTU(C));
4339e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis  }
43401a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
43411a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  return C;
43421a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor}
43431a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
43441f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregorunsigned clang_getNumOverloadedDecls(CXCursor C) {
43457c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (C.kind != CXCursor_OverloadedDeclRef)
43461f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return 0;
43471f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43481f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
43491f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
43501f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return E->getNumDecls();
43511f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43521f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
43531f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
43541f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return S->size();
43551f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43561f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
43571f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4358826faa22bae112e01293a58534a40711043cce65Argyrios Kyrtzidis    return Using->shadow_size();
435995ed7784a335aca53b0c6e952cf31a4cfb633360Fariborz Jahanian  if (isa<ObjCClassDecl>(D))
436095ed7784a335aca53b0c6e952cf31a4cfb633360Fariborz Jahanian    return 1;
43611f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCForwardProtocolDecl *Protocols =dyn_cast<ObjCForwardProtocolDecl>(D))
43621f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return Protocols->protocol_size();
43631f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43641f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return 0;
43651f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
43661f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43671f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas GregorCXCursor clang_getOverloadedDecl(CXCursor cursor, unsigned index) {
43687c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (cursor.kind != CXCursor_OverloadedDeclRef)
43691f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
43701f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43711f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (index >= clang_getNumOverloadedDecls(cursor))
43721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
43731f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4374a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(cursor);
43751f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(cursor).first;
43761f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
4377a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(E->decls_begin()[index], TU);
43781f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43791f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
43801f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
4381a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(S->begin()[index], TU);
43821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43831f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
43841f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D)) {
43851f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // FIXME: This is, unfortunately, linear time.
43861f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    UsingDecl::shadow_iterator Pos = Using->shadow_begin();
43871f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    std::advance(Pos, index);
4388a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<UsingShadowDecl>(*Pos)->getTargetDecl(), TU);
43891f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
43901f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
439195ed7784a335aca53b0c6e952cf31a4cfb633360Fariborz Jahanian    return MakeCXCursor(Classes->getForwardInterfaceDecl(), TU);
43921f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCForwardProtocolDecl *Protocols = dyn_cast<ObjCForwardProtocolDecl>(D))
4393a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(Protocols->protocol_begin()[index], TU);
43941f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43951f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return clang_getNullCursor();
43961f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
43971f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43980d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbarvoid clang_getDefinitionSpellingAndExtent(CXCursor C,
43994ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **startBuf,
44004ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **endBuf,
44014ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startLine,
44024ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startColumn,
44034ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *endLine,
44049ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          unsigned *endColumn) {
4405283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  assert(getCursorDecl(C) && "CXCursor has null decl");
4406283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  NamedDecl *ND = static_cast<NamedDecl *>(getCursorDecl(C));
44074ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  FunctionDecl *FD = dyn_cast<FunctionDecl>(ND);
44084ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
4409f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
44104ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  SourceManager &SM = FD->getASTContext().getSourceManager();
44114ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startBuf = SM.getCharacterData(Body->getLBracLoc());
44124ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endBuf = SM.getCharacterData(Body->getRBracLoc());
44134ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
44144ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
44154ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
44164ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
44174ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff}
4418f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4419430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4420430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas GregorCXSourceRange clang_getCursorReferenceNameRange(CXCursor C, unsigned NameFlags,
4421430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                                                unsigned PieceIndex) {
4422430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  RefNamePieces Pieces;
4423430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4424430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  switch (C.kind) {
4425430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_MemberRefExpr:
4426430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (MemberExpr *E = dyn_cast<MemberExpr>(getCursorExpr(C)))
4427430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Pieces = buildPieces(NameFlags, true, E->getMemberNameInfo(),
4428430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getQualifierLoc().getSourceRange());
4429430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4430430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4431430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_DeclRefExpr:
4432430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (DeclRefExpr *E = dyn_cast<DeclRefExpr>(getCursorExpr(C)))
4433430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Pieces = buildPieces(NameFlags, false, E->getNameInfo(),
4434430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getQualifierLoc().getSourceRange(),
4435430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getExplicitTemplateArgsOpt());
4436430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4437430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4438430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_CallExpr:
4439430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (CXXOperatorCallExpr *OCE =
4440430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        dyn_cast<CXXOperatorCallExpr>(getCursorExpr(C))) {
4441430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Expr *Callee = OCE->getCallee();
4442430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Callee))
4443430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        Callee = ICE->getSubExpr();
4444430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4445430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Callee))
4446430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        Pieces = buildPieces(NameFlags, false, DRE->getNameInfo(),
4447430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                             DRE->getQualifierLoc().getSourceRange());
4448430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    }
4449430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4450430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4451430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  default:
4452430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4453430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  }
4454430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4455430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  if (Pieces.empty()) {
4456430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (PieceIndex == 0)
4457430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      return clang_getCursorExtent(C);
4458430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  } else if (PieceIndex < Pieces.size()) {
4459430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      SourceRange R = Pieces[PieceIndex];
4460430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (R.isValid())
4461430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        return cxloc::translateSourceRange(getCursorContext(C), R);
4462430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  }
4463430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4464430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  return clang_getNullRange();
4465430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor}
4466430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
44670a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregorvoid clang_enableStackTraces(void) {
44680a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  llvm::sys::PrintStackTraceOnErrorSignal();
44690a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor}
44700a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor
4471995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbarvoid clang_executeOnThread(void (*fn)(void*), void *user_data,
4472995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar                           unsigned stack_size) {
4473995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar  llvm::llvm_execute_on_thread(fn, user_data, stack_size);
4474995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar}
4475995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar
4476fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
4477fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
4478fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
4479fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor// Token-based Operations.
4480fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
4481fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4482fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor/* CXToken layout:
4483fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[0]: a CXTokenKind
4484fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[1]: starting token location
4485fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[2]: token length
4486fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[3]: reserved
4487f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek *   ptr_data: for identifiers and keywords, an IdentifierInfo*.
4488fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   otherwise unused.
4489fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor */
4490fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorextern "C" {
4491fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4492fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXTokenKind clang_getTokenKind(CXToken CXTok) {
4493fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return static_cast<CXTokenKind>(CXTok.int_data[0]);
4494fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4495fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4496fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
4497fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  switch (clang_getTokenKind(CXTok)) {
4498fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Identifier:
4499fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Keyword:
4500fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We know we have an IdentifierInfo*, so use that.
4501ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(static_cast<IdentifierInfo *>(CXTok.ptr_data)
4502ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                            ->getNameStart());
4503fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4504fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Literal: {
4505fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We have stashed the starting pointer in the ptr_data field. Use it.
4506fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    const char *Text = static_cast<const char *>(CXTok.ptr_data);
45075f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    return createCXString(StringRef(Text, CXTok.int_data[2]));
4508fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4509f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4510fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Punctuation:
4511fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Comment:
4512fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    break;
4513fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4514f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4515f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // We have to find the starting buffer pointer the hard way, by
4516fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // deconstructing the source location.
4517a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4518fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4519ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
4520f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4521fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
4522fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> LocInfo
4523a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    = CXXUnit->getSourceManager().getDecomposedSpellingLoc(Loc);
4524f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
45255f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  StringRef Buffer
4526f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
4527f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  if (Invalid)
4528aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor    return createCXString("");
4529fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4530f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  return createCXString(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
4531fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4532f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4533fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
4534a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4535fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4536fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return clang_getNullLocation();
4537f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4538fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
4539fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4540fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4541fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4542fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
4543a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
45445352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (!CXXUnit)
45455352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4546f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4547f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return cxloc::translateSourceRange(CXXUnit->getASTContext(),
4548fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4549fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4550f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4551ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisstatic void getTokens(ASTUnit *CXXUnit, SourceRange Range,
4552ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                      SmallVectorImpl<CXToken> &CXTokens) {
4553fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
4554fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
4555ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(Range.getBegin());
4556fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
4557ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(Range.getEnd());
4558f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4559fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Cannot tokenize across files.
4560fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (BeginLocInfo.first != EndLocInfo.first)
4561fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4562f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4563f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Create a lexer
4564f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
45655f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  StringRef Buffer
4566f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
456747a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor  if (Invalid)
456847a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor    return;
4569aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor
4570fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
4571fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor            CXXUnit->getASTContext().getLangOptions(),
4572f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer            Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
4573fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lex.SetCommentRetentionState(true);
4574f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4575fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Lex tokens until we hit the end of the range.
4576f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
4577fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Token Tok;
4578096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall  bool previousWasAt = false;
4579fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  do {
4580fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Lex the next token
4581fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    Lex.LexFromRawLexer(Tok);
4582fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.is(tok::eof))
4583fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      break;
4584f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4585fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Initialize the CXToken.
4586fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXToken CXTok;
4587f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4588fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Common fields
4589fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
4590fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[2] = Tok.getLength();
4591fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[3] = 0;
4592f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4593fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Kind-specific fields
4594fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.isLiteral()) {
4595fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Literal;
4596fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = (void *)Tok.getLiteralData();
4597c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara    } else if (Tok.is(tok::raw_identifier)) {
4598aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor      // Lookup the identifier to determine whether we have a keyword.
4599fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      IdentifierInfo *II
4600c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok);
4601aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek
4602096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall      if ((II->getObjCKeywordID() != tok::objc_not_keyword) && previousWasAt) {
4603aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek        CXTok.int_data[0] = CXToken_Keyword;
4604aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4605aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      else {
4606c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        CXTok.int_data[0] = Tok.is(tok::identifier)
4607c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          ? CXToken_Identifier
4608c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          : CXToken_Keyword;
4609aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4610fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = II;
4611fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else if (Tok.is(tok::comment)) {
4612fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Comment;
4613fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4614fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else {
4615fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Punctuation;
4616fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4617fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    }
4618fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTokens.push_back(CXTok);
4619096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall    previousWasAt = Tok.is(tok::at);
4620fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  } while (Lex.getBufferLocation() <= EffectiveBufferEnd);
4621ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis}
4622ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4623ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisvoid clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
4624ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                    CXToken **Tokens, unsigned *NumTokens) {
4625ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (Tokens)
4626ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    *Tokens = 0;
4627ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (NumTokens)
4628ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    *NumTokens = 0;
4629ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4630ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4631ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (!CXXUnit || !Tokens || !NumTokens)
4632ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
4633ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4634ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
4635ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4636ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SourceRange R = cxloc::translateCXSourceRange(Range);
4637ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (R.isInvalid())
4638ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
4639ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4640ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SmallVector<CXToken, 32> CXTokens;
4641ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  getTokens(CXXUnit, R, CXTokens);
4642f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4643fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (CXTokens.empty())
4644fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4645f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4646fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
4647fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
4648fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *NumTokens = CXTokens.size();
4649fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
46500045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
46516db610934bedc6896393c1e1099525b35380acd6Ted Kremenekvoid clang_disposeTokens(CXTranslationUnit TU,
46526db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                         CXToken *Tokens, unsigned NumTokens) {
46536db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  free(Tokens);
46546db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
46556db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46566db610934bedc6896393c1e1099525b35380acd6Ted Kremenek} // end: extern "C"
46576db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46586db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
46596db610934bedc6896393c1e1099525b35380acd6Ted Kremenek// Token annotation APIs.
46606db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
46616db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46620045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregortypedef llvm::DenseMap<unsigned, CXCursor> AnnotateTokensData;
4663fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
4664fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXCursor parent,
4665fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXClientData client_data);
46666db610934bedc6896393c1e1099525b35380acd6Ted Kremeneknamespace {
46676db610934bedc6896393c1e1099525b35380acd6Ted Kremenekclass AnnotateTokensWorker {
46686db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  AnnotateTokensData &Annotated;
466911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXToken *Tokens;
467011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXCursor *Cursors;
467111949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  unsigned NumTokens;
4672fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned TokIdx;
46734419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  unsigned PreprocessingTokIdx;
4674fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CursorVisitor AnnotateVis;
4675fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceManager &SrcMgr;
4676f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool HasContextSensitiveKeywords;
4677f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4678fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  bool MoreTokens() const { return TokIdx < NumTokens; }
4679fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned NextToken() const { return TokIdx; }
4680fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AdvanceToken() { ++TokIdx; }
4681fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation GetTokenLoc(unsigned tokI) {
4682fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
4683fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
46845f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  bool isFunctionMacroToken(unsigned tokI) const {
4685a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return Tokens[tokI].int_data[3] != 0;
4686a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
46875f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  SourceLocation getFunctionMacroTokenLoc(unsigned tokI) const {
4688a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[3]);
4689a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4690a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4691a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  void annotateAndAdvanceTokens(CXCursor, RangeComparisonResult, SourceRange);
46925f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  void annotateAndAdvanceFunctionMacroTokens(CXCursor, RangeComparisonResult,
46935f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis                                             SourceRange);
4694fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
46956db610934bedc6896393c1e1099525b35380acd6Ted Kremenekpublic:
469611949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  AnnotateTokensWorker(AnnotateTokensData &annotated,
4697fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                       CXToken *tokens, CXCursor *cursors, unsigned numTokens,
4698a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                       CXTranslationUnit tu, SourceRange RegionOfInterest)
469911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    : Annotated(annotated), Tokens(tokens), Cursors(cursors),
47004419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      NumTokens(numTokens), TokIdx(0), PreprocessingTokIdx(0),
4701a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      AnnotateVis(tu,
4702f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                  AnnotateTokensVisitor, this,
4703f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                  /*VisitPreprocessorLast=*/true,
4704e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                  /*VisitIncludedEntities=*/false,
4705f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                  RegionOfInterest),
4706f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      SrcMgr(static_cast<ASTUnit*>(tu->TUData)->getSourceManager()),
4707f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      HasContextSensitiveKeywords(false) { }
470811949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4709fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
47106db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
4711fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AnnotateTokens(CXCursor parent);
4712ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek  void AnnotateTokens() {
4713a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    AnnotateTokens(clang_getTranslationUnitCursor(AnnotateVis.getTU()));
4714ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek  }
4715f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4716f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// \brief Determine whether the annotator saw any cursors that have
4717f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// context-sensitive keywords.
4718f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool hasContextSensitiveKeywords() const {
4719f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    return HasContextSensitiveKeywords;
4720f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
47216db610934bedc6896393c1e1099525b35380acd6Ted Kremenek};
47226db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
47230045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
4724fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekvoid AnnotateTokensWorker::AnnotateTokens(CXCursor parent) {
4725fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Walk the AST within the region of interest, annotating tokens
4726fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // along the way.
4727fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(parent);
4728fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4729fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = 0 ; I < TokIdx ; ++I) {
473011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
47314419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    if (Pos != Annotated.end() &&
47324419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        (clang_isInvalid(Cursors[I].kind) ||
47334419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor         Pos->second.kind != CXCursor_PreprocessingDirective))
4734fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      Cursors[I] = Pos->second;
4735fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4736fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4737fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Finish up annotating any tokens left.
4738fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (!MoreTokens())
4739fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return;
474011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4741fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor &C = clang_getNullCursor();
4742fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = TokIdx ; I < NumTokens ; ++I) {
4743fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
4744fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = (Pos == Annotated.end()) ? C : Pos->second;
474511949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  }
474611949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek}
474711949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4748a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief It annotates and advances tokens with a cursor until the comparison
4749a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis//// between the cursor location and the source range is the same as
4750a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \arg compResult.
4751a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis///
4752a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// Pass RangeBefore to annotate tokens with a cursor until a range is reached.
4753a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// Pass RangeOverlap to annotate tokens inside a range.
4754a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisvoid AnnotateTokensWorker::annotateAndAdvanceTokens(CXCursor updateC,
4755a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               RangeComparisonResult compResult,
4756a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               SourceRange range) {
4757a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  while (MoreTokens()) {
4758a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    const unsigned I = NextToken();
47595f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis    if (isFunctionMacroToken(I))
47605f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis      return annotateAndAdvanceFunctionMacroTokens(updateC, compResult, range);
4761a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4762a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    SourceLocation TokLoc = GetTokenLoc(I);
4763a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
4764a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      Cursors[I] = updateC;
4765a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      AdvanceToken();
4766a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      continue;
4767a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
4768a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    break;
4769a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4770a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
4771a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4772a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief Special annotation handling for macro argument tokens.
47735f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidisvoid AnnotateTokensWorker::annotateAndAdvanceFunctionMacroTokens(
47745f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis                                               CXCursor updateC,
4775a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               RangeComparisonResult compResult,
4776a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               SourceRange range) {
47775f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  assert(MoreTokens());
47785f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  assert(isFunctionMacroToken(NextToken()) &&
4779a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis         "Should be called only for macro arg tokens");
4780a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4781a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // This works differently than annotateAndAdvanceTokens; because expanded
4782a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // macro arguments can have arbitrary translation-unit source order, we do not
4783a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // advance the token index one by one until a token fails the range test.
4784a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // We only advance once past all of the macro arg tokens if all of them
4785a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // pass the range test. If one of them fails we keep the token index pointing
4786a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // at the start of the macro arg tokens so that the failing token will be
4787a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // annotated by a subsequent annotation try.
4788a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4789a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  bool atLeastOneCompFail = false;
4790a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4791a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned I = NextToken();
47925f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  for (; I < NumTokens && isFunctionMacroToken(I); ++I) {
47935f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis    SourceLocation TokLoc = getFunctionMacroTokenLoc(I);
4794a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (TokLoc.isFileID())
4795a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      continue; // not macro arg token, it's parens or comma.
4796a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
4797a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (clang_isInvalid(clang_getCursorKind(Cursors[I])))
4798a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        Cursors[I] = updateC;
4799a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    } else
4800a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      atLeastOneCompFail = true;
4801a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4802a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4803a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  if (!atLeastOneCompFail)
4804a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    TokIdx = I; // All of the tokens were handled, advance beyond all of them.
4805a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
4806a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
48076db610934bedc6896393c1e1099525b35380acd6Ted Kremenekenum CXChildVisitResult
48084419b675577d7c281a659fab1fec10e1bfbe04c5Douglas GregorAnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
4809fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CXSourceLocation Loc = clang_getCursorLocation(cursor);
48104419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  SourceRange cursorRange = getRawCursorExtent(cursor);
481181d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor  if (cursorRange.isInvalid())
481281d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor    return CXChildVisit_Recurse;
4813f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4814f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (!HasContextSensitiveKeywords) {
4815f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C properties can have context-sensitive keywords.
4816f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    if (cursor.kind == CXCursor_ObjCPropertyDecl) {
4817f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCPropertyDecl *Property
4818f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(cursor)))
4819f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        HasContextSensitiveKeywords = Property->getPropertyAttributesAsWritten() != 0;
4820f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4821f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C methods can have context-sensitive keywords.
4822f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_ObjCInstanceMethodDecl ||
4823f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ObjCClassMethodDecl) {
4824f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCMethodDecl *Method
4825f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
4826f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->getObjCDeclQualifier())
4827f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4828f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        else {
4829f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
4830f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                                           PEnd = Method->param_end();
4831f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor               P != PEnd; ++P) {
4832f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((*P)->getObjCDeclQualifier()) {
4833f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              HasContextSensitiveKeywords = true;
4834f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              break;
4835f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            }
4836f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          }
4837f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4838f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4839f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4840f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ methods can have context-sensitive keywords.
4841f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_CXXMethod) {
4842f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (CXXMethodDecl *Method
4843f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(cursor))) {
4844f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->hasAttr<FinalAttr>() || Method->hasAttr<OverrideAttr>())
4845f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4846f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4847f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4848f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ classes can have context-sensitive keywords.
4849f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_StructDecl ||
4850f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassDecl ||
4851f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplate ||
4852f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplatePartialSpecialization) {
4853f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Decl *D = getCursorDecl(cursor))
4854f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (D->hasAttr<FinalAttr>())
4855f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4856f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4857f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
4858f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
48594419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  if (clang_isPreprocessing(cursor.kind)) {
4860cea731a9cb7de3f473d60e5ea544e25621cebd76Chandler Carruth    // For macro expansions, just note where the beginning of the macro
4861cea731a9cb7de3f473d60e5ea544e25621cebd76Chandler Carruth    // expansion occurs.
48629b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth    if (cursor.kind == CXCursor_MacroExpansion) {
48634419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      Annotated[Loc.int_data] = cursor;
48644419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      return CXChildVisit_Recurse;
48654419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
48664419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
48674419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Items in the preprocessing record are kept separate from items in
48684419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // declarations, so we keep a separate token index.
48694419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    unsigned SavedTokIdx = TokIdx;
48704419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = PreprocessingTokIdx;
48714419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
48724419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Skip tokens up until we catch up to the beginning of the preprocessing
48734419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // entry.
48744419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
48754419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
48764419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
48774419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
48784419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
48794419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
48804419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
48814419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
48824419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
48834419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
48844419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
48854419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
48864419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
48874419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
48884419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Look at all of the tokens within this range.
48894419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
48904419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
48914419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
48924419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
48934419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
4894b219cfc4d75f0a03630b7c4509ef791b7e97b2c8David Blaikie        llvm_unreachable("Infeasible");
48954419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
48964419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
48974419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
48984419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        Cursors[I] = cursor;
48994419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
49004419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
49014419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
49024419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
49034419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
49044419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
49054419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Save the preprocessing token index; restore the non-preprocessing
49064419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // token index.
49074419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    PreprocessingTokIdx = TokIdx;
49084419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = SavedTokIdx;
49090045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor    return CXChildVisit_Recurse;
49100045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  }
4911fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4912fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (cursorRange.isInvalid())
4913fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return CXChildVisit_Continue;
4914a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek
4915fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation L = SourceLocation::getFromRawEncoding(Loc.int_data);
4916fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4917a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  // Adjust the annotated range based specific declarations.
4918a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  const enum CXCursorKind cursorK = clang_getCursorKind(cursor);
4919a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  if (cursorK >= CXCursor_FirstDecl && cursorK <= CXCursor_LastDecl) {
492023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    Decl *D = cxcursor::getCursorDecl(cursor);
49212494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
49222494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
492323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
49242494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
49252494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
49262494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
49272494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
49282494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
4929a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek    }
49302494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
49312494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && L.isValid() &&
49322494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, L))
49332494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      cursorRange.setBegin(StartLoc);
4934a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  }
493581d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor
49363f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // If the location of the cursor occurs within a macro instantiation, record
49373f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // the spelling location of the cursor in our annotation map.  We can then
49383f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // paper over the token labelings during a post-processing step to try and
49393f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // get cursor mappings for tokens that are the *arguments* of a macro
49403f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // instantiation.
49413f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  if (L.isMacroID()) {
49423f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    unsigned rawEncoding = SrcMgr.getSpellingLoc(L).getRawEncoding();
49433f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // Only invalidate the old annotation if it isn't part of a preprocessing
49443f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // directive.  Here we assume that the default construction of CXCursor
49453f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // results in CXCursor.kind being an initialized value (i.e., 0).  If
49463f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // this isn't the case, we can fix by doing lookup + insertion.
49474419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
49483f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    CXCursor &oldC = Annotated[rawEncoding];
49493f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    if (!clang_isPreprocessing(oldC.kind))
49503f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek      oldC = cursor;
49513f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  }
49523f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek
4953fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const enum CXCursorKind K = clang_getCursorKind(parent);
4954fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor updateC =
4955d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek    (clang_isInvalid(K) || K == CXCursor_TranslationUnit)
4956d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek     ? clang_getNullCursor() : parent;
4957fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4958a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  annotateAndAdvanceTokens(updateC, RangeBefore, cursorRange);
4959fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
49605517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // Avoid having the cursor of an expression "overwrite" the annotation of the
49615517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // variable declaration that it belongs to.
49625517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // This can happen for C++ constructor expressions whose range generally
49635517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // include the variable declaration, e.g.:
49645517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  //  MyCXXClass foo; // Make sure we don't annotate 'foo' as a CallExpr cursor.
49655517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  if (clang_isExpression(cursorK)) {
49665517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis    Expr *E = getCursorExpr(cursor);
49678ccac3de1335f1cfd7cea56ba1cefcf0b724ce3fArgyrios Kyrtzidis    if (Decl *D = getCursorParentDecl(cursor)) {
49685517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      const unsigned I = NextToken();
49695517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      if (E->getLocStart().isValid() && D->getLocation().isValid() &&
49705517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis          E->getLocStart() == D->getLocation() &&
49715517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis          E->getLocStart() == GetTokenLoc(I)) {
49725517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis        Cursors[I] = updateC;
49735517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis        AdvanceToken();
49745517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      }
49755517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis    }
49765517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  }
49775517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis
4978fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Visit children to get their cursor information.
4979fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned BeforeChildren = NextToken();
4980fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(cursor);
4981fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned AfterChildren = NextToken();
4982fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4983a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // Scan the tokens that are at the end of the cursor, but are not captured
4984a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // but the child cursors.
4985a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  annotateAndAdvanceTokens(cursor, RangeOverlap, cursorRange);
49866db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
4987fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the beginning of the cursor, but are not
4988fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // capture by the child cursors.
4989fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
4990fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
4991fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      break;
49924419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
4993fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
4994fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4995fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4996fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return CXChildVisit_Continue;
49970045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor}
49980045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
49996db610934bedc6896393c1e1099525b35380acd6Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
50006db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXCursor parent,
50016db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXClientData client_data) {
50026db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
50036db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
50046db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
50056628a614c504263ae539462f049d523dd07ac1baTed Kremeneknamespace {
5006a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5007a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief Uses the macro expansions in the preprocessing record to find
5008a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// and mark tokens that are macro arguments. This info is used by the
5009a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// AnnotateTokensWorker.
5010a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisclass MarkMacroArgTokensVisitor {
5011a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  SourceManager &SM;
5012a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  CXToken *Tokens;
5013a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned NumTokens;
5014a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned CurIdx;
5015a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5016a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidispublic:
5017a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  MarkMacroArgTokensVisitor(SourceManager &SM,
5018a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                            CXToken *tokens, unsigned numTokens)
5019a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    : SM(SM), Tokens(tokens), NumTokens(numTokens), CurIdx(0) { }
5020a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5021a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  CXChildVisitResult visit(CXCursor cursor, CXCursor parent) {
5022a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (cursor.kind != CXCursor_MacroExpansion)
5023a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Continue;
5024a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5025a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    SourceRange macroRange = getCursorMacroExpansion(cursor)->getSourceRange();
5026a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (macroRange.getBegin() == macroRange.getEnd())
5027a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Continue; // it's not a function macro.
5028a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5029a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    for (; CurIdx < NumTokens; ++CurIdx) {
5030a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (!SM.isBeforeInTranslationUnit(getTokenLoc(CurIdx),
5031a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                        macroRange.getBegin()))
5032a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        break;
5033a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
5034a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5035a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (CurIdx == NumTokens)
5036a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Break;
5037a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5038a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    for (; CurIdx < NumTokens; ++CurIdx) {
5039a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      SourceLocation tokLoc = getTokenLoc(CurIdx);
5040a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (!SM.isBeforeInTranslationUnit(tokLoc, macroRange.getEnd()))
5041a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        break;
5042a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
50435f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis      setFunctionMacroTokenLoc(CurIdx, SM.getMacroArgExpandedLocation(tokLoc));
5044a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
5045a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5046a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (CurIdx == NumTokens)
5047a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Break;
5048a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5049a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return CXChildVisit_Continue;
5050a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5051a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5052a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisprivate:
5053a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  SourceLocation getTokenLoc(unsigned tokI) {
5054a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
5055a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5056a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
50575f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  void setFunctionMacroTokenLoc(unsigned tokI, SourceLocation loc) {
5058a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // The third field is reserved and currently not used. Use it here
5059a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // to mark macro arg expanded tokens with their expanded locations.
5060a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    Tokens[tokI].int_data[3] = loc.getRawEncoding();
5061a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5062a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis};
5063a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5064a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis} // end anonymous namespace
5065a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5066a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisstatic CXChildVisitResult
5067a676379b26edc959193f9f919ba9c6d296a57824Argyrios KyrtzidisMarkMacroArgTokensVisitorDelegate(CXCursor cursor, CXCursor parent,
5068a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                  CXClientData client_data) {
5069a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  return static_cast<MarkMacroArgTokensVisitor*>(client_data)->visit(cursor,
5070a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                                                     parent);
5071a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
5072a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5073a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisnamespace {
50746628a614c504263ae539462f049d523dd07ac1baTed Kremenek  struct clang_annotateTokens_Data {
50756628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXTranslationUnit TU;
50766628a614c504263ae539462f049d523dd07ac1baTed Kremenek    ASTUnit *CXXUnit;
50776628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXToken *Tokens;
50786628a614c504263ae539462f049d523dd07ac1baTed Kremenek    unsigned NumTokens;
50796628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXCursor *Cursors;
50806628a614c504263ae539462f049d523dd07ac1baTed Kremenek  };
5081ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek}
5082ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek
5083ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisstatic void annotatePreprocessorTokens(CXTranslationUnit TU,
5084ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                       SourceRange RegionOfInterest,
5085ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                       AnnotateTokensData &Annotated) {
5086ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
5087ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5088ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SourceManager &SourceMgr = CXXUnit->getSourceManager();
5089ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  std::pair<FileID, unsigned> BeginLocInfo
5090ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(RegionOfInterest.getBegin());
5091ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  std::pair<FileID, unsigned> EndLocInfo
5092ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(RegionOfInterest.getEnd());
5093ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5094ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (BeginLocInfo.first != EndLocInfo.first)
5095ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
5096ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5097ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  StringRef Buffer;
5098ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  bool Invalid = false;
5099ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
5100ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (Buffer.empty() || Invalid)
5101ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
5102ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5103ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
5104ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            CXXUnit->getASTContext().getLangOptions(),
5105ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            Buffer.begin(), Buffer.data() + BeginLocInfo.second,
5106ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            Buffer.end());
5107ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Lex.SetCommentRetentionState(true);
5108ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5109ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  // Lex tokens in raw mode until we hit the end of the range, to avoid
5110ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  // entering #includes or expanding macros.
5111ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  while (true) {
5112ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    Token Tok;
5113ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    Lex.LexFromRawLexer(Tok);
5114ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5115ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  reprocess:
5116ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
5117ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // We have found a preprocessing directive. Gobble it up so that we
5118ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // don't see it while preprocessing these tokens later, but keep track
5119ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // of all of the token locations inside this preprocessing directive so
5120ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // that we can annotate them appropriately.
5121ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      //
5122ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // FIXME: Some simple tests here could identify macro definitions and
5123ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // #undefs, to provide specific cursor kinds for those.
5124ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      SmallVector<SourceLocation, 32> Locations;
5125ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      do {
5126ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Locations.push_back(Tok.getLocation());
5127ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Lex.LexFromRawLexer(Tok);
5128ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      } while (!Tok.isAtStartOfLine() && !Tok.is(tok::eof));
5129ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5130ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      using namespace cxcursor;
5131ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      CXCursor Cursor
5132ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      = MakePreprocessingDirectiveCursor(SourceRange(Locations.front(),
5133ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                                     Locations.back()),
5134ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                         TU);
5135ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      for (unsigned I = 0, N = Locations.size(); I != N; ++I) {
5136ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Annotated[Locations[I].getRawEncoding()] = Cursor;
5137ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      }
5138ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5139ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      if (Tok.isAtStartOfLine())
5140ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        goto reprocess;
5141ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5142ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      continue;
5143ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    }
5144ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5145ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    if (Tok.is(tok::eof))
5146ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      break;
5147ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
5148ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis}
5149ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
51506628a614c504263ae539462f049d523dd07ac1baTed Kremenek// This gets run a separate thread to avoid stack blowout.
51516628a614c504263ae539462f049d523dd07ac1baTed Kremenekstatic void clang_annotateTokensImpl(void *UserData) {
51526628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXTranslationUnit TU = ((clang_annotateTokens_Data*)UserData)->TU;
51536628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = ((clang_annotateTokens_Data*)UserData)->CXXUnit;
51546628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXToken *Tokens = ((clang_annotateTokens_Data*)UserData)->Tokens;
51556628a614c504263ae539462f049d523dd07ac1baTed Kremenek  const unsigned NumTokens = ((clang_annotateTokens_Data*)UserData)->NumTokens;
51566628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor *Cursors = ((clang_annotateTokens_Data*)UserData)->Cursors;
5157fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
51580396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Determine the region of interest, which contains all of the tokens.
51590045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  SourceRange RegionOfInterest;
51606628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setBegin(
51616628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU, Tokens[0])));
51626628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setEnd(
51636628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU,
51646628a614c504263ae539462f049d523dd07ac1baTed Kremenek                                                         Tokens[NumTokens-1])));
5165fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
51660396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // A mapping from the source locations found when re-lexing or traversing the
51670396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // region of interest to the corresponding cursors.
51680045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  AnnotateTokensData Annotated;
5169ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5170fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Relex the tokens within the source range to look for preprocessing
51710396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // directives.
5172ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  annotatePreprocessorTokens(TU, RegionOfInterest, Annotated);
51736628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5174a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
5175a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // Search and mark tokens that are macro argument expansions.
5176a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    MarkMacroArgTokensVisitor Visitor(CXXUnit->getSourceManager(),
5177a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                      Tokens, NumTokens);
5178a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    CursorVisitor MacroArgMarker(TU,
5179a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                 MarkMacroArgTokensVisitorDelegate, &Visitor,
5180f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                 /*VisitPreprocessorLast=*/true,
5181e70984629f3accf7e1e7187d06bd653dc8e315f2Argyrios Kyrtzidis                                 /*VisitIncludedEntities=*/false,
5182f226ff9fe8c8db6c5b74a61ce649eda1491c3502Argyrios Kyrtzidis                                 RegionOfInterest);
5183a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    MacroArgMarker.visitPreprocessedEntitiesInRegion();
5184a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5185a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
51860396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Annotate all of the source locations in the region of interest that map to
5187fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // a specific cursor.
5188fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  AnnotateTokensWorker W(Annotated, Tokens, Cursors, NumTokens,
5189a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                         TU, RegionOfInterest);
51906628a614c504263ae539462f049d523dd07ac1baTed Kremenek
51916c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // FIXME: We use a ridiculous stack size here because the data-recursion
51926c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm uses a large stack frame than the non-data recursive version,
51936c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // and AnnotationTokensWorker currently transforms the data-recursion
51946c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm back into a traditional recursion by explicitly calling
51956c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // VisitChildren().  We will need to remove this explicit recursive call.
51966628a614c504263ae539462f049d523dd07ac1baTed Kremenek  W.AnnotateTokens();
51976628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5198f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // If we ran into any entities that involve context-sensitive keywords,
5199f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // take another pass through the tokens to mark them as such.
5200f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (W.hasContextSensitiveKeywords()) {
5201f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    for (unsigned I = 0; I != NumTokens; ++I) {
5202f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (clang_getTokenKind(Tokens[I]) != CXToken_Identifier)
5203f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5204f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
5205f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCPropertyDecl) {
5206f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
5207f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (ObjCPropertyDecl *Property
52086628a614c504263ae539462f049d523dd07ac1baTed Kremenek            = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(Cursors[I]))) {
5209f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if (Property->getPropertyAttributesAsWritten() != 0 &&
5210f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              llvm::StringSwitch<bool>(II->getName())
52116628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readonly", true)
52126628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("assign", true)
5213f85e193739c953358c865005855253af4f68a497John McCall              .Case("unsafe_unretained", true)
52146628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readwrite", true)
52156628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("retain", true)
52166628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("copy", true)
52176628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("nonatomic", true)
52186628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("atomic", true)
52196628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("getter", true)
52206628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("setter", true)
5221f85e193739c953358c865005855253af4f68a497John McCall              .Case("strong", true)
5222f85e193739c953358c865005855253af4f68a497John McCall              .Case("weak", true)
52236628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Default(false))
5224f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            Tokens[I].int_data[0] = CXToken_Keyword;
5225f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
5226f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5227f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5228f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
5229f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCInstanceMethodDecl ||
5230f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_ObjCClassMethodDecl) {
5231f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
5232f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (llvm::StringSwitch<bool>(II->getName())
52336628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("in", true)
52346628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("out", true)
52356628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("inout", true)
52366628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("oneway", true)
52376628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("bycopy", true)
52386628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("byref", true)
52396628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Default(false))
5240f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Tokens[I].int_data[0] = CXToken_Keyword;
5241f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5242f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
52436639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis
52446639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      if (Cursors[I].kind == CXCursor_CXXFinalAttr ||
52456639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis          Cursors[I].kind == CXCursor_CXXOverrideAttr) {
52466639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis        Tokens[I].int_data[0] = CXToken_Keyword;
52476628a614c504263ae539462f049d523dd07ac1baTed Kremenek        continue;
5248f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5249f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
5250f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
5251fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
52526628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52536628a614c504263ae539462f049d523dd07ac1baTed Kremenekextern "C" {
52546628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52556628a614c504263ae539462f049d523dd07ac1baTed Kremenekvoid clang_annotateTokens(CXTranslationUnit TU,
52566628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXToken *Tokens, unsigned NumTokens,
52576628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXCursor *Cursors) {
52586628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52596628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (NumTokens == 0 || !Tokens || !Cursors)
52606628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
52616628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52626628a614c504263ae539462f049d523dd07ac1baTed Kremenek  // Any token we don't specifically annotate will have a NULL cursor.
52636628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor C = clang_getNullCursor();
52646628a614c504263ae539462f049d523dd07ac1baTed Kremenek  for (unsigned I = 0; I != NumTokens; ++I)
52656628a614c504263ae539462f049d523dd07ac1baTed Kremenek    Cursors[I] = C;
52666628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52676628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
52686628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!CXXUnit)
52696628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
52706628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52716628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
52726628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52736628a614c504263ae539462f049d523dd07ac1baTed Kremenek  clang_annotateTokens_Data data = { TU, CXXUnit, Tokens, NumTokens, Cursors };
52746628a614c504263ae539462f049d523dd07ac1baTed Kremenek  llvm::CrashRecoveryContext CRC;
52756628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!RunSafely(CRC, clang_annotateTokensImpl, &data,
52766628a614c504263ae539462f049d523dd07ac1baTed Kremenek                 GetSafetyThreadStackSize() * 2)) {
52776628a614c504263ae539462f049d523dd07ac1baTed Kremenek    fprintf(stderr, "libclang: crash detected while annotating tokens\n");
52786628a614c504263ae539462f049d523dd07ac1baTed Kremenek  }
52796628a614c504263ae539462f049d523dd07ac1baTed Kremenek}
52806628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5281fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} // end: extern "C"
5282fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
5283fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
528416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek// Operations for querying linkage of a cursor.
528516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
528616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
528716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenekextern "C" {
528816b4259aecaa22b642d35d36fd89965ed700c1e0Ted KremenekCXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
52890396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
52900396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    return CXLinkage_Invalid;
52910396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
529216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  Decl *D = cxcursor::getCursorDecl(cursor);
529316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  if (NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
529416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    switch (ND->getLinkage()) {
529516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case NoLinkage: return CXLinkage_NoLinkage;
529616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case InternalLinkage: return CXLinkage_Internal;
529716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
529816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case ExternalLinkage: return CXLinkage_External;
529916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    };
530016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
530116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  return CXLinkage_Invalid;
530216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek}
530316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek} // end: extern "C"
530416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
530516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
530645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek// Operations for querying language of a cursor.
530745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
530845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
530945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekstatic CXLanguageKind getDeclLanguage(const Decl *D) {
531045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  switch (D->getKind()) {
531145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    default:
531245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      break;
531345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ImplicitParam:
531445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCAtDefsField:
531545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategory:
531645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategoryImpl:
531745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCClass:
531845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCompatibleAlias:
531945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCForwardProtocol:
532045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCImplementation:
532145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCInterface:
532245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCIvar:
532345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCMethod:
532445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProperty:
532545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCPropertyImpl:
532645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProtocol:
532745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_ObjC;
532845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConstructor:
532945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConversion:
533045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXDestructor:
533145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXMethod:
533245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXRecord:
533345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplate:
533445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplatePartialSpecialization:
533545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplateSpecialization:
533645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Friend:
533745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FriendTemplate:
533845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FunctionTemplate:
533945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::LinkageSpec:
534045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Namespace:
534145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NamespaceAlias:
534245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NonTypeTemplateParm:
534345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::StaticAssert:
534445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTemplateParm:
534545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTypeParm:
534645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingTypename:
534745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingValue:
534845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Using:
534945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingDirective:
535045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingShadow:
535145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_CPlusPlus;
535245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  }
535345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
535445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_C;
535545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
535645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
535745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekextern "C" {
535858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
535958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregorenum CXAvailabilityKind clang_getCursorAvailability(CXCursor cursor) {
536058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  if (clang_isDeclaration(cursor.kind))
536158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    if (Decl *D = cxcursor::getCursorDecl(cursor)) {
53620a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      if (isa<FunctionDecl>(D) && cast<FunctionDecl>(D)->isDeleted())
536358ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Available;
536458ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
53650a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      switch (D->getAvailability()) {
53660a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Available:
53670a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_NotYetIntroduced:
53680a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_Available;
53690a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
53700a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Deprecated:
537158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Deprecated;
53720a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
53730a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Unavailable:
53740a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_NotAvailable;
53750a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      }
537658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    }
53770a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
537858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  return CXAvailability_Available;
537958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor}
538058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
538145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted KremenekCXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
538245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  if (clang_isDeclaration(cursor.kind))
538345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    return getDeclLanguage(cxcursor::getCursorDecl(cursor));
538445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
538545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_Invalid;
538645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
53873910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
53883910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// \brief If the given cursor is the "templated" declaration
53893910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// descibing a class or function template, return the class or
53903910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// function template.
53913910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregorstatic Decl *maybeGetTemplateCursor(Decl *D) {
53923910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (!D)
53933910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    return 0;
53943910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
53953910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
53963910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (FunctionTemplateDecl *FunTmpl = FD->getDescribedFunctionTemplate())
53973910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return FunTmpl;
53983910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
53993910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D))
54003910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (ClassTemplateDecl *ClassTmpl = RD->getDescribedClassTemplate())
54013910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return ClassTmpl;
54023910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54033910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  return D;
54043910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor}
54053910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54062be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorSemanticParent(CXCursor cursor) {
54072be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
54082be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
54092be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getDeclContext();
54103910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
54113910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
54123910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54133910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
54143910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
54152be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
54162be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
54172be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54182be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isStatement(cursor.kind) || clang_isExpression(cursor.kind)) {
54192be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor))
5420a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(D, getCursorTU(cursor));
54212be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
54222be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54232be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
54242be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
54252be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54262be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorLexicalParent(CXCursor cursor) {
54272be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
54282be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
54292be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getLexicalDeclContext();
54303910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
54313910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
54323910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54333910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
54343910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
54352be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
54362be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
54372be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54382be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // FIXME: Note that we can't easily compute the lexical context of a
54392be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // statement or expression, so we return nothing.
54402be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
54412be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
54422be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54439f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_getOverriddenCursors(CXCursor cursor,
54449f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                CXCursor **overridden,
54459f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                unsigned *num_overridden) {
54469f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (overridden)
54479f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *overridden = 0;
54489f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (num_overridden)
54499f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *num_overridden = 0;
54509f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!overridden || !num_overridden)
54519f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
54529f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
5453b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  SmallVector<CXCursor, 8> Overridden;
5454b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  cxcursor::getOverriddenCursors(cursor, Overridden);
54559f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
5456b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  *num_overridden = Overridden.size();
5457b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  *overridden = new CXCursor [Overridden.size()];
5458b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  std::copy(Overridden.begin(), Overridden.end(), *overridden);
54599f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
54609f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
54619f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_disposeOverriddenCursors(CXCursor *overridden) {
54629f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  delete [] overridden;
54639f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
54649f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
5465ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas GregorCXFile clang_getIncludedFile(CXCursor cursor) {
5466ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (cursor.kind != CXCursor_InclusionDirective)
5467ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return 0;
5468ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
5469ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  InclusionDirective *ID = getCursorInclusionDirective(cursor);
5470ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  return (void *)ID->getFile();
5471ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor}
5472ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
547345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek} // end: extern "C"
547445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
54759ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
54769ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
54779ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek// C++ AST instrospection.
54789ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
54799ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
54809ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekextern "C" {
54819ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekunsigned clang_CXXMethod_isStatic(CXCursor C) {
54829ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek  if (!clang_isDeclaration(C.kind))
54839ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek    return 0;
548449f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor
548549f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  CXXMethodDecl *Method = 0;
548649f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
548749f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
548849f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
548949f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  else
549049f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
549149f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  return (Method && Method->isStatic()) ? 1 : 0;
549240b492a43bac3ed0c465772aa6921d011cfc273fTed Kremenek}
5493b12903e1a4b8d1b611b8c7e4f910665d628e68cdTed Kremenek
5494211924b563aa31421836cee7655be729ad02733fDouglas Gregorunsigned clang_CXXMethod_isVirtual(CXCursor C) {
5495211924b563aa31421836cee7655be729ad02733fDouglas Gregor  if (!clang_isDeclaration(C.kind))
5496211924b563aa31421836cee7655be729ad02733fDouglas Gregor    return 0;
5497211924b563aa31421836cee7655be729ad02733fDouglas Gregor
5498211924b563aa31421836cee7655be729ad02733fDouglas Gregor  CXXMethodDecl *Method = 0;
5499211924b563aa31421836cee7655be729ad02733fDouglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
5500211924b563aa31421836cee7655be729ad02733fDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
5501211924b563aa31421836cee7655be729ad02733fDouglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
5502211924b563aa31421836cee7655be729ad02733fDouglas Gregor  else
5503211924b563aa31421836cee7655be729ad02733fDouglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
5504211924b563aa31421836cee7655be729ad02733fDouglas Gregor  return (Method && Method->isVirtual()) ? 1 : 0;
5505211924b563aa31421836cee7655be729ad02733fDouglas Gregor}
55069ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek} // end: extern "C"
55079ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
550845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
550995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek// Attribute introspection.
551095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
551195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
551295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenekextern "C" {
551395f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted KremenekCXType clang_getIBOutletCollectionType(CXCursor C) {
551495f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  if (C.kind != CXCursor_IBOutletCollectionAttr)
5515a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return cxtype::MakeCXType(QualType(), cxcursor::getCursorTU(C));
551695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
551795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  IBOutletCollectionAttr *A =
551895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek    cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(C));
551995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
552018aa2ff4641847d7f8866e8c5912d4d0ddb858ceArgyrios Kyrtzidis  return cxtype::MakeCXType(A->getInterface(), cxcursor::getCursorTU(C));
552195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek}
552295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek} // end: extern "C"
552395f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
552495f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
552559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek// Inspecting memory usage.
552659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek//===----------------------------------------------------------------------===//
552759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5528f787002478f09af1741fb0f82a562002e6799c49Ted Kremenektypedef std::vector<CXTUResourceUsageEntry> MemUsageEntries;
552959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5530f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekstatic inline void createCXTUResourceUsageEntry(MemUsageEntries &entries,
5531f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek                                              enum CXTUResourceUsageKind k,
5532ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek                                              unsigned long amount) {
5533f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  CXTUResourceUsageEntry entry = { k, amount };
553459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  entries.push_back(entry);
553559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
553659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
553759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenekextern "C" {
553859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5539f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekconst char *clang_getTUResourceUsageName(CXTUResourceUsageKind kind) {
554059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  const char *str = "";
554159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  switch (kind) {
5542f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_AST:
554359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: expressions, declarations, and types";
554459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      break;
5545f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_Identifiers:
554659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: identifiers";
554759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      break;
5548f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_Selectors:
554959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: selectors";
5550e294ab717fc9535429ca5d8f575d41ae4441d822Ted Kremenek      break;
5551f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_GlobalCompletionResults:
55524e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek      str = "Code completion: cached global results";
5553e294ab717fc9535429ca5d8f575d41ae4441d822Ted Kremenek      break;
5554457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek    case CXTUResourceUsage_SourceManagerContentCache:
5555457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek      str = "SourceManager: content cache allocator";
5556457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek      break;
5557ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    case CXTUResourceUsage_AST_SideTables:
5558ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek      str = "ASTContext: side tables";
5559ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek      break;
5560f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    case CXTUResourceUsage_SourceManager_Membuffer_Malloc:
5561f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      str = "SourceManager: malloc'ed memory buffers";
5562f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      break;
5563f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    case CXTUResourceUsage_SourceManager_Membuffer_MMap:
5564f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      str = "SourceManager: mmap'ed memory buffers";
5565f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      break;
5566e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    case CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc:
5567e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      str = "ExternalASTSource: malloc'ed memory buffers";
5568e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      break;
5569e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    case CXTUResourceUsage_ExternalASTSource_Membuffer_MMap:
5570e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      str = "ExternalASTSource: mmap'ed memory buffers";
5571e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      break;
55725e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    case CXTUResourceUsage_Preprocessor:
55735e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      str = "Preprocessor: malloc'ed memory";
55745e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      break;
55755e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    case CXTUResourceUsage_PreprocessingRecord:
55765e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      str = "Preprocessor: PreprocessingRecord";
55775e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      break;
5578ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek    case CXTUResourceUsage_SourceManager_DataStructures:
5579ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek      str = "SourceManager: data structures and tables";
5580ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek      break;
5581d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek    case CXTUResourceUsage_Preprocessor_HeaderSearch:
5582d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek      str = "Preprocessor: header search tables";
5583d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek      break;
558459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  }
558559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  return str;
558659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
558759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5588f787002478f09af1741fb0f82a562002e6799c49Ted KremenekCXTUResourceUsage clang_getCXTUResourceUsage(CXTranslationUnit TU) {
558959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  if (!TU) {
5590f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    CXTUResourceUsage usage = { (void*) 0, 0, 0 };
559159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    return usage;
559259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  }
559359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
559459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  ASTUnit *astUnit = static_cast<ASTUnit*>(TU->TUData);
559559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  llvm::OwningPtr<MemUsageEntries> entries(new MemUsageEntries());
559659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  ASTContext &astContext = astUnit->getASTContext();
559759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
559859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used by AST nodes and types?
5599f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST,
5600ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    (unsigned long) astContext.getASTAllocatedMemory());
560159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
560259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used by identifiers?
5603f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Identifiers,
560459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    (unsigned long) astContext.Idents.getAllocator().getTotalMemory());
560559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
560659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used for selectors?
5607f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Selectors,
560859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    (unsigned long) astContext.Selectors.getTotalMemory());
560959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5610ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek  // How much memory is used by ASTContext's side tables?
5611ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST_SideTables,
5612ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    (unsigned long) astContext.getSideTableAllocatedMemory());
5613ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek
56144e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  // How much memory is used for caching global code completion results?
56154e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  unsigned long completionBytes = 0;
56164e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  if (GlobalCodeCompletionAllocator *completionAllocator =
56174e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek      astUnit->getCachedCompletionAllocator().getPtr()) {
56185e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    completionBytes = completionAllocator->getTotalMemory();
56194e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  }
5620457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  createCXTUResourceUsageEntry(*entries,
5621457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek                               CXTUResourceUsage_GlobalCompletionResults,
5622457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek                               completionBytes);
5623457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek
5624457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  // How much memory is being used by SourceManager's content cache?
5625457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  createCXTUResourceUsageEntry(*entries,
5626457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek          CXTUResourceUsage_SourceManagerContentCache,
5627457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek          (unsigned long) astContext.getSourceManager().getContentCacheSize());
5628f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek
5629f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  // How much memory is being used by the MemoryBuffer's in SourceManager?
5630f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  const SourceManager::MemoryBufferSizes &srcBufs =
5631f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    astUnit->getSourceManager().getMemoryBufferSizes();
5632f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek
5633f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5634f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               CXTUResourceUsage_SourceManager_Membuffer_Malloc,
5635f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               (unsigned long) srcBufs.malloc_bytes);
5636ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5637f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               CXTUResourceUsage_SourceManager_Membuffer_MMap,
5638f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               (unsigned long) srcBufs.mmap_bytes);
5639ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5640ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                               CXTUResourceUsage_SourceManager_DataStructures,
5641ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                               (unsigned long) astContext.getSourceManager()
5642ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                                .getDataStructureSizes());
5643e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek
5644e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  // How much memory is being used by the ExternalASTSource?
5645e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  if (ExternalASTSource *esrc = astContext.getExternalSource()) {
5646e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    const ExternalASTSource::MemoryBufferSizes &sizes =
5647e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      esrc->getMemoryBufferSizes();
5648e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek
5649e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    createCXTUResourceUsageEntry(*entries,
5650e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc,
5651e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek                                 (unsigned long) sizes.malloc_bytes);
5652e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    createCXTUResourceUsageEntry(*entries,
5653e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      CXTUResourceUsage_ExternalASTSource_Membuffer_MMap,
5654e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek                                 (unsigned long) sizes.mmap_bytes);
5655e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  }
56565e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
56575e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  // How much memory is being used by the Preprocessor?
56585e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  Preprocessor &pp = astUnit->getPreprocessor();
56595e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  createCXTUResourceUsageEntry(*entries,
56605e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                               CXTUResourceUsage_Preprocessor,
5661c5c5e92ec53f7e6ac7ebbbf77c6d8e4b7d88daecArgyrios Kyrtzidis                               pp.getTotalMemory());
56625e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
56635e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  if (PreprocessingRecord *pRec = pp.getPreprocessingRecord()) {
56645e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    createCXTUResourceUsageEntry(*entries,
56655e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                                 CXTUResourceUsage_PreprocessingRecord,
56665e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                                 pRec->getTotalMemory());
56675e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  }
56685e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
5669d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5670d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek                               CXTUResourceUsage_Preprocessor_HeaderSearch,
5671d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek                               pp.getHeaderSearchInfo().getTotalMemory());
56725e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
5673f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  CXTUResourceUsage usage = { (void*) entries.get(),
567459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek                            (unsigned) entries->size(),
567559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek                            entries->size() ? &(*entries)[0] : 0 };
567659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  entries.take();
567759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  return usage;
567859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
567959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5680f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekvoid clang_disposeCXTUResourceUsage(CXTUResourceUsage usage) {
568159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  if (usage.data)
568259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    delete (MemUsageEntries*) usage.data;
568359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
568459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
568559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek} // end extern "C"
568659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
56876df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregorvoid clang::PrintLibclangResourceUsage(CXTranslationUnit TU) {
56886df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  CXTUResourceUsage Usage = clang_getCXTUResourceUsage(TU);
56896df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  for (unsigned I = 0; I != Usage.numEntries; ++I)
56906df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    fprintf(stderr, "  %s: %lu\n",
56916df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor            clang_getTUResourceUsageName(Usage.entries[I].kind),
56926df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor            Usage.entries[I].amount);
56936df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
56946df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  clang_disposeCXTUResourceUsage(Usage);
56956df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor}
56966df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
569759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek//===----------------------------------------------------------------------===//
569804bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek// Misc. utility functions.
569904bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek//===----------------------------------------------------------------------===//
5700f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
5701abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbar/// Default to using an 8 MB stack size on "safety" threads.
5702abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbarstatic unsigned SafetyStackThreadSize = 8 << 20;
5703bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5704bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarnamespace clang {
5705bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5706bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarbool RunSafely(llvm::CrashRecoveryContext &CRC,
57076c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               void (*Fn)(void*), void *UserData,
57086c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               unsigned Size) {
57096c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (!Size)
57106c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek    Size = GetSafetyThreadStackSize();
57116c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (Size)
5712bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar    return CRC.RunSafelyOnThread(Fn, UserData, Size);
5713bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return CRC.RunSafely(Fn, UserData);
5714bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5715bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5716bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarunsigned GetSafetyThreadStackSize() {
5717bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return SafetyStackThreadSize;
5718bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5719bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5720bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarvoid SetSafetyThreadStackSize(unsigned Value) {
5721bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  SafetyStackThreadSize = Value;
5722bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5723bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5724bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5725bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
572604bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenekextern "C" {
572704bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
5728a2a9d6e4e5b6001b86b7dfc5db1ea296ce29a3d3Ted KremenekCXString clang_getClangVersion() {
5729ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(getClangFullVersion());
573004bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek}
573104bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
573204bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek} // end: extern "C"
573359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5734