CIndex.cpp revision aed123ec3cc37e457fe20a6158fdadf8849ad916
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;
5450398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff
55a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenekstatic CXTranslationUnit MakeCXTranslationUnit(ASTUnit *TU) {
56a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  if (!TU)
57a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return 0;
58a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit D = new CXTranslationUnitImpl();
59a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  D->TUData = TU;
60a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  D->StringPool = createCXStringPool();
61a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return D;
62a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek}
63a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek
6433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \brief The result of comparing two source ranges.
6533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorenum RangeComparisonResult {
6633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief Either the ranges overlap or one of the ranges is invalid.
6733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeOverlap,
68f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The first range ends before the second range starts.
7033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeBefore,
71f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The first range starts after the second range ends.
7333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  RangeAfter
7433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor};
7533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
76f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek/// \brief Compare two source ranges to determine their relative position in
7733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// the translation unit.
78f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekstatic RangeComparisonResult RangeCompare(SourceManager &SM,
79f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                          SourceRange R1,
8033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                                          SourceRange R2) {
8133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R1.isValid() && "First range is invalid?");
8233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  assert(R2.isValid() && "Second range is invalid?");
83a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R1.getEnd() != R2.getBegin() &&
84d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R1.getEnd(), R2.getBegin()))
8533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeBefore;
86a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R2.getEnd() != R1.getBegin() &&
87d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar      SM.isBeforeInTranslationUnit(R2.getEnd(), R1.getBegin()))
8833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor    return RangeAfter;
8933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return RangeOverlap;
9033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
9133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
92fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek/// \brief Determine if a source location falls within, before, or after a
93fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek///   a given source range.
94fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic RangeComparisonResult LocationCompare(SourceManager &SM,
95fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                             SourceLocation L, SourceRange R) {
96fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(R.isValid() && "First range is invalid?");
97fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  assert(L.isValid() && "Second range is invalid?");
98a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (L == R.getBegin() || L == R.getEnd())
99fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeOverlap;
100fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(L, R.getBegin()))
101fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeBefore;
102fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (SM.isBeforeInTranslationUnit(R.getEnd(), L))
103fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return RangeAfter;
104fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return RangeOverlap;
105fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek}
106fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
10776dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// \brief Translate a Clang source range into a CIndex source range.
10876dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar///
10976dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// Clang internally represents ranges where the end location points to the
11076dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// start of the token at the end. However, for external clients it is more
11176dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// useful to have a CXSourceRange be a proper half-open interval. This routine
11276dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar/// does the appropriate translation.
113f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed KremenekCXSourceRange cxloc::translateSourceRange(const SourceManager &SM,
11476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                                          const LangOptions &LangOpts,
1150a76aae8c03cb7dd7bdbe683485560afaf695959Chris Lattner                                          const CharSourceRange &R) {
11676dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  // We want the last character in this location, so we will adjust the
1176a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor  // location accordingly.
11876dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  SourceLocation EndLoc = R.getEnd();
119a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (EndLoc.isValid() && EndLoc.isMacroID())
120edc3dccece244a584f8ebdb81da6c962c08e79beChandler Carruth    EndLoc = SM.getExpansionRange(EndLoc).second;
1210a76aae8c03cb7dd7bdbe683485560afaf695959Chris Lattner  if (R.isTokenRange() && !EndLoc.isInvalid() && EndLoc.isFileID()) {
1226a5a23f8e7fb65e028c8092bc1d1a1d9dfe2e9bcDouglas Gregor    unsigned Length = Lexer::MeasureTokenLength(EndLoc, SM, LangOpts);
123a64ccefdf0ea4e03ec88805d71b0af74950c7472Argyrios Kyrtzidis    EndLoc = EndLoc.getLocWithOffset(Length);
12476dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  }
12576dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar
12676dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  CXSourceRange Result = { { (void *)&SM, (void *)&LangOpts },
12776dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           R.getBegin().getRawEncoding(),
12876dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar                           EndLoc.getRawEncoding() };
12976dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar  return Result;
13076dd3c2ff274772dc5771cb73f856d14aaf5ee2fDaniel Dunbar}
1311db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
1328a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
13333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor// Cursor visitor.
1348a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek//===----------------------------------------------------------------------===//
1358a8da7d17d4eb281b61d08d603c7bb180d280d5aTed Kremenek
13689922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroffnamespace {
137c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
138c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekclass VisitorJob {
139c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekpublic:
140cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek  enum Kind { DeclVisitKind, StmtVisitKind, MemberExprPartsKind,
141e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek              TypeLocVisitKind, OverloadExprPartsKind,
14260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek              DeclRefExprPartsKind, LabelRefVisitKind,
143f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek              ExplicitTemplateArgsVisitKind,
144f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor              NestedNameSpecifierLocVisitKind,
145cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek              DeclarationNameInfoVisitKind,
14694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor              MemberRefVisitKind, SizeOfPackExprPartsKind };
147c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekprotected:
148f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void *data[3];
149c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  CXCursor parent;
150c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  Kind K;
151f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  VisitorJob(CXCursor C, Kind k, void *d1, void *d2 = 0, void *d3 = 0)
152f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : parent(C), K(k) {
153f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    data[0] = d1;
154f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    data[1] = d2;
155f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    data[2] = d3;
156f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
157c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekpublic:
158c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  Kind getKind() const { return K; }
159c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  const CXCursor &getParent() const { return parent; }
160c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  static bool classof(VisitorJob *VJ) { return true; }
161c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek};
162c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
1635f9e272e632e951b1efe824cd16acb4d96077930Chris Lattnertypedef SmallVector<VisitorJob, 10> VisitorWorkList;
164c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
165b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor// Cursor visitor.
1667d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorclass CursorVisitor : public DeclVisitor<CursorVisitor, bool>,
167cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek                      public TypeLocVisitor<CursorVisitor, bool>
1687d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor{
16933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The translation unit we are traversing.
170a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU;
171a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *AU;
172f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
17333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The parent cursor whose children we are traversing.
174b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  CXCursor Parent;
175f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
17633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The declaration that serves at the parent of any statement or
17733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// expression nodes.
178f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  Decl *StmtParent;
179f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
18033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The visitor function.
181b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  CXCursorVisitor Visitor;
182f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
18333e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief The opaque client data, to be passed along to the visitor.
184b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  CXClientData ClientData;
185f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
18604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor  /// \brief Whether we should visit the preprocessing record entries last,
18704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor  /// after visiting other declarations.
18804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor  bool VisitPreprocessorLast;
18904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
19033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// \brief When valid, a source range to which the cursor should restrict
19133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  /// its search.
19233e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  SourceRange RegionOfInterest;
193f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
194d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually remove.  This part of a hack to support proper
195d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // iteration over all Decls contained lexically within an ObjC container.
196d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator *DI_current;
197d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator DE_current;
198d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
199d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  // Cache of pre-allocated worklists for data-recursion walk of Stmts.
2005f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<VisitorWorkList*, 5> WorkListFreeList;
2015f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<VisitorWorkList*, 5> WorkListCache;
202d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek
203b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  using DeclVisitor<CursorVisitor, bool>::Visit;
2047d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  using TypeLocVisitor<CursorVisitor, bool>::Visit;
205f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
206f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  /// \brief Determine whether this particular source range comes before, comes
207f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  /// after, or overlaps the region of interest.
20833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  ///
209d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar  /// \param R a half-open source range retrieved from the abstract syntax tree.
210f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  RangeComparisonResult CompareRegionOfInterest(SourceRange R);
211f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2120f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  class SetParentRAII {
2130f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    CXCursor &Parent;
2140f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    Decl *&StmtParent;
2150f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    CXCursor OldParent;
2160f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek
2170f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  public:
2180f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    SetParentRAII(CXCursor &Parent, Decl *&StmtParent, CXCursor NewParent)
2190f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      : Parent(Parent), StmtParent(StmtParent), OldParent(Parent)
2200f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    {
2210f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      Parent = NewParent;
2220f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      if (clang_isDeclaration(Parent.kind))
2230f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek        StmtParent = getCursorDecl(Parent);
2240f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    }
2250f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek
2260f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    ~SetParentRAII() {
2270f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      Parent = OldParent;
2280f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek      if (clang_isDeclaration(Parent.kind))
2290f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek        StmtParent = getCursorDecl(Parent);
2300f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek    }
2310f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  };
2320f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek
233b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorpublic:
234a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CursorVisitor(CXTranslationUnit TU, CXCursorVisitor Visitor,
235a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                CXClientData ClientData,
23604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                bool VisitPreprocessorLast,
23733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor                SourceRange RegionOfInterest = SourceRange())
238a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    : TU(TU), AU(static_cast<ASTUnit*>(TU->TUData)),
239a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      Visitor(Visitor), ClientData(ClientData),
24008e0bc16b3312c27e87d33be7dcf3d4fe5bdd2e2Douglas Gregor      VisitPreprocessorLast(VisitPreprocessorLast),
24104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      RegionOfInterest(RegionOfInterest), DI_current(0)
242b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  {
243b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.kind = CXCursor_NoDeclFound;
244b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.data[0] = 0;
245b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.data[1] = 0;
246b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Parent.data[2] = 0;
247f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    StmtParent = 0;
248b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
249f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
250d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  ~CursorVisitor() {
251d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    // Free the pre-allocated worklists for data-recursion.
2525f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    for (SmallVectorImpl<VisitorWorkList*>::iterator
253d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek          I = WorkListCache.begin(), E = WorkListCache.end(); I != E; ++I) {
254d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek      delete *I;
255d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    }
256d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
257d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek
258a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *getASTUnit() const { return static_cast<ASTUnit*>(TU->TUData); }
259a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit getTU() const { return TU; }
260ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek
26133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  bool Visit(CXCursor Cursor, bool CheckedRegionOfInterest = false);
262788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
2634c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  bool visitPreprocessedEntitiesInRegion();
2644c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
2654c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  template<typename InputIterator>
2664c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  bool visitPreprocessedEntities(InputIterator First, InputIterator Last);
267788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
268b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  bool VisitChildren(CXCursor Parent);
269f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2707d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  // Declaration visitors
271162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  bool VisitTypeAliasDecl(TypeAliasDecl *D);
27209dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  bool VisitAttributes(Decl *D);
2731ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  bool VisitBlockDecl(BlockDecl *B);
2743064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  bool VisitCXXRecordDecl(CXXRecordDecl *D);
275d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  llvm::Optional<bool> shouldVisitCursor(CXCursor C);
276b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  bool VisitDeclContext(DeclContext *DC);
27779758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTranslationUnitDecl(TranslationUnitDecl *D);
27879758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTypedefDecl(TypedefDecl *D);
27979758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitTagDecl(TagDecl *D);
2800ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool VisitClassTemplateSpecializationDecl(ClassTemplateSpecializationDecl *D);
28174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  bool VisitClassTemplatePartialSpecializationDecl(
28274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                     ClassTemplatePartialSpecializationDecl *D);
283fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D);
2844540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitEnumConstantDecl(EnumConstantDecl *D);
28579758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitDeclaratorDecl(DeclaratorDecl *DD);
2864540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitFunctionDecl(FunctionDecl *ND);
28779758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitFieldDecl(FieldDecl *D);
28879758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitVarDecl(VarDecl *);
28984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  bool VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D);
290fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitFunctionTemplateDecl(FunctionTemplateDecl *D);
29139d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  bool VisitClassTemplateDecl(ClassTemplateDecl *D);
29284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  bool VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D);
29379758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCMethodDecl(ObjCMethodDecl *ND);
2944540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitObjCContainerDecl(ObjCContainerDecl *D);
29579758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCCategoryDecl(ObjCCategoryDecl *ND);
29679758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCProtocolDecl(ObjCProtocolDecl *PID);
29723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  bool VisitObjCPropertyDecl(ObjCPropertyDecl *PD);
29879758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCInterfaceDecl(ObjCInterfaceDecl *D);
2994540c9c73787d6ef736792f24209727b64997c90Ted Kremenek  bool VisitObjCImplDecl(ObjCImplDecl *D);
30079758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D);
3011ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  bool VisitObjCImplementationDecl(ObjCImplementationDecl *D);
30279758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  // FIXME: ObjCCompatibleAliasDecl requires aliased-class locations.
30379758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D);
30479758f614c83ba17757618de1459e2b481f19dd2Ted Kremenek  bool VisitObjCClassDecl(ObjCClassDecl *D);
305a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  bool VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD);
306a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  bool VisitLinkageSpecDecl(LinkageSpecDecl *D);
3078f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  bool VisitNamespaceDecl(NamespaceDecl *D);
3086931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  bool VisitNamespaceAliasDecl(NamespaceAliasDecl *D);
3090a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  bool VisitUsingDirectiveDecl(UsingDirectiveDecl *D);
3107e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUsingDecl(UsingDecl *D);
3117e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D);
3127e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  bool VisitUnresolvedUsingTypenameDecl(UnresolvedUsingTypenameDecl *D);
3130a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
31401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  // Name visitor
31501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  bool VisitDeclarationNameInfo(DeclarationNameInfo Name);
316c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  bool VisitNestedNameSpecifier(NestedNameSpecifier *NNS, SourceRange Range);
317dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  bool VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc NNS);
31801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
319fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Template visitors
320fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateParameters(const TemplateParameterList *Params);
3210b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  bool VisitTemplateName(TemplateName Name, SourceLocation Loc);
322fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  bool VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL);
323fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
3247d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  // Type visitors
325427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis#define ABSTRACT_TYPELOC(CLASS, PARENT)
326427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis#define TYPELOC(CLASS, PARENT) \
327427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  bool Visit##CLASS##TypeLoc(CLASS##TypeLoc TyLoc);
328427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis#include "clang/AST/TypeLocNodes.def"
329427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
330f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitTagTypeLoc(TagTypeLoc TL);
331f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  bool VisitArrayTypeLoc(ArrayTypeLoc TL);
332427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  bool VisitFunctionTypeLoc(FunctionTypeLoc TL, bool SkipResultType = false);
333427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
334c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  // Data-recursive visitor functions.
335c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  bool IsInRegionOfInterest(CXCursor C);
336c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  bool RunVisitorWorkList(VisitorWorkList &WL);
337c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  void EnqueueWorkList(VisitorWorkList &WL, Stmt *S);
338cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  LLVM_ATTRIBUTE_NOINLINE bool Visit(Stmt *S);
339b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor};
340f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
341b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor} // end anonymous namespace
3420d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
343a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C);
3446653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr);
3456653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
346a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
34733e9abd21083a0191a7676a04b497006d2da184dDouglas GregorRangeComparisonResult CursorVisitor::CompareRegionOfInterest(SourceRange R) {
348a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return RangeCompare(AU->getSourceManager(), R, RegionOfInterest);
34933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
35033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
351b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the given cursor and, if requested by the visitor,
352b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// its children.
353b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor///
35433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param Cursor the cursor to visit.
35533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
35633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// \param CheckRegionOfInterest if true, then the caller already checked that
35733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor/// this cursor is within the region of interest.
35833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor///
359b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
360b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
36133e9abd21083a0191a7676a04b497006d2da184dDouglas Gregorbool CursorVisitor::Visit(CXCursor Cursor, bool CheckedRegionOfInterest) {
362b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isInvalid(Cursor.kind))
363b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
364f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
365b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
366b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
367b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    assert(D && "Invalid declaration cursor");
36865ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // Ignore implicit declarations, unless it's an objc method because
36965ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // currently we should report implicit methods for properties when indexing.
37065ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    if (D->isImplicit() && !isa<ObjCMethodDecl>(D))
371b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return false;
372b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
3730d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
37433e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // If we have a range of interest, and this cursor doesn't intersect with it,
37533e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  // we're done.
37633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (RegionOfInterest.isValid() && !CheckedRegionOfInterest) {
377a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    SourceRange Range = getRawCursorExtent(Cursor);
378f408f32aa9ae3d97bc656267dc5d78fa7d03499bDaniel Dunbar    if (Range.isInvalid() || CompareRegionOfInterest(Range))
37933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor      return false;
38033e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  }
381f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
382b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  switch (Visitor(Cursor, Parent, ClientData)) {
383b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Break:
384b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
3850d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
386b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Continue:
387b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return false;
3882e331b938b38057e333fab0ba841130ea8467794Douglas Gregor
389b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  case CXChildVisit_Recurse:
390b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return VisitChildren(Cursor);
391b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
3920d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
393fd64377225a6a140bddb3f997d52a036486f9360Douglas Gregor  return false;
394b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
3950d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
3964c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregorbool CursorVisitor::visitPreprocessedEntitiesInRegion() {
397788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  PreprocessingRecord &PPRec
398a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    = *AU->getPreprocessor().getPreprocessingRecord();
399788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
40092ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  if (RegionOfInterest.isValid()) {
401ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange MappedRange = AU->mapRangeToPreamble(RegionOfInterest);
40292ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis    std::pair<PreprocessingRecord::iterator, PreprocessingRecord::iterator>
403ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      Entities = PPRec.getPreprocessedEntitiesInRange(MappedRange);
40492ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis    return visitPreprocessedEntities(Entities.first, Entities.second);
40592ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  }
40692ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis
407788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  bool OnlyLocalDecls
40832038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor    = !AU->isMainFileAST() && AU->getOnlyLocalDecls();
40932038bb8486a1f31e8bd8e19ef388049669e9ed2Douglas Gregor
41092ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  if (OnlyLocalDecls)
41192ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis    return visitPreprocessedEntities(PPRec.local_begin(), PPRec.local_end());
4124c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
41392ddef1bf843e1e18c040d69f48a6bf0bc7c776aArgyrios Kyrtzidis  return visitPreprocessedEntities(PPRec.begin(), PPRec.end());
4144c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor}
4154c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4164c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregortemplate<typename InputIterator>
4174c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregorbool CursorVisitor::visitPreprocessedEntities(InputIterator First,
4184c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor                                              InputIterator Last) {
4194c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  for (; First != Last; ++First) {
4204c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    if (MacroExpansion *ME = dyn_cast<MacroExpansion>(*First)) {
4214c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeMacroExpansionCursor(ME, TU)))
4224c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
4234c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4244c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
4254c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    }
4264c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4274c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    if (MacroDefinition *MD = dyn_cast<MacroDefinition>(*First)) {
4284c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeMacroDefinitionCursor(MD, TU)))
4294c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
43089d9980bbc2e4a4ac86673e6ec16fb9f5babb63bDouglas Gregor
4314c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
4324c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    }
4334c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4344c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor    if (InclusionDirective *ID = dyn_cast<InclusionDirective>(*First)) {
4354c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (Visit(MakeInclusionDirectiveCursor(ID, TU)))
4364c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        return true;
4374c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor
4384c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      continue;
439788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor    }
440788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor  }
441788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
4424c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor  return false;
443788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor}
444788f5a1242c04762f91eaa7565c07b9865846d88Douglas Gregor
445b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \brief Visit the children of the given cursor.
446a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek///
447b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// \returns true if the visitation should be aborted, false if it
448b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor/// should continue.
449f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekbool CursorVisitor::VisitChildren(CXCursor Cursor) {
450c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (clang_isReference(Cursor.kind) &&
451c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      Cursor.kind != CXCursor_CXXBaseSpecifier) {
452a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    // By definition, references have no children.
453a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return false;
454a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  }
455f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
456f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Set the Parent field to Cursor, then back to its old value once we're
457b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // done.
4580f91f6a2eacf29fb69b1b0c3131eb0385aad5777Ted Kremenek  SetParentRAII SetParent(Parent, StmtParent, Cursor);
459f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
460b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isDeclaration(Cursor.kind)) {
461b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    Decl *D = getCursorDecl(Cursor);
46206d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (!D)
46306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return false;
46406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
465539311e0221df256c70c1c3080c8af847cd29dffTed Kremenek    return VisitAttributes(D) || Visit(D);
466b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
467f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
46806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isStatement(Cursor.kind)) {
46906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Stmt *S = getCursorStmt(Cursor))
47006d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(S);
47106d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
47206d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
47306d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
47406d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
47506d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  if (clang_isExpression(Cursor.kind)) {
47606d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    if (Expr *E = getCursorExpr(Cursor))
47706d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor      return Visit(E);
47806d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor
47906d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor    return false;
48006d9b1ad0bca7230cbae57e3e3207dda77a9eac0Douglas Gregor  }
481f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
482b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (clang_isTranslationUnit(Cursor.kind)) {
483a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    CXTranslationUnit tu = getCursorTU(Cursor);
484a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    ASTUnit *CXXUnit = static_cast<ASTUnit*>(tu->TUData);
48504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
48604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    int VisitOrder[2] = { VisitPreprocessorLast, !VisitPreprocessorLast };
48704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor    for (unsigned I = 0; I != 2; ++I) {
48804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      if (VisitOrder[I]) {
48904a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        if (!CXXUnit->isMainFileAST() && CXXUnit->getOnlyLocalDecls() &&
49004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor            RegionOfInterest.isInvalid()) {
49104a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          for (ASTUnit::top_level_iterator TL = CXXUnit->top_level_begin(),
49204a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                        TLEnd = CXXUnit->top_level_end();
49304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor               TL != TLEnd; ++TL) {
494aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis            if (Visit(MakeCXCursor(*TL, tu, RegionOfInterest), true))
49504a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor              return true;
49604a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor          }
49704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        } else if (VisitDeclContext(
49804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                                CXXUnit->getASTContext().getTranslationUnitDecl()))
4997b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor          return true;
50004a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor        continue;
5017b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor      }
5023178cb674ac8c3b59e1791e14d38d48619a1b621Bob Wilson
50304a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor      // Walk the preprocessing record.
5044c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor      if (CXXUnit->getPreprocessor().getPreprocessingRecord())
5054c30bb148b53c8063e940ca3e049ba4d270dc9d5Douglas Gregor        visitPreprocessedEntitiesInRegion();
5060396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    }
50704a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor
5087b691f33829e6a302e256e138b3917390c2665bbDouglas Gregor    return false;
509b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
510f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
511c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  if (Cursor.kind == CXCursor_CXXBaseSpecifier) {
512c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    if (CXXBaseSpecifier *Base = getCursorCXXBaseSpecifier(Cursor)) {
513c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      if (TypeSourceInfo *BaseTSInfo = Base->getTypeSourceInfo()) {
514c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor        return Visit(BaseTSInfo->getTypeLoc());
515c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor      }
516c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor    }
517c314aa484524c6a310a51e7b83bc194774e95e90Douglas Gregor  }
518221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis
519221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis  if (Cursor.kind == CXCursor_IBOutletCollectionAttr) {
520221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis    IBOutletCollectionAttr *A =
521221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis      cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(Cursor));
522221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis    if (const ObjCInterfaceType *InterT = A->getInterface()->getAs<ObjCInterfaceType>())
523221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis      return Visit(cxcursor::MakeCursorObjCClassRef(InterT->getInterface(),
524221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis                                                    A->getInterfaceLoc(), TU));
525221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis  }
526221d5a5005b3dbde50b8b0a40bb5c5e6da8c1173Argyrios Kyrtzidis
527b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  // Nothing to visit at the moment.
528b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
529dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
530dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5311ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenekbool CursorVisitor::VisitBlockDecl(BlockDecl *B) {
53213c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor  if (TypeSourceInfo *TSInfo = B->getSignatureAsWritten())
53313c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor    if (Visit(TSInfo->getTypeLoc()))
53413c8ccb59b38e9e7133f1c80a00f210b6514a0b1Douglas Gregor        return true;
5351ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
536664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  if (Stmt *Body = B->getBody())
537aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Body, StmtParent, TU, RegionOfInterest));
538664cffd330611d78fc0286f539589920a37ca328Ted Kremenek
539664cffd330611d78fc0286f539589920a37ca328Ted Kremenek  return false;
5401ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek}
5411ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek
542d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekllvm::Optional<bool> CursorVisitor::shouldVisitCursor(CXCursor Cursor) {
543d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (RegionOfInterest.isValid()) {
5446653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange Range = getFullCursorExtent(Cursor, AU->getSourceManager());
545d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Range.isInvalid())
546d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
5476653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
548d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    switch (CompareRegionOfInterest(Range)) {
549d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeBefore:
550d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes before the region of interest; skip it.
551d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return llvm::Optional<bool>();
55223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
553d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeAfter:
554d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration comes after the region of interest; we're done.
555d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
556d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
557d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    case RangeOverlap:
558d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      // This declaration overlaps the region of interest; visit it.
559d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
560d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
561d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
562d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return true;
563d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
564f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
565d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenekbool CursorVisitor::VisitDeclContext(DeclContext *DC) {
566d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  DeclContext::decl_iterator I = DC->decls_begin(), E = DC->decls_end();
567f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
568d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually remove.  This part of a hack to support proper
569d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // iteration over all Decls contained lexically within an ObjC container.
570d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator*> DI_saved(DI_current, &I);
571d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SaveAndRestore<DeclContext::decl_iterator> DE_saved(DE_current, E);
572f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
573d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for ( ; I != E; ++I) {
574d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *D = *I;
575d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (D->getLexicalDeclContext() != DC)
576d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
577aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    CXCursor Cursor = MakeCXCursor(D, TU, RegionOfInterest);
578d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
579d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
580d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
581d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
582d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
583d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar    if (Visit(Cursor, true))
584b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
585b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  }
586b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
587dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
588dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
5891ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTranslationUnitDecl(TranslationUnitDecl *D) {
5901ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  llvm_unreachable("Translation units are visited directly by Visit()");
5911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
5921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
5931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
594162e1c1b487352434552147967c3dd296ebee2f7Richard Smithbool CursorVisitor::VisitTypeAliasDecl(TypeAliasDecl *D) {
595162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
596162e1c1b487352434552147967c3dd296ebee2f7Richard Smith    return Visit(TSInfo->getTypeLoc());
597162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
598162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return false;
599162e1c1b487352434552147967c3dd296ebee2f7Richard Smith}
600162e1c1b487352434552147967c3dd296ebee2f7Richard Smith
6011ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTypedefDecl(TypedefDecl *D) {
6021ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (TypeSourceInfo *TSInfo = D->getTypeSourceInfo())
6031ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return Visit(TSInfo->getTypeLoc());
604f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
6051ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6061ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6071ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6081ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitTagDecl(TagDecl *D) {
6091ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitDeclContext(D);
6101ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6111ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6120ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregorbool CursorVisitor::VisitClassTemplateSpecializationDecl(
6130ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor                                          ClassTemplateSpecializationDecl *D) {
6140ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  bool ShouldVisitBody = false;
6150ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  switch (D->getSpecializationKind()) {
6160ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_Undeclared:
6170ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ImplicitInstantiation:
6180ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    // Nothing to visit
6190ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return false;
6200ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6210ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDeclaration:
6220ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitInstantiationDefinition:
6230ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6240ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6250ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  case TSK_ExplicitSpecialization:
6260ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    ShouldVisitBody = true;
6270ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    break;
6280ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6290ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6300ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  // Visit the template arguments used in the specialization.
6310ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (TypeSourceInfo *SpecType = D->getTypeAsWritten()) {
6320ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    TypeLoc TL = SpecType->getTypeLoc();
6330ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    if (TemplateSpecializationTypeLoc *TSTLoc
6340ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          = dyn_cast<TemplateSpecializationTypeLoc>(&TL)) {
6350ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor      for (unsigned I = 0, N = TSTLoc->getNumArgs(); I != N; ++I)
6360ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor        if (VisitTemplateArgumentLoc(TSTLoc->getArgLoc(I)))
6370ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor          return true;
6380ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    }
6390ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  }
6400ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6410ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  if (ShouldVisitBody && VisitCXXRecordDecl(D))
6420ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor    return true;
6430ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
6440ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor  return false;
6450ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor}
6460ab1e9f672a86cf3f094780d99251553a2b69fc7Douglas Gregor
64774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregorbool CursorVisitor::VisitClassTemplatePartialSpecializationDecl(
64874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor                                   ClassTemplatePartialSpecializationDecl *D) {
64974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
65074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // before visiting these template parameters.
65174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
65274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return true;
65374dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
65474dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  // Visit the partial specialization arguments.
65574dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  const TemplateArgumentLoc *TemplateArgs = D->getTemplateArgsAsWritten();
65674dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  for (unsigned I = 0, N = D->getNumTemplateArgsAsWritten(); I != N; ++I)
65774dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    if (VisitTemplateArgumentLoc(TemplateArgs[I]))
65874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor      return true;
65974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
66074dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  return VisitCXXRecordDecl(D);
66174dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor}
66274dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor
663fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmDecl(TemplateTypeParmDecl *D) {
66484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  // Visit the default argument.
66584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
66684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (TypeSourceInfo *DefArg = D->getDefaultArgumentInfo())
66784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      if (Visit(DefArg->getTypeLoc()))
66884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor        return true;
66984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
670fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
671fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
672fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
6731ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitEnumConstantDecl(EnumConstantDecl *D) {
6741ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInitExpr())
675aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
6761ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
6771ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
6781ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
6797d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitDeclaratorDecl(DeclaratorDecl *DD) {
6807d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  if (TypeSourceInfo *TSInfo = DD->getTypeSourceInfo())
6817d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
6827d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      return true;
6837d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
684c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
685c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = DD->getQualifierLoc())
686c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
687c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
688c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
6897d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor  return false;
6907d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
6917d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
692a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor/// \brief Compare two base or member initializers based on their source order.
693cbb67480094b3bcb5b715acd827cbad55e2a204cSean Huntstatic int CompareCXXCtorInitializers(const void* Xp, const void *Yp) {
694cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *X
695cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Xp);
696cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt  CXXCtorInitializer const * const *Y
697cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt    = static_cast<CXXCtorInitializer const * const *>(Yp);
698a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
699a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  if ((*X)->getSourceOrder() < (*Y)->getSourceOrder())
700a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return -1;
701a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else if ((*X)->getSourceOrder() > (*Y)->getSourceOrder())
702a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 1;
703a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  else
704a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return 0;
705a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor}
706a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
707b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitFunctionDecl(FunctionDecl *ND) {
70801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (TypeSourceInfo *TSInfo = ND->getTypeSourceInfo()) {
70901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function declaration's syntactic components in the order
71001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // written. This requires a bit of work.
711723df245307a530da5433dfb43accf187dc3e243Abramo Bagnara    TypeLoc TL = TSInfo->getTypeLoc().IgnoreParens();
71201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    FunctionTypeLoc *FTL = dyn_cast<FunctionTypeLoc>(&TL);
71301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
71401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // If we have a function declared directly (without the use of a typedef),
71501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // visit just the return type. Otherwise, just visit the function's type
71601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // now.
71701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if ((FTL && !isa<CXXConversionDecl>(ND) && Visit(FTL->getResultLoc())) ||
71801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor        (!FTL && Visit(TL)))
71901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
72001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
721c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // Visit the nested-name-specifier, if present.
722c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (NestedNameSpecifierLoc QualifierLoc = ND->getQualifierLoc())
723c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      if (VisitNestedNameSpecifierLoc(QualifierLoc))
724c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor        return true;
72501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
72601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the declaration name.
72701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (VisitDeclarationNameInfo(ND->getNameInfo()))
72801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
72901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
73001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Visit explicitly-specified template arguments!
73101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
73201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // Visit the function parameters, if we have a function type.
73301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (FTL && VisitFunctionTypeLoc(*FTL, true))
73401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return true;
73501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
73601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Attributes?
73701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
73801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
73910620eb5164e31208fcbf0437cd79ae535ed0559Sean Hunt  if (ND->doesThisDeclarationHaveABody() && !ND->isLateTemplateParsed()) {
740a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    if (CXXConstructorDecl *Constructor = dyn_cast<CXXConstructorDecl>(ND)) {
741a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Find the initializers that were written in the source.
7425f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner      SmallVector<CXXCtorInitializer *, 4> WrittenInits;
743a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (CXXConstructorDecl::init_iterator I = Constructor->init_begin(),
744a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                          IEnd = Constructor->init_end();
745a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor           I != IEnd; ++I) {
746a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (!(*I)->isWritten())
747a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          continue;
748a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
749a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        WrittenInits.push_back(*I);
750a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
751a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
752a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Sort the initializers in source order
753a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      llvm::array_pod_sort(WrittenInits.begin(), WrittenInits.end(),
754cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt                           &CompareCXXCtorInitializers);
755a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
756a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      // Visit the initializers in source order
757a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      for (unsigned I = 0, N = WrittenInits.size(); I != N; ++I) {
758cbb67480094b3bcb5b715acd827cbad55e2a204cSean Hunt        CXXCtorInitializer *Init = WrittenInits[I];
75900eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet        if (Init->isAnyMemberInitializer()) {
76000eb3f9c5b33e3d99aee1f8b75dd9c9678fdd66bFrancois Pichet          if (Visit(MakeCursorMemberRef(Init->getAnyMember(),
761a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor                                        Init->getMemberLocation(), TU)))
762a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
763a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        } else if (TypeSourceInfo *BaseInfo = Init->getBaseClassInfo()) {
764a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor          if (Visit(BaseInfo->getTypeLoc()))
765a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
766a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        }
767a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
768a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        // Visit the initializer value.
769a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor        if (Expr *Initializer = Init->getInit())
770aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis          if (Visit(MakeCXCursor(Initializer, ND, TU, RegionOfInterest)))
771a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor            return true;
772a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      }
773a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
774a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
775aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
776a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return true;
777a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  }
778f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
779b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return false;
780b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
781dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
7821ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitFieldDecl(FieldDecl *D) {
7831ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
7841ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
785f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7861ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *BitWidth = D->getBitWidth())
787aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(BitWidth, StmtParent, TU, RegionOfInterest));
788f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7891ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
7901ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
7911ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
7921ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitVarDecl(VarDecl *D) {
7931ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (VisitDeclaratorDecl(D))
7941ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
795f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7961ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (Expr *Init = D->getInit())
797aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Init, StmtParent, TU, RegionOfInterest));
798f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
7991ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8001ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8011ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
80284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitNonTypeTemplateParmDecl(NonTypeTemplateParmDecl *D) {
80384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitDeclaratorDecl(D))
80484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
80584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
80684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited())
80784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    if (Expr *DefArg = D->getDefaultArgument())
808aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(DefArg, StmtParent, TU, RegionOfInterest));
80984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
81084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
81184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
81284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
813fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitFunctionTemplateDecl(FunctionTemplateDecl *D) {
814fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // FIXME: Visit the "outer" template parameter lists on the FunctionDecl
815fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // before visiting these template parameters.
816fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
817fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return true;
818fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
819fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return VisitFunctionDecl(D->getTemplatedDecl());
820fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
821fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
82239d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregorbool CursorVisitor::VisitClassTemplateDecl(ClassTemplateDecl *D) {
82339d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // FIXME: Visit the "outer" template parameter lists on the TagDecl
82439d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  // before visiting these template parameters.
82539d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
82639d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return true;
82739d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
82839d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  return VisitCXXRecordDecl(D->getTemplatedDecl());
82939d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor}
83039d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor
83184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregorbool CursorVisitor::VisitTemplateTemplateParmDecl(TemplateTemplateParmDecl *D) {
83284b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (VisitTemplateParameters(D->getTemplateParameters()))
83384b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
83484b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
83584b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  if (D->hasDefaultArgument() && !D->defaultArgumentWasInherited() &&
83684b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor      VisitTemplateArgumentLoc(D->getDefaultArgument()))
83784b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor    return true;
83884b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
83984b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor  return false;
84084b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor}
84184b51d77e11aaf6ea4607e01187343423ce6c8aeDouglas Gregor
8421ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCMethodDecl(ObjCMethodDecl *ND) {
8434bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor  if (TypeSourceInfo *TSInfo = ND->getResultTypeSourceInfo())
8444bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor    if (Visit(TSInfo->getTypeLoc()))
8454bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor      return true;
8464bc1cb6aa635a5bf8fae99bf69c56c724c1e786cDouglas Gregor
847f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  for (ObjCMethodDecl::param_iterator P = ND->param_begin(),
8481ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       PEnd = ND->param_end();
8491ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       P != PEnd; ++P) {
850aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
8511ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
8521ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  }
853f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8541ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (ND->isThisDeclarationADefinition() &&
855aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      Visit(MakeCXCursor(ND->getBody(), StmtParent, TU, RegionOfInterest)))
8561ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    return true;
857f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
8581ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
8591ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
8601ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
861d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremeneknamespace {
862d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  struct ContainerDeclsSort {
863d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    SourceManager &SM;
864d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    ContainerDeclsSort(SourceManager &sm) : SM(sm) {}
865d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    bool operator()(Decl *A, Decl *B) {
866d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_A = A->getLocStart();
867d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L_B = B->getLocStart();
868d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      assert(L_A.isValid() && L_B.isValid());
869d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return SM.isBeforeInTranslationUnit(L_A, L_B);
870d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
871d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  };
872d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek}
873d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
874a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregorbool CursorVisitor::VisitObjCContainerDecl(ObjCContainerDecl *D) {
875d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // FIXME: Eventually convert back to just 'VisitDeclContext()'.  Essentially
876d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // an @implementation can lexically contain Decls that are not properly
877d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // nested in the AST.  When we identify such cases, we need to retrofit
878d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // this nesting here.
879d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (!DI_current)
880d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
881d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
882d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Scan the Decls that immediately come after the container
883d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // in the current DeclContext.  If any fall within the
884d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // container's lexical region, stash them into a vector
885d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // for later processing.
8865f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<Decl *, 24> DeclsInContainer;
887d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  SourceLocation EndLoc = D->getSourceRange().getEnd();
888a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  SourceManager &SM = AU->getSourceManager();
889d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (EndLoc.isValid()) {
890d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclContext::decl_iterator next = *DI_current;
891d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    while (++next != DE_current) {
892d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      Decl *D_next = *next;
893d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (!D_next)
894d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        break;
895d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      SourceLocation L = D_next->getLocStart();
896d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (!L.isValid())
897d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        break;
898d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      if (SM.isBeforeInTranslationUnit(L, EndLoc)) {
899d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        *DI_current = next;
900d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        DeclsInContainer.push_back(D_next);
901d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek        continue;
902d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      }
903d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      break;
904d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    }
905d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
906d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
907d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // The common case.
908d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  if (DeclsInContainer.empty())
909d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    return VisitDeclContext(D);
910d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
911d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Get all the Decls in the DeclContext, and sort them with the
912d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // additional ones we've collected.  Then visit them.
913d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  for (DeclContext::decl_iterator I = D->decls_begin(), E = D->decls_end();
914d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek       I!=E; ++I) {
915d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    Decl *subDecl = *I;
9160582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek    if (!subDecl || subDecl->getLexicalDeclContext() != D ||
9170582c897ec7261b4c6af0fe26dc2a0b6b54d266cTed Kremenek        subDecl->getLocStart().isInvalid())
918d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
919d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    DeclsInContainer.push_back(subDecl);
920d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
921d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
922d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now sort the Decls so that they appear in lexical order.
923d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  std::sort(DeclsInContainer.begin(), DeclsInContainer.end(),
924d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek            ContainerDeclsSort(SM));
925d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek
926d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  // Now visit the decls.
9275f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  for (SmallVectorImpl<Decl*>::iterator I = DeclsInContainer.begin(),
928d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek         E = DeclsInContainer.end(); I != E; ++I) {
929aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    CXCursor Cursor = MakeCXCursor(*I, TU, RegionOfInterest);
930d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    const llvm::Optional<bool> &V = shouldVisitCursor(Cursor);
931d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.hasValue())
932d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      continue;
933d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (!V.getValue())
934d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return false;
935d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek    if (Visit(Cursor, true))
936d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek      return true;
937d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  }
938d8c370ca95a3bccc36c4a6b92a4042809ff51ee4Ted Kremenek  return false;
939a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor}
940a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor
941b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCCategoryDecl(ObjCCategoryDecl *ND) {
942b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (Visit(MakeCursorObjCClassRef(ND->getClassInterface(), ND->getLocation(),
943b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                   TU)))
944b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
945f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
94678db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCCategoryDecl::protocol_loc_iterator PL = ND->protocol_loc_begin();
94778db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCCategoryDecl::protocol_iterator I = ND->protocol_begin(),
94878db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = ND->protocol_end(); I != E; ++I, ++PL)
949b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
950b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
951f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
952a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(ND);
953dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
954dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
9551ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCProtocolDecl(ObjCProtocolDecl *PID) {
9561ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCProtocolDecl::protocol_loc_iterator PL = PID->protocol_loc_begin();
9571ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCProtocolDecl::protocol_iterator I = PID->protocol_begin(),
9581ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       E = PID->protocol_end(); I != E; ++I, ++PL)
9591ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
9601ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
961f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
9621ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(PID);
9631ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
9641ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
96523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenekbool CursorVisitor::VisitObjCPropertyDecl(ObjCPropertyDecl *PD) {
96683cb94269015bf2770ade71e616c5322ea7e76e1Douglas Gregor  if (PD->getTypeSourceInfo() && Visit(PD->getTypeSourceInfo()->getTypeLoc()))
967fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall    return true;
968fc929208193eff37e1d3a28b1ea3bd1c9a7913e0John McCall
96923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // FIXME: This implements a workaround with @property declarations also being
97023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // installed in the DeclContext for the @interface.  Eventually this code
97123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // should be removed.
97223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCCategoryDecl *CDecl = dyn_cast<ObjCCategoryDecl>(PD->getDeclContext());
97323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!CDecl || !CDecl->IsClassExtension())
97423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
97523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
97623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCInterfaceDecl *ID = CDecl->getClassInterface();
97723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!ID)
97823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
97923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
98023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  IdentifierInfo *PropertyId = PD->getIdentifier();
98123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  ObjCPropertyDecl *prevDecl =
98223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    ObjCPropertyDecl::findPropertyDecl(cast<DeclContext>(ID), PropertyId);
98323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
98423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (!prevDecl)
98523173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    return false;
98623173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
98723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // Visit synthesized methods since they will be skipped when visiting
98823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  // the @interface.
98923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getGetterMethodDecl())
990a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
991aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
99223173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
99323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
99423173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  if (ObjCMethodDecl *MD = prevDecl->getSetterMethodDecl())
995a054fb46b1fb596d1719b89d2d9a5be3c32a4b0dTed Kremenek    if (MD->isSynthesized() && MD->getLexicalDeclContext() == CDecl)
996aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(MD, TU, RegionOfInterest)))
99723173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek        return true;
99823173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
99923173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek  return false;
100023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek}
100123173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek
1002b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregorbool CursorVisitor::VisitObjCInterfaceDecl(ObjCInterfaceDecl *D) {
1003dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek  // Issue callbacks for super class.
1004b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  if (D->getSuperClass() &&
1005b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1006f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
1007b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor                                        TU)))
1008b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor    return true;
1009f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
101078db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  ObjCInterfaceDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
101178db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor  for (ObjCInterfaceDecl::protocol_iterator I = D->protocol_begin(),
101278db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor         E = D->protocol_end(); I != E; ++I, ++PL)
1013b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1014b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1015f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1016a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor  return VisitObjCContainerDecl(D);
1017dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1018dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10191ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplDecl(ObjCImplDecl *D) {
10201ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCContainerDecl(D);
10211ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10221ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10231ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCCategoryImplDecl(ObjCCategoryImplDecl *D) {
1024ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  // 'ID' could be null when dealing with invalid code.
1025ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek  if (ObjCInterfaceDecl *ID = D->getClassInterface())
1026ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek    if (Visit(MakeCursorObjCClassRef(ID, D->getLocation(), TU)))
1027ebfa339321f8a4df9d5011e591a615d5765107d5Ted Kremenek      return true;
1028f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10291ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
10301ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor}
10311ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor
10321ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCImplementationDecl(ObjCImplementationDecl *D) {
10331ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#if 0
10341ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // Issue callbacks for super class.
10351ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  // FIXME: No source location information!
10361ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  if (D->getSuperClass() &&
10371ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      Visit(MakeCursorObjCSuperClassRef(D->getSuperClass(),
1038f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                                        D->getSuperClassLoc(),
10391ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                        TU)))
1040a59e390ed6d722f0eaaa9f7eb106eaaf470df3f1Douglas Gregor    return true;
10411ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor#endif
1042f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
10431ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return VisitObjCImplDecl(D);
1044dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1045dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10461ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCForwardProtocolDecl(ObjCForwardProtocolDecl *D) {
10471ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  ObjCForwardProtocolDecl::protocol_loc_iterator PL = D->protocol_loc_begin();
10481ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  for (ObjCForwardProtocolDecl::protocol_iterator I = D->protocol_begin(),
10491ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor                                                  E = D->protocol_end();
10501ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor       I != E; ++I, ++PL)
1051b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    if (Visit(MakeCursorObjCProtocolRef(*I, *PL, TU)))
1052b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor      return true;
1053f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
1054f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return false;
1055dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
1056dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek
10571ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregorbool CursorVisitor::VisitObjCClassDecl(ObjCClassDecl *D) {
105895ed7784a335aca53b0c6e952cf31a4cfb633360Fariborz Jahanian  if (Visit(MakeCursorObjCClassRef(D->getForwardInterfaceDecl(),
105995ed7784a335aca53b0c6e952cf31a4cfb633360Fariborz Jahanian                                   D->getForwardDecl()->getLocation(), TU)))
10601ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor      return true;
10611ef2fc1888d8d14e97ebe561ccbd421282de3ce2Douglas Gregor  return false;
1062dd6bcc5f79666b9298d91a0a6ee7a0b537bde601Ted Kremenek}
10635e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramer
1064a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregorbool CursorVisitor::VisitObjCPropertyImplDecl(ObjCPropertyImplDecl *PD) {
1065a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  if (ObjCIvarDecl *Ivar = PD->getPropertyIvarDecl())
1066a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor    return Visit(MakeCursorMemberRef(Ivar, PD->getPropertyIvarDeclLoc(), TU));
1067a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
1068a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor  return false;
1069a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor}
1070a4ffd85a6684e42f900aad5459e58ad91bb88755Douglas Gregor
10718f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenekbool CursorVisitor::VisitNamespaceDecl(NamespaceDecl *D) {
10728f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  return VisitDeclContext(D);
10738f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek}
10748f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek
10756931900f43cea558c6974075256c07728dbfecc6Douglas Gregorbool CursorVisitor::VisitNamespaceAliasDecl(NamespaceAliasDecl *D) {
1076c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
10770cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
10780cfaf6a270ecd0f5c7e541a8047c87948317548bDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1079c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
10806931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
10816931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  return Visit(MakeCursorNamespaceRef(D->getAliasedNamespace(),
10826931900f43cea558c6974075256c07728dbfecc6Douglas Gregor                                      D->getTargetNameLoc(), TU));
10836931900f43cea558c6974075256c07728dbfecc6Douglas Gregor}
10846931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
10857e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUsingDecl(UsingDecl *D) {
1086c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1087dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1088dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1089c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1090dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
10917e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
10921f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (Visit(MakeCursorOverloadedDeclRef(D, D->getLocation(), TU)))
10931f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return true;
10941f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
10957e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
10967e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
10977e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
10980a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregorbool CursorVisitor::VisitUsingDirectiveDecl(UsingDirectiveDecl *D) {
1099c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1100db9924191092b4d426cc066637d81698211846aaDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1101db9924191092b4d426cc066637d81698211846aaDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1102c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
11030a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11040a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  return Visit(MakeCursorNamespaceRef(D->getNominatedNamespaceAsWritten(),
11050a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor                                      D->getIdentLocation(), TU));
11060a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor}
11070a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
11087e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingValueDecl(UnresolvedUsingValueDecl *D) {
1109c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1110dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc()) {
1111dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1112c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1113dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1114c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11157e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return VisitDeclarationNameInfo(D->getNameInfo());
11167e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11177e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
11187e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypenameDecl(
11197e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor                                               UnresolvedUsingTypenameDecl *D) {
1120c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Visit nested-name-specifier.
1121dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1122dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1123c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return true;
1124c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
11257e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  return false;
11267e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor}
11277e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor
112801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitDeclarationNameInfo(DeclarationNameInfo Name) {
112901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  switch (Name.getName().getNameKind()) {
113001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::Identifier:
113101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXLiteralOperatorName:
113201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXOperatorName:
113301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXUsingDirective:
113401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
113501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
113601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConstructorName:
113701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXDestructorName:
113801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::CXXConversionFunctionName:
113901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    if (TypeSourceInfo *TSInfo = Name.getNamedTypeInfo())
114001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor      return Visit(TSInfo->getTypeLoc());
114101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
114201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
114301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCZeroArgSelector:
114401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCOneArgSelector:
114501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case clang::DeclarationName::ObjCMultiArgSelector:
114601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    // FIXME: Per-identifier location info?
114701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return false;
114801829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  }
114901829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
115001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return false;
115101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
115201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1153c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregorbool CursorVisitor::VisitNestedNameSpecifier(NestedNameSpecifier *NNS,
1154c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                             SourceRange Range) {
1155c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // FIXME: This whole routine is a hack to work around the lack of proper
1156c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // source information in nested-name-specifiers (PR5791). Since we do have
1157c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // a beginning source location, we can visit the first component of the
1158c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // nested-name-specifier, if it's a single-token component.
1159c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  if (!NNS)
1160c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return false;
1161c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1162c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  // Get the first component in the nested-name-specifier.
1163c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  while (NestedNameSpecifier *Prefix = NNS->getPrefix())
1164c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    NNS = Prefix;
1165c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1166c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  switch (NNS->getKind()) {
1167c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Namespace:
1168c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(), Range.getBegin(),
1169c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                        TU));
1170c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
117114aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor  case NestedNameSpecifier::NamespaceAlias:
117214aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor    return Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
117314aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor                                        Range.getBegin(), TU));
117414aba76042e041b2c5e439bf4ae353a0a3c7fd73Douglas Gregor
1175c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpec: {
1176c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // If the type has a form where we know that the beginning of the source
1177c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // range matches up with a reference cursor. Visit the appropriate reference
1178c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    // cursor.
1179f4c7371fb1d3cebcfb40abad4537bb82515704eaJohn McCall    const Type *T = NNS->getAsType();
1180c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TypedefType *Typedef = dyn_cast<TypedefType>(T))
1181c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Typedef->getDecl(), Range.getBegin(), TU));
1182c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TagType *Tag = dyn_cast<TagType>(T))
1183c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return Visit(MakeCursorTypeRef(Tag->getDecl(), Range.getBegin(), TU));
1184c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    if (const TemplateSpecializationType *TST
1185c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor                                      = dyn_cast<TemplateSpecializationType>(T))
1186c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor      return VisitTemplateName(TST->getTemplateName(), Range.getBegin());
1187c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1188c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1189c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1190c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::TypeSpecWithTemplate:
1191c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Global:
1192c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  case NestedNameSpecifier::Identifier:
1193c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor    break;
1194c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  }
1195c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1196c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor  return false;
1197c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor}
1198c5ade2e3644a5822df63e442788d68c591ccdc97Douglas Gregor
1199dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregorbool
1200dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas GregorCursorVisitor::VisitNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
12015f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  SmallVector<NestedNameSpecifierLoc, 4> Qualifiers;
1202dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  for (; Qualifier; Qualifier = Qualifier.getPrefix())
1203dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    Qualifiers.push_back(Qualifier);
1204dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1205dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  while (!Qualifiers.empty()) {
1206dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifierLoc Q = Qualifiers.pop_back_val();
1207dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    NestedNameSpecifier *NNS = Q.getNestedNameSpecifier();
1208dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    switch (NNS->getKind()) {
1209dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Namespace:
1210dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespace(),
1211c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1212dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1213dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1214dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1215dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1216dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1217dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::NamespaceAlias:
1218dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(MakeCursorNamespaceRef(NNS->getAsNamespaceAlias(),
1219c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor                                       Q.getLocalBeginLoc(),
1220dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor                                       TU)))
1221dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1222dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1223dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1224dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1225dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpec:
1226dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::TypeSpecWithTemplate:
1227dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      if (Visit(Q.getTypeLoc()))
1228dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor        return true;
1229dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1230dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1231dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1232dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Global:
1233dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    case NestedNameSpecifier::Identifier:
1234dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor      break;
1235dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor    }
1236dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  }
1237dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1238dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor  return false;
1239dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor}
1240dc355713be51fcb4ee52d9fd6b4548ceff47fadfDouglas Gregor
1241fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateParameters(
1242fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          const TemplateParameterList *Params) {
1243fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  if (!Params)
1244fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1245fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1246fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (TemplateParameterList::const_iterator P = Params->begin(),
1247fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                          PEnd = Params->end();
1248fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor       P != PEnd; ++P) {
1249aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (Visit(MakeCXCursor(*P, TU, RegionOfInterest)))
1250fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1251fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1252fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1253fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1254fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1255fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
12560b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregorbool CursorVisitor::VisitTemplateName(TemplateName Name, SourceLocation Loc) {
12570b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  switch (Name.getKind()) {
12580b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::Template:
12590b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(Name.getAsTemplateDecl(), Loc, TU));
12600b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12610b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::OverloadedTemplate:
12621f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // Visit the overloaded template set.
12631f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (Visit(MakeCursorOverloadedDeclRef(Name, Loc, TU)))
12641f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return true;
12651f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
12660b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
12670b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12680b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::DependentTemplate:
12690b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
12700b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return false;
12710b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12720b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case TemplateName::QualifiedTemplate:
12730b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    // FIXME: Visit nested-name-specifier.
12740b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return Visit(MakeCursorTemplateRef(
12750b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                  Name.getAsQualifiedTemplateName()->getDecl(),
12760b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                                       Loc, TU));
1277146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall
1278146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall  case TemplateName::SubstTemplateTemplateParm:
1279146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall    return Visit(MakeCursorTemplateRef(
1280146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall                         Name.getAsSubstTemplateTemplateParm()->getParameter(),
1281146060435c3efce95c95a092c7a1eb651cfb9ae0John McCall                                       Loc, TU));
12821aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor
12831aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor  case TemplateName::SubstTemplateTemplateParmPack:
12841aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor    return Visit(MakeCursorTemplateRef(
12851aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                  Name.getAsSubstTemplateTemplateParmPack()->getParameterPack(),
12861aee05d08b2184acadeb36de300e216390780d6cDouglas Gregor                                       Loc, TU));
12870b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  }
12880b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
12890b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  return false;
12900b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor}
12910b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
1292fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateArgumentLoc(const TemplateArgumentLoc &TAL) {
1293fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  switch (TAL.getArgument().getKind()) {
1294fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Null:
1295fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Integral:
1296fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Pack:
1297fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
129887dd697dcc8ecb64df73ae64d61b8c80ff0c157cDouglas Gregor
1299fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Type:
1300fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (TypeSourceInfo *TSInfo = TAL.getTypeSourceInfo())
1301fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return Visit(TSInfo->getTypeLoc());
1302fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1303fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1304fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Declaration:
1305fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceDeclExpression())
1306aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1307fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1308fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1309fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Expression:
1310fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (Expr *E = TAL.getSourceExpression())
1311aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return Visit(MakeCXCursor(E, StmtParent, TU, RegionOfInterest));
1312fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return false;
1313fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1314fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case TemplateArgument::Template:
1315a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor  case TemplateArgument::TemplateExpansion:
1316b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor    if (VisitNestedNameSpecifierLoc(TAL.getTemplateQualifierLoc()))
1317b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor      return true;
1318b6744efecba58792cce20d2d7b9ee39927c5422eDouglas Gregor
1319a7fc901a2e39bfe55bfcff5934b2d9fdf9656491Douglas Gregor    return VisitTemplateName(TAL.getArgument().getAsTemplateOrTemplatePattern(),
13200b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                             TAL.getTemplateNameLoc());
1321fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  }
1322fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1323fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1324fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1325fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1326a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenekbool CursorVisitor::VisitLinkageSpecDecl(LinkageSpecDecl *D) {
1327a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  return VisitDeclContext(D);
1328a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek}
1329a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek
133001829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitQualifiedTypeLoc(QualifiedTypeLoc TL) {
133101829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  return Visit(TL.getUnqualifiedLoc());
133201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor}
133301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor
1334f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBuiltinTypeLoc(BuiltinTypeLoc TL) {
1335a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTContext &Context = AU->getASTContext();
1336f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1337f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // Some builtin types (such as Objective-C's "id", "sel", and
1338f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  // "Class") have associated declarations. Create cursors for those.
1339f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  QualType VisitType;
1340f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  switch (TL.getType()->getAs<BuiltinType>()->getKind()) {
13416b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Void:
1342f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Bool:
13436b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Char_U:
13446b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UChar:
1345f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char16:
1346f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char32:
13476b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UShort:
13486b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UInt:
13496b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ULong:
13506b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ULongLong:
13516b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::UInt128:
1352f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Char_S:
13536b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::SChar:
13543f59c975aa5d047f7edd1b900b5e885c38af0ef7Chris Lattner  case BuiltinType::WChar_U:
13553f59c975aa5d047f7edd1b900b5e885c38af0ef7Chris Lattner  case BuiltinType::WChar_S:
13566b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Short:
1357f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Int:
1358f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Long:
1359c4174cc4b9b657abb77d0825de473ea29cf48297Ted Kremenek  case BuiltinType::LongLong:
13606b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Int128:
13616b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Float:
13626b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Double:
13636b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::LongDouble:
1364f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::NullPtr:
1365f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::Overload:
1366864c041e118155c2b1ce0ba36942a3da5a4a055eJohn McCall  case BuiltinType::BoundMember:
13676b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::Dependent:
13681de4d4e8cb2e9c88809fea8092bc6e835a5473d2John McCall  case BuiltinType::UnknownAny:
1369f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
13706b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1371f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCId:
1372f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCIdType();
1373f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
13746b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
13756b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek  case BuiltinType::ObjCClass:
13766b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    VisitType = Context.getObjCClassType();
13776b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek    break;
13786b3b514e312b4caa403bbffe5673aa31cfbad051Ted Kremenek
1379f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  case BuiltinType::ObjCSel:
1380f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    VisitType = Context.getObjCSelType();
1381f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    break;
1382f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1383f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1384f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (!VisitType.isNull()) {
1385f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (const TypedefType *Typedef = VisitType->getAs<TypedefType>())
1386f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek      return Visit(MakeCursorTypeRef(Typedef->getDecl(), TL.getBuiltinLoc(),
1387f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                     TU));
1388f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1389f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1390f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1391f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1392f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
13937d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregorbool CursorVisitor::VisitTypedefTypeLoc(TypedefTypeLoc TL) {
1394162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  return Visit(MakeCursorTypeRef(TL.getTypedefNameDecl(), TL.getNameLoc(), TU));
13957d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor}
13967d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
1397f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitUnresolvedUsingTypeLoc(UnresolvedUsingTypeLoc TL) {
1398f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1399f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1400f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1401f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitTagTypeLoc(TagTypeLoc TL) {
14026f155de99c59af890817146ec8526bafb6560f1fArgyrios Kyrtzidis  if (TL.isDefinition())
1403aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(TL.getDecl(), TU, RegionOfInterest));
14046f155de99c59af890817146ec8526bafb6560f1fArgyrios Kyrtzidis
1405f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1406f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1407f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1408fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateTypeParmTypeLoc(TemplateTypeParmTypeLoc TL) {
1409960d13dde337a59dacc9dc3936c26d4aa8478986Chandler Carruth  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1410fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1411fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1412f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCInterfaceTypeLoc(ObjCInterfaceTypeLoc TL) {
1413f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(MakeCursorObjCClassRef(TL.getIFaceDecl(), TL.getNameLoc(), TU)))
1414f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1415f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1416c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return false;
1417c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall}
1418c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1419c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCallbool CursorVisitor::VisitObjCObjectTypeLoc(ObjCObjectTypeLoc TL) {
1420c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  if (TL.hasBaseTypeAsWritten() && Visit(TL.getBaseLoc()))
1421c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall    return true;
1422c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall
1423f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumProtocols(); I != N; ++I) {
1424f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    if (Visit(MakeCursorObjCProtocolRef(TL.getProtocol(I), TL.getProtocolLoc(I),
1425f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor                                        TU)))
1426f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor      return true;
1427f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  }
1428f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1429f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1430f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1431f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1432f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitObjCObjectPointerTypeLoc(ObjCObjectPointerTypeLoc TL) {
1433c12c5bba6ceb6acd4e51e7a0fc03257da9cfd44eJohn McCall  return Visit(TL.getPointeeLoc());
1434f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1435f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1436075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnarabool CursorVisitor::VisitParenTypeLoc(ParenTypeLoc TL) {
1437075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara  return Visit(TL.getInnerLoc());
1438075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara}
1439075f8f1b6bed4d1b224c74f87508534cc6392ce6Abramo Bagnara
1440f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitPointerTypeLoc(PointerTypeLoc TL) {
1441f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1442f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1443f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1444f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitBlockPointerTypeLoc(BlockPointerTypeLoc TL) {
1445f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1446f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1447f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1448f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitMemberPointerTypeLoc(MemberPointerTypeLoc TL) {
1449f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return Visit(TL.getPointeeLoc());
1450f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1451f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1452f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitLValueReferenceTypeLoc(LValueReferenceTypeLoc TL) {
1453f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1454f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1455f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1456f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitRValueReferenceTypeLoc(RValueReferenceTypeLoc TL) {
1457f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return Visit(TL.getPointeeLoc());
1458f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1459f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
14603422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidisbool CursorVisitor::VisitAttributedTypeLoc(AttributedTypeLoc TL) {
14613422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis  return Visit(TL.getModifiedLoc());
14623422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis}
14633422fbc38f35d9e486879850c5bf0175bd2eee16Argyrios Kyrtzidis
146401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregorbool CursorVisitor::VisitFunctionTypeLoc(FunctionTypeLoc TL,
146501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor                                         bool SkipResultType) {
146601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  if (!SkipResultType && Visit(TL.getResultLoc()))
1467f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1468f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1469f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
14705dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek    if (Decl *D = TL.getArg(I))
1471aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      if (Visit(MakeCXCursor(D, TU, RegionOfInterest)))
14725dbacb4179c759eef36bcaa6466b91518e3b98a9Ted Kremenek        return true;
1473f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1474f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1475f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1476f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1477f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregorbool CursorVisitor::VisitArrayTypeLoc(ArrayTypeLoc TL) {
1478f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Visit(TL.getElementLoc()))
1479f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor    return true;
1480f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1481f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  if (Expr *Size = TL.getSizeExpr())
1482aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    return Visit(MakeCXCursor(Size, StmtParent, TU, RegionOfInterest));
1483f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1484f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor  return false;
1485f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor}
1486f20dfbcdaaf96757ea67c94376bdca0bd64db02cDouglas Gregor
1487fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregorbool CursorVisitor::VisitTemplateSpecializationTypeLoc(
1488fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor                                             TemplateSpecializationTypeLoc TL) {
14890b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  // Visit the template name.
14900b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  if (VisitTemplateName(TL.getTypePtr()->getTemplateName(),
14910b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor                        TL.getTemplateNameLoc()))
14920b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    return true;
1493fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1494fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  // Visit the template arguments.
1495fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
1496fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
1497fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor      return true;
1498fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
1499fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  return false;
1500fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor}
1501fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor
15022332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfExprTypeLoc(TypeOfExprTypeLoc TL) {
15032332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return Visit(MakeCXCursor(TL.getUnderlyingExpr(), StmtParent, TU));
15042332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15052332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15062332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregorbool CursorVisitor::VisitTypeOfTypeLoc(TypeOfTypeLoc TL) {
15072332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
1508ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt    return Visit(TSInfo->getTypeLoc());
1509ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt
1510ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt  return false;
1511ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt}
1512ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt
1513ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Huntbool CursorVisitor::VisitUnaryTransformTypeLoc(UnaryTransformTypeLoc TL) {
1514ca63c200346c0ca9e00194ec6e34a5a7b0ed9321Sean Hunt  if (TypeSourceInfo *TSInfo = TL.getUnderlyingTInfo())
15152332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor    return Visit(TSInfo->getTypeLoc());
15162332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15172332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor  return false;
15182332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor}
15192332c117b1ab498322eb99c238630f28c32c7b14Douglas Gregor
15202494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregorbool CursorVisitor::VisitDependentNameTypeLoc(DependentNameTypeLoc TL) {
15212494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15222494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    return true;
15232494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
15242494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor  return false;
15252494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor}
15262494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
152794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregorbool CursorVisitor::VisitDependentTemplateSpecializationTypeLoc(
152894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor                                    DependentTemplateSpecializationTypeLoc TL) {
152994fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the nested-name-specifier, if there is one.
153094fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  if (TL.getQualifierLoc() &&
153194fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
153294fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    return true;
153394fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
153494fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  // Visit the template arguments.
153594fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  for (unsigned I = 0, N = TL.getNumArgs(); I != N; ++I)
153694fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor    if (VisitTemplateArgumentLoc(TL.getArgLoc(I)))
153794fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor      return true;
153894fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
153994fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor  return false;
154094fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor}
154194fdffa4a572fc14ac296f5f1aae9db3734c72f1Douglas Gregor
15429e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregorbool CursorVisitor::VisitElaboratedTypeLoc(ElaboratedTypeLoc TL) {
15439e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  if (VisitNestedNameSpecifierLoc(TL.getQualifierLoc()))
15449e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor    return true;
15459e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15469e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor  return Visit(TL.getNamedTypeLoc());
15479e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor}
15489e876876afc13aa671cc11a17c19907c599b9ab9Douglas Gregor
15497536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregorbool CursorVisitor::VisitPackExpansionTypeLoc(PackExpansionTypeLoc TL) {
15507536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor  return Visit(TL.getPatternLoc());
15517536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor}
15527536dd5e6c99584481b7dab68b7e7d8df9c54054Douglas Gregor
1553427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::VisitDecltypeTypeLoc(DecltypeTypeLoc TL) {
1554427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  if (Expr *E = TL.getUnderlyingExpr())
1555427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis    return Visit(MakeCXCursor(E, StmtParent, TU));
1556427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1557427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return false;
1558427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1559427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1560427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::VisitInjectedClassNameTypeLoc(InjectedClassNameTypeLoc TL) {
1561427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return Visit(MakeCursorTypeRef(TL.getDecl(), TL.getNameLoc(), TU));
1562427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1563427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1564427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis#define DEFAULT_TYPELOC_IMPL(CLASS, PARENT) \
1565427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidisbool CursorVisitor::Visit##CLASS##TypeLoc(CLASS##TypeLoc TL) { \
1566427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis  return Visit##PARENT##Loc(TL); \
1567427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis}
1568427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
1569427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Complex, Type)
1570427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(ConstantArray, ArrayType)
1571427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(IncompleteArray, ArrayType)
1572427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(VariableArray, ArrayType)
1573427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(DependentSizedArray, ArrayType)
1574427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(DependentSizedExtVector, Type)
1575427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Vector, Type)
1576427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(ExtVector, VectorType)
1577427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(FunctionProto, FunctionType)
1578427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(FunctionNoProto, FunctionType)
1579427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Record, TagType)
1580427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Enum, TagType)
1581427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(SubstTemplateTypeParm, Type)
1582427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(SubstTemplateTypeParmPack, Type)
1583427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios KyrtzidisDEFAULT_TYPELOC_IMPL(Auto, Type)
1584427964e15f1b9595659cea3fcb4dd808a00f37b5Argyrios Kyrtzidis
15853064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenekbool CursorVisitor::VisitCXXRecordDecl(CXXRecordDecl *D) {
1586c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  // Visit the nested-name-specifier, if present.
1587c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = D->getQualifierLoc())
1588c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor    if (VisitNestedNameSpecifierLoc(QualifierLoc))
1589c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor      return true;
1590c22b5fff39a7520207f165fb16a27a34b944bd9cDouglas Gregor
15913064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  if (D->isDefinition()) {
15923064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    for (CXXRecordDecl::base_class_iterator I = D->bases_begin(),
15933064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek         E = D->bases_end(); I != E; ++I) {
15943064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      if (Visit(cxcursor::MakeCursorCXXBaseSpecifier(I, TU)))
15953064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek        return true;
15963064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
15973064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  }
15983064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
15993064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  return VisitTagDecl(D);
16003064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek}
16013064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek
160209dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenekbool CursorVisitor::VisitAttributes(Decl *D) {
1603cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt  for (AttrVec::const_iterator i = D->attr_begin(), e = D->attr_end();
1604cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt       i != e; ++i)
1605cf807c4dfdb23e8fa3f400e0b24ef5b79db7a530Sean Hunt    if (Visit(MakeCXCursor(*i, D, TU)))
160609dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek        return true;
160709dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
160809dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek  return false;
160909dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek}
161009dfa37dda8e430fb6129bfd70365a51b4e12243Ted Kremenek
1611c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1612c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Data-recursive visitor methods.
1613c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
1614c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
161528a719433411ef782b582946823bc648ddcc4533Ted Kremeneknamespace {
1616035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#define DEF_JOB(NAME, DATA, KIND)\
1617035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass NAME : public VisitorJob {\
1618035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:\
1619035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  NAME(DATA *d, CXCursor parent) : VisitorJob(parent, VisitorJob::KIND, d) {} \
1620035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) { return VJ->getKind() == KIND; }\
1621f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DATA *get() const { return static_cast<DATA*>(data[0]); }\
1622035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1623035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1624035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(StmtVisit, Stmt, StmtVisitKind)
1625035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(MemberExprParts, MemberExpr, MemberExprPartsKind)
1626e4979ccb5960608edce73f3b274eb7c2de15dac5Ted KremenekDEF_JOB(DeclRefExprParts, DeclRefExpr, DeclRefExprPartsKind)
1627035dc41b509fcc470ceb6764aa64837505a2ece3Ted KremenekDEF_JOB(OverloadExprParts, OverloadExpr, OverloadExprPartsKind)
1628b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios KyrtzidisDEF_JOB(ExplicitTemplateArgsVisit, ASTTemplateArgumentListInfo,
162960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        ExplicitTemplateArgsVisitKind)
163094d96291cd041adc5731a2294828a9c20e450b74Douglas GregorDEF_JOB(SizeOfPackExprParts, SizeOfPackExpr, SizeOfPackExprPartsKind)
1631035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek#undef DEF_JOB
1632035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1633035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass DeclVisit : public VisitorJob {
1634035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1635035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  DeclVisit(Decl *d, CXCursor parent, bool isFirst) :
1636035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::DeclVisitKind,
1637035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               d, isFirst ? (void*) 1 : (void*) 0) {}
1638035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
163982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    return VJ->getKind() == DeclVisitKind;
1640035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1641f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  Decl *get() const { return static_cast<Decl*>(data[0]); }
1642f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  bool isFirst() const { return data[1] ? true : false; }
1643035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1644035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekclass TypeLocVisit : public VisitorJob {
1645035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekpublic:
1646035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  TypeLocVisit(TypeLoc tl, CXCursor parent) :
1647035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    VisitorJob(parent, VisitorJob::TypeLocVisitKind,
1648035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek               tl.getType().getAsOpaquePtr(), tl.getOpaqueData()) {}
1649035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1650035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1651035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return VJ->getKind() == TypeLocVisitKind;
1652035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1653035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
165482f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek  TypeLoc get() const {
1655f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    QualType T = QualType::getFromOpaquePtr(data[0]);
1656f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return TypeLoc(T, data[1]);
1657035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1658035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek};
1659035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek
1660ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekclass LabelRefVisit : public VisitorJob {
1661ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekpublic:
1662ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelRefVisit(LabelDecl *LD, SourceLocation labelLoc, CXCursor parent)
1663ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner    : VisitorJob(parent, VisitorJob::LabelRefVisitKind, LD,
1664dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 labelLoc.getPtrEncoding()) {}
1665ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek
1666ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1667ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek    return VJ->getKind() == VisitorJob::LabelRefVisitKind;
1668ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  }
1669ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  LabelDecl *get() const { return static_cast<LabelDecl*>(data[0]); }
1670ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  SourceLocation getLoc() const {
1671dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin    return SourceLocation::getFromPtrEncoding(data[1]); }
1672f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek};
1673f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1674f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorclass NestedNameSpecifierLocVisit : public VisitorJob {
1675f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorpublic:
1676f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLocVisit(NestedNameSpecifierLoc Qualifier, CXCursor parent)
1677f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    : VisitorJob(parent, VisitorJob::NestedNameSpecifierLocVisitKind,
1678f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getNestedNameSpecifier(),
1679f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                 Qualifier.getOpaqueData()) { }
1680f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1681f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  static bool classof(const VisitorJob *VJ) {
1682f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return VJ->getKind() == VisitorJob::NestedNameSpecifierLocVisitKind;
1683f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1684f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1685f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  NestedNameSpecifierLoc get() const {
1686f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    return NestedNameSpecifierLoc(static_cast<NestedNameSpecifier*>(data[0]),
1687f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor                                  data[1]);
1688f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  }
1689f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor};
1690f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1691f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekclass DeclarationNameInfoVisit : public VisitorJob {
1692f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekpublic:
1693f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfoVisit(Stmt *S, CXCursor parent)
1694f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    : VisitorJob(parent, VisitorJob::DeclarationNameInfoVisitKind, S) {}
1695f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  static bool classof(const VisitorJob *VJ) {
1696f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    return VJ->getKind() == VisitorJob::DeclarationNameInfoVisitKind;
1697f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1698f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  DeclarationNameInfo get() const {
1699f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    Stmt *S = static_cast<Stmt*>(data[0]);
1700f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    switch (S->getStmtClass()) {
1701f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    default:
1702f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      llvm_unreachable("Unhandled Stmt");
1703f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::CXXDependentScopeMemberExprClass:
1704f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<CXXDependentScopeMemberExpr>(S)->getMemberNameInfo();
1705f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    case Stmt::DependentScopeDeclRefExprClass:
1706f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      return cast<DependentScopeDeclRefExpr>(S)->getNameInfo();
1707f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    }
1708f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  }
1709ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek};
1710cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekclass MemberRefVisit : public VisitorJob {
1711cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekpublic:
1712cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  MemberRefVisit(FieldDecl *D, SourceLocation L, CXCursor parent)
1713cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    : VisitorJob(parent, VisitorJob::MemberRefVisitKind, D,
1714dec0984fce504a39a7f085774fb67cfd9957be58Jeffrey Yasskin                 L.getPtrEncoding()) {}
1715cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  static bool classof(const VisitorJob *VJ) {
1716cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return VJ->getKind() == VisitorJob::MemberRefVisitKind;
1717cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1718cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  FieldDecl *get() const {
1719cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return static_cast<FieldDecl*>(data[0]);
1720cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1721cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  SourceLocation getLoc() const {
1722cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    return SourceLocation::getFromRawEncoding((unsigned)(uintptr_t) data[1]);
1723cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1724cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek};
172528a719433411ef782b582946823bc648ddcc4533Ted Kremenekclass EnqueueVisitor : public StmtVisitor<EnqueueVisitor, void> {
172628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitorWorkList &WL;
172728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  CXCursor Parent;
172828a719433411ef782b582946823bc648ddcc4533Ted Kremenekpublic:
172928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueVisitor(VisitorWorkList &wl, CXCursor parent)
173028a719433411ef782b582946823bc648ddcc4533Ted Kremenek    : WL(wl), Parent(parent) {}
173128a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1732ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitAddrLabelExpr(AddrLabelExpr *E);
173373d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitBlockExpr(BlockExpr *B);
173428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCompoundLiteralExpr(CompoundLiteralExpr *E);
1735083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  void VisitCompoundStmt(CompoundStmt *S);
173611b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXDefaultArgExpr(CXXDefaultArgExpr *E) { /* Do nothing. */ }
1737f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E);
173811b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  void VisitCXXNewExpr(CXXNewExpr *E);
17396d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  void VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E);
174028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitCXXOperatorCallExpr(CXXOperatorCallExpr *E);
1741cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E);
174273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E);
1743b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  void VisitCXXTypeidExpr(CXXTypeidExpr *E);
174455b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  void VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr *E);
17451e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  void VisitCXXUuidofExpr(CXXUuidofExpr *E);
1746e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  void VisitDeclRefExpr(DeclRefExpr *D);
1747035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void VisitDeclStmt(DeclStmt *S);
1748f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E);
1749cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitDesignatedInitExpr(DesignatedInitExpr *E);
175028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitExplicitCastExpr(ExplicitCastExpr *E);
175128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitForStmt(ForStmt *FS);
1752ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  void VisitGotoStmt(GotoStmt *GS);
175328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitIfStmt(IfStmt *If);
175428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitInitListExpr(InitListExpr *IE);
175528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitMemberExpr(MemberExpr *M);
1756cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void VisitOffsetOfExpr(OffsetOfExpr *E);
175773d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  void VisitObjCEncodeExpr(ObjCEncodeExpr *E);
175828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitObjCMessageExpr(ObjCMessageExpr *M);
175928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitOverloadExpr(OverloadExpr *E);
1760f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne  void VisitUnaryExprOrTypeTraitExpr(UnaryExprOrTypeTraitExpr *E);
176128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitStmt(Stmt *S);
176228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitSwitchStmt(SwitchStmt *S);
176328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitWhileStmt(WhileStmt *W);
17642939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  void VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E);
17656ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  void VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E);
176621ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley  void VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E);
1767552622067dc45013d240f73952fece703f5e63bdJohn Wiegley  void VisitExpressionTraitExpr(ExpressionTraitExpr *E);
176828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U);
17699d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  void VisitVAArgExpr(VAArgExpr *E);
177094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  void VisitSizeOfPackExpr(SizeOfPackExpr *E);
1771ee8aff06f6a96214731de17b2cb6df407c6c1820Douglas Gregor
177228a719433411ef782b582946823bc648ddcc4533Ted Kremenekprivate:
1773f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  void AddDeclarationNameInfo(Stmt *S);
1774f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  void AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier);
1775b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis  void AddExplicitTemplateArgs(const ASTTemplateArgumentListInfo *A);
1776cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  void AddMemberRef(FieldDecl *D, SourceLocation L);
177728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddStmt(Stmt *S);
1778035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  void AddDecl(Decl *D, bool isFirst = true);
177928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void AddTypeLoc(TypeSourceInfo *TI);
178028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  void EnqueueChildren(Stmt *S);
178128a719433411ef782b582946823bc648ddcc4533Ted Kremenek};
178228a719433411ef782b582946823bc648ddcc4533Ted Kremenek} // end anonyous namespace
178328a719433411ef782b582946823bc648ddcc4533Ted Kremenek
1784f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::AddDeclarationNameInfo(Stmt *S) {
1785f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // 'S' should always be non-null, since it comes from the
1786f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  // statement we are visiting.
1787f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  WL.push_back(DeclarationNameInfoVisit(S, Parent));
1788f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1789f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
1790f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregorvoid
1791f3db29fff6a583ecda823cf909ab7737d8d30129Douglas GregorEnqueueVisitor::AddNestedNameSpecifierLoc(NestedNameSpecifierLoc Qualifier) {
1792f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (Qualifier)
1793f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    WL.push_back(NestedNameSpecifierLocVisit(Qualifier, Parent));
1794f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor}
1795f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
179628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddStmt(Stmt *S) {
179728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (S)
179828a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(StmtVisit(S, Parent));
179928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1800035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::AddDecl(Decl *D, bool isFirst) {
180128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (D)
1802035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    WL.push_back(DeclVisit(D, Parent, isFirst));
180328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
180460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenekvoid EnqueueVisitor::
1805b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis  AddExplicitTemplateArgs(const ASTTemplateArgumentListInfo *A) {
180660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (A)
180760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    WL.push_back(ExplicitTemplateArgsVisit(
1808b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis                        const_cast<ASTTemplateArgumentListInfo*>(A), Parent));
180960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek}
1810cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::AddMemberRef(FieldDecl *D, SourceLocation L) {
1811cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  if (D)
1812cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    WL.push_back(MemberRefVisit(D, L, Parent));
1813cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
181428a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::AddTypeLoc(TypeSourceInfo *TI) {
181528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (TI)
181628a719433411ef782b582946823bc648ddcc4533Ted Kremenek    WL.push_back(TypeLocVisit(TI->getTypeLoc(), Parent));
181728a719433411ef782b582946823bc648ddcc4533Ted Kremenek }
181828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::EnqueueChildren(Stmt *S) {
1819a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  unsigned size = WL.size();
18207502c1d3ce8bb97bcc4f7bebef507040bd93b26fJohn McCall  for (Stmt::child_range Child = S->children(); Child; ++Child) {
182128a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(*Child);
1822a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  }
1823a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  if (size == WL.size())
1824a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek    return;
1825a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // Now reverse the entries we just added.  This will match the DFS
1826a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  // ordering performed by the worklist.
1827a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1828a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek  std::reverse(I, E);
1829a6b70435ae49f2c7da1ad2b200c1aeb9f8761fddTed Kremenek}
1830ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitAddrLabelExpr(AddrLabelExpr *E) {
1831ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(E->getLabel(), E->getLabelLoc(), Parent));
1832ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
183373d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitBlockExpr(BlockExpr *B) {
183473d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddDecl(B->getBlockDecl());
183573d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
183628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCompoundLiteralExpr(CompoundLiteralExpr *E) {
183728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
183828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeSourceInfo());
183928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1840083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenekvoid EnqueueVisitor::VisitCompoundStmt(CompoundStmt *S) {
1841083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  for (CompoundStmt::reverse_body_iterator I = S->body_rbegin(),
1842083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek        E = S->body_rend(); I != E; ++I) {
1843083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek    AddStmt(*I);
1844083c7e2d564033af87e507fbbd02f1c77ff462b1Ted Kremenek  }
184511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
1846f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::
1847f64d80306144f978148ba92f36f7cea7b671dd34Ted KremenekVisitCXXDependentScopeMemberExpr(CXXDependentScopeMemberExpr *E) {
1848f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1849f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
18507c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
18517c3179cf463c3b3b8c21dbb955f933ba50b74f28Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1852f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  if (!E->isImplicitAccess())
1853f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek    AddStmt(E->getBase());
1854f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
185511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenekvoid EnqueueVisitor::VisitCXXNewExpr(CXXNewExpr *E) {
185611b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the initializer or constructor arguments.
185711b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumConstructorArgs(); I > 0; --I)
185811b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getConstructorArg(I-1));
185911b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the array size, if any.
186011b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddStmt(E->getArraySize());
186111b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the allocated type.
186211b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  AddTypeLoc(E->getAllocatedTypeSourceInfo());
186311b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  // Enqueue the placement arguments.
186411b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek  for (unsigned I = E->getNumPlacementArgs(); I > 0; --I)
186511b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek    AddStmt(E->getPlacementArg(I-1));
186611b8e3e7fa67795acc968a9d5ebfd687feaf2b2cTed Kremenek}
186728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitCXXOperatorCallExpr(CXXOperatorCallExpr *CE) {
18688b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek  for (unsigned I = CE->getNumArgs(); I > 1 /* Yes, this is 1 */; --I)
18698b8d8c90f2d8ac651d14b57f116d20b3c911ac7fTed Kremenek    AddStmt(CE->getArg(I-1));
187028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getCallee());
187128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(CE->getArg(0));
187228a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1873cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitCXXPseudoDestructorExpr(CXXPseudoDestructorExpr *E) {
1874cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the name of the type being destroyed.
1875cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getDestroyedTypeInfo());
1876cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the scope type that looks disturbingly like the nested-name-specifier
1877cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // but isn't.
1878cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getScopeTypeInfo());
1879cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the nested-name-specifier.
1880f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor  if (NestedNameSpecifierLoc QualifierLoc = E->getQualifierLoc())
1881f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor    AddNestedNameSpecifierLoc(QualifierLoc);
1882cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit base expression.
1883cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getBase());
1884cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
18856d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenekvoid EnqueueVisitor::VisitCXXScalarValueInitExpr(CXXScalarValueInitExpr *E) {
18866d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
18876d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
188873d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitCXXTemporaryObjectExpr(CXXTemporaryObjectExpr *E) {
188973d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  EnqueueChildren(E);
189073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
189173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
1892b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenekvoid EnqueueVisitor::VisitCXXTypeidExpr(CXXTypeidExpr *E) {
1893b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  EnqueueChildren(E);
1894b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek  if (E->isTypeOperand())
1895b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
1896b8dd1cad52be63b18092bd9d9335cbeee3f5de9fTed Kremenek}
189755b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek
189855b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenekvoid EnqueueVisitor::VisitCXXUnresolvedConstructExpr(CXXUnresolvedConstructExpr
189955b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek                                                     *E) {
190055b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  EnqueueChildren(E);
190155b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
190255b933ae3b3efc29987b8bd92714257f00c4b53aTed Kremenek}
19031e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenekvoid EnqueueVisitor::VisitCXXUuidofExpr(CXXUuidofExpr *E) {
19041e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  EnqueueChildren(E);
19051e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek  if (E->isTypeOperand())
19061e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek    AddTypeLoc(E->getTypeOperandSourceInfo());
19071e7e877091187556bb6d644ab2b7c00a628121ebTed Kremenek}
1908e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenekvoid EnqueueVisitor::VisitDeclRefExpr(DeclRefExpr *DR) {
190960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  if (DR->hasExplicitTemplateArgs()) {
191060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek    AddExplicitTemplateArgs(&DR->getExplicitTemplateArgs());
191160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  }
1912e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek  WL.push_back(DeclRefExprParts(DR, Parent));
1913e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek}
1914f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenekvoid EnqueueVisitor::VisitDependentScopeDeclRefExpr(DependentScopeDeclRefExpr *E) {
1915f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
1916f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek  AddDeclarationNameInfo(E);
191700cf3cc2718671aa48e8da264a523b0058a8591eDouglas Gregor  AddNestedNameSpecifierLoc(E->getQualifierLoc());
1918f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek}
1919035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenekvoid EnqueueVisitor::VisitDeclStmt(DeclStmt *S) {
1920035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  unsigned size = WL.size();
1921035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  bool isFirst = true;
1922035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  for (DeclStmt::decl_iterator D = S->decl_begin(), DEnd = S->decl_end();
1923035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek       D != DEnd; ++D) {
1924035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    AddDecl(*D, isFirst);
1925035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    isFirst = false;
1926035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  }
1927035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  if (size == WL.size())
1928035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek    return;
1929035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // Now reverse the entries we just added.  This will match the DFS
1930035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  // ordering performed by the worklist.
1931035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  VisitorWorkList::iterator I = WL.begin() + size, E = WL.end();
1932035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek  std::reverse(I, E);
1933035dc41b509fcc470ceb6764aa64837505a2ece3Ted Kremenek}
1934cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitDesignatedInitExpr(DesignatedInitExpr *E) {
1935cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddStmt(E->getInit());
1936cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  typedef DesignatedInitExpr::Designator Designator;
1937cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (DesignatedInitExpr::reverse_designators_iterator
1938cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D = E->designators_rbegin(), DEnd = E->designators_rend();
1939cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek         D != DEnd; ++D) {
1940cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isFieldDesignator()) {
1941cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      if (FieldDecl *Field = D->getField())
1942cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        AddMemberRef(Field, D->getFieldLoc());
1943cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1944cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1945cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    if (D->isArrayDesignator()) {
1946cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getArrayIndex(*D));
1947cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
1948cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
1949cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    assert(D->isArrayRangeDesignator() && "Unknown designator kind");
1950cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeEnd(*D));
1951cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    AddStmt(E->getArrayRangeStart(*D));
1952cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
1953cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
195428a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitExplicitCastExpr(ExplicitCastExpr *E) {
195528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(E);
195628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(E->getTypeInfoAsWritten());
195728a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
195828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitForStmt(ForStmt *FS) {
195928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getBody());
196028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInc());
196128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getCond());
196228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(FS->getConditionVariable());
196328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(FS->getInit());
196428a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1965ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenekvoid EnqueueVisitor::VisitGotoStmt(GotoStmt *GS) {
1966ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek  WL.push_back(LabelRefVisit(GS->getLabel(), GS->getLabelLoc(), Parent));
1967ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek}
196828a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitIfStmt(IfStmt *If) {
196928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getElse());
197028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getThen());
197128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(If->getCond());
197228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(If->getConditionVariable());
197328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
197428a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitInitListExpr(InitListExpr *IE) {
197528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  // We care about the syntactic form of the initializer list, only.
197628a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (InitListExpr *Syntactic = IE->getSyntacticForm())
197728a719433411ef782b582946823bc648ddcc4533Ted Kremenek    IE = Syntactic;
197828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(IE);
197928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
198028a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitMemberExpr(MemberExpr *M) {
198189629a746019a42797495b091711a1d68467e88aDouglas Gregor  WL.push_back(MemberExprParts(M, Parent));
198289629a746019a42797495b091711a1d68467e88aDouglas Gregor
198389629a746019a42797495b091711a1d68467e88aDouglas Gregor  // If the base of the member access expression is an implicit 'this', don't
198489629a746019a42797495b091711a1d68467e88aDouglas Gregor  // visit it.
198589629a746019a42797495b091711a1d68467e88aDouglas Gregor  // FIXME: If we ever want to show these implicit accesses, this will be
198689629a746019a42797495b091711a1d68467e88aDouglas Gregor  // unfortunate. However, clang_getCursor() relies on this behavior.
198775e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor  if (!M->isImplicitAccess())
198875e85048e73fcde2ce9d8a48dfdb1220e132eb59Douglas Gregor    AddStmt(M->getBase());
198928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
199073d15c452e675b684b7eee4f2096e386e59397aaTed Kremenekvoid EnqueueVisitor::VisitObjCEncodeExpr(ObjCEncodeExpr *E) {
199173d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek  AddTypeLoc(E->getEncodedTypeSourceInfo());
199273d15c452e675b684b7eee4f2096e386e59397aaTed Kremenek}
199328a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitObjCMessageExpr(ObjCMessageExpr *M) {
199428a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(M);
199528a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddTypeLoc(M->getClassReceiverTypeInfo());
199628a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
1997cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekvoid EnqueueVisitor::VisitOffsetOfExpr(OffsetOfExpr *E) {
1998cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the components of the offsetof expression.
1999cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  for (unsigned N = E->getNumComponents(), I = N; I > 0; --I) {
2000cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    typedef OffsetOfExpr::OffsetOfNode OffsetOfNode;
2001cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    const OffsetOfNode &Node = E->getComponent(I-1);
2002cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    switch (Node.getKind()) {
2003cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Array:
2004cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      AddStmt(E->getIndexExpr(Node.getArrayExprIndex()));
2005cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2006cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Field:
200706dec892b5300b43263d25c5476b506c9d6cfbadAbramo Bagnara      AddMemberRef(Node.getField(), Node.getSourceRange().getEnd());
2008cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      break;
2009cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Identifier:
2010cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    case OffsetOfNode::Base:
2011cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      continue;
2012cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek    }
2013cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  }
2014cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  // Visit the type into which we're computing the offset.
2015cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek  AddTypeLoc(E->getTypeSourceInfo());
2016cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek}
201728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitOverloadExpr(OverloadExpr *E) {
201860608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek  AddExplicitTemplateArgs(E->getOptionalExplicitTemplateArgs());
20196045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek  WL.push_back(OverloadExprParts(E, Parent));
20206045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek}
2021f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbournevoid EnqueueVisitor::VisitUnaryExprOrTypeTraitExpr(
2022f4e3cfbe8abd124be6341ef5d714819b4fbd9082Peter Collingbourne                                              UnaryExprOrTypeTraitExpr *E) {
20236d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  EnqueueChildren(E);
20246d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek  if (E->isArgumentType())
20256d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek    AddTypeLoc(E->getArgumentTypeInfo());
20266d0a00d9b02499d6c1253ea03d4dc7c32f8f289eTed Kremenek}
202728a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitStmt(Stmt *S) {
202828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  EnqueueChildren(S);
202928a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
203028a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitSwitchStmt(SwitchStmt *S) {
203128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getBody());
203228a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(S->getCond());
203328a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(S->getConditionVariable());
203428a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
2035fafa75aebadef8d6b44a920e3f40529f150a5574Ted Kremenek
203628a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitWhileStmt(WhileStmt *W) {
203728a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getBody());
203828a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddStmt(W->getCond());
203928a719433411ef782b582946823bc648ddcc4533Ted Kremenek  AddDecl(W->getConditionVariable());
204028a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
204121ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley
20422939b6f356161f572712d4d6310b65f9599e3675Ted Kremenekvoid EnqueueVisitor::VisitUnaryTypeTraitExpr(UnaryTypeTraitExpr *E) {
20432939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek  AddTypeLoc(E->getQueriedTypeSourceInfo());
20442939b6f356161f572712d4d6310b65f9599e3675Ted Kremenek}
20456ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
20466ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichetvoid EnqueueVisitor::VisitBinaryTypeTraitExpr(BinaryTypeTraitExpr *E) {
20476ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet  AddTypeLoc(E->getRhsTypeSourceInfo());
20480a03a3f98b14006a54bcac9e8908a7c9f50e519fFrancois Pichet  AddTypeLoc(E->getLhsTypeSourceInfo());
20496ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet}
20506ad6f2848d7652ab2991286eb48be440d3493b28Francois Pichet
205121ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegleyvoid EnqueueVisitor::VisitArrayTypeTraitExpr(ArrayTypeTraitExpr *E) {
205221ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley  AddTypeLoc(E->getQueriedTypeSourceInfo());
205321ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley}
205421ff2e516b0e0bc8c1dbf965cb3d44bac3c64330John Wiegley
2055552622067dc45013d240f73952fece703f5e63bdJohn Wiegleyvoid EnqueueVisitor::VisitExpressionTraitExpr(ExpressionTraitExpr *E) {
2056552622067dc45013d240f73952fece703f5e63bdJohn Wiegley  EnqueueChildren(E);
2057552622067dc45013d240f73952fece703f5e63bdJohn Wiegley}
2058552622067dc45013d240f73952fece703f5e63bdJohn Wiegley
205928a719433411ef782b582946823bc648ddcc4533Ted Kremenekvoid EnqueueVisitor::VisitUnresolvedMemberExpr(UnresolvedMemberExpr *U) {
206028a719433411ef782b582946823bc648ddcc4533Ted Kremenek  VisitOverloadExpr(U);
206128a719433411ef782b582946823bc648ddcc4533Ted Kremenek  if (!U->isImplicitAccess())
206228a719433411ef782b582946823bc648ddcc4533Ted Kremenek    AddStmt(U->getBase());
206328a719433411ef782b582946823bc648ddcc4533Ted Kremenek}
20649d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenekvoid EnqueueVisitor::VisitVAArgExpr(VAArgExpr *E) {
20659d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddStmt(E->getSubExpr());
20669d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek  AddTypeLoc(E->getWrittenTypeInfo());
20679d3bf79d94c961af95144ef63bbd6f9f3f32c59aTed Kremenek}
206894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregorvoid EnqueueVisitor::VisitSizeOfPackExpr(SizeOfPackExpr *E) {
206994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  WL.push_back(SizeOfPackExprParts(E, Parent));
207094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor}
20716045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek
2072c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekvoid CursorVisitor::EnqueueWorkList(VisitorWorkList &WL, Stmt *S) {
2073aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  EnqueueVisitor(WL, MakeCXCursor(S, StmtParent, TU,RegionOfInterest)).Visit(S);
2074c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2075c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2076c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::IsInRegionOfInterest(CXCursor C) {
2077c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  if (RegionOfInterest.isValid()) {
2078c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SourceRange Range = getRawCursorExtent(C);
2079c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    if (Range.isInvalid() || CompareRegionOfInterest(Range))
2080c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      return false;
2081c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2082c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return true;
2083c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2084c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2085c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenekbool CursorVisitor::RunVisitorWorkList(VisitorWorkList &WL) {
2086c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  while (!WL.empty()) {
2087c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Dequeue the worklist item.
208882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    VisitorJob LI = WL.back();
208982f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek    WL.pop_back();
209082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek
2091c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    // Set the Parent field, then back to its old value once we're done.
2092c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    SetParentRAII SetParent(Parent, StmtParent, LI.getParent());
2093c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2094c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    switch (LI.getKind()) {
2095f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      case VisitorJob::DeclVisitKind: {
209682f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Decl *D = cast<DeclVisit>(&LI)->get();
2097f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        if (!D)
2098f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek          continue;
2099f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2100f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // For now, perform default visitation for Decls.
2101aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis        if (Visit(MakeCXCursor(D, TU, RegionOfInterest,
2102aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis                               cast<DeclVisit>(&LI)->isFirst())))
2103f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek            return true;
2104f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek
2105f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        continue;
2106f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek      }
210760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      case VisitorJob::ExplicitTemplateArgsVisitKind: {
2108b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis        const ASTTemplateArgumentListInfo *ArgList =
210960608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          cast<ExplicitTemplateArgsVisit>(&LI)->get();
211060608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        for (const TemplateArgumentLoc *Arg = ArgList->getTemplateArgs(),
211160608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               *ArgEnd = Arg + ArgList->NumTemplateArgs;
211260608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek               Arg != ArgEnd; ++Arg) {
211360608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek          if (VisitTemplateArgumentLoc(*Arg))
211460608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek            return true;
211560608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        }
211660608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek        continue;
211760608ec12d17168a3d1f415409a6a6eaf6d94508Ted Kremenek      }
2118cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      case VisitorJob::TypeLocVisitKind: {
2119cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        // Perform default visitation for TypeLocs.
212082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        if (Visit(cast<TypeLocVisit>(&LI)->get()))
2121cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek          return true;
2122cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek        continue;
2123cdb4caf6bd8beb562e169b4d3f6c604c4e4528f8Ted Kremenek      }
2124ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      case VisitorJob::LabelRefVisitKind: {
2125ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner        LabelDecl *LS = cast<LabelRefVisit>(&LI)->get();
2126e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        if (LabelStmt *stmt = LS->getStmt()) {
2127e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          if (Visit(MakeCursorLabelRef(stmt, cast<LabelRefVisit>(&LI)->getLoc(),
2128e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek                                       TU))) {
2129e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek            return true;
2130e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek          }
2131e7455016e5fae3db7e2d88a61633f76ab05fc9d3Ted Kremenek        }
2132ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek        continue;
2133ae1fd6fd47ce3fbb878155af6f517adaeae15a51Ted Kremenek      }
213447695c8ad8424851f62e0d4a983b45b15daee1c5Ted Kremenek
2135f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      case VisitorJob::NestedNameSpecifierLocVisitKind: {
2136f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        NestedNameSpecifierLocVisit *V = cast<NestedNameSpecifierLocVisit>(&LI);
2137f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        if (VisitNestedNameSpecifierLoc(V->get()))
2138f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor          return true;
2139f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor        continue;
2140f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor      }
2141f3db29fff6a583ecda823cf909ab7737d8d30129Douglas Gregor
2142f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      case VisitorJob::DeclarationNameInfoVisitKind: {
2143f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        if (VisitDeclarationNameInfo(cast<DeclarationNameInfoVisit>(&LI)
2144f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek                                     ->get()))
2145f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek          return true;
2146f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek        continue;
2147f64d80306144f978148ba92f36f7cea7b671dd34Ted Kremenek      }
2148cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      case VisitorJob::MemberRefVisitKind: {
2149cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        MemberRefVisit *V = cast<MemberRefVisit>(&LI);
2150cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (Visit(MakeCursorMemberRef(V->get(), V->getLoc(), TU)))
2151cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          return true;
2152cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        continue;
2153cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek      }
2154c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::StmtVisitKind: {
215582f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        Stmt *S = cast<StmtVisit>(&LI)->get();
21568c269ac75569454a049385b1246140db5f2b6faaTed Kremenek        if (!S)
21578c269ac75569454a049385b1246140db5f2b6faaTed Kremenek          continue;
21588c269ac75569454a049385b1246140db5f2b6faaTed Kremenek
2159f1107457c7d5cb57718725c51dfed4bf2ba6ee9eTed Kremenek        // Update the current cursor.
2160aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis        CXCursor Cursor = MakeCXCursor(S, StmtParent, TU, RegionOfInterest);
2161cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        if (!IsInRegionOfInterest(Cursor))
2162cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          continue;
2163cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek        switch (Visitor(Cursor, Parent, ClientData)) {
2164cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Break: return true;
2165cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Continue: break;
2166cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek          case CXChildVisit_Recurse:
2167cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenek            EnqueueWorkList(WL, S);
216882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek            break;
2169c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
217082f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        continue;
2171c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2172c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      case VisitorJob::MemberExprPartsKind: {
2173c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Handle the other pieces in the MemberExpr besides the base.
217482f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        MemberExpr *M = cast<MemberExprParts>(&LI)->get();
2175c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2176c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the nested-name-specifier
217740d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = M->getQualifierLoc())
217840d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2179c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            return true;
2180c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2181c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the declaration name.
2182c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (VisitDeclarationNameInfo(M->getMemberNameInfo()))
2183c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          return true;
2184c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2185c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        // Visit the explicitly-specified template arguments, if any.
2186c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        if (M->hasExplicitTemplateArgs()) {
2187c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          for (const TemplateArgumentLoc *Arg = M->getTemplateArgs(),
2188c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               *ArgEnd = Arg + M->getNumTemplateArgs();
2189c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek               Arg != ArgEnd; ++Arg) {
2190c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek            if (VisitTemplateArgumentLoc(*Arg))
2191c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek              return true;
2192c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek          }
2193c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        }
2194c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek        continue;
2195c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek      }
2196e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      case VisitorJob::DeclRefExprPartsKind: {
219782f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        DeclRefExpr *DR = cast<DeclRefExprParts>(&LI)->get();
2198e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit nested-name-specifier, if present.
219940d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = DR->getQualifierLoc())
220040d96a69c0e1e8c10f92d450c305a7aae696ca9cDouglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
2201e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek            return true;
2202e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        // Visit declaration name.
2203e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        if (VisitDeclarationNameInfo(DR->getNameInfo()))
2204e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek          return true;
2205e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek        continue;
2206e4979ccb5960608edce73f3b274eb7c2de15dac5Ted Kremenek      }
22076045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      case VisitorJob::OverloadExprPartsKind: {
220882f3c50fa163f99d1407849e556d3859a09afd78Ted Kremenek        OverloadExpr *O = cast<OverloadExprParts>(&LI)->get();
22096045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the nested-name-specifier.
22104c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor        if (NestedNameSpecifierLoc QualifierLoc = O->getQualifierLoc())
22114c9be89bb615ec07eb3ed507c8fa9d0baa8a5ad7Douglas Gregor          if (VisitNestedNameSpecifierLoc(QualifierLoc))
22126045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek            return true;
22136045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the declaration name.
22146045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (VisitDeclarationNameInfo(O->getNameInfo()))
22156045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22166045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        // Visit the overloaded declaration reference.
22176045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        if (Visit(MakeCursorOverloadedDeclRef(O, TU)))
22186045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek          return true;
22196045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek        continue;
22206045878f1fa7fad19d826634617991db99d3bd22Ted Kremenek      }
222194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      case VisitorJob::SizeOfPackExprPartsKind: {
222294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        SizeOfPackExpr *E = cast<SizeOfPackExprParts>(&LI)->get();
222394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        NamedDecl *Pack = E->getPack();
222494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTypeParmDecl>(Pack)) {
222594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTypeRef(cast<TemplateTypeParmDecl>(Pack),
222694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                      E->getPackLoc(), TU)))
222794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
222894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
222994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
223094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
223194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
223294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        if (isa<TemplateTemplateParmDecl>(Pack)) {
223394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          if (Visit(MakeCursorTemplateRef(cast<TemplateTemplateParmDecl>(Pack),
223494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor                                          E->getPackLoc(), TU)))
223594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor            return true;
223694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
223794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor          continue;
223894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        }
223994d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
224094d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // Non-type template parameter packs and function parameter packs are
224194d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        // treated like DeclRefExpr cursors.
224294d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        continue;
224394d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      }
2244c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek    }
2245c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  }
2246c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek  return false;
2247c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2248c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
2249cdba6595a61a7bd31f504260abf63c900a759d0fTed Kremenekbool CursorVisitor::Visit(Stmt *S) {
2250d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  VisitorWorkList *WL = 0;
2251d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  if (!WorkListFreeList.empty()) {
2252d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = WorkListFreeList.back();
2253d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL->clear();
2254d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListFreeList.pop_back();
2255d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2256d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  else {
2257d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WL = new VisitorWorkList();
2258d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek    WorkListCache.push_back(WL);
2259d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  }
2260d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  EnqueueWorkList(*WL, S);
2261d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  bool result = RunVisitorWorkList(*WL);
2262d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  WorkListFreeList.push_back(WL);
2263d1ded66c4eda8d170222071dec7ebba78bd86ea4Ted Kremenek  return result;
2264c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek}
2265c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
226648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichetnamespace {
226748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichettypedef llvm::SmallVector<SourceRange, 4> RefNamePieces;
226848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois PichetRefNamePieces buildPieces(unsigned NameFlags, bool IsMemberRefExpr,
226948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                          const DeclarationNameInfo &NI,
227048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                          const SourceRange &QLoc,
2271b0c3e0909bb04af0bfb82ad01ab6909649d68ccaArgyrios Kyrtzidis                          const ASTTemplateArgumentListInfo *TemplateArgs = 0){
227248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantQualifier = NameFlags & CXNameRange_WantQualifier;
227348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantTemplateArgs = NameFlags & CXNameRange_WantTemplateArgs;
227448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const bool WantSinglePiece = NameFlags & CXNameRange_WantSinglePiece;
227548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
227648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  const DeclarationName::NameKind Kind = NI.getName().getNameKind();
227748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
227848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  RefNamePieces Pieces;
227948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
228048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantQualifier && QLoc.isValid())
228148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(QLoc);
228248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
228348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (Kind != DeclarationName::CXXOperatorName || IsMemberRefExpr)
228448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(NI.getLoc());
228548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
228648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantTemplateArgs && TemplateArgs)
228748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceRange(TemplateArgs->LAngleLoc,
228848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                                 TemplateArgs->RAngleLoc));
228948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
229048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (Kind == DeclarationName::CXXOperatorName) {
229148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceLocation::getFromRawEncoding(
229248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                       NI.getInfo().CXXOperatorName.BeginOpNameLoc));
229348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(SourceLocation::getFromRawEncoding(
229448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet                       NI.getInfo().CXXOperatorName.EndOpNameLoc));
229548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  }
229648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
229748a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  if (WantSinglePiece) {
229848a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    SourceRange R(Pieces.front().getBegin(), Pieces.back().getEnd());
229948a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.clear();
230048a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet    Pieces.push_back(R);
230148a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  }
230248a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
230348a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet  return Pieces;
230448a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet}
230548a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet}
230648a8d14fc6f064a5297024c2b34733a4080b2efeFrancois Pichet
2307c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2308c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek// Misc. API hooks.
2309c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek//===----------------------------------------------------------------------===//
2310c0e1d923231d42fb0ac86e56a29ffc5c754657caTed Kremenek
23118c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic llvm::sys::Mutex EnableMultithreadingMutex;
23128c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregorstatic bool EnabledMultithreading;
23138c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
23145e4bc590b0ea010e38372d0b4a0aab578a746fe6Benjamin Kramerextern "C" {
23150a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas GregorCXIndex clang_createIndex(int excludeDeclarationsFromPCH,
23160a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor                          int displayDiagnostics) {
231748615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // Disable pretty stack trace functionality, which will otherwise be a very
231848615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  // poor citizen of the world and set up all sorts of signal handlers.
231948615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar  llvm::DisablePrettyStackTrace = true;
232048615ffe41e41e0cc232dfb61289b707ece37ea1Daniel Dunbar
2321c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // We use crash recovery to make some of our APIs more reliable, implicitly
2322c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  // enable it.
2323c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar  llvm::CrashRecoveryContext::Enable();
2324c7df4f344d78fe0d7591be3756712e777b3d2e8dDaniel Dunbar
23258c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  // Enable support for multithreading in LLVM.
23268c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  {
23278c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    llvm::sys::ScopedLock L(EnableMultithreadingMutex);
23288c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    if (!EnabledMultithreading) {
23298c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      llvm::llvm_start_multithreaded();
23308c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor      EnabledMultithreading = true;
23318c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor    }
23328c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor  }
23338c8d5412cddcc1c45beb0353d91d7894db74e585Douglas Gregor
2334a030b7cf5e6aad5889b1b662b6979840bc75f87fDouglas Gregor  CIndexer *CIdxr = new CIndexer();
2335e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  if (excludeDeclarationsFromPCH)
2336e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    CIdxr->setOnlyLocalDecls();
23370a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  if (displayDiagnostics)
23380a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor    CIdxr->setDisplayDiagnostics();
2339e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  return CIdxr;
2340600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2341600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
23429ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeIndex(CXIndex CIdx) {
23432b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (CIdx)
23442b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    delete static_cast<CIndexer *>(CIdx);
23452bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
23462bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff
2347d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenekvoid clang_toggleCrashRecovery(unsigned isEnabled) {
2348d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  if (isEnabled)
2349d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Enable();
2350d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek  else
2351d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek    llvm::CrashRecoveryContext::Disable();
2352d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek}
2353d2427ddf00aacdc288c386f3882e0821ca9bd27bTed Kremenek
23549ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit clang_createTranslationUnit(CXIndex CIdx,
2355a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                              const char *ast_filename) {
23562b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
23572b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor    return 0;
2358f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
23597d1d49d2971b20a97b3c2a301470b9eaaa130137Douglas Gregor  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2360389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOptions FileSystemOpts;
2361389db16c63eec6ecfa9b235155252d8da766e94eArgyrios Kyrtzidis  FileSystemOpts.WorkingDir = CXXIdx->getWorkingDirectory();
23620d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2363d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie  llvm::IntrusiveRefCntPtr<DiagnosticsEngine> Diags;
2364a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *TU = ASTUnit::LoadFromASTFile(ast_filename, Diags, FileSystemOpts,
2365a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  CXXIdx->getOnlyLocalDecls(),
2366a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                  0, 0, true);
2367a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return MakeCXTranslationUnit(TU);
2368600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
2369600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2370b1c031be513705d924038f497279b9b599868ba1Douglas Gregorunsigned clang_defaultEditingTranslationUnitOptions() {
23712a2c50b330e7754499f42173616a36865b5f313bDouglas Gregor  return CXTranslationUnit_PrecompiledPreamble |
2372b5af843a20e237ad1a13ad66a867e200695b8c8eDouglas Gregor         CXTranslationUnit_CacheCompletionResults;
2373b1c031be513705d924038f497279b9b599868ba1Douglas Gregor}
2374b1c031be513705d924038f497279b9b599868ba1Douglas Gregor
23759ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXTranslationUnit
23769ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarclang_createTranslationUnitFromSourceFile(CXIndex CIdx,
23779ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          const char *source_filename,
23789ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          int num_command_line_args,
23792ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                          const char * const *command_line_args,
23804db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor                                          unsigned num_unsaved_files,
2381a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor                                          struct CXUnsavedFile *unsaved_files) {
2382dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor  unsigned Options = CXTranslationUnit_DetailedPreprocessingRecord |
2383ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth                     CXTranslationUnit_NestedMacroExpansions;
23845a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor  return clang_parseTranslationUnit(CIdx, source_filename,
23855a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    command_line_args, num_command_line_args,
23865a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor                                    unsaved_files, num_unsaved_files,
2387dca8ee8b7bc86076916a3a80f553f7a4e98c14afDouglas Gregor                                    Options);
23885a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor}
238919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
239019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbarstruct ParseTranslationUnitInfo {
239119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx;
239219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename;
23932ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char *const *command_line_args;
239419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args;
239519ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
239619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files;
239719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options;
239819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXTranslationUnit result;
239919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar};
2400b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_parseTranslationUnit_Impl(void *UserData) {
240119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo *PTUI =
240219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    static_cast<ParseTranslationUnitInfo*>(UserData);
240319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  CXIndex CIdx = PTUI->CIdx;
240419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  const char *source_filename = PTUI->source_filename;
24052ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor  const char * const *command_line_args = PTUI->command_line_args;
240619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  int num_command_line_args = PTUI->num_command_line_args;
240719ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  struct CXUnsavedFile *unsaved_files = PTUI->unsaved_files;
240819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned num_unsaved_files = PTUI->num_unsaved_files;
240919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  unsigned options = PTUI->options;
241019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  PTUI->result = 0;
24115a43021ac491bf091494167127772a20d9a9bb48Douglas Gregor
24122b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CIdx)
241319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return;
2414f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2415e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff  CIndexer *CXXIdx = static_cast<CIndexer *>(CIdx);
2416e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff
241744c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  bool PrecompilePreamble = options & CXTranslationUnit_PrecompiledPreamble;
2418467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor  // FIXME: Add a flag for modules.
2419467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor  TranslationUnitKind TUKind
2420467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor    = (options & CXTranslationUnit_Incomplete)? TU_Prefix : TU_Complete;
242187c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor  bool CacheCodeCompetionResults
242287c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor    = options & CXTranslationUnit_CacheCompletionResults;
242387c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor
24245352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  // Configure the diagnostics.
24255352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  DiagnosticOptions DiagOpts;
2426d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie  llvm::IntrusiveRefCntPtr<DiagnosticsEngine>
242725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Diags(CompilerInstance::createDiagnostics(DiagOpts, num_command_line_args,
242825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                                command_line_args));
242925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
243025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
2431d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie  llvm::CrashRecoveryContextCleanupRegistrar<DiagnosticsEngine,
2432d6471f7c1921c7802804ce3ff6fe9768310f72b9David Blaikie    llvm::CrashRecoveryContextReleaseRefCleanup<DiagnosticsEngine> >
243325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    DiagCleanup(Diags.getPtr());
243425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
243525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<ASTUnit::RemappedFile> >
243625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
243725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
243825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
243925a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
244025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
2441f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
24424db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
24435f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2444f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    const llvm::MemoryBuffer *Buffer
2445a0a270c0f1c0a4e3482438bdc5f4a7bd3d25f0a6Chris Lattner      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
244625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
244725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
24484db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor  }
2449f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
245025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<const char *> >
245125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args(new std::vector<const char*>());
245225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
245325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this method.
245425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<std::vector<const char*> >
245525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    ArgsCleanup(Args.get());
245625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
245752ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // Since the Clang C library is primarily used by batch tools dealing with
245852ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // (often very broken) source code, where spell-checking can have a
245952ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // significant negative impact on performance (particularly when
246052ddc5df59a26570fbca47d269f82954ae3397d1Douglas Gregor  // precompiled headers are involved), we disable it by default.
2461b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  // Only do this if we haven't found a spell-checking-related argument.
2462b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  bool FoundSpellCheckingArgument = false;
2463b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  for (int I = 0; I != num_command_line_args; ++I) {
2464b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (strcmp(command_line_args[I], "-fno-spell-checking") == 0 ||
2465b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        strcmp(command_line_args[I], "-fspell-checking") == 0) {
2466b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      FoundSpellCheckingArgument = true;
2467b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      break;
2468e56b4baeba5097852e04bc41ca2e0396cf729955Steve Naroff    }
2469b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  }
2470b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  if (!FoundSpellCheckingArgument)
247125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-fno-spell-checking");
2472b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
247325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  Args->insert(Args->end(), command_line_args,
247425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek               command_line_args + num_command_line_args);
2475d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2476c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // The 'source_filename' argument is optional.  If the caller does not
2477c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // specify it then it is assumed that the source file is specified
2478c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // in the actual argument list.
2479c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // Put the source file after command_line_args otherwise if '-x' flag is
2480c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  // present it will be unused.
2481c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis  if (source_filename)
248225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back(source_filename);
2483c842955140adf7b0a7c73c61390bf580c50a182dArgyrios Kyrtzidis
248444c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  // Do we need the detailed preprocessing record?
2485ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth  bool NestedMacroExpansions = false;
248644c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  if (options & CXTranslationUnit_DetailedPreprocessingRecord) {
248725a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-Xclang");
248825a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    Args->push_back("-detailed-preprocessing-record");
2489ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth    NestedMacroExpansions
2490ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth      = (options & CXTranslationUnit_NestedMacroExpansions);
249144c181aec37789f25f6c15543c164416f72e562aDouglas Gregor  }
249244c181aec37789f25f6c15543c164416f72e562aDouglas Gregor
2493026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  unsigned NumErrors = Diags->getClient()->getNumErrors();
2494b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor  llvm::OwningPtr<ASTUnit> Unit(
24954ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek    ASTUnit::LoadFromCommandLine(Args->size() ? &(*Args)[0] : 0
24964ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 /* vector::data() not portable */,
24974ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 Args->size() ? (&(*Args)[0] + Args->size()) :0,
2498b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 Diags,
2499b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getClangResourcesPath(),
2500b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 CXXIdx->getOnlyLocalDecls(),
2501e47be3e9682e82da15059006f43c7f3c021e4fffDouglas Gregor                                 /*CaptureDiagnostics=*/true,
25024ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                                 RemappedFiles->size() ? &(*RemappedFiles)[0]:0,
250325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                 RemappedFiles->size(),
2504299a4a967b02c9f0d0d94ad8560e3ced893f9116Argyrios Kyrtzidis                                 /*RemappedFilesKeepOriginalName=*/true,
2505b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                 PrecompilePreamble,
2506467dc88512b4ba4bb16e274ea3771dc1415d31daDouglas Gregor                                 TUKind,
250799ba202f659e1885fa5ee114f97c97cf6a857491Douglas Gregor                                 CacheCodeCompetionResults,
2508ba7537febdf1bc1cc617e1f1746f2644feba6274Chandler Carruth                                 NestedMacroExpansions));
2509b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor
2510026f6911bb985c800a54446de9f6da8745ae025aArgyrios Kyrtzidis  if (NumErrors != Diags->getClient()->getNumErrors()) {
2511b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    // Make sure to check that 'Unit' is non-NULL.
2512b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    if (CXXIdx->getDisplayDiagnostics() && Unit.get()) {
2513b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      for (ASTUnit::stored_diag_iterator D = Unit->stored_diag_begin(),
2514b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                      DEnd = Unit->stored_diag_end();
2515b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor           D != DEnd; ++D) {
2516b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        CXStoredDiagnostic Diag(*D, Unit->getASTContext().getLangOptions());
2517b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        CXString Msg = clang_formatDiagnostic(&Diag,
2518b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor                                    clang_defaultDiagnosticDisplayOptions());
2519b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        fprintf(stderr, "%s\n", clang_getCString(Msg));
2520b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor        clang_disposeString(Msg);
2521b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      }
2522274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor#ifdef LLVM_ON_WIN32
2523b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // On Windows, force a flush, since there may be multiple copies of
2524b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // stderr and stdout in the file system, all with different buffers
2525b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      // but writing to the same device.
2526b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor      fflush(stderr);
2527b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor#endif
2528b10daedb8fc4c6f70a66dbc1a6eea01684bb4e77Douglas Gregor    }
2529a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor  }
2530d93256e55673a17d18543397ec462416acb13792Douglas Gregor
2531a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  PTUI->result = MakeCXTranslationUnit(Unit.take());
253219ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar}
253319ffd492a31a25fb691098bf79f317e5f3edf177Daniel DunbarCXTranslationUnit clang_parseTranslationUnit(CXIndex CIdx,
253419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             const char *source_filename,
25352ef6944d529c94824f5bf96f65665f5bee30f5a2Douglas Gregor                                         const char * const *command_line_args,
253619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             int num_command_line_args,
25379e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                            struct CXUnsavedFile *unsaved_files,
253819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned num_unsaved_files,
253919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar                                             unsigned options) {
254019ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  ParseTranslationUnitInfo PTUI = { CIdx, source_filename, command_line_args,
25419e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_command_line_args, unsaved_files,
25429e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                    num_unsaved_files, options, 0 };
254319ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  llvm::CrashRecoveryContext CRC;
254419ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
2545bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_parseTranslationUnit_Impl, &PTUI)) {
254660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "libclang: crash detected during parsing: {\n");
254760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'source_filename' : '%s'\n", source_filename);
254860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'command_line_args' : [");
254960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (int i = 0; i != num_command_line_args; ++i) {
255060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
255160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
255260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "'%s'", command_line_args[i]);
255360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
255460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
255560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'unsaved_files' : [");
255660a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    for (unsigned i = 0; i != num_unsaved_files; ++i) {
255760a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      if (i)
255860a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar        fprintf(stderr, ", ");
255960a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar      fprintf(stderr, "('%s', '...', %ld)", unsaved_files[i].Filename,
256060a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar              unsaved_files[i].Length);
256160a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    }
256260a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "],\n");
256360a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "  'options' : %d,\n", options);
256460a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar    fprintf(stderr, "}\n");
256560a4543d42b6b8564a5650345de7beb7e99c60f3Daniel Dunbar
256619ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar    return 0;
25676df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  } else if (getenv("LIBCLANG_RESOURCE_USAGE")) {
25686df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(PTUI.result);
256919ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  }
25706df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
257119ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar  return PTUI.result;
25725b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff}
25735b7d8e254f6c2855b37b5521c0aee0a560dab237Steve Naroff
25741999844e7a18786e61e619e1dc6c789827541863Douglas Gregorunsigned clang_defaultSaveOptions(CXTranslationUnit TU) {
25751999844e7a18786e61e619e1dc6c789827541863Douglas Gregor  return CXSaveTranslationUnit_None;
25761999844e7a18786e61e619e1dc6c789827541863Douglas Gregor}
25771999844e7a18786e61e619e1dc6c789827541863Douglas Gregor
25781999844e7a18786e61e619e1dc6c789827541863Douglas Gregorint clang_saveTranslationUnit(CXTranslationUnit TU, const char *FileName,
25791999844e7a18786e61e619e1dc6c789827541863Douglas Gregor                              unsigned options) {
25807ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor  if (!TU)
258139c411fa229b2a6747b92f945d1702ee674d3470Douglas Gregor    return CXSaveError_InvalidTU;
25827ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor
258339c411fa229b2a6747b92f945d1702ee674d3470Douglas Gregor  CXSaveError result = static_cast<ASTUnit *>(TU->TUData)->Save(FileName);
25846df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  if (getenv("LIBCLANG_RESOURCE_USAGE"))
25856df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(TU);
25866df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  return result;
25877ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor}
258819ffd492a31a25fb691098bf79f317e5f3edf177Daniel Dunbar
25899ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarvoid clang_disposeTranslationUnit(CXTranslationUnit CTUnit) {
2590ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  if (CTUnit) {
2591ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // If the translation unit has been marked as unsafe to free, just discard
2592ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    // it.
2593a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    if (static_cast<ASTUnit *>(CTUnit->TUData)->isUnsafeToFree())
2594ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar      return;
2595ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2596a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete static_cast<ASTUnit *>(CTUnit->TUData);
2597a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    disposeCXStringPool(CTUnit->StringPool);
2598a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    delete CTUnit;
2599ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  }
26002bd6b9f298afb16a2aec035ebd7b29af7c5c3da8Steve Naroff}
26010d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
2602e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregorunsigned clang_defaultReparseOptions(CXTranslationUnit TU) {
2603e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor  return CXReparse_None;
2604e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor}
2605e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor
2606ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarstruct ReparseTranslationUnitInfo {
2607ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU;
2608ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files;
2609ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files;
2610ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options;
2611ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  int result;
2612ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar};
2613593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2614b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbarstatic void clang_reparseTranslationUnit_Impl(void *UserData) {
2615ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo *RTUI =
2616ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    static_cast<ReparseTranslationUnitInfo*>(UserData);
2617ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  CXTranslationUnit TU = RTUI->TU;
2618ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned num_unsaved_files = RTUI->num_unsaved_files;
2619ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  struct CXUnsavedFile *unsaved_files = RTUI->unsaved_files;
2620ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  unsigned options = RTUI->options;
2621ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  (void) options;
2622ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  RTUI->result = 1;
2623ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2624abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  if (!TU)
2625ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return;
2626593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2627a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
2628593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
2629abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
263025a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::OwningPtr<std::vector<ASTUnit::RemappedFile> >
263125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles(new std::vector<ASTUnit::RemappedFile>());
263225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
263325a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  // Recover resources if we crash before exiting this function.
263425a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek  llvm::CrashRecoveryContextCleanupRegistrar<
263525a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    std::vector<ASTUnit::RemappedFile> > RemappedCleanup(RemappedFiles.get());
263625a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek
2637abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  for (unsigned I = 0; I != num_unsaved_files; ++I) {
26385f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    StringRef Data(unsaved_files[I].Contents, unsaved_files[I].Length);
2639abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor    const llvm::MemoryBuffer *Buffer
26401abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor      = llvm::MemoryBuffer::getMemBufferCopy(Data, unsaved_files[I].Filename);
264125a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek    RemappedFiles->push_back(std::make_pair(unsaved_files[I].Filename,
264225a11e1c5fad62dbad25a265e334720157e3fbc1Ted Kremenek                                            Buffer));
2643abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor  }
2644abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor
26454ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek  if (!CXXUnit->Reparse(RemappedFiles->size() ? &(*RemappedFiles)[0] : 0,
26464ee9926671913ea6189ef9840a244d7c4385a7d5Ted Kremenek                        RemappedFiles->size()))
2647593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor    RTUI->result = 0;
2648abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor}
2649593b0c1047f1323ebbda78ae38e96f976241c663Douglas Gregor
2650ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbarint clang_reparseTranslationUnit(CXTranslationUnit TU,
2651ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned num_unsaved_files,
2652ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 struct CXUnsavedFile *unsaved_files,
2653ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                 unsigned options) {
2654ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  ReparseTranslationUnitInfo RTUI = { TU, num_unsaved_files, unsaved_files,
2655ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar                                      options, 0 };
2656ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  llvm::CrashRecoveryContext CRC;
2657ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2658bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  if (!RunSafely(CRC, clang_reparseTranslationUnit_Impl, &RTUI)) {
2659b1fd3458680bc9c8988dee8967e9c0709fef3945Daniel Dunbar    fprintf(stderr, "libclang: crash detected during reparsing\n");
2660a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    static_cast<ASTUnit *>(TU->TUData)->setUnsafeToFree(true);
2661ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar    return 1;
26626df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  } else if (getenv("LIBCLANG_RESOURCE_USAGE"))
26636df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    PrintLibclangResourceUsage(TU);
26641dfb26af4d6aa4f7818e256659a79f1ec2cba784Ted Kremenek
2665ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar  return RTUI.result;
2666ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar}
2667ea94bbc4769697143e717df9b0310f874102b6c1Daniel Dunbar
2668df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor
26699ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getTranslationUnitSpelling(CXTranslationUnit CTUnit) {
26702b37c9e6cabaf3317922af13e6d12061242f04efDouglas Gregor  if (!CTUnit)
2671ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
2672f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2673a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(CTUnit->TUData);
2674ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(CXXUnit->getOriginalSourceFileName(), true);
2675af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff}
26761eb79b58e56b99cf557d5d353586a10c5360364dDaniel Dunbar
26777eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas GregorCXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
2678aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  CXCursor Result = { CXCursor_TranslationUnit, 0, { 0, 0, TU } };
26797eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return Result;
26807eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
26817eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
2682fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2683600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff
2684fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
26851db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor// CXSourceLocation and CXSourceRange Operations.
26861db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor//===----------------------------------------------------------------------===//
26871db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2688b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregorextern "C" {
2689b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceLocation clang_getNullLocation() {
26905352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  CXSourceLocation Result = { { 0, 0 }, 0 };
2691b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  return Result;
2692b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2693b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
2694b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregorunsigned clang_equalLocations(CXSourceLocation loc1, CXSourceLocation loc2) {
269590a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar  return (loc1.ptr_data[0] == loc2.ptr_data[0] &&
269690a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar          loc1.ptr_data[1] == loc2.ptr_data[1] &&
269790a6b9e1e4d4d1995ada044e319d6e722b07a6b4Daniel Dunbar          loc1.int_data == loc2.int_data);
2698b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2699b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
2700b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceLocation clang_getLocation(CXTranslationUnit tu,
2701b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   CXFile file,
2702b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   unsigned line,
2703b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor                                   unsigned column) {
270442748ec5cb2d75fe0dbb3a6db5aee6c11b5dc190Douglas Gregor  if (!tu || !file)
2705b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return clang_getNullLocation();
270642748ec5cb2d75fe0dbb3a6db5aee6c11b5dc190Douglas Gregor
270786a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  bool Logging = ::getenv("LIBCLANG_LOGGING");
2708a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
270986a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  const FileEntry *File = static_cast<const FileEntry *>(file);
2710507097ec40105ed927cb5a744fad98f5875aacacArgyrios Kyrtzidis  SourceLocation SLoc = CXXUnit->getLocation(File, line, column);
271186a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  if (SLoc.isInvalid()) {
271286a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    if (Logging)
271386a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor      llvm::errs() << "clang_getLocation(\"" << File->getName()
271486a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                   << "\", " << line << ", " << column << ") = invalid\n";
271586a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    return clang_getNullLocation();
271686a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  }
271786a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor
271886a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor  if (Logging)
271986a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor    llvm::errs() << "clang_getLocation(\"" << File->getName()
272086a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                 << "\", " << line << ", " << column << ") = "
272186a4d0dd6a630639aab7715323ed068940e650afDouglas Gregor                 << SLoc.getRawEncoding() << "\n";
272283889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
272383889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  return cxloc::translateSourceLocation(CXXUnit->getASTContext(), SLoc);
272483889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall}
272583889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
272683889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid ChisnallCXSourceLocation clang_getLocationForOffset(CXTranslationUnit tu,
272783889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                            CXFile file,
272883889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall                                            unsigned offset) {
272983889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  if (!tu || !file)
273083889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall    return clang_getNullLocation();
273183889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall
2732a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2733507097ec40105ed927cb5a744fad98f5875aacacArgyrios Kyrtzidis  SourceLocation SLoc
2734507097ec40105ed927cb5a744fad98f5875aacacArgyrios Kyrtzidis    = CXXUnit->getLocation(static_cast<const FileEntry *>(file), offset);
273583889a7f1f338e343ef72aeeef9c27f7b62c0f0fDavid Chisnall  if (SLoc.isInvalid()) return clang_getNullLocation();
2736f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
27371a9a0bc472ee4fec72ee8be8b575fb66ca600d1bTed Kremenek  return cxloc::translateSourceLocation(CXXUnit->getASTContext(), SLoc);
2738b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2739b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
27405352ac06d8f6194825bb2a99ffa009b61bafb503Douglas GregorCXSourceRange clang_getNullRange() {
27415352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  CXSourceRange Result = { { 0, 0 }, 0, 0 };
27425352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  return Result;
27435352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor}
2744d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar
2745b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXSourceRange clang_getRange(CXSourceLocation begin, CXSourceLocation end) {
27465352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (begin.ptr_data[0] != end.ptr_data[0] ||
27475352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor      begin.ptr_data[1] != end.ptr_data[1])
27485352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
2749f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2750f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  CXSourceRange Result = { { begin.ptr_data[0], begin.ptr_data[1] },
27515352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                           begin.int_data, end.int_data };
2752b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  return Result;
2753b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2754ab4e83b904d81d8ab1f8c594655822a023cad87dDouglas Gregor
2755ab4e83b904d81d8ab1f8c594655822a023cad87dDouglas Gregorunsigned clang_equalRanges(CXSourceRange range1, CXSourceRange range2)
2756ab4e83b904d81d8ab1f8c594655822a023cad87dDouglas Gregor{
2757ab4e83b904d81d8ab1f8c594655822a023cad87dDouglas Gregor  return range1.ptr_data[0] == range2.ptr_data[0]
2758ab4e83b904d81d8ab1f8c594655822a023cad87dDouglas Gregor      && range1.ptr_data[1] == range2.ptr_data[1]
2759ab4e83b904d81d8ab1f8c594655822a023cad87dDouglas Gregor      && range1.begin_int_data == range2.begin_int_data
2760ab4e83b904d81d8ab1f8c594655822a023cad87dDouglas Gregor      && range1.end_int_data == range2.end_int_data;
2761ab4e83b904d81d8ab1f8c594655822a023cad87dDouglas Gregor}
2762de5db649fd5f9aedde200f443ad73d62517b1c88Argyrios Kyrtzidis
2763de5db649fd5f9aedde200f443ad73d62517b1c88Argyrios Kyrtzidisint clang_Range_isNull(CXSourceRange range) {
2764de5db649fd5f9aedde200f443ad73d62517b1c88Argyrios Kyrtzidis  return clang_equalRanges(range, clang_getNullRange());
2765de5db649fd5f9aedde200f443ad73d62517b1c88Argyrios Kyrtzidis}
2766de5db649fd5f9aedde200f443ad73d62517b1c88Argyrios Kyrtzidis
27679d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek} // end: extern "C"
2768b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
27699d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenekstatic void createNullLocation(CXFile *file, unsigned *line,
27709d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek                               unsigned *column, unsigned *offset) {
27719d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (file)
27729d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *file = 0;
27739d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (line)
27749d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *line = 0;
27759d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (column)
27769d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *column = 0;
27779d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  if (offset)
27789d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek   *offset = 0;
27799d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  return;
27809d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek}
27819d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek
27829d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenekextern "C" {
278320174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruthvoid clang_getExpansionLocation(CXSourceLocation location,
278420174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                CXFile *file,
278520174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                unsigned *line,
278620174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                unsigned *column,
278720174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                unsigned *offset) {
27881db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(location.int_data);
27891db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2790bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  if (!location.ptr_data[0] || Loc.isInvalid()) {
27919d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    createNullLocation(file, line, column, offset);
279246766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor    return;
279346766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor  }
279446766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor
2795bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  const SourceManager &SM =
2796bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar    *static_cast<const SourceManager*>(location.ptr_data[0]);
279720174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth  SourceLocation ExpansionLoc = SM.getExpansionLoc(Loc);
27981db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2799cea731a9cb7de3f473d60e5ea544e25621cebd76Chandler Carruth  // Check that the FileID is invalid on the expansion location.
28009d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  // This can manifest in invalid code.
280120174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth  FileID fileID = SM.getFileID(ExpansionLoc);
2802e23ac65af568ffe611b0990818ac3a57c856a4d8Douglas Gregor  bool Invalid = false;
2803e23ac65af568ffe611b0990818ac3a57c856a4d8Douglas Gregor  const SrcMgr::SLocEntry &sloc = SM.getSLocEntry(fileID, &Invalid);
2804e23ac65af568ffe611b0990818ac3a57c856a4d8Douglas Gregor  if (!sloc.isFile() || Invalid) {
28059d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    createNullLocation(file, line, column, offset);
28069d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    return;
28079d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek  }
28089d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek
28091db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (file)
28109d5a165d301cc9df68631e624322dd2a962f65b3Ted Kremenek    *file = (void *)SM.getFileEntryForSLocEntry(sloc);
28111db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (line)
281220174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    *line = SM.getExpansionLineNumber(ExpansionLoc);
28131db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  if (column)
281420174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    *column = SM.getExpansionColumnNumber(ExpansionLoc);
2815e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor  if (offset)
281620174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    *offset = SM.getDecomposedLoc(ExpansionLoc).second;
281720174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth}
281820174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth
2819e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidisvoid clang_getPresumedLocation(CXSourceLocation location,
2820e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis                               CXString *filename,
2821e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis                               unsigned *line,
2822e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis                               unsigned *column) {
2823e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis  SourceLocation Loc = SourceLocation::getFromRawEncoding(location.int_data);
2824e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis
2825e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis  if (!location.ptr_data[0] || Loc.isInvalid()) {
2826e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis    if (filename)
2827e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis      *filename = createCXString("");
2828e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis    if (line)
2829e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis      *line = 0;
2830e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis    if (column)
2831e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis      *column = 0;
2832e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis  }
2833e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis  else {
2834e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis	const SourceManager &SM =
2835e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis        *static_cast<const SourceManager*>(location.ptr_data[0]);
2836e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis    PresumedLoc PreLoc = SM.getPresumedLoc(Loc);
2837e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis
2838e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis    if (filename)
2839e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis      *filename = createCXString(PreLoc.getFilename());
2840e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis    if (line)
2841e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis      *line = PreLoc.getLine();
2842e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis    if (column)
2843e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis      *column = PreLoc.getColumn();
2844e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis  }
2845e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis}
2846e6be34d8f77312edf9ed38034e52cb4d22c8e1c1Argyrios Kyrtzidis
284720174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruthvoid clang_getInstantiationLocation(CXSourceLocation location,
284820174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                    CXFile *file,
284920174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                    unsigned *line,
285020174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                    unsigned *column,
285120174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                    unsigned *offset) {
285220174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth  // Redirect to new API.
285320174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth  clang_getExpansionLocation(location, file, line, column, offset);
2854e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor}
2855e69517ce61638f12c9abe4605753a45275ac4e37Douglas Gregor
2856a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregorvoid clang_getSpellingLocation(CXSourceLocation location,
2857a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               CXFile *file,
2858a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *line,
2859a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *column,
2860a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor                               unsigned *offset) {
2861a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(location.int_data);
2862a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
28635adc0515aaacb6c4d4f0c9626d86c1e5c177467cArgyrios Kyrtzidis  if (!location.ptr_data[0] || Loc.isInvalid())
28645adc0515aaacb6c4d4f0c9626d86c1e5c177467cArgyrios Kyrtzidis    return createNullLocation(file, line, column, offset);
2865a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2866a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  const SourceManager &SM =
2867a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *static_cast<const SourceManager*>(location.ptr_data[0]);
2868a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  SourceLocation SpellLoc = Loc;
2869a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (SpellLoc.isMacroID()) {
2870a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    SourceLocation SimpleSpellingLoc = SM.getImmediateSpellingLoc(SpellLoc);
2871a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    if (SimpleSpellingLoc.isFileID() &&
2872a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor        SM.getFileEntryForID(SM.getDecomposedLoc(SimpleSpellingLoc).first))
2873a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor      SpellLoc = SimpleSpellingLoc;
2874a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    else
2875402785357ab053dd53f4fdd858b9630a5e0f8badChandler Carruth      SpellLoc = SM.getExpansionLoc(SpellLoc);
2876a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  }
2877a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
2878a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  std::pair<FileID, unsigned> LocInfo = SM.getDecomposedLoc(SpellLoc);
2879a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  FileID FID = LocInfo.first;
2880a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  unsigned FileOffset = LocInfo.second;
2881a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
28825adc0515aaacb6c4d4f0c9626d86c1e5c177467cArgyrios Kyrtzidis  if (FID.isInvalid())
28835adc0515aaacb6c4d4f0c9626d86c1e5c177467cArgyrios Kyrtzidis    return createNullLocation(file, line, column, offset);
28845adc0515aaacb6c4d4f0c9626d86c1e5c177467cArgyrios Kyrtzidis
2885a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (file)
2886a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *file = (void *)SM.getFileEntryForID(FID);
2887a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (line)
2888a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *line = SM.getLineNumber(FID, FileOffset);
2889a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (column)
2890a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *column = SM.getColumnNumber(FID, FileOffset);
2891a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor  if (offset)
2892a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor    *offset = FileOffset;
2893a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor}
2894a9b06d4c246d6c301e3dd1844f5dba669ed9c631Douglas Gregor
28951db19dea8d221f27be46332d668d1e2decb7f1abDouglas GregorCXSourceLocation clang_getRangeStart(CXSourceRange range) {
2896f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  CXSourceLocation Result = { { range.ptr_data[0], range.ptr_data[1] },
28975352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                              range.begin_int_data };
28981db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  return Result;
28991db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor}
29001db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
29011db19dea8d221f27be46332d668d1e2decb7f1abDouglas GregorCXSourceLocation clang_getRangeEnd(CXSourceRange range) {
2902bb4a61a121ba1ee91eb5725881d98249704bb0aaDaniel Dunbar  CXSourceLocation Result = { { range.ptr_data[0], range.ptr_data[1] },
29035352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor                              range.end_int_data };
29041db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor  return Result;
29051db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor}
29061db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor
2907b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor} // end: extern "C"
2908b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor
29091db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor//===----------------------------------------------------------------------===//
2910fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXFile Operations.
2911fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2912fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2913fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
291474844072411bae91d5dbb89955d200cbe1e0a1c8Ted KremenekCXString clang_getFileName(CXFile SFile) {
291598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
2916a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return createCXString((const char*)NULL);
2917f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
291888145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
291974844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek  return createCXString(FEnt->getName());
292088145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
292188145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff
292288145034694ed5267fa6fa5febc54fadc02bd479Steve Narofftime_t clang_getFileTime(CXFile SFile) {
292398258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (!SFile)
292498258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor    return 0;
2925f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
292688145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  FileEntry *FEnt = static_cast<FileEntry *>(SFile);
292788145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff  return FEnt->getModificationTime();
2928ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff}
2929f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2930b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXFile clang_getFile(CXTranslationUnit tu, const char *file_name) {
2931b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!tu)
2932b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor    return 0;
2933f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2934a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2935f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2936b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  FileManager &FMgr = CXXUnit->getFileManager();
293739b49bcaaddb1049234fca9500c0ac02c088e23dChris Lattner  return const_cast<FileEntry *>(FMgr.getFile(file_name));
2938b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor}
2939f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2940dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregorunsigned clang_isFileMultipleIncludeGuarded(CXTranslationUnit tu, CXFile file) {
2941dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  if (!tu || !file)
2942dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor    return 0;
2943dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor
2944dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  ASTUnit *CXXUnit = static_cast<ASTUnit *>(tu->TUData);
2945dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  FileEntry *FEnt = static_cast<FileEntry *>(file);
2946dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor  return CXXUnit->getPreprocessor().getHeaderSearchInfo()
2947dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor                                          .isFileMultipleIncludeGuarded(FEnt);
2948dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor}
2949dd3e5549e3c11e217078938aacf72f042eea5343Douglas Gregor
2950fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
2951fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2952fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2953fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek// CXCursor Operations.
2954fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
2955fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
2956fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekstatic Decl *getDeclFromExpr(Stmt *E) {
2957c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis  if (ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
2958db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return getDeclFromExpr(CE->getSubExpr());
2959db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2960fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (DeclRefExpr *RefExpr = dyn_cast<DeclRefExpr>(E))
2961fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RefExpr->getDecl();
296238f28c1189142429384996409fffbc57f67b5c60Douglas Gregor  if (BlockDeclRefExpr *RefExpr = dyn_cast<BlockDeclRefExpr>(E))
296338f28c1189142429384996409fffbc57f67b5c60Douglas Gregor    return RefExpr->getDecl();
2964fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (MemberExpr *ME = dyn_cast<MemberExpr>(E))
2965fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return ME->getMemberDecl();
2966fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCIvarRefExpr *RE = dyn_cast<ObjCIvarRefExpr>(E))
2967fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return RE->getDecl();
2968db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCPropertyRefExpr *PRE = dyn_cast<ObjCPropertyRefExpr>(E))
296912f78a6741a4cb3d904340f8d3d2714568b50e7aJohn McCall    return PRE->isExplicitProperty() ? PRE->getExplicitProperty() : 0;
2970db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2971fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (CallExpr *CE = dyn_cast<CallExpr>(E))
2972fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return getDeclFromExpr(CE->getCallee());
29735f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  if (CXXConstructExpr *CE = dyn_cast<CXXConstructExpr>(E))
297493798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    if (!CE->isElidable())
297593798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CE->getConstructor();
2976fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  if (ObjCMessageExpr *OME = dyn_cast<ObjCMessageExpr>(E))
2977fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek    return OME->getMethodDecl();
2978f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
2979db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor  if (ObjCProtocolExpr *PE = dyn_cast<ObjCProtocolExpr>(E))
2980db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor    return PE->getProtocol();
2981c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor  if (SubstNonTypeTemplateParmPackExpr *NTTP
2982c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor                              = dyn_cast<SubstNonTypeTemplateParmPackExpr>(E))
2983c7793c73ba8a343de3f2552d984851985a46f159Douglas Gregor    return NTTP->getParameterPack();
298494d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
298594d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    if (isa<NonTypeTemplateParmDecl>(SizeOfPack->getPack()) ||
298694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor        isa<ParmVarDecl>(SizeOfPack->getPack()))
298794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor      return SizeOfPack->getPack();
2988db1314e3ef76f38de07f1b3c7cdc1100a0678931Douglas Gregor
2989fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek  return 0;
2990fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek}
2991ee9405e807d7c447c0143c2bd865b759192e97b3Steve Naroff
2992c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbarstatic SourceLocation getLocationFromExpr(Expr *E) {
2993c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis  if (ImplicitCastExpr *CE = dyn_cast<ImplicitCastExpr>(E))
2994c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis    return getLocationFromExpr(CE->getSubExpr());
2995c2954616fbd11f5a6117236f58420029b773a639Argyrios Kyrtzidis
2996c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCMessageExpr *Msg = dyn_cast<ObjCMessageExpr>(E))
2997c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return /*FIXME:*/Msg->getLeftLoc();
2998c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(E))
2999c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return DRE->getLocation();
300038f28c1189142429384996409fffbc57f67b5c60Douglas Gregor  if (BlockDeclRefExpr *RefExpr = dyn_cast<BlockDeclRefExpr>(E))
300138f28c1189142429384996409fffbc57f67b5c60Douglas Gregor    return RefExpr->getLocation();
3002c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (MemberExpr *Member = dyn_cast<MemberExpr>(E))
3003c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Member->getMemberLoc();
3004c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  if (ObjCIvarRefExpr *Ivar = dyn_cast<ObjCIvarRefExpr>(E))
3005c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar    return Ivar->getLocation();
300694d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor  if (SizeOfPackExpr *SizeOfPack = dyn_cast<SizeOfPackExpr>(E))
300794d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor    return SizeOfPack->getPackLoc();
300894d96291cd041adc5731a2294828a9c20e450b74Douglas Gregor
3009c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar  return E->getLocStart();
3010c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar}
3011c29f4c396019f51e09511e385d6c07c4ea2a1464Daniel Dunbar
3012fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenekextern "C" {
3013f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3014f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenekunsigned clang_visitChildren(CXCursor parent,
3015b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXCursorVisitor visitor,
3016b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor                             CXClientData client_data) {
3017a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CursorVisitor CursorVis(getCursorTU(parent), visitor, client_data,
301804a9eb3f0dfcc42b317057e236f73e9b196a2ff8Douglas Gregor                          false);
3019b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor  return CursorVis.VisitChildren(parent);
3020b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor}
3021b1373d03e4e38620647311bb87e56c0cf7cffeb7Douglas Gregor
30223387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#ifndef __has_feature
30233387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#define __has_feature(x) 0
30243387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
30253387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#if __has_feature(blocks)
30263387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef enum CXChildVisitResult
30273387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall     (^CXCursorVisitorBlock)(CXCursor cursor, CXCursor parent);
30283387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30293387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
30303387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
30313387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
30323387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block(cursor, parent);
30333387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30343387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#else
30353387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// If we are compiled with a compiler that doesn't have native blocks support,
30363387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall// define and call the block manually, so the
30373387c65a094a02b2a94c05111d035a97d3d5c794David Chisnalltypedef struct _CXChildVisitResult
30383387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall{
30393387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	void *isa;
30403387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int flags;
30413387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall	int reserved;
30429e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar	enum CXChildVisitResult(*invoke)(struct _CXChildVisitResult*, CXCursor,
30439e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                         CXCursor);
30443387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall} *CXCursorVisitorBlock;
30453387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30463387c65a094a02b2a94c05111d035a97d3d5c794David Chisnallstatic enum CXChildVisitResult visitWithBlock(CXCursor cursor, CXCursor parent,
30473387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall    CXClientData client_data) {
30483387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  CXCursorVisitorBlock block = (CXCursorVisitorBlock)client_data;
30493387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return block->invoke(block, cursor, parent);
30503387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30513387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall#endif
30523387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30533387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
30549e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbarunsigned clang_visitChildrenWithBlock(CXCursor parent,
30559e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar                                      CXCursorVisitorBlock block) {
30563387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall  return clang_visitChildren(parent, visitWithBlock, block);
30573387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall}
30583387c65a094a02b2a94c05111d035a97d3d5c794David Chisnall
305978205d4bada39d95097e766af9eb30cdd0159461Douglas Gregorstatic CXString getDeclSpelling(Decl *D) {
306078205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D);
3061e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  if (!ND) {
30625f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    if (ObjCPropertyImplDecl *PropImpl =dyn_cast<ObjCPropertyImplDecl>(D))
3063e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
3064e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return createCXString(Property->getIdentifier()->getName());
3065e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
3066ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
3067e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor  }
3068e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
306978205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCMethodDecl *OMD = dyn_cast<ObjCMethodDecl>(ND))
3070ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(OMD->getSelector().getAsString());
3071f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
307278205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor  if (ObjCCategoryImplDecl *CIMP = dyn_cast<ObjCCategoryImplDecl>(ND))
307378205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // No, this isn't the same as the code below. getIdentifier() is non-virtual
307478205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // and returns different names. NamedDecl returns the class name and
307578205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor    // ObjCCategoryImplDecl returns the category name.
3076ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(CIMP->getIdentifier()->getNameStart());
3077f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30780a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  if (isa<UsingDirectiveDecl>(D))
30790a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("");
30800a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor
308150aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::SmallString<1024> S;
308250aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  llvm::raw_svector_ostream os(S);
308350aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  ND->printName(os);
308450aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek
308550aa6acd0b8d40c8956372a69e0a73f0802a5494Ted Kremenek  return createCXString(os.str());
308678205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor}
3087f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
30889ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXString clang_getCursorSpelling(CXCursor C) {
30897eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  if (clang_isTranslationUnit(C.kind))
3090a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return clang_getTranslationUnitSpelling(
3091a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                            static_cast<CXTranslationUnit>(C.data[2]));
30927eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
3093f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  if (clang_isReference(C.kind)) {
3094f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    switch (C.kind) {
3095acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCSuperClassRef: {
30962e331b938b38057e333fab0ba841130ea8467794Douglas Gregor      ObjCInterfaceDecl *Super = getCursorObjCSuperClassRef(C).first;
3097ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Super->getIdentifier()->getNameStart());
3098acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3099acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCClassRef: {
31001adb082a709f7b588f03672999294e061234b2cfDouglas Gregor      ObjCInterfaceDecl *Class = getCursorObjCClassRef(C).first;
3101ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(Class->getIdentifier()->getNameStart());
3102acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
3103acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    case CXCursor_ObjCProtocolRef: {
310478db0cdd49ec24034a5b2a4210fcda03a0919a81Douglas Gregor      ObjCProtocolDecl *OID = getCursorObjCProtocolRef(C).first;
3105f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      assert(OID && "getCursorSpelling(): Missing protocol decl");
3106ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(OID->getIdentifier()->getNameStart());
3107acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    }
31083064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
31093064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = getCursorCXXBaseSpecifier(C);
31103064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return createCXString(B->getType().getAsString());
31113064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
31127d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    case CXCursor_TypeRef: {
31137d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      TypeDecl *Type = getCursorTypeRef(C).first;
31147d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      assert(Type && "Missing type decl");
31157d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3116ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString(getCursorContext(C).getTypeDeclType(Type).
3117ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                              getAsString());
31187d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
31190b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
31200b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      TemplateDecl *Template = getCursorTemplateRef(C).first;
31216931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(Template && "Missing template decl");
31220b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
31230b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString(Template->getNameAsString());
31240b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
31256931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
31266931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
31276931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      NamedDecl *NS = getCursorNamespaceRef(C).first;
31286931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      assert(NS && "Missing namespace decl");
31296931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
31306931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return createCXString(NS->getNameAsString());
31316931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
31327d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3133a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3134a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      FieldDecl *Field = getCursorMemberRef(C).first;
3135a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      assert(Field && "Missing member decl");
3136a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3137a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return createCXString(Field->getNameAsString());
3138a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3139a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
314036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
314136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      LabelStmt *Label = getCursorLabelRef(C).first;
314236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      assert(Label && "Missing label");
314336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
3144ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
314536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
314636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31471f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef: {
31481f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
31491f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Decl *D = Storage.dyn_cast<Decl *>()) {
31501f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        if (NamedDecl *ND = dyn_cast<NamedDecl>(D))
31511f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor          return createCXString(ND->getNameAsString());
31521f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
31531f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      }
31541f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
31551f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString(E->getName().getAsString());
31561f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      OverloadedTemplateStorage *Ovl
31571f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        = Storage.get<OverloadedTemplateStorage*>();
31581f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      if (Ovl->size() == 0)
31591f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor        return createCXString("");
31601f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return createCXString((*Ovl->begin())->getNameAsString());
31611f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    }
31621f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3163acca725aeff400274fab552fc83d87af8a11995dDaniel Dunbar    default:
3164ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek      return createCXString("<not implemented>");
3165f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff    }
3166f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  }
316797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
316897b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
316997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    Decl *D = getDeclFromExpr(getCursorExpr(C));
317097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    if (D)
317178205d4bada39d95097e766af9eb30cdd0159461Douglas Gregor      return getDeclSpelling(D);
3172ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
317397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
317497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
317536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
317636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
317736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (LabelStmt *Label = dyn_cast_or_null<LabelStmt>(S))
3178ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner      return createCXString(Label->getName());
317936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
318036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("");
318136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
318236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
31839b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion)
31849e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth    return createCXString(getCursorMacroExpansion(C)->getName()
31854ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor                                                           ->getNameStart());
31864ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor
3187572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition)
3188572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString(getCursorMacroDefinition(C)->getName()
3189572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor                                                           ->getNameStart());
3190572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3191ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective)
3192ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString(getCursorInclusionDirective(C)->getFileName());
3193ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
319460cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor  if (clang_isDeclaration(C.kind))
319560cbfacd947590f83257a4191566dda92fbbde69Douglas Gregor    return getDeclSpelling(getCursorDecl(C));
3196e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3197ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString("");
3198f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3199f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
3200358559d8d7b458c5f64941842383a16e61f0828dDouglas GregorCXString clang_getCursorDisplayName(CXCursor C) {
3201358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!clang_isDeclaration(C.kind))
3202358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return clang_getCursorSpelling(C);
3203358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3204358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  Decl *D = getCursorDecl(C);
3205358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (!D)
3206358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString("");
3207358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
320830c42404202d2e2512e51efc6066bd614cfdb5a4Douglas Gregor  PrintingPolicy Policy = getCursorContext(C).getPrintingPolicy();
3209358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast<FunctionTemplateDecl>(D))
3210358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    D = FunTmpl->getTemplatedDecl();
3211358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3212358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (FunctionDecl *Function = dyn_cast<FunctionDecl>(D)) {
3213358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3214358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3215358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << Function->getNameAsString();
3216358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->getPrimaryTemplate())
3217358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "<>";
3218358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "(";
3219358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Function->getNumParams(); I != N; ++I) {
3220358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3221358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3222358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << Function->getParamDecl(I)->getType().getAsString(Policy);
3223358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3224358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3225358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (Function->isVariadic()) {
3226358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Function->getNumParams())
3227358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3228358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      OS << "...";
3229358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3230358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ")";
3231358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3232358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3233358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3234358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateDecl *ClassTemplate = dyn_cast<ClassTemplateDecl>(D)) {
3235358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3236358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3237358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ClassTemplate->getNameAsString();
3238358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << "<";
3239358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    TemplateParameterList *Params = ClassTemplate->getTemplateParameters();
3240358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    for (unsigned I = 0, N = Params->size(); I != N; ++I) {
3241358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (I)
3242358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << ", ";
3243358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3244358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      NamedDecl *Param = Params->getParam(I);
3245358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (Param->getIdentifier()) {
3246358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << Param->getIdentifier()->getName();
3247358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        continue;
3248358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      }
3249358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3250358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // There is no parameter name, which makes this tricky. Try to come up
3251358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      // with something useful that isn't too long.
3252358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      if (TemplateTypeParmDecl *TTP = dyn_cast<TemplateTypeParmDecl>(Param))
3253358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << (TTP->wasDeclaredWithTypename()? "typename" : "class");
3254358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else if (NonTypeTemplateParmDecl *NTTP
3255358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                    = dyn_cast<NonTypeTemplateParmDecl>(Param))
3256358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << NTTP->getType().getAsString(Policy);
3257358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      else
3258358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor        OS << "template<...> class";
3259358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    }
3260358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3261358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ">";
3262358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3263358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3264358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3265358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  if (ClassTemplateSpecializationDecl *ClassSpec
3266358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                              = dyn_cast<ClassTemplateSpecializationDecl>(D)) {
3267358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    // If the type was explicitly written, use that.
3268358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    if (TypeSourceInfo *TSInfo = ClassSpec->getTypeAsWritten())
3269358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor      return createCXString(TSInfo->getType().getAsString(Policy));
3270358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3271358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::SmallString<64> Str;
3272358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    llvm::raw_svector_ostream OS(Str);
3273358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << ClassSpec->getNameAsString();
3274358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    OS << TemplateSpecializationType::PrintTemplateArgumentList(
3275910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().data(),
3276910f8008fea79120489a53593fe971b0b8a4a740Douglas Gregor                                      ClassSpec->getTemplateArgs().size(),
3277358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor                                                                Policy);
3278358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor    return createCXString(OS.str());
3279358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  }
3280358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3281358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor  return clang_getCursorSpelling(C);
3282358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor}
3283358559d8d7b458c5f64941842383a16e61f0828dDouglas Gregor
3284e68fff6fc083c6270d835216a3de0b82c6ef0310Ted KremenekCXString clang_getCursorKindSpelling(enum CXCursorKind Kind) {
328589922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  switch (Kind) {
3286e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FunctionDecl:
3287e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FunctionDecl");
3288e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypedefDecl:
3289e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypedefDecl");
3290e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumDecl:
3291e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumDecl");
3292e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_EnumConstantDecl:
3293e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("EnumConstantDecl");
3294e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_StructDecl:
3295e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("StructDecl");
3296e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnionDecl:
3297e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnionDecl");
3298e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ClassDecl:
3299e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ClassDecl");
3300e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_FieldDecl:
3301e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("FieldDecl");
3302e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_VarDecl:
3303e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("VarDecl");
3304e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ParmDecl:
3305e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ParmDecl");
3306e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInterfaceDecl:
3307e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInterfaceDecl");
3308e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryDecl:
3309e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryDecl");
3310e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolDecl:
3311e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolDecl");
3312e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCPropertyDecl:
3313e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCPropertyDecl");
3314e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCIvarDecl:
3315e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCIvarDecl");
3316e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCInstanceMethodDecl:
3317e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCInstanceMethodDecl");
3318e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassMethodDecl:
3319e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassMethodDecl");
3320e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCImplementationDecl:
3321e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCImplementationDecl");
3322e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCCategoryImplDecl:
3323e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCCategoryImplDecl");
33248bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek  case CXCursor_CXXMethod:
33258bd5a69999cfd06b6b5a58fdd04e4f802b2df5a4Ted Kremenek      return createCXString("CXXMethod");
3326e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedDecl:
3327e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedDecl");
3328e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCSuperClassRef:
3329e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCSuperClassRef");
3330e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCProtocolRef:
3331e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCProtocolRef");
3332e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCClassRef:
3333e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCClassRef");
3334e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TypeRef:
3335e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TypeRef");
33360b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor  case CXCursor_TemplateRef:
33370b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return createCXString("TemplateRef");
33386931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceRef:
33396931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceRef");
3340a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor  case CXCursor_MemberRef:
3341a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    return createCXString("MemberRef");
334236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelRef:
334336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return createCXString("LabelRef");
33441f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case CXCursor_OverloadedDeclRef:
33451f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return createCXString("OverloadedDeclRef");
334642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IntegerLiteral:
334742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IntegerLiteral");
334842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_FloatingLiteral:
334942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("FloatingLiteral");
335042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ImaginaryLiteral:
335142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ImaginaryLiteral");
335242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_StringLiteral:
335342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("StringLiteral");
335442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CharacterLiteral:
335542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CharacterLiteral");
335642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ParenExpr:
335742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ParenExpr");
335842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnaryOperator:
335942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnaryOperator");
336042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ArraySubscriptExpr:
336142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ArraySubscriptExpr");
336242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_BinaryOperator:
336342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("BinaryOperator");
336442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundAssignOperator:
336542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundAssignOperator");
336642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ConditionalOperator:
336742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ConditionalOperator");
336842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CStyleCastExpr:
336942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CStyleCastExpr");
337042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundLiteralExpr:
337142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundLiteralExpr");
337242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_InitListExpr:
337342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("InitListExpr");
337442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_AddrLabelExpr:
337542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("AddrLabelExpr");
337642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_StmtExpr:
337742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("StmtExpr");
337842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GenericSelectionExpr:
337942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GenericSelectionExpr");
338042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GNUNullExpr:
338142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GNUNullExpr");
338242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXStaticCastExpr:
338342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXStaticCastExpr");
338442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXDynamicCastExpr:
338542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXDynamicCastExpr");
338642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXReinterpretCastExpr:
338742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXReinterpretCastExpr");
338842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXConstCastExpr:
338942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXConstCastExpr");
339042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXFunctionalCastExpr:
339142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXFunctionalCastExpr");
339242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXTypeidExpr:
339342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXTypeidExpr");
339442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXBoolLiteralExpr:
339542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXBoolLiteralExpr");
339642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXNullPtrLiteralExpr:
339742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXNullPtrLiteralExpr");
339842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXThisExpr:
339942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXThisExpr");
340042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXThrowExpr:
340142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXThrowExpr");
340242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXNewExpr:
340342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXNewExpr");
340442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXDeleteExpr:
340542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXDeleteExpr");
340642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnaryExpr:
340742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnaryExpr");
340842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCStringLiteral:
340942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCStringLiteral");
341042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCEncodeExpr:
341142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCEncodeExpr");
341242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCSelectorExpr:
341342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCSelectorExpr");
341442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCProtocolExpr:
341542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCProtocolExpr");
341642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCBridgedCastExpr:
341742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCBridgedCastExpr");
34181ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek  case CXCursor_BlockExpr:
34191ee6cad59f017601ea54fbb4f62a6e8d69897e3eTed Kremenek      return createCXString("BlockExpr");
342042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_PackExpansionExpr:
342142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("PackExpansionExpr");
342242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SizeOfPackExpr:
342342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SizeOfPackExpr");
342442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_UnexposedExpr:
342542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("UnexposedExpr");
3426e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_DeclRefExpr:
3427e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("DeclRefExpr");
3428e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_MemberRefExpr:
3429e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("MemberRefExpr");
3430e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_CallExpr:
3431e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("CallExpr");
3432e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_ObjCMessageExpr:
3433e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("ObjCMessageExpr");
3434e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_UnexposedStmt:
3435e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("UnexposedStmt");
343642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DeclStmt:
343742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DeclStmt");
343836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  case CXCursor_LabelStmt:
343936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return createCXString("LabelStmt");
344042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CompoundStmt:
344142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CompoundStmt");
344242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CaseStmt:
344342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CaseStmt");
344442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DefaultStmt:
344542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DefaultStmt");
344642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IfStmt:
344742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IfStmt");
344842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SwitchStmt:
344942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SwitchStmt");
345042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_WhileStmt:
345142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("WhileStmt");
345242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_DoStmt:
345342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("DoStmt");
345442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ForStmt:
345542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ForStmt");
345642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_GotoStmt:
345742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("GotoStmt");
345842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_IndirectGotoStmt:
345942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("IndirectGotoStmt");
346042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ContinueStmt:
346142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ContinueStmt");
346242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_BreakStmt:
346342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("BreakStmt");
346442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ReturnStmt:
346542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ReturnStmt");
346642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_AsmStmt:
346742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("AsmStmt");
346842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtTryStmt:
346942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtTryStmt");
347042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtCatchStmt:
347142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtCatchStmt");
347242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtFinallyStmt:
347342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtFinallyStmt");
347442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtThrowStmt:
347542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtThrowStmt");
347642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAtSynchronizedStmt:
347742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAtSynchronizedStmt");
347842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCAutoreleasePoolStmt:
347942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCAutoreleasePoolStmt");
348042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_ObjCForCollectionStmt:
348142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("ObjCForCollectionStmt");
348242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXCatchStmt:
348342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXCatchStmt");
348442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXTryStmt:
348542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXTryStmt");
348642b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_CXXForRangeStmt:
348742b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("CXXForRangeStmt");
348842b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHTryStmt:
348942b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHTryStmt");
349042b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHExceptStmt:
349142b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHExceptStmt");
349242b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_SEHFinallyStmt:
349342b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("SEHFinallyStmt");
349442b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor  case CXCursor_NullStmt:
349542b2984771a7fd1b17c78bbb2c59fed3db2f1960Douglas Gregor      return createCXString("NullStmt");
3496e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_InvalidFile:
3497e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("InvalidFile");
3498292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek  case CXCursor_InvalidCode:
3499292db6401f040795db3ea4e00fc02622d6c3ba1dTed Kremenek    return createCXString("InvalidCode");
3500e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NoDeclFound:
3501e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NoDeclFound");
3502e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_NotImplemented:
3503e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("NotImplemented");
3504e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  case CXCursor_TranslationUnit:
3505e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek      return createCXString("TranslationUnit");
3506e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_UnexposedAttr:
3507e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("UnexposedAttr");
3508e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek  case CXCursor_IBActionAttr:
3509e77f443dbca8cdc23e5aa94a2653367e4a7cbe47Ted Kremenek      return createCXString("attribute(ibaction)");
35109f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_IBOutletAttr:
35119f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor     return createCXString("attribute(iboutlet)");
3512857e918a8a40deb128840308a318bf623d68295fTed Kremenek  case CXCursor_IBOutletCollectionAttr:
3513857e918a8a40deb128840308a318bf623d68295fTed Kremenek      return createCXString("attribute(iboutletcollection)");
35146639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  case CXCursor_CXXFinalAttr:
35156639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      return createCXString("attribute(final)");
35166639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  case CXCursor_CXXOverrideAttr:
35176639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      return createCXString("attribute(override)");
35189f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  case CXCursor_PreprocessingDirective:
35199f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return createCXString("preprocessing directive");
3520572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  case CXCursor_MacroDefinition:
3521572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return createCXString("macro definition");
35229b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  case CXCursor_MacroExpansion:
35239b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth    return createCXString("macro expansion");
3524ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  case CXCursor_InclusionDirective:
3525ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return createCXString("inclusion directive");
35268f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek  case CXCursor_Namespace:
35278f06e0e9fec3ca501e5fb129f413adbfc88e82f8Ted Kremenek    return createCXString("Namespace");
3528a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek  case CXCursor_LinkageSpec:
3529a0536d8dd900bb48ea886bd68d777b03b061c068Ted Kremenek    return createCXString("LinkageSpec");
35303064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek  case CXCursor_CXXBaseSpecifier:
35313064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    return createCXString("C++ base class specifier");
353201829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Constructor:
353301829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConstructor");
353401829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_Destructor:
353501829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXDestructor");
353601829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor  case CXCursor_ConversionFunction:
353701829d3afafdfd355cbe93537bc408aeeed964c6Douglas Gregor    return createCXString("CXXConversion");
3538fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTypeParameter:
3539fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTypeParameter");
3540fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_NonTypeTemplateParameter:
3541fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("NonTypeTemplateParameter");
3542fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_TemplateTemplateParameter:
3543fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("TemplateTemplateParameter");
3544fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor  case CXCursor_FunctionTemplate:
3545fe72e9ceeae6cc8669cd8bb722425300190638eaDouglas Gregor    return createCXString("FunctionTemplate");
354639d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor  case CXCursor_ClassTemplate:
354739d6f07b056c31e1e6b5946165ed4b23e7887f22Douglas Gregor    return createCXString("ClassTemplate");
354874dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor  case CXCursor_ClassTemplatePartialSpecialization:
354974dbe640021d96a8dbb85c592471c04449ade81cDouglas Gregor    return createCXString("ClassTemplatePartialSpecialization");
35506931900f43cea558c6974075256c07728dbfecc6Douglas Gregor  case CXCursor_NamespaceAlias:
35516931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    return createCXString("NamespaceAlias");
35520a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor  case CXCursor_UsingDirective:
35530a35bceb7768fc0be62cb644a4e31d8bfd9fb44aDouglas Gregor    return createCXString("UsingDirective");
35547e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor  case CXCursor_UsingDeclaration:
35557e24256c95afb64b4d5abf201a0f9f0527cb4cf3Douglas Gregor    return createCXString("UsingDeclaration");
3556162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  case CXCursor_TypeAliasDecl:
3557352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("TypeAliasDecl");
3558352697a87bca664356f21a838b162084013625eaDouglas Gregor  case CXCursor_ObjCSynthesizeDecl:
3559352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("ObjCSynthesizeDecl");
3560352697a87bca664356f21a838b162084013625eaDouglas Gregor  case CXCursor_ObjCDynamicDecl:
3561352697a87bca664356f21a838b162084013625eaDouglas Gregor    return createCXString("ObjCDynamicDecl");
35622dfdb948bef51a601e763191e4becfe59880d382Argyrios Kyrtzidis  case CXCursor_CXXAccessSpecifier:
35632dfdb948bef51a601e763191e4becfe59880d382Argyrios Kyrtzidis    return createCXString("CXXAccessSpecifier");
356489922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  }
3565e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3566deb06bd3566e18f677e76bc435d478b033fe328bTed Kremenek  llvm_unreachable("Unhandled CXCursorKind");
3567a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  return createCXString((const char*) 0);
3568600866cc7d6d9ec2e27d4b6d6ec461f6463b5ab6Steve Naroff}
356989922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff
3570064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidisstruct GetCursorData {
3571064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  SourceLocation TokenBeginLoc;
35724b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  bool PointsAtMacroArgExpansion;
3573064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  CXCursor &BestCursor;
3574064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
35754b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  GetCursorData(SourceManager &SM,
35764b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                SourceLocation tokenBegin, CXCursor &outputCursor)
35774b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    : TokenBeginLoc(tokenBegin), BestCursor(outputCursor) {
35784b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    PointsAtMacroArgExpansion = SM.isMacroArgExpansion(tokenBegin);
35794b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  }
3580064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis};
3581064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
35824b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidisstatic enum CXChildVisitResult GetCursorVisitor(CXCursor cursor,
35834b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                                                CXCursor parent,
35844b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis                                                CXClientData client_data) {
3585064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  GetCursorData *Data = static_cast<GetCursorData *>(client_data);
3586064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  CXCursor *BestCursor = &Data->BestCursor;
35874b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis
35884b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // If we point inside a macro argument we should provide info of what the
35894b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // token is so use the actual cursor, don't replace it with a macro expansion
35904b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  // cursor.
35914b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis  if (cursor.kind == CXCursor_MacroExpansion && Data->PointsAtMacroArgExpansion)
35924b43b305342ae2e49d473d0fa6152e5d0c343765Argyrios Kyrtzidis    return CXChildVisit_Recurse;
359365ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis
359465ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis  if (clang_isDeclaration(cursor.kind)) {
359565ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    // Avoid having the implicit methods override the property decls.
359665ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis    if (ObjCMethodDecl *MD = dyn_cast<ObjCMethodDecl>(getCursorDecl(cursor)))
359765ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis      if (MD->isImplicit())
359865ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis        return CXChildVisit_Break;
359965ab90736ede3932b26848e39c64396c47f2941bArgyrios Kyrtzidis  }
3600064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
3601064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  if (clang_isExpression(cursor.kind) &&
3602064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis      clang_isDeclaration(BestCursor->kind)) {
3603064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    Decl *D = getCursorDecl(*BestCursor);
3604064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
3605064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    // Avoid having the cursor of an expression replace the declaration cursor
3606064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    // when the expression source range overlaps the declaration range.
3607064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    // This can happen for C++ constructor expressions whose range generally
3608064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    // include the variable declaration, e.g.:
3609064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    //  MyCXXClass foo; // Make sure pointing at 'foo' returns a VarDecl cursor.
3610064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis    if (D->getLocation().isValid() && Data->TokenBeginLoc.isValid() &&
3611064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis        D->getLocation() == Data->TokenBeginLoc)
3612064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis      return CXChildVisit_Break;
3613064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis  }
3614064c44b8a17d4c426b87ad1c58de81d76a13fca3Argyrios Kyrtzidis
361593798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // If our current best cursor is the construction of a temporary object,
361693798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // don't replace that cursor with a type reference, because we want
361793798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  // clang_getCursor() to point at the constructor.
361893798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor  if (clang_isExpression(BestCursor->kind) &&
361993798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor      isa<CXXTemporaryObjectExpr>(getCursorExpr(*BestCursor)) &&
3620aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      cursor.kind == CXCursor_TypeRef) {
3621aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    // Keep the cursor pointing at CXXTemporaryObjectExpr but also mark it
3622aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    // as having the actual point on the type reference.
3623aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    *BestCursor = getTypeRefedCallExprCursor(*BestCursor);
362493798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor    return CXChildVisit_Recurse;
3625aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis  }
362693798e25d68e2d146cff9bd0355d4b7b1ca765f1Douglas Gregor
362733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  *BestCursor = cursor;
362833e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  return CXChildVisit_Recurse;
362933e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor}
3630e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3631b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas GregorCXCursor clang_getCursor(CXTranslationUnit TU, CXSourceLocation Loc) {
3632b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor  if (!TU)
3633f462989fe8d6f59ab2d7d0fe2b4b96292ce706eaTed Kremenek    return clang_getNullCursor();
3634e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek
3635a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3636bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
3637bdf6062bc10aa3b73b16402b440b8073310acd06Douglas Gregor
3638a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek  SourceLocation SLoc = cxloc::translateSourceLocation(Loc);
3639671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  CXCursor Result = cxcursor::getCursor(TU, SLoc);
3640a629ea42f6bc095190db2f3932b60a0be14f3d34Ted Kremenek
364140749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  bool Logging = getenv("LIBCLANG_LOGGING");
364240749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  if (Logging) {
364340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile SearchFile;
364440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned SearchLine, SearchColumn;
364540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXFile ResultFile;
364640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    unsigned ResultLine, ResultColumn;
36476653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    CXString SearchFileName, ResultFileName, KindSpelling, USR;
36486653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    const char *IsDef = clang_isCursorDefinition(Result)? " (Definition)" : "";
364940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    CXSourceLocation ResultLoc = clang_getCursorLocation(Result);
365040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
365120174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    clang_getExpansionLocation(Loc, &SearchFile, &SearchLine, &SearchColumn, 0);
365220174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth    clang_getExpansionLocation(ResultLoc, &ResultFile, &ResultLine,
365320174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                               &ResultColumn, 0);
365440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    SearchFileName = clang_getFileName(SearchFile);
365540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    ResultFileName = clang_getFileName(ResultFile);
365640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    KindSpelling = clang_getCursorKindSpelling(Result.kind);
36576653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    USR = clang_getCursorUSR(Result);
36586653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    fprintf(stderr, "clang_getCursor(%s:%d:%d) = %s(%s:%d:%d):%s%s\n",
365940749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(SearchFileName), SearchLine, SearchColumn,
366040749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor            clang_getCString(KindSpelling),
36616653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(ResultFileName), ResultLine, ResultColumn,
36626653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor            clang_getCString(USR), IsDef);
366340749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(SearchFileName);
366440749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(ResultFileName);
366540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor    clang_disposeString(KindSpelling);
36666653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    clang_disposeString(USR);
36670aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor
36680aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    CXCursor Definition = clang_getCursorDefinition(Result);
36690aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    if (!clang_equalCursors(Definition, clang_getNullCursor())) {
36700aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXSourceLocation DefinitionLoc = clang_getCursorLocation(Definition);
36710aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionKindSpelling
36720aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor                                = clang_getCursorKindSpelling(Definition.kind);
36730aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXFile DefinitionFile;
36740aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      unsigned DefinitionLine, DefinitionColumn;
367520174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth      clang_getExpansionLocation(DefinitionLoc, &DefinitionFile,
367620174221af145554b76a0b0f5e4eb3ac70d05945Chandler Carruth                                 &DefinitionLine, &DefinitionColumn, 0);
36770aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      CXString DefinitionFileName = clang_getFileName(DefinitionFile);
36780aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      fprintf(stderr, "  -> %s(%s:%d:%d)\n",
36790aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionKindSpelling),
36800aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              clang_getCString(DefinitionFileName),
36810aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor              DefinitionLine, DefinitionColumn);
36820aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionFileName);
36830aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor      clang_disposeString(DefinitionKindSpelling);
36840aefbd8d6a82fe8f70ec92871ea42fa5240a960fDouglas Gregor    }
368540749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor  }
368640749ee585abc84fbb3c8fdbd8aaac062f153062Douglas Gregor
3687e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek  return Result;
368877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
368977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
3690738855554394a6afcf39cc8345fd22c3756b8dd0Ted KremenekCXCursor clang_getNullCursor(void) {
36915bfb8c128c2ac8eb4032afc180cdc400a0f953caDouglas Gregor  return MakeCXCursorInvalid(CXCursor_InvalidFile);
3692738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
3693738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek
3694738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenekunsigned clang_equalCursors(CXCursor X, CXCursor Y) {
3695283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  return X == Y;
3696738855554394a6afcf39cc8345fd22c3756b8dd0Ted Kremenek}
36970d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbar
36989ce5584553054d0cb934940586aca0186e87fa57Douglas Gregorunsigned clang_hashCursor(CXCursor C) {
36999ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  unsigned Index = 0;
37009ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  if (clang_isExpression(C.kind) || clang_isStatement(C.kind))
37019ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor    Index = 1;
37029ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
37039ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor  return llvm::DenseMapInfo<std::pair<unsigned, void*> >::getHashValue(
37049ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor                                        std::make_pair(C.kind, C.data[Index]));
37059ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor}
37069ce5584553054d0cb934940586aca0186e87fa57Douglas Gregor
37079ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isInvalid(enum CXCursorKind K) {
370877128ddd3077fc045751a55bb3226802b15d5510Steve Naroff  return K >= CXCursor_FirstInvalid && K <= CXCursor_LastInvalid;
370977128ddd3077fc045751a55bb3226802b15d5510Steve Naroff}
371077128ddd3077fc045751a55bb3226802b15d5510Steve Naroff
37119ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isDeclaration(enum CXCursorKind K) {
371289922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff  return K >= CXCursor_FirstDecl && K <= CXCursor_LastDecl;
371389922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff}
37142d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff
37159ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbarunsigned clang_isReference(enum CXCursorKind K) {
3716f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff  return K >= CXCursor_FirstRef && K <= CXCursor_LastRef;
3717f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff}
3718f334b4e3eda5a39f041fe13f805dbb53535daa2fSteve Naroff
371997b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isExpression(enum CXCursorKind K) {
372097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstExpr && K <= CXCursor_LastExpr;
372197b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
372297b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
372397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregorunsigned clang_isStatement(enum CXCursorKind K) {
372497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  return K >= CXCursor_FirstStmt && K <= CXCursor_LastStmt;
372597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor}
372697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
37278be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregorunsigned clang_isAttribute(enum CXCursorKind K) {
37288be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor    return K >= CXCursor_FirstAttr && K <= CXCursor_LastAttr;
37298be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor}
37308be80e1e6effd5a333bc70e7f030dc9397d0554eDouglas Gregor
37317eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregorunsigned clang_isTranslationUnit(enum CXCursorKind K) {
37327eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor  return K == CXCursor_TranslationUnit;
37337eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor}
37347eaa8ae8692c5cd3eed8cb334fe5346470522091Douglas Gregor
37359f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregorunsigned clang_isPreprocessing(enum CXCursorKind K) {
37369f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  return K >= CXCursor_FirstPreprocessing && K <= CXCursor_LastPreprocessing;
37379f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor}
37389f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor
3739ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenekunsigned clang_isUnexposed(enum CXCursorKind K) {
3740ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  switch (K) {
3741ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedDecl:
3742ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedExpr:
3743ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedStmt:
3744ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    case CXCursor_UnexposedAttr:
3745ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return true;
3746ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek    default:
3747ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek      return false;
3748ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek  }
3749ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek}
3750ad6eff611a4391f89fd6c458db16993f76e7f5d0Ted Kremenek
37519ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel DunbarCXCursorKind clang_getCursorKind(CXCursor C) {
37529efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff  return C.kind;
37539efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff}
37549efa767be8e9f2dae509d3a0be93ade01bfa1560Steve Naroff
375598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas GregorCXSourceLocation clang_getCursorLocation(CXCursor C) {
375698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  if (clang_isReference(C.kind)) {
3757f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    switch (C.kind) {
3758f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCSuperClassRef: {
3759f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3760f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCSuperClassRef(C);
3761a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3762f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3763f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3764f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
3765f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCProtocolDecl *, SourceLocation> P
3766f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCProtocolRef(C);
3767a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3768f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
3769f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor
3770f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef: {
3771f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      std::pair<ObjCInterfaceDecl *, SourceLocation> P
3772f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor        = getCursorObjCClassRef(C);
3773a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3774f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
37757d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3776f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef: {
37777d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor      std::pair<TypeDecl *, SourceLocation> P = getCursorTypeRef(C);
3778a297de20bde673b1a2ad67fa5eafd37fea4227fcTed Kremenek      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
37797d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor    }
37800b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
37810b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef: {
37820b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      std::pair<TemplateDecl *, SourceLocation> P = getCursorTemplateRef(C);
37830b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
37840b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    }
37850b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
37866931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef: {
37876931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      std::pair<NamedDecl *, SourceLocation> P = getCursorNamespaceRef(C);
37886931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
37896931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    }
37906931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
3791a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef: {
3792a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      std::pair<FieldDecl *, SourceLocation> P = getCursorMemberRef(C);
3793a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return cxloc::translateSourceLocation(P.first->getASTContext(), P.second);
3794a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    }
3795a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
37963064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
37971b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      CXXBaseSpecifier *BaseSpec = getCursorCXXBaseSpecifier(C);
37981b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (!BaseSpec)
37991b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return clang_getNullLocation();
38001b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
38011b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      if (TypeSourceInfo *TSInfo = BaseSpec->getTypeSourceInfo())
38021b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor        return cxloc::translateSourceLocation(getCursorContext(C),
38031b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                            TSInfo->getTypeLoc().getBeginLoc());
38041b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor
38051b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
38061b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor                                        BaseSpec->getSourceRange().getBegin());
38073064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
3808f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
380936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef: {
381036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      std::pair<LabelStmt *, SourceLocation> P = getCursorLabelRef(C);
381136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C), P.second);
381236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    }
381336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
38141f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
38151f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return cxloc::translateSourceLocation(getCursorContext(C),
38161f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                          getCursorOverloadedDeclRef(C).second);
38171f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3818f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    default:
3819f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3820f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor      llvm_unreachable("Missed a reference kind");
3821f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor    }
382298258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor  }
382397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
382497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3825f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    return cxloc::translateSourceLocation(getCursorContext(C),
382697b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor                                   getLocationFromExpr(getCursorExpr(C)));
382797b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
382836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind))
382936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C),
383036897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor                                          getCursorStmt(C)->getLocStart());
383136897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
38329f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  if (C.kind == CXCursor_PreprocessingDirective) {
38339f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    SourceLocation L = cxcursor::getCursorPreprocessingDirective(C).getBegin();
38349f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
38359f1e3ff3b3095967e2b92b57a53524e2d6bb141cDouglas Gregor  }
38364807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
38379b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion) {
38384ae8f298b1ea51b4c2234f9148e2e4349c9bdd23Douglas Gregor    SourceLocation L
38399e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth      = cxcursor::getCursorMacroExpansion(C)->getSourceRange().getBegin();
38404807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
38414807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor  }
3842572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3843572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  if (C.kind == CXCursor_MacroDefinition) {
3844572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    SourceLocation L = cxcursor::getCursorMacroDefinition(C)->getLocation();
3845572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3846572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor  }
3847ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3848ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (C.kind == CXCursor_InclusionDirective) {
3849ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    SourceLocation L
3850ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor      = cxcursor::getCursorInclusionDirective(C)->getSourceRange().getBegin();
3851ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return cxloc::translateSourceLocation(getCursorContext(C), L);
3852ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  }
3853ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
38549a700d277c38d9afaa7cb3fe93a714bfe9b62eecTed Kremenek  if (C.kind < CXCursor_FirstDecl || C.kind > CXCursor_LastDecl)
38555352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullLocation();
385698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor
3857f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  Decl *D = getCursorDecl(C);
3858f46034af49435a4d1a0085a4738343122aeb6521Douglas Gregor  SourceLocation Loc = D->getLocation();
3859007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // FIXME: Multiple variables declared in a single declaration
3860007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // currently lack the information needed to correctly determine their
3861007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // ranges when accounting for the type-specifier.  We use context
3862007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3863007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  // and if so, whether it is the first decl.
3864007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3865007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (!cxcursor::isFirstInDeclGroup(C))
3866007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      Loc = VD->getLocation();
3867007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
3868007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek
38692ca54feee89d7277fb967e3247a64f40ef155a82Douglas Gregor  return cxloc::translateSourceLocation(getCursorContext(C), Loc);
387088145034694ed5267fa6fa5febc54fadc02bd479Steve Naroff}
3871a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor
3872a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor} // end extern "C"
3873a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
3874671436e9e2794c56f3c2e62739d225571493af37Argyrios KyrtzidisCXCursor cxcursor::getCursor(CXTranslationUnit TU, SourceLocation SLoc) {
3875671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  assert(TU);
3876671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3877671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // Guard against an invalid SourceLocation, or we may assert in one
3878671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // of the following calls.
3879671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  if (SLoc.isInvalid())
3880671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    return clang_getNullCursor();
3881671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3882671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
3883671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3884671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // Translate the given source location to make it point at the beginning of
3885671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  // the token under the cursor.
3886671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  SLoc = Lexer::GetBeginningOfToken(SLoc, CXXUnit->getSourceManager(),
3887671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                                    CXXUnit->getASTContext().getLangOptions());
3888671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3889671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  CXCursor Result = MakeCXCursorInvalid(CXCursor_NoDeclFound);
3890671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  if (SLoc.isValid()) {
3891671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    // FIXME: Would be great to have a "hint" cursor, then walk from that
3892671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    // hint cursor upward until we find a cursor whose source range encloses
3893671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    // the region of interest, rather than starting from the translation unit.
3894671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    GetCursorData ResultData(CXXUnit->getSourceManager(), SLoc, Result);
3895671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    CXCursor Parent = clang_getTranslationUnitCursor(TU);
3896671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    CursorVisitor CursorVis(TU, GetCursorVisitor, &ResultData,
3897671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                            /*VisitPreprocessorLast=*/true,
3898671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis                            SourceLocation(SLoc));
3899671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis    CursorVis.VisitChildren(Parent);
3900671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  }
3901671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3902671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis  return Result;
3903671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis}
3904671436e9e2794c56f3c2e62739d225571493af37Argyrios Kyrtzidis
3905a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorstatic SourceRange getRawCursorExtent(CXCursor C) {
3906a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  if (clang_isReference(C.kind)) {
3907a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    switch (C.kind) {
3908a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCSuperClassRef:
3909a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return  getCursorObjCSuperClassRef(C).second;
3910f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3911a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCProtocolRef:
3912a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCProtocolRef(C).second;
3913f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
3914a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_ObjCClassRef:
3915a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorObjCClassRef(C).second;
39167d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
3917a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    case CXCursor_TypeRef:
3918a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      return getCursorTypeRef(C).second;
39190b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
39200b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
39210b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return getCursorTemplateRef(C).second;
39220b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
39236931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
39246931900f43cea558c6974075256c07728dbfecc6Douglas Gregor      return getCursorNamespaceRef(C).second;
3925a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
3926a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
3927a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor      return getCursorMemberRef(C).second;
3928a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
39293064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier:
39301b0f7af64113b63253ced088a2bc64eb98e6f388Douglas Gregor      return getCursorCXXBaseSpecifier(C)->getSourceRange();
3931f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
393236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
393336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return getCursorLabelRef(C).second;
393436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
39351f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
39361f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return getCursorOverloadedDeclRef(C).second;
39371f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
3938a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    default:
3939a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      // FIXME: Need a way to enumerate all non-reference cases.
3940a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor      llvm_unreachable("Missed a reference kind");
3941a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor    }
3942a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor  }
394397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
394497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind))
3945a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorExpr(C)->getSourceRange();
394633e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor
394733e9abd21083a0191a7676a04b497006d2da184dDouglas Gregor  if (clang_isStatement(C.kind))
3948a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return getCursorStmt(C)->getSourceRange();
3949f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
39506639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis  if (clang_isAttribute(C.kind))
39516639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis    return getCursorAttr(C)->getRange();
39526639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis
3953a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (C.kind == CXCursor_PreprocessingDirective)
3954a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor    return cxcursor::getCursorPreprocessingDirective(C);
39554807231938d8aff28de09f78f301f9ba5845e5e4Douglas Gregor
3956ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_MacroExpansion) {
3957ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
3958ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorMacroExpansion(C)->getSourceRange();
3959ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
3960ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
3961572feb2a190b5e8b04fb06c4ac50ee0f61e93ff0Douglas Gregor
3962ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_MacroDefinition) {
3963ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
3964ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorMacroDefinition(C)->getSourceRange();
3965ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
3966ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
3967ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3968ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (C.kind == CXCursor_InclusionDirective) {
3969ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    ASTUnit *TU = getCursorASTUnit(C);
3970ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    SourceRange Range = cxcursor::getCursorInclusionDirective(C)->getSourceRange();
3971ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return TU->mapRangeFromPreamble(Range);
3972ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
3973ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
3974007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
3975007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    Decl *D = cxcursor::getCursorDecl(C);
3976007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    SourceRange R = D->getSourceRange();
3977007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // FIXME: Multiple variables declared in a single declaration
3978007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // currently lack the information needed to correctly determine their
3979007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // ranges when accounting for the type-specifier.  We use context
3980007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
3981007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    // and if so, whether it is the first decl.
3982007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
3983007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek      if (!cxcursor::isFirstInDeclGroup(C))
3984007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek        R.setBegin(VD->getLocation());
3985007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    }
3986007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek    return R;
3987007a7c9d8dcdb2e9cd94b6075108bfc4c90e6ccdTed Kremenek  }
39886653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return SourceRange();
39896653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
39906653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
39916653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// \brief Retrieves the "raw" cursor extent, which is then extended to include
39926653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor/// the decl-specifier-seq for declarations.
39936653798ff5ce6deb58112777e21307ccc453133dDouglas Gregorstatic SourceRange getFullCursorExtent(CXCursor C, SourceManager &SrcMgr) {
39946653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  if (C.kind >= CXCursor_FirstDecl && C.kind <= CXCursor_LastDecl) {
39956653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    Decl *D = cxcursor::getCursorDecl(C);
39966653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    SourceRange R = D->getSourceRange();
39972494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
39982494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // Adjust the start of the location for declarations preceded by
39992494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // declaration specifiers.
40002494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
40016653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
40022494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
40032494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
40042494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
40052494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
40062494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
40072494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    }
40086653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
40092494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && R.getBegin().isValid() &&
40102494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, R.getBegin()))
40112494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      R.setBegin(StartLoc);
40122494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
40132494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // FIXME: Multiple variables declared in a single declaration
40142494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // currently lack the information needed to correctly determine their
40152494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // ranges when accounting for the type-specifier.  We use context
40162494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // stored in the CXCursor to determine if the VarDecl is in a DeclGroup,
40172494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    // and if so, whether it is the first decl.
40182494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (VarDecl *VD = dyn_cast<VarDecl>(D)) {
40192494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (!cxcursor::isFirstInDeclGroup(C))
40202494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        R.setBegin(VD->getLocation());
40216653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    }
40226653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
40236653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor    return R;
40246653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  }
40256653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor
40266653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor  return getRawCursorExtent(C);
40276653798ff5ce6deb58112777e21307ccc453133dDouglas Gregor}
4028a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
4029a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregorextern "C" {
4030a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor
4031a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas GregorCXSourceRange clang_getCursorExtent(CXCursor C) {
4032a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  SourceRange R = getRawCursorExtent(C);
4033a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  if (R.isInvalid())
40345352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4035f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4036a8e5c5bdbe387b2552c1c23b828f54abcf085a40Douglas Gregor  return cxloc::translateSourceRange(getCursorContext(C), R);
4037a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor}
4038c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
4039c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas GregorCXCursor clang_getCursorReferenced(CXCursor C) {
4040b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
4041b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
4042f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4043a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit tu = getCursorTU(C);
40441f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (clang_isDeclaration(C.kind)) {
40451f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getCursorDecl(C);
40461f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4047a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Using, D->getLocation(), tu);
40481f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
4049a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Classes, D->getLocation(), tu);
40501f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (ObjCForwardProtocolDecl *Protocols
40511f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                                        = dyn_cast<ObjCForwardProtocolDecl>(D))
4052a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Protocols, D->getLocation(), tu);
40535f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    if (ObjCPropertyImplDecl *PropImpl =dyn_cast<ObjCPropertyImplDecl>(D))
4054e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor      if (ObjCPropertyDecl *Property = PropImpl->getPropertyDecl())
4055e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor        return MakeCXCursor(Property, tu);
4056e3c60a7ce9e0f42c7ca2344b33203266aceca1dbDouglas Gregor
4057c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return C;
40581f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
40591f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
406097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isExpression(C.kind)) {
40611f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Expr *E = getCursorExpr(C);
40621f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    Decl *D = getDeclFromExpr(E);
4063aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    if (D) {
4064aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      CXCursor declCursor = MakeCXCursor(D, tu);
4065aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      declCursor = getSelectorIdentifierCursor(getSelectorIdentifierIndex(C),
4066aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis                                               declCursor);
4067aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis      return declCursor;
4068aed123ec3cc37e457fe20a6158fdadf8849ad916Argyrios Kyrtzidis    }
40691f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
40701f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    if (OverloadExpr *Ovl = dyn_cast_or_null<OverloadExpr>(E))
4071a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCursorOverloadedDeclRef(Ovl, tu);
40721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
407397b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor    return clang_getNullCursor();
407497b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  }
407597b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor
407636897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  if (clang_isStatement(C.kind)) {
407736897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    Stmt *S = getCursorStmt(C);
407836897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    if (GotoStmt *Goto = dyn_cast_or_null<GotoStmt>(S))
407937c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek      if (LabelDecl *label = Goto->getLabel())
408037c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        if (LabelStmt *labelS = label->getStmt())
408137c2e9664316b013b9a86f841f143f19ffbc0a02Ted Kremenek        return MakeCXCursor(labelS, getCursorDecl(C), tu);
408236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
408336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    return clang_getNullCursor();
408436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor  }
408536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
40869b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion) {
40879e5bb85ac899eeab7c21b5ff9030c3da6ff4837bChandler Carruth    if (MacroDefinition *Def = getCursorMacroExpansion(C)->getDefinition())
4088a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeMacroDefinitionCursor(Def, tu);
4089bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor  }
4090bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
4091c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  if (!clang_isReference(C.kind))
4092c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    return clang_getNullCursor();
4093f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4094c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  switch (C.kind) {
4095c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    case CXCursor_ObjCSuperClassRef:
4096a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCSuperClassRef(C).first, tu);
4097f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4098f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCProtocolRef: {
4099a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCProtocolRef(C).first, tu);
4100f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4101f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_ObjCClassRef:
4102a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorObjCClassRef(C).first, tu );
41037d0d40e58807f73e06ff5eb637a48e9f978b0e2aDouglas Gregor
4104f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek    case CXCursor_TypeRef:
4105a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTypeRef(C).first, tu );
41060b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
41070b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor    case CXCursor_TemplateRef:
4108a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorTemplateRef(C).first, tu );
41090b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor
41106931900f43cea558c6974075256c07728dbfecc6Douglas Gregor    case CXCursor_NamespaceRef:
4111a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorNamespaceRef(C).first, tu );
41126931900f43cea558c6974075256c07728dbfecc6Douglas Gregor
4113a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor    case CXCursor_MemberRef:
4114a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(getCursorMemberRef(C).first, tu );
4115a67e03fdf1ae8a1f92463a307d0b6281f1161f40Douglas Gregor
41163064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    case CXCursor_CXXBaseSpecifier: {
41173064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      CXXBaseSpecifier *B = cxcursor::getCursorCXXBaseSpecifier(C);
41183064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek      return clang_getTypeDeclaration(cxtype::MakeCXType(B->getType(),
4119a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                                         tu ));
41203064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek    }
4121f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
412236897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor    case CXCursor_LabelRef:
412336897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // FIXME: We end up faking the "parent" declaration here because we
412436897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      // don't want to make CXCursor larger.
412536897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor      return MakeCXCursor(getCursorLabelRef(C).first,
4126a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek               static_cast<ASTUnit*>(tu->TUData)->getASTContext()
4127a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          .getTranslationUnitDecl(),
4128a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          tu);
412936897b05ca2886e287f01802614bc10cbadcec22Douglas Gregor
41301f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    case CXCursor_OverloadedDeclRef:
41311f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor      return C;
41321f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4133c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    default:
4134c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      // We would prefer to enumerate all non-reference cursor kinds here.
4135c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      llvm_unreachable("Unhandled reference cursor kind");
4136c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor      break;
4137c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor    }
4138c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  }
4139f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4140c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor  return clang_getNullCursor();
4141c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor}
4142c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor
4143b699866820102a69d83d6ac6941985c5ef4e8c40Douglas GregorCXCursor clang_getCursorDefinition(CXCursor C) {
4144b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor  if (clang_isInvalid(C.kind))
4145b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return clang_getNullCursor();
4146f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4147a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(C);
4148f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4149b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  bool WasReference = false;
415097b9872d5775446cb8aca1380e437649fe848d91Douglas Gregor  if (clang_isReference(C.kind) || clang_isExpression(C.kind)) {
4151b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    C = clang_getCursorReferenced(C);
4152b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    WasReference = true;
4153b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4154b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
41559b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth  if (C.kind == CXCursor_MacroExpansion)
4156bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor    return clang_getCursorReferenced(C);
4157bf7efa2742dc94363d3561d284c9d634fc5a780fDouglas Gregor
4158b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4159b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4160b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4161b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  Decl *D = getCursorDecl(C);
4162b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!D)
4163b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4164f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4165b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  switch (D->getKind()) {
4166b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't really separate the notions of
4167b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // declaration and definition.
4168b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Namespace:
4169b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Typedef:
4170162e1c1b487352434552147967c3dd296ebee2f7Richard Smith  case Decl::TypeAlias:
41713e4c6c4c79a03f5cb0c4671d7c282d623c6dc35eRichard Smith  case Decl::TypeAliasTemplate:
4172b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTypeParm:
4173b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::EnumConstant:
4174b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Field:
4175d98114647e16796a976b04af79975b4f0eacf22bBenjamin Kramer  case Decl::IndirectField:
4176b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCIvar:
4177b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCAtDefsField:
4178b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ImplicitParam:
4179b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ParmVar:
4180b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NonTypeTemplateParm:
4181b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TemplateTemplateParm:
4182b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategoryImpl:
4183b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCImplementation:
41846206d53f67613958ae1b023aba337ebb46f11a8bAbramo Bagnara  case Decl::AccessSpec:
4185b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::LinkageSpec:
4186b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCPropertyImpl:
4187b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FileScopeAsm:
4188b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::StaticAssert:
4189b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Block:
4190ad8dcf4a9df0e24051dc31bf9e6f3cd138a34298Chris Lattner  case Decl::Label:  // FIXME: Is this right??
4191af0f4d0b2e38c810effc8b024ad2fb6604eec5d3Francois Pichet  case Decl::ClassScopeFunctionSpecialization:
4192b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return C;
4193b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4194b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds that don't make any sense here, but are
4195b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // nonetheless harmless.
4196b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::TranslationUnit:
4197b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
4198b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4199b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  // Declaration kinds for which the definition is not resolvable.
4200b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingTypename:
4201b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UnresolvedUsingValue:
4202b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    break;
4203b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4204b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingDirective:
4205b2cd48756119f4d8d2a865b4b3e0e8efd02e26a0Douglas Gregor    return MakeCXCursor(cast<UsingDirectiveDecl>(D)->getNominatedNamespace(),
4206a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                        TU);
4207b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4208b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::NamespaceAlias:
4209a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<NamespaceAliasDecl>(D)->getNamespace(), TU);
4210b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4211b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Enum:
4212b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Record:
4213b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXRecord:
4214b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplateSpecialization:
4215b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplatePartialSpecialization:
4216952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor    if (TagDecl *Def = cast<TagDecl>(D)->getDefinition())
4217a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
4218b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4219b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4220b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Function:
4221b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXMethod:
4222b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConstructor:
4223b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXDestructor:
4224b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::CXXConversion: {
4225b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
4226b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionDecl>(D)->getBody(Def))
4227a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(const_cast<FunctionDecl *>(Def), TU);
4228b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4229b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4230b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4231b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Var: {
423231310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    // Ask the variable if it has a definition.
423331310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    if (VarDecl *Def = cast<VarDecl>(D)->getDefinition())
4234a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def, TU);
423531310a21fb2a9f13950f864f681c86080b05d5b2Sebastian Redl    return clang_getNullCursor();
4236b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4237f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4238b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FunctionTemplate: {
4239b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    const FunctionDecl *Def = 0;
4240b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (cast<FunctionTemplateDecl>(D)->getTemplatedDecl()->getBody(Def))
4241a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Def->getDescribedFunctionTemplate(), TU);
4242b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4243b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4244f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4245b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ClassTemplate: {
4246b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (RecordDecl *Def = cast<ClassTemplateDecl>(D)->getTemplatedDecl()
4247952b017601f9c82b51119c3a1600f1312a833db9Douglas Gregor                                                            ->getDefinition())
42480b36e614aa19df72885d5e0de996f7fbb9874ec3Douglas Gregor      return MakeCXCursor(cast<CXXRecordDecl>(Def)->getDescribedClassTemplate(),
4249a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                          TU);
4250b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4251b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4252b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
42531f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::Using:
42541f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<UsingDecl>(D),
4255a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
4256b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4257b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::UsingShadow:
4258b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getCursorDefinition(
4259f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek                       MakeCXCursor(cast<UsingShadowDecl>(D)->getTargetDecl(),
4260a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                    TU));
4261b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4262b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCMethod: {
4263b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    ObjCMethodDecl *Method = cast<ObjCMethodDecl>(D);
4264b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (Method->isThisDeclarationADefinition())
4265b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
4266b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4267b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // Dig out the method definition in the associated
4268b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // @implementation, if we have it.
4269b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: The ASTs should make finding the definition easier.
4270b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4271b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                       = dyn_cast<ObjCInterfaceDecl>(Method->getDeclContext()))
4272b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (ObjCImplementationDecl *ClassImpl = Class->getImplementation())
4273b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        if (ObjCMethodDecl *Def = ClassImpl->getMethod(Method->getSelector(),
4274b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                                                  Method->isInstanceMethod()))
4275b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          if (Def->isThisDeclarationADefinition())
4276a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek            return MakeCXCursor(Def, TU);
4277b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4278b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4279b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4280b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4281b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCategory:
4282b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCCategoryImplDecl *Impl
4283b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                               = cast<ObjCCategoryDecl>(D)->getImplementation())
4284a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4285b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4286b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4287b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProtocol:
4288b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (!cast<ObjCProtocolDecl>(D)->isForwardDecl())
4289b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      return C;
4290b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4291b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4292b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCInterface:
4293b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // There are two notions of a "definition" for an Objective-C
4294b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // class: the interface and its implementation. When we resolved a
4295b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // reference to an Objective-C class, produce the @interface as
4296b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // the definition; when we were provided with the interface,
4297b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // produce the @implementation as the definition.
4298b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (WasReference) {
4299b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!cast<ObjCInterfaceDecl>(D)->isForwardDecl())
4300b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor        return C;
4301b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    } else if (ObjCImplementationDecl *Impl
4302b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor                              = cast<ObjCInterfaceDecl>(D)->getImplementation())
4303a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(Impl, TU);
4304b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4305f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4306b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCProperty:
4307b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // FIXME: We don't really know where to find the
4308b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    // ObjCPropertyImplDecls that implement this property.
4309b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4310b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4311b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::ObjCCompatibleAlias:
4312b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (ObjCInterfaceDecl *Class
4313b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor          = cast<ObjCCompatibleAliasDecl>(D)->getClassInterface())
4314b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor      if (!Class->isForwardDecl())
4315a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek        return MakeCXCursor(Class, TU);
4316f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4317b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4318b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
43191f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::ObjCForwardProtocol:
43201f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return MakeCursorOverloadedDeclRef(cast<ObjCForwardProtocolDecl>(D),
4321a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       D->getLocation(), TU);
4322b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
43231f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  case Decl::ObjCClass:
43249e1ebdd97ba0c593900cb7f103324c43471e96eeDaniel Dunbar    return MakeCursorOverloadedDeclRef(cast<ObjCClassDecl>(D), D->getLocation(),
4325a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                                       TU);
4326b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4327b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::Friend:
4328b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendDecl>(D)->getFriendDecl())
4329a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4330b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4331b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4332b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  case Decl::FriendTemplate:
4333b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    if (NamedDecl *Friend = cast<FriendTemplateDecl>(D)->getFriendDecl())
4334a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return clang_getCursorDefinition(MakeCXCursor(Friend, TU));
4335b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return clang_getNullCursor();
4336b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  }
4337b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4338b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getNullCursor();
4339b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4340b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4341b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregorunsigned clang_isCursorDefinition(CXCursor C) {
4342b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  if (!clang_isDeclaration(C.kind))
4343b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor    return 0;
4344b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
4345b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor  return clang_getCursorDefinition(C) == C;
4346b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor}
4347b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor
43481a9d0503b67a499797141af0fd6d315d5045f0eaDouglas GregorCXCursor clang_getCanonicalCursor(CXCursor C) {
43491a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  if (!clang_isDeclaration(C.kind))
43501a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return C;
43511a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
4352e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis  if (Decl *D = getCursorDecl(C)) {
4353debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis    if (ObjCCategoryImplDecl *CatImplD = dyn_cast<ObjCCategoryImplDecl>(D))
4354debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis      if (ObjCCategoryDecl *CatD = CatImplD->getCategoryDecl())
4355debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis        return MakeCXCursor(CatD, getCursorTU(C));
4356debb00f9ce1dd0f855d2b4fff3372b2ceeb20735Argyrios Kyrtzidis
4357e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis    if (ObjCImplDecl *ImplD = dyn_cast<ObjCImplDecl>(D))
4358e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis      if (ObjCInterfaceDecl *IFD = ImplD->getClassInterface())
4359e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis        return MakeCXCursor(IFD, getCursorTU(C));
4360e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis
43611a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor    return MakeCXCursor(D->getCanonicalDecl(), getCursorTU(C));
4362e2f854ddd365e6837cef3e1a1b7621b32200fc71Argyrios Kyrtzidis  }
43631a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
43641a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor  return C;
43651a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor}
43661a9d0503b67a499797141af0fd6d315d5045f0eaDouglas Gregor
43671f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregorunsigned clang_getNumOverloadedDecls(CXCursor C) {
43687c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (C.kind != CXCursor_OverloadedDeclRef)
43691f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return 0;
43701f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43711f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(C).first;
43721f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
43731f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return E->getNumDecls();
43741f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43751f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
43761f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
43771f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return S->size();
43781f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43791f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
43801f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D))
4381826faa22bae112e01293a58534a40711043cce65Argyrios Kyrtzidis    return Using->shadow_size();
438295ed7784a335aca53b0c6e952cf31a4cfb633360Fariborz Jahanian  if (isa<ObjCClassDecl>(D))
438395ed7784a335aca53b0c6e952cf31a4cfb633360Fariborz Jahanian    return 1;
43841f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCForwardProtocolDecl *Protocols =dyn_cast<ObjCForwardProtocolDecl>(D))
43851f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return Protocols->protocol_size();
43861f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43871f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return 0;
43881f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
43891f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43901f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas GregorCXCursor clang_getOverloadedDecl(CXCursor cursor, unsigned index) {
43917c432dd959609a3689c2e4406450c092e6d76d6dDouglas Gregor  if (cursor.kind != CXCursor_OverloadedDeclRef)
43921f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
43931f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
43941f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (index >= clang_getNumOverloadedDecls(cursor))
43951f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    return clang_getNullCursor();
43961f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
4397a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  CXTranslationUnit TU = getCursorTU(cursor);
43981f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  OverloadedDeclRefStorage Storage = getCursorOverloadedDeclRef(cursor).first;
43991f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadExpr *E = Storage.dyn_cast<OverloadExpr *>())
4400a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(E->decls_begin()[index], TU);
44011f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
44021f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (OverloadedTemplateStorage *S
44031f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor                              = Storage.dyn_cast<OverloadedTemplateStorage*>())
4404a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(S->begin()[index], TU);
44051f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
44061f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  Decl *D = Storage.get<Decl*>();
44071f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (UsingDecl *Using = dyn_cast<UsingDecl>(D)) {
44081f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    // FIXME: This is, unfortunately, linear time.
44091f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    UsingDecl::shadow_iterator Pos = Using->shadow_begin();
44101f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor    std::advance(Pos, index);
4411a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(cast<UsingShadowDecl>(*Pos)->getTargetDecl(), TU);
44121f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  }
44131f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCClassDecl *Classes = dyn_cast<ObjCClassDecl>(D))
441495ed7784a335aca53b0c6e952cf31a4cfb633360Fariborz Jahanian    return MakeCXCursor(Classes->getForwardInterfaceDecl(), TU);
44151f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  if (ObjCForwardProtocolDecl *Protocols = dyn_cast<ObjCForwardProtocolDecl>(D))
4416a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return MakeCXCursor(Protocols->protocol_begin()[index], TU);
44171f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
44181f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor  return clang_getNullCursor();
44191f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor}
44201f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor
44210d7dd222381390731151455ff06358b93d04d6d1Daniel Dunbarvoid clang_getDefinitionSpellingAndExtent(CXCursor C,
44224ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **startBuf,
44234ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          const char **endBuf,
44244ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startLine,
44254ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *startColumn,
44264ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff                                          unsigned *endLine,
44279ebfa31222a704690e9b6b30b7a135fe44364bc2Daniel Dunbar                                          unsigned *endColumn) {
4428283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  assert(getCursorDecl(C) && "CXCursor has null decl");
4429283cae37b03047c14ef918503bc46b08405c3b69Douglas Gregor  NamedDecl *ND = static_cast<NamedDecl *>(getCursorDecl(C));
44304ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  FunctionDecl *FD = dyn_cast<FunctionDecl>(ND);
44314ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  CompoundStmt *Body = dyn_cast<CompoundStmt>(FD->getBody());
4432f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
44334ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  SourceManager &SM = FD->getASTContext().getSourceManager();
44344ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startBuf = SM.getCharacterData(Body->getLBracLoc());
44354ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endBuf = SM.getCharacterData(Body->getRBracLoc());
44364ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startLine = SM.getSpellingLineNumber(Body->getLBracLoc());
44374ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *startColumn = SM.getSpellingColumnNumber(Body->getLBracLoc());
44384ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endLine = SM.getSpellingLineNumber(Body->getRBracLoc());
44394ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff  *endColumn = SM.getSpellingColumnNumber(Body->getRBracLoc());
44404ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff}
4441f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4442430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4443430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas GregorCXSourceRange clang_getCursorReferenceNameRange(CXCursor C, unsigned NameFlags,
4444430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                                                unsigned PieceIndex) {
4445430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  RefNamePieces Pieces;
4446430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4447430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  switch (C.kind) {
4448430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_MemberRefExpr:
4449430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (MemberExpr *E = dyn_cast<MemberExpr>(getCursorExpr(C)))
4450430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Pieces = buildPieces(NameFlags, true, E->getMemberNameInfo(),
4451430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getQualifierLoc().getSourceRange());
4452430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4453430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4454430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_DeclRefExpr:
4455430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (DeclRefExpr *E = dyn_cast<DeclRefExpr>(getCursorExpr(C)))
4456430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Pieces = buildPieces(NameFlags, false, E->getNameInfo(),
4457430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getQualifierLoc().getSourceRange(),
4458430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                           E->getExplicitTemplateArgsOpt());
4459430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4460430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4461430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  case CXCursor_CallExpr:
4462430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (CXXOperatorCallExpr *OCE =
4463430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        dyn_cast<CXXOperatorCallExpr>(getCursorExpr(C))) {
4464430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      Expr *Callee = OCE->getCallee();
4465430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (ImplicitCastExpr *ICE = dyn_cast<ImplicitCastExpr>(Callee))
4466430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        Callee = ICE->getSubExpr();
4467430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4468430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (DeclRefExpr *DRE = dyn_cast<DeclRefExpr>(Callee))
4469430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        Pieces = buildPieces(NameFlags, false, DRE->getNameInfo(),
4470430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor                             DRE->getQualifierLoc().getSourceRange());
4471430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    }
4472430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4473430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4474430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  default:
4475430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    break;
4476430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  }
4477430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4478430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  if (Pieces.empty()) {
4479430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor    if (PieceIndex == 0)
4480430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      return clang_getCursorExtent(C);
4481430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  } else if (PieceIndex < Pieces.size()) {
4482430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      SourceRange R = Pieces[PieceIndex];
4483430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor      if (R.isValid())
4484430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor        return cxloc::translateSourceRange(getCursorContext(C), R);
4485430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  }
4486430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
4487430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor  return clang_getNullRange();
4488430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor}
4489430d7a1a621a126a9ffe442ad8987ba02b46dae9Douglas Gregor
44900a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregorvoid clang_enableStackTraces(void) {
44910a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor  llvm::sys::PrintStackTraceOnErrorSignal();
44920a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor}
44930a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor
4494995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbarvoid clang_executeOnThread(void (*fn)(void*), void *user_data,
4495995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar                           unsigned stack_size) {
4496995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar  llvm::llvm_execute_on_thread(fn, user_data, stack_size);
4497995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar}
4498995aaf9c8f0131bef0215a9a0bc794b83a49e0b7Daniel Dunbar
4499fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek} // end: extern "C"
4500fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek
4501fb48049fd5f0457d9052269f6d84768412f5f6dfTed Kremenek//===----------------------------------------------------------------------===//
4502fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor// Token-based Operations.
4503fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
4504fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4505fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor/* CXToken layout:
4506fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[0]: a CXTokenKind
4507fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[1]: starting token location
4508fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[2]: token length
4509fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   int_data[3]: reserved
4510f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek *   ptr_data: for identifiers and keywords, an IdentifierInfo*.
4511fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *   otherwise unused.
4512fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor */
4513fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorextern "C" {
4514fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4515fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXTokenKind clang_getTokenKind(CXToken CXTok) {
4516fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return static_cast<CXTokenKind>(CXTok.int_data[0]);
4517fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4518fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4519fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXString clang_getTokenSpelling(CXTranslationUnit TU, CXToken CXTok) {
4520fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  switch (clang_getTokenKind(CXTok)) {
4521fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Identifier:
4522fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Keyword:
4523fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We know we have an IdentifierInfo*, so use that.
4524ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString(static_cast<IdentifierInfo *>(CXTok.ptr_data)
4525ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek                            ->getNameStart());
4526fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4527fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Literal: {
4528fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // We have stashed the starting pointer in the ptr_data field. Use it.
4529fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    const char *Text = static_cast<const char *>(CXTok.ptr_data);
45305f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner    return createCXString(StringRef(Text, CXTok.int_data[2]));
4531fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4532f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4533fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Punctuation:
4534fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  case CXToken_Comment:
4535fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    break;
4536fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  }
4537f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4538f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // We have to find the starting buffer pointer the hard way, by
4539fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // deconstructing the source location.
4540a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4541fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4542ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek    return createCXString("");
4543f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4544fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceLocation Loc = SourceLocation::getFromRawEncoding(CXTok.int_data[1]);
4545fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> LocInfo
4546a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    = CXXUnit->getSourceManager().getDecomposedSpellingLoc(Loc);
4547f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
45485f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  StringRef Buffer
4549f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = CXXUnit->getSourceManager().getBufferData(LocInfo.first, &Invalid);
4550f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  if (Invalid)
4551aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor    return createCXString("");
4552fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4553f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  return createCXString(Buffer.substr(LocInfo.second, CXTok.int_data[2]));
4554fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4555f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4556fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceLocation clang_getTokenLocation(CXTranslationUnit TU, CXToken CXTok) {
4557a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4558fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (!CXXUnit)
4559fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return clang_getNullLocation();
4560f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4561fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  return cxloc::translateSourceLocation(CXXUnit->getASTContext(),
4562fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4563fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4564fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
4565fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas GregorCXSourceRange clang_getTokenExtent(CXTranslationUnit TU, CXToken CXTok) {
4566a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
45675352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor  if (!CXXUnit)
45685352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor    return clang_getNullRange();
4569f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4570f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  return cxloc::translateSourceRange(CXXUnit->getASTContext(),
4571fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor                        SourceLocation::getFromRawEncoding(CXTok.int_data[1]));
4572fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
4573f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4574ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisstatic void getTokens(ASTUnit *CXXUnit, SourceRange Range,
4575ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                      SmallVectorImpl<CXToken> &CXTokens) {
4576fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  SourceManager &SourceMgr = CXXUnit->getSourceManager();
4577fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> BeginLocInfo
4578ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(Range.getBegin());
4579fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  std::pair<FileID, unsigned> EndLocInfo
4580ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(Range.getEnd());
4581f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4582fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Cannot tokenize across files.
4583fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (BeginLocInfo.first != EndLocInfo.first)
4584fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4585f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4586f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek  // Create a lexer
4587f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor  bool Invalid = false;
45885f9e272e632e951b1efe824cd16acb4d96077930Chris Lattner  StringRef Buffer
4589f715ca12bfc9fddfde75f98a197424434428b821Douglas Gregor    = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
459047a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor  if (Invalid)
459147a3fcd4afe122b23f9e7b6148f147bfa460cfe8Douglas Gregor    return;
4592aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor
4593fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
4594fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor            CXXUnit->getASTContext().getLangOptions(),
4595f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer            Buffer.begin(), Buffer.data() + BeginLocInfo.second, Buffer.end());
4596fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Lex.SetCommentRetentionState(true);
4597f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4598fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  // Lex tokens until we hit the end of the range.
4599f6ac97b101c8840efa92bf29166077ce4049e293Benjamin Kramer  const char *EffectiveBufferEnd = Buffer.data() + EndLocInfo.second;
4600fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  Token Tok;
4601096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall  bool previousWasAt = false;
4602fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  do {
4603fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Lex the next token
4604fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    Lex.LexFromRawLexer(Tok);
4605fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.is(tok::eof))
4606fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      break;
4607f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4608fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    // Initialize the CXToken.
4609fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXToken CXTok;
4610f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4611fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Common fields
4612fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[1] = Tok.getLocation().getRawEncoding();
4613fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[2] = Tok.getLength();
4614fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTok.int_data[3] = 0;
4615f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4616fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    //   - Kind-specific fields
4617fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    if (Tok.isLiteral()) {
4618fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Literal;
4619fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = (void *)Tok.getLiteralData();
4620c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara    } else if (Tok.is(tok::raw_identifier)) {
4621aea67dbd653a2dd6dd5cc2159279e81e855b2482Douglas Gregor      // Lookup the identifier to determine whether we have a keyword.
4622fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      IdentifierInfo *II
4623c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        = CXXUnit->getPreprocessor().LookUpIdentifierInfo(Tok);
4624aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek
4625096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall      if ((II->getObjCKeywordID() != tok::objc_not_keyword) && previousWasAt) {
4626aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek        CXTok.int_data[0] = CXToken_Keyword;
4627aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4628aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      else {
4629c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara        CXTok.int_data[0] = Tok.is(tok::identifier)
4630c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          ? CXToken_Identifier
4631c4bf2b9afb7d47445a9dc6bc848657098a4e3851Abramo Bagnara          : CXToken_Keyword;
4632aa8a66de0e7951ba5a305f3e5a39c9b14a318c42Ted Kremenek      }
4633fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = II;
4634fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else if (Tok.is(tok::comment)) {
4635fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Comment;
4636fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4637fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    } else {
4638fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.int_data[0] = CXToken_Punctuation;
4639fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor      CXTok.ptr_data = 0;
4640fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    }
4641fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    CXTokens.push_back(CXTok);
4642096428b351ebf5de9871ce11e06ba6f2d8276ab5David Chisnall    previousWasAt = Tok.is(tok::at);
4643fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  } while (Lex.getBufferLocation() <= EffectiveBufferEnd);
4644ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis}
4645ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4646ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisvoid clang_tokenize(CXTranslationUnit TU, CXSourceRange Range,
4647ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                    CXToken **Tokens, unsigned *NumTokens) {
4648ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (Tokens)
4649ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    *Tokens = 0;
4650ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (NumTokens)
4651ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    *NumTokens = 0;
4652ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4653ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
4654ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (!CXXUnit || !Tokens || !NumTokens)
4655ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
4656ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4657ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
4658ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4659ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SourceRange R = cxloc::translateCXSourceRange(Range);
4660ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (R.isInvalid())
4661ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
4662ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
4663ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SmallVector<CXToken, 32> CXTokens;
4664ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  getTokens(CXXUnit, R, CXTokens);
4665f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4666fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  if (CXTokens.empty())
4667fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor    return;
4668f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
4669fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *Tokens = (CXToken *)malloc(sizeof(CXToken) * CXTokens.size());
4670fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  memmove(*Tokens, CXTokens.data(), sizeof(CXToken) * CXTokens.size());
4671fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor  *NumTokens = CXTokens.size();
4672fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
46730045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
46746db610934bedc6896393c1e1099525b35380acd6Ted Kremenekvoid clang_disposeTokens(CXTranslationUnit TU,
46756db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                         CXToken *Tokens, unsigned NumTokens) {
46766db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  free(Tokens);
46776db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
46786db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46796db610934bedc6896393c1e1099525b35380acd6Ted Kremenek} // end: extern "C"
46806db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46816db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
46826db610934bedc6896393c1e1099525b35380acd6Ted Kremenek// Token annotation APIs.
46836db610934bedc6896393c1e1099525b35380acd6Ted Kremenek//===----------------------------------------------------------------------===//
46846db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
46850045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregortypedef llvm::DenseMap<unsigned, CXCursor> AnnotateTokensData;
4686fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
4687fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXCursor parent,
4688fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                                                     CXClientData client_data);
46896db610934bedc6896393c1e1099525b35380acd6Ted Kremeneknamespace {
46906db610934bedc6896393c1e1099525b35380acd6Ted Kremenekclass AnnotateTokensWorker {
46916db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  AnnotateTokensData &Annotated;
469211949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXToken *Tokens;
469311949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  CXCursor *Cursors;
469411949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  unsigned NumTokens;
4695fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned TokIdx;
46964419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  unsigned PreprocessingTokIdx;
4697fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CursorVisitor AnnotateVis;
4698fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceManager &SrcMgr;
4699f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool HasContextSensitiveKeywords;
4700f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4701fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  bool MoreTokens() const { return TokIdx < NumTokens; }
4702fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  unsigned NextToken() const { return TokIdx; }
4703fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AdvanceToken() { ++TokIdx; }
4704fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation GetTokenLoc(unsigned tokI) {
4705fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
4706fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
47075f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  bool isFunctionMacroToken(unsigned tokI) const {
4708a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return Tokens[tokI].int_data[3] != 0;
4709a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
47105f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  SourceLocation getFunctionMacroTokenLoc(unsigned tokI) const {
4711a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[3]);
4712a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4713a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4714a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  void annotateAndAdvanceTokens(CXCursor, RangeComparisonResult, SourceRange);
47155f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  void annotateAndAdvanceFunctionMacroTokens(CXCursor, RangeComparisonResult,
47165f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis                                             SourceRange);
4717fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
47186db610934bedc6896393c1e1099525b35380acd6Ted Kremenekpublic:
471911949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  AnnotateTokensWorker(AnnotateTokensData &annotated,
4720fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek                       CXToken *tokens, CXCursor *cursors, unsigned numTokens,
4721a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                       CXTranslationUnit tu, SourceRange RegionOfInterest)
472211949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    : Annotated(annotated), Tokens(tokens), Cursors(cursors),
47234419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      NumTokens(numTokens), TokIdx(0), PreprocessingTokIdx(0),
4724a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      AnnotateVis(tu,
472508e0bc16b3312c27e87d33be7dcf3d4fe5bdd2e2Douglas Gregor                  AnnotateTokensVisitor, this, true, RegionOfInterest),
4726f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      SrcMgr(static_cast<ASTUnit*>(tu->TUData)->getSourceManager()),
4727f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      HasContextSensitiveKeywords(false) { }
472811949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4729fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void VisitChildren(CXCursor C) { AnnotateVis.VisitChildren(C); }
47306db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  enum CXChildVisitResult Visit(CXCursor cursor, CXCursor parent);
4731fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  void AnnotateTokens(CXCursor parent);
4732ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek  void AnnotateTokens() {
4733a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    AnnotateTokens(clang_getTranslationUnitCursor(AnnotateVis.getTU()));
4734ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek  }
4735f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4736f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// \brief Determine whether the annotator saw any cursors that have
4737f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  /// context-sensitive keywords.
4738f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  bool hasContextSensitiveKeywords() const {
4739f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    return HasContextSensitiveKeywords;
4740f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
47416db610934bedc6896393c1e1099525b35380acd6Ted Kremenek};
47426db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
47430045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
4744fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenekvoid AnnotateTokensWorker::AnnotateTokens(CXCursor parent) {
4745fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Walk the AST within the region of interest, annotating tokens
4746fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // along the way.
4747fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(parent);
4748fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4749fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = 0 ; I < TokIdx ; ++I) {
475011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
47514419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    if (Pos != Annotated.end() &&
47524419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        (clang_isInvalid(Cursors[I].kind) ||
47534419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor         Pos->second.kind != CXCursor_PreprocessingDirective))
4754fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      Cursors[I] = Pos->second;
4755fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
4756fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4757fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Finish up annotating any tokens left.
4758fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (!MoreTokens())
4759fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return;
476011949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4761fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor &C = clang_getNullCursor();
4762fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = TokIdx ; I < NumTokens ; ++I) {
4763fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    AnnotateTokensData::iterator Pos = Annotated.find(Tokens[I].int_data[1]);
4764fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = (Pos == Annotated.end()) ? C : Pos->second;
476511949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek  }
476611949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek}
476711949cbae3f42c32a2933fb85b125620666d98eaTed Kremenek
4768a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief It annotates and advances tokens with a cursor until the comparison
4769a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis//// between the cursor location and the source range is the same as
4770a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \arg compResult.
4771a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis///
4772a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// Pass RangeBefore to annotate tokens with a cursor until a range is reached.
4773a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// Pass RangeOverlap to annotate tokens inside a range.
4774a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisvoid AnnotateTokensWorker::annotateAndAdvanceTokens(CXCursor updateC,
4775a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               RangeComparisonResult compResult,
4776a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               SourceRange range) {
4777a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  while (MoreTokens()) {
4778a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    const unsigned I = NextToken();
47795f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis    if (isFunctionMacroToken(I))
47805f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis      return annotateAndAdvanceFunctionMacroTokens(updateC, compResult, range);
4781a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4782a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    SourceLocation TokLoc = GetTokenLoc(I);
4783a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
4784a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      Cursors[I] = updateC;
4785a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      AdvanceToken();
4786a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      continue;
4787a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
4788a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    break;
4789a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4790a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
4791a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4792a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief Special annotation handling for macro argument tokens.
47935f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidisvoid AnnotateTokensWorker::annotateAndAdvanceFunctionMacroTokens(
47945f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis                                               CXCursor updateC,
4795a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               RangeComparisonResult compResult,
4796a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                               SourceRange range) {
47975f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  assert(MoreTokens());
47985f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  assert(isFunctionMacroToken(NextToken()) &&
4799a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis         "Should be called only for macro arg tokens");
4800a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4801a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // This works differently than annotateAndAdvanceTokens; because expanded
4802a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // macro arguments can have arbitrary translation-unit source order, we do not
4803a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // advance the token index one by one until a token fails the range test.
4804a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // We only advance once past all of the macro arg tokens if all of them
4805a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // pass the range test. If one of them fails we keep the token index pointing
4806a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // at the start of the macro arg tokens so that the failing token will be
4807a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // annotated by a subsequent annotation try.
4808a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4809a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  bool atLeastOneCompFail = false;
4810a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4811a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned I = NextToken();
48125f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  for (; I < NumTokens && isFunctionMacroToken(I); ++I) {
48135f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis    SourceLocation TokLoc = getFunctionMacroTokenLoc(I);
4814a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (TokLoc.isFileID())
4815a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      continue; // not macro arg token, it's parens or comma.
4816a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (LocationCompare(SrcMgr, TokLoc, range) == compResult) {
4817a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (clang_isInvalid(clang_getCursorKind(Cursors[I])))
4818a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        Cursors[I] = updateC;
4819a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    } else
4820a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      atLeastOneCompFail = true;
4821a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
4822a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
4823a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  if (!atLeastOneCompFail)
4824a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    TokIdx = I; // All of the tokens were handled, advance beyond all of them.
4825a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
4826a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
48276db610934bedc6896393c1e1099525b35380acd6Ted Kremenekenum CXChildVisitResult
48284419b675577d7c281a659fab1fec10e1bfbe04c5Douglas GregorAnnotateTokensWorker::Visit(CXCursor cursor, CXCursor parent) {
4829fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  CXSourceLocation Loc = clang_getCursorLocation(cursor);
48304419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  SourceRange cursorRange = getRawCursorExtent(cursor);
483181d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor  if (cursorRange.isInvalid())
483281d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor    return CXChildVisit_Recurse;
4833f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
4834f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (!HasContextSensitiveKeywords) {
4835f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C properties can have context-sensitive keywords.
4836f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    if (cursor.kind == CXCursor_ObjCPropertyDecl) {
4837f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCPropertyDecl *Property
4838f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(cursor)))
4839f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        HasContextSensitiveKeywords = Property->getPropertyAttributesAsWritten() != 0;
4840f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4841f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // Objective-C methods can have context-sensitive keywords.
4842f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_ObjCInstanceMethodDecl ||
4843f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ObjCClassMethodDecl) {
4844f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (ObjCMethodDecl *Method
4845f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            = dyn_cast_or_null<ObjCMethodDecl>(getCursorDecl(cursor))) {
4846f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->getObjCDeclQualifier())
4847f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4848f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        else {
4849f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          for (ObjCMethodDecl::param_iterator P = Method->param_begin(),
4850f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                                           PEnd = Method->param_end();
4851f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor               P != PEnd; ++P) {
4852f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            if ((*P)->getObjCDeclQualifier()) {
4853f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              HasContextSensitiveKeywords = true;
4854f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              break;
4855f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            }
4856f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          }
4857f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
4858f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4859f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4860f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ methods can have context-sensitive keywords.
4861f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_CXXMethod) {
4862f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (CXXMethodDecl *Method
4863f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor                  = dyn_cast_or_null<CXXMethodDecl>(getCursorDecl(cursor))) {
4864f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (Method->hasAttr<FinalAttr>() || Method->hasAttr<OverrideAttr>())
4865f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4866f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
4867f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4868f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    // C++ classes can have context-sensitive keywords.
4869f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    else if (cursor.kind == CXCursor_StructDecl ||
4870f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassDecl ||
4871f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplate ||
4872f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor             cursor.kind == CXCursor_ClassTemplatePartialSpecialization) {
4873f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Decl *D = getCursorDecl(cursor))
4874f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (D->hasAttr<FinalAttr>())
4875f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          HasContextSensitiveKeywords = true;
4876f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
4877f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
4878f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
48794419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor  if (clang_isPreprocessing(cursor.kind)) {
4880cea731a9cb7de3f473d60e5ea544e25621cebd76Chandler Carruth    // For macro expansions, just note where the beginning of the macro
4881cea731a9cb7de3f473d60e5ea544e25621cebd76Chandler Carruth    // expansion occurs.
48829b2a0ac970a077bdc0bf08c6c682f80ad733c892Chandler Carruth    if (cursor.kind == CXCursor_MacroExpansion) {
48834419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      Annotated[Loc.int_data] = cursor;
48844419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      return CXChildVisit_Recurse;
48854419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
48864419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
48874419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Items in the preprocessing record are kept separate from items in
48884419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // declarations, so we keep a separate token index.
48894419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    unsigned SavedTokIdx = TokIdx;
48904419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = PreprocessingTokIdx;
48914419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
48924419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Skip tokens up until we catch up to the beginning of the preprocessing
48934419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // entry.
48944419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
48954419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
48964419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
48974419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
48984419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
48994419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
49004419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
49014419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
49024419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
49034419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
49044419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
49054419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
49064419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
49074419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
49084419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Look at all of the tokens within this range.
49094419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    while (MoreTokens()) {
49104419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      const unsigned I = NextToken();
49114419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      SourceLocation TokLoc = GetTokenLoc(I);
49124419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      switch (LocationCompare(SrcMgr, TokLoc, cursorRange)) {
49134419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeBefore:
4914b219cfc4d75f0a03630b7c4509ef791b7e97b2c8David Blaikie        llvm_unreachable("Infeasible");
49154419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeAfter:
49164419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        break;
49174419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      case RangeOverlap:
49184419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        Cursors[I] = cursor;
49194419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        AdvanceToken();
49204419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor        continue;
49214419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      }
49224419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor      break;
49234419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    }
49244419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
49254419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // Save the preprocessing token index; restore the non-preprocessing
49264419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    // token index.
49274419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    PreprocessingTokIdx = TokIdx;
49284419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor    TokIdx = SavedTokIdx;
49290045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor    return CXChildVisit_Recurse;
49300045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  }
4931fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4932fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  if (cursorRange.isInvalid())
4933fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    return CXChildVisit_Continue;
4934a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek
4935fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  SourceLocation L = SourceLocation::getFromRawEncoding(Loc.int_data);
4936fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4937a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  // Adjust the annotated range based specific declarations.
4938a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  const enum CXCursorKind cursorK = clang_getCursorKind(cursor);
4939a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  if (cursorK >= CXCursor_FirstDecl && cursorK <= CXCursor_LastDecl) {
494023173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    Decl *D = cxcursor::getCursorDecl(cursor);
49412494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
49422494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    SourceLocation StartLoc;
494323173d7f029f430611caceea72ae61ba6b80af1cTed Kremenek    if (const DeclaratorDecl *DD = dyn_cast<DeclaratorDecl>(D)) {
49442494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = DD->getTypeSourceInfo())
49452494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
49462494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    } else if (TypedefDecl *Typedef = dyn_cast<TypedefDecl>(D)) {
49472494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      if (TypeSourceInfo *TI = Typedef->getTypeSourceInfo())
49482494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        StartLoc = TI->getTypeLoc().getSourceRange().getBegin();
4949a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek    }
49502494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor
49512494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor    if (StartLoc.isValid() && L.isValid() &&
49522494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor        SrcMgr.isBeforeInTranslationUnit(StartLoc, L))
49532494dd024b392b8def58bf067cc94b51c214cf77Douglas Gregor      cursorRange.setBegin(StartLoc);
4954a333c66b94f6d6dcd5ee7e625d816d3c0300e8e6Ted Kremenek  }
495581d3c04b0934c43518355289ad104d34f6fde06fDouglas Gregor
49563f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // If the location of the cursor occurs within a macro instantiation, record
49573f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // the spelling location of the cursor in our annotation map.  We can then
49583f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // paper over the token labelings during a post-processing step to try and
49593f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // get cursor mappings for tokens that are the *arguments* of a macro
49603f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  // instantiation.
49613f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  if (L.isMacroID()) {
49623f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    unsigned rawEncoding = SrcMgr.getSpellingLoc(L).getRawEncoding();
49633f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // Only invalidate the old annotation if it isn't part of a preprocessing
49643f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // directive.  Here we assume that the default construction of CXCursor
49653f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // results in CXCursor.kind being an initialized value (i.e., 0).  If
49663f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    // this isn't the case, we can fix by doing lookup + insertion.
49674419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
49683f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    CXCursor &oldC = Annotated[rawEncoding];
49693f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek    if (!clang_isPreprocessing(oldC.kind))
49703f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek      oldC = cursor;
49713f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek  }
49723f4046004be223b03f1f895bb934e44921ccf805Ted Kremenek
4973fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const enum CXCursorKind K = clang_getCursorKind(parent);
4974fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const CXCursor updateC =
4975d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek    (clang_isInvalid(K) || K == CXCursor_TranslationUnit)
4976d8b0a84d586bc0a08695968acf2f169c9d01da69Ted Kremenek     ? clang_getNullCursor() : parent;
4977fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
4978a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  annotateAndAdvanceTokens(updateC, RangeBefore, cursorRange);
4979fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
49805517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // Avoid having the cursor of an expression "overwrite" the annotation of the
49815517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // variable declaration that it belongs to.
49825517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // This can happen for C++ constructor expressions whose range generally
49835517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  // include the variable declaration, e.g.:
49845517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  //  MyCXXClass foo; // Make sure we don't annotate 'foo' as a CallExpr cursor.
49855517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  if (clang_isExpression(cursorK)) {
49865517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis    Expr *E = getCursorExpr(cursor);
49878ccac3de1335f1cfd7cea56ba1cefcf0b724ce3fArgyrios Kyrtzidis    if (Decl *D = getCursorParentDecl(cursor)) {
49885517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      const unsigned I = NextToken();
49895517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      if (E->getLocStart().isValid() && D->getLocation().isValid() &&
49905517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis          E->getLocStart() == D->getLocation() &&
49915517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis          E->getLocStart() == GetTokenLoc(I)) {
49925517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis        Cursors[I] = updateC;
49935517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis        AdvanceToken();
49945517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis      }
49955517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis    }
49965517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis  }
49975517b89953e3c9276f161ce29831de388bb2573dArgyrios Kyrtzidis
4998fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Visit children to get their cursor information.
4999fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned BeforeChildren = NextToken();
5000fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  VisitChildren(cursor);
5001fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  const unsigned AfterChildren = NextToken();
5002fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5003a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // Scan the tokens that are at the end of the cursor, but are not captured
5004a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  // but the child cursors.
5005a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  annotateAndAdvanceTokens(cursor, RangeOverlap, cursorRange);
50066db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
5007fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Scan the tokens that are at the beginning of the cursor, but are not
5008fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // capture by the child cursors.
5009fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  for (unsigned I = BeforeChildren; I != AfterChildren; ++I) {
5010fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    if (!clang_isInvalid(clang_getCursorKind(Cursors[I])))
5011fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek      break;
50124419b675577d7c281a659fab1fec10e1bfbe04c5Douglas Gregor
5013fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek    Cursors[I] = cursor;
5014fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  }
5015fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
5016fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  return CXChildVisit_Continue;
50170045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor}
50180045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor
50196db610934bedc6896393c1e1099525b35380acd6Ted Kremenekstatic enum CXChildVisitResult AnnotateTokensVisitor(CXCursor cursor,
50206db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXCursor parent,
50216db610934bedc6896393c1e1099525b35380acd6Ted Kremenek                                                     CXClientData client_data) {
50226db610934bedc6896393c1e1099525b35380acd6Ted Kremenek  return static_cast<AnnotateTokensWorker*>(client_data)->Visit(cursor, parent);
50236db610934bedc6896393c1e1099525b35380acd6Ted Kremenek}
50246db610934bedc6896393c1e1099525b35380acd6Ted Kremenek
50256628a614c504263ae539462f049d523dd07ac1baTed Kremeneknamespace {
5026a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5027a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// \brief Uses the macro expansions in the preprocessing record to find
5028a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// and mark tokens that are macro arguments. This info is used by the
5029a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis/// AnnotateTokensWorker.
5030a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisclass MarkMacroArgTokensVisitor {
5031a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  SourceManager &SM;
5032a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  CXToken *Tokens;
5033a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned NumTokens;
5034a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  unsigned CurIdx;
5035a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5036a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidispublic:
5037a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  MarkMacroArgTokensVisitor(SourceManager &SM,
5038a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                            CXToken *tokens, unsigned numTokens)
5039a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    : SM(SM), Tokens(tokens), NumTokens(numTokens), CurIdx(0) { }
5040a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5041a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  CXChildVisitResult visit(CXCursor cursor, CXCursor parent) {
5042a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (cursor.kind != CXCursor_MacroExpansion)
5043a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Continue;
5044a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5045a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    SourceRange macroRange = getCursorMacroExpansion(cursor)->getSourceRange();
5046a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (macroRange.getBegin() == macroRange.getEnd())
5047a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Continue; // it's not a function macro.
5048a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5049a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    for (; CurIdx < NumTokens; ++CurIdx) {
5050a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (!SM.isBeforeInTranslationUnit(getTokenLoc(CurIdx),
5051a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                        macroRange.getBegin()))
5052a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        break;
5053a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
5054a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5055a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (CurIdx == NumTokens)
5056a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Break;
5057a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5058a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    for (; CurIdx < NumTokens; ++CurIdx) {
5059a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      SourceLocation tokLoc = getTokenLoc(CurIdx);
5060a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      if (!SM.isBeforeInTranslationUnit(tokLoc, macroRange.getEnd()))
5061a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis        break;
5062a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
50635f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis      setFunctionMacroTokenLoc(CurIdx, SM.getMacroArgExpandedLocation(tokLoc));
5064a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    }
5065a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5066a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    if (CurIdx == NumTokens)
5067a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis      return CXChildVisit_Break;
5068a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5069a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return CXChildVisit_Continue;
5070a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5071a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5072a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisprivate:
5073a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  SourceLocation getTokenLoc(unsigned tokI) {
5074a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    return SourceLocation::getFromRawEncoding(Tokens[tokI].int_data[1]);
5075a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5076a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
50775f616b726fdf5db3cbc2fd8d92c42cc624dd4550Argyrios Kyrtzidis  void setFunctionMacroTokenLoc(unsigned tokI, SourceLocation loc) {
5078a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // The third field is reserved and currently not used. Use it here
5079a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // to mark macro arg expanded tokens with their expanded locations.
5080a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    Tokens[tokI].int_data[3] = loc.getRawEncoding();
5081a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5082a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis};
5083a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5084a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis} // end anonymous namespace
5085a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5086a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisstatic CXChildVisitResult
5087a676379b26edc959193f9f919ba9c6d296a57824Argyrios KyrtzidisMarkMacroArgTokensVisitorDelegate(CXCursor cursor, CXCursor parent,
5088a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                  CXClientData client_data) {
5089a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  return static_cast<MarkMacroArgTokensVisitor*>(client_data)->visit(cursor,
5090a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                                                     parent);
5091a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis}
5092a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
5093a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidisnamespace {
50946628a614c504263ae539462f049d523dd07ac1baTed Kremenek  struct clang_annotateTokens_Data {
50956628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXTranslationUnit TU;
50966628a614c504263ae539462f049d523dd07ac1baTed Kremenek    ASTUnit *CXXUnit;
50976628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXToken *Tokens;
50986628a614c504263ae539462f049d523dd07ac1baTed Kremenek    unsigned NumTokens;
50996628a614c504263ae539462f049d523dd07ac1baTed Kremenek    CXCursor *Cursors;
51006628a614c504263ae539462f049d523dd07ac1baTed Kremenek  };
5101ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek}
5102ab97961fb4424d0822076eb0fd4f8faee9992763Ted Kremenek
5103ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidisstatic void annotatePreprocessorTokens(CXTranslationUnit TU,
5104ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                       SourceRange RegionOfInterest,
5105ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                       AnnotateTokensData &Annotated) {
5106ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
5107ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5108ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  SourceManager &SourceMgr = CXXUnit->getSourceManager();
5109ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  std::pair<FileID, unsigned> BeginLocInfo
5110ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(RegionOfInterest.getBegin());
5111ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  std::pair<FileID, unsigned> EndLocInfo
5112ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    = SourceMgr.getDecomposedLoc(RegionOfInterest.getEnd());
5113ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5114ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (BeginLocInfo.first != EndLocInfo.first)
5115ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
5116ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5117ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  StringRef Buffer;
5118ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  bool Invalid = false;
5119ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Buffer = SourceMgr.getBufferData(BeginLocInfo.first, &Invalid);
5120ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  if (Buffer.empty() || Invalid)
5121ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    return;
5122ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5123ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Lexer Lex(SourceMgr.getLocForStartOfFile(BeginLocInfo.first),
5124ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            CXXUnit->getASTContext().getLangOptions(),
5125ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            Buffer.begin(), Buffer.data() + BeginLocInfo.second,
5126ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis            Buffer.end());
5127ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  Lex.SetCommentRetentionState(true);
5128ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5129ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  // Lex tokens in raw mode until we hit the end of the range, to avoid
5130ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  // entering #includes or expanding macros.
5131ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  while (true) {
5132ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    Token Tok;
5133ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    Lex.LexFromRawLexer(Tok);
5134ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5135ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  reprocess:
5136ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    if (Tok.is(tok::hash) && Tok.isAtStartOfLine()) {
5137ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // We have found a preprocessing directive. Gobble it up so that we
5138ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // don't see it while preprocessing these tokens later, but keep track
5139ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // of all of the token locations inside this preprocessing directive so
5140ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // that we can annotate them appropriately.
5141ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      //
5142ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // FIXME: Some simple tests here could identify macro definitions and
5143ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      // #undefs, to provide specific cursor kinds for those.
5144ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      SmallVector<SourceLocation, 32> Locations;
5145ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      do {
5146ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Locations.push_back(Tok.getLocation());
5147ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Lex.LexFromRawLexer(Tok);
5148ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      } while (!Tok.isAtStartOfLine() && !Tok.is(tok::eof));
5149ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5150ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      using namespace cxcursor;
5151ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      CXCursor Cursor
5152ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      = MakePreprocessingDirectiveCursor(SourceRange(Locations.front(),
5153ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                                     Locations.back()),
5154ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis                                         TU);
5155ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      for (unsigned I = 0, N = Locations.size(); I != N; ++I) {
5156ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        Annotated[Locations[I].getRawEncoding()] = Cursor;
5157ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      }
5158ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5159ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      if (Tok.isAtStartOfLine())
5160ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis        goto reprocess;
5161ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5162ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      continue;
5163ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    }
5164ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5165ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis    if (Tok.is(tok::eof))
5166ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis      break;
5167ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  }
5168ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis}
5169ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
51706628a614c504263ae539462f049d523dd07ac1baTed Kremenek// This gets run a separate thread to avoid stack blowout.
51716628a614c504263ae539462f049d523dd07ac1baTed Kremenekstatic void clang_annotateTokensImpl(void *UserData) {
51726628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXTranslationUnit TU = ((clang_annotateTokens_Data*)UserData)->TU;
51736628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = ((clang_annotateTokens_Data*)UserData)->CXXUnit;
51746628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXToken *Tokens = ((clang_annotateTokens_Data*)UserData)->Tokens;
51756628a614c504263ae539462f049d523dd07ac1baTed Kremenek  const unsigned NumTokens = ((clang_annotateTokens_Data*)UserData)->NumTokens;
51766628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor *Cursors = ((clang_annotateTokens_Data*)UserData)->Cursors;
5177fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
51780396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Determine the region of interest, which contains all of the tokens.
51790045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  SourceRange RegionOfInterest;
51806628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setBegin(
51816628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU, Tokens[0])));
51826628a614c504263ae539462f049d523dd07ac1baTed Kremenek  RegionOfInterest.setEnd(
51836628a614c504263ae539462f049d523dd07ac1baTed Kremenek    cxloc::translateSourceLocation(clang_getTokenLocation(TU,
51846628a614c504263ae539462f049d523dd07ac1baTed Kremenek                                                         Tokens[NumTokens-1])));
5185fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek
51860396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // A mapping from the source locations found when re-lexing or traversing the
51870396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // region of interest to the corresponding cursors.
51880045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor  AnnotateTokensData Annotated;
5189ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis
5190fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // Relex the tokens within the source range to look for preprocessing
51910396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // directives.
5192ee0f84fc84ed7de7975e102668d8e53a778f7a8cArgyrios Kyrtzidis  annotatePreprocessorTokens(TU, RegionOfInterest, Annotated);
51936628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5194a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  if (CXXUnit->getPreprocessor().getPreprocessingRecord()) {
5195a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    // Search and mark tokens that are macro argument expansions.
5196a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    MarkMacroArgTokensVisitor Visitor(CXXUnit->getSourceManager(),
5197a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                      Tokens, NumTokens);
5198a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    CursorVisitor MacroArgMarker(TU,
5199a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis                                 MarkMacroArgTokensVisitorDelegate, &Visitor,
520008e0bc16b3312c27e87d33be7dcf3d4fe5bdd2e2Douglas Gregor                                 true, RegionOfInterest);
5201a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis    MacroArgMarker.visitPreprocessedEntitiesInRegion();
5202a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis  }
5203a676379b26edc959193f9f919ba9c6d296a57824Argyrios Kyrtzidis
52040396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  // Annotate all of the source locations in the region of interest that map to
5205fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  // a specific cursor.
5206fbd84caf62a21afa4db5f730e2e2603ead17bb65Ted Kremenek  AnnotateTokensWorker W(Annotated, Tokens, Cursors, NumTokens,
5207a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek                         TU, RegionOfInterest);
52086628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52096c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // FIXME: We use a ridiculous stack size here because the data-recursion
52106c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm uses a large stack frame than the non-data recursive version,
52116c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // and AnnotationTokensWorker currently transforms the data-recursion
52126c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // algorithm back into a traditional recursion by explicitly calling
52136c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  // VisitChildren().  We will need to remove this explicit recursive call.
52146628a614c504263ae539462f049d523dd07ac1baTed Kremenek  W.AnnotateTokens();
52156628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5216f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // If we ran into any entities that involve context-sensitive keywords,
5217f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  // take another pass through the tokens to mark them as such.
5218f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  if (W.hasContextSensitiveKeywords()) {
5219f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    for (unsigned I = 0; I != NumTokens; ++I) {
5220f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (clang_getTokenKind(Tokens[I]) != CXToken_Identifier)
5221f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5222f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
5223f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCPropertyDecl) {
5224f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
5225f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (ObjCPropertyDecl *Property
52266628a614c504263ae539462f049d523dd07ac1baTed Kremenek            = dyn_cast_or_null<ObjCPropertyDecl>(getCursorDecl(Cursors[I]))) {
5227f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          if (Property->getPropertyAttributesAsWritten() != 0 &&
5228f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor              llvm::StringSwitch<bool>(II->getName())
52296628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readonly", true)
52306628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("assign", true)
5231f85e193739c953358c865005855253af4f68a497John McCall              .Case("unsafe_unretained", true)
52326628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("readwrite", true)
52336628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("retain", true)
52346628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("copy", true)
52356628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("nonatomic", true)
52366628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("atomic", true)
52376628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("getter", true)
52386628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Case("setter", true)
5239f85e193739c953358c865005855253af4f68a497John McCall              .Case("strong", true)
5240f85e193739c953358c865005855253af4f68a497John McCall              .Case("weak", true)
52416628a614c504263ae539462f049d523dd07ac1baTed Kremenek              .Default(false))
5242f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor            Tokens[I].int_data[0] = CXToken_Keyword;
5243f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        }
5244f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5245f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5246f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor
5247f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      if (Cursors[I].kind == CXCursor_ObjCInstanceMethodDecl ||
5248f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Cursors[I].kind == CXCursor_ObjCClassMethodDecl) {
5249f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        IdentifierInfo *II = static_cast<IdentifierInfo *>(Tokens[I].ptr_data);
5250f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        if (llvm::StringSwitch<bool>(II->getName())
52516628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("in", true)
52526628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("out", true)
52536628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("inout", true)
52546628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("oneway", true)
52556628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("bycopy", true)
52566628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Case("byref", true)
52576628a614c504263ae539462f049d523dd07ac1baTed Kremenek            .Default(false))
5258f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor          Tokens[I].int_data[0] = CXToken_Keyword;
5259f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor        continue;
5260f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
52616639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis
52626639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis      if (Cursors[I].kind == CXCursor_CXXFinalAttr ||
52636639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis          Cursors[I].kind == CXCursor_CXXOverrideAttr) {
52646639e9255489ad8e10278d5658fdd4b3c0e1e4cdArgyrios Kyrtzidis        Tokens[I].int_data[0] = CXToken_Keyword;
52656628a614c504263ae539462f049d523dd07ac1baTed Kremenek        continue;
5266f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor      }
5267f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor    }
5268f52516038ab5d0b1b90a6dd32f46b7d6dabd04c8Douglas Gregor  }
5269fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor}
52706628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52716628a614c504263ae539462f049d523dd07ac1baTed Kremenekextern "C" {
52726628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52736628a614c504263ae539462f049d523dd07ac1baTed Kremenekvoid clang_annotateTokens(CXTranslationUnit TU,
52746628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXToken *Tokens, unsigned NumTokens,
52756628a614c504263ae539462f049d523dd07ac1baTed Kremenek                          CXCursor *Cursors) {
52766628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52776628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (NumTokens == 0 || !Tokens || !Cursors)
52786628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
52796628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52806628a614c504263ae539462f049d523dd07ac1baTed Kremenek  // Any token we don't specifically annotate will have a NULL cursor.
52816628a614c504263ae539462f049d523dd07ac1baTed Kremenek  CXCursor C = clang_getNullCursor();
52826628a614c504263ae539462f049d523dd07ac1baTed Kremenek  for (unsigned I = 0; I != NumTokens; ++I)
52836628a614c504263ae539462f049d523dd07ac1baTed Kremenek    Cursors[I] = C;
52846628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52856628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit *CXXUnit = static_cast<ASTUnit *>(TU->TUData);
52866628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!CXXUnit)
52876628a614c504263ae539462f049d523dd07ac1baTed Kremenek    return;
52886628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52896628a614c504263ae539462f049d523dd07ac1baTed Kremenek  ASTUnit::ConcurrencyCheck Check(*CXXUnit);
52906628a614c504263ae539462f049d523dd07ac1baTed Kremenek
52916628a614c504263ae539462f049d523dd07ac1baTed Kremenek  clang_annotateTokens_Data data = { TU, CXXUnit, Tokens, NumTokens, Cursors };
52926628a614c504263ae539462f049d523dd07ac1baTed Kremenek  llvm::CrashRecoveryContext CRC;
52936628a614c504263ae539462f049d523dd07ac1baTed Kremenek  if (!RunSafely(CRC, clang_annotateTokensImpl, &data,
52946628a614c504263ae539462f049d523dd07ac1baTed Kremenek                 GetSafetyThreadStackSize() * 2)) {
52956628a614c504263ae539462f049d523dd07ac1baTed Kremenek    fprintf(stderr, "libclang: crash detected while annotating tokens\n");
52966628a614c504263ae539462f049d523dd07ac1baTed Kremenek  }
52976628a614c504263ae539462f049d523dd07ac1baTed Kremenek}
52986628a614c504263ae539462f049d523dd07ac1baTed Kremenek
5299fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} // end: extern "C"
5300fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor
5301fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor//===----------------------------------------------------------------------===//
530216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek// Operations for querying linkage of a cursor.
530316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
530416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
530516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenekextern "C" {
530616b4259aecaa22b642d35d36fd89965ed700c1e0Ted KremenekCXLinkageKind clang_getCursorLinkage(CXCursor cursor) {
53070396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor  if (!clang_isDeclaration(cursor.kind))
53080396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor    return CXLinkage_Invalid;
53090396f466978d65789eca85b2738e9df53adc78f5Douglas Gregor
531016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  Decl *D = cxcursor::getCursorDecl(cursor);
531116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  if (NamedDecl *ND = dyn_cast_or_null<NamedDecl>(D))
531216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    switch (ND->getLinkage()) {
531316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case NoLinkage: return CXLinkage_NoLinkage;
531416b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case InternalLinkage: return CXLinkage_Internal;
531516b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case UniqueExternalLinkage: return CXLinkage_UniqueExternal;
531616b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek      case ExternalLinkage: return CXLinkage_External;
531716b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek    };
531816b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
531916b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek  return CXLinkage_Invalid;
532016b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek}
532116b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek} // end: extern "C"
532216b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek
532316b4259aecaa22b642d35d36fd89965ed700c1e0Ted Kremenek//===----------------------------------------------------------------------===//
532445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek// Operations for querying language of a cursor.
532545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
532645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
532745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekstatic CXLanguageKind getDeclLanguage(const Decl *D) {
532845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  switch (D->getKind()) {
532945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    default:
533045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      break;
533145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ImplicitParam:
533245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCAtDefsField:
533345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategory:
533445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCategoryImpl:
533545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCClass:
533645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCCompatibleAlias:
533745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCForwardProtocol:
533845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCImplementation:
533945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCInterface:
534045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCIvar:
534145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCMethod:
534245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProperty:
534345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCPropertyImpl:
534445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ObjCProtocol:
534545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_ObjC;
534645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConstructor:
534745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXConversion:
534845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXDestructor:
534945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXMethod:
535045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::CXXRecord:
535145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplate:
535245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplatePartialSpecialization:
535345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::ClassTemplateSpecialization:
535445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Friend:
535545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FriendTemplate:
535645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::FunctionTemplate:
535745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::LinkageSpec:
535845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Namespace:
535945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NamespaceAlias:
536045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::NonTypeTemplateParm:
536145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::StaticAssert:
536245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTemplateParm:
536345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::TemplateTypeParm:
536445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingTypename:
536545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UnresolvedUsingValue:
536645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::Using:
536745e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingDirective:
536845e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    case Decl::UsingShadow:
536945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek      return CXLanguage_CPlusPlus;
537045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  }
537145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
537245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_C;
537345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
537445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
537545e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenekextern "C" {
537658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
537758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregorenum CXAvailabilityKind clang_getCursorAvailability(CXCursor cursor) {
537858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  if (clang_isDeclaration(cursor.kind))
537958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    if (Decl *D = cxcursor::getCursorDecl(cursor)) {
53800a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      if (isa<FunctionDecl>(D) && cast<FunctionDecl>(D)->isDeleted())
538158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Available;
538258ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
53830a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      switch (D->getAvailability()) {
53840a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Available:
53850a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_NotYetIntroduced:
53860a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_Available;
53870a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
53880a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Deprecated:
538958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor        return CXAvailability_Deprecated;
53900a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
53910a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      case AR_Unavailable:
53920a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor        return CXAvailability_NotAvailable;
53930a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor      }
539458ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor    }
53950a0d2b179085a52c10402feebeb6db8b4d96a140Douglas Gregor
539658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor  return CXAvailability_Available;
539758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor}
539858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor
539945e1dae500bba7a9ef5b8206263a5609c07c6f03Ted KremenekCXLanguageKind clang_getCursorLanguage(CXCursor cursor) {
540045e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  if (clang_isDeclaration(cursor.kind))
540145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek    return getDeclLanguage(cxcursor::getCursorDecl(cursor));
540245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
540345e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek  return CXLanguage_Invalid;
540445e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek}
54053910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54063910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// \brief If the given cursor is the "templated" declaration
54073910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// descibing a class or function template, return the class or
54083910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor /// function template.
54093910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregorstatic Decl *maybeGetTemplateCursor(Decl *D) {
54103910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (!D)
54113910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    return 0;
54123910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54133910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (FunctionDecl *FD = dyn_cast<FunctionDecl>(D))
54143910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (FunctionTemplateDecl *FunTmpl = FD->getDescribedFunctionTemplate())
54153910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return FunTmpl;
54163910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54173910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  if (CXXRecordDecl *RD = dyn_cast<CXXRecordDecl>(D))
54183910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor    if (ClassTemplateDecl *ClassTmpl = RD->getDescribedClassTemplate())
54193910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return ClassTmpl;
54203910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54213910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor  return D;
54223910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor}
54233910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54242be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorSemanticParent(CXCursor cursor) {
54252be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
54262be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
54272be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getDeclContext();
54283910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
54293910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
54303910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54313910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
54323910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
54332be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
54342be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
54352be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54362be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isStatement(cursor.kind) || clang_isExpression(cursor.kind)) {
54372be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor))
5438a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek      return MakeCXCursor(D, getCursorTU(cursor));
54392be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
54402be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54412be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
54422be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
54432be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54442be5bc9ad3981347a000742f81b91ab3080f1214Douglas GregorCXCursor clang_getCursorLexicalParent(CXCursor cursor) {
54452be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  if (clang_isDeclaration(cursor.kind)) {
54462be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    if (Decl *D = getCursorDecl(cursor)) {
54472be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor      DeclContext *DC = D->getLexicalDeclContext();
54483910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      if (!DC)
54493910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor        return clang_getNullCursor();
54503910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor
54513910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor      return MakeCXCursor(maybeGetTemplateCursor(cast<Decl>(DC)),
54523910cfd17fcd99ac80158e625fc63e4784d26435Douglas Gregor                          getCursorTU(cursor));
54532be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor    }
54542be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  }
54552be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54562be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // FIXME: Note that we can't easily compute the lexical context of a
54572be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  // statement or expression, so we return nothing.
54582be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor  return clang_getNullCursor();
54592be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor}
54602be5bc9ad3981347a000742f81b91ab3080f1214Douglas Gregor
54619f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_getOverriddenCursors(CXCursor cursor,
54629f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                CXCursor **overridden,
54639f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor                                unsigned *num_overridden) {
54649f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (overridden)
54659f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *overridden = 0;
54669f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (num_overridden)
54679f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    *num_overridden = 0;
54689f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  if (!overridden || !num_overridden)
54699f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor    return;
54709f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
5471b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  SmallVector<CXCursor, 8> Overridden;
5472b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  cxcursor::getOverriddenCursors(cursor, Overridden);
54739f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
5474b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  *num_overridden = Overridden.size();
5475b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  *overridden = new CXCursor [Overridden.size()];
5476b11be041e4f05519a2eabf6a99429ba6110f1ca9Argyrios Kyrtzidis  std::copy(Overridden.begin(), Overridden.end(), *overridden);
54779f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
54789f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
54799f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregorvoid clang_disposeOverriddenCursors(CXCursor *overridden) {
54809f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor  delete [] overridden;
54819f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor}
54829f59234a91d057cee7c5e3cee91da8696858c692Douglas Gregor
5483ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas GregorCXFile clang_getIncludedFile(CXCursor cursor) {
5484ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  if (cursor.kind != CXCursor_InclusionDirective)
5485ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor    return 0;
5486ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
5487ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  InclusionDirective *ID = getCursorInclusionDirective(cursor);
5488ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor  return (void *)ID->getFile();
5489ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor}
5490ecdcb883cbc6bb4a2445dc6f02d58d9bdb54a0edDouglas Gregor
549145e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek} // end: extern "C"
549245e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek
54939ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
54949ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
54959ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek// C++ AST instrospection.
54969ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek//===----------------------------------------------------------------------===//
54979ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
54989ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekextern "C" {
54999ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenekunsigned clang_CXXMethod_isStatic(CXCursor C) {
55009ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek  if (!clang_isDeclaration(C.kind))
55019ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek    return 0;
550249f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor
550349f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  CXXMethodDecl *Method = 0;
550449f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
550549f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
550649f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
550749f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  else
550849f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
550949f6f5489483beaffc7ce48dfc000af4e65b9216Douglas Gregor  return (Method && Method->isStatic()) ? 1 : 0;
551040b492a43bac3ed0c465772aa6921d011cfc273fTed Kremenek}
5511b12903e1a4b8d1b611b8c7e4f910665d628e68cdTed Kremenek
5512211924b563aa31421836cee7655be729ad02733fDouglas Gregorunsigned clang_CXXMethod_isVirtual(CXCursor C) {
5513211924b563aa31421836cee7655be729ad02733fDouglas Gregor  if (!clang_isDeclaration(C.kind))
5514211924b563aa31421836cee7655be729ad02733fDouglas Gregor    return 0;
5515211924b563aa31421836cee7655be729ad02733fDouglas Gregor
5516211924b563aa31421836cee7655be729ad02733fDouglas Gregor  CXXMethodDecl *Method = 0;
5517211924b563aa31421836cee7655be729ad02733fDouglas Gregor  Decl *D = cxcursor::getCursorDecl(C);
5518211924b563aa31421836cee7655be729ad02733fDouglas Gregor  if (FunctionTemplateDecl *FunTmpl = dyn_cast_or_null<FunctionTemplateDecl>(D))
5519211924b563aa31421836cee7655be729ad02733fDouglas Gregor    Method = dyn_cast<CXXMethodDecl>(FunTmpl->getTemplatedDecl());
5520211924b563aa31421836cee7655be729ad02733fDouglas Gregor  else
5521211924b563aa31421836cee7655be729ad02733fDouglas Gregor    Method = dyn_cast_or_null<CXXMethodDecl>(D);
5522211924b563aa31421836cee7655be729ad02733fDouglas Gregor  return (Method && Method->isVirtual()) ? 1 : 0;
5523211924b563aa31421836cee7655be729ad02733fDouglas Gregor}
55249ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek} // end: extern "C"
55259ada39a4ac82ff5f5087b0a7fa9ed0d32be55a3bTed Kremenek
552645e1dae500bba7a9ef5b8206263a5609c07c6f03Ted Kremenek//===----------------------------------------------------------------------===//
552795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek// Attribute introspection.
552895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
552995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
553095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenekextern "C" {
553195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted KremenekCXType clang_getIBOutletCollectionType(CXCursor C) {
553295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  if (C.kind != CXCursor_IBOutletCollectionAttr)
5533a60ed47da13393796d8552b9fdca12abbb3eea42Ted Kremenek    return cxtype::MakeCXType(QualType(), cxcursor::getCursorTU(C));
553495f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
553595f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek  IBOutletCollectionAttr *A =
553695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek    cast<IBOutletCollectionAttr>(cxcursor::getCursorAttr(C));
553795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
553818aa2ff4641847d7f8866e8c5912d4d0ddb858ceArgyrios Kyrtzidis  return cxtype::MakeCXType(A->getInterface(), cxcursor::getCursorTU(C));
553995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek}
554095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek} // end: extern "C"
554195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek
554295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek//===----------------------------------------------------------------------===//
554359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek// Inspecting memory usage.
554459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek//===----------------------------------------------------------------------===//
554559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5546f787002478f09af1741fb0f82a562002e6799c49Ted Kremenektypedef std::vector<CXTUResourceUsageEntry> MemUsageEntries;
554759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5548f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekstatic inline void createCXTUResourceUsageEntry(MemUsageEntries &entries,
5549f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek                                              enum CXTUResourceUsageKind k,
5550ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek                                              unsigned long amount) {
5551f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  CXTUResourceUsageEntry entry = { k, amount };
555259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  entries.push_back(entry);
555359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
555459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
555559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenekextern "C" {
555659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5557f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekconst char *clang_getTUResourceUsageName(CXTUResourceUsageKind kind) {
555859fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  const char *str = "";
555959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  switch (kind) {
5560f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_AST:
556159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: expressions, declarations, and types";
556259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      break;
5563f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_Identifiers:
556459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: identifiers";
556559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      break;
5566f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_Selectors:
556759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek      str = "ASTContext: selectors";
5568e294ab717fc9535429ca5d8f575d41ae4441d822Ted Kremenek      break;
5569f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    case CXTUResourceUsage_GlobalCompletionResults:
55704e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek      str = "Code completion: cached global results";
5571e294ab717fc9535429ca5d8f575d41ae4441d822Ted Kremenek      break;
5572457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek    case CXTUResourceUsage_SourceManagerContentCache:
5573457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek      str = "SourceManager: content cache allocator";
5574457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek      break;
5575ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    case CXTUResourceUsage_AST_SideTables:
5576ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek      str = "ASTContext: side tables";
5577ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek      break;
5578f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    case CXTUResourceUsage_SourceManager_Membuffer_Malloc:
5579f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      str = "SourceManager: malloc'ed memory buffers";
5580f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      break;
5581f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    case CXTUResourceUsage_SourceManager_Membuffer_MMap:
5582f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      str = "SourceManager: mmap'ed memory buffers";
5583f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek      break;
5584e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    case CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc:
5585e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      str = "ExternalASTSource: malloc'ed memory buffers";
5586e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      break;
5587e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    case CXTUResourceUsage_ExternalASTSource_Membuffer_MMap:
5588e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      str = "ExternalASTSource: mmap'ed memory buffers";
5589e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      break;
55905e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    case CXTUResourceUsage_Preprocessor:
55915e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      str = "Preprocessor: malloc'ed memory";
55925e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      break;
55935e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    case CXTUResourceUsage_PreprocessingRecord:
55945e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      str = "Preprocessor: PreprocessingRecord";
55955e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek      break;
5596ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek    case CXTUResourceUsage_SourceManager_DataStructures:
5597ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek      str = "SourceManager: data structures and tables";
5598ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek      break;
5599d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek    case CXTUResourceUsage_Preprocessor_HeaderSearch:
5600d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek      str = "Preprocessor: header search tables";
5601d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek      break;
560259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  }
560359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  return str;
560459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
560559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5606f787002478f09af1741fb0f82a562002e6799c49Ted KremenekCXTUResourceUsage clang_getCXTUResourceUsage(CXTranslationUnit TU) {
560759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  if (!TU) {
5608f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek    CXTUResourceUsage usage = { (void*) 0, 0, 0 };
560959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    return usage;
561059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  }
561159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
561259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  ASTUnit *astUnit = static_cast<ASTUnit*>(TU->TUData);
561359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  llvm::OwningPtr<MemUsageEntries> entries(new MemUsageEntries());
561459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  ASTContext &astContext = astUnit->getASTContext();
561559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
561659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used by AST nodes and types?
5617f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST,
5618ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    (unsigned long) astContext.getASTAllocatedMemory());
561959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
562059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used by identifiers?
5621f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Identifiers,
562259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    (unsigned long) astContext.Idents.getAllocator().getTotalMemory());
562359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
562459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  // How much memory is used for selectors?
5625f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_Selectors,
562659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    (unsigned long) astContext.Selectors.getTotalMemory());
562759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5628ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek  // How much memory is used by ASTContext's side tables?
5629ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek  createCXTUResourceUsageEntry(*entries, CXTUResourceUsage_AST_SideTables,
5630ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek    (unsigned long) astContext.getSideTableAllocatedMemory());
5631ba29bd25515fbd99e98ba0fedb9d93617b27609eTed Kremenek
56324e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  // How much memory is used for caching global code completion results?
56334e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  unsigned long completionBytes = 0;
56344e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  if (GlobalCodeCompletionAllocator *completionAllocator =
56354e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek      astUnit->getCachedCompletionAllocator().getPtr()) {
56365e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    completionBytes = completionAllocator->getTotalMemory();
56374e6a3f7310d3d9232877ed6f439247b1054b1e47Ted Kremenek  }
5638457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  createCXTUResourceUsageEntry(*entries,
5639457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek                               CXTUResourceUsage_GlobalCompletionResults,
5640457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek                               completionBytes);
5641457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek
5642457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  // How much memory is being used by SourceManager's content cache?
5643457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek  createCXTUResourceUsageEntry(*entries,
5644457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek          CXTUResourceUsage_SourceManagerContentCache,
5645457aaf0692dfb2d9638f383334b81027f637f20cTed Kremenek          (unsigned long) astContext.getSourceManager().getContentCacheSize());
5646f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek
5647f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  // How much memory is being used by the MemoryBuffer's in SourceManager?
5648f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  const SourceManager::MemoryBufferSizes &srcBufs =
5649f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek    astUnit->getSourceManager().getMemoryBufferSizes();
5650f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek
5651f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5652f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               CXTUResourceUsage_SourceManager_Membuffer_Malloc,
5653f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               (unsigned long) srcBufs.malloc_bytes);
5654ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5655f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               CXTUResourceUsage_SourceManager_Membuffer_MMap,
5656f61b831d7f6a15676b07647f507de80324cb7056Ted Kremenek                               (unsigned long) srcBufs.mmap_bytes);
5657ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5658ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                               CXTUResourceUsage_SourceManager_DataStructures,
5659ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                               (unsigned long) astContext.getSourceManager()
5660ca7dc2b755eb81ac95121ce1a1f1aa44a4a0fe12Ted Kremenek                                .getDataStructureSizes());
5661e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek
5662e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  // How much memory is being used by the ExternalASTSource?
5663e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  if (ExternalASTSource *esrc = astContext.getExternalSource()) {
5664e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    const ExternalASTSource::MemoryBufferSizes &sizes =
5665e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      esrc->getMemoryBufferSizes();
5666e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek
5667e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    createCXTUResourceUsageEntry(*entries,
5668e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      CXTUResourceUsage_ExternalASTSource_Membuffer_Malloc,
5669e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek                                 (unsigned long) sizes.malloc_bytes);
5670e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek    createCXTUResourceUsageEntry(*entries,
5671e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek      CXTUResourceUsage_ExternalASTSource_Membuffer_MMap,
5672e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek                                 (unsigned long) sizes.mmap_bytes);
5673e9b5f3d4acfc2ad6e8b65a4072464e997dea9ed3Ted Kremenek  }
56745e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
56755e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  // How much memory is being used by the Preprocessor?
56765e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  Preprocessor &pp = astUnit->getPreprocessor();
56775e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  createCXTUResourceUsageEntry(*entries,
56785e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                               CXTUResourceUsage_Preprocessor,
5679c5c5e92ec53f7e6ac7ebbbf77c6d8e4b7d88daecArgyrios Kyrtzidis                               pp.getTotalMemory());
56805e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
56815e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  if (PreprocessingRecord *pRec = pp.getPreprocessingRecord()) {
56825e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek    createCXTUResourceUsageEntry(*entries,
56835e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                                 CXTUResourceUsage_PreprocessingRecord,
56845e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek                                 pRec->getTotalMemory());
56855e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek  }
56865e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
5687d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek  createCXTUResourceUsageEntry(*entries,
5688d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek                               CXTUResourceUsage_Preprocessor_HeaderSearch,
5689d1194fbbf65374bfa3578eb40a547e4f97b497d1Ted Kremenek                               pp.getHeaderSearchInfo().getTotalMemory());
56905e1db6a434d0e3fe0fbde0bca2ec44552818fb22Ted Kremenek
5691f787002478f09af1741fb0f82a562002e6799c49Ted Kremenek  CXTUResourceUsage usage = { (void*) entries.get(),
569259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek                            (unsigned) entries->size(),
569359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek                            entries->size() ? &(*entries)[0] : 0 };
569459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  entries.take();
569559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  return usage;
569659fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
569759fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5698f787002478f09af1741fb0f82a562002e6799c49Ted Kremenekvoid clang_disposeCXTUResourceUsage(CXTUResourceUsage usage) {
569959fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek  if (usage.data)
570059fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek    delete (MemUsageEntries*) usage.data;
570159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek}
570259fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
570359fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek} // end extern "C"
570459fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
57056df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregorvoid clang::PrintLibclangResourceUsage(CXTranslationUnit TU) {
57066df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  CXTUResourceUsage Usage = clang_getCXTUResourceUsage(TU);
57076df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  for (unsigned I = 0; I != Usage.numEntries; ++I)
57086df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor    fprintf(stderr, "  %s: %lu\n",
57096df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor            clang_getTUResourceUsageName(Usage.entries[I].kind),
57106df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor            Usage.entries[I].amount);
57116df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
57126df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor  clang_disposeCXTUResourceUsage(Usage);
57136df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor}
57146df78739f3b3f672f9735445741cbcfa7c10fb1fDouglas Gregor
571559fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek//===----------------------------------------------------------------------===//
571604bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek// Misc. utility functions.
571704bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek//===----------------------------------------------------------------------===//
5718f0e23e8ad647f41d090d76cc09684ecf4bb05a5bTed Kremenek
5719abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbar/// Default to using an 8 MB stack size on "safety" threads.
5720abdce7abc8a22dd2fe79a05c0b71864039bd8296Daniel Dunbarstatic unsigned SafetyStackThreadSize = 8 << 20;
5721bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5722bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarnamespace clang {
5723bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5724bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarbool RunSafely(llvm::CrashRecoveryContext &CRC,
57256c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               void (*Fn)(void*), void *UserData,
57266c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek               unsigned Size) {
57276c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (!Size)
57286c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek    Size = GetSafetyThreadStackSize();
57296c53fdd88f0d75875365463822fa817d7b1a9573Ted Kremenek  if (Size)
5730bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar    return CRC.RunSafelyOnThread(Fn, UserData, Size);
5731bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return CRC.RunSafely(Fn, UserData);
5732bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5733bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5734bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarunsigned GetSafetyThreadStackSize() {
5735bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  return SafetyStackThreadSize;
5736bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5737bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5738bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbarvoid SetSafetyThreadStackSize(unsigned Value) {
5739bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar  SafetyStackThreadSize = Value;
5740bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5741bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
5742bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar}
5743bf44c3b099602c9c967f1b20995919fb4ef39a51Daniel Dunbar
574404bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenekextern "C" {
574504bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
5746a2a9d6e4e5b6001b86b7dfc5db1ea296ce29a3d3Ted KremenekCXString clang_getClangVersion() {
5747ee4db4fccbcab7e6bd614fe8757c5453e7ad0e86Ted Kremenek  return createCXString(getClangFullVersion());
574804bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek}
574904bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek
575004bb716aea8fd2372ac10b0c640cabc5e5caa615Ted Kremenek} // end: extern "C"
575159fc1e55da9c856d1703d3d3ac14a36320d26b30Ted Kremenek
5752